mirror of
https://github.com/charlienet/go-mixed.git
synced 2025-07-18 00:22:41 +08:00
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package hmac
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/md5"
|
|
"crypto/sha1"
|
|
"crypto/sha256"
|
|
"crypto/sha512"
|
|
"errors"
|
|
"hash"
|
|
"strings"
|
|
|
|
"github.com/charlienet/go-mixed/bytesconv"
|
|
"github.com/tjfoc/gmsm/sm3"
|
|
)
|
|
|
|
type HMacFunc func(key, msg []byte) bytesconv.BytesResult
|
|
|
|
var hmacFuncs = map[string]HMacFunc{
|
|
"MD5": Md5,
|
|
"SHA1": Sha1,
|
|
"SHA224": Sha224,
|
|
"SHA256": Sha256,
|
|
"SHA384": Sha384,
|
|
"SHA512": Sha512,
|
|
"SM3": Sm3,
|
|
}
|
|
|
|
func ByName(name string) (HMacFunc, error) {
|
|
if f, ok := hmacFuncs[strings.ToUpper(name)]; ok {
|
|
return f, nil
|
|
}
|
|
|
|
return nil, errors.New("Unsupported hash functions")
|
|
}
|
|
|
|
func Md5(key, msg []byte) bytesconv.BytesResult { return sum(md5.New, key, msg) }
|
|
|
|
func Sha1(key, msg []byte) bytesconv.BytesResult { return sum(sha1.New, key, msg) }
|
|
|
|
func Sha224(key, msg []byte) bytesconv.BytesResult { return sum(sha256.New224, key, msg) }
|
|
|
|
func Sha256(key, msg []byte) bytesconv.BytesResult { return sum(sha256.New, key, msg) }
|
|
|
|
func Sha384(key, msg []byte) bytesconv.BytesResult { return sum(sha512.New384, key, msg) }
|
|
|
|
func Sha512(key, msg []byte) bytesconv.BytesResult { return sum(sha512.New, key, msg) }
|
|
|
|
func Sm3(key, msg []byte) bytesconv.BytesResult { return sum(sm3.New, key, msg) }
|
|
|
|
func sum(f func() hash.Hash, msg, key []byte) bytesconv.BytesResult {
|
|
h := hmac.New(f, key)
|
|
|
|
h.Write(msg)
|
|
return h.Sum(nil)
|
|
}
|