github.com/iqoqo/nomad@v0.11.3-0.20200911112621-d7021c74d101/drivers/shared/executor/executor_windows.go (about)

     1  package executor
     2  
     3  import (
     4  	"fmt"
     5  	"os"
     6  	"syscall"
     7  )
     8  
     9  // configure new process group for child process
    10  func (e *UniversalExecutor) setNewProcessGroup() error {
    11  	// We need to check that as build flags includes windows for this file
    12  	if e.childCmd.SysProcAttr == nil {
    13  		e.childCmd.SysProcAttr = &syscall.SysProcAttr{}
    14  	}
    15  	e.childCmd.SysProcAttr.CreationFlags = syscall.CREATE_NEW_PROCESS_GROUP
    16  	return nil
    17  }
    18  
    19  // Cleanup any still hanging user processes
    20  func (e *UniversalExecutor) cleanupChildProcesses(proc *os.Process) error {
    21  	// We must first verify if the process is still running.
    22  	// (Windows process often lingered around after being reported as killed).
    23  	handle, err := syscall.OpenProcess(syscall.PROCESS_TERMINATE|syscall.SYNCHRONIZE|syscall.PROCESS_QUERY_INFORMATION, false, uint32(proc.Pid))
    24  	if err != nil {
    25  		return os.NewSyscallError("OpenProcess", err)
    26  	}
    27  	defer syscall.CloseHandle(handle)
    28  
    29  	result, err := syscall.WaitForSingleObject(syscall.Handle(handle), 0)
    30  
    31  	switch result {
    32  	case syscall.WAIT_OBJECT_0:
    33  		return nil
    34  	case syscall.WAIT_TIMEOUT:
    35  		// Process still running.  Just kill it.
    36  		return proc.Kill()
    37  	default:
    38  		return os.NewSyscallError("WaitForSingleObject", err)
    39  	}
    40  }
    41  
    42  // Send a Ctrl-Break signal for shutting down the process,
    43  // like in https://golang.org/src/os/signal/signal_windows_test.go
    44  func sendCtrlBreak(pid int) error {
    45  	dll, err := syscall.LoadDLL("kernel32.dll")
    46  	if err != nil {
    47  		return fmt.Errorf("Error loading kernel32.dll: %v", err)
    48  	}
    49  	proc, err := dll.FindProc("GenerateConsoleCtrlEvent")
    50  	if err != nil {
    51  		return fmt.Errorf("Cannot find procedure GenerateConsoleCtrlEvent: %v", err)
    52  	}
    53  	result, _, err := proc.Call(syscall.CTRL_BREAK_EVENT, uintptr(pid))
    54  	if result == 0 {
    55  		return fmt.Errorf("Error sending ctrl-break event: %v", err)
    56  	}
    57  	return nil
    58  }
    59  
    60  // Send the process a Ctrl-Break event, allowing it to shutdown by itself
    61  // before being Terminate.
    62  func (e *UniversalExecutor) shutdownProcess(_ os.Signal, proc *os.Process) error {
    63  	if err := sendCtrlBreak(proc.Pid); err != nil {
    64  		return fmt.Errorf("executor shutdown error: %v", err)
    65  	}
    66  	e.logger.Debug("sent Ctrl-Break to process", "pid", proc.Pid)
    67  
    68  	return nil
    69  }