/* SPDX-License-Identifier: MIT * * Copyright (C) 2019 WireGuard LLC. All Rights Reserved. */ package ui import ( "fmt" "runtime" "runtime/debug" "time" "github.com/lxn/walk" "github.com/lxn/win" "golang.zx2c4.com/wireguard/windows/manager" "golang.zx2c4.com/wireguard/windows/version" ) var noTrayAvailable = false var shouldQuitManagerWhenExiting = false var startTime = time.Now() func RunUI() { runtime.LockOSThread() defer func() { if err := recover(); err != nil { walk.MsgBox(nil, "Panic", fmt.Sprint(err, "\n\n", string(debug.Stack())), walk.MsgBoxIconError) panic(err) } }() var err error var ( mtw *ManageTunnelsWindow tray *Tray ) for mtw == nil { mtw, err = NewManageTunnelsWindow() if err != nil { time.Sleep(time.Millisecond * 400) } } for tray == nil { tray, err = NewTray(mtw) if err != nil { if version.OsIsCore() { noTrayAvailable = true break } time.Sleep(time.Millisecond * 400) } } manager.IPCClientRegisterManagerStopping(func() { mtw.Synchronize(func() { walk.App().Exit(0) }) }) onUpdateNotification := func(updateState manager.UpdateState) { if updateState == manager.UpdateStateUnknown { return } mtw.Synchronize(func() { switch updateState { case manager.UpdateStateFoundUpdate: mtw.UpdateFound() tray.UpdateFound() case manager.UpdateStateUpdatesDisabledUnofficialBuild: mtw.SetTitle(mtw.Title() + " (unsigned build, no updates)") } }) } manager.IPCClientRegisterUpdateFound(onUpdateNotification) go func() { updateState, err := manager.IPCClientUpdateState() if err == nil { onUpdateNotification(updateState) } }() if noTrayAvailable { win.ShowWindow(mtw.Handle(), win.SW_MINIMIZE) } mtw.Run() tray.Dispose() mtw.Dispose() if shouldQuitManagerWhenExiting { _, err := manager.IPCClientQuit(true) if err != nil { walk.MsgBox(nil, "Error Exiting WireGuard", fmt.Sprintf("Unable to exit service due to: %v. You may want to stop WireGuard from the service manager.", err), walk.MsgBoxIconError) } } } func onQuit() { shouldQuitManagerWhenExiting = true walk.App().Exit(0) }