anna/IM-server-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加一些异常数据的处理

Browse Source
This commit is contained in:
heavyrian2012
2023-03-20 13:49:29 +08:00
parent 11b7851b9f
commit 68abeed907
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
broker/src/main/java/io/moquette/spi/impl/ProtocolProcessor.java
+3 -2
View File
@@ -608,8 +608,9 @@ public class ProtocolProcessor {
channel.closeFuture();
//disconnect the session
m_sessionsStore.sessionForClient(clientID).disconnect(session.getUsername(), isDup, isRetain);
if(session != null) {
m_sessionsStore.sessionForClient(clientID).disconnect(session.getUsername(), isDup, isRetain);
}
}
broker/src/main/java/io/moquette/spi/impl/Qos1PublishHandler.java
+7
View File
@@ -252,6 +252,13 @@ public class Qos1PublishHandler extends QosPublishHandler {
MemorySessionStore.Session session = m_sessionStore.getSession(clientID);
payloadContent = AES.AESDecrypt(payloadContent, session.getSecret(), true);
if(payloadContent == null) {
ByteBuf ackPayload = Unpooled.buffer();
ackPayload.ensureWritable(1).writeByte(ERROR_CODE_INVALID_DATA.getCode());
sendPubAck(clientID, messageID, ackPayload, ERROR_CODE_INVALID_DATA);
return;
}
imHandler(clientID, username, imtopic, payloadContent, (errorCode, ackPayload) -> sendPubAck(clientID, messageID, ackPayload, errorCode), ProtoConstants.RequestSourceType.Request_From_User);
}