QuerySourceCreateFromKey.cpp File Reference

#include <apitest.h>
#include <shlobj.h>
#include <shlwapi.h>
#include <shlguid_undoc.h>
#include <shlobj_undoc.h>
#include <shlwapi_undoc.h>
#include <versionhelpers.h>

Include dependency graph for QuerySourceCreateFromKey.cpp:

Go to the source code of this file.

Typedefs
typedef HRESULT(WINAPI *	FN_QuerySourceCreateFromKey) (HKEY, PCWSTR, BOOL, REFIID, PVOID *)

Functions
template<class T >
static LONG	SafeRelease (T *p)
static HRESULT	QSCFK (HKEY hKey, PCWSTR pszKey, BOOL Create, REFIID riid, PVOID *ppv)
static HRESULT	ShimverQSCFK (HKEY hKey, PCWSTR pszKey, BOOL Create, PVOID *ppv)
static void	SetupRegistry (void)
static void	CleanupRegistry (void)
static void	Test_EnumValues (void)
static void	Test_EnumSources (void)
static void	Test_CheckValues (void)
void	Test_Version (void)
	START_TEST (QuerySourceCreateFromKey)

Variables
static FN_QuerySourceCreateFromKey	g_pQuerySourceCreateFromKey = NULL
static const WCHAR	k_Root [] = L"Software\\QuerySrcTest"
static const WCHAR	k_SubKeyA [] = L"SubKeyA"
static const WCHAR	k_SubKeyB [] = L"SubKeyB"

Typedef Documentation

FN_QuerySourceCreateFromKey

typedef HRESULT(WINAPI * FN_QuerySourceCreateFromKey) (HKEY, PCWSTR, BOOL, REFIID, PVOID *)

Definition at line 16 of file QuerySourceCreateFromKey.cpp.

Function Documentation

CleanupRegistry()

static void CleanupRegistry ( void  )

static

Definition at line 58 of file QuerySourceCreateFromKey.cpp.

{
    SHDeleteKeyW(HKEY_CURRENT_USER, k_Root);
}

Referenced by START_TEST().

QSCFK()

static HRESULT QSCFK ( HKEY  hKey, PCWSTR  pszKey, BOOL  Create, REFIID  riid, PVOID *  ppv  )

static

Definition at line 24 of file QuerySourceCreateFromKey.cpp.

{
    return g_pQuerySourceCreateFromKey ? g_pQuerySourceCreateFromKey(hKey, pszKey, Create, riid, ppv) : E_NOTIMPL;
}

Referenced by ShimverQSCFK(), and Test_Version().

SafeRelease()

template<class T >

static LONG SafeRelease ( T *  p )

static

Definition at line 19 of file QuerySourceCreateFromKey.cpp.

{
    return p ? p->Release() : 0;
}

Referenced by Test_Version().

SetupRegistry()

static void SetupRegistry ( void  )

static

Definition at line 38 of file QuerySourceCreateFromKey.cpp.

{
    HKEY hRoot;
    RegCreateKeyExW(HKEY_CURRENT_USER, k_Root, 0, NULL, 0, KEY_ALL_ACCESS, NULL, &hRoot, NULL);
 
    static const WCHAR valA[] = L"hello";
    RegSetValueExW(hRoot, L"ValueA", 0, REG_SZ, (const BYTE*)valA, (DWORD)sizeof(valA));
 
    DWORD dwValue = 0xBEEFCAFE;
    RegSetValueExW(hRoot, L"ValueB", 0, REG_DWORD, (const BYTE*)&dwValue, (DWORD)sizeof(dwValue));
 
    HKEY hSub;
    RegCreateKeyExW(hRoot, k_SubKeyA, 0, NULL, 0, KEY_ALL_ACCESS, NULL, &hSub, NULL);
    RegCloseKey(hSub);
    RegCreateKeyExW(hRoot, k_SubKeyB, 0, NULL, 0, KEY_ALL_ACCESS, NULL, &hSub, NULL);
    RegCloseKey(hSub);
 
    RegCloseKey(hRoot);
}

Referenced by START_TEST().

ShimverQSCFK()

static HRESULT ShimverQSCFK ( HKEY  hKey, PCWSTR  pszKey, BOOL  Create, PVOID *  ppv  )

static

Definition at line 29 of file QuerySourceCreateFromKey.cpp.

{
    return QSCFK(hKey, pszKey, Create, IsWindowsVistaOrGreater() ? IID_IQuerySource : IID_IQuerySourceOld, ppv);
}

Referenced by Test_Version().

START_TEST()

START_TEST

(

QuerySourceCreateFromKey

)

Definition at line 261 of file QuerySourceCreateFromKey.cpp.

{
    if (IsWindowsVistaOrGreater())
    {
        skip("Vista+ is not tested well\n");
        return;
    }
 
    g_pQuerySourceCreateFromKey = (FN_QuerySourceCreateFromKey)
        GetProcAddress(GetModuleHandleA("shlwapi"), MAKEINTRESOURCEA(544));
    if (!g_pQuerySourceCreateFromKey)
    {
        skip("QuerySourceCreateFromKey not found\n");
        return;
    }
 
    HRESULT hrCoInit = CoInitialize(NULL);
 
    SetupRegistry();
 
    Test_EnumValues();
    Test_EnumSources();
    Test_CheckValues();
    Test_Version();
 
    CleanupRegistry();
 
    if (SUCCEEDED(hrCoInit))
        CoUninitialize();
}

Test_CheckValues()

static void Test_CheckValues ( void  )

static

Definition at line 166 of file QuerySourceCreateFromKey.cpp.

{
    IQuerySourceOld *pSrc = NULL;
    HRESULT hr = g_pQuerySourceCreateFromKey(HKEY_CURRENT_USER, k_Root, FALSE,
                                             IID_IQuerySourceOld, (PVOID*)&pSrc);
    ok_hr(hr, S_OK);
    ok(pSrc != NULL, "pSrc was NULL\n");
 
    PWSTR pszValue;
 
    // QueryValueString
    hr = 0xDEADFACE;
    pszValue = NULL;
    if (pSrc)
        hr = pSrc->QueryValueString(NULL, L"ValueA", &pszValue);
    ok_hr(hr, S_OK);
    ok(lstrcmpiW(pszValue, L"hello") == 0, "pszValue was %s\n", wine_dbgstr_w(pszValue));
 
    hr = 0xDEADFACE;
    pszValue = NULL;
    if (pSrc)
        hr = pSrc->QueryValueString(NULL, L"ValueB", &pszValue);
    ok_hr(hr, E_DATATYPE_MISMATCH);
    ok(pszValue == NULL, "pszValue was %s\n", wine_dbgstr_w(pszValue));
 
    // QueryValueExists
    hr = 0xDEADFACE;
    if (pSrc)
        hr = pSrc->QueryValueExists(NULL, L"ValueA");
    ok_hr(hr, S_OK);
    if (pSrc)
        hr = pSrc->QueryValueExists(NULL, L"ValueB");
    ok_hr(hr, S_OK);
 
    DWORD dwValue;
 
    // QueryValueDword
    dwValue = 0xDEADFACE;
    hr = 0xDEADFACE;
    if (pSrc)
        hr = pSrc->QueryValueDword(NULL, L"ValueA", &dwValue);
    ok_hr(hr, HRESULT_FROM_WIN32(ERROR_MORE_DATA));
    ok_long(dwValue, 0xDEADFACE);
 
    dwValue = 0xDEADFACE;
    hr = 0xDEADFACE;
    if (pSrc)
        hr = pSrc->QueryValueDword(NULL, L"ValueB", &dwValue);
    ok_hr(hr, S_OK);
    ok_long(dwValue, 0xBEEFCAFE);
 
    FLAGGED_BYTE_BLOB *pBlob;
 
    // QueryValueDirect
    hr = 0xDEADFACE;
    pBlob = NULL;
    if (pSrc)
        hr = pSrc->QueryValueDirect(NULL, L"ValueA", &pBlob);
    ok_hr(hr, S_OK);
    ok(pBlob != NULL, "pBlob was %p\n", pBlob);
    ok(pBlob && pBlob->clSize == 12, "pBlob->clSize was %ld\n", pBlob->clSize);
    ok(pBlob && !memcmp(pBlob->abData, L"hello", 12), "pBlob->abData mismatch\n");
    CoTaskMemFree(pBlob);
 
    hr = 0xDEADFACE;
    pBlob = NULL;
    if (pSrc)
        hr = pSrc->QueryValueDirect(NULL, L"ValueB", &pBlob);
    ok_hr(hr, S_OK);
    ok(pBlob != NULL, "pBlob was %p\n", pBlob);
    ok(pBlob && pBlob->clSize == 4, "pBlob->clSize was %ld\n", pBlob->clSize);
    dwValue = 0xBEEFCAFE;
    ok(pBlob && !memcmp(pBlob->abData, &dwValue, sizeof(dwValue)), "pBlob->abData mismatch\n");
    CoTaskMemFree(pBlob);
 
    if (pSrc)
        pSrc->Release();
}

Referenced by START_TEST().

Test_EnumSources()

static void Test_EnumSources ( void  )

static

Definition at line 116 of file QuerySourceCreateFromKey.cpp.

{
    IQuerySourceOld *pSrc = NULL;
    HRESULT hr = g_pQuerySourceCreateFromKey(HKEY_CURRENT_USER, k_Root, FALSE,
                                             IID_IQuerySourceOld, (PVOID*)&pSrc);
    ok_hr(hr, S_OK);
    ok(pSrc != NULL, "pSrc was NULL\n");
 
    IEnumString *pEnum = NULL;
    hr = pSrc->EnumSources(&pEnum);
    ok_hr(hr, S_OK);
    ok(pEnum != NULL, "pEnum was NULL\n");
 
    LPWSTR psz = NULL;
    ULONG fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_OK);
    ok(lstrcmpiW(psz, k_SubKeyA) == 0, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 1);
    CoTaskMemFree(psz);
 
    psz = NULL;
    fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_OK);
    ok(lstrcmpiW(psz, k_SubKeyB) == 0, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 1);
    CoTaskMemFree(psz);
 
    psz = NULL;
    fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_FALSE);
    ok(psz == NULL, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 0);
    CoTaskMemFree(psz);
 
    if (pSrc)
        pSrc->Release();
}

Referenced by START_TEST().

Test_EnumValues()

static void Test_EnumValues ( void  )

static

Definition at line 63 of file QuerySourceCreateFromKey.cpp.

{
    IQuerySourceOld *pSrc = NULL;
    HRESULT hr = g_pQuerySourceCreateFromKey(HKEY_CURRENT_USER, k_Root, FALSE,
                                             IID_IQuerySourceOld, (PVOID*)&pSrc);
    ok_hr(hr, S_OK);
    ok(pSrc != NULL, "pSrc was NULL\n");
 
    IEnumString *pEnum = NULL;
    hr = 0xDEADFACE;
    if (pSrc)
        hr = pSrc->EnumValues(&pEnum);
    ok(pSrc && hr == S_OK, "EnumValues failed: 0x%08X\n", hr);
 
    LPWSTR psz = NULL;
    ULONG fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_OK);
    ok(lstrcmpiW(psz, L"ValueA") == 0, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 1);
    CoTaskMemFree(psz);
 
    psz = NULL;
    fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_OK);
    ok(lstrcmpiW(psz, L"ValueB") == 0, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 1);
    CoTaskMemFree(psz);
 
    psz = NULL;
    fetched = 0;
    hr = 0xDEADFACE;
 
    if (pEnum)
        hr = pEnum->Next(1, &psz, &fetched);
    ok_hr(hr, S_FALSE);
    ok(psz == NULL, "psz was %s\n", wine_dbgstr_w(psz));
    ok_int(fetched, 0);
    CoTaskMemFree(psz);
 
    if (pEnum)
        pEnum->Release();
    if (pSrc)
        pSrc->Release();
}

Referenced by START_TEST().

Test_Version()

void Test_Version

(

void

)

Definition at line 245 of file QuerySourceCreateFromKey.cpp.

{
    IUnknown *pUnk, *pUnk2;
    REFIID riidQS = IsWindowsVistaOrGreater() ? IID_IQuerySource : IID_IQuerySourceOld;
    HRESULT hr;
    hr = ShimverQSCFK(HKEY_CURRENT_USER, k_Root, FALSE, (pUnk = NULL, (PVOID*)&pUnk));
    ok_hr(hr, S_OK);
    SafeRelease(pUnk);
 
    hr = QSCFK(HKEY_CURRENT_USER, k_Root, FALSE, IID_IUnknown, (pUnk = NULL, (PVOID*)&pUnk));
    hr = pUnk ? pUnk->QueryInterface(riidQS, (void**)&pUnk2) : E_FAIL;
    ok_hr(hr, S_OK);
    SafeRelease(pUnk2);
    SafeRelease(pUnk);
}

Referenced by START_TEST().

Variable Documentation

g_pQuerySourceCreateFromKey

FN_QuerySourceCreateFromKey g_pQuerySourceCreateFromKey = NULL

static

Definition at line 17 of file QuerySourceCreateFromKey.cpp.

Referenced by QSCFK(), START_TEST(), Test_CheckValues(), Test_EnumSources(), and Test_EnumValues().

k_Root

const WCHAR k_Root[] = L"Software\\QuerySrcTest"

static

Definition at line 34 of file QuerySourceCreateFromKey.cpp.

Referenced by CleanupRegistry(), SetupRegistry(), Test_CheckValues(), Test_EnumSources(), Test_EnumValues(), and Test_Version().

k_SubKeyA

const WCHAR k_SubKeyA[] = L"SubKeyA"

static

Definition at line 35 of file QuerySourceCreateFromKey.cpp.

Referenced by SetupRegistry(), and Test_EnumSources().

k_SubKeyB

const WCHAR k_SubKeyB[] = L"SubKeyB"

static

Definition at line 36 of file QuerySourceCreateFromKey.cpp.

Referenced by SetupRegistry(), and Test_EnumSources().