x64dbg
/
TitanEngine
mirror of https://github.com/x64dbg/TitanEngine
1
0
Fork 0
Code Releases Activity

UNICODE-FIX engine improvement

This commit is contained in:
NtQuery 2014-03-11 12:48:16 +01:00
parent a742815814
commit e60e886cf5
2 changed files with 12 additions and 61 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
TitanEngine/Global.Engine.cpp
View File

@ -19,13 +19,11 @@ bool enginePassAllExceptions = true;
bool engineExecutePluginCallBack = true; bool engineExecutePluginCallBack = true;
bool engineAutoHideFromDebugger = false; // hardcoded bool engineAutoHideFromDebugger = false; // hardcoded
char engineExtractedFolderName[512]; char engineFoundDLLName[512] = {0};
char engineFoundDLLName[512]; char engineFoundAPIName[512] = {0};
char engineFoundAPIName[512]; wchar_t engineExtractedFileNameW[512] = {0};
char engineExtractedFileName[512]; wchar_t engineSzEngineFile[MAX_PATH] = {0};
wchar_t engineExtractedFileNameW[512]; wchar_t engineSzEngineFolder[MAX_PATH] = {0};
wchar_t engineSzEngineFile[MAX_PATH];
wchar_t engineSzEngineFolder[MAX_PATH];
HMODULE engineHandle; HMODULE engineHandle;
LPVOID engineExitThreadOneShootCallBack = NULL; LPVOID engineExitThreadOneShootCallBack = NULL;
LPVOID engineDependencyFiles; LPVOID engineDependencyFiles;
@ -36,9 +34,7 @@ void* EngineStartUnpackingCallBack;
void EngineInit() void EngineInit()
{ {
int i; int i;
RtlZeroMemory(&engineSzEngineFile, sizeof engineSzEngineFile); if(GetModuleFileNameW(engineHandle, engineSzEngineFile, _countof(engineSzEngineFile)) > NULL)
RtlZeroMemory(&engineSzEngineFolder, sizeof engineSzEngineFolder);
if(GetModuleFileNameW(engineHandle, engineSzEngineFile, MAX_PATH) > NULL)
{ {
lstrcpyW(engineSzEngineFolder, engineSzEngineFile); lstrcpyW(engineSzEngineFolder, engineSzEngineFile);
i = lstrlenW(engineSzEngineFolder); i = lstrlenW(engineSzEngineFolder);
@ -108,49 +104,6 @@ bool EngineFileExists(char* szFileName)
} }
} }
char* EngineExtractPath(char* szFileName)
{
int i;
RtlZeroMemory(&engineExtractedFolderName, sizeof(engineExtractedFolderName));
lstrcpyA(engineExtractedFolderName, szFileName);
i = lstrlenA(engineExtractedFolderName);
while(i > 0 && engineExtractedFolderName[i] != 0x5C)
{
engineExtractedFolderName[i] = 0x00;
i--;
}
return(engineExtractedFolderName);
}
char* EngineExtractFileName(char* szFileName)
{
int i;
int j;
int x = 0;
i = lstrlenA(szFileName);
RtlZeroMemory(&engineExtractedFileName, sizeof(engineExtractedFileName));
while(i > 0 && szFileName[i] != 0x5C)
{
i--;
}
if(szFileName[i] == 0x5C)
{
for(j = i + 1; j <= lstrlenA(szFileName); j++)
{
engineExtractedFileName[x] = szFileName[j];
x++;
}
}
else
{
return(szFileName);
}
return(engineExtractedFileName);
}
void EngineCreatePathForFile(char* szFileName) void EngineCreatePathForFile(char* szFileName)
{ {
int len=lstrlenA(szFileName); int len=lstrlenA(szFileName);
@ -979,12 +932,11 @@ long long EngineSimulateNtLoaderW(wchar_t* szFileName)
long long EngineSimulateNtLoader(char* szFileName) long long EngineSimulateNtLoader(char* szFileName)
{ {
wchar_t uniFileName[MAX_PATH] = {0};
wchar_t uniFileName[MAX_PATH] = {};
if(szFileName != NULL) if(szFileName != NULL)
{ {
MultiByteToWideChar(CP_ACP, NULL, szFileName, lstrlenA(szFileName)+1, uniFileName, sizeof(uniFileName)/(sizeof(uniFileName[0]))); MultiByteToWideChar(CP_ACP, NULL, szFileName, -1, uniFileName, _countof(uniFileName));
return(EngineSimulateNtLoaderW(uniFileName)); return(EngineSimulateNtLoaderW(uniFileName));
} }
else else
@ -1168,10 +1120,11 @@ long long EngineGetProcAddress(ULONG_PTR ModuleBase, char* szAPIName)
PEXPORTED_DATA ExportedFunctions; PEXPORTED_DATA ExportedFunctions;
PEXPORTED_DATA ExportedFunctionNames; PEXPORTED_DATA ExportedFunctionNames;
PEXPORTED_DATA_WORD ExportedFunctionOrdinals; PEXPORTED_DATA_WORD ExportedFunctionOrdinals;
char szModuleName[MAX_PATH] = {};
bool FileIs64 = false; bool FileIs64 = false;
if(GetModuleFileNameA((HMODULE)ModuleBase, szModuleName, MAX_PATH) == NULL) APIFoundAddress = (ULONG_PTR)GetProcAddress((HMODULE)ModuleBase, szAPIName);
if(APIFoundAddress == 0)
{ {
__try __try
{ {
@ -1224,7 +1177,7 @@ long long EngineGetProcAddress(ULONG_PTR ModuleBase, char* szAPIName)
} }
else else
{ {
return((ULONG_PTR)GetProcAddress((HMODULE)ModuleBase, szAPIName)); return APIFoundAddress;
} }
} }

2
TitanEngine/Global.Engine.h
View File

@ -25,8 +25,6 @@ extern bool engineAutoHideFromDebugger;
void EngineInit(); void EngineInit();
bool EngineIsThereFreeHardwareBreakSlot(LPDWORD FreeRegister); bool EngineIsThereFreeHardwareBreakSlot(LPDWORD FreeRegister);
bool EngineFileExists(char* szFileName); bool EngineFileExists(char* szFileName);
char* EngineExtractPath(char* szFileName);
char* EngineExtractFileName(char* szFileName);
void EngineCreatePathForFile(char* szFileName); void EngineCreatePathForFile(char* szFileName);
void EngineCreatePathForFileW(wchar_t* szFileName); void EngineCreatePathForFileW(wchar_t* szFileName);
wchar_t* EngineExtractFileNameW(wchar_t* szFileName); wchar_t* EngineExtractFileNameW(wchar_t* szFileName);