zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-08-12 10:19:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

SMBServer: SMB1: Added support for splitting SMB_COM_NT_TRANSACT transaction over multiple responses

Browse source
This commit is contained in:
Tal Aloni 2017-07-28 13:48:03 +03:00
parent 6fccba9e0e
commit b08850d7af
1 changed files with 37 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

43
SMBLibrary/Server/SMB1/NTTransactHelper.cs
View file

@ -162,20 +162,53 @@ namespace SMBLibrary.Server.SMB1
internal static List<SMB1Command> GetNTTransactResponse(byte[] responseSetup, byte[] responseParameters, byte[] responseData, int maxBufferSize) internal static List<SMB1Command> GetNTTransactResponse(byte[] responseSetup, byte[] responseParameters, byte[] responseData, int maxBufferSize)
{ {
if (NTTransactResponse.CalculateMessageSize(responseSetup.Length, responseParameters.Length, responseData.Length) <= maxBufferSize) List<SMB1Command> result = new List<SMB1Command>();
{
NTTransactResponse response = new NTTransactResponse(); NTTransactResponse response = new NTTransactResponse();
result.Add(response);
int responseSize = NTTransactResponse.CalculateMessageSize(responseSetup.Length, responseParameters.Length, responseData.Length);
if (responseSize <= maxBufferSize)
{
response.Setup = responseSetup; response.Setup = responseSetup;
response.TotalParameterCount = (ushort)responseParameters.Length; response.TotalParameterCount = (ushort)responseParameters.Length;
response.TotalDataCount = (ushort)responseData.Length; response.TotalDataCount = (ushort)responseData.Length;
response.TransParameters = responseParameters; response.TransParameters = responseParameters;
response.TransData = responseData; response.TransData = responseData;
return response;
} }
else else
{ {
throw new NotImplementedException(); int currentDataLength = maxBufferSize - (responseSize - responseData.Length);
} byte[] buffer = new byte[currentDataLength];
Array.Copy(responseData, 0, buffer, 0, currentDataLength);
response.Setup = responseSetup;
response.TotalParameterCount = (ushort)responseParameters.Length;
response.TotalDataCount = (ushort)responseData.Length;
response.TransParameters = responseParameters;
response.TransData = buffer;
int dataBytesLeftToSend = responseData.Length - currentDataLength;
while (dataBytesLeftToSend > 0)
{
NTTransactResponse additionalResponse = new NTTransactResponse();
currentDataLength = dataBytesLeftToSend;
responseSize = TransactionResponse.CalculateMessageSize(0, 0, dataBytesLeftToSend);
if (responseSize > maxBufferSize)
{
currentDataLength = maxBufferSize - (responseSize - dataBytesLeftToSend);
}
buffer = new byte[currentDataLength];
int dataBytesSent = responseData.Length - dataBytesLeftToSend;
Array.Copy(responseData, dataBytesSent, buffer, 0, currentDataLength);
additionalResponse.TotalParameterCount = (ushort)responseParameters.Length;
additionalResponse.TotalDataCount = (ushort)responseData.Length;
additionalResponse.TransData = buffer;
additionalResponse.ParameterDisplacement = (ushort)response.TransParameters.Length;
additionalResponse.DataDisplacement = (ushort)dataBytesSent;
result.Add(additionalResponse);
dataBytesLeftToSend -= currentDataLength;
}
}
return result;
} }
} }
} }