Security-55163.44.tar.gz

[apple/security.git] / libsecurity_utilities / lib / trackingallocator.cpp

diff --git a/libsecurity_utilities/lib/trackingallocator.cpp b/libsecurity_utilities/lib/trackingallocator.cpp
new file mode 100644 (file)
index 0000000..c207550
--- /dev/null
+++ b/libsecurity_utilities/lib/trackingallocator.cpp
@@ -0,0 +1,89 @@
+/*
+ * Copyright (c) 2000-2004 Apple Computer, 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();
+}