mirror of
https://github.com/fleetdm/fleet
synced 2026-05-22 16:39:01 +00:00
c97a0e2363
For https://github.com/fleetdm/fleet/issues/9943 This will help us avoid issues like this where the log message never worked right: https://github.com/fleetdm/fleet/pull/28296#discussion_r2047505191 Most of the changes are no-op type changes like removing unneeded typecast or disabling gosec on reviewed lines of code # Checklist for submitter If some of the following don't apply, delete the relevant line. <!-- Note that API documentation changes are now addressed by the product design team. --> - [x] Input data is properly validated, `SELECT *` is avoided, SQL injection is prevented (using placeholders for values in statements) - [x] Added/updated automated tests - [x] A detailed QA plan exists on the associated ticket (if it isn't there, work with the product group's QA engineer to add it) - [x] Manual QA for all new/changed functionality - For Orbit and Fleet Desktop changes: - [x] Make sure fleetd is compatible with the latest released version of Fleet (see [Must rule](https://github.com/fleetdm/fleet/blob/main/docs/Contributing/fleetd-development-and-release-strategy.md)). - [x] Orbit runs on macOS, Linux and Windows. Check if the orbit feature/bugfix should only apply to one platform (`runtime.GOOS`). - [x] Manual QA must be performed in the three main OSs, macOS, Windows and Linux. - [x] Auto-update manual QA, from released version of component to new version (see [tools/tuf/test](../tools/tuf/test/README.md)).
78 lines
2.5 KiB
Go
78 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
_ "embed"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog/log"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
// For Windows we must use ico format for the icon,
|
|
// see https://github.com/getlantern/systray/blob/6065fda28be8c8d91aeb5e20de25e1600b8664a3/systray_windows.go#L850-L856.
|
|
|
|
// In the past we implemented some logic to detect the Windows theme but it was buggy,
|
|
// so as a temporary fix we are using the same colored icon for both themes.
|
|
// Such theme detection logic was removed in this PR: https://github.com/fleetdm/fleet/pull/16402.
|
|
|
|
//go:embed windows_app.ico
|
|
var iconDark []byte
|
|
|
|
// blockWaitForStopEvent waits for the named event kernel object to be signalled
|
|
func blockWaitForStopEvent(channelId string) error {
|
|
if channelId == "" {
|
|
return errors.New("communication channel name should not be empty")
|
|
}
|
|
|
|
// converting go string to UTF16 windows compatible string
|
|
targetChannel := "Global\\comm-" + channelId
|
|
ev, err := windows.UTF16PtrFromString(targetChannel)
|
|
if err != nil {
|
|
return fmt.Errorf("there was a problem generating UTF16 string: %w", err)
|
|
}
|
|
|
|
// The right to use the object for synchronization
|
|
// https://learn.microsoft.com/en-us/windows/win32/sync/synchronization-object-security-and-access-rights
|
|
const EVENT_SYNCHRONIZE = 0x00100000
|
|
|
|
// block wait until channel is available
|
|
var handle windows.Handle = windows.InvalidHandle
|
|
for {
|
|
// OpenEvent API opens a named event object from the kernel object manager
|
|
// https://learn.microsoft.com/en-us/windows/win32/api/synchapi/nf-synchapi-openeventw
|
|
handle, err = windows.OpenEvent(EVENT_SYNCHRONIZE, false, ev)
|
|
if (err == nil) && (handle != windows.InvalidHandle) {
|
|
break
|
|
}
|
|
|
|
// wait before next handle check
|
|
time.Sleep(500 * time.Millisecond)
|
|
}
|
|
|
|
defer func() {
|
|
_ = windows.CloseHandle(handle)
|
|
}()
|
|
|
|
// OpenEvent() call was successful and our process got a handle to the named event kernel object
|
|
log.Info().Msg("Comm channel was acquired")
|
|
|
|
// now block wait for the handle to be signaled by Orbit
|
|
// https://learn.microsoft.com/en-us/windows/win32/api/synchapi/nf-synchapi-waitforsingleobject
|
|
s, err := windows.WaitForSingleObject(handle, windows.INFINITE)
|
|
if (err != nil) && (err != windows.ERROR_SUCCESS) {
|
|
return fmt.Errorf("there was a problem calling WaitForSingleObject: %w", err)
|
|
}
|
|
|
|
if s != windows.WAIT_OBJECT_0 {
|
|
return fmt.Errorf("event wait was interrupted for unknown reasons: %d", s)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func trayIconExists() bool {
|
|
log.Debug().Msg("tray icon checker is not implemented for this platform")
|
|
return true
|
|
}
|