AssocQueryKey.c File Reference

#include <apitest.h>
#include <windef.h>
#include <shlwapi.h>
#include <pseh/pseh2.h>
#include <shlobj.h>
#include <shlwapi_undoc.h>

Include dependency graph for AssocQueryKey.c:

Go to the source code of this file.

Classes
struct	_MY_KEY_NAME_INFORMATION

Macros
#define	ok_AqkPath(InitF, AssocKey, pszInit, pszExtra, hrExpected, pszExpectedWin32)

Typedefs
typedef enum _MY_KEY_INFORMATION_CLASS	MY_KEY_INFORMATION_CLASS
typedef struct _MY_KEY_NAME_INFORMATION	MY_KEY_NAME_INFORMATION
typedef struct _MY_KEY_NAME_INFORMATION *	PMY_KEY_NAME_INFORMATION
typedef NTSTATUS(__stdcall *	FN_NtQueryKey) (HANDLE, MY_KEY_INFORMATION_CLASS, PVOID, ULONG, PULONG)

Enumerations
enum	{   HR_FNF = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND) , HR_NF = HRESULT_FROM_WIN32(ERROR_NOT_FOUND) , HR_NOASSOC = HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION) , HR_ACCDENIED = HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED) ,   AF_DEFSTAR = ASSOCF_INIT_DEFAULTTOSTAR , AF_DEFFOLDER = ASSOCF_INIT_DEFAULTTOFOLDER }
enum	_MY_KEY_INFORMATION_CLASS { MyKeyBasicInformation = 0 , MyKeyNodeInformation = 1 , MyKeyFullInformation = 2 , MyKeyNameInformation = 3 }

Functions
static void	Cleanup (void)
static BOOL	InitNtQueryKey (VOID)
static LPWSTR	GetKeyPath (HKEY hKey)
static BOOL	IsValidClassesRootPath (LPCWSTR path)
static HRESULT	SHELL_SetRegString (HKEY hKey, LPCWSTR pszSubKey, LPCWSTR pszName, UINT Type, LPCWSTR pszData)
static HRESULT	AQS (ASSOCF flags, ASSOCSTR str, LPCWSTR pszAsso, LPCWSTR pszExtra, LPWSTR pszOut)
static HRESULT	GetAqkPath (UINT InitF, ASSOCKEY AssocKey, LPCWSTR pszInit, LPCWSTR pszExtra, LPWSTR *ppszKey)
static HKEY	CreateClassesKey (HKEY hRoot)
static HKEY	CreateClassesSubKey (HKEY hRoot, LPCWSTR pszSubKey)
static void	TestClassKey (ASSOCKEY AssocKey)
static void	TEST_AssocKeys (void)
static void	TEST_AssocF_Flags (void)
static void	TEST_PszAssoc (void)
static void	TEST_InvalidArgs (void)
static void	TEST_ByExeName (void)
static void	TEST_Class (void)
static void	TEST_BaseClass (void)
static void	TEST_ShellExecClass (void)
	START_TEST (AssocQueryKey)

Variables
static const WCHAR	pszUniqueExt [] = L".RosTestExtShlwapi"
static const WCHAR	pszUniquePid [] = L"FEF31D544EA64C8EBC50B8719887688A"
static const WCHAR	pszUniquePid2 [] = L"88FDE0879CD24AEBBEB690988BD87706"
static FN_NtQueryKey	g_NtQueryKey = NULL

Macro Definition Documentation

ok_AqkPath

#define ok_AqkPath	(		InitF,
			AssocKey,
			pszInit,
			pszExtra,
			hrExpected,
			pszExpectedWin32
	)

Value:

do \
{ \
    LPCWSTR pszExpected = L"\\REGISTRY\\MACHINE\\" pszExpectedWin32; \
    LPWSTR pszPath; \
    HRESULT hr = GetAqkPath((InitF), (AssocKey), (pszInit), (pszExtra), &pszPath); \
    ok_hr(hr, (hrExpected)); \
    if (SUCCEEDED(hr)) \
    { \
        ok(pszPath && !_wcsicmp(pszPath, pszExpected), "Got \"%s\", expected \"%s\"\n", \
                                                       wine_dbgstr_w(pszPath), wine_dbgstr_w(pszExpected)); \
    } \
    LocalFree(pszPath); \
} while (0)

Definition at line 126 of file AssocQueryKey.c.

Typedef Documentation

FN_NtQueryKey

typedef NTSTATUS(__stdcall * FN_NtQueryKey) (HANDLE, MY_KEY_INFORMATION_CLASS, PVOID, ULONG, PULONG)

Definition at line 55 of file AssocQueryKey.c.

MY_KEY_INFORMATION_CLASS

typedef enum _MY_KEY_INFORMATION_CLASS MY_KEY_INFORMATION_CLASS

MY_KEY_NAME_INFORMATION

typedef struct _MY_KEY_NAME_INFORMATION MY_KEY_NAME_INFORMATION

PMY_KEY_NAME_INFORMATION

typedef struct _MY_KEY_NAME_INFORMATION * PMY_KEY_NAME_INFORMATION

Enumeration Type Documentation

anonymous enum

anonymous enum

Enumerator
HR_FNF
HR_NF
HR_NOASSOC
HR_ACCDENIED
AF_DEFSTAR
AF_DEFFOLDER

Definition at line 30 of file AssocQueryKey.c.

{
    HR_FNF = HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND),
    HR_NF = HRESULT_FROM_WIN32(ERROR_NOT_FOUND),
    HR_NOASSOC = HRESULT_FROM_WIN32(ERROR_NO_ASSOCIATION),
    HR_ACCDENIED = HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED),
    AF_DEFSTAR = ASSOCF_INIT_DEFAULTTOSTAR,
    AF_DEFFOLDER = ASSOCF_INIT_DEFAULTTOFOLDER,
};

_MY_KEY_INFORMATION_CLASS

enum _MY_KEY_INFORMATION_CLASS

Enumerator
MyKeyBasicInformation
MyKeyNodeInformation
MyKeyFullInformation
MyKeyNameInformation

Definition at line 40 of file AssocQueryKey.c.

{
    MyKeyBasicInformation = 0,
    MyKeyNodeInformation  = 1,
    MyKeyFullInformation  = 2,
    MyKeyNameInformation  = 3,
} MY_KEY_INFORMATION_CLASS;

Function Documentation

AQS()

static HRESULT AQS ( ASSOCF  flags, ASSOCSTR  str, LPCWSTR  pszAsso, LPCWSTR  pszExtra, LPWSTR  pszOut  )

static

Definition at line 109 of file AssocQueryKey.c.

{
    DWORD cch = MAX_PATH;
    return AssocQueryStringW(flags, str, pszAsso, pszExtra, pszOut, &cch);
}

Referenced by TestClassKey().

Cleanup()

static void Cleanup ( void  )

static

Definition at line 23 of file AssocQueryKey.c.

{
    SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniqueExt), SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniqueExt);
    SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniquePid), SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniquePid);
    SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniquePid2), SHDeleteKeyW(HKEY_CLASSES_ROOT, pszUniquePid2);
}

Referenced by START_TEST(), and TestClassKey().

CreateClassesKey()

static HKEY CreateClassesKey ( HKEY  hRoot )

static

Definition at line 140 of file AssocQueryKey.c.

{
    if (hRoot == HKEY_CLASSES_ROOT)
        return hRoot;
    if (RegCreateKeyExW(hRoot, L"Software\\Classes", 0, NULL, 0, KEY_WRITE | KEY_READ, NULL, &hRoot, NULL))
        return NULL;
    return hRoot;
}

Referenced by CreateClassesSubKey().

CreateClassesSubKey()

static HKEY CreateClassesSubKey ( HKEY  hRoot, LPCWSTR  pszSubKey  )

static

Definition at line 149 of file AssocQueryKey.c.

{
    hRoot = CreateClassesKey(hRoot);
    if (hRoot && pszSubKey)
    {
        HKEY hSubKey;
        LONG err = RegCreateKeyExW(hRoot, pszSubKey, 0, NULL, 0, KEY_WRITE | KEY_READ, NULL, &hSubKey, NULL);
        if (hRoot != HKEY_CLASSES_ROOT)
            RegCloseKey(hRoot);
        return err ? NULL : hSubKey;
    }
    return hRoot;
}

Referenced by TestClassKey().

GetAqkPath()

static HRESULT GetAqkPath ( UINT  InitF, ASSOCKEY  AssocKey, LPCWSTR  pszInit, LPCWSTR  pszExtra, LPWSTR *  ppszKey  )

static

Definition at line 115 of file AssocQueryKey.c.

{
    HKEY hKey = NULL;
    HRESULT hr = AssocQueryKeyW((ASSOCF)InitF, AssocKey, pszInit, pszExtra, &hKey);
    if (ppszKey)
        *ppszKey = GetKeyPath(hKey);
    if (SUCCEEDED(hr) && hKey)
        RegCloseKey(hKey);
    return hr == S_OK && !hKey ? S_FALSE : hr;
}

Referenced by TEST_ByExeName(), and TestClassKey().

GetKeyPath()

static LPWSTR GetKeyPath ( HKEY  hKey )

static

Definition at line 67 of file AssocQueryKey.c.

{
    if (!g_NtQueryKey || !hKey)
        return NULL;
 
    ULONG needed = 0;
    g_NtQueryKey(hKey, MyKeyNameInformation, NULL, 0, &needed);
 
    if (!needed)
        return NULL;
 
    SIZE_T cb = needed + sizeof(WCHAR);
    PBYTE buf = (PBYTE)_alloca(cb);
    ZeroMemory(buf, cb);
 
    ULONG returned = 0;
    NTSTATUS status = g_NtQueryKey(hKey, MyKeyNameInformation, buf, (ULONG)cb, &returned);
 
    if (status < 0)
        return NULL;
 
    PMY_KEY_NAME_INFORMATION info = (PMY_KEY_NAME_INFORMATION)buf;
    return StrDupW(info->Name); // needs LocalFree
}

Referenced by GetAqkPath(), TEST_AssocF_Flags(), TEST_AssocKeys(), TEST_PszAssoc(), and TEST_ShellExecClass().

InitNtQueryKey()

static BOOL InitNtQueryKey ( VOID  )

static

Definition at line 58 of file AssocQueryKey.c.

{
    HMODULE hNTDLL = GetModuleHandleW(L"ntdll.dll");
    if (!hNTDLL) return FALSE;
    g_NtQueryKey = (FN_NtQueryKey)GetProcAddress(hNTDLL, "NtQueryKey");
    return g_NtQueryKey != NULL;
}

Referenced by START_TEST().

IsValidClassesRootPath()

static BOOL IsValidClassesRootPath ( LPCWSTR  path )

static

Definition at line 92 of file AssocQueryKey.c.

{
    LPCWSTR base;
    if (!path)
        return FALSE;
    if (path == StrStrIW(path, L"\\REGISTRY\\MACHINE\\SOFTWARE\\Classes\\"))
        return TRUE;
    base = StrStrIW(path, L"\\REGISTRY\\USER\\");
    return base && StrStrIW(path, L"_CLASSES\\") > base;
}

Referenced by TEST_ShellExecClass().

SHELL_SetRegString()

static HRESULT SHELL_SetRegString ( HKEY  hKey, LPCWSTR  pszSubKey, LPCWSTR  pszName, UINT  Type, LPCWSTR  pszData  )

static

Definition at line 103 of file AssocQueryKey.c.

{
    ULONG err = SHSetValueW(hKey, pszSubKey, pszName, Type, pszData, (lstrlenW(pszData) + 1) * sizeof(*pszData));
    return HRESULT_FROM_WIN32(err);
}

Referenced by TestClassKey().

START_TEST()

START_TEST

(

AssocQueryKey

)

Definition at line 617 of file AssocQueryKey.c.

{
    if (!InitNtQueryKey())
    {
        skip("NtQueryKey not found\n");
        return;
    }
 
    HRESULT hrCoInit = CoInitialize(NULL);
 
    TEST_AssocKeys();
    TEST_AssocF_Flags();
    TEST_PszAssoc();
    TEST_InvalidArgs();
    TEST_ByExeName();
    TEST_Class();
    TEST_BaseClass();
    TEST_ShellExecClass();
 
    Cleanup();
    if (SUCCEEDED(hrCoInit))
        CoUninitialize();
}

TEST_AssocF_Flags()

static void TEST_AssocF_Flags ( void  )

static

Definition at line 359 of file AssocQueryKey.c.

{
    const ASSOCF cases[] =
    {
        ASSOCF_NONE,
        ASSOCF_INIT_NOREMAPCLSID,
        ASSOCF_INIT_DEFAULTTOSTAR,
        ASSOCF_INIT_DEFAULTTOFOLDER,
        ASSOCF_NOUSERSETTINGS,
        ASSOCF_NOTRUNCATE,
        ASSOCF_VERIFY,
        ASSOCF_REMAPRUNDLL,
        ASSOCF_NOFIXUPS,
        ASSOCF_IGNOREBASECLASS,
        ASSOCF_INIT_IGNOREUNKNOWN,
        ASSOCF_INIT_NOREMAPCLSID | ASSOCF_NOUSERSETTINGS,
        ASSOCF_NOTRUNCATE | ASSOCF_VERIFY,
        ASSOCF_PER_MACHINE_ONLY,
    };
 
    for (size_t i = 0; i < _countof(cases); ++i)
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(cases[i], ASSOCKEY_CLASS, L".reg", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        LPCWSTR pszExpectedLM = L"\\REGISTRY\\MACHINE\\";
        LPCWSTR pszExpectedCU = L"\\REGISTRY\\USER\\";
        if (cases[i] & ASSOCF_PER_MACHINE_ONLY)
            pszExpectedCU = pszExpectedLM;
        ok(path &&
           (StrStrIW(path, pszExpectedLM) || StrStrIW(path, pszExpectedCU)),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
}

Referenced by START_TEST().

TEST_AssocKeys()

static void TEST_AssocKeys ( void  )

static

Definition at line 294 of file AssocQueryKey.c.

{
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_SHELLEXECCLASS, L".reg", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path &&
           (StrStrIW(path, L"\\REGISTRY\\MACHINE\\") || StrStrIW(path, L"\\REGISTRY\\USER\\")) &&
           StrStrIW(path, L"regfile"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_APP, L".reg", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path && StrStrIW(path,
                            L"\\REGISTRY\\MACHINE\\SOFTWARE\\Classes\\Applications\\regedit.exe"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L".reg", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path &&
           (StrStrIW(path, L"\\REGISTRY\\MACHINE\\") || StrStrIW(path, L"\\REGISTRY\\USER\\")) &&
           StrStrIW(path, L"Classes\\regfile"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L".exe", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path && !_wcsicmp(path, L"\\REGISTRY\\MACHINE\\SOFTWARE\\Classes\\exefile"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
}

Referenced by START_TEST().

TEST_BaseClass()

static void TEST_BaseClass ( void  )

static

Definition at line 587 of file AssocQueryKey.c.

{
    ok_AqkPath(AF_DEFSTAR, ASSOCKEY_BASECLASS, pszUniqueExt, NULL, S_OK, L"SOFTWARE\\Classes\\*");
    ok_AqkPath(AF_DEFFOLDER, ASSOCKEY_BASECLASS, L"Folder", NULL, S_OK, L"SOFTWARE\\Classes\\Folder");
    ok_AqkPath(AF_DEFFOLDER, ASSOCKEY_BASECLASS, L"Directory", NULL, S_OK, L"SOFTWARE\\Classes\\Folder");
}

Referenced by START_TEST().

TEST_ByExeName()

static void TEST_ByExeName ( void  )

static

Definition at line 547 of file AssocQueryKey.c.

{
    HRESULT hr;
    BOOL nt5 = LOBYTE(GetVersion()) < 6;
    WCHAR notepadPath[MAX_PATH];
    GetSystemDirectoryW(notepadPath, MAX_PATH);
    lstrcatW(notepadPath, L"\\notepad.exe");
 
    // Full path
    ok_AqkPath(ASSOCF_INIT_BYEXENAME, ASSOCKEY_APP, notepadPath, NULL,
               S_OK, L"SOFTWARE\\Classes\\Applications\\notepad.exe");
 
    // Filename.exe
    ok_AqkPath(ASSOCF_OPEN_BYEXENAME, ASSOCKEY_APP, L"regedit.exe", NULL,
               S_OK, L"SOFTWARE\\Classes\\Applications\\regedit.exe");
 
    // Filename (no extension)
    ok_AqkPath(ASSOCF_OPEN_BYEXENAME, ASSOCKEY_APP, L"regedit", NULL,
               S_OK, L"SOFTWARE\\Classes\\Applications\\regedit.exe");
 
    // Filename.exe with Extra
    ok_AqkPath(ASSOCF_OPEN_BYEXENAME, ASSOCKEY_APP, L"regedit.exe", L"Extra",
               S_OK, L"SOFTWARE\\Classes\\Applications\\regedit.exe");
 
    // Non existent
    ok_AqkPath(ASSOCF_INIT_BYEXENAME, ASSOCKEY_APP, L"__does_not_exist__.exe", NULL,
               nt5 ? HR_FNF : HR_NOASSOC, -0);
 
    // Without BYEXENAME
    hr = GetAqkPath(ASSOCF_NONE, ASSOCKEY_APP, L"regedit.exe", NULL, NULL);
    ok_hr(hr, nt5 ? E_FAIL : HR_NOASSOC);
}

Referenced by START_TEST().

TEST_Class()

static void TEST_Class ( void  )

static

Definition at line 581 of file AssocQueryKey.c.

{
    TestClassKey(ASSOCKEY_CLASS);
}

Referenced by START_TEST().

TEST_InvalidArgs()

static void TEST_InvalidArgs ( void  )

static

Definition at line 465 of file AssocQueryKey.c.

{
    // phkeyOut == NULL
    {
        BOOL threw = FALSE;
        HRESULT hr = E_UNEXPECTED;
 
        _SEH2_TRY
        {
            hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L".txt", NULL, NULL);
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            hr = 0xDEADFACE;
            threw = TRUE;
        }
        _SEH2_END;
 
        ok_hr(hr, 0xDEADFACE);
        ok_int(threw, TRUE);
    }
 
    // pszAssoc == NULL
    {
        BOOL threw = FALSE;
        HRESULT hr = E_UNEXPECTED;
        HKEY hKey = NULL;
 
        _SEH2_TRY
        {
            hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, NULL, NULL, &hKey);
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            hr = 0xDEADFACE;
            threw = TRUE;
        }
        _SEH2_END;
 
        ok(hKey == NULL, "hKey was not NULL\n");
        if (hKey)
            RegCloseKey(hKey);
 
        ok_hr(hr, HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER));
        ok_int(threw, FALSE);
    }
 
    // pszAssoc == ""
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L"", NULL, &hKey);
        if (hKey)
            RegCloseKey(hKey);
        ok_hr(hr, HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER));
    }
 
    // Invalid ASSOCKEY value
    {
        HKEY hKey = NULL;
        BOOL threw = FALSE;
        HRESULT hr = E_UNEXPECTED;
 
        _SEH2_TRY
        {
            hr = AssocQueryKeyW(ASSOCF_NONE, (ASSOCKEY)0xFFFF, L".txt", NULL, &hKey);
        }
        _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
        {
            hr = 0xDEADFACE;
            threw = TRUE;
        }
        _SEH2_END;
 
        ok(hKey == NULL, "hKey was not NULL\n");
        if (hKey)
            RegCloseKey(hKey);
 
        ok_hr(hr, HRESULT_FROM_WIN32(ERROR_INVALID_PARAMETER));
        ok_int(threw, FALSE);
    }
}

Referenced by START_TEST().

TEST_PszAssoc()

static void TEST_PszAssoc ( void  )

static

Definition at line 401 of file AssocQueryKey.c.

{
    const wchar_t* exts[] =
    {
        L".txt", L".htm", L".reg", L".xml", L".png", L".jpg", L".zip"
    };
 
    for (size_t i = 0; i < _countof(exts); ++i)
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, exts[i], NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path &&
           (StrStrIW(path, L"\\REGISTRY\\MACHINE\\") || StrStrIW(path, L"\\REGISTRY\\USER\\")),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    // Direct ProgID
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L"regfile", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path &&
           (StrStrIW(path, L"\\REGISTRY\\MACHINE\\") || StrStrIW(path, L"\\REGISTRY\\USER\\")) &&
           StrStrIW(path, L"regfile"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L"exefile", NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(path && !_wcsicmp(path, L"\\REGISTRY\\MACHINE\\SOFTWARE\\Classes\\exefile"),
           "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    // ProgID: InternetShortcut
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_CLASS, L"InternetShortcut", NULL, &hKey);
        ok_hr(hr, S_OK);
        if (hKey)
            RegCloseKey(hKey);
    }
}

Referenced by START_TEST().

TEST_ShellExecClass()

static void TEST_ShellExecClass ( void  )

static

Definition at line 596 of file AssocQueryKey.c.

{
    static const LPCWSTR cases[] = { L".txt", L".htm", L".html", L"txtfile" };
    for (size_t i = 0; i < _countof(cases); ++i)
    {
        HKEY hKey = NULL;
        HRESULT hr = AssocQueryKeyW(ASSOCF_NONE, ASSOCKEY_SHELLEXECCLASS, cases[i], NULL, &hKey);
        ok_hr(hr, S_OK);
 
        LPWSTR path = GetKeyPath(hKey);
        if (hKey)
            RegCloseKey(hKey);
 
        ok(IsValidClassesRootPath(path), "path was %s\n", wine_dbgstr_w(path));
        LocalFree(path);
    }
 
    TestClassKey(ASSOCKEY_SHELLEXECCLASS);
    // TODO: ASSOCF_VERIFY
}

Referenced by START_TEST().

TestClassKey()

static void TestClassKey ( ASSOCKEY  AssocKey )

static

Definition at line 163 of file AssocQueryKey.c.

{
    BOOL nt5 = LOBYTE(GetVersion()) < 6;
    HKEY hSource, hClass, hClass2;
    HRESULT hr, hrExpect;
    WCHAR buf[MAX_PATH], *pszPath;
    Cleanup();
 
    // A ProgId that does not exist
    hr = GetAqkPath(ASSOCF_NONE, AssocKey, pszUniquePid, NULL, NULL);
    hrExpect = nt5 ? (E_FAIL) : (AssocKey == ASSOCKEY_SHELLEXECCLASS ? HR_NOASSOC : HR_NF);
    ok_hr(hr, hrExpect);
 
    // An extension that does not exist (could map to Unknown)
    hr = GetAqkPath(ASSOCF_NONE, AssocKey, pszUniqueExt, NULL, NULL);
    hrExpect = nt5 ? E_FAIL : S_OK;
    ok_hr(hr, hrExpect);
 
    // An extension that does not exist
    hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, NULL, NULL);
    hrExpect = nt5 ? E_FAIL : (AssocKey == ASSOCKEY_SHELLEXECCLASS ? HR_NOASSOC : S_OK);
    ok_hr(hr, hrExpect);
 
    hSource = CreateClassesSubKey(HKEY_CURRENT_USER, pszUniqueExt);
    if (!hSource)
    {
        skip("Could not create test key\n");
        return;
    }
 
    // An extension with no ProgId (could map to Unknown)
    hr = GetAqkPath(ASSOCF_NONE, AssocKey, pszUniqueExt, NULL, NULL);
    hrExpect = nt5 && AssocKey == ASSOCKEY_SHELLEXECCLASS ? HR_FNF : S_OK;
    ok_hr(hr, hrExpect);
 
    // An extension with no ProgId
    hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, NULL, NULL);
    hrExpect = AssocKey == ASSOCKEY_SHELLEXECCLASS ? (nt5 ? HR_FNF : HR_NOASSOC) : S_OK;
    ok_hr(hr, hrExpect);
 
    // An extension with only a fallback command
    if (SHELL_SetRegString(hSource, L"shell\\open\\command", NULL, REG_SZ, L"calc.exe"))
    {
        skip("Could not create test key\n");
    }
    else
    {
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, NULL, NULL);
        ok_hr(hr, S_OK);
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, L"open", NULL);
        ok_hr(hr, S_OK);
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, L"", NULL);
        ok_hr(hr, S_OK); // Strange but true
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, L"badverb", NULL);
        hrExpect = AssocKey == ASSOCKEY_SHELLEXECCLASS ? (nt5 ? HR_FNF : HR_NOASSOC) : S_OK;
        ok_hr(hr, hrExpect);
        SHDeleteKeyW(hSource, L"shell");
    }
 
    // ProgId
    if ((hClass = CreateClassesSubKey(HKEY_CURRENT_USER, pszUniquePid)) == NULL)
    {
        skip("Could not create test key\n");
    }
    else
    {
        SHELL_SetRegString(hSource, NULL, NULL, REG_SZ, pszUniquePid);
        if (SHELL_SetRegString(hClass, L"shell\\calcverb\\command", NULL, REG_SZ, L"calc.exe"))
        {
            skip("Could not create test key\n");
            goto skip_progidtests;
        }
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, NULL, NULL);
        ok_hr(hr, S_OK);
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, L"calcverb", NULL);
        ok_hr(hr, S_OK);
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN, AssocKey, pszUniqueExt, L"open", NULL);
        hrExpect = AssocKey == ASSOCKEY_SHELLEXECCLASS ? (nt5 ? HR_FNF : HR_NOASSOC) : S_OK;
        ok_hr(hr, hrExpect);
 
        hr = GetAqkPath(ASSOCF_INIT_IGNOREUNKNOWN | ASSOCF_NOUSERSETTINGS, AssocKey, pszUniqueExt, NULL, NULL);
        ok_hr(hr, S_OK);
 
        // CurVer redirection
        if ((hClass2 = CreateClassesSubKey(HKEY_CURRENT_USER, pszUniquePid2)) == NULL)
        {
            skip("Could not create test key\n");
            goto skip_progidtests;
        }
        else
        {
            LPCWSTR pszExe2 = L"calc.exe";
            ASSOCMAKEVERB verbs2[] =
            {
                { L"calcverb2", NULL, NULL, pszExe2, L"CurVerParam" },
            };
            ASSOCMAKESHELL shell2 = { verbs2, _countof(verbs2), 0 };
            AssocMakeShell(0, hClass2, pszExe2, &shell2);
            SHELL_SetRegString(hClass, L"CurVer", NULL, REG_SZ, pszUniquePid2);
 
            hr = GetAqkPath(ASSOCF_NONE, AssocKey, pszUniqueExt, NULL, &pszPath);
            ok_hr(hr, S_OK);
            ok(pszPath && StrStrIW(pszPath, pszUniquePid2), "CurVer ProgId redirection\n");
            LocalFree(pszPath), pszPath = NULL;
 
            hr = AQS(ASSOCF_NONE, ASSOCSTR_COMMAND, pszUniqueExt, NULL, buf);
            ok(SUCCEEDED(hr) && StrStrIW(buf, verbs2[0].pszArgs), "CurVer ProgId redirection\n");
 
            SHDeleteKeyW(hClass2, L"");
            RegCloseKey(hClass2);
        }
 
skip_progidtests:
        SHDeleteKeyW(hClass, L"");
        RegCloseKey(hClass);
    }
 
    if (hSource)
    {
        SHDeleteKeyW(hSource, L"");
        RegCloseKey(hSource);
    }
}

Referenced by TEST_Class(), and TEST_ShellExecClass().

Variable Documentation

g_NtQueryKey

FN_NtQueryKey g_NtQueryKey = NULL

static

Definition at line 56 of file AssocQueryKey.c.

Referenced by GetKeyPath(), and InitNtQueryKey().

pszUniqueExt

const WCHAR pszUniqueExt[] = L".RosTestExtShlwapi"

static

Definition at line 19 of file AssocQueryKey.c.

Referenced by Cleanup(), TEST_BaseClass(), and TestClassKey().

pszUniquePid

const WCHAR pszUniquePid[] = L"FEF31D544EA64C8EBC50B8719887688A"

static

Definition at line 20 of file AssocQueryKey.c.

Referenced by Cleanup(), and TestClassKey().

pszUniquePid2

const WCHAR pszUniquePid2[] = L"88FDE0879CD24AEBBEB690988BD87706"

static

Definition at line 21 of file AssocQueryKey.c.

Referenced by Cleanup(), and TestClassKey().