[SHELL32][APITESTS] Add tests of OpenAs_RunDLL
[reactos.git] / modules / rostests / apitests / shell32 / OpenAs_RunDLL.cpp
1 /*
2 * PROJECT: ReactOS API tests
3 * LICENSE: LGPLv2.1+ - See COPYING.LIB in the top level directory
4 * PURPOSE: Test for OpenAs_RunDLL
5 * PROGRAMMERS: Katayama Hirofumi MZ <katayama.hirofumi.mz@gmail.com>
6 */
7
8 #include "shelltest.h"
9 #include <stdio.h>
10 #include <process.h>
11
12 // OpenAs_RunDLLA
13 typedef void (WINAPI *OPENAS_RUNDLLA)(HWND, HINSTANCE, LPCSTR, int);
14 // OpenAs_RunDLLW
15 typedef void (WINAPI *OPENAS_RUNDLLW)(HWND, HINSTANCE, LPCWSTR, int);
16
17 static OPENAS_RUNDLLA pOpenAs_RunDLLA = NULL;
18 static OPENAS_RUNDLLW pOpenAs_RunDLLW = NULL;
19
20 struct TEST_ENTRY
21 {
22 int nLine;
23 BOOL bWide;
24 HINSTANCE hInst;
25 int nRet;
26 BOOL bCreateFile;
27 LPCSTR pszFileA;
28 LPCWSTR pszFileW;
29 DWORD dwError;
30 };
31
32 #define COUNT 5
33 #define INTERVAL 200
34
35 #define DEATH 999
36 #define OPENED 1
37 #define NOT_OPENED 0
38
39 #define BAD_INST ((HINSTANCE)0xDEAD)
40 #define BAD_SZ_A ((LPCSTR)0xDEAD)
41 #define BAD_SZ_W ((LPCWSTR)0xDEAD)
42
43 static const TEST_ENTRY s_TestEntries[] =
44 {
45 // ANSI
46 {__LINE__, FALSE, NULL, OPENED, FALSE, NULL, NULL, 0 },
47 {__LINE__, FALSE, NULL, OPENED, FALSE, "", NULL, 0 },
48 {__LINE__, FALSE, NULL, OPENED, FALSE, "invalid file name.txt", NULL, 0 },
49 {__LINE__, FALSE, NULL, DEATH, FALSE, BAD_SZ_A, NULL, 0 },
50 {__LINE__, FALSE, NULL, OPENED, TRUE, "created file.txt", NULL, 0 },
51 {__LINE__, FALSE, BAD_INST, OPENED, FALSE, NULL, NULL, 0 },
52 {__LINE__, FALSE, BAD_INST, OPENED, FALSE, "invalid file name.txt", NULL, 0 },
53 {__LINE__, FALSE, BAD_INST, DEATH, FALSE, BAD_SZ_A, NULL, 0xDEADFACE },
54 {__LINE__, FALSE, BAD_INST, OPENED, TRUE, "created file.txt", NULL, 0 },
55 // WIDE
56 {__LINE__, TRUE, NULL, OPENED, FALSE, NULL, NULL, 0x80070490 },
57 {__LINE__, TRUE, NULL, OPENED, FALSE, NULL, L"", ERROR_NO_SCROLLBARS },
58 {__LINE__, TRUE, NULL, OPENED, FALSE, NULL, L"invalid file name.txt", ERROR_NO_SCROLLBARS },
59 {__LINE__, TRUE, NULL, OPENED, FALSE, NULL, BAD_SZ_W, 0xDEADFACE },
60 {__LINE__, TRUE, NULL, OPENED, TRUE, NULL, L"created file.txt", ERROR_NO_SCROLLBARS },
61 {__LINE__, TRUE, BAD_INST, OPENED, FALSE, NULL, NULL, 0x80070490 },
62 {__LINE__, TRUE, BAD_INST, OPENED, FALSE, NULL, L"invalid file name.txt", ERROR_NO_SCROLLBARS },
63 {__LINE__, TRUE, BAD_INST, OPENED, FALSE, NULL, BAD_SZ_W, 0xDEADFACE },
64 {__LINE__, TRUE, BAD_INST, OPENED, TRUE, NULL, L"created file.txt", ERROR_NO_SCROLLBARS },
65 };
66
67 static HANDLE s_hThread = NULL;
68 static INT s_nRet = NOT_OPENED;
69
70 static unsigned __stdcall
71 watch_thread_proc(void *arg)
72 {
73 for (int i = 0; i < COUNT; ++i)
74 {
75 Sleep(INTERVAL);
76
77 HWND hwnd = FindWindowA("#32770", "Open With");
78 if (hwnd == NULL)
79 continue;
80
81 if (!IsWindowVisible(hwnd))
82 {
83 trace("not visible\n");
84 continue;
85 }
86
87 s_nRet = OPENED;
88 PostMessage(hwnd, WM_COMMAND, MAKEWPARAM(IDCANCEL, BN_CLICKED), 0);
89 break;
90 }
91 return 0;
92 }
93
94 static void StartWatchGUI(const TEST_ENTRY *pEntry)
95 {
96 s_nRet = NOT_OPENED;
97 s_hThread = (HANDLE)_beginthreadex(NULL, 0, watch_thread_proc, NULL, 0, NULL);
98 }
99
100 static void DoEntry(const TEST_ENTRY *pEntry)
101 {
102 if (pEntry->bWide)
103 {
104 if (pEntry->bCreateFile)
105 {
106 FILE *fp = _wfopen(pEntry->pszFileW, L"wb");
107 ok(fp != NULL, "Line %d: Unable to create file '%s'\n", pEntry->nLine, wine_dbgstr_w(pEntry->pszFileW));
108 fclose(fp);
109 }
110
111 StartWatchGUI(pEntry);
112 SetLastError(0xDEADFACE);
113
114 _SEH2_TRY
115 {
116 (*pOpenAs_RunDLLW)(NULL, pEntry->hInst, pEntry->pszFileW, SW_SHOWDEFAULT);
117 }
118 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
119 {
120 s_nRet = DEATH;
121 }
122 _SEH2_END;
123
124 DWORD dwError = GetLastError();
125 ok(pEntry->dwError == dwError, "Line %d: error expected %ld, was %ld\n", pEntry->nLine, pEntry->dwError, dwError);
126
127 WaitForSingleObject(s_hThread, INFINITE);
128 CloseHandle(s_hThread);
129
130 if (pEntry->bCreateFile)
131 {
132 ok(DeleteFileW(pEntry->pszFileW), "Line %d: DeleteFileA failed\n", pEntry->nLine);
133 }
134 }
135 else
136 {
137 if (pEntry->bCreateFile)
138 {
139 FILE *fp = fopen(pEntry->pszFileA, "wb");
140 ok(fp != NULL, "Line %d: Unable to create file '%s'\n", pEntry->nLine, pEntry->pszFileA);
141 fclose(fp);
142 }
143
144 StartWatchGUI(pEntry);
145 SetLastError(0xDEADFACE);
146
147 _SEH2_TRY
148 {
149 (*pOpenAs_RunDLLA)(NULL, pEntry->hInst, pEntry->pszFileA, SW_SHOWDEFAULT);
150 }
151 _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
152 {
153 s_nRet = DEATH;
154 }
155 _SEH2_END;
156
157 DWORD dwError = GetLastError();
158 ok(pEntry->dwError == dwError, "Line %d: error expected %ld, was %ld\n", pEntry->nLine, pEntry->dwError, dwError);
159
160 WaitForSingleObject(s_hThread, INFINITE);
161 CloseHandle(s_hThread);
162
163 if (pEntry->bCreateFile)
164 {
165 ok(DeleteFileA(pEntry->pszFileA), "Line %d: DeleteFileA failed\n", pEntry->nLine);
166 }
167 }
168
169 ok(pEntry->nRet == s_nRet, "Line %d: s_nRet expected %d, was %d\n", pEntry->nLine, pEntry->nRet, s_nRet);
170 }
171
172 START_TEST(OpenAs_RunDLL)
173 {
174 HINSTANCE hShell32 = LoadLibraryA("shell32.dll");
175 if (hShell32 == NULL)
176 {
177 skip("Unable to load shell32.dll\n");
178 return;
179 }
180
181 pOpenAs_RunDLLA = (OPENAS_RUNDLLA)GetProcAddress(hShell32, "OpenAs_RunDLLA");
182 if (pOpenAs_RunDLLA == NULL)
183 {
184 skip("Unable to get OpenAs_RunDLLA\n");
185 return;
186 }
187
188 pOpenAs_RunDLLW = (OPENAS_RUNDLLW)GetProcAddress(hShell32, "OpenAs_RunDLLW");
189 if (pOpenAs_RunDLLW == NULL)
190 {
191 skip("Unable to get OpenAs_RunDLLW\n");
192 return;
193 }
194
195 TCHAR szCurDir[MAX_PATH], szTempDir[MAX_PATH];
196
197 ok(GetCurrentDirectory(_countof(szCurDir), szCurDir), "GetCurrentDirectory failed\n");
198 ok(GetTempPath(_countof(szTempDir), szTempDir), "GetTempPath failed\n");
199
200 if (!SetCurrentDirectory(szTempDir))
201 {
202 skip("SetCurrentDirectory failed\n");
203 }
204 else
205 {
206 for (size_t i = 0; i < _countof(s_TestEntries); ++i)
207 {
208 const TEST_ENTRY *pEntry = &s_TestEntries[i];
209 DoEntry(pEntry);
210 }
211
212 ok(SetCurrentDirectory(szCurDir), "SetCurrentDirectory failed\n");
213 }
214
215 FreeLibrary(hShell32);
216 }