[mirror_edk2.git] / MdePkg / Library / BaseLib / X64 / Thunk16.asm

;------------------------------------------------------------------------------\r
;\r
; Copyright (c) 2006, Intel Corporation\r
; All rights reserved. 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
; Module Name:\r
;\r
;   Thunk.asm\r
;\r
; Abstract:\r
;\r
;   Real mode thunk\r
;\r
;------------------------------------------------------------------------------\r
\r
EXTERNDEF   m16Start:BYTE\r
EXTERNDEF   m16Size:WORD\r
EXTERNDEF   mThunk16Attr:WORD\r
EXTERNDEF   m16Gdt:WORD\r
EXTERNDEF   m16GdtrBase:WORD\r
EXTERNDEF   mTransition:WORD\r
\r
IA32_REGS   STRUC   4t\r
_EDI        DD      ?\r
_ESI        DD      ?\r
_EBP        DD      ?\r
_ESP        DD      ?\r
_EBX        DD      ?\r
_EDX        DD      ?\r
_ECX        DD      ?\r
_EAX        DD      ?\r
_DS         DW      ?\r
_ES         DW      ?\r
_FS         DW      ?\r
_GS         DW      ?\r
_EFLAGS     DQ      ?\r
_EIP        DD      ?\r
_CS         DW      ?\r
_SS         DW      ?\r
IA32_REGS   ENDS\r
\r
    .const\r
\r
m16Size         DW      InternalAsmThunk16 - m16Start\r
mThunk16Attr    DW      _ThunkAttr - m16Start\r
m16Gdt          DW      _NullSeg - m16Start\r
m16GdtrBase     DW      _16GdtrBase - m16Start\r
mTransition     DW      _EntryPoint - m16Start\r
\r
    .code\r
\r
m16Start    LABEL   BYTE\r
\r
SavedGdt    LABEL   FWORD\r
            DW      ?\r
            DQ      ?\r
\r
;------------------------------------------------------------------------------\r
; _BackFromUserCode() takes control in real mode after 'retf' has been executed\r
; by user code. It will be shadowed to somewhere in memory below 1MB.\r
;------------------------------------------------------------------------------\r
_BackFromUserCode   PROC\r
    DB      16h                         ; push ss\r
    DB      0eh                         ; push cs\r
    DB      66h\r
    call    @Base                       ; push eip\r
@Base:\r
    DB      66h\r
    push    0                           ; reserved high order 32 bits of EFlags\r
    pushf                               ; pushfd actually\r
    cli                                 ; disable interrupts\r
    push    gs\r
    push    fs\r
    DB      6                           ; push es\r
    DB      1eh                         ; push ds\r
    DB      66h, 60h                    ; pushad\r
    DB      66h, 0bah                   ; mov edx, imm32\r
_ThunkAttr  DD      ?\r
    test    dl, THUNK_ATTRIBUTE_DISABLE_A20_MASK_INT_15\r
    jz      @1\r
    mov     eax, 15cd2401h              ; mov ax, 2401h & int 15h\r
    cli                                 ; disable interrupts\r
    jnc     @2\r
@1:\r
    test    dl, THUNK_ATTRIBUTE_DISABLE_A20_MASK_KBD_CTRL\r
    jz      @2\r
    in      al, 92h\r
    or      al, 2\r
    out     92h, al                     ; deactivate A20M#\r
@2:\r
    mov     eax, ss\r
    lea     bp, [esp + sizeof (IA32_REGS)]\r
    mov     word ptr (IA32_REGS ptr [rsi - sizeof (IA32_REGS)])._ESP, bp\r
    mov     ebx, (IA32_REGS ptr [rsi - sizeof (IA32_REGS)])._EIP\r
    shl     ax, 4                       ; shl eax, 4\r
    add     bp, ax                      ; add ebp, eax\r
    mov     ax, cs\r
    shl     ax, 4\r
    lea     ax, [eax + ebx + (@64BitCode - @Base)]\r
    DB      2eh                         ; cs:\r
    mov     [rdi + (@64Eip - @Base)], ax\r
    DB      66h, 0b8h                   ; mov eax, imm32\r
SavedCr4    DD      ?\r
    mov     cr4, rax\r
    DB      66h, 2eh\r
    lgdt    fword ptr [rdi + (SavedGdt - @Base)]\r
    DB      66h\r
    mov     ecx, 0c0000080h\r
    rdmsr\r
    or      ah, 1\r
    wrmsr\r
    DB      66h, 0b8h                   ; mov eax, imm32\r
SavedCr0    DD      ?\r
    mov     cr0, rax\r
    DB      66h, 0eah                   ; jmp far cs:@64Bit\r
@64Eip      DD      ?\r
SavedCs     DW      ?\r
@64BitCode:\r
    DB      48h, 0b8h                   ; mov rax, imm64\r
SavedRip    DQ      ?\r
    jmp     rax                         ; return to caller\r
_BackFromUserCode   ENDP\r
\r
_EntryPoint DD      _ToUserCode - m16Start\r
            DW      CODE16\r
_16Gdtr     LABEL   FWORD\r
            DW      GDT_SIZE - 1\r
_16GdtrBase DQ      _NullSeg\r
_16Idtr     FWORD   (1 SHL 10) - 1\r
\r
;------------------------------------------------------------------------------\r
; _ToUserCode() takes control in real mode before passing control to user code.\r
; It will be shadowed to somewhere in memory below 1MB.\r
;------------------------------------------------------------------------------\r
_ToUserCode PROC\r
    mov     ss, edx                     ; set new segment selectors\r
    mov     ds, edx\r
    mov     es, edx\r
    mov     fs, edx\r
    mov     gs, edx\r
    DB      66h\r
    mov     ecx, 0c0000080h\r
    mov     cr0, rax                    ; real mode starts at next instruction\r
    rdmsr\r
    and     ah, NOT 1\r
    wrmsr\r
    mov     cr4, rbp\r
    mov     ss, esi                     ; set up 16-bit stack segment\r
    mov     sp, bx                      ; set up 16-bit stack pointer\r
    DB      66h\r
    call    @Base                       ; push eip\r
@Base:\r
    pop     bp                          ; ebp <- address of @Base\r
    push    [esp + sizeof (IA32_REGS) + 2]\r
    lea     eax, [rsi + (@RealMode - @Base)]\r
    push    rax\r
    retf\r
@RealMode:\r
    DB      66h, 2eh                    ; CS and operand size override\r
    lidt    fword ptr [rsi + (_16Idtr - @Base)]\r
    DB      66h, 61h                    ; popad\r
    DB      1fh                         ; pop ds\r
    DB      07h                         ; pop es\r
    pop     fs\r
    pop     gs\r
    popf                                ; popfd\r
    lea     sp, [esp + 4]               ; skip high order 32 bits of EFlags\r
    DB      66h\r
    retf                                ; transfer control to user code\r
_ToUserCode ENDP\r
\r
CODE16  = _16Code - $\r
DATA16  = _16Data - $\r
DATA32  = _32Data - $\r
\r
_NullSeg    DQ      0\r
_16Code     LABEL   QWORD\r
            DW      -1\r
            DW      0\r
            DB      0\r
            DB      9bh\r
            DB      8fh                 ; 16-bit segment, 4GB limit\r
            DB      0\r
_16Data     LABEL   QWORD\r
            DW      -1\r
            DW      0\r
            DB      0\r
            DB      93h\r
            DB      8fh                 ; 16-bit segment, 4GB limit\r
            DB      0\r
_32Data     LABEL   QWORD\r
            DW      -1\r
            DW      0\r
            DB      0\r
            DB      93h\r
            DB      0cfh                ; 16-bit segment, 4GB limit\r
            DB      0\r
\r
GDT_SIZE = $ - _NullSeg\r
\r
;------------------------------------------------------------------------------\r
; IA32_REGISTER_SET *\r
; EFIAPI\r
; InternalAsmThunk16 (\r
;   IN      IA32_REGISTER_SET         *RegisterSet,\r
;   IN OUT  VOID                      *Transition\r
;   );\r
;------------------------------------------------------------------------------\r
InternalAsmThunk16  PROC    USES    rbp rbx rsi rdi\r
    mov     r10d, ds\r
    mov     r11d, es\r
    mov     r9d, ss\r
    push    fs\r
    push    gs\r
    mov     rsi, rcx\r
    movzx   r8d, (IA32_REGS ptr [rsi])._SS\r
    mov     edi, (IA32_REGS ptr [rsi])._ESP\r
    lea     rdi, [edi - (sizeof (IA32_REGS) + 4)]\r
    imul    eax, r8d, 16                ; eax <- r8d(stack segment) * 16\r
    mov     ebx, edi                    ; ebx <- stack for 16-bit code\r
    push    sizeof (IA32_REGS) / 4\r
    add     edi, eax                    ; edi <- linear address of 16-bit stack\r
    pop     rcx\r
    rep     movsd                       ; copy RegSet\r
    lea     ecx, [rdx + (SavedCr4 - m16Start)]\r
    mov     eax, edx                    ; eax <- transition code address\r
    and     edx, 0fh\r
    shl     eax, 12\r
    lea     ax, [rdx + (_BackFromUserCode - m16Start)]\r
    stosd                               ; [edi] <- return address of user code\r
    sgdt    fword ptr [rcx + (SavedGdt - SavedCr4)]\r
    sidt    fword ptr [rsp + 38h]       ; save IDT stack in argument space\r
    mov     rax, cr0\r
    mov     [rcx + (SavedCr0 - SavedCr4)], eax\r
    and     eax, 7ffffffeh              ; clear PE, PG bits\r
    mov     rbp, cr4\r
    mov     [rcx], ebp                  ; save CR4 in SavedCr4\r
    and     ebp, 300h                   ; clear all but PCE and OSFXSR bits\r
    mov     esi, r8d                    ; esi <- 16-bit stack segment\r
    DB      6ah, DATA32                 ; push DATA32\r
    pop     rdx                         ; rdx <- 32-bit data segment selector\r
    lgdt    fword ptr [rcx + (_16Gdtr - SavedCr4)]\r
    mov     ss, edx\r
    pushfq\r
    lea     edx, [rdx + DATA16 - DATA32]\r
    lea     r8, @RetFromRealMode\r
    mov     [rcx + (SavedRip - SavedCr4)], r8\r
    mov     r8d, cs\r
    mov     [rcx + (SavedCs - SavedCr4)], r8w\r
    mov     r8, rsp\r
    jmp     fword ptr [rcx + (_EntryPoint - SavedCr4)]\r
@RetFromRealMode:\r
    mov     rsp, r8\r
    popfq\r
    lidt    fword ptr [rsp + 38h]       ; restore protected mode IDTR\r
    lea     eax, [rbp - sizeof (IA32_REGS)]\r
    pop     gs\r
    pop     fs\r
    mov     ss, r9d\r
    mov     es, r11d\r
    mov     ds, r10d\r
    ret\r
InternalAsmThunk16  ENDP\r
\r
    END\r
Commit	Line	Data
878ddf1f	1	;------------------------------------------------------------------------------\r
	2	;\r
	3	; Copyright (c) 2006, Intel Corporation\r
	4	; All rights reserved. This program and the accompanying materials\r
	5	; are licensed and made available under the terms and conditions of the BSD License\r
	6	; which accompanies this distribution. The full text of the license may be found at\r
	7	; http://opensource.org/licenses/bsd-license.php\r
	8	;\r
	9	; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	10	; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	11	;\r
	12	; Module Name:\r
	13	;\r
	14	; Thunk.asm\r
	15	;\r
	16	; Abstract:\r
	17	;\r
	18	; Real mode thunk\r
	19	;\r
	20	;------------------------------------------------------------------------------\r
	21	\r
97d92bda	22	EXTERNDEF m16Start:BYTE\r
	23	EXTERNDEF m16Size:WORD\r
	24	EXTERNDEF mThunk16Attr:WORD\r
	25	EXTERNDEF m16Gdt:WORD\r
	26	EXTERNDEF m16GdtrBase:WORD\r
	27	EXTERNDEF mTransition:WORD\r
878ddf1f	28	\r
878ddf1f	29	IA32_REGS STRUC 4t\r
	30	_EDI DD ?\r
	31	_ESI DD ?\r
	32	_EBP DD ?\r
	33	_ESP DD ?\r
	34	_EBX DD ?\r
	35	_EDX DD ?\r
	36	_ECX DD ?\r
	37	_EAX DD ?\r
	38	_DS DW ?\r
	39	_ES DW ?\r
	40	_FS DW ?\r
	41	_GS DW ?\r
97d92bda	42	_EFLAGS DQ ?\r
878ddf1f	43	_EIP DD ?\r
	44	_CS DW ?\r
	45	_SS DW ?\r
	46	IA32_REGS ENDS\r
	47	\r
97d92bda	48	.const\r
97d92bda	49	\r
bbe4ba81	50	m16Size DW InternalAsmThunk16 - m16Start\r
bbe4ba81	51	mThunk16Attr DW _ThunkAttr - m16Start\r
f23d790a	52	m16Gdt DW _NullSeg - m16Start\r
bbe4ba81	53	m16GdtrBase DW _16GdtrBase - m16Start\r
bbe4ba81	54	mTransition DW _EntryPoint - m16Start\r
97d92bda	55	\r
	56	.code\r
	57	\r
	58	m16Start LABEL BYTE\r
	59	\r
bbe4ba81	60	SavedGdt LABEL FWORD\r
	61	DW ?\r
	62	DQ ?\r
97d92bda	63	\r
3f566587	64	;------------------------------------------------------------------------------\r
	65	; _BackFromUserCode() takes control in real mode after 'retf' has been executed\r
	66	; by user code. It will be shadowed to somewhere in memory below 1MB.\r
	67	;------------------------------------------------------------------------------\r
97d92bda	68	_BackFromUserCode PROC\r
	69	DB 16h ; push ss\r
	70	DB 0eh ; push cs\r
878ddf1f	71	DB 66h\r
97d92bda	72	call @Base ; push eip\r
97d92bda	73	@Base:\r
878ddf1f	74	DB 66h\r
97d92bda	75	push 0 ; reserved high order 32 bits of EFlags\r
878ddf1f	76	pushf ; pushfd actually\r
97d92bda	77	cli ; disable interrupts\r
878ddf1f	78	push gs\r
	79	push fs\r
	80	DB 6 ; push es\r
	81	DB 1eh ; push ds\r
	82	DB 66h, 60h ; pushad\r
97d92bda	83	DB 66h, 0bah ; mov edx, imm32\r
	84	_ThunkAttr DD ?\r
	85	test dl, THUNK_ATTRIBUTE_DISABLE_A20_MASK_INT_15\r
	86	jz @1\r
	87	mov eax, 15cd2401h ; mov ax, 2401h & int 15h\r
	88	cli ; disable interrupts\r
	89	jnc @2\r
	90	@1:\r
	91	test dl, THUNK_ATTRIBUTE_DISABLE_A20_MASK_KBD_CTRL\r
	92	jz @2\r
	93	in al, 92h\r
	94	or al, 2\r
	95	out 92h, al ; deactivate A20M#\r
	96	@2:\r
	97	mov eax, ss\r
	98	lea bp, [esp + sizeof (IA32_REGS)]\r
	99	mov word ptr (IA32_REGS ptr [rsi - sizeof (IA32_REGS)])._ESP, bp\r
	100	mov ebx, (IA32_REGS ptr [rsi - sizeof (IA32_REGS)])._EIP\r
	101	shl ax, 4 ; shl eax, 4\r
	102	add bp, ax ; add ebp, eax\r
38b5365e	103	mov ax, cs\r
	104	shl ax, 4\r
	105	lea ax, [eax + ebx + (@64BitCode - @Base)]\r
	106	DB 2eh ; cs:\r
	107	mov [rdi + (@64Eip - @Base)], ax\r
97d92bda	108	DB 66h, 0b8h ; mov eax, imm32\r
	109	SavedCr4 DD ?\r
	110	mov cr4, rax\r
	111	DB 66h, 2eh\r
bbe4ba81	112	lgdt fword ptr [rdi + (SavedGdt - @Base)]\r
878ddf1f	113	DB 66h\r
	114	mov ecx, 0c0000080h\r
	115	rdmsr\r
97d92bda	116	or ah, 1\r
878ddf1f	117	wrmsr\r
97d92bda	118	DB 66h, 0b8h ; mov eax, imm32\r
	119	SavedCr0 DD ?\r
	120	mov cr0, rax\r
38b5365e	121	DB 66h, 0eah ; jmp far cs:@64Bit\r
	122	@64Eip DD ?\r
	123	SavedCs DW ?\r
	124	@64BitCode:\r
	125	DB 48h, 0b8h ; mov rax, imm64\r
	126	SavedRip DQ ?\r
	127	jmp rax ; return to caller\r
97d92bda	128	_BackFromUserCode ENDP\r
97d92bda	129	\r
bbe4ba81	130	_EntryPoint DD _ToUserCode - m16Start\r
f23d790a	131	DW CODE16\r
bbe4ba81	132	_16Gdtr LABEL FWORD\r
f23d790a	133	DW GDT_SIZE - 1\r
f23d790a	134	_16GdtrBase DQ _NullSeg\r
bbe4ba81	135	_16Idtr FWORD (1 SHL 10) - 1\r
97d92bda	136	\r
3f566587	137	;------------------------------------------------------------------------------\r
	138	; _ToUserCode() takes control in real mode before passing control to user code.\r
	139	; It will be shadowed to somewhere in memory below 1MB.\r
	140	;------------------------------------------------------------------------------\r
97d92bda	141	_ToUserCode PROC\r
97d92bda	142	mov ss, edx ; set new segment selectors\r
	143	mov ds, edx\r
	144	mov es, edx\r
	145	mov fs, edx\r
	146	mov gs, edx\r
878ddf1f	147	DB 66h\r
97d92bda	148	mov ecx, 0c0000080h\r
	149	mov cr0, rax ; real mode starts at next instruction\r
	150	rdmsr\r
	151	and ah, NOT 1\r
	152	wrmsr\r
	153	mov cr4, rbp\r
	154	mov ss, esi ; set up 16-bit stack segment\r
38b5365e	155	mov sp, bx ; set up 16-bit stack pointer\r
878ddf1f	156	DB 66h\r
97d92bda	157	call @Base ; push eip\r
97d92bda	158	@Base:\r
bbe4ba81	159	pop bp ; ebp <- address of @Base\r
18c319ae	160	push [esp + sizeof (IA32_REGS) + 2]\r
bbe4ba81	161	lea eax, [rsi + (@RealMode - @Base)]\r
18c319ae	162	push rax\r
	163	retf\r
	164	@RealMode:\r
97d92bda	165	DB 66h, 2eh ; CS and operand size override\r
bbe4ba81	166	lidt fword ptr [rsi + (_16Idtr - @Base)]\r
97d92bda	167	DB 66h, 61h ; popad\r
	168	DB 1fh ; pop ds\r
	169	DB 07h ; pop es\r
	170	pop fs\r
	171	pop gs\r
	172	popf ; popfd\r
	173	lea sp, [esp + 4] ; skip high order 32 bits of EFlags\r
878ddf1f	174	DB 66h\r
97d92bda	175	retf ; transfer control to user code\r
	176	_ToUserCode ENDP\r
	177	\r
f23d790a	178	CODE16 = _16Code - $\r
	179	DATA16 = _16Data - $\r
	180	DATA32 = _32Data - $\r
	181	\r
	182	_NullSeg DQ 0\r
	183	_16Code LABEL QWORD\r
	184	DW -1\r
	185	DW 0\r
	186	DB 0\r
	187	DB 9bh\r
	188	DB 8fh ; 16-bit segment, 4GB limit\r
	189	DB 0\r
	190	_16Data LABEL QWORD\r
	191	DW -1\r
	192	DW 0\r
	193	DB 0\r
	194	DB 93h\r
	195	DB 8fh ; 16-bit segment, 4GB limit\r
	196	DB 0\r
	197	_32Data LABEL QWORD\r
	198	DW -1\r
	199	DW 0\r
	200	DB 0\r
	201	DB 93h\r
	202	DB 0cfh ; 16-bit segment, 4GB limit\r
	203	DB 0\r
	204	\r
	205	GDT_SIZE = $ - _NullSeg\r
97d92bda	206	\r
3f566587	207	;------------------------------------------------------------------------------\r
	208	; IA32_REGISTER_SET *\r
	209	; EFIAPI\r
	210	; InternalAsmThunk16 (\r
	211	; IN IA32_REGISTER_SET *RegisterSet,\r
	212	; IN OUT VOID *Transition\r
	213	; );\r
	214	;------------------------------------------------------------------------------\r
97d92bda	215	InternalAsmThunk16 PROC USES rbp rbx rsi rdi\r
	216	mov r10d, ds\r
	217	mov r11d, es\r
f23d790a	218	mov r9d, ss\r
97d92bda	219	push fs\r
	220	push gs\r
	221	mov rsi, rcx\r
	222	movzx r8d, (IA32_REGS ptr [rsi])._SS\r
	223	mov edi, (IA32_REGS ptr [rsi])._ESP\r
	224	lea rdi, [edi - (sizeof (IA32_REGS) + 4)]\r
	225	imul eax, r8d, 16 ; eax <- r8d(stack segment) * 16\r
bbe4ba81	226	mov ebx, edi ; ebx <- stack for 16-bit code\r
97d92bda	227	push sizeof (IA32_REGS) / 4\r
	228	add edi, eax ; edi <- linear address of 16-bit stack\r
	229	pop rcx\r
	230	rep movsd ; copy RegSet\r
bbe4ba81	231	lea ecx, [rdx + (SavedCr4 - m16Start)]\r
97d92bda	232	mov eax, edx ; eax <- transition code address\r
	233	and edx, 0fh\r
	234	shl eax, 12\r
38b5365e	235	lea ax, [rdx + (_BackFromUserCode - m16Start)]\r
97d92bda	236	stosd ; [edi] <- return address of user code\r
bbe4ba81	237	sgdt fword ptr [rcx + (SavedGdt - SavedCr4)]\r
97d92bda	238	sidt fword ptr [rsp + 38h] ; save IDT stack in argument space\r
97d92bda	239	mov rax, cr0\r
bbe4ba81	240	mov [rcx + (SavedCr0 - SavedCr4)], eax\r
97d92bda	241	and eax, 7ffffffeh ; clear PE, PG bits\r
	242	mov rbp, cr4\r
	243	mov [rcx], ebp ; save CR4 in SavedCr4\r
	244	and ebp, 300h ; clear all but PCE and OSFXSR bits\r
	245	mov esi, r8d ; esi <- 16-bit stack segment\r
f23d790a	246	DB 6ah, DATA32 ; push DATA32\r
f23d790a	247	pop rdx ; rdx <- 32-bit data segment selector\r
bbe4ba81	248	lgdt fword ptr [rcx + (_16Gdtr - SavedCr4)]\r
f23d790a	249	mov ss, edx\r
18c319ae	250	pushfq\r
f23d790a	251	lea edx, [rdx + DATA16 - DATA32]\r
38b5365e	252	lea r8, @RetFromRealMode\r
	253	mov [rcx + (SavedRip - SavedCr4)], r8\r
	254	mov r8d, cs\r
	255	mov [rcx + (SavedCs - SavedCr4)], r8w\r
	256	mov r8, rsp\r
	257	jmp fword ptr [rcx + (_EntryPoint - SavedCr4)]\r
	258	@RetFromRealMode:\r
	259	mov rsp, r8\r
18c319ae	260	popfq\r
97d92bda	261	lidt fword ptr [rsp + 38h] ; restore protected mode IDTR\r
97d92bda	262	lea eax, [rbp - sizeof (IA32_REGS)]\r
878ddf1f	263	pop gs\r
878ddf1f	264	pop fs\r
f23d790a	265	mov ss, r9d\r
97d92bda	266	mov es, r11d\r
97d92bda	267	mov ds, r10d\r
878ddf1f	268	ret\r
	269	InternalAsmThunk16 ENDP\r
	270	\r
	271	END\r