Add windows service name lookup to procstat input (#4811)

This commit is contained in:
pytimer
2018-10-06 02:14:44 +08:00
committed by Daniel Nelson
parent db3967eb97
commit f81696b6b5
4 changed files with 87 additions and 1 deletions

View File

@@ -0,0 +1,49 @@
// +build windows
package procstat
import (
"unsafe"
"golang.org/x/sys/windows"
"golang.org/x/sys/windows/svc/mgr"
)
func getService(name string) (*mgr.Service, error) {
m, err := mgr.Connect()
if err != nil {
return nil, err
}
defer m.Disconnect()
srv, err := m.OpenService(name)
if err != nil {
return nil, err
}
return srv, nil
}
func queryPidWithWinServiceName(winServiceName string) (uint32, error) {
srv, err := getService(winServiceName)
if err != nil {
return 0, err
}
var p *windows.SERVICE_STATUS_PROCESS
var bytesNeeded uint32
var buf []byte
if err := windows.QueryServiceStatusEx(srv.Handle, windows.SC_STATUS_PROCESS_INFO, nil, 0, &bytesNeeded); err != windows.ERROR_INSUFFICIENT_BUFFER {
return 0, err
}
buf = make([]byte, bytesNeeded)
p = (*windows.SERVICE_STATUS_PROCESS)(unsafe.Pointer(&buf[0]))
if err := windows.QueryServiceStatusEx(srv.Handle, windows.SC_STATUS_PROCESS_INFO, &buf[0], uint32(len(buf)), &bytesNeeded); err != nil {
return 0, err
}
return p.ProcessId, nil
}