Add RemoveRegistryKey

3 files changed, 117 insertions, 0 deletions

diff --git a/src/ca/RemoveRegistryKeysEx.cpp b/src/ca/RemoveRegistryKeysEx.cpp
new file mode 100644
index 00000000..478c0779
--- /dev/null
+++ b/src/ca/RemoveRegistryKeysEx.cpp
@@ -0,0 +1,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 `Wix4RemoveRegistryKeyEx`, `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);
+}
diff --git a/src/ca/utilca.def b/src/ca/utilca.def
index 97d5776f..337c3a68 100644
--- a/src/ca/utilca.def
+++ b/src/ca/utilca.def
@@ -35,6 +35,8 @@ EXPORTS
     WixSilentExec64
 ; RemoveFoldersEx.cpp
     WixRemoveFoldersEx
+; RemoveRegistryKeysEx.cpp
+    WixRemoveRegistryKeysEx
 ;scaexec.cpp
     RegisterPerfCounterData
     UnregisterPerfCounterData
diff --git a/src/ca/utilca.vcxproj b/src/ca/utilca.vcxproj
index 4fafba68..6a076f2f 100644
--- a/src/ca/utilca.vcxproj
+++ b/src/ca/utilca.vcxproj
@@ -59,6 +59,7 @@
     <ClCompile Include="OsInfo.cpp" />
     <ClCompile Include="qtexecca.cpp" />
     <ClCompile Include="RemoveFoldersEx.cpp" />
+    <ClCompile Include="RemoveRegistryKeysEx.cpp" />
     <ClCompile Include="RestartManager.cpp" />
     <ClCompile Include="scaexec.cpp" />
     <ClCompile Include="scamanifest.cpp" />