mirrors/winsw
1
0
Fork 0
mirror of https://github.com/winsw/winsw.git synced 2025-12-10 18:37:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add dev ps command

Browse Source
This commit is contained in:
NextTurn
2020-08-02 00:00:00 +08:00
committed by Next Turn
parent fcbc087b4b
commit 9573a78668
4 changed files with 170 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
src/WinSW.Core/Util/ProcessExtensions.cs
View File

@@ -15,20 +15,72 @@ namespace WinSW.Util
{
Stop(process, stopTimeout);
foreach (Process child in GetDescendants(process))
foreach (Process child in GetChildren(process))
{
StopTree(child, stopTimeout);
using (child)
{
StopTree(child, stopTimeout);
}
}
}
internal static void StopDescendants(this Process process, TimeSpan stopTimeout)
{
foreach (Process child in GetDescendants(process))
foreach (Process child in GetChildren(process))
{
StopTree(child, stopTimeout);
using (child)
{
StopTree(child, stopTimeout);
}
}
}
internal static unsafe List<Process> GetChildren(this Process process)
{
DateTime startTime = process.StartTime;
int processId = process.Id;
var children = new List<Process>();
foreach (Process other in Process.GetProcesses())
{
try
{
if (other.StartTime <= startTime)
{
goto Next;
}
IntPtr handle = other.Handle;
if (NtQueryInformationProcess(
handle,
PROCESSINFOCLASS.ProcessBasicInformation,
out PROCESS_BASIC_INFORMATION information,
sizeof(PROCESS_BASIC_INFORMATION)) != 0)
{
goto Next;
}
if ((int)information.InheritedFromUniqueProcessId == processId)
{
Logger.Info($"Found child process '{other.Format()}'.");
children.Add(other);
continue;
}
Next:
other.Dispose();
}
catch (Exception e) when (e is InvalidOperationException || e is Win32Exception)
{
other.Dispose();
}
}
return children;
}
private static void Stop(Process process, TimeSpan stopTimeout)
{
Logger.Info("Stopping process " + process.Id);
@@ -61,51 +113,5 @@ namespace WinSW.Util
// TODO: Propagate error if process kill fails? Currently we use the legacy behavior
}
private static unsafe List<Process> GetDescendants(Process root)
{
DateTime startTime = root.StartTime;
int processId = root.Id;
var children = new List<Process>();
foreach (Process process in Process.GetProcesses())
{
try
{
if (process.StartTime <= startTime)
{
goto Next;
}
IntPtr handle = process.Handle;
if (NtQueryInformationProcess(
handle,
PROCESSINFOCLASS.ProcessBasicInformation,
out PROCESS_BASIC_INFORMATION information,
sizeof(PROCESS_BASIC_INFORMATION)) != 0)
{
goto Next;
}
if ((int)information.InheritedFromUniqueProcessId == processId)
{
Logger.Info($"Found child process '{process.Format()}'.");
children.Add(process);
continue;
}
Next:
process.Dispose();
}
catch (Exception e) when (e is InvalidOperationException || e is Win32Exception)
{
process.Dispose();
}
}
return children;
}
}
}