接口测试中验签与接口参数加密

验签：

　　背景：

　　　　在实际工作中，有些敏感接口如支付接口，接口定义了一个入参为auth，在请求接口时，必须传入正确的auth，接口才会返回正确的数据，否则不处理，目的是为了加强接口的安全性

 

　　原理：

　　　　每次发送请求时都需要携带一个参数，可能是header里面传，也可能是在body里面，一般定义名为auth

　　　　根据提供的算法，生成一个签名，每次请求的时候带上这个签名，后端验证签名通过后，这个请求才会被处理，否则不处理这条请求

 

　　例子：

　　使用fastapi本地mock一个接口

import fastapi
import uvicorn
import hashlib
server = fastapi.FastAPI()

salt = 'ssz-abc'

@server.post('/api/test')
def test(name:str,addr:str,phone:str,auth:str):
    s = '%s%s%s%s'%(name,addr,phone,salt)
    s = hashlib.md5(s.encode()).hexdigest()
    print(s)
    if auth == s:
        return {"code":0,"msg":"操作成功","name":name,"addr":addr,"phone":phone,"auth":auth}
    else:
        return {"code":-1,"msg":"验签失败"}


if __name__ == '__main__':
    uvicorn.run('验签:server',port=9000,host='0.0.0.0',debug=True)

 

　　当使用postman请求时，如果auth与算法生成的结果不一致，则不会处理这条请求

 

加密：

　　背景：实际工作中，有些接口不希望被其他人读懂入参和出参报文，经常会将参数使用加密后的数据来传递入参，比如常用的AES加密，加密和解密使用不同的密钥

　　原理：每次请求传过来的都是一个加密字符串，1.先知道加密的算法是什么  2.每次发请求之前，都把参数加密

#pip install pycrypto

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

server = fastapi.FASTAPI()
 
class Prpcrypt():
    def __init__(self, key):
        '''
        :param key:security_key,一个16位的随机字符串，不要让别人知道
        '''
        self.key = key.encode()
 
    # 加密函数，如果text不是16的倍数【加密文本text必须为16的倍数！】，那就补足为16的倍数
    def encrypt(self, text):
        text = text.encode()
        cryptor = AES.new(self.key, AES.MODE_CBC, self.key)  # 生成一个加密对象
 
        # 这里密钥key 长度必须为16（AES-128）、24（AES-192）、或32（AES-256）Bytes 长度.目前AES-128足够用
        length = 16
        count = len(text)
        if count % length != 0:  # 判断加密的字符串长度是不是16的倍数
            add = length - (count % length)  # 如果不是的话，就在前面补多少个/0， 这里的算法是
            # 举例：53/16 商3 余5; 16-5 = 11, 53+11 = 64 ;，64是53上面最小的一个16的倍数
            # 53是加密内容的长度，不是16的倍数，通过 16 - (53 % 16 ) 算出需要补多少个/0，加密内容的长度才是16的倍数
        else:
            add = 0
        text = text + (b'\0' * add)
        ciphertext = cryptor.encrypt(text)
        # 因为AES加密时候得到的字符串不一定是ascii字符集的，输出到终端或者保存时候可能存在问题
        # 所以这里统一把加密后的字符串转化为16进制字符串,就是b9925b80185497ccbcc3d501604845ee这样的密文
        # b2a_hex这个函数的作用就是把bytes类型的转成16进制
        result = b2a_hex(ciphertext).decode()
        return result
 
    def decrypt(self, text):
        cryptor = AES.new(self.key, AES.MODE_CBC, self.key)  # 生成一个加密对象
        plain_text = cryptor.decrypt(a2b_hex(text))  # a2b_hex就是把16进制的转成bytes类型的
        return plain_text.decode()

@server.get('/api/test1')
    def test1('/api/test1'):
        p = Prpcrypt('1234567890123456')
        miwen = p.encrypt('{"code":1}')
        result = p.decrypt(miwen)
        return {"code":0,"msg":result}