X-Git-Url: https://git.saurik.com/apple/security.git/blobdiff_plain/80e2389990082500d76eb566d4946be3e786c3ef..d8f41ccd20de16f8ebe2ccc84d47bf1cb2b26bbb:/Security/libsecurity_utilities/lib/trackingallocator.cpp

diff --git a/Security/libsecurity_utilities/lib/trackingallocator.cpp b/Security/libsecurity_utilities/lib/trackingallocator.cpp
new file mode 100644
index 00000000..33650037
--- /dev/null
+++ b/Security/libsecurity_utilities/lib/trackingallocator.cpp
@@ -0,0 +1,89 @@
+/*
+ * 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@
+ */
+
+
+//
+// cssmalloc - memory allocation in the CDSA world.
+//
+// Don't eat heavily before inspecting this code.
+//
+#include <security_utilities/trackingallocator.h>
+
+
+//
+// The default action of the destructor is to free all memory.
+//
+TrackingAllocator::~TrackingAllocator()
+{
+	reset();
+}
+
+
+//
+// Standard allocation operations.
+// We pass them down to our subAllocator and keep track of what we've got.
+//
+void *TrackingAllocator::malloc(size_t inSize) throw(std::bad_alloc)
+{
+	void *anAddress = subAllocator.malloc(inSize);
+	mAllocSet.insert(anAddress);
+	return anAddress;
+}
+
+void TrackingAllocator::free(void *inAddress) throw()
+{
+	subAllocator.free(inAddress);
+	mAllocSet.erase(inAddress);
+}
+
+void *TrackingAllocator::realloc(void *inAddress, size_t inNewSize) throw(std::bad_alloc)
+{
+	void *anAddress = subAllocator.realloc(inAddress, inNewSize);
+	if (anAddress != inAddress)
+	{
+		mAllocSet.erase(inAddress);
+		mAllocSet.insert(anAddress);
+	}
+
+	return anAddress;
+}
+
+
+//
+// Free all memory allocated through this Allocator (since the last commit(), if any)
+//
+void TrackingAllocator::reset()
+{
+	AllocSet::iterator first = mAllocSet.begin(), last = mAllocSet.end();
+	for (; first != last; ++first)
+		subAllocator.free(*first);
+}
+
+
+//
+// Forget about all allocated memory. It's now your responsibility.
+//
+void TrackingAllocator::commit()
+{
+	mAllocSet.clear();
+}