4 Commits

Author SHA1 Message Date
徐涛 be1009ef58 refactor(encryption): 优化 Spiral 加解密算法,移除不必要的 IV 生成逻辑 2026-01-27 09:31:19 +08:00
徐涛 b57b3afb2c fix(hash): 修正 BLAKE2B 哈希长度配置错误
将 blake2b.New256 替换为 blake2b.New 并传入正确的字节长度参数 28,
以确保生成 224 位的哈希值。同时清理了文件末尾多余的空行。
2025-10-09 11:21:17 +08:00
徐涛 30ddec3409 feat(hash): 添加 BLAKE3 校验和算法支持
- 在 `README.md` 中将 BLAKE3 校验和算法标记为已完成
- 引入 `lukechampine.com/blake3` 依赖以实现 BLAKE3 算法功能
- 新增 `hash/blake3` 包,提供多种 BLAKE3 哈希计算函数
  - 支持 BLAKE3、BLAKE3/224、BLAKE3/256、BLAKE3/384 等变种
  - 提供字节数组与文件的哈希计算及十六进制编码结果
- 更新 `go.mod` 和 `go.sum` 以包含新增依赖项
2025-10-07 22:38:43 +08:00
徐涛 357b19d9ad feat(hash): 添加 BLAKE2b 校验和算法支持
- 在 `hash` 模块中新增 `blake2b` 子包,实现 BLAKE2b 系列散列算法
- 支持多种输出长度:224、256、384 和 512 位
- 提供字节及十六进制字符串两种格式的散列计算函数
- 支持对字节数组和文件进行散列计算
- 更新 go.mod 使用 Go 1.24.0 并引入 golang.org/x/crypto 和 golang.org/x/sys 依赖
- 更新 README.md 标记 BLAKE2b 功能为已完成
2025-10-07 22:29:05 +08:00
6 changed files with 283 additions and 24 deletions
+3 -3
View File
@@ -22,7 +22,7 @@ Golang 中可以使用的常用辅助功能工具箱。主要配备以下功能
- [x] KeyPair 生成器
- [x] Key 导入与导出
- [x] RSA 签名算法
- 散列及校验和算法
- 散列及校验和算法
- [x] Sha512 散列算法(便捷封装)
- [x] Sha256 散列算法(便捷封装)
- [x] Sha1 散列算法(便捷封装)
@@ -32,8 +32,8 @@ Golang 中可以使用的常用辅助功能工具箱。主要配备以下功能
- [x] CRC32 校验和算法(便捷封装)
- [x] CRC64 校验和算法(便捷封装)
- [x] pHash 图像感知算法(便捷封装)
- [ ] BLAKE2 校验和算法(便捷封装)
- [ ] BLAKE3 校验和算法(便捷封装)
- [x] BLAKE2b 校验和算法(便捷封装)
- [x] BLAKE3 校验和算法(便捷封装)
- [ ] BlockHash 散列算法
- 唯一序列号生成器
- [x] 冰雹 ID 生成器(短主机精简日期版雪花 ID)
+38 -20
View File
@@ -2,12 +2,13 @@
package spiral
import (
"crypto/aes"
"crypto/cipher"
"errors"
"fmt"
"strings"
"archgrid.xyz/ag/toolsbox/encryption"
"archgrid.xyz/ag/toolsbox/encryption/aes"
"archgrid.xyz/ag/toolsbox/hash/sha512"
verifyCode "archgrid.xyz/ag/toolsbox/random/verify_code"
"archgrid.xyz/ag/toolsbox/serialize/base64"
@@ -21,25 +22,34 @@ const (
)
// 根据给定的密钥字符串生成加解密使用的密钥。
// 与Rust版本兼容:使用SHA512 hex字符串的字节表示。
func generateKey(key string) []byte {
keyBytes := sha512.Sha512([]byte(key))
return keyBytes[4:36]
hexStr := sha512.Sha512Hex([]byte(key))
// 取hex字符串的第4-36字节(对应Rust版本)
return []byte(hexStr[4:36])
}
// 对给定的数据进行加密。
func Encrypt(data string, strength ...Strength) (string, error) {
var ivGen aes.IVGenerator
if append(strength, Enhanced)[0] == Compatible {
ivGen = aes.PrefixIVGenerator
} else {
ivGen = aes.XorIVGenerator
}
key := verifyCode.RandStr(20)
keyBytes := generateKey(key)
cipherData, err := aes.Encrypt([]byte(data), keyBytes, encryption.PKCS7Padding, ivGen)
// 直接使用crypto/aes,避免二次SHA256哈希
block, err := aes.NewCipher(keyBytes)
if err != nil {
return "", fmt.Errorf("加密计算失败,%w", err)
return "", fmt.Errorf("创建加密单元失败,%w", err)
}
// 使用key的前16字节作为IV(与Rust版本PrefixIVGenerator对应)
iv := keyBytes[:16]
// PKCS7 padding
plainText := encryption.Padding([]byte(data), block.BlockSize(), encryption.PKCS7Padding)
cipherData := make([]byte, len(plainText))
mode := cipher.NewCBCEncrypter(block, iv)
mode.CryptBlocks(cipherData, plainText)
var result strings.Builder
result.WriteString("[")
result.WriteString(key)
@@ -49,24 +59,32 @@ func Encrypt(data string, strength ...Strength) (string, error) {
// 对给定的数据进行解密。
func Decrypt(data string, strength ...Strength) (string, error) {
var ivGen aes.IVGenerator
if append(strength, Enhanced)[0] == Compatible {
ivGen = aes.PrefixIVGenerator
} else {
ivGen = aes.XorIVGenerator
}
if message, found := strings.CutPrefix(data, "["); found {
if len(message) > 20 {
keySeed := message[:20]
key := generateKey(keySeed)
keyBytes := generateKey(keySeed)
cipherData, err := base64.FromBase64(message[20:])
if err != nil {
return "", fmt.Errorf("密文损坏无法解析,%w", err)
}
plainText, err := aes.Decrypt(cipherData, key, encryption.PKCS7Padding, ivGen)
// 直接使用crypto/aes,避免二次SHA256哈希
block, err := aes.NewCipher(keyBytes)
if err != nil {
return "", fmt.Errorf("密文解密计算失败,%w", err)
return "", fmt.Errorf("创建加密单元失败,%w", err)
}
// 使用key的前16字节作为IV(与Rust版本对应)
iv := keyBytes[:16]
plainText := make([]byte, len(cipherData))
mode := cipher.NewCBCDecrypter(block, iv)
mode.CryptBlocks(plainText, cipherData)
// PKCS7 unpadding
plainText = encryption.Unpadding(plainText, encryption.PKCS7Padding)
return string(plainText), nil
}
return "", errors.New("密文缺损,无法完成解密。")
+5 -1
View File
@@ -1,13 +1,15 @@
module archgrid.xyz/ag/toolsbox
go 1.20
go 1.24.0
require go.uber.org/zap v1.24.0
require (
github.com/azr/gift v1.1.2 // indirect
github.com/disintegration/imaging v1.6.2 // indirect
github.com/klauspost/cpuid/v2 v2.0.9 // indirect
golang.org/x/image v0.0.0-20220722155232-062f8c9fd539 // indirect
golang.org/x/sys v0.36.0 // indirect
)
require (
@@ -17,4 +19,6 @@ require (
github.com/sigurn/crc8 v0.0.0-20220107193325-2243fe600f9f
go.uber.org/atomic v1.7.0 // indirect
go.uber.org/multierr v1.6.0 // indirect
golang.org/x/crypto v0.42.0
lukechampine.com/blake3 v1.4.1
)
+8
View File
@@ -10,6 +10,8 @@ github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/howeyc/crc16 v0.0.0-20171223171357-2b2a61e366a6 h1:IIVxLyDUYErC950b8kecjoqDet8P5S4lcVRUOM6rdkU=
github.com/howeyc/crc16 v0.0.0-20171223171357-2b2a61e366a6/go.mod h1:JslaLRrzGsOKJgFEPBP65Whn+rdwDQSk0I0MCRFe2Zw=
github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4=
github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/sigurn/crc8 v0.0.0-20220107193325-2243fe600f9f h1:1R9KdKjCNSd7F8iGTxIpoID9prlYH8nuNYKt0XvweHA=
github.com/sigurn/crc8 v0.0.0-20220107193325-2243fe600f9f/go.mod h1:vQhwQ4meQEDfahT5kd61wLAF5AAeh5ZPLVI4JJ/tYo8=
@@ -21,9 +23,15 @@ go.uber.org/multierr v1.6.0 h1:y6IPFStTAIT5Ytl7/XYmHvzXQ7S3g/IeZW9hyZ5thw4=
go.uber.org/multierr v1.6.0/go.mod h1:cdWPpRnG4AhwMwsgIHip0KRBQjJy5kYEpYjJxpXp9iU=
go.uber.org/zap v1.24.0 h1:FiJd5l1UOLj0wCgbSE0rwwXHzEdAZS6hiiSnxJN/D60=
go.uber.org/zap v1.24.0/go.mod h1:2kMP+WWQ8aoFoedH3T2sq6iJ2yDWpHbP0f6MQbS9Gkg=
golang.org/x/crypto v0.42.0 h1:chiH31gIWm57EkTXpwnqf8qeuMUi0yekh6mT2AvFlqI=
golang.org/x/crypto v0.42.0/go.mod h1:4+rDnOTJhQCx2q7/j6rAN5XDw8kPjeaXEUR2eL94ix8=
golang.org/x/image v0.0.0-20191009234506-e7c1f5e7dbb8/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0=
golang.org/x/image v0.0.0-20220722155232-062f8c9fd539 h1:/eM0PCrQI2xd471rI+snWuu251/+/jpBpZqir2mPdnU=
golang.org/x/image v0.0.0-20220722155232-062f8c9fd539/go.mod h1:doUCurBvlfPMKfmIpRIywoHmhN3VyhnoFDbvIEWF4hY=
golang.org/x/sys v0.36.0 h1:KVRy2GtZBrk1cBYA7MKu5bEZFxQk4NIDV6RLVcC8o0k=
golang.org/x/sys v0.36.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
lukechampine.com/blake3 v1.4.1 h1:I3Smz7gso8w4/TunLKec6K2fn+kyKtDxr/xcQEN84Wg=
lukechampine.com/blake3 v1.4.1/go.mod h1:QFosUxmjB8mnrWFSNwKmvxHpfY72bmD2tQ0kBMM3kwo=
+117
View File
@@ -0,0 +1,117 @@
// 提供Blake2B系列散列算法函数和校验和函数。
package blake2b
import (
"encoding/hex"
"fmt"
"hash"
"io"
"os"
"golang.org/x/crypto/blake2b"
)
// 根据给定的位数返回一个散列算法的Hash实例。
func hasherSelect(bitSize int) hash.Hash {
switch bitSize {
case 224:
hasher, _ := blake2b.New(28, nil)
return hasher
case 256:
hasher, _ := blake2b.New256(nil)
return hasher
case 384:
hasher, _ := blake2b.New384(nil)
return hasher
case 512:
hasher, _ := blake2b.New512(nil)
return hasher
default:
hasher, _ := blake2b.New256(nil)
return hasher
}
}
// 计算给定字节数组的Blake2B校验和,返回字节数组。
func Blake2b(data []byte) []byte {
hasher := hasherSelect(512)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake2B/256校验和,返回字节数组。
func Blake2b_256(data []byte) []byte {
hasher := hasherSelect(256)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake2B/384校验和,返回字节数组。
func Blake2b_384(data []byte) []byte {
hasher := hasherSelect(384)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake2B/224校验和,返回字节数组。
func Blake2b_224(data []byte) []byte {
hasher := hasherSelect(224)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake2B校验和,返回十六进制字符串。
func Blake2bHex(data []byte) string {
return hex.EncodeToString(Blake2b(data))
}
// 计算给定字节数组的Blake2B/256校验和,返回十六进制字符串。
func Blake2b_256Hex(data []byte) string {
return hex.EncodeToString(Blake2b_256(data))
}
// 计算给定字节数组的Blake2B/384校验和,返回十六进制字符串。
func Blake2b_384Hex(data []byte) string {
return hex.EncodeToString(Blake2b_384(data))
}
// 计算给定字节数组的Blake2B/224校验和,返回十六进制字符串。
func Blake2b_224Hex(data []byte) string {
return hex.EncodeToString(Blake2b_224(data))
}
// 根据给定位数计算一个字节数组的Blake2B校验和,返回字节数组。
func Sum(data []byte, bitSize ...int) []byte {
hasher := hasherSelect(append(bitSize, 512)[0])
hasher.Write(data)
return hasher.Sum(nil)
}
// 根据给定位数计算一个字节数组的Blake2B校验和,返回十六进制字符串。
func SumHex(data []byte, bitSize ...int) string {
return hex.EncodeToString(Sum(data, bitSize...))
}
// 根据给定位数计算一个文件的Blake2B校验和,返回字节数组。
func SumFile(file string, bitSize ...int) ([]byte, error) {
f, err := os.Open(file)
if err != nil {
return nil, fmt.Errorf("未能打开指定文件,%w", err)
}
defer f.Close()
hasher := hasherSelect(append(bitSize, 512)[0])
if _, err := io.Copy(hasher, f); err != nil {
return nil, fmt.Errorf("未能读取指定文件的内容,%w", err)
}
return hasher.Sum(nil), nil
}
// 根据给定位数计算一个文件的Blake2B校验和,返回十六进制字符串。
func SumFileHex(file string, bitSize ...int) (string, error) {
hash, err := SumFile(file, bitSize...)
if err != nil {
return "", err
}
return hex.EncodeToString(hash), nil
}
+112
View File
@@ -0,0 +1,112 @@
// 提供Blake3系列散列算法函数和校验和函数。
package blake3
import (
"encoding/hex"
"fmt"
"hash"
"io"
"os"
"lukechampine.com/blake3"
)
// 根据给定的位数返回一个散列算法的Hash实例。
func hasherSelect(bitSize int) hash.Hash {
switch bitSize {
case 224:
return blake3.New(28, nil)
case 256:
return blake3.New(32, nil)
case 384:
return blake3.New(48, nil)
case 512:
return blake3.New(64, nil)
default:
return blake3.New(32, nil)
}
}
// 计算给定字节数组的Blake3校验和,返回字节数组。
func Blake3(data []byte) []byte {
hasher := hasherSelect(512)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake3/256校验和,返回字节数组。
func Blake3_256(data []byte) []byte {
hasher := hasherSelect(256)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake3/384校验和,返回字节数组。
func Blake3_384(data []byte) []byte {
hasher := hasherSelect(384)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake3/224校验和,返回字节数组。
func Blake3_224(data []byte) []byte {
hasher := hasherSelect(224)
hasher.Write(data)
return hasher.Sum(nil)
}
// 计算给定字节数组的Blake3校验和,返回十六进制字符串。
func Blake3Hex(data []byte) string {
return hex.EncodeToString(Blake3(data))
}
// 计算给定字节数组的Blake3/256校验和,返回十六进制字符串。
func Blake3_256Hex(data []byte) string {
return hex.EncodeToString(Blake3_256(data))
}
// 计算给定字节数组的Blake3/384校验和,返回十六进制字符串。
func Blake3_384Hex(data []byte) string {
return hex.EncodeToString(Blake3_384(data))
}
// 计算给定字节数组的Blake3/224校验和,返回十六进制字符串。
func Blake3_224Hex(data []byte) string {
return hex.EncodeToString(Blake3_224(data))
}
// 根据给定位数计算一个字节数组的Blake3校验和,返回字节数组。
func Sum(data []byte, bitSize ...int) []byte {
hasher := hasherSelect(append(bitSize, 512)[0])
hasher.Write(data)
return hasher.Sum(nil)
}
// 根据给定位数计算一个字节数组的Blake3校验和,返回十六进制字符串。
func SumHex(data []byte, bitSize ...int) string {
return hex.EncodeToString(Sum(data, bitSize...))
}
// 根据给定位数计算一个文件的Blake3校验和,返回字节数组。
func SumFile(file string, bitSize ...int) ([]byte, error) {
f, err := os.Open(file)
if err != nil {
return nil, fmt.Errorf("未能打开指定文件,%w", err)
}
defer f.Close()
hasher := hasherSelect(append(bitSize, 512)[0])
if _, err := io.Copy(hasher, f); err != nil {
return nil, fmt.Errorf("未能读取指定文件的内容,%w", err)
}
return hasher.Sum(nil), nil
}
// 根据给定位数计算一个文件的Blake3校验和,返回十六进制字符串。
func SumFileHex(file string, bitSize ...int) (string, error) {
hash, err := SumFile(file, bitSize...)
if err != nil {
return "", err
}
return hex.EncodeToString(hash), nil
}