116 lines
2.5 KiB
Go
116 lines
2.5 KiB
Go
package crypto
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/md5"
|
|
"crypto/rand"
|
|
"crypto/sha1"
|
|
"crypto/sha256"
|
|
"crypto/sha512"
|
|
|
|
"apigo.cc/go/encoding"
|
|
)
|
|
|
|
// MD5 返回数据的原始 MD5 字节
|
|
func MD5(data ...[]byte) []byte {
|
|
h := md5.New()
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
// MD5ToHex 返回 MD5 的 Hex 字符串
|
|
func MD5ToHex(data []byte) string {
|
|
return encoding.Hex(MD5(data))
|
|
}
|
|
|
|
// MD5ToBase64 返回 MD5 的 Base64 字符串
|
|
func MD5ToBase64(data []byte) string {
|
|
return encoding.Base64(MD5(data))
|
|
}
|
|
|
|
// MD5ToURLBase64 返回 MD5 的 URL 安全 Base64 字符串
|
|
func MD5ToURLBase64(data []byte) string {
|
|
return encoding.URLBase64(MD5(data))
|
|
}
|
|
|
|
// SHA1 系列
|
|
func SHA1(data ...[]byte) []byte {
|
|
h := sha1.New()
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func SHA1ToHex(data []byte) string { return encoding.Hex(SHA1(data)) }
|
|
func SHA1ToBase64(data []byte) string { return encoding.Base64(SHA1(data)) }
|
|
func SHA1ToURLBase64(data []byte) string { return encoding.URLBase64(SHA1(data)) }
|
|
|
|
// SHA256 系列
|
|
func SHA256(data ...[]byte) []byte {
|
|
h := sha256.New()
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func SHA256ToHex(data []byte) string { return encoding.Hex(SHA256(data)) }
|
|
func SHA256ToBase64(data []byte) string { return encoding.Base64(SHA256(data)) }
|
|
func SHA256ToURLBase64(data []byte) string { return encoding.URLBase64(SHA256(data)) }
|
|
|
|
// SHA512 系列
|
|
func SHA512(data ...[]byte) []byte {
|
|
h := sha512.New()
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func SHA512ToHex(data []byte) string { return encoding.Hex(SHA512(data)) }
|
|
func SHA512ToBase64(data []byte) string { return encoding.Base64(SHA512(data)) }
|
|
func SHA512ToURLBase64(data []byte) string { return encoding.URLBase64(SHA512(data)) }
|
|
|
|
// HMAC 系列
|
|
func HMACMD5(key []byte, data ...[]byte) []byte {
|
|
h := hmac.New(md5.New, key)
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func HMACSHA1(key []byte, data ...[]byte) []byte {
|
|
h := hmac.New(sha1.New, key)
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func HMACSHA256(key []byte, data ...[]byte) []byte {
|
|
h := hmac.New(sha256.New, key)
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
func HMACSHA512(key []byte, data ...[]byte) []byte {
|
|
h := hmac.New(sha512.New, key)
|
|
for _, v := range data {
|
|
h.Write(v)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|
|
|
|
// MakeToken 生成随机令牌
|
|
func MakeToken(size int) []byte {
|
|
token := make([]byte, size)
|
|
rand.Read(token)
|
|
return token
|
|
}
|