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