mirror of
https://github.com/reactos/reactos.git
synced 2025-07-31 19:11:55 +00:00
[SHLWAPI_APITEST] Add test showing SHLoadIndirectString should load a dll as datafile
This commit is contained in:
parent
07f98eb10a
commit
922a956805
7 changed files with 190 additions and 0 deletions
134
modules/rostests/apitests/shlwapi/SHLoadIndirectString.c
Normal file
134
modules/rostests/apitests/shlwapi/SHLoadIndirectString.c
Normal file
|
@ -0,0 +1,134 @@
|
|||
/*
|
||||
* 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"
|
||||
|
||||
static void execute_test(LPCWSTR DllFile)
|
||||
{
|
||||
WCHAR DllBuffer[MAX_PATH + 20];
|
||||
WCHAR Buffer[MAX_PATH * 2] = {0};
|
||||
HRESULT hr;
|
||||
HANDLE hEvent;
|
||||
DWORD dwRet;
|
||||
|
||||
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...
|
||||
HMODULE 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);
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue