30ee36b4bfd6254894d37bc1e909de3169d6f5ab
[reactos.git] / reactos / dll / win32 / msi / classes.c
1 /*
2 * Implementation of the Microsoft Installer (msi.dll)
3 *
4 * Copyright 2005 Aric Stewart for CodeWeavers
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 /* Actions handled in this module:
22 *
23 * RegisterClassInfo
24 * RegisterProgIdInfo
25 * RegisterExtensionInfo
26 * RegisterMIMEInfo
27 * UnregisterClassInfo
28 * UnregisterProgIdInfo
29 * UnregisterExtensionInfo
30 * UnregisterMIMEInfo
31 */
32
33 #include <stdarg.h>
34
35 #include "windef.h"
36 #include "winbase.h"
37 #include "winerror.h"
38 #include "winreg.h"
39 #include "wine/debug.h"
40 #include "msipriv.h"
41 #include "winuser.h"
42 #include "wine/unicode.h"
43
44 WINE_DEFAULT_DEBUG_CHANNEL(msi);
45
46 static MSIAPPID *load_appid( MSIPACKAGE* package, MSIRECORD *row )
47 {
48 LPCWSTR buffer;
49 MSIAPPID *appid;
50
51 /* fill in the data */
52
53 appid = msi_alloc_zero( sizeof(MSIAPPID) );
54 if (!appid)
55 return NULL;
56
57 appid->AppID = msi_dup_record_field( row, 1 );
58 TRACE("loading appid %s\n", debugstr_w( appid->AppID ));
59
60 buffer = MSI_RecordGetString(row,2);
61 deformat_string( package, buffer, &appid->RemoteServerName );
62
63 appid->LocalServer = msi_dup_record_field(row,3);
64 appid->ServiceParameters = msi_dup_record_field(row,4);
65 appid->DllSurrogate = msi_dup_record_field(row,5);
66
67 appid->ActivateAtStorage = !MSI_RecordIsNull(row,6);
68 appid->RunAsInteractiveUser = !MSI_RecordIsNull(row,7);
69
70 list_add_tail( &package->appids, &appid->entry );
71
72 return appid;
73 }
74
75 static MSIAPPID *load_given_appid( MSIPACKAGE *package, LPCWSTR name )
76 {
77 MSIRECORD *row;
78 MSIAPPID *appid;
79 static const WCHAR ExecSeqQuery[] =
80 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
81 '`','A','p','p','I','d','`',' ','W','H','E','R','E',' ',
82 '`','A','p','p','I','d','`',' ','=',' ','\'','%','s','\'',0};
83
84 if (!name)
85 return NULL;
86
87 /* check for appids already loaded */
88 LIST_FOR_EACH_ENTRY( appid, &package->appids, MSIAPPID, entry )
89 {
90 if (lstrcmpiW( appid->AppID, name )==0)
91 {
92 TRACE("found appid %s %p\n", debugstr_w(name), appid);
93 return appid;
94 }
95 }
96
97 row = MSI_QueryGetRecord(package->db, ExecSeqQuery, name);
98 if (!row)
99 return NULL;
100
101 appid = load_appid(package, row);
102 msiobj_release(&row->hdr);
103
104 return appid;
105 }
106
107 static MSIPROGID *load_given_progid(MSIPACKAGE *package, LPCWSTR progid);
108 static MSICLASS *load_given_class( MSIPACKAGE *package, LPCWSTR classid );
109
110 static MSIPROGID *load_progid( MSIPACKAGE* package, MSIRECORD *row )
111 {
112 MSIPROGID *progid;
113 LPCWSTR buffer;
114
115 /* fill in the data */
116
117 progid = msi_alloc_zero( sizeof(MSIPROGID) );
118 if (!progid)
119 return NULL;
120
121 list_add_tail( &package->progids, &progid->entry );
122
123 progid->ProgID = msi_dup_record_field(row,1);
124 TRACE("loading progid %s\n",debugstr_w(progid->ProgID));
125
126 buffer = MSI_RecordGetString(row,2);
127 progid->Parent = load_given_progid(package,buffer);
128 if (progid->Parent == NULL && buffer)
129 FIXME("Unknown parent ProgID %s\n",debugstr_w(buffer));
130
131 buffer = MSI_RecordGetString(row,3);
132 progid->Class = load_given_class(package,buffer);
133 if (progid->Class == NULL && buffer)
134 FIXME("Unknown class %s\n",debugstr_w(buffer));
135
136 progid->Description = msi_dup_record_field(row,4);
137
138 if (!MSI_RecordIsNull(row,6))
139 {
140 INT icon_index = MSI_RecordGetInteger(row,6);
141 LPCWSTR FileName = MSI_RecordGetString(row,5);
142 LPWSTR FilePath;
143 static const WCHAR fmt[] = {'%','s',',','%','i',0};
144
145 FilePath = build_icon_path(package,FileName);
146
147 progid->IconPath = msi_alloc( (strlenW(FilePath)+10)* sizeof(WCHAR) );
148
149 sprintfW(progid->IconPath,fmt,FilePath,icon_index);
150
151 msi_free(FilePath);
152 }
153 else
154 {
155 buffer = MSI_RecordGetString(row,5);
156 if (buffer)
157 progid->IconPath = build_icon_path(package,buffer);
158 }
159
160 progid->CurVer = NULL;
161 progid->VersionInd = NULL;
162
163 /* if we have a parent then we may be that parents CurVer */
164 if (progid->Parent && progid->Parent != progid)
165 {
166 MSIPROGID *parent = progid->Parent;
167
168 while (parent->Parent && parent->Parent != parent)
169 parent = parent->Parent;
170
171 /* FIXME: need to determine if we are really the CurVer */
172
173 progid->CurVer = parent;
174 parent->VersionInd = progid;
175 }
176
177 return progid;
178 }
179
180 static MSIPROGID *load_given_progid(MSIPACKAGE *package, LPCWSTR name)
181 {
182 MSIPROGID *progid;
183 MSIRECORD *row;
184 static const WCHAR ExecSeqQuery[] =
185 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
186 '`','P','r','o','g','I','d','`',' ','W','H','E','R','E',' ',
187 '`','P','r','o','g','I','d','`',' ','=',' ','\'','%','s','\'',0};
188
189 if (!name)
190 return NULL;
191
192 /* check for progids already loaded */
193 LIST_FOR_EACH_ENTRY( progid, &package->progids, MSIPROGID, entry )
194 {
195 if (strcmpiW( progid->ProgID,name )==0)
196 {
197 TRACE("found progid %s (%p)\n",debugstr_w(name), progid );
198 return progid;
199 }
200 }
201
202 row = MSI_QueryGetRecord( package->db, ExecSeqQuery, name );
203 if (!row)
204 return NULL;
205
206 progid = load_progid(package, row);
207 msiobj_release(&row->hdr);
208
209 return progid;
210 }
211
212 static MSICLASS *load_class( MSIPACKAGE* package, MSIRECORD *row )
213 {
214 MSICLASS *cls;
215 DWORD i;
216 LPCWSTR buffer;
217
218 /* fill in the data */
219
220 cls = msi_alloc_zero( sizeof(MSICLASS) );
221 if (!cls)
222 return NULL;
223
224 list_add_tail( &package->classes, &cls->entry );
225
226 cls->clsid = msi_dup_record_field( row, 1 );
227 TRACE("loading class %s\n",debugstr_w(cls->clsid));
228 cls->Context = msi_dup_record_field( row, 2 );
229 buffer = MSI_RecordGetString(row,3);
230 cls->Component = get_loaded_component(package, buffer);
231
232 cls->ProgIDText = msi_dup_record_field(row,4);
233 cls->ProgID = load_given_progid(package, cls->ProgIDText);
234
235 cls->Description = msi_dup_record_field(row,5);
236
237 buffer = MSI_RecordGetString(row,6);
238 if (buffer)
239 cls->AppID = load_given_appid(package, buffer);
240
241 cls->FileTypeMask = msi_dup_record_field(row,7);
242
243 if (!MSI_RecordIsNull(row,9))
244 {
245
246 INT icon_index = MSI_RecordGetInteger(row,9);
247 LPCWSTR FileName = MSI_RecordGetString(row,8);
248 LPWSTR FilePath;
249 static const WCHAR fmt[] = {'%','s',',','%','i',0};
250
251 FilePath = build_icon_path(package,FileName);
252
253 cls->IconPath = msi_alloc( (strlenW(FilePath)+5)* sizeof(WCHAR) );
254
255 sprintfW(cls->IconPath,fmt,FilePath,icon_index);
256
257 msi_free(FilePath);
258 }
259 else
260 {
261 buffer = MSI_RecordGetString(row,8);
262 if (buffer)
263 cls->IconPath = build_icon_path(package,buffer);
264 }
265
266 if (!MSI_RecordIsNull(row,10))
267 {
268 i = MSI_RecordGetInteger(row,10);
269 if (i != MSI_NULL_INTEGER && i > 0 && i < 4)
270 {
271 static const WCHAR ole2[] = {'o','l','e','2','.','d','l','l',0};
272 static const WCHAR ole32[] = {'o','l','e','3','2','.','d','l','l',0};
273
274 switch(i)
275 {
276 case 1:
277 cls->DefInprocHandler = strdupW(ole2);
278 break;
279 case 2:
280 cls->DefInprocHandler32 = strdupW(ole32);
281 break;
282 case 3:
283 cls->DefInprocHandler = strdupW(ole2);
284 cls->DefInprocHandler32 = strdupW(ole32);
285 break;
286 }
287 }
288 else
289 {
290 cls->DefInprocHandler32 = msi_dup_record_field( row, 10);
291 reduce_to_longfilename(cls->DefInprocHandler32);
292 }
293 }
294 buffer = MSI_RecordGetString(row,11);
295 deformat_string(package,buffer,&cls->Argument);
296
297 buffer = MSI_RecordGetString(row,12);
298 cls->Feature = get_loaded_feature(package,buffer);
299
300 cls->Attributes = MSI_RecordGetInteger(row,13);
301
302 return cls;
303 }
304
305 /*
306 * the Class table has 3 primary keys. Generally it is only
307 * referenced through the first CLSID key. However when loading
308 * all of the classes we need to make sure we do not ignore rows
309 * with other Context and ComponentIndexs
310 */
311 static MSICLASS *load_given_class(MSIPACKAGE *package, LPCWSTR classid)
312 {
313 MSICLASS *cls;
314 MSIRECORD *row;
315 static const WCHAR ExecSeqQuery[] =
316 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
317 '`','C','l','a','s','s','`',' ','W','H','E','R','E',' ',
318 '`','C','L','S','I','D','`',' ','=',' ','\'','%','s','\'',0};
319
320 if (!classid)
321 return NULL;
322
323 /* check for classes already loaded */
324 LIST_FOR_EACH_ENTRY( cls, &package->classes, MSICLASS, entry )
325 {
326 if (lstrcmpiW( cls->clsid, classid )==0)
327 {
328 TRACE("found class %s (%p)\n",debugstr_w(classid), cls);
329 return cls;
330 }
331 }
332
333 row = MSI_QueryGetRecord(package->db, ExecSeqQuery, classid);
334 if (!row)
335 return NULL;
336
337 cls = load_class(package, row);
338 msiobj_release(&row->hdr);
339
340 return cls;
341 }
342
343 static MSIEXTENSION *load_given_extension( MSIPACKAGE *package, LPCWSTR extension );
344
345 static MSIMIME *load_mime( MSIPACKAGE* package, MSIRECORD *row )
346 {
347 LPCWSTR extension;
348 MSIMIME *mt;
349
350 /* fill in the data */
351
352 mt = msi_alloc_zero( sizeof(MSIMIME) );
353 if (!mt)
354 return mt;
355
356 mt->ContentType = msi_dup_record_field( row, 1 );
357 TRACE("loading mime %s\n", debugstr_w(mt->ContentType));
358
359 extension = MSI_RecordGetString( row, 2 );
360 mt->Extension = load_given_extension( package, extension );
361 mt->suffix = strdupW( extension );
362
363 mt->clsid = msi_dup_record_field( row, 3 );
364 mt->Class = load_given_class( package, mt->clsid );
365
366 list_add_tail( &package->mimes, &mt->entry );
367
368 return mt;
369 }
370
371 static MSIMIME *load_given_mime( MSIPACKAGE *package, LPCWSTR mime )
372 {
373 MSIRECORD *row;
374 static const WCHAR ExecSeqQuery[] =
375 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
376 '`','M','I','M','E','`',' ','W','H','E','R','E',' ',
377 '`','C','o','n','t','e','n','t','T','y','p','e','`',' ','=',' ',
378 '\'','%','s','\'',0};
379 MSIMIME *mt;
380
381 if (!mime)
382 return NULL;
383
384 /* check for mime already loaded */
385 LIST_FOR_EACH_ENTRY( mt, &package->mimes, MSIMIME, entry )
386 {
387 if (strcmpiW(mt->ContentType,mime)==0)
388 {
389 TRACE("found mime %s (%p)\n",debugstr_w(mime), mt);
390 return mt;
391 }
392 }
393
394 row = MSI_QueryGetRecord(package->db, ExecSeqQuery, mime);
395 if (!row)
396 return NULL;
397
398 mt = load_mime(package, row);
399 msiobj_release(&row->hdr);
400
401 return mt;
402 }
403
404 static MSIEXTENSION *load_extension( MSIPACKAGE* package, MSIRECORD *row )
405 {
406 MSIEXTENSION *ext;
407 LPCWSTR buffer;
408
409 /* fill in the data */
410
411 ext = msi_alloc_zero( sizeof(MSIEXTENSION) );
412 if (!ext)
413 return NULL;
414
415 list_init( &ext->verbs );
416
417 list_add_tail( &package->extensions, &ext->entry );
418
419 ext->Extension = msi_dup_record_field( row, 1 );
420 TRACE("loading extension %s\n", debugstr_w(ext->Extension));
421
422 buffer = MSI_RecordGetString( row, 2 );
423 ext->Component = get_loaded_component( package,buffer );
424
425 ext->ProgIDText = msi_dup_record_field( row, 3 );
426 ext->ProgID = load_given_progid( package, ext->ProgIDText );
427
428 buffer = MSI_RecordGetString( row, 4 );
429 ext->Mime = load_given_mime( package, buffer );
430
431 buffer = MSI_RecordGetString(row,5);
432 ext->Feature = get_loaded_feature( package, buffer );
433
434 return ext;
435 }
436
437 /*
438 * While the extension table has 2 primary keys, this function is only looking
439 * at the Extension key which is what is referenced as a foreign key
440 */
441 static MSIEXTENSION *load_given_extension( MSIPACKAGE *package, LPCWSTR name )
442 {
443 MSIRECORD *row;
444 MSIEXTENSION *ext;
445 static const WCHAR ExecSeqQuery[] =
446 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
447 '`','E','x','t','e','n','s','i','o','n','`',' ',
448 'W','H','E','R','E',' ',
449 '`','E','x','t','e','n','s','i','o','n','`',' ','=',' ',
450 '\'','%','s','\'',0};
451
452 if (!name)
453 return NULL;
454
455 if (name[0] == '.')
456 name++;
457
458 /* check for extensions already loaded */
459 LIST_FOR_EACH_ENTRY( ext, &package->extensions, MSIEXTENSION, entry )
460 {
461 if (strcmpiW( ext->Extension, name )==0)
462 {
463 TRACE("extension %s already loaded %p\n", debugstr_w(name), ext);
464 return ext;
465 }
466 }
467
468 row = MSI_QueryGetRecord( package->db, ExecSeqQuery, name );
469 if (!row)
470 return NULL;
471
472 ext = load_extension(package, row);
473 msiobj_release(&row->hdr);
474
475 return ext;
476 }
477
478 static UINT iterate_load_verb(MSIRECORD *row, LPVOID param)
479 {
480 MSIPACKAGE* package = param;
481 MSIVERB *verb;
482 LPCWSTR buffer;
483 MSIEXTENSION *extension;
484
485 buffer = MSI_RecordGetString(row,1);
486 extension = load_given_extension( package, buffer );
487 if (!extension)
488 {
489 ERR("Verb unable to find loaded extension %s\n", debugstr_w(buffer));
490 return ERROR_SUCCESS;
491 }
492
493 /* fill in the data */
494
495 verb = msi_alloc_zero( sizeof(MSIVERB) );
496 if (!verb)
497 return ERROR_OUTOFMEMORY;
498
499 verb->Verb = msi_dup_record_field(row,2);
500 TRACE("loading verb %s\n",debugstr_w(verb->Verb));
501 verb->Sequence = MSI_RecordGetInteger(row,3);
502
503 buffer = MSI_RecordGetString(row,4);
504 deformat_string(package,buffer,&verb->Command);
505
506 buffer = MSI_RecordGetString(row,5);
507 deformat_string(package,buffer,&verb->Argument);
508
509 /* associate the verb with the correct extension */
510 list_add_tail( &extension->verbs, &verb->entry );
511
512 return ERROR_SUCCESS;
513 }
514
515 static UINT iterate_all_classes(MSIRECORD *rec, LPVOID param)
516 {
517 MSICOMPONENT *comp;
518 LPCWSTR clsid;
519 LPCWSTR context;
520 LPCWSTR buffer;
521 MSIPACKAGE* package = param;
522 MSICLASS *cls;
523 BOOL match = FALSE;
524
525 clsid = MSI_RecordGetString(rec,1);
526 context = MSI_RecordGetString(rec,2);
527 buffer = MSI_RecordGetString(rec,3);
528 comp = get_loaded_component(package,buffer);
529
530 LIST_FOR_EACH_ENTRY( cls, &package->classes, MSICLASS, entry )
531 {
532 if (strcmpiW( clsid, cls->clsid ))
533 continue;
534 if (strcmpW( context, cls->Context ))
535 continue;
536 if (comp == cls->Component)
537 {
538 match = TRUE;
539 break;
540 }
541 }
542
543 if (!match)
544 load_class(package, rec);
545
546 return ERROR_SUCCESS;
547 }
548
549 static VOID load_all_classes(MSIPACKAGE *package)
550 {
551 UINT rc = ERROR_SUCCESS;
552 MSIQUERY *view;
553
554 static const WCHAR ExecSeqQuery[] =
555 {'S','E','L','E','C','T',' ','*',' ', 'F','R','O','M',' ',
556 '`','C','l','a','s','s','`',0};
557
558 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
559 if (rc != ERROR_SUCCESS)
560 return;
561
562 rc = MSI_IterateRecords(view, NULL, iterate_all_classes, package);
563 msiobj_release(&view->hdr);
564 }
565
566 static UINT iterate_all_extensions(MSIRECORD *rec, LPVOID param)
567 {
568 MSICOMPONENT *comp;
569 LPCWSTR buffer;
570 LPCWSTR extension;
571 MSIPACKAGE* package = param;
572 BOOL match = FALSE;
573 MSIEXTENSION *ext;
574
575 extension = MSI_RecordGetString(rec,1);
576 buffer = MSI_RecordGetString(rec,2);
577 comp = get_loaded_component(package,buffer);
578
579 LIST_FOR_EACH_ENTRY( ext, &package->extensions, MSIEXTENSION, entry )
580 {
581 if (strcmpiW(extension,ext->Extension))
582 continue;
583 if (comp == ext->Component)
584 {
585 match = TRUE;
586 break;
587 }
588 }
589
590 if (!match)
591 load_extension(package, rec);
592
593 return ERROR_SUCCESS;
594 }
595
596 static VOID load_all_extensions(MSIPACKAGE *package)
597 {
598 UINT rc = ERROR_SUCCESS;
599 MSIQUERY *view;
600
601 static const WCHAR ExecSeqQuery[] =
602 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
603 '`','E','x','t','e','n','s','i','o','n','`',0};
604
605 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
606 if (rc != ERROR_SUCCESS)
607 return;
608
609 rc = MSI_IterateRecords(view, NULL, iterate_all_extensions, package);
610 msiobj_release(&view->hdr);
611 }
612
613 static UINT iterate_all_progids(MSIRECORD *rec, LPVOID param)
614 {
615 LPCWSTR buffer;
616 MSIPACKAGE* package = param;
617
618 buffer = MSI_RecordGetString(rec,1);
619 load_given_progid(package,buffer);
620 return ERROR_SUCCESS;
621 }
622
623 static VOID load_all_progids(MSIPACKAGE *package)
624 {
625 UINT rc = ERROR_SUCCESS;
626 MSIQUERY *view;
627
628 static const WCHAR ExecSeqQuery[] =
629 {'S','E','L','E','C','T',' ','`','P','r','o','g','I','d','`',' ',
630 'F','R','O','M',' ', '`','P','r','o','g','I','d','`',0};
631
632 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
633 if (rc != ERROR_SUCCESS)
634 return;
635
636 rc = MSI_IterateRecords(view, NULL, iterate_all_progids, package);
637 msiobj_release(&view->hdr);
638 }
639
640 static VOID load_all_verbs(MSIPACKAGE *package)
641 {
642 UINT rc = ERROR_SUCCESS;
643 MSIQUERY *view;
644
645 static const WCHAR ExecSeqQuery[] =
646 {'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
647 '`','V','e','r','b','`',0};
648
649 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
650 if (rc != ERROR_SUCCESS)
651 return;
652
653 rc = MSI_IterateRecords(view, NULL, iterate_load_verb, package);
654 msiobj_release(&view->hdr);
655 }
656
657 static UINT iterate_all_mimes(MSIRECORD *rec, LPVOID param)
658 {
659 LPCWSTR buffer;
660 MSIPACKAGE* package = param;
661
662 buffer = MSI_RecordGetString(rec,1);
663 load_given_mime(package,buffer);
664 return ERROR_SUCCESS;
665 }
666
667 static VOID load_all_mimes(MSIPACKAGE *package)
668 {
669 UINT rc = ERROR_SUCCESS;
670 MSIQUERY *view;
671
672 static const WCHAR ExecSeqQuery[] =
673 {'S','E','L','E','C','T',' ',
674 '`','C','o','n','t','e','n','t','T','y','p','e','`',
675 ' ','F','R','O','M',' ',
676 '`','M','I','M','E','`',0};
677
678 rc = MSI_DatabaseOpenViewW(package->db, ExecSeqQuery, &view);
679 if (rc != ERROR_SUCCESS)
680 return;
681
682 rc = MSI_IterateRecords(view, NULL, iterate_all_mimes, package);
683 msiobj_release(&view->hdr);
684 }
685
686 static void load_classes_and_such(MSIPACKAGE *package)
687 {
688 TRACE("Loading all the class info and related tables\n");
689
690 /* check if already loaded */
691 if (!list_empty( &package->classes ) ||
692 !list_empty( &package->mimes ) ||
693 !list_empty( &package->extensions ) ||
694 !list_empty( &package->progids ) )
695 return;
696
697 load_all_classes(package);
698 load_all_extensions(package);
699 load_all_progids(package);
700 /* these loads must come after the other loads */
701 load_all_verbs(package);
702 load_all_mimes(package);
703 }
704
705 static void mark_progid_for_install( MSIPACKAGE* package, MSIPROGID *progid )
706 {
707 MSIPROGID *child;
708
709 if (!progid)
710 return;
711
712 if (progid->InstallMe)
713 return;
714
715 progid->InstallMe = TRUE;
716
717 /* all children if this is a parent also install */
718 LIST_FOR_EACH_ENTRY( child, &package->progids, MSIPROGID, entry )
719 {
720 if (child->Parent == progid)
721 mark_progid_for_install( package, child );
722 }
723 }
724
725 static void mark_progid_for_uninstall( MSIPACKAGE *package, MSIPROGID *progid )
726 {
727 MSIPROGID *child;
728
729 if (!progid)
730 return;
731
732 if (!progid->InstallMe)
733 return;
734
735 progid->InstallMe = FALSE;
736
737 LIST_FOR_EACH_ENTRY( child, &package->progids, MSIPROGID, entry )
738 {
739 if (child->Parent == progid)
740 mark_progid_for_uninstall( package, child );
741 }
742 }
743
744 static void mark_mime_for_install( MSIMIME *mime )
745 {
746 if (!mime)
747 return;
748 mime->InstallMe = TRUE;
749 }
750
751 static void mark_mime_for_uninstall( MSIMIME *mime )
752 {
753 if (!mime)
754 return;
755 mime->InstallMe = FALSE;
756 }
757
758 static UINT register_appid(const MSIAPPID *appid, LPCWSTR app )
759 {
760 static const WCHAR szRemoteServerName[] =
761 {'R','e','m','o','t','e','S','e','r','v','e','r','N','a','m','e',0};
762 static const WCHAR szLocalService[] =
763 {'L','o','c','a','l','S','e','r','v','i','c','e',0};
764 static const WCHAR szService[] =
765 {'S','e','r','v','i','c','e','P','a','r','a','m','e','t','e','r','s',0};
766 static const WCHAR szDLL[] =
767 {'D','l','l','S','u','r','r','o','g','a','t','e',0};
768 static const WCHAR szActivate[] =
769 {'A','c','t','i','v','a','t','e','A','s','S','t','o','r','a','g','e',0};
770 static const WCHAR szY[] = {'Y',0};
771 static const WCHAR szRunAs[] = {'R','u','n','A','s',0};
772 static const WCHAR szUser[] =
773 {'I','n','t','e','r','a','c','t','i','v','e',' ','U','s','e','r',0};
774
775 HKEY hkey2,hkey3;
776
777 RegCreateKeyW(HKEY_CLASSES_ROOT,szAppID,&hkey2);
778 RegCreateKeyW( hkey2, appid->AppID, &hkey3 );
779 RegCloseKey(hkey2);
780 msi_reg_set_val_str( hkey3, NULL, app );
781
782 if (appid->RemoteServerName)
783 msi_reg_set_val_str( hkey3, szRemoteServerName, appid->RemoteServerName );
784
785 if (appid->LocalServer)
786 msi_reg_set_val_str( hkey3, szLocalService, appid->LocalServer );
787
788 if (appid->ServiceParameters)
789 msi_reg_set_val_str( hkey3, szService, appid->ServiceParameters );
790
791 if (appid->DllSurrogate)
792 msi_reg_set_val_str( hkey3, szDLL, appid->DllSurrogate );
793
794 if (appid->ActivateAtStorage)
795 msi_reg_set_val_str( hkey3, szActivate, szY );
796
797 if (appid->RunAsInteractiveUser)
798 msi_reg_set_val_str( hkey3, szRunAs, szUser );
799
800 RegCloseKey(hkey3);
801 return ERROR_SUCCESS;
802 }
803
804 UINT ACTION_RegisterClassInfo(MSIPACKAGE *package)
805 {
806 static const WCHAR szFileType_fmt[] = {'F','i','l','e','T','y','p','e','\\','%','s','\\','%','i',0};
807 MSIRECORD *uirow;
808 HKEY hkey,hkey2,hkey3;
809 MSICLASS *cls;
810
811 load_classes_and_such(package);
812 if (RegCreateKeyW(HKEY_CLASSES_ROOT, szCLSID, &hkey) != ERROR_SUCCESS)
813 return ERROR_FUNCTION_FAILED;
814
815 LIST_FOR_EACH_ENTRY( cls, &package->classes, MSICLASS, entry )
816 {
817 MSICOMPONENT *comp;
818 MSIFILE *file;
819 DWORD size;
820 LPWSTR argument;
821 MSIFEATURE *feature;
822
823 comp = cls->Component;
824 if ( !comp )
825 continue;
826
827 if (!comp->Enabled)
828 {
829 TRACE("component is disabled\n");
830 continue;
831 }
832
833 feature = cls->Feature;
834 if (!feature)
835 continue;
836
837 if (feature->ActionRequest != INSTALLSTATE_LOCAL &&
838 feature->ActionRequest != INSTALLSTATE_ADVERTISED )
839 {
840 TRACE("Feature %s not scheduled for installation, skipping registration of class %s\n",
841 debugstr_w(feature->Feature), debugstr_w(cls->clsid));
842 continue;
843 }
844 feature->Action = feature->ActionRequest;
845
846 file = get_loaded_file( package, comp->KeyPath );
847 if (!file)
848 {
849 TRACE("COM server not provided, skipping class %s\n", debugstr_w(cls->clsid));
850 continue;
851 }
852
853 TRACE("Registering class %s (%p)\n", debugstr_w(cls->clsid), cls);
854
855 cls->Installed = TRUE;
856 mark_progid_for_install( package, cls->ProgID );
857
858 RegCreateKeyW( hkey, cls->clsid, &hkey2 );
859
860 if (cls->Description)
861 msi_reg_set_val_str( hkey2, NULL, cls->Description );
862
863 RegCreateKeyW( hkey2, cls->Context, &hkey3 );
864
865 /*
866 * FIXME: Implement install on demand (advertised components).
867 *
868 * ole32.dll should call msi.MsiProvideComponentFromDescriptor()
869 * when it needs an InProcServer that doesn't exist.
870 * The component advertise string should be in the "InProcServer" value.
871 */
872 size = lstrlenW( file->TargetPath )+1;
873 if (cls->Argument)
874 size += lstrlenW(cls->Argument)+1;
875
876 argument = msi_alloc( size * sizeof(WCHAR) );
877 lstrcpyW( argument, file->TargetPath );
878
879 if (cls->Argument)
880 {
881 lstrcatW( argument, szSpace );
882 lstrcatW( argument, cls->Argument );
883 }
884
885 msi_reg_set_val_str( hkey3, NULL, argument );
886 msi_free(argument);
887
888 RegCloseKey(hkey3);
889
890 if (cls->ProgID || cls->ProgIDText)
891 {
892 LPCWSTR progid;
893
894 if (cls->ProgID)
895 progid = cls->ProgID->ProgID;
896 else
897 progid = cls->ProgIDText;
898
899 msi_reg_set_subkey_val( hkey2, szProgID, NULL, progid );
900
901 if (cls->ProgID && cls->ProgID->VersionInd)
902 {
903 msi_reg_set_subkey_val( hkey2, szVIProgID, NULL,
904 cls->ProgID->VersionInd->ProgID );
905 }
906 }
907
908 if (cls->AppID)
909 {
910 MSIAPPID *appid = cls->AppID;
911 msi_reg_set_val_str( hkey2, szAppID, appid->AppID );
912 register_appid( appid, cls->Description );
913 }
914
915 if (cls->IconPath)
916 msi_reg_set_subkey_val( hkey2, szDefaultIcon, NULL, cls->IconPath );
917
918 if (cls->DefInprocHandler)
919 msi_reg_set_subkey_val( hkey2, szInprocHandler, NULL, cls->DefInprocHandler );
920
921 if (cls->DefInprocHandler32)
922 msi_reg_set_subkey_val( hkey2, szInprocHandler32, NULL, cls->DefInprocHandler32 );
923
924 RegCloseKey(hkey2);
925
926 /* if there is a FileTypeMask, register the FileType */
927 if (cls->FileTypeMask)
928 {
929 LPWSTR ptr, ptr2;
930 LPWSTR keyname;
931 INT index = 0;
932 ptr = cls->FileTypeMask;
933 while (ptr && *ptr)
934 {
935 ptr2 = strchrW(ptr,';');
936 if (ptr2)
937 *ptr2 = 0;
938 keyname = msi_alloc( (strlenW(szFileType_fmt) + strlenW(cls->clsid) + 4) * sizeof(WCHAR));
939 sprintfW( keyname, szFileType_fmt, cls->clsid, index );
940
941 msi_reg_set_subkey_val( HKEY_CLASSES_ROOT, keyname, NULL, ptr );
942 msi_free(keyname);
943
944 if (ptr2)
945 ptr = ptr2+1;
946 else
947 ptr = NULL;
948
949 index ++;
950 }
951 }
952
953 uirow = MSI_CreateRecord(1);
954 MSI_RecordSetStringW( uirow, 1, cls->clsid );
955 ui_actiondata(package,szRegisterClassInfo,uirow);
956 msiobj_release(&uirow->hdr);
957 }
958
959 RegCloseKey(hkey);
960 return ERROR_SUCCESS;
961 }
962
963 UINT ACTION_UnregisterClassInfo( MSIPACKAGE *package )
964 {
965 static const WCHAR szFileType[] = {'F','i','l','e','T','y','p','e','\\',0};
966 MSIRECORD *uirow;
967 MSICLASS *cls;
968 HKEY hkey, hkey2;
969
970 load_classes_and_such( package );
971 if (RegOpenKeyW( HKEY_CLASSES_ROOT, szCLSID, &hkey ) != ERROR_SUCCESS)
972 return ERROR_SUCCESS;
973
974 LIST_FOR_EACH_ENTRY( cls, &package->classes, MSICLASS, entry )
975 {
976 MSIFEATURE *feature;
977 MSICOMPONENT *comp;
978 LPWSTR filetype;
979 LONG res;
980
981 comp = cls->Component;
982 if (!comp)
983 continue;
984
985 if (!comp->Enabled)
986 {
987 TRACE("component is disabled\n");
988 continue;
989 }
990
991 feature = cls->Feature;
992 if (!feature)
993 continue;
994
995 if (feature->ActionRequest != INSTALLSTATE_ABSENT)
996 {
997 TRACE("Feature %s not scheduled for removal, skipping unregistration of class %s\n",
998 debugstr_w(feature->Feature), debugstr_w(cls->clsid));
999 continue;
1000 }
1001 feature->Action = feature->ActionRequest;
1002
1003 TRACE("Unregistering class %s (%p)\n", debugstr_w(cls->clsid), cls);
1004
1005 cls->Installed = FALSE;
1006 mark_progid_for_uninstall( package, cls->ProgID );
1007
1008 res = RegDeleteTreeW( hkey, cls->clsid );
1009 if (res != ERROR_SUCCESS)
1010 WARN("Failed to delete class key %d\n", res);
1011
1012 if (cls->AppID)
1013 {
1014 res = RegOpenKeyW( HKEY_CLASSES_ROOT, szAppID, &hkey2 );
1015 if (res == ERROR_SUCCESS)
1016 {
1017 res = RegDeleteKeyW( hkey2, cls->AppID->AppID );
1018 if (res != ERROR_SUCCESS)
1019 WARN("Failed to delete appid key %d\n", res);
1020 RegCloseKey( hkey2 );
1021 }
1022 }
1023 if (cls->FileTypeMask)
1024 {
1025 filetype = msi_alloc( (strlenW( szFileType ) + strlenW( cls->clsid ) + 1) * sizeof(WCHAR) );
1026 if (filetype)
1027 {
1028 strcpyW( filetype, szFileType );
1029 strcatW( filetype, cls->clsid );
1030 res = RegDeleteTreeW( HKEY_CLASSES_ROOT, filetype );
1031 msi_free( filetype );
1032
1033 if (res != ERROR_SUCCESS)
1034 WARN("Failed to delete file type %d\n", res);
1035 }
1036 }
1037
1038 uirow = MSI_CreateRecord( 1 );
1039 MSI_RecordSetStringW( uirow, 1, cls->clsid );
1040 ui_actiondata( package, szUnregisterClassInfo, uirow );
1041 msiobj_release( &uirow->hdr );
1042 }
1043
1044 RegCloseKey( hkey );
1045 return ERROR_SUCCESS;
1046 }
1047
1048 static LPCWSTR get_clsid_of_progid( const MSIPROGID *progid )
1049 {
1050 while (progid)
1051 {
1052 if (progid->Class)
1053 return progid->Class->clsid;
1054 if (progid->Parent == progid)
1055 break;
1056 progid = progid->Parent;
1057 }
1058 return NULL;
1059 }
1060
1061 static UINT register_progid( const MSIPROGID* progid )
1062 {
1063 static const WCHAR szCurVer[] = {'C','u','r','V','e','r',0};
1064 HKEY hkey = 0;
1065 UINT rc;
1066
1067 rc = RegCreateKeyW( HKEY_CLASSES_ROOT, progid->ProgID, &hkey );
1068 if (rc == ERROR_SUCCESS)
1069 {
1070 LPCWSTR clsid = get_clsid_of_progid( progid );
1071
1072 if (clsid)
1073 msi_reg_set_subkey_val( hkey, szCLSID, NULL, clsid );
1074 else
1075 ERR("%s has no class\n", debugstr_w( progid->ProgID ) );
1076
1077 if (progid->Description)
1078 msi_reg_set_val_str( hkey, NULL, progid->Description );
1079
1080 if (progid->IconPath)
1081 msi_reg_set_subkey_val( hkey, szDefaultIcon, NULL, progid->IconPath );
1082
1083 /* write out the current version */
1084 if (progid->CurVer)
1085 msi_reg_set_subkey_val( hkey, szCurVer, NULL, progid->CurVer->ProgID );
1086
1087 RegCloseKey(hkey);
1088 }
1089 else
1090 ERR("failed to create key %s\n", debugstr_w( progid->ProgID ) );
1091
1092 return rc;
1093 }
1094
1095 UINT ACTION_RegisterProgIdInfo(MSIPACKAGE *package)
1096 {
1097 MSIPROGID *progid;
1098 MSIRECORD *uirow;
1099
1100 load_classes_and_such(package);
1101
1102 LIST_FOR_EACH_ENTRY( progid, &package->progids, MSIPROGID, entry )
1103 {
1104 /* check if this progid is to be installed */
1105 if (progid->Class && progid->Class->Installed)
1106 progid->InstallMe = TRUE;
1107
1108 if (!progid->InstallMe)
1109 {
1110 TRACE("progid %s not scheduled to be installed\n",
1111 debugstr_w(progid->ProgID));
1112 continue;
1113 }
1114
1115 TRACE("Registering progid %s\n", debugstr_w(progid->ProgID));
1116
1117 register_progid( progid );
1118
1119 uirow = MSI_CreateRecord( 1 );
1120 MSI_RecordSetStringW( uirow, 1, progid->ProgID );
1121 ui_actiondata( package, szRegisterProgIdInfo, uirow );
1122 msiobj_release( &uirow->hdr );
1123 }
1124
1125 return ERROR_SUCCESS;
1126 }
1127
1128 UINT ACTION_UnregisterProgIdInfo( MSIPACKAGE *package )
1129 {
1130 MSIPROGID *progid;
1131 MSIRECORD *uirow;
1132 LONG res;
1133
1134 load_classes_and_such( package );
1135
1136 LIST_FOR_EACH_ENTRY( progid, &package->progids, MSIPROGID, entry )
1137 {
1138 /* check if this progid is to be removed */
1139 if (progid->Class && !progid->Class->Installed)
1140 progid->InstallMe = FALSE;
1141
1142 if (progid->InstallMe)
1143 {
1144 TRACE("progid %s not scheduled to be removed\n", debugstr_w(progid->ProgID));
1145 continue;
1146 }
1147
1148 TRACE("Unregistering progid %s\n", debugstr_w(progid->ProgID));
1149
1150 res = RegDeleteTreeW( HKEY_CLASSES_ROOT, progid->ProgID );
1151 if (res != ERROR_SUCCESS)
1152 TRACE("Failed to delete progid key %d\n", res);
1153
1154 uirow = MSI_CreateRecord( 1 );
1155 MSI_RecordSetStringW( uirow, 1, progid->ProgID );
1156 ui_actiondata( package, szUnregisterProgIdInfo, uirow );
1157 msiobj_release( &uirow->hdr );
1158 }
1159
1160 return ERROR_SUCCESS;
1161 }
1162
1163 static UINT register_verb(MSIPACKAGE *package, LPCWSTR progid,
1164 MSICOMPONENT* component, const MSIEXTENSION* extension,
1165 MSIVERB* verb, INT* Sequence )
1166 {
1167 LPWSTR keyname;
1168 HKEY key;
1169 static const WCHAR szShell[] = {'s','h','e','l','l',0};
1170 static const WCHAR szCommand[] = {'c','o','m','m','a','n','d',0};
1171 static const WCHAR fmt[] = {'\"','%','s','\"',' ','%','s',0};
1172 static const WCHAR fmt2[] = {'\"','%','s','\"',0};
1173 LPWSTR command;
1174 DWORD size;
1175 LPWSTR advertise;
1176
1177 keyname = build_directory_name(4, progid, szShell, verb->Verb, szCommand);
1178
1179 TRACE("Making Key %s\n",debugstr_w(keyname));
1180 RegCreateKeyW(HKEY_CLASSES_ROOT, keyname, &key);
1181 size = strlenW(component->FullKeypath);
1182 if (verb->Argument)
1183 size += strlenW(verb->Argument);
1184 size += 4;
1185
1186 command = msi_alloc(size * sizeof (WCHAR));
1187 if (verb->Argument)
1188 sprintfW(command, fmt, component->FullKeypath, verb->Argument);
1189 else
1190 sprintfW(command, fmt2, component->FullKeypath);
1191
1192 msi_reg_set_val_str( key, NULL, command );
1193 msi_free(command);
1194
1195 advertise = create_component_advertise_string(package, component,
1196 extension->Feature->Feature);
1197
1198 size = strlenW(advertise);
1199
1200 if (verb->Argument)
1201 size += strlenW(verb->Argument);
1202 size += 4;
1203
1204 command = msi_alloc_zero(size * sizeof (WCHAR));
1205
1206 strcpyW(command,advertise);
1207 if (verb->Argument)
1208 {
1209 strcatW(command,szSpace);
1210 strcatW(command,verb->Argument);
1211 }
1212
1213 msi_reg_set_val_multi_str( key, szCommand, command );
1214
1215 RegCloseKey(key);
1216 msi_free(keyname);
1217 msi_free(advertise);
1218 msi_free(command);
1219
1220 if (verb->Command)
1221 {
1222 keyname = build_directory_name(3, progid, szShell, verb->Verb);
1223 msi_reg_set_subkey_val( HKEY_CLASSES_ROOT, keyname, NULL, verb->Command );
1224 msi_free(keyname);
1225 }
1226
1227 if (verb->Sequence != MSI_NULL_INTEGER)
1228 {
1229 if (*Sequence == MSI_NULL_INTEGER || verb->Sequence < *Sequence)
1230 {
1231 *Sequence = verb->Sequence;
1232 keyname = build_directory_name(2, progid, szShell);
1233 msi_reg_set_subkey_val( HKEY_CLASSES_ROOT, keyname, NULL, verb->Verb );
1234 msi_free(keyname);
1235 }
1236 }
1237 return ERROR_SUCCESS;
1238 }
1239
1240 UINT ACTION_RegisterExtensionInfo(MSIPACKAGE *package)
1241 {
1242 static const WCHAR szContentType[] =
1243 {'C','o','n','t','e','n','t',' ','T','y','p','e',0 };
1244 HKEY hkey = NULL;
1245 MSIEXTENSION *ext;
1246 MSIRECORD *uirow;
1247 BOOL install_on_demand = TRUE;
1248 LONG res;
1249
1250 load_classes_and_such(package);
1251
1252 /* We need to set install_on_demand based on if the shell handles advertised
1253 * shortcuts and the like. Because Mike McCormack is working on this i am
1254 * going to default to TRUE
1255 */
1256
1257 LIST_FOR_EACH_ENTRY( ext, &package->extensions, MSIEXTENSION, entry )
1258 {
1259 LPWSTR extension;
1260 MSIFEATURE *feature;
1261
1262 if (!ext->Component)
1263 continue;
1264
1265 if (!ext->Component->Enabled)
1266 {
1267 TRACE("component is disabled\n");
1268 continue;
1269 }
1270
1271 feature = ext->Feature;
1272 if (!feature)
1273 continue;
1274
1275 /*
1276 * yes. MSDN says that these are based on _Feature_ not on
1277 * Component. So verify the feature is to be installed
1278 */
1279 if (feature->ActionRequest != INSTALLSTATE_LOCAL &&
1280 !(install_on_demand && feature->ActionRequest == INSTALLSTATE_ADVERTISED))
1281 {
1282 TRACE("Feature %s not scheduled for installation, skipping registration of extension %s\n",
1283 debugstr_w(feature->Feature), debugstr_w(ext->Extension));
1284 continue;
1285 }
1286 feature->Action = feature->ActionRequest;
1287
1288 TRACE("Registering extension %s (%p)\n", debugstr_w(ext->Extension), ext);
1289
1290 ext->Installed = TRUE;
1291
1292 /* this is only registered if the extension has at least 1 verb
1293 * according to MSDN
1294 */
1295 if (ext->ProgID && !list_empty( &ext->verbs ) )
1296 mark_progid_for_install( package, ext->ProgID );
1297
1298 mark_mime_for_install(ext->Mime);
1299
1300 extension = msi_alloc( (strlenW( ext->Extension ) + 2) * sizeof(WCHAR) );
1301 if (extension)
1302 {
1303 extension[0] = '.';
1304 strcpyW( extension + 1, ext->Extension );
1305 res = RegCreateKeyW( HKEY_CLASSES_ROOT, extension, &hkey );
1306 msi_free( extension );
1307 if (res != ERROR_SUCCESS)
1308 WARN("Failed to create extension key %d\n", res);
1309 }
1310
1311 if (ext->Mime)
1312 msi_reg_set_val_str( hkey, szContentType, ext->Mime->ContentType );
1313
1314 if (ext->ProgID || ext->ProgIDText)
1315 {
1316 static const WCHAR szSN[] =
1317 {'\\','S','h','e','l','l','N','e','w',0};
1318 HKEY hkey2;
1319 LPWSTR newkey;
1320 LPCWSTR progid;
1321 MSIVERB *verb;
1322 INT Sequence = MSI_NULL_INTEGER;
1323
1324 if (ext->ProgID)
1325 progid = ext->ProgID->ProgID;
1326 else
1327 progid = ext->ProgIDText;
1328
1329 msi_reg_set_val_str( hkey, NULL, progid );
1330
1331 newkey = msi_alloc( (strlenW(progid)+strlenW(szSN)+1) * sizeof(WCHAR));
1332
1333 strcpyW(newkey,progid);
1334 strcatW(newkey,szSN);
1335 RegCreateKeyW(hkey,newkey,&hkey2);
1336 RegCloseKey(hkey2);
1337
1338 msi_free(newkey);
1339
1340 /* do all the verbs */
1341 LIST_FOR_EACH_ENTRY( verb, &ext->verbs, MSIVERB, entry )
1342 {
1343 register_verb( package, progid, ext->Component,
1344 ext, verb, &Sequence);
1345 }
1346 }
1347
1348 RegCloseKey(hkey);
1349
1350 uirow = MSI_CreateRecord(1);
1351 MSI_RecordSetStringW( uirow, 1, ext->Extension );
1352 ui_actiondata(package,szRegisterExtensionInfo,uirow);
1353 msiobj_release(&uirow->hdr);
1354 }
1355
1356 return ERROR_SUCCESS;
1357 }
1358
1359 UINT ACTION_UnregisterExtensionInfo( MSIPACKAGE *package )
1360 {
1361 MSIEXTENSION *ext;
1362 MSIRECORD *uirow;
1363 LONG res;
1364
1365 load_classes_and_such( package );
1366
1367 LIST_FOR_EACH_ENTRY( ext, &package->extensions, MSIEXTENSION, entry )
1368 {
1369 LPWSTR extension;
1370 MSIFEATURE *feature;
1371
1372 if (!ext->Component)
1373 continue;
1374
1375 if (!ext->Component->Enabled)
1376 {
1377 TRACE("component is disabled\n");
1378 continue;
1379 }
1380
1381 feature = ext->Feature;
1382 if (!feature)
1383 continue;
1384
1385 if (feature->ActionRequest != INSTALLSTATE_ABSENT)
1386 {
1387 TRACE("Feature %s not scheduled for removal, skipping unregistration of extension %s\n",
1388 debugstr_w(feature->Feature), debugstr_w(ext->Extension));
1389 continue;
1390 }
1391
1392 TRACE("Unregistering extension %s\n", debugstr_w(ext->Extension));
1393
1394 ext->Installed = FALSE;
1395
1396 if (ext->ProgID && !list_empty( &ext->verbs ))
1397 mark_progid_for_uninstall( package, ext->ProgID );
1398
1399 mark_mime_for_uninstall( ext->Mime );
1400
1401 extension = msi_alloc( (strlenW( ext->Extension ) + 2) * sizeof(WCHAR) );
1402 if (extension)
1403 {
1404 extension[0] = '.';
1405 strcpyW( extension + 1, ext->Extension );
1406 res = RegDeleteTreeW( HKEY_CLASSES_ROOT, extension );
1407 msi_free( extension );
1408 if (res != ERROR_SUCCESS)
1409 WARN("Failed to delete extension key %d\n", res);
1410 }
1411
1412 if (ext->ProgID || ext->ProgIDText)
1413 {
1414 static const WCHAR shellW[] = {'\\','s','h','e','l','l',0};
1415 LPCWSTR progid;
1416 LPWSTR progid_shell;
1417
1418 if (ext->ProgID)
1419 progid = ext->ProgID->ProgID;
1420 else
1421 progid = ext->ProgIDText;
1422
1423 progid_shell = msi_alloc( (strlenW( progid ) + strlenW( shellW ) + 1) * sizeof(WCHAR) );
1424 if (progid_shell)
1425 {
1426 strcpyW( progid_shell, progid );
1427 strcatW( progid_shell, shellW );
1428 res = RegDeleteTreeW( HKEY_CLASSES_ROOT, progid_shell );
1429 msi_free( progid_shell );
1430 if (res != ERROR_SUCCESS)
1431 WARN("Failed to delete shell key %d\n", res);
1432 RegDeleteKeyW( HKEY_CLASSES_ROOT, progid );
1433 }
1434 }
1435
1436 uirow = MSI_CreateRecord( 1 );
1437 MSI_RecordSetStringW( uirow, 1, ext->Extension );
1438 ui_actiondata( package, szUnregisterExtensionInfo, uirow );
1439 msiobj_release( &uirow->hdr );
1440 }
1441
1442 return ERROR_SUCCESS;
1443 }
1444
1445 UINT ACTION_RegisterMIMEInfo(MSIPACKAGE *package)
1446 {
1447 static const WCHAR szExten[] =
1448 {'E','x','t','e','n','s','i','o','n',0 };
1449 MSIRECORD *uirow;
1450 MSIMIME *mt;
1451
1452 load_classes_and_such(package);
1453
1454 LIST_FOR_EACH_ENTRY( mt, &package->mimes, MSIMIME, entry )
1455 {
1456 LPWSTR extension;
1457 LPWSTR key;
1458
1459 /*
1460 * check if the MIME is to be installed. Either as requested by an
1461 * extension or Class
1462 */
1463 mt->InstallMe = (mt->InstallMe ||
1464 (mt->Class && mt->Class->Installed) ||
1465 (mt->Extension && mt->Extension->Installed));
1466
1467 if (!mt->InstallMe)
1468 {
1469 TRACE("MIME %s not scheduled to be installed\n", debugstr_w(mt->ContentType));
1470 continue;
1471 }
1472
1473 TRACE("Registering MIME type %s\n", debugstr_w(mt->ContentType));
1474
1475 extension = msi_alloc( (strlenW( mt->Extension->Extension ) + 2) * sizeof(WCHAR) );
1476 key = msi_alloc( (strlenW( mt->ContentType ) + strlenW( szMIMEDatabase ) + 1) * sizeof(WCHAR) );
1477
1478 if (extension && key)
1479 {
1480 extension[0] = '.';
1481 strcpyW( extension + 1, mt->Extension->Extension );
1482
1483 strcpyW( key, szMIMEDatabase );
1484 strcatW( key, mt->ContentType );
1485 msi_reg_set_subkey_val( HKEY_CLASSES_ROOT, key, szExten, extension );
1486
1487 if (mt->clsid)
1488 msi_reg_set_subkey_val( HKEY_CLASSES_ROOT, key, szCLSID, mt->clsid );
1489 }
1490 msi_free( extension );
1491 msi_free( key );
1492
1493 uirow = MSI_CreateRecord( 2 );
1494 MSI_RecordSetStringW( uirow, 1, mt->ContentType );
1495 MSI_RecordSetStringW( uirow, 2, mt->suffix );
1496 ui_actiondata( package, szRegisterMIMEInfo, uirow );
1497 msiobj_release( &uirow->hdr );
1498 }
1499
1500 return ERROR_SUCCESS;
1501 }
1502
1503 UINT ACTION_UnregisterMIMEInfo( MSIPACKAGE *package )
1504 {
1505 MSIRECORD *uirow;
1506 MSIMIME *mime;
1507
1508 load_classes_and_such( package );
1509
1510 LIST_FOR_EACH_ENTRY( mime, &package->mimes, MSIMIME, entry )
1511 {
1512 LONG res;
1513 LPWSTR mime_key;
1514
1515 mime->InstallMe = (mime->InstallMe ||
1516 (mime->Class && mime->Class->Installed) ||
1517 (mime->Extension && mime->Extension->Installed));
1518
1519 if (mime->InstallMe)
1520 {
1521 TRACE("MIME %s not scheduled to be removed\n", debugstr_w(mime->ContentType));
1522 continue;
1523 }
1524
1525 TRACE("Unregistering MIME type %s\n", debugstr_w(mime->ContentType));
1526
1527 mime_key = msi_alloc( (strlenW( szMIMEDatabase ) + strlenW( mime->ContentType ) + 1) * sizeof(WCHAR) );
1528 if (mime_key)
1529 {
1530 strcpyW( mime_key, szMIMEDatabase );
1531 strcatW( mime_key, mime->ContentType );
1532 res = RegDeleteKeyW( HKEY_CLASSES_ROOT, mime_key );
1533 if (res != ERROR_SUCCESS)
1534 WARN("Failed to delete MIME key %d\n", res);
1535 msi_free( mime_key );
1536 }
1537
1538 uirow = MSI_CreateRecord( 2 );
1539 MSI_RecordSetStringW( uirow, 1, mime->ContentType );
1540 MSI_RecordSetStringW( uirow, 2, mime->suffix );
1541 ui_actiondata( package, szUnregisterMIMEInfo, uirow );
1542 msiobj_release( &uirow->hdr );
1543 }
1544
1545 return ERROR_SUCCESS;
1546 }