mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
240 lines
8.6 KiB
240 lines
8.6 KiB
/***
|
|
*chkesp.c
|
|
*
|
|
* Copyright (c) 1997-2001, Microsoft Corporation. All rights reserved.
|
|
*
|
|
*Purpose:
|
|
* Defines _chkesp() and other run-time error checking support routines.
|
|
*
|
|
*Revision History:
|
|
* 05-22-98 JWM Support added for KFrei's RTC work; header added.
|
|
* 07-28-98 JWM RTC update.
|
|
* 10-30-98 KBF Quit messing with the CRT's Debug Heap flags
|
|
* 11-19-98 KBF Added stuff to handle multiple callbacks
|
|
* 11-24-98 KBF Added 3rd callback for memory/string function checks
|
|
* 12-03-98 KBF Added 4th callback to disable mem/string function
|
|
* checks temporarily
|
|
* 05-11-99 KBF Wrap RTC support in #ifdef.
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#include <malloc.h>
|
|
#include <dbgint.h>
|
|
#include <windows.h>
|
|
#include <rtcsup.h>
|
|
|
|
/***
|
|
*void __chkesp() - check to make sure esp was properly restored
|
|
*
|
|
*Purpose:
|
|
* A debugging check called after every function call to make sure esp has
|
|
* the same value before and after the call.
|
|
*
|
|
*Entry:
|
|
* condition code: the ZF flag should be cleared if esp has changed
|
|
*
|
|
*Return:
|
|
* <void>
|
|
*
|
|
*******************************************************************************/
|
|
void __declspec(naked) _chkesp() {
|
|
__asm {
|
|
jne esperror ;
|
|
ret
|
|
|
|
esperror:
|
|
; function prolog
|
|
|
|
push ebp
|
|
mov ebp, esp
|
|
sub esp, __LOCAL_SIZE
|
|
|
|
push eax ; save the old return value
|
|
push edx
|
|
|
|
push ebx
|
|
push esi
|
|
push edi
|
|
}
|
|
|
|
/**
|
|
* let the user know that there is a problem, and allow them to debug the
|
|
* program.
|
|
*/
|
|
#ifdef _DEBUG
|
|
if (_CrtDbgReport(_CRT_ERROR, __FILE__, __LINE__, "",
|
|
"The value of ESP was not properly saved across a function "
|
|
"call. This is usually a result of calling a function "
|
|
"declared with one calling convention with a function "
|
|
"pointer declared with a different calling convention. " ) == 1)
|
|
#endif
|
|
{
|
|
|
|
/* start the debugger */
|
|
__asm int 3;
|
|
}
|
|
|
|
__asm {
|
|
; function epilog
|
|
|
|
pop edi
|
|
pop esi
|
|
pop ebx
|
|
|
|
pop edx ; restore the old return value
|
|
pop eax
|
|
|
|
mov esp, ebp
|
|
pop ebp
|
|
ret
|
|
}
|
|
}
|
|
|
|
#ifdef _RTC
|
|
/***
|
|
*void __CRT_RTC_INIT() - Initialize the RTC subsystem (in section .CRT$XIC)
|
|
*
|
|
*Purpose:
|
|
* Setup anything involving the RTC subsystem
|
|
*
|
|
*Entry:
|
|
* The allocation hook - function to use to allocate memory
|
|
* The release hook - function to use to release memory
|
|
*
|
|
*Return:
|
|
* The default error reporting function
|
|
*
|
|
*******************************************************************************/
|
|
|
|
#ifdef _RTC_ADVMEM
|
|
// This stuff is currently disabled
|
|
#define mk_list(type) \
|
|
typedef struct type##_l { \
|
|
int version; \
|
|
type##_hook_fp funcptr; \
|
|
} type##_l; \
|
|
static struct { \
|
|
int size; \
|
|
int max; \
|
|
type##_l *hooks; \
|
|
} type##_list = {0,0,0}; \
|
|
type##_hook_fp type##_hook = 0; \
|
|
static int type##_version = 0
|
|
|
|
mk_list(_RTC_Allocate);
|
|
mk_list(_RTC_Free);
|
|
mk_list(_RTC_MemCheck);
|
|
mk_list(_RTC_FuncCheckSet);
|
|
HANDLE _RTC_api_change_mutex = NULL;
|
|
|
|
#define add_func(type, vers, fp) { \
|
|
if (type##_version < vers) \
|
|
{ \
|
|
type##_version = vers; \
|
|
type##_hook = (type##_hook_fp)fp; \
|
|
} \
|
|
if (!type##_list.hooks) \
|
|
{ \
|
|
type##_list.hooks = (type##_l*) \
|
|
VirtualAlloc(0, 65536, MEM_RESERVE, PAGE_READWRITE); \
|
|
} \
|
|
if (type##_list.size == type##_list.max) \
|
|
{ \
|
|
type##_list.max += 4096/sizeof(type##_l); \
|
|
VirtualAlloc(type##_list.hooks, type##_list.max*sizeof(type##_l),\
|
|
MEM_COMMIT, PAGE_READWRITE); \
|
|
} \
|
|
type##_list.hooks[type##_list.size].funcptr = (type##_hook_fp)fp; \
|
|
type##_list.hooks[type##_list.size++].version = vers; \
|
|
}
|
|
|
|
#define del_func(type, fp) { \
|
|
int i; \
|
|
for (i = 0; i < type##_list.size; i++) \
|
|
{ \
|
|
if (type##_list.hooks[i].funcptr == fp) \
|
|
{ \
|
|
for (i++; i < type##_list.size; i++) \
|
|
{ \
|
|
type##_list.hooks[i-1].funcptr = type##_list.hooks[i].funcptr; \
|
|
type##_list.hooks[i-1].version = type##_list.hooks[i].version; \
|
|
} \
|
|
type##_list.size--; \
|
|
break; \
|
|
} \
|
|
} \
|
|
if (fp == (void*)type##_hook) \
|
|
{ \
|
|
int hiver = 0; \
|
|
type##_hook_fp candidate = 0; \
|
|
for (i = 0; i < type##_list.size; i++) \
|
|
{ \
|
|
if (type##_list.hooks[i].version > hiver) \
|
|
{ \
|
|
hiver = type##_list.hooks[i].version; \
|
|
candidate = type##_list.hooks[i].funcptr; \
|
|
} \
|
|
} \
|
|
type##_hook = candidate; \
|
|
type##_version = hiver; \
|
|
} \
|
|
}
|
|
|
|
#endif
|
|
|
|
/*
|
|
funcs is a list of function pointers that are currently defined as:
|
|
funcs[0] = Allocation hook
|
|
funcs[1] = Free hook
|
|
funcs[2] = Memory Check hook
|
|
funcs[3] = Function Check enabler/disabler hook
|
|
*/
|
|
|
|
_RTC_error_fn __cdecl
|
|
_CRT_RTC_INIT(HANDLE mutex, void **funcs, int funccount, int version, int unloading)
|
|
{
|
|
#ifdef _RTC_ADVMEM
|
|
// This stuff is currently disabled
|
|
if (mutex && !_RTC_api_change_mutex)
|
|
_RTC_api_change_mutex = mutex;
|
|
if (funccount > 0)
|
|
{
|
|
if (!unloading)
|
|
{
|
|
switch (funccount)
|
|
{
|
|
default:
|
|
case 4:
|
|
add_func(_RTC_FuncCheckSet, version, funcs[3]);
|
|
case 3:
|
|
add_func(_RTC_MemCheck, version, funcs[2]);
|
|
case 2:
|
|
add_func(_RTC_Free, version, funcs[1]);
|
|
case 1:
|
|
add_func(_RTC_Allocate, version, funcs[0]);
|
|
}
|
|
} else {
|
|
switch (funccount)
|
|
{
|
|
default:
|
|
case 4:
|
|
del_func(_RTC_FuncCheckSet, funcs[3]);
|
|
case 3:
|
|
del_func(_RTC_MemCheck, funcs[2]);
|
|
case 2:
|
|
del_func(_RTC_Free, funcs[1]);
|
|
case 1:
|
|
del_func(_RTC_Allocate, funcs[0]);
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#ifdef _DEBUG
|
|
return &_CrtDbgReport;
|
|
#else
|
|
return 0;
|
|
#endif
|
|
}
|
|
|
|
#endif
|