55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
package crypto
|
||
|
||
import (
|
||
"bytes"
|
||
"testing"
|
||
)
|
||
|
||
func TestDefaultAES(t *testing.T) {
|
||
// 1. 测试初始默认值
|
||
var confAES *Symmetric
|
||
OnSetDefaultAES(func(aes *Symmetric) {
|
||
confAES = aes
|
||
})
|
||
|
||
if confAES == nil {
|
||
t.Fatal("confAES should be initialized by OnSetDefaultAES")
|
||
}
|
||
|
||
// 2. 测试 SetDefaultAES 触发更新与锁定
|
||
rawKey := []byte("12345678901234567890123456789012")
|
||
newKey := bytes.Clone(rawKey)
|
||
newIv := []byte("123456789012")
|
||
SetDefaultAES(newKey, newIv)
|
||
|
||
// 验证密钥已被擦除 (ZeroMemory 会用随机 junk 覆盖，所以检查是否不再等于原始值)
|
||
if bytes.Equal(newKey, rawKey) {
|
||
t.Error("newKey should be overwritten after SetDefaultAES")
|
||
}
|
||
|
||
// 此时 confAES 应该已经被回调更新了
|
||
data := []byte("hello world")
|
||
encrypted, err := confAES.EncryptAndErase(bytes.Clone(data))
|
||
if err != nil {
|
||
t.Fatalf("Encrypt failed: %v", err)
|
||
}
|
||
|
||
// 3. 测试安全性：SetDefaultAES 之后不再允许 OnSetDefaultAES
|
||
var blockedAES *Symmetric
|
||
OnSetDefaultAES(func(aes *Symmetric) {
|
||
blockedAES = aes
|
||
})
|
||
if blockedAES != nil {
|
||
t.Error("OnSetDefaultAES should be blocked after SetDefaultAES (auto-lock)")
|
||
}
|
||
|
||
// 4. 测试 SetDefaultAES 仅允许一次
|
||
anotherKey := []byte("another key 32 bytes long.......")
|
||
SetDefaultAES(anotherKey, newIv)
|
||
// 验证密钥没有改变（通过解密验证）
|
||
_, err = confAES.DecryptBytes(encrypted)
|
||
if err != nil {
|
||
t.Errorf("Decryption should still work with the first injected key: %v", err)
|
||
}
|
||
}
|