Fun/Snap.Hutao
1
0
Fork 0
mirror of https://jihulab.com/DGP-Studio/Snap.Hutao.git synced 2025-11-19 21:02:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Align HMODULE ptr

Browse Source
This commit is contained in:
Lightczx
2024-06-05 17:17:52 +08:00
parent 3e8e109123
commit b392a6f8e5
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/Snap.Hutao/Snap.Hutao/Service/Game/Unlocker/RequiredLocalModule.cs
View File

@@ -4,6 +4,7 @@
using Snap.Hutao.Win32.Foundation; using Snap.Hutao.Win32.Foundation;
using Snap.Hutao.Win32.System.Diagnostics.Debug; using Snap.Hutao.Win32.System.Diagnostics.Debug;
using Snap.Hutao.Win32.System.SystemService; using Snap.Hutao.Win32.System.SystemService;
using System.Runtime.CompilerServices;
using static Snap.Hutao.Win32.Kernel32; using static Snap.Hutao.Win32.Kernel32;
namespace Snap.Hutao.Service.Game.Unlocker; namespace Snap.Hutao.Service.Game.Unlocker;
@@ -14,11 +15,16 @@ internal readonly struct RequiredLocalModule : IDisposable
public readonly Module UnityPlayer; public readonly Module UnityPlayer;
public readonly Module UserAssembly; public readonly Module UserAssembly;
public unsafe RequiredLocalModule(in HMODULE unityPlayer, in HMODULE userAssembly) [SuppressMessage("", "SH002")]
public RequiredLocalModule(HMODULE unityPlayer, HMODULE userAssembly)
{ {
// Align the pointer
unityPlayer = (nint)(unityPlayer & ~0x3L);
userAssembly = (nint)(userAssembly & ~0x3L);
HasValue = true; HasValue = true;
UnityPlayer = new((nuint)(nint)unityPlayer, ((IMAGE_NT_HEADERS64*)((IMAGE_DOS_HEADER*)(nint)unityPlayer)->e_lfanew)->OptionalHeader.SizeOfImage); UnityPlayer = new((nuint)(nint)unityPlayer, GetImageSize(unityPlayer));
UserAssembly = new((nuint)(nint)userAssembly, ((IMAGE_NT_HEADERS64*)((IMAGE_DOS_HEADER*)(nint)userAssembly)->e_lfanew)->OptionalHeader.SizeOfImage); UserAssembly = new((nuint)(nint)userAssembly, GetImageSize(userAssembly));
} }
public void Dispose() public void Dispose()
@@ -26,4 +32,11 @@ internal readonly struct RequiredLocalModule : IDisposable
FreeLibrary((nint)UnityPlayer.Address); FreeLibrary((nint)UnityPlayer.Address);
FreeLibrary((nint)UserAssembly.Address); FreeLibrary((nint)UserAssembly.Address);
} }
[SuppressMessage("", "SH002")]
[MethodImpl(MethodImplOptions.AggressiveInlining)]
private unsafe uint GetImageSize(HMODULE hModule)
{
return ((IMAGE_NT_HEADERS64*)((IMAGE_DOS_HEADER*)(nint)hModule)->e_lfanew)->OptionalHeader.SizeOfImage;
}
} }