wehub-resource-sync
94 lines
2.7 KiB
Go
94 lines
2.7 KiB
Go
//go:build windows
|
|
|
|
package proc
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"os/exec"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// A launcher (cmd.exe) that spawns a long-lived grandchild (ping) which inherits
|
|
// the stdout pipe: killing only the direct child leaves the grandchild holding
|
|
// the pipe, so cmd.Wait blocks until the grandchild exits. KillTree must take
|
|
// the whole tree down so Wait returns promptly.
|
|
func TestKillTreeUnblocksWaitOnSurvivingGrandchild(t *testing.T) {
|
|
cmd := exec.Command("cmd", "/c", "ping", "-n", "30", "127.0.0.1")
|
|
HideWindow(cmd)
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
t.Fatalf("StdoutPipe: %v", err)
|
|
}
|
|
if err := cmd.Start(); err != nil {
|
|
t.Fatalf("Start: %v", err)
|
|
}
|
|
go func() { _, _ = io.Copy(io.Discard, stdout) }()
|
|
time.Sleep(500 * time.Millisecond) // let cmd.exe exec the ping grandchild
|
|
|
|
KillTree(cmd)
|
|
|
|
done := make(chan error, 1)
|
|
go func() { done <- cmd.Wait() }()
|
|
select {
|
|
case <-done:
|
|
case <-time.After(8 * time.Second):
|
|
t.Fatal("cmd.Wait blocked after KillTree — grandchild survived holding the pipe")
|
|
}
|
|
}
|
|
|
|
// StartTracked must create a Job Object for the started process, and KillTracked
|
|
// must take the whole tracked tree down, including descendants a plain taskkill
|
|
// /T would miss.
|
|
func TestKillTrackedReapsTrackedTree(t *testing.T) {
|
|
cmd := exec.Command("cmd", "/c", "ping", "-n", "30", "127.0.0.1")
|
|
HideWindow(cmd)
|
|
stdout, err := cmd.StdoutPipe()
|
|
if err != nil {
|
|
t.Fatalf("StdoutPipe: %v", err)
|
|
}
|
|
job, err := StartTracked(cmd)
|
|
if err != nil {
|
|
t.Fatalf("StartTracked: %v", err)
|
|
}
|
|
if job == 0 {
|
|
t.Fatal("StartTracked returned 0 — job object not created")
|
|
}
|
|
go func() { _, _ = io.Copy(io.Discard, stdout) }()
|
|
time.Sleep(500 * time.Millisecond) // let cmd.exe exec the ping grandchild into the job
|
|
|
|
KillTracked(cmd, job)
|
|
|
|
done := make(chan error, 1)
|
|
go func() { done <- cmd.Wait() }()
|
|
select {
|
|
case <-done:
|
|
case <-time.After(8 * time.Second):
|
|
t.Fatal("cmd.Wait blocked after KillTracked — tracked tree survived")
|
|
}
|
|
}
|
|
|
|
// StartTracked creates the child suspended to win the job-assignment race, so it
|
|
// must resume it or the process hangs forever. A child that exits with a known
|
|
// code proves the resume fired: Wait returns that code instead of timing out.
|
|
func TestStartTrackedResumesSuspendedChild(t *testing.T) {
|
|
cmd := exec.Command("cmd", "/c", "exit", "7")
|
|
HideWindow(cmd)
|
|
if _, err := StartTracked(cmd); err != nil {
|
|
t.Fatalf("StartTracked: %v", err)
|
|
}
|
|
|
|
done := make(chan error, 1)
|
|
go func() { done <- cmd.Wait() }()
|
|
select {
|
|
case err := <-done:
|
|
var ee *exec.ExitError
|
|
if !errors.As(err, &ee) || ee.ExitCode() != 7 {
|
|
t.Fatalf("exit = %v, want exit status 7", err)
|
|
}
|
|
case <-time.After(8 * time.Second):
|
|
t.Fatal("cmd.Wait blocked — StartTracked left the child suspended")
|
|
}
|
|
}
|