Kaynağa Gözat

change base64 map

党树林 3 hafta önce
ebeveyn
işleme
0837e909d2
1 değiştirilmiş dosya ile 5 ekleme ve 2 silme
  1. 5 2
      pkg/utils/encdec.go

+ 5 - 2
pkg/utils/encdec.go

@@ -13,15 +13,18 @@ import (
 // 加密密钥(32字节,用于AES-256)
 var key = []byte("change-me-MUST-32-BIT-1234567890")
 
+// 自定义Base64编码表,这里排除了'+'和'/'
+const customBase64Table = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_"
+
 func CustomBase64Encode(src []byte) string {
 	// 创建一个使用自定义编码表的Base64编码器
-	encoder := base64.URLEncoding
+	encoder := base64.NewEncoding(customBase64Table)
 	return encoder.EncodeToString(src)
 }
 
 func CustomBase64Decode(encoded string) ([]byte, error) {
 	// 创建一个使用自定义编码表的Base64解码器
-	decoder := base64.URLEncoding
+	decoder := base64.NewEncoding(customBase64Table)
 	return decoder.DecodeString(encoded)
 }