python加密解密算法_Python-AES加密算法接口测试

前言

先前已经学过了Python-SHA256加密算法接口测试，今天我跟大家讲解一下AES加密算法的接口如何写python脚本进行测试。

一：安装pycryptodome模块

pip install pycryptodome

二：定义AES加密的方法，本次使用ECB模式加密

import base64

from Crypto.Cipher import AES

from Crypto.Util.Padding import pad, unpad

#自定义填充模式-zeropadding

#字符长度不足16(128位)

def add_to_16(value):

while len(value)%16 !=0:

value += '\0'

return value.encode()#转成字节形式

#定义加密方法

def encrypt_ecb(pt,key='hello123321olleh'):

#1初始化加密器：128位key,ECB模式加密

aes=AES.new(add_to_16(key),AES.MODE_ECB)

#2.处理数据块-128位,pkcs7模式

block=pad(pt.encode('utf8'),16)

#3.加密数据块

tmp=aes.encrypt(block)

#4.base64编码数据

ct=base64.encodebytes(tmp).decode()

return ct

三：定义解密的方法

#定义解密方法

def decrypt_ecb(ct,key='hello123321olleh'):

# 1.初始化加密器：128位key,ECB模式加密

aes = AES.new(add_to_16(key), AES.MODE_ECB)

# 2.解码base64

block=base64.decodebytes(ct.encode('utf8'))

#3.解密数据块

tmp = aes.decrypt(block)

# print(tmp)

#4.反填充数据-转成字符串

pt=unpad(tmp,16).decode('utf8')

return pt

四：代码如下

import base64

from Crypto.Cipher import AES

from Crypto.Util.Padding import pad, unpad

#自定义填充模式-zeropadding

#字符长度不足16(128位)

def add_to_16(value):

while len(value)%16 !=0:

value += '\0'

return value.encode()#转成字节形式

#定义加密方法

def encrypt_ecb(pt,key='hello123321olleh'):

#1初始化加密器：128位key,ECB模式加密

aes=AES.new(add_to_16(key),AES.MODE_ECB)

#2.处理数据块-128位,pkcs7模式

block=pad(pt.encode('utf8'),16)

#3.加密数据块

tmp=aes.encrypt(block)

#4.base64编码数据

ct=base64.encodebytes(tmp).decode()

return ct

#定义解密方法

def decrypt_ecb(ct,key='hello123321olleh'):

# 1.初始化加密器：128位key,ECB模式加密

aes = AES.new(add_to_16(key), AES.MODE_ECB)

# 2.解码base64

block=base64.decodebytes(ct.encode('utf8'))

#3.解密数据块

tmp = aes.decrypt(block)

# print(tmp)

#4.反填充数据-转成字符串

pt=unpad(tmp,16).decode('utf8')

return pt

#完成加密接口测试

if __name__ == '__main__':

import requests

payload = '''

{

"action":"add_case",

"data":{

"title": "test12345",

"summary": "test11111111111111111",

"tag": "test",

"protocol": "HTTP",

"method": "GET",

"path": "/",

"params": "test"

}

}

'''

# 加密原始报文

ct = encrypt_ecb(payload)

body = {'code': ct}

# 发送加密报文

resp = requests.post('http://localhost:9090/api/aes', data=body)

code=resp.json()['code']

print(decrypt_ecb(code))