95daabb40bbe3ac48eca25a654d6f2486271312f
[reactos.git] / reactos / dll / win32 / msi / appsearch.c
1 /*
2 * Implementation of the AppSearch action of the Microsoft Installer (msi.dll)
3 *
4 * Copyright 2005 Juan Lang
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 #include <stdarg.h>
21
22 #define COBJMACROS
23
24 #include "windef.h"
25 #include "winbase.h"
26 #include "winreg.h"
27 #include "msi.h"
28 #include "msiquery.h"
29 #include "msidefs.h"
30 #include "winver.h"
31 #include "shlwapi.h"
32 #include "wine/unicode.h"
33 #include "wine/debug.h"
34 #include "msipriv.h"
35
36 WINE_DEFAULT_DEBUG_CHANNEL(msi);
37
38 typedef struct tagMSISIGNATURE
39 {
40 LPCWSTR Name; /* NOT owned by this structure */
41 LPWSTR File;
42 DWORD MinVersionMS;
43 DWORD MinVersionLS;
44 DWORD MaxVersionMS;
45 DWORD MaxVersionLS;
46 DWORD MinSize;
47 DWORD MaxSize;
48 FILETIME MinTime;
49 FILETIME MaxTime;
50 LPWSTR Languages;
51 }MSISIGNATURE;
52
53 static void ACTION_VerStrToInteger(LPCWSTR verStr, PDWORD ms, PDWORD ls)
54 {
55 const WCHAR *ptr;
56 int x1 = 0, x2 = 0, x3 = 0, x4 = 0;
57
58 x1 = atoiW(verStr);
59 ptr = strchrW(verStr, '.');
60 if (ptr)
61 {
62 x2 = atoiW(ptr + 1);
63 ptr = strchrW(ptr + 1, '.');
64 }
65 if (ptr)
66 {
67 x3 = atoiW(ptr + 1);
68 ptr = strchrW(ptr + 1, '.');
69 }
70 if (ptr)
71 x4 = atoiW(ptr + 1);
72 /* FIXME: byte-order dependent? */
73 *ms = x1 << 16 | x2;
74 *ls = x3 << 16 | x4;
75 }
76
77 /* Fills in sig with the values from the Signature table, where name is the
78 * signature to find. Upon return, sig->File will be NULL if the record is not
79 * found, and not NULL if it is found.
80 * Warning: clears all fields in sig!
81 * Returns ERROR_SUCCESS upon success (where not finding the record counts as
82 * success), something else on error.
83 */
84 static UINT ACTION_AppSearchGetSignature(MSIPACKAGE *package, MSISIGNATURE *sig, LPCWSTR name)
85 {
86 static const WCHAR query[] = {
87 's','e','l','e','c','t',' ','*',' ',
88 'f','r','o','m',' ',
89 'S','i','g','n','a','t','u','r','e',' ',
90 'w','h','e','r','e',' ','S','i','g','n','a','t','u','r','e',' ','=',' ',
91 '\'','%','s','\'',0};
92 LPWSTR minVersion, maxVersion;
93 MSIRECORD *row;
94 DWORD time;
95
96 TRACE("package %p, sig %p\n", package, sig);
97
98 memset(sig, 0, sizeof(*sig));
99 sig->Name = name;
100 row = MSI_QueryGetRecord( package->db, query, name );
101 if (!row)
102 {
103 TRACE("failed to query signature for %s\n", debugstr_w(name));
104 return ERROR_SUCCESS;
105 }
106
107 /* get properties */
108 sig->File = msi_dup_record_field(row,2);
109 minVersion = msi_dup_record_field(row,3);
110 if (minVersion)
111 {
112 ACTION_VerStrToInteger(minVersion, &sig->MinVersionMS, &sig->MinVersionLS);
113 msi_free( minVersion );
114 }
115 maxVersion = msi_dup_record_field(row,4);
116 if (maxVersion)
117 {
118 ACTION_VerStrToInteger(maxVersion, &sig->MaxVersionMS, &sig->MaxVersionLS);
119 msi_free( maxVersion );
120 }
121 sig->MinSize = MSI_RecordGetInteger(row,5);
122 if (sig->MinSize == MSI_NULL_INTEGER)
123 sig->MinSize = 0;
124 sig->MaxSize = MSI_RecordGetInteger(row,6);
125 if (sig->MaxSize == MSI_NULL_INTEGER)
126 sig->MaxSize = 0;
127 sig->Languages = msi_dup_record_field(row,9);
128 time = MSI_RecordGetInteger(row,7);
129 if (time != MSI_NULL_INTEGER)
130 DosDateTimeToFileTime(HIWORD(time), LOWORD(time), &sig->MinTime);
131 time = MSI_RecordGetInteger(row,8);
132 if (time != MSI_NULL_INTEGER)
133 DosDateTimeToFileTime(HIWORD(time), LOWORD(time), &sig->MaxTime);
134
135 TRACE("Found file name %s for Signature_ %s;\n",
136 debugstr_w(sig->File), debugstr_w(name));
137 TRACE("MinVersion is %d.%d.%d.%d\n", HIWORD(sig->MinVersionMS),
138 LOWORD(sig->MinVersionMS), HIWORD(sig->MinVersionLS),
139 LOWORD(sig->MinVersionLS));
140 TRACE("MaxVersion is %d.%d.%d.%d\n", HIWORD(sig->MaxVersionMS),
141 LOWORD(sig->MaxVersionMS), HIWORD(sig->MaxVersionLS),
142 LOWORD(sig->MaxVersionLS));
143 TRACE("MinSize is %d, MaxSize is %d;\n", sig->MinSize, sig->MaxSize);
144 TRACE("Languages is %s\n", debugstr_w(sig->Languages));
145
146 msiobj_release( &row->hdr );
147
148 return ERROR_SUCCESS;
149 }
150
151 /* Frees any memory allocated in sig */
152 static void ACTION_FreeSignature(MSISIGNATURE *sig)
153 {
154 msi_free(sig->File);
155 msi_free(sig->Languages);
156 }
157
158 static LPWSTR app_search_file(LPWSTR path, MSISIGNATURE *sig)
159 {
160 VS_FIXEDFILEINFO *info;
161 DWORD attr, handle, size;
162 LPWSTR val = NULL;
163 LPBYTE buffer;
164
165 if (!sig->File)
166 {
167 PathRemoveFileSpecW(path);
168 PathAddBackslashW(path);
169
170 attr = GetFileAttributesW(path);
171 if (attr != INVALID_FILE_ATTRIBUTES &&
172 (attr & FILE_ATTRIBUTE_DIRECTORY))
173 return strdupW(path);
174
175 return NULL;
176 }
177
178 attr = GetFileAttributesW(path);
179 if (attr == INVALID_FILE_ATTRIBUTES ||
180 (attr & FILE_ATTRIBUTE_DIRECTORY))
181 return NULL;
182
183 size = GetFileVersionInfoSizeW(path, &handle);
184 if (!size)
185 return strdupW(path);
186
187 buffer = msi_alloc(size);
188 if (!buffer)
189 return NULL;
190
191 if (!GetFileVersionInfoW(path, 0, size, buffer))
192 goto done;
193
194 if (!VerQueryValueW(buffer, szBackSlash, (LPVOID)&info, &size) || !info)
195 goto done;
196
197 if (sig->MinVersionLS || sig->MinVersionMS)
198 {
199 if (info->dwFileVersionMS < sig->MinVersionMS)
200 goto done;
201
202 if (info->dwFileVersionMS == sig->MinVersionMS &&
203 info->dwFileVersionLS < sig->MinVersionLS)
204 goto done;
205 }
206
207 if (sig->MaxVersionLS || sig->MaxVersionMS)
208 {
209 if (info->dwFileVersionMS > sig->MaxVersionMS)
210 goto done;
211
212 if (info->dwFileVersionMS == sig->MaxVersionMS &&
213 info->dwFileVersionLS > sig->MaxVersionLS)
214 goto done;
215 }
216
217 val = strdupW(path);
218
219 done:
220 msi_free(buffer);
221 return val;
222 }
223
224 static UINT ACTION_AppSearchComponents(MSIPACKAGE *package, LPWSTR *appValue, MSISIGNATURE *sig)
225 {
226 static const WCHAR query[] = {
227 'S','E','L','E','C','T',' ','*',' ',
228 'F','R','O','M',' ',
229 '`','C','o','m','p','L','o','c','a','t','o','r','`',' ',
230 'W','H','E','R','E',' ','`','S','i','g','n','a','t','u','r','e','_','`',' ','=',' ',
231 '\'','%','s','\'',0};
232 static const WCHAR sigquery[] = {
233 'S','E','L','E','C','T',' ','*',' ','F','R','O','M',' ',
234 '`','S','i','g','n','a','t','u','r','e','`',' ',
235 'W','H','E','R','E',' ','`','S','i','g','n','a','t','u','r','e','`',' ','=',' ',
236 '\'','%','s','\'',0};
237
238 MSIRECORD *row, *rec;
239 LPCWSTR signature, guid;
240 BOOL sigpresent = TRUE;
241 BOOL isdir;
242 UINT type;
243 WCHAR path[MAX_PATH];
244 DWORD size = MAX_PATH;
245 LPWSTR ptr;
246 DWORD attr;
247
248 TRACE("%s\n", debugstr_w(sig->Name));
249
250 *appValue = NULL;
251
252 row = MSI_QueryGetRecord(package->db, query, sig->Name);
253 if (!row)
254 {
255 TRACE("failed to query CompLocator for %s\n", debugstr_w(sig->Name));
256 return ERROR_SUCCESS;
257 }
258
259 signature = MSI_RecordGetString(row, 1);
260 guid = MSI_RecordGetString(row, 2);
261 type = MSI_RecordGetInteger(row, 3);
262
263 rec = MSI_QueryGetRecord(package->db, sigquery, signature);
264 if (!rec)
265 sigpresent = FALSE;
266
267 *path = '\0';
268 MsiLocateComponentW(guid, path, &size);
269 if (!*path)
270 goto done;
271
272 attr = GetFileAttributesW(path);
273 if (attr == INVALID_FILE_ATTRIBUTES)
274 goto done;
275
276 isdir = (attr & FILE_ATTRIBUTE_DIRECTORY);
277
278 if (type != msidbLocatorTypeDirectory && sigpresent && !isdir)
279 {
280 *appValue = app_search_file(path, sig);
281 }
282 else if (!sigpresent && (type != msidbLocatorTypeDirectory || isdir))
283 {
284 if (type == msidbLocatorTypeFileName)
285 {
286 ptr = strrchrW(path, '\\');
287 *(ptr + 1) = '\0';
288 }
289 else
290 PathAddBackslashW(path);
291
292 *appValue = strdupW(path);
293 }
294 else if (sigpresent)
295 {
296 PathAddBackslashW(path);
297 lstrcatW(path, MSI_RecordGetString(rec, 2));
298
299 attr = GetFileAttributesW(path);
300 if (attr != INVALID_FILE_ATTRIBUTES &&
301 !(attr & FILE_ATTRIBUTE_DIRECTORY))
302 *appValue = strdupW(path);
303 }
304
305 done:
306 if (rec) msiobj_release(&rec->hdr);
307 msiobj_release(&row->hdr);
308 return ERROR_SUCCESS;
309 }
310
311 static void ACTION_ConvertRegValue(DWORD regType, const BYTE *value, DWORD sz,
312 LPWSTR *appValue)
313 {
314 static const WCHAR dwordFmt[] = { '#','%','d','\0' };
315 static const WCHAR binPre[] = { '#','x','\0' };
316 static const WCHAR binFmt[] = { '%','0','2','X','\0' };
317 LPWSTR ptr;
318 DWORD i;
319
320 switch (regType)
321 {
322 case REG_SZ:
323 if (*(LPCWSTR)value == '#')
324 {
325 /* escape leading pound with another */
326 *appValue = msi_alloc(sz + sizeof(WCHAR));
327 (*appValue)[0] = '#';
328 strcpyW(*appValue + 1, (LPCWSTR)value);
329 }
330 else
331 {
332 *appValue = msi_alloc(sz);
333 strcpyW(*appValue, (LPCWSTR)value);
334 }
335 break;
336 case REG_DWORD:
337 /* 7 chars for digits, 1 for NULL, 1 for #, and 1 for sign
338 * char if needed
339 */
340 *appValue = msi_alloc(10 * sizeof(WCHAR));
341 sprintfW(*appValue, dwordFmt, *(const DWORD *)value);
342 break;
343 case REG_EXPAND_SZ:
344 sz = ExpandEnvironmentStringsW((LPCWSTR)value, NULL, 0);
345 *appValue = msi_alloc(sz * sizeof(WCHAR));
346 ExpandEnvironmentStringsW((LPCWSTR)value, *appValue, sz);
347 break;
348 case REG_BINARY:
349 /* #x<nibbles>\0 */
350 *appValue = msi_alloc((sz * 2 + 3) * sizeof(WCHAR));
351 lstrcpyW(*appValue, binPre);
352 ptr = *appValue + lstrlenW(binPre);
353 for (i = 0; i < sz; i++, ptr += 2)
354 sprintfW(ptr, binFmt, value[i]);
355 break;
356 default:
357 WARN("unimplemented for values of type %d\n", regType);
358 *appValue = NULL;
359 }
360 }
361
362 static UINT ACTION_SearchDirectory(MSIPACKAGE *package, MSISIGNATURE *sig,
363 LPCWSTR path, int depth, LPWSTR *appValue);
364
365 static UINT ACTION_AppSearchReg(MSIPACKAGE *package, LPWSTR *appValue, MSISIGNATURE *sig)
366 {
367 static const WCHAR query[] = {
368 's','e','l','e','c','t',' ','*',' ',
369 'f','r','o','m',' ',
370 'R','e','g','L','o','c','a','t','o','r',' ',
371 'w','h','e','r','e',' ',
372 'S','i','g','n','a','t','u','r','e','_',' ','=',' ', '\'','%','s','\'',0};
373 LPWSTR keyPath = NULL, valueName = NULL;
374 LPWSTR deformatted = NULL;
375 LPWSTR ptr = NULL, end;
376 int root, type;
377 HKEY rootKey, key = NULL;
378 DWORD sz = 0, regType;
379 LPBYTE value = NULL;
380 MSIRECORD *row;
381 UINT rc;
382
383 TRACE("%s\n", debugstr_w(sig->Name));
384
385 *appValue = NULL;
386
387 row = MSI_QueryGetRecord( package->db, query, sig->Name );
388 if (!row)
389 {
390 TRACE("failed to query RegLocator for %s\n", debugstr_w(sig->Name));
391 return ERROR_SUCCESS;
392 }
393
394 root = MSI_RecordGetInteger(row,2);
395 keyPath = msi_dup_record_field(row,3);
396 valueName = msi_dup_record_field(row,4);
397 type = MSI_RecordGetInteger(row,5);
398
399 deformat_string(package, keyPath, &deformatted);
400
401 switch (root)
402 {
403 case msidbRegistryRootClassesRoot:
404 rootKey = HKEY_CLASSES_ROOT;
405 break;
406 case msidbRegistryRootCurrentUser:
407 rootKey = HKEY_CURRENT_USER;
408 break;
409 case msidbRegistryRootLocalMachine:
410 rootKey = HKEY_LOCAL_MACHINE;
411 break;
412 case msidbRegistryRootUsers:
413 rootKey = HKEY_USERS;
414 break;
415 default:
416 WARN("Unknown root key %d\n", root);
417 goto end;
418 }
419
420 rc = RegOpenKeyW(rootKey, deformatted, &key);
421 if (rc)
422 {
423 TRACE("RegOpenKeyW returned %d\n", rc);
424 goto end;
425 }
426
427 rc = RegQueryValueExW(key, valueName, NULL, NULL, NULL, &sz);
428 if (rc)
429 {
430 TRACE("RegQueryValueExW returned %d\n", rc);
431 goto end;
432 }
433 /* FIXME: sanity-check sz before allocating (is there an upper-limit
434 * on the value of a property?)
435 */
436 value = msi_alloc( sz );
437 rc = RegQueryValueExW(key, valueName, NULL, &regType, value, &sz);
438 if (rc)
439 {
440 TRACE("RegQueryValueExW returned %d\n", rc);
441 goto end;
442 }
443
444 /* bail out if the registry key is empty */
445 if (sz == 0)
446 goto end;
447
448 if ((regType == REG_SZ || regType == REG_EXPAND_SZ) &&
449 (ptr = strchrW((LPWSTR)value, '"')) && (end = strchrW(++ptr, '"')))
450 *end = '\0';
451 else
452 ptr = (LPWSTR)value;
453
454 switch (type & 0x0f)
455 {
456 case msidbLocatorTypeDirectory:
457 rc = ACTION_SearchDirectory(package, sig, ptr, 0, appValue);
458 break;
459 case msidbLocatorTypeFileName:
460 *appValue = app_search_file(ptr, sig);
461 break;
462 case msidbLocatorTypeRawValue:
463 ACTION_ConvertRegValue(regType, value, sz, appValue);
464 break;
465 default:
466 FIXME("AppSearch unimplemented for type %d (key path %s, value %s)\n",
467 type, debugstr_w(keyPath), debugstr_w(valueName));
468 }
469 end:
470 msi_free( value );
471 RegCloseKey( key );
472
473 msi_free( keyPath );
474 msi_free( valueName );
475 msi_free( deformatted );
476
477 msiobj_release(&row->hdr);
478
479 return ERROR_SUCCESS;
480 }
481
482 static LPWSTR get_ini_field(LPWSTR buf, int field)
483 {
484 LPWSTR beg, end;
485 int i = 1;
486
487 if (field == 0)
488 return strdupW(buf);
489
490 beg = buf;
491 while ((end = strchrW(beg, ',')) && i < field)
492 {
493 beg = end + 1;
494 while (*beg && *beg == ' ')
495 beg++;
496
497 i++;
498 }
499
500 end = strchrW(beg, ',');
501 if (!end)
502 end = beg + lstrlenW(beg);
503
504 *end = '\0';
505 return strdupW(beg);
506 }
507
508 static UINT ACTION_AppSearchIni(MSIPACKAGE *package, LPWSTR *appValue,
509 MSISIGNATURE *sig)
510 {
511 static const WCHAR query[] = {
512 's','e','l','e','c','t',' ','*',' ',
513 'f','r','o','m',' ',
514 'I','n','i','L','o','c','a','t','o','r',' ',
515 'w','h','e','r','e',' ',
516 'S','i','g','n','a','t','u','r','e','_',' ','=',' ','\'','%','s','\'',0};
517 MSIRECORD *row;
518 LPWSTR fileName, section, key;
519 int field, type;
520 WCHAR buf[MAX_PATH];
521
522 TRACE("%s\n", debugstr_w(sig->Name));
523
524 *appValue = NULL;
525
526 row = MSI_QueryGetRecord( package->db, query, sig->Name );
527 if (!row)
528 {
529 TRACE("failed to query IniLocator for %s\n", debugstr_w(sig->Name));
530 return ERROR_SUCCESS;
531 }
532
533 fileName = msi_dup_record_field(row, 2);
534 section = msi_dup_record_field(row, 3);
535 key = msi_dup_record_field(row, 4);
536 field = MSI_RecordGetInteger(row, 5);
537 type = MSI_RecordGetInteger(row, 6);
538 if (field == MSI_NULL_INTEGER)
539 field = 0;
540 if (type == MSI_NULL_INTEGER)
541 type = 0;
542
543 GetPrivateProfileStringW(section, key, NULL, buf, MAX_PATH, fileName);
544 if (buf[0])
545 {
546 switch (type & 0x0f)
547 {
548 case msidbLocatorTypeDirectory:
549 ACTION_SearchDirectory(package, sig, buf, 0, appValue);
550 break;
551 case msidbLocatorTypeFileName:
552 *appValue = app_search_file(buf, sig);
553 break;
554 case msidbLocatorTypeRawValue:
555 *appValue = get_ini_field(buf, field);
556 break;
557 }
558 }
559
560 msi_free(fileName);
561 msi_free(section);
562 msi_free(key);
563
564 msiobj_release(&row->hdr);
565
566 return ERROR_SUCCESS;
567 }
568
569 /* Expands the value in src into a path without property names and only
570 * containing long path names into dst. Replaces at most len characters of dst,
571 * and always NULL-terminates dst if dst is not NULL and len >= 1.
572 * May modify src.
573 * Assumes src and dst are non-overlapping.
574 * FIXME: return code probably needed:
575 * - what does AppSearch return if the table values are invalid?
576 * - what if dst is too small?
577 */
578 static void ACTION_ExpandAnyPath(MSIPACKAGE *package, WCHAR *src, WCHAR *dst,
579 size_t len)
580 {
581 WCHAR *ptr, *deformatted;
582
583 if (!src || !dst || !len)
584 {
585 if (dst) *dst = '\0';
586 return;
587 }
588
589 dst[0] = '\0';
590
591 /* Ignore the short portion of the path */
592 if ((ptr = strchrW(src, '|')))
593 ptr++;
594 else
595 ptr = src;
596
597 deformat_string(package, ptr, &deformatted);
598 if (!deformatted || strlenW(deformatted) > len - 1)
599 {
600 msi_free(deformatted);
601 return;
602 }
603
604 lstrcpyW(dst, deformatted);
605 dst[lstrlenW(deformatted)] = '\0';
606 msi_free(deformatted);
607 }
608
609 /* Sets *matches to whether the file (whose path is filePath) matches the
610 * versions set in sig.
611 * Return ERROR_SUCCESS in case of success (whether or not the file matches),
612 * something else if an install-halting error occurs.
613 */
614 static UINT ACTION_FileVersionMatches(const MSISIGNATURE *sig, LPCWSTR filePath,
615 BOOL *matches)
616 {
617 UINT rc = ERROR_SUCCESS;
618
619 *matches = FALSE;
620 if (sig->Languages)
621 {
622 FIXME(": need to check version for languages %s\n",
623 debugstr_w(sig->Languages));
624 }
625 else
626 {
627 DWORD zero, size = GetFileVersionInfoSizeW(filePath, &zero);
628
629 if (size)
630 {
631 LPVOID buf = msi_alloc( size);
632
633 if (buf)
634 {
635 UINT versionLen;
636 LPVOID subBlock = NULL;
637
638 if (GetFileVersionInfoW(filePath, 0, size, buf))
639 VerQueryValueW(buf, szBackSlash, &subBlock, &versionLen);
640 if (subBlock)
641 {
642 VS_FIXEDFILEINFO *info = subBlock;
643
644 TRACE("Comparing file version %d.%d.%d.%d:\n",
645 HIWORD(info->dwFileVersionMS),
646 LOWORD(info->dwFileVersionMS),
647 HIWORD(info->dwFileVersionLS),
648 LOWORD(info->dwFileVersionLS));
649 if (info->dwFileVersionMS < sig->MinVersionMS
650 || (info->dwFileVersionMS == sig->MinVersionMS &&
651 info->dwFileVersionLS < sig->MinVersionLS))
652 {
653 TRACE("Less than minimum version %d.%d.%d.%d\n",
654 HIWORD(sig->MinVersionMS),
655 LOWORD(sig->MinVersionMS),
656 HIWORD(sig->MinVersionLS),
657 LOWORD(sig->MinVersionLS));
658 }
659 else if ((sig->MaxVersionMS || sig->MaxVersionLS) &&
660 (info->dwFileVersionMS > sig->MaxVersionMS ||
661 (info->dwFileVersionMS == sig->MaxVersionMS &&
662 info->dwFileVersionLS > sig->MaxVersionLS)))
663 {
664 TRACE("Greater than maximum version %d.%d.%d.%d\n",
665 HIWORD(sig->MaxVersionMS),
666 LOWORD(sig->MaxVersionMS),
667 HIWORD(sig->MaxVersionLS),
668 LOWORD(sig->MaxVersionLS));
669 }
670 else
671 *matches = TRUE;
672 }
673 msi_free( buf);
674 }
675 else
676 rc = ERROR_OUTOFMEMORY;
677 }
678 }
679 return rc;
680 }
681
682 /* Sets *matches to whether the file in findData matches that in sig.
683 * fullFilePath is assumed to be the full path of the file specified in
684 * findData, which may be necessary to compare the version.
685 * Return ERROR_SUCCESS in case of success (whether or not the file matches),
686 * something else if an install-halting error occurs.
687 */
688 static UINT ACTION_FileMatchesSig(const MSISIGNATURE *sig,
689 const WIN32_FIND_DATAW *findData, LPCWSTR fullFilePath, BOOL *matches)
690 {
691 UINT rc = ERROR_SUCCESS;
692
693 *matches = TRUE;
694 /* assumes the caller has already ensured the filenames match, so check
695 * the other fields..
696 */
697 if (sig->MinTime.dwLowDateTime || sig->MinTime.dwHighDateTime)
698 {
699 if (findData->ftCreationTime.dwHighDateTime <
700 sig->MinTime.dwHighDateTime ||
701 (findData->ftCreationTime.dwHighDateTime == sig->MinTime.dwHighDateTime
702 && findData->ftCreationTime.dwLowDateTime <
703 sig->MinTime.dwLowDateTime))
704 *matches = FALSE;
705 }
706 if (*matches && (sig->MaxTime.dwLowDateTime || sig->MaxTime.dwHighDateTime))
707 {
708 if (findData->ftCreationTime.dwHighDateTime >
709 sig->MaxTime.dwHighDateTime ||
710 (findData->ftCreationTime.dwHighDateTime == sig->MaxTime.dwHighDateTime
711 && findData->ftCreationTime.dwLowDateTime >
712 sig->MaxTime.dwLowDateTime))
713 *matches = FALSE;
714 }
715 if (*matches && sig->MinSize && findData->nFileSizeLow < sig->MinSize)
716 *matches = FALSE;
717 if (*matches && sig->MaxSize && findData->nFileSizeLow > sig->MaxSize)
718 *matches = FALSE;
719 if (*matches && (sig->MinVersionMS || sig->MinVersionLS ||
720 sig->MaxVersionMS || sig->MaxVersionLS))
721 rc = ACTION_FileVersionMatches(sig, fullFilePath, matches);
722 return rc;
723 }
724
725 /* Recursively searches the directory dir for files that match the signature
726 * sig, up to (depth + 1) levels deep. That is, if depth is 0, it searches dir
727 * (and only dir). If depth is 1, searches dir and its immediate
728 * subdirectories.
729 * Assumes sig->File is not NULL.
730 * Returns ERROR_SUCCESS on success (which may include non-critical errors),
731 * something else on failures which should halt the install.
732 */
733 static UINT ACTION_RecurseSearchDirectory(MSIPACKAGE *package, LPWSTR *appValue,
734 MSISIGNATURE *sig, LPCWSTR dir, int depth)
735 {
736 HANDLE hFind;
737 WIN32_FIND_DATAW findData;
738 UINT rc = ERROR_SUCCESS;
739 size_t dirLen = lstrlenW(dir), fileLen = lstrlenW(sig->File);
740 WCHAR subpath[MAX_PATH];
741 WCHAR *buf;
742
743 static const WCHAR starDotStarW[] = { '*','.','*',0 };
744
745 TRACE("Searching directory %s for file %s, depth %d\n", debugstr_w(dir),
746 debugstr_w(sig->File), depth);
747
748 if (depth < 0)
749 return ERROR_SUCCESS;
750
751 *appValue = NULL;
752 /* We need the buffer in both paths below, so go ahead and allocate it
753 * here. Add two because we might need to add a backslash if the dir name
754 * isn't backslash-terminated.
755 */
756 buf = msi_alloc( (dirLen + max(fileLen, strlenW(starDotStarW)) + 2) * sizeof(WCHAR));
757 if (!buf)
758 return ERROR_OUTOFMEMORY;
759
760 lstrcpyW(buf, dir);
761 PathAddBackslashW(buf);
762 lstrcatW(buf, sig->File);
763
764 hFind = FindFirstFileW(buf, &findData);
765 if (hFind != INVALID_HANDLE_VALUE)
766 {
767 if (!(findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
768 {
769 BOOL matches;
770
771 rc = ACTION_FileMatchesSig(sig, &findData, buf, &matches);
772 if (rc == ERROR_SUCCESS && matches)
773 {
774 TRACE("found file, returning %s\n", debugstr_w(buf));
775 *appValue = buf;
776 }
777 }
778 FindClose(hFind);
779 }
780
781 if (rc == ERROR_SUCCESS && !*appValue)
782 {
783 lstrcpyW(buf, dir);
784 PathAddBackslashW(buf);
785 lstrcatW(buf, starDotStarW);
786
787 hFind = FindFirstFileW(buf, &findData);
788 if (hFind != INVALID_HANDLE_VALUE)
789 {
790 if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY &&
791 lstrcmpW(findData.cFileName, szDot) &&
792 lstrcmpW(findData.cFileName, szDotDot))
793 {
794 lstrcpyW(subpath, dir);
795 PathAppendW(subpath, findData.cFileName);
796 rc = ACTION_RecurseSearchDirectory(package, appValue, sig,
797 subpath, depth - 1);
798 }
799
800 while (rc == ERROR_SUCCESS && !*appValue &&
801 FindNextFileW(hFind, &findData) != 0)
802 {
803 if (!lstrcmpW(findData.cFileName, szDot) ||
804 !lstrcmpW(findData.cFileName, szDotDot))
805 continue;
806
807 lstrcpyW(subpath, dir);
808 PathAppendW(subpath, findData.cFileName);
809 if (findData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
810 rc = ACTION_RecurseSearchDirectory(package, appValue,
811 sig, subpath, depth - 1);
812 }
813
814 FindClose(hFind);
815 }
816 }
817
818 if (!*appValue)
819 msi_free(buf);
820
821 return rc;
822 }
823
824 static UINT ACTION_CheckDirectory(MSIPACKAGE *package, LPCWSTR dir,
825 LPWSTR *appValue)
826 {
827 DWORD attr = GetFileAttributesW(dir);
828
829 if (attr != INVALID_FILE_ATTRIBUTES && (attr & FILE_ATTRIBUTE_DIRECTORY))
830 {
831 TRACE("directory exists, returning %s\n", debugstr_w(dir));
832 *appValue = strdupW(dir);
833 }
834
835 return ERROR_SUCCESS;
836 }
837
838 static BOOL ACTION_IsFullPath(LPCWSTR path)
839 {
840 WCHAR first = toupperW(path[0]);
841 BOOL ret;
842
843 if (first >= 'A' && first <= 'Z' && path[1] == ':')
844 ret = TRUE;
845 else if (path[0] == '\\' && path[1] == '\\')
846 ret = TRUE;
847 else
848 ret = FALSE;
849 return ret;
850 }
851
852 static UINT ACTION_SearchDirectory(MSIPACKAGE *package, MSISIGNATURE *sig,
853 LPCWSTR path, int depth, LPWSTR *appValue)
854 {
855 UINT rc;
856 DWORD attr;
857 LPWSTR val = NULL;
858
859 TRACE("%p, %p, %s, %d, %p\n", package, sig, debugstr_w(path), depth,
860 appValue);
861
862 if (ACTION_IsFullPath(path))
863 {
864 if (sig->File)
865 rc = ACTION_RecurseSearchDirectory(package, &val, sig, path, depth);
866 else
867 {
868 /* Recursively searching a directory makes no sense when the
869 * directory to search is the thing you're trying to find.
870 */
871 rc = ACTION_CheckDirectory(package, path, &val);
872 }
873 }
874 else
875 {
876 WCHAR pathWithDrive[MAX_PATH] = { 'C',':','\\',0 };
877 DWORD drives = GetLogicalDrives();
878 int i;
879
880 rc = ERROR_SUCCESS;
881 for (i = 0; rc == ERROR_SUCCESS && !val && i < 26; i++)
882 {
883 if (!(drives & (1 << i)))
884 continue;
885
886 pathWithDrive[0] = 'A' + i;
887 if (GetDriveTypeW(pathWithDrive) != DRIVE_FIXED)
888 continue;
889
890 lstrcpynW(pathWithDrive + 3, path,
891 sizeof(pathWithDrive) / sizeof(pathWithDrive[0]) - 3);
892
893 if (sig->File)
894 rc = ACTION_RecurseSearchDirectory(package, &val, sig,
895 pathWithDrive, depth);
896 else
897 rc = ACTION_CheckDirectory(package, pathWithDrive, &val);
898 }
899 }
900
901 attr = GetFileAttributesW(val);
902 if (attr != INVALID_FILE_ATTRIBUTES &&
903 (attr & FILE_ATTRIBUTE_DIRECTORY) &&
904 val && val[lstrlenW(val) - 1] != '\\')
905 {
906 val = msi_realloc(val, (lstrlenW(val) + 2) * sizeof(WCHAR));
907 if (!val)
908 rc = ERROR_OUTOFMEMORY;
909 else
910 PathAddBackslashW(val);
911 }
912
913 *appValue = val;
914
915 TRACE("returning %d\n", rc);
916 return rc;
917 }
918
919 static UINT ACTION_AppSearchSigName(MSIPACKAGE *package, LPCWSTR sigName,
920 MSISIGNATURE *sig, LPWSTR *appValue);
921
922 static UINT ACTION_AppSearchDr(MSIPACKAGE *package, LPWSTR *appValue, MSISIGNATURE *sig)
923 {
924 static const WCHAR query[] = {
925 's','e','l','e','c','t',' ','*',' ',
926 'f','r','o','m',' ',
927 'D','r','L','o','c','a','t','o','r',' ',
928 'w','h','e','r','e',' ',
929 'S','i','g','n','a','t','u','r','e','_',' ','=',' ', '\'','%','s','\'',0};
930 LPWSTR parentName = NULL, parent = NULL;
931 WCHAR path[MAX_PATH];
932 WCHAR expanded[MAX_PATH];
933 MSIRECORD *row;
934 int depth;
935 DWORD sz, attr;
936 UINT rc;
937
938 TRACE("%s\n", debugstr_w(sig->Name));
939
940 *appValue = NULL;
941
942 row = MSI_QueryGetRecord( package->db, query, sig->Name );
943 if (!row)
944 {
945 TRACE("failed to query DrLocator for %s\n", debugstr_w(sig->Name));
946 return ERROR_SUCCESS;
947 }
948
949 /* check whether parent is set */
950 parentName = msi_dup_record_field(row,2);
951 if (parentName)
952 {
953 MSISIGNATURE parentSig;
954
955 rc = ACTION_AppSearchSigName(package, parentName, &parentSig, &parent);
956 ACTION_FreeSignature(&parentSig);
957 msi_free(parentName);
958 }
959
960 sz = MAX_PATH;
961 MSI_RecordGetStringW(row, 3, path, &sz);
962
963 if (MSI_RecordIsNull(row,4))
964 depth = 0;
965 else
966 depth = MSI_RecordGetInteger(row,4);
967
968 if (sz)
969 ACTION_ExpandAnyPath(package, path, expanded, MAX_PATH);
970 else
971 strcpyW(expanded, path);
972
973 if (parent)
974 {
975 attr = GetFileAttributesW(parent);
976 if (attr != INVALID_FILE_ATTRIBUTES &&
977 !(attr & FILE_ATTRIBUTE_DIRECTORY))
978 {
979 PathRemoveFileSpecW(parent);
980 PathAddBackslashW(parent);
981 }
982
983 strcpyW(path, parent);
984 strcatW(path, expanded);
985 }
986 else if (sz)
987 strcpyW(path, expanded);
988
989 PathAddBackslashW(path);
990
991 rc = ACTION_SearchDirectory(package, sig, path, depth, appValue);
992
993 msi_free(parent);
994 msiobj_release(&row->hdr);
995
996 TRACE("returning %d\n", rc);
997 return rc;
998 }
999
1000 static UINT ACTION_AppSearchSigName(MSIPACKAGE *package, LPCWSTR sigName,
1001 MSISIGNATURE *sig, LPWSTR *appValue)
1002 {
1003 UINT rc;
1004
1005 *appValue = NULL;
1006 rc = ACTION_AppSearchGetSignature(package, sig, sigName);
1007 if (rc == ERROR_SUCCESS)
1008 {
1009 rc = ACTION_AppSearchComponents(package, appValue, sig);
1010 if (rc == ERROR_SUCCESS && !*appValue)
1011 {
1012 rc = ACTION_AppSearchReg(package, appValue, sig);
1013 if (rc == ERROR_SUCCESS && !*appValue)
1014 {
1015 rc = ACTION_AppSearchIni(package, appValue, sig);
1016 if (rc == ERROR_SUCCESS && !*appValue)
1017 rc = ACTION_AppSearchDr(package, appValue, sig);
1018 }
1019 }
1020 }
1021 return rc;
1022 }
1023
1024 static UINT iterate_appsearch(MSIRECORD *row, LPVOID param)
1025 {
1026 MSIPACKAGE *package = param;
1027 LPWSTR propName, sigName, value = NULL;
1028 MSISIGNATURE sig;
1029 UINT r;
1030
1031 /* get property and signature */
1032 propName = msi_dup_record_field(row,1);
1033 sigName = msi_dup_record_field(row,2);
1034
1035 TRACE("%s %s\n", debugstr_w(propName), debugstr_w(sigName));
1036
1037 r = ACTION_AppSearchSigName(package, sigName, &sig, &value);
1038 if (value)
1039 {
1040 MSI_SetPropertyW(package, propName, value);
1041 msi_free(value);
1042 }
1043 ACTION_FreeSignature(&sig);
1044 msi_free(propName);
1045 msi_free(sigName);
1046
1047 return r;
1048 }
1049
1050 UINT ACTION_AppSearch(MSIPACKAGE *package)
1051 {
1052 static const WCHAR query[] = {
1053 's','e','l','e','c','t',' ','*',' ',
1054 'f','r','o','m',' ',
1055 'A','p','p','S','e','a','r','c','h',0};
1056 MSIQUERY *view = NULL;
1057 UINT r;
1058
1059 r = MSI_OpenQuery( package->db, &view, query );
1060 if (r != ERROR_SUCCESS)
1061 return ERROR_SUCCESS;
1062
1063 r = MSI_IterateRecords( view, NULL, iterate_appsearch, package );
1064 msiobj_release( &view->hdr );
1065
1066 return r;
1067 }
1068
1069 static UINT ITERATE_CCPSearch(MSIRECORD *row, LPVOID param)
1070 {
1071 MSIPACKAGE *package = param;
1072 LPCWSTR signature;
1073 LPWSTR value = NULL;
1074 MSISIGNATURE sig;
1075 UINT r = ERROR_SUCCESS;
1076
1077 static const WCHAR success[] = {'C','C','P','_','S','u','c','c','e','s','s',0};
1078
1079 signature = MSI_RecordGetString(row, 1);
1080
1081 TRACE("%s\n", debugstr_w(signature));
1082
1083 ACTION_AppSearchSigName(package, signature, &sig, &value);
1084 if (value)
1085 {
1086 TRACE("Found signature %s\n", debugstr_w(signature));
1087 MSI_SetPropertyW(package, success, szOne);
1088 msi_free(value);
1089 r = ERROR_NO_MORE_ITEMS;
1090 }
1091
1092 ACTION_FreeSignature(&sig);
1093
1094 return r;
1095 }
1096
1097 UINT ACTION_CCPSearch(MSIPACKAGE *package)
1098 {
1099 static const WCHAR query[] = {
1100 's','e','l','e','c','t',' ','*',' ',
1101 'f','r','o','m',' ',
1102 'C','C','P','S','e','a','r','c','h',0};
1103 MSIQUERY *view = NULL;
1104 UINT r;
1105
1106 r = MSI_OpenQuery(package->db, &view, query);
1107 if (r != ERROR_SUCCESS)
1108 return ERROR_SUCCESS;
1109
1110 r = MSI_IterateRecords(view, NULL, ITERATE_CCPSearch, package);
1111 msiobj_release(&view->hdr);
1112
1113 return r;
1114 }