Make starting & cleaning process group Windows compatible.

client/driver/executor/executor.go

@@ -445,6 +445,17 @@ func ClientCleanup(ic *dstructs.IsolationConfig, pid int) error {
 	return clientCleanup(ic, pid)
 }
 
+// Cleanup any still hanging user processes
+func (e *UniversalExecutor) cleanupUserLeftovers(proc *os.Process) error {
+	// If new process group was created upon command execution
+	// we can kill the whole process group now to cleanup any leftovers.
+	if e.cmd.SysProcAttr != nil && e.cmd.SysProcAttr.Setpgid {
+		return syscall.Kill(-proc.Pid, syscall.SIGKILL)
+	} else {
+		return proc.Kill()
+	}
+}
+
 // Exit cleans up the alloc directory, destroys resource container and kills the
 // user process
 func (e *UniversalExecutor) Exit() error {
@@ -472,7 +483,7 @@ func (e *UniversalExecutor) Exit() error {
 		if err != nil {
 			e.logger.Printf("[ERR] executor: can't find process with pid: %v, err: %v",
 				e.cmd.Process.Pid, err)
-		} else if err := proc.Kill(); err != nil && err.Error() != finishedErr {
+		} else if err := e.cleanupUserLeftovers(proc); err != nil && err.Error() != finishedErr {
 			merr.Errors = append(merr.Errors,
 				fmt.Errorf("can't kill process with pid: %v, err: %v", e.cmd.Process.Pid, err))
 		}

client/driver/executor/executor_linux.go

@@ -330,15 +330,6 @@ func DestroyCgroup(groups *cgroupConfig.Cgroup, cgPaths map[string]string, execu
 	return mErrs.ErrorOrNil()
 }
 
-// configure new process group for child process
-func (e *UniversalExecutor) setNewProcessGroup() error {
-	if e.cmd.SysProcAttr == nil {
-		e.cmd.SysProcAttr = &syscall.SysProcAttr{}
-	}
-	e.cmd.SysProcAttr.Setpgid = true
-	return nil
-}
-
 // getCgroupManager returns the correct libcontainer cgroup manager.
 func getCgroupManager(groups *cgroupConfig.Cgroup, paths map[string]string) cgroups.Manager {
 	return &cgroupFs.Manager{Cgroups: groups, Paths: paths}

client/driver/executor/executor_unix.go

@@ -5,6 +5,7 @@ package executor
 import (
 	"fmt"
 	"io"
+	"syscall"
 
 	syslog "github.com/RackSec/srslog"
 
@@ -47,3 +48,12 @@ func (e *UniversalExecutor) collectLogs(we io.Writer, wo io.Writer) {
 		}
 	}
 }
+
+// configure new process group for child process
+func (e *UniversalExecutor) setNewProcessGroup() error {
+	if e.cmd.SysProcAttr == nil {
+		e.cmd.SysProcAttr = &syscall.SysProcAttr{}
+	}
+	e.cmd.SysProcAttr.Setpgid = true
+	return nil
+}

client/driver/raw_exec.go

@@ -292,8 +292,8 @@ func (h *rawExecHandle) run() {
 	ps, werr := h.executor.Wait()
 	close(h.doneCh)
 	if ps.ExitCode == 0 && werr != nil {
-		if e := killProcessGroup(h.userPid); e != nil {
-			h.logger.Printf("[ERR] driver.raw_exec: error killing user process group: %v", e)
+		if e := killProcess(h.userPid); e != nil {
+			h.logger.Printf("[ERR] driver.raw_exec: error killing user process: %v", e)
 		}
 	}
 

client/driver/utils.go

@@ -8,7 +8,6 @@ import (
 	"os/exec"
 	"path/filepath"
 	"strings"
-	"syscall"
 	"time"
 
 	"github.com/hashicorp/go-multierror"
@@ -110,15 +109,6 @@ func killProcess(pid int) error {
 	return proc.Kill()
 }
 
-// killProcessGroup kills a process group with the given pid
-func killProcessGroup(pid int) error {
-	proc, err := os.FindProcess(pid)
-	if err != nil {
-		return err
-	}
-	return syscall.Kill(-proc.Pid, syscall.SIGKILL)
-}
-
 // destroyPlugin kills the plugin with the given pid and also kills the user
 // process
 func destroyPlugin(pluginPid int, userPid int) error {