jemalloc_mock.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#define NEW_BASE_ADDR 98765

void *mymalloc(size_t size) {
        return (void *)size;
}

void myfree(void *ptr) {
        (void)ptr;
}

void *myrealloc(void *ptr, size_t size) {
        myfree(ptr);
        return (void*)mymalloc(size);
}

int main(void) {
        void *ptr = NULL;
        void *new_ptr = NULL;
        for (int i = 1; i <= 10; i++) {
                ptr = mymalloc(i);
                new_ptr = myrealloc(ptr, NEW_BASE_ADDR + i);
                myfree(new_ptr);
                sleep(1);
        }
        return 0;
}

probe begin {
    warn("Begin to trace malloc.\n")
}

probe process.function("mymalloc").return {
    printf("malloc: %d, bytes: %d\n", $return, @entry($size))
}

probe process.function("myrealloc").return {
    printf("realloc: %d, bytes: %d\n", $return, @entry($size))
}

probe process.function("myfree") {
    if ($ptr == 0) next
    printf("free: %d\n", $ptr)
}

probe timer.s(10) {
    exit()
}