python實現aes加密解密

引子：

　　　在windows中python3使用  pycryptodemo 模塊實現Aes加密解密。Aes加密有多種方式，自己可以上網了解下。 AES是每16位一加密，並且保證key和vi都是16位的。廢話不多說直接上代碼。

import base64
from Crypto.Cipher import AES
from binascii import b2a_hex,a2b_hex


class PrpCrypt(object):
    def __init__(self,key,iv):
        self.key=key.encode('utf-8')
        self.mode=AES.MODE_CBC
        self.iv=iv.encode('utf-8')

    def pad_byte(self, b):
        '''
        1 先計算所傳入bytes類型文本與16的余數
        2 在將此余數轉成bytes 當然用0補位也可以
        3 已知了 余數 那么就用余數*被轉成的余數，就得到了需要補全的bytes
        4 拼接原有文本和補位
        :param b: bytes類型的文本
        :return: 返回補全后的bytes文本
        '''
        bytes_num_to_pad = AES.block_size - (len(b) % AES.block_size)
        # python3 中默認unicode轉碼
        # 實際上byte_to_pad 就已經 將 數字轉成了unicode 對應的字符  即使你的入參正好是16的倍數，那么bytes也是把列表整體的轉碼也是有值的
        # 后邊解密的匿名函數 拿到最后一個數字后，就知道應該截取的長度，在反着切片就行了
        # 這樣保證了數據的完整性
        byte_to_pad = bytes([bytes_num_to_pad])
        padding = byte_to_pad * bytes_num_to_pad
        padded = b + padding
        return padded

    def encrypt(self,text):
        '''
        1 先生成aes實例
        2 對傳入的text轉成bytes
        3 對傳入的text補全
        4 調用encrypt 加密 得到密文
        5 先將密文轉16進制，在將16進制用base64轉碼，然后在將得到的base64解碼
        其實在步驟4 就已經完成了aes加密，我所在的公司加密比較復雜 ，需要的可以直接返回步驟4的值

        :param text:
        :return:
        '''
        cryptor = AES.new(self.key,self.mode,self.iv)
        text = text.encode('utf-8')
        text = self.pad_byte(text)
        self.ciphertext = cryptor.encrypt(text)
        cryptbase64 = base64.b64encode(b2a_hex(self.ciphertext)).decode('utf8')
        return cryptbase64


    def decrypt(self,text):
        '''
        解密和加密的順序是相反的
        1 定義匿名函數，去掉補位
        2 base64解碼
        3 生成aes實例
        4 16進制轉2進制
        5 使用decrypt解碼  得到補全的bytes類型明文

        :param text:
        :return:  解密且去掉補位的明文
        '''
        unpad = lambda s: s[:-ord(s[len(s) - 1:])]
        base64Str = base64.b64decode(text.encode('utf8'))
        cryptor = AES.new(self.key, self.mode, self.iv)
        aesStr = cryptor.decrypt(a2b_hex(base64Str))
        aesStr = str(unpad(aesStr), encoding='utf8')
        return aesStr



if __name__ == '__main__':
    data='sadfsasdgsdfgdsf123'
    pc=PrpCrypt('daajanbaqa654321','1234567890123456')
    redata = pc.encrypt(data)
    print(redata)
    result=pc.decrypt(redata)
    print(result)

　　思路：

　　　　　　這里引用一句前輩的話，思路很重要。具體的怎么實現反而在其次。

　　　　　　1 因為Aes是按照16位加密的，key和iv都好說，可以直接設置成定長的16位。

　　　　　　2 因為傳入的數據和傳回的數據不是定長的，所以加密的時候要先判斷是否是16位的倍數，如果不夠那么就補全。

　　　　　　3 問題是對端拿到數據后怎么知道從哪里截取。

　　　　　　　　1 在對傳入數據做是否16位判斷的時候，你就知道了要補多少位，將這個數字轉成bytes類型，python3中默認unicode碼。

　　　　　　　　2 對端拿到數據后，先截取最后一位，在根據unicode轉成對應數字，那么對端就知道了從哪開始截取。

　　　　　　　　3 看下unicode和utf8的關系，utf8是unicode實現的一種方式