Runtime/testcase/tstRTMemCache.cpp

64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/* $Id$ */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** @file
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * IPRT Testcase - RTMemCache.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/*
e64031e20c39650a7bc902a3e1aba613b9415deevboxsync * Copyright (C) 2010 Oracle Corporation
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * available from http://www.virtualbox.org. This file is free software;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * you can redistribute it and/or modify it under the terms of the GNU
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * General Public License (GPL) as published by the Free Software
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * The contents of this file may alternatively be used under the terms
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * of the Common Development and Distribution License Version 1.0
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * VirtualBox OSE distribution, in which case the provisions of the
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * CDDL are applicable instead of those of the GPL.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * You may elect to license modified versions of this file under the
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * terms and conditions of either the GPL or the CDDL or both.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/*******************************************************************************
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*   Header Files                                                               *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*******************************************************************************/
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/memcache.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/asm.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/err.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/initterm.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/mem.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/param.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/rand.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/string.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/semaphore.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/test.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/time.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync#include <iprt/thread.h>
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/*******************************************************************************
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*   Structures and Typedefs                                                    *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*******************************************************************************/
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsynctypedef struct TST3THREAD
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTHREAD            hThread;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTSEMEVENTMULTI     hEvt;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint64_t volatile   cIterations;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint32_t            cbObject;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    bool                fUseCache;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync} TST3THREAD, *PTST3THREAD;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/*******************************************************************************
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*   Global Variables                                                           *
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync*******************************************************************************/
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** The test handle */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic RTTEST               g_hTest;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** Global mem cache handle for use in some of the testcases. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic RTMEMCACHE           g_hMemCache;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** Stop indicator for tst3 threads.  */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic bool volatile        g_fTst3Stop;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/**
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * Basic API checks.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * We'll return if any of these fails.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic void tst1(void)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTestISub("Basics");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Create one without constructor or destructor. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint32_t const cObjects = PAGE_SIZE * 2 / 256;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTMEMCACHE hMemCache;
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    RTTESTI_CHECK_RC_RETV(RTMemCacheCreate(&hMemCache, 256, cObjects, 32, NULL, NULL, NULL, 0 /*fFlags*/), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RETV(hMemCache != NIL_RTMEMCACHE);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Allocate a bit and free it again. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    void *pv = NULL;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC_RETV(RTMemCacheAllocEx(hMemCache, &pv), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RETV(pv != NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RETV(RT_ALIGN_P(pv, 32) == pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTMemCacheFree(hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK((pv = RTMemCacheAlloc(hMemCache)) != NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTMemCacheFree(hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Allocate everything and free it again, checking size constraints. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    for (uint32_t iLoop = 0; iLoop < 20; iLoop++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        /* Allocate everything. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        void *apv[cObjects];
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        for (uint32_t i = 0; i < cObjects; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            apv[i] = NULL;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTTESTI_CHECK_RC(RTMemCacheAllocEx(hMemCache, &apv[i]), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        /* Check that we've got it all. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        int rc;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTTESTI_CHECK_RC(rc = RTMemCacheAllocEx(hMemCache, &pv), VERR_MEM_CACHE_MAX_SIZE);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        if (RT_SUCCESS(rc))
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTMemCacheFree(hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTTESTI_CHECK((pv = RTMemCacheAlloc(hMemCache)) == NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTMemCacheFree(hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        /* Free all the allocations. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        for (uint32_t i = 0; i < cObjects; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTMemCacheFree(hMemCache, apv[i]);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTTESTI_CHECK((pv = RTMemCacheAlloc(hMemCache)) != NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTMemCacheFree(hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Destroy it. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC(RTMemCacheDestroy(hMemCache), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC(RTMemCacheDestroy(NIL_RTMEMCACHE), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** Constructor for tst2. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic DECLCALLBACK(int) tst2Ctor(RTMEMCACHE hMemCache, void *pvObj, void *pvUser)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK(hMemCache == g_hMemCache);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK(ASMMemIsAll8(pvObj, 256, 0) == NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    if (*(bool *)pvUser)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        return VERR_RESOURCE_BUSY;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    strcat((char *)pvObj, "ctor was called\n");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    return VINF_SUCCESS;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/** Destructor for tst2.  Checks that it was constructed and used twice. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic DECLCALLBACK(void) tst2Dtor(RTMEMCACHE hMemCache, void *pvObj, void *pvUser)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK(!strcmp((char *)pvObj, "ctor was called\nused\nused\n"));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    strcat((char *)pvObj, "dtor was called\n");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/**
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * Test constructor / destructor.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic void tst2(void)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTestISub("Ctor/Dtor");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Create one without constructor or destructor. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    bool            fFail    = false;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint32_t const  cObjects = PAGE_SIZE * 2 / 256;
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    RTTESTI_CHECK_RC_RETV(RTMemCacheCreate(&g_hMemCache, 256, cObjects, 32, tst2Ctor, tst2Dtor, &fFail, 0 /*fFlags*/), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* A failure run first. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    fFail = true;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    void *pv = (void *)0x42;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC_RETV(RTMemCacheAllocEx(g_hMemCache, &pv), VERR_RESOURCE_BUSY);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK(pv == (void *)0x42);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    fFail = false;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* To two rounds where we allocate all the objects and free them again. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    for (uint32_t iLoop = 0; iLoop < 2; iLoop++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        void *apv[cObjects];
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        for (uint32_t i = 0; i < cObjects; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            apv[i] = NULL;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTTESTI_CHECK_RC_RETV(RTMemCacheAllocEx(g_hMemCache, &apv[i]), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            if (iLoop == 0)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTTESTI_CHECK(!strcmp((char *)apv[i], "ctor was called\n"));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            else
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTTESTI_CHECK(!strcmp((char *)apv[i], "ctor was called\nused\n"));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            strcat((char *)apv[i], "used\n");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTTESTI_CHECK_RETV((pv = RTMemCacheAlloc(g_hMemCache)) == NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTMemCacheFree(g_hMemCache, pv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        for (uint32_t i = 0; i < cObjects; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            RTMemCacheFree(g_hMemCache, apv[i]);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* Cone, destroy the cache. */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC(RTMemCacheDestroy(g_hMemCache), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/**
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * Thread that allocates
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * @returns
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * @param   hThreadSelf         The thread.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * @param   pvArg               Pointer to fUseCache.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsyncstatic DECLCALLBACK(int) tst3Thread(RTTHREAD hThreadSelf, void *pvArg)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    PTST3THREAD     pThread     = (PTST3THREAD)(pvArg);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    size_t          cbObject    = pThread->cbObject;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint64_t        cIterations = 0;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* wait for the kick-off */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTEST_CHECK_RC_OK(g_hTest, RTSemEventMultiWait(pThread->hEvt, RT_INDEFINITE_WAIT));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* allocate and free loop */
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    if (pThread->fUseCache)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    {
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        while (!g_fTst3Stop)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        {
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync            void *apv[64];
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            for (unsigned i = 0; i < RT_ELEMENTS(apv); i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                apv[i] = RTMemCacheAlloc(g_hMemCache);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTTEST_CHECK(g_hTest, apv[i] != NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            for (unsigned i = 0; i < RT_ELEMENTS(apv); i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTMemCacheFree(g_hMemCache, apv[i]);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync            cIterations += RT_ELEMENTS(apv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        }
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    }
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    else
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    {
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        while (!g_fTst3Stop)
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        {
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync            void *apv[64];
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            for (unsigned i = 0; i < RT_ELEMENTS(apv); i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                apv[i] = RTMemAlloc(cbObject);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTTEST_CHECK(g_hTest, apv[i] != NULL);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync            for (unsigned i = 0; i < RT_ELEMENTS(apv); i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                RTMemFree(apv[i]);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync            cIterations += RT_ELEMENTS(apv);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* report back the status */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    pThread->cIterations = cIterations;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    return VINF_SUCCESS;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync/**
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync * Time constrained test with and unlimited  N threads.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync */
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsyncstatic void tst3(uint32_t cThreads, uint32_t cbObject, int iMethod, uint32_t cSecs)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    RTTestISubF("Benchmark - %u threads, %u bytes, %u secs, %s", cThreads, cbObject, cSecs,
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync                iMethod == 0 ? "RTMemCache"
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync                : "RTMemAlloc");
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /*
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     * Create a cache with unlimited space, a start semaphore and line up
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     * the threads.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     */
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    RTTESTI_CHECK_RC_RETV(RTMemCacheCreate(&g_hMemCache, cbObject, 0 /*cbAlignment*/, UINT32_MAX, NULL, NULL, NULL, 0 /*fFlags*/), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTSEMEVENTMULTI hEvt;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC_OK_RETV(RTSemEventMultiCreate(&hEvt));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    TST3THREAD aThreads[64];
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RETV(cThreads < RT_ELEMENTS(aThreads));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    ASMAtomicWriteBool(&g_fTst3Stop, false);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    for (uint32_t i = 0; i < cThreads; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    {
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        aThreads[i].hThread     = NIL_RTTHREAD;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        aThreads[i].cIterations = 0;
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        aThreads[i].fUseCache   = iMethod == 0;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        aThreads[i].cbObject    = cbObject;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        aThreads[i].hEvt        = hEvt;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTTESTI_CHECK_RC_OK_RETV(RTThreadCreateF(&aThreads[i].hThread, tst3Thread, &aThreads[i], 0,
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync                                                 RTTHREADTYPE_DEFAULT, RTTHREADFLAGS_WAITABLE, "tst3-%u", i));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /*
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     * Start the race.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTimeNanoTS(); /* warmup */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint64_t uStartTS = RTTimeNanoTS();
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC_OK_RETV(RTSemEventMultiSignal(hEvt));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTThreadSleep(cSecs * 1000);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    ASMAtomicWriteBool(&g_fTst3Stop, true);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    for (uint32_t i = 0; i < cThreads; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        RTTESTI_CHECK_RC_OK_RETV(RTThreadWait(aThreads[i].hThread, 60*1000, NULL));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint64_t cElapsedNS = RTTimeNanoTS() - uStartTS;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /*
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     * Sum up the counts.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    uint64_t cIterations = 0;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    for (uint32_t i = 0; i < cThreads; i++)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        cIterations += aThreads[i].cIterations;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
09e11be404479b989812ad9da85a16f3654ab320vboxsync    RTTestIPrintf(RTTESTLVL_ALWAYS, "%'8u iterations per second, %'llu ns on avg\n",
09e11be404479b989812ad9da85a16f3654ab320vboxsync                  (unsigned)((long double)cIterations * 1000000000.0 / cElapsedNS),
09e11be404479b989812ad9da85a16f3654ab320vboxsync                  cElapsedNS / cIterations);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /* clean up */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC(RTMemCacheDestroy(g_hMemCache), VINF_SUCCESS);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTESTI_CHECK_RC_OK(RTSemEventMultiDestroy(hEvt));
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsyncstatic void tst3AllMethods(uint32_t cThreads, uint32_t cbObject, uint32_t cSecs)
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync{
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    tst3(cThreads, cbObject, 0, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync    tst3(cThreads, cbObject, 1, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync}
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsyncint main(int argc, char **argv)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync{
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTEST hTest;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    int rc = RTTestInitAndCreate("tstRTMemCache", &hTest);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    if (rc)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync        return rc;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    RTTestBanner(hTest);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    g_hTest = hTest;
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    tst1();
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    tst2();
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    if (RTTestIErrorCount() == 0)
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    {
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        uint32_t cSecs = argc == 1 ? 5 : 2;
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        /*            threads, cbObj, cSecs */
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     1,   256, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     1,    32, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     1,     8, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     1,     2, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     1,     1, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,   256, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,   128, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,    64, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,    32, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,     2, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(     3,     1, cSecs);
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync
db2893e30b129fbadd201b8f56cd7bc1974b395dvboxsync        tst3AllMethods(    16,    32, cSecs);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    }
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    /*
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     * Summary.
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync     */
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync    return RTTestSummaryAndDestroy(hTest);
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync}
64babf0f32eaf36212d54af4a3ce5fe193b24825vboxsync