make windows "kill children process" code simpler

Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
This commit is contained in:
Nicolas De Loof 2021-03-31 14:24:11 +02:00
parent 487bf96ed2
commit a691cc98e7
No known key found for this signature in database
GPG Key ID: 9858809D6F8F6E7E
2 changed files with 17 additions and 69 deletions

View File

@ -18,9 +18,9 @@ package mobycli
import ( import (
"fmt" "fmt"
"os"
"syscall"
"unsafe" "unsafe"
"golang.org/x/sys/windows"
) )
func init() { func init() {
@ -29,81 +29,28 @@ func init() {
} }
} }
var (
kernel32 = syscall.NewLazyDLL("kernel32.dll")
)
type jobObjectExtendedLimitInformation struct {
BasicLimitInformation struct {
PerProcessUserTimeLimit uint64
PerJobUserTimeLimit uint64
LimitFlags uint32
MinimumWorkingSetSize uintptr
MaximumWorkingSetSize uintptr
ActiveProcessLimit uint32
Affinity uintptr
PriorityClass uint32
SchedulingClass uint32
}
IoInfo struct {
ReadOperationCount uint64
WriteOperationCount uint64
OtherOperationCount uint64
ReadTransferCount uint64
WriteTransferCount uint64
OtherTransferCount uint64
}
ProcessMemoryLimit uintptr
JobMemoryLimit uintptr
PeakProcessMemoryUsed uintptr
PeakJobMemoryUsed uintptr
}
// killSubProcessesOnClose will ensure on windows that all child processes of the current process are killed if parent is killed. // killSubProcessesOnClose will ensure on windows that all child processes of the current process are killed if parent is killed.
func killSubProcessesOnClose() error { func killSubProcessesOnClose() error {
job, err := createJobObject() job, err := windows.CreateJobObject(nil, nil)
if err != nil { if err != nil {
return err return err
} }
info := jobObjectExtendedLimitInformation{} info := windows.JOBOBJECT_EXTENDED_LIMIT_INFORMATION{
info.BasicLimitInformation.LimitFlags = 0x2000 BasicLimitInformation: windows.JOBOBJECT_BASIC_LIMIT_INFORMATION{
if err := setInformationJobObject(job, info); err != nil { LimitFlags: windows.JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE,
_ = syscall.CloseHandle(job) },
}
if _, err := windows.SetInformationJobObject(
job,
windows.JobObjectExtendedLimitInformation,
uintptr(unsafe.Pointer(&info)),
uint32(unsafe.Sizeof(info))); err != nil {
_ = windows.CloseHandle(job)
return err return err
} }
proc, err := syscall.GetCurrentProcess() if err := windows.AssignProcessToJobObject(job, windows.CurrentProcess()); err != nil {
if err != nil { _ = windows.CloseHandle(job)
_ = syscall.CloseHandle(job)
return err
}
if err := assignProcessToJobObject(job, proc); err != nil {
_ = syscall.CloseHandle(job)
return err return err
} }
return nil return nil
} }
func createJobObject() (syscall.Handle, error) {
res, _, err := kernel32.NewProc("CreateJobObjectW").Call(uintptr(unsafe.Pointer(nil)), uintptr(unsafe.Pointer(nil)))
if res == 0 {
return syscall.InvalidHandle, os.NewSyscallError("CreateJobObject", err)
}
return syscall.Handle(res), nil
}
func setInformationJobObject(job syscall.Handle, info jobObjectExtendedLimitInformation) error {
infoClass := uint32(9)
res, _, err := kernel32.NewProc("SetInformationJobObject").Call(uintptr(job), uintptr(infoClass), uintptr(unsafe.Pointer(&info)), uintptr(uint32(unsafe.Sizeof(info))))
if res == 0 {
return os.NewSyscallError("SetInformationJobObject", err)
}
return nil
}
func assignProcessToJobObject(job syscall.Handle, process syscall.Handle) error {
res, _, err := kernel32.NewProc("AssignProcessToJobObject").Call(uintptr(job), uintptr(process))
if res == 0 {
return os.NewSyscallError("AssignProcessToJobObject", err)
}
return nil
}

1
go.mod
View File

@ -56,6 +56,7 @@ require (
github.com/valyala/fasttemplate v1.2.1 // indirect github.com/valyala/fasttemplate v1.2.1 // indirect
golang.org/x/oauth2 v0.0.0-20201109201403-9fd604954f58 golang.org/x/oauth2 v0.0.0-20201109201403-9fd604954f58
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c golang.org/x/sync v0.0.0-20210220032951-036812b2e83c
golang.org/x/sys v0.0.0-20210119212857-b64e53b001e4
google.golang.org/grpc v1.33.2 google.golang.org/grpc v1.33.2
google.golang.org/protobuf v1.25.0 google.golang.org/protobuf v1.25.0
gopkg.in/ini.v1 v1.62.0 gopkg.in/ini.v1 v1.62.0