httprunner学习23-加解密
前言
有些接口的请求参数是加密的，返回的接口内容也是需要解密才能看到。
加密接口
比如当我们访问下面这个登陆的接口时，请求参数账号和密码都是需要加密，通过parms参数传过去，服务器才能识别到
没加密的时候，请求参数应该是
{
"params": {
"username": "test",
"password": "123456"
}
}
返回的结果，解密后应该是
{
"code": 0,
"msg": "login success!",
"datas": {
"username": "test",
"token": "a003442ffc9645af181d8c768bd8758a250ba6d6"
}
}
像这种接口，做接口自动化的时候，请求参数应该用未加密之前的测试参数，这样方便维护和修改，在发请求的过程中先对请求参数加密。
具体的加密规则和方法，得看开发的加密规则了，比如参考这篇的加密https://www.cnblogs.com/yoyoketang/p/11717282.html
hrun脚本编写
写脚本的时候，yaml文件写未加密的时候，原始的测试数据
- config:
name: testcase description
variables: {}
- test:
name: /api/v2/login
request:
headers:
Content-Type: application/json
User-Agent: Fiddler
json:
params:
password: '123456'
username: test
method: POST
url: http://49.235.92.12:9000/api/v2/login
setup_hooks:
- ${setup_request($request)}
validate:
- eq:
- status_code
- 200
- eq:
- headers.Content-Type
- application/json
- eq:
- content.code
- 0
- eq:
- content.msg
- login success!
接下来只需要把 params 下的参数做加密就可以了，这里用到setup_hooks函数实现, ${setup_request($request)}函数在debugtalk.py里面去写一个
setup_hooks
setup_hooks用于在 HTTP 请求发送前执行 hook 函数，主要用于准备工作；也可以实现对请求的 request 内容进行预处理。
以下是官方文档给的案例
def setup_hook_prepare_kwargs(request):
if request["method"] == "POST":
content_type = request.get("headers", {}).get("content-type")
if content_type and "data" in request:
# if request content-type is application/json, request data should be dumped
if content_type.startswith("application/json") and isinstance(request["data"], (dict, list)):
request["data"] = json.dumps(request["data"])
if isinstance(request["data"], str):
request["data"] = request["data"].encode('utf-8')
可以依着葫芦画瓢
from cryptography.hazmat.primitives import padding
from cryptography.hazmat.primitives.ciphers import algorithms
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
import json
'''
AES/CBC/PKCS7Padding 加密解密
环境需求:
pip3 install pycryptodome
'''
class PrpCrypt(object):
def __init__(self, key='1234123412341234'):
self.key = key.encode('utf-8')
self.mode = AES.MODE_CBC
self.iv = b'0102030405060708'
# block_size 128位
# 加密函数，如果text不足16位就用空格补足为16位，
# 如果大于16当时不是16的倍数，那就补足为16的倍数。
def encrypt(self, text):
cryptor = AES.new(self.key, self.mode, self.iv)
text = text.encode('utf-8')
# 这里密钥key 长度必须为16（AES-128）,
# 24（AES-192）,或者32 （AES-256）Bytes 长度
# 目前AES-128 足够目前使用
text=self.pkcs7_padding(text)
self.ciphertext = cryptor.encrypt(text)
# 因为AES加密时候得到的字符串不一定是ascii字符集的，输出到终端或者保存时候可能存在问题
# 所以这里统一把加密后的字符串转化为16进制字符串
return b2a_hex(self.ciphertext).decode().upper()
@staticmethod
def pkcs7_padding(data):
if not isinstance(data, bytes):
data = data.encode()
padder = padding.PKCS7(algorithms.AES.block_size).padder()
padded_data = padder.update(data) + padder.finalize()
return padded_data
@staticmethod
def pkcs7_unpadding(padded_data):
unpadder = padding.PKCS7(algorithms.AES.block_size).unpadder()
data = unpadder.update(padded_data)
try:
uppadded_data = data + unpadder.finalize()
except ValueError:
raise Exception('无效的加密信息!')
else:
return uppadded_data
# 解密后，去掉补足的空格用strip() 去掉
def decrypt(self, text):
# 偏移量'iv'
cryptor = AES.new(self.key, self.mode, self.iv)
plain_text = cryptor.decrypt(a2b_hex(text))
# return plain_text.rstrip('\0')
return bytes.decode(plain_text).rstrip("\x01").\
rstrip("\x02").rstrip("\x03").rstrip("\x04").rstrip("\x05").\
rstrip("\x06").rstrip("\x07").rstrip("\x08").rstrip("\x09").\
rstrip("\x0a").rstrip("\x0b").rstrip("\x0c").rstrip("\x0d").\
rstrip("\x0e").rstrip("\x0f").rstrip("\x10")
def dict_json(self, d):
'''python字典转json字符串, 去掉一些空格'''
j = json.dumps(d).replace('": ', '":').replace(', "', ',"').replace(", {", ",{")
return j
def setup_request(request):
pc = PrpCrypt('12345678\0\0\0\0\0\0\0\0') # 初始化密钥
params = request.get("json").get("params")
print("加密前：%s" % params)
# 对params加密
en_params = pc.encrypt(json.dumps(params))
print("解密后：%s" % en_params)
request["json"]["params"] = en_params
运行用例
运行用例结构
D:\soft\HELL\DEMO>hrun login_decrype.yml
/api/v2/login
加密前：{'password': '123456', 'username': 'test'}
解密后：3DBF6BE0F8549A98AA81B629A028E487BA2EBE85EF1BC9FE7105F5FE833F1DF26F022E404EFBDAD5A5DF1A7B7FEDA16C
INFO POST http://49.235.92.12:9000/api/v2/login
INFO status_code: 200, response_time(ms): 467.89 ms, response_length: 209 bytes
INFO start to validate.
.
----------------------------------------------------------------------
Ran 1 test in 0.574s
OK
INFO Start to render Html report ...
INFO Generated Html report: D:\soft\HELL\DEMO\reports\1571850793.html