raven/testdata/native_loader/module_path_client.c

#include <windows.h>
#include <lzexpand.h>

/* A library loaded by its full path from a folder beside the program reports that path as its file name, so it finds the files it ships
   with; and a process it starts can open it to wait on it, as a bootstrapper's elevated half watches the half that started it. The
   child's exit code reaches the parent whole, as an installer's 1603 does. */

static int child(const WCHAR *digits)
{
	DWORD parent = 0;
	while (*digits >= L'0' && *digits <= L'9')
		parent = parent * 10 + (DWORD)(*digits++ - L'0');
	HANDLE process = OpenProcess(SYNCHRONIZE, FALSE, parent);
	if (!process)
		return 20;
	CloseHandle(process);
	return 1603;
}

static int search_directory(WCHAR *folder)
{
	WCHAR current[MAX_PATH], reported[MAX_PATH], expected[MAX_PATH], probe[MAX_PATH], cwd[MAX_PATH], excluded[MAX_PATH];
	char ansi[MAX_PATH];
	if (!GetCurrentDirectoryW(MAX_PATH,current)) return 30;
	if (LoadLibraryW(L"module_plugin.dll") || GetLastError() != ERROR_MOD_NOT_FOUND) return 31;
	if (!SetDllDirectoryW(folder)) return 32;
	if (!SetDllDirectoryW(NULL) || LoadLibraryW(L"module_plugin.dll") || GetLastError() != ERROR_MOD_NOT_FOUND) return 35;
	if (!SetDllDirectoryW(folder)) return 32;
	HMODULE loaded = LoadLibraryW(L"MODULE_PLUGIN.DLL");
	lstrcpyW(expected,folder);
	lstrcatW(expected,L"\\module_plugin.dll");
	if (!loaded || !GetModuleFileNameW(loaded,reported,MAX_PATH) || lstrcmpiW(reported,expected)) return 33;
	int (*answer)(void) = (int (*)(void))GetProcAddress(loaded,"plugin_answer");
	if (!answer || answer() != 42 || !FreeLibrary(loaded)) return 34;
	lstrcpyW(probe,folder); lstrcatW(probe,L"\\module_probe.dll");
	lstrcpyW(cwd,folder); lstrcatW(cwd,L"\\module_cwd.dll");
	lstrcpyW(excluded,folder); lstrcatW(excluded,L"\\module_excluded.dll");
	if (!CopyFileW(expected,probe,FALSE) || !CopyFileW(expected,cwd,FALSE) || !CopyFileW(expected,excluded,FALSE)) return 47;
	unsigned i = 0;
	for (; folder[i] && i < MAX_PATH-1; ++i) {
		if (folder[i] > 127) return 36;
		ansi[i] = (char)folder[i];
	}
	ansi[i] = 0;
	if (!SetDllDirectoryA(ansi)) return 37;
	if (LoadLibraryExW(L"module_probe.dll",NULL,LOAD_LIBRARY_SEARCH_SYSTEM32) || GetLastError() != ERROR_MOD_NOT_FOUND) return 38;
	loaded = LoadLibraryExW(L"module_probe.dll",NULL,LOAD_LIBRARY_SEARCH_USER_DIRS);
	if (!loaded || !FreeLibrary(loaded)) return 39;
	if (!SetCurrentDirectoryW(folder) || !SetDllDirectoryA("")) return 40;
	if (LoadLibraryW(L"module_cwd.dll") || GetLastError() != ERROR_MOD_NOT_FOUND) return 41;
	if (!SetDllDirectoryA(NULL)) return 42;
	loaded = LoadLibraryW(L"module_cwd.dll");
	if (!loaded || !FreeLibrary(loaded)) return 43;
	if (!SetDllDirectoryW(L"")) return 44;
	if (LoadLibraryW(L"module_excluded.dll") || GetLastError() != ERROR_MOD_NOT_FOUND) return 45;
	if (!SetDllDirectoryW(NULL) || !SetCurrentDirectoryW(current)) return 46;
	if (!DeleteFileW(probe) || !DeleteFileW(cwd) || !DeleteFileW(excluded)) return 48;
	return 0;
}

int raven_entry(void)
{
	WCHAR self[MAX_PATH], path[MAX_PATH], reported[MAX_PATH], command[MAX_PATH + 64];
	const WCHAR *line = GetCommandLineW();
	for (const WCHAR *at = line; *at; ++at)
		if (at[0] == L' ' && at[1] == L'-' && at[2] == L'-' && at[3] == L'c' && at[4] == L' ')
			return child(at + 5);
	if (!GetModuleFileNameW(NULL, self, MAX_PATH))
		return 1;
	lstrcpyW(path, self);
	WCHAR *slash = path;
	for (WCHAR *at = path; *at; ++at)
		if (*at == L'\\')
			slash = at;
	lstrcpyW(slash + 1, L"plugins");
	int search = search_directory(path);
	if (search) return search;
	lstrcpyW(slash + 1, L"plugins\\module_plugin.dll");
	HMODULE plugin = LoadLibraryW(path);
	if (!plugin)
		return 2;
	if (!GetModuleFileNameW(plugin, reported, MAX_PATH) || lstrcmpiW(reported, path))
		return 3;
	int (*answer)(void) = (int (*)(void))GetProcAddress(plugin, "plugin_answer");
	if (!answer || answer() != 42)
		return 4;
	/* A file that is no program image, as a type library tried as a resource module first, fails to load with ERROR_BAD_EXE_FORMAT. */
	lstrcpyW(slash + 1, L"plugins\\notes.tlb");
	HANDLE notes = CreateFileW(path, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
	DWORD written = 0;
	if (notes == INVALID_HANDLE_VALUE || !WriteFile(notes, "not a program", 13, &written, NULL))
		return 8;
	CloseHandle(notes);
	if (LoadLibraryExW(path, NULL, LOAD_LIBRARY_AS_DATAFILE) || GetLastError() != ERROR_BAD_EXE_FORMAT)
		return 9;
	/* The type library loader reads such a file through the LZ functions, which give a plain file as it is. */
	OFSTRUCT record;
	char text[4] = {0};
	INT packed = LZOpenFileW(path, &record, OF_READ);
	if (packed < 0 || LZSeek(packed, 4, 0) != 4 || LZRead(packed, text, 3) != 3 || text[0] != 'a' || text[2] != 'p')
		return 10;
	LZClose(packed);
	wsprintfW(command, L"\"%s\" --c %u", self, GetCurrentProcessId());
	STARTUPINFOW startup = {sizeof startup};
	PROCESS_INFORMATION started;
	if (!CreateProcessW(self, command, NULL, NULL, FALSE, 0, NULL, NULL, &startup, &started))
		return 5;
	DWORD code = 1;
	if (WaitForSingleObject(started.hProcess, 30000) != WAIT_OBJECT_0 || !GetExitCodeProcess(started.hProcess, &code))
		return 6;
	CloseHandle(started.hThread);
	CloseHandle(started.hProcess);
	return code == 1603 ? 0 : 7;
}