From 31949136df477c5699d7476e909d9bf30a0904cc Mon Sep 17 00:00:00 2001
From: Simon Rozman <simon@rozman.si>
Date: Thu, 31 Jan 2019 15:20:11 +0100
Subject: [PATCH] Introduce SetupAPI - Windows device and driver management API

Signed-off-by: Simon Rozman <simon@rozman.si>
---
 setupapi/mksyscall.go         |  8 +++++
 setupapi/setupapi.go          |  6 ++++
 setupapi/setupapi_windows.go  | 41 ++++++++++++++++++++++
 setupapi/zsetupapi_windows.go | 65 +++++++++++++++++++++++++++++++++++
 4 files changed, 120 insertions(+)
 create mode 100644 setupapi/mksyscall.go
 create mode 100644 setupapi/setupapi.go
 create mode 100644 setupapi/setupapi_windows.go
 create mode 100644 setupapi/zsetupapi_windows.go

diff --git a/setupapi/mksyscall.go b/setupapi/mksyscall.go
new file mode 100644
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
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
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
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
+}