OSX/include/security_utilities/trackingallocator.cpp

   1 /*
   2  * Copyright (c) 2000-2004,2011,2014 Apple 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 //
  36 TrackingAllocator::~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 //
  46 void *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
  53 void TrackingAllocator::free(void *inAddress) throw()
  54 {
  55         subAllocator.free(inAddress);
  56         mAllocSet.erase(inAddress);
  57 }
  58
  59 void *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 //
  75 void 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 //
  86 void TrackingAllocator::commit()
  87 {
  88         mAllocSet.clear();
  89 }