接口测试之AES数据加密

旧城冷巷雨未停 提交于 2020-01-15 12:28:42

在接口测试中,会遇到加密的请求数据,例如:常用的base64加密,AES加密,在这里,简述用Python转化AES的加密方法

原理

  • 官网链接:https://pycryptodome.readthedocs.io/en/latest/src/cipher/aes.html
  • 在线加密/解密:https://www.sojson.com/encrypt_aes.html
  • AES加密主要包括两个步骤:密钥扩展和明文加密。
    密钥扩展:将输入的密钥(16字节、24字节和32字节)进行扩展,根据密钥长度的不同,得到扩展后的密钥进行加密的轮数也不相同,个人理解为补码。
  •   例如:对用户名进行AES加密,6位的用户名不满足16个字节,就需要补充位数。
    


Python实现:Crypto算法库

算法库详解: https://segmentfault.com/a/1190000016851912

安装
Crypto不是自带的模块,需要下载。http://www.voidspace.org.uk/python/modules.shtml#pycrypto

安装好引用的时候,提示找不到Crypto,找了很多资料,原因是
C:\Python27\Lib\site-packages在这个路径下面有一个文件夹叫做crypto,把它的首字母改成大写,即是Crypto
就没有问题了

简单使用

from Crypto.Cipher import AES 
import base64 
secret = "12345678912345678912345678912345"   #由用户输入的16位或24位或32位长的初始密码字符串 
cipher = AES.new(secret)            #通过AES处理初始密码字符串,并返回cipher对象 
s = cipher.encrypt("1234567891234567")     #输入需要加密的字符串,注意字符串长度要是16的倍数。16,32,48.. 
print s                     #输出加密后的字符串 
print base64.b64encode(s)            #输出加密后的字符串的base64编码。 
print cipher.decrypt(s)             #解密

接口自动化中实现

class AesMethod:
    def __init__(self):
        self.key=key
    def pkcs7padding(self,text):
        """
        明文使用PKCS7填充,如果块长度是16,数据长度9,那么还差7个字节,就在后面补充7个0x07
        数据:  FF FF FF FF FF FF FF FF FF
        填充后:FF FF FF FF FF FF FF FF FF 07 07 07 07 07 07 07
        最终调用AES加密方法时,传入的是一个byte数组,要求是16的整数倍,因此需要对明文进行处理
        :param text: 待加密内容(明文)
        :return:填充后的数据
        """
        bs = AES.block_size  # 16
        length = len(text)
        bytes_length = len(bytes(text, encoding='utf-8'))
        # tips:utf-8编码时,英文占1个byte,而中文占3个byte
        padding_size = length if(bytes_length == length) else bytes_length
        padding = bs - padding_size % bs
        # tips:chr(padding)看与其它语言的约定,有的会使用'\0'
        padding_text = chr(padding) * padding
        return text + padding_text

    def aes_encrypt(self, data):
        key_bytes=bytes(self.key, encoding='utf-8')
        cipher = AES.new(key_bytes,mode=1)
        # 处理明文
        content_padding = self.pkcs7padding(data)
        # 加密
        encrypt_bytes = cipher.encrypt(bytes(content_padding, encoding='utf-8'))
        # 重新编码
        result = str(base64.b64encode(encrypt_bytes), encoding='utf-8')
        return result

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!