mirror of
https://git.zx2c4.com/wireguard-windows
synced 2024-11-10 16:59:18 +00:00
80 lines
2.6 KiB
Go
80 lines
2.6 KiB
Go
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2019-2022 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package ui
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
|
|
"github.com/lxn/win"
|
|
"golang.org/x/sys/windows"
|
|
|
|
"golang.zx2c4.com/wireguard/windows/l18n"
|
|
)
|
|
|
|
func raise(hwnd win.HWND) {
|
|
if win.IsIconic(hwnd) {
|
|
win.ShowWindow(hwnd, win.SW_RESTORE)
|
|
}
|
|
|
|
win.SetActiveWindow(hwnd)
|
|
win.SetWindowPos(hwnd, win.HWND_TOPMOST, 0, 0, 0, 0, win.SWP_NOMOVE|win.SWP_NOSIZE|win.SWP_SHOWWINDOW)
|
|
win.SetForegroundWindow(hwnd)
|
|
win.SetWindowPos(hwnd, win.HWND_NOTOPMOST, 0, 0, 0, 0, win.SWP_NOMOVE|win.SWP_NOSIZE|win.SWP_SHOWWINDOW)
|
|
}
|
|
|
|
func raiseRemote(hwnd win.HWND) {
|
|
runtime.LockOSThread()
|
|
defer runtime.UnlockOSThread()
|
|
win.SendMessage(hwnd, raiseMsg, 0, 0)
|
|
currentForegroundHwnd := win.GetForegroundWindow()
|
|
currentForegroundThreadId := win.GetWindowThreadProcessId(currentForegroundHwnd, nil)
|
|
currentThreadId := win.GetCurrentThreadId()
|
|
win.AttachThreadInput(int32(currentForegroundThreadId), int32(currentThreadId), true)
|
|
win.SetWindowPos(hwnd, win.HWND_TOPMOST, 0, 0, 0, 0, win.SWP_NOMOVE|win.SWP_NOSIZE|win.SWP_SHOWWINDOW)
|
|
win.SetWindowPos(hwnd, win.HWND_NOTOPMOST, 0, 0, 0, 0, win.SWP_NOMOVE|win.SWP_NOSIZE|win.SWP_SHOWWINDOW)
|
|
win.SetForegroundWindow(hwnd)
|
|
win.AttachThreadInput(int32(currentForegroundThreadId), int32(currentThreadId), false)
|
|
win.SetFocus(hwnd)
|
|
win.SetActiveWindow(hwnd)
|
|
}
|
|
|
|
func RaiseUI() bool {
|
|
hwnd := win.FindWindow(windows.StringToUTF16Ptr(manageWindowWindowClass), nil)
|
|
if hwnd == 0 {
|
|
return false
|
|
}
|
|
raiseRemote(hwnd)
|
|
return true
|
|
}
|
|
|
|
func WaitForRaiseUIThenQuit() {
|
|
var handle win.HWINEVENTHOOK
|
|
runtime.LockOSThread()
|
|
defer runtime.UnlockOSThread()
|
|
handle, err := win.SetWinEventHook(win.EVENT_OBJECT_CREATE, win.EVENT_OBJECT_CREATE, 0, func(hWinEventHook win.HWINEVENTHOOK, event uint32, hwnd win.HWND, idObject, idChild int32, idEventThread, dwmsEventTime uint32) uintptr {
|
|
class := make([]uint16, len(manageWindowWindowClass)+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(manageWindowWindowClass) || windows.UTF16ToString(class) != manageWindowWindowClass {
|
|
return 0
|
|
}
|
|
win.UnhookWinEvent(handle)
|
|
raiseRemote(hwnd)
|
|
os.Exit(0)
|
|
return 0
|
|
}, 0, 0, win.WINEVENT_SKIPOWNPROCESS|win.WINEVENT_OUTOFCONTEXT)
|
|
if err != nil {
|
|
showErrorCustom(nil, l18n.Sprintf("WireGuard Detection Error"), l18n.Sprintf("Unable to wait for WireGuard window to appear: %v", err))
|
|
}
|
|
for {
|
|
var msg win.MSG
|
|
if m := win.GetMessage(&msg, 0, 0, 0); m != 0 {
|
|
win.TranslateMessage(&msg)
|
|
win.DispatchMessage(&msg)
|
|
}
|
|
}
|
|
}
|