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.
402 lines
13 KiB
402 lines
13 KiB
// TITLE("Call Out to User Mode")
|
|
//++
|
|
//
|
|
// Copyright (c) 1994 Microsoft Corporation
|
|
//
|
|
// Module Name:
|
|
//
|
|
// callout.s
|
|
//
|
|
// Abstract:
|
|
//
|
|
// This module implements the code necessary to call out from kernel
|
|
// mode to user mode.
|
|
//
|
|
// Author:
|
|
//
|
|
// John Vert (jvert) 2-Nov-1994
|
|
//
|
|
// Environment:
|
|
//
|
|
// Kernel mode only.
|
|
//
|
|
// Revision History:
|
|
//
|
|
//--
|
|
|
|
#include "ksalpha.h"
|
|
|
|
//
|
|
// Define external variables that can be addressed using GP.
|
|
//
|
|
|
|
.extern KeUserCallbackDispatcher
|
|
|
|
SBTTL("Call User Mode Function")
|
|
//++
|
|
//
|
|
// NTSTATUS
|
|
// KiCallUserMode (
|
|
// IN PVOID *OutputBuffer,
|
|
// IN PULONG OutputLength
|
|
// )
|
|
//
|
|
// Routine Description:
|
|
//
|
|
// This function calls a user mode function.
|
|
//
|
|
// N.B. This function calls out to user mode and the NtCallbackReturn
|
|
// function returns back to the caller of this function. Therefore,
|
|
// the stack layout must be consistent between the two routines.
|
|
//
|
|
// Arguments:
|
|
//
|
|
// OutputBuffer (a0) - Supplies a pointer to the variable that receivies
|
|
// the address of the output buffer.
|
|
//
|
|
// OutputLength (a1) - Supplies a pointer to a variable that receives
|
|
// the length of the output buffer.
|
|
//
|
|
// Return Value:
|
|
//
|
|
// The final status of the call out function is returned as the status
|
|
// of the function.
|
|
//
|
|
// N.B. This function does not return to its caller. A return to the
|
|
// caller is executed when a NtCallbackReturn system service is
|
|
// executed.
|
|
//
|
|
// N.B. This function does return to its caller if a kernel stack
|
|
// expansion is required and the attempted expansion fails.
|
|
//
|
|
//--
|
|
|
|
NESTED_ENTRY(KiCallUserMode, CuFrameLength, zero)
|
|
|
|
lda sp, -CuFrameLength(sp) // allocate stack frame
|
|
stq ra, CuRa(sp) // save return address
|
|
|
|
//
|
|
// Save nonvolatile integer registers.
|
|
//
|
|
|
|
stq s0, CuS0(sp) // save integer registers s0 - s5
|
|
stq s1, CuS1(sp) //
|
|
stq s2, CuS2(sp) //
|
|
stq s3, CuS3(sp) //
|
|
stq s4, CuS4(sp) //
|
|
stq s5, CuS5(sp) //
|
|
stq fp, CuFP(sp) // save FP
|
|
|
|
//
|
|
// Save nonvolatile floating registers.
|
|
//
|
|
|
|
stt f2, CuF2(sp) // save floating registers f2 - f9
|
|
stt f3, CuF3(sp) //
|
|
stt f4, CuF4(sp) //
|
|
stt f5, CuF5(sp) //
|
|
stt f6, CuF6(sp) //
|
|
stt f7, CuF7(sp) //
|
|
stt f8, CuF8(sp) //
|
|
stt f9, CuF9(sp) //
|
|
|
|
PROLOGUE_END
|
|
|
|
//
|
|
// Save argument registers.
|
|
//
|
|
|
|
stq a0, CuA0(sp) // save output buffer address
|
|
stq a1, CuA1(sp) // save output length address
|
|
|
|
//
|
|
// Check if sufficient room is available on the kernel stack for another
|
|
// system call.
|
|
//
|
|
|
|
GET_CURRENT_THREAD // get current thread address
|
|
|
|
bis v0, zero, t0 // save current thread address
|
|
LDP t1, ThInitialStack(t0) // get initial stack address
|
|
LDP t2, ThStackLimit(t0) // get current stack limit
|
|
SUBP sp, KERNEL_LARGE_STACK_COMMIT, t3 // compute bottom address
|
|
cmpult t2, t3, t4 // check if limit exceeded
|
|
bne t4, 10f // if ne, limit not exceeded
|
|
bis sp, zero, a0 // set current kernel stack address
|
|
bsr ra, MmGrowKernelStack // attempt to grow the kernel stack
|
|
bne v0, 20f // if ne, attempt to grow failed
|
|
|
|
GET_CURRENT_THREAD // get current thread address
|
|
|
|
bis v0, zero, t0 // save current thread address
|
|
LDP t1, ThInitialStack(t0) // get initial stack address
|
|
10: LDP fp, ThTrapFrame(t0) // get trap frame address
|
|
LDP t2, ThCallbackStack(t0) // get callback stack address
|
|
STP t1, CuInStk(sp) // save initial stack address
|
|
STP fp, CuTrFr(sp) // save trap frame address
|
|
STP t2, CuCbStk(sp) // save callback stack address
|
|
STP sp, ThCallbackStack(t0) // set callback stack address
|
|
|
|
//
|
|
// Restore state and callback to user mode.
|
|
//
|
|
// N.B. Interrupts are disabled to prevent get/set context APCs from
|
|
// occurring.
|
|
//
|
|
|
|
DISABLE_INTERRUPTS // disable interrupts
|
|
|
|
STP sp, ThInitialStack(t0) // reset initial stack address
|
|
ldq t3, TrFir(fp) // get old PC
|
|
STP t3, CuTrFir(sp) // save old PC
|
|
LDP t4, KeUserCallbackDispatcher // get continuation address
|
|
stq t4, TrFir(fp) // set continuation address
|
|
|
|
//
|
|
// If a user mode APC is pending, then request an APC interrupt.
|
|
//
|
|
|
|
bis zero, zero, a1 // assume no user APC pending
|
|
ldq_u t1, ThApcState+AsUserApcPending(t0) // get user APC pending
|
|
extbl t1, (ThApcState+AsUserApcPending) % 8, t1 //
|
|
ZeroByte(ThAlerted(t0)) // clear kernel mode alerted
|
|
cmovne t1, APC_INTERRUPT, a1 // if pending set APC interrupt
|
|
|
|
//
|
|
// Set initial kernel stack for this thread.
|
|
//
|
|
|
|
bis sp, zero, a0 // set kernel stack address
|
|
|
|
SET_INITIAL_KERNEL_STACK // set kernel stack pointer
|
|
|
|
ldl a0, TrPsr(fp) // get previous processor status
|
|
|
|
//
|
|
// a0 = previous psr
|
|
// a1 = sfw interrupt requests
|
|
//
|
|
|
|
RETURN_FROM_SYSTEM_CALL // return to user mode
|
|
|
|
ret zero, (ra) //
|
|
|
|
//
|
|
// An attempt to grow the kernel stack failed.
|
|
//
|
|
|
|
20: ldq ra, CuRa(sp) // restore return address
|
|
lda sp, CuFrameLength(sp) // deallocate stack frame
|
|
ret zero, (ra)
|
|
|
|
.end KiCallUserMode
|
|
|
|
SBTTL("Switch Kernel Stack")
|
|
//++
|
|
//
|
|
// PVOID
|
|
// KeSwitchKernelStack (
|
|
// IN PVOID StackBase,
|
|
// IN PVOID StackLimit
|
|
// )
|
|
//
|
|
// Routine Description:
|
|
//
|
|
// This function switches to the specified large kernel stack.
|
|
//
|
|
// N.B. This function can ONLY be called when there are no variables
|
|
// in the stack that refer to other variables in the stack, i.e.,
|
|
// there are no pointers into the stack.
|
|
//
|
|
// Arguments:
|
|
//
|
|
// StackBase (a0) - Supplies a pointer to the base of the new kernel
|
|
// stack.
|
|
//
|
|
// StackLimit (a1) - supplies a pointer to the limit of the new kernel
|
|
// stack.
|
|
//
|
|
// Return Value:
|
|
//
|
|
// The old kernel stack is returned as the function value.
|
|
//
|
|
//--
|
|
|
|
.struct 0
|
|
SsRa: .space 8 // saved return address
|
|
SsSp: .space 8 // saved new stack pointer
|
|
SsA0: .space 8 // saved argument registers a0-a1
|
|
SsA1: .space 8 //
|
|
SsFrameLength: // length of stack frame
|
|
|
|
NESTED_ENTRY(KeSwitchKernelStack, SsFrameLength, zero)
|
|
|
|
lda sp, -SsFrameLength(sp) // allocate stack frame
|
|
stq ra, SsRa(sp) // save return address
|
|
|
|
PROLOGUE_END
|
|
|
|
//
|
|
// Save the address of the new stack and copy the old stack to the new
|
|
// stack.
|
|
//
|
|
|
|
GET_CURRENT_THREAD // get current thread address
|
|
|
|
stq a0, SsA0(sp) // save new stack base address
|
|
stq a1, SsA1(sp) // save new stack limit address
|
|
LDP a2, ThStackBase(v0) // get current stack base address
|
|
LDP a3, ThTrapFrame(v0) // get current trap frame address
|
|
ADDP a3, a0, a3 // relocate trap frame address
|
|
SUBP a3, a2, a3 //
|
|
STP a3, ThTrapFrame(v0) // set current trap frame address
|
|
bis sp, zero, a1 // set source address of copy
|
|
SUBP a2, sp, a2 // compute length of copy
|
|
SUBP a0, a2, a0 // set destination address of copy
|
|
stq a0, SsSp(sp) // save new stack pointer address
|
|
bsr ra, RtlMoveMemory // copy old stack to new stack
|
|
|
|
//
|
|
// Switch to new kernel stack and return the address of the old kernel stack.
|
|
//
|
|
|
|
GET_CURRENT_THREAD // get current thread address
|
|
|
|
DISABLE_INTERRUPTS // disable interrupts
|
|
|
|
LDP t0, ThStackBase(v0) // get old stack base address
|
|
ldq a0, SsA0(sp) // get new stack base address
|
|
ldq a1, SsA1(sp) // get new stack limit address
|
|
STP a0, ThInitialStack(v0) // set new initial stack address
|
|
STP a0, ThStackBase(v0) // set new stack base address
|
|
STP a1, ThStackLimit(v0) // set new stack limit address
|
|
ldil t1, TRUE // set large kernel stack TRUE
|
|
StoreByte(t1, ThLargeStack(v0)) //
|
|
ldq sp, SsSp(sp) // set initial stack address
|
|
|
|
SET_INITIAL_KERNEL_STACK // set initial kernel stack address
|
|
|
|
ENABLE_INTERRUPTS // enable interrupts
|
|
|
|
ldq ra, SsRa(sp) // restore return address
|
|
lda sp, SsFrameLength(sp) // deallocate stack frame
|
|
ret zero, (ra) // return
|
|
|
|
.end KeSwitchKernelStack
|
|
|
|
SBTTL("Return from User Mode Callback")
|
|
//++
|
|
//
|
|
// NTSTATUS
|
|
// NtCallbackReturn (
|
|
// IN PVOID OutputBuffer OPTIONAL,
|
|
// IN ULONG OutputLength,
|
|
// IN NTSTATUS Status
|
|
// )
|
|
//
|
|
// Routine Description:
|
|
//
|
|
// This function returns from a user mode callout to the kernel
|
|
// mode caller of the user mode callback function.
|
|
//
|
|
// N.B. This function returns to the function that called out to user
|
|
// mode and the KiCallUserMode function calls out to user mode.
|
|
// Therefore, the stack layout must be consistent between the
|
|
// two routines.
|
|
//
|
|
// Arguments:
|
|
//
|
|
// OutputBuffer - Supplies an optional pointer to an output buffer.
|
|
//
|
|
// OutputLength - Supplies the length of the output buffer.
|
|
//
|
|
// Status - Supplies the status value returned to the caller of the
|
|
// callback function.
|
|
//
|
|
// Return Value:
|
|
//
|
|
// If the callback return cannot be executed, then an error status is
|
|
// returned. Otherwise, the specified callback status is returned to
|
|
// the caller of the callback function.
|
|
//
|
|
// N.B. This function returns to the function that called out to user
|
|
// mode is a callout is currently active.
|
|
//
|
|
//--
|
|
|
|
LEAF_ENTRY(NtCallbackReturn)
|
|
|
|
GET_CURRENT_THREAD // get current thread address
|
|
|
|
LDP t1, ThCallbackStack(v0) // get callback stack address
|
|
beq t1, 10f // if eq, no callback stack present
|
|
|
|
//
|
|
// Restore nonvolatile integer registers.
|
|
//
|
|
|
|
ldq s0, CuS0(t1) // restore integer registers s0 - s5
|
|
ldq s1, CuS1(t1) //
|
|
ldq s2, CuS2(t1) //
|
|
ldq s3, CuS3(t1) //
|
|
ldq s4, CuS4(t1) //
|
|
ldq s5, CuS5(t1) //
|
|
ldq fp, CuFP(t1) // restore FP
|
|
|
|
//
|
|
// Restore nonvolatile floating registers.
|
|
//
|
|
|
|
ldt f2, CuF2(t1) // restore floating registers f2 - f9
|
|
ldt f3, CuF3(t1) //
|
|
ldt f4, CuF4(t1) //
|
|
ldt f5, CuF5(t1) //
|
|
ldt f6, CuF6(t1) //
|
|
ldt f7, CuF7(t1) //
|
|
ldt f8, CuF8(t1) //
|
|
ldt f9, CuF9(t1) //
|
|
|
|
//
|
|
// Restore the trap frame and callback stack addresses, and store the output
|
|
// buffer address and length.
|
|
//
|
|
|
|
LDP t2, CuTrFr(t1) // get previous trap frame address
|
|
LDP t3, CuCbStk(t1) // get previous callback stack address
|
|
LDP t4, CuA0(t1) // get address to store output address
|
|
LDP t5, CuA1(t1) // get address to store output length
|
|
LDP t6, CuTrFir(t1) // get old trap frame PC
|
|
STP t2, ThTrapFrame(v0) // restore trap frame address
|
|
STP t3, ThCallbackStack(v0) // restore callback stack address
|
|
STP a0, 0(t4) // store output buffer address
|
|
STP a1, 0(t5) // store output buffer length
|
|
stq t6, TrFir(t2) // restore old trap frame PC
|
|
|
|
//
|
|
// **** this is the place where the current stack could be trimmed back.
|
|
//
|
|
// Restore initial stack pointer, trim stackback to callback frame,
|
|
// deallocate callback stack frame, and return to callback caller.
|
|
//
|
|
|
|
LDP a0, CuInStk(t1) // get previous initial stack
|
|
STP a0, ThInitialStack(v0) //
|
|
|
|
SET_INITIAL_KERNEL_STACK // set initial kernel stack address
|
|
|
|
bis t1, zero, sp // trim stack callback frame
|
|
bis a2, zero, v0 // set callback service status
|
|
ldq ra, CuRa(sp) // restore return address
|
|
lda sp, CuFrameLength(sp) // deallocate stack frame
|
|
ret zero, (ra) // return
|
|
|
|
//
|
|
// No callback is currently active.
|
|
//
|
|
|
|
10: ldil v0, STATUS_NO_CALLBACK_ACTIVE // set service status
|
|
ret zero, (ra) // return
|
|
|
|
.end NtCallbackReturn
|
|
|