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_classnameimport 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.MessageProBufimport com.google.protobuf.InvalidProtocolBufferExceptionimport com.google.protobuf.MessageLiteimport java.util.function.Functionimport org.apache.logging.log4j.LogManagerimport org.apache.logging.log4j.Loggerclass 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 groovygroovyc -cp \"lib/*\" src/MessageBuilder.groovy -d build/messagegroovyc -cp \"lib/*\" src/ProtobufParser.groovy -d build/protobuf

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

:: 切换到目录cd groovyjar cf MessageBuilder.jar -C build/message .jar cf ProtobufParser.jar -C build/protobuf .

示例完整命令汇总

PS D:\\System\\Desktop\\csdn> cd groovyPS D:\\System\\Desktop\\csdn\\groovy> groovyc -cp \"lib/*\" src/MessageBuilder.groovy -d build/messagePS D:\\System\\Desktop\\csdn\\groovy> groovyc -cp \"lib/*\" src/ProtobufParser.groovy -d build/protobufPS 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.LogManagerimport 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); // 消息解密

设置显示日志视图

运行测试

运行结果

直播技巧分享