Write EngineVersion, ProtocolVersion, and Win64 to the Burn manifest.

1 files changed, 50 insertions, 0 deletions

diff --git a/src/burn/engine/manifest.cpp b/src/burn/engine/manifest.cpp
index 7e16de13..0cd10dbb 100644
--- a/src/burn/engine/manifest.cpp
+++ b/src/burn/engine/manifest.cpp
@@ -7,6 +7,11 @@ static HRESULT ParseFromXml(
     __in IXMLDOMDocument* pixdDocument,
     __in BURN_ENGINE_STATE* pEngineState
     );
+#if DEBUG
+static void ValidateHarvestingAttributes(
+    __in IXMLDOMDocument* pixdDocument
+    );
+#endif
 
 // function definitions
 
@@ -43,6 +48,10 @@ extern "C" HRESULT ManifestLoadXmlFromBuffer(
     hr = XmlLoadDocumentFromBuffer(pbBuffer, cbBuffer, &pixdDocument);
     ExitOnFailure(hr, "Failed to load manifest as XML document.");
 
+#if DEBUG
+    ValidateHarvestingAttributes(pixdDocument);
+#endif
+
     hr = ParseFromXml(pixdDocument, pEngineState);
 
 LExit:
@@ -162,3 +171,44 @@ LExit:
     ReleaseObject(pixeBundle);
     return hr;
 }
+
+#if DEBUG
+static void ValidateHarvestingAttributes(
+    __in IXMLDOMDocument* pixdDocument
+    )
+{
+    HRESULT hr = S_OK;
+    IXMLDOMElement* pixeBundle = NULL;
+    LPWSTR sczEngineVersion = NULL;
+    DWORD dwProtocolVersion = 0;
+    BOOL fWin64 = FALSE;
+
+    hr = pixdDocument->get_documentElement(&pixeBundle);
+    ExitOnFailure(hr, "Failed to get document element.");
+
+    hr = XmlGetAttributeEx(pixeBundle, L"EngineVersion", &sczEngineVersion);
+    ExitOnRequiredXmlQueryFailure(hr, "Failed to get BurnManifest/@EngineVersion attribute.");
+
+    hr = XmlGetAttributeUInt32(pixeBundle, L"ProtocolVersion", &dwProtocolVersion);
+    ExitOnRequiredXmlQueryFailure(hr, "Failed to get BurnManifest/@ProtocolVersion attribute.");
+
+    hr = XmlGetYesNoAttribute(pixeBundle, L"Win64", &fWin64);
+    ExitOnRequiredXmlQueryFailure(hr, "Failed to get BurnManifest/@Win64 attribute.");
+
+    Assert(CSTR_EQUAL == ::CompareStringW(LOCALE_INVARIANT, 0, sczEngineVersion, -1, wzVerMajorMinorBuild, -1));
+
+    Assert(BURN_PROTOCOL_VERSION == dwProtocolVersion);
+
+#if !defined(_WIN64)
+    Assert(!fWin64);
+#else
+    Assert(fWin64);
+#endif
+
+LExit:
+    AssertSz(SUCCEEDED(hr), "Failed to get harvesting attributes.");
+
+    ReleaseStr(sczEngineVersion);
+    ReleaseObject(pixeBundle);
+}
+#endif