Go中使用exec.Command执行PowerShell命令的终止问题

两种方法失效的原因 #

第一种方法的问题 #

你的代码在runUntilStopped函数中，每次进入default分支都会调用cmd.Run()——这是一个阻塞方法，会一直等待ping命令执行完毕才会回到select逻辑。当stopChan发送停止信号时，goroutine正被cmd.Run()卡住，根本无法响应信号，更无法终止正在运行的ping进程。

第二种方法的问题 #

Windows系统中，PowerShell的子进程（比如你执行的ping）不会随父PowerShell进程的终止而自动结束。直接调用cmd.Process.Kill()只会杀死PowerShell主进程，其启动的ping进程仍会在后台运行。

解决方案 #

方案1：单次命令的可控执行（针对第一种场景） #

使用exec.CommandContext创建带上下文的命令，结合进程组设置，实现收到停止信号时立即终止命令：

package main

import (
	"context"
	"fmt"
	"os/exec"
	"syscall"
	"time"
)

type customOutput struct{}

func (c customOutput) Write(b []byte) (int, error) {
	fmt.Println(string(b))
	return len(b), nil
}

func runCmd(ctx context.Context, cmdStr string) {
	fmt.Println("Command started.")
	cmd := exec.CommandContext(ctx, "powershell.exe", "-Command", cmdStr)
	cmd.SysProcAttr = &syscall.SysProcAttr{
		HideWindow:            true,
		CreateNewProcessGroup: true, // Windows下必须设置，确保能终止子进程
	}
	cmd.Stdout = customOutput{}

	if err := cmd.Start(); err != nil {
		fmt.Printf("Failed to start command: %v\n", err)
		return
	}

	// 同时监听上下文取消和命令完成信号
	select {
	case <-ctx.Done():
		fmt.Println("Stopping command...")
		if err := cmd.Process.Kill(); err != nil {
			fmt.Printf("Failed to kill process: %v\n", err)
		}
	case err := <-func() <-chan error {
		ch := make(chan error)
		go func() { ch <- cmd.Wait() }()
		return ch
	}():
		if err != nil {
			fmt.Printf("Command exited with error: %v\n", err)
		} else {
			fmt.Println("Command completed successfully.")
		}
	}
}

func main() {
	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()

	go runCmd(ctx, "ping -n 50 google.com")

	time.Sleep(3 * time.Second)

	fmt.Println("Sending stop signal...")
	cancel() // 触发命令终止
	time.Sleep(2 * time.Second)

	fmt.Println("Program finished.")
}

方案2：交互式PowerShell的进程树终止（针对第二种场景） #

通过设置进程组并发送SIGBREAK信号（对应Windows的CTRL+BREAK），实现终止PowerShell及其所有子进程：

package main

import (
	"fmt"
	"io"
	"log"
	"os/exec"
	"syscall"
)

func main() {
	cmd := exec.Command("powershell", "-nologo")
	cmd.SysProcAttr = &syscall.SysProcAttr{
		CreateNewProcessGroup: true, // 创建独立进程组
	}

	stdin, err := cmd.StdinPipe()
	if err != nil {
		panic(err)
	}
	stdout, err := cmd.StdoutPipe()
	if err != nil {
		panic(err)
	}

	if err := cmd.Start(); err != nil {
		panic(err)
	}
	log.Println("[+] Process created with PID:", cmd.Process.Pid)

	// 发送ping命令
	fmt.Fprintln(stdin, "ping -n 50 google.com")

	count := 0
	buf := make([]byte, 1024)
	done := make(chan struct{})

	// 单独goroutine读取输出
	go func() {
		defer close(done)
		for {
			n, err := stdout.Read(buf)
			if err == io.EOF {
				break
			}
			if err != nil {
				fmt.Printf("Read error: %v\n", err)
				break
			}
			fmt.Print(string(buf[:n]))
			count += 1

			if count >= 15 && count <= 20 {
				fmt.Println("\n[!] Stopping process...")
				// 发送SIGBREAK终止整个进程组（PowerShell+ping）
				if err := syscall.Kill(cmd.Process.Pid, syscall.SIGBREAK); err != nil {
					fmt.Printf("Failed to send break signal: %v\n", err)
					// 兜底：直接Kill主进程
					cmd.Process.Kill()
				}
			}
		}
	}()

	<-done
	stdout.Close()
	stdin.Close()
	cmd.Wait()
}

内容的提问来源于stack exchange，提问作者Niu