]> git.proxmox.com Git - mirror_edk2.git/blob - MdeModulePkg/Core/Pei/Dispatcher/Dispatcher.c
Add assert check to make sure new stack size larger than old stack size.
[mirror_edk2.git] / MdeModulePkg / Core / Pei / Dispatcher / Dispatcher.c
1 /** @file
2 EFI PEI Core dispatch services
3
4 Copyright (c) 2006, Intel Corporation
5 All rights reserved. This program and the accompanying materials
6 are licensed and made available under the terms and conditions of the BSD License
7 which accompanies this distribution. The full text of the license may be found at
8 http://opensource.org/licenses/bsd-license.php
9
10 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
11 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
12
13 **/
14
15 #include <PeiMain.h>
16
17 ///
18 /// CAR is filled with this initial value during SEC phase
19 ///
20 #define INIT_CAR_VALUE 0x5AA55AA5
21
22 typedef struct {
23 EFI_STATUS_CODE_DATA DataHeader;
24 EFI_HANDLE Handle;
25 } PEIM_FILE_HANDLE_EXTENDED_DATA;
26
27 /**
28
29 Discover all Peims and optional Apriori file in one FV. There is at most one
30 Apriori file in one FV.
31
32
33 @param Private - Pointer to the private data passed in from caller
34 @param VolumeHandle - Fv handle.
35
36 **/
37 VOID
38 DiscoverPeimsAndOrderWithApriori (
39 IN PEI_CORE_INSTANCE *Private,
40 IN EFI_PEI_FV_HANDLE VolumeHandle
41 )
42 {
43 EFI_STATUS Status;
44 EFI_PEI_FV_HANDLE FileHandle;
45 EFI_PEI_FILE_HANDLE AprioriFileHandle;
46 EFI_GUID *Apriori;
47 UINTN Index;
48 UINTN Index2;
49 UINTN PeimIndex;
50 UINTN PeimCount;
51 EFI_GUID *Guid;
52 EFI_PEI_FV_HANDLE TempFileHandles[FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv)];
53 EFI_GUID FileGuid[FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv)];
54
55 //
56 // Walk the FV and find all the PEIMs and the Apriori file.
57 //
58 AprioriFileHandle = NULL;
59 Private->CurrentFvFileHandles[0] = NULL;
60 Guid = NULL;
61 FileHandle = NULL;
62
63 //
64 // If the current Fv has been scanned, directly get its cachable record.
65 //
66 if (Private->Fv[Private->CurrentPeimFvCount].ScanFv) {
67 CopyMem (Private->CurrentFvFileHandles, Private->Fv[Private->CurrentPeimFvCount].FvFileHandles, sizeof (Private->CurrentFvFileHandles));
68 return;
69 }
70
71 //
72 // Go ahead to scan this Fv, and cache FileHandles within it.
73 //
74 for (PeimCount = 0; PeimCount < FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv); PeimCount++) {
75 Status = PeiFindFileEx (
76 VolumeHandle,
77 NULL,
78 PEI_CORE_INTERNAL_FFS_FILE_DISPATCH_TYPE,
79 &FileHandle,
80 &AprioriFileHandle
81 );
82 if (Status != EFI_SUCCESS) {
83 break;
84 }
85
86 Private->CurrentFvFileHandles[PeimCount] = FileHandle;
87 }
88
89 Private->AprioriCount = 0;
90 if (AprioriFileHandle != NULL) {
91 //
92 // Read the Apriori file
93 //
94 Status = PeiServicesFfsFindSectionData (EFI_SECTION_RAW, AprioriFileHandle, (VOID **) &Apriori);
95 if (!EFI_ERROR (Status)) {
96 //
97 // Calculate the number of PEIMs in the A Priori list
98 //
99 Private->AprioriCount = *(UINT32 *)(((EFI_FFS_FILE_HEADER *)AprioriFileHandle)->Size) & 0x00FFFFFF;
100 Private->AprioriCount -= sizeof (EFI_FFS_FILE_HEADER) - sizeof (EFI_COMMON_SECTION_HEADER);
101 Private->AprioriCount /= sizeof (EFI_GUID);
102
103 ZeroMem (FileGuid, sizeof (FileGuid));
104 for (Index = 0; Index < PeimCount; Index++) {
105 //
106 // Make an array of file name guids that matches the FileHandle array so we can convert
107 // quickly from file name to file handle
108 //
109 CopyMem (&FileGuid[Index], &((EFI_FFS_FILE_HEADER *)Private->CurrentFvFileHandles[Index])->Name,sizeof(EFI_GUID));
110 }
111
112 //
113 // Walk through FileGuid array to find out who is invalid PEIM guid in Apriori file.
114 // Add avalible PEIMs in Apriori file into TempFileHandles array at first.
115 //
116 Index2 = 0;
117 for (Index = 0; Index2 < Private->AprioriCount; Index++) {
118 while (Index2 < Private->AprioriCount) {
119 Guid = ScanGuid (FileGuid, PeimCount * sizeof (EFI_GUID), &Apriori[Index2++]);
120 if (Guid != NULL) {
121 break;
122 }
123 }
124 if (Guid == NULL) {
125 break;
126 }
127 PeimIndex = ((UINTN)Guid - (UINTN)&FileGuid[0])/sizeof (EFI_GUID);
128 TempFileHandles[Index] = Private->CurrentFvFileHandles[PeimIndex];
129
130 //
131 // Since we have copied the file handle we can remove it from this list.
132 //
133 Private->CurrentFvFileHandles[PeimIndex] = NULL;
134 }
135
136 //
137 // Update valid Aprioricount
138 //
139 Private->AprioriCount = Index;
140
141 //
142 // Add in any PEIMs not in the Apriori file
143 //
144 for (;Index < PeimCount; Index++) {
145 for (Index2 = 0; Index2 < PeimCount; Index2++) {
146 if (Private->CurrentFvFileHandles[Index2] != NULL) {
147 TempFileHandles[Index] = Private->CurrentFvFileHandles[Index2];
148 Private->CurrentFvFileHandles[Index2] = NULL;
149 break;
150 }
151 }
152 }
153 //
154 //Index the end of array contains re-range Pei moudle.
155 //
156 TempFileHandles[Index] = NULL;
157
158 //
159 // Private->CurrentFvFileHandles is currently in PEIM in the FV order.
160 // We need to update it to start with files in the A Priori list and
161 // then the remaining files in PEIM order.
162 //
163 CopyMem (Private->CurrentFvFileHandles, TempFileHandles, sizeof (Private->CurrentFvFileHandles));
164 }
165 }
166 //
167 // Cache the current Fv File Handle. So that we don't have to scan the Fv again.
168 // Instead, we can retrieve the file handles within this Fv from cachable data.
169 //
170 Private->Fv[Private->CurrentPeimFvCount].ScanFv = TRUE;
171 CopyMem (Private->Fv[Private->CurrentPeimFvCount].FvFileHandles, Private->CurrentFvFileHandles, sizeof (Private->CurrentFvFileHandles));
172
173 }
174
175 /**
176 Shadow PeiCore module from flash to installed memory.
177
178 @param PeiServices An indirect pointer to the EFI_PEI_SERVICES table published by the PEI Foundation.
179 @param PrivateInMem PeiCore's private data structure
180
181 @return PeiCore function address after shadowing.
182 **/
183 VOID*
184 ShadowPeiCore(
185 EFI_PEI_SERVICES **PeiServices,
186 PEI_CORE_INSTANCE *PrivateInMem
187 )
188 {
189 EFI_PEI_FILE_HANDLE PeiCoreFileHandle;
190 EFI_PHYSICAL_ADDRESS EntryPoint;
191 EFI_STATUS Status;
192 UINT32 AuthenticationState;
193
194 PeiCoreFileHandle = NULL;
195
196 //
197 // Find the PEI Core in the BFV
198 //
199 Status = PeiFindFileEx (
200 (EFI_PEI_FV_HANDLE)PrivateInMem->Fv[0].FvHeader,
201 NULL,
202 EFI_FV_FILETYPE_PEI_CORE,
203 &PeiCoreFileHandle,
204 NULL
205 );
206 ASSERT_EFI_ERROR (Status);
207
208 //
209 // Shadow PEI Core into memory so it will run faster
210 //
211 Status = PeiLoadImage (
212 PeiServices,
213 *((EFI_PEI_FILE_HANDLE*)&PeiCoreFileHandle),
214 &EntryPoint,
215 &AuthenticationState
216 );
217 ASSERT_EFI_ERROR (Status);
218
219 //
220 // Compute the PeiCore's function address after shaowed PeiCore.
221 // _ModuleEntryPoint is PeiCore main function entry
222 //
223 return (VOID*) ((UINTN) EntryPoint + (UINTN) PeiCore - (UINTN) _ModuleEntryPoint);
224 }
225
226 /**
227 Conduct PEIM dispatch.
228
229 @param SecCoreData Points to a data structure containing information about the PEI core's operating
230 environment, such as the size and location of temporary RAM, the stack location and
231 the BFV location.
232 @param Private Pointer to the private data passed in from caller
233
234 @retval EFI_SUCCESS - Successfully dispatched PEIM.
235 @retval EFI_NOT_FOUND - The dispatch failed.
236
237 **/
238 VOID
239 PeiDispatcher (
240 IN CONST EFI_SEC_PEI_HAND_OFF *SecCoreData,
241 IN PEI_CORE_INSTANCE *Private
242 )
243 {
244 EFI_STATUS Status;
245 UINT32 Index1;
246 UINT32 Index2;
247 EFI_PEI_SERVICES **PeiServices;
248 EFI_PEI_FV_HANDLE VolumeHandle;
249 EFI_PEI_FILE_HANDLE PeimFileHandle;
250 UINTN FvCount;
251 UINTN PeimCount;
252 UINT32 AuthenticationState;
253 EFI_PHYSICAL_ADDRESS EntryPoint;
254 EFI_PEIM_ENTRY_POINT2 PeimEntryPoint;
255 UINTN SaveCurrentPeimCount;
256 UINTN SaveCurrentFvCount;
257 EFI_PEI_FILE_HANDLE SaveCurrentFileHandle;
258 PEIM_FILE_HANDLE_EXTENDED_DATA ExtendedData;
259 EFI_PHYSICAL_ADDRESS NewPermenentMemoryBase;
260 TEMPORARY_RAM_SUPPORT_PPI *TemporaryRamSupportPpi;
261 EFI_HOB_HANDOFF_INFO_TABLE *OldHandOffTable;
262 EFI_HOB_HANDOFF_INFO_TABLE *NewHandOffTable;
263 INTN StackOffset;
264 INTN HeapOffset;
265 PEI_CORE_INSTANCE *PrivateInMem;
266 UINT64 NewPeiStackSize;
267 UINT64 OldPeiStackSize;
268 UINT64 StackGap;
269 EFI_FV_FILE_INFO FvFileInfo;
270 UINTN OldCheckingTop;
271 UINTN OldCheckingBottom;
272
273
274 PeiServices = &Private->PS;
275 PeimEntryPoint = NULL;
276 PeimFileHandle = NULL;
277 EntryPoint = 0;
278
279 if ((Private->PeiMemoryInstalled) && (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME)) {
280 //
281 // Once real memory is available, shadow the RegisterForShadow modules. And meanwhile
282 // update the modules' status from PEIM_STATE_REGISITER_FOR_SHADOW to PEIM_STATE_DONE.
283 //
284 SaveCurrentPeimCount = Private->CurrentPeimCount;
285 SaveCurrentFvCount = Private->CurrentPeimFvCount;
286 SaveCurrentFileHandle = Private->CurrentFileHandle;
287
288 for (Index1 = 0; Index1 <= SaveCurrentFvCount; Index1++) {
289 for (Index2 = 0; (Index2 < FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv)) && (Private->Fv[Index1].FvFileHandles[Index2] != NULL); Index2++) {
290 if (Private->Fv[Index1].PeimState[Index2] == PEIM_STATE_REGISITER_FOR_SHADOW) {
291 PeimFileHandle = Private->Fv[Index1].FvFileHandles[Index2];
292 Status = PeiLoadImage (
293 &Private->PS,
294 PeimFileHandle,
295 &EntryPoint,
296 &AuthenticationState
297 );
298 if (Status == EFI_SUCCESS) {
299 //
300 // PEIM_STATE_REGISITER_FOR_SHADOW move to PEIM_STATE_DONE
301 //
302 Private->Fv[Index1].PeimState[Index2]++;
303 Private->CurrentFileHandle = PeimFileHandle;
304 Private->CurrentPeimFvCount = Index1;
305 Private->CurrentPeimCount = Index2;
306 //
307 // Call the PEIM entry point
308 //
309 PeimEntryPoint = (EFI_PEIM_ENTRY_POINT2)(UINTN)EntryPoint;
310
311 PERF_START (0, "PEIM", NULL, 0);
312 PeimEntryPoint(PeimFileHandle, (const EFI_PEI_SERVICES **) &Private->PS);
313 PERF_END (0, "PEIM", NULL, 0);
314 }
315
316 //
317 // Process the Notify list and dispatch any notifies for
318 // newly installed PPIs.
319 //
320 ProcessNotifyList (Private);
321 }
322 }
323 }
324 Private->CurrentFileHandle = SaveCurrentFileHandle;
325 Private->CurrentPeimFvCount = SaveCurrentFvCount;
326 Private->CurrentPeimCount = SaveCurrentPeimCount;
327 }
328
329 //
330 // This is the main dispatch loop. It will search known FVs for PEIMs and
331 // attempt to dispatch them. If any PEIM gets dispatched through a single
332 // pass of the dispatcher, it will start over from the Bfv again to see
333 // if any new PEIMs dependencies got satisfied. With a well ordered
334 // FV where PEIMs are found in the order their dependencies are also
335 // satisfied, this dipatcher should run only once.
336 //
337 do {
338 //
339 // In case that reenter PeiCore happens, the last pass record is still available.
340 //
341 if (!Private->PeimDispatcherReenter) {
342 Private->PeimNeedingDispatch = FALSE;
343 Private->PeimDispatchOnThisPass = FALSE;
344 } else {
345 Private->PeimDispatcherReenter = FALSE;
346 }
347
348 for (FvCount = Private->CurrentPeimFvCount; FvCount < Private->FvCount; FvCount++) {
349 Private->CurrentPeimFvCount = FvCount;
350 //
351 // Get this Fv Handle by PeiService FvFindNextVolume.
352 //
353 PeiFvFindNextVolume ((CONST EFI_PEI_SERVICES **) PeiServices, FvCount, &VolumeHandle);
354
355 if (Private->CurrentPeimCount == 0) {
356 //
357 // When going through each FV, at first, search Apriori file to
358 // reorder all PEIMs to ensure the PEIMs in Apriori file to get
359 // dispatch at first.
360 //
361 DiscoverPeimsAndOrderWithApriori (Private, VolumeHandle);
362 }
363
364 //
365 // Start to dispatch all modules within the current Fv.
366 //
367 for (PeimCount = Private->CurrentPeimCount;
368 (PeimCount < FixedPcdGet32 (PcdPeiCoreMaxPeimPerFv)) && (Private->CurrentFvFileHandles[PeimCount] != NULL);
369 PeimCount++) {
370 Private->CurrentPeimCount = PeimCount;
371 PeimFileHandle = Private->CurrentFileHandle = Private->CurrentFvFileHandles[PeimCount];
372
373 if (Private->Fv[FvCount].PeimState[PeimCount] == PEIM_STATE_NOT_DISPATCHED) {
374 if (!DepexSatisfied (Private, PeimFileHandle, PeimCount)) {
375 Private->PeimNeedingDispatch = TRUE;
376 } else {
377 Status = PeiFfsGetFileInfo (PeimFileHandle, &FvFileInfo);
378 ASSERT_EFI_ERROR (Status);
379 if (FvFileInfo.FileType == EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE) {
380 //
381 // For Fv type file, Produce new FV PPI and FV hob
382 //
383 Status = ProcessFvFile (PeiServices, PeimFileHandle, &AuthenticationState);
384 } else {
385 //
386 // For PEIM driver, Load its entry point
387 //
388 Status = PeiLoadImage (
389 PeiServices,
390 PeimFileHandle,
391 &EntryPoint,
392 &AuthenticationState
393 );
394 }
395
396 if ((Status == EFI_SUCCESS)) {
397 //
398 // The PEIM has its dependencies satisfied, and its entry point
399 // has been found, so invoke it.
400 //
401 PERF_START (0, "PEIM", NULL, 0);
402
403 ExtendedData.Handle = (EFI_HANDLE)PeimFileHandle;
404
405 REPORT_STATUS_CODE_WITH_EXTENDED_DATA (
406 EFI_PROGRESS_CODE,
407 FixedPcdGet32(PcdStatusCodeValuePeimDispatch),
408 (VOID *)(&ExtendedData),
409 sizeof (ExtendedData)
410 );
411
412 Status = VerifyPeim (Private, VolumeHandle, PeimFileHandle);
413 if (Status != EFI_SECURITY_VIOLATION && (AuthenticationState == 0)) {
414 //
415 // PEIM_STATE_NOT_DISPATCHED move to PEIM_STATE_DISPATCHED
416 //
417 Private->Fv[FvCount].PeimState[PeimCount]++;
418
419 if (FvFileInfo.FileType != EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE) {
420 //
421 // Call the PEIM entry point for PEIM driver
422 //
423 PeimEntryPoint = (EFI_PEIM_ENTRY_POINT2)(UINTN)EntryPoint;
424 PeimEntryPoint (PeimFileHandle, (const EFI_PEI_SERVICES **) PeiServices);
425 }
426
427 Private->PeimDispatchOnThisPass = TRUE;
428 }
429
430 REPORT_STATUS_CODE_WITH_EXTENDED_DATA (
431 EFI_PROGRESS_CODE,
432 FixedPcdGet32(PcdStatusCodeValuePeimDispatch),
433 (VOID *)(&ExtendedData),
434 sizeof (ExtendedData)
435 );
436 PERF_END (0, "PEIM", NULL, 0);
437
438 }
439
440 if (Private->SwitchStackSignal) {
441 //
442 // Before switch stack from CAR to permenent memory, caculate the heap and stack
443 // usage in temporary memory for debuging.
444 //
445 DEBUG_CODE_BEGIN ();
446 UINT32 *StackPointer;
447
448 for (StackPointer = (UINT32*)SecCoreData->StackBase;
449 (StackPointer < (UINT32*)((UINTN)SecCoreData->StackBase + SecCoreData->StackSize)) \
450 && (*StackPointer == INIT_CAR_VALUE);
451 StackPointer ++);
452
453 DEBUG ((EFI_D_INFO, "Total Cache as RAM: %d bytes.\n", SecCoreData->TemporaryRamSize));
454 DEBUG ((EFI_D_INFO, " CAR stack ever used: %d bytes.\n",
455 (SecCoreData->StackSize - ((UINTN) StackPointer - (UINTN)SecCoreData->StackBase))
456 ));
457 DEBUG ((EFI_D_INFO, " CAR heap used: %d bytes.\n",
458 ((UINTN) Private->HobList.HandoffInformationTable->EfiFreeMemoryBottom -
459 (UINTN) Private->HobList.Raw)
460 ));
461 DEBUG_CODE_END ();
462
463 //
464 // Reserve the size of new stack at bottom of physical memory
465 //
466 OldPeiStackSize = (UINT64) SecCoreData->StackSize;
467 NewPeiStackSize = (RShiftU64 (Private->PhysicalMemoryLength, 1) + EFI_PAGE_MASK) & ~EFI_PAGE_MASK;
468 if (FixedPcdGet32(PcdPeiCoreMaxPeiStackSize) > (UINT32) NewPeiStackSize) {
469 Private->StackSize = NewPeiStackSize;
470 } else {
471 Private->StackSize = FixedPcdGet32(PcdPeiCoreMaxPeiStackSize);
472 }
473
474 //
475 // In theory, the size of new stack in permenent memory should large than
476 // size of old stack in temporary memory.
477 // But if new stack is smaller than the size of old stack, we also reserve
478 // the size of old stack at bottom of permenent memory.
479 //
480 ASSERT (Private->StackSize >= OldPeiStackSize);
481 StackGap = Private->StackSize - OldPeiStackSize;
482
483 //
484 // Update HandOffHob for new installed permenent memory
485 //
486 OldHandOffTable = Private->HobList.HandoffInformationTable;
487 OldCheckingBottom = (UINTN)(SecCoreData->TemporaryRamBase);
488 OldCheckingTop = (UINTN)(OldCheckingBottom + SecCoreData->TemporaryRamSize);
489
490 //
491 // The whole temporary memory will be migrated to physical memory.
492 // CAUTION: The new base is computed accounding to gap of new stack.
493 //
494 NewPermenentMemoryBase = Private->PhysicalMemoryBegin + StackGap;
495
496 //
497 // Caculate stack offset and heap offset between CAR and new permement
498 // memory seperately.
499 //
500 StackOffset = (UINTN) NewPermenentMemoryBase - (UINTN) SecCoreData->StackBase;
501 HeapOffset = (INTN) ((UINTN) Private->PhysicalMemoryBegin + Private->StackSize - \
502 (UINTN) SecCoreData->PeiTemporaryRamBase);
503 DEBUG ((EFI_D_INFO, "Heap Offset = 0x%X Stack Offset = 0x%X\n", HeapOffset, StackOffset));
504
505 //
506 // Caculate new HandOffTable and PrivateData address in permenet memory's stack
507 //
508 NewHandOffTable = (EFI_HOB_HANDOFF_INFO_TABLE *)((UINTN)OldHandOffTable + HeapOffset);
509 PrivateInMem = (PEI_CORE_INSTANCE *)((UINTN) (VOID*) Private + StackOffset);
510
511 //
512 // TemporaryRamSupportPpi is produced by platform's SEC
513 //
514 Status = PeiLocatePpi (
515 (CONST EFI_PEI_SERVICES **) PeiServices,
516 &gEfiTemporaryRamSupportPpiGuid,
517 0,
518 NULL,
519 (VOID**)&TemporaryRamSupportPpi
520 );
521
522
523 if (!EFI_ERROR (Status)) {
524 //
525 // Temporary Ram support Ppi is provided by platform, it will copy
526 // temporary memory to permenent memory and do stack switching.
527 // After invoken temporary Ram support, following code's stack is in
528 // memory but not in CAR.
529 //
530 TemporaryRamSupportPpi->TemporaryRamMigration (
531 (CONST EFI_PEI_SERVICES **) PeiServices,
532 (EFI_PHYSICAL_ADDRESS)(UINTN) SecCoreData->TemporaryRamBase,
533 (EFI_PHYSICAL_ADDRESS)(UINTN) NewPermenentMemoryBase,
534 SecCoreData->TemporaryRamSize
535 );
536
537 } else {
538 //
539 // In IA32/x64/Itanium architecture, we need platform provide
540 // TEMPORAY_RAM_MIGRATION_PPI.
541 //
542 ASSERT (FALSE);
543 }
544
545
546 //
547 //
548 // Fixup the PeiCore's private data
549 //
550 PrivateInMem->PS = &PrivateInMem->ServiceTableShadow;
551 PrivateInMem->CpuIo = &PrivateInMem->ServiceTableShadow.CpuIo;
552 PrivateInMem->HobList.Raw = (VOID*) ((UINTN) PrivateInMem->HobList.Raw + HeapOffset);
553 PrivateInMem->StackBase = (EFI_PHYSICAL_ADDRESS)(((UINTN)PrivateInMem->PhysicalMemoryBegin + EFI_PAGE_MASK) & ~EFI_PAGE_MASK);
554
555 PeiServices = &PrivateInMem->PS;
556
557 //
558 // Fixup for PeiService's address
559 //
560 SetPeiServicesTablePointer(PeiServices);
561
562 //
563 // Update HandOffHob for new installed permenent memory
564 //
565 NewHandOffTable->EfiEndOfHobList =
566 (EFI_PHYSICAL_ADDRESS)((UINTN) NewHandOffTable->EfiEndOfHobList + HeapOffset);
567 NewHandOffTable->EfiMemoryTop = PrivateInMem->PhysicalMemoryBegin +
568 PrivateInMem->PhysicalMemoryLength;
569 NewHandOffTable->EfiMemoryBottom = PrivateInMem->PhysicalMemoryBegin;
570 NewHandOffTable->EfiFreeMemoryTop = PrivateInMem->FreePhysicalMemoryTop;
571 NewHandOffTable->EfiFreeMemoryBottom = NewHandOffTable->EfiEndOfHobList +
572 sizeof (EFI_HOB_GENERIC_HEADER);
573
574 //
575 // We need convert the PPI desciptor's pointer
576 //
577 ConvertPpiPointers (PrivateInMem,
578 OldCheckingBottom,
579 OldCheckingTop,
580 HeapOffset
581 );
582
583 DEBUG ((EFI_D_INFO, "Stack Hob: BaseAddress=0x%X Length=0x%X\n",
584 (UINTN)PrivateInMem->StackBase,
585 PrivateInMem->StackSize));
586 BuildStackHob (PrivateInMem->StackBase, PrivateInMem->StackSize);
587
588 //
589 // After the whole temporary memory is migrated, then we can allocate page in
590 // permenent memory.
591 //
592 PrivateInMem->PeiMemoryInstalled = TRUE;
593
594 //
595 // Indicate that PeiCore reenter
596 //
597 PrivateInMem->PeimDispatcherReenter = TRUE;
598
599 //
600 // Shadow PEI Core. When permanent memory is avaiable, shadow
601 // PEI Core and PEIMs to get high performance.
602 //
603 PrivateInMem->ShadowedPeiCore = ShadowPeiCore (
604 PeiServices,
605 PrivateInMem
606 );
607 //
608 // Process the Notify list and dispatch any notifies for
609 // newly installed PPIs.
610 //
611 ProcessNotifyList (PrivateInMem);
612
613 //
614 // Entry PEI Phase 2
615 //
616 PeiCore (SecCoreData, NULL, PrivateInMem);
617
618 //
619 // Code should not come here
620 //
621 ASSERT_EFI_ERROR(FALSE);
622 }
623
624 //
625 // Process the Notify list and dispatch any notifies for
626 // newly installed PPIs.
627 //
628 ProcessNotifyList (Private);
629
630 if ((Private->PeiMemoryInstalled) && (Private->Fv[FvCount].PeimState[PeimCount] == PEIM_STATE_REGISITER_FOR_SHADOW) && \
631 (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME)) {
632 //
633 // If memory is availble we shadow images by default for performance reasons.
634 // We call the entry point a 2nd time so the module knows it's shadowed.
635 //
636 //PERF_START (PeiServices, L"PEIM", PeimFileHandle, 0);
637 PeimEntryPoint (PeimFileHandle, (const EFI_PEI_SERVICES **) PeiServices);
638 //PERF_END (PeiServices, L"PEIM", PeimFileHandle, 0);
639
640 //
641 // PEIM_STATE_REGISITER_FOR_SHADOW move to PEIM_STATE_DONE
642 //
643 Private->Fv[FvCount].PeimState[PeimCount]++;
644
645 //
646 // Process the Notify list and dispatch any notifies for
647 // newly installed PPIs.
648 //
649 ProcessNotifyList (Private);
650 }
651 }
652 }
653 }
654
655 //
656 // We set to NULL here to optimize the 2nd entry to this routine after
657 // memory is found. This reprevents rescanning of the FV. We set to
658 // NULL here so we start at the begining of the next FV
659 //
660 Private->CurrentFileHandle = NULL;
661 Private->CurrentPeimCount = 0;
662 //
663 // Before walking through the next FV,Private->CurrentFvFileHandles[]should set to NULL
664 //
665 SetMem (Private->CurrentFvFileHandles, sizeof (Private->CurrentFvFileHandles), 0);
666 }
667
668 //
669 // Before making another pass, we should set Private->CurrentPeimFvCount =0 to go
670 // through all the FV.
671 //
672 Private->CurrentPeimFvCount = 0;
673
674 //
675 // PeimNeedingDispatch being TRUE means we found a PEIM that did not get
676 // dispatched. So we need to make another pass
677 //
678 // PeimDispatchOnThisPass being TRUE means we dispatched a PEIM on this
679 // pass. If we did not dispatch a PEIM there is no point in trying again
680 // as it will fail the next time too (nothing has changed).
681 //
682 } while (Private->PeimNeedingDispatch && Private->PeimDispatchOnThisPass);
683
684 }
685
686 /**
687 Initialize the Dispatcher's data members
688
689 @param PrivateData PeiCore's private data structure
690 @param OldCoreData Old data from SecCore
691 NULL if being run in non-permament memory mode.
692 @param SecCoreData Points to a data structure containing information about the PEI core's operating
693 environment, such as the size and location of temporary RAM, the stack location and
694 the BFV location.
695
696 @return None.
697
698 **/
699 VOID
700 InitializeDispatcherData (
701 IN PEI_CORE_INSTANCE *PrivateData,
702 IN PEI_CORE_INSTANCE *OldCoreData,
703 IN CONST EFI_SEC_PEI_HAND_OFF *SecCoreData
704 )
705 {
706 if (OldCoreData == NULL) {
707 PrivateData->PeimDispatcherReenter = FALSE;
708 PeiInitializeFv (PrivateData, SecCoreData);
709 }
710
711 return;
712 }
713
714 /**
715 This routine parses the Dependency Expression, if available, and
716 decides if the module can be executed.
717
718
719 @param Private PeiCore's private data structure
720 @param FileHandle PEIM's file handle
721 @param PeimCount Peim count in all dispatched PEIMs.
722
723 @retval TRUE Can be dispatched
724 @retval FALSE Cannot be dispatched
725
726 **/
727 BOOLEAN
728 DepexSatisfied (
729 IN PEI_CORE_INSTANCE *Private,
730 IN EFI_PEI_FILE_HANDLE FileHandle,
731 IN UINTN PeimCount
732 )
733 {
734 EFI_STATUS Status;
735 VOID *DepexData;
736
737 if (PeimCount < Private->AprioriCount) {
738 //
739 // If its in the A priori file then we set Depex to TRUE
740 //
741 return TRUE;
742 }
743
744 //
745 // Depex section not in the encapsulated section.
746 //
747 Status = PeiServicesFfsFindSectionData (
748 EFI_SECTION_PEI_DEPEX,
749 FileHandle,
750 (VOID **)&DepexData
751 );
752
753 if (EFI_ERROR (Status)) {
754 //
755 // If there is no DEPEX, assume the module can be executed
756 //
757 return TRUE;
758 }
759
760 //
761 // Evaluate a given DEPEX
762 //
763 return PeimDispatchReadiness (&Private->PS, DepexData);
764 }
765
766 /**
767 This routine enable a PEIM to register itself to shadow when PEI Foundation
768 discovery permanent memory.
769
770 @param FileHandle File handle of a PEIM.
771
772 @retval EFI_NOT_FOUND The file handle doesn't point to PEIM itself.
773 @retval EFI_ALREADY_STARTED Indicate that the PEIM has been registered itself.
774 @retval EFI_SUCCESS Successfully to register itself.
775
776 **/
777 EFI_STATUS
778 EFIAPI
779 PeiRegisterForShadow (
780 IN EFI_PEI_FILE_HANDLE FileHandle
781 )
782 {
783 PEI_CORE_INSTANCE *Private;
784 Private = PEI_CORE_INSTANCE_FROM_PS_THIS (GetPeiServicesTablePointer ());
785
786 if (Private->CurrentFileHandle != FileHandle) {
787 //
788 // The FileHandle must be for the current PEIM
789 //
790 return EFI_NOT_FOUND;
791 }
792
793 if (Private->Fv[Private->CurrentPeimFvCount].PeimState[Private->CurrentPeimCount] >= PEIM_STATE_REGISITER_FOR_SHADOW) {
794 //
795 // If the PEIM has already entered the PEIM_STATE_REGISTER_FOR_SHADOW or PEIM_STATE_DONE then it's already been started
796 //
797 return EFI_ALREADY_STARTED;
798 }
799
800 Private->Fv[Private->CurrentPeimFvCount].PeimState[Private->CurrentPeimCount] = PEIM_STATE_REGISITER_FOR_SHADOW;
801
802 return EFI_SUCCESS;
803 }
804