public void SimpleFile()
{
byte[] randomData = new byte[65 * 4096];
_random.NextBytes(randomData);
using (TempFile tmpFile = new TempFile())
{
// Create a file
File.WriteAllBytes(tmpFile.File.FullName, randomData);
// Discover it via the NTFS lib
char driveLetter = tmpFile.File.FullName[0];
RawDisk disk = new RawDisk(driveLetter);
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, 0);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, tmpFile.File.DirectoryName);
NtfsFile ntfsFile = NTFSHelpers.OpenFile(ntfsDir, tmpFile.File.Name);
Assert.IsNotNull(ntfsFile);
// Read it
using (Stream actualStream = File.OpenRead(tmpFile.File.FullName))
using (Stream ntfsStream = ntfsFile.OpenRead())
{
bool equal = StreamUtils.CompareStreams(actualStream, ntfsStream);
Assert.IsTrue(equal);
}
}
}
public void EnumerateChilds()
{
Random rand = new Random();
using (TempDir tmpDir = new TempDir())
{
// Make files
byte[] data = new byte[1024 * 1024];
for (int i = 0; i < 10; i++)
{
rand.NextBytes(data);
File.WriteAllBytes(Path.Combine(tmpDir.Directory.FullName, i + ".bin"), data);
}
// Make dirs
for (int i = 0; i < 10; i++)
{
rand.NextBytes(data);
tmpDir.Directory.CreateSubdirectory("dir" + i);
}
// Discover dir in NTFSLib
char driveLetter = tmpDir.Directory.FullName[0];
RawDisk disk = new RawDisk(driveLetter);
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, 0);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, tmpDir.Directory.FullName);
// Enumerate files
List <NtfsFile> ntfsFiles = ntfsDir.ListFiles().ToList();
Assert.AreEqual(10, ntfsFiles.Count);
for (int i = 0; i < 10; i++)
{
Assert.AreEqual(1, ntfsFiles.Count(s => s.Name == i + ".bin"));
}
// Enumerate dirs
List <NtfsDirectory> ntfsDirs = ntfsDir.ListDirectories().ToList();
Assert.AreEqual(10, ntfsDirs.Count);
for (int i = 0; i < 10; i++)
{
Assert.AreEqual(1, ntfsDirs.Count(s => s.Name == "dir" + i));
}
}
}
public void AlternateDatastreamDirectory()
{
Random rand = new Random();
byte[][] data = new byte[10][];
for (int i = 0; i < 10; i++)
{
data[i] = new byte[1024 * 1024];
rand.NextBytes(data[i]);
}
using (TempDir tmpDir = new TempDir())
{
// Make file
for (int i = 0; i < 10; i++)
{
using (SafeFileHandle fileHandle = Win32.CreateFile(tmpDir.Directory.FullName + ":alternate" + i + ":$DATA"))
using (FileStream fs = new FileStream(fileHandle, FileAccess.ReadWrite))
{
fs.Write(data[i], 0, data[i].Length);
}
}
// Discover dir in NTFSLib
char driveLetter = tmpDir.Directory.FullName[0];
RawDisk disk = new RawDisk(driveLetter);
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, 0);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, tmpDir.Directory.FullName);
// Check streams
string[] streams = ntfsWrapper.ListDatastreams(ntfsDir.MFTRecord);
Assert.AreEqual(10, streams.Length);
for (int i = 0; i < 10; i++)
{
Assert.AreEqual(1, streams.Count(s => s == "alternate" + i));
}
// Check data
for (int i = 0; i < 10; i++)
{
using (Stream memStream = new MemoryStream(data[i]))
using (Stream fileStream = ntfsWrapper.OpenFileRecord(ntfsDir.MFTRecord, "alternate" + i))
{
StreamUtils.CompareStreams(memStream, fileStream);
}
}
}
}
public void SparseFile()
{
byte[] randomData = new byte[35 * 4096];
_random.NextBytes(randomData);
// Clear the 16 * 4096 -> 32 * 4096 range
Array.Clear(randomData, 16 * 4096, 16 * 4096);
using (TempFile tmpFile = new TempFile())
{
// Create a file
File.WriteAllBytes(tmpFile.File.FullName, randomData);
using (FilesystemDeviceWrapper wrapper = Win32.GetFileWrapper(tmpFile.File.FullName))
{
wrapper.FileSystemSetSparseFile(true);
wrapper.FileSystemSetZeroData(16 * 4096, 16 * 4096);
FILE_ALLOCATED_RANGE_BUFFER[] rangesData = wrapper.FileSystemQueryAllocatedRanges(0, randomData.Length);
// We should have 2 ranges on non-zero data
Assert.AreEqual(2, rangesData.Length);
}
// Discover it via the NTFS lib
char driveLetter = tmpFile.File.FullName[0];
RawDisk disk = new RawDisk(driveLetter);
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, 0);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, tmpFile.File.DirectoryName);
NtfsFile ntfsFile = NTFSHelpers.OpenFile(ntfsDir, tmpFile.File.Name);
Assert.IsNotNull(ntfsFile);
Assert.IsTrue(tmpFile.File.Attributes.HasFlag(FileAttributes.SparseFile));
AttributeData attributeData = ntfsFile.MFTRecord.Attributes.OfType <AttributeData>().Single();
Assert.IsTrue(attributeData.DataFragments.Length > 1);
Assert.IsTrue(attributeData.DataFragments.Any(s => s.IsSparseFragment));
// Read it
using (Stream actualStream = File.OpenRead(tmpFile.File.FullName))
using (Stream ntfsStream = ntfsFile.OpenRead())
{
bool equal = StreamUtils.CompareStreams(actualStream, ntfsStream);
Assert.IsTrue(equal);
}
}
}
public static IEnumerable <NtfsDirectory> GetDirectories(char driveLetter, string directory)
{
if (string.IsNullOrWhiteSpace(directory))
{
throw new ArgumentException("String cannot be null, empty or whitespace.", nameof(directory));
}
using (RawDisk disk = new RawDisk(driveLetter, FileAccess.Read))
{
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, rawDiskCacheRecordSize);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, directory);
return(ntfsDir.ListDirectories());
}
}
public void CompressedFile()
{
using (TempFile tmpFile = new TempFile())
{
// Create a file
// Write file data
using (FileStream fs = File.OpenWrite(tmpFile.File.FullName))
{
byte[] data = Encoding.ASCII.GetBytes("The white bunny jumps over the brown dog in a carparking lot");
for (int i = 0; i < 20000; i++)
{
fs.Write(data, 0, data.Length);
}
}
using (FilesystemDeviceWrapper wrapper = Win32.GetFileWrapper(tmpFile.File.FullName))
{
wrapper.FileSystemSetCompression(COMPRESSION_FORMAT.LZNT1);
}
// Discover it via the NTFS lib
char driveLetter = tmpFile.File.FullName[0];
RawDisk disk = new RawDisk(driveLetter);
NTFSDiskProvider provider = new NTFSDiskProvider(disk);
NTFSWrapper ntfsWrapper = new NTFSWrapper(provider, 0);
NtfsDirectory ntfsDir = NTFSHelpers.OpenDir(ntfsWrapper, tmpFile.File.DirectoryName);
NtfsFile ntfsFile = NTFSHelpers.OpenFile(ntfsDir, tmpFile.File.Name);
Assert.IsNotNull(ntfsFile);
Assert.IsTrue(tmpFile.File.Attributes.HasFlag(FileAttributes.Compressed));
AttributeData attributeData = ntfsFile.MFTRecord.Attributes.OfType <AttributeData>().Single();
Assert.IsTrue(attributeData.DataFragments.Any(s => s.IsCompressed));
// Read it
using (Stream actualStream = File.OpenRead(tmpFile.File.FullName))
using (Stream ntfsStream = ntfsFile.OpenRead())
{
bool equal = StreamUtils.CompareStreams(actualStream, ntfsStream);
Assert.IsTrue(equal);
}
}
}
