CpuFeaturesDxe.c 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
  1. /** @file
  2. CPU Features DXE driver to initialize CPU features.
  3. Copyright (c) 2017 - 2018, Intel Corporation. All rights reserved.<BR>
  4. SPDX-License-Identifier: BSD-2-Clause-Patent
  5. **/
  6. #include <PiDxe.h>
  7. #include <Library/BaseLib.h>
  8. #include <Library/DebugLib.h>
  9. #include <Library/UefiLib.h>
  10. #include <Library/UefiBootServicesTableLib.h>
  11. #include <Library/RegisterCpuFeaturesLib.h>
  12. #include <Library/HobLib.h>
  13. #include <Protocol/SmmConfiguration.h>
  14. #include <Guid/CpuFeaturesInitDone.h>
  15. /**
  16. Worker function to perform CPU feature initialization.
  17. **/
  18. VOID
  19. CpuFeaturesInitializeWorker (
  20. VOID
  21. )
  22. {
  23. EFI_STATUS Status;
  24. EFI_HANDLE Handle;
  25. CpuFeaturesDetect ();
  26. CpuFeaturesInitialize ();
  27. //
  28. // Install CPU Features Init Done Protocol
  29. //
  30. Handle = NULL;
  31. Status = gBS->InstallProtocolInterface (
  32. &Handle,
  33. &gEdkiiCpuFeaturesInitDoneGuid,
  34. EFI_NATIVE_INTERFACE,
  35. NULL
  36. );
  37. ASSERT_EFI_ERROR (Status);
  38. }
  39. /**
  40. Event notification that initialize CPU features when gEfiSmmConfigurationProtocol installs.
  41. @param[in] Event The Event that is being processed, not used.
  42. @param[in] Context Event Context, not used.
  43. **/
  44. VOID
  45. EFIAPI
  46. SmmConfigurationEventNotify (
  47. IN EFI_EVENT Event,
  48. IN VOID *Context
  49. )
  50. {
  51. EFI_STATUS Status;
  52. EFI_SMM_CONFIGURATION_PROTOCOL *SmmConfiguration;
  53. //
  54. // Make sure this notification is for this handler
  55. //
  56. Status = gBS->LocateProtocol (&gEfiSmmConfigurationProtocolGuid, NULL, (VOID **)&SmmConfiguration);
  57. if (EFI_ERROR (Status)) {
  58. return;
  59. }
  60. CpuFeaturesInitializeWorker ();
  61. }
  62. /**
  63. CPU Features driver entry point function.
  64. If PcdCpuFeaturesInitAfterSmmRelocation is TRUE, it will register one
  65. SMM Configuration Protocol notify function to perform CPU features
  66. initialization. Otherwise, it will perform CPU features initialization
  67. directly.
  68. @param ImageHandle Image handle this driver.
  69. @param SystemTable Pointer to the System Table.
  70. @retval EFI_SUCCESS CPU Features is initialized successfully.
  71. **/
  72. EFI_STATUS
  73. EFIAPI
  74. CpuFeaturesDxeInitialize (
  75. IN EFI_HANDLE ImageHandle,
  76. IN EFI_SYSTEM_TABLE *SystemTable
  77. )
  78. {
  79. VOID *Registration;
  80. EFI_STATUS Status;
  81. EFI_HANDLE Handle;
  82. if (GetFirstGuidHob (&gEdkiiCpuFeaturesInitDoneGuid) != NULL) {
  83. //
  84. // Try to find HOB first. This HOB exist means CPU features have
  85. // been initialized by CpuFeaturesPei driver, just install
  86. // gEdkiiCpuFeaturesInitDoneGuid.
  87. //
  88. Handle = NULL;
  89. Status = gBS->InstallProtocolInterface (
  90. &Handle,
  91. &gEdkiiCpuFeaturesInitDoneGuid,
  92. EFI_NATIVE_INTERFACE,
  93. NULL
  94. );
  95. ASSERT_EFI_ERROR (Status);
  96. return Status;
  97. }
  98. if (PcdGetBool (PcdCpuFeaturesInitAfterSmmRelocation)) {
  99. //
  100. // Install notification callback on SMM Configuration Protocol
  101. //
  102. EfiCreateProtocolNotifyEvent (
  103. &gEfiSmmConfigurationProtocolGuid,
  104. TPL_CALLBACK,
  105. SmmConfigurationEventNotify,
  106. NULL,
  107. &Registration
  108. );
  109. } else {
  110. CpuFeaturesInitializeWorker ();
  111. }
  112. return EFI_SUCCESS;
  113. }