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)
}
另外有需要云服务器可以了解下创新互联scvps.cn，海内外云服务器15元起步，三天无理由+7*72小时售后在线，公司持有idc许可证，提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案，具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势，专为企业上云打造定制，能够满足用户丰富、多元化的应用场景需求。
分享标题：golang获取exe、dll、apk版本号-创新互联
转载来源：http://pwwzsj.com/article/dgejjc.html
