-void *realloc( void *ptr, size_t size ){
- return mspace_realloc( mySpace, ptr, size );
+/** @brief Snapshotting malloc implementation for user programs */
+void *malloc(size_t size)
+{
+ if (user_snapshot_space) {
+ void *tmp = mspace_malloc(user_snapshot_space, size);
+ ASSERT(tmp);
+ return tmp;
+ } else
+ return HandleEarlyAllocationRequest(size);
+}
+
+/** @brief Snapshotting free implementation for user programs */
+void free(void * ptr)
+{
+ if (!DontFree(ptr))
+ mspace_free(user_snapshot_space, ptr);
+}
+
+/** @brief Snapshotting realloc implementation for user programs */
+void *realloc(void *ptr, size_t size)
+{
+ void *tmp = mspace_realloc(user_snapshot_space, ptr, size);
+ ASSERT(tmp);
+ return tmp;
+}
+
+/** @brief Snapshotting calloc implementation for user programs */
+void * calloc(size_t num, size_t size)
+{
+ if (user_snapshot_space) {
+ void *tmp = mspace_calloc(user_snapshot_space, num, size);
+ ASSERT(tmp);
+ return tmp;
+ } else {
+ void *tmp = HandleEarlyAllocationRequest(size * num);
+ std::memset(tmp, 0, size * num);
+ return tmp;
+ }