在AWS API网关中,可以通过Lambda函数来自定义数据/负载加密。以下是一个Python示例代码,可以将请求体加密为JSON格式:
import json
import base64
import hashlib
from Crypto.Cipher import AES
def encrypt(event, context):
# the encryption key
key = b'mysecretpassword'
# pad the data to be encrypted
padding = b"{"
data = event['body']
padded_data = data + padding * (AES.block_size - len(data) % AES.block_size)
# encrypt the data
iv = hashlib.md5(event['headers']['User-Agent'].encode()).hexdigest()[:16].encode()
cipher = AES.new(key, AES.MODE_CBC, iv)
encrypted_data = cipher.encrypt(padded_data)
# encode the encrypted data and return the response
encoded_data = base64.b64encode(encrypted_data).decode()
return {
'statusCode': 200,
'body': json.dumps({'data': encoded_data})
}
在此示例中,使用AES-256 CBC模式进行加密。要求请求头必须包含“User-Agent”属性值,以用于生成16字节的IV(初始化向量)。加密过程中,先对要加密的数据进行填充,以确保数据块长度为AES块大小的倍数。然后使用生成的IV和密钥进行AES加密。最后,将加密后的数据进行Base64编码,以确保它可以以JSON格式进行传输。
这样,就可以在API网关中使用此Lambda函数来对所有传入请求体进行加密,确保数据的安全性。