-/*
- * Copyright (c) 2000-2004,2011,2014 Apple Inc. All Rights Reserved.
- *
- * @APPLE_LICENSE_HEADER_START@
- *
- * This file contains Original Code and/or Modifications of Original Code
- * as defined in and that are subject to the Apple Public Source License
- * Version 2.0 (the 'License'). You may not use this file except in
- * compliance with the License. Please obtain a copy of the License at
- * http://www.opensource.apple.com/apsl/ and read it before using this
- * file.
- *
- * The 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.
- *
- * @APPLE_LICENSE_HEADER_END@
- */
-
-
-//
-// trackingallocator - Allocator that keeps track of allocations
-//
-#ifndef _H_TRACKINGALLOCATOR
-#define _H_TRACKINGALLOCATOR
-
-#include <security_utilities/alloc.h>
-#include <set>
-
-namespace Security
-{
-
-//
-// A Allocator that keeps track of allocations and can throw everything
-// away unless explicitly committed.
-//
-class TrackingAllocator : public Allocator
-{
-public:
- TrackingAllocator(Allocator &inAllocator) : subAllocator(inAllocator) {}
- virtual ~TrackingAllocator();
-
- Allocator &subAllocator;
-
- // standard Allocator operations
- void *malloc(size_t inSize) throw(std::bad_alloc);
- void free(void *inAddress) throw();
- void *realloc(void *inAddress, size_t inNewSize) throw(std::bad_alloc);
-
- // reset frees all memory; commit forgets about it all
- void reset();
- void commit();
-
-private:
- typedef std::set<void *> AllocSet;
- AllocSet mAllocSet;
-};
-
-} // end namespace Security
-
-#endif //_H_TRACKINGALLOCATOR