Update main.go
This commit is contained in:
Pedro Albanese
359
main.go
359
main.go
@@ -5,6 +5,7 @@ import (
|
|||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"golang.org/x/crypto/sha3"
|
"golang.org/x/crypto/sha3"
|
||||||
|
"hash"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
@@ -12,8 +13,8 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
var check = flag.String("c", "", "Check hashsum file.")
|
|
||||||
var bits = flag.Int("b", 224, "Bits: 224, 256, 384 and 512.")
|
var bits = flag.Int("b", 224, "Bits: 224, 256, 384 and 512.")
|
||||||
|
var check = flag.String("c", "", "Check hashsum file.")
|
||||||
var recursive = flag.Bool("r", false, "Process directories recursively.")
|
var recursive = flag.Bool("r", false, "Process directories recursively.")
|
||||||
var target = flag.String("t", "", "Target file/wildcard to generate hashsum list.")
|
var target = flag.String("t", "", "Target file/wildcard to generate hashsum list.")
|
||||||
var verbose = flag.Bool("v", false, "Verbose mode. (The exit code is always 0 in this mode)")
|
var verbose = flag.Bool("v", false, "Verbose mode. (The exit code is always 0 in this mode)")
|
||||||
@@ -30,14 +31,23 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == false && *bits == 224 {
|
if *target != "" && *recursive == false {
|
||||||
files, err := filepath.Glob(*target)
|
files, err := filepath.Glob(*target)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, match := range files {
|
for _, match := range files {
|
||||||
h := sha3.New224()
|
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)
|
f, err := os.Open(match)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
@@ -66,7 +76,16 @@ func main() {
|
|||||||
fmt.Println(err)
|
fmt.Println(err)
|
||||||
}
|
}
|
||||||
if matched {
|
if matched {
|
||||||
h := sha3.New224()
|
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)
|
f, err := os.Open(path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
@@ -85,172 +104,7 @@ func main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == false && *bits == 256 {
|
if *check != "" {
|
||||||
files, err := filepath.Glob(*target)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, match := range files {
|
|
||||||
h := sha3.New256()
|
|
||||||
f, err := os.Open(match)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if _, err := io.Copy(h, f); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == true && *bits == 256 {
|
|
||||||
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 {
|
|
||||||
h := sha3.New256()
|
|
||||||
f, err := os.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if _, err := io.Copy(h, f); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == false && *bits == 384 {
|
|
||||||
files, err := filepath.Glob(*target)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, match := range files {
|
|
||||||
h := sha3.New384()
|
|
||||||
f, err := os.Open(match)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if _, err := io.Copy(h, f); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == true && *bits == 384 {
|
|
||||||
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 {
|
|
||||||
h := sha3.New384()
|
|
||||||
f, err := os.Open(path)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if _, err := io.Copy(h, f); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == false && *bits == 512 {
|
|
||||||
files, err := filepath.Glob(*target)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, match := range files {
|
|
||||||
h := sha3.New512()
|
|
||||||
f, err := os.Open(match)
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
if _, err := io.Copy(h, f); err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *target != "" && *recursive == true && *bits == 512 {
|
|
||||||
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 {
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
fmt.Println(hex.EncodeToString(h.Sum(nil)), "*" + f.Name())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *check != "" && *bits == 224 {
|
|
||||||
file, err := os.Open(*check)
|
file, err := os.Open(*check)
|
||||||
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -267,7 +121,16 @@ func main() {
|
|||||||
for _, eachline := range txtlines {
|
for _, eachline := range txtlines {
|
||||||
lines := strings.Split(string(eachline), " *")
|
lines := strings.Split(string(eachline), " *")
|
||||||
if strings.Contains(string(eachline), " *") {
|
if strings.Contains(string(eachline), " *") {
|
||||||
h := sha3.New224()
|
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])
|
_, err := os.Stat(lines[1])
|
||||||
if err == nil {
|
if err == nil {
|
||||||
f, err := os.Open(lines[1])
|
f, err := os.Open(lines[1])
|
||||||
@@ -297,159 +160,5 @@ func main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *check != "" && *bits == 256 {
|
|
||||||
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), " *") {
|
|
||||||
h := sha3.New256()
|
|
||||||
_, err := os.Stat(lines[1])
|
|
||||||
if err == nil {
|
|
||||||
f, err := os.Open(lines[1])
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
io.Copy(h, f)
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *check != "" && *bits == 384 {
|
|
||||||
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), " *") {
|
|
||||||
h := sha3.New384()
|
|
||||||
_, err := os.Stat(lines[1])
|
|
||||||
if err == nil {
|
|
||||||
f, err := os.Open(lines[1])
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal(err)
|
|
||||||
}
|
|
||||||
io.Copy(h, f)
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
if *check != "" && *bits == 512 {
|
|
||||||
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), " *") {
|
|
||||||
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)
|
|
||||||
|
|
||||||
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)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user