#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
 */
#include <stdlib.h>

#ifdef __cplusplus
extern "C" {
#endif

#ifdef _DCC
#define STDARGS __stkargs
#else
#define STDARGS __stdargs
#endif

typedef void (*OutputFuncPtr)(char *);

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

int STDARGS Fortify_OutputAllMemory(const char *file, unsigned long line);
int STDARGS Fortify_CheckAllMemory(const char *file, unsigned long line);
int STDARGS Fortify_CheckPointer(void *uptr, const char *file, unsigned long line);
int STDARGS Fortify_Disable(const char *file, unsigned long line);
int STDARGS Fortify_SetMallocFailRate(int Percent);
int STDARGS Fortify_EnterScope(const char *file, unsigned long line);
int STDARGS Fortify_LeaveScope(const char *file, unsigned long line);
int STDARGS Fortify_DumpAllMemory(int scope, const 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__)

#else /* FORTIFY Define the special fortify functions away to nothing */

#define Fortify_OutputAllMemory()     0
#define Fortify_CheckAllMemory()      0
#define Fortify_CheckPointer(ptr)     (ptr != NULL)
#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__ */