dlls/msi/tests/custom.c

   1 /*
   2  * DLL for testing type 1 custom actions
   3  *
   4  * Copyright 2017 Zebediah Figura
   5  *
   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.
  10  *
  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.
  15  *
  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
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #include <windef.h>
  25 #include <winbase.h>
  26 #define COBJMACROS
  27 #include <msxml.h>
  28 #include <msi.h>
  29 #include <msiquery.h>
  30
  31 static void ok_(MSIHANDLE hinst, int todo, const char *file, int line, int condition, const char *msg, ...)
  32 {
  33     static char buffer[2000];
  34     MSIHANDLE record;
  35     va_list valist;
  36
  37     va_start(valist, msg);
  38     vsprintf(buffer, msg, valist);
  39     va_end(valist);
  40
  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);
  49 }
  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__)
  52
  53 static const char *dbgstr_w(WCHAR *str)
  54 {
  55     static char buffer[300], *p;
  56
  57     if (!str) return "(null)";
  58
  59     p = buffer;
  60     *p++ = 'L';
  61     *p++ = '"';
  62     while ((*p++ = *str++));
  63     *p++ = '"';
  64     *p++ = 0;
  65
  66     return buffer;
  67 }
  68
  69 static void check_prop(MSIHANDLE hinst, const char *prop, const char *expect)
  70 {
  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);
  77 }
  78
  79 static void test_props(MSIHANDLE hinst)
  80 {
  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;
  88
  89     /* test invalid values */
  90     r = MsiGetPropertyA(hinst, NULL, NULL, NULL);
  91     ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
  92
  93     r = MsiGetPropertyA(hinst, "boo", NULL, NULL);
  94     ok(hinst, !r, "got %u\n", r);
  95
  96     r = MsiGetPropertyA(hinst, "boo", buffer, NULL );
  97     ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
  98
  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);
 103
 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);
 110
 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);
 117
 118     /* set the property to something */
 119     r = MsiSetPropertyA(hinst, NULL, NULL);
 120     ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
 121
 122     r = MsiSetPropertyA(hinst, "", NULL);
 123     ok(hinst, !r, "got %u\n", r);
 124
 125     r = MsiSetPropertyA(hinst, "", "asdf");
 126     ok(hinst, r == ERROR_FUNCTION_FAILED, "got %u\n", r);
 127
 128     r = MsiSetPropertyA(hinst, "=", "asdf");
 129     ok(hinst, !r, "got %u\n", r);
 130     check_prop(hinst, "=", "asdf");
 131
 132     r = MsiSetPropertyA(hinst, " ", "asdf");
 133     ok(hinst, !r, "got %u\n", r);
 134     check_prop(hinst, " ", "asdf");
 135
 136     r = MsiSetPropertyA(hinst, "'", "asdf");
 137     ok(hinst, !r, "got %u\n", r);
 138     check_prop(hinst, "'", "asdf");
 139
 140     r = MsiSetPropertyA(hinst, "boo", NULL);
 141     ok(hinst, !r, "got %u\n", r);
 142     check_prop(hinst, "boo", "");
 143
 144     r = MsiSetPropertyA(hinst, "boo", "");
 145     ok(hinst, !r, "got %u\n", r);
 146     check_prop(hinst, "boo", "");
 147
 148     r = MsiSetPropertyA(hinst, "boo", "xyz");
 149     ok(hinst, !r, "got %u\n", r);
 150     check_prop(hinst, "boo", "xyz");
 151
 152     r = MsiGetPropertyA(hinst, "boo", NULL, NULL);
 153     ok(hinst, !r, "got %u\n", r);
 154
 155     r = MsiGetPropertyA(hinst, "boo", buffer, NULL );
 156     ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
 157
 158     /* Returned size is in bytes, not chars, but only for custom actions.
 159      * Seems to be a casualty of RPC... */
 160
 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);
 165
 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);
 172
 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);
 179
 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);
 186
 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);
 193
 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);
 198
 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);
 205
 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);
 212
 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);
 219
 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);
 226
 227     r = MsiSetPropertyA(hinst, "boo", NULL);
 228     ok(hinst, !r, "got %u\n", r);
 229     check_prop(hinst, "boo", "");
 230
 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);
 235
 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");
 242 }
 243
 244 static void test_db(MSIHANDLE hinst)
 245 {
 246     MSIHANDLE hdb, view, rec, rec2;
 247     char buffer[10];
 248     DWORD sz;
 249     UINT r;
 250
 251     hdb = MsiGetActiveDatabase(hinst);
 252     ok(hinst, hdb, "MsiGetActiveDatabase failed\n");
 253
 254     r = MsiDatabaseIsTablePersistentA(hdb, "Test");
 255     ok(hinst, r == MSICONDITION_TRUE, "got %u\n", r);
 256
 257     r = MsiDatabaseOpenViewA(hdb, NULL, &view);
 258     ok(hinst, r == ERROR_BAD_QUERY_SYNTAX, "got %u\n", r);
 259
 260     r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test`", NULL);
 261     ok(hinst, r == ERROR_INVALID_PARAMETER, "got %u\n", r);
 262
 263     r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test`", &view);
 264     ok(hinst, !r, "got %u\n", r);
 265
 266     r = MsiViewGetColumnInfo(view, MSICOLINFO_NAMES, &rec2);
 267     ok(hinst, !r, "got %u\n", r);
 268
 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);
 274
 275     r = MsiCloseHandle(rec2);
 276     ok(hinst, !r, "got %u\n", r);
 277
 278     r = MsiViewExecute(view, 0);
 279     ok(hinst, !r, "got %u\n", r);
 280
 281     r = MsiViewFetch(view, &rec2);
 282     ok(hinst, !r, "got %u\n", r);
 283
 284     r = MsiRecordGetFieldCount(rec2);
 285     ok(hinst, r == 3, "got %u\n", r);
 286
 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);
 292
 293     r = MsiRecordGetInteger(rec2, 2);
 294     ok(hinst, r == 1, "got %d\n", r);
 295
 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");
 300
 301     r = MsiCloseHandle(rec2);
 302     ok(hinst, !r, "got %u\n", r);
 303
 304     r = MsiViewFetch(view, &rec2);
 305     ok(hinst, !r, "got %u\n", r);
 306
 307     r = MsiRecordGetFieldCount(rec2);
 308     ok(hinst, r == 3, "got %u\n", r);
 309
 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);
 315
 316     r = MsiRecordGetInteger(rec2, 2);
 317     ok(hinst, r == 2, "got %d\n", r);
 318
 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");
 323
 324     r = MsiCloseHandle(rec2);
 325     ok(hinst, !r, "got %u\n", r);
 326
 327     r = MsiViewFetch(view, &rec2);
 328     ok(hinst, r == ERROR_NO_MORE_ITEMS, "got %u\n", r);
 329     ok(hinst, !rec2, "got %u\n", rec2);
 330
 331     r = MsiViewClose(view);
 332     ok(hinst, !r, "got %u\n", r);
 333
 334     r = MsiCloseHandle(view);
 335     ok(hinst, !r, "got %u\n", r);
 336
 337     r = MsiDatabaseOpenViewA(hdb, "SELECT * FROM `Test` WHERE `Name` = ?", &view);
 338     ok(hinst, !r, "got %u\n", r);
 339
 340     rec = MsiCreateRecord(1);
 341     MsiRecordSetStringA(rec, 1, "one");
 342
 343     r = MsiViewExecute(view, rec);
 344     ok(hinst, !r, "got %u\n", r);
 345
 346     r = MsiViewFetch(view, &rec2);
 347     ok(hinst, !r, "got %u\n", r);
 348
 349     r = MsiRecordGetInteger(rec2, 2);
 350     ok(hinst, r == 1, "got %d\n", r);
 351
 352     r = MsiCloseHandle(rec2);
 353     ok(hinst, !r, "got %u\n", r);
 354
 355     r = MsiViewFetch(view, &rec2);
 356     ok(hinst, r == ERROR_NO_MORE_ITEMS, "got %u\n", r);
 357     ok(hinst, !rec2, "got %u\n", rec2);
 358
 359     r = MsiCloseHandle(rec);
 360     ok(hinst, !r, "got %u\n", r);
 361
 362     r = MsiCloseHandle(view);
 363     ok(hinst, !r, "got %u\n", r);
 364
 365     r = MsiCloseHandle(hdb);
 366     ok(hinst, !r, "got %u\n", r);
 367 }
 368
 369 /* Main test. Anything that doesn't depend on a specific install configuration
 370  * or have undesired side effects should go here. */
 371 UINT WINAPI main_test(MSIHANDLE hinst)
 372 {
 373     UINT res;
 374     IUnknown *unk = NULL;
 375     HRESULT hr;
 376
 377     /* Test for an MTA apartment */
 378     hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER, &IID_IUnknown, (void **)&unk);
 379     todo_wine_ok(hinst, hr == S_OK, "CoCreateInstance failed with %08x\n", hr);
 380
 381     if (unk) IUnknown_Release(unk);
 382
 383     /* but ours is uninitialized */
 384     hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
 385     ok(hinst, hr == S_OK, "got %#x\n", hr);
 386     CoUninitialize();
 387
 388     /* Test MsiGetDatabaseState() */
 389     res = MsiGetDatabaseState(hinst);
 390     todo_wine_ok(hinst, res == MSIDBSTATE_ERROR, "expected MSIDBSTATE_ERROR, got %u\n", res);
 391
 392     test_props(hinst);
 393     test_db(hinst);
 394
 395     return ERROR_SUCCESS;
 396 }
 397
 398 UINT WINAPI test_retval(MSIHANDLE hinst)
 399 {
 400     char prop[10];
 401     DWORD len = sizeof(prop);
 402     UINT retval;
 403
 404     MsiGetPropertyA(hinst, "TEST_RETVAL", prop, &len);
 405     sscanf(prop, "%u", &retval);
 406     return retval;
 407 }
 408
 409 static void append_file(MSIHANDLE hinst, const char *filename, const char *text)
 410 {
 411     DWORD size;
 412     HANDLE file = CreateFileA(filename, GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
 413     ok(hinst, file != INVALID_HANDLE_VALUE, "CreateFile failed, error %u\n", GetLastError());
 414
 415     SetFilePointer(file, 0, NULL, FILE_END);
 416     WriteFile(file, text, strlen(text), &size, NULL);
 417     CloseHandle(file);
 418 }
 419
 420 UINT WINAPI da_immediate(MSIHANDLE hinst)
 421 {
 422     char prop[300];
 423     DWORD len = sizeof(prop);
 424
 425     MsiGetPropertyA(hinst, "TESTPATH", prop, &len);
 426
 427     append_file(hinst, prop, "one");
 428
 429     ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_SCHEDULED), "shouldn't be scheduled\n");
 430     ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_ROLLBACK), "shouldn't be rollback\n");
 431     ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_COMMIT), "shouldn't be commit\n");
 432
 433     return ERROR_SUCCESS;
 434 }
 435
 436 UINT WINAPI da_deferred(MSIHANDLE hinst)
 437 {
 438     char prop[300];
 439     DWORD len = sizeof(prop);
 440     LANGID lang;
 441     UINT r;
 442
 443     /* Test that we were in fact deferred */
 444     r = MsiGetPropertyA(hinst, "CustomActionData", prop, &len);
 445     ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
 446     ok(hinst, prop[0], "CustomActionData was empty\n");
 447
 448     append_file(hinst, prop, "two");
 449
 450     /* Test available properties */
 451     len = sizeof(prop);
 452     r = MsiGetPropertyA(hinst, "ProductCode", prop, &len);
 453     ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
 454     ok(hinst, prop[0], "got %s\n", prop);
 455
 456     len = sizeof(prop);
 457     r = MsiGetPropertyA(hinst, "UserSID", prop, &len);
 458     ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
 459     ok(hinst, prop[0], "got %s\n", prop);
 460
 461     len = sizeof(prop);
 462     r = MsiGetPropertyA(hinst, "TESTPATH", prop, &len);
 463     ok(hinst, r == ERROR_SUCCESS, "got %u\n", r);
 464     todo_wine_ok(hinst, !prop[0], "got %s\n", prop);
 465
 466     /* Test modes */
 467     ok(hinst, MsiGetMode(hinst, MSIRUNMODE_SCHEDULED), "should be scheduled\n");
 468     ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_ROLLBACK), "shouldn't be rollback\n");
 469     ok(hinst, !MsiGetMode(hinst, MSIRUNMODE_COMMIT), "shouldn't be commit\n");
 470
 471     lang = MsiGetLanguage(hinst);
 472     ok(hinst, lang != ERROR_INVALID_HANDLE, "MsiGetLanguage failed\n");
 473
 474     return ERROR_SUCCESS;
 475 }