89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
//go:build windows
|
|
|
|
package procstat
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"unsafe"
|
|
|
|
"github.com/shirou/gopsutil/v3/process"
|
|
"golang.org/x/sys/windows"
|
|
"golang.org/x/sys/windows/svc/mgr"
|
|
)
|
|
|
|
func processName(p *process.Process) (string, error) {
|
|
return p.Name()
|
|
}
|
|
|
|
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
|
|
|
|
err = windows.QueryServiceStatusEx(srv.Handle, windows.SC_STATUS_PROCESS_INFO, nil, 0, &bytesNeeded)
|
|
if !errors.Is(err, windows.ERROR_INSUFFICIENT_BUFFER) {
|
|
return 0, err
|
|
}
|
|
|
|
buf = make([]byte, bytesNeeded)
|
|
p = (*windows.SERVICE_STATUS_PROCESS)(unsafe.Pointer(&buf[0])) //nolint:gosec // G103: Valid use of unsafe call to create SERVICE_STATUS_PROCESS
|
|
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
|
|
}
|
|
|
|
func collectMemmap(Process, string, map[string]any) {}
|
|
|
|
func findBySystemdUnits(_ []string) ([]processGroup, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func findByWindowsServices(services []string) ([]processGroup, error) {
|
|
groups := make([]processGroup, 0, len(services))
|
|
for _, service := range services {
|
|
pid, err := queryPidWithWinServiceName(service)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to query PID of service %q: %w", service, err)
|
|
}
|
|
|
|
p, err := process.NewProcess(int32(pid))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to find process for PID %d of service %q: %w", pid, service, err)
|
|
}
|
|
|
|
groups = append(groups, processGroup{
|
|
processes: []*process.Process{p},
|
|
tags: map[string]string{"win_service": service},
|
|
})
|
|
}
|
|
|
|
return groups, nil
|
|
}
|
|
|
|
func collectTotalReadWrite(_ Process) (r, w uint64, err error) {
|
|
return 0, 0, errors.ErrUnsupported
|
|
}
|