mempool-generic.cpp revision 4110da50b47fcdcce178bdd9e4c6e198469d9de9
20593760b116c90f3e439552763eef632a3bbb17vboxsync/* $Id$ */
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** @file
20593760b116c90f3e439552763eef632a3bbb17vboxsync * IPRT - Memory Allocation Pool.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/*
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Copyright (C) 2009 Sun Microsystems, Inc.
20593760b116c90f3e439552763eef632a3bbb17vboxsync *
20593760b116c90f3e439552763eef632a3bbb17vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
20593760b116c90f3e439552763eef632a3bbb17vboxsync * available from http://www.virtualbox.org. This file is free software;
20593760b116c90f3e439552763eef632a3bbb17vboxsync * you can redistribute it and/or modify it under the terms of the GNU
20593760b116c90f3e439552763eef632a3bbb17vboxsync * General Public License (GPL) as published by the Free Software
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
20593760b116c90f3e439552763eef632a3bbb17vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
20593760b116c90f3e439552763eef632a3bbb17vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
20593760b116c90f3e439552763eef632a3bbb17vboxsync *
20593760b116c90f3e439552763eef632a3bbb17vboxsync * The contents of this file may alternatively be used under the terms
20593760b116c90f3e439552763eef632a3bbb17vboxsync * of the Common Development and Distribution License Version 1.0
20593760b116c90f3e439552763eef632a3bbb17vboxsync * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20593760b116c90f3e439552763eef632a3bbb17vboxsync * VirtualBox OSE distribution, in which case the provisions of the
20593760b116c90f3e439552763eef632a3bbb17vboxsync * CDDL are applicable instead of those of the GPL.
20593760b116c90f3e439552763eef632a3bbb17vboxsync *
20593760b116c90f3e439552763eef632a3bbb17vboxsync * You may elect to license modified versions of this file under the
20593760b116c90f3e439552763eef632a3bbb17vboxsync * terms and conditions of either the GPL or the CDDL or both.
20593760b116c90f3e439552763eef632a3bbb17vboxsync *
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Clara, CA 95054 USA or visit http://www.sun.com if you need
20593760b116c90f3e439552763eef632a3bbb17vboxsync * additional information or have any questions.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/*******************************************************************************
20593760b116c90f3e439552763eef632a3bbb17vboxsync* Header Files *
20593760b116c90f3e439552763eef632a3bbb17vboxsync*******************************************************************************/
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/mempool.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/asm.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/assert.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/err.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/mem.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/spinlock.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include <iprt/string.h>
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync#include "internal/magics.h"
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/*******************************************************************************
20593760b116c90f3e439552763eef632a3bbb17vboxsync* Structures and Typedefs *
20593760b116c90f3e439552763eef632a3bbb17vboxsync*******************************************************************************/
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** Pointer to a memory pool instance. */
20593760b116c90f3e439552763eef632a3bbb17vboxsynctypedef struct RTMEMPOOLINT *PRTMEMPOOLINT;
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** Pointer to a memory pool entry. */
20593760b116c90f3e439552763eef632a3bbb17vboxsynctypedef struct RTMEMPOOLENTRY *PRTMEMPOOLENTRY;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/**
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Memory pool entry.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsynctypedef struct RTMEMPOOLENTRY
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Pointer to the pool */
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Pointer to the next entry. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY volatile pNext;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Pointer to the previous entry. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY volatile pPrev;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** The number of references to the pool entry. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync uint32_t volatile cRefs;
20593760b116c90f3e439552763eef632a3bbb17vboxsync} RTMEMPOOLENTRY;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/**
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Memory pool instance data.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsynctypedef struct RTMEMPOOLINT
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Magic number (RTMEMPOOL_MAGIC). */
20593760b116c90f3e439552763eef632a3bbb17vboxsync uint32_t u32Magic;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Spinlock protecting the pool entry list updates. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSPINLOCK hSpinLock;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** Head entry pointer. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY volatile pHead;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** The number of entries in the pool (for statistical purposes). */
20593760b116c90f3e439552763eef632a3bbb17vboxsync uint32_t volatile cEntries;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** User data assoicated with the pool. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync void *pvUser;
20593760b116c90f3e439552763eef632a3bbb17vboxsync /** The pool name. (variable length) */
20593760b116c90f3e439552763eef632a3bbb17vboxsync char szName[8];
20593760b116c90f3e439552763eef632a3bbb17vboxsync} RTMEMPOOLINT;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/*******************************************************************************
20593760b116c90f3e439552763eef632a3bbb17vboxsync* Defined Constants And Macros *
20593760b116c90f3e439552763eef632a3bbb17vboxsync*******************************************************************************/
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** Validates a memory pool handle, translating RTMEMPOOL_DEFAULT when found,
20593760b116c90f3e439552763eef632a3bbb17vboxsync * and returns rc if not valid. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync#define RTMEMPOOL_VALID_RETURN_RC(pMemPool, rc) \
20593760b116c90f3e439552763eef632a3bbb17vboxsync do { \
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pMemPool == RTMEMPOOL_DEFAULT) \
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool = &g_rtMemPoolDefault; \
20593760b116c90f3e439552763eef632a3bbb17vboxsync else \
20593760b116c90f3e439552763eef632a3bbb17vboxsync { \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtrReturn((pMemPool), (rc)); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertReturn((pMemPool)->u32Magic == RTMEMPOOL_MAGIC, (rc)); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync } \
20593760b116c90f3e439552763eef632a3bbb17vboxsync } while (0)
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** Validates a memory pool entry and returns rc if not valid. */
20593760b116c90f3e439552763eef632a3bbb17vboxsync#define RTMEMPOOL_VALID_ENTRY_RETURN_RC(pEntry, rc) \
20593760b116c90f3e439552763eef632a3bbb17vboxsync do { \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtrReturn(pEntry, (rc)); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtrNullReturn((pEntry)->pMemPool, (rc)); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertReturn((pEntry)->pMemPool->u32Magic == RTMEMPOOL_MAGIC, (rc)); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtrNull((pEntry)->pNext); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtrNull((pEntry)->pPrev); \
20593760b116c90f3e439552763eef632a3bbb17vboxsync } while (0)
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync/*******************************************************************************
20593760b116c90f3e439552763eef632a3bbb17vboxsync* Global Variables *
20593760b116c90f3e439552763eef632a3bbb17vboxsync*******************************************************************************/
20593760b116c90f3e439552763eef632a3bbb17vboxsync/** The */
20593760b116c90f3e439552763eef632a3bbb17vboxsyncstatic RTMEMPOOLINT g_rtMemPoolDefault =
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .u32Magic = */ RTMEMPOOL_MAGIC,
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .hSpinLock = */ NIL_RTSPINLOCK,
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .pHead = */ NULL,
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .cEntries = */ 0,
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .pvUser = */ NULL,
20593760b116c90f3e439552763eef632a3bbb17vboxsync /* .szName = */ "default"
20593760b116c90f3e439552763eef632a3bbb17vboxsync};
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(int) RTMemPoolCreate(PRTMEMPOOL phMemPool, const char *pszName)
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtr(phMemPool);
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertPtr(pszName);
20593760b116c90f3e439552763eef632a3bbb17vboxsync Assert(*pszName);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync size_t cchName = strlen(pszName);
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = (PRTMEMPOOLINT)RTMemAlloc(RT_OFFSETOF(RTMEMPOOLINT, szName[cchName + 1]));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pMemPool)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return VERR_NO_MEMORY;
20593760b116c90f3e439552763eef632a3bbb17vboxsync int rc = RTSpinlockCreate(&pMemPool->hSpinLock);
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (RT_SUCCESS(rc))
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->u32Magic = RTMEMPOOL_MAGIC;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->pHead = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->cEntries = 0;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->pvUser = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync memcpy(pMemPool->szName, pszName, cchName);
20593760b116c90f3e439552763eef632a3bbb17vboxsync *phMemPool = pMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync return VINF_SUCCESS;
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMemFree(pMemPool);
20593760b116c90f3e439552763eef632a3bbb17vboxsync return rc;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(int) RTMemPoolDestroy(RTMEMPOOL hMemPool)
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync if (hMemPool == NIL_RTMEMPOOL)
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync return VINF_SUCCESS;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync PRTMEMPOOLINT pMemPool = hMemPool;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync RTMEMPOOL_VALID_RETURN_RC(pMemPool, VERR_INVALID_HANDLE);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync if (pMemPool == &g_rtMemPoolDefault)
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync return VINF_SUCCESS;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync /*
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync * Invalidate the handle and free all associated resources.
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync */
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync ASMAtomicWriteU32(&pMemPool->u32Magic, RTMEMPOOL_MAGIC_DEAD);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync int rc = RTSpinlockDestroy(pMemPool->hSpinLock); AssertRC(rc);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pMemPool->hSpinLock = NIL_RTSPINLOCK;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync PRTMEMPOOLENTRY pEntry = pMemPool->pHead;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pMemPool->pHead = NULL;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync while (pEntry)
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync {
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync PRTMEMPOOLENTRY pFree = pEntry;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pEntry = pEntry->pNext;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pFree->pMemPool = NULL;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pFree->pNext = NULL;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pFree->pPrev = NULL;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pFree->cRefs = UINT32_MAX - 3;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync RTMemFree(pFree);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync }
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync RTMemFree(pMemPool);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync return VINF_SUCCESS;
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncDECLINLINE(void) rtMemPoolInitAndLink(PRTMEMPOOLINT pMemPool, PRTMEMPOOLENTRY pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pMemPool = pMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pNext = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pPrev = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->cRefs = 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pMemPool->hSpinLock != NIL_RTSPINLOCK)
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSPINLOCKTMP Tmp = RTSPINLOCKTMP_INITIALIZER;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSpinlockAcquire(pMemPool->hSpinLock, &Tmp);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pHead = pMemPool->pHead;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pNext = pHead;
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pHead)
20593760b116c90f3e439552763eef632a3bbb17vboxsync pHead->pPrev = pEntry;
20593760b116c90f3e439552763eef632a3bbb17vboxsync else
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->pHead = pEntry;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSpinlockRelease(pMemPool->hSpinLock, &Tmp);
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync ASMAtomicIncU32(&pMemPool->cEntries);
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncDECLINLINE(void) rtMemPoolUnlink(PRTMEMPOOLENTRY pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = pEntry->pMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pMemPool->hSpinLock != NIL_RTSPINLOCK)
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSPINLOCKTMP Tmp = RTSPINLOCKTMP_INITIALIZER;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSpinlockAcquire(pMemPool->hSpinLock, &Tmp);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pEntry->pNext)
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pNext->pPrev = pEntry->pPrev;
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (pEntry->pPrev)
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pPrev->pNext = pEntry->pNext;
20593760b116c90f3e439552763eef632a3bbb17vboxsync else
20593760b116c90f3e439552763eef632a3bbb17vboxsync pMemPool->pHead = pEntry->pNext;
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pMemPool = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTSpinlockRelease(pMemPool->hSpinLock, &Tmp);
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync else
20593760b116c90f3e439552763eef632a3bbb17vboxsync pEntry->pMemPool = NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync ASMAtomicDecU32(&pMemPool->cEntries);
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void *) RTMemPoolAlloc(RTMEMPOOL hMemPool, size_t cb) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = hMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_RETURN_RC(pMemPool, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)RTMemAlloc(cb + sizeof(*pEntry));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pMemPool, pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return pEntry + 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void *) RTMemPoolAllocZ(RTMEMPOOL hMemPool, size_t cb) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = hMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_RETURN_RC(pMemPool, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)RTMemAllocZ(cb + sizeof(*pEntry));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pMemPool, pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return pEntry + 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void *) RTMemPoolDup(RTMEMPOOL hMemPool, const void *pvSrc, size_t cb) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = hMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_RETURN_RC(pMemPool, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)RTMemAlloc(cb + sizeof(*pEntry));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync memcpy(pEntry + 1, pvSrc, cb);
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pMemPool, pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return pEntry + 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void *) RTMemPoolDupEx(RTMEMPOOL hMemPool, const void *pvSrc, size_t cbSrc, size_t cbExtra) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pMemPool = hMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_RETURN_RC(pMemPool, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)RTMemAlloc(cbSrc + cbExtra + sizeof(*pEntry));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync memcpy(pEntry + 1, pvSrc, cbSrc);
20593760b116c90f3e439552763eef632a3bbb17vboxsync memset((uint8_t *)(pEntry + 1) + cbSrc, '\0', cbExtra);
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pMemPool, pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return pEntry + 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void *) RTMemPoolRealloc(RTMEMPOOL hMemPool, void *pvOld, size_t cbNew) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync /*
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Fend off the odd cases.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!cbNew)
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMemPoolRelease(hMemPool, pvOld);
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pvOld)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return RTMemPoolAlloc(hMemPool, cbNew);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync /*
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Real realloc.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pNewMemPool = hMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_RETURN_RC(pNewMemPool, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pOldEntry = (PRTMEMPOOLENTRY)pvOld - 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_ENTRY_RETURN_RC(pOldEntry, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLINT pOldMemPool = pOldEntry->pMemPool;
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertReturn(pOldEntry->cRefs == 1, NULL);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync /*
20593760b116c90f3e439552763eef632a3bbb17vboxsync * Unlink it from the current pool and try reallocate it.
20593760b116c90f3e439552763eef632a3bbb17vboxsync */
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolUnlink(pOldEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)RTMemRealloc(pOldEntry, cbNew + sizeof(*pEntry));
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pEntry)
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pOldMemPool, pOldEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync return NULL;
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolInitAndLink(pNewMemPool, pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return pEntry + 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(void) RTMemPoolFree(RTMEMPOOL hMemPool, void *pv) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMemPoolRelease(hMemPool, pv);
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(uint32_t) RTMemPoolRetain(void *pv) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)pv - 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_ENTRY_RETURN_RC(pEntry, UINT32_MAX);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync uint32_t cRefs = ASMAtomicIncU32(&pEntry->cRefs);
20593760b116c90f3e439552763eef632a3bbb17vboxsync Assert(cRefs < UINT32_MAX / 2);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return cRefs;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsyncRTDECL(uint32_t) RTMemPoolRelease(RTMEMPOOL hMemPool, void *pv) RT_NO_THROW
20593760b116c90f3e439552763eef632a3bbb17vboxsync{
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!pv)
20593760b116c90f3e439552763eef632a3bbb17vboxsync return 0;
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync PRTMEMPOOLENTRY pEntry = (PRTMEMPOOLENTRY)pv - 1;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMEMPOOL_VALID_ENTRY_RETURN_RC(pEntry, UINT32_MAX);
20593760b116c90f3e439552763eef632a3bbb17vboxsync Assert( hMemPool == NIL_RTMEMPOOL
20593760b116c90f3e439552763eef632a3bbb17vboxsync || hMemPool == pEntry->pMemPool
20593760b116c90f3e439552763eef632a3bbb17vboxsync || (hMemPool == RTMEMPOOL_DEFAULT && pEntry->pMemPool == &g_rtMemPoolDefault));
20593760b116c90f3e439552763eef632a3bbb17vboxsync AssertReturn(pEntry->cRefs > 0, UINT32_MAX);
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync uint32_t cRefs = ASMAtomicDecU32(&pEntry->cRefs);
20593760b116c90f3e439552763eef632a3bbb17vboxsync Assert(cRefs < UINT32_MAX / 2);
20593760b116c90f3e439552763eef632a3bbb17vboxsync if (!cRefs)
20593760b116c90f3e439552763eef632a3bbb17vboxsync {
20593760b116c90f3e439552763eef632a3bbb17vboxsync rtMemPoolUnlink(pEntry);
4110da50b47fcdcce178bdd9e4c6e198469d9de9vboxsync pEntry->cRefs = UINT32_MAX - 2;
20593760b116c90f3e439552763eef632a3bbb17vboxsync RTMemFree(pEntry);
20593760b116c90f3e439552763eef632a3bbb17vboxsync }
20593760b116c90f3e439552763eef632a3bbb17vboxsync
20593760b116c90f3e439552763eef632a3bbb17vboxsync return cRefs;
20593760b116c90f3e439552763eef632a3bbb17vboxsync}
20593760b116c90f3e439552763eef632a3bbb17vboxsync