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

Golang不知道大家是否熟悉？今天我将给大家介绍《是否可以在运行时检测Go二进制文件是否使用-H=windowsgui编译》，这篇文章主要会讲到等等知识点，如果你在看完本篇文章后，有更好的建议或者发现哪里有问题，希望大家都能积极评论指出，谢谢！希望我们能一起加油进步！

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

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

解决方案

According to the Go source，链接时好像设置了optionalheader的subsystem。

因此，您可以通过debug/pe获取。

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

请注意，os.executable() 可能会返回符号链接的路径，因此可能不可靠。参考：the document of 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")
    }
}

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。