#include <apitest.h>
#include <shlwapi.h>

Include dependency graph for SHLoadRegUIString.c:

Typedefs
typedef HRESULT(WINAPI *	FN_SHLoadRegUIStringA) (HKEY hkey, LPCSTR value, LPSTR buf, DWORD size)

typedef HRESULT(WINAPI *	FN_SHLoadRegUIStringW) (HKEY hkey, LPCWSTR value, LPWSTR buf, DWORD size)

Functions
static void	test_SHLoadRegUIStringA (HKEY hKey)

static void	test_SHLoadRegUIStringW (HKEY hKey)

BOOL	extract_resource (const WCHAR *Filename, LPCWSTR ResourceName)

	START_TEST (SHLoadRegUIString)

Variables
static FN_SHLoadRegUIStringA	pSHLoadRegUIStringA = NULL

static FN_SHLoadRegUIStringW	pSHLoadRegUIStringW = NULL

Typedef Documentation

◆ FN_SHLoadRegUIStringA

typedef HRESULT(WINAPI * FN_SHLoadRegUIStringA) (HKEY hkey, LPCSTR value, LPSTR buf, DWORD size)

Definition at line 11 of file SHLoadRegUIString.c.

◆ FN_SHLoadRegUIStringW

typedef HRESULT(WINAPI * FN_SHLoadRegUIStringW) (HKEY hkey, LPCWSTR value, LPWSTR buf, DWORD size)

Definition at line 12 of file SHLoadRegUIString.c.

Function Documentation

◆ extract_resource()

BOOL extract_resource	(	const WCHAR *	Filename,
		LPCWSTR	ResourceName
	)

Definition at line 239 of file load_notifications.c.

{
    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;
}

Referenced by START_TEST().

◆ START_TEST()

START_TEST ( SHLoadRegUIString )

Definition at line 47 of file SHLoadRegUIString.c.

{
    LONG error;
    HKEY hKey;
    DWORD cbValue;
    static const WCHAR s_szTestValue1[] = L"%WINDIR%\\TEST";
    static const WCHAR s_szTestValue2[] = L"@SHLoadRegUIString.dll%EmptyEnvVar%,-3";
    HMODULE hSHLWAPI;
 
    SetEnvironmentVariableW(L"EmptyEnvVar", L"");
 
    /* Get procedures */
    hSHLWAPI = GetModuleHandleW(L"shlwapi");
    pSHLoadRegUIStringA = (FN_SHLoadRegUIStringA)GetProcAddress(hSHLWAPI, (LPCSTR)438);
    pSHLoadRegUIStringW = (FN_SHLoadRegUIStringW)GetProcAddress(hSHLWAPI, (LPCSTR)439);
    if (!pSHLoadRegUIStringA || !pSHLoadRegUIStringW)
    {
        skip("No procedure found\n");
        return;
    }
 
    if (!extract_resource(L"SHLoadRegUIString.dll", MAKEINTRESOURCEW(101)))
    {
        skip("File 'SHLoadRegUIString.dll' cannot be extracted\n");
        return;
    }
 
    /* Open registry key and write some test values */
    error = RegOpenKeyExW(HKEY_CURRENT_USER, L"Software", 0, KEY_READ | KEY_WRITE, &hKey);
    ok_long(error, ERROR_SUCCESS);
 
    cbValue = (lstrlenW(s_szTestValue1) + 1) * sizeof(WCHAR);
    error = RegSetValueExW(hKey, L"TestValue1", 0, REG_SZ, (LPBYTE)s_szTestValue1, cbValue);
    ok_long(error, ERROR_SUCCESS);
 
    cbValue = (lstrlenW(s_szTestValue2) + 1) * sizeof(WCHAR);
    error = RegSetValueExW(hKey, L"TestValue2", 0, REG_SZ, (LPBYTE)s_szTestValue2, cbValue);
    ok_long(error, ERROR_SUCCESS);
 
    /* The main dish */
    test_SHLoadRegUIStringA(hKey);
    test_SHLoadRegUIStringW(hKey);
 
    /* Delete the test values and close the key */
    RegDeleteValueW(hKey, L"TestValue1");
    RegDeleteValueW(hKey, L"TestValue2");
    RegCloseKey(hKey);
 
    DeleteFileW(L"SHLoadRegUIString.dll");
}

◆ test_SHLoadRegUIStringA()

static void test_SHLoadRegUIStringA ( HKEY hKey )

static

Definition at line 17 of file SHLoadRegUIString.c.

{
    HRESULT hr;
    CHAR szBuff[MAX_PATH];
 
    hr = pSHLoadRegUIStringA(hKey, "TestValue1", szBuff, _countof(szBuff));
    ok_long(hr, S_OK);
    ok_str(szBuff, "%WINDIR%\\TEST");
 
    hr = pSHLoadRegUIStringA(hKey, "TestValue2", szBuff, _countof(szBuff));
    ok_long(hr, S_OK);
    ok_str(szBuff, "Test string one.");
}

Referenced by START_TEST().

◆ test_SHLoadRegUIStringW()

static void test_SHLoadRegUIStringW ( HKEY hKey )

static

Definition at line 31 of file SHLoadRegUIString.c.

{
    HRESULT hr;
    WCHAR szBuff[MAX_PATH];
 
    hr = pSHLoadRegUIStringW(hKey, L"TestValue1", szBuff, _countof(szBuff));
    ok_long(hr, S_OK);
    ok_wstr(szBuff, L"%WINDIR%\\TEST");
 
    hr = pSHLoadRegUIStringW(hKey, L"TestValue2", szBuff, _countof(szBuff));
    ok_long(hr, S_OK);
    ok_wstr(szBuff, L"Test string one.");
}