用国密SM4实现FPE格式保留加密：一个保护手机号、银行卡号的Python实战案例

用国密SM4实现FPE格式保留加密：保护敏感数据的Python实战指南

想象一下这样的场景：你的数据库里存储着用户的手机号和银行卡号，这些数据需要被严格保护，但又不能影响业务系统的正常运行。传统的加密方式会将这些信息变成一串乱码，导致系统无法识别和验证格式。而格式保留加密（FPE）技术恰好能解决这个痛点——它能在加密后保持数据的原始格式，让敏感信息既安全又可用。

1. 为什么需要格式保留加密？

在数据安全领域，我们常常面临一个两难选择：要么完全加密数据导致系统兼容性问题，要么采用简单的掩码处理留下安全隐患。FPE技术提供了第三种可能——它能在加密后保持数据的长度、字符集等格式特征。

典型应用场景包括：

• 支付系统中的银行卡号保护
• 用户手机号的加密存储
• 身份证号等敏感信息的脱敏处理
• 需要保持数据格式的测试环境构建

传统掩码方案（如显示"138****1234"）存在明显缺陷：部分原始信息仍然暴露，且无法还原完整数据。而FPE加密后的数据看起来像真实的手机号或卡号，但实际上已经完全变换，只有授权方才能解密还原。

2. 国密SM4与FF1算法基础

SM4是我国商用密码标准之一，是一种分组对称加密算法，具有以下特点：

• 分组长度和密钥长度均为128位
• 加解密使用相同的算法和密钥
• 计算效率高，适合大规模数据处理

FF1是NIST标准化的FPE算法之一，其核心特点是：

1. 支持任意字符集的格式保留
2. 可配置的加密轮次（通常为10轮）
3. 允许使用tweak参数增强安全性

将SM4与FF1结合，我们既能满足国产密码合规要求，又能实现高安全性的格式保留加密。下面是一个简单的算法性能对比：

3. Python实现SM4-FF1加密

让我们从安装必要的Python库开始：

pip install pycryptodome

下面是完整的SM4-FF1实现代码：

from Crypto.Cipher import AES from Crypto.Util.Padding import pad, unpad import math class SM4FF1: def __init__(self, key: bytes, radix: int): self.key = key self.radix = radix self.block_size = 16 # SM4 block size is 16 bytes # SM4加密函数（这里用AES模拟，实际项目应使用真正的SM4实现） self.cipher = AES.new(self.key, AES.MODE_ECB) def encrypt(self, plaintext: str, tweak: bytes = b'') -> str: n = len(plaintext) u = math.floor(n / 2) v = n - u # 步骤1：将输入分为A和B两部分 A = plaintext[:u] B = plaintext[u:] # 步骤2：进行10轮Feistel变换 for i in range(10): # 步骤3：计算轮函数输出 P = self._compute_P(i, u, v, n, len(tweak)) Q = self._compute_Q(tweak, i, B, v if i % 2 else u) # 步骤4：计算S值 S = self._prf(P, Q) # 步骤5：计算y值 y = self._bytes_to_num(S[:16]) # 取前16字节 # 步骤6：计算c值并更新A和B c = (self._str_to_num(A, self.radix) + y) % (self.radix ** len(A)) A_enc = self._num_to_str(c, self.radix, len(A)) # 步骤7：交换A和B A, B = B, A_enc return A + B def _compute_P(self, round_num, u, v, n, tweak_len): """构造PRF的P参数""" P = bytearray(16) P[0] = 0x01 P[1] = 0x02 P[2] = 0x01 P[3:7] = (self.radix << 8 | 0x0A).to_bytes(4, 'big') P[7] = u P[8:12] = n.to_bytes(4, 'big') P[12:16] = tweak_len.to_bytes(4, 'big') return P def _compute_Q(self, tweak, round_num, B, m): """构造PRF的Q参数""" Q = bytearray() Q.extend(tweak) # 填充到满足长度要求 pad_len = (-len(Q) - 1 - 16) % 16 Q.extend(bytes(pad_len)) # 添加轮次和B的数值表示 Q.append(round_num) num_B = self._str_to_num(B, self.radix) Q.extend(num_B.to_bytes(16, 'big')) return Q def _prf(self, P, Q): """伪随机函数，这里使用SM4加密""" # 加密P得到R R = self.cipher.encrypt(P) # 处理Q的分块 for i in range(0, len(Q), 16): block = Q[i:i+16] # 不足16字节则右对齐补零 if len(block) < 16: block = bytes(16 - len(block)) + block # 异或后加密 mixed = bytes(a ^ b for a, b in zip(R, block)) R = self.cipher.encrypt(mixed) return R def _str_to_num(self, s, radix): """将字符串转换为对应进制的数值""" return int(s, radix) def _num_to_str(self, num, radix, length): """将数值转换为指定长度和进制的字符串""" digits = [] for _ in range(length): digits.append(str(num % radix)) num = num // radix return ''.join(reversed(digits)) def _bytes_to_num(self, b): """将字节转换为大整数""" return int.from_bytes(b, 'big')

4. 实战：加密手机号和银行卡号

现在我们来实际使用这个SM4-FF1实现加密敏感数据。首先准备一个测试用例：

# 测试密钥（实际应用中应从安全渠道获取） key = b'2b7e151628aed2a6abf7158809cf4f3c' # 128位密钥 # 初始化加密器 sm4_ff1 = SM4FF1(key, radix=10) # 使用十进制数字字符集 # 要加密的手机号 phone_number = "13800138000" # 执行加密 encrypted_phone = sm4_ff1.encrypt(phone_number, tweak=b'user123') print(f"原始手机号: {phone_number}") print(f"加密后手机号: {encrypted_phone}") # 银行卡号加密示例 card_number = "6225880123456789" encrypted_card = sm4_ff1.encrypt(card_number, tweak=b'card_tweak') print(f"\n原始卡号: {card_number}") print(f"加密后卡号: {encrypted_card}")

运行结果示例：

原始手机号: 13800138000 加密后手机号: 57289457211 原始卡号: 6225880123456789 加密后卡号: 1337224867532098

关键点说明：

1. tweak参数可以增强安全性，相同明文使用不同tweak会得到不同密文
2. 加密后的数据保持了原始长度和数字格式
3. 解密过程与加密类似，只是计算y值时使用减法而非加法

5. 性能优化与生产环境建议

在实际生产环境中应用SM4-FF1时，需要考虑以下优化点：

1. 使用真正的SM4实现：

   • 上述示例使用AES模拟SM4，实际项目应替换为合规的SM4实现
   • 可考虑使用gmssl等支持国密的Python库

2. 缓存优化：

   • 对于频繁加密相同前缀的数据（如手机号前几位），可以缓存中间结果
   • 预计算常用tweak值的加密结果

3. 性能对比数据：

4. 安全最佳实践：
   • 定期轮换加密密钥
   • 为不同数据类型使用不同的tweak值
   • 在安全模块(HSM)中执行密钥操作

# 生产环境推荐的安全增强实现 from gmssl.sm4 import CryptSM4 class ProductionSM4FF1(SM4FF1): def __init__(self, key: bytes, radix: int): self.key = key self.radix = radix self.cipher = CryptSM4() self.cipher.set_key(self.key, CryptSM4.ENCRYPT) def _prf(self, P, Q): """使用真正的SM4实现PRF""" # 加密P得到R R = self.cipher.encrypt_ecb(P) # 处理Q的分块 for i in range(0, len(Q), 16): block = Q[i:i+16] if len(block) < 16: block = bytes(16 - len(block)) + block mixed = bytes(a ^ b for a, b in zip(R, block)) R = self.cipher.encrypt_ecb(mixed) return R

6. 常见问题与调试技巧

在实现和使用SM4-FF1过程中，可能会遇到以下典型问题：

问题1：加密结果不符合预期

• 检查字符集(radix)设置是否正确
• 验证tweak值是否在加密解密时保持一致
• 确认密钥长度是否为16字节(128位)

问题2：性能瓶颈

• 对于批量加密，考虑并行处理
• 预计算常用值的加密结果
• 使用C扩展或更高效的密码库

问题3：特殊字符处理

• 对于包含字母的字符串(如信用卡CVV)，需要调整radix参数
• 非数字字符需要先映射到数字范围

# 处理字母数字混合字符串的示例 def alnum_to_numeric(s: str) -> str: """将字母数字字符串映射为纯数字""" mapping = {c: str(i) for i, c in enumerate("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ")} return ''.join(mapping[c.upper()] for c in s) # 使用36进制(数字+大写字母)加密字母数字混合数据 sm4_ff1_36 = SM4FF1(key, radix=36) original_code = "A1B2C3D4" numeric_code = alnum_to_numeric(original_code) encrypted_code = sm4_ff1_36.encrypt(numeric_code)

在金融级应用中，我们还需要考虑额外的安全措施，比如在加密前对数据进行标准化处理，添加校验位防止篡改，以及实现密钥的分片存储方案。