mirror of
https://github.com/acidanthera/audk.git
synced 2025-07-25 22:54:51 +02:00
UefiCpuPkg: Move AsmRelocateApLoopStart from Mpfuncs.nasm to AmdSev.nasm
Cc: Guo Dong <guo.dong@intel.com> Cc: Ray Ni <ray.ni@intel.com> Cc: Sean Rhodes <sean@starlabs.systems> Cc: James Lu <james.lu@intel.com> Cc: Gua Guo <gua.guo@intel.com> Signed-off-by: Yuanhao Xie <yuanhao.xie@intel.com> Acked-by: Gerd Hoffmann <kraxel@redhat.com> Tested-by: Gerd Hoffmann <kraxel@redhat.com> Reviewed-by: Ray Ni <ray.ni@intel.com>
This commit is contained in:
Xie, Yuanhao
mergify[bot]
parent
0d1ad06c27
commit
c7c2599759
@ -346,3 +346,173 @@ PM16Mode:
|
|||||||
iret
|
iret
|
||||||
|
|
||||||
SwitchToRealProcEnd:
|
SwitchToRealProcEnd:
|
||||||
|
|
||||||
|
;-------------------------------------------------------------------------------------
|
||||||
|
; AsmRelocateApLoopAmdSev (MwaitSupport, ApTargetCState, PmCodeSegment, TopOfApStack, CountTofinish, Pm16CodeSegment, SevEsAPJumpTable, WakeupBuffer);
|
||||||
|
;-------------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
AsmRelocateApLoopAmdSevStart:
|
||||||
|
BITS 64
|
||||||
|
cmp qword [rsp + 56], 0 ; SevEsAPJumpTable
|
||||||
|
je NoSevEsAmdSev
|
||||||
|
|
||||||
|
;
|
||||||
|
; Perform some SEV-ES related setup before leaving 64-bit mode
|
||||||
|
;
|
||||||
|
push rcx
|
||||||
|
push rdx
|
||||||
|
|
||||||
|
;
|
||||||
|
; Get the RDX reset value using CPUID
|
||||||
|
;
|
||||||
|
mov rax, 1
|
||||||
|
cpuid
|
||||||
|
mov rsi, rax ; Save off the reset value for RDX
|
||||||
|
|
||||||
|
;
|
||||||
|
; Prepare the GHCB for the AP_HLT_LOOP VMGEXIT call
|
||||||
|
; - Must be done while in 64-bit long mode so that writes to
|
||||||
|
; the GHCB memory will be unencrypted.
|
||||||
|
; - No NAE events can be generated once this is set otherwise
|
||||||
|
; the AP_RESET_HOLD SW_EXITCODE will be overwritten.
|
||||||
|
;
|
||||||
|
mov rcx, 0xc0010130
|
||||||
|
rdmsr ; Retrieve current GHCB address
|
||||||
|
shl rdx, 32
|
||||||
|
or rdx, rax
|
||||||
|
|
||||||
|
mov rdi, rdx
|
||||||
|
xor rax, rax
|
||||||
|
mov rcx, 0x800
|
||||||
|
shr rcx, 3
|
||||||
|
rep stosq ; Clear the GHCB
|
||||||
|
|
||||||
|
mov rax, 0x80000004 ; VMGEXIT AP_RESET_HOLD
|
||||||
|
mov [rdx + 0x390], rax
|
||||||
|
mov rax, 114 ; Set SwExitCode valid bit
|
||||||
|
bts [rdx + 0x3f0], rax
|
||||||
|
inc rax ; Set SwExitInfo1 valid bit
|
||||||
|
bts [rdx + 0x3f0], rax
|
||||||
|
inc rax ; Set SwExitInfo2 valid bit
|
||||||
|
bts [rdx + 0x3f0], rax
|
||||||
|
|
||||||
|
pop rdx
|
||||||
|
pop rcx
|
||||||
|
|
||||||
|
NoSevEsAmdSev:
|
||||||
|
cli ; Disable interrupt before switching to 32-bit mode
|
||||||
|
mov rax, [rsp + 40] ; CountTofinish
|
||||||
|
lock dec dword [rax] ; (*CountTofinish)--
|
||||||
|
|
||||||
|
mov r10, [rsp + 48] ; Pm16CodeSegment
|
||||||
|
mov rax, [rsp + 56] ; SevEsAPJumpTable
|
||||||
|
mov rbx, [rsp + 64] ; WakeupBuffer
|
||||||
|
mov rsp, r9 ; TopOfApStack
|
||||||
|
|
||||||
|
push rax ; Save SevEsAPJumpTable
|
||||||
|
push rbx ; Save WakeupBuffer
|
||||||
|
push r10 ; Save Pm16CodeSegment
|
||||||
|
push rcx ; Save MwaitSupport
|
||||||
|
push rdx ; Save ApTargetCState
|
||||||
|
|
||||||
|
lea rax, [PmEntryAmdSev] ; rax <- The start address of transition code
|
||||||
|
|
||||||
|
push r8
|
||||||
|
push rax
|
||||||
|
|
||||||
|
;
|
||||||
|
; Clear R8 - R15, for reset, before going into 32-bit mode
|
||||||
|
;
|
||||||
|
xor r8, r8
|
||||||
|
xor r9, r9
|
||||||
|
xor r10, r10
|
||||||
|
xor r11, r11
|
||||||
|
xor r12, r12
|
||||||
|
xor r13, r13
|
||||||
|
xor r14, r14
|
||||||
|
xor r15, r15
|
||||||
|
|
||||||
|
;
|
||||||
|
; Far return into 32-bit mode
|
||||||
|
;
|
||||||
|
o64 retf
|
||||||
|
|
||||||
|
BITS 32
|
||||||
|
PmEntryAmdSev:
|
||||||
|
mov eax, cr0
|
||||||
|
btr eax, 31 ; Clear CR0.PG
|
||||||
|
mov cr0, eax ; Disable paging and caches
|
||||||
|
|
||||||
|
mov ecx, 0xc0000080
|
||||||
|
rdmsr
|
||||||
|
and ah, ~ 1 ; Clear LME
|
||||||
|
wrmsr
|
||||||
|
mov eax, cr4
|
||||||
|
and al, ~ (1 << 5) ; Clear PAE
|
||||||
|
mov cr4, eax
|
||||||
|
|
||||||
|
pop edx
|
||||||
|
add esp, 4
|
||||||
|
pop ecx,
|
||||||
|
add esp, 4
|
||||||
|
|
||||||
|
MwaitCheckAmdSev:
|
||||||
|
cmp cl, 1 ; Check mwait-monitor support
|
||||||
|
jnz HltLoopAmdSev
|
||||||
|
mov ebx, edx ; Save C-State to ebx
|
||||||
|
MwaitLoopAmdSev:
|
||||||
|
cli
|
||||||
|
mov eax, esp ; Set Monitor Address
|
||||||
|
xor ecx, ecx ; ecx = 0
|
||||||
|
xor edx, edx ; edx = 0
|
||||||
|
monitor
|
||||||
|
mov eax, ebx ; Mwait Cx, Target C-State per eax[7:4]
|
||||||
|
shl eax, 4
|
||||||
|
mwait
|
||||||
|
jmp MwaitLoopAmdSev
|
||||||
|
|
||||||
|
HltLoopAmdSev:
|
||||||
|
pop edx ; PM16CodeSegment
|
||||||
|
add esp, 4
|
||||||
|
pop ebx ; WakeupBuffer
|
||||||
|
add esp, 4
|
||||||
|
pop eax ; SevEsAPJumpTable
|
||||||
|
add esp, 4
|
||||||
|
cmp eax, 0 ; Check for SEV-ES
|
||||||
|
je DoHltAmdSev
|
||||||
|
|
||||||
|
cli
|
||||||
|
;
|
||||||
|
; SEV-ES is enabled, use VMGEXIT (GHCB information already
|
||||||
|
; set by caller)
|
||||||
|
;
|
||||||
|
BITS 64
|
||||||
|
rep vmmcall
|
||||||
|
BITS 32
|
||||||
|
|
||||||
|
;
|
||||||
|
; Back from VMGEXIT AP_HLT_LOOP
|
||||||
|
; Push the FLAGS/CS/IP values to use
|
||||||
|
;
|
||||||
|
push word 0x0002 ; EFLAGS
|
||||||
|
xor ecx, ecx
|
||||||
|
mov cx, [eax + 2] ; CS
|
||||||
|
push cx
|
||||||
|
mov cx, [eax] ; IP
|
||||||
|
push cx
|
||||||
|
push word 0x0000 ; For alignment, will be discarded
|
||||||
|
|
||||||
|
push edx
|
||||||
|
push ebx
|
||||||
|
|
||||||
|
mov edx, esi ; Restore RDX reset value
|
||||||
|
|
||||||
|
retf
|
||||||
|
|
||||||
|
DoHltAmdSev:
|
||||||
|
cli
|
||||||
|
hlt
|
||||||
|
jmp DoHltAmdSev
|
||||||
|
|
||||||
|
BITS 64
|
||||||
|
AsmRelocateApLoopAmdSevEnd:
|
||||||
|
|||||||
@ -277,175 +277,6 @@ CProcedureInvoke:
|
|||||||
%include "AmdSev.nasm"
|
%include "AmdSev.nasm"
|
||||||
|
|
||||||
RendezvousFunnelProcEnd:
|
RendezvousFunnelProcEnd:
|
||||||
;-------------------------------------------------------------------------------------
|
|
||||||
; AsmRelocateApLoopAmdSev (MwaitSupport, ApTargetCState, PmCodeSegment, TopOfApStack, CountTofinish, Pm16CodeSegment, SevEsAPJumpTable, WakeupBuffer);
|
|
||||||
;-------------------------------------------------------------------------------------
|
|
||||||
|
|
||||||
AsmRelocateApLoopAmdSevStart:
|
|
||||||
BITS 64
|
|
||||||
cmp qword [rsp + 56], 0 ; SevEsAPJumpTable
|
|
||||||
je NoSevEsAmdSev
|
|
||||||
|
|
||||||
;
|
|
||||||
; Perform some SEV-ES related setup before leaving 64-bit mode
|
|
||||||
;
|
|
||||||
push rcx
|
|
||||||
push rdx
|
|
||||||
|
|
||||||
;
|
|
||||||
; Get the RDX reset value using CPUID
|
|
||||||
;
|
|
||||||
mov rax, 1
|
|
||||||
cpuid
|
|
||||||
mov rsi, rax ; Save off the reset value for RDX
|
|
||||||
|
|
||||||
;
|
|
||||||
; Prepare the GHCB for the AP_HLT_LOOP VMGEXIT call
|
|
||||||
; - Must be done while in 64-bit long mode so that writes to
|
|
||||||
; the GHCB memory will be unencrypted.
|
|
||||||
; - No NAE events can be generated once this is set otherwise
|
|
||||||
; the AP_RESET_HOLD SW_EXITCODE will be overwritten.
|
|
||||||
;
|
|
||||||
mov rcx, 0xc0010130
|
|
||||||
rdmsr ; Retrieve current GHCB address
|
|
||||||
shl rdx, 32
|
|
||||||
or rdx, rax
|
|
||||||
|
|
||||||
mov rdi, rdx
|
|
||||||
xor rax, rax
|
|
||||||
mov rcx, 0x800
|
|
||||||
shr rcx, 3
|
|
||||||
rep stosq ; Clear the GHCB
|
|
||||||
|
|
||||||
mov rax, 0x80000004 ; VMGEXIT AP_RESET_HOLD
|
|
||||||
mov [rdx + 0x390], rax
|
|
||||||
mov rax, 114 ; Set SwExitCode valid bit
|
|
||||||
bts [rdx + 0x3f0], rax
|
|
||||||
inc rax ; Set SwExitInfo1 valid bit
|
|
||||||
bts [rdx + 0x3f0], rax
|
|
||||||
inc rax ; Set SwExitInfo2 valid bit
|
|
||||||
bts [rdx + 0x3f0], rax
|
|
||||||
|
|
||||||
pop rdx
|
|
||||||
pop rcx
|
|
||||||
|
|
||||||
NoSevEsAmdSev:
|
|
||||||
cli ; Disable interrupt before switching to 32-bit mode
|
|
||||||
mov rax, [rsp + 40] ; CountTofinish
|
|
||||||
lock dec dword [rax] ; (*CountTofinish)--
|
|
||||||
|
|
||||||
mov r10, [rsp + 48] ; Pm16CodeSegment
|
|
||||||
mov rax, [rsp + 56] ; SevEsAPJumpTable
|
|
||||||
mov rbx, [rsp + 64] ; WakeupBuffer
|
|
||||||
mov rsp, r9 ; TopOfApStack
|
|
||||||
|
|
||||||
push rax ; Save SevEsAPJumpTable
|
|
||||||
push rbx ; Save WakeupBuffer
|
|
||||||
push r10 ; Save Pm16CodeSegment
|
|
||||||
push rcx ; Save MwaitSupport
|
|
||||||
push rdx ; Save ApTargetCState
|
|
||||||
|
|
||||||
lea rax, [PmEntryAmdSev] ; rax <- The start address of transition code
|
|
||||||
|
|
||||||
push r8
|
|
||||||
push rax
|
|
||||||
|
|
||||||
;
|
|
||||||
; Clear R8 - R15, for reset, before going into 32-bit mode
|
|
||||||
;
|
|
||||||
xor r8, r8
|
|
||||||
xor r9, r9
|
|
||||||
xor r10, r10
|
|
||||||
xor r11, r11
|
|
||||||
xor r12, r12
|
|
||||||
xor r13, r13
|
|
||||||
xor r14, r14
|
|
||||||
xor r15, r15
|
|
||||||
|
|
||||||
;
|
|
||||||
; Far return into 32-bit mode
|
|
||||||
;
|
|
||||||
o64 retf
|
|
||||||
|
|
||||||
BITS 32
|
|
||||||
PmEntryAmdSev:
|
|
||||||
mov eax, cr0
|
|
||||||
btr eax, 31 ; Clear CR0.PG
|
|
||||||
mov cr0, eax ; Disable paging and caches
|
|
||||||
|
|
||||||
mov ecx, 0xc0000080
|
|
||||||
rdmsr
|
|
||||||
and ah, ~ 1 ; Clear LME
|
|
||||||
wrmsr
|
|
||||||
mov eax, cr4
|
|
||||||
and al, ~ (1 << 5) ; Clear PAE
|
|
||||||
mov cr4, eax
|
|
||||||
|
|
||||||
pop edx
|
|
||||||
add esp, 4
|
|
||||||
pop ecx,
|
|
||||||
add esp, 4
|
|
||||||
|
|
||||||
MwaitCheckAmdSev:
|
|
||||||
cmp cl, 1 ; Check mwait-monitor support
|
|
||||||
jnz HltLoopAmdSev
|
|
||||||
mov ebx, edx ; Save C-State to ebx
|
|
||||||
MwaitLoopAmdSev:
|
|
||||||
cli
|
|
||||||
mov eax, esp ; Set Monitor Address
|
|
||||||
xor ecx, ecx ; ecx = 0
|
|
||||||
xor edx, edx ; edx = 0
|
|
||||||
monitor
|
|
||||||
mov eax, ebx ; Mwait Cx, Target C-State per eax[7:4]
|
|
||||||
shl eax, 4
|
|
||||||
mwait
|
|
||||||
jmp MwaitLoopAmdSev
|
|
||||||
|
|
||||||
HltLoopAmdSev:
|
|
||||||
pop edx ; PM16CodeSegment
|
|
||||||
add esp, 4
|
|
||||||
pop ebx ; WakeupBuffer
|
|
||||||
add esp, 4
|
|
||||||
pop eax ; SevEsAPJumpTable
|
|
||||||
add esp, 4
|
|
||||||
cmp eax, 0 ; Check for SEV-ES
|
|
||||||
je DoHltAmdSev
|
|
||||||
|
|
||||||
cli
|
|
||||||
;
|
|
||||||
; SEV-ES is enabled, use VMGEXIT (GHCB information already
|
|
||||||
; set by caller)
|
|
||||||
;
|
|
||||||
BITS 64
|
|
||||||
rep vmmcall
|
|
||||||
BITS 32
|
|
||||||
|
|
||||||
;
|
|
||||||
; Back from VMGEXIT AP_HLT_LOOP
|
|
||||||
; Push the FLAGS/CS/IP values to use
|
|
||||||
;
|
|
||||||
push word 0x0002 ; EFLAGS
|
|
||||||
xor ecx, ecx
|
|
||||||
mov cx, [eax + 2] ; CS
|
|
||||||
push cx
|
|
||||||
mov cx, [eax] ; IP
|
|
||||||
push cx
|
|
||||||
push word 0x0000 ; For alignment, will be discarded
|
|
||||||
|
|
||||||
push edx
|
|
||||||
push ebx
|
|
||||||
|
|
||||||
mov edx, esi ; Restore RDX reset value
|
|
||||||
|
|
||||||
retf
|
|
||||||
|
|
||||||
DoHltAmdSev:
|
|
||||||
cli
|
|
||||||
hlt
|
|
||||||
jmp DoHltAmdSev
|
|
||||||
|
|
||||||
BITS 64
|
|
||||||
AsmRelocateApLoopAmdSevEnd:
|
|
||||||
|
|
||||||
;-------------------------------------------------------------------------------------
|
;-------------------------------------------------------------------------------------
|
||||||
; AsmRelocateApLoop (MwaitSupport, ApTargetCState, TopOfApStack, CountTofinish, Cr3);
|
; AsmRelocateApLoop (MwaitSupport, ApTargetCState, TopOfApStack, CountTofinish, Cr3);
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user