go - 是否可以检测 Go 二进制文件是否在运行时使用 -H=windowsgui 编译

Question

-tray我构建了一个工具，如果在启动时传递了标志，可以从控制台运行或在系统托盘中运行。仅在go build <list of go files>没有构建标志的情况下运行将创建一个可以在系统托盘中运行的二进制文件，但它也会产生一个控制台窗口。如果我通过了构建标志-H=windowsgui，那么控制台窗口将被隐藏，但我仍然需要通过-tray才能让它在系统托盘中实际运行。

有没有办法在运行时检测到二进制文件是用标志构建的，-H=windowsgui这样我就可以自动做正确的事情并启用托盘，而无需通过-tray标志？

标签： go

Answer 1

根据 Go 源，它似乎在链接时设置Subsystem。OptionalHeader

因此，您可以使用debug/pe来获取它。

以下代码将it is windows GUI在使用 编译时go build -ldflags "-H windowsgui"打印，it is windows CUI否则打印。

请注意，os.Executable()可能返回符号链接的路径，因此可能不可靠。参考：os.Executable()的文档

package main

import (
    "debug/pe"
    "fmt"
    "os"
)

// these constants are copied from https://github.com/golang/go/blob/6219b48e11f36329de801f62f18448bb4b1cd1a5/src/cmd/link/internal/ld/pe.go#L92-L93  
const (
    IMAGE_SUBSYSTEM_WINDOWS_GUI = 2
    IMAGE_SUBSYSTEM_WINDOWS_CUI = 3
)

func main() {
    fileName, err := os.Executable()
    if err != nil {
        panic(err)
    }
    fl, err := pe.Open(fileName)
    if err != nil {
        panic(err) // maybe not windows binary, or unreadable for some reasons
    }
    defer fl.Close()

    var subsystem uint16
    if header, ok := fl.OptionalHeader.(*pe.OptionalHeader64); ok {
        subsystem = header.Subsystem
    } else if header, ok := fl.OptionalHeader.(*pe.OptionalHeader32); ok {
        subsystem = header.Subsystem
    }

    if subsystem == IMAGE_SUBSYSTEM_WINDOWS_GUI {
        fmt.Println("it is windows GUI")
    } else if subsystem == IMAGE_SUBSYSTEM_WINDOWS_CUI {
        fmt.Println("it is windows CUI")
    } else {
        fmt.Println("binary type unknown")
    }
}