]> git.proxmox.com Git - mirror_edk2.git/blame_incremental - MdeModulePkg/Universal/Acpi/BootScriptExecutorDxe/X64/SetIdtEntry.c
MdeModulePkg: Replace BSD License with BSD+Patent License
[mirror_edk2.git] / MdeModulePkg / Universal / Acpi / BootScriptExecutorDxe / X64 / SetIdtEntry.c
... / ...
CommitLineData
1/** @file\r
2 Set a IDT entry for debug purpose\r
3\r
4 Set a IDT entry for interrupt vector 3 for debug purpose for x64 platform\r
5\r
6Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>\r
7Copyright (c) 2017, AMD Incorporated. All rights reserved.<BR>\r
8\r
9\r
10SPDX-License-Identifier: BSD-2-Clause-Patent\r
11\r
12**/\r
13#include "ScriptExecute.h"\r
14\r
15//\r
16// 8 extra pages for PF handler.\r
17//\r
18#define EXTRA_PAGE_TABLE_PAGES 8\r
19\r
20#define IA32_PG_P BIT0\r
21#define IA32_PG_RW BIT1\r
22#define IA32_PG_PS BIT7\r
23\r
24UINT64 mPhyMask;\r
25VOID *mOriginalHandler;\r
26UINTN mPageFaultBuffer;\r
27UINTN mPageFaultIndex = 0;\r
28//\r
29// Store the uplink information for each page being used.\r
30//\r
31UINT64 *mPageFaultUplink[EXTRA_PAGE_TABLE_PAGES];\r
32\r
33/**\r
34 Page fault handler.\r
35\r
36**/\r
37VOID\r
38EFIAPI\r
39PageFaultHandlerHook (\r
40 VOID\r
41 );\r
42\r
43/**\r
44 Hook IDT with our page fault handler so that the on-demand paging works on page fault.\r
45\r
46 @param IdtEntry a pointer to IDT entry\r
47\r
48**/\r
49VOID\r
50HookPageFaultHandler (\r
51 IN IA32_IDT_GATE_DESCRIPTOR *IdtEntry\r
52 )\r
53{\r
54 UINT32 RegEax;\r
55 UINT8 PhysicalAddressBits;\r
56 UINTN PageFaultHandlerHookAddress;\r
57\r
58 AsmCpuid (0x80000000, &RegEax, NULL, NULL, NULL);\r
59 if (RegEax >= 0x80000008) {\r
60 AsmCpuid (0x80000008, &RegEax, NULL, NULL, NULL);\r
61 PhysicalAddressBits = (UINT8) RegEax;\r
62 } else {\r
63 PhysicalAddressBits = 36;\r
64 }\r
65 mPhyMask = LShiftU64 (1, PhysicalAddressBits) - 1;\r
66 mPhyMask &= (1ull << 48) - SIZE_4KB;\r
67\r
68 //\r
69 // Set Page Fault entry to catch >4G access\r
70 //\r
71 PageFaultHandlerHookAddress = (UINTN)PageFaultHandlerHook;\r
72 mOriginalHandler = (VOID *)(UINTN)(LShiftU64 (IdtEntry->Bits.OffsetUpper, 32) + IdtEntry->Bits.OffsetLow + (IdtEntry->Bits.OffsetHigh << 16));\r
73 IdtEntry->Bits.OffsetLow = (UINT16)PageFaultHandlerHookAddress;\r
74 IdtEntry->Bits.Selector = (UINT16)AsmReadCs ();\r
75 IdtEntry->Bits.Reserved_0 = 0;\r
76 IdtEntry->Bits.GateType = IA32_IDT_GATE_TYPE_INTERRUPT_32;\r
77 IdtEntry->Bits.OffsetHigh = (UINT16)(PageFaultHandlerHookAddress >> 16);\r
78 IdtEntry->Bits.OffsetUpper = (UINT32)(PageFaultHandlerHookAddress >> 32);\r
79 IdtEntry->Bits.Reserved_1 = 0;\r
80\r
81 if (mPage1GSupport) {\r
82 mPageFaultBuffer = (UINTN)(AsmReadCr3 () & mPhyMask) + EFI_PAGES_TO_SIZE(2);\r
83 }else {\r
84 mPageFaultBuffer = (UINTN)(AsmReadCr3 () & mPhyMask) + EFI_PAGES_TO_SIZE(6);\r
85 }\r
86 ZeroMem (mPageFaultUplink, sizeof (mPageFaultUplink));\r
87}\r
88\r
89/**\r
90 The function will check if current waking vector is long mode.\r
91\r
92 @param AcpiS3Context a pointer to a structure of ACPI_S3_CONTEXT\r
93\r
94 @retval TRUE Current context need long mode waking vector.\r
95 @retval FALSE Current context need not long mode waking vector.\r
96**/\r
97BOOLEAN\r
98IsLongModeWakingVector (\r
99 IN ACPI_S3_CONTEXT *AcpiS3Context\r
100 )\r
101{\r
102 EFI_ACPI_4_0_FIRMWARE_ACPI_CONTROL_STRUCTURE *Facs;\r
103\r
104 Facs = (EFI_ACPI_4_0_FIRMWARE_ACPI_CONTROL_STRUCTURE *) ((UINTN) (AcpiS3Context->AcpiFacsTable));\r
105 if ((Facs == NULL) ||\r
106 (Facs->Signature != EFI_ACPI_4_0_FIRMWARE_ACPI_CONTROL_STRUCTURE_SIGNATURE) ||\r
107 ((Facs->FirmwareWakingVector == 0) && (Facs->XFirmwareWakingVector == 0)) ) {\r
108 // Something wrong with FACS\r
109 return FALSE;\r
110 }\r
111 if (Facs->XFirmwareWakingVector != 0) {\r
112 if ((Facs->Version == EFI_ACPI_4_0_FIRMWARE_ACPI_CONTROL_STRUCTURE_VERSION) &&\r
113 ((Facs->Flags & EFI_ACPI_4_0_64BIT_WAKE_SUPPORTED_F) != 0) &&\r
114 ((Facs->OspmFlags & EFI_ACPI_4_0_OSPM_64BIT_WAKE__F) != 0)) {\r
115 // Both BIOS and OS wants 64bit vector\r
116 if (FeaturePcdGet (PcdDxeIplSwitchToLongMode)) {\r
117 return TRUE;\r
118 }\r
119 }\r
120 }\r
121 return FALSE;\r
122}\r
123\r
124/**\r
125 Set a IDT entry for interrupt vector 3 for debug purpose.\r
126\r
127 @param AcpiS3Context a pointer to a structure of ACPI_S3_CONTEXT\r
128\r
129**/\r
130VOID\r
131SetIdtEntry (\r
132 IN ACPI_S3_CONTEXT *AcpiS3Context\r
133 )\r
134{\r
135 IA32_IDT_GATE_DESCRIPTOR *IdtEntry;\r
136 IA32_DESCRIPTOR *IdtDescriptor;\r
137 UINTN S3DebugBuffer;\r
138 EFI_STATUS Status;\r
139\r
140 //\r
141 // Restore IDT for debug\r
142 //\r
143 IdtDescriptor = (IA32_DESCRIPTOR *) (UINTN) (AcpiS3Context->IdtrProfile);\r
144 AsmWriteIdtr (IdtDescriptor);\r
145\r
146 //\r
147 // Setup the default CPU exception handlers\r
148 //\r
149 Status = InitializeCpuExceptionHandlers (NULL);\r
150 ASSERT_EFI_ERROR (Status);\r
151\r
152 DEBUG_CODE (\r
153 //\r
154 // Update IDT entry INT3 if the instruction is valid in it\r
155 //\r
156 S3DebugBuffer = (UINTN) (AcpiS3Context->S3DebugBufferAddress);\r
157 if (*(UINTN *)S3DebugBuffer != (UINTN) -1) {\r
158 IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *)(IdtDescriptor->Base + (3 * sizeof (IA32_IDT_GATE_DESCRIPTOR)));\r
159 IdtEntry->Bits.OffsetLow = (UINT16)S3DebugBuffer;\r
160 IdtEntry->Bits.Selector = (UINT16)AsmReadCs ();\r
161 IdtEntry->Bits.Reserved_0 = 0;\r
162 IdtEntry->Bits.GateType = IA32_IDT_GATE_TYPE_INTERRUPT_32;\r
163 IdtEntry->Bits.OffsetHigh = (UINT16)(S3DebugBuffer >> 16);\r
164 IdtEntry->Bits.OffsetUpper = (UINT32)(S3DebugBuffer >> 32);\r
165 IdtEntry->Bits.Reserved_1 = 0;\r
166 }\r
167 );\r
168\r
169 //\r
170 // If both BIOS and OS wants long mode waking vector,\r
171 // S3ResumePei should have established 1:1 Virtual to Physical identity mapping page table,\r
172 // no need to hook page fault handler.\r
173 //\r
174 if (!IsLongModeWakingVector (AcpiS3Context)) {\r
175 IdtEntry = (IA32_IDT_GATE_DESCRIPTOR *)(IdtDescriptor->Base + (14 * sizeof (IA32_IDT_GATE_DESCRIPTOR)));\r
176 HookPageFaultHandler (IdtEntry);\r
177 }\r
178}\r
179\r
180/**\r
181 Acquire page for page fault.\r
182\r
183 @param[in, out] Uplink Pointer to up page table entry.\r
184\r
185**/\r
186VOID\r
187AcquirePage (\r
188 IN OUT UINT64 *Uplink\r
189 )\r
190{\r
191 UINTN Address;\r
192\r
193 Address = mPageFaultBuffer + EFI_PAGES_TO_SIZE (mPageFaultIndex);\r
194 ZeroMem ((VOID *) Address, EFI_PAGES_TO_SIZE (1));\r
195\r
196 //\r
197 // Cut the previous uplink if it exists and wasn't overwritten.\r
198 //\r
199 if ((mPageFaultUplink[mPageFaultIndex] != NULL) &&\r
200 ((*mPageFaultUplink[mPageFaultIndex] & ~mAddressEncMask & mPhyMask) == Address)) {\r
201 *mPageFaultUplink[mPageFaultIndex] = 0;\r
202 }\r
203\r
204 //\r
205 // Link & Record the current uplink.\r
206 //\r
207 *Uplink = Address | mAddressEncMask | IA32_PG_P | IA32_PG_RW;\r
208 mPageFaultUplink[mPageFaultIndex] = Uplink;\r
209\r
210 mPageFaultIndex = (mPageFaultIndex + 1) % EXTRA_PAGE_TABLE_PAGES;\r
211}\r
212\r
213/**\r
214 The page fault handler that on-demand read >4G memory/MMIO.\r
215\r
216 @retval TRUE The page fault is correctly handled.\r
217 @retval FALSE The page fault is not handled and is passed through to original handler.\r
218\r
219**/\r
220BOOLEAN\r
221EFIAPI\r
222PageFaultHandler (\r
223 VOID\r
224 )\r
225{\r
226 UINT64 *PageTable;\r
227 UINT64 PFAddress;\r
228 UINTN PTIndex;\r
229\r
230 PFAddress = AsmReadCr2 ();\r
231 DEBUG ((DEBUG_INFO, "BootScript - PageFaultHandler: Cr2 - %lx\n", PFAddress));\r
232\r
233 if (PFAddress >= mPhyMask + SIZE_4KB) {\r
234 return FALSE;\r
235 }\r
236 PFAddress &= mPhyMask;\r
237\r
238 PageTable = (UINT64*)(UINTN)(AsmReadCr3 () & mPhyMask);\r
239\r
240 PTIndex = BitFieldRead64 (PFAddress, 39, 47);\r
241 // PML4E\r
242 if ((PageTable[PTIndex] & IA32_PG_P) == 0) {\r
243 AcquirePage (&PageTable[PTIndex]);\r
244 }\r
245 PageTable = (UINT64*)(UINTN)(PageTable[PTIndex] & ~mAddressEncMask & mPhyMask);\r
246 PTIndex = BitFieldRead64 (PFAddress, 30, 38);\r
247 // PDPTE\r
248 if (mPage1GSupport) {\r
249 PageTable[PTIndex] = ((PFAddress | mAddressEncMask) & ~((1ull << 30) - 1)) | IA32_PG_P | IA32_PG_RW | IA32_PG_PS;\r
250 } else {\r
251 if ((PageTable[PTIndex] & IA32_PG_P) == 0) {\r
252 AcquirePage (&PageTable[PTIndex]);\r
253 }\r
254 PageTable = (UINT64*)(UINTN)(PageTable[PTIndex] & ~mAddressEncMask & mPhyMask);\r
255 PTIndex = BitFieldRead64 (PFAddress, 21, 29);\r
256 // PD\r
257 PageTable[PTIndex] = ((PFAddress | mAddressEncMask) & ~((1ull << 21) - 1)) | IA32_PG_P | IA32_PG_RW | IA32_PG_PS;\r
258 }\r
259\r
260 return TRUE;\r
261}\r