如何在JMeter中使用BeanShell Sampler编写加密Java代码？

Hey there! Let's break down how to implement encryption using the BeanShell Sampler in JMeter, plus share some handy tips for writing Java-based code here effectively.

Step 1: Add the BeanShell Sampler to Your Test Plan #

First, get the sampler set up in your test:

• Right-click on your Test Plan (or a Thread Group) → Add → Sampler → BeanShell Sampler
• Give it a meaningful name, like "MD5 Encryption Sampler" or "AES Encryption Sampler"

Step 2: Write Encryption Code (Java/BeanShell Syntax) #

BeanShell supports most Java syntax, so you can leverage standard Java encryption libraries. Below are two common examples:

Example 1: MD5 Encryption #

MD5 is a simple hashing algorithm great for non-sensitive data (note: it's not secure for confidential data anymore, but useful for checksums):

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

// Get input (can also fetch from a JMeter variable with vars.get("inputVariable"))
String input = "your-sensitive-data-here";
MessageDigest md = MessageDigest.getInstance("MD5");
byte[] digestBytes = md.digest(input.getBytes());

// Convert byte array to a hex string
BigInteger hashInt = new BigInteger(1, digestBytes);
String encryptedText = hashInt.toString(16);

// Pad with leading zeros to ensure 32-character length
while (encryptedText.length() < 32) {
    encryptedText = "0" + encryptedText;
}

// Store result in a JMeter variable for use in other samplers
vars.put("MD5EncryptedValue", encryptedText);
// Log the result for debugging
log.info("MD5 Encrypted Result: " + encryptedText);

Example 2: AES Encryption (For Confidential Data) #

AES is a secure symmetric encryption algorithm. Here's how to implement it with a 128-bit key:

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import javax.crypto.spec.IvParameterSpec;
import java.util.Base64;

try {
    String input = "your-confidential-text";
    // 16-byte key for AES-128 (use 24 for AES-192, 32 for AES-256)
    String secretKey = "my16ByteSecretKey";
    // 16-byte initialization vector (IV)
    String iv = "my16ByteIVValue";

    SecretKeySpec keySpec = new SecretKeySpec(secretKey.getBytes(), "AES");
    IvParameterSpec ivSpec = new IvParameterSpec(iv.getBytes());

    // Initialize cipher in encryption mode with CBC padding
    Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
    cipher.init(Cipher.ENCRYPT_MODE, keySpec, ivSpec);

    byte[] encryptedBytes = cipher.doFinal(input.getBytes());
    // Encode to Base64 for easy storage/transmission
    String encryptedText = Base64.getEncoder().encodeToString(encryptedBytes);

    vars.put("AESEncryptedValue", encryptedText);
    log.info("AES Encrypted Result: " + encryptedText);
} catch (Exception e) {
    log.error("AES Encryption Failed: " + e.getMessage());
    throw e; // Mark sampler as failed if encryption goes wrong
}

Pro Tips for Writing Java/BeanShell Code in JMeter #

• Always Import Required Classes: BeanShell doesn't auto-import most Java libraries, so add import statements at the top for any classes you use (like MessageDigest, Cipher, etc.).
• Leverage JMeter's Built-in Variables:
  • vars: Use this to get/set JMeter variables (e.g., vars.get("existingVariable") to fetch a value, vars.put("newVariable", value) to save results).
  • log: Write debug messages to JMeter's log file (use log.info() for general info, log.error() for errors) to troubleshoot issues.
  • props: Access global JMeter properties if you want to reuse keys or configuration across your test.
• Handle Exceptions: Wrap your code in a try-catch block to catch errors (like invalid key lengths) and log them properly—this prevents your entire sampler from failing silently.
• Test with Logs: Print intermediate values (like input, key, encrypted result) using log.info() so you can verify the encryption works as expected. Check the jmeter.log file or enable the Log Viewer in JMeter to see these logs.
• Consider JSR223 for High Load: If you're running performance tests with high throughput, switch to the JSR223 Sampler with Groovy. It's faster than BeanShell, supports modern Java features, and uses almost identical syntax.
• Reuse Code: If you need the same encryption logic across multiple samplers, write it in a separate BeanShell script file and load it with source("path/to/your-script.bsh") in the sampler.

Verify Your Encryption #

After running the test:

• Use a Debug Sampler and View Results Tree listener to check the variables you stored (e.g., ${MD5EncryptedValue}).
• For AES, you can adjust the code to use Cipher.DECRYPT_MODE to decrypt the value and confirm it matches the original input.

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