最近公司產(chǎn)品需要重構(gòu),之前的2.0版本全部重新推到重做,產(chǎn)生了現(xiàn)今全新的3.0版本。所以我之前寫的自動化腳本也全部需要重寫。3.0版本與服務(wù)端的交互全部使用了全新的加密方式(DES加密+BSAE64加密+MD5加密),公司后臺是用java編寫,我的腳本使用Python編寫,故需要用Python對加密方式封裝一下,具體如下:
Java的DES加密(java):
//DES加密
public static String des_encrypt(String data, String key) {
try {
if (StringUtils.isBlank(key)) key = CONSTANTS._DES_PWD;
byte[] bt = encrypt(data.getBytes(CONSTANTS.UTF8), key.getBytes(CONSTANTS.UTF8));
String strs = Base64.encodeBase64String(bt);
return strs;
}catch (Exception e){
e.printStackTrace();
return "";
}
}
這里的邏輯是,如果傳入的key為空,則使用默認(rèn)的key,然后調(diào)用encrypt方法進(jìn)行des加密,后再用Base64加密,就算加密成功。
encrypt方法代碼(java):
private static byte[] encrypt(byte[] data, byte[] key) throws Exception {
SecureRandom sr = new SecureRandom();
DESKeySpec dks = new DESKeySpec(key);
SecretKeyFactory keyFactory = SecretKeyFactory.getInstance("DES");
SecretKey securekey = keyFactory.generateSecret(dks);
Cipher cipher = Cipher.getInstance("DES");
cipher.init(Cipher.ENCRYPT_MODE, securekey, sr);
return cipher.doFinal(data);
}
這里要注意的是 DESKeySpec() 類,看了下代碼,其規(guī)定了秘鑰的長度只能是8個字節(jié),大于8字節(jié)則只取前8個字節(jié)
這里java里DES加密采用的是默認(rèn)模式:ECB模式,采用 PKCS5Padding 的填充模式。故沒有初始化向量(如果是其他模式,需要加上初始化向量)
知曉了Java的代碼,則把他轉(zhuǎn)成Python。
Python代碼實現(xiàn):
# -*- coding:utf-8 -*-
import base64
import hashlib
from pyDes import des, ECB, PAD_PKCS5
class Encryption:
def __init__(self, key):
if key == '':
self.key = "12345678"
else:
self.key = key
self.key = self.key[0:8] # 只取前8位
def des_encrypt(self, data):
"""
DES加密后再base64加密
:param data: 原始數(shù)據(jù)
:return: 返回des+base64加密后的字串
"""
bt = self.encrypt(data)
strs = bytes.decode(base64.b64encode(bt))
return strs
def encrypt(self, data):
"""
DES 加密
:param data: 原始字符串
:return: 返回加密后的字符(byte)
"""
k = des(self.key, ECB, padmode=PAD_PKCS5)
en = k.encrypt(data, padmode=PAD_PKCS5)
return en
這里使用了PyDes包,我直接取key的前8位(這里直接截取不知道是否有問題,但實際運行來看暫時沒碰到問題)。加密模式也是ECB模式,填充模式為PAD_PKCS5,測試一下兩邊一對比,能對上,OK!
注:兩邊的編碼需一致,這都是UTF8的編碼。另外,加密都是byte類型,需注意轉(zhuǎn)換
參考文章(感謝作者):http://blog.csdn.net/lihao21/article/details/78557461