ReactOS 0.4.17-dev-684-ga6524ef
AssocQueryString.c
Go to the documentation of this file.
1/*
2 * PROJECT: ReactOS api tests
3 * LICENSE: LGPL-2.0-or-later (https://spdx.org/licenses/LGPL-2.0-or-later)
4 * PURPOSE: Test for AssocQueryStringA/W
5 * COPYRIGHT: Copyright 2024-2026 Katayama Hirofumi MZ (katayama.hirofumi.mz@gmail.com)
6 */
7
8#include <apitest.h>
9#include <shlwapi.h>
10#include <stdio.h>
11#include <versionhelpers.h>
12
13static CHAR s_szTextFileA[MAX_PATH] = "";
14static WCHAR s_szTextFileW[MAX_PATH] = L"";
15#define NON_EXISTENT_FILENAME_A "C:\\ThisIsNotExistentFile.txt"
16#define NON_EXISTENT_FILENAME_W L"C:\\ThisIsNotExistentFile.txt"
17
18static void TEST_Start(void)
19{
20 ExpandEnvironmentStringsA( "%APPDATA%\\This is a test.txt", s_szTextFileA, _countof(s_szTextFileA));
21 ExpandEnvironmentStringsW(L"%APPDATA%\\This is a test.txt", s_szTextFileW, _countof(s_szTextFileW));
22 fclose(_wfopen(s_szTextFileW, L"w"));
23 trace("%s\n", wine_dbgstr_a(s_szTextFileA));
24 trace("%s\n", wine_dbgstr_w(s_szTextFileW));
25}
26
27static void TEST_End(void)
28{
29 DeleteFileW(s_szTextFileW);
30}
31
32/*
33 * "shlwapi_winetest assoc" already has many tests.
34 * We just do additional tests in here.
35 */
36
37static void TEST_AssocQueryStringA(void)
38{
39 /* Note: NT5 differs from NT6 because NT5 converts a NULL pszExtra to L"" */
40 CHAR szPath[MAX_PATH], szAnswer[MAX_PATH];
41 CHAR szDebug1[MAX_PATH], szDebug2[MAX_PATH];
42 HRESULT hr;
43 DWORD cch;
44
45 /* ".txt" */
46 lstrcpynA(szPath, ".txt", _countof(szPath));
47 cch = _countof(szPath);
48 hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
49 if (IsWindowsVistaOrGreater())
50 {
51 ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
52 ok_long(hr, S_OK);
53 ok_int(cch, lstrlenA(szAnswer) + 1);
54 }
55 else
56 {
57 lstrcpynA(szAnswer, ".txt", _countof(szAnswer));
58 ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
59 ok_int(cch, _countof(szPath));
60 }
61 lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
62 lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
63 ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
64
65 /* s_szTextFileA */
66 lstrcpynA(szPath, s_szTextFileA, _countof(szPath));
67 cch = _countof(szPath);
68 hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
69 if (IsWindowsVistaOrGreater())
70 {
71 ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
72 ok_long(hr, S_OK);
73 ok_int(cch, lstrlenA(szAnswer) + 1);
74 }
75 else
76 {
77 lstrcpynA(szAnswer, s_szTextFileA, _countof(szAnswer));
78 ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
79 ok_int(cch, _countof(szPath));
80 }
81 lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
82 lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
83 ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
84
85 /* NON_EXISTENT_FILENAME_A */
86 lstrcpynA(szPath, NON_EXISTENT_FILENAME_A, _countof(szPath));
87 cch = _countof(szPath);
88 hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
89 if (IsWindowsVistaOrGreater())
90 {
91 ExpandEnvironmentStringsA("%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
92 ok_long(hr, S_OK);
93 ok_int(cch, lstrlenA(szAnswer) + 1);
94 }
95 else
96 {
97 lstrcpynA(szAnswer, NON_EXISTENT_FILENAME_A, _countof(szAnswer));
98 ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
99 ok_int(cch, _countof(szPath));
100 }
101 lstrcpynA(szDebug1, wine_dbgstr_a(szPath), _countof(szDebug1));
102 lstrcpynA(szDebug2, wine_dbgstr_a(szAnswer), _countof(szDebug2));
103 ok(lstrcmpiA(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
104}
105
106static void TEST_AssocQueryStringW(void)
107{
108 WCHAR szPath[MAX_PATH], szAnswer[MAX_PATH];
109 CHAR szDebug1[MAX_PATH], szDebug2[MAX_PATH];
110 HRESULT hr;
111 DWORD cch;
112
113 /* ".txt" */
114 lstrcpynW(szPath, L".txt", _countof(szPath));
115 cch = _countof(szPath);
116 hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
117 ok_long(hr, S_OK);
118 ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
119 lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
120 lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
121 ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
122 ok_int(cch, lstrlenW(szAnswer) + 1);
123
124 /* s_szTextFileW */
125 lstrcpynW(szPath, s_szTextFileW, _countof(szPath));
126 cch = _countof(szPath);
127 hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
128 ok_long(hr, S_OK);
129 ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
130 lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
131 lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
132 ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
133 ok_int(cch, lstrlenW(szAnswer) + 1);
134
135 /* NON_EXISTENT_FILENAME_W */
136 lstrcpynW(szPath, NON_EXISTENT_FILENAME_W, _countof(szPath));
137 cch = _countof(szPath);
138 hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPath, NULL, szPath, &cch);
139 ok_long(hr, S_OK);
140 ExpandEnvironmentStringsW(L"%WINDIR%\\system32\\notepad.exe", szAnswer, _countof(szAnswer));
141 lstrcpynA(szDebug1, wine_dbgstr_w(szPath), _countof(szDebug1));
142 lstrcpynA(szDebug2, wine_dbgstr_w(szAnswer), _countof(szDebug2));
143 ok(lstrcmpiW(szPath, szAnswer) == 0, "%s vs %s\n", szDebug1, szDebug2);
144 ok_int(cch, lstrlenW(szAnswer) + 1);
145}
146
147static void TEST_TrickyPointers(void)
148{
149 HRESULT hr;
150 CHAR szPathA[MAX_PATH], szOutputA[MAX_PATH];
151 WCHAR szPathW[MAX_PATH], szOutputW[MAX_PATH];
152 DWORD cch;
153 BOOL bVistaPlus = IsWindowsVistaOrGreater();
154
155 lstrcpynA(szPathA, s_szTextFileA, _countof(szPathA));
156 cch = _countof(szOutputA);
157 _SEH2_TRY
158 {
159 hr = AssocQueryStringA(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPathA, NULL, szOutputA, (PDWORD)UlongToPtr(cch));
160 }
161 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
162 {
163 hr = 0xDEADFACE;
164 }
165 _SEH2_END;
166
167 if (bVistaPlus)
168 ok_long(hr, 0xDEADFACE);
169 else
170 ok_long(hr, HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND));
171
172 lstrcpynW(szPathW, s_szTextFileW, _countof(szPathW));
173 cch = _countof(szOutputW);
174 _SEH2_TRY
175 {
176 hr = AssocQueryStringW(ASSOCF_NONE, ASSOCSTR_EXECUTABLE, szPathW, NULL, szOutputW, (PDWORD)UlongToPtr(cch));
177 }
178 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
179 {
180 hr = 0xDEADFACE;
181 }
182 _SEH2_END;
183
184 if (bVistaPlus)
185 ok_long(hr, 0xDEADFACE);
186 else
187 ok_long(hr, S_OK);
188}
189
190START_TEST(AssocQueryString)
191{
192 HRESULT hrCoInit = CoInitialize(NULL);
193
194 TEST_Start();
195 TEST_AssocQueryStringA();
196 TEST_AssocQueryStringW();
197 TEST_TrickyPointers();
198 TEST_End();
199
200 if (SUCCEEDED(hrCoInit))
201 CoUninitialize();
202}
TEST_TrickyPointers
static void TEST_TrickyPointers(void)
Definition: AssocQueryString.c:147
s_szTextFileW
static WCHAR s_szTextFileW[MAX_PATH]
Definition: AssocQueryString.c:14
TEST_Start
static void TEST_Start(void)
Definition: AssocQueryString.c:18
TEST_AssocQueryStringA
static void TEST_AssocQueryStringA(void)
Definition: AssocQueryString.c:37
TEST_AssocQueryStringW
static void TEST_AssocQueryStringW(void)
Definition: AssocQueryString.c:106
NON_EXISTENT_FILENAME_A
#define NON_EXISTENT_FILENAME_A
Definition: AssocQueryString.c:15
TEST_End
static void TEST_End(void)
Definition: AssocQueryString.c:27
s_szTextFileA
static CHAR s_szTextFileA[MAX_PATH]
Definition: AssocQueryString.c:13
NON_EXISTENT_FILENAME_W
#define NON_EXISTENT_FILENAME_W
Definition: AssocQueryString.c:16
ok_long
#define ok_long(expression, result)
Definition: atltest.h:133
trace
#define trace
Definition: atltest.h:70
ok
#define ok(value,...)
Definition: atltest.h:57
START_TEST
#define START_TEST(x)
Definition: atltest.h:75
ok_int
#define ok_int(expression, result)
Definition: atltest.h:134
hr
HRESULT hr
Definition: delayimp.cpp:582
NULL
#define NULL
Definition: types.h:112
CoUninitialize
void WINAPI DECLSPEC_HOTPATCH CoUninitialize(void)
Definition: combase.c:2842
lstrcpynA
#define lstrcpynA
Definition: compat.h:751
MAX_PATH
#define MAX_PATH
Definition: compat.h:34
lstrcpynW
#define lstrcpynW
Definition: compat.h:738
lstrlenW
#define lstrlenW
Definition: compat.h:750
ExpandEnvironmentStringsA
DWORD WINAPI ExpandEnvironmentStringsA(IN LPCSTR lpSrc, IN LPSTR lpDst, IN DWORD nSize)
Definition: environ.c:372
ExpandEnvironmentStringsW
DWORD WINAPI ExpandEnvironmentStringsW(IN LPCWSTR lpSrc, IN LPWSTR lpDst, IN DWORD nSize)
Definition: environ.c:492
DeleteFileW
BOOL WINAPI DeleteFileW(IN LPCWSTR lpFileName)
Definition: delete.c:39
lstrcmpiW
int WINAPI lstrcmpiW(LPCWSTR str1, LPCWSTR str2)
Definition: locale.c:4171
lstrcmpiA
int WINAPI lstrcmpiA(LPCSTR str1, LPCSTR str2)
Definition: locale.c:4133
fclose
int CDECL fclose(FILE *file)
Definition: file.c:3757
_wfopen
FILE *CDECL _wfopen(const wchar_t *path, const wchar_t *mode)
Definition: file.c:4335
CoInitialize
HRESULT WINAPI CoInitialize(LPVOID lpReserved)
Definition: compobj.c:531
AssocQueryStringA
HRESULT WINAPI AssocQueryStringA(ASSOCF cfFlags, ASSOCSTR str, LPCSTR pszAssoc, LPCSTR pszExtra, LPSTR pszOut, DWORD *pcchOut)
Definition: assoc.c:612
AssocQueryStringW
HRESULT WINAPI AssocQueryStringW(ASSOCF cfFlags, ASSOCSTR str, LPCWSTR pszAssoc, LPCWSTR pszExtra, LPWSTR pszOut, DWORD *pcchOut)
Definition: assoc.c:569
L
#define L(x)
Definition: resources.c:13
UlongToPtr
#define UlongToPtr(u)
Definition: config.h:106
BOOL
unsigned int BOOL
Definition: ntddk_ex.h:94
DWORD
unsigned long DWORD
Definition: ntddk_ex.h:95
EXCEPTION_EXECUTE_HANDLER
#define EXCEPTION_EXECUTE_HANDLER
Definition: excpt.h:90
S_OK
#define S_OK
Definition: intsafe.h:52
SUCCEEDED
#define SUCCEEDED(hr)
Definition: intsafe.h:50
wine_dbgstr_w
#define wine_dbgstr_w
Definition: kernel32.h:34
lstrlenA
int WINAPI lstrlenA(LPCSTR lpString)
Definition: lstring.c:145
ERROR_FILE_NOT_FOUND
#define ERROR_FILE_NOT_FOUND
Definition: disk.h:79
szPath
LPCWSTR szPath
Definition: env.c:37
WCHAR
short WCHAR
Definition: pedump.c:58
PDWORD
DWORD * PDWORD
Definition: pedump.c:68
CHAR
char CHAR
Definition: pedump.c:57
ASSOCSTR_EXECUTABLE
@ ASSOCSTR_EXECUTABLE
Definition: shlwapi.h:887
ASSOCF_NONE
@ ASSOCF_NONE
Definition: shlwapi.h:858
AssocQueryString
#define AssocQueryString
Definition: shlwapi.h:1024
_SEH2_EXCEPT
#define _SEH2_EXCEPT(...)
Definition: pseh2_64.h:104
_SEH2_END
#define _SEH2_END
Definition: pseh2_64.h:194
_SEH2_TRY
#define _SEH2_TRY
Definition: pseh2_64.h:93
wine_dbgstr_a
const char int int int static __inline const char * wine_dbgstr_a(const char *s)
Definition: debug.h:152
cch
_In_ UINT _In_ UINT cch
Definition: shellapi.h:432
_countof
#define _countof(array)
Definition: sndvol32.h:70
IsWindowsVistaOrGreater
VERSIONHELPERAPI IsWindowsVistaOrGreater()
Definition: versionhelpers.h:70
HRESULT_FROM_WIN32
static HRESULT HRESULT_FROM_WIN32(unsigned int x)
Definition: winerror.h:210