?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
comctl32: Close previously selected treeview item at common ancestor.
[wine.git] / dlls / msi / patch.c
blob545635987e0465262b3d8ff41b6157d41f71f076
1 /*
2 * Implementation of the Microsoft Installer (msi.dll)
3 *
4 * Copyright 2004,2005 Aric Stewart for CodeWeavers
5 * Copyright 2011 Hans Leidekker for CodeWeavers
6 *
7 * This library is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * This library is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with this library; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 */
21
22 #include <stdarg.h>
23 #define COBJMACROS
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winreg.h"
27 #include "objbase.h"
28 #include "shlwapi.h"
29 #include "wine/debug.h"
30 #include "wine/unicode.h"
31 #include "msipriv.h"
32
33 WINE_DEFAULT_DEBUG_CHANNEL(msi);
34
35 static BOOL match_language( MSIPACKAGE *package, LANGID langid )
36 {
37 UINT i;
38
39 if (!package->num_langids || !langid) return TRUE;
40 for (i = 0; i < package->num_langids; i++)
41 {
42 if (package->langids[i] == langid) return TRUE;
43 }
44 return FALSE;
45 }
46
47 struct transform_desc
48 {
49 WCHAR *product_code_from;
50 WCHAR *product_code_to;
51 WCHAR *version_from;
52 WCHAR *version_to;
53 WCHAR *upgrade_code;
54 };
55
56 static void free_transform_desc( struct transform_desc *desc )
57 {
58 msi_free( desc->product_code_from );
59 msi_free( desc->product_code_to );
60 msi_free( desc->version_from );
61 msi_free( desc->version_to );
62 msi_free( desc->upgrade_code );
63 msi_free( desc );
64 }
65
66 static struct transform_desc *parse_transform_desc( const WCHAR *str )
67 {
68 struct transform_desc *ret;
69 const WCHAR *p = str, *q;
70 UINT len;
71
72 if (!(ret = msi_alloc_zero( sizeof(*ret) ))) return NULL;
73
74 q = strchrW( p, '}' );
75 if (*p != '{' || !q) goto error;
76
77 len = q - p + 1;
78 if (!(ret->product_code_from = msi_alloc( (len + 1) * sizeof(WCHAR) ))) goto error;
79 memcpy( ret->product_code_from, p, len * sizeof(WCHAR) );
80 ret->product_code_from[len] = 0;
81
82 p = q + 1;
83 if (!(q = strchrW( p, ';' ))) goto error;
84 len = q - p;
85 if (!(ret->version_from = msi_alloc( (len + 1) * sizeof(WCHAR) ))) goto error;
86 memcpy( ret->version_from, p, len * sizeof(WCHAR) );
87 ret->version_from[len] = 0;
88
89 p = q + 1;
90 q = strchrW( p, '}' );
91 if (*p != '{' || !q) goto error;
92
93 len = q - p + 1;
94 if (!(ret->product_code_to = msi_alloc( (len + 1) * sizeof(WCHAR) ))) goto error;
95 memcpy( ret->product_code_to, p, len * sizeof(WCHAR) );
96 ret->product_code_to[len] = 0;
97
98 p = q + 1;
99 if (!(q = strchrW( p, ';' ))) goto error;
100 len = q - p;
101 if (!(ret->version_to = msi_alloc( (len + 1) * sizeof(WCHAR) ))) goto error;
102 memcpy( ret->version_to, p, len * sizeof(WCHAR) );
103 ret->version_to[len] = 0;
104
105 p = q + 1;
106 q = strchrW( p, '}' );
107 if (*p != '{' || !q) goto error;
108
109 len = q - p + 1;
110 if (!(ret->upgrade_code = msi_alloc( (len + 1) * sizeof(WCHAR) ))) goto error;
111 memcpy( ret->upgrade_code, p, len * sizeof(WCHAR) );
112 ret->upgrade_code[len] = 0;
113
114 return ret;
115
116 error:
117 free_transform_desc( ret );
118 return NULL;
119 }
120
121 static UINT check_transform_applicable( MSIPACKAGE *package, IStorage *transform )
122 {
123 static const UINT supported_flags =
124 MSITRANSFORM_VALIDATE_PRODUCT | MSITRANSFORM_VALIDATE_LANGUAGE |
125 MSITRANSFORM_VALIDATE_PLATFORM | MSITRANSFORM_VALIDATE_MAJORVERSION |
126 MSITRANSFORM_VALIDATE_MINORVERSION | MSITRANSFORM_VALIDATE_UPGRADECODE;
127 MSISUMMARYINFO *si;
128 UINT r, valid_flags = 0, wanted_flags = 0;
129 WCHAR *template, *product, *p;
130 struct transform_desc *desc;
131
132 r = msi_get_suminfo( transform, 0, &si );
133 if (r != ERROR_SUCCESS)
134 {
135 WARN("no summary information!\n");
136 return r;
137 }
138 wanted_flags = msi_suminfo_get_int32( si, PID_CHARCOUNT );
139 wanted_flags &= 0xffff; /* mask off error condition flags */
140 TRACE("validation flags 0x%04x\n", wanted_flags);
141
142 if (wanted_flags & ~supported_flags)
143 {
144 FIXME("unsupported validation flags 0x%04x\n", wanted_flags);
145 msiobj_release( &si->hdr );
146 return ERROR_FUNCTION_FAILED;
147 }
148 if (!(template = msi_suminfo_dup_string( si, PID_TEMPLATE )))
149 {
150 WARN("no template property!\n");
151 msiobj_release( &si->hdr );
152 return ERROR_FUNCTION_FAILED;
153 }
154 TRACE("template property: %s\n", debugstr_w(template));
155 if (!(product = msi_get_suminfo_product( transform )))
156 {
157 WARN("no product property!\n");
158 msi_free( template );
159 msiobj_release( &si->hdr );
160 return ERROR_FUNCTION_FAILED;
161 }
162 TRACE("product property: %s\n", debugstr_w(product));
163 if (!(desc = parse_transform_desc( product )))
164 {
165 msi_free( template );
166 msiobj_release( &si->hdr );
167 return ERROR_FUNCTION_FAILED;
168 }
169 msi_free( product );
170
171 if (wanted_flags & MSITRANSFORM_VALIDATE_LANGUAGE)
172 {
173 if (!template[0] || ((p = strchrW( template, ';' )) && match_language( package, atoiW( p + 1 ) )))
174 {
175 valid_flags |= MSITRANSFORM_VALIDATE_LANGUAGE;
176 }
177 }
178 if (wanted_flags & MSITRANSFORM_VALIDATE_PRODUCT)
179 {
180 WCHAR *product_code_installed = msi_dup_property( package->db, szProductCode );
181
182 if (!product_code_installed)
183 {
184 msi_free( template );
185 free_transform_desc( desc );
186 msiobj_release( &si->hdr );
187 return ERROR_INSTALL_PACKAGE_INVALID;
188 }
189 if (!strcmpW( desc->product_code_from, product_code_installed ))
190 {
191 valid_flags |= MSITRANSFORM_VALIDATE_PRODUCT;
192 }
193 msi_free( product_code_installed );
194 }
195 if (wanted_flags & MSITRANSFORM_VALIDATE_PLATFORM)
196 {
197 if ((p = strchrW( template, ';' )))
198 {
199 *p = 0;
200 if (package->platform == parse_platform( template ))
201 valid_flags |= MSITRANSFORM_VALIDATE_PLATFORM;
202 }
203 }
204 msi_free( template );
205 if (wanted_flags & MSITRANSFORM_VALIDATE_MAJORVERSION)
206 {
207 WCHAR *product_version_installed = msi_dup_property( package->db, szProductVersion );
208 DWORD major_installed, minor_installed, major, minor;
209
210 if (!product_version_installed)
211 {
212 free_transform_desc( desc );
213 msiobj_release( &si->hdr );
214 return ERROR_INSTALL_PACKAGE_INVALID;
215 }
216 msi_parse_version_string( product_version_installed, &major_installed, &minor_installed );
217 msi_parse_version_string( desc->version_from, &major, &minor );
218
219 if (major_installed == major)
220 {
221 valid_flags |= MSITRANSFORM_VALIDATE_MAJORVERSION;
222 wanted_flags &= ~MSITRANSFORM_VALIDATE_MINORVERSION;
223 }
224 msi_free( product_version_installed );
225 }
226 else if (wanted_flags & MSITRANSFORM_VALIDATE_MINORVERSION)
227 {
228 WCHAR *product_version_installed = msi_dup_property( package->db, szProductVersion );
229 DWORD major_installed, minor_installed, major, minor;
230
231 if (!product_version_installed)
232 {
233 free_transform_desc( desc );
234 msiobj_release( &si->hdr );
235 return ERROR_INSTALL_PACKAGE_INVALID;
236 }
237 msi_parse_version_string( product_version_installed, &major_installed, &minor_installed );
238 msi_parse_version_string( desc->version_from, &major, &minor );
239
240 if (major_installed == major && minor_installed == minor)
241 valid_flags |= MSITRANSFORM_VALIDATE_MINORVERSION;
242 msi_free( product_version_installed );
243 }
244 if (wanted_flags & MSITRANSFORM_VALIDATE_UPGRADECODE)
245 {
246 WCHAR *upgrade_code_installed = msi_dup_property( package->db, szUpgradeCode );
247
248 if (!upgrade_code_installed)
249 {
250 free_transform_desc( desc );
251 msiobj_release( &si->hdr );
252 return ERROR_INSTALL_PACKAGE_INVALID;
253 }
254 if (!strcmpW( desc->upgrade_code, upgrade_code_installed ))
255 valid_flags |= MSITRANSFORM_VALIDATE_UPGRADECODE;
256 msi_free( upgrade_code_installed );
257 }
258
259 free_transform_desc( desc );
260 msiobj_release( &si->hdr );
261 if ((valid_flags & wanted_flags) != wanted_flags) return ERROR_FUNCTION_FAILED;
262 TRACE("applicable transform\n");
263 return ERROR_SUCCESS;
264 }
265
266 static UINT apply_substorage_transform( MSIPACKAGE *package, MSIDATABASE *patch_db, LPCWSTR name )
267 {
268 UINT ret = ERROR_FUNCTION_FAILED;
269 IStorage *stg = NULL;
270 HRESULT r;
271
272 TRACE("%p %s\n", package, debugstr_w(name));
273
274 if (*name++ != ':')
275 {
276 ERR("expected a colon in %s\n", debugstr_w(name));
277 return ERROR_FUNCTION_FAILED;
278 }
279 r = IStorage_OpenStorage( patch_db->storage, name, NULL, STGM_SHARE_EXCLUSIVE, NULL, 0, &stg );
280 if (SUCCEEDED(r))
281 {
282 ret = check_transform_applicable( package, stg );
283 if (ret == ERROR_SUCCESS)
284 msi_table_apply_transform( package->db, stg );
285 else
286 TRACE("substorage transform %s wasn't applicable\n", debugstr_w(name));
287 IStorage_Release( stg );
288 }
289 else
290 {
291 ERR("failed to open substorage %s\n", debugstr_w(name));
292 }
293 return ERROR_SUCCESS;
294 }
295
296 UINT msi_check_patch_applicable( MSIPACKAGE *package, MSISUMMARYINFO *si )
297 {
298 LPWSTR guid_list, *guids, product_code;
299 UINT i, ret = ERROR_FUNCTION_FAILED;
300
301 product_code = msi_dup_property( package->db, szProductCode );
302 if (!product_code)
303 {
304 /* FIXME: the property ProductCode should be written into the DB somewhere */
305 ERR("no product code to check\n");
306 return ERROR_SUCCESS;
307 }
308 guid_list = msi_suminfo_dup_string( si, PID_TEMPLATE );
309 guids = msi_split_string( guid_list, ';' );
310 for (i = 0; guids[i] && ret != ERROR_SUCCESS; i++)
311 {
312 if (!strcmpW( guids[i], product_code )) ret = ERROR_SUCCESS;
313 }
314 msi_free( guids );
315 msi_free( guid_list );
316 msi_free( product_code );
317 return ret;
318 }
319
320 static UINT msi_parse_patch_summary( MSISUMMARYINFO *si, MSIPATCHINFO **patch )
321 {
322 MSIPATCHINFO *pi;
323 UINT r = ERROR_SUCCESS;
324 WCHAR *p;
325
326 if (!(pi = msi_alloc_zero( sizeof(MSIPATCHINFO) )))
327 {
328 return ERROR_OUTOFMEMORY;
329 }
330 if (!(pi->patchcode = msi_suminfo_dup_string( si, PID_REVNUMBER )))
331 {
332 msi_free( pi );
333 return ERROR_OUTOFMEMORY;
334 }
335 p = pi->patchcode;
336 if (*p != '{')
337 {
338 msi_free( pi->patchcode );
339 msi_free( pi );
340 return ERROR_PATCH_PACKAGE_INVALID;
341 }
342 if (!(p = strchrW( p + 1, '}' )))
343 {
344 msi_free( pi->patchcode );
345 msi_free( pi );
346 return ERROR_PATCH_PACKAGE_INVALID;
347 }
348 if (p[1])
349 {
350 FIXME("patch obsoletes %s\n", debugstr_w(p + 1));
351 p[1] = 0;
352 }
353 TRACE("patch code %s\n", debugstr_w(pi->patchcode));
354 if (!(pi->products = msi_suminfo_dup_string( si, PID_TEMPLATE )))
355 {
356 msi_free( pi->patchcode );
357 msi_free( pi );
358 return ERROR_OUTOFMEMORY;
359 }
360 if (!(pi->transforms = msi_suminfo_dup_string( si, PID_LASTAUTHOR )))
361 {
362 msi_free( pi->patchcode );
363 msi_free( pi->products );
364 msi_free( pi );
365 return ERROR_OUTOFMEMORY;
366 }
367 *patch = pi;
368 return r;
369 }
370
371 static UINT patch_set_media_source_prop( MSIPACKAGE *package )
372 {
373 static const WCHAR query[] = {
374 'S','E','L','E','C','T',' ','`','S','o','u','r','c','e','`',' ','F','R','O','M',' ',
375 '`','M','e','d','i','a','`',' ','W','H','E','R','E',' ','`','S','o','u','r','c','e','`',' ',
376 'I','S',' ','N','O','T',' ','N','U','L','L',0};
377 MSIQUERY *view;
378 MSIRECORD *rec;
379 const WCHAR *property;
380 WCHAR *patch;
381 UINT r;
382
383 r = MSI_DatabaseOpenViewW( package->db, query, &view );
384 if (r != ERROR_SUCCESS)
385 return r;
386
387 r = MSI_ViewExecute( view, 0 );
388 if (r != ERROR_SUCCESS)
389 goto done;
390
391 if (MSI_ViewFetch( view, &rec ) == ERROR_SUCCESS)
392 {
393 property = MSI_RecordGetString( rec, 1 );
394 patch = msi_dup_property( package->db, szPatch );
395 msi_set_property( package->db, property, patch, -1 );
396 msi_free( patch );
397 msiobj_release( &rec->hdr );
398 }
399
400 done:
401 msiobj_release( &view->hdr );
402 return r;
403 }
404
405 struct patch_offset
406 {
407 struct list entry;
408 WCHAR *name;
409 UINT sequence;
410 };
411
412 struct patch_offset_list
413 {
414 struct list files;
415 struct list patches;
416 UINT count, min, max;
417 UINT offset_to_apply;
418 };
419
420 static struct patch_offset_list *patch_offset_list_create( void )
421 {
422 struct patch_offset_list *pos = msi_alloc( sizeof(struct patch_offset_list) );
423 list_init( &pos->files );
424 list_init( &pos->patches );
425 pos->count = pos->max = 0;
426 pos->min = 999999;
427 return pos;
428 }
429
430 static void patch_offset_list_free( struct patch_offset_list *pos )
431 {
432 struct patch_offset *po, *po2;
433
434 LIST_FOR_EACH_ENTRY_SAFE( po, po2, &pos->files, struct patch_offset, entry )
435 {
436 msi_free( po->name );
437 msi_free( po );
438 }
439 LIST_FOR_EACH_ENTRY_SAFE( po, po2, &pos->patches, struct patch_offset, entry )
440 {
441 msi_free( po->name );
442 msi_free( po );
443 }
444 msi_free( pos );
445 }
446
447 static void patch_offset_get_filepatches( MSIDATABASE *db, UINT last_sequence, struct patch_offset_list *pos )
448 {
449 static const WCHAR query[] = {
450 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','P','a','t','c','h',' ',
451 'W','H','E','R','E',' ','S','e','q','u','e','n','c','e',' ','<','=',' ','?',' ',
452 'O','R','D','E','R',' ','B','Y',' ','S','e','q','u','e','n','c','e',0};
453 MSIQUERY *view;
454 MSIRECORD *rec;
455 UINT r;
456
457 r = MSI_DatabaseOpenViewW( db, query, &view );
458 if (r != ERROR_SUCCESS)
459 return;
460
461 rec = MSI_CreateRecord( 1 );
462 MSI_RecordSetInteger( rec, 1, last_sequence );
463
464 r = MSI_ViewExecute( view, rec );
465 msiobj_release( &rec->hdr );
466 if (r != ERROR_SUCCESS)
467 return;
468
469 while (MSI_ViewFetch( view, &rec ) == ERROR_SUCCESS)
470 {
471 struct patch_offset *po = msi_alloc( sizeof(struct patch_offset) );
472
473 po->name = msi_dup_record_field( rec, 1 );
474 po->sequence = MSI_RecordGetInteger( rec, 2 );
475 pos->min = min( pos->min, po->sequence );
476 pos->max = max( pos->max, po->sequence );
477 list_add_tail( &pos->patches, &po->entry );
478 pos->count++;
479
480 msiobj_release( &rec->hdr );
481 }
482 msiobj_release( &view->hdr );
483 }
484
485 static void patch_offset_get_files( MSIDATABASE *db, UINT last_sequence, struct patch_offset_list *pos )
486 {
487 static const WCHAR query[] = {
488 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','F','i','l','e',' ',
489 'W','H','E','R','E',' ','S','e','q','u','e','n','c','e',' ','<','=',' ','?',' ',
490 'O','R','D','E','R',' ','B','Y',' ','S','e','q','u','e','n','c','e',0};
491 MSIQUERY *view;
492 MSIRECORD *rec;
493 UINT r;
494
495 r = MSI_DatabaseOpenViewW( db, query, &view );
496 if (r != ERROR_SUCCESS)
497 return;
498
499 rec = MSI_CreateRecord( 1 );
500 MSI_RecordSetInteger( rec, 1, last_sequence );
501
502 r = MSI_ViewExecute( view, rec );
503 msiobj_release( &rec->hdr );
504 if (r != ERROR_SUCCESS)
505 return;
506
507 while (MSI_ViewFetch( view, &rec ) == ERROR_SUCCESS)
508 {
509 UINT attributes = MSI_RecordGetInteger( rec, 7 );
510 if (attributes & msidbFileAttributesPatchAdded)
511 {
512 struct patch_offset *po = msi_alloc( sizeof(struct patch_offset) );
513
514 po->name = msi_dup_record_field( rec, 1 );
515 po->sequence = MSI_RecordGetInteger( rec, 8 );
516 pos->min = min( pos->min, po->sequence );
517 pos->max = max( pos->max, po->sequence );
518 list_add_tail( &pos->files, &po->entry );
519 pos->count++;
520 }
521 msiobj_release( &rec->hdr );
522 }
523 msiobj_release( &view->hdr );
524 }
525
526 static UINT patch_update_file_sequence( MSIDATABASE *db, const struct patch_offset_list *pos,
527 MSIQUERY *view, MSIRECORD *rec )
528 {
529 struct patch_offset *po;
530 const WCHAR *file = MSI_RecordGetString( rec, 1 );
531 UINT r = ERROR_SUCCESS, seq = MSI_RecordGetInteger( rec, 8 );
532
533 LIST_FOR_EACH_ENTRY( po, &pos->files, struct patch_offset, entry )
534 {
535 if (!strcmpiW( file, po->name ))
536 {
537 MSI_RecordSetInteger( rec, 8, seq + pos->offset_to_apply );
538 r = MSI_ViewModify( view, MSIMODIFY_UPDATE, rec );
539 if (r != ERROR_SUCCESS)
540 ERR("Failed to update offset for file %s (%u)\n", debugstr_w(file), r);
541 break;
542 }
543 }
544 return r;
545 }
546
547 static UINT patch_update_filepatch_sequence( MSIDATABASE *db, const struct patch_offset_list *pos,
548 MSIQUERY *view, MSIRECORD *rec )
549 {
550 static const WCHAR delete_query[] = {
551 'D','E','L','E','T','E',' ','F','R','O','M',' ','`','P','a','t','c','h','`',' ',
552 'W','H','E','R','E',' ','`','F','i','l','e','_','`',' ','=',' ','?',' ',
553 'A','N','D',' ','`','S','e','q','u','e','n','c','e','`',' ','=',' ','?',0};
554 static const WCHAR insert_query[] = {
555 'I','N','S','E','R','T',' ','I','N','T','O',' ','`','P','a','t','c','h','`',' ',
556 '(','`','F','i','l','e','_','`',',','`','S','e','q','u','e','n','c','e','`',',',
557 '`','P','a','t','c','h','S','i','z','e','`',',','`','A','t','t','r','i','b','u','t','e','s','`',',',
558 '`','H','e','a','d','e','r','`',',','`','S','t','r','e','a','m','R','e','f','_','`',')',' ',
559 'V','A','L','U','E','S',' ','(','?',',','?',',','?',',','?',',','?',',','?',')',0};
560 struct patch_offset *po;
561 const WCHAR *file = MSI_RecordGetString( rec, 1 );
562 UINT r = ERROR_SUCCESS, seq = MSI_RecordGetInteger( rec, 2 );
563
564 LIST_FOR_EACH_ENTRY( po, &pos->patches, struct patch_offset, entry )
565 {
566 if (seq == po->sequence && !strcmpiW( file, po->name ))
567 {
568 MSIQUERY *delete_view, *insert_view;
569 MSIRECORD *rec2;
570
571 r = MSI_DatabaseOpenViewW( db, delete_query, &delete_view );
572 if (r != ERROR_SUCCESS) return r;
573
574 rec2 = MSI_CreateRecord( 2 );
575 MSI_RecordSetStringW( rec2, 1, po->name );
576 MSI_RecordSetInteger( rec2, 2, po->sequence );
577 r = MSI_ViewExecute( delete_view, rec2 );
578 msiobj_release( &delete_view->hdr );
579 msiobj_release( &rec2->hdr );
580 if (r != ERROR_SUCCESS) return r;
581
582 r = MSI_DatabaseOpenViewW( db, insert_query, &insert_view );
583 if (r != ERROR_SUCCESS) return r;
584
585 MSI_RecordSetInteger( rec, 2, po->sequence + pos->offset_to_apply );
586
587 r = MSI_ViewExecute( insert_view, rec );
588 msiobj_release( &insert_view->hdr );
589 if (r != ERROR_SUCCESS)
590 ERR("Failed to update offset for filepatch %s (%u)\n", debugstr_w(file), r);
591 break;
592 }
593 }
594 return r;
595 }
596
597 static UINT patch_offset_modify_db( MSIDATABASE *db, struct patch_offset_list *pos )
598 {
599 static const WCHAR file_query[] = {
600 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','`','F','i','l','e','`',' ',
601 'W','H','E','R','E',' ','`','S','e','q','u','e','n','c','e','`',' ','>','=',' ','?',' ',
602 'A','N','D',' ','`','S','e','q','u','e','n','c','e','`',' ','<','=',' ','?',' ',
603 'O','R','D','E','R',' ','B','Y',' ','`','S','e','q','u','e','n','c','e','`',0};
604 static const WCHAR patch_query[] = {
605 'S','E','L','E','C','T',' ','*','F','R','O','M',' ','`','P','a','t','c','h','`',' ',
606 'W','H','E','R','E',' ','`','S','e','q','u','e','n','c','e','`',' ','>','=',' ','?',' ',
607 'A','N','D',' ','`','S','e','q','u','e','n','c','e','`',' ','<','=',' ','?',' ',
608 'O','R','D','E','R',' ','B','Y',' ','`','S','e','q','u','e','n','c','e','`',0};
609 MSIRECORD *rec;
610 MSIQUERY *view;
611 UINT r, min = pos->min, max = pos->max, r_fetch;
612
613 r = MSI_DatabaseOpenViewW( db, file_query, &view );
614 if (r != ERROR_SUCCESS)
615 return ERROR_SUCCESS;
616
617 rec = MSI_CreateRecord( 2 );
618 MSI_RecordSetInteger( rec, 1, min );
619 MSI_RecordSetInteger( rec, 2, max );
620
621 r = MSI_ViewExecute( view, rec );
622 msiobj_release( &rec->hdr );
623 if (r != ERROR_SUCCESS)
624 goto done;
625
626 while ((r_fetch = MSI_ViewFetch( view, &rec )) == ERROR_SUCCESS)
627 {
628 r = patch_update_file_sequence( db, pos, view, rec );
629 msiobj_release( &rec->hdr );
630 if (r != ERROR_SUCCESS) goto done;
631 }
632 msiobj_release( &view->hdr );
633
634 r = MSI_DatabaseOpenViewW( db, patch_query, &view );
635 if (r != ERROR_SUCCESS)
636 return ERROR_SUCCESS;
637
638 rec = MSI_CreateRecord( 2 );
639 MSI_RecordSetInteger( rec, 1, min );
640 MSI_RecordSetInteger( rec, 2, max );
641
642 r = MSI_ViewExecute( view, rec );
643 msiobj_release( &rec->hdr );
644 if (r != ERROR_SUCCESS)
645 goto done;
646
647 while ((r_fetch = MSI_ViewFetch( view, &rec )) == ERROR_SUCCESS)
648 {
649 r = patch_update_filepatch_sequence( db, pos, view, rec );
650 msiobj_release( &rec->hdr );
651 if (r != ERROR_SUCCESS) goto done;
652 }
653
654 done:
655 msiobj_release( &view->hdr );
656 return r;
657 }
658
659 static const WCHAR patch_media_query[] = {
660 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ','`','M','e','d','i','a','`',' ',
661 'W','H','E','R','E',' ','`','S','o','u','r','c','e','`',' ','I','S',' ','N','O','T',' ','N','U','L','L',' ',
662 'A','N','D',' ','`','C','a','b','i','n','e','t','`',' ','I','S',' ','N','O','T',' ','N','U','L','L',' ',
663 'O','R','D','E','R',' ','B','Y',' ','`','D','i','s','k','I','d','`',0};
664
665 struct patch_media
666 {
667 struct list entry;
668 UINT disk_id;
669 UINT last_sequence;
670 WCHAR *prompt;
671 WCHAR *cabinet;
672 WCHAR *volume;
673 WCHAR *source;
674 };
675
676 static UINT patch_add_media( MSIPACKAGE *package, IStorage *storage, MSIPATCHINFO *patch )
677 {
678 static const WCHAR delete_query[] = {
679 'D','E','L','E','T','E',' ','F','R','O','M',' ','`','M','e','d','i','a','`',' ',
680 'W','H','E','R','E',' ','`','D','i','s','k','I','d','`','=','?',0};
681 static const WCHAR insert_query[] = {
682 'I','N','S','E','R','T',' ','I','N','T','O',' ','`','M','e','d','i','a','`',' ',
683 '(','`','D','i','s','k','I','d','`',',','`','L','a','s','t','S','e','q','u','e','n','c','e','`',',',
684 '`','D','i','s','k','P','r','o','m','p','t','`',',','`','C','a','b','i','n','e','t','`',',',
685 '`','V','o','l','u','m','e','L','a','b','e','l','`',',','`','S','o','u','r','c','e','`',')',' ',
686 'V','A','L','U','E','S',' ','(','?',',','?',',','?',',','?',',','?',',','?',')',0};
687 MSIQUERY *view;
688 MSIRECORD *rec;
689 UINT r, disk_id;
690 struct list media_list;
691 struct patch_media *media, *next;
692
693 r = MSI_DatabaseOpenViewW( package->db, patch_media_query, &view );
694 if (r != ERROR_SUCCESS) return r;
695
696 r = MSI_ViewExecute( view, 0 );
697 if (r != ERROR_SUCCESS)
698 {
699 msiobj_release( &view->hdr );
700 TRACE("query failed %u\n", r);
701 return r;
702 }
703 list_init( &media_list );
704 while (MSI_ViewFetch( view, &rec ) == ERROR_SUCCESS)
705 {
706 disk_id = MSI_RecordGetInteger( rec, 1 );
707 TRACE("disk_id %u\n", disk_id);
708 if (disk_id >= MSI_INITIAL_MEDIA_TRANSFORM_DISKID)
709 {
710 msiobj_release( &rec->hdr );
711 continue;
712 }
713 if (!(media = msi_alloc( sizeof( *media )))) goto done;
714 media->disk_id = disk_id;
715 media->last_sequence = MSI_RecordGetInteger( rec, 2 );
716 media->prompt = msi_dup_record_field( rec, 3 );
717 media->cabinet = msi_dup_record_field( rec, 4 );
718 media->volume = msi_dup_record_field( rec, 5 );
719 media->source = msi_dup_record_field( rec, 6 );
720
721 list_add_tail( &media_list, &media->entry );
722 msiobj_release( &rec->hdr );
723 }
724 LIST_FOR_EACH_ENTRY( media, &media_list, struct patch_media, entry )
725 {
726 MSIQUERY *delete_view, *insert_view;
727
728 r = MSI_DatabaseOpenViewW( package->db, delete_query, &delete_view );
729 if (r != ERROR_SUCCESS) goto done;
730
731 rec = MSI_CreateRecord( 1 );
732 MSI_RecordSetInteger( rec, 1, media->disk_id );
733
734 r = MSI_ViewExecute( delete_view, rec );
735 msiobj_release( &delete_view->hdr );
736 msiobj_release( &rec->hdr );
737 if (r != ERROR_SUCCESS) goto done;
738
739 r = MSI_DatabaseOpenViewW( package->db, insert_query, &insert_view );
740 if (r != ERROR_SUCCESS) goto done;
741
742 disk_id = package->db->media_transform_disk_id;
743 TRACE("disk id %u\n", disk_id);
744 TRACE("last sequence %u\n", media->last_sequence);
745 TRACE("prompt %s\n", debugstr_w(media->prompt));
746 TRACE("cabinet %s\n", debugstr_w(media->cabinet));
747 TRACE("volume %s\n", debugstr_w(media->volume));
748 TRACE("source %s\n", debugstr_w(media->source));
749
750 rec = MSI_CreateRecord( 6 );
751 MSI_RecordSetInteger( rec, 1, disk_id );
752 MSI_RecordSetInteger( rec, 2, media->last_sequence );
753 MSI_RecordSetStringW( rec, 3, media->prompt );
754 MSI_RecordSetStringW( rec, 4, media->cabinet );
755 MSI_RecordSetStringW( rec, 5, media->volume );
756 MSI_RecordSetStringW( rec, 6, media->source );
757
758 r = MSI_ViewExecute( insert_view, rec );
759 msiobj_release( &insert_view->hdr );
760 msiobj_release( &rec->hdr );
761 if (r != ERROR_SUCCESS) goto done;
762
763 r = msi_add_cabinet_stream( package, disk_id, storage, media->cabinet );
764 if (r != ERROR_SUCCESS) ERR("failed to add cabinet stream %u\n", r);
765 else
766 {
767 patch->disk_id = disk_id;
768 package->db->media_transform_disk_id++;
769 }
770 }
771
772 done:
773 msiobj_release( &view->hdr );
774 LIST_FOR_EACH_ENTRY_SAFE( media, next, &media_list, struct patch_media, entry )
775 {
776 list_remove( &media->entry );
777 msi_free( media->prompt );
778 msi_free( media->cabinet );
779 msi_free( media->volume );
780 msi_free( media->source );
781 msi_free( media );
782 }
783 return r;
784 }
785
786 static UINT patch_set_offsets( MSIDATABASE *db, MSIPATCHINFO *patch )
787 {
788 MSIQUERY *view;
789 MSIRECORD *rec;
790 UINT r;
791
792 r = MSI_DatabaseOpenViewW( db, patch_media_query, &view );
793 if (r != ERROR_SUCCESS)
794 return r;
795
796 r = MSI_ViewExecute( view, 0 );
797 if (r != ERROR_SUCCESS)
798 goto done;
799
800 while (MSI_ViewFetch( view, &rec ) == ERROR_SUCCESS)
801 {
802 UINT offset, last_sequence = MSI_RecordGetInteger( rec, 2 );
803 struct patch_offset_list *pos;
804
805 /* FIXME: set/check Source field instead? */
806 if (last_sequence >= MSI_INITIAL_MEDIA_TRANSFORM_OFFSET)
807 {
808 msiobj_release( &rec->hdr );
809 continue;
810 }
811 pos = patch_offset_list_create();
812 patch_offset_get_files( db, last_sequence, pos );
813 patch_offset_get_filepatches( db, last_sequence, pos );
814
815 offset = db->media_transform_offset - pos->min;
816 last_sequence = offset + pos->max;
817
818 last_sequence += pos->min;
819 pos->offset_to_apply = offset;
820 if (pos->count)
821 {
822 r = patch_offset_modify_db( db, pos );
823 if (r != ERROR_SUCCESS)
824 ERR("Failed to set offsets, expect breakage (%u)\n", r);
825 }
826 MSI_RecordSetInteger( rec, 2, last_sequence );
827 r = MSI_ViewModify( view, MSIMODIFY_UPDATE, rec );
828 if (r != ERROR_SUCCESS)
829 ERR("Failed to update Media table entry, expect breakage (%u)\n", r);
830
831 db->media_transform_offset = last_sequence + 1;
832
833 patch_offset_list_free( pos );
834 msiobj_release( &rec->hdr );
835 }
836
837 done:
838 msiobj_release( &view->hdr );
839 return r;
840 }
841
842 static UINT msi_apply_patch_db( MSIPACKAGE *package, MSIDATABASE *patch_db, MSIPATCHINFO *patch )
843 {
844 UINT i, r = ERROR_SUCCESS;
845 WCHAR **substorage;
846
847 /* apply substorage transforms */
848 substorage = msi_split_string( patch->transforms, ';' );
849 for (i = 0; substorage && substorage[i] && r == ERROR_SUCCESS; i++)
850 {
851 r = apply_substorage_transform( package, patch_db, substorage[i] );
852 if (r == ERROR_SUCCESS)
853 {
854 r = patch_set_offsets( package->db, patch );
855 if (r == ERROR_SUCCESS)
856 r = patch_add_media( package, patch_db->storage, patch );
857 }
858 }
859 msi_free( substorage );
860 if (r != ERROR_SUCCESS)
861 return r;
862
863 r = patch_set_media_source_prop( package );
864 if (r != ERROR_SUCCESS)
865 return r;
866
867 patch->state = MSIPATCHSTATE_APPLIED;
868 list_add_tail( &package->patches, &patch->entry );
869 return ERROR_SUCCESS;
870 }
871
872 void msi_free_patchinfo( MSIPATCHINFO *patch )
873 {
874 msi_free( patch->patchcode );
875 msi_free( patch->products );
876 msi_free( patch->transforms );
877 msi_free( patch->filename );
878 msi_free( patch->localfile );
879 msi_free( patch );
880 }
881
882 static UINT msi_apply_patch_package( MSIPACKAGE *package, const WCHAR *file )
883 {
884 static const WCHAR dotmsp[] = {'.','m','s','p',0};
885 MSIDATABASE *patch_db = NULL;
886 WCHAR localfile[MAX_PATH];
887 MSISUMMARYINFO *si;
888 MSIPATCHINFO *patch = NULL;
889 UINT r;
890
891 TRACE("%p, %s\n", package, debugstr_w(file));
892
893 r = MSI_OpenDatabaseW( file, MSIDBOPEN_READONLY + MSIDBOPEN_PATCHFILE, &patch_db );
894 if (r != ERROR_SUCCESS)
895 {
896 ERR("failed to open patch collection %s\n", debugstr_w( file ) );
897 return r;
898 }
899 r = msi_get_suminfo( patch_db->storage, 0, &si );
900 if (r != ERROR_SUCCESS)
901 {
902 msiobj_release( &patch_db->hdr );
903 return r;
904 }
905 r = msi_check_patch_applicable( package, si );
906 if (r != ERROR_SUCCESS)
907 {
908 TRACE("patch not applicable\n");
909 r = ERROR_SUCCESS;
910 goto done;
911 }
912 r = msi_parse_patch_summary( si, &patch );
913 if ( r != ERROR_SUCCESS )
914 goto done;
915
916 r = msi_create_empty_local_file( localfile, dotmsp );
917 if ( r != ERROR_SUCCESS )
918 goto done;
919
920 r = ERROR_OUTOFMEMORY;
921 patch->registered = FALSE;
922 if (!(patch->filename = strdupW( file ))) goto done;
923 if (!(patch->localfile = strdupW( localfile ))) goto done;
924
925 r = msi_apply_patch_db( package, patch_db, patch );
926 if (r != ERROR_SUCCESS) WARN("patch failed to apply %u\n", r);
927
928 done:
929 msiobj_release( &si->hdr );
930 msiobj_release( &patch_db->hdr );
931 if (patch && r != ERROR_SUCCESS)
932 {
933 DeleteFileW( patch->localfile );
934 msi_free_patchinfo( patch );
935 }
936 return r;
937 }
938
939 /* get the PATCH property, and apply all the patches it specifies */
940 UINT msi_apply_patches( MSIPACKAGE *package )
941 {
942 LPWSTR patch_list, *patches;
943 UINT i, r = ERROR_SUCCESS;
944
945 patch_list = msi_dup_property( package->db, szPatch );
946
947 TRACE("patches to be applied: %s\n", debugstr_w(patch_list));
948
949 patches = msi_split_string( patch_list, ';' );
950 for (i = 0; patches && patches[i] && r == ERROR_SUCCESS; i++)
951 r = msi_apply_patch_package( package, patches[i] );
952
953 msi_free( patches );
954 msi_free( patch_list );
955 return r;
956 }
957
958 UINT msi_apply_transforms( MSIPACKAGE *package )
959 {
960 static const WCHAR szTransforms[] = {'T','R','A','N','S','F','O','R','M','S',0};
961 LPWSTR xform_list, *xforms;
962 UINT i, r = ERROR_SUCCESS;
963
964 xform_list = msi_dup_property( package->db, szTransforms );
965 xforms = msi_split_string( xform_list, ';' );
966
967 for (i = 0; xforms && xforms[i] && r == ERROR_SUCCESS; i++)
968 {
969 if (xforms[i][0] == ':')
970 r = apply_substorage_transform( package, package->db, xforms[i] );
971 else
972 {
973 WCHAR *transform;
974
975 if (!PathIsRelativeW( xforms[i] )) transform = xforms[i];
976 else
977 {
978 WCHAR *p = strrchrW( package->PackagePath, '\\' );
979 DWORD len = p - package->PackagePath + 1;
980
981 if (!(transform = msi_alloc( (len + strlenW( xforms[i] ) + 1) * sizeof(WCHAR)) ))
982 {
983 msi_free( xforms );
984 msi_free( xform_list );
985 return ERROR_OUTOFMEMORY;
986 }
987 memcpy( transform, package->PackagePath, len * sizeof(WCHAR) );
988 memcpy( transform + len, xforms[i], (strlenW( xforms[i] ) + 1) * sizeof(WCHAR) );
989 }
990 r = MSI_DatabaseApplyTransformW( package->db, transform, 0 );
991 if (transform != xforms[i]) msi_free( transform );
992 }
993 }
994 msi_free( xforms );
995 msi_free( xform_list );
996 return r;
997 }
998
999 UINT msi_apply_registered_patch( MSIPACKAGE *package, LPCWSTR patch_code )
1000 {
1001 UINT r;
1002 DWORD len;
1003 WCHAR patch_file[MAX_PATH];
1004 MSIDATABASE *patch_db;
1005 MSIPATCHINFO *patch_info;
1006 MSISUMMARYINFO *si;
1007
1008 TRACE("%p, %s\n", package, debugstr_w(patch_code));
1009
1010 len = sizeof(patch_file) / sizeof(WCHAR);
1011 r = MsiGetPatchInfoExW( patch_code, package->ProductCode, NULL, package->Context,
1012 INSTALLPROPERTY_LOCALPACKAGEW, patch_file, &len );
1013 if (r != ERROR_SUCCESS)
1014 {
1015 ERR("failed to get patch filename %u\n", r);
1016 return r;
1017 }
1018 r = MSI_OpenDatabaseW( patch_file, MSIDBOPEN_READONLY + MSIDBOPEN_PATCHFILE, &patch_db );
1019 if (r != ERROR_SUCCESS)
1020 {
1021 ERR("failed to open patch database %s\n", debugstr_w( patch_file ));
1022 return r;
1023 }
1024 r = msi_get_suminfo( patch_db->storage, 0, &si );
1025 if (r != ERROR_SUCCESS)
1026 {
1027 msiobj_release( &patch_db->hdr );
1028 return r;
1029 }
1030 r = msi_parse_patch_summary( si, &patch_info );
1031 msiobj_release( &si->hdr );
1032 if (r != ERROR_SUCCESS)
1033 {
1034 ERR("failed to parse patch summary %u\n", r);
1035 msiobj_release( &patch_db->hdr );
1036 return r;
1037 }
1038 patch_info->registered = TRUE;
1039 patch_info->localfile = strdupW( patch_file );
1040 if (!patch_info->localfile)
1041 {
1042 msiobj_release( &patch_db->hdr );
1043 msi_free_patchinfo( patch_info );
1044 return ERROR_OUTOFMEMORY;
1045 }
1046 r = msi_apply_patch_db( package, patch_db, patch_info );
1047 msiobj_release( &patch_db->hdr );
1048 if (r != ERROR_SUCCESS)
1049 {
1050 ERR("failed to apply patch %u\n", r);
1051 msi_free_patchinfo( patch_info );
1052 }
1053 return r;
1054 }
Windows API implementation