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

SMB1FileStore, SMB2FileStore: Return STATUS_IO_TIMEOUT instead of STATUS_INVALID_SMB when server does not reply and there is no protocol violation

Browse source
This commit is contained in:
Tal Aloni 2024-06-12 12:27:26 +03:00
parent d8b1791212
commit 7ecc9e667a
2 changed files with 54 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
SMBLibrary/Client/SMB1FileStore.cs
View file

@ -36,7 +36,7 @@ namespace SMBLibrary.Client
request.ImpersonationLevel = ImpersonationLevel.Impersonation; request.ImpersonationLevel = ImpersonationLevel.Impersonation;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_CREATE_ANDX); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_CREATE_ANDX, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Commands[0] is NTCreateAndXResponse) if (reply.Commands[0] is NTCreateAndXResponse)
@ -51,7 +51,7 @@ namespace SMBLibrary.Client
return reply.Header.Status; return reply.Header.Status;
} }
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus CloseFile(object handle) public NTStatus CloseFile(object handle)
@ -59,12 +59,12 @@ namespace SMBLibrary.Client
CloseRequest request = new CloseRequest(); CloseRequest request = new CloseRequest();
request.FID = (ushort)handle; request.FID = (ushort)handle;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_CLOSE); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_CLOSE, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus ReadFile(out byte[] data, object handle, long offset, int maxCount) public NTStatus ReadFile(out byte[] data, object handle, long offset, int maxCount)
@ -76,7 +76,7 @@ namespace SMBLibrary.Client
request.MaxCountLarge = (uint)maxCount; request.MaxCountLarge = (uint)maxCount;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_READ_ANDX); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_READ_ANDX, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is ReadAndXResponse) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is ReadAndXResponse)
@ -85,7 +85,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus WriteFile(out int numberOfBytesWritten, object handle, long offset, byte[] data) public NTStatus WriteFile(out int numberOfBytesWritten, object handle, long offset, byte[] data)
@ -97,7 +97,7 @@ namespace SMBLibrary.Client
request.Data = data; request.Data = data;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_WRITE_ANDX); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_WRITE_ANDX, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is WriteAndXResponse) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is WriteAndXResponse)
@ -106,7 +106,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus FlushFileBuffers(object handle) public NTStatus FlushFileBuffers(object handle)
@ -150,7 +150,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
@ -180,10 +180,10 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out connectionTerminated);
if (reply == null) if (reply == null)
{ {
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
else if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) else if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
{ {
@ -201,7 +201,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus GetFileInformation(out FileInformation result, object handle, FileInformationClass informationClass) public NTStatus GetFileInformation(out FileInformation result, object handle, FileInformationClass informationClass)
@ -224,7 +224,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
@ -244,7 +244,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
else else
{ {
@ -277,7 +277,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
@ -288,7 +288,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetFileInformation(object handle, FileInformation information) public NTStatus SetFileInformation(object handle, FileInformation information)
@ -319,12 +319,12 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
else else
{ {
@ -349,12 +349,12 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus GetFileSystemInformation(out FileSystemInformation result, FileSystemInformationClass informationClass) public NTStatus GetFileSystemInformation(out FileSystemInformation result, FileSystemInformationClass informationClass)
@ -376,7 +376,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
@ -387,7 +387,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
else else
{ {
@ -412,7 +412,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (ushort)maxOutputLength; request.MaxDataCount = (ushort)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION2, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is Transaction2Response)
@ -423,7 +423,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetFileSystemInformation(FileSystemInformation information) public NTStatus SetFileSystemInformation(FileSystemInformation information)
@ -450,7 +450,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (uint)maxOutputLength; request.MaxDataCount = (uint)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_TRANSACT); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_TRANSACT, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is NTTransactResponse) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is NTTransactResponse)
@ -461,7 +461,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetSecurityInformation(object handle, SecurityInformation securityInformation, SecurityDescriptor securityDescriptor) public NTStatus SetSecurityInformation(object handle, SecurityInformation securityInformation, SecurityDescriptor securityDescriptor)
@ -504,7 +504,7 @@ namespace SMBLibrary.Client
request.MaxDataCount = (uint)maxOutputLength; request.MaxDataCount = (uint)maxOutputLength;
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_TRANSACT); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_NT_TRANSACT, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is NTTransactResponse) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is NTTransactResponse)
@ -515,7 +515,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus FsCtlPipeTranscieve(object handle, byte[] input, out byte[] output, int maxOutputLength) public NTStatus FsCtlPipeTranscieve(object handle, byte[] input, out byte[] output, int maxOutputLength)
@ -536,7 +536,7 @@ namespace SMBLibrary.Client
request.Name = @"\PIPE\"; request.Name = @"\PIPE\";
TrySendMessage(request); TrySendMessage(request);
SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION); SMB1Message reply = m_client.WaitForMessage(CommandName.SMB_COM_TRANSACTION, out bool connectionTerminated);
if (reply != null) if (reply != null)
{ {
if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is TransactionResponse) if (reply.Header.Status == NTStatus.STATUS_SUCCESS && reply.Commands[0] is TransactionResponse)
@ -547,7 +547,7 @@ namespace SMBLibrary.Client
} }
return reply.Header.Status; return reply.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus Disconnect() public NTStatus Disconnect()

48
SMBLibrary/Client/SMB2FileStore.cs
View file

@ -40,7 +40,7 @@ namespace SMBLibrary.Client
request.ImpersonationLevel = ImpersonationLevel.Impersonation; request.ImpersonationLevel = ImpersonationLevel.Impersonation;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is CreateResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is CreateResponse)
@ -52,7 +52,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus CloseFile(object handle) public NTStatus CloseFile(object handle)
@ -60,13 +60,13 @@ namespace SMBLibrary.Client
CloseRequest request = new CloseRequest(); CloseRequest request = new CloseRequest();
request.FileId = (FileID)handle; request.FileId = (FileID)handle;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus ReadFile(out byte[] data, object handle, long offset, int maxCount) public NTStatus ReadFile(out byte[] data, object handle, long offset, int maxCount)
@ -79,7 +79,7 @@ namespace SMBLibrary.Client
request.ReadLength = (uint)maxCount; request.ReadLength = (uint)maxCount;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is ReadResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is ReadResponse)
@ -89,7 +89,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus WriteFile(out int numberOfBytesWritten, object handle, long offset, byte[] data) public NTStatus WriteFile(out int numberOfBytesWritten, object handle, long offset, byte[] data)
@ -102,7 +102,7 @@ namespace SMBLibrary.Client
request.Data = data; request.Data = data;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is WriteResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is WriteResponse)
@ -112,7 +112,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus FlushFileBuffers(object handle) public NTStatus FlushFileBuffers(object handle)
@ -121,7 +121,7 @@ namespace SMBLibrary.Client
request.FileId = (FileID) handle; request.FileId = (FileID) handle;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is FlushResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is FlushResponse)
@ -130,7 +130,7 @@ namespace SMBLibrary.Client
} }
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus LockFile(object handle, long byteOffset, long length, bool exclusiveLock) public NTStatus LockFile(object handle, long byteOffset, long length, bool exclusiveLock)
@ -155,7 +155,7 @@ namespace SMBLibrary.Client
request.FileName = fileName; request.FileName = fileName;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
while (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryDirectoryResponse) while (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryDirectoryResponse)
@ -164,16 +164,16 @@ namespace SMBLibrary.Client
result.AddRange(page); result.AddRange(page);
request.Reopen = false; request.Reopen = false;
TrySendCommand(request); TrySendCommand(request);
response = m_client.WaitForCommand(request.MessageID); response = m_client.WaitForCommand(request.MessageID, out connectionTerminated);
if (response == null) if (response == null)
{ {
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
} }
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus GetFileInformation(out FileInformation result, object handle, FileInformationClass informationClass) public NTStatus GetFileInformation(out FileInformation result, object handle, FileInformationClass informationClass)
@ -186,7 +186,7 @@ namespace SMBLibrary.Client
request.FileId = (FileID)handle; request.FileId = (FileID)handle;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse)
@ -196,7 +196,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetFileInformation(object handle, FileInformation information) public NTStatus SetFileInformation(object handle, FileInformation information)
@ -208,13 +208,13 @@ namespace SMBLibrary.Client
request.SetFileInformation(information); request.SetFileInformation(information);
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus GetFileSystemInformation(out FileSystemInformation result, FileSystemInformationClass informationClass) public NTStatus GetFileSystemInformation(out FileSystemInformation result, FileSystemInformationClass informationClass)
@ -243,7 +243,7 @@ namespace SMBLibrary.Client
request.FileId = (FileID)handle; request.FileId = (FileID)handle;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse)
@ -253,7 +253,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetFileSystemInformation(FileSystemInformation information) public NTStatus SetFileSystemInformation(FileSystemInformation information)
@ -271,7 +271,7 @@ namespace SMBLibrary.Client
request.FileId = (FileID)handle; request.FileId = (FileID)handle;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse) if (response.Header.Status == NTStatus.STATUS_SUCCESS && response is QueryInfoResponse)
@ -281,7 +281,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus SetSecurityInformation(object handle, SecurityInformation securityInformation, SecurityDescriptor securityDescriptor) public NTStatus SetSecurityInformation(object handle, SecurityInformation securityInformation, SecurityDescriptor securityDescriptor)
@ -310,7 +310,7 @@ namespace SMBLibrary.Client
request.Input = input; request.Input = input;
request.MaxOutputResponse = (uint)maxOutputLength; request.MaxOutputResponse = (uint)maxOutputLength;
TrySendCommand(request); TrySendCommand(request);
SMB2Command response = m_client.WaitForCommand(request.MessageID); SMB2Command response = m_client.WaitForCommand(request.MessageID, out bool connectionTerminated);
if (response != null) if (response != null)
{ {
if ((response.Header.Status == NTStatus.STATUS_SUCCESS || response.Header.Status == NTStatus.STATUS_BUFFER_OVERFLOW) && response is IOCtlResponse) if ((response.Header.Status == NTStatus.STATUS_SUCCESS || response.Header.Status == NTStatus.STATUS_BUFFER_OVERFLOW) && response is IOCtlResponse)
@ -320,7 +320,7 @@ namespace SMBLibrary.Client
return response.Header.Status; return response.Header.Status;
} }
return NTStatus.STATUS_INVALID_SMB; return connectionTerminated ? NTStatus.STATUS_INVALID_SMB : NTStatus.STATUS_IO_TIMEOUT;
} }
public NTStatus Disconnect() public NTStatus Disconnect()