Enforce schema restrictions for numeric and image thmutil attributes.

1 files changed, 17 insertions, 2 deletions

diff --git a/src/libs/dutil/WixToolset.DUtil/inc/xmlutil.h b/src/libs/dutil/WixToolset.DUtil/inc/xmlutil.h
index ba92ada9..227c54f3 100644
--- a/src/libs/dutil/WixToolset.DUtil/inc/xmlutil.h
+++ b/src/libs/dutil/WixToolset.DUtil/inc/xmlutil.h
@@ -106,10 +106,25 @@ HRESULT DAPI XmlGetAttributeNumberBase(
     __in int nBase,
     __out DWORD* pdwValue
     );
-HRESULT DAPI XmlGetAttributeLargeNumber(
+HRESULT DAPI XmlGetAttributeUInt16(
     __in IXMLDOMNode* pixnNode,
     __in_z LPCWSTR pwzAttribute,
-    __out DWORD64* pdw64Value
+    __out WORD* pwValue
+    );
+HRESULT DAPI XmlGetAttributeInt32(
+    __in IXMLDOMNode* pixnNode,
+    __in_z LPCWSTR pwzAttribute,
+    __out int* piValue
+    );
+HRESULT DAPI XmlGetAttributeUInt32(
+    __in IXMLDOMNode* pixnNode,
+    __in_z LPCWSTR pwzAttribute,
+    __out DWORD* pdwValue
+    );
+HRESULT DAPI XmlGetAttributeUInt64(
+    __in IXMLDOMNode* pixnNode,
+    __in_z LPCWSTR pwzAttribute,
+    __out DWORD64* pqwValue
     );
 HRESULT DAPI XmlGetNamedItem(
     __in IXMLDOMNamedNodeMap *pixnmAttributes,