cdsa/cdsa_utilities/trackingallocator.h

   1 /*
   2  * Copyright (c) 2000-2001 Apple Computer, Inc. All Rights Reserved.
   3  *
   4  * The contents of this file constitute Original Code as defined in and are
   5  * subject to the Apple Public Source License Version 1.2 (the 'License').
   6  * You may not use this file except in compliance with the License. Please obtain
   7  * a copy of the License at http://www.apple.com/publicsource and read it before
   8  * using this file.
   9  *
  10  * This Original Code and all software distributed under the License are
  11  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER EXPRESS
  12  * OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES, INCLUDING WITHOUT
  13  * LIMITATION, ANY WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
  14  * PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT. Please see the License for the
  15  * specific language governing rights and limitations under the License.
  16  */
  17
  18
  19 //
  20 // trackingallocator - CssmAllocator that keeps track of allocations
  21 //
  22 #ifndef _H_TRACKINGALLOCATOR
  23 #define _H_TRACKINGALLOCATOR
  24
  25 #include <Security/cssmalloc.h>
  26 #include <set>
  27
  28 namespace Security
  29 {
  30
  31 //
  32 // A CssmAllocator that keeps track of allocations and can throw everything
  33 // away unless explicitly committed.
  34 //
  35 class TrackingAllocator : public CssmAllocator
  36 {
  37 public:
  38         TrackingAllocator(CssmAllocator &inAllocator) : mAllocator(inAllocator) {}
  39         virtual ~TrackingAllocator();
  40
  41         void *malloc(size_t inSize) throw(std::bad_alloc)
  42         {
  43                 void *anAddress = mAllocator.malloc(inSize);
  44                 mAllocSet.insert(anAddress);
  45                 return anAddress;
  46         }
  47
  48         void free(void *inAddress) throw()
  49         {
  50                 mAllocator.free(inAddress);
  51                 mAllocSet.erase(inAddress);
  52         }
  53
  54         void *realloc(void *inAddress, size_t inNewSize) throw(std::bad_alloc)
  55         {
  56                 void *anAddress = mAllocator.realloc(inAddress, inNewSize);
  57                 if (anAddress != inAddress)
  58                 {
  59                         mAllocSet.erase(inAddress);
  60                         mAllocSet.insert(anAddress);
  61                 }
  62
  63                 return anAddress;
  64         }
  65
  66         void commit() { mAllocSet.clear(); }
  67 private:
  68         typedef std::set<void *> AllocSet;
  69
  70         CssmAllocator &mAllocator;
  71         AllocSet mAllocSet;
  72 };
  73
  74 } // end namespace Security
  75
  76 #endif //_H_TRACKINGALLOCATOR