calc_hmac代码解释-CFANZ编程社区

让我们逐步解释这段代码的含义、传入的参数以及最终的输出。

函数定义

python复制代码
 def calc_hmac(algo_name, key, data):

这定义了一个名为calc_hmac的函数，它接受三个参数：

algo_name：一个字符串，指定要使用的哈希算法的名称。在这个例子中，我们考虑了两种算法——"SHA256"和"SM3"。
key：一个字符串，用作HMAC计算的密钥。
data：一个字符串，包含要计算HMAC的数据。

函数内部逻辑

算法选择：

python复制代码
 if algo_name == "SHA256":
 
     algorithm = hashlib.sha256
 
 elif algo_name == "SM3":
 
     # 注意：Python标准库不支持SM3，这里需要替换为实际支持SM3的哈希函数
 
     raise NotImplementedError("SM3 algorithm is not supported by the standard Python library")
 
 else:
 
     raise ValueError("Unsupported algorithm: " + algo_name)

这段代码根据algo_name参数的值选择哈希算法。如果algo_name是"SHA256"，则使用Python标准库中的hashlib.sha256。如果algo_name是"SM3"，则抛出一个NotImplementedError异常，因为Python标准库不支持SM3。如果algo_name既不是"SHA256"也不是"SM3"，则抛出一个ValueError异常。

HMAC计算：

python复制代码
 hmac_obj = hmac.new(key.encode('utf-8'), digestmod=algorithm)
 
 hmac_obj.update(data.encode('utf-8'))

这里，我们使用hmac.new函数创建一个HMAC对象，传入密钥（经过UTF-8编码）和哈希算法（作为digestmod参数）。然后，我们使用update方法将数据（也经过UTF-8编码）添加到HMAC对象中。

获取HMAC值并编码：

python复制代码
 hmac_digest = hmac_obj.digest()
 
 encoded_hmac = base64.urlsafe_b64encode(hmac_digest).decode('utf-8')

使用digest方法获取HMAC的二进制表示，然后使用base64.urlsafe_b64encode函数将其编码为Base64 URL安全的字符串，最后使用decode('utf-8')将结果转换为UTF-8编码的字符串。

最终输出

python复制代码
 return encoded_hmac

函数返回计算并编码后的HMAC值。

示例用法

python复制代码
 try:
 
     print(calc_hmac("SHA256", "your_secret_key", "your_data"))
 
 except Exception as e:
 
     print(e)

这段代码尝试计算一个使用SHA256算法、密钥为"your_secret_key"、数据为"your_data"的HMAC值，并打印出来。如果发生任何异常（例如，由于算法不受支持），则捕获异常并打印错误消息。

总结

传入的参数：算法名称、密钥和数据。
最终的输出：经过计算并Base64 URL编码后的HMAC值。
用途：生成一个基于哈希的消息认证码（HMAC），用于验证数据的完整性和真实性。