add configuration of max packet size for network

Author: JavaSaBr

rlib-network/src/main/java/javasabr/rlib/network/NetworkConfig.java

@@ -30,6 +30,12 @@ class SimpleNetworkConfig implements NetworkConfig {
  private int writeBufferSize = 2048;
  @Builder.Default
  private int retryDelayInMs = 1000;
+ @Builder.Default
+ private int maxPacketSize = 5 * 1024 * 1024;
+ @Builder.Default
+ private int maxEmptyReadsBeforeClose = 3;
+ @Builder.Default
+ private boolean useDirectByteBuffer = false;
  }
 
  NetworkConfig DEFAULT_CLIENT = new NetworkConfig() {
@@ -76,6 +82,13 @@ default int writeBufferSize() {
  return 2048;
  }
 
+ /**
+ * Gets the max size of one single network packet.
+ */
+ default int maxPacketSize() {
+ return 5 * 1024 * 1024;
+ }
+
  /**
  * Gets a timeout for retry read/write operation.
  */

rlib-network/src/main/java/javasabr/rlib/network/ServerNetworkConfig.java

@@ -34,6 +34,12 @@ class SimpleServerNetworkConfig implements ServerNetworkConfig {
  @Builder.Default
  private int retryDelayInMs = 1000;
  @Builder.Default
+ private int maxPacketSize = 5 * 1024 * 1024;
+ @Builder.Default
+ private int maxEmptyReadsBeforeClose = 3;
+ @Builder.Default
+ private boolean useDirectByteBuffer = false;
+ @Builder.Default
  private int threadGroupMinSize = 1;
  @Builder.Default
  private int threadGroupMaxSize = 1;

rlib-network/src/main/java/javasabr/rlib/network/packet/impl/AbstractNetworkPacketReader.java

@@ -153,11 +153,15 @@ protected int readPackets(ByteBuffer receivedBuffer) {
  */
  protected int readPackets(ByteBuffer receivedBuffer, ByteBuffer pendingBuffer) {
  String remoteAddress = remoteAddress();
+ NetworkConfig networkConfig = connection
+ .network()
+ .config();
 
  log.debug(remoteAddress, receivedBuffer,
  "[%s] Start reading packets from received buffer:[%s]"::formatted);
 
  int waitedBytes = pendingBuffer.position();
+
  ByteBuffer bufferToRead = receivedBuffer;
  ByteBuffer tempBigBuffer = tempBigBuffer();
 
@@ -214,6 +218,11 @@ else if (waitedBytes > 0) {
 
  int positionBeforeRead = endPosition;
  int packetFullLength = readFullPacketLength(bufferToRead);
+ if (packetFullLength > networkConfig.maxPacketSize()) {
+ throw new IllegalStateException(
+ "Received to big packet:[" + packetFullLength + ">" + networkConfig.maxPacketSize() + "]");
+ }
+
  int alreadyReadBytes = bufferToRead.position() - endPosition;
  int packetDataLength = calculatePacketDataLength(packetFullLength, alreadyReadBytes, bufferToRead);