liblttng-ust-libc-wrapper/lttng-ust-malloc.c

   1 /*
   2  * Copyright (C) 2009  Pierre-Marc Fournier
   3  * Copyright (C) 2011  Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2.1 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public
  16  * License along with this library; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  18  */
  19
  20 #define _GNU_SOURCE
  21 #include <dlfcn.h>
  22 #include <sys/types.h>
  23 #include <stdio.h>
  24
  25 #define TRACEPOINT_DEFINE
  26 #define TRACEPOINT_CREATE_PROBES
  27 #include "ust_libc.h"
  28
  29 void *malloc(size_t size)
  30 {
  31         static void *(*plibc_malloc)(size_t size) = NULL;
  32         void *retval;
  33
  34         if (plibc_malloc == NULL) {
  35                 plibc_malloc = dlsym(RTLD_NEXT, "malloc");
  36                 if (plibc_malloc == NULL) {
  37                         fprintf(stderr, "mallocwrap: unable to find malloc\n");
  38                         return NULL;
  39                 }
  40         }
  41         retval = plibc_malloc(size);
  42         tracepoint(ust_libc, malloc, size, retval);
  43         return retval;
  44 }
  45
  46 void free(void *ptr)
  47 {
  48         static void *(*plibc_free)(void *ptr) = NULL;
  49
  50         if (plibc_free == NULL) {
  51                 plibc_free = dlsym(RTLD_NEXT, "free");
  52                 if (plibc_free == NULL) {
  53                         fprintf(stderr, "mallocwrap: unable to find free\n");
  54                         return;
  55                 }
  56         }
  57         tracepoint(ust_libc, free, ptr);
  58         plibc_free(ptr);
  59 }