Add WindowsBuildNumber variable which is set to dwBuildNumber from RtlGetVersion()

1 files changed, 5 insertions, 0 deletions

diff --git a/src/engine/variable.cpp b/src/engine/variable.cpp
index dc5a569a..6322942e 100644
--- a/src/engine/variable.cpp
+++ b/src/engine/variable.cpp
@@ -36,6 +36,7 @@ enum OS_INFO_VARIABLE
     OS_INFO_VARIABLE_CompatibilityMode,
     OS_INFO_VARIABLE_TerminalServer,
     OS_INFO_VARIABLE_ProcessorArchitecture,
+    OS_INFO_VARIABLE_WindowsBuildNumber,
 };
 
 enum SET_VARIABLE
@@ -253,6 +254,7 @@ extern "C" HRESULT VariableInitialize(
         {L"VersionMsi", InitializeVariableVersionMsi, 0},
         {L"VersionNT", InitializeVariableVersionNT, OS_INFO_VARIABLE_VersionNT},
         {L"VersionNT64", InitializeVariableVersionNT, OS_INFO_VARIABLE_VersionNT64},
+        {L"WindowsBuildNumber", InitializeVariableVersionNT, OS_INFO_VARIABLE_WindowsBuildNumber},
         {L"WindowsFolder", InitializeVariableCsidlFolder, CSIDL_WINDOWS},
         {L"WindowsVolume", InitializeVariableWindowsVolumeFolder, 0},
         {BURN_BUNDLE_ACTION, InitializeVariableNumeric, 0, FALSE, TRUE},
@@ -1680,6 +1682,9 @@ static HRESULT InitializeVariableVersionNT(
             }
         }
         break;
+    case OS_INFO_VARIABLE_WindowsBuildNumber:
+        value.qwValue = static_cast<DWORD64>(ovix.dwBuildNumber);
+        value.Type = BURN_VARIANT_TYPE_NUMERIC;
     default:
         AssertSz(FALSE, "Unknown OS info type.");
         break;