AstralTransRocketries/minosoft
1
0
Fork 0
mirror of https://gitlab.bixilon.de/bixilon/minosoft.git synced 2025-09-15 02:15:34 -04:00
Code Issues Packages Projects Releases Wiki Activity

check packet length against max packet length

Browse Source
This commit is contained in:
Bixilon 2020-09-06 13:41:11 +02:00
parent 32fb5092c3
commit f7dd739aad
No known key found for this signature in database
GPG Key ID: 5CAD791931B09AC4
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/main/java/de/bixilon/minosoft/protocol/network/Network.java
View File

@ -45,7 +45,6 @@ public class Network {
int compressionThreshold = -1;
Socket socket;
OutputStream outputStream;
InputStream cipherInputStream;
InputStream inputStream;
boolean encryptionEnabled = false;
SecretKey secretKey;
@ -78,7 +77,6 @@ public class Network {
socket.setKeepAlive(true);
outputStream = socket.getOutputStream();
inputStream = socket.getInputStream();
cipherInputStream = inputStream;
socketRThread.setName(String.format("%d/SocketR", connection.getConnectionId()));
@ -153,7 +151,7 @@ public class Network {
int length = 0;
int read;
do {
read = cipherInputStream.read();
read = inputStream.read();
if (read == -1) {
disconnect();
return;
@ -166,8 +164,12 @@ public class Network {
throw new RuntimeException("VarInt is too big");
}
} while ((read & 0b10000000) != 0);
byte[] data = cipherInputStream.readNBytes(length);
if (length > ProtocolDefinition.PROTOCOL_PACKET_MAX_SIZE) {
Log.protocol(String.format("Server sent us a to big packet (%d bytes > %d bytes)", length, ProtocolDefinition.PROTOCOL_PACKET_MAX_SIZE));
inputStream.skip(length);
continue;
}
byte[] data = inputStream.readNBytes(length);
if (compressionThreshold >= 0) {
// compression is enabled
@ -257,7 +259,7 @@ public class Network {
public void enableEncryption(SecretKey secretKey) {
Cipher cipherEncrypt = CryptManager.createNetCipherInstance(Cipher.ENCRYPT_MODE, secretKey);
Cipher cipherDecrypt = CryptManager.createNetCipherInstance(Cipher.DECRYPT_MODE, secretKey);
cipherInputStream = new CipherInputStream(inputStream, cipherDecrypt);
inputStream = new CipherInputStream(inputStream, cipherDecrypt);
outputStream = new CipherOutputStream(outputStream, cipherEncrypt);
encryptionEnabled = true;
Log.debug("Encryption enabled!");