[mirror_edk2.git] / UefiCpuPkg / CpuMpPei / Ia32 / MpFuncs.asm

;------------------------------------------------------------------------------ ;\r
; Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>\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
; Module Name:\r
;\r
;   MpFuncs32.asm\r
;\r
; Abstract:\r
;\r
;   This is the assembly code for MP support\r
;\r
;-------------------------------------------------------------------------------\r
\r
.686p\r
.model  flat\r
\r
include  MpEqu.inc\r
InitializeFloatingPointUnits PROTO C\r
\r
.code\r
\r
;-------------------------------------------------------------------------------------\r
;RendezvousFunnelProc  procedure follows. All APs execute their procedure. This\r
;procedure serializes all the AP processors through an Init sequence. It must be\r
;noted that APs arrive here very raw...ie: real mode, no stack.\r
;ALSO THIS PROCEDURE IS EXECUTED BY APs ONLY ON 16 BIT MODE. HENCE THIS PROC\r
;IS IN MACHINE CODE.\r
;-------------------------------------------------------------------------------------\r
RendezvousFunnelProc   PROC  PUBLIC\r
RendezvousFunnelProcStart::\r
; At this point CS = 0x(vv00) and ip= 0x0.\r
; Save BIST information to ebp firstly\r
    db 66h,  08bh, 0e8h                 ; mov        ebp, eax    ; save BIST information\r
\r
    db 8ch,0c8h                         ; mov        ax,cs\r
    db 8eh,0d8h                         ; mov        ds,ax\r
    db 8eh,0c0h                         ; mov        es,ax\r
    db 8eh,0d0h                         ; mov        ss,ax\r
    db 33h,0c0h                         ; xor        ax,ax\r
    db 8eh,0e0h                         ; mov        fs,ax\r
    db 8eh,0e8h                         ; mov        gs,ax\r
\r
    db 0BEh                             ; opcode of mov si, mem16\r
    dw BufferStartLocation              ; mov        si, BufferStartLocation\r
    db 66h,  8Bh, 1Ch                   ; mov        ebx,dword ptr [si]\r
\r
    db 0BFh                             ; opcode of mov di, mem16\r
    dw PmodeOffsetLocation              ; mov        di, PmodeOffsetLocation\r
    db 66h,  8Bh, 05h                   ; mov        eax,dword ptr [di]\r
    db 8Bh,  0F8h                       ; mov        di, ax\r
    db 83h,  0EFh,06h                   ; sub        di, 06h\r
    db 66h,  03h, 0C3h                  ; add        eax, ebx\r
    db 66h,  89h, 05h                   ; mov        dword ptr [di],eax\r
\r
    db 0BEh                             ; opcode of mov si, mem16\r
    dw GdtrLocation                     ; mov        si, GdtrLocation\r
    db 66h                              ; db         66h\r
    db 2Eh,  0Fh, 01h, 14h              ; lgdt       fword ptr cs:[si]\r
\r
    db 0BEh\r
    dw IdtrLocation                     ; mov        si, IdtrLocation\r
    db 66h                              ; db         66h\r
    db 2Eh,0Fh, 01h, 1Ch                ; lidt       fword ptr cs:[si]\r
\r
    db 33h,  0C0h                       ; xor        ax,  ax\r
    db 8Eh,  0D8h                       ; mov        ds,  ax\r
\r
    db 0Fh,  20h, 0C0h                  ; mov        eax, cr0            ;Get control register 0\r
    db 66h,  83h, 0C8h, 03h             ; or         eax, 000000003h     ;Set PE bit (bit #0) & MP\r
    db 0Fh,  22h, 0C0h                  ; mov        cr0, eax\r
\r
    db 66h,  67h, 0EAh                  ; far jump\r
    dd 0h                               ; 32-bit offset\r
    dw PROTECT_MODE_CS                  ; 16-bit selector\r
\r
Flat32Start::                           ; protected mode entry point\r
    mov        ax, PROTECT_MODE_DS\r
    mov        ds, ax\r
    mov        es, ax\r
    mov        fs, ax\r
    mov        gs, ax\r
    mov        ss, ax\r
\r
    mov        esi, ebx\r
    mov        edi, esi\r
    add        edi, LockLocation\r
    mov        eax, NotVacantFlag\r
\r
TestLock:\r
    xchg       dword ptr [edi], eax\r
    cmp        eax, NotVacantFlag\r
    jz         TestLock\r
\r
    mov        edi, esi\r
    add        edi, NumApsExecutingLoction\r
    inc        dword ptr [edi]\r
    mov        ebx, dword ptr [edi]\r
\r
ProgramStack:\r
    mov        edi, esi\r
    add        edi, StackSizeLocation\r
    mov        eax, dword ptr [edi]\r
    mov        edi, esi\r
    add        edi, StackStartAddressLocation\r
    add        eax, dword ptr [edi]\r
    mov        esp, eax\r
    mov        dword ptr [edi], eax\r
\r
Releaselock:\r
    mov        eax, VacantFlag\r
    mov        edi, esi\r
    add        edi, LockLocation\r
    xchg       dword ptr [edi], eax\r
\r
CProcedureInvoke:\r
    push       ebp               ; push BIST data at top of AP stack\r
    xor        ebp, ebp          ; clear ebp for call stack trace\r
    push       ebp\r
    mov        ebp, esp\r
\r
    mov        eax, InitializeFloatingPointUnits\r
    call       eax               ; Call assembly function to initialize FPU per UEFI spec\r
\r
    push       ebx               ; Push NumApsExecuting\r
    mov        eax, esi\r
    add        eax, LockLocation\r
    push       eax               ; push address of exchange info data buffer\r
\r
    mov        edi, esi\r
    add        edi, ApProcedureLocation\r
    mov        eax, dword ptr [edi]\r
\r
    call       eax               ; invoke C function\r
\r
    jmp        $                  ; never reach here\r
\r
RendezvousFunnelProc   ENDP\r
RendezvousFunnelProcEnd::\r
\r
AsmCliHltLoop PROC near C PUBLIC\r
    cli\r
    hlt\r
    jmp        $-2\r
AsmCliHltLoop ENDP\r
\r
;-------------------------------------------------------------------------------------\r
;  AsmGetAddressMap (&AddressMap);\r
;-------------------------------------------------------------------------------------\r
AsmGetAddressMap   PROC  near C  PUBLIC\r
    pushad\r
    mov        ebp,esp\r
\r
    mov        ebx, dword ptr [ebp+24h]\r
    mov        dword ptr [ebx], RendezvousFunnelProcStart\r
    mov        dword ptr [ebx +  4h], Flat32Start - RendezvousFunnelProcStart\r
    mov        dword ptr [ebx +  8h], 0\r
    mov        dword ptr [ebx + 0ch], RendezvousFunnelProcEnd - RendezvousFunnelProcStart\r
\r
    popad\r
    ret\r
AsmGetAddressMap   ENDP\r
\r
PAUSE32   MACRO\r
    DB      0F3h\r
    DB      090h\r
    ENDM\r
\r
;-------------------------------------------------------------------------------------\r
;AsmExchangeRole procedure follows. This procedure executed by current BSP, that is\r
;about to become an AP. It switches it'stack with the current AP.\r
;AsmExchangeRole (IN   CPU_EXCHANGE_INFO    *MyInfo, IN   CPU_EXCHANGE_INFO    *OthersInfo);\r
;-------------------------------------------------------------------------------------\r
AsmExchangeRole   PROC  near C  PUBLIC\r
    ; DO NOT call other functions in this function, since 2 CPU may use 1 stack\r
    ; at the same time. If 1 CPU try to call a function, stack will be corrupted.\r
    pushad\r
    mov        ebp,esp\r
\r
    ; esi contains MyInfo pointer\r
    mov        esi, dword ptr [ebp+24h]\r
\r
    ; edi contains OthersInfo pointer\r
    mov        edi, dword ptr [ebp+28h]\r
\r
    ;Store EFLAGS, GDTR and IDTR register to stack\r
    pushfd\r
    mov        eax, cr4\r
    push       eax       ; push cr4 firstly\r
    mov        eax, cr0\r
    push       eax\r
\r
    sgdt       fword ptr [esi+8]\r
    sidt       fword ptr [esi+14]\r
\r
    ; Store the its StackPointer\r
    mov        dword ptr [esi+4],esp\r
\r
    ; update its switch state to STORED\r
    mov        byte ptr [esi], CPU_SWITCH_STATE_STORED\r
\r
WaitForOtherStored:\r
    ; wait until the other CPU finish storing its state\r
    cmp        byte ptr [edi], CPU_SWITCH_STATE_STORED\r
    jz         OtherStored\r
    PAUSE32\r
    jmp        WaitForOtherStored\r
\r
OtherStored:\r
    ; Since another CPU already stored its state, load them\r
    ; load GDTR value\r
    lgdt       fword ptr [edi+8]\r
\r
    ; load IDTR value\r
    lidt       fword ptr [edi+14]\r
\r
    ; load its future StackPointer\r
    mov        esp, dword ptr [edi+4]\r
\r
    ; update the other CPU's switch state to LOADED\r
    mov        byte ptr [edi], CPU_SWITCH_STATE_LOADED\r
\r
WaitForOtherLoaded:\r
    ; wait until the other CPU finish loading new state,\r
    ; otherwise the data in stack may corrupt\r
    cmp        byte ptr [esi], CPU_SWITCH_STATE_LOADED\r
    jz         OtherLoaded\r
    PAUSE32\r
    jmp        WaitForOtherLoaded\r
\r
OtherLoaded:\r
    ; since the other CPU already get the data it want, leave this procedure\r
    pop        eax\r
    mov        cr0, eax\r
    pop        eax\r
    mov        cr4, eax\r
    popfd\r
\r
    popad\r
    ret\r
AsmExchangeRole   ENDP\r
\r
AsmInitializeGdt   PROC  near C  PUBLIC\r
  push         ebp\r
  mov          ebp, esp\r
  pushad\r
  mov          edi, [ebp + 8]      ; Load GDT register\r
\r
  mov          ax,cs               ; Get the selector data from our code image\r
  mov          es,ax\r
  lgdt         FWORD PTR es:[edi]  ; and update the GDTR\r
\r
  push         PROTECT_MODE_CS\r
  lea          eax, SetCodeSelectorFarJump\r
  push         eax\r
  retf\r
SetCodeSelectorFarJump:\r
  mov          ax, PROTECT_MODE_DS ; Update the Base for the new selectors, too\r
  mov          ds, ax\r
  mov          es, ax\r
  mov          fs, ax\r
  mov          gs, ax\r
  mov          ss, ax\r
\r
  popad\r
  pop          ebp\r
  ret\r
AsmInitializeGdt  ENDP\r
\r
END\r
Commit	Line	Data
ea0f431c JF	1	;------------------------------------------------------------------------------ ;\r
	2	; Copyright (c) 2015, Intel Corporation. All rights reserved.<BR>\r
	3	; This program and the accompanying materials\r
	4	; are licensed and made available under the terms and conditions of the BSD License\r
	5	; which accompanies this distribution. The full text of the license may be found at\r
	6	; http://opensource.org/licenses/bsd-license.php.\r
	7	;\r
	8	; THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
	9	; WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
	10	;\r
	11	; Module Name:\r
	12	;\r
	13	; MpFuncs32.asm\r
	14	;\r
	15	; Abstract:\r
	16	;\r
	17	; This is the assembly code for MP support\r
	18	;\r
	19	;-------------------------------------------------------------------------------\r
	20	\r
	21	.686p\r
	22	.model flat\r
	23	\r
	24	include MpEqu.inc\r
	25	InitializeFloatingPointUnits PROTO C\r
	26	\r
	27	.code\r
	28	\r
	29	;-------------------------------------------------------------------------------------\r
	30	;RendezvousFunnelProc procedure follows. All APs execute their procedure. This\r
	31	;procedure serializes all the AP processors through an Init sequence. It must be\r
	32	;noted that APs arrive here very raw...ie: real mode, no stack.\r
	33	;ALSO THIS PROCEDURE IS EXECUTED BY APs ONLY ON 16 BIT MODE. HENCE THIS PROC\r
	34	;IS IN MACHINE CODE.\r
	35	;-------------------------------------------------------------------------------------\r
	36	RendezvousFunnelProc PROC PUBLIC\r
	37	RendezvousFunnelProcStart::\r
	38	; At this point CS = 0x(vv00) and ip= 0x0.\r
	39	; Save BIST information to ebp firstly\r
	40	db 66h, 08bh, 0e8h ; mov ebp, eax ; save BIST information\r
	41	\r
	42	db 8ch,0c8h ; mov ax,cs\r
	43	db 8eh,0d8h ; mov ds,ax\r
	44	db 8eh,0c0h ; mov es,ax\r
	45	db 8eh,0d0h ; mov ss,ax\r
	46	db 33h,0c0h ; xor ax,ax\r
	47	db 8eh,0e0h ; mov fs,ax\r
	48	db 8eh,0e8h ; mov gs,ax\r
	49	\r
	50	db 0BEh ; opcode of mov si, mem16\r
	51	dw BufferStartLocation ; mov si, BufferStartLocation\r
	52	db 66h, 8Bh, 1Ch ; mov ebx,dword ptr [si]\r
	53	\r
	54	db 0BFh ; opcode of mov di, mem16\r
	55	dw PmodeOffsetLocation ; mov di, PmodeOffsetLocation\r
	56	db 66h, 8Bh, 05h ; mov eax,dword ptr [di]\r
	57	db 8Bh, 0F8h ; mov di, ax\r
	58	db 83h, 0EFh,06h ; sub di, 06h\r
	59	db 66h, 03h, 0C3h ; add eax, ebx\r
	60	db 66h, 89h, 05h ; mov dword ptr [di],eax\r
	61	\r
	62	db 0BEh ; opcode of mov si, mem16\r
	63	dw GdtrLocation ; mov si, GdtrLocation\r
	64	db 66h ; db 66h\r
65	db 2Eh, 0Fh, 01h, 14h ; lgdt fword ptr cs:[si]\r
66	\r
67	db 0BEh\r
68	dw IdtrLocation ; mov si, IdtrLocation\r
69	db 66h ; db 66h\r
70	db 2Eh,0Fh, 01h, 1Ch ; lidt fword ptr cs:[si]\r
71	\r
72	db 33h, 0C0h ; xor ax, ax\r
73	db 8Eh, 0D8h ; mov ds, ax\r
74	\r
75	db 0Fh, 20h, 0C0h ; mov eax, cr0 ;Get control register 0\r
76	db 66h, 83h, 0C8h, 03h ; or eax, 000000003h ;Set PE bit (bit #0) & MP\r
77	db 0Fh, 22h, 0C0h ; mov cr0, eax\r
78	\r
79	db 66h, 67h, 0EAh ; far jump\r
80	dd 0h ; 32-bit offset\r
81	dw PROTECT_MODE_CS ; 16-bit selector\r
82	\r
83	Flat32Start:: ; protected mode entry point\r
84	mov ax, PROTECT_MODE_DS\r
85	mov ds, ax\r
86	mov es, ax\r
87	mov fs, ax\r
88	mov gs, ax\r
89	mov ss, ax\r
90	\r
91	mov esi, ebx\r
92	mov edi, esi\r
93	add edi, LockLocation\r
94	mov eax, NotVacantFlag\r
95	\r
96	TestLock:\r
97	xchg dword ptr [edi], eax\r
98	cmp eax, NotVacantFlag\r
99	jz TestLock\r
100	\r
101	mov edi, esi\r
102	add edi, NumApsExecutingLoction\r
103	inc dword ptr [edi]\r
104	mov ebx, dword ptr [edi]\r
105	\r
106	ProgramStack:\r
107	mov edi, esi\r
108	add edi, StackSizeLocation\r
109	mov eax, dword ptr [edi]\r
110	mov edi, esi\r
111	add edi, StackStartAddressLocation\r
112	add eax, dword ptr [edi]\r
113	mov esp, eax\r
114	mov dword ptr [edi], eax\r
115	\r
116	Releaselock:\r
117	mov eax, VacantFlag\r
118	mov edi, esi\r
119	add edi, LockLocation\r
120	xchg dword ptr [edi], eax\r
121	\r
122	CProcedureInvoke:\r
123	push ebp ; push BIST data at top of AP stack\r
124	xor ebp, ebp ; clear ebp for call stack trace\r
125	push ebp\r
126	mov ebp, esp\r
127	\r
128	mov eax, InitializeFloatingPointUnits\r
129	call eax ; Call assembly function to initialize FPU per UEFI spec\r
130	\r
131	push ebx ; Push NumApsExecuting\r
132	mov eax, esi\r
133	add eax, LockLocation\r
134	push eax ; push address of exchange info data buffer\r
135	\r
136	mov edi, esi\r
137	add edi, ApProcedureLocation\r
138	mov eax, dword ptr [edi]\r
139	\r
140	call eax ; invoke C function\r
141	\r
142	jmp $ ; never reach here\r
143	\r
144	RendezvousFunnelProc ENDP\r
145	RendezvousFunnelProcEnd::\r
146	\r
147	AsmCliHltLoop PROC near C PUBLIC\r
148	cli\r
149	hlt\r
150	jmp $-2\r
151	AsmCliHltLoop ENDP\r
152	\r
153	;-------------------------------------------------------------------------------------\r
154	; AsmGetAddressMap (&AddressMap);\r
155	;-------------------------------------------------------------------------------------\r
156	AsmGetAddressMap PROC near C PUBLIC\r
157	pushad\r
158	mov ebp,esp\r
159	\r
160	mov ebx, dword ptr [ebp+24h]\r
161	mov dword ptr [ebx], RendezvousFunnelProcStart\r
162	mov dword ptr [ebx + 4h], Flat32Start - RendezvousFunnelProcStart\r
163	mov dword ptr [ebx + 8h], 0\r
164	mov dword ptr [ebx + 0ch], RendezvousFunnelProcEnd - RendezvousFunnelProcStart\r
165	\r
166	popad\r
167	ret\r
168	AsmGetAddressMap ENDP\r
169	\r
170	PAUSE32 MACRO\r
171	DB 0F3h\r
172	DB 090h\r
173	ENDM\r
174	\r
175	;-------------------------------------------------------------------------------------\r
176	;AsmExchangeRole procedure follows. This procedure executed by current BSP, that is\r
177	;about to become an AP. It switches it'stack with the current AP.\r
178	;AsmExchangeRole (IN CPU_EXCHANGE_INFO MyInfo, IN CPU_EXCHANGE_INFO OthersInfo);\r
179	;-------------------------------------------------------------------------------------\r
180	AsmExchangeRole PROC near C PUBLIC\r
181	; DO NOT call other functions in this function, since 2 CPU may use 1 stack\r
182	; at the same time. If 1 CPU try to call a function, stack will be corrupted.\r
183	pushad\r
184	mov ebp,esp\r
185	\r
186	; esi contains MyInfo pointer\r
187	mov esi, dword ptr [ebp+24h]\r
188	\r
189	; edi contains OthersInfo pointer\r
190	mov edi, dword ptr [ebp+28h]\r
191	\r
192	;Store EFLAGS, GDTR and IDTR register to stack\r
193	pushfd\r
194	mov eax, cr4\r
195	push eax ; push cr4 firstly\r
196	mov eax, cr0\r
197	push eax\r
198	\r
199	sgdt fword ptr [esi+8]\r
200	sidt fword ptr [esi+14]\r
201	\r
202	; Store the its StackPointer\r
203	mov dword ptr [esi+4],esp\r
204	\r
205	; update its switch state to STORED\r
206	mov byte ptr [esi], CPU_SWITCH_STATE_STORED\r
207	\r
208	WaitForOtherStored:\r
209	; wait until the other CPU finish storing its state\r
210	cmp byte ptr [edi], CPU_SWITCH_STATE_STORED\r
211	jz OtherStored\r
212	PAUSE32\r
213	jmp WaitForOtherStored\r
214	\r
215	OtherStored:\r
216	; Since another CPU already stored its state, load them\r
217	; load GDTR value\r
218	lgdt fword ptr [edi+8]\r
219	\r
220	; load IDTR value\r
221	lidt fword ptr [edi+14]\r
222	\r
223	; load its future StackPointer\r
224	mov esp, dword ptr [edi+4]\r
225	\r
226	; update the other CPU's switch state to LOADED\r
227	mov byte ptr [edi], CPU_SWITCH_STATE_LOADED\r
228	\r
229	WaitForOtherLoaded:\r
230	; wait until the other CPU finish loading new state,\r
231	; otherwise the data in stack may corrupt\r
232	cmp byte ptr [esi], CPU_SWITCH_STATE_LOADED\r
233	jz OtherLoaded\r
234	PAUSE32\r
235	jmp WaitForOtherLoaded\r
236	\r
237	OtherLoaded:\r
238	; since the other CPU already get the data it want, leave this procedure\r
239	pop eax\r
240	mov cr0, eax\r
241	pop eax\r
242	mov cr4, eax\r
243	popfd\r
244	\r
245	popad\r
246	ret\r
247	AsmExchangeRole ENDP\r
248	\r
249	AsmInitializeGdt PROC near C PUBLIC\r
250	push ebp\r
251	mov ebp, esp\r
252	pushad\r
253	mov edi, [ebp + 8] ; Load GDT register\r
254	\r
255	mov ax,cs ; Get the selector data from our code image\r
256	mov es,ax\r
257	lgdt FWORD PTR es:[edi] ; and update the GDTR\r
258	\r
259	push PROTECT_MODE_CS\r
260	lea eax, SetCodeSelectorFarJump\r
261	push eax\r
262	retf\r
263	SetCodeSelectorFarJump:\r
264	mov ax, PROTECT_MODE_DS ; Update the Base for the new selectors, too\r
265	mov ds, ax\r
266	mov es, ax\r
267	mov fs, ax\r
268	mov gs, ax\r
269	mov ss, ax\r
270	\r
271	popad\r
272	pop ebp\r
273	ret\r
274	AsmInitializeGdt ENDP\r
275	\r
276	END\r