Leaked source code of windows server 2003
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.
 
 
 
 
 
 

467 lines
16 KiB

/***
*trnsctrl.cpp -
*
* Copyright (c) 1990-2001, Microsoft Corporation. All rights reserved.
*
*Purpose:
*
*Revision History:
* 06-01-97 Created by TiborL.
* 07-12-99 RDL Image relative fixes under CC_P7_SOFT25.
* 10-07-99 SAH utc_p7#1126: fix ipsr.ri reset.
* 10-19-99 TGL Miscellaneous unwind fixes.
* 03-15-00 PML Remove CC_P7_SOFT25, which is now on permanently.
* 03-30-00 SAH New version of GetLanguageSpecificData from ntia64.h.
* 06-08-00 RDL VS#111429: IA64 workaround for AV while handling throw.
* 06-05-01 GB AMD64 Eh support Added.
* 12-07-01 BWT Remove NTSUBSET
*
****/
extern "C" {
#include <windows.h>
};
#include <winnt.h>
#include <mtdll.h>
#include <ehassert.h>
#include <ehdata.h>
#include <trnsctrl.h>
#include <ehstate.h>
#include <eh.h>
#include <ehhooks.h>
#pragma hdrstop
#ifdef _MT
#define pFrameInfoChain (*((FRAMEINFO **) &(_getptd()->_pFrameInfoChain)))
#define _ImageBase (_getptd()->_ImageBase)
#define _ThrowImageBase (_getptd()->_ThrowImageBase)
#define _pCurrentException (*((EHExceptionRecord **)&(_getptd()->_curexception)))
#define _pForeignExcept (*((EHExceptionRecord **)&(_getptd()->_pForeignException)))
#else
static FRAMEINFO *pFrameInfoChain = NULL; // used to remember nested frames
static unsigned __int64 _ImageBase = 0;
static unsigned __int64 _ThrowImageBase = 0;
extern EHExceptionRecord *_pCurrentException;
extern EHExceptionRecord *_pForeignExcept;
#endif
// Should be used out of ntamd64.h, but can't figure out how to allow that with
// existing dependencies.
#define GetLanguageSpecificData(f, base) \
(((PUNWIND_INFO)(f->UnwindInfoAddress + base))->UnwindCode + ((((PUNWIND_INFO)(f->UnwindInfoAddress + base))->CountOfCodes + 1)&~1) +1)
extern "C" VOID RtlRestoreContext (PCONTEXT ContextRecord,PEXCEPTION_RECORD ExceptionRecord OPTIONAL);
extern "C" void __FrameUnwindToState(EHRegistrationNode *, DispatcherContext *, FuncInfo *, __ehstate_t);
//
// Returns the establisher frame pointers. For catch handlers it is the parent's frame pointer.
//
EHRegistrationNode *_GetEstablisherFrame(
EHRegistrationNode *pRN,
DispatcherContext *pDC,
FuncInfo *pFuncInfo,
EHRegistrationNode *pEstablisher
) {
TryBlockMapEntry *pEntry;
HandlerType *pHandler;
unsigned __int64 HandlerAdd, ImageBase;
unsigned num_of_try_blocks = FUNC_NTRYBLOCKS(*pFuncInfo);
unsigned index, i;
__ehstate_t curState;
curState = __StateFromControlPc(pFuncInfo, pDC);
*pEstablisher = *pRN;
for (index = num_of_try_blocks; index > 0; index--) {
pEntry = FUNC_PTRYBLOCK(*pFuncInfo, index -1, pDC->ImageBase);
if (curState > TBME_HIGH(*pEntry) && curState <= TBME_CATCHHIGH(*pEntry)) {
// Get catch handler address.
HandlerAdd = (*RtlLookupFunctionEntry(pDC->ControlPc, &ImageBase, pDC->HistoryTable)).BeginAddress;
pHandler = TBME_PLIST(*pEntry, ImageBase);
for ( i = 0;
i < (unsigned)TBME_NCATCHES(*pEntry) &&
pHandler[i].dispOfHandler != HandlerAdd
; i++);
if ( i < (unsigned)TBME_NCATCHES(*pEntry)) {
*pEstablisher = *(EHRegistrationNode *)OffsetToAddress(pHandler[i].dispFrame, *pRN);
break;
}
}
}
return pEstablisher;
}
extern "C" unsigned __int64 _GetImageBase()
{
return _ImageBase;
}
extern "C" unsigned __int64 _GetThrowImageBase()
{
return _ThrowImageBase;
}
extern "C" VOID _SetThrowImageBase(unsigned __int64 NewThrowImageBase)
{
_ThrowImageBase = NewThrowImageBase;
}
extern "C" VOID _MoveContext(CONTEXT* pTarget, CONTEXT* pSource)
{
RtlMoveMemory(pTarget, pSource, sizeof(CONTEXT));
}
// This function returns the try block for the given state if the state is in a
// catch; otherwise, NULL is returned.
static __inline TryBlockMapEntry *_CatchTryBlock(
FuncInfo *pFuncInfo,
__ehstate_t curState
) {
TryBlockMapEntry *pEntry;
unsigned num_of_try_blocks = FUNC_NTRYBLOCKS(*pFuncInfo);
unsigned index;
for (index = num_of_try_blocks; index > 0; index--) {
pEntry = FUNC_PTRYBLOCK(*pFuncInfo, index -1, _ImageBase);
if (curState > TBME_HIGH(*pEntry) && curState <= TBME_CATCHHIGH(*pEntry)) {
return pEntry;
}
}
return NULL;
}
//
// This routine returns TRUE if we are executing from within a catch. Otherwise, FALSE is returned.
//
BOOL _ExecutionInCatch(
DispatcherContext *pDC,
FuncInfo *pFuncInfo
) {
__ehstate_t curState = __StateFromControlPc(pFuncInfo, pDC);
return _CatchTryBlock(pFuncInfo, curState)? TRUE : FALSE;
}
// This function unwinds to the empty state.
VOID __FrameUnwindToEmptyState(
EHRegistrationNode *pRN,
DispatcherContext *pDC,
FuncInfo *pFuncInfo
) {
__ehstate_t stateFromControlPC;
TryBlockMapEntry *pEntry;
EHRegistrationNode EstablisherFramePointers, *pEstablisher;
pEstablisher = _GetEstablisherFrame(pRN, pDC, pFuncInfo, &EstablisherFramePointers);
stateFromControlPC = __StateFromControlPc(pFuncInfo, pDC);
pEntry = _CatchTryBlock(pFuncInfo, stateFromControlPC);
__FrameUnwindToState(pEstablisher, pDC, pFuncInfo,
pEntry == NULL ? EH_EMPTY_STATE : TBME_HIGH(*pEntry));
}
//
// Prototype for the internal handler
//
extern "C" EXCEPTION_DISPOSITION __InternalCxxFrameHandler(
EHExceptionRecord *pExcept, // Information for this exception
EHRegistrationNode *pRN, // Dynamic information for this frame
CONTEXT *pContext, // Context info
DispatcherContext *pDC, // More dynamic info for this frame
FuncInfo *pFuncInfo, // Static information for this frame
int CatchDepth, // How deeply nested are we?
EHRegistrationNode *pMarkerRN, // Marker node for when checking inside catch block
BOOL recursive); // True if this is a translation exception
//
// __CxxFrameHandler - Real entry point to the runtime
//
extern "C" _CRTIMP EXCEPTION_DISPOSITION __CxxFrameHandler(
EHExceptionRecord *pExcept, // Information for this exception
EHRegistrationNode RN, // Dynamic information for this frame
CONTEXT *pContext, // Context info
DispatcherContext *pDC // More dynamic info for this frame
) {
FuncInfo *pFuncInfo;
EXCEPTION_DISPOSITION result;
EHRegistrationNode EstablisherFrame = RN;
_ImageBase = pDC->ImageBase;
_ThrowImageBase = (unsigned __int64)pExcept->params.pThrowImageBase;
pFuncInfo = (FuncInfo*)(_ImageBase +*(PULONG)pDC->HandlerData);
result = __InternalCxxFrameHandler( pExcept, &EstablisherFrame, pContext, pDC, pFuncInfo, 0, NULL, FALSE );
return result;
}
// Call the SEH to EH translator.
int __SehTransFilter( EXCEPTION_POINTERS *ExPtrs,
EHExceptionRecord *pExcept,
EHRegistrationNode *pRN,
CONTEXT *pContext,
DispatcherContext *pDC,
FuncInfo *pFuncInfo,
BOOL *pResult)
{
_pForeignExcept = pExcept;
_ThrowImageBase = (unsigned __int64)((EHExceptionRecord *)ExPtrs->ExceptionRecord)->params.pThrowImageBase;
__InternalCxxFrameHandler( (EHExceptionRecord *)ExPtrs->ExceptionRecord,
pRN,
pContext,
pDC,
pFuncInfo,
0,
NULL,
TRUE );
_pForeignExcept = NULL;
*pResult = TRUE;
return EXCEPTION_EXECUTE_HANDLER;
}
BOOL _CallSETranslator(
EHExceptionRecord *pExcept, // The exception to be translated
EHRegistrationNode *pRN, // Dynamic info of function with catch
CONTEXT *pContext, // Context info
DispatcherContext *pDC, // More dynamic info of function with catch (ignored)
FuncInfo *pFuncInfo, // Static info of function with catch
ULONG CatchDepth, // How deeply nested in catch blocks are we?
EHRegistrationNode *pMarkerRN // Marker for parent context
) {
BOOL result = FALSE;
pRN;
pDC;
pFuncInfo;
CatchDepth;
// Call the translator.
_EXCEPTION_POINTERS excptr = { (PEXCEPTION_RECORD)pExcept, pContext };
__try {
__pSETranslator(PER_CODE(pExcept), &excptr);
result = FALSE;
} __except(__SehTransFilter( exception_info(),
pExcept,
pRN,
pContext,
pDC,
pFuncInfo,
&result
)) {}
// If we got back, then we were unable to translate it.
return result;
}
/////////////////////////////////////////////////////////////////////////////
//
// _GetRangeOfTrysToCheck - determine which try blocks are of interest, given
// the current catch block nesting depth. We only check the trys at a single
// depth.
//
// Returns:
// Address of first try block of interest is returned
// pStart and pEnd get the indices of the range in question
//
TryBlockMapEntry* _GetRangeOfTrysToCheck(
EHRegistrationNode *pRN,
FuncInfo *pFuncInfo,
int CatchDepth,
__ehstate_t curState,
unsigned *pStart,
unsigned *pEnd,
DispatcherContext *pDC
) {
TryBlockMapEntry *pEntry, *pCurCatchEntry = NULL;
unsigned num_of_try_blocks = FUNC_NTRYBLOCKS(*pFuncInfo);
unsigned int index;
__ehstate_t ipState = __StateFromControlPc(pFuncInfo, pDC);
DASSERT( num_of_try_blocks > 0 );
*pStart = *pEnd = -1;
for (index = num_of_try_blocks; index > 0; index--) {
pEntry = FUNC_PTRYBLOCK(*pFuncInfo, index -1, pDC->ImageBase);
if (ipState > TBME_HIGH(*pEntry) && ipState <= TBME_CATCHHIGH(*pEntry)) {
break;
}
}
if (index) {
pCurCatchEntry = FUNC_PTRYBLOCK(*pFuncInfo, index -1, pDC->ImageBase);
}
for(index = 0; index < num_of_try_blocks; index++ ) {
pEntry = FUNC_PTRYBLOCK(*pFuncInfo, index, pDC->ImageBase);
// if in catch block, check for try-catch only in current block
if (pCurCatchEntry) {
if (TBME_LOW(*pEntry) <= TBME_HIGH(*pCurCatchEntry) ||
TBME_HIGH(*pEntry) > TBME_CATCHHIGH(*pCurCatchEntry))
continue;
}
if( curState >= TBME_LOW(*pEntry) && curState <= TBME_HIGH(*pEntry) ) {
if (*pStart == -1) {
*pStart = index;
}
*pEnd = index+1;
}
}
if ( *pStart == -1){
*pStart = 0;
*pEnd = 0;
return NULL;
}
else
return FUNC_PTRYBLOCK(*pFuncInfo, *pStart, pDC->ImageBase);
}
FRAMEINFO * _CreateFrameInfo(
FRAMEINFO * pFrameInfo,
PVOID pExceptionObject
) {
pFrameInfo->pExceptionObject = pExceptionObject;
pFrameInfo->pNext = (pFrameInfo < pFrameInfoChain)? pFrameInfoChain : NULL;
pFrameInfoChain = pFrameInfo;
return pFrameInfo;
}
/////////////////////////////////////////////////////////////////////////////
//
// IsExceptionObjectToBeDestroyed - Determine if an exception object is still
// in use by a more deeply nested catch frame, or if it unused and should be
// destroyed on exiting from the current catch block.
//
// Returns:
// TRUE if exception object not found and should be destroyed.
//
BOOL _IsExceptionObjectToBeDestroyed(
PVOID pExceptionObject
) {
FRAMEINFO * pFrameInfo;
for (pFrameInfo = pFrameInfoChain; pFrameInfo != NULL; pFrameInfo = pFrameInfo->pNext ) {
if( pFrameInfo->pExceptionObject == pExceptionObject ) {
return FALSE;
}
}
return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
//
// _FindAndUnlinkFrame - Pop the frame information for this scope that was
// pushed by _CreateFrameInfo. This should be the first frame in the list,
// but the code will look for a nested frame and pop all frames, just in
// case.
//
void _FindAndUnlinkFrame(
FRAMEINFO * pFrameInfo
) {
DASSERT(pFrameInfo == pFrameInfoChain);
for (FRAMEINFO *pCurFrameInfo = pFrameInfoChain;
pCurFrameInfo != NULL;
pCurFrameInfo = pCurFrameInfo->pNext)
{
if (pFrameInfo == pCurFrameInfo) {
pFrameInfoChain = pCurFrameInfo->pNext;
return;
}
}
// Should never be reached.
DASSERT(0);
}
extern "C" void _UnwindNestedFrames(
EHRegistrationNode *pFrame, // Unwind up to (but not including) this frame
EHExceptionRecord *pExcept, // The exception that initiated this unwind
CONTEXT *pContext, // Context info for current exception
EHRegistrationNode *pEstablisher,
void *Handler,
__ehstate_t TargetUnwindState,
FuncInfo *pFuncInfo,
DispatcherContext *pDC,
BOOLEAN recursive
) {
static const EXCEPTION_RECORD ExceptionTemplate = { // A generic exception record
0x80000029L, // STATUS_UNWIND_CONSOLIDATE
EXCEPTION_NONCONTINUABLE, // Exception flags (we don't do resume)
NULL, // Additional record (none)
NULL, // Address of exception (OS fills in)
15, // Number of parameters
{ EH_MAGIC_NUMBER1, // Our version control magic number
NULL, NULL, NULL, NULL, NULL, NULL, NULL,
NULL, NULL, NULL, NULL, NULL, NULL, NULL
} // pThrowInfo
};
CONTEXT Context;
EXCEPTION_RECORD ExceptionRecord = ExceptionTemplate;
ExceptionRecord.ExceptionInformation[0] = (ULONG_PTR)__CxxCallCatchBlock;
// Address of call back function
ExceptionRecord.ExceptionInformation[1] = (ULONG_PTR)pEstablisher;
// Used by callback funciton
ExceptionRecord.ExceptionInformation[2] = (ULONG_PTR)Handler;
// Used by callback function to call catch block
ExceptionRecord.ExceptionInformation[3] = (ULONG_PTR)TargetUnwindState;
// Used by CxxFrameHandler to unwind to target_state
ExceptionRecord.ExceptionInformation[4] = (ULONG_PTR)pContext;
// used to set pCurrentExContext in callback function
ExceptionRecord.ExceptionInformation[5] = (ULONG_PTR)pFuncInfo;
// Used in callback function to set state on stack to -2
ExceptionRecord.ExceptionInformation[6] = (ULONG_PTR)pExcept;
// Used for passing currne t Exception
ExceptionRecord.ExceptionInformation[7] = (ULONG_PTR)recursive;
// Used for translated Exceptions
RtlUnwindEx((void *)*pFrame,
(void *)pDC->ControlPc, // Address where control left function
&ExceptionRecord,
NULL,
&Context,
pDC->HistoryTable);
}
typedef
PVOID
(*PCallSettingFrameHandler) (
ULONG NonUsed,
EHRegistrationNode Establisher
);
extern "C"
PVOID
_CallSettingFrame (
PVOID Handler,
EHRegistrationNode *Establisher,
ULONG NLG_CODE
)
/*++
Routine Description:
This function calls a catch handler or destructor passing the establisher
frame.
Arguments:
Handler - Supplies the address of the handler.
Establisher - Supplies a pointer the establisher frame.
NLG_CODE - Supplies a non-local goto code.
Return Value:
The continuation address is returned if catch handler returns.
--*/
{
return ((PCallSettingFrameHandler)Handler)(0, *Establisher);
}