janitorial: Remove remaining NULL checks before free() (found by Smatch).
[wine] / dlls / msi / tests / install.c
1 /*
2  * Copyright (C) 2006 James Hawkins
3  *
4  * A test program for installing MSI products.
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 <stdio.h>
22
23 #include <windows.h>
24 #include <msiquery.h>
25 #include <msidefs.h>
26 #include <msi.h>
27 #include <fci.h>
28
29 #include "wine/test.h"
30
31 static const char *msifile = "winetest.msi";
32 CHAR CURR_DIR[MAX_PATH];
33 CHAR PROG_FILES_DIR[MAX_PATH];
34
35 /* msi database data */
36
37 static const CHAR admin_exec_seq_dat[] = "Action\tCondition\tSequence\n"
38                                          "s72\tS255\tI2\n"
39                                          "AdminExecuteSequence\tAction\n"
40                                          "CostFinalize\t\t1000\n"
41                                          "CostInitialize\t\t800\n"
42                                          "FileCost\t\t900\n"
43                                          "InstallAdminPackage\t\t3900\n"
44                                          "InstallFiles\t\t4000\n"
45                                          "InstallFinalize\t\t6600\n"
46                                          "InstallInitialize\t\t1500\n"
47                                          "InstallValidate\t\t1400";
48
49 static const CHAR advt_exec_seq_dat[] = "Action\tCondition\tSequence\n"
50                                         "s72\tS255\tI2\n"
51                                         "AdvtExecuteSequence\tAction\n"
52                                         "CostFinalize\t\t1000\n"
53                                         "CostInitialize\t\t800\n"
54                                         "CreateShortcuts\t\t4500\n"
55                                         "InstallFinalize\t\t6600\n"
56                                         "InstallInitialize\t\t1500\n"
57                                         "InstallValidate\t\t1400\n"
58                                         "PublishComponents\t\t6200\n"
59                                         "PublishFeatures\t\t6300\n"
60                                         "PublishProduct\t\t6400\n"
61                                         "RegisterClassInfo\t\t4600\n"
62                                         "RegisterExtensionInfo\t\t4700\n"
63                                         "RegisterMIMEInfo\t\t4900\n"
64                                         "RegisterProgIdInfo\t\t4800";
65
66 static const CHAR component_dat[] = "Component\tComponentId\tDirectory_\tAttributes\tCondition\tKeyPath\n"
67                                     "s72\tS38\ts72\ti2\tS255\tS72\n"
68                                     "Component\tComponent\n"
69                                     "Five\t{8CC92E9D-14B2-4CA4-B2AA-B11D02078087}\tNEWDIR\t2\t\tfive.txt\n"
70                                     "Four\t{FD37B4EA-7209-45C0-8917-535F35A2F080}\tCABOUTDIR\t2\t\tfour.txt\n"
71                                     "One\t{783B242E-E185-4A56-AF86-C09815EC053C}\tMSITESTDIR\t2\t\tone.txt\n"
72                                     "Three\t{010B6ADD-B27D-4EDD-9B3D-34C4F7D61684}\tCHANGEDDIR\t2\t\tthree.txt\n"
73                                     "Two\t{BF03D1A6-20DA-4A65-82F3-6CAC995915CE}\tFIRSTDIR\t2\t\ttwo.txt\n"
74                                     "dangler\t{6091DF25-EF96-45F1-B8E9-A9B1420C7A3C}\tTARGETDIR\t4\t\tregdata";
75
76 static const CHAR directory_dat[] = "Directory\tDirectory_Parent\tDefaultDir\n"
77                                     "s72\tS72\tl255\n"
78                                     "Directory\tDirectory\n"
79                                     "CABOUTDIR\tMSITESTDIR\tcabout\n"
80                                     "CHANGEDDIR\tMSITESTDIR\tchanged:second\n"
81                                     "FIRSTDIR\tMSITESTDIR\tfirst\n"
82                                     "MSITESTDIR\tProgramFilesFolder\tmsitest\n"
83                                     "NEWDIR\tCABOUTDIR\tnew\n"
84                                     "ProgramFilesFolder\tTARGETDIR\t.\n"
85                                     "TARGETDIR\t\tSourceDir";
86
87 static const CHAR feature_dat[] = "Feature\tFeature_Parent\tTitle\tDescription\tDisplay\tLevel\tDirectory_\tAttributes\n"
88                                   "s38\tS38\tL64\tL255\tI2\ti2\tS72\ti2\n"
89                                   "Feature\tFeature\n"
90                                   "Five\t\tFive\tThe Five Feature\t5\t3\tNEWDIR\t0\n"
91                                   "Four\t\tFour\tThe Four Feature\t4\t3\tCABOUTDIR\t0\n"
92                                   "One\t\tOne\tThe One Feature\t1\t3\tMSITESTDIR\t0\n"
93                                   "Three\t\tThree\tThe Three Feature\t3\t3\tCHANGEDDIR\t0\n"
94                                   "Two\t\tTwo\tThe Two Feature\t2\t3\tFIRSTDIR\t0";
95
96 static const CHAR feature_comp_dat[] = "Feature_\tComponent_\n"
97                                        "s38\ts72\n"
98                                        "FeatureComponents\tFeature_\tComponent_\n"
99                                        "Five\tFive\n"
100                                        "Four\tFour\n"
101                                        "One\tOne\n"
102                                        "Three\tThree\n"
103                                        "Two\tTwo";
104
105 static const CHAR file_dat[] = "File\tComponent_\tFileName\tFileSize\tVersion\tLanguage\tAttributes\tSequence\n"
106                                "s72\ts72\tl255\ti4\tS72\tS20\tI2\ti2\n"
107                                "File\tFile\n"
108                                "five.txt\tFive\tfive.txt\t1000\t\t\t16384\t5\n"
109                                "four.txt\tFour\tfour.txt\t1000\t\t\t16384\t4\n"
110                                "one.txt\tOne\tone.txt\t1000\t\t\t0\t1\n"
111                                "three.txt\tThree\tthree.txt\t1000\t\t\t0\t3\n"
112                                "two.txt\tTwo\ttwo.txt\t1000\t\t\t0\t2";
113
114 static const CHAR install_exec_seq_dat[] = "Action\tCondition\tSequence\n"
115                                            "s72\tS255\tI2\n"
116                                            "InstallExecuteSequence\tAction\n"
117                                            "AllocateRegistrySpace\tNOT Installed\t1550\n"
118                                            "CostFinalize\t\t1000\n"
119                                            "CostInitialize\t\t800\n"
120                                            "FileCost\t\t900\n"
121                                            "InstallFiles\t\t4000\n"
122                                            "InstallFinalize\t\t6600\n"
123                                            "InstallInitialize\t\t1500\n"
124                                            "InstallValidate\t\t1400\n"
125                                            "LaunchConditions\t\t100\n"
126                                            "WriteRegistryValues\t\t5000";
127
128 static const CHAR media_dat[] = "DiskId\tLastSequence\tDiskPrompt\tCabinet\tVolumeLabel\tSource\n"
129                                 "i2\ti4\tL64\tS255\tS32\tS72\n"
130                                 "Media\tDiskId\n"
131                                 "1\t3\t\t\tDISK1\t\n"
132                                 "2\t5\t\tmsitest.cab\tDISK2\t\n";
133
134 static const CHAR property_dat[] = "Property\tValue\n"
135                                    "s72\tl0\n"
136                                    "Property\tProperty\n"
137                                    "DefaultUIFont\tDlgFont8\n"
138                                    "INSTALLLEVEL\t3\n"
139                                    "InstallMode\tTypical\n"
140                                    "Manufacturer\tWine\n"
141                                    "PIDTemplate\t12345<###-%%%%%%%>@@@@@\n"
142                                    "ProductCode\t{F1C3AF50-8B56-4A69-A00C-00773FE42F30}\n"
143                                    "ProductID\tnone\n"
144                                    "ProductLanguage\t1033\n"
145                                    "ProductName\tMSITEST\n"
146                                    "ProductVersion\t1.1.1\n"
147                                    "PROMPTROLLBACKCOST\tP\n"
148                                    "Setup\tSetup\n"
149                                    "UpgradeCode\t{CE067E8D-2E1A-4367-B734-4EB2BDAD6565}";
150
151 static const CHAR registry_dat[] = "Registry\tRoot\tKey\tName\tValue\tComponent_\n"
152                                    "s72\ti2\tl255\tL255\tL0\ts72\n"
153                                    "Registry\tRegistry\n"
154                                    "Apples\t2\tSOFTWARE\\Wine\\msitest\tName\timaname\tOne\n"
155                                    "Oranges\t2\tSOFTWARE\\Wine\\msitest\tnumber\t#314\tTwo\n"
156                                    "regdata\t2\tSOFTWARE\\Wine\\msitest\tblah\tbad\tdangler";
157
158 typedef struct _msi_table
159 {
160     const CHAR *filename;
161     const CHAR *data;
162     int size;
163 } msi_table;
164
165 #define ADD_TABLE(x) {#x".idt", x##_dat, sizeof(x##_dat)}
166
167 static const msi_table tables[] =
168 {
169     ADD_TABLE(admin_exec_seq),
170     ADD_TABLE(advt_exec_seq),
171     ADD_TABLE(component),
172     ADD_TABLE(directory),
173     ADD_TABLE(feature),
174     ADD_TABLE(feature_comp),
175     ADD_TABLE(file),
176     ADD_TABLE(install_exec_seq),
177     ADD_TABLE(media),
178     ADD_TABLE(property),
179     ADD_TABLE(registry)
180 };
181
182 /* cabinet definitions */
183
184 /* make the max size large so there is only one cab file */
185 #define MEDIA_SIZE          999999999
186 #define FOLDER_THRESHOLD    900000
187
188 /* The following defintions were copied from dlls/cabinet/cabinet.h
189  * because they are undocumented in windows.
190  */
191
192 /* EXTRACTdest flags */
193 #define EXTRACT_FILLFILELIST  0x00000001
194 #define EXTRACT_EXTRACTFILES  0x00000002
195
196 struct ExtractFileList {
197     LPSTR  filename;
198     struct ExtractFileList *next;
199     BOOL   unknown;  /* always 1L */
200 };
201
202 /* the first parameter of the function extract */
203 typedef struct {
204     long   result1;          /* 0x000 */
205     long   unknown1[3];      /* 0x004 */
206     struct ExtractFileList *filelist; /* 0x010 */
207     long   filecount;        /* 0x014 */
208     long   flags;            /* 0x018 */
209     char   directory[0x104]; /* 0x01c */
210     char   lastfile[0x20c];  /* 0x120 */
211 } EXTRACTDEST;
212
213 /* cabinet function pointers */
214 HMODULE hCabinet;
215 static HRESULT (WINAPI *pExtract)(EXTRACTDEST*, LPCSTR);
216
217 /* the FCI callbacks */
218
219 static void *mem_alloc(ULONG cb)
220 {
221     return HeapAlloc(GetProcessHeap(), 0, cb);
222 }
223
224 static void mem_free(void *memory)
225 {
226     HeapFree(GetProcessHeap(), 0, memory);
227 }
228
229 static BOOL get_next_cabinet(PCCAB pccab, ULONG  cbPrevCab, void *pv)
230 {
231     return TRUE;
232 }
233
234 static long progress(UINT typeStatus, ULONG cb1, ULONG cb2, void *pv)
235 {
236     return 0;
237 }
238
239 static int file_placed(PCCAB pccab, char *pszFile, long cbFile,
240                        BOOL fContinuation, void *pv)
241 {
242     return 0;
243 }
244
245 static INT_PTR fci_open(char *pszFile, int oflag, int pmode, int *err, void *pv)
246 {
247     HANDLE handle;
248     DWORD dwAccess = 0;
249     DWORD dwShareMode = 0;
250     DWORD dwCreateDisposition = OPEN_EXISTING;
251     
252     dwAccess = GENERIC_READ | GENERIC_WRITE;
253     /* FILE_SHARE_DELETE is not supported by Windows Me/98/95 */
254     dwShareMode = FILE_SHARE_READ | FILE_SHARE_WRITE;
255
256     if (GetFileAttributesA(pszFile) != INVALID_FILE_ATTRIBUTES)
257         dwCreateDisposition = OPEN_EXISTING;
258     else
259         dwCreateDisposition = CREATE_NEW;
260
261     handle = CreateFileA(pszFile, dwAccess, dwShareMode, NULL,
262                          dwCreateDisposition, 0, NULL);
263
264     ok(handle != INVALID_HANDLE_VALUE, "Failed to CreateFile %s\n", pszFile);
265
266     return (INT_PTR)handle;
267 }
268
269 static UINT fci_read(INT_PTR hf, void *memory, UINT cb, int *err, void *pv)
270 {
271     HANDLE handle = (HANDLE)hf;
272     DWORD dwRead;
273     BOOL res;
274     
275     res = ReadFile(handle, memory, cb, &dwRead, NULL);
276     ok(res, "Failed to ReadFile\n");
277
278     return dwRead;
279 }
280
281 static UINT fci_write(INT_PTR hf, void *memory, UINT cb, int *err, void *pv)
282 {
283     HANDLE handle = (HANDLE)hf;
284     DWORD dwWritten;
285     BOOL res;
286
287     res = WriteFile(handle, memory, cb, &dwWritten, NULL);
288     ok(res, "Failed to WriteFile\n");
289
290     return dwWritten;
291 }
292
293 static int fci_close(INT_PTR hf, int *err, void *pv)
294 {
295     HANDLE handle = (HANDLE)hf;
296     ok(CloseHandle(handle), "Failed to CloseHandle\n");
297
298     return 0;
299 }
300
301 static long fci_seek(INT_PTR hf, long dist, int seektype, int *err, void *pv)
302 {
303     HANDLE handle = (HANDLE)hf;
304     DWORD ret;
305     
306     ret = SetFilePointer(handle, dist, NULL, seektype);
307     ok(ret != INVALID_SET_FILE_POINTER, "Failed to SetFilePointer\n");
308
309     return ret;
310 }
311
312 static int fci_delete(char *pszFile, int *err, void *pv)
313 {
314     BOOL ret = DeleteFileA(pszFile);
315     ok(ret, "Failed to DeleteFile %s\n", pszFile);
316
317     return 0;
318 }
319
320 static BOOL check_record(MSIHANDLE rec, UINT field, LPCSTR val)
321 {
322     CHAR buffer[0x20];
323     UINT r;
324     DWORD sz;
325
326     sz = sizeof buffer;
327     r = MsiRecordGetString(rec, field, buffer, &sz);
328     return (r == ERROR_SUCCESS ) && !strcmp(val, buffer);
329 }
330
331 static BOOL get_temp_file(char *pszTempName, int cbTempName, void *pv)
332 {
333     LPSTR tempname;
334
335     tempname = HeapAlloc(GetProcessHeap(), 0, MAX_PATH);
336     GetTempFileNameA(".", "xx", 0, tempname);
337
338     if (tempname && (strlen(tempname) < (unsigned)cbTempName))
339     {
340         lstrcpyA(pszTempName, tempname);
341         HeapFree(GetProcessHeap(), 0, tempname);
342         return TRUE;
343     }
344
345     HeapFree(GetProcessHeap(), 0, tempname);
346
347     return FALSE;
348 }
349
350 static INT_PTR get_open_info(char *pszName, USHORT *pdate, USHORT *ptime,
351                              USHORT *pattribs, int *err, void *pv)
352 {
353     BY_HANDLE_FILE_INFORMATION finfo;
354     FILETIME filetime;
355     HANDLE handle;
356     DWORD attrs;
357     BOOL res;
358
359     handle = CreateFile(pszName, GENERIC_READ, FILE_SHARE_READ, NULL,
360                         OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
361
362     ok(handle != INVALID_HANDLE_VALUE, "Failed to CreateFile %s\n", pszName);
363
364     res = GetFileInformationByHandle(handle, &finfo);
365     ok(res, "Expected GetFileInformationByHandle to succeed\n");
366    
367     FileTimeToLocalFileTime(&finfo.ftLastWriteTime, &filetime);
368     FileTimeToDosDateTime(&filetime, pdate, ptime);
369
370     attrs = GetFileAttributes(pszName);
371     ok(attrs != INVALID_FILE_ATTRIBUTES, "Failed to GetFileAttributes\n");
372
373     return (INT_PTR)handle;
374 }
375
376 static void add_file(HFCI hfci, char *file)
377 {
378     char path[MAX_PATH];
379     BOOL res;
380
381     lstrcpyA(path, CURR_DIR);
382     lstrcatA(path, "\\");
383     lstrcatA(path, file);
384
385     res = FCIAddFile(hfci, path, file, FALSE, get_next_cabinet, progress,
386                      get_open_info, tcompTYPE_MSZIP);
387     ok(res, "Expected FCIAddFile to succeed\n");
388 }
389
390 static void set_cab_parameters(PCCAB pCabParams, const CHAR *name)
391 {
392     ZeroMemory(pCabParams, sizeof(CCAB));
393
394     pCabParams->cb = MEDIA_SIZE;
395     pCabParams->cbFolderThresh = FOLDER_THRESHOLD;
396     pCabParams->setID = 0xbeef;
397     lstrcpyA(pCabParams->szCabPath, CURR_DIR);
398     lstrcatA(pCabParams->szCabPath, "\\");
399     lstrcpyA(pCabParams->szCab, name);
400 }
401
402 static void create_cab_file(const CHAR *name)
403 {
404     CCAB cabParams;
405     HFCI hfci;
406     ERF erf;
407     static CHAR four_txt[] = "four.txt",
408                 five_txt[] = "five.txt";
409     BOOL res;
410
411     set_cab_parameters(&cabParams, name);
412
413     hfci = FCICreate(&erf, file_placed, mem_alloc, mem_free, fci_open,
414                       fci_read, fci_write, fci_close, fci_seek, fci_delete,
415                       get_temp_file, &cabParams, NULL);
416
417     ok(hfci != NULL, "Failed to create an FCI context\n");
418
419     add_file(hfci, four_txt);
420     add_file(hfci, five_txt);
421
422     res = FCIFlushCabinet(hfci, FALSE, get_next_cabinet, progress);
423     ok(res, "Failed to flush the cabinet\n");
424
425     res = FCIDestroy(hfci);
426     ok(res, "Failed to destroy the cabinet\n");
427 }
428
429 static BOOL init_function_pointers(void)
430 {
431     hCabinet = LoadLibraryA("cabinet.dll");
432     if (!hCabinet)
433         return FALSE;
434
435     pExtract = (void *)GetProcAddress(hCabinet, "Extract");
436     if (!pExtract)
437         return FALSE;
438
439     return TRUE;
440 }
441
442 static BOOL get_program_files_dir(LPSTR buf)
443 {
444     HKEY hkey;
445     CHAR temp[MAX_PATH];
446     DWORD type = REG_EXPAND_SZ, size;
447
448     if (RegOpenKey(HKEY_LOCAL_MACHINE,
449                    "Software\\Microsoft\\Windows\\CurrentVersion", &hkey))
450         return FALSE;
451
452     size = MAX_PATH;
453     if (RegQueryValueEx(hkey, "ProgramFilesPath", 0, &type, (LPBYTE)temp, &size))
454         return FALSE;
455
456     ExpandEnvironmentStrings(temp, buf, MAX_PATH);
457
458     RegCloseKey(hkey);
459     return TRUE;
460 }
461
462 static void create_file(const CHAR *name)
463 {
464     HANDLE file;
465     DWORD written;
466
467     file = CreateFileA(name, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
468     ok(file != INVALID_HANDLE_VALUE, "Failure to open file %s\n", name);
469     WriteFile(file, name, strlen(name), &written, NULL);
470     WriteFile(file, "\n", strlen("\n"), &written, NULL);
471     CloseHandle(file);
472 }
473
474 static void create_test_files(void)
475 {
476     int len;
477
478     GetCurrentDirectoryA(MAX_PATH, CURR_DIR);
479     len = lstrlenA(CURR_DIR);
480
481     if(len && (CURR_DIR[len-1] == '\\'))
482         CURR_DIR[len - 1] = 0;
483
484     get_program_files_dir(PROG_FILES_DIR);
485
486     CreateDirectoryA("msitest", NULL);
487     create_file("msitest\\one.txt");
488     CreateDirectoryA("msitest\\first", NULL);
489     create_file("msitest\\first\\two.txt");
490     CreateDirectoryA("msitest\\second", NULL);
491     create_file("msitest\\second\\three.txt");
492
493     create_file("four.txt");
494     create_file("five.txt");
495     create_cab_file("msitest.cab");
496
497     DeleteFileA("four.txt");
498     DeleteFileA("five.txt");
499 }
500
501 static BOOL delete_pf(const CHAR *rel_path, BOOL is_file)
502 {
503     CHAR path[MAX_PATH];
504
505     lstrcpyA(path, PROG_FILES_DIR);
506     lstrcatA(path, "\\");
507     lstrcatA(path, rel_path);
508
509     if (is_file)
510         return DeleteFileA(path);
511     else
512         return RemoveDirectoryA(path);
513 }
514
515 static void delete_test_files(void)
516 {
517     DeleteFileA("msitest.msi");
518     DeleteFileA("msitest.cab");
519     DeleteFileA("msitest\\second\\three.txt");
520     DeleteFileA("msitest\\first\\two.txt");
521     DeleteFileA("msitest\\one.txt");
522     RemoveDirectoryA("msitest\\second");
523     RemoveDirectoryA("msitest\\first");
524     RemoveDirectoryA("msitest");
525 }
526
527 static void write_file(const CHAR *filename, const char *data, int data_size)
528 {
529     DWORD size;
530
531     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
532                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
533
534     WriteFile(hf, data, data_size, &size, NULL);
535     CloseHandle(hf);
536 }
537
538 static void write_msi_summary_info(MSIHANDLE db)
539 {
540     MSIHANDLE summary;
541     UINT r;
542
543     r = MsiGetSummaryInformationA(db, NULL, 4, &summary);
544     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
545
546     r = MsiSummaryInfoSetPropertyA(summary, PID_TEMPLATE, VT_LPSTR, 0, NULL, ";1033");
547     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
548
549     r = MsiSummaryInfoSetPropertyA(summary, PID_REVNUMBER, VT_LPSTR, 0, NULL,
550                                    "{004757CA-5092-49c2-AD20-28E1CE0DF5F2}");
551     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
552
553     r = MsiSummaryInfoSetPropertyA(summary, PID_PAGECOUNT, VT_I4, 100, NULL, NULL);
554     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
555
556     r = MsiSummaryInfoSetPropertyA(summary, PID_WORDCOUNT, VT_I4, 0, NULL, NULL);
557     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
558
559     /* write the summary changes back to the stream */
560     r = MsiSummaryInfoPersist(summary);
561     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
562
563     MsiCloseHandle(summary);
564 }
565
566 static void create_database(const CHAR *name, const msi_table *tables, int num_tables)
567 {
568     MSIHANDLE db;
569     UINT r;
570     int j;
571
572     r = MsiOpenDatabaseA(name, MSIDBOPEN_CREATE, &db);
573     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
574
575     /* import the tables into the database */
576     for (j = 0; j < num_tables; j++)
577     {
578         const msi_table *table = &tables[j];
579
580         write_file(table->filename, table->data, (table->size - 1) * sizeof(char));
581
582         r = MsiDatabaseImportA(db, CURR_DIR, table->filename);
583         ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
584
585         DeleteFileA(table->filename);
586     }
587
588     write_msi_summary_info(db);
589
590     r = MsiDatabaseCommit(db);
591     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
592
593     MsiCloseHandle(db);
594 }
595
596 static void test_MsiInstallProduct(void)
597 {
598     UINT r;
599     CHAR path[MAX_PATH];
600     LONG res;
601     HKEY hkey;
602     DWORD num, size, type;
603
604     r = MsiInstallProductA(msifile, NULL);
605     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
606
607     todo_wine
608     {
609         ok(delete_pf("msitest\\cabout\\new\\five.txt", TRUE), "File not installed\n");
610         ok(delete_pf("msitest\\cabout\\new", FALSE), "File not installed\n");
611         ok(delete_pf("msitest\\cabout\\four.txt", TRUE), "File not installed\n");
612         ok(delete_pf("msitest\\cabout", FALSE), "File not installed\n");
613         ok(delete_pf("msitest\\changed\\three.txt", TRUE), "File not installed\n");
614         ok(delete_pf("msitest\\changed", FALSE), "File not installed\n");
615         ok(delete_pf("msitest\\first\\two.txt", TRUE), "File not installed\n");
616         ok(delete_pf("msitest\\first", FALSE), "File not installed\n");
617         ok(delete_pf("msitest\\one.txt", TRUE), "File not installed\n");
618         ok(delete_pf("msitest", FALSE), "File not installed\n");
619     }
620
621     res = RegOpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\\Wine\\msitest", &hkey);
622     ok(res == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %ld\n", res);
623
624     size = MAX_PATH;
625     type = REG_SZ;
626     res = RegQueryValueExA(hkey, "Name", NULL, &type, (LPBYTE)path, &size);
627     ok(res == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %ld\n", res);
628     ok(!lstrcmpA(path, "imaname"), "Expected imaname, got %s\n", path);
629
630     size = MAX_PATH;
631     type = REG_SZ;
632     res = RegQueryValueExA(hkey, "blah", NULL, &type, (LPBYTE)path, &size);
633     todo_wine
634     {
635         ok(res == ERROR_FILE_NOT_FOUND, "Expected ERROR_FILE_NOT_FOUND, got %ld\n", res);
636     }
637
638     size = sizeof(num);
639     type = REG_DWORD;
640     res = RegQueryValueExA(hkey, "number", NULL, &type, (LPBYTE)&num, &size);
641     ok(res == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %ld\n", res);
642     ok(num == 314, "Expected 314, got %ld\n", num);
643
644     RegDeleteKeyA(HKEY_LOCAL_MACHINE, "SOFTWARE\\Wine\\msitest");
645 }
646
647 static void test_MsiSetComponentState(void)
648 {
649     MSIHANDLE package;
650     char path[MAX_PATH];
651     UINT r;
652
653     CoInitialize(NULL);
654
655     lstrcpy(path, CURR_DIR);
656     lstrcat(path, "\\");
657     lstrcat(path, msifile);
658
659     r = MsiOpenPackage(path, &package);
660     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
661
662     r = MsiDoAction(package, "CostInitialize");
663     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
664
665     r = MsiDoAction(package, "FileCost");
666     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
667
668     r = MsiDoAction(package, "CostFinalize");
669     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
670
671     r = MsiSetComponentState(package, "dangler", INSTALLSTATE_SOURCE);
672     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
673
674     MsiCloseHandle(package);
675     CoUninitialize();
676 }
677
678 static void test_packagecoltypes(void)
679 {
680     MSIHANDLE hdb, view, rec;
681     char path[MAX_PATH];
682     LPCSTR query;
683     UINT r, count;
684
685     CoInitialize(NULL);
686
687     lstrcpy(path, CURR_DIR);
688     lstrcat(path, "\\");
689     lstrcat(path, msifile);
690
691     r = MsiOpenDatabase(path, MSIDBOPEN_READONLY, &hdb);
692     ok(r == ERROR_SUCCESS, "Expected ERROR_SUCCESS, got %u\n", r);
693
694     query = "SELECT * FROM `Media`";
695     r = MsiDatabaseOpenView( hdb, query, &view );
696     ok(r == ERROR_SUCCESS, "MsiDatabaseOpenView failed\n");
697
698     r = MsiViewGetColumnInfo( view, MSICOLINFO_NAMES, &rec );
699     count = MsiRecordGetFieldCount( rec );
700     ok(r == ERROR_SUCCESS, "MsiViewGetColumnInfo failed\n");
701     ok(count == 6, "Expected 6, got %d\n", count);
702     ok(check_record(rec, 1, "DiskId"), "wrong column label\n");
703     ok(check_record(rec, 2, "LastSequence"), "wrong column label\n");
704     ok(check_record(rec, 3, "DiskPrompt"), "wrong column label\n");
705     ok(check_record(rec, 4, "Cabinet"), "wrong column label\n");
706     ok(check_record(rec, 5, "VolumeLabel"), "wrong column label\n");
707     ok(check_record(rec, 6, "Source"), "wrong column label\n");
708     MsiCloseHandle(rec);
709
710     r = MsiViewGetColumnInfo( view, MSICOLINFO_TYPES, &rec );
711     count = MsiRecordGetFieldCount( rec );
712     ok(r == ERROR_SUCCESS, "MsiViewGetColumnInfo failed\n");
713     ok(count == 6, "Expected 6, got %d\n", count);
714     ok(check_record(rec, 4, "S255"), "wrong column label\n");
715     ok(check_record(rec, 5, "S32"), "wrong column label\n");
716     ok(check_record(rec, 6, "S72"), "wrong column label\n");
717     todo_wine
718     {
719         ok(check_record(rec, 1, "i2"), "wrong column label\n");
720         ok(check_record(rec, 2, "i4"), "wrong column label\n");
721         ok(check_record(rec, 3, "L64"), "wrong column label\n");
722     }
723
724     MsiCloseHandle(rec);
725     MsiCloseHandle(view);
726     MsiCloseHandle(hdb);
727     DeleteFile(msifile);
728 }
729
730 START_TEST(install)
731 {
732     if (!init_function_pointers())
733         return;
734
735     create_test_files();
736     create_database(msifile, tables, sizeof(tables) / sizeof(msi_table));
737
738     test_MsiInstallProduct();
739     test_MsiSetComponentState();
740     test_packagecoltypes();
741
742     delete_test_files();
743 }