AssocQueryString.c File Reference

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

Include dependency graph for AssocQueryString.c:

Go to the source code of this file.

Macros
#define	NON_EXISTENT_FILENAME_A   "C:\\ThisIsNotExistentFile.txt"
#define	NON_EXISTENT_FILENAME_W   L"C:\\ThisIsNotExistentFile.txt"

Functions
static void	TEST_Start (void)
static void	TEST_End (void)
static void	TEST_AssocQueryStringA (void)
static void	TEST_AssocQueryStringW (void)
static void	TEST_TrickyPointers (void)
	START_TEST (AssocQueryString)

Variables
static CHAR	s_szTextFileA [MAX_PATH] = ""
static WCHAR	s_szTextFileW [MAX_PATH] = L""

Macro Definition Documentation

NON_EXISTENT_FILENAME_A

#define NON_EXISTENT_FILENAME_A   "C:\\ThisIsNotExistentFile.txt"

Definition at line 15 of file AssocQueryString.c.

NON_EXISTENT_FILENAME_W

#define NON_EXISTENT_FILENAME_W   L"C:\\ThisIsNotExistentFile.txt"

Definition at line 16 of file AssocQueryString.c.

Function Documentation

START_TEST()

START_TEST

(

AssocQueryString

)

Definition at line 190 of file AssocQueryString.c.

{
    HRESULT hrCoInit = CoInitialize(NULL);
 
    TEST_Start();
    TEST_AssocQueryStringA();
    TEST_AssocQueryStringW();
    TEST_TrickyPointers();
    TEST_End();
 
    if (SUCCEEDED(hrCoInit))
        CoUninitialize();
}

TEST_AssocQueryStringA()

static void TEST_AssocQueryStringA ( void  )

static

Definition at line 37 of file AssocQueryString.c.

{
    /* Note: NT5 differs from NT6 because NT5 converts a NULL pszExtra to L"" */
    CHAR szPath[MAX_PATH], szAnswer[MAX_PATH];
    CHAR szDebug1[MAX_PATH], szDebug2[MAX_PATH];
    HRESULT hr;
    DWORD cch;
 
    /* ".txt" */
    lstrcpynA(szPath, ".txt", _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    if (IsWindowsVistaOrGreater())
    {
        ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
        ok_long(hr, S_OK);
        ok_int(cch, lstrlenA(szAnswer) + 1);
    }
    else
    {
        lstrcpynA(szAnswer, ".txt", _countof(szAnswer));
        ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
        ok_int(cch, _countof(szPath));
    }
    lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
    ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
 
    /* s_szTextFileA */
    lstrcpynA(szPath, s_szTextFileA, _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    if (IsWindowsVistaOrGreater())
    {
        ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
        ok_long(hr, S_OK);
        ok_int(cch, lstrlenA(szAnswer) + 1);
    }
    else
    {
        lstrcpynA(szAnswer, s_szTextFileA, _countof(szAnswer));
        ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
        ok_int(cch, _countof(szPath));
    }
    lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
    ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
 
    /* NON_EXISTENT_FILENAME_A */
    lstrcpynA(szPath, NON_EXISTENT_FILENAME_A, _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    if (IsWindowsVistaOrGreater())
    {
        ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
        ok_long(hr, S_OK);
        ok_int(cch, lstrlenA(szAnswer) + 1);
    }
    else
    {
        lstrcpynA(szAnswer, NON_EXISTENT_FILENAME_A, _countof(szAnswer));
        ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
        ok_int(cch, _countof(szPath));
    }
    lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
    ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
}

Referenced by START_TEST().

TEST_AssocQueryStringW()

static void TEST_AssocQueryStringW ( void  )

static

Definition at line 106 of file AssocQueryString.c.

{
    WCHAR szPath[MAX_PATH], szAnswer[MAX_PATH];
    CHAR szDebug1[MAX_PATH], szDebug2[MAX_PATH];
    HRESULT hr;
    DWORD cch;
 
    /* ".txt" */
    lstrcpynW(szPath, L".txt", _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    ok_long(hr, S_OK);
    ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
    lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
    ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
    ok_int(cch, lstrlenW(szAnswer) + 1);
 
    /* s_szTextFileW */
    lstrcpynW(szPath, s_szTextFileW, _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    ok_long(hr, S_OK);
    ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
    lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
    ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
    ok_int(cch, lstrlenW(szAnswer) + 1);
 
    /* NON_EXISTENT_FILENAME_W */
    lstrcpynW(szPath, NON_EXISTENT_FILENAME_W, _countof(szPath));
    cch = _countof(szPath);
    hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
    ok_long(hr, S_OK);
    ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
    lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
    lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
    ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
    ok_int(cch, lstrlenW(szAnswer) + 1);
}

Referenced by START_TEST().

TEST_End()

static void TEST_End ( void  )

static

Definition at line 27 of file AssocQueryString.c.

{
    DeleteFileW(s_szTextFileW);
}

Referenced by START_TEST().

TEST_Start()

static void TEST_Start ( void  )

static

Definition at line 18 of file AssocQueryString.c.

{
    ExpandEnvironmentStringsA( "%APPDATA%\\This is a test.txt", s_szTextFileA, _countof(s_szTextFileA));
    ExpandEnvironmentStringsW(L"%APPDATA%\\This is a test.txt", s_szTextFileW, _countof(s_szTextFileW));
    fclose(_wfopen(s_szTextFileW, L"w"));
    trace("%s\n", wine_dbgstr_a(s_szTextFileA));
    trace("%s\n", wine_dbgstr_w(s_szTextFileW));
}

Referenced by START_TEST().

TEST_TrickyPointers()

static void TEST_TrickyPointers ( void  )

static

Definition at line 147 of file AssocQueryString.c.

{
    HRESULT hr;
    CHAR  szPathA[MAX_PATH], szOutputA[MAX_PATH];
    WCHAR szPathW[MAX_PATH], szOutputW[MAX_PATH];
    DWORD cch;
    BOOL bVistaPlus = IsWindowsVistaOrGreater();
 
    lstrcpynA(szPathA, s_szTextFileA, _countof(szPathA));
    cch = _countof(szOutputA);
    _SEH2_TRY
    {
        hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPathA, NULL, szOutputA, (PDWORD)UlongToPtr(cch));
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        hr = 0xDEADFACE;
    }
    _SEH2_END;
 
    if (bVistaPlus)
        ok_long(hr, 0xDEADFACE);
    else
        ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
 
    lstrcpynW(szPathW, s_szTextFileW, _countof(szPathW));
    cch = _countof(szOutputW);
    _SEH2_TRY
    {
        hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPathW, NULL, szOutputW, (PDWORD)UlongToPtr(cch));
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        hr = 0xDEADFACE;
    }
    _SEH2_END;
 
    if (bVistaPlus)
        ok_long(hr, 0xDEADFACE);
    else
        ok_long(hr, S_OK);
}

Referenced by START_TEST().

Variable Documentation

s_szTextFileA

CHAR s_szTextFileA[MAX_PATH] = ""

static

Definition at line 13 of file AssocQueryString.c.

Referenced by TEST_AssocQueryStringA(), TEST_Start(), and TEST_TrickyPointers().

s_szTextFileW

WCHAR s_szTextFileW[MAX_PATH] = L""

static

Definition at line 14 of file AssocQueryString.c.

Referenced by TEST_AssocQueryStringW(), TEST_End(), TEST_Start(), and TEST_TrickyPointers().