]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/SnpDxe/Snp.c
BaseTools:Change the path of the file that Binary Cache
[mirror_edk2.git] / MdeModulePkg / Universal / Network / SnpDxe / Snp.c
index e01a135708bd2eab013fe67e79c06e875156aa45..a23af05078bc8939c7ea42dc83f2d26ad9489888 100644 (file)
 /** @file\r
-Copyright (c) 2004 - 2005, Intel Corporation\r
-All rights reserved. This program and the accompanying materials\r
-are licensed and made available under the terms and conditions of the BSD License\r
-which accompanies this distribution.  The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
-\r
-THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
-WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
-\r
-Module name:\r
-    snp.c\r
-\r
-Abstract:\r
+  Implementation of driver entry point and driver binding protocol.\r
 \r
+Copyright (c) 2004 - 2018, Intel Corporation. All rights reserved.<BR>\r
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
 #include "Snp.h"\r
 \r
-EFI_STATUS\r
-pxe_start (\r
-  SNP_DRIVER *snp\r
-  );\r
-EFI_STATUS\r
-pxe_stop (\r
-  SNP_DRIVER *snp\r
-  );\r
-EFI_STATUS\r
-pxe_init (\r
-  SNP_DRIVER *snp,\r
-  UINT16     OpFlags\r
-  );\r
-EFI_STATUS\r
-pxe_shutdown (\r
-  SNP_DRIVER *snp\r
-  );\r
-EFI_STATUS\r
-pxe_get_stn_addr (\r
-  SNP_DRIVER *snp\r
-  );\r
-\r
-EFI_STATUS\r
-EFIAPI\r
-InitializeSnpNiiDriver (\r
-  IN EFI_HANDLE       image_handle,\r
-  IN EFI_SYSTEM_TABLE *system_table\r
-  );\r
-\r
-EFI_STATUS\r
-EFIAPI\r
-SimpleNetworkDriverSupported (\r
-  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
-  IN EFI_HANDLE                     Controller,\r
-  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
-  );\r
-\r
-EFI_STATUS\r
-EFIAPI\r
-SimpleNetworkDriverStart (\r
-  IN EFI_DRIVER_BINDING_PROTOCOL    *This,\r
-  IN EFI_HANDLE                     Controller,\r
-  IN EFI_DEVICE_PATH_PROTOCOL       *RemainingDevicePath\r
-  );\r
-\r
-EFI_STATUS\r
-EFIAPI\r
-SimpleNetworkDriverStop (\r
-  IN  EFI_DRIVER_BINDING_PROTOCOL    *This,\r
-  IN  EFI_HANDLE                     Controller,\r
-  IN  UINTN                          NumberOfChildren,\r
-  IN  EFI_HANDLE                     *ChildHandleBuffer\r
-  );\r
-\r
-//\r
-// Simple Network Protocol Driver Global Variables\r
-//\r
-EFI_DRIVER_BINDING_PROTOCOL mSimpleNetworkDriverBinding = {\r
-  SimpleNetworkDriverSupported,\r
-  SimpleNetworkDriverStart,\r
-  SimpleNetworkDriverStop,\r
-  0xa,\r
-  NULL,\r
-  NULL\r
-};\r
-\r
-//\r
-//  Module global variables needed to support undi 3.0 interface\r
-//\r
-EFI_PCI_IO_PROTOCOL         *mPciIoFncs;\r
-struct s_v2p                *_v2p = NULL; // undi3.0 map_list head\r
-// End Global variables\r
-//\r
-\r
 /**\r
-  This routine maps the given CPU address to a Device address. It creates a\r
-  an entry in the map list with the virtual and physical addresses and the\r
-  un map cookie.\r
+  One notified function to stop UNDI device when gBS->ExitBootServices() called.\r
 \r
-  @param  v2p                  pointer to return a map list node pointer.\r
-  @param  type                 the direction in which the data flows from the given\r
-                               virtual address device->cpu or cpu->device or both\r
-                               ways.\r
-  @param  vaddr                virtual address (or CPU address) to be mapped\r
-  @param  bsize                size of the buffer to be mapped.\r
-\r
-  @retval EFI_SUCEESS          routine has completed the mapping\r
-  @retval other                error as indicated.\r
+  @param  Event                   Pointer to this event\r
+  @param  Context                 Event handler private data\r
 \r
 **/\r
-EFI_STATUS\r
-add_v2p (\r
-  IN OUT struct s_v2p           **v2p,\r
-  EFI_PCI_IO_PROTOCOL_OPERATION type,\r
-  VOID                          *vaddr,\r
-  UINTN                         bsize\r
-  )\r
-{\r
-  EFI_STATUS  Status;\r
-\r
-  if ((v2p == NULL) || (vaddr == NULL) || (bsize == 0)) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  *v2p = AllocatePool (sizeof (struct s_v2p));\r
-  if (*v2p != NULL) {\r
-    return EFI_OUT_OF_RESOURCES;\r
-  }\r
-\r
-  Status = mPciIoFncs->Map (\r
-                        mPciIoFncs,\r
-                        type,\r
-                        vaddr,\r
-                        &bsize,\r
-                        &(*v2p)->paddr,\r
-                        &(*v2p)->unmap\r
-                        );\r
-  if (Status != EFI_SUCCESS) {\r
-    FreePool (*v2p);\r
-    return Status;\r
-  }\r
-  (*v2p)->vaddr = vaddr;\r
-  (*v2p)->bsize = bsize;\r
-  (*v2p)->next  = _v2p;\r
-  _v2p          = *v2p;\r
-\r
-  return EFI_SUCCESS;\r
-}\r
-\r
-\r
-/**\r
-  This routine searches the linked list of mapped address nodes (for undi3.0\r
-  interface) to find the node that corresponds to the given virtual address and\r
-  returns a pointer to that node.\r
-\r
-  @param  v2p                  pointer to return a map list node pointer.\r
-  @param  vaddr                virtual address (or CPU address) to be searched in\r
-                               the map list\r
-\r
-  @retval EFI_SUCEESS          if a match found!\r
-  @retval Other                match not found\r
-\r
-**/\r
-EFI_STATUS\r
-find_v2p (\r
-  struct s_v2p **v2p,\r
-  VOID         *vaddr\r
+VOID\r
+EFIAPI\r
+SnpNotifyExitBootServices (\r
+  EFI_EVENT Event,\r
+  VOID      *Context\r
   )\r
 {\r
-  struct s_v2p  *v;\r
-\r
-  if (v2p == NULL || vaddr == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
+  SNP_DRIVER             *Snp;\r
 \r
-  for (v = _v2p; v != NULL; v = v->next) {\r
-    if (v->vaddr == vaddr) {\r
-      *v2p = v;\r
-      return EFI_SUCCESS;\r
-    }\r
-  }\r
+  Snp  = (SNP_DRIVER *)Context;\r
 \r
-  return EFI_NOT_FOUND;\r
+  //\r
+  // Shutdown and stop UNDI driver\r
+  //\r
+  PxeShutdown (Snp);\r
+  PxeStop (Snp);\r
 }\r
 \r
-\r
 /**\r
-  This routine unmaps the given virtual address and frees the memory allocated\r
-  for the map list node corresponding to that address.\r
+  Send command to UNDI. It does nothing currently.\r
 \r
-  @param  vaddr                virtual address (or CPU address) to be unmapped\r
+  @param Cdb   command to be sent to UNDI.\r
 \r
-  @retval EFI_SUCEESS          if successfully unmapped\r
-  @retval Other                as indicated by the error\r
+  @retval EFI_INVALID_PARAMETER  The command is 0.\r
+  @retval EFI_UNSUPPORTED        Default return status because it's not\r
+                                 supported currently.\r
 \r
 **/\r
 EFI_STATUS\r
-del_v2p (\r
-  VOID *vaddr\r
-  )\r
-{\r
-  struct s_v2p  *v;\r
-  struct s_v2p  *t;\r
-  EFI_STATUS    Status;\r
-\r
-  if (vaddr == NULL) {\r
-    return EFI_INVALID_PARAMETER;\r
-  }\r
-\r
-  if (_v2p == NULL) {\r
-    return EFI_NOT_FOUND;\r
-  }\r
-  //\r
-  // Is our node at the head of the list??\r
-  //\r
-  if ((v = _v2p)->vaddr == vaddr) {\r
-    _v2p    = _v2p->next;\r
-\r
-    Status  = mPciIoFncs->Unmap (mPciIoFncs, v->unmap);\r
-\r
-    FreePool (v);\r
-\r
-    if (Status) {\r
-      DEBUG ((EFI_D_ERROR, "Unmap failed with status = %r\n", Status));\r
-    }\r
-    return Status;\r
-  }\r
-\r
-  for (; v->next != NULL; v = t) {\r
-    if ((t = v->next)->vaddr == vaddr) {\r
-      v->next = t->next;\r
-      Status  = mPciIoFncs->Unmap (mPciIoFncs, t->unmap);\r
-      FreePool (t);\r
-\r
-      if (Status) {\r
-        DEBUG ((EFI_D_ERROR, "Unmap failed with status = %r\n", Status));\r
-      }\r
-      return Status;\r
-    }\r
-  }\r
-\r
-  return EFI_NOT_FOUND;\r
-}\r
-\r
-STATIC\r
-EFI_STATUS\r
-issue_hwundi_command (\r
-  UINT64 cdb\r
+EFIAPI\r
+IssueHwUndiCommand (\r
+  UINT64 Cdb\r
   )\r
-/*++\r
-\r
-Routine Description:\r
-\r
-Arguments:\r
-\r
-Returns:\r
-\r
---*/\r
 {\r
-  DEBUG ((EFI_D_ERROR, "\nissue_hwundi_command() - This should not be called!"));\r
+  DEBUG ((EFI_D_ERROR, "\nIssueHwUndiCommand() - This should not be called!"));\r
 \r
-  if (cdb == 0) {\r
+  if (Cdb == 0) {\r
     return EFI_INVALID_PARAMETER;\r
 \r
   }\r
@@ -273,49 +65,52 @@ Returns:
 /**\r
   Compute 8-bit checksum of a buffer.\r
 \r
-  @param  ptr                  Pointer to buffer.\r
-  @param  len                  Length of buffer in bytes.\r
+  @param  Buffer               Pointer to buffer.\r
+  @param  Length               Length of buffer in bytes.\r
 \r
-  @return 8-bit checksum of all bytes in buffer.\r
-  @return If ptr is NULL or len is zero, zero is returned.\r
+  @return 8-bit checksum of all bytes in buffer, or zero if ptr is NULL or len\r
+          is zero.\r
 \r
 **/\r
-STATIC\r
 UINT8\r
-calc_8bit_cksum (\r
-  VOID  *ptr,\r
-  UINTN len\r
+Calc8BitCksum (\r
+  VOID  *Buffer,\r
+  UINTN Length\r
   )\r
 {\r
-  UINT8 *bptr;\r
-  UINT8 cksum;\r
+  UINT8 *Ptr;\r
+  UINT8 Cksum;\r
 \r
-  bptr  = ptr;\r
-  cksum = 0;\r
+  Ptr   = Buffer;\r
+  Cksum = 0;\r
 \r
-  if (ptr == NULL || len == 0) {\r
+  if (Ptr == NULL || Length == 0) {\r
     return 0;\r
   }\r
 \r
-  while (len--) {\r
-    cksum = (UINT8) (cksum +*bptr++);\r
+  while (Length-- != 0) {\r
+    Cksum = (UINT8) (Cksum + *Ptr++);\r
   }\r
 \r
-  return cksum;\r
+  return Cksum;\r
 }\r
 \r
-\r
 /**\r
-  Test to see if this driver supports Controller. Any Controller\r
-  that contains a Nii protocol can be supported.\r
-\r
-  @param  This                 Protocol instance pointer.\r
-  @param  Controller           Handle of device to test.\r
-  @param  RemainingDevicePath  Not used.\r
-\r
-  @retval EFI_SUCCESS          This driver supports this device.\r
-  @retval EFI_ALREADY_STARTED  This driver is already running on this device.\r
-  @retval other                This driver does not support this device.\r
+  Test to see if this driver supports ControllerHandle. This service\r
+  is called by the EFI boot service ConnectController(). In\r
+  order to make drivers as small as possible, there are a few calling\r
+  restrictions for this service. ConnectController() must\r
+  follow these calling restrictions. If any other agent wishes to call\r
+  Supported() it must also follow these calling restrictions.\r
+\r
+  @param  This                Protocol instance pointer.\r
+  @param  ControllerHandle    Handle of device to test.\r
+  @param  RemainingDevicePath Optional parameter use to pick a specific child\r
+                              device to start.\r
+\r
+  @retval EFI_SUCCESS         This driver supports this device.\r
+  @retval EFI_ALREADY_STARTED This driver is already running on this device.\r
+  @retval other               This driver does not support this device.\r
 \r
 **/\r
 EFI_STATUS\r
@@ -328,7 +123,7 @@ SimpleNetworkDriverSupported (
 {\r
   EFI_STATUS                                Status;\r
   EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL *NiiProtocol;\r
-  PXE_UNDI                                  *pxe;\r
+  PXE_UNDI                                  *Pxe;\r
 \r
   Status = gBS->OpenProtocol (\r
                   Controller,\r
@@ -370,36 +165,36 @@ SimpleNetworkDriverSupported (
   //\r
   // Check to see if !PXE structure is valid. Paragraph alignment of !PXE structure is required.\r
   //\r
-  if (NiiProtocol->ID & 0x0F) {\r
+  if ((NiiProtocol->Id & 0x0F) != 0) {\r
     DEBUG ((EFI_D_NET, "\n!PXE structure is not paragraph aligned.\n"));\r
     Status = EFI_UNSUPPORTED;\r
     goto Done;\r
   }\r
 \r
-  pxe = (PXE_UNDI *) (UINTN) (NiiProtocol->ID);\r
+  Pxe = (PXE_UNDI *) (UINTN) (NiiProtocol->Id);\r
 \r
   //\r
   //  Verify !PXE revisions.\r
   //\r
-  if (pxe->hw.Signature != PXE_ROMID_SIGNATURE) {\r
+  if (Pxe->hw.Signature != PXE_ROMID_SIGNATURE) {\r
     DEBUG ((EFI_D_NET, "\n!PXE signature is not valid.\n"));\r
     Status = EFI_UNSUPPORTED;\r
     goto Done;\r
   }\r
 \r
-  if (pxe->hw.Rev < PXE_ROMID_REV) {\r
+  if (Pxe->hw.Rev < PXE_ROMID_REV) {\r
     DEBUG ((EFI_D_NET, "\n!PXE.Rev is not supported.\n"));\r
     Status = EFI_UNSUPPORTED;\r
     goto Done;\r
   }\r
 \r
-  if (pxe->hw.MajorVer < PXE_ROMID_MAJORVER) {\r
+  if (Pxe->hw.MajorVer < PXE_ROMID_MAJORVER) {\r
 \r
     DEBUG ((EFI_D_NET, "\n!PXE.MajorVer is not supported.\n"));\r
     Status = EFI_UNSUPPORTED;\r
     goto Done;\r
 \r
-  } else if (pxe->hw.MajorVer == PXE_ROMID_MAJORVER && pxe->hw.MinorVer < PXE_ROMID_MINORVER) {\r
+  } else if (Pxe->hw.MajorVer == PXE_ROMID_MAJORVER && Pxe->hw.MinorVer < PXE_ROMID_MINORVER) {\r
     DEBUG ((EFI_D_NET, "\n!PXE.MinorVer is not supported."));\r
     Status = EFI_UNSUPPORTED;\r
     goto Done;\r
@@ -407,14 +202,14 @@ SimpleNetworkDriverSupported (
   //\r
   // Do S/W UNDI specific checks.\r
   //\r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_HW_UNDI) == 0) {\r
-    if (pxe->sw.EntryPoint < pxe->sw.Len) {\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_HW_UNDI) == 0) {\r
+    if (Pxe->sw.EntryPoint < Pxe->sw.Len) {\r
       DEBUG ((EFI_D_NET, "\n!PXE S/W entry point is not valid."));\r
       Status = EFI_UNSUPPORTED;\r
       goto Done;\r
     }\r
 \r
-    if (pxe->sw.BusCnt == 0) {\r
+    if (Pxe->sw.BusCnt == 0) {\r
       DEBUG ((EFI_D_NET, "\n!PXE.BusCnt is zero."));\r
       Status = EFI_UNSUPPORTED;\r
       goto Done;\r
@@ -435,16 +230,22 @@ Done:
   return Status;\r
 }\r
 \r
-\r
 /**\r
-  called for any handle that we said "supported" in the above call!\r
+  Start this driver on ControllerHandle. This service is called by the\r
+  EFI boot service ConnectController(). In order to make\r
+  drivers as small as possible, there are a few calling restrictions for\r
+  this service. ConnectController() must follow these\r
+  calling restrictions. If any other agent wishes to call Start() it\r
+  must also follow these calling restrictions.\r
 \r
   @param  This                 Protocol instance pointer.\r
-  @param  Controller           Handle of device to start\r
-  @param  RemainingDevicePath  Not used.\r
+  @param  ControllerHandle     Handle of device to bind driver to.\r
+  @param  RemainingDevicePath  Optional parameter use to pick a specific child\r
+                               device to start.\r
 \r
-  @retval EFI_SUCCESS          This driver supports this device.\r
-  @retval other                This driver failed to start this device.\r
+  @retval EFI_SUCCESS          This driver is added to ControllerHandle\r
+  @retval EFI_DEVICE_ERROR     This driver could not be started due to a device error\r
+  @retval other                This driver does not support this device\r
 \r
 **/\r
 EFI_STATUS\r
@@ -458,15 +259,16 @@ SimpleNetworkDriverStart (
   EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL *Nii;\r
   EFI_DEVICE_PATH_PROTOCOL                  *NiiDevicePath;\r
   EFI_STATUS                                Status;\r
-  PXE_UNDI                                  *pxe;\r
-  SNP_DRIVER                                *snp;\r
-  VOID                                      *addr;\r
+  PXE_UNDI                                  *Pxe;\r
+  SNP_DRIVER                                *Snp;\r
+  VOID                                      *Address;\r
   EFI_HANDLE                                Handle;\r
-  PXE_PCI_CONFIG_INFO                       ConfigInfo;\r
-  PCI_TYPE00                                *ConfigHeader;\r
-  UINT32                                    *TempBar;\r
   UINT8                                     BarIndex;\r
   PXE_STATFLAGS                             InitStatFlags;\r
+  EFI_PCI_IO_PROTOCOL                       *PciIo;\r
+  EFI_ACPI_ADDRESS_SPACE_DESCRIPTOR         *BarDesc;\r
+  BOOLEAN                                   FoundIoBar;\r
+  BOOLEAN                                   FoundMemoryBar;\r
 \r
   DEBUG ((EFI_D_NET, "\nSnpNotifyNetworkInterfaceIdentifier()  "));\r
 \r
@@ -496,7 +298,7 @@ SimpleNetworkDriverStart (
   Status = gBS->OpenProtocol (\r
                   Handle,\r
                   &gEfiPciIoProtocolGuid,\r
-                  (VOID **) &mPciIoFncs,\r
+                  (VOID **) &PciIo,\r
                   This->DriverBindingHandle,\r
                   Controller,\r
                   EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
@@ -527,18 +329,18 @@ SimpleNetworkDriverStart (
 \r
   DEBUG ((EFI_D_INFO, "Start(): UNDI3.1 found\n"));\r
 \r
-  pxe = (PXE_UNDI *) (UINTN) (Nii->ID);\r
+  Pxe = (PXE_UNDI *) (UINTN) (Nii->Id);\r
 \r
-  if (calc_8bit_cksum (pxe, pxe->hw.Len) != 0) {\r
+  if (Calc8BitCksum (Pxe, Pxe->hw.Len) != 0) {\r
     DEBUG ((EFI_D_NET, "\n!PXE checksum is not correct.\n"));\r
     goto NiiError;\r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_RX_SUPPORTED) != 0) {\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_RX_SUPPORTED) != 0) {\r
     //\r
     //  We can get any packets.\r
     //\r
-  } else if ((pxe->hw.Implementation & PXE_ROMID_IMP_BROADCAST_RX_SUPPORTED) != 0) {\r
+  } else if ((Pxe->hw.Implementation & PXE_ROMID_IMP_BROADCAST_RX_SUPPORTED) != 0) {\r
     //\r
     //  We need to be able to get broadcast packets for DHCP.\r
     //  If we do not have promiscuous support, we must at least have\r
@@ -552,62 +354,75 @@ SimpleNetworkDriverStart (
   // OK, we like this UNDI, and we know snp is not already there on this handle\r
   // Allocate and initialize a new simple network protocol structure.\r
   //\r
-  Status = mPciIoFncs->AllocateBuffer (\r
-                        mPciIoFncs,\r
-                        AllocateAnyPages,\r
-                        EfiBootServicesData,\r
-                        SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
-                        &addr,\r
-                        0\r
-                        );\r
+  Status = PciIo->AllocateBuffer (\r
+                    PciIo,\r
+                    AllocateAnyPages,\r
+                    EfiBootServicesData,\r
+                    SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
+                    &Address,\r
+                    0\r
+                    );\r
 \r
   if (Status != EFI_SUCCESS) {\r
     DEBUG ((EFI_D_NET, "\nCould not allocate SNP_DRIVER structure.\n"));\r
     goto NiiError;\r
   }\r
 \r
-  snp = (SNP_DRIVER *) (UINTN) addr;\r
+  Snp = (SNP_DRIVER *) (UINTN) Address;\r
+\r
+  ZeroMem (Snp, sizeof (SNP_DRIVER));\r
 \r
-  ZeroMem (snp, sizeof (SNP_DRIVER));\r
+  Snp->PciIo      = PciIo;\r
+  Snp->Signature  = SNP_DRIVER_SIGNATURE;\r
 \r
-  snp->IoFncs     = mPciIoFncs;\r
-  snp->Signature  = SNP_DRIVER_SIGNATURE;\r
+  EfiInitializeLock (&Snp->Lock, TPL_NOTIFY);\r
 \r
-  EfiInitializeLock (&snp->lock, TPL_NOTIFY);\r
+  Snp->Snp.Revision       = EFI_SIMPLE_NETWORK_PROTOCOL_REVISION;\r
+  Snp->Snp.Start          = SnpUndi32Start;\r
+  Snp->Snp.Stop           = SnpUndi32Stop;\r
+  Snp->Snp.Initialize     = SnpUndi32Initialize;\r
+  Snp->Snp.Reset          = SnpUndi32Reset;\r
+  Snp->Snp.Shutdown       = SnpUndi32Shutdown;\r
+  Snp->Snp.ReceiveFilters = SnpUndi32ReceiveFilters;\r
+  Snp->Snp.StationAddress = SnpUndi32StationAddress;\r
+  Snp->Snp.Statistics     = SnpUndi32Statistics;\r
+  Snp->Snp.MCastIpToMac   = SnpUndi32McastIpToMac;\r
+  Snp->Snp.NvData         = SnpUndi32NvData;\r
+  Snp->Snp.GetStatus      = SnpUndi32GetStatus;\r
+  Snp->Snp.Transmit       = SnpUndi32Transmit;\r
+  Snp->Snp.Receive        = SnpUndi32Receive;\r
+  Snp->Snp.WaitForPacket  = NULL;\r
 \r
-  snp->snp.Revision       = EFI_SIMPLE_NETWORK_PROTOCOL_REVISION;\r
-  snp->snp.Start          = snp_undi32_start;\r
-  snp->snp.Stop           = snp_undi32_stop;\r
-  snp->snp.Initialize     = snp_undi32_initialize;\r
-  snp->snp.Reset          = snp_undi32_reset;\r
-  snp->snp.Shutdown       = snp_undi32_shutdown;\r
-  snp->snp.ReceiveFilters = snp_undi32_receive_filters;\r
-  snp->snp.StationAddress = snp_undi32_station_address;\r
-  snp->snp.Statistics     = snp_undi32_statistics;\r
-  snp->snp.MCastIpToMac   = snp_undi32_mcast_ip_to_mac;\r
-  snp->snp.NvData         = snp_undi32_nvdata;\r
-  snp->snp.GetStatus      = snp_undi32_get_status;\r
-  snp->snp.Transmit       = snp_undi32_transmit;\r
-  snp->snp.Receive        = snp_undi32_receive;\r
-  snp->snp.WaitForPacket  = NULL;\r
+  Snp->Snp.Mode           = &Snp->Mode;\r
 \r
-  snp->snp.Mode           = &snp->mode;\r
+  Snp->TxRxBufferSize     = 0;\r
+  Snp->TxRxBuffer         = NULL;\r
+\r
+  Snp->RecycledTxBuf = AllocatePool (sizeof (UINT64) * SNP_TX_BUFFER_INCREASEMENT);\r
+  if (Snp->RecycledTxBuf == NULL) {\r
+    Status = EFI_OUT_OF_RESOURCES;\r
+    goto Error_DeleteSNP;\r
+  }\r
+  Snp->MaxRecycledTxBuf    = SNP_TX_BUFFER_INCREASEMENT;\r
+  Snp->RecycledTxBufCount  = 0;\r
 \r
-  snp->tx_rx_bufsize      = 0;\r
-  snp->tx_rx_buffer       = NULL;\r
+  if (Nii->Revision >= EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL_REVISION) {\r
+    Snp->IfNum = Nii->IfNum;\r
 \r
-  snp->if_num             = Nii->IfNum;\r
+  } else {\r
+    Snp->IfNum = (UINT8) (Nii->IfNum & 0xFF);\r
+  }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_HW_UNDI) != 0) {\r
-    snp->is_swundi            = FALSE;\r
-    snp->issue_undi32_command = &issue_hwundi_command;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_HW_UNDI) != 0) {\r
+    Snp->IsSwUndi             = FALSE;\r
+    Snp->IssueUndi32Command   = &IssueHwUndiCommand;\r
   } else {\r
-    snp->is_swundi = TRUE;\r
+    Snp->IsSwUndi = TRUE;\r
 \r
-    if ((pxe->sw.Implementation & PXE_ROMID_IMP_SW_VIRT_ADDR) != 0) {\r
-      snp->issue_undi32_command = (issue_undi32_command) (UINTN) pxe->sw.EntryPoint;\r
+    if ((Pxe->sw.Implementation & PXE_ROMID_IMP_SW_VIRT_ADDR) != 0) {\r
+      Snp->IssueUndi32Command = (ISSUE_UNDI32_COMMAND) (UINTN) Pxe->sw.EntryPoint;\r
     } else {\r
-      snp->issue_undi32_command = (issue_undi32_command) (UINTN) ((UINT8) (UINTN) pxe + pxe->sw.EntryPoint);\r
+      Snp->IssueUndi32Command = (ISSUE_UNDI32_COMMAND) (UINTN) ((UINT8) (UINTN) Pxe + Pxe->sw.EntryPoint);\r
     }\r
   }\r
   //\r
@@ -626,223 +441,226 @@ SimpleNetworkDriverStart (
   // -it is OK to allocate one global set of CPB, DB pair for each UNDI\r
   // interface as EFI does not multi-task and so SNP will not be re-entered!\r
   //\r
-  Status = mPciIoFncs->AllocateBuffer (\r
-                        mPciIoFncs,\r
-                        AllocateAnyPages,\r
-                        EfiBootServicesData,\r
-                        SNP_MEM_PAGES (4096),\r
-                        &addr,\r
-                        0\r
-                        );\r
+  Status = PciIo->AllocateBuffer (\r
+                    PciIo,\r
+                    AllocateAnyPages,\r
+                    EfiBootServicesData,\r
+                    SNP_MEM_PAGES (4096),\r
+                    &Address,\r
+                    0\r
+                    );\r
 \r
   if (Status != EFI_SUCCESS) {\r
     DEBUG ((EFI_D_NET, "\nCould not allocate CPB and DB structures.\n"));\r
     goto Error_DeleteSNP;\r
   }\r
 \r
-  snp->cpb  = (VOID *) (UINTN) addr;\r
-  snp->db   = (VOID *) ((UINTN) addr + 2048);\r
-\r
-  //\r
-  // pxe_start call is going to give the callback functions to UNDI, these callback\r
-  // functions use the BarIndex values from the snp structure, so these must be initialized\r
-  // with default values before doing a pxe_start. The correct values can be obtained after\r
-  // getting the config information from UNDI\r
-  //\r
-  snp->MemoryBarIndex = 0;\r
-  snp->IoBarIndex     = 1;\r
+  Snp->Cpb  = (VOID *) (UINTN) Address;\r
+  Snp->Db   = (VOID *) ((UINTN) Address + 2048);\r
 \r
   //\r
-  // we need the undi init information many times in this snp code, just get it\r
-  // once here and store it in the snp driver structure. to get Init Info\r
-  // from UNDI we have to start undi first.\r
+  // Find the correct BAR to do IO.\r
   //\r
-  Status = pxe_start (snp);\r
-\r
-  if (Status != EFI_SUCCESS) {\r
-    goto Error_DeleteSNP;\r
-  }\r
-\r
-  snp->cdb.OpCode     = PXE_OPCODE_GET_INIT_INFO;\r
-  snp->cdb.OpFlags    = PXE_OPFLAGS_NOT_USED;\r
-\r
-  snp->cdb.CPBsize    = PXE_CPBSIZE_NOT_USED;\r
-  snp->cdb.CPBaddr    = PXE_DBADDR_NOT_USED;\r
-\r
-  snp->cdb.DBsize     = sizeof snp->init_info;\r
-  snp->cdb.DBaddr     = (UINT64)(UINTN) &snp->init_info;\r
-\r
-  snp->cdb.StatCode   = PXE_STATCODE_INITIALIZE;\r
-  snp->cdb.StatFlags  = PXE_STATFLAGS_INITIALIZE;\r
+  // Enumerate through the PCI BARs for the device to determine which one is\r
+  // the IO BAR.  Save the index of the BAR into the adapter info structure.\r
+  // for regular 32bit BARs, 0 is memory mapped, 1 is io mapped\r
+  //\r
+  Snp->MemoryBarIndex = 0;\r
+  Snp->IoBarIndex     = 1;\r
+  FoundMemoryBar      = FALSE;\r
+  FoundIoBar          = FALSE;\r
+  for (BarIndex = 0; BarIndex < PCI_MAX_BAR; BarIndex++) {\r
+    Status = PciIo->GetBarAttributes (\r
+                      PciIo,\r
+                      BarIndex,\r
+                      NULL,\r
+                      (VOID**) &BarDesc\r
+                      );\r
+    if (Status == EFI_UNSUPPORTED) {\r
+      continue;\r
+    } else if (EFI_ERROR (Status)) {\r
+      goto Error_DeleteSNP;\r
+    }\r
 \r
-  snp->cdb.IFnum      = snp->if_num;\r
-  snp->cdb.Control    = PXE_CONTROL_LAST_CDB_IN_LIST;\r
+    if ((!FoundMemoryBar) && (BarDesc->ResType == ACPI_ADDRESS_SPACE_TYPE_MEM)) {\r
+      Snp->MemoryBarIndex = BarIndex;\r
+      FoundMemoryBar      = TRUE;\r
+    } else if ((!FoundIoBar) && (BarDesc->ResType == ACPI_ADDRESS_SPACE_TYPE_IO)) {\r
+      Snp->IoBarIndex = BarIndex;\r
+      FoundIoBar      = TRUE;\r
+    }\r
 \r
-  DEBUG ((EFI_D_NET, "\nsnp->undi.get_init_info()  "));\r
+    FreePool (BarDesc);\r
 \r
-  (*snp->issue_undi32_command) ((UINT64)(UINTN) &snp->cdb);\r
+    if (FoundMemoryBar && FoundIoBar) {\r
+      break;\r
+    }\r
+  }\r
 \r
-  //\r
-  // Save the INIT Stat Code...\r
-  //\r
-  InitStatFlags = snp->cdb.StatFlags;\r
+  Status = PxeStart (Snp);\r
 \r
-  if (snp->cdb.StatCode != PXE_STATCODE_SUCCESS) {\r
-    DEBUG ((EFI_D_NET, "\nsnp->undi.init_info()  %xh:%xh\n", snp->cdb.StatFlags, snp->cdb.StatCode));\r
-    pxe_stop (snp);\r
+  if (Status != EFI_SUCCESS) {\r
     goto Error_DeleteSNP;\r
   }\r
 \r
-  snp->cdb.OpCode     = PXE_OPCODE_GET_CONFIG_INFO;\r
-  snp->cdb.OpFlags    = PXE_OPFLAGS_NOT_USED;\r
+  Snp->Cdb.OpCode     = PXE_OPCODE_GET_INIT_INFO;\r
+  Snp->Cdb.OpFlags    = PXE_OPFLAGS_NOT_USED;\r
 \r
-  snp->cdb.CPBsize    = PXE_CPBSIZE_NOT_USED;\r
-  snp->cdb.CPBaddr    = PXE_DBADDR_NOT_USED;\r
+  Snp->Cdb.CPBsize    = PXE_CPBSIZE_NOT_USED;\r
+  Snp->Cdb.CPBaddr    = PXE_DBADDR_NOT_USED;\r
 \r
-  snp->cdb.DBsize     = sizeof ConfigInfo;\r
-  snp->cdb.DBaddr     = (UINT64)(UINTN) &ConfigInfo;\r
+  Snp->Cdb.DBsize     = (UINT16) sizeof (Snp->InitInfo);\r
+  Snp->Cdb.DBaddr     = (UINT64)(UINTN) (&Snp->InitInfo);\r
 \r
-  snp->cdb.StatCode   = PXE_STATCODE_INITIALIZE;\r
-  snp->cdb.StatFlags  = PXE_STATFLAGS_INITIALIZE;\r
+  Snp->Cdb.StatCode   = PXE_STATCODE_INITIALIZE;\r
+  Snp->Cdb.StatFlags  = PXE_STATFLAGS_INITIALIZE;\r
 \r
-  snp->cdb.IFnum      = snp->if_num;\r
-  snp->cdb.Control    = PXE_CONTROL_LAST_CDB_IN_LIST;\r
+  Snp->Cdb.IFnum      = Snp->IfNum;\r
+  Snp->Cdb.Control    = PXE_CONTROL_LAST_CDB_IN_LIST;\r
 \r
-  DEBUG ((EFI_D_NET, "\nsnp->undi.get_config_info()  "));\r
+  DEBUG ((EFI_D_NET, "\nSnp->undi.get_init_info()  "));\r
 \r
-  (*snp->issue_undi32_command) ((UINT64)(UINTN) &snp->cdb);\r
+  (*Snp->IssueUndi32Command) ((UINT64)(UINTN) &Snp->Cdb);\r
 \r
-  if (snp->cdb.StatCode != PXE_STATCODE_SUCCESS) {\r
-    DEBUG ((EFI_D_NET, "\nsnp->undi.config_info()  %xh:%xh\n", snp->cdb.StatFlags, snp->cdb.StatCode));\r
-    pxe_stop (snp);\r
-    goto Error_DeleteSNP;\r
-  }\r
-  //\r
-  // Find the correct BAR to do IO.\r
   //\r
+  // Save the INIT Stat Code...\r
   //\r
-  // Enumerate through the PCI BARs for the device to determine which one is\r
-  // the IO BAR.  Save the index of the BAR into the adapter info structure.\r
-  // for  regular 32bit BARs, 0 is memory mapped, 1 is io mapped\r
-  //\r
-  ConfigHeader  = (PCI_TYPE00 *) &ConfigInfo.Config.Byte[0];\r
-  TempBar       = (UINT32 *) &ConfigHeader->Device.Bar[0];\r
-  for (BarIndex = 0; BarIndex <= 5; BarIndex++) {\r
-    if ((*TempBar & PCI_BAR_MEM_MASK) == PCI_BAR_MEM_64BIT) {\r
-      //\r
-      // This is a 64-bit memory bar, skip this and the\r
-      // next bar as well.\r
-      //\r
-      TempBar++;\r
-    }\r
+  InitStatFlags = Snp->Cdb.StatFlags;\r
 \r
-    if ((*TempBar & PCI_BAR_IO_MASK) == PCI_BAR_IO_MODE) {\r
-      snp->IoBarIndex = BarIndex;\r
-      break;\r
-    }\r
-\r
-    TempBar++;\r
+  if (Snp->Cdb.StatCode != PXE_STATCODE_SUCCESS) {\r
+    DEBUG ((EFI_D_NET, "\nSnp->undi.init_info()  %xh:%xh\n", Snp->Cdb.StatFlags, Snp->Cdb.StatCode));\r
+    PxeStop (Snp);\r
+    goto Error_DeleteSNP;\r
   }\r
 \r
   //\r
   //  Initialize simple network protocol mode structure\r
   //\r
-  snp->mode.State               = EfiSimpleNetworkStopped;\r
-  snp->mode.HwAddressSize       = snp->init_info.HWaddrLen;\r
-  snp->mode.MediaHeaderSize     = snp->init_info.MediaHeaderLen;\r
-  snp->mode.MaxPacketSize       = snp->init_info.FrameDataLen;\r
-  snp->mode.NvRamAccessSize     = snp->init_info.NvWidth;\r
-  snp->mode.NvRamSize           = snp->init_info.NvCount * snp->mode.NvRamAccessSize;\r
-  snp->mode.IfType              = snp->init_info.IFtype;\r
-  snp->mode.MaxMCastFilterCount = snp->init_info.MCastFilterCnt;\r
-  snp->mode.MCastFilterCount    = 0;\r
+  Snp->Mode.State               = EfiSimpleNetworkStopped;\r
+  Snp->Mode.HwAddressSize       = Snp->InitInfo.HWaddrLen;\r
+  Snp->Mode.MediaHeaderSize     = Snp->InitInfo.MediaHeaderLen;\r
+  Snp->Mode.MaxPacketSize       = Snp->InitInfo.FrameDataLen;\r
+  Snp->Mode.NvRamAccessSize     = Snp->InitInfo.NvWidth;\r
+  Snp->Mode.NvRamSize           = Snp->InitInfo.NvCount * Snp->Mode.NvRamAccessSize;\r
+  Snp->Mode.IfType              = Snp->InitInfo.IFtype;\r
+  Snp->Mode.MaxMCastFilterCount = Snp->InitInfo.MCastFilterCnt;\r
+  Snp->Mode.MCastFilterCount    = 0;\r
 \r
   switch (InitStatFlags & PXE_STATFLAGS_CABLE_DETECT_MASK) {\r
   case PXE_STATFLAGS_CABLE_DETECT_SUPPORTED:\r
-    snp->mode.MediaPresentSupported = TRUE;\r
+    Snp->CableDetectSupported = TRUE;\r
     break;\r
 \r
   case PXE_STATFLAGS_CABLE_DETECT_NOT_SUPPORTED:\r
   default:\r
-    snp->mode.MediaPresentSupported = FALSE;\r
+    Snp->CableDetectSupported = FALSE;\r
+  }\r
+\r
+  switch (InitStatFlags & PXE_STATFLAGS_GET_STATUS_NO_MEDIA_MASK) {\r
+  case PXE_STATFLAGS_GET_STATUS_NO_MEDIA_SUPPORTED:\r
+    Snp->MediaStatusSupported = TRUE;\r
+    break;\r
+\r
+  case PXE_STATFLAGS_GET_STATUS_NO_MEDIA_NOT_SUPPORTED:\r
+  default:\r
+    Snp->MediaStatusSupported = FALSE;\r
+  }\r
+\r
+  if (Snp->CableDetectSupported || Snp->MediaStatusSupported) {\r
+    Snp->Mode.MediaPresentSupported = TRUE;\r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_STATION_ADDR_SETTABLE) != 0) {\r
-    snp->mode.MacAddressChangeable = TRUE;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_STATION_ADDR_SETTABLE) != 0) {\r
+    Snp->Mode.MacAddressChangeable = TRUE;\r
   } else {\r
-    snp->mode.MacAddressChangeable = FALSE;\r
+    Snp->Mode.MacAddressChangeable = FALSE;\r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_MULTI_FRAME_SUPPORTED) != 0) {\r
-    snp->mode.MultipleTxSupported = TRUE;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_MULTI_FRAME_SUPPORTED) != 0) {\r
+    Snp->Mode.MultipleTxSupported = TRUE;\r
   } else {\r
-    snp->mode.MultipleTxSupported = FALSE;\r
+    Snp->Mode.MultipleTxSupported = FALSE;\r
   }\r
 \r
-  snp->mode.ReceiveFilterMask = EFI_SIMPLE_NETWORK_RECEIVE_UNICAST;\r
+  Snp->Mode.ReceiveFilterMask = EFI_SIMPLE_NETWORK_RECEIVE_UNICAST;\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_MULTICAST_RX_SUPPORTED) != 0) {\r
-    snp->mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS_MULTICAST;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_MULTICAST_RX_SUPPORTED) != 0) {\r
+    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS_MULTICAST;\r
 \r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_RX_SUPPORTED) != 0) {\r
-    snp->mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_RX_SUPPORTED) != 0) {\r
+    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS;\r
 \r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_BROADCAST_RX_SUPPORTED) != 0) {\r
-    snp->mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_BROADCAST;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_BROADCAST_RX_SUPPORTED) != 0) {\r
+    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_BROADCAST;\r
 \r
   }\r
 \r
-  if ((pxe->hw.Implementation & PXE_ROMID_IMP_FILTERED_MULTICAST_RX_SUPPORTED) != 0) {\r
-    snp->mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_MULTICAST;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_FILTERED_MULTICAST_RX_SUPPORTED) != 0) {\r
+    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_MULTICAST;\r
 \r
   }\r
 \r
-  if (pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_MULTICAST_RX_SUPPORTED) {\r
-    snp->mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS_MULTICAST;\r
+  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_MULTICAST_RX_SUPPORTED) != 0) {\r
+    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS_MULTICAST;\r
 \r
   }\r
 \r
-  snp->mode.ReceiveFilterSetting = 0;\r
+  Snp->Mode.ReceiveFilterSetting = 0;\r
 \r
   //\r
   //  need to get the station address to save in the mode structure. we need to\r
   // initialize the UNDI first for this.\r
   //\r
-  snp->tx_rx_bufsize  = snp->init_info.MemoryRequired;\r
-  Status              = pxe_init (snp, PXE_OPFLAGS_INITIALIZE_DO_NOT_DETECT_CABLE);\r
+  Snp->TxRxBufferSize = Snp->InitInfo.MemoryRequired;\r
+  Status              = PxeInit (Snp, PXE_OPFLAGS_INITIALIZE_DO_NOT_DETECT_CABLE);\r
 \r
-  if (Status) {\r
-    pxe_stop (snp);\r
+  if (EFI_ERROR (Status)) {\r
+    PxeStop (Snp);\r
     goto Error_DeleteSNP;\r
   }\r
 \r
-  Status = pxe_get_stn_addr (snp);\r
+  Status = PxeGetStnAddr (Snp);\r
 \r
   if (Status != EFI_SUCCESS) {\r
-    DEBUG ((EFI_D_ERROR, "\nsnp->undi.get_station_addr()  failed.\n"));\r
-    pxe_shutdown (snp);\r
-    pxe_stop (snp);\r
+    DEBUG ((EFI_D_ERROR, "\nSnp->undi.get_station_addr() failed.\n"));\r
+    PxeShutdown (Snp);\r
+    PxeStop (Snp);\r
     goto Error_DeleteSNP;\r
   }\r
 \r
-  snp->mode.MediaPresent = FALSE;\r
+  Snp->Mode.MediaPresent = FALSE;\r
 \r
   //\r
   // We should not leave UNDI started and initialized here. this DriverStart()\r
   // routine must only find and attach the SNP interface to UNDI layer that it\r
   // finds on the given handle!\r
-  // The UNDI layer will be started when upper layers call snp->start.\r
+  // The UNDI layer will be started when upper layers call Snp->start.\r
   // How ever, this DriverStart() must fill up the snp mode structure which\r
   // contains the MAC address of the NIC. For this reason we started and\r
   // initialized UNDI here, now we are done, do a shutdown and stop of the\r
   // UNDI interface!\r
   //\r
-  pxe_shutdown (snp);\r
-  pxe_stop (snp);\r
+  PxeShutdown (Snp);\r
+  PxeStop (Snp);\r
+\r
+  //\r
+  // Create EXIT_BOOT_SERIVES Event\r
+  //\r
+  Status = gBS->CreateEventEx (\r
+                  EVT_NOTIFY_SIGNAL,\r
+                  TPL_NOTIFY,\r
+                  SnpNotifyExitBootServices,\r
+                  Snp,\r
+                  &gEfiEventExitBootServicesGuid,\r
+                  &Snp->ExitBootServicesEvent\r
+                  );\r
+  if (EFI_ERROR (Status)) {\r
+    goto Error_DeleteSNP;\r
+  }\r
 \r
   //\r
   //  add SNP to the undi handle\r
@@ -851,26 +669,30 @@ SimpleNetworkDriverStart (
                   &Controller,\r
                   &gEfiSimpleNetworkProtocolGuid,\r
                   EFI_NATIVE_INTERFACE,\r
-                  &(snp->snp)\r
+                  &(Snp->Snp)\r
                   );\r
 \r
   if (!EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
 \r
-  Status = mPciIoFncs->FreeBuffer (\r
-                        mPciIoFncs,\r
-                        SNP_MEM_PAGES (4096),\r
-                        snp->cpb\r
-                        );\r
+  PciIo->FreeBuffer (\r
+           PciIo,\r
+           SNP_MEM_PAGES (4096),\r
+           Snp->Cpb\r
+           );\r
 \r
 Error_DeleteSNP:\r
 \r
-  mPciIoFncs->FreeBuffer (\r
-                mPciIoFncs,\r
-                SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
-                snp\r
-                );\r
+  if (Snp->RecycledTxBuf != NULL) {\r
+    FreePool (Snp->RecycledTxBuf);\r
+  }\r
+\r
+  PciIo->FreeBuffer (\r
+           PciIo,\r
+           SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
+           Snp\r
+           );\r
 NiiError:\r
   gBS->CloseProtocol (\r
         Controller,\r
@@ -886,13 +708,32 @@ NiiError:
         Controller\r
         );\r
 \r
+  //\r
+  // If we got here that means we are in error state.\r
+  //\r
+  if (!EFI_ERROR (Status)) {\r
+    Status = EFI_DEVICE_ERROR;\r
+  }\r
+\r
   return Status;\r
 }\r
 \r
-\r
 /**\r
-\r
-\r
+  Stop this driver on ControllerHandle. This service is called by the\r
+  EFI boot service DisconnectController(). In order to\r
+  make drivers as small as possible, there are a few calling\r
+  restrictions for this service. DisconnectController()\r
+  must follow these calling restrictions. If any other agent wishes\r
+  to call Stop() it must also follow these calling restrictions.\r
+\r
+  @param  This              Protocol instance pointer.\r
+  @param  ControllerHandle  Handle of device to stop driver on\r
+  @param  NumberOfChildren  Number of Handles in ChildHandleBuffer. If number of\r
+                            children is zero stop the entire bus driver.\r
+  @param  ChildHandleBuffer List of Child Handles to Stop.\r
+\r
+  @retval EFI_SUCCESS       This driver is removed ControllerHandle\r
+  @retval other             This driver was not removed from this device\r
 \r
 **/\r
 EFI_STATUS\r
@@ -907,6 +748,7 @@ SimpleNetworkDriverStop (
   EFI_STATUS                  Status;\r
   EFI_SIMPLE_NETWORK_PROTOCOL *SnpProtocol;\r
   SNP_DRIVER                  *Snp;\r
+  EFI_PCI_IO_PROTOCOL         *PciIo;\r
 \r
   //\r
   // Get our context back.\r
@@ -929,13 +771,18 @@ SimpleNetworkDriverStop (
   Status = gBS->UninstallProtocolInterface (\r
                   Controller,\r
                   &gEfiSimpleNetworkProtocolGuid,\r
-                  &Snp->snp\r
+                  &Snp->Snp\r
                   );\r
 \r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
 \r
+  //\r
+  // Close EXIT_BOOT_SERIVES Event\r
+  //\r
+  gBS->CloseEvent (Snp->ExitBootServicesEvent);\r
+\r
   Status = gBS->CloseProtocol (\r
                   Controller,\r
                   &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
@@ -950,36 +797,51 @@ SimpleNetworkDriverStop (
                   Controller\r
                   );\r
 \r
-  pxe_shutdown (Snp);\r
-  pxe_stop (Snp);\r
+  PxeShutdown (Snp);\r
+  PxeStop (Snp);\r
 \r
-  mPciIoFncs->FreeBuffer (\r
-                mPciIoFncs,\r
-                SNP_MEM_PAGES (4096),\r
-                Snp->cpb\r
-                );\r
+  FreePool (Snp->RecycledTxBuf);\r
 \r
-  mPciIoFncs->FreeBuffer (\r
-                mPciIoFncs,\r
-                SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
-                Snp\r
-                );\r
+  PciIo = Snp->PciIo;\r
+  PciIo->FreeBuffer (\r
+           PciIo,\r
+           SNP_MEM_PAGES (4096),\r
+           Snp->Cpb\r
+           );\r
+\r
+  PciIo->FreeBuffer (\r
+           PciIo,\r
+           SNP_MEM_PAGES (sizeof (SNP_DRIVER)),\r
+           Snp\r
+           );\r
 \r
   return Status;\r
 }\r
 \r
+//\r
+// Simple Network Protocol Driver Global Variables\r
+//\r
+EFI_DRIVER_BINDING_PROTOCOL gSimpleNetworkDriverBinding = {\r
+  SimpleNetworkDriverSupported,\r
+  SimpleNetworkDriverStart,\r
+  SimpleNetworkDriverStop,\r
+  0xa,\r
+  NULL,\r
+  NULL\r
+};\r
 \r
 /**\r
-  Install all the driver protocol\r
+  The SNP driver entry point.\r
 \r
-  @param  entry                EFI_IMAGE_ENTRY_POINT)\r
+  @param ImageHandle       The driver image handle.\r
+  @param SystemTable       The system table.\r
 \r
-  @retval EFI_SUCEESS          Initialization routine has found UNDI hardware,\r
-                               loaded it's ROM, and installed a notify event for\r
-                               the Network Indentifier Interface Protocol\r
-                               successfully.\r
-  @retval Other                Return value from HandleProtocol for\r
-                               DeviceIoProtocol or LoadedImageProtocol\r
+  @retval EFI_SUCEESS      Initialization routine has found UNDI hardware,\r
+                           loaded it's ROM, and installed a notify event for\r
+                           the Network Indentifier Interface Protocol\r
+                           successfully.\r
+  @retval Other            Return value from HandleProtocol for\r
+                           DeviceIoProtocol or LoadedImageProtocol\r
 \r
 **/\r
 EFI_STATUS\r
@@ -992,8 +854,8 @@ InitializeSnpNiiDriver (
   return EfiLibInstallDriverBindingComponentName2 (\r
            ImageHandle,\r
            SystemTable,\r
-           &mSimpleNetworkDriverBinding,\r
-           NULL,\r
+           &gSimpleNetworkDriverBinding,\r
+           ImageHandle,\r
            &gSimpleNetworkComponentName,\r
            &gSimpleNetworkComponentName2\r
            );\r