JMeter中Groovy与Python SHA-256哈希结果不一致问题排查与解决

在将Python SHA-256哈希逻辑迁移到JMeter Groovy脚本时出现哈希值不一致的情况，经排查代码逻辑本身无错误，核心原因是输入字符串未统一执行首尾空白字符的修剪处理。以下是具体分析和修正方案：

原代码对比 #

Python 实现 #

import hashlib

def calculate_sha256(input_str):
    sha256_hash = hashlib.sha256()
    sha256_hash.update(input_str.encode('utf-8'))
    return sha256_hash.hexdigest()

# 示例调用（若实际调用时隐式执行了strip()，会与Groovy未处理的输入产生差异）
input_text = "test input".strip()
print(calculate_sha256(input_text))

原Groovy实现（未处理输入空白） #

import java.security.MessageDigest
import java.math.BigInteger

def calculateSHA256(String input) {
    MessageDigest digest = MessageDigest.getInstance("SHA-256")
    byte[] hash = digest.digest(input.getBytes("UTF-8"))
    BigInteger number = new BigInteger(1, hash)
    String hex = number.toString(16)
    while (hex.length() < 64) {
        hex = "0" + hex
    }
    return hex
}

// JMeter调用示例（输入未trim）
def input = vars.get("input_variable")
log.info(calculateSHA256(input))

问题核心 #

如果Python端调用时对输入执行了strip()（去除首尾空格、换行等空白），而Groovy端直接使用原始输入，两者的实际哈希计算内容会存在差异，最终导致哈希值不一致。

修正方案 #

1. 统一输入处理（强制trim） #

无论Python还是Groovy，都对输入字符串执行空白修剪，确保计算的原始内容完全一致。

修正后的Groovy代码（原写法优化）

import java.security.MessageDigest
import java.math.BigInteger

def calculateSHA256(String input) {
    // 处理null输入，同时修剪首尾空白
    String trimmedInput = input?.trim() ?: ""
    MessageDigest digest = MessageDigest.getInstance("SHA-256")
    byte[] hash = digest.digest(trimmedInput.getBytes("UTF-8"))
    BigInteger number = new BigInteger(1, hash)
    // 用String.format自动补全64位前导零，替代循环拼接
    return String.format("%064x", number)
}

// JMeter调用：存储结果到变量
def input = vars.get("input_variable")
vars.put("sha256_result", calculateSHA256(input))

2. Groovy简洁实现（推荐） #

利用Groovy的扩展方法简化代码，无需手动处理BigInteger和前导零：

import java.security.MessageDigest

def calculateSHA256(String input) {
    String trimmedInput = input?.trim() ?: ""
    MessageDigest digest = MessageDigest.getInstance("SHA-256")
    // Groovy内置encodeHex()方法直接生成标准64位十六进制哈希字符串
    return digest.digest(trimmedInput.bytes).encodeHex().toString()
}

// JMeter调用
def input = vars.get("input_variable")
vars.put("sha256_result", calculateSHA256(input))

内容的提问来源于stack exchange，提问作者phamthainb