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

Tests: Added tests to NegotiateRequestParsingTests and NegotiateResponseParsingTests

Browse source
This commit is contained in:
Tal Aloni 2024-08-03 01:42:00 +03:00
parent 1485c365d5
commit 3f4381b5f9
2 changed files with 75 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
SMBLibrary.Tests/SMB2/NegotiateRequestParsingTests.cs
View file

@ -6,6 +6,7 @@
*/ */
using Microsoft.VisualStudio.TestTools.UnitTesting; using Microsoft.VisualStudio.TestTools.UnitTesting;
using SMBLibrary.SMB2; using SMBLibrary.SMB2;
using System;
namespace SMBLibrary.Tests.SMB2 namespace SMBLibrary.Tests.SMB2
{ {
@ -13,9 +14,47 @@ namespace SMBLibrary.Tests.SMB2
public class NegotiateRequestParsingTests public class NegotiateRequestParsingTests
{ {
[TestMethod] [TestMethod]
public void ParseNegotiateRequestWithNegotiateContextList() public void ParseNegotiateRequestWithNegotiateContextList_WhenOffsetIsZero()
{ {
byte[] negotiateRequestCommandBytes = new byte[] byte[] negotiateRequestCommandBytes = GetNegotiateRequestWithNegotiateContextListBytes();
NegotiateRequest negotiateRequest = new NegotiateRequest(negotiateRequestCommandBytes, 0);
Assert.AreEqual(5, negotiateRequest.Dialects.Count);
Assert.AreEqual(4, negotiateRequest.NegotiateContextList.Count);
Assert.AreEqual(NegotiateContextType.SMB2_PREAUTH_INTEGRITY_CAPABILITIES, negotiateRequest.NegotiateContextList[0].ContextType);
Assert.AreEqual(NegotiateContextType.SMB2_ENCRYPTION_CAPABILITIES, negotiateRequest.NegotiateContextList[1].ContextType);
}
[TestMethod]
public void ParseNegotiateRequestWithNegotiateContextList_WhenOffsetIsNonZero()
{
// Test non-zero offset
byte[] negotiateRequestCommandBytes = GetNegotiateRequestWithNegotiateContextListBytes();
byte[] buffer = new byte[negotiateRequestCommandBytes.Length + 2];
Array.Copy(negotiateRequestCommandBytes, 0, buffer, 2, negotiateRequestCommandBytes.Length);
NegotiateRequest negotiateRequest = new NegotiateRequest(buffer, 2);
Assert.AreEqual(5, negotiateRequest.Dialects.Count);
Assert.AreEqual(4, negotiateRequest.NegotiateContextList.Count);
Assert.AreEqual(NegotiateContextType.SMB2_PREAUTH_INTEGRITY_CAPABILITIES, negotiateRequest.NegotiateContextList[0].ContextType);
Assert.AreEqual(NegotiateContextType.SMB2_ENCRYPTION_CAPABILITIES, negotiateRequest.NegotiateContextList[1].ContextType);
}
[TestMethod]
public void ParseRewrittenNegotiateRequestWithNegotiateContextList()
{
byte[] negotiateRequestCommandBytes = GetNegotiateRequestWithNegotiateContextListBytes();
NegotiateRequest negotiateRequest = new NegotiateRequest(negotiateRequestCommandBytes, 0);
negotiateRequestCommandBytes = negotiateRequest.GetBytes();
negotiateRequest = new NegotiateRequest(negotiateRequestCommandBytes, 0);
Assert.AreEqual(5, negotiateRequest.Dialects.Count);
Assert.AreEqual(4, negotiateRequest.NegotiateContextList.Count);
Assert.AreEqual(NegotiateContextType.SMB2_PREAUTH_INTEGRITY_CAPABILITIES, negotiateRequest.NegotiateContextList[0].ContextType);
Assert.AreEqual(NegotiateContextType.SMB2_ENCRYPTION_CAPABILITIES, negotiateRequest.NegotiateContextList[1].ContextType);
}
private static byte[] GetNegotiateRequestWithNegotiateContextListBytes()
{
return new byte[]
{ {
0xfe,0x53,0x4d,0x42,0x40,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00, 0xfe,0x53,0x4d,0x42,0x40,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00, 0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,
@ -35,12 +74,6 @@ namespace SMBLibrary.Tests.SMB2
0x73,0x00,0x65,0x00,0x72,0x00,0x73,0x00,0x2e,0x00,0x6c,0x00,0x6f,0x00,0x63,0x00, 0x73,0x00,0x65,0x00,0x72,0x00,0x73,0x00,0x2e,0x00,0x6c,0x00,0x6f,0x00,0x63,0x00,
0x61,0x00,0x6c,0x00 0x61,0x00,0x6c,0x00
}; };
NegotiateRequest negotiateRequest = new NegotiateRequest(negotiateRequestCommandBytes, 0);
Assert.AreEqual(5, negotiateRequest.Dialects.Count);
Assert.AreEqual(4, negotiateRequest.NegotiateContextList.Count);
Assert.AreEqual(NegotiateContextType.SMB2_PREAUTH_INTEGRITY_CAPABILITIES, negotiateRequest.NegotiateContextList[0].ContextType);
Assert.AreEqual(NegotiateContextType.SMB2_ENCRYPTION_CAPABILITIES, negotiateRequest.NegotiateContextList[1].ContextType);
} }
} }
} }

40
SMBLibrary.Tests/SMB2/NegotiateResponseParsingTests.cs
View file

@ -6,6 +6,7 @@
*/ */
using Microsoft.VisualStudio.TestTools.UnitTesting; using Microsoft.VisualStudio.TestTools.UnitTesting;
using SMBLibrary.SMB2; using SMBLibrary.SMB2;
using System;
namespace SMBLibrary.Tests.SMB2 namespace SMBLibrary.Tests.SMB2
{ {
@ -13,9 +14,40 @@ namespace SMBLibrary.Tests.SMB2
public class NegotiateResponseParsingTests public class NegotiateResponseParsingTests
{ {
[TestMethod] [TestMethod]
public void ParseNegotiateResponseWithNegotiateContextList() public void ParseNegotiateResponseWithNegotiateContextList_WhenOffsetIsZero()
{ {
byte[] negotiateResponseCommandBytes = new byte[] byte[] negotiateResponseCommandBytes = GetNegotiateResponseWithNegotiateContextListBytes();
NegotiateResponse negotiateResponse = new NegotiateResponse(negotiateResponseCommandBytes, 0);
Assert.AreEqual(SMB2Dialect.SMB311, negotiateResponse.DialectRevision);
Assert.AreEqual(2, negotiateResponse.NegotiateContextList.Count);
}
[TestMethod]
public void ParseNegotiateResponseWithNegotiateContextList_WhenOffsetIsNonZero()
{
byte[] negotiateResponseCommandBytes = GetNegotiateResponseWithNegotiateContextListBytes();
byte[] buffer = new byte[negotiateResponseCommandBytes.Length + 2];
Array.Copy(negotiateResponseCommandBytes, 0, buffer, 2, negotiateResponseCommandBytes.Length);
NegotiateResponse negotiateResponse = new NegotiateResponse(buffer, 2);
Assert.AreEqual(SMB2Dialect.SMB311, negotiateResponse.DialectRevision);
Assert.AreEqual(2, negotiateResponse.NegotiateContextList.Count);
}
[TestMethod]
public void ParseRewrittenNegotiateResponseWithNegotiateContextList()
{
byte[] negotiateResponseCommandBytes = GetNegotiateResponseWithNegotiateContextListBytes();
NegotiateResponse negotiateResponse = new NegotiateResponse(negotiateResponseCommandBytes, 0);
negotiateResponseCommandBytes = negotiateResponse.GetBytes();
negotiateResponse = new NegotiateResponse(negotiateResponseCommandBytes, 0);
Assert.AreEqual(SMB2Dialect.SMB311, negotiateResponse.DialectRevision);
Assert.AreEqual(2, negotiateResponse.NegotiateContextList.Count);
}
private static byte[] GetNegotiateResponseWithNegotiateContextListBytes()
{
return new byte[]
{ {
0xfe, 0x53, 0x4d, 0x42, 0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0xfe, 0x53, 0x4d, 0x42, 0x40, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00,
0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
@ -50,10 +82,6 @@ namespace SMBLibrary.Tests.SMB2
0xe5, 0x74, 0x26, 0x7b, 0x55, 0xe9, 0x14, 0x7e, 0x96, 0xea, 0x2e, 0x56, 0x41, 0xfe, 0x00, 0x00, 0xe5, 0x74, 0x26, 0x7b, 0x55, 0xe9, 0x14, 0x7e, 0x96, 0xea, 0x2e, 0x56, 0x41, 0xfe, 0x00, 0x00,
0x02, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00 0x02, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x02, 0x00
}; };
NegotiateResponse negotiateResponse = new NegotiateResponse(negotiateResponseCommandBytes, 0);
Assert.AreEqual(SMB2Dialect.SMB311, negotiateResponse.DialectRevision);
Assert.AreEqual(2, negotiateResponse.NegotiateContextList.Count);
} }
} }
} }