huwcampbell.com
Reactos
1/*
2 * PROJECT: ReactOS API tests
3 * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
4 * PURPOSE: Test for ShellExecCmdLine
5 * PROGRAMMERS: Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
6 */
7#include "shelltest.h"
8#include <shlwapi.h>
9#include <strsafe.h>
10#include <versionhelpers.h>
11#include "shell32_apitest_sub.h"
12#include "closewnd.h"
13
14#define NDEBUG
15#include <debug.h>
16#include <stdio.h>
17
18#ifndef SECL_NO_UI
19 #define SECL_NO_UI 0x2
20 #define SECL_LOG_USAGE 0x8
21 #define SECL_USE_IDLIST 0x10
22 #define SECL_ALLOW_NONEXE 0x20
23 #define SECL_RUNAS 0x40
24#endif
25
26#define ShellExecCmdLine proxy_ShellExecCmdLine
27
28#define shell32_hInstance GetModuleHandle(NULL)
29#define IDS_FILE_NOT_FOUND (-1)
30
31static const WCHAR wszExe[] = L".exe";
32static const WCHAR wszCom[] = L".com";
33
34static __inline void __SHCloneStrW(WCHAR **target, const WCHAR *source)
35{
36 *target = (WCHAR *)SHAlloc((lstrlenW(source) + 1) * sizeof(WCHAR) );
37 lstrcpyW(*target, source);
38}
39
40// NOTE: You have to sync the following code to dll/win32/shell32/shlexec.cpp.
41static LPCWSTR
42SplitParams(LPCWSTR psz, LPWSTR pszArg0, size_t cchArg0)
43{
44 LPCWSTR pch;
45 size_t ich = 0;
46 if (*psz == L'"')
47 {
48 // 1st argument is quoted. the string in quotes is quoted 1st argument.
49 // [pch] --> [pszArg0+ich]
50 for (pch = psz + 1; *pch && ich + 1 < cchArg0; ++ich, ++pch)
51 {
52 if (*pch == L'"' && pch[1] == L'"')
53 {
54 // doubled double quotations found!
55 pszArg0[ich] = L'"';
56 }
57 else if (*pch == L'"')
58 {
59 // single double quotation found!
60 ++pch;
61 break;
62 }
63 else
64 {
65 // otherwise
66 pszArg0[ich] = *pch;
67 }
68 }
69 }
70 else
71 {
72 // 1st argument is unquoted. non-space sequence is 1st argument.
73 // [pch] --> [pszArg0+ich]
74 for (pch = psz; *pch && !iswspace(*pch) && ich + 1 < cchArg0; ++ich, ++pch)
75 {
76 pszArg0[ich] = *pch;
77 }
78 }
79 pszArg0[ich] = 0;
80
81 // skip space
82 while (iswspace(*pch))
83 ++pch;
84
85 return pch;
86}
87
88HRESULT WINAPI ShellExecCmdLine(
89 HWND hwnd,
90 LPCWSTR pwszCommand,
91 LPCWSTR pwszStartDir,
92 int nShow,
93 LPVOID pUnused,
94 DWORD dwSeclFlags)
95{
96 SHELLEXECUTEINFOW info;
97 DWORD dwSize, dwError, dwType, dwFlags = SEE_MASK_DOENVSUBST | SEE_MASK_NOASYNC;
98 LPCWSTR pszVerb = NULL;
99 WCHAR szFile[MAX_PATH], szFile2[MAX_PATH];
100 HRESULT hr;
101 LPCWSTR pchParams;
102 LPWSTR lpCommand = NULL;
103
104 if (pwszCommand == NULL)
105 RaiseException(EXCEPTION_ACCESS_VIOLATION, EXCEPTION_NONCONTINUABLE,
106 1, (ULONG_PTR*)pwszCommand);
107
108 __SHCloneStrW(&lpCommand, pwszCommand);
109 StrTrimW(lpCommand, L" \t");
110
111 if (dwSeclFlags & SECL_NO_UI)
112 dwFlags |= SEE_MASK_FLAG_NO_UI;
113 if (dwSeclFlags & SECL_LOG_USAGE)
114 dwFlags |= SEE_MASK_FLAG_LOG_USAGE;
115 if (dwSeclFlags & SECL_USE_IDLIST)
116 dwFlags |= SEE_MASK_INVOKEIDLIST;
117
118 if (dwSeclFlags & SECL_RUNAS)
119 {
120 dwSize = 0;
121 hr = AssocQueryStringW(0, ASSOCSTR_COMMAND, lpCommand, L"RunAs", NULL, &dwSize);
122 if (SUCCEEDED(hr) && dwSize != 0)
123 {
124 pszVerb = L"runas";
125 }
126 }
127
128 if (UrlIsFileUrlW(lpCommand))
129 {
130 StringCchCopyW(szFile, _countof(szFile), lpCommand);
131 pchParams = NULL;
132 }
133 else
134 {
135 pchParams = SplitParams(lpCommand, szFile, _countof(szFile));
136 if (szFile[0] != UNICODE_NULL && szFile[1] == L':' &&
137 szFile[2] == UNICODE_NULL)
138 {
139 PathAddBackslashW(szFile);
140 }
141
142 WCHAR szCurDir[MAX_PATH];
143 GetCurrentDirectoryW(_countof(szCurDir), szCurDir);
144 if (pwszStartDir)
145 {
146 SetCurrentDirectoryW(pwszStartDir);
147 }
148
149 if (PathIsRelativeW(szFile) &&
150 GetFullPathNameW(szFile, _countof(szFile2), szFile2, NULL) &&
151 PathFileExistsW(szFile2))
152 {
153 StringCchCopyW(szFile, _countof(szFile), szFile2);
154 }
155 else if (SearchPathW(NULL, szFile, NULL, _countof(szFile2), szFile2, NULL) ||
156 SearchPathW(NULL, szFile, wszExe, _countof(szFile2), szFile2, NULL) ||
157 SearchPathW(NULL, szFile, wszCom, _countof(szFile2), szFile2, NULL) ||
158 SearchPathW(pwszStartDir, szFile, NULL, _countof(szFile2), szFile2, NULL) ||
159 SearchPathW(pwszStartDir, szFile, wszExe, _countof(szFile2), szFile2, NULL) ||
160 SearchPathW(pwszStartDir, szFile, wszCom, _countof(szFile2), szFile2, NULL))
161 {
162 StringCchCopyW(szFile, _countof(szFile), szFile2);
163 }
164 else if (SearchPathW(NULL, lpCommand, NULL, _countof(szFile2), szFile2, NULL) ||
165 SearchPathW(NULL, lpCommand, wszExe, _countof(szFile2), szFile2, NULL) ||
166 SearchPathW(NULL, lpCommand, wszCom, _countof(szFile2), szFile2, NULL) ||
167 SearchPathW(pwszStartDir, lpCommand, NULL, _countof(szFile2), szFile2, NULL) ||
168 SearchPathW(pwszStartDir, lpCommand, wszExe, _countof(szFile2), szFile2, NULL) ||
169 SearchPathW(pwszStartDir, lpCommand, wszCom, _countof(szFile2), szFile2, NULL))
170 {
171 StringCchCopyW(szFile, _countof(szFile), szFile2);
172 pchParams = NULL;
173 }
174
175 if (pwszStartDir)
176 {
177 SetCurrentDirectoryW(szCurDir);
178 }
179
180 if (!(dwSeclFlags & SECL_ALLOW_NONEXE))
181 {
182 if (!GetBinaryTypeW(szFile, &dwType))
183 {
184 SHFree(lpCommand);
185
186 if (!(dwSeclFlags & SECL_NO_UI))
187 {
188 WCHAR szText[128 + MAX_PATH], szFormat[128];
189 LoadStringW(shell32_hInstance, IDS_FILE_NOT_FOUND, szFormat, _countof(szFormat));
190 StringCchPrintfW(szText, _countof(szText), szFormat, szFile);
191 MessageBoxW(hwnd, szText, NULL, MB_ICONERROR);
192 }
193 return CO_E_APPNOTFOUND;
194 }
195 }
196 else
197 {
198 if (GetFileAttributesW(szFile) == INVALID_FILE_ATTRIBUTES)
199 {
200 SHFree(lpCommand);
201
202 if (!(dwSeclFlags & SECL_NO_UI))
203 {
204 WCHAR szText[128 + MAX_PATH], szFormat[128];
205 LoadStringW(shell32_hInstance, IDS_FILE_NOT_FOUND, szFormat, _countof(szFormat));
206 StringCchPrintfW(szText, _countof(szText), szFormat, szFile);
207 MessageBoxW(hwnd, szText, NULL, MB_ICONERROR);
208 }
209 return HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND);
210 }
211 }
212 }
213
214 ZeroMemory(&info, sizeof(info));
215 info.cbSize = sizeof(info);
216 info.fMask = dwFlags;
217 info.hwnd = hwnd;
218 info.lpVerb = pszVerb;
219 info.lpFile = szFile;
220 info.lpParameters = (pchParams && *pchParams) ? pchParams : NULL;
221 info.lpDirectory = pwszStartDir;
222 info.nShow = nShow;
223 if (ShellExecuteExW(&info))
224 {
225 if (info.lpIDList)
226 CoTaskMemFree(info.lpIDList);
227
228 SHFree(lpCommand);
229
230 return S_OK;
231 }
232
233 dwError = GetLastError();
234
235 SHFree(lpCommand);
236
237 return HRESULT_FROM_WIN32(dwError);
238}
239
240#undef ShellExecCmdLine
241
242typedef HRESULT (WINAPI *SHELLEXECCMDLINE)(HWND, LPCWSTR, LPCWSTR, INT, LPVOID, DWORD);
243SHELLEXECCMDLINE g_pShellExecCmdLine = NULL;
244
245typedef struct TEST_ENTRY
246{
247 INT lineno;
248 BOOL result;
249 BOOL bAllowNonExe;
250 LPCWSTR pwszCommand;
251 LPCWSTR pwszStartDir;
252} TEST_ENTRY;
253
254static WCHAR s_sub_program[MAX_PATH];
255static WCHAR s_win_test_exe[MAX_PATH];
256static WCHAR s_sys_bat_file[MAX_PATH];
257static WCHAR s_cur_dir[MAX_PATH];
258
259static const TEST_ENTRY s_entries_1[] =
260{
261 // NULL
262 { __LINE__, 0xBADFACE, FALSE, NULL, NULL },
263 { __LINE__, 0xBADFACE, FALSE, NULL, L"." },
264 { __LINE__, 0xBADFACE, FALSE, NULL, L"system32" },
265 { __LINE__, 0xBADFACE, FALSE, NULL, L"C:\\Program Files" },
266 { __LINE__, 0xBADFACE, TRUE, NULL, NULL },
267 { __LINE__, 0xBADFACE, TRUE, NULL, L"." },
268 { __LINE__, 0xBADFACE, TRUE, NULL, L"system32" },
269 { __LINE__, 0xBADFACE, TRUE, NULL, L"C:\\Program Files" },
270 // notepad
271 { __LINE__, TRUE, FALSE, L"notepad", NULL },
272 { __LINE__, TRUE, FALSE, L"notepad", L"." },
273 { __LINE__, TRUE, FALSE, L"notepad", L"system32" },
274 { __LINE__, TRUE, FALSE, L"notepad", L"C:\\Program Files" },
275 { __LINE__, TRUE, FALSE, L"notepad \"Test File.txt\"", NULL },
276 { __LINE__, TRUE, FALSE, L"notepad \"Test File.txt\"", L"." },
277 { __LINE__, TRUE, TRUE, L"notepad", NULL },
278 { __LINE__, TRUE, TRUE, L"notepad", L"." },
279 { __LINE__, TRUE, TRUE, L"notepad", L"system32" },
280 { __LINE__, TRUE, TRUE, L"notepad", L"C:\\Program Files" },
281 { __LINE__, TRUE, TRUE, L"notepad \"Test File.txt\"", NULL },
282 { __LINE__, TRUE, TRUE, L"notepad \"Test File.txt\"", L"." },
283 // notepad.exe
284 { __LINE__, TRUE, FALSE, L"notepad.exe", NULL },
285 { __LINE__, TRUE, FALSE, L"notepad.exe", L"." },
286 { __LINE__, TRUE, FALSE, L"notepad.exe", L"system32" },
287 { __LINE__, TRUE, FALSE, L"notepad.exe", L"C:\\Program Files" },
288 { __LINE__, TRUE, FALSE, L"notepad.exe \"Test File.txt\"", NULL },
289 { __LINE__, TRUE, FALSE, L"notepad.exe \"Test File.txt\"", L"." },
290 { __LINE__, TRUE, TRUE, L"notepad.exe", NULL },
291 { __LINE__, TRUE, TRUE, L"notepad.exe", L"." },
292 { __LINE__, TRUE, TRUE, L"notepad.exe", L"system32" },
293 { __LINE__, TRUE, TRUE, L"notepad.exe", L"C:\\Program Files" },
294 { __LINE__, TRUE, TRUE, L"notepad.exe \"Test File.txt\"", NULL },
295 { __LINE__, TRUE, TRUE, L"notepad.exe \"Test File.txt\"", L"." },
296 // C:\notepad.exe
297 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe", NULL },
298 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe", L"." },
299 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe", L"system32" },
300 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe", L"C:\\Program Files" },
301 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe \"Test File.txt\"", NULL },
302 { __LINE__, FALSE, FALSE, L"C:\\notepad.exe \"Test File.txt\"", L"." },
303 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe", NULL },
304 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe", L"." },
305 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe", L"system32" },
306 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe", L"C:\\Program Files" },
307 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe \"Test File.txt\"", NULL },
308 { __LINE__, FALSE, TRUE, L"C:\\notepad.exe \"Test File.txt\"", L"." },
309 // "notepad"
310 { __LINE__, TRUE, FALSE, L"\"notepad\"", NULL },
311 { __LINE__, TRUE, FALSE, L"\"notepad\"", L"." },
312 { __LINE__, TRUE, FALSE, L"\"notepad\"", L"system32" },
313 { __LINE__, TRUE, FALSE, L"\"notepad\"", L"C:\\Program Files" },
314 { __LINE__, TRUE, FALSE, L"\"notepad\" \"Test File.txt\"", NULL },
315 { __LINE__, TRUE, FALSE, L"\"notepad\" \"Test File.txt\"", L"." },
316 { __LINE__, TRUE, TRUE, L"\"notepad\"", NULL },
317 { __LINE__, TRUE, TRUE, L"\"notepad\"", L"." },
318 { __LINE__, TRUE, TRUE, L"\"notepad\"", L"system32" },
319 { __LINE__, TRUE, TRUE, L"\"notepad\"", L"C:\\Program Files" },
320 { __LINE__, TRUE, TRUE, L"\"notepad\" \"Test File.txt\"", NULL },
321 { __LINE__, TRUE, TRUE, L"\"notepad\" \"Test File.txt\"", L"." },
322 // "notepad.exe"
323 { __LINE__, TRUE, FALSE, L"\"notepad.exe\"", NULL },
324 { __LINE__, TRUE, FALSE, L"\"notepad.exe\"", L"." },
325 { __LINE__, TRUE, FALSE, L"\"notepad.exe\"", L"system32" },
326 { __LINE__, TRUE, FALSE, L"\"notepad.exe\"", L"C:\\Program Files" },
327 { __LINE__, TRUE, FALSE, L"\"notepad.exe\" \"Test File.txt\"", NULL },
328 { __LINE__, TRUE, FALSE, L"\"notepad.exe\" \"Test File.txt\"", L"." },
329 { __LINE__, TRUE, TRUE, L"\"notepad.exe\"", NULL },
330 { __LINE__, TRUE, TRUE, L"\"notepad.exe\"", L"." },
331 { __LINE__, TRUE, TRUE, L"\"notepad.exe\"", L"system32" },
332 { __LINE__, TRUE, TRUE, L"\"notepad.exe\"", L"C:\\Program Files" },
333 { __LINE__, TRUE, TRUE, L"\"notepad.exe\" \"Test File.txt\"", NULL },
334 { __LINE__, TRUE, TRUE, L"\"notepad.exe\" \"Test File.txt\"", L"." },
335 // test program
336 { __LINE__, FALSE, FALSE, L"test program", NULL },
337 { __LINE__, FALSE, FALSE, L"test program", L"." },
338 { __LINE__, FALSE, FALSE, L"test program", L"system32" },
339 { __LINE__, FALSE, FALSE, L"test program", L"C:\\Program Files" },
340 { __LINE__, FALSE, FALSE, L"test program \"Test File.txt\"", NULL },
341 { __LINE__, FALSE, FALSE, L"test program \"Test File.txt\"", L"." },
342 { __LINE__, FALSE, TRUE, L"test program", NULL },
343 { __LINE__, FALSE, TRUE, L"test program", L"." },
344 { __LINE__, FALSE, TRUE, L"test program", L"system32" },
345 { __LINE__, FALSE, TRUE, L"test program", L"C:\\Program Files" },
346 { __LINE__, FALSE, TRUE, L"test program \"Test File.txt\"", NULL },
347 { __LINE__, FALSE, TRUE, L"test program \"Test File.txt\"", L"." },
348 // test program.exe
349 { __LINE__, FALSE, FALSE, L"test program.exe", NULL },
350 { __LINE__, FALSE, FALSE, L"test program.exe", L"." },
351 { __LINE__, FALSE, FALSE, L"test program.exe", L"system32" },
352 { __LINE__, FALSE, FALSE, L"test program.exe", L"C:\\Program Files" },
353 { __LINE__, FALSE, FALSE, L"test program.exe \"Test File.txt\"", NULL },
354 { __LINE__, FALSE, FALSE, L"test program.exe \"Test File.txt\"", L"." },
355 { __LINE__, FALSE, TRUE, L"test program.exe", NULL },
356 { __LINE__, FALSE, TRUE, L"test program.exe", L"." },
357 { __LINE__, FALSE, TRUE, L"test program.exe", L"system32" },
358 { __LINE__, FALSE, TRUE, L"test program.exe", L"C:\\Program Files" },
359 { __LINE__, FALSE, TRUE, L"test program.exe \"Test File.txt\"", NULL },
360 { __LINE__, FALSE, TRUE, L"test program.exe \"Test File.txt\"", L"." },
361 // test program.bat
362 { __LINE__, FALSE, FALSE, L"test program.bat", NULL },
363 { __LINE__, FALSE, FALSE, L"test program.bat", L"." },
364 { __LINE__, FALSE, FALSE, L"test program.bat", L"system32" },
365 { __LINE__, FALSE, FALSE, L"test program.bat", L"C:\\Program Files" },
366 { __LINE__, FALSE, FALSE, L"test program.bat \"Test File.txt\"", NULL },
367 { __LINE__, FALSE, FALSE, L"test program.bat \"Test File.txt\"", L"." },
368 { __LINE__, FALSE, TRUE, L"test program.bat", NULL },
369 { __LINE__, FALSE, TRUE, L"test program.bat", L"." },
370 { __LINE__, FALSE, TRUE, L"test program.bat", L"system32" },
371 { __LINE__, FALSE, TRUE, L"test program.bat", L"C:\\Program Files" },
372 { __LINE__, FALSE, TRUE, L"test program.bat \"Test File.txt\"", NULL },
373 { __LINE__, FALSE, TRUE, L"test program.bat \"Test File.txt\"", L"." },
374 // "test program"
375 { __LINE__, FALSE, FALSE, L"\"test program\"", NULL },
376 { __LINE__, FALSE, FALSE, L"\"test program\"", L"." },
377 { __LINE__, FALSE, FALSE, L"\"test program\"", L"system32" },
378 { __LINE__, FALSE, FALSE, L"\"test program\"", L"C:\\Program Files" },
379 { __LINE__, FALSE, FALSE, L"\"test program\" \"Test File.txt\"", NULL },
380 { __LINE__, FALSE, FALSE, L"\"test program\" \"Test File.txt\"", L"." },
381 { __LINE__, FALSE, TRUE, L"\"test program\"", NULL },
382 { __LINE__, FALSE, TRUE, L"\"test program\"", L"." },
383 { __LINE__, FALSE, TRUE, L"\"test program\"", L"system32" },
384 { __LINE__, FALSE, TRUE, L"\"test program\"", L"C:\\Program Files" },
385 { __LINE__, FALSE, TRUE, L"\"test program\" \"Test File.txt\"", NULL },
386 { __LINE__, FALSE, TRUE, L"\"test program\" \"Test File.txt\"", L"." },
387 // "test program.exe"
388 { __LINE__, TRUE, FALSE, L"\"test program.exe\"", NULL },
389 { __LINE__, TRUE, FALSE, L"\"test program.exe\"", L"." },
390 { __LINE__, TRUE, FALSE, L"\"test program.exe\"", L"system32" },
391 { __LINE__, TRUE, FALSE, L"\"test program.exe\"", L"C:\\Program Files" },
392 { __LINE__, TRUE, FALSE, L"\"test program.exe\" \"Test File.txt\"", NULL },
393 { __LINE__, TRUE, FALSE, L"\"test program.exe\" \"Test File.txt\"", L"." },
394 { __LINE__, TRUE, TRUE, L"\"test program.exe\"", NULL },
395 { __LINE__, TRUE, TRUE, L"\"test program.exe\"", L"." },
396 { __LINE__, TRUE, TRUE, L"\"test program.exe\"", L"system32" },
397 { __LINE__, TRUE, TRUE, L"\"test program.exe\"", L"C:\\Program Files" },
398 { __LINE__, TRUE, TRUE, L"\"test program.exe\" \"Test File.txt\"", NULL },
399 { __LINE__, TRUE, TRUE, L"\"test program.exe\" \"Test File.txt\"", L"." },
400 // "test program.bat"
401 { __LINE__, FALSE, FALSE, L"\"test program.bat\"", NULL },
402 { __LINE__, FALSE, FALSE, L"\"test program.bat\"", L"." },
403 { __LINE__, FALSE, FALSE, L"\"test program.bat\"", L"system32" },
404 { __LINE__, FALSE, FALSE, L"\"test program.bat\"", L"C:\\Program Files" },
405 { __LINE__, FALSE, FALSE, L"\"test program.bat\" \"Test File.txt\"", NULL },
406 { __LINE__, FALSE, FALSE, L"\"test program.bat\" \"Test File.txt\"", L"." },
407 { __LINE__, FALSE, TRUE, L"\"test program.bat\"", NULL },
408 { __LINE__, FALSE, TRUE, L"\"test program.bat\"", L"." },
409 { __LINE__, FALSE, TRUE, L"\"test program.bat\"", L"system32" },
410 { __LINE__, FALSE, TRUE, L"\"test program.bat\"", L"C:\\Program Files" },
411 { __LINE__, FALSE, TRUE, L"\"test program.bat\" \"Test File.txt\"", NULL },
412 { __LINE__, FALSE, TRUE, L"\"test program.bat\" \"Test File.txt\"", L"." },
413 // invalid program
414 { __LINE__, FALSE, FALSE, L"invalid program", NULL },
415 { __LINE__, FALSE, FALSE, L"invalid program", L"." },
416 { __LINE__, FALSE, FALSE, L"invalid program", L"system32" },
417 { __LINE__, FALSE, FALSE, L"invalid program", L"C:\\Program Files" },
418 { __LINE__, FALSE, FALSE, L"invalid program \"Test File.txt\"", NULL },
419 { __LINE__, FALSE, FALSE, L"invalid program \"Test File.txt\"", L"." },
420 { __LINE__, FALSE, TRUE, L"invalid program", NULL },
421 { __LINE__, FALSE, TRUE, L"invalid program", L"." },
422 { __LINE__, FALSE, TRUE, L"invalid program", L"system32" },
423 { __LINE__, FALSE, TRUE, L"invalid program", L"C:\\Program Files" },
424 { __LINE__, FALSE, TRUE, L"invalid program \"Test File.txt\"", NULL },
425 { __LINE__, FALSE, TRUE, L"invalid program \"Test File.txt\"", L"." },
426 // \"invalid program.exe\"
427 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\"", NULL },
428 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\"", L"." },
429 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\"", L"system32" },
430 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\"", L"C:\\Program Files" },
431 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\" \"Test File.txt\"", NULL },
432 { __LINE__, FALSE, FALSE, L"\"invalid program.exe\" \"Test File.txt\"", L"." },
433 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\"", NULL },
434 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\"", L"." },
435 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\"", L"system32" },
436 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\"", L"C:\\Program Files" },
437 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\" \"Test File.txt\"", NULL },
438 { __LINE__, FALSE, TRUE, L"\"invalid program.exe\" \"Test File.txt\"", L"." },
439 // My Documents
440 { __LINE__, TRUE, TRUE, L"::{450d8fba-ad25-11d0-98a8-0800361b1103}", NULL },
441 { __LINE__, TRUE, TRUE, L"shell:::{450d8fba-ad25-11d0-98a8-0800361b1103}", NULL },
442 // shell:sendto
443 { __LINE__, TRUE, TRUE, L"shell:sendto", NULL },
444 // https://google.com
445 { __LINE__, TRUE, FALSE, L"https://google.com", NULL },
446 { __LINE__, TRUE, TRUE, L"https://google.com", NULL },
447 // Test File 1.txt
448 { __LINE__, FALSE, FALSE, L"Test File 1.txt", NULL },
449 { __LINE__, FALSE, FALSE, L"Test File 1.txt", L"." },
450 { __LINE__, FALSE, FALSE, L"Test File 1.txt", L"system32" },
451 { __LINE__, FALSE, FALSE, L"Test File 1.txt", s_cur_dir },
452 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", NULL },
453 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", L"." },
454 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", L"system32" },
455 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", s_cur_dir },
456 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", NULL },
457 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", L"." },
458 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", L"system32" },
459 { __LINE__, FALSE, TRUE, L"Test File 1.txt", NULL },
460 // { __LINE__, TRUE, TRUE, L"Test File 1.txt", L"." }, // Fails on Vista, 7, 8.1
461 { __LINE__, FALSE, TRUE, L"Test File 1.txt", L"system32" },
462 { __LINE__, TRUE, TRUE, L"Test File 1.txt", s_cur_dir },
463 { __LINE__, FALSE, TRUE, L"\"Test File 1.txt\"", NULL },
464 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", L"." },
465 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", L"system32" },
466 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", s_cur_dir },
467 { __LINE__, FALSE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", NULL },
468 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", L"." },
469 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", L"system32" },
470 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", s_cur_dir },
471 // Test File 2.bat
472 { __LINE__, FALSE, FALSE, L"Test File 2.bat", NULL },
473 { __LINE__, FALSE, FALSE, L"Test File 2.bat", L"." },
474 { __LINE__, FALSE, FALSE, L"Test File 2.bat", L"system32" },
475 { __LINE__, FALSE, FALSE, L"Test File 2.bat", s_cur_dir },
476 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", NULL },
477 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", L"." },
478 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", L"system32" },
479 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", s_cur_dir },
480 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", NULL },
481 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", L"." },
482 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", L"system32" },
483 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", s_cur_dir },
484 { __LINE__, FALSE, TRUE, L"Test File 2.bat", NULL },
485 { __LINE__, FALSE, TRUE, L"Test File 2.bat", L"." },
486 { __LINE__, FALSE, TRUE, L"Test File 2.bat", L"system32" },
487 { __LINE__, FALSE, TRUE, L"Test File 2.bat", s_cur_dir },
488 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", NULL },
489 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", L"." },
490 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", L"system32" },
491 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", s_cur_dir },
492 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", NULL },
493 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", L"." },
494 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", L"system32" },
495 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", s_cur_dir },
496};
497
498static const TEST_ENTRY s_entries_2[] =
499{
500 // Test File 1.txt (with setting path)
501 { __LINE__, FALSE, FALSE, L"Test File 1.txt", NULL },
502 { __LINE__, FALSE, FALSE, L"Test File 1.txt", L"." },
503 { __LINE__, FALSE, FALSE, L"Test File 1.txt", L"system32" },
504 { __LINE__, FALSE, FALSE, L"Test File 1.txt", s_cur_dir },
505 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", NULL },
506 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", L"." },
507 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", L"system32" },
508 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\"", s_cur_dir },
509 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", NULL },
510 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", L"." },
511 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", L"system32" },
512 { __LINE__, FALSE, FALSE, L"\"Test File 1.txt\" \"Test File.txt\"", s_cur_dir },
513 { __LINE__, FALSE, TRUE, L"Test File 1.txt", NULL },
514 // { __LINE__, TRUE, TRUE, L"Test File 1.txt", L"." }, // Fails on Vista, 7, 8.1
515 { __LINE__, FALSE, TRUE, L"Test File 1.txt", L"system32" },
516 { __LINE__, TRUE, TRUE, L"Test File 1.txt", s_cur_dir },
517 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", NULL },
518 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", L"." },
519 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", L"system32" },
520 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\"", s_cur_dir },
521 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", NULL },
522 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", L"." },
523 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", L"system32" },
524 { __LINE__, TRUE, TRUE, L"\"Test File 1.txt\" \"Test File.txt\"", s_cur_dir },
525 // Test File 2.bat (with setting path)
526 { __LINE__, FALSE, FALSE, L"Test File 2.bat", NULL },
527 { __LINE__, FALSE, FALSE, L"Test File 2.bat", L"." },
528 { __LINE__, FALSE, FALSE, L"Test File 2.bat", L"system32" },
529 { __LINE__, FALSE, FALSE, L"Test File 2.bat", s_cur_dir },
530 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", NULL },
531 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", L"." },
532 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", L"system32" },
533 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\"", s_cur_dir },
534 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", NULL },
535 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", L"." },
536 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", L"system32" },
537 { __LINE__, FALSE, FALSE, L"\"Test File 2.bat\" \"Test File.txt\"", s_cur_dir },
538 { __LINE__, FALSE, TRUE, L"Test File 2.bat", NULL },
539 { __LINE__, FALSE, TRUE, L"Test File 2.bat", L"." },
540 { __LINE__, FALSE, TRUE, L"Test File 2.bat", L"system32" },
541 { __LINE__, FALSE, TRUE, L"Test File 2.bat", s_cur_dir },
542 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", NULL },
543 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", L"." },
544 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", L"system32" },
545 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\"", s_cur_dir },
546 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", NULL },
547 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", L"." },
548 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", L"system32" },
549 { __LINE__, FALSE, TRUE, L"\"Test File 2.bat\" \"Test File.txt\"", s_cur_dir },
550};
551
552static void DoEntry(const TEST_ENTRY *pEntry)
553{
554 HRESULT hr;
555 DWORD dwSeclFlags;
556 BOOL result;
557
558 if (pEntry->bAllowNonExe)
559 dwSeclFlags = SECL_NO_UI | SECL_ALLOW_NONEXE;
560 else
561 dwSeclFlags = SECL_NO_UI;
562
563 _SEH2_TRY
564 {
565 if (IsReactOS())
566 {
567 hr = proxy_ShellExecCmdLine(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
568 SW_SHOWNORMAL, NULL, dwSeclFlags);
569 }
570 else
571 {
572 hr = (*g_pShellExecCmdLine)(NULL, pEntry->pwszCommand, pEntry->pwszStartDir,
573 SW_SHOWNORMAL, NULL, dwSeclFlags);
574 }
575 }
576 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
577 {
578 hr = 0xBADFACE;
579 }
580 _SEH2_END;
581
582 if (hr == 0xBADFACE)
583 result = hr;
584 else
585 result = (hr == S_OK);
586
587 ok(result == pEntry->result, "Line %d: result expected %d, was %d\n",
588 pEntry->lineno, pEntry->result, result);
589}
590
591static WINDOW_LIST s_List1, s_List2;
592
593static VOID TEST_ShellExecCmdLine(VOID)
594{
595 GetWindowList(&s_List1);
596
597 // do tests
598 for (size_t i = 0; i < _countof(s_entries_1); ++i)
599 {
600 DoEntry(&s_entries_1[i]);
601 }
602 SetEnvironmentVariableW(L"PATH", s_cur_dir);
603 for (size_t i = 0; i < _countof(s_entries_2); ++i)
604 {
605 DoEntry(&s_entries_2[i]);
606 }
607
608 // Execution can be asynchronous; you have to wait for it to finish.
609 Sleep(2000);
610
611 // Close newly-opened window(s)
612 GetWindowList(&s_List2);
613 CloseNewWindows(&s_List1, &s_List2);
614 FreeWindowList(&s_List1);
615 FreeWindowList(&s_List2);
616}
617
618START_TEST(ShellExecCmdLine)
619{
620 using namespace std;
621
622 if (!IsReactOS())
623 {
624 if (!IsWindowsVistaOrGreater())
625 {
626 skip("ShellExecCmdLine is not available on this platform\n");
627 return;
628 }
629
630 HMODULE hShell32 = GetModuleHandleA("shell32");
631 g_pShellExecCmdLine = (SHELLEXECCMDLINE)GetProcAddress(hShell32, (LPCSTR)(INT_PTR)265);
632 if (!g_pShellExecCmdLine)
633 {
634 skip("ShellExecCmdLine is not found\n");
635 return;
636 }
637 }
638
639 if (!FindSubProgram(s_sub_program, _countof(s_sub_program)))
640 {
641 skip("shell32_apitest_sub.exe is not found\n");
642 return;
643 }
644
645 // s_win_test_exe
646 GetWindowsDirectoryW(s_win_test_exe, _countof(s_win_test_exe));
647 PathAppendW(s_win_test_exe, L"test program.exe");
648 SetFileAttributesW(s_win_test_exe, FILE_ATTRIBUTE_NORMAL); // Clear readonly flag if it exists.
649 BOOL ret = CopyFileW(s_sub_program, s_win_test_exe, FALSE);
650 if (!ret)
651 {
652 skip("Cannot copy test files to the system root directory. (Error: %lu)\n", GetLastError());
653 return;
654 }
655
656 FILE *fp;
657
658 // s_sys_bat_file
659 GetSystemDirectoryW(s_sys_bat_file, _countof(s_sys_bat_file));
660 PathAppendW(s_sys_bat_file, L"test program.bat");
661 fp = _wfopen(s_sys_bat_file, L"wb");
662 fclose(fp);
663 ok_int(PathFileExistsW(s_sys_bat_file), TRUE);
664
665 // "Test File 1.txt"
666 fp = fopen("Test File 1.txt", "wb");
667 ok(fp != NULL, "failed to create a test file\n");
668 fclose(fp);
669 ok_int(PathFileExistsA("Test File 1.txt"), TRUE);
670
671 // "Test File 2.bat"
672 fp = fopen("Test File 2.bat", "wb");
673 ok(fp != NULL, "failed to create a test file\n");
674 fclose(fp);
675 ok_int(PathFileExistsA("Test File 2.bat"), TRUE);
676
677 // s_cur_dir
678 GetCurrentDirectoryW(_countof(s_cur_dir), s_cur_dir);
679
680 TEST_ShellExecCmdLine();
681
682 // Some process can lock the file of s_win_test_exe
683 Sleep(1000);
684
685 // clean up
686 SetFileAttributesW(s_win_test_exe, FILE_ATTRIBUTE_NORMAL); // Clear readonly flag
687 ok(DeleteFileW(s_win_test_exe), "Failed to delete \"%S\".\n", s_win_test_exe);
688 ok(DeleteFileW(s_sys_bat_file), "Failed to delete \"%S\".\n", s_sys_bat_file);
689 ok(DeleteFileA("Test File 1.txt"), "Failed to delete \"Test File 1.txt\".\n");
690 ok(DeleteFileA("Test File 2.bat"), "Failed to delete \"Test File 2.bat\".\n");
691}