You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Victorique/vtqe/tools/hash.go

72 lines
1.9 KiB
Go

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

package tools
import (
"fmt"
"os"
"b612.me/starainrt"
"b612.me/starlog"
"github.com/spf13/cobra"
)
var hashcmd = &cobra.Command{
Use: "hash",
Short: "多种方法哈希值校验",
Long: "进行多种方法哈希值校验",
Run: func(this *cobra.Command, args []string) {
var cumethod, method []string
var result map[string]string
var err error
crypto := new(starainrt.StarCrypto)
cumethod = []string{"md5", "crc32", "sha512", "sha384", "sha256", "sha224", "sha1"}
if ok, _ := this.Flags().GetBool("all"); ok {
method = cumethod
} else {
if len(args) == 0 {
this.Usage()
os.Exit(1)
}
for _, v := range cumethod {
if ok, _ := this.Flags().GetBool(v); ok {
method = append(method, v)
}
}
if len(method) == 0 {
method = append(method, "md5")
}
}
if ok, _ := this.Flags().GetBool("file"); ok {
result, err = crypto.FileSumAll(args[0], method, func(pect float64) {
if pect != 100.0 {
fmt.Printf("校验已完成:%f%%\r", pect)
} else {
fmt.Printf("校验已完成:%f%%\n", pect)
}
})
} else {
result, err = crypto.SumAll([]byte(args[0]), method)
}
if err != nil {
starlog.StdPrintln(0, starlog.RED, "错误:"+err.Error())
}
for _, v := range method {
fmt.Printf("%s%s\n", v, result[v])
}
},
}
func init() {
hashcmd.Flags().BoolP("all", "a", false, "使用所有的校验方法")
hashcmd.Flags().BoolP("file", "f", false, "对指定文件进行校验")
hashcmd.Flags().BoolP("md5", "m", false, "进行MD5校验默认")
hashcmd.Flags().BoolP("crc32", "c", false, "进行CRC32校验")
hashcmd.Flags().BoolP("sha512", "s", false, "进行SHA512校验")
hashcmd.Flags().Bool("sha384", false, "进行SHA384校验")
hashcmd.Flags().Bool("sha256", false, "进行SHA256校验")
hashcmd.Flags().Bool("sha224", false, "进行SHA224校验")
hashcmd.Flags().Bool("sha1", false, "进行SHA1校验")
Maincmd.AddCommand(hashcmd)
}