Add in some features for the thunk layer:

[mirror_edk2.git] / EdkCompatibilityPkg / Compatibility / FrameworkHiiToUefiHiiThunk / HiiDatabase.c

diff --git a/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/HiiDatabase.c b/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/HiiDatabase.c
index d5ef19c356aeea52e3ba4963e4e28a8a58cd4465..e2e2ded7c3ee20f85360168f56826895f83fa0e1 100644 (file)
--- a/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/HiiDatabase.c
+++ b/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/HiiDatabase.c
@@ -56,10 +56,12 @@ EFI_HII_THUNK_PRIVATE_DATA HiiThunkPrivateDataTempate = {
   },\r
 };\r
 \r
-EFI_HII_DATABASE_PROTOCOL *mUefiHiiDatabaseProtocol;\r
-EFI_HII_FONT_PROTOCOL     *mUefiHiiFontProtocol;\r
-EFI_HII_IMAGE_PROTOCOL    *mUefiHiiImageProtocol;\r
-EFI_HII_STRING_PROTOCOL   *mUefiStringProtocol;\r
+CONST EFI_HII_DATABASE_PROTOCOL            *mUefiHiiDatabaseProtocol;\r
+CONST EFI_HII_FONT_PROTOCOL                *mUefiHiiFontProtocol;\r
+CONST EFI_HII_IMAGE_PROTOCOL               *mUefiHiiImageProtocol;\r
+CONST EFI_HII_STRING_PROTOCOL              *mUefiStringProtocol;\r
+CONST EFI_HII_CONFIG_ROUTING_PROTOCOL      *mUefiConfigRoutingProtocol;\r
+\r
 \r
 EFI_STATUS\r
 EFIAPI\r
@@ -119,6 +121,13 @@ Returns:
                   );\r
   ASSERT_EFI_ERROR (Status);\r
 \r
+  Status = gBS->LocateProtocol (\r
+                  &gEfiHiiConfigRoutingProtocolGuid,\r
+                  NULL,\r
+                  (VOID **) &mUefiConfigRoutingProtocol\r
+                  );\r
+  ASSERT_EFI_ERROR (Status);\r
+\r
   //\r
   // Install protocol interface\r
   //\r