MdeModulePkg/Universal/Acpi/BootScriptExecutorDxe: Add support for PCD PcdPteMemoryEn...

[mirror_edk2.git] / MdeModulePkg / Universal / Acpi / BootScriptExecutorDxe / X64 / S3Asm.asm

;; @file\r
;   This is the assembly code for transferring to control to OS S3 waking vector\r
;   for X64 platform\r
;\r
; Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>\r
;\r
; This program and the accompanying materials\r
; are licensed and made available under the terms and conditions of the BSD License\r
; which accompanies this distribution.  The full text of the license may be found at\r
; http://opensource.org/licenses/bsd-license.php\r
;\r
; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
;\r
;;\r
\r
EXTERN mOriginalHandler:QWORD\r
EXTERN PageFaultHandler:PROC\r
\r
    .code\r
    \r
PUBLIC   AsmFixAddress16\r
PUBLIC   AsmJmpAddr32\r
    \r
AsmTransferControl  PROC\r
    ; rcx S3WakingVector    :DWORD\r
    ; rdx AcpiLowMemoryBase :DWORD\r
    lea   eax, @F\r
    mov   r8, 2800000000h\r
    or    rax, r8\r
    push  rax\r
    shrd  ebx, ecx, 20\r
    and   ecx, 0fh          \r
    mov   bx, cx          \r
    mov   [@jmp_addr], ebx\r
    retf\r
@@:\r
    DB    0b8h, 30h, 0      ; mov ax, 30h as selector\r
    mov   ds, eax\r
    mov   es, eax\r
    mov   fs, eax\r
    mov   gs, eax\r
    mov   ss, eax\r
    mov   rax, cr0\r
    mov   rbx, cr4        \r
    DB    66h\r
    and   eax, ((NOT 080000001h) AND 0ffffffffh)\r
    and   bl, NOT (1 SHL 5)\r
    mov   cr0, rax\r
    DB    66h\r
    mov   ecx, 0c0000080h\r
    rdmsr\r
    and   ah, NOT 1\r
    wrmsr\r
    mov   cr4, rbx\r
    DB    0eah              ; jmp far @jmp_addr\r
@jmp_addr DD  ?\r
AsmTransferControl  ENDP\r
\r
AsmTransferControl32  PROC\r
    ; S3WakingVector    :DWORD\r
    ; AcpiLowMemoryBase :DWORD\r
    push  rbp\r
    mov   ebp, esp    \r
    DB    8dh, 05h          ;  lea   eax, AsmTransferControl16\r
AsmFixAddress16  DD ?\r
    push  28h               ; CS\r
    push  rax\r
    retf\r
AsmTransferControl32  ENDP\r
\r
AsmTransferControl16  PROC\r
    DB    0b8h, 30h, 0      ; mov ax, 30h as selector\r
    mov   ds, ax\r
    mov   es, ax\r
    mov   fs, ax\r
    mov   gs, ax\r
    mov   ss, ax\r
    mov   rax, cr0          ; Get control register 0  \r
    DB    66h\r
    DB    83h, 0e0h, 0feh   ; and    eax, 0fffffffeh  ; Clear PE bit (bit #0)\r
    DB    0fh, 22h, 0c0h    ; mov    cr0, eax         ; Activate real mode\r
    DB    0eah              ; jmp far AsmJmpAddr32\r
AsmJmpAddr32 DD  ?\r
AsmTransferControl16  ENDP\r
\r
PageFaultHandlerHook PROC\r
    push    rax                         ; save all volatile registers\r
    push    rcx\r
    push    rdx\r
    push    r8\r
    push    r9\r
    push    r10\r
    push    r11\r
    ; save volatile fp registers\r
    add     rsp, -68h\r
    stmxcsr [rsp + 60h]\r
    movdqa  [rsp + 0h], xmm0\r
    movdqa  [rsp + 10h], xmm1\r
    movdqa  [rsp + 20h], xmm2\r
    movdqa  [rsp + 30h], xmm3\r
    movdqa  [rsp + 40h], xmm4\r
    movdqa  [rsp + 50h], xmm5\r
\r
    add     rsp, -20h\r
    call    PageFaultHandler\r
    add     rsp, 20h\r
    \r
    ; load volatile fp registers\r
    ldmxcsr [rsp + 60h]\r
    movdqa  xmm0,  [rsp + 0h]\r
    movdqa  xmm1,  [rsp + 10h]\r
    movdqa  xmm2,  [rsp + 20h]\r
    movdqa  xmm3,  [rsp + 30h]\r
    movdqa  xmm4,  [rsp + 40h]\r
    movdqa  xmm5,  [rsp + 50h]\r
    add     rsp, 68h\r
\r
    test    al, al\r
    \r
    pop     r11\r
    pop     r10\r
    pop     r9\r
    pop     r8\r
    pop     rdx\r
    pop     rcx\r
    pop     rax                         ; restore all volatile registers\r
    jnz     @F\r
    jmp     mOriginalHandler\r
@@:\r
    add     rsp, 08h                    ; skip error code for PF\r
    iretq\r
PageFaultHandlerHook ENDP\r
\r
    END\r