pulumi/pkg/util/cmdutil/child.go
Sean Gillespie 402a599fc7
Don't use shebangs to launch providers and correctly kill child process trees on Unix (#934)
* Don't use shebangs to launch providers and correctly kill child process trees on Unix

* Link to relevant documentation
2018-02-14 13:56:07 -08:00

29 lines
1.1 KiB
Go

// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
// +build !windows
package cmdutil
import (
"os/exec"
"syscall"
)
// KillChildren calls os.Process.Kill() on every child process of `pid`'s, stoping after the first error (if any). It
// also only kills direct child process, not any children they may have.
func KillChildren(pid int) error {
// A subprocess that was launched after calling `RegisterProcessGroup` below will
// belong to a process group whose ID is the same as the PID. Passing the negation
// of our PID (same as the PGID) sends a SIGKILL to all processes in our group.
//
// Relevant documentation: https://linux.die.net/man/2/kill
// "If pid is less than -1, then sig is sent to every process in the
// process group whose ID is -pid. "
return syscall.Kill(-pid, syscall.SIGKILL)
}
// RegisterProcessGroup informs the OS that it needs to call `setpgid` on this
// child process. When it comes time to kill this process, we'll kill all processes
// in the same process group.
func RegisterProcessGroup(cmd *exec.Cmd) {
cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
}