msi: Include the null terminator when writing data to the stream.
[wine] / 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( !strcmp(sval, ""), "value incorrect\n");
395     todo_wine {
396     ok( type == VT_LPSTR, "type wrong\n");
397     ok( sz == 0, "length wrong\n");
398     }
399
400     ival = -1;
401     r = MsiSummaryInfoGetProperty(hsuminfo, PID_WORDCOUNT, &type, &ival, NULL, NULL, NULL);
402     ok(r == ERROR_SUCCESS, "MsiSummaryInfoGetProperty failed\n");
403     todo_wine ok( ival == 0, "value incorrect\n");
404
405     /* looks like msi adds some of its own values in here */
406     count = 0;
407     r = MsiSummaryInfoGetPropertyCount( hsuminfo, &count );
408     ok(r == ERROR_SUCCESS, "getpropcount failed\n");
409     todo_wine ok(count == 10, "prop count incorrect\n");
410
411     MsiCloseHandle( hsuminfo );
412     MsiCloseHandle( hdb );
413
414     DeleteFile( msifile );
415 }
416
417 START_TEST(suminfo)
418 {
419     test_suminfo();
420     test_summary_binary();
421 }