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

Added proper Length property to NetBIOS session packets

Browse source
This commit is contained in:
Tal Aloni 2017-03-10 20:01:07 +02:00
parent 801719172d
commit 62d240d166
7 changed files with 61 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
SMBLibrary/NetBios/SessionPackets/NegativeSessionResponsePacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -35,5 +34,13 @@ namespace SMBLibrary.NetBios
return base.GetBytes();
}
public override int Length
{
get
{
return HeaderLength + 1;
}
}
}
}

9
SMBLibrary/NetBios/SessionPackets/PositiveSessionResponsePacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -30,5 +29,13 @@ namespace SMBLibrary.NetBios
this.Trailer = new byte[0];
return base.GetBytes();
}
public override int Length
{
get
{
return HeaderLength;
}
}
}
}

9
SMBLibrary/NetBios/SessionPackets/SessionKeepAlivePacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -30,5 +29,13 @@ namespace SMBLibrary.NetBios
this.Trailer = new byte[0];
return base.GetBytes();
}
public override int Length
{
get
{
return HeaderLength;
}
}
}
}

1
SMBLibrary/NetBios/SessionPackets/SessionMessagePacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios

31
SMBLibrary/NetBios/SessionPackets/SessionPacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -16,11 +15,12 @@ namespace SMBLibrary.NetBios
/// </summary>
public abstract class SessionPacket
{
public const int HeaderLength = 4;
public const int MaxSessionPacketLength = 131075;
public SessionPacketTypeName Type;
public byte Flags;
public int Length; // 2 bytes + length extension bit
private int TrailerLength; // 2 bytes + length extension bit
public byte[] Trailer;
public SessionPacket()
@ -31,30 +31,37 @@ namespace SMBLibrary.NetBios
{
Type = (SessionPacketTypeName)ByteReader.ReadByte(buffer, offset + 0);
Flags = ByteReader.ReadByte(buffer, offset + 1);
Length = (Flags & 0x01) << 16 | BigEndianConverter.ToUInt16(buffer, offset + 2);
this.Trailer = ByteReader.ReadBytes(buffer, offset + 4, Length);
TrailerLength = (Flags & 0x01) << 16 | BigEndianConverter.ToUInt16(buffer, offset + 2);
Trailer = ByteReader.ReadBytes(buffer, offset + 4, TrailerLength);
}
public virtual byte[] GetBytes()
{
Length = this.Trailer.Length;
if (Length > 0x1FFFF)
TrailerLength = this.Trailer.Length;
if (TrailerLength > 0x1FFFF)
{
throw new ArgumentException("Invalid NBT packet length");
}
Flags = Convert.ToByte(Length > 0xFFFF);
Flags = Convert.ToByte(TrailerLength > 0xFFFF);
byte[] buffer = new byte[4 + Trailer.Length];
ByteWriter.WriteByte(buffer, 0, (byte)this.Type);
byte[] buffer = new byte[HeaderLength + Trailer.Length];
ByteWriter.WriteByte(buffer, 0, (byte)Type);
ByteWriter.WriteByte(buffer, 1, Flags);
BigEndianWriter.WriteUInt16(buffer, 2, (ushort)(Length & 0xFFFF));
ByteWriter.WriteBytes(buffer, 4, this.Trailer);
BigEndianWriter.WriteUInt16(buffer, 2, (ushort)(TrailerLength & 0xFFFF));
ByteWriter.WriteBytes(buffer, 4, Trailer);
return buffer;
}
public virtual int Length
{
get
{
return HeaderLength + Trailer.Length;
}
}
public static SessionPacket GetSessionPacket(byte[] buffer, int offset)
{
SessionPacketTypeName type = (SessionPacketTypeName)ByteReader.ReadByte(buffer, offset);

11
SMBLibrary/NetBios/SessionPackets/SessionRequestPacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -39,5 +38,15 @@ namespace SMBLibrary.NetBios
ByteWriter.WriteBytes(this.Trailer, part1.Length, part2);
return base.GetBytes();
}
public override int Length
{
get
{
byte[] part1 = NetBiosUtils.EncodeName(CalledName, String.Empty);
byte[] part2 = NetBiosUtils.EncodeName(CallingName, String.Empty);
return HeaderLength + part1.Length + part2.Length;
}
}
}
}

9
SMBLibrary/NetBios/SessionPackets/SessionRetargetResponsePacket.cs
View file

@ -6,7 +6,6 @@
*/
using System;
using System.Collections.Generic;
using System.Text;
using Utilities;
namespace SMBLibrary.NetBios
@ -37,5 +36,13 @@ namespace SMBLibrary.NetBios
BigEndianWriter.WriteUInt16(this.Trailer, 4, Port);
return base.GetBytes();
}
public override int Length
{
get
{
return HeaderLength + 6;
}
}
}
}