Update main.go
This commit is contained in:
25
main.go
25
main.go
@@ -1,4 +1,5 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
@@ -13,18 +14,20 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
var bits = flag.Int("b", 224, "Bits: 224, 256, 384 and 512.")
|
var (
|
||||||
var check = flag.String("c", "", "Check hashsum file.")
|
bits = flag.Int("b", 224, "Bits: 224, 256, 384 and 512.")
|
||||||
var recursive = flag.Bool("r", false, "Process directories recursively.")
|
check = flag.String("c", "", "Check hashsum file.")
|
||||||
var target = flag.String("t", "", "Target file/wildcard to generate hashsum list.")
|
recursive = flag.Bool("r", false, "Process directories recursively.")
|
||||||
var verbose = flag.Bool("v", false, "Verbose mode. (The exit code is always 0 in this mode)")
|
target = flag.String("t", "", "Target file/wildcard to generate hashsum list.")
|
||||||
|
verbose = flag.Bool("v", false, "Verbose mode. (The exit code is always 0 in this mode)")
|
||||||
|
)
|
||||||
|
|
||||||
func main() {
|
func main() {
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
if (len(os.Args) < 2) || (*bits != 224 && *bits != 256 && *bits != 384 && *bits != 512) {
|
if (len(os.Args) < 2) || (*bits != 224 && *bits != 256 && *bits != 384 && *bits != 512) {
|
||||||
fmt.Println("SHA3 Hashsum Tool - ALBANESE Lab (c) 2020-2021\n")
|
fmt.Println("SHA3 Hashsum Tool - ALBANESE Lab (c) 2020-2021\n")
|
||||||
fmt.Println("Usage of",os.Args[0]+":")
|
fmt.Println("Usage of", os.Args[0]+":")
|
||||||
fmt.Printf("%s [-v] [-b N] [-c <hash.ext>] [-r] -t <file.ext>\n\n", os.Args[0])
|
fmt.Printf("%s [-v] [-b N] [-c <hash.ext>] [-r] -t <file.ext>\n\n", os.Args[0])
|
||||||
flag.PrintDefaults()
|
flag.PrintDefaults()
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
@@ -56,7 +59,7 @@ func main() {
|
|||||||
if _, err := io.Copy(h, f); err != nil {
|
if _, err := io.Copy(h, f); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*"+f.Name())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -94,7 +97,7 @@ func main() {
|
|||||||
if _, err := io.Copy(h, f); err != nil {
|
if _, err := io.Copy(h, f); err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*"+f.Name())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
@@ -141,9 +144,9 @@ func main() {
|
|||||||
|
|
||||||
if *verbose {
|
if *verbose {
|
||||||
if hex.EncodeToString(h.Sum(nil)) == lines[0] {
|
if hex.EncodeToString(h.Sum(nil)) == lines[0] {
|
||||||
fmt.Println(lines[1] + "\t", "OK")
|
fmt.Println(lines[1]+"\t", "OK")
|
||||||
} else {
|
} else {
|
||||||
fmt.Println(lines[1] + "\t", "FAILED")
|
fmt.Println(lines[1]+"\t", "FAILED")
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if hex.EncodeToString(h.Sum(nil)) == lines[0] {
|
if hex.EncodeToString(h.Sum(nil)) == lines[0] {
|
||||||
@@ -153,7 +156,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
if *verbose {
|
if *verbose {
|
||||||
fmt.Println(lines[1] + "\t", "Not found!")
|
fmt.Println(lines[1]+"\t", "Not found!")
|
||||||
} else {
|
} else {
|
||||||
os.Exit(1)
|
os.Exit(1)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user