diff options
Diffstat (limited to 'src/VBox/Runtime/testcase/tstRTMemWipe.cpp')
-rw-r--r-- | src/VBox/Runtime/testcase/tstRTMemWipe.cpp | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/src/VBox/Runtime/testcase/tstRTMemWipe.cpp b/src/VBox/Runtime/testcase/tstRTMemWipe.cpp new file mode 100644 index 00000000..b259f5e7 --- /dev/null +++ b/src/VBox/Runtime/testcase/tstRTMemWipe.cpp @@ -0,0 +1,94 @@ +/* $Id: tstRTMemWipe.cpp $ */ +/** @file + * IPRT Testcase - RTMemWipe* functions. + */ + +/* + * Copyright (C) 2012-2019 Oracle Corporation + * + * This file is part of VirtualBox Open Source Edition (OSE), as + * available from http://www.virtualbox.org. This file is free software; + * you can redistribute it and/or modify it under the terms of the GNU + * General Public License (GPL) as published by the Free Software + * Foundation, in version 2 as it comes in the "COPYING" file of the + * VirtualBox OSE distribution. VirtualBox OSE is distributed in the + * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind. + * + * The contents of this file may alternatively be used under the terms + * of the Common Development and Distribution License Version 1.0 + * (CDDL) only, as it comes in the "COPYING.CDDL" file of the + * VirtualBox OSE distribution, in which case the provisions of the + * CDDL are applicable instead of those of the GPL. + * + * You may elect to license modified versions of this file under the + * terms and conditions of either the GPL or the CDDL or both. + */ + + +/********************************************************************************************************************************* +* Header Files * +*********************************************************************************************************************************/ +#include <iprt/path.h> +#include <iprt/rand.h> +#include <iprt/string.h> +#include <iprt/stream.h> +#include <iprt/initterm.h> +#include <iprt/mem.h> +#include <iprt/test.h> + + +/********************************************************************************************************************************* +* Global Variables * +*********************************************************************************************************************************/ + +static void doMemWipeThoroughly(RTTEST hTest) +{ + for (uint32_t p = 0; p < RTRandU32Ex(1, 64); p++) + { + uint32_t cbAlloc = RTRandS32Ex(1, _1M) * sizeof(uint8_t); + + RTTestPrintf(hTest, RTTESTLVL_ALWAYS, "Testing wipe #%.02RU32 (%u bytes) ...\n", + p + 1, cbAlloc); + + void *pvBuf = RTMemAlloc(cbAlloc); + if (!pvBuf) + { + RTTestIFailed("No memory for first buffer (%z bytes)\n", + cbAlloc); + continue; + } + RTRandBytes(pvBuf, cbAlloc); + + void *pvWipe = RTMemDup(pvBuf, cbAlloc); + if (!pvWipe) + { + RTMemFree(pvBuf); + + RTTestIFailed("No memory for second buffer (%z bytes)\n", + cbAlloc); + continue; + } + uint32_t cbWipeRand = RTRandU32Ex(1, cbAlloc); + RTMemWipeThoroughly(pvWipe, RT_MIN(cbAlloc, cbWipeRand), p /* Passes */); + if (!memcmp(pvWipe, pvBuf, cbAlloc)) + RTTestIFailed("Memory blocks must differ (%z bytes, 0x%p vs. 0x%p)!\n", + cbAlloc, pvWipe, pvBuf); + + RTMemFree(pvWipe); + RTMemFree(pvBuf); + } +} + +int main() +{ + RTTEST hTest; + RTEXITCODE rcExit = RTTestInitAndCreate("memwipe", &hTest); + if (rcExit != RTEXITCODE_SUCCESS) + return rcExit; + RTTestBanner(hTest); + + doMemWipeThoroughly(hTest); + + return RTTestSummaryAndDestroy(hTest); +} + |