1
0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Sonarr/NzbDrone.Common/ProcessProvider.cs

264 lines
7.7 KiB
C#
Raw Normal View History

using System;
2013-08-14 07:40:34 +03:00
using System.Collections.Generic;
using System.ComponentModel;
2011-10-07 06:37:41 +03:00
using System.Diagnostics;
using System.IO;
2011-10-07 06:37:41 +03:00
using System.Linq;
using NLog;
2013-08-14 07:25:38 +03:00
using NzbDrone.Common.EnvironmentInfo;
2013-08-31 04:42:30 +03:00
using NzbDrone.Common.Instrumentation;
using NzbDrone.Common.Model;
2011-10-07 06:37:41 +03:00
namespace NzbDrone.Common
2011-10-07 06:37:41 +03:00
{
2013-05-11 02:53:50 +03:00
public interface IProcessProvider
{
ProcessInfo GetCurrentProcess();
ProcessInfo GetProcessById(int id);
2013-08-14 08:20:24 +03:00
void OpenDefaultBrowser(string url);
2013-05-11 02:53:50 +03:00
void WaitForExit(Process process);
void SetPriority(int processId, ProcessPriorityClass priority);
void KillAll(string processName);
bool Exists(string processName);
ProcessPriorityClass GetCurrentProcessPriority();
2013-08-14 08:20:24 +03:00
Process Start(string path, string args = null, Action<string> onOutputDataReceived = null, Action<string> onErrorDataReceived = null);
Process SpawnNewProcess(string path, string args = null);
2013-05-11 02:53:50 +03:00
}
public class ProcessProvider : IProcessProvider
2011-10-07 06:37:41 +03:00
{
private static readonly Logger Logger = NzbDroneLogger.GetLogger();
2011-10-07 06:37:41 +03:00
2013-08-14 08:20:24 +03:00
public const string NZB_DRONE_PROCESS_NAME = "NzbDrone";
public const string NZB_DRONE_CONSOLE_PROCESS_NAME = "NzbDrone.Console";
2011-10-07 06:37:41 +03:00
2013-08-14 07:40:34 +03:00
private static List<Process> GetProcessesByName(string name)
{
var monoProcesses = Process.GetProcessesByName("mono")
2013-08-14 08:02:25 +03:00
.Where(process => process.Modules.Cast<ProcessModule>().Any(module => module.ModuleName.ToLower() == name.ToLower() + ".exe"));
2013-08-14 07:40:34 +03:00
return Process.GetProcessesByName(name)
.Union(monoProcesses).ToList();
}
public ProcessInfo GetCurrentProcess()
2011-10-07 06:37:41 +03:00
{
2011-10-07 09:36:04 +03:00
return ConvertToProcessInfo(Process.GetCurrentProcess());
}
2011-10-07 06:37:41 +03:00
public bool Exists(string processName)
{
2013-08-14 07:40:34 +03:00
return GetProcessesByName(processName).Any();
}
public ProcessPriorityClass GetCurrentProcessPriority()
{
return Process.GetCurrentProcess().PriorityClass;
}
public ProcessInfo GetProcessById(int id)
2011-10-07 09:36:04 +03:00
{
Logger.Trace("Finding process with Id:{0}", id);
2013-05-07 08:54:21 +03:00
var processInfo = ConvertToProcessInfo(Process.GetProcesses().FirstOrDefault(p => p.Id == id));
if (processInfo == null)
{
Logger.Warn("Unable to find process with ID {0}", id);
}
else
{
Logger.Trace("Found process {0}", processInfo.ToString());
}
return processInfo;
2011-10-07 09:36:04 +03:00
}
2011-10-07 06:37:41 +03:00
2013-08-14 08:20:24 +03:00
public void OpenDefaultBrowser(string url)
2011-10-07 06:37:41 +03:00
{
2013-08-14 08:20:24 +03:00
Logger.Info("Opening URL [{0}]", url);
var process = new Process
{
StartInfo = new ProcessStartInfo(url)
{
UseShellExecute = true
}
};
process.Start();
2011-10-07 06:37:41 +03:00
}
2013-08-14 08:20:24 +03:00
public Process Start(string path, string args = null, Action<string> onOutputDataReceived = null, Action<string> onErrorDataReceived = null)
{
2013-08-14 07:25:38 +03:00
if (OsInfo.IsMono && path.EndsWith(".exe", StringComparison.InvariantCultureIgnoreCase))
{
args = path + " " + args;
path = "mono";
}
var logger = LogManager.GetLogger(new FileInfo(path).Name);
var startInfo = new ProcessStartInfo(path, args)
{
CreateNoWindow = true,
UseShellExecute = false,
RedirectStandardError = true,
RedirectStandardOutput = true,
2013-08-14 06:22:28 +03:00
RedirectStandardInput = true
};
logger.Info("Starting {0} {1}", path, args);
var process = new Process
{
StartInfo = startInfo
};
process.OutputDataReceived += (sender, eventArgs) =>
{
if (string.IsNullOrWhiteSpace(eventArgs.Data)) return;
logger.Debug(eventArgs.Data);
if (onOutputDataReceived != null)
{
onOutputDataReceived(eventArgs.Data);
}
};
process.ErrorDataReceived += (sender, eventArgs) =>
{
if (string.IsNullOrWhiteSpace(eventArgs.Data)) return;
logger.Error(eventArgs.Data);
if (onErrorDataReceived != null)
{
onErrorDataReceived(eventArgs.Data);
}
};
2013-08-14 06:22:28 +03:00
process.Start();
process.BeginErrorReadLine();
process.BeginOutputReadLine();
2013-08-14 06:22:28 +03:00
return process;
}
public Process SpawnNewProcess(string path, string args = null)
{
if (OsInfo.IsMono && path.EndsWith(".exe", StringComparison.InvariantCultureIgnoreCase))
{
args = path + " " + args;
path = "mono";
}
Logger.Info("Starting {0} {1}", path, args);
var startInfo = new ProcessStartInfo(path, args);
var process = new Process
{
StartInfo = startInfo
};
process.Start();
return process;
}
public void WaitForExit(Process process)
{
Logger.Trace("Waiting for process {0} to exit.", process.ProcessName);
process.WaitForExit();
}
public void SetPriority(int processId, ProcessPriorityClass priority)
2011-10-07 09:36:04 +03:00
{
var process = Process.GetProcessById(processId);
Logger.Info("Updating [{0}] process priority from {1} to {2}",
process.ProcessName,
process.PriorityClass,
priority);
process.PriorityClass = priority;
}
2013-07-27 03:56:49 +03:00
public void KillAll(string processName)
{
2013-08-14 07:40:34 +03:00
var processToKill = GetProcessesByName(processName);
2013-07-27 03:56:49 +03:00
foreach (var processInfo in processToKill)
{
Kill(processInfo.Id);
}
}
2011-10-07 09:36:04 +03:00
private static ProcessInfo ConvertToProcessInfo(Process process)
{
if (process == null) return null;
process.Refresh();
try
{
if (process.Id <= 0 || process.HasExited) return null;
return new ProcessInfo
{
Id = process.Id,
StartPath = GetExeFileName(process),
Name = process.ProcessName
};
}
catch (Win32Exception)
{
Logger.Warn("Coudn't get process info for " + process.ProcessName);
}
return null;
2011-10-07 06:37:41 +03:00
}
private static string GetExeFileName(Process process)
{
if (process.MainModule.FileName != "mono.exe")
{
return process.MainModule.FileName;
}
return process.Modules.Cast<ProcessModule>().FirstOrDefault(module => module.ModuleName.ToLower().EndsWith(".exe")).FileName;
}
private void Kill(int processId)
{
2013-08-14 06:22:28 +03:00
var process = Process.GetProcesses().FirstOrDefault(p => p.Id == processId);
if (process == null)
{
Logger.Warn("Cannot find process with id: {0}", processId);
return;
}
2013-08-14 06:22:28 +03:00
process.Refresh();
if (process.HasExited)
{
return;
}
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);
}
2011-10-07 06:37:41 +03:00
}
}