mirror of https://github.com/shirou/gopsutil
[windows]service: add windows service feature
parent
eeb1d38d69
commit
26add8e657
@ -0,0 +1,32 @@
|
|||||||
|
// +build windows
|
||||||
|
|
||||||
|
package winservices
|
||||||
|
|
||||||
|
import (
|
||||||
|
"golang.org/x/sys/windows/svc/mgr"
|
||||||
|
)
|
||||||
|
|
||||||
|
type scmanager struct {
|
||||||
|
mgr *mgr.Mgr
|
||||||
|
}
|
||||||
|
|
||||||
|
func openSCManager() (*scmanager, error) {
|
||||||
|
m, err := mgr.Connect()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &scmanager{m}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (sc *scmanager) close() error {
|
||||||
|
return sc.mgr.Disconnect()
|
||||||
|
}
|
||||||
|
|
||||||
|
func getService(serviceName string) (*mgr.Service, error) {
|
||||||
|
m, err := openSCManager()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer m.close()
|
||||||
|
return m.mgr.OpenService(serviceName)
|
||||||
|
}
|
@ -0,0 +1,126 @@
|
|||||||
|
// +build windows
|
||||||
|
|
||||||
|
package winservices
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
"golang.org/x/sys/windows/svc"
|
||||||
|
"golang.org/x/sys/windows/svc/mgr"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Service represent a windows service.
|
||||||
|
type Service struct {
|
||||||
|
Name string
|
||||||
|
Config mgr.Config
|
||||||
|
Status ServiceStatus
|
||||||
|
srv *mgr.Service
|
||||||
|
}
|
||||||
|
|
||||||
|
// ServiceStatus combines State and Accepted commands to fully describe running service.
|
||||||
|
type ServiceStatus struct {
|
||||||
|
State svc.State
|
||||||
|
Accepts svc.Accepted
|
||||||
|
Pid uint32
|
||||||
|
Win32ExitCode uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewService create and return a windows Service
|
||||||
|
func NewService(name string) (*Service, error) {
|
||||||
|
// call windows service function need to OpenService handler,
|
||||||
|
// so first call func OpenService to get the specified service handler.
|
||||||
|
service, err := getService(name)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &Service{
|
||||||
|
Name: name,
|
||||||
|
srv: service,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetServiceDetail get a windows service by name
|
||||||
|
func (s *Service) GetServiceDetail() error {
|
||||||
|
return s.GetServiceDetailWithContext(context.Background())
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetServiceDetailWithContext get a windows service by name
|
||||||
|
func (s *Service) GetServiceDetailWithContext(ctx context.Context) error {
|
||||||
|
config, err := s.QueryServiceConfigWithContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s.Config = config
|
||||||
|
|
||||||
|
status, err := s.QueryStatusWithContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
s.Status = status
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// QueryServiceConfig return the specified service config
|
||||||
|
func (s *Service) QueryServiceConfig() (mgr.Config, error) {
|
||||||
|
return s.QueryServiceConfigWithContext(context.Background())
|
||||||
|
}
|
||||||
|
|
||||||
|
// QueryServiceConfigWithContext call QueryServiceConfig() and QueryServiceConfig2()
|
||||||
|
// implement windows https://msdn.microsoft.com/en-us/library/windows/desktop/ms684932(v=vs.85).aspx
|
||||||
|
func (s *Service) QueryServiceConfigWithContext(ctx context.Context) (mgr.Config, error) {
|
||||||
|
return s.srv.Config()
|
||||||
|
}
|
||||||
|
|
||||||
|
// QueryStatus return the specified name service currentState and ControlsAccepted
|
||||||
|
func (s *Service) QueryStatus() (ServiceStatus, error) {
|
||||||
|
return s.QueryStatusWithContext(context.Background())
|
||||||
|
}
|
||||||
|
|
||||||
|
// QueryStatusWithContext return the specified name service currentState and ControlsAccepted
|
||||||
|
func (s *Service) QueryStatusWithContext(ctx context.Context) (ServiceStatus, error) {
|
||||||
|
var p *windows.SERVICE_STATUS_PROCESS
|
||||||
|
var bytesNeeded uint32
|
||||||
|
var buf []byte
|
||||||
|
|
||||||
|
if err := QueryServiceStatusEx(s.srv.Handle, SC_STATUS_PROCESS_INFO, nil, 0, &bytesNeeded); err != windows.ERROR_INSUFFICIENT_BUFFER {
|
||||||
|
return ServiceStatus{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
buf = make([]byte, bytesNeeded)
|
||||||
|
p = (*windows.SERVICE_STATUS_PROCESS)(unsafe.Pointer(&buf[0]))
|
||||||
|
if err := QueryServiceStatusEx(s.srv.Handle, SC_STATUS_PROCESS_INFO, &buf[0], uint32(len(buf)), &bytesNeeded); err != nil {
|
||||||
|
return ServiceStatus{}, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return ServiceStatus{
|
||||||
|
State: svc.State(p.CurrentState),
|
||||||
|
Accepts: svc.Accepted(p.ControlsAccepted),
|
||||||
|
Pid: p.ProcessId,
|
||||||
|
Win32ExitCode: p.Win32ExitCode,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListServices return all windows service
|
||||||
|
// reference to golang.org/x/sys/windows/svc/mgr#ListServices()
|
||||||
|
func ListServices() ([]Service, error) {
|
||||||
|
m, err := openSCManager()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer m.close()
|
||||||
|
|
||||||
|
names, err := m.mgr.ListServices()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
services := make([]Service, 0)
|
||||||
|
for _, name := range names {
|
||||||
|
services = append(services, Service{Name: name})
|
||||||
|
}
|
||||||
|
|
||||||
|
return services, nil
|
||||||
|
}
|
@ -0,0 +1,57 @@
|
|||||||
|
// +build windows
|
||||||
|
|
||||||
|
package winservices
|
||||||
|
|
||||||
|
import (
|
||||||
|
"syscall"
|
||||||
|
"unsafe"
|
||||||
|
|
||||||
|
"golang.org/x/sys/windows"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
modadvapi32 = windows.NewLazySystemDLL("advapi32.dll")
|
||||||
|
procQueryServiceStatusEx = modadvapi32.NewProc("QueryServiceStatusEx")
|
||||||
|
)
|
||||||
|
|
||||||
|
// Do the interface allocations only once for common
|
||||||
|
// Errno values.
|
||||||
|
const (
|
||||||
|
errnoERROR_IO_PENDING = 997
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
errERROR_IO_PENDING error = syscall.Errno(errnoERROR_IO_PENDING)
|
||||||
|
)
|
||||||
|
|
||||||
|
// errnoErr returns common boxed Errno values, to prevent
|
||||||
|
// allocations at runtime.
|
||||||
|
func errnoErr(e syscall.Errno) error {
|
||||||
|
switch e {
|
||||||
|
case 0:
|
||||||
|
return nil
|
||||||
|
case errnoERROR_IO_PENDING:
|
||||||
|
return errERROR_IO_PENDING
|
||||||
|
}
|
||||||
|
// TODO: add more here, after collecting data on the common
|
||||||
|
// error values see on Windows. (perhaps when running
|
||||||
|
// all.bat?)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
// SC_STATUS_PROCESS_INFO reference to https://msdn.microsoft.com/en-us/library/windows/desktop/ms684941(v=vs.85).aspx,
|
||||||
|
// Use SC_STATUS_PROCESS_INFO to retrieve the service status information.
|
||||||
|
const SC_STATUS_PROCESS_INFO int = 0
|
||||||
|
|
||||||
|
// QueryServiceStatusEx golang/sys/windows standard library can not implement QueryServiceStatusEx.
|
||||||
|
func QueryServiceStatusEx(service windows.Handle, infoLevel int, buff *byte, buffSize uint32, bytesNeeded *uint32) (err error) {
|
||||||
|
r1, _, e1 := syscall.Syscall6(procQueryServiceStatusEx.Addr(), 5, uintptr(service), uintptr(infoLevel), uintptr(unsafe.Pointer(buff)), uintptr(buffSize), uintptr(unsafe.Pointer(bytesNeeded)), 0)
|
||||||
|
if r1 == 0 {
|
||||||
|
if e1 != 0 {
|
||||||
|
err = errnoErr(e1)
|
||||||
|
} else {
|
||||||
|
err = syscall.EINVAL
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
Loading…
Reference in New Issue