有没有办法在Go中的Windows上找到进程ID?

时间:2015-01-19 21:41:59

标签: go

在OS软件包中有一个FindProcess(),您可以在该过程的ID中传递以获取进程。然后,您可以在进程上调用kill,但有没有办法根据名称查找进程? (在Windows中)

例如,我希望能够做到这样的事情。

p, perr := os.FindProcessByName("Itunes")
if perr != nil {
    fmt.Println(perr)
}
p.Kill()

我只需要在Windows上使用它。

1 个答案:

答案 0 :(得分:3)

它不漂亮,但你可以使用w32绑定:(github.com/AllenDang/w32)

package main

import (
    "fmt"
    "github.com/AllenDang/w32"
    "unsafe"
)

func GetProcessName(id uint32) string {
    snapshot := w32.CreateToolhelp32Snapshot(w32.TH32CS_SNAPMODULE, id)
    if snapshot == w32.ERROR_INVALID_HANDLE {
        return "<UNKNOWN>"
    }
    defer w32.CloseHandle(snapshot)

    var me w32.MODULEENTRY32
    me.Size = uint32(unsafe.Sizeof(me))
    if w32.Module32First(snapshot, &me) {
        return w32.UTF16PtrToString(&me.SzModule[0])
    }

    return "<UNKNOWN>"
}

func ListProcesses() []uint32 {
    sz := uint32(1000)
    procs := make([]uint32, sz)
    var bytesReturned uint32
    if w32.EnumProcesses(procs, sz, &bytesReturned) {
        return procs[:int(bytesReturned)/4]
    }
    return []uint32{}
}

func FindProcessByName(name string) (uint32, error) {
    for _, pid := range ListProcesses() {
        if GetProcessName(pid) == name {
            return pid, nil
        }
    }
    return 0, fmt.Errorf("unknown process")
}

func main() {
    fmt.Println(FindProcessByName("chrome.exe"))
}