mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-31 07:52:37 +01:00
146 lines
4.2 KiB
C#
146 lines
4.2 KiB
C#
using System;
|
|
using NLog;
|
|
using NetFwTypeLib;
|
|
|
|
namespace NzbDrone.Common
|
|
{
|
|
public interface IFirewallAdapter
|
|
{
|
|
void MakeAccessible();
|
|
bool IsNzbDronePortOpen();
|
|
}
|
|
|
|
public class FirewallAdapter : IFirewallAdapter
|
|
{
|
|
private readonly IConfigFileProvider _configFileProvider;
|
|
private readonly Logger _logger;
|
|
|
|
public FirewallAdapter(IConfigFileProvider configFileProvider, Logger logger)
|
|
{
|
|
_configFileProvider = configFileProvider;
|
|
_logger = logger;
|
|
}
|
|
|
|
public void MakeAccessible()
|
|
{
|
|
int port = 0;
|
|
|
|
if (IsFirewallEnabled())
|
|
{
|
|
if (IsNzbDronePortOpen())
|
|
{
|
|
_logger.Trace("NzbDrone port is already open, skipping.");
|
|
return;
|
|
}
|
|
|
|
CloseFirewallPort();
|
|
|
|
//Open the new port
|
|
OpenFirewallPort(_configFileProvider.Port);
|
|
}
|
|
}
|
|
|
|
|
|
public bool IsNzbDronePortOpen()
|
|
{
|
|
try
|
|
{
|
|
var netFwMgrType = Type.GetTypeFromProgID("HNetCfg.FwMgr", false);
|
|
|
|
var mgr = (INetFwMgr)Activator.CreateInstance(netFwMgrType);
|
|
|
|
if (!mgr.LocalPolicy.CurrentProfile.FirewallEnabled)
|
|
return false;
|
|
|
|
var ports = mgr.LocalPolicy.CurrentProfile.GloballyOpenPorts;
|
|
|
|
foreach (INetFwOpenPort p in ports)
|
|
{
|
|
if (p.Port == _configFileProvider.Port)
|
|
return true;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.WarnException("Failed to check for open port in firewall", ex);
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
|
|
private void OpenFirewallPort(int portNumber)
|
|
{
|
|
try
|
|
{
|
|
var type = Type.GetTypeFromProgID("HNetCfg.FWOpenPort", false);
|
|
var port = Activator.CreateInstance(type) as INetFwOpenPort;
|
|
|
|
port.Port = portNumber;
|
|
port.Name = "NzbDrone";
|
|
port.Protocol = NET_FW_IP_PROTOCOL_.NET_FW_IP_PROTOCOL_TCP;
|
|
port.Enabled = true;
|
|
|
|
var netFwMgrType = Type.GetTypeFromProgID("HNetCfg.FwMgr", false);
|
|
var mgr = (INetFwMgr)Activator.CreateInstance(netFwMgrType);
|
|
var ports = mgr.LocalPolicy.CurrentProfile.GloballyOpenPorts;
|
|
|
|
ports.Add(port);
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.WarnException("Failed to open port in firewall for NzbDrone " + portNumber, ex);
|
|
}
|
|
}
|
|
|
|
private int CloseFirewallPort()
|
|
{
|
|
try
|
|
{
|
|
var netFwMgrType = Type.GetTypeFromProgID("HNetCfg.FwMgr", false);
|
|
var mgr = (INetFwMgr)Activator.CreateInstance(netFwMgrType);
|
|
var ports = mgr.LocalPolicy.CurrentProfile.GloballyOpenPorts;
|
|
|
|
var portNumber = 8989;
|
|
|
|
foreach (INetFwOpenPort p in ports)
|
|
{
|
|
if (p.Name == "NzbDrone")
|
|
{
|
|
portNumber = p.Port;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (portNumber != _configFileProvider.Port)
|
|
{
|
|
ports.Remove(portNumber, NET_FW_IP_PROTOCOL_.NET_FW_IP_PROTOCOL_TCP);
|
|
return portNumber;
|
|
}
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.WarnException("Failed to close port in firewall for NzbDrone", ex);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
private bool IsFirewallEnabled()
|
|
{
|
|
try
|
|
{
|
|
var netFwMgrType = Type.GetTypeFromProgID("HNetCfg.FwMgr", false);
|
|
var mgr = (INetFwMgr)Activator.CreateInstance(netFwMgrType);
|
|
return mgr.LocalPolicy.CurrentProfile.FirewallEnabled;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_logger.WarnException("Failed to check if the firewall is enabled", ex);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
}
|