RtlVirtualUnwind.c File Reference

#include <rtltests.h>

Include dependency graph for RtlVirtualUnwind.c:

Go to the source code of this file.

Classes
union	_TEST_UNWIND_CODE
struct	_TEST_UNWIND_INFO
struct	_IMAGE_STRUCT

Macros
#define	UWOP_PUSH_NONVOL   0
#define	UWOP_ALLOC_LARGE   1
#define	UWOP_ALLOC_SMALL   2
#define	UWOP_SET_FPREG   3
#define	UWOP_SAVE_NONVOL   4
#define	UWOP_SAVE_NONVOL_FAR   5
#define	UWOP_EPILOG   6
#define	UWOP_SPARE_CODE   7
#define	UWOP_SAVE_XMM128   8
#define	UWOP_SAVE_XMM128_FAR   9
#define	UWOP_PUSH_MACHFRAME   10
#define	PARENT_OFFSET   0x100 /* Offset of the parent unwind info inside Unwind[]. */
#define	INIT_RAX   0xDEADBEEFDEADBEEFULL

Typedefs
typedef union _TEST_UNWIND_CODE	TEST_UNWIND_CODE
typedef struct _TEST_UNWIND_INFO	TEST_UNWIND_INFO
typedef struct _IMAGE_STRUCT	IMAGE_STRUCT

Enumerations
enum	{   REG_RAX = 0 , REG_RCX , REG_RDX , REG_RBX ,   REG_RSP , REG_RBP , REG_RSI , REG_RDI ,   REG_R8 , REG_R9 , REG_R10 , REG_R11 ,   REG_R12 , REG_R13 , REG_R14 , REG_R15 }

Functions
static	DECLSPEC_ALIGN (16)
static VOID	SetCode (TEST_UNWIND_INFO *Info, ULONG Index, UCHAR CodeOffset, UCHAR UnwindOp, UCHAR OpInfo)
static VOID	InitContext (PCONTEXT Context, ULONG_PTR Rsp)
static ULONG_PTR	DoUnwind (PCONTEXT Context, ULONG CodeOffset, PKNONVOLATILE_CONTEXT_POINTERS Pointers)
static ULONG_PTR	RunUnwind (PCONTEXT Context, ULONG_PTR Rsp, ULONG CodeOffset, PKNONVOLATILE_CONTEXT_POINTERS Pointers)
static VOID	Test_PushNonvol (VOID)
static VOID	Test_AllocSmall (VOID)
static VOID	Test_AllocLargeScaled (VOID)
static VOID	Test_AllocLargeUnscaled (VOID)
static VOID	Test_SetFpReg (VOID)
static VOID	Test_SaveNonvol (VOID)
static VOID	Test_SaveXmm128 (VOID)
static VOID	Test_PushMachframe (VOID)
static VOID	Test_TypicalProlog (VOID)
static VOID	Test_PartialProlog (VOID)
static VOID	Test_ContextPointers (VOID)
static VOID	Test_ChainedInfo (VOID)
static VOID	Test_SaveNonvolFar (VOID)
static VOID	Test_SaveXmm128Far (VOID)
static VOID	Test_Epilog (VOID)
static VOID	Test_SpareCode (VOID)
	START_TEST (RtlVirtualUnwind)

Macro Definition Documentation

INIT_RAX

#define INIT_RAX   0xDEADBEEFDEADBEEFULL

Definition at line 69 of file RtlVirtualUnwind.c.

PARENT_OFFSET

#define PARENT_OFFSET   0x100 /* Offset of the parent unwind info inside Unwind[]. */

Definition at line 67 of file RtlVirtualUnwind.c.

UWOP_ALLOC_LARGE

#define UWOP_ALLOC_LARGE   1

Definition at line 11 of file RtlVirtualUnwind.c.

UWOP_ALLOC_SMALL

#define UWOP_ALLOC_SMALL   2

Definition at line 12 of file RtlVirtualUnwind.c.

UWOP_EPILOG

#define UWOP_EPILOG   6

Definition at line 16 of file RtlVirtualUnwind.c.

UWOP_PUSH_MACHFRAME

#define UWOP_PUSH_MACHFRAME   10

Definition at line 20 of file RtlVirtualUnwind.c.

UWOP_PUSH_NONVOL

#define UWOP_PUSH_NONVOL   0

Definition at line 10 of file RtlVirtualUnwind.c.

UWOP_SAVE_NONVOL

#define UWOP_SAVE_NONVOL   4

Definition at line 14 of file RtlVirtualUnwind.c.

UWOP_SAVE_NONVOL_FAR

#define UWOP_SAVE_NONVOL_FAR   5

Definition at line 15 of file RtlVirtualUnwind.c.

UWOP_SAVE_XMM128

#define UWOP_SAVE_XMM128   8

Definition at line 18 of file RtlVirtualUnwind.c.

UWOP_SAVE_XMM128_FAR

#define UWOP_SAVE_XMM128_FAR   9

Definition at line 19 of file RtlVirtualUnwind.c.

UWOP_SET_FPREG

#define UWOP_SET_FPREG   3

Definition at line 13 of file RtlVirtualUnwind.c.

UWOP_SPARE_CODE

#define UWOP_SPARE_CODE   7

Definition at line 17 of file RtlVirtualUnwind.c.

Typedef Documentation

IMAGE_STRUCT

typedef struct _IMAGE_STRUCT IMAGE_STRUCT

TEST_UNWIND_CODE

typedef union _TEST_UNWIND_CODE TEST_UNWIND_CODE

TEST_UNWIND_INFO

typedef struct _TEST_UNWIND_INFO TEST_UNWIND_INFO

Enumeration Type Documentation

anonymous enum

anonymous enum

Enumerator
REG_RAX
REG_RCX
REG_RDX
REG_RBX
REG_RSP
REG_RBP
REG_RSI
REG_RDI
REG_R8
REG_R9
REG_R10
REG_R11
REG_R12
REG_R13
REG_R14
REG_R15

Definition at line 24 of file RtlVirtualUnwind.c.

{
    REG_RAX = 0, REG_RCX, REG_RDX, REG_RBX,
    REG_RSP,     REG_RBP, REG_RSI, REG_RDI,
    REG_R8,      REG_R9,  REG_R10, REG_R11,
    REG_R12,     REG_R13, REG_R14, REG_R15,
};

Function Documentation

DECLSPEC_ALIGN()

static DECLSPEC_ALIGN ( 16  )

static

Definition at line 71 of file RtlVirtualUnwind.c.

{
    TEST_UNWIND_INFO *Info = (TEST_UNWIND_INFO *)g_Image.Unwind;
    RtlZeroMemory(Info, sizeof(*Info));
    Info->Version = Version;
    Info->SizeOfProlog = SizeOfProlog;
    Info->CountOfCodes = CountOfCodes;
    return Info;
}

DoUnwind()

static ULONG_PTR DoUnwind ( PCONTEXT  Context, ULONG  CodeOffset, PKNONVOLATILE_CONTEXT_POINTERS  Pointers  )

static

Definition at line 108 of file RtlVirtualUnwind.c.

{
    RUNTIME_FUNCTION Func =
    {
        FIELD_OFFSET(IMAGE_STRUCT, Code),
        FIELD_OFFSET(IMAGE_STRUCT, Data),
        FIELD_OFFSET(IMAGE_STRUCT, Unwind)
    };
    PVOID HandlerData = NULL;
    ULONG_PTR EstablisherFrame = 0;
 
    RtlVirtualUnwind(UNW_FLAG_NHANDLER,
                     (ULONG_PTR)&g_Image,
                     (ULONG_PTR)g_Image.Code + CodeOffset,
                     &Func,
                     Context,
                     &HandlerData,
                     &EstablisherFrame,
                     Pointers);
    return EstablisherFrame;
}

Referenced by RunUnwind(), and Test_SetFpReg().

InitContext()

static VOID InitContext ( PCONTEXT  Context, ULONG_PTR  Rsp  )

static

Definition at line 100 of file RtlVirtualUnwind.c.

{
    RtlZeroMemory(Context, sizeof(*Context));
    Context->Rsp = Rsp;
    Context->Rax = INIT_RAX;
}

Referenced by RunUnwind(), and Test_SetFpReg().

RunUnwind()

static ULONG_PTR RunUnwind ( PCONTEXT  Context, ULONG_PTR  Rsp, ULONG  CodeOffset, PKNONVOLATILE_CONTEXT_POINTERS  Pointers  )

static

Definition at line 132 of file RtlVirtualUnwind.c.

{
    InitContext(Context, Rsp);
    return DoUnwind(Context, CodeOffset, Pointers);
}

Referenced by Test_AllocLargeScaled(), Test_AllocLargeUnscaled(), Test_AllocSmall(), Test_ChainedInfo(), Test_ContextPointers(), Test_Epilog(), Test_PartialProlog(), Test_PushMachframe(), Test_PushNonvol(), Test_SaveNonvol(), Test_SaveNonvolFar(), Test_SaveXmm128(), Test_SaveXmm128Far(), Test_SpareCode(), and Test_TypicalProlog().

SetCode()

static VOID SetCode ( TEST_UNWIND_INFO *  Info, ULONG  Index, UCHAR  CodeOffset, UCHAR  UnwindOp, UCHAR  OpInfo  )

static

Definition at line 91 of file RtlVirtualUnwind.c.

{
    Info->UnwindCode[Index].CodeOffset = CodeOffset;
    Info->UnwindCode[Index].UnwindOp = UnwindOp;
    Info->UnwindCode[Index].OpInfo = OpInfo;
}

Referenced by Test_AllocLargeScaled(), Test_AllocLargeUnscaled(), Test_AllocSmall(), Test_ChainedInfo(), Test_ContextPointers(), Test_Epilog(), Test_PartialProlog(), Test_PushMachframe(), Test_PushNonvol(), Test_SaveNonvol(), Test_SaveNonvolFar(), Test_SaveXmm128(), Test_SaveXmm128Far(), Test_SetFpReg(), Test_SpareCode(), and Test_TypicalProlog().

START_TEST()

START_TEST

(

RtlVirtualUnwind

)

Definition at line 534 of file RtlVirtualUnwind.c.

{
    Test_PushNonvol();
    Test_AllocSmall();
    Test_AllocLargeScaled();
    Test_AllocLargeUnscaled();
    Test_SetFpReg();
    Test_SaveNonvol();
    Test_SaveXmm128();
    Test_PushMachframe();
    Test_TypicalProlog();
    Test_PartialProlog();
    Test_ContextPointers();
    Test_ChainedInfo();
    Test_SaveNonvolFar();
    Test_SaveXmm128Far();
    Test_Epilog();
    Test_SpareCode();
}

Test_AllocLargeScaled()

static VOID Test_AllocLargeScaled ( VOID  )

static

Definition at line 176 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0x100 / 8] = 0x12345678ABCDEF00ULL;
 
    /* sub rsp, 0x100 with ALLOC_LARGE OpInfo=0: next slot * 8 */
    Info = ResetUnwindInfo(1, 7, 2);
    SetCode(Info, 0, 7, UWOP_ALLOC_LARGE, 0);
    Info->UnwindCode[1].FrameOffset = 0x100 / 8;
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 7, NULL), Stack);
    ok_eq_hex64(Ctx.Rip, 0x12345678ABCDEF00ULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 0x100 + 8);
}

Referenced by START_TEST().

Test_AllocLargeUnscaled()

static VOID Test_AllocLargeUnscaled ( VOID  )

static

Definition at line 195 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
    ULONG Size = 0x300;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[Size / 8] = 0xAA55AA55AA55AA55ULL;
 
    /* sub rsp, Size with ALLOC_LARGE OpInfo=1: next 2 slots = full ULONG */
    Info = ResetUnwindInfo(1, 7, 3);
    SetCode(Info, 0, 7, UWOP_ALLOC_LARGE, 1);
    *(ULONG *)&Info->UnwindCode[1] = Size;
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 7, NULL), Stack);
    ok_eq_hex64(Ctx.Rip, 0xAA55AA55AA55AA55ULL);
    ok_eq_hex64(Ctx.Rsp, Stack + Size + 8);
}

Referenced by START_TEST().

Test_AllocSmall()

static VOID Test_AllocSmall ( VOID  )

static

Definition at line 158 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[8] = 0xFEEDFACEFEEDFACEULL;
 
    /* sub rsp, 0x40 -> OpInfo = (0x40/8) - 1 = 7 */
    Info = ResetUnwindInfo(1, 4, 1);
    SetCode(Info, 0, 4, UWOP_ALLOC_SMALL, 7);
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 4, NULL), Stack);
    ok_eq_hex64(Ctx.Rip, 0xFEEDFACEFEEDFACEULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 0x40 + 8);
}

Referenced by START_TEST().

Test_ChainedInfo()

static VOID Test_ChainedInfo ( VOID  )

static

Definition at line 383 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Child, *Parent;
    PRUNTIME_FUNCTION ParentFunc;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* Child runs PUSH_NONVOL(RBX), then chains to a parent that runs
     * PUSH_NONVOL(RBP).  Final tail step pops Rip. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0] = 0xAA00AA00AA00AA00ULL; /* saved RBX */
    g_StackBuffer[1] = 0xBB00BB00BB00BB00ULL; /* saved RBP */
    g_StackBuffer[2] = 0xCC00CC00CC00CC00ULL; /* return Rip */
 
    Child = ResetUnwindInfo(1, 1, 1);
    Child->Flags = UNW_FLAG_CHAININFO;
    SetCode(Child, 0, 1, UWOP_PUSH_NONVOL, REG_RBX);
 
    /* Parent's RUNTIME_FUNCTION sits right after the rounded-up child
     * UnwindCode array, at &UnwindCode[(CountOfCodes + 1) & ~1]. */
    ParentFunc = (PRUNTIME_FUNCTION)&Child->UnwindCode[2];
    ParentFunc->BeginAddress = FIELD_OFFSET(IMAGE_STRUCT, Code);
    ParentFunc->EndAddress   = FIELD_OFFSET(IMAGE_STRUCT, Data);
    ParentFunc->UnwindData   = FIELD_OFFSET(IMAGE_STRUCT, Unwind) + PARENT_OFFSET;
 
    Parent = (TEST_UNWIND_INFO *)(g_Image.Unwind + PARENT_OFFSET);
    RtlZeroMemory(Parent, sizeof(*Parent));
    Parent->Version = 1;
    Parent->SizeOfProlog = 1;
    Parent->CountOfCodes = 1;
    SetCode(Parent, 0, 1, UWOP_PUSH_NONVOL, REG_RBP);
 
    RunUnwind(&Ctx, Stack, 1, NULL);
    ok_eq_hex64(Ctx.Rbx, 0xAA00AA00AA00AA00ULL);
    ok_eq_hex64(Ctx.Rbp, 0xBB00BB00BB00BB00ULL);
    ok_eq_hex64(Ctx.Rip, 0xCC00CC00CC00CC00ULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 0x18);
}

Referenced by START_TEST().

Test_ContextPointers()

static VOID Test_ContextPointers ( VOID  )

static

Definition at line 364 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    KNONVOLATILE_CONTEXT_POINTERS Ptrs;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    RtlZeroMemory(&Ptrs, sizeof(Ptrs));
    g_StackBuffer[0] = 0xAABBCCDDAABBCCDDULL;
 
    Info = ResetUnwindInfo(1, 1, 1);
    SetCode(Info, 0, 1, UWOP_PUSH_NONVOL, REG_RBX);
 
    RunUnwind(&Ctx, Stack, 1, &Ptrs);
    ok_eq_pointer(Ptrs.Rbx, &g_StackBuffer[0]);
    ok_eq_hex64(Ctx.Rbx, 0xAABBCCDDAABBCCDDULL);
}

Referenced by START_TEST().

Test_Epilog()

static VOID Test_Epilog ( VOID  )

static

Definition at line 478 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* UWOP_EPILOG consumes 2 slots.  If the unwinder advances by only 1,
     * the trailing zero slot is misread as UWOP_PUSH_NONVOL(RAX) and the
     * subsequent PUSH_NONVOL(RDI) sees a shifted Rsp. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0] = 0xAAAAAAAAAAAAAAAAULL;
    g_StackBuffer[1] = 0xBBBBBBBBBBBBBBBBULL;
 
    Info = ResetUnwindInfo(2, 0, 3);
    SetCode(Info, 0, 0, UWOP_EPILOG, 0);
    Info->UnwindCode[1].FrameOffset = 0;
    SetCode(Info, 2, 0, UWOP_PUSH_NONVOL, REG_RDI);
 
    RunUnwind(&Ctx, Stack, 0, NULL);
    ok_eq_hex64(Ctx.Rdi, 0xAAAAAAAAAAAAAAAAULL);
    ok_eq_hex64(Ctx.Rax, INIT_RAX);
}

Referenced by START_TEST().

Test_PartialProlog()

static VOID Test_PartialProlog ( VOID  )

static

Definition at line 337 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* Same prolog but ControlPc is between push rbx and sub rsp.
     * Only push rbp / push rbx have executed, so sub rsp must NOT undo. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0] = 0x2222222222222222ULL;
    g_StackBuffer[1] = 0x3333333333333333ULL;
    g_StackBuffer[2] = 0x4444444444444444ULL;
 
    Info = ResetUnwindInfo(1, 7, 4);
    SetCode(Info, 0, 7, UWOP_ALLOC_SMALL, 7);
    SetCode(Info, 1, 3, UWOP_PUSH_NONVOL, REG_RSI);
    SetCode(Info, 2, 2, UWOP_PUSH_NONVOL, REG_RBX);
    SetCode(Info, 3, 1, UWOP_PUSH_NONVOL, REG_RBP);
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 2, NULL), Stack);
    ok_eq_hex64(Ctx.Rbx, 0x2222222222222222ULL);
    ok_eq_hex64(Ctx.Rbp, 0x3333333333333333ULL);
    ok_eq_hex64(Ctx.Rip, 0x4444444444444444ULL);
    ok_eq_hex64(Ctx.Rsi, 0); /* push rsi not yet executed */
    ok_eq_hex64(Ctx.Rsp, Stack + 0x18);
}

Referenced by START_TEST().

Test_PushMachframe()

static VOID Test_PushMachframe ( VOID  )

static

Definition at line 284 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* MACHINE_FRAME layout: [Rip][Cs][EFlags][Rsp][Ss], OpInfo=1 means an
     * error code was pushed (one extra QWORD before the frame). */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[1] = 0xCAFEBABE00000000ULL; /* Rip */
    g_StackBuffer[4] = Stack + 0x100;         /* Rsp */
 
    Info = ResetUnwindInfo(1, 1, 1);
    SetCode(Info, 0, 1, UWOP_PUSH_MACHFRAME, 1);
 
    RunUnwind(&Ctx, Stack, 1, NULL);
    ok_eq_hex64(Ctx.Rip, 0xCAFEBABE00000000ULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 0x100);
}

Referenced by START_TEST().

Test_PushNonvol()

static VOID Test_PushNonvol ( VOID  )

static

Definition at line 139 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0] = 0x1111111122222222ULL;
    g_StackBuffer[1] = 0xDEADC0DEDEADC0DEULL;
 
    Info = ResetUnwindInfo(1, 1, 1);
    SetCode(Info, 0, 1, UWOP_PUSH_NONVOL, REG_RBP);
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 1, NULL), Stack);
    ok_eq_hex64(Ctx.Rbp, 0x1111111122222222ULL);
    ok_eq_hex64(Ctx.Rip, 0xDEADC0DEDEADC0DEULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 16);
}

Referenced by START_TEST().

Test_SaveNonvol()

static VOID Test_SaveNonvol ( VOID  )

static

Definition at line 241 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0x80 / 8] = 0x4242424242424242ULL;
    g_StackBuffer[0]        = 0xDEADC0DEDEADC0DEULL;
 
    /* mov [rsp + 0x80], rsi; FrameOffset (in 8-byte units) = 0x10 */
    Info = ResetUnwindInfo(1, 8, 2);
    SetCode(Info, 0, 8, UWOP_SAVE_NONVOL, REG_RSI);
    Info->UnwindCode[1].FrameOffset = 0x10;
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 8, NULL), Stack);
    ok_eq_hex64(Ctx.Rsi, 0x4242424242424242ULL);
    ok_eq_hex64(Ctx.Rip, 0xDEADC0DEDEADC0DEULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 8);
}

Referenced by START_TEST().

Test_SaveNonvolFar()

static VOID Test_SaveNonvolFar ( VOID  )

static

Definition at line 422 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* Expected at unscaled byte offset 8, decoy at the buggy
     * (DWORD64*)Rsp + 8 == Rsp + 64 location. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[1] = 0xAAAAAAAABBBBBBBBULL; /* byte offset 8 */
    g_StackBuffer[8] = 0xCCCCCCCCDDDDDDDDULL; /* byte offset 64 */
 
    Info = ResetUnwindInfo(2, 0, 3);
    SetCode(Info, 0, 0, UWOP_SAVE_NONVOL_FAR, REG_RBX);
    *(ULONG *)&Info->UnwindCode[1] = 8;
 
    RunUnwind(&Ctx, Stack, 0, NULL);
    ok_eq_hex64(Ctx.Rbx, 0xAAAAAAAABBBBBBBBULL);
    ok(Ctx.Rbx != 0xCCCCCCCCDDDDDDDDULL,
       "Rbx looks scaled-by-8: %I64x\n", Ctx.Rbx);
    /* If the unwinder mis-advances i, trailing zero slots decode as
     * UWOP_PUSH_NONVOL(RAX) and shift Rsp / clobber Rax. */
    ok_eq_hex64(Ctx.Rax, INIT_RAX);
    ok_eq_hex64(Ctx.Rsp, Stack + 8);
}

Referenced by START_TEST().

Test_SaveXmm128()

static VOID Test_SaveXmm128 ( VOID  )

static

Definition at line 262 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
    M128A Expected;
 
    Expected.Low  = 0x0011223344556677ULL;
    Expected.High = 0x8899AABBCCDDEEFFULL;
 
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    *(M128A *)&g_StackBuffer[0x40 / 8] = Expected;
 
    /* movaps [rsp + 0x40], xmm7; FrameOffset (in 16-byte units) = 4 */
    Info = ResetUnwindInfo(1, 9, 2);
    SetCode(Info, 0, 9, UWOP_SAVE_XMM128, 7 /* XMM7 */);
    Info->UnwindCode[1].FrameOffset = 4;
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 9, NULL), Stack);
    ok_eq_xmm(Ctx.Xmm7, Expected);
}

Referenced by START_TEST().

Test_SaveXmm128Far()

static VOID Test_SaveXmm128Far ( VOID  )

static

Definition at line 448 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
    M128A Expected, Decoy;
 
    Expected.Low  = 0x1111111122222222ULL;
    Expected.High = 0x3333333344444444ULL;
    Decoy.Low     = 0x5555555566666666ULL;
    Decoy.High    = 0x7777777788888888ULL;
 
    /* Expected at unscaled byte offset 16, decoy at the buggy
     * (M128A*)Rsp + 16 == Rsp + 256 location. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    *(M128A *)&g_StackBuffer[16 / 8]  = Expected;
    *(M128A *)&g_StackBuffer[256 / 8] = Decoy;
 
    Info = ResetUnwindInfo(2, 0, 3);
    SetCode(Info, 0, 0, UWOP_SAVE_XMM128_FAR, 6 /* XMM6 */);
    *(ULONG *)&Info->UnwindCode[1] = 16;
 
    RunUnwind(&Ctx, Stack, 0, NULL);
    ok_eq_xmm(Ctx.Xmm6, Expected);
    ok(Ctx.Xmm6.Low != Decoy.Low,
       "Xmm6 looks scaled-by-16: %I64x\n", Ctx.Xmm6.Low);
    ok_eq_hex64(Ctx.Rax, INIT_RAX);
    ok_eq_hex64(Ctx.Rsp, Stack + 8);
}

Referenced by START_TEST().

Test_SetFpReg()

static VOID Test_SetFpReg ( VOID  )

static

Definition at line 215 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR EstFrame;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
    ULONG_PTR FrameTop = Stack + 0x100;
 
    /* lea rbp, [rsp + 0x20]; FrameOffset (in 16-byte units) = 2 */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0x100 / 8] = 0x9999AAAA9999AAAAULL;
 
    Info = ResetUnwindInfo(1, 4, 1);
    Info->FrameRegister = REG_RBP;
    Info->FrameOffset = 2;
    SetCode(Info, 0, 4, UWOP_SET_FPREG, 0);
 
    InitContext(&Ctx, Stack);
    Ctx.Rbp = FrameTop + 0x20;
    EstFrame = DoUnwind(&Ctx, 4, NULL);
 
    ok_eq_hex64(EstFrame, FrameTop);
    ok_eq_hex64(Ctx.Rip, 0x9999AAAA9999AAAAULL);
    ok_eq_hex64(Ctx.Rsp, FrameTop + 8);
}

Referenced by START_TEST().

Test_SpareCode()

static VOID Test_SpareCode ( VOID  )

static

Definition at line 501 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* Same idea as Test_Epilog, but UWOP_SPARE_CODE consumes 3 slots.
     * The unwinder hits ASSERT(FALSE) before advancing i, so on debug
     * (DBG=1) builds the fix path is unreachable; catch and skip. */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0] = 0xAAAAAAAAAAAAAAAAULL;
    g_StackBuffer[1] = 0xBBBBBBBBBBBBBBBBULL;
 
    Info = ResetUnwindInfo(2, 0, 4);
    SetCode(Info, 0, 0, UWOP_SPARE_CODE, 0);
    Info->UnwindCode[1].FrameOffset = 0;
    Info->UnwindCode[2].FrameOffset = 0;
    SetCode(Info, 3, 0, UWOP_PUSH_NONVOL, REG_RDI);
 
    _SEH2_TRY
    {
        RunUnwind(&Ctx, Stack, 0, NULL);
        ok_eq_hex64(Ctx.Rdi, 0xAAAAAAAAAAAAAAAAULL);
        ok_eq_hex64(Ctx.Rax, INIT_RAX);
    }
    _SEH2_EXCEPT(EXCEPTION_EXECUTE_HANDLER)
    {
        skip("UWOP_SPARE_CODE triggered exception 0x%08lx (debug build ASSERT)\n",
             _SEH2_GetExceptionCode());
    }
    _SEH2_END;
}

Referenced by START_TEST().

Test_TypicalProlog()

static VOID Test_TypicalProlog ( VOID  )

static

Definition at line 304 of file RtlVirtualUnwind.c.

{
    CONTEXT Ctx;
    TEST_UNWIND_INFO *Info;
    ULONG_PTR Stack = (ULONG_PTR)g_StackBuffer;
 
    /* Equivalent to:
     *      push rbp        ; CodeOffset 1
     *      push rbx        ; CodeOffset 2
     *      push rsi        ; CodeOffset 3
     *      sub  rsp, 0x40  ; CodeOffset 7  (4-byte instruction)
     */
    RtlZeroMemory(g_StackBuffer, sizeof(g_StackBuffer));
    g_StackBuffer[0x40 / 8] = 0x1111111111111111ULL; /* saved RSI */
    g_StackBuffer[0x48 / 8] = 0x2222222222222222ULL; /* saved RBX */
    g_StackBuffer[0x50 / 8] = 0x3333333333333333ULL; /* saved RBP */
    g_StackBuffer[0x58 / 8] = 0x4444444444444444ULL; /* return Rip */
 
    Info = ResetUnwindInfo(1, 7, 4);
    SetCode(Info, 0, 7, UWOP_ALLOC_SMALL, 7);
    SetCode(Info, 1, 3, UWOP_PUSH_NONVOL, REG_RSI);
    SetCode(Info, 2, 2, UWOP_PUSH_NONVOL, REG_RBX);
    SetCode(Info, 3, 1, UWOP_PUSH_NONVOL, REG_RBP);
 
    ok_eq_hex64(RunUnwind(&Ctx, Stack, 7, NULL), Stack);
    ok_eq_hex64(Ctx.Rsi, 0x1111111111111111ULL);
    ok_eq_hex64(Ctx.Rbx, 0x2222222222222222ULL);
    ok_eq_hex64(Ctx.Rbp, 0x3333333333333333ULL);
    ok_eq_hex64(Ctx.Rip, 0x4444444444444444ULL);
    ok_eq_hex64(Ctx.Rsp, Stack + 0x60);
    ok_eq_hex64(Ctx.Rax, INIT_RAX);
}

Referenced by START_TEST().