1 files changed, 54 insertions, 0 deletions
diff --git a/ui/raise.go b/ui/raise.go
new file mode 100644
index 00000000..b53757c7
--- /dev/null
+++ b/ui/raise.go
@@ -0,0 +1,54 @@
+/* SPDX-License-Identifier: MIT
+ *
+ * Copyright (C) 2019 WireGuard LLC. All Rights Reserved.
+ */
+
+package ui
+
+import (
+	"fmt"
+	"golang.org/x/sys/windows"
+	"golang.zx2c4.com/wireguard/windows/ui/internal/walk"
+	"golang.zx2c4.com/wireguard/windows/ui/internal/walk/win"
+	"os"
+	"runtime"
+)
+
+const wireguardUIClass = "WireGuard UI - MainWindow"
+
+func RaiseUI() bool {
+	hwnd := win.FindWindow(windows.StringToUTF16Ptr(wireguardUIClass), windows.StringToUTF16Ptr("WireGuard"))
+	if hwnd == 0 {
+		return false
+	}
+	win.ShowWindow(hwnd, win.SW_NORMAL)
+	win.SetForegroundWindow(hwnd)
+	return true
+}
+
+func WaitForRaiseUIThenQuit() {
+	var handle win.HWINEVENTHOOK
+	runtime.LockOSThread()
+	handle, err := win.SetWinEventHook(win.EVENT_OBJECT_CREATE, win.EVENT_OBJECT_CREATE, 0, func(hWinEventHook win.HWINEVENTHOOK, event uint32, hwnd win.HWND, idObject int32, idChild int32, idEventThread uint32, dwmsEventTime uint32) uintptr {
+		class := make([]uint16, len(wireguardUIClass)+2) /* Plus 2, one for the null terminator, and one to see if this is only a prefix */
+		n, err := win.GetClassName(hwnd, &class[0], len(class))
+		if err != nil || n != len(wireguardUIClass) || windows.UTF16ToString(class) != wireguardUIClass {
+			return 0
+		}
+		win.UnhookWinEvent(handle)
+		win.ShowWindow(hwnd, win.SW_NORMAL)
+		win.SetForegroundWindow(hwnd)
+		os.Exit(0)
+		return 0
+	}, 0, 0, win.WINEVENT_SKIPOWNPROCESS|win.WINEVENT_OUTOFCONTEXT)
+	if err != nil {
+		walk.MsgBox(nil, "WireGuard Detection Error", fmt.Sprintf("Unable to wait for WireGuard window to appear: %v", err), walk.MsgBoxIconError)
+	}
+	for {
+		var msg win.MSG
+		if m := win.GetMessage(&msg, 0, 0, 0); m != 0 {
+			win.TranslateMessage(&msg)
+			win.DispatchMessage(&msg)
+		}
+	}
+}