Golang - 按名称杀死进程

时间:2016-12-09 12:24:55

标签: go process command kill

如果您只知道进程名称,那么使用Go代码终止进程的有效方法是什么?我看到os包提供的一些功能如:

func FindProcess(pid int) (*Process, error)
func (p *Process) Kill() error
func (p *Process) Signal(sig Signal) error

有没有一个好的/通常的做法来获取pid而不必执行命令然后解析输出?

我找到了一种使用如下命令取回pid的方法:

  • echo $(ps cax | grep myapp | grep -o '^[ ]*[0-9]*')

我有used it with exec.Command()但如果有更好的方法,我想避免使用它。

4 个答案:

答案 0 :(得分:6)

运行外部命令可能是执行此操作的最佳方法。但是,只要您是要杀死的进程的所有者,以下代码至少在Ubuntu上运行。

// killprocess project main.go
package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "os"
    "path/filepath"
    "strconv"
    "strings"
)

// args holds the commandline args
var args []string

// findAndKillProcess walks iterative through the /process directory tree
// looking up the process name found in each /proc/<pid>/status file. If
// the name matches the name in the argument the process with the corresponding
// <pid> will be killed.
func findAndKillProcess(path string, info os.FileInfo, err error) error {
    // We just return in case of errors, as they are likely due to insufficient
    // privileges. We shouldn't get any errors for accessing the information we
    // are interested in. Run as root (sudo) and log the error, in case you want
    // this information.
    if err != nil {
        // log.Println(err)
        return nil
    }

    // We are only interested in files with a path looking like /proc/<pid>/status.
    if strings.Count(path, "/") == 3 {
        if strings.Contains(path, "/status") {

            // Let's extract the middle part of the path with the <pid> and
            // convert the <pid> into an integer. Log an error if it fails.
            pid, err := strconv.Atoi(path[6:strings.LastIndex(path, "/")])
            if err != nil {
                log.Println(err)
                return nil
            }

            // The status file contains the name of the process in its first line.
            // The line looks like "Name: theProcess".
            // Log an error in case we cant read the file.
            f, err := ioutil.ReadFile(path)
            if err != nil {
                log.Println(err)
                return nil
            }

            // Extract the process name from within the first line in the buffer
            name := string(f[6:bytes.IndexByte(f, '\n')])

            if name == args[1] {
                fmt.Printf("PID: %d, Name: %s will be killed.\n", pid, name)
                proc, err := os.FindProcess(pid)
                if err != nil {
                    log.Println(err)
                }
                // Kill the process
                proc.Kill()

                // Let's return a fake error to abort the walk through the
                // rest of the /proc directory tree
                return io.EOF
            }

        }
    }

    return nil
}

// main is the entry point of any go application
func main() {
    args = os.Args
    if len(args) != 2 {
        log.Fatalln("Usage: killprocess <processname>")
    }
    fmt.Printf("trying to kill process \"%s\"\n", args[1])

    err := filepath.Walk("/proc", findAndKillProcess)
    if err != nil {
        if err == io.EOF {
            // Not an error, just a signal when we are done
            err = nil
        } else {
            log.Fatal(err)
        }
    }
}

这只是一个肯定可以改进的例子。我为Linux编写了这个,并在Ubuntu 15.10上测试了代码。它不会在Windows上运行。

答案 1 :(得分:4)

我终于使用了以下内容:

// `echo "sudo_password" | sudo -S [command]`
// is used in order to run the command with `sudo`

_, err := exec.Command("sh", "-c", "echo '"+ sudopassword +"' | sudo -S pkill -SIGINT my_app_name").Output()

if err != nil {
    // ...
} else {
    // ...
}

我使用SIGINT信号来优雅地停止应用。

来自wikipedia

  • SIGINT

    当用户希望中断该过程时,SIGINT信号由其控制终端发送到进程。这通常是通过按Ctrl + C启动的,但在某些系统上,可以使用“删除”字符或“中断”键。

  • SIGKILL

    SIGKILL信号被发送到进程以使其立即终止(kill)。与SIGTERM和SIGINT相反,此信号无法捕获或忽略,接收过程无法在接收到此信号时执行任何清理。以下例外情况适用:

答案 2 :(得分:2)

跨平台(第 3 方)解决方案

几个月来我已经实施了各种解决方案来做到这一点,但出于某种原因,我花了很长时间才找到 gopsutil。它是一个 3rd 方库,对您来说可能是也可能不是交易破坏者,但它在我们的跨平台项目中完美运行。下面的示例将杀死具有匹配名称的第一个进程,但可以轻松地进行调整以杀死具有该名称的所有进程。

import "github.com/shirou/gopsutil/v3/process"

func KillProcess(name string) error {
    processes, err := process.Processes()
    if err != nil {
        return err
    }
    for _, p := range processes {
        n, err := p.Name()
        if err != nil {
            return err
        }
        if n == name {
            return p.Kill()
        }
    }
    return fmt.Errorf("process not found")
}

有上下文支持

作为一个额外的好处,该库还支持所有进程相关操作的上下文取消,包括进程查询和终止进程。

func KillAllProcessesCtx(ctx context.Context, name string) error {
    processes, err := process.ProcessesWithContext(ctx)
    if err != nil {
        return err
    }
    for _, p := range processes {
        n, err := p.NameWithContext(ctx)
        if err != nil {
            return err
        }
        if n == name {
            err = p.KillWithContext(ctx)
            if err != nil {
                return err
            }
        }
    }
    return nil
}

正常终止

该库还通过向进程发送您自己的信号来支持正常终止。

// Do this
err = p.SendSignal(syscall.SIGINT)
        
// Instead of this
err = p.Kill()

答案 3 :(得分:0)

你已经可以用 Go 通过进程 ID 杀死一个进程,所以真正的问题 这是从进程名称中获取进程 ID。这是示例 视窗:

package main

import (
   "fmt"
   "golang.org/x/sys/windows"
)

// unsafe.Sizeof(windows.ProcessEntry32{})
const processEntrySize = 568

func processID(name string) (uint32, error) {
   h, e := windows.CreateToolhelp32Snapshot(windows.TH32CS_SNAPPROCESS, 0)
   if e != nil { return 0, e }
   p := windows.ProcessEntry32{Size: processEntrySize}
   for {
      e := windows.Process32Next(h, &p)
      if e != nil { return 0, e }
      if windows.UTF16ToString(p.ExeFile[:]) == name {
         return p.ProcessID, nil
      }
   }
   return 0, fmt.Errorf("%q not found", name)
}

func main() {
   n, e := processID("WindowsTerminal.exe")
   if e != nil {
      panic(e)
   }
   println(n)
}

https://pkg.go.dev/golang.org/x/sys/windows#CreateToolhelp32Snapshot