diff --git a/injector/launcher_payload/src/dll.c b/injector/launcher_payload/src/dll.c index 097721f..bcd9ce3 100644 --- a/injector/launcher_payload/src/dll.c +++ b/injector/launcher_payload/src/dll.c @@ -4,11 +4,6 @@ #include -static inline void read_env(const char *env, char *dest, size_t size) { - GetEnvironmentVariableA(env, dest, size); - SetEnvironmentVariableA(env, ""); -} - BOOL WINAPI DllMain(HINSTANCE inst, DWORD reason, LPVOID reserved) { // Only listen for attach if (reason != DLL_PROCESS_ATTACH) { @@ -16,12 +11,10 @@ BOOL WINAPI DllMain(HINSTANCE inst, DWORD reason, LPVOID reserved) { } // Get target EXE path - char targetExe[MAX_PATH]; - read_env(ENV_EXE_PATH, targetExe, sizeof(targetExe)); + char *targetExe = getenv(ENV_EXE_PATH); // Get the path of the DLL to inject - char injectDll[MAX_PATH]; - read_env(ENV_DLL_PATH, injectDll, sizeof(injectDll)); + char *injectDll = getenv(ENV_DLL_PATH); // Compute the working directory path char workdir[MAX_PATH]; diff --git a/injector/src/injector.c b/injector/src/injector.c index c2c0937..9885a5a 100644 --- a/injector/src/injector.c +++ b/injector/src/injector.c @@ -34,8 +34,7 @@ int main(int argc, char **argv) { char injectorPath[MAX_PATH]; GetModuleFileNameA(GetModuleHandleA(NULL), injectorPath, sizeof(injectorPath)); - char *lastSep = strrchr(injectorPath, '\\'); - *lastSep = '\0'; + *(strrchr(injectorPath, '\\')) = '\0'; SetCurrentDirectoryA(injectorPath); @@ -49,7 +48,7 @@ int main(int argc, char **argv) { char launcherPayloadPath[MAX_PATH]; GetFullPathNameA(LAUNCHER_INJECT_DLL, sizeof(launcherPayloadPath), launcherPayloadPath, NULL); - printf("Starting \"%s\" via \"%s\"\n", gameExePath, launcherPath); + printf("Starting '%s' via '%s'\n", gameExePath, launcherPath); // Set envvars SetEnvironmentVariableA(ENV_EXE_PATH, gameExePath);