mirror of
https://github.com/HIllya51/LunaHook.git
synced 2024-11-26 23:34:01 +08:00
456 lines
9.3 KiB
C++
456 lines
9.3 KiB
C++
|
|
namespace
|
|
{
|
|
std::optional<HookParam> ParseRCode(std::wstring RCode)
|
|
{
|
|
RCode.erase(0, 1);
|
|
std::wsmatch match;
|
|
HookParam hp;
|
|
hp.type |= DIRECT_READ;
|
|
|
|
// {S|Q|V|M}
|
|
switch (RCode[0])
|
|
{
|
|
case L'S':
|
|
break;
|
|
case L'Q':
|
|
hp.type |= CODEC_UTF16;
|
|
break;
|
|
case L'U':
|
|
hp.type |= CODEC_UTF32;
|
|
break;
|
|
case L'V':
|
|
hp.type |= CODEC_UTF8;
|
|
break;
|
|
default:
|
|
return {};
|
|
}
|
|
RCode.erase(0, 1);
|
|
|
|
// [codepage#]
|
|
if (std::regex_search(RCode, match, std::wregex(L"^([0-9]+)#")))
|
|
{
|
|
hp.codepage = std::stoi(match[1]);
|
|
RCode.erase(0, match[0].length());
|
|
}
|
|
|
|
// @addr
|
|
if (!std::regex_match(RCode, match, std::wregex(L"@([[:xdigit:]]+)")))
|
|
return {};
|
|
hp.address = std::stoull(match[1], nullptr, 16);
|
|
return hp;
|
|
}
|
|
|
|
std::optional<HookParam> ParseHCode(std::wstring HCode, std::optional<HookParam> hpo = {})
|
|
{
|
|
auto hp = hpo ? hpo.value() : HookParam{};
|
|
if (HCode[0] == 'L')
|
|
{
|
|
hp.type |= HOOK_RETURN;
|
|
HCode.erase(0, 1);
|
|
}
|
|
switch (HCode[0])
|
|
{
|
|
case L'B':
|
|
hp.type |= BREAK_POINT;
|
|
case L'H':
|
|
break;
|
|
default:
|
|
return {};
|
|
}
|
|
|
|
HCode.erase(0, 1);
|
|
|
|
if (endWith(HCode, L":JIT:YUZU"))
|
|
hp.jittype = JITTYPE::YUZU;
|
|
else if (endWith(HCode, L":JIT:PPSSPP"))
|
|
hp.jittype = JITTYPE::PPSSPP;
|
|
else if (endWith(HCode, L":JIT:VITA3K"))
|
|
hp.jittype = JITTYPE::VITA3K;
|
|
else if (endWith(HCode, L":JIT:RPCS3"))
|
|
hp.jittype = JITTYPE::RPCS3;
|
|
else if (endWith(HCode, L":JIT:UNITY"))
|
|
hp.jittype = JITTYPE::UNITY;
|
|
|
|
// {A|B|W|H|S|Q|V|M}
|
|
switch (HCode[0])
|
|
{
|
|
case L'A':
|
|
hp.type |= CODEC_ANSI_BE;
|
|
break;
|
|
case L'B':
|
|
// ANSI LE
|
|
break;
|
|
case L'W':
|
|
hp.type |= CODEC_UTF16;
|
|
break;
|
|
case L'I':
|
|
hp.type |= CODEC_UTF32;
|
|
break;
|
|
case L'S':
|
|
hp.type |= USING_STRING;
|
|
break;
|
|
case L'Q':
|
|
hp.type |= USING_STRING | CODEC_UTF16;
|
|
break;
|
|
case L'M':
|
|
hp.type |= SPECIAL_JIT_STRING | USING_STRING | CODEC_UTF16;
|
|
break;
|
|
case L'U':
|
|
hp.type |= USING_STRING | CODEC_UTF32;
|
|
break;
|
|
case L'V':
|
|
hp.type |= USING_STRING | CODEC_UTF8;
|
|
break;
|
|
default:
|
|
return {};
|
|
}
|
|
HCode.erase(0, 1);
|
|
|
|
if (hp.type & USING_STRING)
|
|
{
|
|
if (HCode[0] == L'F')
|
|
{
|
|
hp.type |= FULL_STRING;
|
|
HCode.erase(0, 1);
|
|
}
|
|
}
|
|
|
|
// [N]
|
|
if (HCode[0] == L'N')
|
|
{
|
|
hp.type |= NO_CONTEXT;
|
|
HCode.erase(0, 1);
|
|
}
|
|
|
|
std::wsmatch match;
|
|
// [codepage#]
|
|
if (std::regex_search(HCode, match, std::wregex(L"^([0-9]+)#")))
|
|
{
|
|
hp.codepage = std::stoi(match[1]);
|
|
HCode.erase(0, match[0].length());
|
|
}
|
|
|
|
// [padding+]
|
|
if (std::regex_search(HCode, match, std::wregex(L"^([[:xdigit:]]+)\\+")))
|
|
{
|
|
hp.padding = std::stoull(match[1], nullptr, 16);
|
|
HCode.erase(0, match[0].length());
|
|
}
|
|
|
|
auto ConsumeHexInt = [&HCode]
|
|
{
|
|
size_t size = 0;
|
|
int value = 0;
|
|
try
|
|
{
|
|
value = std::stoi(HCode, &size, 16);
|
|
}
|
|
catch (std::invalid_argument)
|
|
{
|
|
}
|
|
HCode.erase(0, size);
|
|
return value;
|
|
};
|
|
|
|
// data_offset
|
|
hp.offset = ConsumeHexInt();
|
|
|
|
// [*deref_offset1]
|
|
if (HCode[0] == L'*')
|
|
{
|
|
hp.type |= DATA_INDIRECT;
|
|
HCode.erase(0, 1);
|
|
hp.index = ConsumeHexInt();
|
|
}
|
|
|
|
// [:split_offset[*deref_offset2]]
|
|
if (HCode[0] == L':')
|
|
{
|
|
hp.type |= USING_SPLIT;
|
|
HCode.erase(0, 1);
|
|
hp.split = ConsumeHexInt();
|
|
|
|
if (HCode[0] == L'*')
|
|
{
|
|
hp.type |= SPLIT_INDIRECT;
|
|
HCode.erase(0, 1);
|
|
hp.split_index = ConsumeHexInt();
|
|
}
|
|
}
|
|
if (hp.jittype == JITTYPE::UNITY)
|
|
{
|
|
if (HCode[0] != L'@')
|
|
return {};
|
|
HCode.erase(0, 1);
|
|
HCode = HCode.substr(0, HCode.size() - wcslen(L":JIT:UNITY"));
|
|
hp.argidx = hp.offset;
|
|
hp.offset = 0;
|
|
hp.address = 0;
|
|
hp.type &= ~MODULE_OFFSET;
|
|
hp.type &= ~FUNCTION_OFFSET;
|
|
strcpy(hp.function, "");
|
|
wcscpy(hp.module, L"");
|
|
strcpy(hp.unityfunctioninfo, wcasta(HCode).c_str());
|
|
}
|
|
else
|
|
{
|
|
|
|
// @addr[:module[:func]]
|
|
if (!std::regex_match(HCode, match, std::wregex(L"^@([[:xdigit:]]+)(:.+?)?(:.+)?")))
|
|
return {};
|
|
hp.address = std::stoull(match[1], nullptr, 16);
|
|
if (match[2].matched)
|
|
{
|
|
hp.type |= MODULE_OFFSET;
|
|
wcsncpy_s(hp.module, match[2].str().erase(0, 1).c_str(), MAX_MODULE_SIZE - 1);
|
|
}
|
|
if (match[3].matched)
|
|
{
|
|
hp.type |= FUNCTION_OFFSET;
|
|
std::wstring func = match[3];
|
|
strncpy_s(hp.function, std::string(func.begin(), func.end()).erase(0, 1).c_str(), MAX_MODULE_SIZE - 1);
|
|
}
|
|
|
|
// ITH has registers offset by 4 vs AGTH: need this to correct
|
|
if (hp.offset < 0)
|
|
hp.offset -= 4;
|
|
if (hp.split < 0)
|
|
hp.split -= 4;
|
|
|
|
if (hp.jittype != JITTYPE::PC)
|
|
{
|
|
hp.emu_addr = hp.address;
|
|
hp.argidx = hp.offset;
|
|
hp.offset = 0;
|
|
hp.address = 0;
|
|
hp.type &= ~MODULE_OFFSET;
|
|
hp.type &= ~FUNCTION_OFFSET;
|
|
strcpy(hp.function, "");
|
|
wcscpy(hp.module, L"");
|
|
}
|
|
}
|
|
return hp;
|
|
}
|
|
|
|
std::optional<HookParam> ParseECode(std::wstring code)
|
|
{
|
|
code.erase(0, 1);
|
|
HookParam hp;
|
|
hp.type |= EMBED_ABLE;
|
|
|
|
if (code[0] == L'D')
|
|
{
|
|
hp.type |= EMBED_DYNA_SJIS;
|
|
code.erase(0, 1);
|
|
}
|
|
if (code[0] == L'S')
|
|
{
|
|
code.erase(0, 1);
|
|
hp.type |= EMBED_BEFORE_SIMPLE;
|
|
|
|
if (code[0] == L'N')
|
|
hp.type |= EMBED_AFTER_NEW;
|
|
else if (code[0] == L'O')
|
|
hp.type |= EMBED_AFTER_OVERWRITE;
|
|
else
|
|
return ParseHCode(code, hp);
|
|
code.erase(0, 1);
|
|
}
|
|
|
|
return ParseHCode(code, hp);
|
|
}
|
|
std::wstring HexString(int64_t num)
|
|
{
|
|
if (num < 0)
|
|
return FormatString(L"-%I64X", -num);
|
|
return FormatString(L"%I64X", num);
|
|
}
|
|
|
|
std::wstring GenerateRCode(HookParam hp)
|
|
{
|
|
std::wstring RCode = L"R";
|
|
|
|
if (hp.type & CODEC_UTF16)
|
|
RCode += L'Q';
|
|
else if (hp.type & CODEC_UTF32)
|
|
RCode += L'U';
|
|
else if (hp.type & CODEC_UTF8)
|
|
RCode += L'V';
|
|
else
|
|
{
|
|
RCode += L'S';
|
|
if (hp.codepage != 0)
|
|
RCode += std::to_wstring(hp.codepage) + L'#';
|
|
}
|
|
|
|
RCode += L'@' + HexString(hp.address);
|
|
|
|
return RCode;
|
|
}
|
|
|
|
std::wstring GenerateHCode(HookParam hp, DWORD processId)
|
|
{
|
|
std::wstring HCode;
|
|
if (hp.type & EMBED_ABLE)
|
|
{
|
|
HCode += L"E";
|
|
|
|
if (hp.hook_before || hp.hook_after)
|
|
HCode += L'X';
|
|
else
|
|
{
|
|
if (hp.type & EMBED_DYNA_SJIS)
|
|
HCode += L"D";
|
|
if (hp.type & EMBED_BEFORE_SIMPLE)
|
|
HCode += L"S";
|
|
|
|
if (hp.type & EMBED_AFTER_NEW)
|
|
HCode += L"N";
|
|
else if (hp.type & EMBED_AFTER_OVERWRITE)
|
|
HCode += L"O";
|
|
}
|
|
}
|
|
if (hp.type & BREAK_POINT)
|
|
HCode += L"B";
|
|
else
|
|
HCode += L"H";
|
|
|
|
if (hp.type & USING_STRING)
|
|
{
|
|
if (hp.type & SPECIAL_JIT_STRING)
|
|
HCode += L'M';
|
|
else if (hp.type & CODEC_UTF16)
|
|
HCode += L'Q';
|
|
else if (hp.type & CODEC_UTF8)
|
|
HCode += L'V';
|
|
else if (hp.type & CODEC_UTF32)
|
|
HCode += L'U';
|
|
else
|
|
HCode += L'S';
|
|
}
|
|
else
|
|
{
|
|
if (hp.type & CODEC_UTF16)
|
|
HCode += L'W';
|
|
else if (hp.type & CODEC_UTF32)
|
|
HCode += L'I';
|
|
else if (hp.type & CODEC_ANSI_BE)
|
|
HCode += L'A';
|
|
else
|
|
HCode += L'B';
|
|
}
|
|
|
|
if (hp.text_fun || hp.filter_fun)
|
|
HCode += L'X';
|
|
|
|
if (hp.type & FULL_STRING)
|
|
HCode += L'F';
|
|
|
|
if (hp.type & NO_CONTEXT)
|
|
HCode += L'N';
|
|
|
|
if (hp.codepage != 0 && !(hp.type & CODEC_UTF8) && !(hp.type & CODEC_UTF16) && !(hp.type & CODEC_UTF32))
|
|
HCode += std::to_wstring(hp.codepage) + L'#';
|
|
|
|
if (hp.padding)
|
|
HCode += HexString(hp.padding) + L'+';
|
|
|
|
if (hp.offset < 0)
|
|
hp.offset += 4;
|
|
if (hp.split < 0)
|
|
hp.split += 4;
|
|
|
|
if (hp.jittype == JITTYPE::PC)
|
|
{
|
|
HCode += HexString(hp.offset);
|
|
}
|
|
else
|
|
{
|
|
HCode += HexString(hp.argidx);
|
|
}
|
|
|
|
if (hp.type & DATA_INDIRECT)
|
|
HCode += L'*' + HexString(hp.index);
|
|
if (hp.type & USING_SPLIT)
|
|
HCode += L':' + HexString(hp.split);
|
|
if (hp.type & SPLIT_INDIRECT)
|
|
HCode += L'*' + HexString(hp.split_index);
|
|
|
|
// Attempt to make the address relative
|
|
if (hp.jittype == JITTYPE::PC)
|
|
{
|
|
if (processId && !(hp.type & MODULE_OFFSET))
|
|
if (AutoHandle<> process = OpenProcess(PROCESS_VM_READ | PROCESS_QUERY_INFORMATION, FALSE, processId))
|
|
if (MEMORY_BASIC_INFORMATION info = {}; VirtualQueryEx(process, (LPCVOID)hp.address, &info, sizeof(info)))
|
|
if (auto moduleName = getModuleFilename(processId, (HMODULE)info.AllocationBase))
|
|
{
|
|
hp.type |= MODULE_OFFSET;
|
|
hp.address -= (uint64_t)info.AllocationBase;
|
|
wcsncpy_s(hp.module, moduleName->c_str() + moduleName->rfind(L'\\') + 1, MAX_MODULE_SIZE - 1);
|
|
}
|
|
|
|
HCode += L'@' + HexString(hp.address);
|
|
if (hp.type & MODULE_OFFSET)
|
|
HCode += L':' + std::wstring(hp.module);
|
|
if (hp.type & FUNCTION_OFFSET)
|
|
HCode += L':' + acastw(hp.function);
|
|
}
|
|
else
|
|
{
|
|
if (hp.jittype == JITTYPE::UNITY)
|
|
{
|
|
HCode += L'@';
|
|
HCode += acastw(hp.unityfunctioninfo);
|
|
HCode += L":JIT:UNITY";
|
|
}
|
|
else
|
|
{
|
|
|
|
HCode += L'@' + HexString(hp.emu_addr);
|
|
switch (hp.jittype)
|
|
{
|
|
case JITTYPE::YUZU:
|
|
HCode += L":JIT:YUZU";
|
|
break;
|
|
case JITTYPE::PPSSPP:
|
|
HCode += L":JIT:PPSSPP";
|
|
break;
|
|
case JITTYPE::VITA3K:
|
|
HCode += L":JIT:VITA3K";
|
|
break;
|
|
case JITTYPE::RPCS3:
|
|
HCode += L":JIT:RPCS3";
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
return HCode;
|
|
}
|
|
}
|
|
|
|
namespace HookCode
|
|
{
|
|
std::optional<HookParam> Parse(std::wstring code)
|
|
{
|
|
if (code[0] == L'/')
|
|
code.erase(0, 1);
|
|
code.erase(std::find(code.begin(), code.end(), L'/'), code.end()); // legacy/AGTH compatibility
|
|
Trim(code);
|
|
if (code[0] == L'R')
|
|
return ParseRCode(code);
|
|
else if (code[0] == L'E')
|
|
return ParseECode(code);
|
|
else
|
|
return ParseHCode(code);
|
|
}
|
|
|
|
std::wstring Generate(HookParam hp, DWORD processId)
|
|
{
|
|
std::wstring HCode = L"";
|
|
return HCode += (hp.type & DIRECT_READ ? GenerateRCode(hp) : GenerateHCode(hp, processId));
|
|
}
|
|
|
|
}
|