Merge branch 'master' of http://www.scserverddns.top:13000/sc/DyeingComputer

DyeingComputer.csproj

@@ -145,6 +145,7 @@
     <Compile Include="UserClass\SqliteHelper.cs" />
     <Compile Include="UserClass\StrToInt.cs" />
     <Compile Include="UserClass\AsyncTcpServer.cs" />
+    <Compile Include="UserClass\NetFwManger.cs" />
     <Compile Include="ViewModel\CurveDiagramViewModel.cs" />
     <Compile Include="ViewModel\HistoryRecordsViewModel.cs" />
     <Compile Include="ViewModel\MainWindowViewModel.cs" />
@@ -418,6 +419,17 @@
       <Install>false</Install>
     </BootstrapperPackage>
   </ItemGroup>
+  <ItemGroup>
+    <COMReference Include="NetFwTypeLib">
+      <Guid>{58FBCF7C-E7A9-467C-80B3-FC65E8FCCA08}</Guid>
+      <VersionMajor>1</VersionMajor>
+      <VersionMinor>0</VersionMinor>
+      <Lcid>0</Lcid>
+      <WrapperTool>tlbimp</WrapperTool>
+      <Isolated>False</Isolated>
+      <EmbedInteropTypes>True</EmbedInteropTypes>
+    </COMReference>
+  </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />
   <Target Name="AfterResolveReferences">
     <ItemGroup>

UserClass/AsyncTcpServer.cs

@@ -59,7 +59,8 @@ namespace DyeingComputer.UserClass
         }
 
         public static async Task Main()
-        {            
+        {
+            NetFwManger.AllowPort(7790,"TCP");//开放7790端口
             TcpService service = new TcpService();
             service.Connecting = (client, e) => { return EasyTask.CompletedTask; };//有客户端正在连接
             service.Connected = (client, e) => { return EasyTask.CompletedTask; };//有客户端成功连接

UserClass/NetFwManger.cs

@@ -0,0 +1,71 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using NetFwTypeLib;
+
+namespace DyeingComputer.UserClass
+{
+    public class NetFwManger
+    {
+        private static string GetName(int port, string protocol)
+        {
+            return "800-port-" + protocol + "-" + port;
+        }
+
+        public static void AllowPort(int port, string protocol)
+        {
+            
+            DelPort(port, protocol);
+            //创建一个INetFwRule对象
+            Type type = Type.GetTypeFromProgID("HNetCfg.FwRule");
+            INetFwRule rule = (INetFwRule)Activator.CreateInstance(type);
+
+            //设置规则的属性
+            rule.Action = NET_FW_ACTION_.NET_FW_ACTION_ALLOW; //允许连接
+            rule.Direction = NET_FW_RULE_DIRECTION_.NET_FW_RULE_DIR_IN; //入站规则
+            rule.Enabled = true; //启用规则
+            rule.InterfaceTypes = "All"; //适用于所有网络接口
+            rule.Name = GetName(port, protocol); //规则名称
+            if (protocol.ToLower() == "tcp")
+            {
+                rule.Protocol = (int)NET_FW_IP_PROTOCOL_.NET_FW_IP_PROTOCOL_TCP; //TCP协议
+            }
+            else
+            {
+                rule.Protocol = (int)NET_FW_IP_PROTOCOL_.NET_FW_IP_PROTOCOL_UDP; //UDP协议
+            }
+
+            rule.LocalPorts = "" + port; //本地端口号
+
+            //获取FirewallPolicy对象
+            Type policyType = Type.GetTypeFromProgID("HNetCfg.FwPolicy2");
+            INetFwPolicy2 policy = (INetFwPolicy2)Activator.CreateInstance(policyType);
+
+            //将规则添加到防火墙策略中
+            policy.Rules.Add(rule);
+        }
+
+        public static void DelPort(int port, string protocol)
+        {
+            //获取FirewallPolicy对象
+            Type policyType = Type.GetTypeFromProgID("HNetCfg.FwPolicy2");
+            INetFwPolicy2 policy = (INetFwPolicy2)Activator.CreateInstance(policyType);
+
+            //获取现有的规则集合
+            INetFwRules rules = policy.Rules;
+
+            //查找名称的规则并删除它
+            foreach (INetFwRule rule in rules)
+            {
+                if (rule.Name == GetName(port, protocol))
+                {
+                    rules.Remove(rule.Name);
+                    Console.WriteLine(@"Firewall rule deleted successfully.");
+                    break;
+                }
+            }
+        }
+    }
+}