tutti/packages/agent/daemon/runtime/connector_process_group_windows.go

//go:build windows

package agentruntime

import (
	"errors"
	"os/exec"
	"strconv"
	"syscall"

	"golang.org/x/sys/windows"
)

func prepareConnectorProcessGroup(command *exec.Cmd) {
	command.SysProcAttr = &syscall.SysProcAttr{CreationFlags: windows.CREATE_NEW_PROCESS_GROUP}
}

func terminateConnectorProcessGroup(command *exec.Cmd) error {
	return terminateConnectorWindowsProcessTree(command, false)
}

func killConnectorProcessGroup(command *exec.Cmd) error {
	return terminateConnectorWindowsProcessTree(command, true)
}

func terminateConnectorWindowsProcessTree(command *exec.Cmd, force bool) error {
	if command == nil || command.Process == nil {
		return nil
	}
	arguments := []string{"/PID", strconv.Itoa(command.Process.Pid), "/T"}
	if force {
		arguments = append(arguments, "/F")
	}
	err := exec.Command("taskkill.exe", arguments...).Run()
	if err == nil {
		return nil
	}
	if killErr := command.Process.Kill(); killErr != nil {
		return errors.Join(err, killErr)
	}
	return nil
}