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

Properly handle NegotiateRequest

Browse source
This commit is contained in:
Tal Aloni 2017-01-17 21:54:49 +02:00
parent 4722555728
commit 30ce7ff79c
2 changed files with 33 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
SMBLibrary/Server/ConnectionState/ConnectionState.cs
View file

@ -15,17 +15,25 @@ namespace SMBLibrary.Server
{
public delegate void LogDelegate(Severity severity, string message);
public enum SMBDialect
{
NotSet,
NTLM012, // NT LM 0.12
}
public class ConnectionState
{
public Socket ClientSocket;
public IPEndPoint ClientEndPoint;
public NBTConnectionReceiveBuffer ReceiveBuffer;
protected LogDelegate LogToServerHandler;
public SMBDialect ServerDialect;
public ConnectionState(LogDelegate logToServerHandler)
{
ReceiveBuffer = new NBTConnectionReceiveBuffer();
LogToServerHandler = logToServerHandler;
ServerDialect = SMBDialect.NotSet;
}
public ConnectionState(ConnectionState state)
@ -34,6 +42,7 @@ namespace SMBLibrary.Server
ClientEndPoint = state.ClientEndPoint;
ReceiveBuffer = state.ReceiveBuffer;
LogToServerHandler = state.LogToServerHandler;
ServerDialect = state.ServerDialect;
}
public void LogToServer(Severity severity, string message)

31
SMBLibrary/Server/SMBServer.SMB1.cs
View file

@ -53,25 +53,42 @@ namespace SMBLibrary.Server
/// </summary>
public SMB1Command ProcessSMB1Command(SMB1Header header, SMB1Command command, SMB1ConnectionState state, List<SMB1Command> sendQueue)
{
if (command is NegotiateRequest)
if (state.ServerDialect == SMBDialect.NotSet)
{
NegotiateRequest request = (NegotiateRequest)command;
if (request.Dialects.Contains(SMBServer.NTLanManagerDialect))
if (command is NegotiateRequest)
{
if (EnableExtendedSecurity && header.ExtendedSecurityFlag)
NegotiateRequest request = (NegotiateRequest)command;
if (request.Dialects.Contains(SMBServer.NTLanManagerDialect))
{
return NegotiateHelper.GetNegotiateResponseExtended(request, m_serverGuid);
state.ServerDialect = SMBDialect.NTLM012;
if (EnableExtendedSecurity && header.ExtendedSecurityFlag)
{
return NegotiateHelper.GetNegotiateResponseExtended(request, m_serverGuid);
}
else
{
return NegotiateHelper.GetNegotiateResponse(header, request, m_users);
}
}
else
{
return NegotiateHelper.GetNegotiateResponse(header, request, m_users);
return new NegotiateResponseNotSupported();
}
}
else
{
return new NegotiateResponseNotSupported();
// [MS-CIFS] An SMB_COM_NEGOTIATE exchange MUST be completed before any other SMB messages are sent to the server
header.Status = NTStatus.STATUS_SMB_BAD_COMMAND;
return new ErrorResponse(command.CommandName);
}
}
else if (command is NegotiateRequest)
{
// There MUST be only one SMB_COM_NEGOTIATE exchange per SMB connection.
// Subsequent SMB_COM_NEGOTIATE requests received by the server MUST be rejected with error responses.
header.Status = NTStatus.STATUS_SMB_BAD_COMMAND;
return new ErrorResponse(command.CommandName);
}
else if (command is SessionSetupAndXRequest)
{
SessionSetupAndXRequest request = (SessionSetupAndXRequest)command;