mymemory.cc

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