添加参数

utils/aes128.go

@@ -13,9 +13,16 @@ const (
 )
 
 //加密
-func PswEncrypt(src string)(string){
-	key := []byte(sKey)
-	iv := []byte(ivParameter)
+func PswEncrypt(src, skey, ivParam string)(string){
+	if skey == ""{
+		skey = sKey
+	}
+	if ivParam == ""{
+		ivParam = ivParameter
+	}
+
+	key := []byte(skey)
+	iv := []byte(ivParam)
 
 	result, err := Aes128Encrypt([]byte(src), key, iv)
 	if err != nil {
@@ -25,10 +32,16 @@ func PswEncrypt(src string)(string){
 }
 
 //解密
-func PswDecrypt(src string)(string) {
+func PswDecrypt(src, skey, ivParam string)(string) {
+	if skey == ""{
+		skey = sKey
+	}
+	if ivParam == ""{
+		ivParam = ivParameter
+	}
 
-	key := []byte(sKey)
-	iv := []byte(ivParameter)
+	key := []byte(skey)
+	iv := []byte(ivParam)
 
 	var result []byte
 	var err error