/* * Copyright (c) 2000-2001 Apple Computer, Inc. All Rights Reserved. * * The contents of this file constitute Original Code as defined in and are * subject to the Apple Public Source License Version 1.2 (the 'License'). * You may not use this file except in compliance with the License. Please obtain * a copy of the License at http://www.apple.com/publicsource and read it before * using this file. * * This Original Code and all software distributed under the License are * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS * OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, INCLUDING WITHOUT * LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR * PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. Please see the License for the * specific language governing rights and limitations under the License. */ // // trackingallocator - CssmAllocator that keeps track of allocations // #ifndef _H_TRACKINGALLOCATOR #define _H_TRACKINGALLOCATOR #include <Security/cssmalloc.h> #include <set> namespace Security { // // A CssmAllocator that keeps track of allocations and can throw everything // away unless explicitly committed. // class TrackingAllocator : public CssmAllocator { public: TrackingAllocator(CssmAllocator &inAllocator) : mAllocator(inAllocator) {} virtual ~TrackingAllocator(); void *malloc(size_t inSize) throw(std::bad_alloc) { void *anAddress = mAllocator.malloc(inSize); mAllocSet.insert(anAddress); return anAddress; } void free(void *inAddress) throw() { mAllocator.free(inAddress); mAllocSet.erase(inAddress); } void *realloc(void *inAddress, size_t inNewSize) throw(std::bad_alloc) { void *anAddress = mAllocator.realloc(inAddress, inNewSize); if (anAddress != inAddress) { mAllocSet.erase(inAddress); mAllocSet.insert(anAddress); } return anAddress; } void commit() { mAllocSet.clear(); } private: typedef std::set<void *> AllocSet; CssmAllocator &mAllocator; AllocSet mAllocSet; }; } // end namespace Security #endif //_H_TRACKINGALLOCATOR