735e250326
unix.Kill() does not produce an error for PID 0, -1. As a result, checking process.Alive() would return "true" for both 0 and -1 on macOS (and previously on Linux as well). Let's shortcut these values to consider them "not alive", to prevent someone trying to kill them. A basic test was added to check the behavior. Given that the intent of these functions is to handle single processes, this patch also prevents 0 and negative values to be used. From KILL(2): https://man7.org/linux/man-pages/man2/kill.2.html If pid is positive, then signal sig is sent to the process with the ID specified by pid. If pid equals 0, then sig is sent to every process in the process group of the calling process. If pid equals -1, then sig is sent to every process for which the calling process has permission to send signals, except for process 1 (init), but see below. If pid is less than -1, then sig is sent to every process in the process group whose ID is -pid. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
82 lines
2.5 KiB
Go
82 lines
2.5 KiB
Go
//go:build !windows
|
|
// +build !windows
|
|
|
|
package process
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
"strconv"
|
|
|
|
"golang.org/x/sys/unix"
|
|
)
|
|
|
|
// Alive returns true if process with a given pid is running. It only considers
|
|
// positive PIDs; 0 (all processes in the current process group), -1 (all processes
|
|
// with a PID larger than 1), and negative (-n, all processes in process group
|
|
// "n") values for pid are never considered to be alive.
|
|
func Alive(pid int) bool {
|
|
if pid < 1 {
|
|
return false
|
|
}
|
|
switch runtime.GOOS {
|
|
case "darwin":
|
|
// OS X does not have a proc filesystem. Use kill -0 pid to judge if the
|
|
// process exists. From KILL(2): https://www.freebsd.org/cgi/man.cgi?query=kill&sektion=2&manpath=OpenDarwin+7.2.1
|
|
//
|
|
// Sig may be one of the signals specified in sigaction(2) or it may
|
|
// be 0, in which case error checking is performed but no signal is
|
|
// actually sent. This can be used to check the validity of pid.
|
|
err := unix.Kill(pid, 0)
|
|
|
|
// Either the PID was found (no error) or we get an EPERM, which means
|
|
// the PID exists, but we don't have permissions to signal it.
|
|
return err == nil || err == unix.EPERM
|
|
default:
|
|
_, err := os.Stat(filepath.Join("/proc", strconv.Itoa(pid)))
|
|
return err == nil
|
|
}
|
|
}
|
|
|
|
// Kill force-stops a process. It only considers positive PIDs; 0 (all processes
|
|
// in the current process group), -1 (all processes with a PID larger than 1),
|
|
// and negative (-n, all processes in process group "n") values for pid are
|
|
// ignored. Refer to [KILL(2)] for details.
|
|
//
|
|
// [KILL(2)]: https://man7.org/linux/man-pages/man2/kill.2.html
|
|
func Kill(pid int) error {
|
|
if pid < 1 {
|
|
return fmt.Errorf("invalid PID (%d): only positive PIDs are allowed", pid)
|
|
}
|
|
err := unix.Kill(pid, unix.SIGKILL)
|
|
if err != nil && err != unix.ESRCH {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// Zombie return true if process has a state with "Z". It only considers positive
|
|
// PIDs; 0 (all processes in the current process group), -1 (all processes with
|
|
// a PID larger than 1), and negative (-n, all processes in process group "n")
|
|
// values for pid are ignored. Refer to [PROC(5)] for details.
|
|
//
|
|
// [PROC(5)]: https://man7.org/linux/man-pages/man5/proc.5.html
|
|
func Zombie(pid int) (bool, error) {
|
|
if pid < 1 {
|
|
return false, nil
|
|
}
|
|
data, err := os.ReadFile(fmt.Sprintf("/proc/%d/stat", pid))
|
|
if err != nil {
|
|
if os.IsNotExist(err) {
|
|
return false, nil
|
|
}
|
|
return false, err
|
|
}
|
|
if cols := bytes.SplitN(data, []byte(" "), 4); len(cols) >= 3 && string(cols[2]) == "Z" {
|
|
return true, nil
|
|
}
|
|
return false, nil
|
|
}
|