From 6ac6017fe8cb4478aee135799c3b5f9862935250 Mon Sep 17 00:00:00 2001
From: erictian <erictian@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Fri, 23 Sep 2011 05:17:31 +0000
Subject: error handling enhancement for possible null pointer dereference

Signed-off-by: erictian
Reviewed-by: ydong10


git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@12434 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Bus/Usb/UsbBusPei/UsbPeim.c | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

(limited to 'MdeModulePkg/Bus/Usb/UsbBusPei')

diff --git a/MdeModulePkg/Bus/Usb/UsbBusPei/UsbPeim.c b/MdeModulePkg/Bus/Usb/UsbBusPei/UsbPeim.c
index 4a5d8e8bc4..b63173d181 100644
--- a/MdeModulePkg/Bus/Usb/UsbBusPei/UsbPeim.c
+++ b/MdeModulePkg/Bus/Usb/UsbBusPei/UsbPeim.c
@@ -1,7 +1,7 @@
 /** @file
 The module to produce Usb Bus PPI.
 
-Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>
+Copyright (c) 2006 - 2011, Intel Corporation. All rights reserved.<BR>
   
 This program and the accompanying materials
 are licensed and made available under the terms and conditions
@@ -307,13 +307,13 @@ PeiHubEnumeration (
         NewPeiUsbDevice->DeviceSpeed = (UINT8)IsPortLowSpeedDeviceAttached (PortStatus.PortStatus);
 
         if(NewPeiUsbDevice->DeviceSpeed != EFI_USB_SPEED_HIGH) {
-    	    if (PeiUsbDevice->DeviceSpeed == EFI_USB_SPEED_HIGH) {
-             NewPeiUsbDevice->Translator.TranslatorPortNumber = (UINT8)Index;
-             NewPeiUsbDevice->Translator.TranslatorHubAddress = *CurrentAddress;
-    	    } else {
-    	        CopyMem(&(NewPeiUsbDevice->Translator), &(PeiUsbDevice->Translator), sizeof(EFI_USB2_HC_TRANSACTION_TRANSLATOR));
+          if (PeiUsbDevice->DeviceSpeed == EFI_USB_SPEED_HIGH) {
+            NewPeiUsbDevice->Translator.TranslatorPortNumber = (UINT8)Index;
+            NewPeiUsbDevice->Translator.TranslatorHubAddress = *CurrentAddress;
+          } else {
+            CopyMem(&(NewPeiUsbDevice->Translator), &(PeiUsbDevice->Translator), sizeof(EFI_USB2_HC_TRANSACTION_TRANSLATOR));
           }
-      	}
+        }
 
         //
         // Configure that Usb Device
@@ -380,18 +380,21 @@ PeiUsbEnumeration (
 
 
   CurrentAddress = 0;
-  if (Usb2HcPpi != NULL){
+  if (Usb2HcPpi != NULL) {
     Usb2HcPpi->GetRootHubPortNumber (
                 PeiServices,
                 Usb2HcPpi,
                 (UINT8 *) &NumOfRootPort
                 );    
-  } else {
+  } else if (UsbHcPpi != NULL) {
     UsbHcPpi->GetRootHubPortNumber (
                 PeiServices,
                 UsbHcPpi,
                 (UINT8 *) &NumOfRootPort
                 );
+  } else {
+    ASSERT (FALSE);
+    return EFI_INVALID_PARAMETER;
   }
 
   for (Index = 0; Index < NumOfRootPort; Index++) {
-- 
cgit v1.2.3