Skip to content

Commit

Permalink
Fix Service Access is denied in collector useApi
Browse files Browse the repository at this point in the history
Signed-off-by: Alvaro Cabanas <[email protected]>
  • Loading branch information
alvarocabanas committed Jul 29, 2022
1 parent c5ec339 commit 0de37af
Showing 1 changed file with 23 additions and 9 deletions.
32 changes: 23 additions & 9 deletions collector/service.go
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ package collector
import (
"fmt"
"strings"
"syscall"

"github.com/StackExchange/wmi"
"github.com/prometheus-community/windows_exporter/log"
Expand Down Expand Up @@ -234,30 +235,43 @@ func (c *serviceCollector) collectAPI(ch chan<- prometheus.Metric) error {
}
defer svcmgrConnection.Disconnect() //nolint:errcheck

// List All Services from the Services Manager
// List All Services from the Services Manager.
serviceList, err := svcmgrConnection.ListServices()
if err != nil {
return err
}

// Iterate through the Services List
// Iterate through the Services List.
for _, service := range serviceList {
// Retrieve handle for each service
serviceHandle, err := svcmgrConnection.OpenService(service)
// Get UTF16 service name.
serviceName, err := syscall.UTF16PtrFromString(service)
if err != nil {
log.Warnf("Service %s get name error: %#v", service, err)
continue
}
defer serviceHandle.Close()

// Get Service Configuration
serviceConfig, err := serviceHandle.Config()
// Open connection for service handler.
serviceHandle, err := windows.OpenService(svcmgrConnection.Handle, serviceName, windows.GENERIC_READ)
if err != nil {
log.Warnf("Open service %s error: %#v", service, err)
continue
}

// Get Service Current Status
serviceStatus, err := serviceHandle.Query()
// Create handle for each service.
serviceManager := &mgr.Service{Name: service, Handle: serviceHandle}
defer serviceManager.Close()

// Get Service Configuration.
serviceConfig, err := serviceManager.Config()
if err != nil {
log.Warnf("Get ervice %s config error: %#v", service, err)
continue
}

// Get Service Current Status.
serviceStatus, err := serviceManager.Query()
if err != nil {
log.Warnf("Get service %s status error: %#v", service, err)
continue
}

Expand Down

0 comments on commit 0de37af

Please sign in to comment.