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

merge in upstream player resource reloading

Browse source
This commit is contained in:
Adam 2020-09-02 22:54:21 +10:00
parent 0e7650f89b
commit 36250d5e54
4 changed files with 65 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

193
Penumbra/Penumbra.cs
View file

@ -1,193 +0,0 @@
using System;
using System.IO;
using System.Runtime.InteropServices;
using System.Text;
using Dalamud.Hooking;
using Dalamud.Plugin;
using Penumbra.Structs;
using Penumbra.Util;
using FileMode = Penumbra.Structs.FileMode;
using Penumbra.Extensions;
namespace Penumbra
{
public class Penumbra : IDisposable
{
public Plugin Plugin { get; set; }
public bool IsEnabled { get; set; }
public Crc32 Crc32 { get; }
// Delegate prototypes
public unsafe delegate byte ReadFilePrototype( IntPtr pFileHandler, SeFileDescriptor* pFileDesc, int priority, bool isSync );
public unsafe delegate byte ReadSqpackPrototype( IntPtr pFileHandler, SeFileDescriptor* pFileDesc, int priority, bool isSync );
public unsafe delegate void* GetResourceSyncPrototype( IntPtr pFileManager, uint* pCategoryId, char* pResourceType,
uint* pResourceHash, char* pPath, void* pUnknown );
public unsafe delegate void* GetResourceAsyncPrototype( IntPtr pFileManager, uint* pCategoryId, char* pResourceType,
uint* pResourceHash, char* pPath, void* pUnknown, bool isUnknown );
// Hooks
public Hook< GetResourceSyncPrototype > GetResourceSyncHook { get; private set; }
public Hook< GetResourceAsyncPrototype > GetResourceAsyncHook { get; private set; }
public Hook< ReadSqpackPrototype > ReadSqpackHook { get; private set; }
// Unmanaged functions
public ReadFilePrototype ReadFile { get; private set; }
public Penumbra( Plugin plugin )
{
Plugin = plugin;
Crc32 = new Crc32();
}
public unsafe void Init()
{
var scanner = Plugin.PluginInterface.TargetModuleScanner;
var readFileAddress =
scanner.ScanText( "E8 ?? ?? ?? ?? 84 C0 0F 84 ?? 00 00 00 4C 8B C3 BA 05" );
var readSqpackAddress =
scanner.ScanText( "E8 ?? ?? ?? ?? EB 05 E8 ?? ?? ?? ?? 84 C0 0F 84 ?? 00 00 00 4C 8B C3" );
var getResourceSyncAddress =
scanner.ScanText( "E8 ?? ?? 00 00 48 8D 4F ?? 48 89 87 ?? ?? 00 00" );
var getResourceAsyncAddress =
scanner.ScanText( "E8 ?? ?? ?? 00 48 8B D8 EB ?? F0 FF 83 ?? ?? 00 00" );
ReadSqpackHook = new Hook< ReadSqpackPrototype >( readSqpackAddress, new ReadSqpackPrototype( ReadSqpackHandler ) );
GetResourceSyncHook = new Hook< GetResourceSyncPrototype >( getResourceSyncAddress,
new GetResourceSyncPrototype( GetResourceSyncHandler ) );
GetResourceAsyncHook = new Hook< GetResourceAsyncPrototype >( getResourceAsyncAddress,
new GetResourceAsyncPrototype( GetResourceAsyncHandler ) );
ReadFile = Marshal.GetDelegateForFunctionPointer< ReadFilePrototype >( readFileAddress );
}
public unsafe void* GetResourceSyncHandler( IntPtr pFileManager, uint* pCategoryId,
char* pResourceType, uint* pResourceHash, char* pPath, void* pUnknown )
{
return GetResourceHandler( true, pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown, false );
}
public unsafe void* GetResourceAsyncHandler( IntPtr pFileManager, uint* pCategoryId,
char* pResourceType, uint* pResourceHash, char* pPath, void* pUnknown, bool isUnknown )
{
return GetResourceHandler( false, pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown, isUnknown );
}
private unsafe void* GetResourceHandler( bool isSync, IntPtr pFileManager, uint* pCategoryId,
char* pResourceType, uint* pResourceHash, char* pPath, void* pUnknown, bool isUnknown )
{
var gameFsPath = Marshal.PtrToStringAnsi( new IntPtr( pPath ) );
var candidate = Plugin.ModManager.GetCandidateForGameFile( gameFsPath );
// path must be < 260 because statically defined array length :(
if( candidate == null || candidate.FullName.Length >= 260 || !candidate.Exists )
{
return isSync
? GetResourceSyncHook.Original( pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown )
: GetResourceAsyncHook.Original( pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown, isUnknown );
}
var cleanPath = candidate.FullName.Replace( '\\', '/' );
var asciiPath = Encoding.ASCII.GetBytes( cleanPath );
var bPath = stackalloc byte[asciiPath.Length + 1];
Marshal.Copy( asciiPath, 0, new IntPtr( bPath ), asciiPath.Length );
pPath = ( char* )bPath;
Crc32.Init();
Crc32.Update( asciiPath );
*pResourceHash = Crc32.Checksum;
return isSync
? GetResourceSyncHook.Original( pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown )
: GetResourceAsyncHook.Original( pFileManager, pCategoryId, pResourceType,
pResourceHash, pPath, pUnknown, isUnknown );
}
public unsafe byte ReadSqpackHandler( IntPtr pFileHandler, SeFileDescriptor* pFileDesc, int priority, bool isSync )
{
var gameFsPath = Marshal.PtrToStringAnsi( new IntPtr( pFileDesc->ResourceHandle->FileName ) );
var isRooted = Path.IsPathRooted( gameFsPath );
if( gameFsPath == null || gameFsPath.Length >= 260 || !isRooted )
{
return ReadSqpackHook.Original( pFileHandler, pFileDesc, priority, isSync );
}
#if DEBUG
PluginLog.Log( "loading modded file: {GameFsPath}", gameFsPath );
#endif
pFileDesc->FileMode = FileMode.LoadUnpackedResource;
var utfPath = Encoding.Unicode.GetBytes( gameFsPath );
Marshal.Copy( utfPath, 0, new IntPtr( &pFileDesc->UtfFileName ), utfPath.Length );
var fd = stackalloc byte[0x20 + utfPath.Length + 0x16];
Marshal.Copy( utfPath, 0, new IntPtr( fd + 0x21 ), utfPath.Length );
pFileDesc->FileDescriptor = fd;
return ReadFile( pFileHandler, pFileDesc, priority, isSync );
}
public void Enable()
{
if( IsEnabled )
return;
ReadSqpackHook.Enable();
GetResourceSyncHook.Enable();
GetResourceAsyncHook.Enable();
IsEnabled = true;
}
public void Disable()
{
if( !IsEnabled )
return;
ReadSqpackHook.Disable();
GetResourceSyncHook.Disable();
GetResourceAsyncHook.Disable();
IsEnabled = false;
}
public void Dispose()
{
if( IsEnabled )
Disable();
ReadSqpackHook.Dispose();
GetResourceSyncHook.Dispose();
GetResourceAsyncHook.Dispose();
}
}
}