Public-Mirror/gbe_fork
1
0
Fork 0
mirror of https://github.com/Detanup01/gbe_fork.git synced 2024-10-22 20:48:17 +08:00
Code Issues 1 Projects Releases Wiki Activity

sloppy code is problematic! (at least verify PE signature)

Browse Source
This commit is contained in:
otavepto 2024-02-13 02:08:14 +02:00
parent 6cc937d4b4
commit 018d4aa2c1
3 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
CHANGELOG.md
View File

@ -1,3 +1,9 @@
## 2024/2/13
* cold client loader: validate the PE signature before attempting to detect arch
---
## 2024/2/10
* a hacky fix for the overlay on directx12, currently very slow when loading images

16
helpers/pe_helpers.cpp
View File

@ -24,19 +24,23 @@ PIMAGE_NT_HEADERS pe_helpers::get_nt_header(HMODULE hModule)
{
// https://dev.to/wireless90/validating-the-pe-signature-my-av-flagged-me-windows-pe-internals-2m5o/
PIMAGE_DOS_HEADER dosHeader = (PIMAGE_DOS_HEADER)(char*)hModule;
if (dosHeader->e_magic != 0x5A4D) { // "MZ"
return nullptr;
}
LONG newExeHeaderOffset = dosHeader->e_lfanew;
return (PIMAGE_NT_HEADERS)((char*)hModule + newExeHeaderOffset);
}
PIMAGE_FILE_HEADER pe_helpers::get_file_header(HMODULE hModule)
{
return &get_nt_header(hModule)->FileHeader;
auto nt_header = get_nt_header(hModule);
return nt_header ? &nt_header->FileHeader : nullptr;
}
PIMAGE_OPTIONAL_HEADER pe_helpers::get_optional_header(HMODULE hModule)
{
return &get_nt_header(hModule)->OptionalHeader;
auto nt_header = get_nt_header(hModule);
return nt_header ? &nt_header->OptionalHeader : nullptr;
}
uint8_t* pe_helpers::search_memory(uint8_t *mem, size_t size, const std::string &search_patt)
@ -234,12 +238,14 @@ std::string pe_helpers::get_err_string(DWORD code)
bool pe_helpers::is_module_64(HMODULE hModule)
{
return (get_file_header(hModule)->Machine == IMAGE_FILE_MACHINE_AMD64);
auto file_header = get_file_header(hModule);
return file_header ? (file_header->Machine == IMAGE_FILE_MACHINE_AMD64) : false;
}
bool pe_helpers::is_module_32(HMODULE hModule)
{
return (get_file_header(hModule)->Machine == IMAGE_FILE_MACHINE_I386);
auto file_header = get_file_header(hModule);
return file_header ? (file_header->Machine == IMAGE_FILE_MACHINE_I386) : false;
}
pe_helpers::SectionHeadersResult pe_helpers::get_section_headers(HMODULE hModule)

8
tools/steamclient_loader/win/ColdClientLoader.cpp
View File

@ -58,8 +58,7 @@ std::wstring get_ini_value(LPCWSTR section, LPCWSTR key, LPCWSTR default_val = N
static std::vector<uint8_t> get_pe_header(const std::wstring &filepath)
{
try
{
try {
std::ifstream file(filepath, std::ios::binary);
if (!file.is_open()) throw;
@ -71,9 +70,8 @@ static std::vector<uint8_t> get_pe_header(const std::wstring &filepath)
file.close();
return data;
}
catch(const std::exception& e)
{
} catch(const std::exception& e) {
dbg_log::write(std::string("Error reading PE header: ") + e.what());
return std::vector<uint8_t>();
}
}