SabreTools/BinaryObjectScanner
1
0
Fork 0
mirror of https://github.com/SabreTools/BinaryObjectScanner.git synced 2026-07-16 05:45:03 +00:00
Code Issues 295 Packages Projects Releases 20 Wiki Activity

Add NE segment table parsing to builder

Browse Source
This commit is contained in:
Matt Nadareski
2022-11-06 21:45:37 -08:00
parent c9b3a67c8b
commit fbab512975
1 changed files with 55 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
BurnOutSharp.Builder/NewExecutable.cs
View File

@@ -46,6 +46,19 @@ namespace BurnOutSharp.Builder
// Set the executable header
executable.Header = executableHeader;
// If the offset for the segment table doesn't exist
int tableAddress = initialOffset + executableHeader.SegmentTableOffset;
if (tableAddress >= data.Length)
return executable;
// Try to parse the segment table
var relocationTable = ParseSegmentTable(data, tableAddress, executableHeader.FileSegmentCount);
if (relocationTable == null)
return null;
// Set the segment table
executable.SegmentTable = relocationTable;
// TODO: Implement NE parsing
return null;
}
@@ -112,8 +125,20 @@ namespace BurnOutSharp.Builder
/// <returns>Filled segment table on success, null on error</returns>
private static SegmentTableEntry[] ParseSegmentTable(byte[] data, int offset, int count)
{
// TODO: Implement segment table reading
return null;
// TODO: Use marshalling here instead of building
var segmentTable = new SegmentTableEntry[count];
for (int i = 0; i < count; i++)
{
var entry = new SegmentTableEntry();
entry.Offset = data.ReadUInt16(ref offset);
entry.Length = data.ReadUInt16(ref offset);
entry.FlagWord = (SegmentTableEntryFlag)data.ReadUInt16(ref offset);
entry.MinimumAllocationSize = data.ReadUInt16(ref offset);
segmentTable[i] = entry;
}
return segmentTable;
}
#endregion
@@ -157,6 +182,20 @@ namespace BurnOutSharp.Builder
// Set the executable header
executable.Header = executableHeader;
// If the offset for the segment table doesn't exist
int tableAddress = initialOffset + executableHeader.SegmentTableOffset;
if (tableAddress >= data.Length)
return executable;
// Try to parse the segment table
data.Seek(tableAddress, SeekOrigin.Begin);
var relocationTable = ParseSegmentTable(data, executableHeader.FileSegmentCount);
if (relocationTable == null)
return null;
// Set the segment table
executable.SegmentTable = relocationTable;
// TODO: Implement NE parsing
return null;
}
@@ -221,8 +260,20 @@ namespace BurnOutSharp.Builder
/// <returns>Filled segment table on success, null on error</returns>
private static SegmentTableEntry[] ParseSegmentTable(Stream data, int count)
{
// TODO: Implement segment table reading
return null;
// TODO: Use marshalling here instead of building
var segmentTable = new SegmentTableEntry[count];
for (int i = 0; i < count; i++)
{
var entry = new SegmentTableEntry();
entry.Offset = data.ReadUInt16();
entry.Length = data.ReadUInt16();
entry.FlagWord = (SegmentTableEntryFlag)data.ReadUInt16();
entry.MinimumAllocationSize = data.ReadUInt16();
segmentTable[i] = entry;
}
return segmentTable;
}
#endregion