-
Notifications
You must be signed in to change notification settings - Fork 373
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix(tools): kill child process on timeout
- Loading branch information
1 parent
30aa4a8
commit 92787fb
Showing
3 changed files
with
131 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,114 @@ | ||
// Copyright 2023 Maintainers of NUKE. | ||
// Distributed under the MIT License. | ||
// https://github.com/nuke-build/nuke/blob/master/LICENSE | ||
|
||
using System; | ||
using System.Collections.Generic; | ||
using System.Diagnostics; | ||
using System.IO; | ||
using System.Runtime.InteropServices; | ||
|
||
namespace Nuke.Common.Tooling; | ||
|
||
// https://raw.githubusercontent.com/dotnet/cli/master/test/Microsoft.DotNet.Tools.Tests.Utilities/Extensions/ProcessExtensions.cs | ||
internal static class NativeProcessExtension | ||
{ | ||
private static readonly bool _isWindows = RuntimeInformation.IsOSPlatform(OSPlatform.Windows); | ||
private static readonly TimeSpan _defaultTimeout = TimeSpan.FromSeconds(30); | ||
|
||
internal static void KillTree(this Process process) | ||
{ | ||
process.KillTree(_defaultTimeout); | ||
} | ||
|
||
internal static void KillTree(this Process process, TimeSpan timeout) | ||
{ | ||
string stdout; | ||
if (_isWindows) | ||
{ | ||
RunProcessAndWaitForExit( | ||
"taskkill", | ||
$"/T /F /PID {process.Id}", | ||
timeout, | ||
out stdout); | ||
} | ||
else | ||
{ | ||
var children = new HashSet<int>(); | ||
GetAllChildIdsUnix(process.Id, children, timeout); | ||
foreach (var childId in children) | ||
{ | ||
KillProcessUnix(childId, timeout); | ||
} | ||
KillProcessUnix(process.Id, timeout); | ||
} | ||
} | ||
|
||
private static void GetAllChildIdsUnix(int parentId, ISet<int> children, TimeSpan timeout) | ||
{ | ||
string stdout; | ||
var exitCode = RunProcessAndWaitForExit( | ||
"pgrep", | ||
$"-P {parentId}", | ||
timeout, | ||
out stdout); | ||
|
||
if (exitCode == 0 && !string.IsNullOrEmpty(stdout)) | ||
{ | ||
using (var reader = new StringReader(stdout)) | ||
{ | ||
while (true) | ||
{ | ||
var text = reader.ReadLine(); | ||
if (text == null) | ||
{ | ||
return; | ||
} | ||
|
||
int id; | ||
if (int.TryParse(text, out id)) | ||
{ | ||
children.Add(id); | ||
// Recursively get the children | ||
GetAllChildIdsUnix(id, children, timeout); | ||
} | ||
} | ||
} | ||
} | ||
} | ||
|
||
private static void KillProcessUnix(int processId, TimeSpan timeout) | ||
{ | ||
string stdout; | ||
RunProcessAndWaitForExit( | ||
"kill", | ||
$"-TERM {processId}", | ||
timeout, | ||
out stdout); | ||
} | ||
|
||
private static int RunProcessAndWaitForExit(string fileName, string arguments, TimeSpan timeout, out string stdout) | ||
{ | ||
var startInfo = new ProcessStartInfo | ||
{ | ||
FileName = fileName, | ||
Arguments = arguments, | ||
RedirectStandardOutput = true, | ||
UseShellExecute = false | ||
}; | ||
|
||
var process = Process.Start(startInfo); | ||
|
||
stdout = null; | ||
if (process.WaitForExit((int)timeout.TotalMilliseconds)) | ||
{ | ||
stdout = process.StandardOutput.ReadToEnd(); | ||
} | ||
else | ||
{ | ||
process.Kill(); | ||
} | ||
|
||
return process.ExitCode; | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters