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.
351 lines
9.9 KiB
351 lines
9.9 KiB
/*++
|
|
|
|
Copyright (c) 1990 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
thredini.c
|
|
|
|
Abstract:
|
|
|
|
This module implements the machine dependent functions to set the initial
|
|
context and data alignment handling mode for a process or thread object.
|
|
|
|
Author:
|
|
|
|
David N. Cutler (davec) 1-Apr-1990
|
|
|
|
Environment:
|
|
|
|
Kernel mode only.
|
|
|
|
Revision History:
|
|
|
|
3-19-96 Bernard Lint (blint) Conversion to IA64 (from PPC and MIPS versions)
|
|
|
|
--*/
|
|
|
|
#include "ki.h"
|
|
|
|
VOID
|
|
KeContextToKframesSpecial (
|
|
IN PKTHREAD Thread,
|
|
IN OUT PKTRAP_FRAME TrapFrame,
|
|
IN OUT PKEXCEPTION_FRAME ExceptionFrame,
|
|
IN PCONTEXT ContextFrame,
|
|
IN ULONG ContextFlags
|
|
);
|
|
|
|
//
|
|
// The following assert macros are used to check that an input object is
|
|
// really the proper type.
|
|
//
|
|
|
|
#define ASSERT_PROCESS(E) { \
|
|
ASSERT((E)->Header.Type == ProcessObject); \
|
|
}
|
|
|
|
#define ASSERT_THREAD(E) { \
|
|
ASSERT((E)->Header.Type == ThreadObject); \
|
|
}
|
|
|
|
|
|
|
|
VOID
|
|
KiInitializeContextThread (
|
|
IN PKTHREAD Thread,
|
|
IN PKSYSTEM_ROUTINE SystemRoutine,
|
|
IN PKSTART_ROUTINE StartRoutine OPTIONAL,
|
|
IN PVOID StartContext OPTIONAL,
|
|
IN PCONTEXT ContextRecord OPTIONAL
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function initializes the machine dependent context of a thread object.
|
|
|
|
Actually, what it does is to lay out the stack for the thread so that
|
|
it contains a stack frame that will be picked up by SwapContext and
|
|
returned thru, resulting in a transfer of control to KiThreadStartup.
|
|
In otherwords, we lay out a stack with a stack frame that looks as if
|
|
SwapContext had been called just before the first instruction in
|
|
KiThreadStartup.
|
|
|
|
N.B. This function does not check the accessibility of the context record.
|
|
It is assumed the the caller of this routine is either prepared to
|
|
handle access violations or has probed and copied the context record
|
|
as appropriate.
|
|
|
|
N.B. Arguments to the new thread are passed in the Swap Frame preserved registers
|
|
s0 - s3 which are restored by Swap Context when thread execution begins.
|
|
|
|
Arguments:
|
|
|
|
Thread - Supplies a pointer to a dispatcher object of type thread.
|
|
|
|
SystemRoutine - Supplies a pointer to the system function that is to be
|
|
called when the thread is first scheduled for execution.
|
|
|
|
N.B. This is the routine entry point, not a function pointer (plabel pointer).
|
|
|
|
StartRoutine - Supplies an optional pointer to a function that is to be
|
|
called after the system has finished initializing the thread. This
|
|
parameter is specified if the thread is a system thread and will
|
|
execute totally in kernel mode.
|
|
|
|
N.B. This is the routine function pointer (plabel pointer).
|
|
|
|
StartContext - Supplies an optional pointer to an arbitrary data structure
|
|
which will be passed to the StartRoutine as a parameter. This
|
|
parameter is specified if the thread is a system thread and will
|
|
execute totally in kernel mode.
|
|
|
|
ContextRecord - Supplies an optional pointer a context frame which contains
|
|
the initial user mode state of the thread. This parameter is specified
|
|
if the thread is a user thread and will execute in user mode. If this
|
|
parameter is not specified, then the Teb parameter is ignored.
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
PKSWITCH_FRAME SwFrame;
|
|
PKEXCEPTION_FRAME ExFrame;
|
|
ULONG_PTR InitialStack;
|
|
PKTRAP_FRAME TrFrame;
|
|
|
|
//
|
|
// Set up the thread backing store pointers from the initial stack pointers.
|
|
//
|
|
|
|
InitialStack = (ULONG_PTR)Thread->InitialStack;
|
|
Thread->InitialBStore = (PVOID)InitialStack;
|
|
Thread->BStoreLimit = (PVOID)(InitialStack + KERNEL_BSTORE_SIZE);
|
|
|
|
//
|
|
// If a context frame is specified, then initialize a trap frame and
|
|
// and an exception frame with the specified user mode context. Also
|
|
// allocate the switch frame.
|
|
//
|
|
|
|
if (ARGUMENT_PRESENT(ContextRecord)) {
|
|
|
|
TrFrame = (PKTRAP_FRAME)((InitialStack)
|
|
- KTHREAD_STATE_SAVEAREA_LENGTH
|
|
- KTRAP_FRAME_LENGTH);
|
|
|
|
ExFrame = (PKEXCEPTION_FRAME)(((ULONG_PTR)TrFrame +
|
|
STACK_SCRATCH_AREA -
|
|
sizeof(KEXCEPTION_FRAME)) & ~((ULONG_PTR)15));
|
|
|
|
SwFrame = (PKSWITCH_FRAME)(((ULONG_PTR)ExFrame -
|
|
sizeof(KSWITCH_FRAME)) & ~((ULONG_PTR)15));
|
|
|
|
KeContextToKframesSpecial(Thread, TrFrame, ExFrame,
|
|
ContextRecord,
|
|
ContextRecord->ContextFlags | CONTEXT_CONTROL);
|
|
|
|
//
|
|
// Set the saved previous processor mode in the trap frame and the
|
|
// previous processor mode in the thread object to user mode.
|
|
//
|
|
|
|
TrFrame->PreviousMode = UserMode;
|
|
Thread->PreviousMode = UserMode;
|
|
|
|
//
|
|
// Initialize the FPSR for user mode
|
|
//
|
|
|
|
TrFrame->StFPSR = USER_FPSR_INITIAL;
|
|
|
|
//
|
|
// Initialize the user TEB pointer in the trap frame
|
|
//
|
|
|
|
TrFrame->IntTeb = (ULONGLONG)Thread->Teb;
|
|
|
|
} else {
|
|
|
|
SwFrame = (PKSWITCH_FRAME)((InitialStack) - sizeof(KSWITCH_FRAME));
|
|
|
|
//
|
|
// Set the previous mode in thread object to kernel.
|
|
//
|
|
|
|
Thread->PreviousMode = KernelMode;
|
|
}
|
|
|
|
//
|
|
// Initialize context switch frame and set thread start up parameters.
|
|
// The Swap return pointer and SystemRoutine are entry points, not function pointers.
|
|
//
|
|
|
|
RtlZeroMemory((PVOID)SwFrame, sizeof(KSWITCH_FRAME)); // init all to 0
|
|
|
|
SwFrame->SwitchRp = ((PPLABEL_DESCRIPTOR)KiThreadStartup)->EntryPoint;
|
|
SwFrame->SwitchExceptionFrame.IntS0 = (ULONGLONG)ContextRecord;
|
|
SwFrame->SwitchExceptionFrame.IntS1 = (ULONGLONG)StartContext;
|
|
SwFrame->SwitchExceptionFrame.IntS2 = (ULONGLONG)StartRoutine;
|
|
SwFrame->SwitchExceptionFrame.IntS3 =
|
|
((PPLABEL_DESCRIPTOR)SystemRoutine)->EntryPoint;
|
|
SwFrame->SwitchFPSR = FPSR_FOR_KERNEL;
|
|
SwFrame->SwitchBsp = (ULONGLONG)Thread->InitialBStore;
|
|
|
|
Thread->KernelBStore = Thread->InitialBStore;
|
|
Thread->KernelStack = (PVOID)((ULONG_PTR)SwFrame-STACK_SCRATCH_AREA);
|
|
|
|
if (Thread->Teb) {
|
|
PKAPPLICATION_REGISTERS AppRegs;
|
|
|
|
AppRegs = GET_APPLICATION_REGISTER_SAVEAREA(Thread->StackBase);
|
|
|
|
AppRegs->Ar21 = 0; // ContextRecord->StFCR;
|
|
|
|
//
|
|
// Ar24 is EFlags. Need to set up a value that is good for iVE
|
|
// Based on i386 eflags SANITIZE_FLAGS
|
|
// This is simplier, though since never running i386 in kernel
|
|
// mode
|
|
AppRegs->Ar24 = EFLAGS_INTERRUPT_MASK | (((ULONG) ContextRecord->Eflag) & EFLAGS_USER_SANITIZE);
|
|
|
|
AppRegs->Ar25 = USER_CODE_DESCRIPTOR;
|
|
AppRegs->Ar26 = USER_DATA_DESCRIPTOR;
|
|
AppRegs->Ar27 = (((ULONGLONG) CR4_VME | CR4_FXSR | CR4_XMMEXCPT) << 32)
|
|
| (CR0_PE | CFLG_II);
|
|
AppRegs->Ar28 = SANITIZE_AR28_FSR (ContextRecord->StFSR, UserMode);
|
|
AppRegs->Ar29 = SANITIZE_AR29_FIR (ContextRecord->StFIR, UserMode);
|
|
AppRegs->Ar30 = SANITIZE_AR30_FDR (ContextRecord->StFDR, UserMode);
|
|
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
BOOLEAN
|
|
KeSetAutoAlignmentProcess (
|
|
IN PRKPROCESS Process,
|
|
IN BOOLEAN Enable
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function sets the data alignment handling mode for the specified
|
|
process and returns the previous data alignment handling mode.
|
|
|
|
Arguments:
|
|
|
|
Process - Supplies a pointer to a dispatcher object of type process.
|
|
|
|
Enable - Supplies a boolean value that determines the handling of data
|
|
alignment exceptions for the process. A value of TRUE causes all
|
|
data alignment exceptions to be automatically handled by the kernel.
|
|
A value of FALSE causes all data alignment exceptions to be actually
|
|
raised as exceptions.
|
|
|
|
Return Value:
|
|
|
|
A value of TRUE is returned if data alignment exceptions were
|
|
previously automatically handled by the kernel. Otherwise, a value
|
|
of FALSE is returned.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
KIRQL OldIrql;
|
|
BOOLEAN Previous;
|
|
|
|
ASSERT_PROCESS(Process);
|
|
|
|
//
|
|
// Raise IRQL to dispatcher level and lock dispatcher database.
|
|
//
|
|
|
|
KiLockDispatcherDatabase(&OldIrql);
|
|
|
|
//
|
|
// Capture the previous data alignment handling mode and set the
|
|
// specified data alignment mode.
|
|
//
|
|
|
|
Previous = Process->AutoAlignment;
|
|
Process->AutoAlignment = Enable;
|
|
|
|
//
|
|
// Unlock dispatcher database, lower IRQL to its previous value, and
|
|
// return the previous data alignment mode.
|
|
//
|
|
|
|
KiUnlockDispatcherDatabase(OldIrql);
|
|
return Previous;
|
|
}
|
|
|
|
BOOLEAN
|
|
KeSetAutoAlignmentThread (
|
|
IN PKTHREAD Thread,
|
|
IN BOOLEAN Enable
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This function sets the data alignment handling mode for the specified
|
|
thread and returns the previous data alignment handling mode.
|
|
|
|
Arguments:
|
|
|
|
Thread - Supplies a pointer to a dispatcher object of type thread.
|
|
|
|
Enable - Supplies a boolean value that determines the handling of data
|
|
alignment exceptions for the thread. A value of TRUE causes all
|
|
data alignment exceptions to be automatically handled by the kernel.
|
|
A value of FALSE causes all data alignment exceptions to be actually
|
|
raised as exceptions.
|
|
|
|
Return Value:
|
|
|
|
A value of TRUE is returned if data alignment exceptions were
|
|
previously automatically handled by the kernel. Otherwise, a value
|
|
of FALSE is returned.
|
|
|
|
--*/
|
|
|
|
{
|
|
|
|
KIRQL OldIrql;
|
|
BOOLEAN Previous;
|
|
|
|
ASSERT_THREAD(Thread);
|
|
|
|
//
|
|
// Raise IRQL to dispatcher level and lock dispatcher database.
|
|
//
|
|
|
|
KiLockDispatcherDatabase(&OldIrql);
|
|
|
|
//
|
|
// Capture the previous data alignment handling mode and set the
|
|
// specified data alignment mode.
|
|
//
|
|
|
|
Previous = Thread->AutoAlignment;
|
|
Thread->AutoAlignment = Enable;
|
|
|
|
//
|
|
// Unlock dispatcher database, lower IRQL to its previous value, and
|
|
// return the previous data alignment mode.
|
|
//
|
|
|
|
KiUnlockDispatcherDatabase(OldIrql);
|
|
return Previous;
|
|
}
|