+++ /dev/null
-/*
- * 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