NBTConnectionReceiveBuffer: Added support for SMB 2.1 Large MTU

SMBLibrary/Client/ConnectionState.cs

@@ -19,10 +19,10 @@ namespace SMBLibrary.Client
         private Socket m_clientSocket;
         private NBTConnectionReceiveBuffer m_receiveBuffer;
 
-        public ConnectionState(Socket clientSocket)
+        public ConnectionState(Socket clientSocket, bool isLargeMTU)
         {
             m_clientSocket = clientSocket;
-            m_receiveBuffer = new NBTConnectionReceiveBuffer();
+            m_receiveBuffer = new NBTConnectionReceiveBuffer(isLargeMTU);
         }
 
         public Socket ClientSocket

SMBLibrary/Client/SMB1Client.cs

@@ -142,7 +142,7 @@ namespace SMBLibrary.Client
                 return false;
             }
 
-            ConnectionState state = new ConnectionState(m_clientSocket);
+            ConnectionState state = new ConnectionState(m_clientSocket, false);
             NBTConnectionReceiveBuffer buffer = state.ReceiveBuffer;
             m_clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state);
             return true;

SMBLibrary/Client/SMB2Client.cs

@@ -134,7 +134,7 @@ namespace SMBLibrary.Client
                 return false;
             }
 
-            ConnectionState state = new ConnectionState(m_clientSocket);
+            ConnectionState state = new ConnectionState(m_clientSocket, true);
             NBTConnectionReceiveBuffer buffer = state.ReceiveBuffer;
             m_clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state);
             return true;

SMBLibrary/NetBios/NBTConnectionReceiveBuffer.cs

@@ -18,7 +18,7 @@ namespace SMBLibrary.NetBios
         private int m_bytesInBuffer = 0;
         private int? m_packetLength;
 
-        public NBTConnectionReceiveBuffer() : this(SessionPacket.MaxSessionPacketLength)
+        public NBTConnectionReceiveBuffer(bool isLargeMTU) : this(isLargeMTU ? SessionPacket.MaxDirectTcpPacketLength : SessionPacket.MaxSessionPacketLength)
         {
         }
 

SMBLibrary/Server/ConnectionState/ConnectionState.cs

@@ -1,4 +1,4 @@
-/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
+/* Copyright (C) 2014-2020 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
  * 
  * You can redistribute this program and/or modify it under the terms of
  * the GNU Lesser Public License as published by the Free Software Foundation,
@@ -29,11 +29,11 @@ namespace SMBLibrary.Server
         public SMBDialect Dialect;
         public GSSContext AuthenticationContext;
 
-        public ConnectionState(Socket clientSocket, IPEndPoint clientEndPoint, LogDelegate logToServerHandler)
+        public ConnectionState(Socket clientSocket, IPEndPoint clientEndPoint, SMBTransportType transportType, LogDelegate logToServerHandler)
         {
             m_clientSocket = clientSocket;
             m_clientEndPoint = clientEndPoint;
-            m_receiveBuffer = new NBTConnectionReceiveBuffer();
+            m_receiveBuffer = new NBTConnectionReceiveBuffer(transportType == SMBTransportType.DirectTCPTransport);
             m_sendQueue = new BlockingQueue<SessionPacket>();
             m_creationDT = DateTime.UtcNow;
             m_lastReceiveDT = DateTime.UtcNow;

SMBLibrary/Server/SMBServer.cs

@@ -1,4 +1,4 @@
-/* Copyright (C) 2014-2019 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
+/* Copyright (C) 2014-2020 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
  * 
  * You can redistribute this program and/or modify it under the terms of
  * the GNU Lesser Public License as published by the Free Software Foundation,
@@ -160,7 +160,7 @@ namespace SMBLibrary.Server
 
             if (acceptConnection)
             {
-                ConnectionState state = new ConnectionState(clientSocket, clientEndPoint, Log);
+                ConnectionState state = new ConnectionState(clientSocket, clientEndPoint, m_transport, Log);
                 state.LogToServer(Severity.Verbose, "New connection request accepted");
                 Thread senderThread = new Thread(delegate()
                 {