fergalmoran/DnsServer
1
0
Fork 0
mirror of https://github.com/fergalmoran/DnsServer.git synced 2025-12-22 09:29:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

added windows firewall auto entry support.

Browse Source
This commit is contained in:
Shreyas Zare
2020-05-23 18:05:50 +05:30
parent 6763bdd5fa
commit 1ff00e7ace
4 changed files with 265 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
DnsServerApp/DnsServerApp.csproj
View File

@@ -39,6 +39,9 @@
</PropertyGroup>
<ItemGroup>
<Reference Include="System" />
<Reference Include="TechnitiumLibrary.Net.Firewall">
<HintPath>..\..\TechnitiumLibrary\bin\TechnitiumLibrary.Net.Firewall.dll</HintPath>
</Reference>
</ItemGroup>
<ItemGroup>
<Compile Include="Program.cs" />

133
DnsServerApp/Program.cs
View File

@@ -1,6 +1,6 @@
/*
Technitium DNS Server
Copyright (C) 2019 Shreyas Zare (shreyas@technitium.com)
Copyright (C) 2020 Shreyas Zare (shreyas@technitium.com)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
@@ -19,7 +19,11 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
using DnsServerCore;
using System;
using System.Diagnostics;
using System.Reflection;
using System.Security.Principal;
using System.Threading;
using TechnitiumLibrary.Net.Firewall;
namespace DnsServerApp
{
@@ -27,6 +31,37 @@ namespace DnsServerApp
{
static void Main(string[] args)
{
#region check windows firewall entry
string appPath = Assembly.GetEntryAssembly().Location;
if (!WindowsFirewallEntryExists(appPath))
{
bool isAdmin = new WindowsPrincipal(WindowsIdentity.GetCurrent()).IsInRole(WindowsBuiltInRole.Administrator);
if (isAdmin)
{
AddWindowsFirewallEntry(appPath);
}
else
{
ProcessStartInfo processInfo = new ProcessStartInfo(appPath, string.Join(" ", args));
processInfo.UseShellExecute = true;
processInfo.Verb = "runas";
try
{
Process.Start(processInfo);
return;
}
catch
{ }
}
}
#endregion
string configFolder = null;
if (args.Length == 1)
@@ -86,5 +121,101 @@ namespace DnsServerApp
}
}
}
#region private
private static bool WindowsFirewallEntryExists(string appPath)
{
switch (Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
if (Environment.OSVersion.Version.Major > 5)
{
//vista and above
try
{
return WindowsFirewall.RuleExistsVista("", appPath) == RuleStatus.Allowed;
}
catch
{
return false;
}
}
else
{
try
{
return WindowsFirewall.ApplicationExists(appPath) == RuleStatus.Allowed;
}
catch
{
return false;
}
}
default:
return false;
}
}
private static bool AddWindowsFirewallEntry(string appPath)
{
switch (Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
if (Environment.OSVersion.Version.Major > 5)
{
//vista and above
try
{
RuleStatus status = WindowsFirewall.RuleExistsVista("", appPath);
switch (status)
{
case RuleStatus.Blocked:
case RuleStatus.Disabled:
WindowsFirewall.RemoveRuleVista("", appPath);
break;
case RuleStatus.Allowed:
return true;
}
WindowsFirewall.AddRuleVista("Technitium DNS Server", "Allow incoming connection request to the DNS server.", FirewallAction.Allow, appPath, Protocol.ANY, null, null, null, null, InterfaceTypeFlags.All, true, Direction.Inbound, true);
return true;
}
catch
{ }
}
else
{
try
{
RuleStatus status = WindowsFirewall.ApplicationExists(appPath);
switch (status)
{
case RuleStatus.Disabled:
WindowsFirewall.RemoveApplication(appPath);
break;
case RuleStatus.Allowed:
return true;
}
WindowsFirewall.AddApplication("Technitium DNS Server", appPath);
return true;
}
catch
{ }
}
break;
}
return false;
}
#endregion
}
}

3
DnsService/DnsService.csproj
View File

@@ -41,6 +41,9 @@
<Reference Include="System" />
<Reference Include="System.Configuration.Install" />
<Reference Include="System.ServiceProcess" />
<Reference Include="TechnitiumLibrary.Net.Firewall">
<HintPath>..\..\TechnitiumLibrary\bin\TechnitiumLibrary.Net.Firewall.dll</HintPath>
</Reference>
</ItemGroup>
<ItemGroup>
<Compile Include="ProjectInstaller.cs">

128
DnsService/Program.cs
View File

@@ -1,4 +1,26 @@
using System.ServiceProcess;
/*
Technitium DNS Server
Copyright (C) 2020 Shreyas Zare (shreyas@technitium.com)
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
using System;
using System.Reflection;
using System.ServiceProcess;
using TechnitiumLibrary.Net.Firewall;
namespace DnsService
{
@@ -9,6 +31,15 @@ namespace DnsService
/// </summary>
static void Main()
{
#region check windows firewall entry
string appPath = Assembly.GetEntryAssembly().Location;
if (!WindowsFirewallEntryExists(appPath))
AddWindowsFirewallEntry(appPath);
#endregion
ServiceBase[] ServicesToRun;
ServicesToRun = new ServiceBase[]
{
@@ -16,5 +47,100 @@ namespace DnsService
};
ServiceBase.Run(ServicesToRun);
}
#region private
private static bool WindowsFirewallEntryExists(string appPath)
{
switch (Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
if (Environment.OSVersion.Version.Major > 5)
{
//vista and above
try
{
return WindowsFirewall.RuleExistsVista("", appPath) == RuleStatus.Allowed;
}
catch
{
return false;
}
}
else
{
try
{
return WindowsFirewall.ApplicationExists(appPath) == RuleStatus.Allowed;
}
catch
{
return false;
}
}
default:
return false;
}
}
private static bool AddWindowsFirewallEntry(string appPath)
{
switch (Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
if (Environment.OSVersion.Version.Major > 5)
{
//vista and above
try
{
RuleStatus status = WindowsFirewall.RuleExistsVista("", appPath);
switch (status)
{
case RuleStatus.Blocked:
case RuleStatus.Disabled:
WindowsFirewall.RemoveRuleVista("", appPath);
break;
case RuleStatus.Allowed:
return true;
}
WindowsFirewall.AddRuleVista("Technitium DNS Server", "Allow incoming connection request to the DNS server.", FirewallAction.Allow, appPath, Protocol.ANY, null, null, null, null, InterfaceTypeFlags.All, true, Direction.Inbound, true);
return true;
}
catch
{ }
}
else
{
try
{
RuleStatus status = WindowsFirewall.ApplicationExists(appPath);
switch (status)
{
case RuleStatus.Disabled:
WindowsFirewall.RemoveApplication(appPath);
break;
case RuleStatus.Allowed:
return true;
}
WindowsFirewall.AddApplication("Technitium DNS Server", appPath);
return true;
}
catch
{ }
}
break;
}
return false;
}
#endregion
}
}