Introduce SetupAPI - Windows device and driver management API

Signed-off-by: Simon Rozman <simon@rozman.si>

diff --git a/setupapi/mksyscall.go b/setupapi/mksyscall.go
new file mode 100644 (file)
index 0000000..3447dee
--- /dev/null
+++ b/setupapi/mksyscall.go
@@ -0,0 +1,8 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package setupapi
+
+//go:generate go run $GOROOT/src/syscall/mksyscall_windows.go -output zsetupapi_windows.go setupapi_windows.go

diff --git a/setupapi/setupapi.go b/setupapi/setupapi.go
new file mode 100644 (file)
index 0000000..69703b3
--- /dev/null
+++ b/setupapi/setupapi.go
@@ -0,0 +1,6 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package setupapi

diff --git a/setupapi/setupapi_windows.go b/setupapi/setupapi_windows.go
new file mode 100644 (file)
index 0000000..22cacee
--- /dev/null
+++ b/setupapi/setupapi_windows.go
@@ -0,0 +1,41 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package setupapi
+
+import (
+       "golang.org/x/sys/windows"
+)
+
+const (
+       SP_MAX_MACHINENAME_LENGTH = windows.MAX_PATH + 3
+)
+
+type DIGCF uint32
+
+const (
+       Default         DIGCF = 0x00000001
+       Present         DIGCF = 0x00000002
+       AllClasses      DIGCF = 0x00000004
+       Profile         DIGCF = 0x00000008
+       DeviceInterface DIGCF = 0x00000010
+       InterfaceDevice DIGCF = 0x00000010
+)
+
+type DevInfo windows.Handle
+
+//sys  setupDiGetClassDevsEx(ClassGuid *windows.GUID, Enumerator *string, hwndParent uintptr, Flags uint32, DeviceInfoSet uintptr, MachineName string, reserved uint32) (handle windows.Handle, err error) = setupapi.SetupDiGetClassDevsExW
+
+// The SetupDiGetClassDevsEx function returns a handle to a device information set that contains requested device information elements for a local or a remote computer.
+func SetupDiGetClassDevsEx(ClassGuid *windows.GUID, Enumerator string, hwndParent uintptr, Flags DIGCF, DeviceInfoSet DevInfo, MachineName string) (DevInfo, error) {
+       enumerator := &Enumerator
+
+       if Enumerator == "" {
+               enumerator = nil
+       }
+
+       h, err := setupDiGetClassDevsEx(ClassGuid, enumerator, hwndParent, uint32(Flags), uintptr(DeviceInfoSet), MachineName, 0)
+       return DevInfo(h), err
+}

diff --git a/setupapi/zsetupapi_windows.go b/setupapi/zsetupapi_windows.go
new file mode 100644 (file)
index 0000000..84ead2d
--- /dev/null
+++ b/setupapi/zsetupapi_windows.go
@@ -0,0 +1,65 @@
+// Code generated by 'go generate'; DO NOT EDIT.
+
+package setupapi
+
+import (
+       "syscall"
+       "unsafe"
+
+       "golang.org/x/sys/windows"
+)
+
+var _ unsafe.Pointer
+
+// 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
+}
+
+var (
+       modsetupapi = windows.NewLazySystemDLL("setupapi.dll")
+
+       procSetupDiGetClassDevsExW = modsetupapi.NewProc("SetupDiGetClassDevsExW")
+)
+
+func setupDiGetClassDevsEx(ClassGuid *windows.GUID, Enumerator *string, hwndParent uintptr, Flags uint32, DeviceInfoSet uintptr, MachineName string, reserved uint32) (handle windows.Handle, err error) {
+       var _p0 *uint16
+       _p0, err = syscall.UTF16PtrFromString(MachineName)
+       if err != nil {
+               return
+       }
+       return _setupDiGetClassDevsEx(ClassGuid, Enumerator, hwndParent, Flags, DeviceInfoSet, _p0, reserved)
+}
+
+func _setupDiGetClassDevsEx(ClassGuid *windows.GUID, Enumerator *string, hwndParent uintptr, Flags uint32, DeviceInfoSet uintptr, MachineName *uint16, reserved uint32) (handle windows.Handle, err error) {
+       r0, _, e1 := syscall.Syscall9(procSetupDiGetClassDevsExW.Addr(), 7, uintptr(unsafe.Pointer(ClassGuid)), uintptr(unsafe.Pointer(Enumerator)), uintptr(hwndParent), uintptr(Flags), uintptr(DeviceInfoSet), uintptr(unsafe.Pointer(MachineName)), uintptr(reserved), 0, 0)
+       handle = windows.Handle(r0)
+       if handle == 0 {
+               if e1 != 0 {
+                       err = errnoErr(e1)
+               } else {
+                       err = syscall.EINVAL
+               }
+       }
+       return
+}