mirror of
https://github.com/charlienet/go-mixed.git
synced 2025-07-17 08:02:40 +08:00
118 lines
2.3 KiB
Go
118 lines
2.3 KiB
Go
package hash
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/md5"
|
|
"crypto/sha1"
|
|
"crypto/sha256"
|
|
"crypto/sha512"
|
|
"errors"
|
|
"hash"
|
|
"hash/fnv"
|
|
"strings"
|
|
|
|
"github.com/cespare/xxhash/v2"
|
|
"github.com/charlienet/go-mixed/bytesconv"
|
|
"github.com/charlienet/go-mixed/crypto"
|
|
"github.com/spaolacci/murmur3"
|
|
"github.com/tjfoc/gmsm/sm3"
|
|
)
|
|
|
|
var _ crypto.Signer = &hashComparer{}
|
|
|
|
type HashFunc func([]byte) bytesconv.BytesResult
|
|
|
|
var hashFuncs = map[string]HashFunc{
|
|
"MD5": Md5,
|
|
"SHA1": Sha1,
|
|
"SHA224": Sha224,
|
|
"SHA256": Sha256,
|
|
"SHA384": Sha384,
|
|
"SHA512": Sha512,
|
|
"SM3": Sm3,
|
|
}
|
|
|
|
type hashComparer struct {
|
|
hashFunc HashFunc
|
|
}
|
|
|
|
func New(fname string) (*hashComparer, error) {
|
|
f, err := ByName(fname)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &hashComparer{
|
|
hashFunc: f,
|
|
}, nil
|
|
}
|
|
|
|
func (c *hashComparer) Sign(msg []byte) ([]byte, error) {
|
|
ret := c.hashFunc(msg)
|
|
return ret.Bytes(), nil
|
|
}
|
|
|
|
func (c *hashComparer) Verify(msg, target []byte) bool {
|
|
ret := c.hashFunc(msg)
|
|
ret.Bytes()
|
|
|
|
return bytes.Compare(ret.Bytes(), target) == 0
|
|
}
|
|
|
|
func ByName(name string) (HashFunc, error) {
|
|
if f, ok := hashFuncs[strings.ToUpper(name)]; ok {
|
|
return f, nil
|
|
}
|
|
|
|
return nil, errors.New("Unsupported hash functions")
|
|
}
|
|
|
|
func Md5(msg []byte) bytesconv.BytesResult { return sum(md5.New, msg) }
|
|
|
|
func Sha1(msg []byte) bytesconv.BytesResult { return sum(sha1.New, msg) }
|
|
|
|
func Sha224(msg []byte) bytesconv.BytesResult { return sum(sha256.New224, msg) }
|
|
|
|
func Sha256(msg []byte) bytesconv.BytesResult { return sum(sha256.New, msg) }
|
|
|
|
func Sha384(msg []byte) bytesconv.BytesResult { return sum(sha512.New384, msg) }
|
|
|
|
func Sha512(msg []byte) bytesconv.BytesResult { return sum(sha512.New, msg) }
|
|
|
|
func Sm3(msg []byte) bytesconv.BytesResult { return sum(sm3.New, msg) }
|
|
|
|
func Murmur3(msg []byte) uint64 {
|
|
return murmur3.Sum64(msg)
|
|
}
|
|
|
|
func XXhash(msg []byte) []byte {
|
|
d := xxhash.New()
|
|
_, _ = d.Write(msg)
|
|
return d.Sum(nil)
|
|
}
|
|
|
|
func XXHashUint64(msg []byte) uint64 {
|
|
h := xxhash.New()
|
|
_, _ = h.Write(msg)
|
|
return h.Sum64()
|
|
}
|
|
|
|
func Funv32(msg []byte) uint32 {
|
|
h := fnv.New32()
|
|
_, _ = h.Write(msg)
|
|
return h.Sum32()
|
|
}
|
|
|
|
func Funv64(msg []byte) uint64 {
|
|
h := fnv.New64()
|
|
_, _ = h.Write(msg)
|
|
return h.Sum64()
|
|
}
|
|
|
|
func sum(f func() hash.Hash, msg []byte) bytesconv.BytesResult {
|
|
h := f()
|
|
|
|
_, _ = h.Write(msg)
|
|
return h.Sum(nil)
|
|
}
|