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

JMeter 实现 Protobuf 加密解密

news 2025/7/23 7:13:07

一、 .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.dtcms.com/a/292374.html

相关文章：

  • AI 音频产品开发模板及流程（一）
  • 网络安全第三次作业搭建前端页面并解析
  • allegro 16.6配置CIS库报错 ORCIS-6129 ORCIS-6469
  • LeetCode 658.找到K个最接近的元素
  • .NET使用EPPlus导出EXCEL的接口中，文件流缺少文件名信息
  • Unity笔记——事件中心
  • 力扣-300.最长递增子序列
  • 以太坊网络发展分析：技术升级与市场动态的双重驱动
  • 快手开源 Kwaipilot-AutoThink 思考模型，有效解决过度思考问题
  • Cy3-COOH 花菁染料Cy3-羧基
  • linux-日志服务
  • Gitlab-CI实现组件自动推送
  • 常用 Flutter 命令大全：从开发到发布全流程总结
  • 检索增强型生成助力无人机精准数学推理！RAG-UAV：基于RAG的复杂算术推理方法
  • Lua语言
  • MybatisPlus-16.扩展功能-枚举处理器
  • ORACLE DATABASE 11.2.0.4 RAC Install
  • Vue-22-通过flask接口提供的数据使用plotly.js绘图(一)
  • Oracle定时清理归档日志
  • RAG（检索增强生成）里的文档管理
  • 二次供水管理系统：远程监控+智能调控+故障预警解决方案
  • 用ffmpeg 进行视频的拼接
  • 【指南】网络安全领域：HW 行动（国家网络安全攻防演练）是什么？
  • 一些Avalonia与WPF内容的对应关系和不同用法
  • 单一职责原则（SRP）：构建高质量软件的基石
  • Java 并发性深度解析
  • JS WebAPIs DOM节点概述
  • 网络层传输
  • 4G车载录像机的作用详解：提升行车安全与智能管理的核心技术
  • 数字签名（Digital Signature）