mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-10-30 23:42:33 +01:00
213 lines
7.3 KiB
C#
213 lines
7.3 KiB
C#
using System;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using System.Runtime.Remoting;
|
|
using System.Timers;
|
|
using System.Xml.Linq;
|
|
using System.Xml.XPath;
|
|
using NLog;
|
|
|
|
namespace NzbDrone
|
|
{
|
|
internal class IISController
|
|
{
|
|
private static readonly Logger IISLogger = LogManager.GetLogger("IISExpress");
|
|
private static readonly Logger Logger = LogManager.GetLogger("IISController");
|
|
private static readonly string IISFolder = Path.Combine(Config.ProjectRoot, @"IISExpress\");
|
|
private static readonly string IISExe = Path.Combine(IISFolder, @"iisexpress.exe");
|
|
private static readonly string IISConfigPath = Path.Combine(IISFolder, "AppServer", "applicationhost.config");
|
|
|
|
private static Timer _pingTimer;
|
|
private static int _pingFailCounter;
|
|
|
|
public static Process IISProcess { get; private set; }
|
|
|
|
|
|
internal static string AppUrl
|
|
{
|
|
get { return string.Format("http://localhost:{0}/", Config.Port); }
|
|
}
|
|
|
|
internal static Process StartServer()
|
|
{
|
|
Logger.Info("Preparing IISExpress Server...");
|
|
IISProcess = new Process();
|
|
|
|
IISProcess.StartInfo.FileName = IISExe;
|
|
IISProcess.StartInfo.Arguments = String.Format("/config:\"{0}\" /trace:i", IISConfigPath);//"/config:"""" /trace:i";
|
|
IISProcess.StartInfo.WorkingDirectory = Config.ProjectRoot;
|
|
|
|
IISProcess.StartInfo.UseShellExecute = false;
|
|
IISProcess.StartInfo.RedirectStandardOutput = true;
|
|
IISProcess.StartInfo.RedirectStandardError = true;
|
|
IISProcess.StartInfo.CreateNoWindow = true;
|
|
|
|
|
|
IISProcess.OutputDataReceived += (OnOutputDataReceived);
|
|
IISProcess.ErrorDataReceived += (OnErrorDataReceived);
|
|
|
|
//Set Variables for the config file.
|
|
IISProcess.StartInfo.EnvironmentVariables.Add("NZBDRONE_PATH", Config.ProjectRoot);
|
|
IISProcess.StartInfo.EnvironmentVariables.Add("NZBDRONE_PID", Process.GetCurrentProcess().Id.ToString());
|
|
|
|
try
|
|
{
|
|
UpdateIISConfig();
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.ErrorException("An error has occurred while trying to update the config file.", e);
|
|
}
|
|
|
|
|
|
Logger.Info("Starting process. [{0}]", IISProcess.StartInfo.FileName);
|
|
|
|
|
|
|
|
IISProcess.Start();
|
|
IISProcess.PriorityClass = ProcessPriorityClass.AboveNormal;
|
|
|
|
IISProcess.BeginErrorReadLine();
|
|
IISProcess.BeginOutputReadLine();
|
|
|
|
//Start Ping
|
|
_pingTimer = new Timer(300000) { AutoReset = true };
|
|
_pingTimer.Elapsed += (PingServer);
|
|
_pingTimer.Start();
|
|
|
|
return IISProcess;
|
|
}
|
|
|
|
private static void OnErrorDataReceived(object sender, DataReceivedEventArgs e)
|
|
{
|
|
if (e == null || String.IsNullOrWhiteSpace(e.Data))
|
|
return;
|
|
|
|
IISLogger.Error(e.Data);
|
|
}
|
|
|
|
internal static void StopServer()
|
|
{
|
|
KillProcess(IISProcess);
|
|
|
|
Logger.Info("Finding orphaned IIS Processes.");
|
|
foreach (var process in Process.GetProcessesByName("IISExpress"))
|
|
{
|
|
string processPath = process.MainModule.FileName;
|
|
Logger.Info("[{0}]IIS Process found. Path:{1}", process.Id, processPath);
|
|
if (NormalizePath(processPath) == NormalizePath(IISExe))
|
|
{
|
|
Logger.Info("[{0}]Process is considered orphaned.", process.Id);
|
|
KillProcess(process);
|
|
}
|
|
else
|
|
{
|
|
Logger.Info("[{0}]Process has a different start-up path. skipping.", process.Id);
|
|
}
|
|
}
|
|
}
|
|
|
|
private static void RestartServer()
|
|
{
|
|
_pingTimer.Stop();
|
|
Logger.Warn("Attempting to restart server.");
|
|
StopServer();
|
|
StartServer();
|
|
}
|
|
|
|
private static void PingServer(object sender, ElapsedEventArgs e)
|
|
{
|
|
try
|
|
{
|
|
var response = new WebClient().DownloadString(AppUrl + "/health");
|
|
|
|
if (!response.Contains("OK"))
|
|
{
|
|
throw new ServerException("Health services responded with an invalid response.");
|
|
}
|
|
if (_pingFailCounter > 0)
|
|
{
|
|
Logger.Info("Application pool has been successfully recovered.");
|
|
}
|
|
_pingFailCounter = 0;
|
|
}
|
|
catch (Exception ex)
|
|
{
|
|
_pingFailCounter++;
|
|
Logger.ErrorException("Application pool is not responding. Count " + _pingFailCounter, ex);
|
|
if (_pingFailCounter > 2)
|
|
{
|
|
RestartServer();
|
|
}
|
|
}
|
|
}
|
|
|
|
private static void OnOutputDataReceived(object s, DataReceivedEventArgs e)
|
|
{
|
|
if (e == null || String.IsNullOrWhiteSpace(e.Data) || e.Data.StartsWith("Request started:") ||
|
|
e.Data.StartsWith("Request ended:") || e.Data == ("IncrementMessages called"))
|
|
return;
|
|
|
|
if (e.Data.Contains(" NzbDrone."))
|
|
{
|
|
Console.WriteLine(e.Data);
|
|
return;
|
|
}
|
|
|
|
IISLogger.Trace(e.Data);
|
|
}
|
|
|
|
private static void UpdateIISConfig()
|
|
{
|
|
string configPath = Path.Combine(IISFolder, @"AppServer\applicationhost.config");
|
|
|
|
Logger.Info(@"Server configuration file: {0}", configPath);
|
|
Logger.Info(@"Configuring server to: [http://localhost:{0}]", Config.Port);
|
|
|
|
var configXml = XDocument.Load(configPath);
|
|
|
|
var bindings =
|
|
configXml.XPathSelectElement("configuration/system.applicationHost/sites").Elements("site").Where(
|
|
d => d.Attribute("name").Value.ToLowerInvariant() == "nzbdrone").First().Element("bindings");
|
|
bindings.Descendants().Remove();
|
|
bindings.Add(
|
|
new XElement("binding",
|
|
new XAttribute("protocol", "http"),
|
|
new XAttribute("bindingInformation", String.Format("*:{0}:", Config.Port))
|
|
));
|
|
|
|
configXml.Save(configPath);
|
|
}
|
|
|
|
private static void KillProcess(Process process)
|
|
{
|
|
if (process != null && !process.HasExited)
|
|
{
|
|
Logger.Info("[{0}]Killing process", process.Id);
|
|
process.Kill();
|
|
Logger.Info("[{0}]Waiting for exit", process.Id);
|
|
process.WaitForExit();
|
|
Logger.Info("[{0}]Process terminated successfully", process.Id);
|
|
}
|
|
}
|
|
|
|
public static string NormalizePath(string path)
|
|
{
|
|
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();
|
|
}
|
|
|
|
|
|
}
|
|
} |