klish/lub/heap/heap_post_realloc.c

#include <string.h>

#include "private.h"
#include "context.h"
#include "node.h"

void
lub_heap_post_realloc(lub_heap_t        *this,
                      char             **ptr)
{
    /* only act if we are about to re-enable the monitoring */
    if((0 < lub_heap_frame_count) && ( 2 > this->suppress))
    {
        if(NULL != *ptr)
        {
            stackframe_t              frame;
            lub_heap_node_t          *node = (lub_heap_node_t*)*ptr;
            const lub_heap_context_t *context;

            lub_heap__get_stackframe(&frame,lub_heap_frame_count);

            /* make sure we break any recursive behaviour */
            ++this->suppress;
            context = lub_heap_context_find_or_create(this,&frame);
            --this->suppress;
            
            /* initialise the node instance */
            lub_heap_node_init(node,(lub_heap_context_t*)context);

            /* make sure we doctor the pointer given back to the client */
            *ptr = lub_heap_node__get_ptr(node);
        }
    }
}

void
lub_heap_leak_suppress_detection(lub_heap_t *this)
{
    ++this->suppress;
}

void
lub_heap_leak_restore_detection(lub_heap_t *this)
{
    --this->suppress;
}