Использование Malloc Hooks

Question

Я пытаюсь использовать ловушку malloc для создания пользовательской функции my_malloc (). В моей основной программе, когда я вызываю malloc (), я хочу, чтобы он вызывал my_malloc (), может кто-нибудь дать мне пример того, как это сделать в C

Answer 1

Вы смотрели здесь? http://www.gnu.org/software/libtool/manual/libc/Hooks-for-Malloc.html

Answer 2

Согласно http://www.gnu.org/software/libtool/manual/libc/Hooks-for-Malloc.html, вот как это сделать с библиотеками GCC.

/* Prototypes for __malloc_hook, __free_hook */
 #include <malloc.h>

 /* Prototypes for our hooks.  */
 static void my_init_hook (void);
 static void *my_malloc_hook (size_t, const void *);
 static void my_free_hook (void*, const void *);

 /* Override initializing hook from the C library. */
 void (*__malloc_initialize_hook) (void) = my_init_hook;

 static void
 my_init_hook (void)
 {
   old_malloc_hook = __malloc_hook;
   old_free_hook = __free_hook;
   __malloc_hook = my_malloc_hook;
   __free_hook = my_free_hook;
 }

 static void *
 my_malloc_hook (size_t size, const void *caller)
 {
   void *result;
   /* Restore all old hooks */
   __malloc_hook = old_malloc_hook;
   __free_hook = old_free_hook;
   /* Call recursively */
   result = malloc (size);
   /* Save underlying hooks */
   old_malloc_hook = __malloc_hook;
   old_free_hook = __free_hook;
   /* printf might call malloc, so protect it too. */
   printf ("malloc (%u) returns %p\n", (unsigned int) size, result);
   /* Restore our own hooks */
   __malloc_hook = my_malloc_hook;
   __free_hook = my_free_hook;
   return result;
 }

 static void
 my_free_hook (void *ptr, const void *caller)
 {
   /* Restore all old hooks */
   __malloc_hook = old_malloc_hook;
   __free_hook = old_free_hook;
   /* Call recursively */
   free (ptr);
   /* Save underlying hooks */
   old_malloc_hook = __malloc_hook;
   old_free_hook = __free_hook;
   /* printf might call free, so protect it too. */
   printf ("freed pointer %p\n", ptr);
   /* Restore our own hooks */
   __malloc_hook = my_malloc_hook;
   __free_hook = my_free_hook;
 }

 main ()
 {
   ...
 }

Answer 3

Просто обратите внимание, что решение my_malloc_hook () на самом деле не работает в многопоточной программе - см. http://www.phpman.info/index.php/man/malloc_hook/3.

Answer 4

Если ваша функция вызывает sbrk напрямую, вы можете просто вызвать ее как malloc и убедиться, что она связана до библиотеки malloc. Это работает на всех операционных системах типа Unix. Для окон см. Есть ли способ переопределить malloc во время соединения в Windows?

Если ваша функция является оболочкой для malloc библиотеки, предложение Алексея #define сработает.

Answer 5

#undef malloc
#define malloc my_malloc

Просто поместите это наверху любого файла, для которого вам нужно это сделать.