mi: replace all WMI calls with MI calls (#1714)

internal/mi/instance.go

@@ -0,0 +1,179 @@
+//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
+}