IP输入检查

xljiulang · Oct 17, 2022 · 527b63e · 527b63e
1 parent 016f30b
commit 527b63e
Show file tree

Hide file tree

Showing 3 changed files with 51 additions and 0 deletions.
diff --git a/WindivertDotnet/IPV4Header.cs b/WindivertDotnet/IPV4Header.cs
@@ -166,6 +166,11 @@ private unsafe static IPAddress GetIPAddress(void* ptr)
 
         private unsafe static void SetIPAddress(void* ptr, IPAddress value)
         {
+            if (value.AddressFamily != AddressFamily.InterNetwork)
+            {
+                throw new ArgumentException($"AddressFamily要求必须为{AddressFamily.InterNetwork}", nameof(value));
+            }
+
             var span = new Span<byte>(ptr, IPV4_SIZE);
             value.TryWriteBytes(span, out _);
         }

diff --git a/WindivertDotnet/IPV6Header.cs b/WindivertDotnet/IPV6Header.cs
@@ -2,6 +2,7 @@
 using System.Buffers.Binary;
 using System.Diagnostics;
 using System.Net;
+using System.Net.Sockets;
 
 namespace WindivertDotnet
 {
@@ -148,6 +149,11 @@ private unsafe static IPAddress GetIPAddress(void* ptr)
 
         private unsafe static void SetIPAddress(void* ptr, IPAddress value)
         {
+            if (value.AddressFamily != AddressFamily.InterNetworkV6)
+            {
+                throw new ArgumentException($"AddressFamily要求必须为{AddressFamily.InterNetworkV6}", nameof(value));
+            }
+
             var span = new Span<byte>(ptr, IPV6_SIZE);
             value.TryWriteBytes(span, out _);
         }

diff --git a/WindivertDotnet/WinDivertRouter.cs b/WindivertDotnet/WinDivertRouter.cs
@@ -41,6 +41,7 @@ public unsafe class WinDivertRouter
         /// WinDivert路由
         /// </summary> 
         /// <param name="dstAddr">目标地址</param> 
+        /// <exception cref="ArgumentException"></exception> 
         /// <exception cref="NetworkInformationException"></exception>
         /// <exception cref="NotSupportedException"></exception>
         public WinDivertRouter(IPAddress dstAddr)
@@ -53,6 +54,7 @@ public WinDivertRouter(IPAddress dstAddr)
         /// </summary> 
         /// <param name="dstAddr">目标地址</param>
         /// <param name="srcAddr">源地址</param> 
+        /// <exception cref="ArgumentException"></exception> 
         /// <exception cref="NetworkInformationException"></exception>
         /// <exception cref="NotSupportedException"></exception>
         public WinDivertRouter(IPAddress dstAddr, IPAddress srcAddr)
@@ -67,6 +69,8 @@ public WinDivertRouter(IPAddress dstAddr, IPAddress srcAddr)
         /// <param name="dstAddr">目标地址</param>
         /// <param name="srcAddr">源地址</param> 
         /// <param name="interfaceIndex">网络接口索引</param>
+        /// <exception cref="ArgumentException"></exception>
+        /// <exception cref="ArgumentOutOfRangeException"></exception>
         /// <exception cref="NetworkInformationException"></exception>
         /// <exception cref="NotSupportedException"></exception>
         public WinDivertRouter(IPAddress dstAddr, IPAddress srcAddr, int interfaceIndex)
@@ -80,10 +84,25 @@ public WinDivertRouter(IPAddress dstAddr, IPAddress srcAddr, int interfaceIndex)
         /// <param name="dstAddr">目标地址</param>
         /// <param name="srcAddr">源地址</param>
         /// <param name="interfaceIndex">网络接口索引</param>
+        /// <exception cref="ArgumentException"></exception>
+        /// <exception cref="ArgumentOutOfRangeException"></exception>
         /// <exception cref="NetworkInformationException"></exception>
         /// <exception cref="NotSupportedException"></exception>
         private WinDivertRouter(IPAddress dstAddr, IPAddress? srcAddr, int? interfaceIndex)
         {
+            EnsureNotAny(dstAddr, nameof(dstAddr));
+            EnsureNotAny(srcAddr, nameof(srcAddr));
+
+            if (srcAddr != null && srcAddr.AddressFamily != dstAddr.AddressFamily)
+            {
+                throw new ArgumentException($"{srcAddr}和{dstAddr}的AddressFamily不一致", nameof(srcAddr));
+            }
+
+            if (InterfaceIndex < 0)
+            {
+                throw new ArgumentOutOfRangeException(nameof(interfaceIndex));
+            }
+
             byte* srcSockAddr = null;
             if (srcAddr != null)
             {
@@ -125,14 +144,35 @@ private WinDivertRouter(IPAddress dstAddr, IPAddress? srcAddr, int? interfaceInd
         /// </summary>
         /// <param name="dstAddr">目标地址</param>
         /// <returns></returns>
+        /// <exception cref="ArgumentException"></exception>
         /// <exception cref="NetworkInformationException"></exception>
         public static int GetInterfaceIndex(IPAddress dstAddr)
         {
+            EnsureNotAny(dstAddr, nameof(dstAddr));
+
             var dstSockAddr = stackalloc byte[SocketAddress_SIZE];
             SetIPAddress(dstSockAddr, dstAddr);
             return GetInterfaceIndex(dstSockAddr);
         }
 
+        /// <summary>
+        /// 确保不是任意Ip
+        /// </summary>
+        /// <param name="address"></param>
+        /// <param name="name"></param>
+        /// <exception cref="ArgumentException"></exception>
+        private static void EnsureNotAny(IPAddress? address, string name)
+        {
+            if (address != null)
+            {
+                if (address.Equals(IPAddress.Any) || address.Equals(IPAddress.IPv6Any))
+                {
+                    throw new ArgumentException($"值不能为{address}", name);
+                }
+            }
+        }
+
+
         /// <summary>
         /// 获取网络接口索引
         /// </summary>