* Bump projects to .net 9 and update nugets * add VersionedSerialization + source generator * migrate versioning to StructVersion class, add handling/detection for 29.2/31.2 * add new struct definitions * rename serialization methods and add BinaryObjectStreamReader for interop * Rework metadata struct loading to use new struct versioning * move 29/31.1/.2 to use tags (-2022,-2023) instead of minor versions * fix metadata usage validity checks * rework code registration offsetting a bit and add second 29/31.1 condition * tweak .1 condition (again) * 29/31.2 was a psyop * also remove 29.2 from the readme * remove loading of packed dlls - this was a very unsafe feature * support auto-recovering type indices from type handles fixes loading of memory-dumped v29+ libraries since those replacee their class indices on load with a pointer to the corresponding type * support loading PEs without an export table * also read UnresolvedVirtualCallCount on regular v31 * Disable plugin loading for now * Overhaul disassembler script + add Binary Ninja target (#12) * Overhaul diassembler scripts: - No longer defines top level functions - Split into three classes: StatusHandler (like before), DisassemblerInterface (for interfacing with the used program API), ScriptContext (for definiting general functions that use the disassembler interface) - Add type annotations to all class methods and remove 2.7 compatibility stuff (Ghidra now supports Python 3 so this is unnecessary anymore) - Disassembler backends are now responsible for launching metadata/script processing, to better support disassembler differences - String handling is back in the base ScriptContext class, disassembler interfaces opt into the fake string segment creation and fall back to the old method if it isn't supported * Add Binary Ninja disassembler script backend This uses the new backend-controlled execution to launch metadata processing on a background thread to keep the ui responsive * make binary ninja script use own _BINARYNINJA_ define and add define helpers to header * Update README to account for new script and binary ninja backend * implement fake string segment functions for binary ninja but don't advertise support * also cache API function types in binary ninja backend * fix ida script and disable folders again * Fix metadata usage issues caused by it being a value type now * make TryMapVATR overrideable and implement it for ELFs * Make field offset reading use TryMapVATR to reduce exceptions * Fix NRE in Assembly ctor on < v24.2 * Update actions workflow to produce cross-platform CLI binaries, update readme to reflect .net 9 changes * workflow: only restore packages for projects that are being built * workflow: tweak caching and fix gui compilation * workflow: remove double .zip in CLI artifact name * 29/31.2 don't actually exist, this logic is not needed
151 lines
5.2 KiB
C#
151 lines
5.2 KiB
C#
using System.Buffers.Binary;
|
|
using System.Collections.Immutable;
|
|
using System.Runtime.CompilerServices;
|
|
using System.Runtime.InteropServices;
|
|
using System.Text;
|
|
|
|
namespace VersionedSerialization;
|
|
|
|
// ReSharper disable ReplaceSliceWithRangeIndexer | The range indexer gets compiled into .Slice(x, y) and not .Slice(x) which worsens performance
|
|
public ref struct SpanReader(ReadOnlySpan<byte> data, int offset = 0, bool littleEndian = true, bool is32Bit = false) : IReader
|
|
{
|
|
public int Offset = offset;
|
|
public readonly byte Peek => _data[Offset];
|
|
public readonly bool IsLittleEndian => _littleEndian;
|
|
public readonly bool Is32Bit => _is32Bit;
|
|
public readonly int Length => _data.Length;
|
|
public readonly int PointerSize => Is32Bit ? sizeof(uint) : sizeof(ulong);
|
|
|
|
private readonly ReadOnlySpan<byte> _data = data;
|
|
private readonly bool _littleEndian = littleEndian;
|
|
private readonly bool _is32Bit = is32Bit;
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
private T ReadInternal<T>() where T : unmanaged
|
|
{
|
|
var value = MemoryMarshal.Read<T>(_data.Slice(Offset));
|
|
Offset += Unsafe.SizeOf<T>();
|
|
return value;
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
private static TTo Cast<TFrom, TTo>(in TFrom from) => Unsafe.As<TFrom, TTo>(ref Unsafe.AsRef(in from));
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public ReadOnlySpan<byte> ReadBytes(int length)
|
|
{
|
|
var val = _data.Slice(Offset, length);
|
|
Offset += length;
|
|
return val;
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public T ReadPrimitive<T>() where T : unmanaged
|
|
{
|
|
if (typeof(T) == typeof(byte))
|
|
return Cast<byte, T>(_data[Offset++]);
|
|
|
|
var value = ReadInternal<T>();
|
|
if (!_littleEndian)
|
|
{
|
|
if (value is ulong val)
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(val);
|
|
value = Cast<ulong, T>(converted);
|
|
}
|
|
else if (typeof(T) == typeof(long))
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(Cast<T, long>(value));
|
|
value = Cast<long, T>(converted);
|
|
}
|
|
else if (typeof(T) == typeof(uint))
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(Cast<T, uint>(value));
|
|
value = Cast<uint, T>(converted);
|
|
}
|
|
else if (typeof(T) == typeof(int))
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(Cast<T, int>(value));
|
|
value = Cast<int, T>(converted);
|
|
}
|
|
else if (typeof(T) == typeof(ushort))
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(Cast<T, ushort>(value));
|
|
value = Cast<ushort, T>(converted);
|
|
}
|
|
else if (typeof(T) == typeof(short))
|
|
{
|
|
var converted = BinaryPrimitives.ReverseEndianness(Cast<T, short>(value));
|
|
value = Cast<short, T>(converted);
|
|
}
|
|
}
|
|
|
|
return value;
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public ImmutableArray<T> ReadPrimitiveArray<T>(long count) where T : unmanaged
|
|
{
|
|
var array = ImmutableArray.CreateBuilder<T>(checked((int)count));
|
|
for (long i = 0; i < count; i++)
|
|
array.Add(ReadPrimitive<T>());
|
|
|
|
return array.MoveToImmutable();
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public T ReadVersionedObject<T>(in StructVersion version = default) where T : IReadable, new()
|
|
{
|
|
var obj = new T();
|
|
obj.Read(ref this, in version);
|
|
return obj;
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public ImmutableArray<T> ReadVersionedObjectArray<T>(long count, in StructVersion version = default) where T : IReadable, new()
|
|
{
|
|
var array = ImmutableArray.CreateBuilder<T>(checked((int)count));
|
|
for (long i = 0; i < count; i++)
|
|
array.Add(ReadVersionedObject<T>(in version));
|
|
|
|
return array.MoveToImmutable();
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public string ReadString()
|
|
{
|
|
var length = _data.Slice(Offset).IndexOf(byte.MinValue);
|
|
|
|
if (length == -1)
|
|
throw new InvalidDataException("Failed to find string in span.");
|
|
|
|
var val = _data.Slice(Offset, length);
|
|
Offset += length + 1; // Skip null terminator
|
|
|
|
return Encoding.UTF8.GetString(val);
|
|
}
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public bool ReadBoolean() => ReadPrimitive<byte>() != 0;
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public ulong ReadNUInt() => _is32Bit ? ReadPrimitive<uint>() : ReadPrimitive<ulong>();
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
public long ReadNInt() => _is32Bit ? ReadPrimitive<int>() : ReadPrimitive<long>();
|
|
|
|
public void Align(int alignment = 0)
|
|
{
|
|
if (alignment == 0)
|
|
alignment = Is32Bit ? 4 : 8;
|
|
|
|
var rem = Offset % alignment;
|
|
if (rem != 0)
|
|
Offset += alignment - rem;
|
|
}
|
|
|
|
public void Skip(int count)
|
|
{
|
|
Offset += count;
|
|
}
|
|
} |