From 0bf47d3ddf96a69244f9d0c605e5dd5d76a64e9f Mon Sep 17 00:00:00 2001
From: xdu2 <xdu2@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Tue, 2 Feb 2010 11:18:33 +0000
Subject: 1. add error handling in case DHCP success but boot request fail 2.
 add UnloadImage to ease debug

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@9910 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c      | 7 +++++++
 MdeModulePkg/Universal/Network/UefiPxeBcDxe/UefiPxeBcDxe.inf | 1 +
 2 files changed, 8 insertions(+)

(limited to 'MdeModulePkg')

diff --git a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c
index d2e60b6a59..f406c507c3 100644
--- a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c
+++ b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/PxeBcDhcp.c
@@ -1217,6 +1217,13 @@ PxeBcDiscvBootService (
     }
   }
 
+  if (TryIndex > PXEBC_BOOT_REQUEST_RETRIES) {
+    //
+    // No server response our PXE request
+    //
+    Status = EFI_TIMEOUT;
+  }
+
   if (!EFI_ERROR (Status)) {
     //
     // Find Pxe Reply
diff --git a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/UefiPxeBcDxe.inf b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/UefiPxeBcDxe.inf
index eefc5ac666..4df333e1a6 100644
--- a/MdeModulePkg/Universal/Network/UefiPxeBcDxe/UefiPxeBcDxe.inf
+++ b/MdeModulePkg/Universal/Network/UefiPxeBcDxe/UefiPxeBcDxe.inf
@@ -22,6 +22,7 @@
   MODULE_TYPE                    = UEFI_DRIVER
   VERSION_STRING                 = 1.0
   ENTRY_POINT                    = PxeBcDriverEntryPoint
+  UNLOAD_IMAGE                   = NetLibDefaultUnload
 
 #
 # The following information is for reference only and not required by the build tools.
-- 
cgit v1.2.3