X86UnitTestHost.c

Go to the documentation of this file.

 
#include "UnitTestHost.h"
 
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_CS    0
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_DS    1
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_ES    2
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_FS    3
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_GS    4
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_SS    5
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_TR    6
#define UNIT_TEST_HOST_BASE_LIB_SEGMENT_LDTR  7
 
STATIC UINT64           mUnitTestHostBaseLibMsr[2][0x1000];
STATIC UINTN            mUnitTestHostBaseLibCr[5];
STATIC UINTN            mUnitTestHostBaseLibDr[8];
STATIC UINT16           mUnitTestHostBaseLibSegment[8];
STATIC IA32_DESCRIPTOR  mUnitTestHostBaseLibGdtr;
STATIC IA32_DESCRIPTOR  mUnitTestHostBaseLibIdtr;
 
UINT32
EFIAPI
UnitTestHostBaseLibAsmCpuid (
  IN      UINT32  Index,
  OUT     UINT32  *Eax   OPTIONAL,
  OUT     UINT32  *Ebx   OPTIONAL,
  OUT     UINT32  *Ecx   OPTIONAL,
  OUT     UINT32  *Edx   OPTIONAL
  )
{
  UINT32  RetEcx;
 
  RetEcx = 0;
  switch (Index) {
    case 1:
      RetEcx |= BIT30; /* RdRand */
      break;
  }
 
  if (Eax != NULL) {
    *Eax = 0;
  }
 
  if (Ebx != NULL) {
    *Ebx = 0;
  }
 
  if (Ecx != NULL) {
    *Ecx = RetEcx;
  }
 
  if (Edx != NULL) {
    *Edx = 0;
  }
 
  return Index;
}
 
UINT32
EFIAPI
UnitTestHostBaseLibAsmCpuidEx (
  IN      UINT32  Index,
  IN      UINT32  SubIndex,
  OUT     UINT32  *Eax   OPTIONAL,
  OUT     UINT32  *Ebx   OPTIONAL,
  OUT     UINT32  *Ecx   OPTIONAL,
  OUT     UINT32  *Edx   OPTIONAL
  )
{
  if (Eax != NULL) {
    *Eax = 0;
  }
 
  if (Ebx != NULL) {
    *Ebx = 0;
  }
 
  if (Ecx != NULL) {
    *Ecx = 0;
  }
 
  if (Edx != NULL) {
    *Edx = 0;
  }
 
  return Index;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmDisableCache (
  VOID
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmEnableCache (
  VOID
  )
{
}
 
UINT64
EFIAPI
UnitTestHostBaseLibAsmReadMsr64 (
  IN      UINT32  Index
  )
{
  if (Index < 0x1000) {
    return mUnitTestHostBaseLibMsr[0][Index];
  }
 
  if ((Index >= 0xC0000000) && (Index < 0xC0001000)) {
    return mUnitTestHostBaseLibMsr[1][Index];
  }
 
  return 0;
}
 
UINT64
EFIAPI
UnitTestHostBaseLibAsmWriteMsr64 (
  IN      UINT32  Index,
  IN      UINT64  Value
  )
{
  if (Index < 0x1000) {
    mUnitTestHostBaseLibMsr[0][Index] = Value;
  }
 
  if ((Index >= 0xC0000000) && (Index < 0xC0001000)) {
    mUnitTestHostBaseLibMsr[1][Index - 0xC00000000] = Value;
  }
 
  return Value;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadCr0 (
  VOID
  )
{
  return mUnitTestHostBaseLibCr[0];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadCr2 (
  VOID
  )
{
  return mUnitTestHostBaseLibCr[2];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadCr3 (
  VOID
  )
{
  return mUnitTestHostBaseLibCr[3];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadCr4 (
  VOID
  )
{
  return mUnitTestHostBaseLibCr[4];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteCr0 (
  UINTN  Cr0
  )
{
  mUnitTestHostBaseLibCr[0] = Cr0;
  return Cr0;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteCr2 (
  UINTN  Cr2
  )
{
  mUnitTestHostBaseLibCr[2] = Cr2;
  return Cr2;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteCr3 (
  UINTN  Cr3
  )
{
  mUnitTestHostBaseLibCr[3] = Cr3;
  return Cr3;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteCr4 (
  UINTN  Cr4
  )
{
  mUnitTestHostBaseLibCr[4] = Cr4;
  return Cr4;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr0 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[0];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr1 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[1];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr2 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[2];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr3 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[3];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr4 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[4];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr5 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[5];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr6 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[6];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmReadDr7 (
  VOID
  )
{
  return mUnitTestHostBaseLibDr[7];
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr0 (
  UINTN  Dr0
  )
{
  mUnitTestHostBaseLibDr[0] = Dr0;
  return Dr0;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr1 (
  UINTN  Dr1
  )
{
  mUnitTestHostBaseLibDr[1] = Dr1;
  return Dr1;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr2 (
  UINTN  Dr2
  )
{
  mUnitTestHostBaseLibDr[2] = Dr2;
  return Dr2;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr3 (
  UINTN  Dr3
  )
{
  mUnitTestHostBaseLibDr[3] = Dr3;
  return Dr3;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr4 (
  UINTN  Dr4
  )
{
  mUnitTestHostBaseLibDr[4] = Dr4;
  return Dr4;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr5 (
  UINTN  Dr5
  )
{
  mUnitTestHostBaseLibDr[5] = Dr5;
  return Dr5;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr6 (
  UINTN  Dr6
  )
{
  mUnitTestHostBaseLibDr[6] = Dr6;
  return Dr6;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmWriteDr7 (
  UINTN  Dr7
  )
{
  mUnitTestHostBaseLibDr[7] = Dr7;
  return Dr7;
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadCs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_CS];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadDs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_DS];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadEs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_ES];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadFs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_FS];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadGs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_GS];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadSs (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_SS];
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadTr (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_TR];
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmReadGdtr (
  OUT     IA32_DESCRIPTOR  *Gdtr
  )
{
  Gdtr = &mUnitTestHostBaseLibGdtr;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmWriteGdtr (
  IN      CONST IA32_DESCRIPTOR  *Gdtr
  )
{
  CopyMem (&mUnitTestHostBaseLibGdtr, Gdtr, sizeof (IA32_DESCRIPTOR));
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmReadIdtr (
  OUT     IA32_DESCRIPTOR  *Idtr
  )
{
  Idtr = &mUnitTestHostBaseLibIdtr;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmWriteIdtr (
  IN      CONST IA32_DESCRIPTOR  *Idtr
  )
{
  CopyMem (&mUnitTestHostBaseLibIdtr, Idtr, sizeof (IA32_DESCRIPTOR));
}
 
UINT16
EFIAPI
UnitTestHostBaseLibAsmReadLdtr (
  VOID
  )
{
  return mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_LDTR];
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmWriteLdtr (
  IN      UINT16  Ldtr
  )
{
  mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_LDTR] = Ldtr;
}
 
UINT64
EFIAPI
UnitTestHostBaseLibAsmReadPmc (
  IN      UINT32  Index
  )
{
  return 0;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmMonitor (
  IN      UINTN  Eax,
  IN      UINTN  Ecx,
  IN      UINTN  Edx
  )
{
  return Eax;
}
 
UINTN
EFIAPI
UnitTestHostBaseLibAsmMwait (
  IN      UINTN  Eax,
  IN      UINTN  Ecx
  )
{
  return Eax;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmWbinvd (
  VOID
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmInvd (
  VOID
  )
{
}
 
VOID *
EFIAPI
UnitTestHostBaseLibAsmFlushCacheLine (
  IN      VOID  *LinearAddress
  )
{
  return LinearAddress;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmEnablePaging32 (
  IN      SWITCH_STACK_ENTRY_POINT  EntryPoint,
  IN      VOID                      *Context1   OPTIONAL,
  IN      VOID                      *Context2   OPTIONAL,
  IN      VOID                      *NewStack
  )
{
  EntryPoint (Context1, Context2);
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmDisablePaging32 (
  IN      SWITCH_STACK_ENTRY_POINT  EntryPoint,
  IN      VOID                      *Context1   OPTIONAL,
  IN      VOID                      *Context2   OPTIONAL,
  IN      VOID                      *NewStack
  )
{
  EntryPoint (Context1, Context2);
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmEnablePaging64 (
  IN      UINT16  Cs,
  IN      UINT64  EntryPoint,
  IN      UINT64  Context1   OPTIONAL,
  IN      UINT64  Context2   OPTIONAL,
  IN      UINT64  NewStack
  )
{
  SWITCH_STACK_ENTRY_POINT  NewEntryPoint;
 
  NewEntryPoint = (SWITCH_STACK_ENTRY_POINT)(UINTN)(EntryPoint);
  NewEntryPoint ((VOID *)(UINTN)Context1, (VOID *)(UINTN)Context2);
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmDisablePaging64 (
  IN      UINT16  Cs,
  IN      UINT32  EntryPoint,
  IN      UINT32  Context1   OPTIONAL,
  IN      UINT32  Context2   OPTIONAL,
  IN      UINT32  NewStack
  )
{
  SWITCH_STACK_ENTRY_POINT  NewEntryPoint;
 
  NewEntryPoint = (SWITCH_STACK_ENTRY_POINT)(UINTN)(EntryPoint);
  NewEntryPoint ((VOID *)(UINTN)Context1, (VOID *)(UINTN)Context2);
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmGetThunk16Properties (
  OUT     UINT32  *RealModeBufferSize,
  OUT     UINT32  *ExtraStackSize
  )
{
  *RealModeBufferSize = 0;
  *ExtraStackSize     = 0;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmPrepareThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmPrepareAndThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmWriteTr (
  IN UINT16  Selector
  )
{
  mUnitTestHostBaseLibSegment[UNIT_TEST_HOST_BASE_LIB_SEGMENT_TR] = Selector;
}
 
VOID
EFIAPI
UnitTestHostBaseLibAsmLfence (
  VOID
  )
{
}
 
VOID
EFIAPI
UnitTestHostBaseLibPatchInstructionX86 (
  OUT X86_ASSEMBLY_PATCH_LABEL  *InstructionEnd,
  IN  UINT64                    PatchValue,
  IN  UINTN                     ValueSize
  )
{
}
 
UINT32
EFIAPI
AsmCpuid (
  IN      UINT32  Index,
  OUT     UINT32  *Eax   OPTIONAL,
  OUT     UINT32  *Ebx   OPTIONAL,
  OUT     UINT32  *Ecx   OPTIONAL,
  OUT     UINT32  *Edx   OPTIONAL
  )
{
  return gUnitTestHostBaseLib.X86->AsmCpuid (Index, Eax, Ebx, Ecx, Edx);
}
 
UINT32
EFIAPI
AsmCpuidEx (
  IN      UINT32  Index,
  IN      UINT32  SubIndex,
  OUT     UINT32  *Eax   OPTIONAL,
  OUT     UINT32  *Ebx   OPTIONAL,
  OUT     UINT32  *Ecx   OPTIONAL,
  OUT     UINT32  *Edx   OPTIONAL
  )
{
  return gUnitTestHostBaseLib.X86->AsmCpuidEx (Index, SubIndex, Eax, Ebx, Ecx, Edx);
}
 
VOID
EFIAPI
AsmDisableCache (
  VOID
  )
{
  gUnitTestHostBaseLib.X86->AsmDisableCache ();
}
 
VOID
EFIAPI
AsmEnableCache (
  VOID
  )
{
  gUnitTestHostBaseLib.X86->AsmEnableCache ();
}
 
UINT64
EFIAPI
AsmReadMsr64 (
  IN      UINT32  Index
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadMsr64 (Index);
}
 
UINT64
EFIAPI
AsmWriteMsr64 (
  IN      UINT32  Index,
  IN      UINT64  Value
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteMsr64 (Index, Value);
}
 
UINTN
EFIAPI
AsmReadCr0 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadCr0 ();
}
 
UINTN
EFIAPI
AsmReadCr2 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadCr2 ();
}
 
UINTN
EFIAPI
AsmReadCr3 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadCr3 ();
}
 
UINTN
EFIAPI
AsmReadCr4 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadCr4 ();
}
 
UINTN
EFIAPI
AsmWriteCr0 (
  UINTN  Cr0
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteCr0 (Cr0);
}
 
UINTN
EFIAPI
AsmWriteCr2 (
  UINTN  Cr2
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteCr2 (Cr2);
}
 
UINTN
EFIAPI
AsmWriteCr3 (
  UINTN  Cr3
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteCr3 (Cr3);
}
 
UINTN
EFIAPI
AsmWriteCr4 (
  UINTN  Cr4
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteCr4 (Cr4);
}
 
UINTN
EFIAPI
AsmReadDr0 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr0 ();
}
 
UINTN
EFIAPI
AsmReadDr1 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr1 ();
}
 
UINTN
EFIAPI
AsmReadDr2 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr2 ();
}
 
UINTN
EFIAPI
AsmReadDr3 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr3 ();
}
 
UINTN
EFIAPI
AsmReadDr4 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr4 ();
}
 
UINTN
EFIAPI
AsmReadDr5 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr5 ();
}
 
UINTN
EFIAPI
AsmReadDr6 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr6 ();
}
 
UINTN
EFIAPI
AsmReadDr7 (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDr7 ();
}
 
UINTN
EFIAPI
AsmWriteDr0 (
  UINTN  Dr0
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr0 (Dr0);
}
 
UINTN
EFIAPI
AsmWriteDr1 (
  UINTN  Dr1
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr1 (Dr1);
}
 
UINTN
EFIAPI
AsmWriteDr2 (
  UINTN  Dr2
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr2 (Dr2);
}
 
UINTN
EFIAPI
AsmWriteDr3 (
  UINTN  Dr3
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr3 (Dr3);
}
 
UINTN
EFIAPI
AsmWriteDr4 (
  UINTN  Dr4
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr4 (Dr4);
}
 
UINTN
EFIAPI
AsmWriteDr5 (
  UINTN  Dr5
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr5 (Dr5);
}
 
UINTN
EFIAPI
AsmWriteDr6 (
  UINTN  Dr6
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr6 (Dr6);
}
 
UINTN
EFIAPI
AsmWriteDr7 (
  UINTN  Dr7
  )
{
  return gUnitTestHostBaseLib.X86->AsmWriteDr7 (Dr7);
}
 
UINT16
EFIAPI
AsmReadCs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadCs ();
}
 
UINT16
EFIAPI
AsmReadDs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadDs ();
}
 
UINT16
EFIAPI
AsmReadEs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadEs ();
}
 
UINT16
EFIAPI
AsmReadFs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadFs ();
}
 
UINT16
EFIAPI
AsmReadGs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadGs ();
}
 
UINT16
EFIAPI
AsmReadSs (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadSs ();
}
 
UINT16
EFIAPI
AsmReadTr (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadTr ();
}
 
VOID
EFIAPI
AsmReadGdtr (
  OUT     IA32_DESCRIPTOR  *Gdtr
  )
{
  gUnitTestHostBaseLib.X86->AsmReadGdtr (Gdtr);
}
 
VOID
EFIAPI
AsmWriteGdtr (
  IN      CONST IA32_DESCRIPTOR  *Gdtr
  )
{
  gUnitTestHostBaseLib.X86->AsmWriteGdtr (Gdtr);
}
 
VOID
EFIAPI
AsmReadIdtr (
  OUT     IA32_DESCRIPTOR  *Idtr
  )
{
  gUnitTestHostBaseLib.X86->AsmReadIdtr (Idtr);
}
 
VOID
EFIAPI
AsmWriteIdtr (
  IN      CONST IA32_DESCRIPTOR  *Idtr
  )
{
  gUnitTestHostBaseLib.X86->AsmWriteIdtr (Idtr);
}
 
UINT16
EFIAPI
AsmReadLdtr (
  VOID
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadLdtr ();
}
 
VOID
EFIAPI
AsmWriteLdtr (
  IN      UINT16  Ldtr
  )
{
  gUnitTestHostBaseLib.X86->AsmWriteLdtr (Ldtr);
}
 
UINT64
EFIAPI
AsmReadPmc (
  IN      UINT32  Index
  )
{
  return gUnitTestHostBaseLib.X86->AsmReadPmc (Index);
}
 
UINTN
EFIAPI
AsmMonitor (
  IN      UINTN  Eax,
  IN      UINTN  Ecx,
  IN      UINTN  Edx
  )
{
  return gUnitTestHostBaseLib.X86->AsmMonitor (Eax, Ecx, Edx);
}
 
UINTN
EFIAPI
AsmMwait (
  IN      UINTN  Eax,
  IN      UINTN  Ecx
  )
{
  return gUnitTestHostBaseLib.X86->AsmMwait (Eax, Ecx);
}
 
VOID
EFIAPI
AsmWbinvd (
  VOID
  )
{
  gUnitTestHostBaseLib.X86->AsmWbinvd ();
}
 
VOID
EFIAPI
AsmInvd (
  VOID
  )
{
  gUnitTestHostBaseLib.X86->AsmInvd ();
}
 
VOID *
EFIAPI
AsmFlushCacheLine (
  IN      VOID  *LinearAddress
  )
{
  return gUnitTestHostBaseLib.X86->AsmFlushCacheLine (LinearAddress);
}
 
VOID
EFIAPI
AsmEnablePaging32 (
  IN      SWITCH_STACK_ENTRY_POINT  EntryPoint,
  IN      VOID                      *Context1   OPTIONAL,
  IN      VOID                      *Context2   OPTIONAL,
  IN      VOID                      *NewStack
  )
{
  gUnitTestHostBaseLib.X86->AsmEnablePaging32 (EntryPoint, Context1, Context2, NewStack);
}
 
VOID
EFIAPI
AsmDisablePaging32 (
  IN      SWITCH_STACK_ENTRY_POINT  EntryPoint,
  IN      VOID                      *Context1   OPTIONAL,
  IN      VOID                      *Context2   OPTIONAL,
  IN      VOID                      *NewStack
  )
{
  gUnitTestHostBaseLib.X86->AsmDisablePaging32 (EntryPoint, Context1, Context2, NewStack);
}
 
VOID
EFIAPI
AsmEnablePaging64 (
  IN      UINT16  Cs,
  IN      UINT64  EntryPoint,
  IN      UINT64  Context1   OPTIONAL,
  IN      UINT64  Context2   OPTIONAL,
  IN      UINT64  NewStack
  )
{
  gUnitTestHostBaseLib.X86->AsmEnablePaging64 (Cs, EntryPoint, Context1, Context2, NewStack);
}
 
VOID
EFIAPI
AsmDisablePaging64 (
  IN      UINT16  Cs,
  IN      UINT32  EntryPoint,
  IN      UINT32  Context1   OPTIONAL,
  IN      UINT32  Context2   OPTIONAL,
  IN      UINT32  NewStack
  )
{
  gUnitTestHostBaseLib.X86->AsmDisablePaging64 (Cs, EntryPoint, Context1, Context2, NewStack);
}
 
VOID
EFIAPI
AsmGetThunk16Properties (
  OUT     UINT32  *RealModeBufferSize,
  OUT     UINT32  *ExtraStackSize
  )
{
  gUnitTestHostBaseLib.X86->AsmGetThunk16Properties (RealModeBufferSize, ExtraStackSize);
}
 
VOID
EFIAPI
AsmPrepareThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
  gUnitTestHostBaseLib.X86->AsmPrepareThunk16 (ThunkContext);
}
 
VOID
EFIAPI
AsmThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
  gUnitTestHostBaseLib.X86->AsmThunk16 (ThunkContext);
}
 
VOID
EFIAPI
AsmPrepareAndThunk16 (
  IN OUT  THUNK_CONTEXT  *ThunkContext
  )
{
  gUnitTestHostBaseLib.X86->AsmPrepareAndThunk16 (ThunkContext);
}
 
VOID
EFIAPI
AsmWriteTr (
  IN UINT16  Selector
  )
{
  gUnitTestHostBaseLib.X86->AsmWriteTr (Selector);
}
 
VOID
EFIAPI
AsmLfence (
  VOID
  )
{
  gUnitTestHostBaseLib.X86->AsmLfence ();
}
 
VOID
EFIAPI
PatchInstructionX86 (
  OUT X86_ASSEMBLY_PATCH_LABEL  *InstructionEnd,
  IN  UINT64                    PatchValue,
  IN  UINTN                     ValueSize
  )
{
  gUnitTestHostBaseLib.X86->PatchInstructionX86 (InstructionEnd, PatchValue, ValueSize);
}
 
STATIC UNIT_TEST_HOST_BASE_LIB_COMMON  mUnitTestHostBaseLibCommon = {
  UnitTestHostBaseLibEnableInterrupts,
  UnitTestHostBaseLibDisableInterrupts,
  UnitTestHostBaseLibEnableDisableInterrupts,
  UnitTestHostBaseLibGetInterruptState,
};
 
STATIC UNIT_TEST_HOST_BASE_LIB_X86  mUnitTestHostBaseLibX86 = {
  UnitTestHostBaseLibAsmCpuid,
  UnitTestHostBaseLibAsmCpuidEx,
  UnitTestHostBaseLibAsmDisableCache,
  UnitTestHostBaseLibAsmEnableCache,
  UnitTestHostBaseLibAsmReadMsr64,
  UnitTestHostBaseLibAsmWriteMsr64,
  UnitTestHostBaseLibAsmReadCr0,
  UnitTestHostBaseLibAsmReadCr2,
  UnitTestHostBaseLibAsmReadCr3,
  UnitTestHostBaseLibAsmReadCr4,
  UnitTestHostBaseLibAsmWriteCr0,
  UnitTestHostBaseLibAsmWriteCr2,
  UnitTestHostBaseLibAsmWriteCr3,
  UnitTestHostBaseLibAsmWriteCr4,
  UnitTestHostBaseLibAsmReadDr0,
  UnitTestHostBaseLibAsmReadDr1,
  UnitTestHostBaseLibAsmReadDr2,
  UnitTestHostBaseLibAsmReadDr3,
  UnitTestHostBaseLibAsmReadDr4,
  UnitTestHostBaseLibAsmReadDr5,
  UnitTestHostBaseLibAsmReadDr6,
  UnitTestHostBaseLibAsmReadDr7,
  UnitTestHostBaseLibAsmWriteDr0,
  UnitTestHostBaseLibAsmWriteDr1,
  UnitTestHostBaseLibAsmWriteDr2,
  UnitTestHostBaseLibAsmWriteDr3,
  UnitTestHostBaseLibAsmWriteDr4,
  UnitTestHostBaseLibAsmWriteDr5,
  UnitTestHostBaseLibAsmWriteDr6,
  UnitTestHostBaseLibAsmWriteDr7,
  UnitTestHostBaseLibAsmReadCs,
  UnitTestHostBaseLibAsmReadDs,
  UnitTestHostBaseLibAsmReadEs,
  UnitTestHostBaseLibAsmReadFs,
  UnitTestHostBaseLibAsmReadGs,
  UnitTestHostBaseLibAsmReadSs,
  UnitTestHostBaseLibAsmReadTr,
  UnitTestHostBaseLibAsmReadGdtr,
  UnitTestHostBaseLibAsmWriteGdtr,
  UnitTestHostBaseLibAsmReadIdtr,
  UnitTestHostBaseLibAsmWriteIdtr,
  UnitTestHostBaseLibAsmReadLdtr,
  UnitTestHostBaseLibAsmWriteLdtr,
  UnitTestHostBaseLibAsmReadPmc,
  UnitTestHostBaseLibAsmMonitor,
  UnitTestHostBaseLibAsmMwait,
  UnitTestHostBaseLibAsmWbinvd,
  UnitTestHostBaseLibAsmInvd,
  UnitTestHostBaseLibAsmFlushCacheLine,
  UnitTestHostBaseLibAsmEnablePaging32,
  UnitTestHostBaseLibAsmDisablePaging32,
  UnitTestHostBaseLibAsmEnablePaging64,
  UnitTestHostBaseLibAsmDisablePaging64,
  UnitTestHostBaseLibAsmGetThunk16Properties,
  UnitTestHostBaseLibAsmPrepareThunk16,
  UnitTestHostBaseLibAsmThunk16,
  UnitTestHostBaseLibAsmPrepareAndThunk16,
  UnitTestHostBaseLibAsmWriteTr,
  UnitTestHostBaseLibAsmLfence,
  UnitTestHostBaseLibPatchInstructionX86
};
 
UNIT_TEST_HOST_BASE_LIB  gUnitTestHostBaseLib = {
  &mUnitTestHostBaseLibCommon,
  &mUnitTestHostBaseLibX86
};