PolarSSL/MbedTLS 中使用 PBKDF2-HMAC 密钥派生函数的注意事项

在 PolarSSL/MbedTLS 加密库中，mbedtls_pkcs5_pbkdf2_hmac 函数是一个用于实现 PBKDF2 密钥派生算法的重要函数。本文将深入分析该函数的使用方法及常见问题解决方案。

函数原型解析

mbedtls_pkcs5_pbkdf2_hmac 函数原型如下：

int mbedtls_pkcs5_pbkdf2_hmac(mbedtls_md_context_t *ctx,
                              const unsigned char *password,
                              size_t plen,
                              const unsigned char *salt,
                              size_t slen,
                              unsigned int iteration_count,
                              uint32_t key_length,
                              unsigned char *output);

该函数通过 PBKDF2 算法从密码和盐值派生加密密钥，支持多种哈希算法。

常见错误分析

开发者在使用过程中常会遇到错误代码 -20736（MBEDTLS_ERR_MD_BAD_INPUT_DATA），这通常由以下原因导致：

1. 未正确初始化哈希上下文：在使用 mbedtls_pkcs5_pbkdf2_hmac 前，必须正确初始化并配置哈希上下文。

2. 未指定哈希算法：哈希上下文必须明确指定所使用的哈希算法（如 SHA-256）。

正确使用方法

推荐使用 mbedtls_pkcs5_pbkdf2_hmac_ext 替代旧版函数，该函数直接接受哈希算法标识符，简化了使用流程：

int mbedtls_pkcs5_pbkdf2_hmac_ext(mbedtls_md_type_t md_alg,
                                  const unsigned char *password,
                                  size_t plen,
                                  const unsigned char *salt,
                                  size_t slen,
                                  unsigned int iteration_count,
                                  uint32_t key_length,
                                  unsigned char *output);

完整示例代码

以下是使用 PBKDF2-HMAC-SHA256 的正确实现：

#include "mbedtls/md.h"
#include "mbedtls/pkcs5.h"

int derive_key(const char *password, const char *salt, 
              int iterations, unsigned char *key, size_t key_len)
{
    return mbedtls_pkcs5_pbkdf2_hmac_ext(MBEDTLS_MD_SHA256,
                                        (const unsigned char *)password,
                                        strlen(password),
                                        (const unsigned char *)salt,
                                        strlen(salt),
                                        iterations,
                                        key_len,
                                        key);
}

性能优化建议

1. 迭代次数选择：PBKDF2 的安全性依赖于迭代次数，通常建议至少 10000 次，但需要平衡安全性和性能。

2. 盐值生成：盐值应使用密码学安全的随机数生成器生成，长度建议至少 16 字节。

3. 内存清理：派生密钥使用后应及时从内存中清除，防止信息泄露。

总结

正确使用 PolarSSL/MbedTLS 的 PBKDF2 函数需要注意哈希算法的选择和上下文的初始化。新版 mbedtls_pkcs5_pbkdf2_hmac_ext 函数提供了更简洁的接口，推荐开发者使用。在实际应用中，还需注意迭代次数、盐值生成等安全参数的合理配置，以确保密钥派生的安全性。