From 0f9931107ecf9e1f6714e6fd2cabc76d2ddb1153 Mon Sep 17 00:00:00 2001 From: Sean Hall Date: Thu, 26 May 2022 17:32:45 -0500 Subject: Add MemSizeChecked. --- src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp') diff --git a/src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp b/src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp index 09692bfb..520ed426 100644 --- a/src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp +++ b/src/libs/dutil/test/DUtilUnitTest/MemUtilTest.cpp @@ -23,7 +23,7 @@ namespace DutilTests void MemUtilAppendTest() { HRESULT hr = S_OK; - DWORD dwSize; + SIZE_T cbSize = 0; ArrayValue *rgValues = NULL; DWORD cValues = 0; @@ -65,8 +65,11 @@ namespace DutilTests // and make sure it doesn't grow since we already have enough space hr = MemEnsureArraySize(reinterpret_cast(&rgValues), cValues, sizeof(ArrayValue), 5); NativeAssert::Succeeded(hr, "Failed to ensure array size matches what it should already be"); - dwSize = MemSize(rgValues); - if (dwSize != 6 * sizeof(ArrayValue)) + + hr = MemSizeChecked(rgValues, &cbSize); + NativeAssert::Succeeded(hr, "Failed to get current array size"); + + if (cbSize != 6 * sizeof(ArrayValue)) { hr = E_FAIL; ExitOnFailure(hr, "MemEnsureArraySize is growing an array that is already big enough!"); -- cgit v1.2.3-55-g6feb