zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-05-02 11:57:48 +02:00
Code Issues Projects Releases Packages Wiki Activity

Client: Fix ArgumentNullException following recent ArrayPool usage

Browse source
This commit is contained in:
Tal Aloni 2023-12-30 20:43:09 +02:00
parent cb2c50a7c3
commit e31df2a797
2 changed files with 34 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
SMBLibrary/Client/SMB1Client.cs
View file

@ -458,6 +458,8 @@ namespace SMBLibrary.Client
buffer.SetNumberOfBytesReceived(numberOfBytesReceived); buffer.SetNumberOfBytesReceived(numberOfBytesReceived);
ProcessConnectionBuffer(state); ProcessConnectionBuffer(state);
if (clientSocket.Connected)
{
try try
{ {
clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state); clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state);
@ -476,6 +478,7 @@ namespace SMBLibrary.Client
} }
} }
} }
}
private void ProcessConnectionBuffer(ConnectionState state) private void ProcessConnectionBuffer(ConnectionState state)
{ {

3
SMBLibrary/Client/SMB2Client.cs
View file

@ -396,6 +396,8 @@ namespace SMBLibrary.Client
buffer.SetNumberOfBytesReceived(numberOfBytesReceived); buffer.SetNumberOfBytesReceived(numberOfBytesReceived);
ProcessConnectionBuffer(state); ProcessConnectionBuffer(state);
if (clientSocket.Connected)
{
try try
{ {
clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state); clientSocket.BeginReceive(buffer.Buffer, buffer.WriteOffset, buffer.AvailableLength, SocketFlags.None, new AsyncCallback(OnClientSocketReceive), state);
@ -414,6 +416,7 @@ namespace SMBLibrary.Client
} }
} }
} }
}
private void ProcessConnectionBuffer(ConnectionState state) private void ProcessConnectionBuffer(ConnectionState state)
{ {