msi: Make MsiDoAction() RPC-compatible.
[wine.git] / dlls / msi / tests / custom.c
blob5bdb8d2cc5538db3aa752ed369e40299e9f669ad
1 /*
2 * DLL for testing type 1 custom actions
4 * Copyright 2017 Zebediah Figura
6 * This library is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * This library is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with this library; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 #include <stdarg.h>
22 #include <stdio.h>
24 #include <windef.h>
25 #include <winbase.h>
26 #define COBJMACROS
27 #include <msxml.h>
28 #include <msi.h>
29 #include <msiquery.h>
31 static void ok_(MSIHANDLE hinst, int todo, const char *file, int line, int condition, const char *msg, ...)
33 static char buffer[2000];
34 MSIHANDLE record;
35 va_list valist;
37 va_start(valist, msg);
38 vsprintf(buffer, msg, valist);
39 va_end(valist);
41 record = MsiCreateRecord(5);
42 MsiRecordSetInteger(record, 1, todo);
43 MsiRecordSetStringA(record, 2, file);
44 MsiRecordSetInteger(record, 3, line);
45 MsiRecordSetInteger(record, 4, condition);
46 MsiRecordSetStringA(record, 5, buffer);
47 MsiProcessMessage(hinst, INSTALLMESSAGE_USER, record);
48 MsiCloseHandle(record);
50 #define ok(hinst, condition, ...) ok_(hinst, 0, __FILE__, __LINE__, condition, __VA_ARGS__)
51 #define todo_wine_ok(hinst, condition, ...) ok_(hinst, 1, __FILE__, __LINE__, condition, __VA_ARGS__)
53 static const char *dbgstr_w(WCHAR *str)
55 static char buffer[300], *p;
57 if (!str) return "(null)";
59 p = buffer;
60 *p++ = 'L';
61 *p++ = '"';
62 while ((*p++ = *str++));
63 *p++ = '"';
64 *p++ = 0;
66 return buffer;
69 static void check_prop(MSIHANDLE hinst, const char *prop, const char *expect)
71 char buffer[10] = "x";
72 DWORD sz = sizeof(buffer);
73 UINT r = MsiGetPropertyA(hinst, prop, buffer, &sz);
74 ok(hinst, !r, "'%s': got %u\n", prop, r);
75 ok(hinst, sz == strlen(buffer), "'%s': expected %u, got %u\n", prop, strlen(buffer), sz);
76 ok(hinst, !strcmp(buffer, expect), "expected '%s', got '%s'\n", expect, buffer);
79 static void test_props(MSIHANDLE hinst)
81 static const WCHAR booW[] = {'b','o','o',0};
82 static const WCHAR xyzW[] = {'x','y','z',0};
83 static const WCHAR xyW[] = {'x','y',0};
84 char buffer[10];
85 WCHAR bufferW[10];
86 DWORD sz;
87 UINT r;
89 /* test invalid values */
90 r = MsiGetPropertyA(hinst, NULL, NULL, NULL);
91 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
93 r = MsiGetPropertyA(hinst, "boo", NULL, NULL);
94 ok(hinst, !r, "got %u\n", r);
96 r = MsiGetPropertyA(hinst, "boo", buffer, NULL );
97 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
99 sz = 0;
100 r = MsiGetPropertyA(hinst, "boo", NULL, &sz);
101 ok(hinst, !r, "got %u\n", r);
102 ok(hinst, sz == 0, "got size %u\n", sz);
104 sz = 0;
105 strcpy(buffer,"x");
106 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
107 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
108 ok(hinst, !strcmp(buffer, "x"), "got \"%s\"\n", buffer);
109 ok(hinst, sz == 0, "got size %u\n", sz);
111 sz = 1;
112 strcpy(buffer,"x");
113 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
114 ok(hinst, !r, "got %u\n", r);
115 ok(hinst, !buffer[0], "got \"%s\"\n", buffer);
116 ok(hinst, sz == 0, "got size %u\n", sz);
118 /* set the property to something */
119 r = MsiSetPropertyA(hinst, NULL, NULL);
120 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
122 r = MsiSetPropertyA(hinst, "", NULL);
123 ok(hinst, !r, "got %u\n", r);
125 r = MsiSetPropertyA(hinst, "", "asdf");
126 ok(hinst, r == ERROR_FUNCTION_FAILED, "got %u\n", r);
128 r = MsiSetPropertyA(hinst, "=", "asdf");
129 ok(hinst, !r, "got %u\n", r);
130 check_prop(hinst, "=", "asdf");
132 r = MsiSetPropertyA(hinst, " ", "asdf");
133 ok(hinst, !r, "got %u\n", r);
134 check_prop(hinst, " ", "asdf");
136 r = MsiSetPropertyA(hinst, "'", "asdf");
137 ok(hinst, !r, "got %u\n", r);
138 check_prop(hinst, "'", "asdf");
140 r = MsiSetPropertyA(hinst, "boo", NULL);
141 ok(hinst, !r, "got %u\n", r);
142 check_prop(hinst, "boo", "");
144 r = MsiSetPropertyA(hinst, "boo", "");
145 ok(hinst, !r, "got %u\n", r);
146 check_prop(hinst, "boo", "");
148 r = MsiSetPropertyA(hinst, "boo", "xyz");
149 ok(hinst, !r, "got %u\n", r);
150 check_prop(hinst, "boo", "xyz");
152 r = MsiGetPropertyA(hinst, "boo", NULL, NULL);
153 ok(hinst, !r, "got %u\n", r);
155 r = MsiGetPropertyA(hinst, "boo", buffer, NULL );
156 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
158 /* Returned size is in bytes, not chars, but only for custom actions.
159 * Seems to be a casualty of RPC... */
161 sz = 0;
162 r = MsiGetPropertyA(hinst, "boo", NULL, &sz);
163 ok(hinst, !r, "got %u\n", r);
164 ok(hinst, sz == 6, "got size %u\n", sz);
166 sz = 0;
167 strcpy(buffer,"q");
168 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
169 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
170 ok(hinst, !strcmp(buffer, "q"), "got \"%s\"\n", buffer);
171 todo_wine_ok(hinst, sz == 6, "got size %u\n", sz);
173 sz = 1;
174 strcpy(buffer,"x");
175 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
176 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
177 ok(hinst, !buffer[0], "got \"%s\"\n", buffer);
178 todo_wine_ok(hinst, sz == 6, "got size %u\n", sz);
180 sz = 3;
181 strcpy(buffer,"x");
182 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
183 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
184 ok(hinst, !strcmp(buffer, "xy"), "got \"%s\"\n", buffer);
185 todo_wine_ok(hinst, sz == 6, "got size %u\n", sz);
187 sz = 4;
188 strcpy(buffer,"x");
189 r = MsiGetPropertyA(hinst, "boo", buffer, &sz);
190 ok(hinst, !r, "got %u\n", r);
191 ok(hinst, !strcmp(buffer, "xyz"), "got \"%s\"\n", buffer);
192 ok(hinst, sz == 3, "got size %u\n", sz);
194 sz = 0;
195 r = MsiGetPropertyW(hinst, booW, NULL, &sz);
196 ok(hinst, !r, "got %u\n", r);
197 ok(hinst, sz == 3, "got size %u\n", sz);
199 sz = 0;
200 lstrcpyW(bufferW, booW);
201 r = MsiGetPropertyW(hinst, booW, bufferW, &sz);
202 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
203 ok(hinst, !lstrcmpW(bufferW, booW), "got %s\n", dbgstr_w(bufferW));
204 ok(hinst, sz == 3, "got size %u\n", sz);
206 sz = 1;
207 lstrcpyW(bufferW, booW);
208 r = MsiGetPropertyW(hinst, booW, bufferW, &sz);
209 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
210 ok(hinst, !bufferW[0], "got %s\n", dbgstr_w(bufferW));
211 ok(hinst, sz == 3, "got size %u\n", sz);
213 sz = 3;
214 lstrcpyW(bufferW, booW);
215 r = MsiGetPropertyW(hinst, booW, bufferW, &sz);
216 ok(hinst, r == ERROR_MORE_DATA, "got %u\n", r);
217 ok(hinst, !lstrcmpW(bufferW, xyW), "got %s\n", dbgstr_w(bufferW));
218 ok(hinst, sz == 3, "got size %u\n", sz);
220 sz = 4;
221 lstrcpyW(bufferW, booW);
222 r = MsiGetPropertyW(hinst, booW, bufferW, &sz);
223 ok(hinst, !r, "got %u\n", r);
224 ok(hinst, !lstrcmpW(bufferW, xyzW), "got %s\n", dbgstr_w(bufferW));
225 ok(hinst, sz == 3, "got size %u\n", sz);
227 r = MsiSetPropertyA(hinst, "boo", NULL);
228 ok(hinst, !r, "got %u\n", r);
229 check_prop(hinst, "boo", "");
231 sz = 0;
232 r = MsiGetPropertyA(hinst, "embednullprop", NULL, &sz);
233 ok(hinst, !r, "got %u\n", r);
234 ok(hinst, sz == 6, "got size %u\n", sz);
236 sz = 4;
237 memset(buffer, 0xcc, sizeof(buffer));
238 r = MsiGetPropertyA(hinst, "embednullprop", buffer, &sz);
239 ok(hinst, !r, "got %u\n", r);
240 ok(hinst, sz == 3, "got size %u\n", sz);
241 ok(hinst, !memcmp(buffer, "a\0\0\0\xcc", 5), "wrong data\n");
244 static void test_db(MSIHANDLE hinst)
246 MSIHANDLE hdb, view, rec, rec2, suminfo;
247 char buffer[10];
248 DWORD sz;
249 UINT r;
251 hdb = MsiGetActiveDatabase(hinst);
252 ok(hinst, hdb, "MsiGetActiveDatabase failed\n");
254 r = MsiDatabaseIsTablePersistentA(hdb, "Test");
255 ok(hinst, r == MSICONDITION_TRUE, "got %u\n", r);
257 r = MsiDatabaseOpenViewA(hdb, NULL, &view);
258 ok(hinst, r == ERROR_BAD_QUERY_SYNTAX, "got %u\n", r);
260 r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test`", NULL);
261 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
263 r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test`", &view);
264 ok(hinst, !r, "got %u\n", r);
266 r = MsiViewGetColumnInfo(view, MSICOLINFO_NAMES, &rec2);
267 ok(hinst, !r, "got %u\n", r);
269 sz = sizeof(buffer);
270 r = MsiRecordGetStringA(rec2, 1, buffer, &sz);
271 ok(hinst, !r, "got %u\n", r);
272 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
273 ok(hinst, !strcmp(buffer, "Name"), "got '%s'\n", buffer);
275 r = MsiCloseHandle(rec2);
276 ok(hinst, !r, "got %u\n", r);
278 r = MsiViewExecute(view, 0);
279 ok(hinst, !r, "got %u\n", r);
281 r = MsiViewFetch(view, &rec2);
282 ok(hinst, !r, "got %u\n", r);
284 r = MsiRecordGetFieldCount(rec2);
285 ok(hinst, r == 3, "got %u\n", r);
287 sz = sizeof(buffer);
288 r = MsiRecordGetStringA(rec2, 1, buffer, &sz);
289 ok(hinst, !r, "got %u\n", r);
290 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
291 ok(hinst, !strcmp(buffer, "one"), "got '%s'\n", buffer);
293 r = MsiRecordGetInteger(rec2, 2);
294 ok(hinst, r == 1, "got %d\n", r);
296 sz = sizeof(buffer);
297 r = MsiRecordReadStream(rec2, 3, buffer, &sz);
298 ok(hinst, !r, "got %u\n", r);
299 ok(hinst, !memcmp(buffer, "unus", 4), "wrong data\n");
301 r = MsiCloseHandle(rec2);
302 ok(hinst, !r, "got %u\n", r);
304 r = MsiViewFetch(view, &rec2);
305 ok(hinst, !r, "got %u\n", r);
307 r = MsiRecordGetFieldCount(rec2);
308 ok(hinst, r == 3, "got %u\n", r);
310 sz = sizeof(buffer);
311 r = MsiRecordGetStringA(rec2, 1, buffer, &sz);
312 ok(hinst, !r, "got %u\n", r);
313 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
314 ok(hinst, !strcmp(buffer, "two"), "got '%s'\n", buffer);
316 r = MsiRecordGetInteger(rec2, 2);
317 ok(hinst, r == 2, "got %d\n", r);
319 sz = sizeof(buffer);
320 r = MsiRecordReadStream(rec2, 3, buffer, &sz);
321 ok(hinst, !r, "got %u\n", r);
322 ok(hinst, !memcmp(buffer, "duo", 3), "wrong data\n");
324 r = MsiViewModify(view, MSIMODIFY_REFRESH, 0);
325 ok(hinst, r == ERROR_INVALID_HANDLE, "got %u\n", r);
327 r = MsiRecordSetStringA(rec2, 1, "three");
328 ok(hinst, !r, "got %u\n", r);
330 r = MsiRecordSetInteger(rec2, 2, 3);
331 ok(hinst, !r, "got %u\n", r);
333 r = MsiRecordSetInteger(rec2, 3, 3);
334 ok(hinst, !r, "got %u\n", r);
336 r = MsiViewModify(view, MSIMODIFY_REFRESH, rec2);
337 ok(hinst, !r, "got %d\n", r);
339 sz = sizeof(buffer);
340 r = MsiRecordGetStringA(rec2, 1, buffer, &sz);
341 ok(hinst, !r, "got %u\n", r);
342 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
343 ok(hinst, !strcmp(buffer, "two"), "got '%s'\n", buffer);
345 r = MsiRecordGetInteger(rec2, 2);
346 ok(hinst, r == 2, "got %d\n", r);
348 sz = sizeof(buffer);
349 r = MsiRecordReadStream(rec2, 3, buffer, &sz);
350 ok(hinst, !r, "got %u\n", r);
351 ok(hinst, !memcmp(buffer, "duo", 3), "wrong data\n");
353 r = MsiCloseHandle(rec2);
354 ok(hinst, !r, "got %u\n", r);
356 r = MsiViewFetch(view, &rec2);
357 ok(hinst, r == ERROR_NO_MORE_ITEMS, "got %u\n", r);
358 ok(hinst, !rec2, "got %u\n", rec2);
360 r = MsiViewClose(view);
361 ok(hinst, !r, "got %u\n", r);
363 r = MsiCloseHandle(view);
364 ok(hinst, !r, "got %u\n", r);
366 r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test` WHERE `Name` = ?", &view);
367 ok(hinst, !r, "got %u\n", r);
369 rec = MsiCreateRecord(1);
370 MsiRecordSetStringA(rec, 1, "one");
372 r = MsiViewExecute(view, rec);
373 ok(hinst, !r, "got %u\n", r);
375 r = MsiViewFetch(view, &rec2);
376 ok(hinst, !r, "got %u\n", r);
378 r = MsiRecordGetInteger(rec2, 2);
379 ok(hinst, r == 1, "got %d\n", r);
381 r = MsiCloseHandle(rec2);
382 ok(hinst, !r, "got %u\n", r);
384 r = MsiViewFetch(view, &rec2);
385 ok(hinst, r == ERROR_NO_MORE_ITEMS, "got %u\n", r);
386 ok(hinst, !rec2, "got %u\n", rec2);
388 r = MsiCloseHandle(rec);
389 ok(hinst, !r, "got %u\n", r);
391 r = MsiCloseHandle(view);
392 ok(hinst, !r, "got %u\n", r);
394 /* test MsiDatabaseGetPrimaryKeys() */
395 r = MsiDatabaseGetPrimaryKeysA(hdb, "Test", &rec);
396 ok(hinst, !r, "got %u\n", r);
398 r = MsiRecordGetFieldCount(rec);
399 ok(hinst, r == 1, "got %d\n", r);
401 sz = sizeof(buffer);
402 r = MsiRecordGetStringA(rec, 0, buffer, &sz);
403 ok(hinst, !r, "got %u\n", r);
404 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
405 ok(hinst, !strcmp(buffer, "Test"), "got '%s'\n", buffer);
407 sz = sizeof(buffer);
408 r = MsiRecordGetStringA(rec, 1, buffer, &sz);
409 ok(hinst, !r, "got %u\n", r);
410 ok(hinst, sz == strlen(buffer), "got size %u\n", sz);
411 ok(hinst, !strcmp(buffer, "Name"), "got '%s'\n", buffer);
413 r = MsiCloseHandle(rec);
414 ok(hinst, !r, "got %u\n", r);
416 r = MsiGetSummaryInformationA(hdb, NULL, 1, NULL);
417 ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
419 r = MsiGetSummaryInformationA(hdb, NULL, 1, &suminfo);
420 ok(hinst, !r, "got %u\n", r);
422 r = MsiCloseHandle(suminfo);
423 ok(hinst, !r, "got %u\n", r);
425 r = MsiCloseHandle(hdb);
426 ok(hinst, !r, "got %u\n", r);
429 static void test_doaction(MSIHANDLE hinst)
431 UINT r;
433 r = MsiDoActionA(hinst, "nested51");
434 ok(hinst, !r, "got %u\n", r);
435 check_prop(hinst, "nested", "1");
437 r = MsiDoActionA(hinst, "nested1");
438 ok(hinst, !r, "got %u\n", r);
439 check_prop(hinst, "nested", "2");
442 UINT WINAPI nested(MSIHANDLE hinst)
444 MsiSetPropertyA(hinst, "nested", "2");
446 return ERROR_SUCCESS;
449 /* Main test. Anything that doesn't depend on a specific install configuration
450 * or have undesired side effects should go here. */
451 UINT WINAPI main_test(MSIHANDLE hinst)
453 UINT res;
454 IUnknown *unk = NULL;
455 HRESULT hr;
457 /* Test for an MTA apartment */
458 hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&unk);
459 todo_wine_ok(hinst, hr == S_OK, "CoCreateInstance failed with %08x\n", hr);
461 if (unk) IUnknown_Release(unk);
463 /* but ours is uninitialized */
464 hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
465 ok(hinst, hr == S_OK, "got %#x\n", hr);
466 CoUninitialize();
468 /* Test MsiGetDatabaseState() */
469 res = MsiGetDatabaseState(hinst);
470 todo_wine_ok(hinst, res == MSIDBSTATE_ERROR, "expected MSIDBSTATE_ERROR, got %u\n", res);
472 test_props(hinst);
473 test_db(hinst);
474 test_doaction(hinst);
476 return ERROR_SUCCESS;
479 UINT WINAPI test_retval(MSIHANDLE hinst)
481 char prop[10];
482 DWORD len = sizeof(prop);
483 UINT retval;
485 MsiGetPropertyA(hinst, "TEST_RETVAL", prop, &len);
486 sscanf(prop, "%u", &retval);
487 return retval;
490 static void append_file(MSIHANDLE hinst, const char *filename, const char *text)
492 DWORD size;
493 HANDLE file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
494 ok(hinst, file != INVALID_HANDLE_VALUE, "CreateFile failed, error %u\n", GetLastError());
496 SetFilePointer(file, 0, NULL, FILE_END);
497 WriteFile(file, text, strlen(text), &size, NULL);
498 CloseHandle(file);
501 UINT WINAPI da_immediate(MSIHANDLE hinst)
503 char prop[300];
504 DWORD len = sizeof(prop);
506 MsiGetPropertyA(hinst, "TESTPATH", prop, &len);
508 append_file(hinst, prop, "one");
510 ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_SCHEDULED), "shouldn't be scheduled\n");
511 ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_ROLLBACK), "shouldn't be rollback\n");
512 ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_COMMIT), "shouldn't be commit\n");
514 return ERROR_SUCCESS;
517 UINT WINAPI da_deferred(MSIHANDLE hinst)
519 char prop[300];
520 DWORD len = sizeof(prop);
521 LANGID lang;
522 UINT r;
524 /* Test that we were in fact deferred */
525 r = MsiGetPropertyA(hinst, "CustomActionData", prop, &len);
526 ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
527 ok(hinst, prop[0], "CustomActionData was empty\n");
529 append_file(hinst, prop, "two");
531 /* Test available properties */
532 len = sizeof(prop);
533 r = MsiGetPropertyA(hinst, "ProductCode", prop, &len);
534 ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
535 ok(hinst, prop[0], "got %s\n", prop);
537 len = sizeof(prop);
538 r = MsiGetPropertyA(hinst, "UserSID", prop, &len);
539 ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
540 ok(hinst, prop[0], "got %s\n", prop);
542 len = sizeof(prop);
543 r = MsiGetPropertyA(hinst, "TESTPATH", prop, &len);
544 ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
545 todo_wine_ok(hinst, !prop[0], "got %s\n", prop);
547 /* Test modes */
548 ok(hinst, MsiGetMode(hinst, MSIRUNMODE_SCHEDULED), "should be scheduled\n");
549 ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_ROLLBACK), "shouldn't be rollback\n");
550 ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_COMMIT), "shouldn't be commit\n");
552 lang = MsiGetLanguage(hinst);
553 ok(hinst, lang != ERROR_INVALID_HANDLE, "MsiGetLanguage failed\n");
555 return ERROR_SUCCESS;