openmpi/src/mem/malloc.h

/*
 * $HEADER$
 */

/** @file */

#ifndef LAM_MALLOC_H
#define LAM_MALLOC_H

#include <stdlib.h>

/*
 * THIS FILE CANNOT INCLUDE ANY OTHER LAM HEADER FILES!!!
 *
 * It is included via <lam_config_bottom.h>.  Hence, it should not
 * include ANY other files, nor should it include "lam_config.h".
 *
 */

/*
 * Set LAM_MALLOC_DEBUG_LEVEL to
 * 0 for no checking
 * 1 for basic error checking
 * 2 for more error checking
 */

#ifndef LAM_MALLOC_DEBUG_LEVEL
#define LAM_MALLOC_DEBUG_LEVEL 2
#endif

#if defined(c_plusplus) || defined(__cplusplus)
extern "C" {
#endif
  /**
   * Shut down malloc debug output.
   *
   * This function is invoked as part of lam_finalize() to shut down the
   * output stream for malloc debug messages.
   */
  void lam_malloc_init(void);

  /**
   * Initialize malloc debug output.
   *
   * This function is invoked to setup a dedicated output stream for
   * malloc debug functions.  It does \em not (currently) do anything
   * other than that (i.e., no internal accounting for tracking
   * malloc/free statements, etc.).
   *
   * It is invoked as part of lam_init().  Although this function is
   * not \em necessary for LAM_MALLOC() and LAM_FREE(), it is strong
   * recommended because no output messages -- regardless of the
   * malloc debug level set by lam_malloc_debug() -- will be displayed
   * unless this function is invoked first.
   */
  void lam_malloc_finalize(void);

  /**
   * \internal
   *
   * Back-end error-checking malloc function for LAM (you should use
   * the normal malloc() instead of this function).
   *
   * @param size The number of bytes to allocate
   * @param file Typically the __FILE__ macro
   * @param line Typically the __LINE__ macro
   *
   * This function is only used when --enable-mem-debug was specified to
   * configure (or by default if you're building in a CVS checkout).
   */
  void *lam_malloc(size_t size, char *file, int line);

  /**
   * \internal
   *
   * Back-end error-checking realloc function for LAM (you should use
   * the normal realloc() instead of this function).
   *
   * @param ptr Pointer to reallocate
   * @param size The number of bytes to allocate
   * @param file Typically the __FILE__ macro
   * @param line Typically the __LINE__ macro
   *
   * This function is only used when --enable-mem-debug was specified to
   * configure (or by default if you're building in a CVS checkout).
   */
  void *lam_realloc(void *ptr, size_t size, char *file, int line);

  /**
   * \internal
   *
   * Back-end error-checking free function for LAM (you should use
   * free() instead of this function).
   *
   * @param addr Address on the heap to free()
   * @param file Typically the __FILE__ macro
   * @param line Typically the __LINE__ macro
   *
   * This function is only used when --enable-mem-debug was specified
   * to configure (or by default if you're building in a CVS
   * checkout).
   */
  void lam_free(void *addr, char *file, int line);
#if defined(c_plusplus) || defined(__cplusplus)
}
#endif

extern int lam_malloc_debug_level;
extern int lam_malloc_output;

static inline void lam_malloc_debug(int level);


/**
 * Used to set the debug level for malloc debug.
 *
 * @param level The level of debugging (0 = none, 1 = some, 2 = more)
 *
 * This value defaults to the LAM_MALLOC_DEBUG_LEVEL.
 */
static inline void lam_malloc_debug(int level)
{
  lam_malloc_debug_level = level;
}

#endif /* LAM_MALLOC_H */