1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package marketnetwork
import (
"bytes"
"os"
"path/filepath"
"strconv"
"syscall"
"time"
)
func processesUnder(dir string) []int {
entries, err := os.ReadDir("/proc")
if err != nil {
return nil
}
needle := []byte(filepath.Clean(dir) + string(filepath.Separator))
self := os.Getpid()
var out []int
for _, e := range entries {
pid, err := strconv.Atoi(e.Name())
if err != nil || pid == self {
continue
}
cmdline, err := os.ReadFile(filepath.Join("/proc", e.Name(), "cmdline"))
if err != nil || !bytes.Contains(cmdline, needle) {
continue
}
out = append(out, pid)
}
return out
}
func KillUnder(dir string, grace time.Duration) []int {
pids := processesUnder(dir)
exited := exits(pids)
for _, pid := range pids {
if p, err := os.FindProcess(pid); err == nil {
p.Signal(syscall.SIGTERM)
}
}
awaitExits(exited, grace)
for _, pid := range processesUnder(dir) {
if p, err := os.FindProcess(pid); err == nil {
p.Kill()
}
}
return pids
}
