WIXFEAT:5355 - Don't clear variable for missing key or value.

2 files changed, 8 insertions, 4 deletions

diff --git a/src/engine/search.cpp b/src/engine/search.cpp
index 065003c7..6d5f8d49 100644
--- a/src/engine/search.cpp
+++ b/src/engine/search.cpp
@@ -946,8 +946,7 @@ static HRESULT RegistrySearchValue(
     {
         // What if there is a hidden variable in sczKey?
         LogStringLine(REPORT_STANDARD, "Registry key not found. Key = '%ls'", sczKey);
-        hr = VariableSetVariant(pVariables, pSearch->sczVariable, &value);
-        ExitOnFailure(hr, "Failed to clear variable.");
+
         ExitFunction1(hr = S_OK);
     }
     ExitOnFailure(hr, "Failed to open registry key.");
@@ -958,8 +957,7 @@ static HRESULT RegistrySearchValue(
     {
         // What if there is a hidden variable in sczKey or sczValue?
         LogStringLine(REPORT_STANDARD, "Registry value not found. Key = '%ls', Value = '%ls'", sczKey, sczValue);
-        hr = VariableSetVariant(pVariables, pSearch->sczVariable, &value);
-        ExitOnFailure(hr, "Failed to clear variable.");
+
         ExitFunction1(hr = S_OK);
     }
     ExitOnWin32Error(er, hr, "Failed to query registry key value size.");
diff --git a/src/test/BurnUnitTest/SearchTest.cpp b/src/test/BurnUnitTest/SearchTest.cpp
index d868190d..eca01f5f 100644
--- a/src/test/BurnUnitTest/SearchTest.cpp
+++ b/src/test/BurnUnitTest/SearchTest.cpp
@@ -198,6 +198,8 @@ namespace Bootstrapper
 
                 VariableSetStringHelper(&variables, L"MyKey", L"SOFTWARE\\Microsoft\\WiX_Burn_UnitTest\\Value", FALSE);
                 VariableSetStringHelper(&variables, L"MyValue", L"String", FALSE);
+                VariableSetStringHelper(&variables, L"Variable27", L"Default27", FALSE);
+                VariableSetStringHelper(&variables, L"Variable28", L"Default28", FALSE);
 
                 LPCWSTR wzDocument =
                     L"<Bundle>"
@@ -227,6 +229,8 @@ namespace Bootstrapper
                     L"    <RegistrySearch Id='Search24' Type='exists' Root='HKU' Key='.DEFAULT\\System\\NetworkServiceSidSubkeyDoesNotExist' Variable='Variable24' />"
                     L"    <RegistrySearch Id='Search25' Type='value' Root='HKCR' Key='.msi' Variable='Variable25' VariableType='string' />"
                     L"    <RegistrySearch Id='Search26' Type='value' Root='HKCR' Key='.msi' Variable='Variable26' VariableType='formatted' />"
+                    L"    <RegistrySearch Id='Search27' Type='value' Root='HKCU' Key='SOFTWARE\\Microsoft\\WiX_Burn_UnitTest\\StringDoesNotExist' Value='String' Variable='Variable27' VariableType='string' />"
+                    L"    <RegistrySearch Id='Search28' Type='value' Root='HKCU' Key='SOFTWARE\\Microsoft\\WiX_Burn_UnitTest\\String' Value='DoesNotExist' Variable='Variable28' VariableType='string' />"
                     L"</Bundle>";
 
                 // load XML document
@@ -270,6 +274,8 @@ namespace Bootstrapper
                 Assert::Equal(0ll, VariableGetNumericHelper(&variables, L"Variable24"));
                 Assert::Equal<String^>(gcnew String(L"Msi.Package"), VariableGetStringHelper(&variables, L"Variable25"));
                 Assert::Equal<String^>(gcnew String(L"Msi.Package"), VariableGetStringHelper(&variables, L"Variable26"));
+                Assert::Equal<String^>(gcnew String(L"Default27"), VariableGetStringHelper(&variables, L"Variable27"));
+                Assert::Equal<String^>(gcnew String(L"Default28"), VariableGetStringHelper(&variables, L"Variable28"));
             }
             finally
             {