2013-05-23 08:12:01 +03:00
|
|
|
using System;
|
2013-09-21 02:56:17 +03:00
|
|
|
using System.Linq;
|
2013-05-20 05:35:48 +03:00
|
|
|
using NLog;
|
2013-05-23 08:12:01 +03:00
|
|
|
using NzbDrone.Common;
|
2013-06-28 03:04:52 +03:00
|
|
|
using NzbDrone.Common.EnvironmentInfo;
|
2013-09-21 02:56:17 +03:00
|
|
|
using NzbDrone.Common.Processes;
|
2013-05-23 08:12:01 +03:00
|
|
|
using NzbDrone.Core.Configuration;
|
2013-05-20 05:35:48 +03:00
|
|
|
|
2013-08-10 23:30:27 +03:00
|
|
|
namespace NzbDrone.Host.AccessControl
|
2013-05-20 05:35:48 +03:00
|
|
|
{
|
|
|
|
public interface IUrlAclAdapter
|
|
|
|
{
|
2013-09-21 05:50:57 +03:00
|
|
|
void ConfigureUrl();
|
2013-08-31 01:55:01 +03:00
|
|
|
string UrlAcl { get; }
|
2013-05-20 05:35:48 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
public class UrlAclAdapter : IUrlAclAdapter
|
|
|
|
{
|
2013-09-21 02:56:17 +03:00
|
|
|
private const string URL_ACL = "http://{0}:{1}/";
|
|
|
|
|
2013-05-20 05:35:48 +03:00
|
|
|
private readonly IProcessProvider _processProvider;
|
|
|
|
private readonly IConfigFileProvider _configFileProvider;
|
2013-09-21 05:50:57 +03:00
|
|
|
private readonly IRuntimeInfo _runtimeInfo;
|
2013-05-20 05:35:48 +03:00
|
|
|
private readonly Logger _logger;
|
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
public string UrlAcl { get; private set; }
|
|
|
|
private string _localUrl;
|
|
|
|
private string _wildcardUrl;
|
|
|
|
|
|
|
|
public UrlAclAdapter(IProcessProvider processProvider,
|
|
|
|
IConfigFileProvider configFileProvider,
|
|
|
|
IRuntimeInfo runtimeInfo,
|
|
|
|
Logger logger)
|
2013-05-20 05:35:48 +03:00
|
|
|
{
|
|
|
|
_processProvider = processProvider;
|
|
|
|
_configFileProvider = configFileProvider;
|
2013-09-21 05:50:57 +03:00
|
|
|
_runtimeInfo = runtimeInfo;
|
2013-05-20 05:35:48 +03:00
|
|
|
_logger = logger;
|
2013-09-21 02:56:17 +03:00
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
_localUrl = String.Format(URL_ACL, "localhost", _configFileProvider.Port);
|
|
|
|
_wildcardUrl = String.Format(URL_ACL, "*", _configFileProvider.Port);
|
2013-09-21 02:56:17 +03:00
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
UrlAcl = _wildcardUrl;
|
2013-09-21 02:56:17 +03:00
|
|
|
}
|
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
public void ConfigureUrl()
|
2013-08-31 01:55:01 +03:00
|
|
|
{
|
2013-09-21 05:50:57 +03:00
|
|
|
if (!_runtimeInfo.IsAdmin && !IsRegistered)
|
2013-08-31 01:55:01 +03:00
|
|
|
{
|
2013-09-21 05:50:57 +03:00
|
|
|
UrlAcl = _localUrl;
|
2013-09-21 02:56:17 +03:00
|
|
|
}
|
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
if (_runtimeInfo.IsAdmin)
|
2013-09-21 02:56:17 +03:00
|
|
|
{
|
2013-09-21 05:50:57 +03:00
|
|
|
RefreshRegistration();
|
2013-08-31 01:55:01 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
private void RefreshRegistration()
|
2013-05-20 05:35:48 +03:00
|
|
|
{
|
2013-06-28 03:04:52 +03:00
|
|
|
if (OsInfo.Version.Major < 6)
|
2013-05-20 05:35:48 +03:00
|
|
|
return;
|
|
|
|
|
2013-08-31 01:55:01 +03:00
|
|
|
RegisterUrl();
|
2013-05-20 05:35:48 +03:00
|
|
|
}
|
|
|
|
|
2013-09-21 05:50:57 +03:00
|
|
|
private bool IsRegistered
|
|
|
|
{
|
|
|
|
get
|
|
|
|
{
|
|
|
|
var arguments = String.Format("http show urlacl {0}", _wildcardUrl);
|
|
|
|
var output = RunNetsh(arguments);
|
|
|
|
|
|
|
|
if (output == null || !output.Standard.Any()) return false;
|
|
|
|
|
|
|
|
return output.Standard.Any(line => line.Contains(_wildcardUrl));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-08-31 01:55:01 +03:00
|
|
|
private void RegisterUrl()
|
2013-05-20 05:35:48 +03:00
|
|
|
{
|
2013-08-31 01:55:01 +03:00
|
|
|
var arguments = String.Format("http add urlacl {0} sddl=D:(A;;GX;;;S-1-1-0)", UrlAcl);
|
2013-05-20 05:35:48 +03:00
|
|
|
RunNetsh(arguments);
|
|
|
|
}
|
|
|
|
|
2013-09-21 02:56:17 +03:00
|
|
|
private ProcessOutput RunNetsh(string arguments)
|
2013-05-20 05:35:48 +03:00
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
2013-09-21 02:56:17 +03:00
|
|
|
var output = _processProvider.StartAndCapture("netsh.exe", arguments);
|
|
|
|
|
|
|
|
return output;
|
2013-05-20 05:35:48 +03:00
|
|
|
}
|
|
|
|
catch (Exception ex)
|
|
|
|
{
|
|
|
|
_logger.WarnException("Error executing netsh with arguments: " + arguments, ex);
|
|
|
|
}
|
2013-09-21 02:56:17 +03:00
|
|
|
|
|
|
|
return null;
|
2013-05-20 05:35:48 +03:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|