python3 AES加密模块使用

文章是 2016-07-25 11:08:10 写的，现在python3才是用的最多的，把代码升级到python 3.x能跑起来。

AES是新的一种加密模块。在上次介绍过在C语言中如何来OpenSSL中的DES。这次我们来看看python自带的库如何来使用AES来加解密。其实二者的原理还是非常像，只是说在python中来做这个事情会比C语言要简单点，但是比起C#/JAVA还是有点点啰嗦。在C#/JAVA这种语言中，对于加密的源数据的处理，padding一般都会有完整的实现。我在上次C语言中也处理过这个问题。在python库中，也是需要自己来处理这个。

# -*- coding: utf-8 -*-
# 使用3.x版本的python
# pip3 install pycryptodome
# 在windows下，注意需要将crypto目录，小写的c修改成C，
# 否则会找不到库文件
from Crypto.Cipher import AES
import math
import base64

# padding算法，原理就是按照16字节对齐
BS = 16
pad = lambda s: s + (BS - len(s) % BS) * chr(0) 
unpad = lambda s : s[0:-ord(s[-1])]

# 解析加密的key
key=b'68b329da9893e34099c7d8ad5cb9c940'
# 16byte=2个uint32
# IV must be 16 bytes long
iv=b'68b329da9893e340'


def encrypt_test(text):
    aes_obj = AES.new(key, AES.MODE_CBC,iv)
    padding_zero=pad(text)
    e_buf=aes_obj.encrypt(str.encode(padding_zero))
    return e_buf

def decrypt_test(buff):
    aes_obj = AES.new(key, AES.MODE_CBC,iv)
    text=aes_obj.decrypt(buff).decode()
    return text

def auto_test():
    buff = encrypt_test("123")
    decode_text = decrypt_test(buff)
    print(decode_text)

if __name__ == "__main__":
    auto_test()

这个代码中padding写的还是非常漂亮的。如果在C里面实现这个，还需要啰嗦写不少。