65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
// Package checksum computes checksums, like MD5 or SHA256, for large files
|
|
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"crypto/md5"
|
|
"crypto/sha1"
|
|
"crypto/sha256"
|
|
"hash"
|
|
"os"
|
|
|
|
"golang.org/x/crypto/blake2s"
|
|
)
|
|
|
|
// MD5sum returns MD5 checksum of filename
|
|
func MD5sum(filename string) (string, error) {
|
|
return sum(md5.New(), filename)
|
|
}
|
|
|
|
// SHA256sum returns SHA256 checksum of filename
|
|
func SHA256sum(filename string) (string, error) {
|
|
return sum(sha256.New(), filename)
|
|
}
|
|
|
|
// SHA1sum returns SHA1 checksum of filename
|
|
func SHA1sum(filename string) (string, error) {
|
|
return sum(sha1.New(), filename)
|
|
}
|
|
|
|
// Blake2s256 returns BLAKE2s-256 checksum of filename
|
|
func Blake2s256(filename string) (string, error) {
|
|
hash, _ := blake2s.New256([]byte{})
|
|
return sum(hash, filename)
|
|
}
|
|
|
|
// CRC32 returns CRC-32-IEEE checksum of filename
|
|
func CRC32(filename string) (string, error) {
|
|
if info, err := os.Stat(filename); err != nil || info.IsDir() {
|
|
return "", err
|
|
}
|
|
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer func() { _ = file.Close() }()
|
|
|
|
return CRCReader(bufio.NewReader(file))
|
|
}
|
|
|
|
// sum calculates the hash based on a provided hash provider
|
|
func sum(hashAlgorithm hash.Hash, filename string) (string, error) {
|
|
if info, err := os.Stat(filename); err != nil || info.IsDir() {
|
|
return "", err
|
|
}
|
|
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer func() { _ = file.Close() }()
|
|
|
|
return sumReader(hashAlgorithm, bufio.NewReader(file))
|
|
}
|