Merge pull request #389 from laozc/wmi-process-system-api-service

feat: Use WMI to implement Service related System APIs to reduce PowerShell overhead

Author: k8s-ci-robot

pkg/cim/system.go

@@ -10,6 +10,22 @@ import (
 	"github.com/microsoft/wmi/server2019/root/cimv2"
 )
 
+var (
+	BIOSSelectorList    = []string{"SerialNumber"}
+	ServiceSelectorList = []string{"DisplayName", "State", "StartMode"}
+)
+
+type ServiceInterface interface {
+	GetPropertyName() (string, error)
+	GetPropertyDisplayName() (string, error)
+	GetPropertyState() (string, error)
+	GetPropertyStartMode() (string, error)
+	GetDependents() ([]ServiceInterface, error)
+	StartService() (result uint32, err error)
+	StopService() (result uint32, err error)
+	Refresh() error
+}
+
 // QueryBIOSElement retrieves the BIOS element.
 //
 // The equivalent WMI query is:
@@ -33,6 +49,11 @@ func QueryBIOSElement(selectorList []string) (*cimv2.CIM_BIOSElement, error) {
 	return bios, err
 }
 
+// GetBIOSSerialNumber returns the BIOS serial number.
+func GetBIOSSerialNumber(bios *cimv2.CIM_BIOSElement) (string, error) {
+	return bios.GetPropertySerialNumber()
+}
+
 // QueryServiceByName retrieves a specific service by its name.
 //
 // The equivalent WMI query is:
@@ -55,3 +76,60 @@ func QueryServiceByName(name string, selectorList []string) (*cimv2.Win32_Servic
 
 	return service, err
 }
+
+// GetServiceName returns the name of a service.
+func GetServiceName(service ServiceInterface) (string, error) {
+	return service.GetPropertyName()
+}
+
+// GetServiceDisplayName returns the display name of a service.
+func GetServiceDisplayName(service ServiceInterface) (string, error) {
+	return service.GetPropertyDisplayName()
+}
+
+// GetServiceState returns the state of a service.
+func GetServiceState(service ServiceInterface) (string, error) {
+	return service.GetPropertyState()
+}
+
+// GetServiceStartMode returns the start mode of a service.
+func GetServiceStartMode(service ServiceInterface) (string, error) {
+	return service.GetPropertyStartMode()
+}
+
+// Win32Service wraps the WMI class Win32_Service (mainly for testing)
+type Win32Service struct {
+	*cimv2.Win32_Service
+}
+
+func (s *Win32Service) GetDependents() ([]ServiceInterface, error) {
+	collection, err := s.GetAssociated("Win32_DependentService", "Win32_Service", "Dependent", "Antecedent")
+	if err != nil {
+		return nil, err
+	}
+
+	var result []ServiceInterface
+	for _, coll := range collection {
+		service, err := cimv2.NewWin32_ServiceEx1(coll)
+		if err != nil {
+			return nil, err
+		}
+
+		result = append(result, &Win32Service{
+			service,
+		})
+	}
+	return result, nil
+}
+
+type Win32ServiceFactory struct {
+}
+
+func (impl Win32ServiceFactory) GetService(name string) (ServiceInterface, error) {
+	service, err := QueryServiceByName(name, ServiceSelectorList)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Win32Service{Win32_Service: service}, nil
+}