5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync/* $Id$ */
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync/** @file
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * IPRT Testcase - MemPool.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync */
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync/*
c7814cf6e1240a519cbec0441e033d0e2470ed00vboxsync * Copyright (C) 2009-2010 Oracle Corporation
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync *
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * available from http://www.virtualbox.org. This file is free software;
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * you can redistribute it and/or modify it under the terms of the GNU
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * General Public License (GPL) as published by the Free Software
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync *
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * The contents of this file may alternatively be used under the terms
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * of the Common Development and Distribution License Version 1.0
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * VirtualBox OSE distribution, in which case the provisions of the
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * CDDL are applicable instead of those of the GPL.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync *
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * You may elect to license modified versions of this file under the
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * terms and conditions of either the GPL or the CDDL or both.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync */
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync/*******************************************************************************
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync* Header Files *
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync*******************************************************************************/
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync#include <iprt/mempool.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync#include <iprt/asm.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync#include <iprt/err.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync#include <iprt/initterm.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync#include <iprt/string.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync#include <iprt/test.h>
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync#include <iprt/thread.h>
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync#include <iprt/rand.h>
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/*******************************************************************************
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync* Global Variables *
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync*******************************************************************************/
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/** The test handle */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic RTTEST g_hTest;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/** Memory pool for tst4. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic RTMEMPOOL g_hMemPool4;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/**
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Basic API checks.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * We'll return if any of these fails.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic void tst1(RTMEMPOOL hMemPool)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync{
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync void *pv;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Normal alloc. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAlloc(hMemPool, 1));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAlloc(hMemPool, 0));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Zeroed allocation. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < 512; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAllocZ(hMemPool, 1024));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(ASMMemIsAllU32(pv, 1024, 0) == NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync memset(pv, 'a', 1024);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK_RETV(RTMemPoolRefCount(pv) == 1);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAllocZ(hMemPool, 0));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Duped allocation. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync static const char szTest[] = "test string abcdef";
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolDup(hMemPool, szTest, sizeof(szTest)));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(memcmp(pv, szTest, sizeof(szTest)) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < 512; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync size_t const cb = 256 - sizeof(szTest);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolDupEx(hMemPool, szTest, sizeof(szTest), cb));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(memcmp(pv, szTest, sizeof(szTest)) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(ASMMemIsAll8((uint8_t *)pv + sizeof(szTest), cb, 0) == NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync memset(pv, 'b', sizeof(szTest) + cb);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK_RETV(RTMemPoolRefCount(pv) == 1);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Reallocation */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolRealloc(hMemPool, NULL, 1));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolRealloc(hMemPool, pv, 2));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAlloc(hMemPool, 42));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolRealloc(hMemPool, pv, 32));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, pv) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolRealloc(hMemPool, NULL, 128));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolRealloc(hMemPool, pv, 256));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRealloc(hMemPool, pv, 0) == NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Free (a bit hard to test) */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMemPoolFree(hMemPool, NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMemPoolFree(hMemPool, RTMemPoolAlloc(hMemPool, 42));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* Memory referencing. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 1; i <= 4096; i *= 3)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync void *pv2;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv = RTMemPoolAlloc(hMemPool, i));
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 1);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync memset(pv, 'a', i);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 2);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 2);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 3);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 3);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 4);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 4);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRelease(hMemPool, pv) == 3);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 3);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 4);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 4);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 5);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 5);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRetain(pv) == 6);
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == 6);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRelease(NIL_RTMEMPOOL, pv) == 5);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRelease(NIL_RTMEMPOOL, pv) == 4);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t cRefs = 3;; cRefs--)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRelease(hMemPool, pv) == cRefs);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (cRefs == 0)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync break;
3f286f1a573fddc997ce03a37282536899ec4446vboxsync RTTESTI_CHECK(RTMemPoolRefCount(pv) == cRefs);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x cRefs=%d\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv, cRefs));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t j = 0; j < 42; j++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv2 = RTMemPoolAlloc(hMemPool, i));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(pv2 != pv);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync memset(pv2, 'f', i);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(RTMemPoolRelease(hMemPool, pv2) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_MSG_RETV((pv2 = ASMMemIsAll8(pv, i, 'a')) == NULL, ("i=%#x pv=%p off=%#x cRefs=%d\n", i, pv, (uintptr_t)pv2 - (uintptr_t)pv, cRefs));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync}
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/**
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Test automatic cleanup upon destruction.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic void tst3(void)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync{
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestISub("Destroy non-empty pool");
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /*
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Nothing freed.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMEMPOOL hMemPool;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_RETV(RTMemPoolCreate(&hMemPool, "test 3a"), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolAlloc(hMemPool, 10));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolAlloc(hMemPool, 20));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolAlloc(hMemPool, 40));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolAlloc(hMemPool, 80));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_RETV(RTMemPoolDestroy(hMemPool), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /*
ad27e1d5e48ca41245120c331cc88b50464813cevboxsync * Pseudo random freeing to test list maintenance.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTRAND hRand;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_OK_RETV(RTRandAdvCreateParkMiller(&hRand));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < 10; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_RETV(RTMemPoolCreate(&hMemPool, "test 3b"), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync void *apvHistory[256];
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RT_ZERO(apvHistory);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync uint32_t cBlocks = 0;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync uint32_t j;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (j = 0; j < RT_ELEMENTS(apvHistory) - i * 7; j++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(apvHistory[j] = RTMemPoolAlloc(hMemPool, j));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync memset(apvHistory[j], 'a', j);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync cBlocks++;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (RTRandAdvU32Ex(hRand, 0, 4) == 4)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync uint32_t iFree = RTRandAdvU32Ex(hRand, 0, j);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync cBlocks -= apvHistory[iFree] != NULL;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RETV(RTMemPoolRelease(hMemPool, apvHistory[iFree]) == 0);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync apvHistory[iFree] = NULL;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_RETV(RTMemPoolDestroy(hMemPool), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestIPrintf(RTTESTLVL_INFO, "cBlocks=%u j=%u\n", cBlocks, j);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTRandAdvDestroy(hRand);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync}
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/** Thread function for tst4. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic DECLCALLBACK(int) tst4Thread(RTTHREAD hSelf, void *pvArg)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync{
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync// uint32_t iThread = (uint32_t)(uintptr_t)pvArg;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMEMPOOL hMemPool = g_hMemPool4;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* setup. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestSetDefault(g_hTest, NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* wait for the kick-off */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTThreadUserWait(hSelf, RT_INDEFINITE_WAIT);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* do the work. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < 1024; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync void *apvHistory[256];
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RT_ZERO(apvHistory);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync uint32_t j;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (j = 0; j < RT_ELEMENTS(apvHistory) - (i % 200); j++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RET(apvHistory[j] = RTMemPoolAlloc(hMemPool, (i & 15) + (j & 63)), VERR_NO_MEMORY);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t k = i & 7; k < j; k += 3)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RET(RTMemPoolRelease(hMemPool, apvHistory[k]) == 0, VERR_INTERNAL_ERROR);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync apvHistory[k] = NULL;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync while (j-- > 0)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RET(RTMemPoolRelease(hMemPool, apvHistory[j]) == 0, VERR_INTERNAL_ERROR);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync return VINF_SUCCESS;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync}
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/** sub test */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic void tst4Sub(uint32_t cThreads)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync{
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestISubF("Serialization - %u threads", cThreads);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMEMPOOL hMemPool;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_RETV(RTMemPoolCreate(&hMemPool, "test 2a"), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync g_hMemPool4 = hMemPool;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync PRTTHREAD pahThreads = (PRTTHREAD)RTMemPoolAlloc(hMemPool, cThreads * sizeof(RTTHREAD));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK(pahThreads);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (pahThreads)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* start them. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < cThreads; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync int rc = RTThreadCreateF(&pahThreads[i], tst4Thread, (void *)(uintptr_t)i, 0,
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE, "tst4-%u/%u", i, cThreads);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_OK(rc);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (RT_FAILURE(rc))
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync pahThreads[i] = NIL_RTTHREAD;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTThreadYield();
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* kick them off. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < cThreads; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (pahThreads[i] != NIL_RTTHREAD)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_OK(RTThreadUserSignal(pahThreads[i]));
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /* wait for them. */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync for (uint32_t i = 0; i < cThreads; i++)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (pahThreads[i] != NIL_RTTHREAD)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync int rc = RTThreadWait(pahThreads[i], 2*60*1000, NULL);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC_OK(rc);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(RTMemPoolDestroy(hMemPool), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync}
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync/**
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Starts a bunch of threads beating on a pool to test serialization.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsyncstatic void tst4(void)
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync{
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /*
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Test it with a few different thread counts.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(1);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(2);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(3);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(4);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(8);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4Sub(16);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync}
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsyncint main()
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync{
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync RTTEST hTest;
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync int rc = RTTestInitAndCreate("tstRTMemPool", &hTest);
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync if (rc)
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync return rc;
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync RTTestBanner(hTest);
176991ff77f632fbe0e92e3aa17cf249367a309bvboxsync g_hTest = hTest;
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync /*
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * Smoke tests using first the default and then a custom pool.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestSub(hTest, "Smoke test on default pool");
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst1(RTMEMPOOL_DEFAULT);
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTestSub(hTest, "Smoke test on custom pool");
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTMEMPOOL hMemPool;
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolCreate(&hMemPool, "test 2a"), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (RT_SUCCESS(rc))
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolDestroy(hMemPool), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolDestroy(NIL_RTMEMPOOL), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolDestroy(RTMEMPOOL_DEFAULT), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolDestroy(RTMEMPOOL_DEFAULT), VINF_SUCCESS);
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolCreate(&hMemPool, "test 2b"), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync if (RT_SUCCESS(rc))
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync {
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst1(hMemPool);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync RTTESTI_CHECK_RC(rc = RTMemPoolDestroy(hMemPool), VINF_SUCCESS);
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync }
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync /*
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync * Further tests.
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync */
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst3();
8ca5eef802d34e34f85919c5d39b510eeab3de3bvboxsync tst4();
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync /*
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync * Summary.
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync */
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync return RTTestSummaryAndDestroy(hTest);
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync}
5ff501b3f5f8ba5172a6d8ad1a2a99b103a5d058vboxsync