cmd.go

package vproxy

import (
	"fmt"
	"log"
	"os"
	"os/exec"

	"github.com/shirou/gopsutil/process"
)

func runCommand(args []string) *exec.Cmd {
	cmd := exec.Command(args[0], args[1:]...)
	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr
	setProcAttr(cmd)

	fmt.Println("[*] running command:", cmd)
	err := cmd.Start()
	if err != nil {
		log.Fatal("error starting command: ", err)
	}
	return cmd
}

func stopCommand(cmd *exec.Cmd) {
	if cmd == nil {
		return
	}

	fmt.Println("[*] stopping process", cmd.Process.Pid)
	proc, err := process.NewProcess(int32(cmd.Process.Pid))
	if err != nil {
		if err.Error() == "process does not exist" {
			return
		}
		fmt.Println("[*] warning: error finding child process:", err)
	} else {
		err = proc.Terminate()
		if err != nil {
			fmt.Println("[*] error killing child process:", err)
		}
	}
	cmd.Process.Wait()
}