From 806b6f31d9e43ed798938a4a9dc900bdca008c70 Mon Sep 17 00:00:00 2001
From: qhuang8 <qhuang8@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Fri, 28 Sep 2007 09:12:52 +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@3970 6f19259b-4bc3-4df7-8a09-765794883524
---
 Nt32Pkg/SnpNt32Dxe/SnpNt32.c      | 5 ++---
 Nt32Pkg/SnpNt32Dxe/SnpNt32Dxe.inf | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

(limited to 'Nt32Pkg')

diff --git a/Nt32Pkg/SnpNt32Dxe/SnpNt32.c b/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
index 67a3ac0038..6559d316e7 100644
--- a/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
+++ b/Nt32Pkg/SnpNt32Dxe/SnpNt32.c
@@ -1018,15 +1018,14 @@ Returns:
   // Install the Driver Protocols
   //
 
-  Status = NetLibInstallAllDriverProtocolsWithUnload (
+  Status = EfiLibInstallAllDriverProtocols (
             ImageHandle,
             SystemTable,
             &gSnpNt32DriverBinding,
             ImageHandle,
             &gSnpNt32DriverComponentName,
             NULL,
-            NULL,
-            SnpNt32Unload
+            NULL
             );
   if (EFI_ERROR (Status)) {
     return Status;
diff --git a/Nt32Pkg/SnpNt32Dxe/SnpNt32Dxe.inf b/Nt32Pkg/SnpNt32Dxe/SnpNt32Dxe.inf
index 2896a83c34..abc0644db1 100644
--- a/Nt32Pkg/SnpNt32Dxe/SnpNt32Dxe.inf
+++ b/Nt32Pkg/SnpNt32Dxe/SnpNt32Dxe.inf
@@ -24,7 +24,7 @@
   EFI_SPECIFICATION_VERSION      = 0x00020000
 
   ENTRY_POINT                    = InitializeSnpNt32river
-
+  UNLOAD_IMAGE                   = SnpNt32Unload
 #
 # The following information is for reference only and not required by the build tools.
 #
-- 
cgit v1.2.3