Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138/* * PROJECT: ReactOS api tests * LICENSE: GPL-2.0-or-later (https://spdx.org/licenses/GPL-2.0-or-later) * PURPOSE: Tests for SHLoadIndirectString * COPYRIGHT: Copyright 2020 Mark Jansen (mark.jansen@reactos.org) */
#include <apitest.h>#include <shlwapi.h>#include "resource.h"
#include <pseh/pseh2.h>
static void execute_test(LPCWSTR DllFile){ WCHAR DllBuffer[MAX_PATH + 20]; WCHAR Buffer[MAX_PATH * 2] = {0}; HRESULT hr; HANDLE hEvent; DWORD dwRet; HMODULE mod;
hEvent = CreateEventA(NULL, TRUE, FALSE, "Local\\shlwapi_apitest_evt");
// Show that it is not signaled dwRet = WaitForSingleObject(hEvent, 1); ok_hex(dwRet, WAIT_TIMEOUT);
// Ensure the module is not loaded yet... mod = GetModuleHandleW(DllFile); if (mod != NULL) { CloseHandle(hEvent); skip("%S loaded, cannot continue\n", DllFile); return; }
wsprintfW(DllBuffer, L"@%s,-3", DllFile);
// Load a string from the dll hr = SHLoadIndirectString(DllBuffer, Buffer, _countof(Buffer), NULL); ok_hex(hr, S_OK); if (SUCCEEDED(hr)) { // Module should still not be loaded mod = GetModuleHandleW(DllFile); ok_ptr(mod, 0);
ok_wstr(Buffer, L"Test string one.");
// DllMain not called.. dwRet = WaitForSingleObject(hEvent, 1); ok_hex(dwRet, WAIT_TIMEOUT);
// Show that calling DllMain will set the event mod = LoadLibraryW(DllFile); ok(mod != NULL, "Failed to load %S\n", DllFile);
if (mod) { dwRet = WaitForSingleObject(hEvent, 1); ok_hex(dwRet, WAIT_OBJECT_0); FreeLibrary(mod); } } CloseHandle(hEvent);}
BOOL extract_resource(const WCHAR* Filename, LPCWSTR ResourceName){ BOOL Success; DWORD dwWritten, Size; HGLOBAL hGlobal; LPVOID pData; HANDLE Handle; HRSRC hRsrc = FindResourceW(GetModuleHandleW(NULL), ResourceName, (LPCWSTR)10); ok(!!hRsrc, "Unable to find %s\n", wine_dbgstr_w(ResourceName)); if (!hRsrc) return FALSE;
hGlobal = LoadResource(GetModuleHandleW(NULL), hRsrc); Size = SizeofResource(GetModuleHandleW(NULL), hRsrc); pData = LockResource(hGlobal);
ok(Size && !!pData, "Unable to load %s\n", wine_dbgstr_w(ResourceName)); if (!Size || !pData) return FALSE;
Handle = CreateFileW(Filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (Handle == INVALID_HANDLE_VALUE) { skip("Failed to create temp file %ls, error %lu\n", Filename, GetLastError()); return FALSE; } Success = WriteFile(Handle, pData, Size, &dwWritten, NULL); ok(Success == TRUE, "WriteFile failed with %lu\n", GetLastError()); ok(dwWritten == Size, "WriteFile wrote %lu bytes instead of %lu\n", dwWritten, Size); CloseHandle(Handle); Success = Success && (dwWritten == Size);
UnlockResource(pData); return Success;}
START_TEST(SHLoadIndirectString){ WCHAR workdir[MAX_PATH], dllpath[MAX_PATH]; BOOL ret; UINT Length;
ret = GetTempPathW(_countof(workdir), workdir); ok(ret, "GetTempPathW error: %lu\n", GetLastError());
Length = GetTempFileNameW(workdir, L"ntdll", 0, dllpath); ok(Length != 0, "GetTempFileNameW failed with %lu\n", GetLastError());
if (extract_resource(dllpath, (LPCWSTR)101)) { _SEH2_TRY { execute_test(dllpath); } _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER) { ok(0, "Ldr didnt handle exception\n"); } _SEH2_END; } else { ok(0, "Failed to extract resource\n"); }
DeleteFileW(dllpath);}