完成数据库敏感信息的加密

server/utils/util_test.go

@@ -1,7 +1,10 @@
 package utils_test
 
 import (
+	"crypto/md5"
 	"encoding/base64"
+	"encoding/hex"
+	"fmt"
 	"net"
 	"testing"
 
@@ -51,3 +54,26 @@ func TestAesDecryptCBC(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, "Hello Next Terminal", string(decryptCBC))
 }
+
+func TestPbkdf2(t *testing.T) {
+	pbkdf2, err := utils.Pbkdf2("1234")
+	assert.NoError(t, err)
+	println(hex.EncodeToString(pbkdf2))
+}
+
+func TestAesEncryptCBCWithAnyKey(t *testing.T) {
+	origData := []byte("admin")                                        // 待加密的数据
+	key := []byte(fmt.Sprintf("%x", md5.Sum([]byte("next-terminal")))) // 加密的密钥
+	encryptedCBC, err := utils.AesEncryptCBC(origData, key)
+	assert.NoError(t, err)
+	assert.Equal(t, "3qwawlPxghyiLS5hdr/p0g==", base64.StdEncoding.EncodeToString(encryptedCBC))
+}
+
+func TestAesDecryptCBCWithAnyKey(t *testing.T) {
+	origData, err := base64.StdEncoding.DecodeString("3qwawlPxghyiLS5hdr/p0g==") // 待解密的数据
+	assert.NoError(t, err)
+	key := []byte(fmt.Sprintf("%x", md5.Sum([]byte("next-terminal")))) // 加密的密钥
+	decryptCBC, err := utils.AesDecryptCBC(origData, key)
+	assert.NoError(t, err)
+	assert.Equal(t, "admin", string(decryptCBC))
+}