ToxicMushroom/Penumbra
1
0
Fork 0
mirror of https://github.com/xivdev/Penumbra.git synced 2025-12-12 18:27:24 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Switch CreateFileW hook to hooking from import table.

Browse source
This commit is contained in:
Ottermandias 2023-03-09 15:37:43 +01:00
parent 5c6c96b6c0
commit cd894e415d
1 changed files with 16 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

27
Penumbra/Interop/Loader/CreateFileWHook.cs
View file

@ -22,12 +22,6 @@ public unsafe class CreateFileWHook : IDisposable
private const char Prefix = ( char )( ( byte )'P' | ( ( '?' & 0x00FF ) << 8 ) ); private const char Prefix = ( char )( ( byte )'P' | ( ( '?' & 0x00FF ) << 8 ) );
private const int BufferSize = Utf8GamePath.MaxGamePathLength; private const int BufferSize = Utf8GamePath.MaxGamePathLength;
[DllImport( "kernel32.dll" )]
private static extern nint LoadLibrary( string dllName );
[DllImport( "kernel32.dll" )]
private static extern nint GetProcAddress( nint hModule, string procName );
private delegate nint CreateFileWDelegate( char* fileName, uint access, uint shareMode, nint security, uint creation, uint flags, nint template ); private delegate nint CreateFileWDelegate( char* fileName, uint access, uint shareMode, nint security, uint creation, uint flags, nint template );
private readonly Hook< CreateFileWDelegate > _createFileWHook; private readonly Hook< CreateFileWDelegate > _createFileWHook;
@ -36,11 +30,7 @@ public unsafe class CreateFileWHook : IDisposable
private readonly ThreadLocal< nint > _fileNameStorage = new(SetupStorage, true); private readonly ThreadLocal< nint > _fileNameStorage = new(SetupStorage, true);
public CreateFileWHook() public CreateFileWHook()
{ => _createFileWHook = Hook< CreateFileWDelegate >.FromImport( null, "KERNEL32.dll", "CreateFileW", 0, CreateFileWDetour );
var userApi = LoadLibrary( "kernel32.dll" );
var createFileAddress = GetProcAddress( userApi, "CreateFileW" );
_createFileWHook = Hook< CreateFileWDelegate >.FromAddress( createFileAddress, CreateFileWDetour );
}
/// <remarks> Long paths in windows need to start with "\\?\", so we keep this static in the pointers. </remarks> /// <remarks> Long paths in windows need to start with "\\?\", so we keep this static in the pointers. </remarks>
private static nint SetupStorage() private static nint SetupStorage()
@ -169,4 +159,19 @@ public unsafe class CreateFileWHook : IDisposable
fileName = new ReadOnlySpan< byte >( ( void* )address, ( int )length ); fileName = new ReadOnlySpan< byte >( ( void* )address, ( int )length );
return true; return true;
} }
// ***** Old method *****
//[DllImport( "kernel32.dll" )]
//private static extern nint LoadLibrary( string dllName );
//
//[DllImport( "kernel32.dll" )]
//private static extern nint GetProcAddress( nint hModule, string procName );
//
//public CreateFileWHookOld()
//{
// var userApi = LoadLibrary( "kernel32.dll" );
// var createFileAddress = GetProcAddress( userApi, "CreateFileW" );
// _createFileWHook = Hook<CreateFileWDelegate>.FromAddress( createFileAddress, CreateFileWDetour );
//}
} }