当前位置： 首页 > ai >正文

JMeter 实现 Protobuf 加密解密

ai 2025/7/23 10:09:29

一、 .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.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) commandSystem.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] & 0xFFbyte[] 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 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); // 消息解密

在这里插入图片描述

设置显示日志视图

在这里插入图片描述

运行测试

在这里插入图片描述

运行结果

在这里插入图片描述

http://www.xdnf.cn/news/15960.html

相关文章：

  • UE5 UI 水平框
  • ansible 批量 scp 和 load 镜像
  • MybatisPlus-16.扩展功能-枚举处理器
  • Windows PE文件内未用空间学习
  • DNS应用层协议
  • Linux驱动-中断-共享队列
  • 两个android，一个客户端一个服务器端
  • 2025.7.22 测试 总结
  • Web服务器（Tomcat、项目部署）
  • C# 中的装箱与拆箱
  • 今日行情明日机会——20250722
  • 基于AutoJawSegment项目的CBCT图像分割实践指南
  • 【bug】Yolo11在使用tensorrt推理numpy报错
  • Java 中 String 类的常用方法
  • OneCode 3.0 @TreeAnnotation及@ChildTreeAnnotation子树注解速查手册
  • 生存分析机器学习问题
  • 数据交换---JSON格式
  • IDEA-通过IDEA导入第三方的依赖包
  • Android常用的adb和logcat命令
  • Qt/C++源码/监控设备模拟器/支持onvif和gb28181/多路批量模拟/虚拟监控摄像头
  • RedisJSON 指令精讲JSON.TOGGLE 键翻转布尔值
  • Python趣味算法：实现任意进制转换算法原理+源码
  • 【无标题】buuctf-re3
  • 企业级IIS配置手册：安全加固/负载均衡/性能优化最佳实践
  • PyQt5—QLabel 学习笔记
  • 常用 Flutter 命令大全：从开发到发布全流程总结
  • ELF 文件操作手册
  • Java 动态导出 Word 登记表：多人员、分页、动态表格的最佳实践
  • C++11--锁分析
  • ospf技术