Initial commit

injector/launcher_payload/src/dll.c

@@ -0,0 +1,67 @@
+#include <injshared.h>
+
+#include <lpayload.h>
+
+const char EXE_ENV[] = "JADEITE_TARGET_EXE_PATH";
+const char INJECT_DLL_ENV[] = "JADEITE_INJECT_DLL_PATH";
+
+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) {
+        return TRUE;
+    }
+
+    // Get target EXE path
+    char targetExe[MAX_PATH];
+    read_env(EXE_ENV, targetExe, sizeof(targetExe));
+    
+    // Get the path of the DLL to inject
+    char injectDll[MAX_PATH];
+    read_env(INJECT_DLL_ENV, injectDll, sizeof(injectDll));
+
+    // Compute the working directory path
+    char workdir[MAX_PATH];
+    strcpy(workdir, targetExe);
+    *(strrchr(workdir, '\\')) = '\0';
+
+    // Start the game
+    STARTUPINFO si;
+    ZeroMemory(&si, sizeof(si));
+
+    PROCESS_INFORMATION pi;
+    si.cb = sizeof(si);
+    ZeroMemory(&pi, sizeof(pi));
+
+    if (!CreateProcessA(
+        targetExe,
+        NULL,
+        NULL,
+        NULL,
+        FALSE,
+        CREATE_SUSPENDED,
+        NULL,
+        workdir,
+        &si,
+        &pi
+    )) {
+        exit(1);
+    }
+
+    // Inject
+    void *payloadStart = &_binary_lpayload_o_p_payload_bin_start;
+    size_t payloadSize = (size_t)&_binary_lpayload_o_p_payload_bin_size;
+    inject(pi.hProcess, payloadStart, payloadSize, injectDll);
+
+    // Resume the process
+    ResumeThread(pi.hThread);
+
+    // The launcher process should now hang untill the game terminates
+    WaitForSingleObject(pi.hProcess, INFINITE);
+
+    return TRUE;
+}