/* * Unit test of the ShellExecute function. * * Copyright 2005 Francois Gouget for CodeWeavers * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ /* TODO: * - test the default verb selection * - test selection of an alternate class * - try running executables in more ways * - try passing arguments to executables * - ShellExecute("foo.shlexec") with no path should work if foo.shlexec is * in the PATH * - test associations that use %l, %L or "%1" instead of %1 * - we may want to test ShellExecuteEx() instead of ShellExecute() * and then we could also check its return value * - ShellExecuteEx() also calls SetLastError() with meaningful values which * we could check */ #include #include #include "wtypes.h" #include "winbase.h" #include "windef.h" #include "shellapi.h" #include "shlwapi.h" #include "wine/test.h" #include "shell32_test.h" static char argv0[MAX_PATH]; static int myARGC; static char** myARGV; static char tmpdir[MAX_PATH]; static char child_file[MAX_PATH]; static DLLVERSIONINFO dllver; /*** * * ShellExecute wrappers * ***/ static void dump_child(void); static HANDLE hEvent; static void init_event(const char* child_file) { char* event_name; event_name=strrchr(child_file, '\\')+1; hEvent=CreateEvent(NULL, FALSE, FALSE, event_name); } static void strcat_param(char* str, const char* param) { if (param!=NULL) { strcat(str, "\""); strcat(str, param); strcat(str, "\""); } else { strcat(str, "null"); } } static char shell_call[2048]=""; static int shell_execute(LPCSTR operation, LPCSTR file, LPCSTR parameters, LPCSTR directory) { int rc; strcpy(shell_call, "ShellExecute("); strcat_param(shell_call, operation); strcat(shell_call, ", "); strcat_param(shell_call, file); strcat(shell_call, ", "); strcat_param(shell_call, parameters); strcat(shell_call, ", "); strcat_param(shell_call, directory); strcat(shell_call, ")"); if (winetest_debug > 1) trace("%s\n", shell_call); DeleteFile(child_file); SetLastError(0xcafebabe); /* FIXME: We cannot use ShellExecuteEx() here because if there is no * association it displays the 'Open With' dialog and I could not find * a flag to prevent this. */ rc=(int)ShellExecute(NULL, operation, file, parameters, directory, SW_SHOWNORMAL); if (rc>=32) { int wait_rc; wait_rc=WaitForSingleObject(hEvent, 5000); ok(wait_rc==WAIT_OBJECT_0, "WaitForSingleObject returned %d\n", wait_rc); } /* The child process may have changed the result file, so let profile * functions know about it */ WritePrivateProfileStringA(NULL, NULL, NULL, child_file); if (rc>=32) dump_child(); return rc; } static int shell_execute_ex(DWORD mask, LPCSTR operation, LPCSTR file, LPCSTR parameters, LPCSTR directory) { SHELLEXECUTEINFO sei; BOOL success; int rc; strcpy(shell_call, "ShellExecuteEx("); strcat_param(shell_call, operation); strcat(shell_call, ", "); strcat_param(shell_call, file); strcat(shell_call, ", "); strcat_param(shell_call, parameters); strcat(shell_call, ", "); strcat_param(shell_call, directory); strcat(shell_call, ")"); if (winetest_debug > 1) trace("%s\n", shell_call); sei.cbSize=sizeof(sei); sei.fMask=SEE_MASK_NOCLOSEPROCESS | mask; sei.hwnd=NULL; sei.lpVerb=operation; sei.lpFile=file; sei.lpParameters=parameters; sei.lpDirectory=directory; sei.nShow=SW_SHOWNORMAL; sei.hInstApp=NULL; /* Out */ sei.lpIDList=NULL; sei.lpClass=NULL; sei.hkeyClass=NULL; sei.dwHotKey=0; U(sei).hIcon=NULL; sei.hProcess=NULL; /* Out */ DeleteFile(child_file); SetLastError(0xcafebabe); success=ShellExecuteEx(&sei); rc=(int)sei.hInstApp; ok((success && rc >= 32) || (!success && rc < 32), "%s rc=%d and hInstApp=%d is not allowed\n", shell_call, success, rc); if (rc>=32) { int wait_rc; if (sei.hProcess!=NULL) { wait_rc=WaitForSingleObject(sei.hProcess, 5000); ok(wait_rc==WAIT_OBJECT_0, "WaitForSingleObject(hProcess) returned %d\n", wait_rc); } wait_rc=WaitForSingleObject(hEvent, 5000); ok(wait_rc==WAIT_OBJECT_0, "WaitForSingleObject returned %d\n", wait_rc); } /* The child process may have changed the result file, so let profile * functions know about it */ WritePrivateProfileStringA(NULL, NULL, NULL, child_file); if (rc>=32) dump_child(); return rc; } /*** * * Functions to create / delete associations wrappers * ***/ static void create_test_association(const char* extension) { HKEY hkey, hkey_shell; char class[MAX_PATH]; LONG rc; sprintf(class, "shlexec%s", extension); rc=RegCreateKeyEx(HKEY_CLASSES_ROOT, extension, 0, NULL, 0, KEY_SET_VALUE, NULL, &hkey, NULL); assert(rc==ERROR_SUCCESS); rc=RegSetValueEx(hkey, NULL, 0, REG_SZ, (LPBYTE) class, strlen(class)+1); assert(rc==ERROR_SUCCESS); CloseHandle(hkey); rc=RegCreateKeyEx(HKEY_CLASSES_ROOT, class, 0, NULL, 0, KEY_CREATE_SUB_KEY | KEY_ENUMERATE_SUB_KEYS, NULL, &hkey, NULL); assert(rc==ERROR_SUCCESS); rc=RegCreateKeyEx(hkey, "shell", 0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hkey_shell, NULL); assert(rc==ERROR_SUCCESS); CloseHandle(hkey); CloseHandle(hkey_shell); } static void delete_test_association(const char* extension) { char class[MAX_PATH]; sprintf(class, "shlexec%s", extension); SHDeleteKey(HKEY_CLASSES_ROOT, class); SHDeleteKey(HKEY_CLASSES_ROOT, extension); } static void create_test_verb(const char* extension, const char* verb, const char* cmdtail) { HKEY hkey_shell, hkey_verb, hkey_cmd; char shell[MAX_PATH]; char* cmd; LONG rc; sprintf(shell, "shlexec%s\\shell", extension); rc=RegOpenKeyEx(HKEY_CLASSES_ROOT, shell, 0, KEY_CREATE_SUB_KEY, &hkey_shell); assert(rc==ERROR_SUCCESS); rc=RegCreateKeyEx(hkey_shell, verb, 0, NULL, 0, KEY_CREATE_SUB_KEY, NULL, &hkey_verb, NULL); assert(rc==ERROR_SUCCESS); rc=RegCreateKeyEx(hkey_verb, "command", 0, NULL, 0, KEY_SET_VALUE, NULL, &hkey_cmd, NULL); assert(rc==ERROR_SUCCESS); cmd=malloc(strlen(argv0)+10+strlen(child_file)+2+strlen(cmdtail)+1); sprintf(cmd,"%s shlexec \"%s\" %s", argv0, child_file, cmdtail); rc=RegSetValueEx(hkey_cmd, NULL, 0, REG_SZ, (LPBYTE) cmd, strlen(cmd)+1); assert(rc==ERROR_SUCCESS); free(cmd); CloseHandle(hkey_shell); CloseHandle(hkey_verb); CloseHandle(hkey_cmd); } /*** * * Functions to check that the child process was started just right * (borrowed from dlls/kernel32/tests/process.c) * ***/ static const char* encodeA(const char* str) { static char encoded[2*1024+1]; char* ptr; size_t len,i; if (!str) return ""; len = strlen(str) + 1; if (len >= sizeof(encoded)/2) { fprintf(stderr, "string is too long!\n"); assert(0); } ptr = encoded; for (i = 0; i < len; i++) sprintf(&ptr[i * 2], "%02x", (unsigned char)str[i]); ptr[2 * len] = '\0'; return ptr; } static unsigned decode_char(char c) { if (c >= '0' && c <= '9') return c - '0'; if (c >= 'a' && c <= 'f') return c - 'a' + 10; assert(c >= 'A' && c <= 'F'); return c - 'A' + 10; } static char* decodeA(const char* str) { static char decoded[1024]; char* ptr; size_t len,i; len = strlen(str) / 2; if (!len--) return NULL; if (len >= sizeof(decoded)) { fprintf(stderr, "string is too long!\n"); assert(0); } ptr = decoded; for (i = 0; i < len; i++) ptr[i] = (decode_char(str[2 * i]) << 4) | decode_char(str[2 * i + 1]); ptr[len] = '\0'; return ptr; } static void childPrintf(HANDLE h, const char* fmt, ...) { va_list valist; char buffer[1024]; DWORD w; va_start(valist, fmt); vsprintf(buffer, fmt, valist); va_end(valist); WriteFile(h, buffer, strlen(buffer), &w, NULL); } static void doChild(int argc, char** argv) { char* filename; HANDLE hFile; int i; filename=argv[2]; hFile=CreateFileA(filename, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, 0); if (hFile == INVALID_HANDLE_VALUE) return; /* Arguments */ childPrintf(hFile, "[Arguments]\n"); if (winetest_debug > 2) trace("argcA=%d\n", argc); childPrintf(hFile, "argcA=%d\n", argc); for (i = 0; i < argc; i++) { if (winetest_debug > 2) trace("argvA%d=%s\n", i, argv[i]); childPrintf(hFile, "argvA%d=%s\n", i, encodeA(argv[i])); } CloseHandle(hFile); init_event(filename); SetEvent(hEvent); CloseHandle(hEvent); } static char* getChildString(const char* sect, const char* key) { char buf[1024]; char* ret; GetPrivateProfileStringA(sect, key, "-", buf, sizeof(buf), child_file); if (buf[0] == '\0' || (buf[0] == '-' && buf[1] == '\0')) return NULL; assert(!(strlen(buf) & 1)); ret = decodeA(buf); return ret; } static void dump_child(void) { if (winetest_debug > 1) { char key[18]; char* str; int i, c; c=GetPrivateProfileIntA("Arguments", "argcA", -1, child_file); trace("argcA=%d\n",c); for (i=0;ibasename) { sprintf(filename, test->basename, tmpdir); if (strchr(filename, '/')) { c=filename; while (*c) { if (*c=='\\') *c='/'; c++; } } rc=shell_execute(test->verb, filename, NULL, NULL); if (rc>=32) rc=0; if ((test->todo & 0x1)==0) { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } else todo_wine { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } if (rc==0) { const char* verb; if ((test->todo & 0x2)==0) { okChildInt("argcA", 5); } else todo_wine { okChildInt("argcA", 5); } verb=(test->verb ? test->verb : "Open"); if ((test->todo & 0x4)==0) { okChildString("argvA3", verb); } else todo_wine { okChildString("argvA3", verb); } if ((test->todo & 0x8)==0) { okChildPath("argvA4", filename); } else todo_wine { okChildPath("argvA4", filename); } } test++; } test=noquotes_tests; while (test->basename) { sprintf(filename, test->basename, tmpdir); rc=shell_execute(test->verb, filename, NULL, NULL); if (rc>=32) rc=0; if ((test->todo & 0x1)==0) { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } else todo_wine { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } if (rc==0) { int count; const char* verb; char* str; verb=(test->verb ? test->verb : "Open"); if ((test->todo & 0x4)==0) { okChildString("argvA3", verb); } else todo_wine { okChildString("argvA3", verb); } count=4; str=filename; while (1) { char attrib[18]; char* space; space=strchr(str, ' '); if (space) *space='\0'; sprintf(attrib, "argvA%d", count); if ((test->todo & 0x8)==0) { okChildPath(attrib, str); } else todo_wine { okChildPath(attrib, str); } count++; if (!space) break; str=space+1; } if ((test->todo & 0x2)==0) { okChildInt("argcA", count); } else todo_wine { okChildInt("argcA", count); } } test++; } if (dllver.dwMajorVersion != 0) { /* The more recent versions of shell32.dll accept quoted filenames * while older ones (e.g. 4.00) don't. Still we want to test this * because IE 6 depends on the new behavior. * One day we may need to check the exact version of the dll but for * now making sure DllGetVersion() is present is sufficient. */ sprintf(filename, "\"%s\\test file.shlexec\"", tmpdir); rc=shell_execute(NULL, filename, NULL, NULL); ok(rc>=32, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); okChildInt("argcA", 5); okChildString("argvA3", "Open"); sprintf(filename, "%s\\test file.shlexec", tmpdir); okChildPath("argvA4", filename); } } static filename_tests_t lnk_tests[]= { /* Pass bad / nonexistent filenames as a parameter */ {NULL, "%s\\nonexistent.shlexec", 0, 0xa}, {NULL, "%s\\nonexistent.noassoc", 0, 0xa}, /* Pass regular paths as a parameter */ {NULL, "%s\\test file.shlexec", 0, 0xa}, {NULL, "%s/%%nasty%% $file.shlexec", 0, 0xa}, /* Pass filenames with no association as a parameter */ {NULL, "%s\\test file.noassoc", 0, 0xa}, {NULL, NULL, 0, 0} }; static void test_lnks(void) { char filename[MAX_PATH]; char params[MAX_PATH]; const filename_tests_t* test; int rc; sprintf(filename, "%s\\test_shortcut_shlexec.lnk", tmpdir); rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, filename, NULL, NULL); ok(rc>=32, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); okChildInt("argcA", 5); okChildString("argvA3", "Open"); sprintf(filename, "%s\\test file.shlexec", tmpdir); okChildPath("argvA4", filename); sprintf(filename, "%s\\test_shortcut_exe.lnk", tmpdir); rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, filename, NULL, NULL); ok(rc>=32, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); okChildInt("argcA", 4); okChildString("argvA3", "Lnk"); if (dllver.dwMajorVersion>=6) { char* c; /* Recent versions of shell32.dll accept '/'s in shortcut paths. * Older versions don't or are quite buggy in this regard. */ sprintf(filename, "%s\\test_shortcut_exe.lnk", tmpdir); c=filename; while (*c) { if (*c=='\\') *c='/'; c++; } rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, filename, NULL, NULL); ok(rc>=32, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); okChildInt("argcA", 4); okChildString("argvA3", "Lnk"); } sprintf(filename, "%s\\test_shortcut_exe.lnk", tmpdir); test=lnk_tests; while (test->basename) { params[0]='\"'; sprintf(params+1, test->basename, tmpdir); strcat(params,"\""); rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, filename, params, NULL); if (rc>=32) rc=0; if ((test->todo & 0x1)==0) { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } else todo_wine { ok(rc==test->rc, "%s failed: rc=%d err=%d\n", shell_call, rc, GetLastError()); } if (rc==0) { if ((test->todo & 0x2)==0) { okChildInt("argcA", 5); } else { okChildInt("argcA", 5); } if ((test->todo & 0x4)==0) { okChildString("argvA3", "Lnk"); } else todo_wine { okChildString("argvA3", "Lnk"); } sprintf(params, test->basename, tmpdir); if ((test->todo & 0x8)==0) { okChildPath("argvA4", params); } else { okChildPath("argvA4", params); } } test++; } } static void test_exes(void) { char filename[MAX_PATH]; char params[1024]; int rc; sprintf(params, "shlexec \"%s\" Exec", child_file); /* We need NOZONECHECKS on Win2003 to block a dialog */ rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, argv0, params, NULL); ok(rc>=32, "%s returned %d\n", shell_call, rc); okChildInt("argcA", 4); okChildString("argvA3", "Exec"); sprintf(filename, "%s\\test file.noassoc", tmpdir); if (CopyFile(argv0, filename, FALSE)) { rc=shell_execute(NULL, filename, params, NULL); todo_wine { ok(rc==SE_ERR_NOASSOC, "%s succeeded: rc=%d\n", shell_call, rc); } } } static void test_exes_long(void) { char filename[MAX_PATH]; char params[2024]; char longparam[MAX_PATH]; int rc; for (rc = 0; rc < MAX_PATH; rc++) longparam[rc]='a'+rc%26; longparam[MAX_PATH-1]=0; sprintf(params, "shlexec \"%s\" %s", child_file,longparam); /* We need NOZONECHECKS on Win2003 to block a dialog */ rc=shell_execute_ex(SEE_MASK_NOZONECHECKS, NULL, argv0, params, NULL); ok(rc>=32, "%s returned %d\n", shell_call, rc); okChildInt("argcA", 4); okChildString("argvA3", longparam); sprintf(filename, "%s\\test file.noassoc", tmpdir); if (CopyFile(argv0, filename, FALSE)) { rc=shell_execute(NULL, filename, params, NULL); todo_wine { ok(rc==SE_ERR_NOASSOC, "%s succeeded: rc=%d\n", shell_call, rc); } } } static void init_test(void) { HMODULE hdll; HRESULT (WINAPI *pDllGetVersion)(DLLVERSIONINFO*); char filename[MAX_PATH]; WCHAR lnkfile[MAX_PATH]; char params[1024]; const char* const * testfile; lnk_desc_t desc; DWORD rc; HRESULT r; hdll=GetModuleHandleA("shell32.dll"); pDllGetVersion=(void*)GetProcAddress(hdll, "DllGetVersion"); if (pDllGetVersion) { dllver.cbSize=sizeof(dllver); pDllGetVersion(&dllver); trace("major=%d minor=%d build=%d platform=%d\n", dllver.dwMajorVersion, dllver.dwMinorVersion, dllver.dwBuildNumber, dllver.dwPlatformID); } else { memset(&dllver, 0, sizeof(dllver)); } r = CoInitialize(NULL); ok(SUCCEEDED(r), "CoInitialize failed (0x%08x)\n", r); if (!SUCCEEDED(r)) exit(1); rc=GetModuleFileName(NULL, argv0, sizeof(argv0)); assert(rc!=0 && rc= 3) { doChild(myARGC, myARGV); exit(0); } init_test(); test_filename(); test_lnks(); test_exes(); test_exes_long(); cleanup_test(); }