#ifndef __FORTIFY_H__
#define __FORTIFY_H__
/*
 * FILE:
 *   fortify.h
 *
 * DESCRIPTION:
 *     Header file for fortify.c - A fortified shell for malloc, realloc, 
 *   calloc & free
 *
 * WRITTEN:
 *   spb 29/4/94
 *
 * VERSION:
 *   1.0 29/4/94
 *
 * MODIFIED: <string.h>, Fortify_strdup, strdup added, Mark Koennecke, 9.6.97
 * Fortify_STRDUP is in strdup.c
 */
#include <stdlib.h>
#include <string.h>

#ifdef __cplusplus
extern "C" {
#endif

  typedef void (*OutputFuncPtr) (char *);

  void *Fortify_malloc(size_t size, char *file, unsigned long line);
  void *Fortify_realloc(void *ptr, size_t new_size, char *file,
                        unsigned long line);
  void *Fortify_calloc(size_t num, size_t size, char *file,
                       unsigned long line);
  void Fortify_free(void *uptr, char *file, unsigned long line);
  char *Fortify_STRDUP(const char *in, char *file, unsigned long line);

  int Fortify_OutputAllMemory(char *file, unsigned long line);
  int Fortify_CheckAllMemory(char *file, unsigned long line);
  int Fortify_CheckPointer(void *uptr, char *file, unsigned long line);
  int Fortify_Disable(char *file, unsigned long line);
  int Fortify_SetMallocFailRate(int Percent);
  int Fortify_EnterScope(char *file, unsigned long line);
  int Fortify_LeaveScope(char *file, unsigned long line);
  int Fortify_DumpAllMemory(int scope, char *file, unsigned long line);

  typedef void (*Fortify_OutputFuncPtr) (const char *);
  Fortify_OutputFuncPtr Fortify_SetOutputFunc(Fortify_OutputFuncPtr
                                              Output);

#ifdef __cplusplus
}
#endif
#ifndef __FORTIFY_C__           /* Only define the macros if we're NOT in fortify.c */
#ifdef FORTIFY                  /* Add file and line information to the fortify calls */
#define malloc(size)                  Fortify_malloc(size, __FILE__, __LINE__)
#define realloc(ptr,new_size)         Fortify_realloc(ptr, new_size, __FILE__, __LINE__)
#define calloc(num,size)              Fortify_calloc(num, size, __FILE__, __LINE__)
#define free(ptr)                     Fortify_free(ptr, __FILE__, __LINE__)
#define Fortify_OutputAllMemory()     Fortify_OutputAllMemory(__FILE__, __LINE__)
#define Fortify_CheckAllMemory()      Fortify_CheckAllMemory(__FILE__, __LINE__)
#define Fortify_CheckPointer(ptr)     Fortify_CheckPointer(ptr, __FILE__, __LINE__)
#define Fortify_Disable()             Fortify_Disable(__FILE__, __LINE__)
#define Fortify_EnterScope()          Fortify_EnterScope(__FILE__, __LINE__)
#define Fortify_LeaveScope()          Fortify_LeaveScope(__FILE__, __LINE__)
#define Fortify_DumpAllMemory(s)      Fortify_DumpAllMemory(s,__FILE__, __LINE__)
#define strdup(s)                     Fortify_STRDUP(s,__FILE__,__LINE__)
#else                           /* FORTIFY Define the special fortify functions away to nothing */
#define Fortify_OutputAllMemory()     0
#define Fortify_CheckAllMemory()      0
#define Fortify_CheckPointer(ptr)     1
#define Fortify_Disable()             1
#define Fortify_SetOutputFunc()       0
#define Fortify_SetMallocFailRate(p)  0
#define Fortify_EnterScope()          0
#define Fortify_LeaveScope()          0
#define Fortify_DumpAllMemory(s)      0
#endif                          /*   FORTIFY     */
#endif                          /* __FORTIFY_C__ */
#endif                          /* __FORTIFY_H__ */