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