1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
#include "precomp.h"
LPCWSTR vcsRemoveRegistryKeyExQuery =
L"SELECT `RemoveRegistryKeyEx`, `Component_`, `Root`, `Key`, `InstallMode`, `Condition` FROM `Wix4RemoveRegistryKeyEx`";
enum eRemoveRegistryKeyExQuery { rrxqId = 1, rrxqComponent, rrxqRoot, rrxqKey, rrxqMode, rrxqCondition };
extern "C" UINT WINAPI WixRemoveRegistryKeysEx(
__in MSIHANDLE hInstall
)
{
//AssertSz(FALSE, "debug WixRemoveRegistryKeyEx");
HRESULT hr = S_OK;
PMSIHANDLE hView;
PMSIHANDLE hRec;
LPWSTR sczId = NULL;
LPWSTR sczComponent = NULL;
LPWSTR sczCondition = NULL;
LPWSTR sczKey = NULL;
int iRoot = 0;
int iMode = 0;
MSIHANDLE hTable = NULL;
MSIHANDLE hColumns = NULL;
hr = WcaInitialize(hInstall, __FUNCTION__);
ExitOnFailure(hr, "Failed to initialize " __FUNCTION__);
// anything to do?
if (S_OK != WcaTableExists(L"Wix4RemoveRegistryKeyEx"))
{
WcaLog(LOGMSG_STANDARD, "Wix4RemoveRegistryKeyEx table doesn't exist, so there are no registry keys to remove.");
ExitFunction();
}
hr = WcaOpenExecuteView(vcsRemoveRegistryKeyExQuery, &hView);
ExitOnFailure(hr, "Failed to open view on Wix4RemoveRegistryKeyEx table");
while (S_OK == (hr = WcaFetchRecord(hView, &hRec)))
{
hr = WcaGetRecordString(hRec, rrxqId, &sczId);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx identity.");
hr = WcaGetRecordString(hRec, rrxqCondition, &sczCondition);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx condition.");
if (sczCondition && *sczCondition)
{
MSICONDITION condition = ::MsiEvaluateConditionW(hInstall, sczCondition);
if (MSICONDITION_TRUE == condition)
{
WcaLog(LOGMSG_STANDARD, "True condition for row %S: %S; processing.", sczId, sczCondition);
}
else
{
WcaLog(LOGMSG_STANDARD, "False or invalid condition for row %S: %S; skipping.", sczId, sczCondition);
continue;
}
}
hr = WcaGetRecordString(hRec, rrxqComponent, &sczComponent);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx component.");
hr = WcaGetRecordInteger(hRec, rrxqRoot, &iRoot);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx root.");
hr = WcaGetRecordString(hRec, rrxqKey, &sczKey);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx key.");
hr = WcaGetRecordInteger(hRec, rrxqMode, &iMode);
ExitOnFailure(hr, "Failed to get Wix4RemoveRegistryKeyEx mode.");
switch (iMode)
{
case 1: // remove on install
WcaLog(LOGMSG_STANDARD, "Adding RemoveRegistry row: %ls/%d/%ls/-/%ls", sczId, iRoot, sczKey, sczComponent);
hr = WcaAddTempRecord(&hTable, &hColumns, L"RemoveRegistry", NULL, 0, 5, sczId, iRoot, sczKey, L"-", sczComponent);
ExitOnFailure(hr, "Failed to add RemoveRegistry row for remove-on-install Wix4RemoveRegistryKeyEx row: %ls:", sczId);
break;
case 2: // remove on uninstall
WcaLog(LOGMSG_STANDARD, "Adding Registry row: %ls/%d/%ls/-/null/%ls", sczId, iRoot, sczKey, sczComponent);
hr = WcaAddTempRecord(&hTable, &hColumns, L"Registry", NULL, 0, 6, sczId, iRoot, sczKey, L"-", NULL, sczComponent);
ExitOnFailure(hr, "Failed to add Registry row for remove-on-uninstall Wix4RemoveRegistryKeyEx row: %ls:", sczId);
break;
}
}
// reaching the end of the list is actually a good thing, not an error
if (E_NOMOREITEMS == hr)
{
hr = S_OK;
}
ExitOnFailure(hr, "Failure occured while processing Wix4RemoveRegistryKeyEx table.");
LExit:
if (hColumns)
{
::MsiCloseHandle(hColumns);
}
if (hTable)
{
::MsiCloseHandle(hTable);
}
ReleaseStr(sczKey);
ReleaseStr(sczComponent);
ReleaseStr(sczCondition);
ReleaseStr(sczId);
DWORD er = SUCCEEDED(hr) ? ERROR_SUCCESS : ERROR_INSTALL_FAILURE;
return WcaFinalize(er);
}
|