golang 获取exe、dll、apk 版本号

通过go获取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)
}

本文来自:CSDN博客

感谢作者:guoxiao219

查看原文:golang 获取exe、dll、apk 版本号

• 编辑
• 预览