mirror of
https://github.com/prometheus-community/windows_exporter
synced 2024-12-22 14:32:12 +00:00
180 lines
3.6 KiB
Go
180 lines
3.6 KiB
Go
//go:build windows
|
|
|
|
package mi
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"syscall"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
type Instance struct {
|
|
ft *InstanceFT
|
|
classDecl *ClassDecl
|
|
serverName *uint16
|
|
nameSpace *uint16
|
|
_ [4]uintptr
|
|
}
|
|
|
|
type InstanceFT struct {
|
|
Clone uintptr
|
|
Destruct uintptr
|
|
Delete uintptr
|
|
IsA uintptr
|
|
GetClassName uintptr
|
|
SetNameSpace uintptr
|
|
GetNameSpace uintptr
|
|
GetElementCount uintptr
|
|
AddElement uintptr
|
|
SetElement uintptr
|
|
SetElementAt uintptr
|
|
GetElement uintptr
|
|
GetElementAt uintptr
|
|
ClearElement uintptr
|
|
ClearElementAt uintptr
|
|
GetServerName uintptr
|
|
SetServerName uintptr
|
|
GetClass uintptr
|
|
}
|
|
|
|
type ClassDecl struct {
|
|
Flags uint32
|
|
Code uint32
|
|
Name *uint16
|
|
Mqualifiers uintptr
|
|
NumQualifiers uint32
|
|
Mproperties uintptr
|
|
NumProperties uint32
|
|
Size uint32
|
|
SuperClass *uint16
|
|
SuperClassDecl uintptr
|
|
Methods uintptr
|
|
NumMethods uint32
|
|
|
|
Schema uintptr
|
|
ProviderFT uintptr
|
|
OwningClass uintptr
|
|
}
|
|
|
|
func (instance *Instance) Delete() error {
|
|
if instance == nil || instance.ft == nil {
|
|
return ErrNotInitialized
|
|
}
|
|
|
|
r0, _, _ := syscall.SyscallN(instance.ft.Delete, uintptr(unsafe.Pointer(instance)))
|
|
|
|
if result := ResultError(r0); !errors.Is(result, MI_RESULT_OK) {
|
|
return result
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (instance *Instance) GetElement(elementName string) (*Element, error) {
|
|
if instance == nil || instance.ft == nil {
|
|
return nil, ErrNotInitialized
|
|
}
|
|
|
|
elementNameUTF16, err := windows.UTF16PtrFromString(elementName)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to convert element name %s to UTF-16: %w", elementName, err)
|
|
}
|
|
|
|
var (
|
|
value uintptr
|
|
valueType ValueType
|
|
)
|
|
|
|
r0, _, _ := syscall.SyscallN(
|
|
instance.ft.GetElement,
|
|
uintptr(unsafe.Pointer(instance)),
|
|
uintptr(unsafe.Pointer(elementNameUTF16)),
|
|
uintptr(unsafe.Pointer(&value)),
|
|
uintptr(unsafe.Pointer(&valueType)),
|
|
0,
|
|
0,
|
|
)
|
|
|
|
if result := ResultError(r0); !errors.Is(result, MI_RESULT_OK) {
|
|
return nil, result
|
|
}
|
|
|
|
return &Element{
|
|
value: value,
|
|
valueType: valueType,
|
|
}, nil
|
|
}
|
|
|
|
func (instance *Instance) GetElementCount() (uint32, error) {
|
|
if instance == nil || instance.ft == nil {
|
|
return 0, ErrNotInitialized
|
|
}
|
|
|
|
var count uint32
|
|
|
|
r0, _, _ := syscall.SyscallN(
|
|
instance.ft.GetElementCount,
|
|
uintptr(unsafe.Pointer(instance)),
|
|
uintptr(unsafe.Pointer(&count)),
|
|
)
|
|
|
|
if result := ResultError(r0); !errors.Is(result, MI_RESULT_OK) {
|
|
return 0, result
|
|
}
|
|
|
|
return count, nil
|
|
}
|
|
|
|
func (instance *Instance) GetClassName() (string, error) {
|
|
if instance == nil || instance.ft == nil {
|
|
return "", ErrNotInitialized
|
|
}
|
|
|
|
var classNameUTF16 *uint16
|
|
|
|
r0, _, _ := syscall.SyscallN(
|
|
instance.ft.GetClassName,
|
|
uintptr(unsafe.Pointer(instance)),
|
|
uintptr(unsafe.Pointer(&classNameUTF16)),
|
|
)
|
|
|
|
if result := ResultError(r0); !errors.Is(result, MI_RESULT_OK) {
|
|
return "", result
|
|
}
|
|
|
|
if classNameUTF16 == nil {
|
|
return "", errors.New("class name is nil")
|
|
}
|
|
|
|
return windows.UTF16PtrToString(classNameUTF16), nil
|
|
}
|
|
|
|
func Instance_Print(instance *Instance) (string, error) {
|
|
elementMap := map[string]any{}
|
|
|
|
properties := instance.classDecl.Properties()
|
|
|
|
count, err := instance.GetElementCount()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
if count == 0 {
|
|
return "", nil
|
|
}
|
|
|
|
for _, property := range properties {
|
|
name := windows.UTF16PtrToString(property.Name)
|
|
|
|
element, _ := instance.GetElement(name)
|
|
value, _ := element.GetValue()
|
|
|
|
elementMap[windows.UTF16PtrToString(property.Name)] = value
|
|
}
|
|
|
|
return fmt.Sprintf("%v", elementMap), nil
|
|
}
|