防止 Ctrl+C 中断 Golang 中的 exec.Command

Posted 2023-02-23

【中文标题】防止 Ctrl+C 中断 Golang 中的 exec.Command

【英文标题】：Prevent Ctrl+C from interrupting exec.Command in Golang

【发布时间】：2016-01-14 21:56:51

【问题描述】：

我注意到以exec.Command 开始的进程即使通过signal.Notify 拦截了中断调用也会被中断。我已经完成了以下示例来说明问题：

package main

import (
    "log"
    "os"
    "os/exec"
    "os/signal"
    "syscall"
)

func sleep() 
    log.Println("Sleep start")
    cmd := exec.Command("sleep", "60")
    cmd.Run()
    log.Println("Sleep stop")


func main() 
    var doneChannel = make(chan bool)

    go sleep()

    c := make(chan os.Signal, 1)
    signal.Notify(c, os.Interrupt)
    signal.Notify(c, syscall.SIGTERM)
    go func() 
        <-c
        log.Println("Receved Ctrl + C")
    ()

    <-doneChannel

如果在此程序运行时按下 Ctrl+C，它将打印：

2015/10/16 10:05:50 Sleep start
^C2015/10/16 10:05:52 Receved Ctrl + C
2015/10/16 10:05:52 Sleep stop

显示sleep 命令被中断。虽然 Ctrl+C 被成功捕获，但主程序并没有退出，只是 sleep 命令受到影响。

知道如何防止这种情况发生吗？

【参考方案1】：

当您按下ctrl+c 时，shell 将向整个进程组发出信号。如果直接给父进程发信号，子进程不会收到信号。

为了防止 shell 向子进程发出信号，您需要在启动进程之前使用 syscall.SysProcAttr 中的 Setpgid 和 Pgid 字段在其自己的进程组中启动命令

cmd := exec.Command("sleep", "60")
cmd.SysProcAttr = &syscall.SysProcAttr
    Setpgid: true,

【讨论】：

这对windows不起作用，因为windows下SysProcAttr中没有Setpgid。有其他选择吗？

@tkausl 是的。在 Windows 上，您需要将 syscall.SysProcAttr 中的 CreationFlags 设置为 syscall.CREATE_NEW_PROCESS_GROUP。

【参考方案2】：

您可以忽略syscall.SIGINT 信号，然后它不会传递给exec.Command。

func main() 
    var doneChannel = make(chan bool)

    signal.Ignore(syscall.SIGINT)

    go func() 
        log.Println("Sleep start")
        cmd := exec.Command("sleep", "10")
        cmd.Run()
        log.Println("Sleep stop")
        doneChannel <- true
    ()

    <-doneChannel

【讨论】：

import("os/signal" "syscall")

这只是因为孩子（在本例中为sleep）继承了父母的信号处置。信号仍然被传递，如果目标应用程序使用例如恢复默认信号行为，这可能会失败。 signal(SIGINT, SIG_DFL) 或设置它自己的信号处理程序。