update fastfile load code

This commit is contained in:
quaK 2024-01-21 03:29:01 +02:00
parent 3ffcd854db
commit f45b25dde2

View File

@ -112,24 +112,14 @@ namespace fastfiles
namespace zone_loading
{
utils::hook::detour db_is_patch_hook;
utils::hook::detour db_read_stream_file_hook;
utils::hook::detour db_auth_load_inflate_hook;
utils::hook::detour db_auth_load_inflate_init_hook;
utils::hook::detour db_auth_load_inflate_end_hook;
utils::hook::detour db_patch_mem_fix_stream_alignment_hook;
game::db_z_stream_s db_zlib_stream;
char db_zlib_memory[0x20000];
bool is_reading_stream_file;
bool check_missing_content_func(const char* zone_name)
{
const char* lang_code = game::SEH_GetCurrentLanguageCode();
char buffer[0x100]{ 0 };
sprintf_s(buffer, "%s_", lang_code);
const auto len = sprintf_s(buffer, "%s_", lang_code);
if (!strncmp(zone_name, buffer, strlen(buffer)))
if (!strncmp(zone_name, buffer, len))
{
printf("Tried to load missing language zone: %s\n", zone_name);
return true;
@ -168,93 +158,6 @@ namespace fastfiles
a.and_(edi, r12d);
a.jmp(0x3BAC06_b);
}
void db_read_stream_file_stub(int a1, int a2)
{
if (!game::Sys_IsDatabaseThread()) // sanity check
{
__debugbreak();
}
is_reading_stream_file = true;
db_read_stream_file_hook.invoke<void>(a1, a2);
is_reading_stream_file = false;
}
template <class T1, class T2>
void stream_copy(T1 dest, T2 src)
{
dest->next_in = static_cast<decltype(dest->next_in)>(src->next_in);
dest->avail_in = static_cast<decltype(dest->avail_in)>(src->avail_in);
dest->total_in = static_cast<decltype(dest->total_in)>(src->total_in);
dest->next_out = static_cast<decltype(dest->next_out)>(src->next_out);
dest->avail_out = static_cast<decltype(dest->avail_out)>(src->avail_out);
dest->total_out = static_cast<decltype(dest->total_out)>(src->total_out);
}
bool is_custom_zone_read;
__int64 db_auth_load_inflate_stub(game::DB_ReadStream* stream)
{
if (!is_custom_zone_read) return db_auth_load_inflate_hook.invoke<__int64>(stream);
__int64 result{};
stream_copy(&db_zlib_stream, stream);
if (db_zlib_stream.avail_in)
{
result = game::db_inflate(&db_zlib_stream, Z_SYNC_FLUSH);
if (result == 1 && !db_zlib_stream.avail_out) result = 0;
}
stream_copy(stream, &db_zlib_stream);
return result;
}
void db_auth_load_inflate_init_stub(game::DB_ReadStream* stream, int isSecure, const char* filename)
{
is_custom_zone_read = false;
db_auth_load_inflate_init_hook.invoke<void>(stream, isSecure, filename);
if (stream->avail_in >= 4 &&
(stream->next_in[1] == 'I' && stream->next_in[2] == 'W' && stream->next_in[3] == 'C'))
{
return;
}
if (is_reading_stream_file || game::g_load->file->isSecure) return;
is_custom_zone_read = true;
memset(&db_zlib_stream, 0, sizeof(game::db_z_stream_s));
stream_copy(&db_zlib_stream, stream);
game::DB_Zip_InitThreadMemory(&db_zlib_stream, &db_zlib_memory, 0x20000);
game::db_inflateInit_(&db_zlib_stream, "1.1.4", 88);
stream_copy(stream, &db_zlib_stream);
}
void db_auth_load_inflate_end_stub()
{
db_auth_load_inflate_end_hook.invoke<void>();
if (!is_custom_zone_read) return;
game::db_inflateEnd(&db_zlib_stream);
game::DB_Zip_ShutdownThreadMemory(&db_zlib_stream);
is_custom_zone_read = false;
}
unsigned __int64 db_patch_mem_fix_stream_alignment_stub(int alignment) // probably unneeded
{
if (!is_custom_zone_read) return db_patch_mem_fix_stream_alignment_hook.invoke<unsigned __int64>(alignment);
*game::g_streamPos = (~alignment & (alignment + *game::g_streamPos));
return *game::g_streamPos;
}
}
using namespace zone_loading;
@ -298,11 +201,9 @@ namespace fastfiles
// Allow loading of unsigned fastfiles
utils::hook::set<uint8_t>(0x9E8CAE_b, 0xEB); // DB_InflateInit
db_read_stream_file_hook.create(0x0A7F370_b, db_read_stream_file_stub);
db_auth_load_inflate_hook.create(0x9E6AE0_b, db_auth_load_inflate_stub);
db_auth_load_inflate_init_hook.create(0x9E6970_b, db_auth_load_inflate_init_stub);
db_auth_load_inflate_end_hook.create(0x9E8CF0_b, db_auth_load_inflate_end_stub);
db_patch_mem_fix_stream_alignment_hook.create(0x0A0A80_b, db_patch_mem_fix_stream_alignment_stub);
// Skip signature validation
utils::hook::set(0x1409E6390, 0xC301B0); // signature
command::add("loadzone", [](const command::params& params)
{