public void IncreasesRVA()
{
var stream = new MemoryStream();
var rw = new StreamStructReaderWriter(stream);
rw.Write(123);
rw.Write('c');
rw.Write(true);
Assert.AreEqual(7, stream.Length);
}
public void KeepsInitialRVA()
{
var stream = new MemoryStream();
stream.WriteByte(2);
stream.WriteByte(1);
stream.WriteByte(0);
var rw = new StreamStructReaderWriter(stream);
var read = rw.Read <TestStruct1>(rw.Write(_testStruct1));
var read2 = rw.Read <TestStruct2>(rw.Write(_testStruct2));
Assert.AreEqual(_testStruct1, read, $"Can not read struct of type {nameof(TestStruct1)} correctly after writing it");
Assert.AreEqual(_testStruct2, read2, $"Can not read struct of type {nameof(TestStruct2)} correctly after writing it");
}
public void IsCorrectSize()
{
var stream = new MemoryStream();
var rw = new StreamStructReaderWriter(stream);
rw.Write(_testStruct2);
Assert.AreEqual(9, stream.Length);
}
private void CanWriteAndRead <T>(T testValue) where T : struct
{
var stream = new MemoryStream();
var rw = new StreamStructReaderWriter(stream);
rw.Write(testValue);
var readStruct = rw.Read <T>(0);
Assert.AreEqual(testValue, readStruct, $"Can not read struct of type {typeof(T).Name} after writing it");
}
public unsafe IOptionalHeaderResult Package(IOptionalHeaderInfo param)
{
var codeSection = param.Sections.SectionHeaders.First(sec => sec.NameString.StartsWith(".text"));
var initializedDataSection = param.Sections.SectionHeaders.FirstOrDefault(sec => sec.NameString.StartsWith(".data"));
var uninitializedDataSection = param.Sections.SectionHeaders.FirstOrDefault(sec => sec.NameString.StartsWith(".bss"));
var optHeaderStandard = new PE32PlusOptionalHeaderStandard
{
ExecutableKind = ExecutableKind.PE32Plus,
BaseOfCodeRVA = codeSection.VirtualAddress,
EntryPointRVA = param.Sections.EntryPointRVA,
MajorLinkerVersion = param.MajorLinkerVersion,
MinorLinkerVersion = param.MinorLinkerVersion,
SizeOfCode = codeSection.VirtualSize,
SizeOfInitializedData = initializedDataSection != default ? initializedDataSection.VirtualSize : 0,
SizeOfUninitializedData = uninitializedDataSection != default ? uninitializedDataSection.VirtualSize : 0
};
var optHeaderWinNT = new PE32PlusOptionalHeaderWinNT
{
FileAlignment = param.FileAlignment,
SectionAlignment = param.SectionAlignment,
Subsystem = param.Subsystem,
DllCharacteristics = param.DllCharacteristics,
HeapSizeReserve = param.HeapSizeReserve,
HeapSizeCommit = param.HeapSizeCommit,
StackSizeReserve = param.StackSizeReserve,
StackSizeCommit = param.StackSizeCommit,
SizeOfImage = param.Sections.SectionHeaders.Max(sec => Round.Up(sec.VirtualAddress + sec.VirtualSize, param.SectionAlignment)),
ImageBaseOffset = param.ImageBase,
NumberOfDataDirectoryEntries = 0x10,
SizeOfHeaders = Round.Up(
(uint)(param.MZHeader.RawData.Length +
sizeof(PEHeader) +
sizeof(PE32PlusOptionalHeader) +
sizeof(SectionHeader) * param.Sections.SectionHeaders.Count),
param.FileAlignment
),
MajorOSVersion = param.MajorOperatingSystemVersion,
MinorOSVersion = param.MinorOperatingSystemVersion,
MajorSubsystemVersion = param.MajorSubSystemVersion,
MinorSubsystemVersion = param.MinorSubSystemVersion,
MajorImageVersion = param.MajorImageVersion,
MinorImageVersion = param.MinorImageVersion
};
var optHeaderDataDictionary = new PE32PlusOptionalHeaderDataDirectories
{
IAT = param.Sections.IAT,
ImportTable = param.Sections.ImportTable,
Debug = param.Sections.Debug,
BaseRelocationTable = param.Sections.BaseRelocationTable
//todo
};
var rawData = new MemoryStream();
var structWriter = new StreamStructReaderWriter(rawData);
structWriter.Write(optHeaderStandard);
structWriter.Write(optHeaderWinNT);
structWriter.Write(optHeaderDataDictionary);
return(new OptionalHeadersResult(
rawData.ToArray(),
param.Sections.Debug.VirtualAddress != 0,
param.Sections.BaseRelocationTable.VirtualAddress != 0
));
}
}
