CrazyRedMachine/popnhax
1
0
Fork 0
forked from Popn_Tools/popnhax
Code Issues Pull Requests Packages Projects Releases Wiki Activity

replace fuzzy search with Boyer-Moore

Browse Source
This commit is contained in:
CrazyRedMachine 2023-07-09 21:34:07 +02:00
parent 273bb4b69a
commit 2127e8c928
7 changed files with 227 additions and 493 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

476
popnhax/dllmain.cc
View File

@ -12,7 +12,7 @@
#define F_OK 0
#define access _access
#include "util/fuzzy_search.h"
#include "util/search.h"
#include "minhook/hde32.h"
#include "minhook/include/MinHook.h"
@ -737,12 +737,7 @@ static bool patch_purelong()
char *data = getDllData(g_game_dll_fn, &dllSize);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x80\x1A\x06\x00\x83\xFA\x08\x77\x08", 9)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x80\x1A\x06\x00\x83\xFA\x08\x77\x08", 9, 0);
if (pattern_offset == -1) {
LOG("popnhax: Couldn't find score increment function\n");
return false;
@ -773,12 +768,7 @@ static bool patch_datecode(char *datecode) {
g_datecode_override = strdup(datecode);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8D\x44\x24\x10\x88\x4C\x24\x10\x88\x5C\x24\x11\x8D\x50\x01", 15)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x8D\x44\x24\x10\x88\x4C\x24\x10\x88\x5C\x24\x11\x8D\x50\x01", 15, 0);
if (pattern_offset != -1) {
uint64_t patch_addr = (int64_t)data + pattern_offset + 0x08;
MH_CreateHook((LPVOID)patch_addr, (LPVOID)asm_patch_datecode,
@ -801,12 +791,7 @@ static bool patch_datecode(char *datecode) {
DWORD avsdllSize = 0;
char *avsdata = getDllData("libavs-win32.dll", &avsdllSize);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x57\x56\x89\x34\x24\x8B\xF2\x8B\xD0\x0F\xB6\x46\x2E", 13)
int64_t pattern_offset = find_block(avsdata, avsdllSize, &task, 0);
int64_t pattern_offset = search(avsdata, avsdllSize, "\x57\x56\x89\x34\x24\x8B\xF2\x8B\xD0\x0F\xB6\x46\x2E", 13, 0);
if (pattern_offset != -1) {
uint64_t patch_addr = (int64_t)avsdata + pattern_offset;
MH_CreateHook((LPVOID)patch_addr, (LPVOID)asm_patch_datecode_libavs,
@ -828,12 +813,7 @@ static bool patch_database(uint8_t force_unlocks) {
patch_purelong();
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8D\x44\x24\x10\x88\x4C\x24\x10\x88\x5C\x24\x11\x8D\x50\x01", 15)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x8D\x44\x24\x10\x88\x4C\x24\x10\x88\x5C\x24\x11\x8D\x50\x01", 15, 0);
if (pattern_offset != -1) {
uint64_t patch_addr = (int64_t)data + pattern_offset;
MH_CreateHook((LPVOID)patch_addr, (LPVOID)omnimix_patch_jbx,
@ -1125,34 +1105,18 @@ static bool patch_unset_volume() {
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
int64_t first_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x04\x00\x81\xC4\x00\x01\x00\x00\xC3\xCC", 10)
first_loc = find_block(data, dllSize, &task, 0);
int64_t first_loc = search(data, dllSize, "\x04\x00\x81\xC4\x00\x01\x00\x00\xC3\xCC", 10, 0);
if (first_loc == -1) {
return false;
}
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83", 1)
int64_t pattern_offset = find_block(data, 0x10, &task, first_loc);
int64_t pattern_offset = search(data, 0x10, "\x83", 1, first_loc);
if (pattern_offset == -1) {
return false;
}
uint64_t patch_addr = (int64_t)data + pattern_offset;
patch_memory(patch_addr, (char *)"\xC3", 1);
}
LOG("popnhax: windows volume untouched\n");
@ -1175,34 +1139,18 @@ static bool patch_remove_timer() {
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
int64_t first_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8B\xAC\x24\x68\x01", 5)
first_loc = find_block(data, dllSize, &task, 0);
int64_t first_loc = search(data, dllSize, "\x8B\xAC\x24\x68\x01", 5, 0);
if (first_loc == -1) {
return false;
}
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x0F", 1)
int64_t pattern_offset = find_block(data, 0x15, &task, first_loc);
int64_t pattern_offset = search(data, 0x15, "\x0F", 1, first_loc);
if (pattern_offset == -1) {
return false;
}
uint64_t patch_addr = (int64_t)data + pattern_offset;
patch_memory(patch_addr, (char *)"\x90\xE9", 2);
}
LOG("popnhax: timer removed\n");
@ -1224,27 +1172,13 @@ static bool patch_skip_tutorials() {
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
int64_t first_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xFD\xFF\x5E\xC2\x04\x00\xE8", 7)
first_loc = find_block(data, dllSize, &task, 0);
int64_t first_loc = search(data, dllSize, "\xFD\xFF\x5E\xC2\x04\x00\xE8", 7, 0);
if (first_loc == -1) {
return false;
}
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x74", 1)
int64_t pattern_offset = find_block(data, 0x10, &task, first_loc);
int64_t pattern_offset = search(data, 0x10, "\x74", 1, first_loc);
if (pattern_offset == -1) {
return false;
}
@ -1254,12 +1188,7 @@ static bool patch_skip_tutorials() {
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x66\x85\xC0\x75\x5E\x6A", 6)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x66\x85\xC0\x75\x5E\x6A", 6, 0);
if (pattern_offset == -1) {
return false;
}
@ -1269,12 +1198,7 @@ static bool patch_skip_tutorials() {
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x00\x5F\x5E\x66\x83\xF8\x01\x75", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x00\x5F\x5E\x66\x83\xF8\x01\x75", 8, 0);
if (pattern_offset == -1) {
return false;
}
@ -1307,18 +1231,14 @@ bool force_unlock_songs() {
int music_unlocks = 0, chart_unlocks = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x69\xC0\xAC\x00\x00\x00\x8B\x80", 8) // 0xac here is the size of music_entry. May change in the future
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
// 0xac here is the size of music_entry. May change in the future
int64_t pattern_offset = search(data, dllSize, "\x69\xC0\xAC\x00\x00\x00\x8B\x80", 8, 0);
if (pattern_offset == -1) {
LOG("popnhax: couldn't unlock songs and charts\n");
return false;
}
uint32_t buffer_offset = *(uint32_t*)((int64_t)data + pattern_offset + task.blocks[0].data[0].length);
uint32_t buffer_offset = *(uint32_t*)((int64_t)data + pattern_offset + 8);
buffer_offset -= 0x1c; // The difference between music_entry.mask and music_entry.fw_genre_ptr to put it at the beginning of the entry
music_entry *entry = (music_entry*)buffer_offset;
@ -1361,18 +1281,14 @@ bool force_unlock_charas() {
int chara_unlocks = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x98\x6B\xC0\x4C\x8B\x80", 6) // 0x4c here is the size of character_entry. May change in the future
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
// 0x4c here is the size of character_entry. May change in the future
int64_t pattern_offset = search(data, dllSize, "\x98\x6B\xC0\x4C\x8B\x80", 6, 0);
if (pattern_offset == -1) {
LOG("popnhax: couldn't unlock characters\n");
return false;
}
uint32_t buffer_offset = *(uint32_t*)((int64_t)data + pattern_offset + task.blocks[0].data[0].length);
uint32_t buffer_offset = *(uint32_t*)((int64_t)data + pattern_offset + 6);
buffer_offset -= 0x48; // The difference between character_entry.game_version and character_entry.chara_id_ptr to put it at the beginning of the entry
character_entry *entry = (character_entry*)buffer_offset;
@ -1413,12 +1329,7 @@ static bool patch_unlocks_offline() {
char *data = getDllData(g_game_dll_fn, &dllSize);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xB8\x49\x06\x00\x00\x66\x3B", 7)
int64_t pattern_offset = find_block(data, dllSize-0xE0000, &task, 0xE0000);
int64_t pattern_offset = search(data, dllSize-0xE0000, "\xB8\x49\x06\x00\x00\x66\x3B", 7, 0xE0000);
if (pattern_offset == -1) {
LOG("Couldn't find first song unlock\n");
return false;
@ -1466,11 +1377,7 @@ static bool get_addr_icca(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xE8\x4B\x14\x00\x00\x84\xC0\x74\x03\x33\xC0\xC3\x8B\x0D", 14)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xE8\x4B\x14\x00\x00\x84\xC0\x74\x03\x33\xC0\xC3\x8B\x0D", 14, 0);
if (pattern_offset == -1) {
return false;
}
@ -1497,12 +1404,7 @@ static bool get_addr_timing_offset(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xB8\xB4\xFF\xFF\xFF", 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xB8\xB4\xFF\xFF\xFF", 5, 0);
if (pattern_offset == -1) {
return false;
}
@ -1530,11 +1432,7 @@ static bool get_addr_beam_brightness(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xB8\x64\x00\x00\x00\xD9", 6)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xB8\x64\x00\x00\x00\xD9", 6, 0);
if (pattern_offset == -1) {
return false;
}
@ -1561,11 +1459,7 @@ static bool get_addr_sd_timing(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xB8\xC4\xFF\xFF\xFF", 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xB8\xC4\xFF\xFF\xFF", 5, 0);
if (pattern_offset == -1) {
return false;
}
@ -1592,11 +1486,7 @@ static bool get_addr_hd_timing(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xB8\xB4\xFF\xFF\xFF", 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xB8\xB4\xFF\xFF\xFF", 5, 0);
if (pattern_offset == -1) {
return false;
}
@ -1637,20 +1527,11 @@ static bool patch_hidden_is_offset()
patch_memory(hiddencommitoptionaddr+placeholder_offset+1, eax_to_offset, 5);
/* find option commit function (unilab) */
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x03\xC7\x8D\x44\x01\x2A\x89\x10", 8)
uint8_t shift = 6;
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x03\xC7\x8D\x44\x01\x2A\x89\x10", 8, 0);
if (pattern_offset == -1) {
/* wasn't found, look for older function */
fuzzy_search_task fallbacktask;
FUZZY_START(fallbacktask, 1)
FUZZY_CODE(fallbacktask, 0, "\x0F\xB6\xC3\x03\xCF\x8D", 6)
pattern_offset = find_block(data, dllSize, &fallbacktask, 0);
pattern_offset = search(data, dllSize, "\x0F\xB6\xC3\x03\xCF\x8D", 6, 0);
shift = 14;
if (pattern_offset == -1) {
@ -1673,12 +1554,7 @@ static bool patch_hidden_is_offset()
uint32_t *cast_code = (uint32_t*) &set_offset_fun[1];
*cast_code = timing_addr;
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, set_offset_fun, 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, set_offset_fun, 5, 0);
if (pattern_offset == -1) {
LOG("popnhax: hidden is offset: cannot find offset update function\n");
return false;
@ -1701,27 +1577,12 @@ static bool patch_show_hidden_adjust_result_screen() {
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
int64_t first_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x6A\x00\x0F\xBE\xCB", 5)
first_loc = find_block(data, dllSize, &task, 0);
if (first_loc == -1) {
int64_t first_loc = search(data, dllSize, "\x6A\x00\x0F\xBE\xCB", 5, 0);
if (first_loc == -1)
return false;
}
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x80\xBC\x24", 3)
int64_t pattern_offset = find_block(data, 0x200, &task, first_loc);
int64_t pattern_offset = search(data, 0x200, "\x80\xBC\x24", 3, first_loc);
if (pattern_offset == -1) {
return false;
}
@ -1729,7 +1590,7 @@ static bool patch_show_hidden_adjust_result_screen() {
uint64_t hook_addr = (int64_t)data + pattern_offset;
MH_CreateHook((LPVOID)(hook_addr), (LPVOID)asm_show_hidden_result,
(void **)&real_show_hidden_result);
}
LOG("popnhax: show hidden/adjust value on result screen\n");
@ -1740,27 +1601,13 @@ static bool force_show_fast_slow() {
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
int64_t first_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x6A\x00\x0F\xBE\xCB", 5)
first_loc = find_block(data, dllSize, &task, 0);
int64_t first_loc = search(data, dllSize, "\x6A\x00\x0F\xBE\xCB", 5, 0);
if (first_loc == -1) {
return false;
}
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x0F\x85", 2)
int64_t pattern_offset = find_block(data, 0x50, &task, first_loc);
int64_t pattern_offset = search(data, 0x50, "\x0F\x85", 2, first_loc);
if (pattern_offset == -1) {
return false;
}
@ -1780,11 +1627,7 @@ static bool patch_pfree() {
/* stop stage counter (2 matches, 1st one is the good one) */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xF8\x04\x77\x3E", 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xF8\x04\x77\x3E", 5, 0);
if (pattern_offset == -1) {
LOG("couldn't find stop stage counter\n");
return false;
@ -1804,14 +1647,7 @@ static bool patch_pfree() {
char offset_from_stage1[2] = {0x00, 0x00};
int64_t child_fun_loc = 0;
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\x8D\x46\xFF\x83\xF8\x0A\x0F",
7)
int64_t offset = find_block(data, dllSize, &task, 0);
int64_t offset = search(data, dllSize, "\x8D\x46\xFF\x83\xF8\x0A\x0F", 7, 0);
if (offset == -1) {
#if DEBUG == 1
LOG("popnhax: pfree: failed to retrieve struct size and offset\n");
@ -1827,14 +1663,7 @@ static bool patch_pfree() {
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\xCB\x69",
2)
int64_t pattern_offset = find_block(data, 0x40, &task, child_fun_loc);
int64_t pattern_offset = search(data, 0x40, "\xCB\x69", 2, child_fun_loc);
if (pattern_offset == -1) {
LOG("popnhax: pfree: failed to retrieve offset from stage1 (child_fun_loc = %llx\n",child_fun_loc);
return false;
@ -1848,14 +1677,7 @@ static bool patch_pfree() {
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\x8d\x74\x01",
3)
int64_t pattern_offset = find_block(data, 0x40, &task, child_fun_loc);
int64_t pattern_offset = search(data, 0x40, "\x8d\x74\x01", 3, child_fun_loc);
if (pattern_offset == -1) {
LOG("popnhax: pfree: failed to retrieve offset from base\n");
return false;
@ -1871,14 +1693,7 @@ pfree_apply:
int64_t first_loc = 0;
/* cleanup score and stats part1 */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\xFE\x46\x0E\x80",
4)
first_loc = find_block(data, dllSize, &task, 0);
first_loc = search(data, dllSize, "\xFE\x46\x0E\x80", 4, 0);
if (first_loc == -1) {
LOG("popnhax: pfree: cannot find stage update function\n");
return false;
@ -1890,14 +1705,7 @@ pfree_apply:
/* cleanup score and stats part2 */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\x83\xC4\x08\x8A",
4)
int64_t pattern_offset = find_block(data, 0x40, &task, first_loc);
int64_t pattern_offset = search(data, 0x40, "\x83\xC4\x08\x8A", 4, first_loc);
if (pattern_offset == -1) {
LOG("popnhax: pfree: cannot find stage update function\n");
return false;
@ -1931,11 +1739,7 @@ static bool patch_quick_retire(bool pfree)
/* pfree already installs this hook
*/
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xF8\x04\x77\x3E", 5)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xF8\x04\x77\x3E", 5, 0);
if (pattern_offset == -1) {
LOG("couldn't find stop stage counter\n");
return false;
@ -1953,12 +1757,7 @@ static bool patch_quick_retire(bool pfree)
/* instant retire with numpad 9 in song */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x55\x8B\xEC\x83\xE4\xF8\x83\xEC\x08\x0F\xBF\x05", 12)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x55\x8B\xEC\x83\xE4\xF8\x83\xEC\x08\x0F\xBF\x05", 12, 0);
if (pattern_offset == -1) {
LOG("popnhax: cannot retrieve song loop\n");
@ -1982,12 +1781,7 @@ static bool patch_quick_retire(bool pfree)
/* instant exit with numpad 9 on result screen */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xF8\x53\x55\x56\x57\x8B\xE9\x8B\x75\x00\x8B\x85", 12)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xF8\x53\x55\x56\x57\x8B\xE9\x8B\x75\x00\x8B\x85", 12, 0);
if (pattern_offset == -1) {
LOG("popnhax: cannot retrieve result screen loop\n");
@ -2013,12 +1807,7 @@ static bool patch_quick_retire(bool pfree)
/* retrieve songstart function pointer for quick retry */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xE9\x0C\x01\x00\x00\x8B\x85\x10\x0A\x00\x00", 11)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xE9\x0C\x01\x00\x00\x8B\x85\x10\x0A\x00\x00", 11, 0);
if (pattern_offset == -1) {
LOG("popnhax: quick retry: cannot retrieve song start function\n");
@ -2031,12 +1820,7 @@ static bool patch_quick_retire(bool pfree)
/* instant retry (go back to option select) with numpad 7 */
{
/* retrieve current stage score addr for cleanup (also used to fix quick retire medal) */
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xF3\xA5\x5F\x5E\x5B\xC2\x04\x00", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xF3\xA5\x5F\x5E\x5B\xC2\x04\x00", 8, 0);
if (pattern_offset == -1) {
LOG("popnhax: quick retry: cannot retrieve score addr\n");
@ -2049,12 +1833,7 @@ static bool patch_quick_retire(bool pfree)
}
{
/* hook quick retire transition to go back to option select instead */
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x6A\x01\x8B\xCE\xFF\xD0\xE8", 7)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x6A\x01\x8B\xCE\xFF\xD0\xE8", 7, 0);
if (pattern_offset == -1) {
LOG("popnhax: quick retry: cannot retrieve screen transition function\n");
@ -2068,12 +1847,7 @@ static bool patch_quick_retire(bool pfree)
/* instant launch song with numpad 7 on option select (hold 7 during song for quick retry) */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x51\x50\x8B\x83\x0C\x0A\x00\x00\xEB\x09\x33\xD2", 12)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x51\x50\x8B\x83\x0C\x0A\x00\x00\xEB\x09\x33\xD2", 12, 0);
if (pattern_offset == -1) {
LOG("popnhax: quick retry: cannot retrieve option screen loop\n");
@ -2155,12 +1929,7 @@ static bool patch_keysound_offset(int8_t value)
patch_add_to_base_offset(value);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x51\x53\x56\x57\x0f\xb7\xf8\x8b\x34\xfd", 10)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x51\x53\x56\x57\x0f\xb7\xf8\x8b\x34\xfd", 10, 0);
if (pattern_offset == -1) {
LOG("popnhax: keysound offset: cannot prepatch\n");
return false;
@ -2304,12 +2073,8 @@ static bool patch_score_challenge()
/* Part1: retrieve course id and song id, useful and will simplify a little */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x81\xC6\xCC\x08\x00\x00\xC7\x44\x24", 9)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x81\xC6\xCC\x08\x00\x00\xC7\x44\x24", 9, 0);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find course/song address\n");
return false;
@ -2323,12 +2088,7 @@ static bool patch_score_challenge()
/* Part2: retrieve subfunctions which used to be called by the now stubbed function */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x66\x89\x08\x88\x50\x02", 6)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x66\x89\x08\x88\x50\x02", 6, 0);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find song data prep function\n");
return false;
@ -2339,12 +2099,7 @@ static bool patch_score_challenge()
score_challenge_prep_songdata = (void(*)())patch_addr;
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8B\x4F\x0C\x83\xEC\x10\x56\x85\xC9\x75\x04\x33\xC0\xEB\x08\x8B\x47\x14\x2B\xC1\xC1\xF8\x02\x8B\x77\x10\x8B\xD6\x2B\xD1\xC1\xFA\x02\x3B\xD0\x73\x2B", 37)
int64_t pattern_offset = find_block(data, dllSize-0x60000, &task, 0x60000);
int64_t pattern_offset = search(data, dllSize-0x60000, "\x8B\x4F\x0C\x83\xEC\x10\x56\x85\xC9\x75\x04\x33\xC0\xEB\x08\x8B\x47\x14\x2B\xC1\xC1\xF8\x02\x8B\x77\x10\x8B\xD6\x2B\xD1\xC1\xFA\x02\x3B\xD0\x73\x2B", 37, 0x60000);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find category song inject function\n");
return false;
@ -2355,12 +2110,7 @@ static bool patch_score_challenge()
score_challenge_song_inject = (void(*)())patch_addr;
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8B\x01\x8B\x50\x14\xFF\xE2\xC3\xCC\xCC\xCC\xCC", 12)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x8B\x01\x8B\x50\x14\xFF\xE2\xC3\xCC\xCC\xCC\xCC", 12, 0);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find check if logged function\n");
return false;
@ -2371,12 +2121,7 @@ static bool patch_score_challenge()
score_challenge_test_if_logged1 = (void(*)())patch_addr;
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xF7\xD8\x1B\xC0\x40\xC3\xE8", 7)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xF7\xD8\x1B\xC0\x40\xC3\xE8", 7, 0);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find check if logged function\n");
return false;
@ -2389,12 +2134,7 @@ static bool patch_score_challenge()
/* Part3: "unstub" the score challenge category creation */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xF8\x10\x77\x75\xFF\x24\x85", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xF8\x10\x77\x75\xFF\x24\x85", 8, 0);
if (pattern_offset == -1) {
LOG("popnhax: score challenge: cannot find category building loop\n");
return false;
@ -2526,12 +2266,7 @@ static bool patch_options()
char *data = getDllData(g_game_dll_fn, &dllSize);
/* starting value */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xFF\xD0\xB8\x01\x00\x00\x00\x01\x46\x34\x01\x46\x38\xC3\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x8B\x4C\x24\x04\x8B\x54\x24\x08\x89\x48\x20\x89\x50\x24\xC7\x40\x2C\x00\x00\x00\x00\xC6\x40\x30\x01\xC2\x08\x00\xCC\xCC\xCC\xCC\x83\xEC\x18\x33\xC0", 60)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xFF\xD0\xB8\x01\x00\x00\x00\x01\x46\x34\x01\x46\x38\xC3\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x8B\x4C\x24\x04\x8B\x54\x24\x08\x89\x48\x20\x89\x50\x24\xC7\x40\x2C\x00\x00\x00\x00\xC6\x40\x30\x01\xC2\x08\x00\xCC\xCC\xCC\xCC\x83\xEC\x18\x33\xC0", 60, 0);
if (pattern_offset == -1) {
LOG("popnhax: always full options: cannot find function call\n");
return false;
@ -2545,12 +2280,7 @@ static bool patch_options()
}
/* prevent switching with numpad 0 */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xC6\x85\x1F\x0A\x00\x00\x00\xC6\x85\x20\x0A\x00\x00\x00\xE9\x3E\x01\x00\x00\x33\xC9", 21)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xC6\x85\x1F\x0A\x00\x00\x00\xC6\x85\x20\x0A\x00\x00\x00\xE9\x3E\x01\x00\x00\x33\xC9", 21, 0);
if (pattern_offset == -1) {
LOG("popnhax: always full options: cannot find numpad0 check\n");
return false;
@ -2586,11 +2316,7 @@ static bool get_addr_pldata(uint32_t *res)
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x14\xFF\xE2\xC3\xCC\xCC", 6) // +10hからポインタ群がある
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x14\xFF\xE2\xC3\xCC\xCC", 6, 0); // +10hからポインタ群がある
if (pattern_offset == -1) {
return false;
}
@ -2611,12 +2337,7 @@ static bool unilab_check()
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x03\xC7\x8D\x44\x01\x2A\x89\x10", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x03\xC7\x8D\x44\x01\x2A\x89\x10", 8, 0);
if (pattern_offset == -1) {
return false;
}
@ -2631,11 +2352,7 @@ static bool get_addr_numkey()
DWORD dllSize = 0;
char *data = getDllData(g_game_dll_fn, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x85\xC9\x74\x08\x8B\x01\x8B\x40\x24\x52\xFF\xD0", 12)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x85\xC9\x74\x08\x8B\x01\x8B\x40\x24\x52\xFF\xD0", 12, 0);
if (pattern_offset == -1) {
return false;
}
@ -2657,12 +2374,7 @@ static bool get_addr_random()
char *data = getDllData(g_game_dll_fn, &dllSize); // hook \x83\xC4\x04\xB9\x02\x00\x00\x00 button \x03\xC5\x83\xF8\x09\x7C\xDE
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xC4\x04\xB9\x02\x00\x00\x00", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xC4\x04\xB9\x02\x00\x00\x00", 8, 0);
if (pattern_offset == -1) {
return false;
}
@ -2671,24 +2383,14 @@ static bool get_addr_random()
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x51\x55\x56\xC7\x44\x24\x08\x00\x00\x00", 10)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x51\x55\x56\xC7\x44\x24\x08\x00\x00\x00", 10, 0);
if (pattern_offset == -1) {
return false;
}
ran_func = (uint32_t *) ((int64_t)data + pattern_offset);
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x03\xC5\x83\xF8\x09\x7C\xDE", 7)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x03\xC5\x83\xF8\x09\x7C\xDE", 7, 0);
if (pattern_offset == -1) {
return false;
}
@ -2792,27 +2494,16 @@ static bool get_rendaddr()
char *data = getDllData(g_game_dll_fn, &dllSize);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0,
"\x3b\xC3\x74\x13\xC7\x00\x02\x00\x00\x00\x89\x58\x04\x89\x58\x08", 16)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x3b\xC3\x74\x13\xC7\x00\x02\x00\x00\x00\x89\x58\x04\x89\x58\x08", 16, 0);
if (pattern_offset == -1) {
return false;
}
g_rend_addr = (uint32_t *)((int64_t)data + pattern_offset -4);
font_color = (uint32_t *)((int64_t)data + pattern_offset +36);
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\xC3\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x8B\x4C\x24\x0C\x8D\x44\x24\x10", 24)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\xC3\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x8B\x4C\x24\x0C\x8D\x44\x24\x10", 24, 0);
if (pattern_offset == -1) {
return false;
}
@ -2845,12 +2536,7 @@ static bool get_speedaddr()
char *data = getDllData(g_game_dll_fn, &dllSize);
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xC4\x0C\x8B\xC3\x8D\x7C\x24", 8)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xC4\x0C\x8B\xC3\x8D\x7C\x24", 8, 0);
if (pattern_offset == -1) {
return false;
}
@ -2859,12 +2545,7 @@ static bool get_speedaddr()
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x8B\x74\x24\x18\x8D\x0C\x5B\x8B\x54\x8E\xF4", 11)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x8B\x74\x24\x18\x8D\x0C\x5B\x8B\x54\x8E\xF4", 11, 0);
if (pattern_offset == -1) {
return false;
}
@ -2873,12 +2554,7 @@ static bool get_speedaddr()
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x0F\xBF\xC5\x83\xC4\x0C\x66\x89\x6E\x08", 10)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x0F\xBF\xC5\x83\xC4\x0C\x66\x89\x6E\x08", 10, 0);
if (pattern_offset == -1) {
return false;
}
@ -3155,12 +2831,7 @@ static bool patch_practice_mode()
/* AGING MODE to Practice Mode */
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x83\xEC\x40\x53\x56\x57", 6)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x83\xEC\x40\x53\x56\x57", 6, 0);
if (pattern_offset == -1) {
LOG("popnhax: cannot retrieve aging loop\n");
@ -3217,12 +2888,7 @@ static bool patch_practice_mode()
(void **)&real_get_random);
}
{
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, "\x5E\x8B\xE5\x5D\xC2\x04\x00\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x55\x8B\xEC\x83\xE4\xF8\x51\x56\x8B\xF1\x8B", 32)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, "\x5E\x8B\xE5\x5D\xC2\x04\x00\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\xCC\x55\x8B\xEC\x83\xE4\xF8\x51\x56\x8B\xF1\x8B", 32, 0);
if (pattern_offset == -1) {
LOG("popnhax: Cannot find address for restore addr\n");
return false;
@ -3500,7 +3166,7 @@ BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserv
MH_EnableHook(MH_ALL_HOOKS);
LOG("popnhax: done patching game, enjoy!");
LOG("popnhax: done patching game, enjoy!\n");
if (g_log_fp != stderr)
fclose(g_log_fp);

69
popnhax/translation.cc
View File

@ -3,7 +3,7 @@
#include <io.h>
#include <windows.h>
#include "util/fuzzy_search.h"
#include "util/search.h"
#include "util/log.h"
#include "util/patch.h"
@ -14,7 +14,7 @@
#define MAX_REPLACE_SIZE 16384
FILE *g_dict_applied_fp;
bool patch_sjis(const char *dllFilename, const uint8_t *find, uint8_t find_size, int64_t *offset, uint8_t *replace, uint8_t replace_size, bool dump_dict) {
bool patch_sjis(const char *dllFilename, const char *find, uint8_t find_size, int64_t *offset, char *replace, uint8_t replace_size, bool dump_dict) {
static DWORD dllSize = 0;
static char *data = getDllData(dllFilename, &dllSize);
static uint8_t first_offset = 0;
@ -23,12 +23,7 @@ bool patch_sjis(const char *dllFilename, const uint8_t *find, uint8_t find_size,
uint64_t patch_addr;
bool valid_sjis = false;
do {
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, find, find_size)
*offset = find_block(data, dllSize-*offset-0x100000, &task, *offset);
*offset = search(data, dllSize-*offset, find, find_size, *offset);
if (*offset == -1) {
*offset = offset_orig;
return false;
@ -41,10 +36,6 @@ bool patch_sjis(const char *dllFilename, const uint8_t *find, uint8_t find_size,
LOG("popnhax: dump_dict: dump applied patches in dict_applied.txt\n");
g_dict_applied_fp = fopen("dict_applied.txt","wb");
}
/* limit search to a 0x100000-wide zone starting from first string found to speedup the process
* make sure to put the first string first (usually 種類順)
*/
dllSize = *offset + 0x100000;
first_offset = 1;
}
@ -133,9 +124,9 @@ static void perform_reloc(char *data, int32_t delta, uint32_t ext_base, uint32_t
*p += delta;
if ( ext_base > 0 && *p >= ((int64_t)data+ext_base) )
{
fprintf(stderr,"reloc rva %lx to ext ", *p);
//fprintf(stderr,"reloc rva %lx to ext ", *p);
*p += ext_delta;
fprintf(stderr," %lx\n", *p);
//fprintf(stderr," %lx\n", *p);
}
VirtualProtect((LPVOID)p, 4, old_prot, &old_prot);
}
@ -169,10 +160,6 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
DWORD dllSize = 0;
char *data = getDllData(dllFilename, &dllSize);
/* some of the patches might require fixups */
PIMAGE_NT_HEADERS headers = (PIMAGE_NT_HEADERS)((int64_t)data + ((PIMAGE_DOS_HEADER)data)->e_lfanew);
DWORD_PTR reloc_delta = (DWORD_PTR)((int64_t)data - headers->OptionalHeader.ImageBase);
char dict_filepath[64];
sprintf(dict_filepath, "%s%s%s", "data_mods\\", foldername, "\\popn22.ips");
FILE *ips_fp = fopen(dict_filepath, "rb");
@ -184,18 +171,7 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
LOG("popnhax: translation: popn22.ips found\n");
if (dump_dll)
{
LOG("popnhax: translation debug: dump dll before patch\n");
FILE* dllrtp = fopen("dllruntime.dll", "wb");
fwrite(data, 1, dllSize, dllrtp);
fclose(dllrtp);
LOG("popnhax: translation debug: dllruntime.dll generated\n");
}
perform_reloc(data, -1*reloc_delta, 0, 0);
//check header
/* check .ips header */
uint8_t buffer[8];
if (fread(&buffer, 1, 5, ips_fp) != 5)
return false;
@ -206,9 +182,24 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
return false;
}
if (dump_dll)
{
LOG("popnhax: translation debug: dump dll before patch\n");
FILE* dllrtp = fopen("dllruntime.dll", "wb");
fwrite(data, 1, dllSize, dllrtp);
fclose(dllrtp);
LOG("popnhax: translation debug: dllruntime.dll generated\n");
}
/* undo all relocation so you can apply ips patch with correct values, we'll reapply them later */
PIMAGE_NT_HEADERS headers = (PIMAGE_NT_HEADERS)((int64_t)data + ((PIMAGE_DOS_HEADER)data)->e_lfanew);
DWORD_PTR reloc_delta = (DWORD_PTR)((int64_t)data - headers->OptionalHeader.ImageBase);
perform_reloc(data, -1*reloc_delta, 0, 0);
uint32_t trans_base = 0; /* eclale patch adds new section header which I'm relocating */
uint32_t trans_base_offset = 0;
uint32_t trans_rebase = 0;
uint32_t offset = 0;
uint16_t size = 0;
uint16_t count = 0;
@ -218,7 +209,7 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
{
bool skip = false;
/* need to convert offset to rva before applying patch */
/* need to convert offset to rva before applying patch since the dll is already in memory */
uint8_t *bp = (uint8_t *)&offset;
offset = BYTE3_TO_UINT(bp);
uint32_t rva;
@ -250,7 +241,7 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
IPS_READ(&value, 1, "RLE VALUE");
LOG("RLE PATCH! size %d value %d\n", size, value);
//LOG("RLE PATCH! size %d value %d\n", size, value);
//fprintf(stderr, "rle value %d (%d bytes)\n",value, size);
if ( size > MAX_REPLACE_SIZE )
{
@ -299,15 +290,11 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
}
}
// if ( rva+size > dllSize
//LOG( "%02x %02x %02x %02x...\n", replace[0], replace[1], replace[2], replace[3]);
if ( trans_base_offset != 0 && offset >= trans_base_offset )
{
//LOG( "WARNING: patching into new section.\n");
/* patching into new section */
if ( trans_rebase == 0 )
{
//LOG( "ALLOCATING NEW SECTION FOR TRANS\n");
HANDLE hProc = GetCurrentProcess();
LPVOID myAlloc = VirtualAllocEx(hProc, NULL, 16384, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
if (myAlloc == NULL)
@ -317,6 +304,8 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
}
trans_rebase = (uint32_t)myAlloc - (uint32_t)data;
//LOG( "virtualalloc worked; address %x (%p)\n", trans_rebase, myAlloc);
/* seems useless */
//memcpy(replace, &trans_rebase, 4);
//patch_memory((int64_t)data+0x02d4, (char *)replace, 4);
//LOG( "patched .trans section address to %02x %02x %02x %02x\n", replace[0], replace[1], replace[2], replace[3]);
@ -332,6 +321,7 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
}
}
/* redo all relocation now the dll is patched */
perform_reloc(data, reloc_delta, trans_base, trans_rebase-trans_base);
LOG("popnhax: translation: IPS patch applied.\n");
@ -351,8 +341,8 @@ static bool patch_translation_ips(const char *dllFilename, const char *foldernam
static bool patch_translation_dict(const char *dllFilename, const char *foldername, bool dump_dict)
{
uint8_t original[128];
uint8_t translation[128];
char original[128];
char translation[128];
uint8_t buffer;
int64_t curr_offset = 0;
uint8_t word_count = 0;
@ -471,7 +461,6 @@ bool patch_translate(const char *dllFilename, const char *folder, bool debug)
bool dict_done = false;
ips_done = patch_translation_ips(dllFilename, folder, debug);
dict_done = patch_translation_dict(dllFilename, folder, debug);
return ips_done || dict_done;

1
util/Module.mk
View File

@ -2,5 +2,6 @@ libs += util
srcpp_util := \
fuzzy_search.cc \
search.cc \
cmdline.cc \
patch.cc \

4
util/log.h
View File

@ -1,6 +1,7 @@
#ifndef __LOG_H__
#define __LOG_H__
#include <cstdio>
#include <stdbool.h>
#include <stddef.h>
#include <stdint.h>
@ -8,8 +9,11 @@
extern FILE *g_log_fp;
#define LOG(...) do { \
if (g_log_fp != stderr) { \
fprintf(g_log_fp, __VA_ARGS__); \
fflush(g_log_fp);\
}\
fprintf(stderr, __VA_ARGS__); \
} while (0)
#endif

16
util/patch.cc
View File

@ -3,7 +3,7 @@
#include <psapi.h>
// clang-format on
#include "util/fuzzy_search.h"
#include "util/search.h"
#include "patch.h"
void patch_memory(uint64_t patch_addr, char *data, size_t len) {
@ -88,12 +88,7 @@ void find_and_patch_string(const char *dllFilename, const char *input_string, co
char *data = getDllData(dllFilename, &dllSize);
while (1) {
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, input_string, strlen(input_string))
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, input_string, strlen(input_string), 0);
if (pattern_offset == -1) {
break;
}
@ -110,12 +105,7 @@ bool find_and_patch_hex(const char *dllFilename, const char *find, uint8_t find_
DWORD dllSize = 0;
char *data = getDllData(dllFilename, &dllSize);
fuzzy_search_task task;
FUZZY_START(task, 1)
FUZZY_CODE(task, 0, find, find_size)
int64_t pattern_offset = find_block(data, dllSize, &task, 0);
int64_t pattern_offset = search(data, dllSize, find, find_size, 0);
if (pattern_offset == -1) {
return false;
}

78
util/search.cc Normal file
View File

@ -0,0 +1,78 @@
#include <memory.h>
#include <stdint.h>
#include <cstdio>
#include "util/log.h"
#define NO_OF_CHARS 256
// A utility function to get maximum of two integers
static int max(int a, int b) {
return (a > b) ? a : b;
}
// The preprocessing function for Boyer Moore's bad character heuristic
static void badCharHeuristic(const unsigned char *str, int size, int* badchar) {
int i;
// Initialize all occurrences as -1
for (i = 0; i < NO_OF_CHARS; i++)
badchar[i] = -1;
// Fill the actual value of last occurrence of a character
for (i = 0; i < size; i++)
badchar[(int) str[i]] = i;
}
#define DEBUG_SEARCH 0
int _search(unsigned char *haystack, size_t haystack_size, const unsigned char *needle, size_t needle_size, int orig_offset, int debug) {
int badchar[NO_OF_CHARS];
badCharHeuristic(needle, needle_size, badchar);
int64_t s = 0; // s is shift of the pattern with respect to text
while (s <= (haystack_size - needle_size)) {
int j = needle_size - 1;
if (debug == 2)
{
LOG("--------------------------------\n");
LOG("txt...");
for (size_t i = 0; i < needle_size; i++)
{
LOG("%02x ", haystack[orig_offset+s+i]);
}
LOG("\n");
LOG("pat...");
for (size_t i = 0; i < needle_size; i++)
{
LOG("%02x ", needle[i]);
}
LOG("\n");
}
while (j >= 0 && needle[j] == haystack[orig_offset + s + j])
j--;
if (j < 0) {
//printf("\n pattern occurs at shift = %d", s);
//s += (s + needle_size < haystack_size) ? needle_size - badchar[txt[s + needle_size]] : 1;
if (debug)
LOG("found string at offset %llx!\n", orig_offset +s);
return orig_offset + s;
}
else
{
s += max(1, j - badchar[(int)haystack[orig_offset + s + j]]);
if (debug)
LOG("mismatch at pos %d, new offset %llx\n\n", j, orig_offset+s);
}
}
return -1;
}
int search(char *haystack, size_t haystack_size, const char *needle, size_t needle_size, size_t orig_offset) {
int res = _search((unsigned char*) haystack, haystack_size, (const unsigned char *)needle, needle_size, orig_offset, 0);
return res;
}

6
util/search.h Normal file
View File

@ -0,0 +1,6 @@
#ifndef __SEARCH_H__
#define __SEARCH_H__
int search(char *haystack, size_t haystack_size, const char *needle, size_t needle_size, size_t orig_offset);
#endif