dlls/msi/tests/suminfo.c

   1 /*
   2  * Copyright (C) 2005 Mike McCormack for CodeWeavers
   3  *
   4  * A test program for MSI database files.
   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 #define COBJMACROS
  22
  23 #include <stdio.h>
  24 #include <windows.h>
  25 #include <msi.h>
  26 #include <msiquery.h>
  27 #include <objidl.h>
  28
  29 #include "wine/test.h"
  30
  31 /*
  32  * The following are defined in Windows SDK's msidefs.h
  33  * but that file doesn't exist in the msvc6 header set.
  34  *
  35  * Some are already defined in PropIdl.h - undefine them
  36  */
  37 #undef PID_DICTIONARY
  38 #undef PID_CODEPAGE
  39 #undef PID_SUBJECT
  40 #undef PID_SECURITY
  41
  42 #define PID_DICTIONARY 0
  43 #define PID_CODEPAGE 1
  44 #define PID_TITLE 2
  45 #define PID_SUBJECT 3
  46 #define PID_AUTHOR 4
  47 #define PID_KEYWORDS 5
  48 #define PID_COMMENTS 6
  49 #define PID_TEMPLATE 7
  50 #define PID_LASTAUTHOR 8
  51 #define PID_REVNUMBER 9
  52 #define PID_EDITTINE 10
  53 #define PID_LASTPRINTED 11
  54 #define PID_CREATE_DTM 12
  55 #define PID_LASTSAVE_DTM 13
  56 #define PID_PAGECOUNT 14
  57 #define PID_WORDCOUNT 15
  58 #define PID_CHARCOUNT 16
  59 #define PID_THUMBNAIL 17
  60 #define PID_APPNAME 18
  61 #define PID_SECURITY 19
  62 #define PID_MSIVERSION PID_PAGECOUNT
  63 #define PID_MSISOURCE PID_WORDCOUNT
  64 #define PID_MSIRESTRICT PID_CHARCOUNT
  65
  66 static void test_suminfo(void)
  67 {
  68     const char *msifile = "winetest.msi";
  69     MSIHANDLE hdb = 0, hsuminfo;
  70     UINT r, count, type;
  71     DWORD sz;
  72     INT val;
  73     FILETIME ft;
  74     char buf[0x10];
  75
  76     DeleteFile(msifile);
  77
  78     /* just MsiOpenDatabase should not create a file */
  79     r = MsiOpenDatabase(msifile, MSIDBOPEN_CREATE, &hdb);
  80     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
  81
  82     r = MsiGetSummaryInformation(hdb, NULL, 0, NULL);
  83     ok(r == ERROR_INVALID_PARAMETER, "MsiGetSummaryInformation wrong error\n");
  84
  85     r = MsiGetSummaryInformation(hdb, NULL, 0, &hsuminfo);
  86     ok(r == ERROR_SUCCESS, "MsiGetSummaryInformation failed\n");
  87
  88     r = MsiSummaryInfoGetPropertyCount(0, NULL);
  89     ok(r == ERROR_INVALID_HANDLE, "getpropcount failed\n");
  90
  91     r = MsiSummaryInfoGetPropertyCount(hsuminfo, NULL);
  92     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
  93
  94     count = -1;
  95     r = MsiSummaryInfoGetPropertyCount(hsuminfo, &count);
  96     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
  97     ok(count == 0, "count should be zero\n");
  98
  99     r = MsiSummaryInfoGetProperty(hsuminfo, 0, NULL, NULL, NULL, 0, NULL);
 100     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
 101
 102     r = MsiSummaryInfoGetProperty(hsuminfo, -1, NULL, NULL, NULL, 0, NULL);
 103     ok(r == ERROR_UNKNOWN_PROPERTY, "MsiSummaryInfoGetProperty wrong error\n");
 104
 105     r = MsiSummaryInfoGetProperty(hsuminfo, PID_SECURITY+1, NULL, NULL, NULL, 0, NULL);
 106     ok(r == ERROR_UNKNOWN_PROPERTY, "MsiSummaryInfoGetProperty wrong error\n");
 107
 108     type = -1;
 109     r = MsiSummaryInfoGetProperty(hsuminfo, 0, &type, NULL, NULL, 0, NULL);
 110     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
 111     ok(type == 0, "wrong type\n");
 112
 113     type = -1;
 114     val = 1234;
 115     r = MsiSummaryInfoGetProperty(hsuminfo, 0, &type, &val, NULL, 0, NULL);
 116     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
 117     ok(type == 0, "wrong type\n");
 118     ok(val == 1234, "wrong val\n");
 119
 120     buf[0]='x';
 121     buf[1]=0;
 122     sz = 0x10;
 123     r = MsiSummaryInfoGetProperty(hsuminfo, PID_REVNUMBER, &type, &val, NULL, buf, &sz);
 124     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
 125     ok(buf[0]=='x', "cleared buffer\n");
 126     ok(sz == 0x10, "count wasn't zero\n");
 127     ok(type == VT_EMPTY, "should be empty\n");
 128
 129     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 0, NULL, "Mike");
 130     ok(r == ERROR_FUNCTION_FAILED, "MsiSummaryInfoSetProperty wrong error\n");
 131
 132     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 1, NULL, "JungAh");
 133     ok(r == ERROR_FUNCTION_FAILED, "MsiSummaryInfoSetProperty wrong error\n");
 134
 135     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 1, &ft, "Mike");
 136     ok(r == ERROR_FUNCTION_FAILED, "MsiSummaryInfoSetProperty wrong error\n");
 137
 138     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_I2, 1, &ft, "JungAh");
 139     ok(r == ERROR_FUNCTION_FAILED, "MsiSummaryInfoSetProperty wrong error\n");
 140
 141     r = MsiCloseHandle(hsuminfo);
 142     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 143
 144     /* try again with the update count set */
 145     r = MsiGetSummaryInformation(hdb, NULL, 1, &hsuminfo);
 146     ok(r == ERROR_SUCCESS, "MsiGetSummaryInformation failed\n");
 147
 148     r = MsiSummaryInfoSetProperty(hsuminfo, 0, VT_LPSTR, 1, NULL, NULL);
 149     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 150
 151     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_LPSTR, 1, NULL, NULL);
 152     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 153
 154     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_I4, 0, NULL, "Mike");
 155     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 156
 157     r = MsiSummaryInfoSetProperty(hsuminfo, PID_AUTHOR, VT_I4, 0, NULL, "JungAh");
 158     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 159
 160     r = MsiSummaryInfoSetProperty(hsuminfo, PID_KEYWORDS, VT_I2, 0, NULL, "Mike");
 161     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 162
 163     r = MsiSummaryInfoSetProperty(hsuminfo, PID_COMMENTS, VT_FILETIME, 0, NULL, "JungAh");
 164     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 165
 166     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TEMPLATE, VT_I2, 0, NULL, "Mike");
 167     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 168
 169     r = MsiSummaryInfoSetProperty(hsuminfo, PID_LASTAUTHOR, VT_LPSTR, 0, NULL, NULL);
 170     ok(r == ERROR_INVALID_PARAMETER, "MsiSummaryInfoSetProperty wrong error\n");
 171
 172     r = MsiSummaryInfoSetProperty(hsuminfo, PID_LASTSAVE_DTM, VT_FILETIME, 0, NULL, NULL);
 173     ok(r == ERROR_INVALID_PARAMETER, "MsiSummaryInfoSetProperty wrong error\n");
 174
 175     r = MsiSummaryInfoSetProperty(hsuminfo, PID_LASTAUTHOR, VT_LPWSTR, 0, NULL, "h\0i\0\0");
 176     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 177
 178     r = MsiSummaryInfoSetProperty(hsuminfo, PID_REVNUMBER, VT_I4, 0, NULL, "Jungah");
 179     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 180
 181     r = MsiSummaryInfoSetProperty(hsuminfo, PID_PAGECOUNT, VT_LPSTR, 1, NULL, NULL);
 182     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 183
 184     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 0, NULL, "Mike");
 185     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty failed\n");
 186
 187     sz = 2;
 188     strcpy(buf,"x");
 189     r = MsiSummaryInfoGetProperty(hsuminfo, PID_TITLE, &type, NULL, NULL, buf, &sz );
 190     ok(r == ERROR_MORE_DATA, "MsiSummaryInfoSetProperty failed\n");
 191     ok(sz == 4, "count was wrong\n");
 192     ok(type == VT_LPSTR, "type was wrong\n");
 193     ok(!strcmp(buf,"M"), "buffer was wrong\n");
 194
 195     sz = 4;
 196     strcpy(buf,"x");
 197     r = MsiSummaryInfoGetProperty(hsuminfo, PID_TITLE, &type, NULL, NULL, buf, &sz );
 198     ok(r == ERROR_MORE_DATA, "MsiSummaryInfoSetProperty failed\n");
 199     ok(sz == 4, "count was wrong\n");
 200     ok(type == VT_LPSTR, "type was wrong\n");
 201     ok(!strcmp(buf,"Mik"), "buffer was wrong\n");
 202
 203     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 0, NULL, "JungAh");
 204     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty failed\n");
 205
 206     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_I2, 1, &ft, "Mike");
 207     ok(r == ERROR_FUNCTION_FAILED, "MsiSummaryInfoSetProperty wrong error\n");
 208
 209     r = MsiCloseHandle(hsuminfo);
 210     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 211
 212     /* try again with a higher update count */
 213     r = MsiGetSummaryInformation(hdb, NULL, 10, &hsuminfo);
 214     ok(r == ERROR_SUCCESS, "MsiGetSummaryInformation failed\n");
 215
 216     r = MsiSummaryInfoSetProperty(hsuminfo, PID_TITLE, VT_LPSTR, 0, NULL, "JungAh");
 217     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty failed\n");
 218
 219     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_LPSTR, 1, NULL, NULL);
 220     ok(r == ERROR_DATATYPE_MISMATCH, "MsiSummaryInfoSetProperty wrong error\n");
 221
 222     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_I2, 1, NULL, NULL);
 223     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty wrong error\n");
 224
 225     r = MsiSummaryInfoSetProperty(hsuminfo, PID_CODEPAGE, VT_I2, 1, &ft, "Mike");
 226     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty wrong error\n");
 227
 228     r = MsiSummaryInfoSetProperty(hsuminfo, PID_AUTHOR, VT_LPSTR, 1, &ft, "Mike");
 229     ok(r == ERROR_SUCCESS, "MsiSummaryInfoSetProperty wrong error\n");
 230
 231     r = MsiSummaryInfoPersist(hsuminfo);
 232     ok(r == ERROR_SUCCESS, "MsiSummaryInfoPersist failed\n");
 233
 234     MsiDatabaseCommit(hdb);
 235
 236     r = MsiCloseHandle(hsuminfo);
 237     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 238
 239     r = MsiCloseHandle(hdb);
 240     ok(r == ERROR_SUCCESS, "MsiCloseHandle failed\n");
 241
 242     r = DeleteFile(msifile);
 243     ok(r, "DeleteFile failed\n");
 244 }
 245
 246 static const WCHAR tb[] = { 0x4840, 0x3f7f, 0x4164, 0x422f, 0x4836, 0 }; /* _Tables */
 247 static const WCHAR sd[] = { 0x4840, 0x3f3f, 0x4577, 0x446c, 0x3b6a, 0x45e4, 0x4824, 0 }; /* _StringData */
 248 static const WCHAR sp[] = { 0x4840, 0x3f3f, 0x4577, 0x446c, 0x3e6a, 0x44b2, 0x482f, 0 }; /* _StringPool */
 249
 250 #define LOSE_CONST(x) ((LPSTR)(UINT_PTR)(x))
 251
 252 static void test_create_database_binary(void)
 253 {
 254     static const CLSID CLSID_MsiDatabase =
 255         { 0xc1084, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46 } };
 256     static const CLSID IID_IPropertySetStorage =
 257         { 0x13a, 0, 0, {0xc0, 0, 0, 0, 0, 0, 0, 0x46 } };
 258     static const CLSID FMTID_SummaryInformation =
 259         { 0xf29f85e0, 0x4ff9, 0x1068, {0xab, 0x91, 0x08, 0x00, 0x2b, 0x27, 0xb3, 0xd9}};
 260     DWORD mode = STGM_CREATE | STGM_READWRITE | STGM_DIRECT | STGM_SHARE_EXCLUSIVE;
 261     static const WCHAR msifile[] = {
 262         'w','i','n','e','t','e','s','t','.','m','s','i',0 };
 263     IPropertySetStorage *pss = NULL;
 264     IPropertyStorage *ps = NULL;
 265     IStorage *stg = NULL;
 266     IStream *stm = NULL;
 267     HRESULT r;
 268     PROPSPEC propspec[10];
 269     PROPVARIANT propvar[10];
 270     USHORT data[2] = { 0, 0 };
 271
 272     r = StgCreateDocfile( msifile, mode, 0, &stg );
 273     ok( r == S_OK, "failed to create database\n");
 274
 275     r = IStorage_SetClass( stg, &CLSID_MsiDatabase );
 276     ok( r == S_OK, "failed to set clsid\n");
 277
 278     /* create the _StringData stream */
 279     r = IStorage_CreateStream( stg, sd, STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm );
 280     ok( r == S_OK, "failed to create stream\n");
 281
 282     IStream_Release( stm );
 283
 284     /* create the _StringPool stream */
 285     r = IStorage_CreateStream( stg, sp, STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm );
 286     ok( r == S_OK, "failed to create stream\n");
 287
 288     r = IStream_Write( stm, data, sizeof data, NULL );
 289     ok( r == S_OK, "failed to write stream\n");
 290
 291     IStream_Release( stm );
 292
 293     /* create the _Tables stream */
 294     r = IStorage_CreateStream( stg, tb, STGM_WRITE | STGM_SHARE_EXCLUSIVE, 0, 0, &stm );
 295     ok( r == S_OK, "failed to create stream\n");
 296
 297     IStream_Release( stm );
 298
 299     r = IStorage_QueryInterface( stg, &IID_IPropertySetStorage, (void**) &pss );
 300     ok( r == S_OK, "failed to set clsid\n");
 301
 302     r = IPropertySetStorage_Create( pss, &FMTID_SummaryInformation, NULL, 0, mode, &ps );
 303     ok( r == S_OK, "failed to create property set\n");
 304
 305     r = IPropertyStorage_SetClass( ps, &FMTID_SummaryInformation );
 306     ok( r == S_OK, "failed to set class\n");
 307
 308     propspec[0].ulKind = PRSPEC_PROPID;
 309     U(propspec[0]).propid = PID_TITLE;
 310     propvar[0].vt = VT_LPSTR;
 311     U(propvar[0]).pszVal = LOSE_CONST("test title");
 312
 313     propspec[1].ulKind = PRSPEC_PROPID;
 314     U(propspec[1]).propid = PID_SUBJECT;
 315     propvar[1].vt = VT_LPSTR;
 316     U(propvar[1]).pszVal = LOSE_CONST("msi suminfo / property storage test");
 317
 318     propspec[2].ulKind = PRSPEC_PROPID;
 319     U(propspec[2]).propid = PID_AUTHOR;
 320     propvar[2].vt = VT_LPSTR;
 321     U(propvar[2]).pszVal = LOSE_CONST("mike_m");
 322
 323     propspec[3].ulKind = PRSPEC_PROPID;
 324     U(propspec[3]).propid = PID_TEMPLATE;
 325     propvar[3].vt = VT_LPSTR;
 326     U(propvar[3]).pszVal = LOSE_CONST(";1033");  /* actually the string table's codepage */
 327
 328     propspec[4].ulKind = PRSPEC_PROPID;
 329     U(propspec[4]).propid = PID_REVNUMBER;
 330     propvar[4].vt = VT_LPSTR;
 331     U(propvar[4]).pszVal = LOSE_CONST("{913B8D18-FBB6-4CAC-A239-C74C11E3FA74}");
 332
 333     propspec[5].ulKind = PRSPEC_PROPID;
 334     U(propspec[5]).propid = PID_PAGECOUNT;
 335     propvar[5].vt = VT_I4;
 336     U(propvar[5]).lVal = 100;
 337
 338     propspec[6].ulKind = PRSPEC_PROPID;
 339     U(propspec[6]).propid = PID_WORDCOUNT;
 340     propvar[6].vt = VT_I4;
 341     U(propvar[6]).lVal = 0;
 342
 343     /* MSDN says that PID_LASTPRINTED should be a VT_FILETIME... */
 344     propspec[7].ulKind = PRSPEC_PROPID;
 345     U(propspec[7]).propid = PID_LASTPRINTED;
 346     propvar[7].vt = VT_LPSTR;
 347     U(propvar[7]).pszVal = LOSE_CONST("7/1/1999 5:17");
 348
 349     r = IPropertyStorage_WriteMultiple( ps, 8, propspec, propvar, PID_FIRST_USABLE );
 350     ok( r == S_OK, "failed to write properties\n");
 351
 352     IPropertyStorage_Commit( ps, STGC_DEFAULT );
 353
 354     IPropertyStorage_Release( ps );
 355     IPropertySetStorage_Release( pss );
 356
 357     IStorage_Commit( stg, STGC_DEFAULT );
 358     IStorage_Release( stg );
 359 }
 360
 361 static void test_summary_binary(void)
 362 {
 363     const char *msifile = "winetest.msi";
 364     MSIHANDLE hdb = 0, hsuminfo = 0;
 365     UINT r, type, count;
 366     INT ival;
 367     DWORD sz;
 368     char sval[20];
 369
 370     DeleteFile( msifile );
 371
 372     test_create_database_binary();
 373
 374     ok( INVALID_FILE_ATTRIBUTES != GetFileAttributes(msifile), "file doesn't exist!\n");
 375
 376     /* just MsiOpenDatabase should not create a file */
 377     r = MsiOpenDatabase(msifile, MSIDBOPEN_READONLY, &hdb);
 378     ok(r == ERROR_SUCCESS, "MsiOpenDatabase failed\n");
 379
 380     r = MsiGetSummaryInformation(hdb, NULL, 0, &hsuminfo);
 381     ok(r == ERROR_SUCCESS, "MsiGetSummaryInformation failed\n");
 382
 383     /*
 384      * Check what reading PID_LASTPRINTED does...
 385      * The string value is written to the msi file
 386      * but it appears that we're not allowed to read it back again.
 387      * We can still read its type though...?
 388      */
 389     sz = sizeof sval;
 390     sval[0] = 0;
 391     type = 0;
 392     r = MsiSummaryInfoGetProperty(hsuminfo, PID_LASTPRINTED, &type, NULL, NULL, sval, &sz);
 393     ok(r == ERROR_SUCCESS, "MsiSummaryInfoGetProperty failed\n");
 394     ok(!lstrcmpA(sval, "") || !lstrcmpA(sval, "7"),
 395         "Expected empty string or \"7\", got \"%s\"\n", sval);
 396     todo_wine {
 397     ok(type == VT_LPSTR, "Expected VT_LPSTR, got %d\n", type);
 398     ok(sz == 0 || sz == 1, "Expected 0 or 1, got %d\n", sz);
 399     }
 400
 401     ival = -1;
 402     r = MsiSummaryInfoGetProperty(hsuminfo, PID_WORDCOUNT, &type, &ival, NULL, NULL, NULL);
 403     ok(r == ERROR_SUCCESS, "MsiSummaryInfoGetProperty failed\n");
 404     todo_wine ok( ival == 0, "value incorrect\n");
 405
 406     /* looks like msi adds some of its own values in here */
 407     count = 0;
 408     r = MsiSummaryInfoGetPropertyCount( hsuminfo, &count );
 409     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
 410     todo_wine ok(count == 10, "prop count incorrect\n");
 411
 412     MsiCloseHandle( hsuminfo );
 413     MsiCloseHandle( hdb );
 414
 415     DeleteFile( msifile );
 416 }
 417
 418 START_TEST(suminfo)
 419 {
 420     test_suminfo();
 421     test_summary_binary();
 422 }