80 lines
3.3 KiB
C
80 lines
3.3 KiB
C
#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__ */
|