diff --git a/main.go b/main.go deleted file mode 100644 index 9fe5ace..0000000 --- a/main.go +++ /dev/null @@ -1,170 +0,0 @@ -package main - -import ( - "bufio" - "encoding/hex" - "flag" - "fmt" - "golang.org/x/crypto/sha3" - "hash" - "io" - "log" - "os" - "path/filepath" - "strings" -) - -var ( - bits = flag.Int("b", 224, "Bits: 224, 256, 384 and 512.") - check = flag.String("c", "", "Check hashsum file.") - recursive = flag.Bool("r", false, "Process directories recursively.") - 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() { - flag.Parse() - - 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("Usage of", os.Args[0]+":") - fmt.Printf("%s [-v] [-b N] [-c ] [-r] -t \n\n", os.Args[0]) - flag.PrintDefaults() - os.Exit(1) - } - - if *target != "" && *recursive == false { - files, err := filepath.Glob(*target) - if err != nil { - log.Fatal(err) - } - for _, match := range files { - var h hash.Hash - if *bits == 224 { - h = sha3.New224() - } else if *bits == 256 { - h = sha3.New256() - } else if *bits == 384 { - h = sha3.New384() - } else if *bits == 512 { - h = sha3.New512() - } - f, err := os.Open(match) - if err != nil { - log.Fatal(err) - } - file, err := os.Stat(match) - if file.IsDir() { - } else { - if _, err := io.Copy(h, f); err != nil { - log.Fatal(err) - } - fmt.Println(hex.EncodeToString(h.Sum(nil)), "*"+f.Name()) - } - f.Close() - } - } - - if *target != "" && *recursive == true && *bits == 224 { - err := filepath.Walk(filepath.Dir(*target), - func(path string, info os.FileInfo, err error) error { - if err != nil { - return err - } - file, err := os.Stat(path) - if file.IsDir() { - } else { - filename := filepath.Base(path) - pattern := filepath.Base(*target) - matched, err := filepath.Match(pattern, filename) - if err != nil { - fmt.Println(err) - } - if matched { - var h hash.Hash - if *bits == 224 { - h = sha3.New224() - } else if *bits == 256 { - h = sha3.New256() - } else if *bits == 384 { - h = sha3.New384() - } else if *bits == 512 { - h = sha3.New512() - } - f, err := os.Open(path) - if err != nil { - log.Fatal(err) - } - if _, err := io.Copy(h, f); err != nil { - log.Fatal(err) - } - f.Close() - fmt.Println(hex.EncodeToString(h.Sum(nil)), "*"+f.Name()) - } - } - return nil - }) - if err != nil { - log.Println(err) - } - } - - if *check != "" { - file, err := os.Open(*check) - - if err != nil { - log.Fatalf("failed opening file: %s", err) - } - scanner := bufio.NewScanner(file) - scanner.Split(bufio.ScanLines) - var txtlines []string - - for scanner.Scan() { - txtlines = append(txtlines, scanner.Text()) - } - file.Close() - for _, eachline := range txtlines { - lines := strings.Split(string(eachline), " *") - if strings.Contains(string(eachline), " *") { - var h hash.Hash - if *bits == 224 { - h = sha3.New224() - } else if *bits == 256 { - h = sha3.New256() - } else if *bits == 384 { - h = sha3.New384() - } else if *bits == 512 { - h = sha3.New512() - } - _, err := os.Stat(lines[1]) - if err == nil { - f, err := os.Open(lines[1]) - if err != nil { - log.Fatal(err) - } - io.Copy(h, f) - f.Close() - - if *verbose { - if hex.EncodeToString(h.Sum(nil)) == lines[0] { - fmt.Println(lines[1]+"\t", "OK") - } else { - fmt.Println(lines[1]+"\t", "FAILED") - } - } else { - if hex.EncodeToString(h.Sum(nil)) == lines[0] { - } else { - os.Exit(1) - } - } - } else { - if *verbose { - fmt.Println(lines[1]+"\t", "Not found!") - } else { - os.Exit(1) - } - } - } - } - } -}