96 lines
2.5 KiB
Go
96 lines
2.5 KiB
Go
|
package securityUtil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
base64Table = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||
|
|
|
||
|
|
// const encodeStd = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
|
||
|
|
// const encodeURL = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_"
|
||
|
|
)
|
||
|
|
|
||
|
|
var coder = base64.NewEncoding(base64Table)
|
||
|
|
|
||
|
|
// 对字符串进行Base64编码
|
||
|
|
func Base64Encode(src string) string {
|
||
|
|
if src == "" {
|
||
|
|
return src
|
||
|
|
}
|
||
|
|
|
||
|
|
return base64.StdEncoding.EncodeToString([]byte(src))
|
||
|
|
}
|
||
|
|
|
||
|
|
// 对字符串进行Base64解码
|
||
|
|
func Base64Encode2(src []byte) []byte {
|
||
|
|
if len(src) == 0 {
|
||
|
|
return src
|
||
|
|
}
|
||
|
|
|
||
|
|
return []byte(base64.StdEncoding.EncodeToString(src))
|
||
|
|
}
|
||
|
|
|
||
|
|
// 对字符数组进行Base64编码
|
||
|
|
func Base64Decode(src string) (string, error) {
|
||
|
|
if src == "" {
|
||
|
|
return src, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
bytes, err := coder.DecodeString(src)
|
||
|
|
if err != nil {
|
||
|
|
return "", err
|
||
|
|
}
|
||
|
|
|
||
|
|
return string(bytes), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// 对字符数组进行Base64解码
|
||
|
|
func Base64Decode2(src []byte) ([]byte, error) {
|
||
|
|
if len(src) == 0 {
|
||
|
|
return src, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return coder.DecodeString(string(src))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEncryptAndDecrypt(t *testing.T) {
|
||
|
|
x := []byte("6d15fbbf-4913-40f5-afd1-c02effc2373a")
|
||
|
|
key := []byte("WB6aEKK5LoGpetJv")
|
||
|
|
x1, _ := AESEncrypt_CBC_Pkcs7(x, key)
|
||
|
|
x1_base64 := Base64Encode2(x1)
|
||
|
|
x1_base64_str := string(x1_base64)
|
||
|
|
fmt.Printf("Base64 of encrypted data:%s\n", x1_base64_str)
|
||
|
|
|
||
|
|
// x1_base64_str = "OoV781MTCRIEKBaWDn4NDuS3Iq1stwnORQA30Ip/eewTEzaNDQl/TgVQU09Bm7pcIP6GxGfzO7vKhRITgKCghpTi9/D+oz/GdKn8KjF/gmE="
|
||
|
|
x2_init_str, _ := Base64Decode(x1_base64_str)
|
||
|
|
x2_init := []byte(x2_init_str)
|
||
|
|
fmt.Printf("x2_init:%d\n", len(x2_init))
|
||
|
|
|
||
|
|
x2, _ := AESDecrypt_CBC_Pkcs7(x2_init, key)
|
||
|
|
fmt.Printf("Decrypted data:%s\n", string(x2))
|
||
|
|
if string(x) != string(x2) {
|
||
|
|
t.Errorf("Expected %s, but got %s", string(x), string(x2))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestDecrypt(t *testing.T) {
|
||
|
|
key := []byte("WB6aEKK5LoGpetJv")
|
||
|
|
x2_init_str, _ := Base64Decode("Oit72+aWraykW7i0e/q+zZ77w5yEU/5KuNpRYoJaxpw93i6zWKiFB8c6/PIedZxz")
|
||
|
|
x2_init := []byte(x2_init_str)
|
||
|
|
x2, _ := AESDecrypt_CBC_Pkcs7(x2_init, key)
|
||
|
|
fmt.Printf("Decrypted data:%s\n", string(x2))
|
||
|
|
|
||
|
|
x2_init_str, _ = Base64Decode("r0KHpfSmQ8jmx/FR4IJPOGBLTYF9lDRWbo9P8lIwekjkkU8BOO0QvfypgHZRIJWS")
|
||
|
|
x2_init = []byte(x2_init_str)
|
||
|
|
x2, _ = AESDecrypt_CBC_Pkcs7(x2_init, key)
|
||
|
|
fmt.Printf("Decrypted data:%s\n", string(x2))
|
||
|
|
|
||
|
|
x2_init_str, _ = Base64Decode("ev2P8zTEnebpBm43Dd8YlA==")
|
||
|
|
x2_init = []byte(x2_init_str)
|
||
|
|
x2, _ = AESDecrypt_CBC_Pkcs7(x2_init, key)
|
||
|
|
fmt.Printf("Decrypted data:%s\n", string(x2))
|
||
|
|
}
|