|
|
/*++
Copyright (c) 1989 Microsoft Corporation
Module Name:
sxsctxact.c
Abstract:
Side-by-side activation support for Windows/NT Implementation of context activation/deactivation
Author:
Michael Grier (MGrier) 2/2/2000
Revision History:
--*/
#include <ntos.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <sxstypes.h>
#include "sxsp.h"
#if NT_SXS_PERF_COUNTERS_ENABLED
#if defined(_X86_)
__inline ULONGLONG RtlpGetCycleCount(void) { __asm { RDTSC } } #else
__inline ULONGLONG RtlpGetCycleCount(void) { return 0; } #endif // defined(_X86_)
#endif // NT_SXS_PERF_COUNTERS_ENABLED
// DWORD just so that in the debugger we don't have to guess the size...
ULONG RtlpCaptureActivationContextActivationStacks = #if DBG
TRUE #else
FALSE #endif
;
//
// APPCOMPAT: Setting this flag to TRUE indicates that we no longer allow
// skipping over "unactivated" (ie: multiple activation) context frames.
// The default action should be FALSE, which will let mutliply-activated
// contexts slide by.
//
// WARNING: This allows app authors to be a little sleazy about their activate
// and deactivate pairs.
//
BOOLEAN RtlpNotAllowingMultipleActivation = FALSE;
NTSTATUS RtlpAllocateActivationContextStackFrame( IN ULONG Flags, PTEB Teb, OUT PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME *FrameOut ) { NTSTATUS Status = STATUS_INTERNAL_ERROR; LIST_ENTRY *ple; PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME Frame = NULL; ULONG i;
if (FrameOut != NULL) *FrameOut = NULL;
ASSERT((Flags == 0) && (FrameOut != NULL) && (Teb != NULL)); if ((Flags != 0) || (FrameOut == NULL) || (Teb == NULL)) { Status = STATUS_INVALID_PARAMETER; goto Exit; }
for (ple = Teb->ActivationContextStack.FrameListCache.Flink; ple != &Teb->ActivationContextStack.FrameListCache; ple = ple->Flink) { PACTIVATION_CONTEXT_STACK_FRAMELIST FrameList = CONTAINING_RECORD(ple, ACTIVATION_CONTEXT_STACK_FRAMELIST, Links);
for (i=0; i<RTL_NUMBER_OF(FrameList->Frames); i++) { if (FrameList->Frames[i].Frame.Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST) { ASSERT(FrameList->FramesInUse != NUMBER_OF(FrameList->Frames)); FrameList->FramesInUse++; Frame = &FrameList->Frames[i]; break; } }
if (Frame != NULL) break; }
if (Frame == NULL) { // No space left; allocate a new framelist...
PACTIVATION_CONTEXT_STACK_FRAMELIST FrameList = RtlAllocateHeap(RtlProcessHeap(), 0, sizeof(ACTIVATION_CONTEXT_STACK_FRAMELIST));
if (FrameList == NULL) { Status = STATUS_NO_MEMORY; goto Exit; }
for (i=0; i<RTL_NUMBER_OF(FrameList->Frames); i++) { FrameList->Frames[i].Frame.Previous = NULL; FrameList->Frames[i].Frame.ActivationContext = NULL; FrameList->Frames[i].Frame.Flags = RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST | RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED; FrameList->Frames[i].Cookie = 0; }
Frame = &FrameList->Frames[0];
FrameList->FramesInUse = 1;
InsertHeadList(&Teb->ActivationContextStack.FrameListCache, &FrameList->Links); }
ASSERT((Frame != NULL) && (Frame->Frame.Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST));
Frame->Frame.Flags = RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED; *FrameOut = Frame; Status = STATUS_SUCCESS;
Exit: return Status; }
VOID RtlpFreeActivationContextStackFrame( PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME Frame ) { LIST_ENTRY *ple = NULL; PTEB Teb = NtCurrentTeb();
ASSERT(Frame != NULL); if (Frame != NULL) { // If this assert fires, someone's trying to free an already freed frame. Or someone's set the
// "I'm on the free list" flag in the frame data.
ASSERT(!(Frame->Frame.Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST)); if (!(Frame->Frame.Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST)) { for (ple = Teb->ActivationContextStack.FrameListCache.Flink; ple != &Teb->ActivationContextStack.FrameListCache; ple = ple->Flink) { PACTIVATION_CONTEXT_STACK_FRAMELIST FrameList = CONTAINING_RECORD(ple, ACTIVATION_CONTEXT_STACK_FRAMELIST, Links);
// A frame list with zero frames shouldn't be on the list!
ASSERT(FrameList->FramesInUse > 0);
if ((Frame >= &FrameList->Frames[0]) && (Frame < &FrameList->Frames[RTL_NUMBER_OF(FrameList->Frames)])) { // It's in this frame list; look for it!
ULONG i = (ULONG)(Frame - FrameList->Frames);
// If this assert fires, it means that the frame pointer passed in should have been a frame
// in this framelist, but it actually didn't point to any of the array entries exactly.
// Probably someone munged the pointer.
ASSERT(Frame == &FrameList->Frames[i]);
if ((Frame == &FrameList->Frames[i]) && (FrameList->FramesInUse > 0)) { FrameList->FramesInUse--;
Frame->Frame.Flags = RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_ON_FREE_LIST;
if (FrameList->FramesInUse == 0) { RemoveEntryList(ple); RtlFreeHeap(RtlProcessHeap(), 0, FrameList); } }
// No sense continuing the search on the list; we've found the one.
break; } }
// If we ran off the end of the list, it must have been a bogus frame pointer.
ASSERT(ple != &Teb->ActivationContextStack.FrameListCache); } } }
#if !defined(INVALID_HANDLE_VALUE)
#define INVALID_HANDLE_VALUE ((HANDLE)(LONG_PTR)-1)
#endif // !defined(INVALID_HANDLE_VALUE)
//
// Define magic cookie values returned by RtlActivateActivationContext*() that
// represent a failure to activate the requested context. The notable thing
// is that on deactivation via the cookie, we need to know whether to leave
// querying disabled or whether to enable it, thus the two magic values.
//
// The top nibble of the cookie denotes its type: normal, default-pushed or failed
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_NORMAL (1)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_DUPLICATE_ACTIVATION (2)
#define ULONG_PTR_IZE(_x) ((ULONG_PTR) (_x))
#define ULONG_PTR_IZE_SHIFT_AND_MASK(_x, _shift, _mask) ((ULONG_PTR) ((ULONG_PTR_IZE((_x)) & (_mask)) << (_shift)))
//
// We only use the lower 12 bits of the thread id, but that should be unique enough;
// this is really for debugging aids; if your tests pass such that you're
// erroneously passing activation context cookies between threads that happen to
// match up in their lower 12 bits of their thread id, you're pretty darned
// lucky.
//
#define CHAR_BITS 8
#define BIT_LENGTH(x) (sizeof(x) * CHAR_BITS)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_LENGTH (4)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_OFFSET (BIT_LENGTH(PVOID) - ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_LENGTH)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_MASK ((1 << ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_LENGTH) - 1)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_LENGTH ((BIT_LENGTH(PVOID) / 2) - ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_LENGTH)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_OFFSET (ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_OFFSET - ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_LENGTH)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_MASK ((1 << ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_LENGTH) - 1)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_LENGTH (BIT_LENGTH(PVOID) - (ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_LENGTH + ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_LENGTH))
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_OFFSET (0)
#define ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_MASK ((1 << ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_LENGTH) - 1)
// Never try to use more than 32 bits for the TID field.
C_ASSERT(ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_LENGTH <= BIT_LENGTH(ULONG));
#define MAKE_ACTIVATION_CONTEXT_ACTIVATION_COOKIE(_type, _teb, _code) \
((ULONG_PTR) ( \ ULONG_PTR_IZE_SHIFT_AND_MASK((_type), ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_OFFSET, ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_BIT_MASK) | \ ULONG_PTR_IZE_SHIFT_AND_MASK((HandleToUlong((_teb)->ClientId.UniqueThread)), ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_OFFSET, ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_MASK) | \ ULONG_PTR_IZE_SHIFT_AND_MASK((_code), ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_OFFSET, ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE_BIT_MASK)))
#define EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_FIELD(_x, _fieldname) (ULONG_PTR_IZE((ULONG_PTR_IZE((_x)) >> ACTIVATION_CONTEXT_ACTIVATION_COOKIE_ ## _fieldname ## _BIT_OFFSET)) & ACTIVATION_CONTEXT_ACTIVATION_COOKIE_ ## _fieldname ## _BIT_MASK)
#define EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE(_x) EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_FIELD((_x), TYPE)
#define EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID(_x) EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_FIELD((_x), TID)
#define EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_CODE(_x) EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_FIELD((_x), CODE)
#define ACTIVATION_CONTEXT_TRUNCATED_TID_(_teb) (HandleToUlong((_teb)->ClientId.UniqueThread) & ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID_BIT_MASK)
#define ACTIVATION_CONTEXT_TRUNCATED_TID() ACTIVATION_CONTEXT_TRUNCATED_TID_(NtCurrentTeb())
PACTIVATION_CONTEXT RtlpMapSpecialValuesToBuiltInActivationContexts( PACTIVATION_CONTEXT ActivationContext ) { if (ActivationContext == ACTCTX_EMPTY) { ActivationContext = &RtlpTheEmptyActivationContext; } return ActivationContext; }
// Disable FPO optimization so that captured call stacks are more complete
#if defined(_X86_)
#pragma optimize( "y", off ) // disable FPO for consistent stack traces
#endif
NTSTATUS NTAPI RtlActivateActivationContext( ULONG Flags, PACTIVATION_CONTEXT ActivationContext, ULONG_PTR *CookieOut ) { NTSTATUS Status = STATUS_INTERNAL_ERROR;
ASSERT(Flags == 0); ASSERT(CookieOut != NULL);
if (CookieOut != NULL) *CookieOut = INVALID_ACTIVATION_CONTEXT_ACTIVATION_COOKIE;
if ((Flags != 0) || (CookieOut == NULL)) { Status = STATUS_INVALID_PARAMETER; goto Exit; }
if (!NT_SUCCESS(Status = RtlActivateActivationContextEx( 0, NtCurrentTeb(), ActivationContext, CookieOut))) goto Exit;
Status = STATUS_SUCCESS; Exit: return Status; }
NTSTATUS NTAPI RtlActivateActivationContextEx( ULONG Flags, PTEB Teb, PACTIVATION_CONTEXT ActivationContext, ULONG_PTR *Cookie ) { #if NT_SXS_PERF_COUNTERS_ENABLED
ULONGLONG InitialCycleCount = RtlpGetCycleCount(); #endif
NTSTATUS Status = STATUS_SUCCESS; ULONG_PTR NewCookie = INVALID_ACTIVATION_CONTEXT_ACTIVATION_COOKIE; PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME Frame; ULONG CapturedFrameCount, CapturedFrameHash;
ASSERT(Cookie != NULL);
if (Cookie != NULL) *Cookie = INVALID_ACTIVATION_CONTEXT_ACTIVATION_COOKIE;
ASSERT((Flags & ~(RTL_ACTIVATE_ACTIVATION_CONTEXT_EX_FLAG_RELEASE_ON_STACK_DEALLOCATION)) == 0); ASSERT(Teb != NULL); ASSERT(ActivationContext != INVALID_HANDLE_VALUE);
ActivationContext = RtlpMapSpecialValuesToBuiltInActivationContexts(ActivationContext);
if (((Flags & ~(RTL_ACTIVATE_ACTIVATION_CONTEXT_EX_FLAG_RELEASE_ON_STACK_DEALLOCATION)) != 0) || (Teb == NULL) || (ActivationContext == INVALID_HANDLE_VALUE) || (Cookie == NULL)) { Status = STATUS_INVALID_PARAMETER; goto Exit; }
Status = RtlpAllocateActivationContextStackFrame(0, Teb, &Frame); if (!NT_SUCCESS(Status)) goto Exit;
Frame->Frame.Flags = RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED;
if (Flags & RTL_ACTIVATE_ACTIVATION_CONTEXT_EX_FLAG_RELEASE_ON_STACK_DEALLOCATION) { Frame->Frame.Flags |= RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_NO_DEACTIVATE | RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_RELEASE_ON_DEACTIVATION; RtlAddRefActivationContext(ActivationContext); }
if (RtlpCaptureActivationContextActivationStacks) CapturedFrameCount = RtlCaptureStackBackTrace(2, NUMBER_OF(Frame->ActivationStackBackTrace), Frame->ActivationStackBackTrace, &CapturedFrameHash); else CapturedFrameCount = 0;
while (CapturedFrameCount < NUMBER_OF(Frame->ActivationStackBackTrace)) Frame->ActivationStackBackTrace[CapturedFrameCount++] = NULL;
Frame->Frame.Previous = Teb->ActivationContextStack.ActiveFrame; Frame->Frame.ActivationContext = ActivationContext;
NewCookie = MAKE_ACTIVATION_CONTEXT_ACTIVATION_COOKIE(ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_NORMAL, Teb, Teb->ActivationContextStack.NextCookieSequenceNumber); Teb->ActivationContextStack.NextCookieSequenceNumber++; Frame->Cookie = NewCookie; *Cookie = NewCookie; Teb->ActivationContextStack.ActiveFrame = &Frame->Frame;
Status = STATUS_SUCCESS;
Exit: #if NT_SXS_PERF_COUNTERS_ENABLED
Teb->ActivationContextCounters.ActivationCycles += RtlpGetCycleCount() - InitialCycleCount; Teb->ActivationContextCounters.Activations++; #endif // NT_SXS_PERF_COUNTERS_ENABLED
return Status; }
#if defined(_X86_)
#pragma optimize("", on)
#endif
VOID NTAPI RtlDeactivateActivationContext( ULONG Flags, ULONG_PTR Cookie ) { #if NT_SXS_PERF_COUNTERS_ENABLED
ULONGLONG InitialCycleCount = RtlpGetCycleCount(); #endif // NT_SXS_PERF_COUNTERS_ENABLED
PTEB Teb = NtCurrentTeb(); PRTL_ACTIVATION_CONTEXT_STACK_FRAME Frame; PRTL_ACTIVATION_CONTEXT_STACK_FRAME UnwindEndFrame; PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME HeapFrame = NULL;
if ((Flags & ~(RTL_DEACTIVATE_ACTIVATION_CONTEXT_FLAG_FORCE_EARLY_DEACTIVATION)) != 0) { DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_ERROR_LEVEL, "SXS: %s() called with invalid flags 0x%08lx\n", __FUNCTION__, Flags); RtlRaiseStatus(STATUS_INVALID_PARAMETER); }
// Fast exit
if (Cookie == INVALID_ACTIVATION_CONTEXT_ACTIVATION_COOKIE) return;
if (EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE(Cookie) != ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TYPE_NORMAL) { DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_ERROR_LEVEL, "SXS: %s() called with invalid cookie type 0x%08I64x\n", __FUNCTION__, Cookie); RtlRaiseStatus(STATUS_INVALID_PARAMETER); }
if (EXTRACT_ACTIVATION_CONTEXT_ACTIVATION_COOKIE_TID(Cookie) != ACTIVATION_CONTEXT_TRUNCATED_TID()) { DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_ERROR_LEVEL, "SXS: %s() called with invalid cookie tid 0x%08I64x - should be %08lx\n", __FUNCTION__, Cookie, ACTIVATION_CONTEXT_TRUNCATED_TID()); RtlRaiseStatus(STATUS_INVALID_PARAMETER); }
Frame = (PRTL_ACTIVATION_CONTEXT_STACK_FRAME) Teb->ActivationContextStack.ActiveFrame; // Do the "downcast", but don't use HeapFrame unless the RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED
// flag is set...
if (Frame != NULL) { HeapFrame = (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) ? CONTAINING_RECORD(Frame, RTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME, Frame) : NULL; }
RTL_SOFT_ASSERT((Frame != NULL) && (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) && (HeapFrame->Cookie == Cookie));
if (Frame != NULL) { if (((Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) == 0) || (HeapFrame->Cookie != Cookie)) { ULONG InterveningFrameCount = 0;
// The cookie wasn't current. Let's see if we can figure out what frame it was for...
PRTL_ACTIVATION_CONTEXT_STACK_FRAME CandidateFrame = Frame->Previous; PRTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME CandidateHeapFrame = (CandidateFrame != NULL) ? (CandidateFrame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) ? CONTAINING_RECORD(CandidateFrame, RTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME, Frame) : NULL : NULL;
while ((CandidateFrame != NULL) && ((CandidateHeapFrame == NULL) || (CandidateHeapFrame->Cookie != Cookie))) { InterveningFrameCount++; CandidateFrame = CandidateFrame->Previous; CandidateHeapFrame = (CandidateFrame != NULL) ? (CandidateFrame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) ? CONTAINING_RECORD(CandidateFrame, RTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME, Frame) : NULL : NULL; }
RTL_SOFT_ASSERT(CandidateFrame != NULL); if (CandidateFrame == NULL) { RtlRaiseStatus(STATUS_SXS_INVALID_DEACTIVATION); } else { // Otherwise someone left some dirt around.
EXCEPTION_RECORD ExceptionRecord;
ExceptionRecord.ExceptionRecord = NULL; ExceptionRecord.NumberParameters = 3; ExceptionRecord.ExceptionInformation[0] = InterveningFrameCount; ExceptionRecord.ExceptionInformation[1] = (ULONG_PTR) CandidateFrame; ExceptionRecord.ExceptionInformation[2] = (ULONG_PTR) Teb->ActivationContextStack.ActiveFrame; ExceptionRecord.ExceptionCode = STATUS_SXS_EARLY_DEACTIVATION; ExceptionRecord.ExceptionFlags = 0; // this exception *is* continuable since we can actually put the activation stack into a reasonable state
RtlRaiseException(&ExceptionRecord);
// If they continue the exception, just do the unwinds.
UnwindEndFrame = CandidateFrame->Previous; } } else UnwindEndFrame = Frame->Previous;
do { PRTL_ACTIVATION_CONTEXT_STACK_FRAME Previous = Frame->Previous;
// This is a weird one. A no-deactivate frame is typically only used to propogate
// active activation context state to a newly created thread. As such, it'll always
// be the topmost frame... thus how did we come to decide to deactivate it?
ASSERTMSG( "Unwinding through a no-deactivate frame", !(Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_NO_DEACTIVATE));
if (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_RELEASE_ON_DEACTIVATION) { RtlReleaseActivationContext(Frame->ActivationContext); }
if (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) { RtlpFreeActivationContextStackFrame(CONTAINING_RECORD(Frame, RTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME, Frame)); }
Frame = Previous; } while (Frame != UnwindEndFrame);
Teb->ActivationContextStack.ActiveFrame = UnwindEndFrame; }
#if NT_SXS_PERF_COUNTERS_ENABLED
Teb->ActivationContextCounters.DeactivationCycles += RtlpGetCycleCount() - InitialCycleCount; Teb->ActivationContextCounters.Deactivations++; #endif
}
VOID RtlActivateActivationContextUnsafeFast( PRTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME Frame, PACTIVATION_CONTEXT ActivationContext ) { const PTEB Teb = NtCurrentTeb(); const PRTL_ACTIVATION_CONTEXT_STACK_FRAME pStackFrame = (PRTL_ACTIVATION_CONTEXT_STACK_FRAME)Teb->ActivationContextStack.ActiveFrame; ASSERT(Frame->Format == RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_FORMAT_WHISTLER); ASSERT(Frame->Size >= sizeof(RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME));
if ( ( ( ( pStackFrame == NULL ) && ( ActivationContext == NULL ) ) || ( pStackFrame && ( pStackFrame->ActivationContext == ActivationContext ) ) ) && !RtlpNotAllowingMultipleActivation ) { // this uses too much stack in stress
//DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_TRACE_LEVEL, "SXS: %s() Not really activating\n", __FUNCTION__ );
Frame->Frame.Flags |= RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_NOT_REALLY_ACTIVATED; } else { // this uses too much stack in stress
//DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_TRACE_LEVEL, "SXS: %s() Really activating\n", __FUNCTION__ );
Frame->Frame.Previous = Teb->ActivationContextStack.ActiveFrame; Frame->Frame.ActivationContext = ActivationContext; Frame->Frame.Flags = 0; Teb->ActivationContextStack.ActiveFrame = &Frame->Frame; } }
VOID RtlDeactivateActivationContextUnsafeFast( PRTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME Frame ) { const PTEB Teb = NtCurrentTeb();
//
// Was this "not really activated" above (AppCompat problem) from above?
//
if ( !RtlpNotAllowingMultipleActivation && ( ( Frame->Frame.Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_NOT_REALLY_ACTIVATED ) != 0 ) ) { return; } // Make sure that the deactivation matches. If it does not (the exceptional
// condition) we'll throw an exception to make people deal with their
// coding errors.
else if (Teb->ActivationContextStack.ActiveFrame != &Frame->Frame) { EXCEPTION_RECORD ExceptionRecord; ULONG InterveningFrameCount = 0;
// What's the deal? Look to see if we're further up the stack.
// Actually, what we'll do is see if we can find our parent up the stack.
// This will also handle the double-deactivation case and let us continue
// nicely.
PRTL_ACTIVATION_CONTEXT_STACK_FRAME SearchFrame = Teb->ActivationContextStack.ActiveFrame; const PRTL_ACTIVATION_CONTEXT_STACK_FRAME Previous = Frame->Frame.Previous;
DbgPrintEx(DPFLTR_SXS_ID, DPFLTR_TRACE_LEVEL, "SXS: %s() Active frame is not the frame being deactivated %p != %p\n", __FUNCTION__, Teb->ActivationContextStack.ActiveFrame, &Frame->Frame); while ((SearchFrame != NULL) && (SearchFrame != Previous)) { InterveningFrameCount++; SearchFrame = SearchFrame->Previous; }
ExceptionRecord.ExceptionRecord = NULL; ExceptionRecord.NumberParameters = 3; ExceptionRecord.ExceptionInformation[0] = InterveningFrameCount; ExceptionRecord.ExceptionInformation[1] = (ULONG_PTR) &Frame->Frame; ExceptionRecord.ExceptionInformation[2] = (ULONG_PTR) Teb->ActivationContextStack.ActiveFrame;
if (SearchFrame != NULL) { // We're there. That's actually good; it just probably means that a function that our caller called
// activated an activation context and forgot to deactivate. Throw the exception and if it's continued,
// we're good to go.
if (InterveningFrameCount == 0) { // Wow, the frame-to-deactivate's previous is the active one. It must be that the caller
// already deactivated and is now deactivating again.
ExceptionRecord.ExceptionCode = STATUS_SXS_MULTIPLE_DEACTIVATION; } else { // Otherwise someone left some dirt around.
ExceptionRecord.ExceptionCode = STATUS_SXS_EARLY_DEACTIVATION; }
ExceptionRecord.ExceptionFlags = 0; // this exception *is* continuable since we can actually put the activation stack into a reasonable state
} else { // It wasn't there. It's almost certainly the wrong thing to try to set this
ExceptionRecord.ExceptionCode = STATUS_SXS_INVALID_DEACTIVATION; ExceptionRecord.ExceptionFlags = EXCEPTION_NONCONTINUABLE; // this exception is NOT continuable
}
RtlRaiseException(&ExceptionRecord); }
Teb->ActivationContextStack.ActiveFrame = Frame->Frame.Previous;
}
NTSTATUS NTAPI RtlGetActiveActivationContext( PACTIVATION_CONTEXT *ActivationContext ) { NTSTATUS Status = STATUS_SUCCESS; PTEB Teb = NULL; PACTIVATION_CONTEXT_STACK Stack = NULL; PRTL_ACTIVATION_CONTEXT_STACK_FRAME Frame = NULL;
if (ActivationContext != NULL) *ActivationContext = NULL;
if (ActivationContext == NULL) { Status = STATUS_INVALID_PARAMETER; goto Exit; }
Teb = NtCurrentTeb(); Frame = (PRTL_ACTIVATION_CONTEXT_STACK_FRAME) Teb->ActivationContextStack.ActiveFrame;
if (Frame != NULL) { RtlAddRefActivationContext(Frame->ActivationContext); *ActivationContext = Frame->ActivationContext; }
Status = STATUS_SUCCESS; Exit: return Status; }
VOID NTAPI RtlFreeThreadActivationContextStack( VOID ) { PTEB Teb = NtCurrentTeb(); PRTL_ACTIVATION_CONTEXT_STACK_FRAME Frame = (PRTL_ACTIVATION_CONTEXT_STACK_FRAME) Teb->ActivationContextStack.ActiveFrame;
while (Frame != NULL) { PRTL_ACTIVATION_CONTEXT_STACK_FRAME Previous = Frame->Previous;
// Release any lingering frames. The notable case when this happens is when a thread that
// has a non-default activation context active creates another thread which inherits the
// first thread's activation context, adding a reference to the activation context. When
// the new thread eventually dies, that initial frame is still active.
if (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_RELEASE_ON_DEACTIVATION) { RtlReleaseActivationContext(Frame->ActivationContext); }
if (Frame->Flags & RTL_ACTIVATION_CONTEXT_STACK_FRAME_FLAG_HEAP_ALLOCATED) { RtlpFreeActivationContextStackFrame(CONTAINING_RECORD(Frame, RTL_HEAP_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME, Frame)); }
Frame = Previous; }
Teb->ActivationContextStack.ActiveFrame = NULL; }
BOOLEAN NTAPI RtlIsActivationContextActive( PACTIVATION_CONTEXT ActivationContext ) { PTEB Teb = NtCurrentTeb(); PRTL_ACTIVATION_CONTEXT_STACK_FRAME Frame = (PRTL_ACTIVATION_CONTEXT_STACK_FRAME) Teb->ActivationContextStack.ActiveFrame;
while (Frame != NULL) { if (Frame->ActivationContext == ActivationContext) { return TRUE; }
Frame = Frame->Previous; }
return FALSE; }
|