MdeModulePkg XhciPei: Minor refinement about IoMmu

1. Call IoMmuInit() after locating gPeiUsbControllerPpiGuid.
2. Call XhcPeiFreeSched() to do cleanup in XhcEndOfPei.

Cc: Jiewen Yao <jiewen.yao@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Star Zeng <star.zeng@intel.com>
Reviewed-by: Jiewen Yao <jiewen.yao@intel.com>

diff --git a/MdeModulePkg/Bus/Pci/XhciPei/XhcPeim.c b/MdeModulePkg/Bus/Pci/XhciPei/XhcPeim.c
index 99f69f730b3fe4d052b24ba79db75e16a9157629..c5631e87cacc1e4e0a4e38a90603d77a1bb19b8b 100644 (file)
--- a/MdeModulePkg/Bus/Pci/XhciPei/XhcPeim.c
+++ b/MdeModulePkg/Bus/Pci/XhciPei/XhcPeim.c
@@ -1427,6 +1427,8 @@ XhcEndOfPei (
 \r
   XhcPeiHaltHC (Xhc, XHC_GENERIC_TIMEOUT);\r
 \r
+  XhcPeiFreeSched (Xhc);\r
+\r
   return EFI_SUCCESS;\r
 }\r
 \r
@@ -1461,8 +1463,6 @@ XhcPeimEntry (
     return EFI_SUCCESS;\r
   }\r
 \r
-  IoMmuInit ();\r
-\r
   Status = PeiServicesLocatePpi (\r
              &gPeiUsbControllerPpiGuid,\r
              0,\r
@@ -1473,6 +1473,8 @@ XhcPeimEntry (
     return EFI_UNSUPPORTED;\r
   }\r
 \r
+  IoMmuInit ();\r
+\r
   Index = 0;\r
   while (TRUE) {\r
     Status = UsbControllerPpi->GetUsbController (\r