From f3de0a607dd36d045ef8d1782db5e9c296cf6b43 Mon Sep 17 00:00:00 2001
From: xdu2 <xdu2@6f19259b-4bc3-4df7-8a09-765794883524>
Date: Mon, 4 Jul 2011 07:33:34 +0000
Subject: [PATCH] MdeModulePkg: Add unload support for SnpDxe and fix NULL
 ImageHandle in EfiLibInstallDriverBindingComponentName2() in driver entry
 point.

Signed-off-by: xdu2
Reviewed-by: vanjeff

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@11967 6f19259b-4bc3-4df7-8a09-765794883524
---
 MdeModulePkg/Universal/Network/SnpDxe/Snp.c      | 4 ++--
 MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
index e90e4e0244..2dc0536117 100644
--- a/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
+++ b/MdeModulePkg/Universal/Network/SnpDxe/Snp.c
@@ -1,7 +1,7 @@
 /** @file
   Implementation of driver entry point and driver binding protocol.
 
-Copyright (c) 2004 - 2010, Intel Corporation. All rights reserved.<BR>
+Copyright (c) 2004 - 2011, Intel Corporation. All rights reserved.<BR>
 This program and the accompanying materials are licensed
 and made available under the terms and conditions of the BSD License which
 accompanies this distribution. The full text of the license may be found at
@@ -1014,7 +1014,7 @@ InitializeSnpNiiDriver (
            ImageHandle,
            SystemTable,
            &mSimpleNetworkDriverBinding,
-           NULL,
+           ImageHandle,
            &gSimpleNetworkComponentName,
            &gSimpleNetworkComponentName2
            );
diff --git a/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf b/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
index 70dd3eeca7..a7f81c83c5 100644
--- a/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
+++ b/MdeModulePkg/Universal/Network/SnpDxe/SnpDxe.inf
@@ -1,7 +1,7 @@
 ## @file
 #    Component description file for SNP module.
 #
-# 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 of the BSD License which 
 # accompanies this distribution. The full text of the license may be found at 
@@ -19,6 +19,7 @@
   MODULE_TYPE                    = UEFI_DRIVER
   VERSION_STRING                 = 1.0
   ENTRY_POINT                    = InitializeSnpNiiDriver
+  UNLOAD_IMAGE                   = NetLibDefaultUnload
 
 #
 # The following information is for reference only and not required by the build tools.
@@ -63,7 +64,7 @@
   UefiDriverEntryPoint
   BaseMemoryLib
   DebugLib
-
+  NetLib
 
 [Guids]
   gEfiEventExitBootServicesGuid                 ## CONSUMES
-- 
2.39.2