From 259c6b7420c5e95258f563eca14859a182ffe76a Mon Sep 17 00:00:00 2001
From: Sean Hall <r.sean.hall@gmail.com>
Date: Sat, 23 May 2020 16:22:44 +1000
Subject: Add WindowsBuildNumber variable which is set to dwBuildNumber from
 RtlGetVersion()

---
 src/engine/variable.cpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'src/engine')

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;
-- 
cgit v1.2.3-55-g6feb