package cn.iocoder.yudao.module.iot.gateway.protocol.tcp.codec.delimiter;
|
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.module.iot.gateway.protocol.tcp.IotTcpConfig;
|
import cn.iocoder.yudao.module.iot.gateway.protocol.tcp.codec.IotTcpCodecTypeEnum;
|
import cn.iocoder.yudao.module.iot.gateway.protocol.tcp.codec.IotTcpFrameCodec;
|
import io.vertx.core.Handler;
|
import io.vertx.core.buffer.Buffer;
|
import io.vertx.core.parsetools.RecordParser;
|
import lombok.extern.slf4j.Slf4j;
|
import cn.hutool.core.lang.Assert;
|
|
/**
|
* IoT TCP 分隔符帧编解码器
|
* <p>
|
* 基于分隔符的拆包策略，消息格式：消息内容 + 分隔符
|
* <p>
|
* 支持的分隔符：
|
* <ul>
|
* <li>\n - 换行符</li>
|
* <li>\r - 回车符</li>
|
* <li>\r\n - 回车换行</li>
|
* <li>自定义字符串</li>
|
* </ul>
|
*
|
* @author 芋道源码
|
*/
|
@Slf4j
|
public class IotTcpDelimiterFrameCodec implements IotTcpFrameCodec {
|
|
/**
|
* 最大记录大小（64KB），防止 DoS 攻击
|
*/
|
private static final int MAX_RECORD_SIZE = 65536;
|
|
/**
|
* 解析后的分隔符字节数组
|
*/
|
private final byte[] delimiterBytes;
|
|
public IotTcpDelimiterFrameCodec(IotTcpConfig.CodecConfig config) {
|
Assert.notBlank(config.getDelimiter(), "delimiter 不能为空");
|
this.delimiterBytes = parseDelimiter(config.getDelimiter());
|
}
|
|
@Override
|
public IotTcpCodecTypeEnum getType() {
|
return IotTcpCodecTypeEnum.DELIMITER;
|
}
|
|
@Override
|
public RecordParser createDecodeParser(Handler<Buffer> handler) {
|
RecordParser parser = RecordParser.newDelimited(Buffer.buffer(delimiterBytes));
|
parser.maxRecordSize(MAX_RECORD_SIZE); // 设置最大记录大小，防止 DoS 攻击
|
// 处理完整消息（不包含分隔符）
|
parser.handler(handler);
|
parser.exceptionHandler(ex -> {
|
throw new RuntimeException("[createDecodeParser][解析异常]", ex);
|
});
|
return parser;
|
}
|
|
@Override
|
public Buffer encode(byte[] data) {
|
Buffer buffer = Buffer.buffer();
|
buffer.appendBytes(data);
|
buffer.appendBytes(delimiterBytes);
|
return buffer;
|
}
|
|
/**
|
* 解析分隔符字符串为字节数组
|
* <p>
|
* 支持转义字符：\n、\r、\r\n、\t
|
*
|
* @param delimiter 分隔符字符串
|
* @return 分隔符字节数组
|
*/
|
private byte[] parseDelimiter(String delimiter) {
|
// 处理转义字符
|
String parsed = delimiter
|
.replace("\\r\\n", "\r\n")
|
.replace("\\r", "\r")
|
.replace("\\n", "\n")
|
.replace("\\t", "\t");
|
return StrUtil.utf8Bytes(parsed);
|
}
|
|
}
|
