From da1d02010d73465a9c2b21f619e8c01cb00bd1fa Mon Sep 17 00:00:00 2001
From: qhuang8 <qhuang8@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Fri, 28 Sep 2007 06:02:01 +0000
Subject: Clear up network stack to use R9 MdeLib: 1.Use driver model APIs in
 UefiLib 2.Register Unload Image in INF/MSA to be included in autogen code.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@3962 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c | 4 ++--
 MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.inf  | 2 +-
 MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.msa  | 1 +
 3 files changed, 4 insertions(+), 3 deletions(-)

(limited to 'MdeModulePkg/Universal/Network/Tcp4Dxe')

diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
index 1eefb38f91..bbd5a30a7b 100644
--- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
+++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.c
@@ -176,7 +176,7 @@ Returns:
   //
   // Install the TCP4 Driver Binding Protocol
   //
-  Status = NetLibInstallAllDriverProtocols (
+  Status = EfiLibInstallAllDriverProtocols (
              ImageHandle,
              SystemTable,
              &mTcp4DriverBinding,
@@ -185,7 +185,7 @@ Returns:
              NULL,
              NULL
              );
-
+  ASSERT_EFI_ERROR (Status);
   //
   // Initialize ISS and random port.
   //
diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.inf b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.inf
index 86c88a9506..6e013f79f0 100644
--- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.inf
+++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.inf
@@ -25,7 +25,7 @@
   EFI_SPECIFICATION_VERSION      = 0x00020000
 
   ENTRY_POINT                    = Tcp4DriverEntryPoint
-
+  UNLOAD_IMAGE                   = NetLibDefaultUnload
 #
 # The following information is for reference only and not required by the build tools.
 #
diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.msa b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.msa
index 4285d62c08..7fca1f5872 100644
--- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.msa
+++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Dxe.msa
@@ -85,6 +85,7 @@
     <Specification>EDK_RELEASE_VERSION 0x00020000</Specification>
     <Extern>
       <ModuleEntryPoint>Tcp4DriverEntryPoint</ModuleEntryPoint>
+      <ModuleUnloadImage>NetLibDefaultUnload</ModuleUnloadImage>
     </Extern>
   </Externs>
 </ModuleSurfaceArea>
\ No newline at end of file
-- 
cgit v1.2.3