mymemory.cc

   1 #include "mymemory.h"
   2 #include "snapshot.h"
   3 #include "snapshotimp.h"
   4 #include <stdio.h>
   5 #include <dlfcn.h>
   6 #if !USE_CHECKPOINTING
   7 static mspace sStaticSpace = NULL;
   8 #endif
   9
  10 void *MYMALLOC(size_t size) {
  11 #if USE_CHECKPOINTING
  12   static void *(*mallocp)(size_t size);
  13   char *error;
  14   void *ptr;
  15
  16   /* get address of libc malloc */
  17   if (!mallocp) {
  18     mallocp = ( void * ( * )( size_t ) )dlsym(RTLD_NEXT, "malloc");
  19     if ((error = dlerror()) != NULL) {
  20       fputs(error, stderr);
  21       exit(1);
  22     }
  23   }
  24   ptr = mallocp(size);
  25   return ptr;
  26 #else
  27   if( !sTheRecord ){
  28     createSharedLibrary();
  29   }
  30   if( NULL == sStaticSpace )
  31     sStaticSpace = create_mspace_with_base( ( void * )( sTheRecord->mSharedMemoryBase ), SHARED_MEMORY_DEFAULT -sizeof( struct Snapshot_t ), 1 );
  32   return mspace_malloc( sStaticSpace, size );
  33 #endif
  34 }
  35
  36 void MYFREE(void *ptr) {
  37 #if USE_CHECKPOINTING
  38   static void (*freep)(void *);
  39   char *error;
  40
  41   /* get address of libc free */
  42   if (!freep) {
  43     freep = ( void  ( * )( void * ) )dlsym(RTLD_NEXT, "free");
  44     if ((error = dlerror()) != NULL) {
  45       fputs(error, stderr);
  46       exit(1);
  47     }
  48   }
  49   freep(ptr);
  50 #else
  51   mspace_free( sStaticSpace, ptr );
  52 #endif
  53 }
  54 mspace mySpace = NULL;
  55 void *malloc( size_t size ) {
  56   return mspace_malloc( mySpace, size );
  57 }
  58
  59 void free( void * ptr ){
  60   mspace_free( mySpace, ptr );
  61 }
  62
  63 void * operator new(size_t size) throw(std::bad_alloc) {
  64   return MYMALLOC(size);
  65 }
  66
  67 void operator delete(void *p) throw() {
  68   MYFREE(p);
  69 }