JMeter 实现 Protobuf 加密解密

一、 .proto文件编译成.jar文件

相关依赖下载详见：将 message.proto 编译成 .jar文件

1.依赖于java编译环境
2.依赖protoc编译jar包

编译目录

1.创建一个根目录：protobuf
2.在protobuf下创建build、output、lib、src目录

lib：放 protobuf-java-4.31.1.jar
src：放 xxx.proto

准备.proto 文件

将准备好的.protp 文件放入src目录下

syntax = "proto3"; 

option java_package = "com.message.proto";
option java_outer_classname = "MessageProBuf";

package Message;

//消息的头信息
message MessageHead
{

	 string from = 1;

	 string to = 2;

	 string messageId = 3;

	 int32 chatType = 4;
	
}



//消息的 body 
message ChatMessage
{
	 MessageHead messageHead=1;

	 string fromUserId = 2;

	 string toUserId= 3;

	 int64 timeSend= 4;

	 int32 type= 5;
	
	 string content=6;
}

将.proto文件编译.java 文件

执行命令：

cd .\src\
protoc --java_out=../output message.proto

命令执行完成后，根据.proto 文件中的 java_package、java_outer_classname 生成对应的包目录和文件名，示例如下：
在这里插入图片描述

使用javac命令将MessageProBuf.java编译成.class文件

这里依赖protobuf-java-4.31.1.jar 库，所以需要根目录下创建一个lib目录，并将protobuf-java-4.31.1.jar文件放到lib目录下

cd ..\lib\
javac -encoding UTF-8 -cp ".;protobuf-java-4.31.1.jar" -d ..\build ..\output\com\message\proto\MessageProBuf.java 

使用 jar 命令将所有 .class 文件打包成 .jar 文件

cd ..\build\
jar cf message-protobuf.jar com/

示例完整命令汇总

:: .proto文件编译为.java文件
cd .\src\
protoc --java_out=../output message.proto

:: .java文件编译为.class文件
cd ..\lib\
javac -encoding UTF-8 -cp ".;protobuf-java-4.31.1.jar" -d ..\build ..\output\com\message\proto\MessageProBuf.java 

:: .class文件编译为.jar包
cd ..\build\
jar cf message-protobuf.jar com/

二、编写 .groovy文件编译成.jar文件

相关依赖下载详见：JMeter groovy 编译成.jar 文件

编译目录

1.创建一个根目录：groovy
2.在protobuf下创建build、lib、src目录

lib：放 message-protobuf.jar、protobuf-java-4.31.1.jar 等jar包
src：放 .groovy 文件

准备 MessageBuilder.groovy （消息构建文件）

// 导入 .proto 文件中的包名和类名
import com.message.proto.MessageProBuf
java_package ：com.message.proto
java_outer_classname：MessageProBuf

import com.message.proto.MessageProBuf  // .proto 中的java_package、java_outer_classname
import com.google.protobuf.MessageLite

/**
 * 消息构建工具类
 */
class MessageBuilder {

    /**
     * 构建单聊消息
     * @param fromUserId 发送者ID
     * @param fromUserName 发送者名称
     * @param toUserId 接收者ID
     * @param toUserName 接收者名称
     * @param content 消息内容
     * @param messageId 消息ID（可选）
     * @return 构建完成的 MessageLite 对象
     */
    static MessageLite buildChatMessage(String fromUserId, String fromUserName,
                                        String toUserId, String toUserName,
                                        String content, String messageId = null) {
        if (messageId == null) {
            messageId = UUID.randomUUID().toString()
        }

        return MessageProBuf.ChatMessage.newBuilder()
            .setMessageHead(MessageProBuf.MessageHead.newBuilder()
                .setFrom("${fromUserId}/pc")
                .setTo(toUserId)
                .setMessageId(messageId)
                .setChatType(1)) // 单聊
            .setFromUserId(fromUserId)
            .setFromUserName(fromUserName)
            .setToUserId(toUserId)
            .setToUserName(toUserName)
            .setContent(content)
            .setType(1) // 文本消息
            .setTimeSend(System.currentTimeMillis())
            .setIsReadDel(false)
            .setIsEncrypt(false)
            .setSeqNo(-1)
            .build()
    }

   // ...  构建其他消息类型省略
}

准备 ProtobufParser.groovy （加密解密文件）

import com.message.proto.MessageProBuf
import com.google.protobuf.InvalidProtocolBufferException
import com.google.protobuf.MessageLite
import java.util.function.Function
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Logger


class ProtobufParser {
    private static final Logger log = LogManager.getLogger(ProtobufParser::class)


    // 支持扩展的命令解析映射
    private static final Map<Integer, Function<byte[], MessageLite>> MESSAGE_PARSERS = new HashMap<>()

    static {
        MESSAGE_PARSERS.put(10, { data -> MessageProBuf.ChatMessage.parseFrom(data) });
        // 更多指令码和解析函数的映射可以添加到这里

    }

    /**
     * 构建加密后的消息字节数组
     * @param command 指令码
     * @param message Protobuf 消息体
     * @return 拼接后的字节数组
     */
    static byte[] encryptMessage(int command, MessageLite message) {
        if (message == null) {
            throw new IllegalArgumentException("Message cannot be null")
        }

        byte[] buffer = message.toByteArray()
        byte[] bytes = new byte[buffer.length + 1]
        bytes[0] = (byte) command
        System.arraycopy(buffer, 0, bytes, 1, buffer.length)
        return bytes
    }

    /**
     * 解析响应数据中的 Protobuf 消息
     * @param responseData 响应字节数组数据
     * @return 解析后的 MessageLite 对象（可扩展）
     */
    static Optional<MessageLite> parseProtobufMessage(byte[] responseData) {
        if (responseData == null || responseData.length < 1) {
            throw new IllegalArgumentException("Response data is null or empty")
        }

        int command = responseData[0] & 0xFF

        byte[] messageData = Arrays.copyOfRange(responseData, 1, responseData.length)

        try {
            Function<byte[], MessageLite> parser = MESSAGE_PARSERS.get(command)
            if (parser != null) {
                MessageLite message = parser.apply(messageData)
                log.info("Decoded message: ${message.toString()}")
                return Optional.of(message)
            } else {
                log.warn("Unknown command: $command")
                return Optional.empty()
            }
        } catch (InvalidProtocolBufferException e) {
            log.error("Failed to parse Protobuf message", e)
            throw new RuntimeException("Failed to parse Protobuf message with command: $command", e)
        }
    }
}

编译 Groovy 文件为 .class

1、将MessageBuilder.groovy 构件成 MessageBuilder.class 文件到指定的build/message 目录下
2、将ProtobufParser.groovy 构件成 ProtobufParser.class 文件到指定的build/protobuf 目录下

:: 切换到目录
cd groovy

groovyc -cp "lib/*" src/MessageBuilder.groovy -d build/message

groovyc -cp "lib/*" src/ProtobufParser.groovy -d build/protobuf

将目录build/message 、build/protobuf 目录下所有的.class 文件分别构建成 .jar文件

:: 切换到目录
cd groovy

jar cf MessageBuilder.jar -C build/message .

jar cf ProtobufParser.jar -C build/protobuf .

示例完整命令汇总

PS D:\System\Desktop\csdn> cd groovy
PS D:\System\Desktop\csdn\groovy> groovyc -cp "lib/*" src/MessageBuilder.groovy -d build/message
PS D:\System\Desktop\csdn\groovy> groovyc -cp "lib/*" src/ProtobufParser.groovy -d build/protobuf
PS D:\System\Desktop\csdn\groovy> jar cf MessageBuilder.jar -C build/message .
PS D:\System\Desktop\csdn\groovy> jar cf ProtobufParser.jar -C build/protobuf .
PS D:\System\Desktop\csdn\groovy>

三、JMeter 使用加解密

将 protobuf-java-4.31.1.jar 文件复制到 jmeter lib 目录下

将message-protobuf.jar、MessageBuilder.jar、ProtobufParser.jar 三个文件复制到lib/ext 目录下

四、重启JMeter

五、在JMeter 中使用

在JMeter 中添加线程组

在线程组上添加取样器/JSR223 Sampler

在JSR223 Sampler 中的script 脚本添加如下代码

import com.message.proto.MessageProBuf;
import com.google.protobuf.MessageLite;
import org.apache.logging.log4j.LogManager
import org.apache.logging.log4j.Logger

// 创建日志记录器
Logger logger = LogManager.getLogger("test-protpbuf-jsr223-sampler")


// 单聊消息
MessageLite chatMessage = MessageBuilder.buildChatMessage("${fromUserId}","fromUserName","${userId}","toUserName","我是xxx"); // 构建消息体
byte[] chatMessageEncryptedBytes = ProtobufParser.encryptMessage(10,chatMessage); // 加密
logger.info("单聊消息byte: {}", chatMessageEncryptedBytes)
def chatMessageDecrypt = ProtobufParser.parseProtobufMessage(chatMessageEncryptedBytes); // 消息解密

设置显示日志视图

运行测试

运行结果