zydronium/SMBLibrary
1
0
Fork 0
mirror of https://github.com/TalAloni/SMBLibrary.git synced 2025-07-03 16:23:19 +02:00
Code Issues Projects Releases Packages Wiki Activity

SMB1: Improved FindInformation / QueryFSInformation implementations

Browse source
This commit is contained in:
Tal Aloni 2017-02-10 12:08:17 +02:00
parent a3017c1b50
commit a8ccae65ad
11 changed files with 120 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

8
SMBLibrary/SMB1/Transaction2Subcommands/Structures/FindInformation/FindFileBothDirectoryInfo.cs
View file

@ -98,5 +98,13 @@ namespace SMBLibrary.SMB1
}
return length;
}
public override FindInformationLevel InformationLevel
{
get
{
return FindInformationLevel.SMB_FIND_FILE_BOTH_DIRECTORY_INFO;
}
}
}
}

8
SMBLibrary/SMB1/Transaction2Subcommands/Structures/FindInformation/FindFileDirectoryInfo.cs
View file

@ -80,5 +80,13 @@ namespace SMBLibrary.SMB1
}
return length;
}
public override FindInformationLevel InformationLevel
{
get
{
return FindInformationLevel.SMB_FIND_FILE_DIRECTORY_INFO;
}
}
}
}

8
SMBLibrary/SMB1/Transaction2Subcommands/Structures/FindInformation/FindFileFullDirectoryInfo.cs
View file

@ -83,5 +83,13 @@ namespace SMBLibrary.SMB1
}
return length;
}
public override FindInformationLevel InformationLevel
{
get
{
return FindInformationLevel.SMB_FIND_FILE_FULL_DIRECTORY_INFO;
}
}
}
}

10
SMBLibrary/SMB1/Transaction2Subcommands/Structures/FindInformation/FindFileNamesInfo.cs
View file

@ -1,4 +1,4 @@
/* Copyright (C) 2014 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
*
* You can redistribute this program and/or modify it under the terms of
* the GNU Lesser Public License as published by the Free Software Foundation,
@ -59,5 +59,13 @@ namespace SMBLibrary.SMB1
}
return length;
}
public override FindInformationLevel InformationLevel
{
get
{
return FindInformationLevel.SMB_FIND_FILE_NAMES_INFO;
}
}
}
}

9
SMBLibrary/SMB1/Transaction2Subcommands/Structures/FindInformation/FindInformation.cs
View file

@ -1,4 +1,4 @@
/* Copyright (C) 2014 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
*
* You can redistribute this program and/or modify it under the terms of
* the GNU Lesser Public License as published by the Free Software Foundation,
@ -31,6 +31,11 @@ namespace SMBLibrary.SMB1
}
}
public abstract FindInformationLevel InformationLevel
{
get;
}
public static FindInformation ReadEntry(byte[] buffer, ref int offset, FindInformationLevel informationLevel, bool isUnicode, bool returnResumeKeys)
{
switch (informationLevel)
@ -44,7 +49,7 @@ namespace SMBLibrary.SMB1
case FindInformationLevel.SMB_FIND_FILE_BOTH_DIRECTORY_INFO:
return new FindFileBothDirectoryInfo(buffer, ref offset, isUnicode);
default:
throw new InvalidRequestException();;
throw new UnsupportedInformationLevelException();;
}
}
}

12
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryFSInformation/QueryFSAttibuteInfo.cs
View file

@ -1,4 +1,4 @@
/* Copyright (C) 2014 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
*
* You can redistribute this program and/or modify it under the terms of
* the GNU Lesser Public License as published by the Free Software Foundation,
@ -46,12 +46,20 @@ namespace SMBLibrary.SMB1
return buffer;
}
public int Length
public override int Length
{
get
{
return FixedLength + FileSystemName.Length * 2;
}
}
public override QueryFSInformationLevel InformationLevel
{
get
{
return QueryFSInformationLevel.SMB_QUERY_FS_ATTRIBUTE_INFO;
}
}
}
}

20
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryFSInformation/QueryFSDeviceInfo.cs
View file

@ -1,4 +1,4 @@
/* Copyright (C) 2014 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
*
* You can redistribute this program and/or modify it under the terms of
* the GNU Lesser Public License as published by the Free Software Foundation,
@ -16,7 +16,7 @@ namespace SMBLibrary.SMB1
/// </summary>
public class QueryFSDeviceInfo : QueryFSInformation
{
public const int Length = 8;
public const int FixedLength = 8;
public DeviceType DeviceType;
public DeviceCharacteristics DeviceCharacteristics;
@ -38,5 +38,21 @@ namespace SMBLibrary.SMB1
LittleEndianWriter.WriteUInt32(buffer, 4, (uint)DeviceCharacteristics);
return buffer;
}
public override int Length
{
get
{
return FixedLength;
}
}
public override QueryFSInformationLevel InformationLevel
{
get
{
return QueryFSInformationLevel.SMB_QUERY_FS_DEVICE_INFO;
}
}
}
}

14
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryFSInformation/QueryFSInformation.cs
View file

@ -1,4 +1,4 @@
/* Copyright (C) 2014 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
/* Copyright (C) 2014-2017 Tal Aloni <tal.aloni.il@gmail.com>. All rights reserved.
*
* You can redistribute this program and/or modify it under the terms of
* the GNU Lesser Public License as published by the Free Software Foundation,
@ -15,6 +15,16 @@ namespace SMBLibrary.SMB1
{
public abstract byte[] GetBytes(bool isUnicode);
public abstract int Length
{
get;
}
public abstract QueryFSInformationLevel InformationLevel
{
get;
}
public static QueryFSInformation GetQueryFSInformation(byte[] buffer, QueryFSInformationLevel informationLevel, bool isUnicode)
{
switch (informationLevel)
@ -28,7 +38,7 @@ namespace SMBLibrary.SMB1
case QueryFSInformationLevel.SMB_QUERY_FS_ATTRIBUTE_INFO:
return new QueryFSAttibuteInfo(buffer, 0);
default:
throw new InvalidRequestException();
throw new UnsupportedInformationLevelException();
}
}
}

18
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryFSInformation/QueryFSSizeInfo.cs
View file

@ -16,7 +16,7 @@ namespace SMBLibrary.SMB1
/// </summary>
public class QueryFSSizeInfo : QueryFSInformation
{
public const int Length = 24;
public const int FixedLength = 24;
public long TotalAllocationUnits;
public long TotalFreeAllocationUnits;
@ -44,5 +44,21 @@ namespace SMBLibrary.SMB1
LittleEndianWriter.WriteUInt32(buffer, 20, BytesPerSector);
return buffer;
}
public override int Length
{
get
{
return FixedLength;
}
}
public override QueryFSInformationLevel InformationLevel
{
get
{
return QueryFSInformationLevel.SMB_QUERY_FS_SIZE_INFO;
}
}
}
}

28
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryFSInformation/QueryFSVolumeInfo.cs
View file

@ -20,7 +20,7 @@ namespace SMBLibrary.SMB1
public DateTime? VolumeCreationTime;
public uint SerialNumber;
//uint VolumeLabelSize;
private uint VolumeLabelSize;
public ushort Reserved;
public string VolumeLabel; // Unicode
@ -33,22 +33,38 @@ namespace SMBLibrary.SMB1
{
VolumeCreationTime = FileTimeHelper.ReadNullableFileTime(buffer, offset + 0);
SerialNumber = LittleEndianConverter.ToUInt32(buffer, offset + 8);
uint volumeLabelSize = LittleEndianConverter.ToUInt32(buffer, offset + 12);
VolumeLabelSize = LittleEndianConverter.ToUInt32(buffer, offset + 12);
Reserved = LittleEndianConverter.ToUInt16(buffer, offset + 16);
VolumeLabel = ByteReader.ReadUTF16String(buffer, offset + 18, (int)volumeLabelSize);
VolumeLabel = ByteReader.ReadUTF16String(buffer, offset + 18, (int)VolumeLabelSize);
}
public override byte[] GetBytes(bool isUnicode)
{
uint volumeLabelSize = (uint)(VolumeLabel.Length * 2);
VolumeLabelSize = (uint)(VolumeLabel.Length * 2);
byte[] buffer = new byte[FixedLength + volumeLabelSize];
byte[] buffer = new byte[this.Length];
FileTimeHelper.WriteFileTime(buffer, 0, VolumeCreationTime);
LittleEndianWriter.WriteUInt32(buffer, 8, SerialNumber);
LittleEndianWriter.WriteUInt32(buffer, 12, volumeLabelSize);
LittleEndianWriter.WriteUInt32(buffer, 12, VolumeLabelSize);
LittleEndianWriter.WriteUInt16(buffer, 16, Reserved);
ByteWriter.WriteUTF16String(buffer, 18, VolumeLabel);
return buffer;
}
public override int Length
{
get
{
return FixedLength + VolumeLabel.Length * 2;
}
}
public override QueryFSInformationLevel InformationLevel
{
get
{
return QueryFSInformationLevel.SMB_QUERY_FS_VOLUME_INFO;
}
}
}
}

2
SMBLibrary/SMB1/Transaction2Subcommands/Structures/QueryInformation/QueryInformation.cs
View file

@ -41,7 +41,7 @@ namespace SMBLibrary.SMB1
case QueryInformationLevel.SMB_QUERY_FILE_COMPRESSION_INFO:
return new QueryFileCompressionInfo(buffer, 0);
default:
throw new InvalidRequestException();
throw new UnsupportedInformationLevelException();
}
}
}