2010-10-15 09:10:44 +02:00
|
|
|
|
using System;
|
|
|
|
|
using System.Diagnostics;
|
|
|
|
|
using System.IO;
|
|
|
|
|
using NLog;
|
2011-10-07 08:36:04 +02:00
|
|
|
|
using Ninject;
|
2010-10-15 09:10:44 +02:00
|
|
|
|
|
2011-10-07 03:30:44 +02:00
|
|
|
|
namespace NzbDrone.Providers
|
2010-10-15 09:10:44 +02:00
|
|
|
|
{
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public class IISProvider
|
2010-10-15 09:10:44 +02:00
|
|
|
|
{
|
2011-10-07 08:43:35 +02:00
|
|
|
|
private static readonly Logger IISLogger = LogManager.GetLogger("Host.IISExpress");
|
|
|
|
|
private static readonly Logger Logger = LogManager.GetLogger("Host.IISProvider");
|
2011-10-07 08:57:43 +02:00
|
|
|
|
private readonly ConfigProvider _configProvider;
|
|
|
|
|
private readonly ProcessProvider _processProvider;
|
2011-10-08 06:51:35 +02:00
|
|
|
|
private readonly EnviromentProvider _enviromentProvider;
|
2011-10-07 03:30:44 +02:00
|
|
|
|
|
2011-04-24 05:02:20 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
[Inject]
|
2011-10-08 06:51:35 +02:00
|
|
|
|
public IISProvider(ConfigProvider configProvider, ProcessProvider processProvider, EnviromentProvider enviromentProvider)
|
2011-10-07 08:36:04 +02:00
|
|
|
|
{
|
|
|
|
|
_configProvider = configProvider;
|
|
|
|
|
_processProvider = processProvider;
|
2011-10-08 06:51:35 +02:00
|
|
|
|
_enviromentProvider = enviromentProvider;
|
2011-10-07 08:36:04 +02:00
|
|
|
|
}
|
2010-10-15 09:10:44 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public IISProvider()
|
2011-10-07 03:30:44 +02:00
|
|
|
|
{
|
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public string AppUrl
|
2010-10-15 09:10:44 +02:00
|
|
|
|
{
|
2011-10-08 06:51:35 +02:00
|
|
|
|
get { return string.Format("http://localhost:{0}/", _configProvider.PortNumber); }
|
2010-10-15 09:10:44 +02:00
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public int IISProcessId { get; private set; }
|
2011-10-07 05:37:41 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public bool ServerStarted { get; private set; }
|
2011-10-07 05:37:41 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public void StartServer()
|
2010-10-15 09:10:44 +02:00
|
|
|
|
{
|
|
|
|
|
Logger.Info("Preparing IISExpress Server...");
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
var startInfo = new ProcessStartInfo();
|
2010-10-15 09:10:44 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
startInfo.FileName = _configProvider.IISExePath;
|
|
|
|
|
startInfo.Arguments = String.Format("/config:\"{0}\" /trace:i", _configProvider.IISConfigPath);
|
2011-10-08 06:51:35 +02:00
|
|
|
|
startInfo.WorkingDirectory = _enviromentProvider.ApplicationPath;
|
2010-10-17 19:22:48 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
startInfo.UseShellExecute = false;
|
|
|
|
|
startInfo.RedirectStandardOutput = true;
|
|
|
|
|
startInfo.RedirectStandardError = true;
|
|
|
|
|
startInfo.CreateNoWindow = true;
|
2010-10-17 19:22:48 +02:00
|
|
|
|
|
2010-10-15 09:10:44 +02:00
|
|
|
|
//Set Variables for the config file.
|
2011-10-08 06:51:35 +02:00
|
|
|
|
startInfo.EnvironmentVariables.Add("NZBDRONE_PATH", _enviromentProvider.ApplicationPath);
|
2011-10-07 08:36:04 +02:00
|
|
|
|
startInfo.EnvironmentVariables.Add("NZBDRONE_PID", Process.GetCurrentProcess().Id.ToString());
|
2010-10-21 03:49:23 +02:00
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
2011-10-07 08:36:04 +02:00
|
|
|
|
_configProvider.UpdateIISConfig(_configProvider.IISConfigPath);
|
2010-10-21 03:49:23 +02:00
|
|
|
|
}
|
|
|
|
|
catch (Exception e)
|
|
|
|
|
{
|
2011-04-20 09:44:13 +02:00
|
|
|
|
Logger.ErrorException("An error has occurred while trying to update the config file.", e);
|
2010-10-21 03:49:23 +02:00
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
var iisProcess = _processProvider.Start(startInfo);
|
|
|
|
|
IISProcessId = iisProcess.Id;
|
2010-10-15 09:10:44 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
iisProcess.OutputDataReceived += (OnOutputDataReceived);
|
|
|
|
|
iisProcess.ErrorDataReceived += (OnErrorDataReceived);
|
2011-07-17 22:01:37 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
iisProcess.BeginErrorReadLine();
|
|
|
|
|
iisProcess.BeginOutputReadLine();
|
2010-10-15 09:10:44 +02:00
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
ServerStarted = true;
|
2010-10-15 09:10:44 +02:00
|
|
|
|
}
|
|
|
|
|
|
2011-04-25 05:51:18 +02:00
|
|
|
|
private static void OnErrorDataReceived(object sender, DataReceivedEventArgs e)
|
|
|
|
|
{
|
|
|
|
|
if (e == null || String.IsNullOrWhiteSpace(e.Data))
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
IISLogger.Error(e.Data);
|
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public void StopServer()
|
2011-04-22 04:23:31 +02:00
|
|
|
|
{
|
2011-10-07 08:36:04 +02:00
|
|
|
|
_processProvider.Kill(IISProcessId);
|
2011-04-22 08:46:26 +02:00
|
|
|
|
|
|
|
|
|
Logger.Info("Finding orphaned IIS Processes.");
|
2011-10-07 08:36:04 +02:00
|
|
|
|
foreach (var process in _processProvider.GetProcessByName("IISExpress"))
|
2011-04-22 08:46:26 +02:00
|
|
|
|
{
|
2011-10-07 08:36:04 +02:00
|
|
|
|
Logger.Info("[{0}]IIS Process found. Path:{1}", process.Id, process.StartPath);
|
|
|
|
|
if (NormalizePath(process.StartPath) == NormalizePath(_configProvider.IISExePath))
|
2011-04-22 08:46:26 +02:00
|
|
|
|
{
|
|
|
|
|
Logger.Info("[{0}]Process is considered orphaned.", process.Id);
|
2011-10-07 08:36:04 +02:00
|
|
|
|
_processProvider.Kill(process.Id);
|
2011-04-22 08:46:26 +02:00
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
Logger.Info("[{0}]Process has a different start-up path. skipping.", process.Id);
|
|
|
|
|
}
|
|
|
|
|
}
|
2011-04-22 04:23:31 +02:00
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
public void RestartServer()
|
2011-04-22 08:46:26 +02:00
|
|
|
|
{
|
2011-10-07 08:36:04 +02:00
|
|
|
|
ServerStarted = false;
|
2011-04-22 08:46:26 +02:00
|
|
|
|
Logger.Warn("Attempting to restart server.");
|
|
|
|
|
StopServer();
|
|
|
|
|
StartServer();
|
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 03:30:44 +02:00
|
|
|
|
private void OnOutputDataReceived(object s, DataReceivedEventArgs e)
|
2010-10-17 19:22:48 +02:00
|
|
|
|
{
|
2011-04-24 05:02:20 +02:00
|
|
|
|
if (e == null || String.IsNullOrWhiteSpace(e.Data) || e.Data.StartsWith("Request started:") ||
|
2011-04-10 04:44:01 +02:00
|
|
|
|
e.Data.StartsWith("Request ended:") || e.Data == ("IncrementMessages called"))
|
2010-10-17 19:22:48 +02:00
|
|
|
|
return;
|
|
|
|
|
|
2011-06-13 05:45:22 +02:00
|
|
|
|
if (e.Data.Contains(" NzbDrone."))
|
|
|
|
|
{
|
2011-10-09 04:16:11 +02:00
|
|
|
|
System.Console.WriteLine(e.Data);
|
2011-06-13 05:45:22 +02:00
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
2010-10-17 19:22:48 +02:00
|
|
|
|
IISLogger.Trace(e.Data);
|
|
|
|
|
}
|
|
|
|
|
|
2011-10-07 08:36:04 +02:00
|
|
|
|
private string NormalizePath(string path)
|
2010-10-15 09:10:44 +02:00
|
|
|
|
{
|
2011-04-25 05:51:18 +02:00
|
|
|
|
if (String.IsNullOrWhiteSpace(path))
|
|
|
|
|
throw new ArgumentException("Path can not be null or empty");
|
|
|
|
|
|
|
|
|
|
var info = new FileInfo(path);
|
|
|
|
|
|
|
|
|
|
if (info.FullName.StartsWith(@"\\")) //UNC
|
|
|
|
|
{
|
|
|
|
|
return info.FullName.TrimEnd('/', '\\', ' ');
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
return info.FullName.Trim('/', '\\', ' ').ToLower();
|
2010-10-15 09:10:44 +02:00
|
|
|
|
}
|
|
|
|
|
}
|
2011-04-10 04:44:01 +02:00
|
|
|
|
}
|