1/* 2 * Copyright (c) 2000-2004 Apple Computer, Inc. All Rights Reserved. 3 * 4 * @APPLE_LICENSE_HEADER_START@ 5 * 6 * This file contains Original Code and/or Modifications of Original Code 7 * as defined in and that are subject to the Apple Public Source License 8 * Version 2.0 (the 'License'). You may not use this file except in 9 * compliance with the License. Please obtain a copy of the License at 10 * http://www.opensource.apple.com/apsl/ and read it before using this 11 * file. 12 * 13 * The Original Code and all software distributed under the License are 14 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER 15 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, 16 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, 17 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. 18 * Please see the License for the specific language governing rights and 19 * limitations under the License. 20 * 21 * @APPLE_LICENSE_HEADER_END@ 22 */ 23 24 25// 26// cssmalloc - memory allocation in the CDSA world. 27// 28// Don't eat heavily before inspecting this code. 29// 30#include <security_utilities/trackingallocator.h> 31 32 33// 34// The default action of the destructor is to free all memory. 35// 36TrackingAllocator::~TrackingAllocator() 37{ 38 reset(); 39} 40 41 42// 43// Standard allocation operations. 44// We pass them down to our subAllocator and keep track of what we've got. 45// 46void *TrackingAllocator::malloc(size_t inSize) throw(std::bad_alloc) 47{ 48 void *anAddress = subAllocator.malloc(inSize); 49 mAllocSet.insert(anAddress); 50 return anAddress; 51} 52 53void TrackingAllocator::free(void *inAddress) throw() 54{ 55 subAllocator.free(inAddress); 56 mAllocSet.erase(inAddress); 57} 58 59void *TrackingAllocator::realloc(void *inAddress, size_t inNewSize) throw(std::bad_alloc) 60{ 61 void *anAddress = subAllocator.realloc(inAddress, inNewSize); 62 if (anAddress != inAddress) 63 { 64 mAllocSet.erase(inAddress); 65 mAllocSet.insert(anAddress); 66 } 67 68 return anAddress; 69} 70 71 72// 73// Free all memory allocated through this Allocator (since the last commit(), if any) 74// 75void TrackingAllocator::reset() 76{ 77 AllocSet::iterator first = mAllocSet.begin(), last = mAllocSet.end(); 78 for (; first != last; ++first) 79 subAllocator.free(*first); 80} 81 82 83// 84// Forget about all allocated memory. It's now your responsibility. 85// 86void TrackingAllocator::commit() 87{ 88 mAllocSet.clear(); 89} 90