golang 获取exe、dll、apk 版本号

// golang 实现读取exe dll apk 版本号
package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "path/filepath"
)

import (
    "github.com/lunny/axmlParser"
)

var (
    file fileInfo
)

const (
    MZ       = "MZ"
    PE       = "PE"
    RSRC     = ".rsrc"
    TYPET    = 16
    PEOFFSET = 64
    MACHINE  = 332
    DEFAULT  = `C:\Windows\System32\cmd.exe`
)

type fileInfo struct {
    FilePath string
    Version  string
    Debug    bool
}

func (f *fileInfo) checkError(err error) {
    if err != nil {
        log.Fatalln(err)
    }
}

// 获取exe dll版本
func (f *fileInfo) GetExeVersion() (err error) {
    file, err := os.Open(f.FilePath)
    f.checkError(err)

    // 第一次读取64 byte
    buffer := make([]byte, 64)
    _, err = file.Read(buffer)
    f.checkError(err)
    defer file.Close()

    str := string(buffer[0]) + string(buffer[1])
    if str != MZ {
        log.Fatalln("读取exe错误,找不到 MZ.", f.FilePath)
    }

    peOffset := f.unpack([]byte{buffer[60], buffer[61], buffer[62], buffer[63]})
    if peOffset < PEOFFSET {
        log.Fatalln("peOffset 读取错误.", f.FilePath)
    }

    // 读取从文件开头移位到 peOffset，第二次读取 24 byte
    _, err = file.Seek(int64(peOffset), 0)
    buffer = make([]byte, 24)
    _, err = file.Read(buffer)
    f.checkError(err)

    str = string(buffer[0]) + string(buffer[1])
    if str != PE {
        log.Fatalln("读取exe错误,找不到 PE.", f.FilePath)
    }

    machine := f.unpack([]byte{buffer[4], buffer[5]})
    if machine != MACHINE {
        log.Fatalln("machine 读取错误.", f.FilePath)
    }

    noSections := f.unpack([]byte{buffer[6], buffer[7]})
    optHdrSize := f.unpack([]byte{buffer[20], buffer[21]})

    // 读取从当前位置移位到 optHdrSize，第三次读取 40 byte
    file.Seek(int64(optHdrSize), 1)
    resFound := false
    for i := 0; i < int(noSections); i++ {
        buffer = make([]byte, 40)
        file.Read(buffer)
        str = string(buffer[0]) + string(buffer[1]) + string(buffer[2]) + string(buffer[3]) + string(buffer[4])
        if str == RSRC {
            resFound = true
            break
        }
    }
    if !resFound {
        log.Fatalln("读取exe错误,找不到 .rsrc.", f.FilePath)
    }

    infoVirt := f.unpack([]byte{buffer[12], buffer[13], buffer[14], buffer[15]})
    infoSize := f.unpack([]byte{buffer[16], buffer[17], buffer[18], buffer[19]})
    infoOff := f.unpack([]byte{buffer[20], buffer[21], buffer[22], buffer[23]})

    // 读取从文件开头位置移位到 infoOff，第四次读取 infoSize byte
    file.Seek(int64(infoOff), 0)
    buffer = make([]byte, infoSize)
    _, err = file.Read(buffer)
    f.checkError(err)

    nameEntries := f.unpack([]byte{buffer[12], buffer[13]})
    idEntries := f.unpack([]byte{buffer[14], buffer[15]})

    var infoFound bool
    var subOff, i int64
    for i = 0; i < (nameEntries + idEntries); i++ {
        typeT := f.unpack([]byte{buffer[i*8+16], buffer[i*8+17], buffer[i*8+18], buffer[i*8+19]})
        if typeT == TYPET {
            infoFound = true
            subOff = int64(f.unpack([]byte{buffer[i*8+20], buffer[i*8+21], buffer[i*8+22], buffer[i*8+23]}))
            break
        }
    }
    if !infoFound {
        log.Fatalln("读取exe错误,找不到 typeT == 16.", f.FilePath)
    }

    subOff = subOff & 0x7fffffff
    infoOff = f.unpack([]byte{buffer[subOff+20], buffer[subOff+21], buffer[subOff+22], buffer[subOff+23]}) //offset of first FILEINFO
    infoOff = infoOff & 0x7fffffff
    infoOff = f.unpack([]byte{buffer[infoOff+20], buffer[infoOff+21], buffer[infoOff+22], buffer[infoOff+23]}) //offset to data
    dataOff := f.unpack([]byte{buffer[infoOff], buffer[infoOff+1], buffer[infoOff+2], buffer[infoOff+3]})
    dataOff = dataOff - infoVirt

    version1 := f.unpack([]byte{buffer[dataOff+48], buffer[dataOff+48+1]})
    version2 := f.unpack([]byte{buffer[dataOff+48+2], buffer[dataOff+48+3]})
    version3 := f.unpack([]byte{buffer[dataOff+48+4], buffer[dataOff+48+5]})
    version4 := f.unpack([]byte{buffer[dataOff+48+6], buffer[dataOff+48+7]})

    version := fmt.Sprintf("%d.%d.%d.%d", version2, version1, version4, version3)
    f.Version = version

    return nil
}

func (f *fileInfo) unpack(b []byte) (num int64) {
    for i := 0; i < len(b); i++ {
        num = 256*num + int64((b[len(b)-1-i] & 0xff))
    }
    return
}

// 获取 apk 版本
func (f *fileInfo) GetApkVersion() (err error) {
    listener := new(axmlParser.AppNameListener)
    _, err = axmlParser.ParseApk(f.FilePath, listener)
    f.checkError(err)

    f.Version = listener.VersionName
    return nil
}

func init() {
    flag.StringVar(&file.FilePath, "path", DEFAULT, "Get exe or dll or apk version information.")
    flag.BoolVar(&file.Debug, "d", false, "if true print exe or dll file name.")
}

func main() {
    flag.Parse()

    suffix := filepath.Ext(file.FilePath)
    switch suffix {
    case ".exe", ".dll":
        file.GetExeVersion()
    case ".apk":
        file.GetApkVersion()
    default:
        log.Fatalln("仅能获取exe、dll、apk版本号,请重新输入程序路径.", file.FilePath)
    }

    switch {
    case file.Debug:
        fmt.Printf("%s 的版本号为: ", file.FilePath)
    case file.FilePath == DEFAULT:
        flag.PrintDefaults()
        fmt.Printf("%s 的版本号为: ", file.FilePath)
    }

    fmt.Printf("%s", file.Version)
}