[mirror_edk2.git] / NetworkPkg / SnpDxe / Snp.c

/** @file\r
  Implementation of driver entry point and driver binding protocol.\r
\r
Copyright (c) 2004 - 2019, Intel Corporation. All rights reserved.<BR>\r
Copyright (c) Microsoft Corporation.<BR>\r
SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "Snp.h"\r
\r
/**\r
  One notified function to stop UNDI device when gBS->ExitBootServices() called.\r
\r
  @param  Event                   Pointer to this event\r
  @param  Context                 Event handler private data\r
\r
**/\r
VOID\r
EFIAPI\r
SnpNotifyExitBootServices (\r
  EFI_EVENT  Event,\r
  VOID       *Context\r
  )\r
{\r
  SNP_DRIVER  *Snp;\r
\r
  Snp = (SNP_DRIVER *)Context;\r
\r
  //\r
  // Shutdown and stop UNDI driver\r
  //\r
  PxeShutdown (Snp);\r
  PxeStop (Snp);\r
}\r
\r
/**\r
  Send command to UNDI. It does nothing currently.\r
\r
  @param Cdb   command to be sent to UNDI.\r
\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
EFIAPI\r
IssueHwUndiCommand (\r
  UINT64  Cdb\r
  )\r
{\r
  DEBUG ((DEBUG_ERROR, "\nIssueHwUndiCommand() - This should not be called!"));\r
\r
  if (Cdb == 0) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  //  %%TBD - For now, nothing is done.\r
  //\r
  return EFI_UNSUPPORTED;\r
}\r
\r
/**\r
  Compute 8-bit checksum of a buffer.\r
\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, or zero if ptr is NULL or len\r
          is zero.\r
\r
**/\r
UINT8\r
Calc8BitCksum (\r
  VOID   *Buffer,\r
  UINTN  Length\r
  )\r
{\r
  UINT8  *Ptr;\r
  UINT8  Cksum;\r
\r
  Ptr   = Buffer;\r
  Cksum = 0;\r
\r
  if ((Ptr == NULL) || (Length == 0)) {\r
    return 0;\r
  }\r
\r
  while (Length-- != 0) {\r
    Cksum = (UINT8)(Cksum + *Ptr++);\r
  }\r
\r
  return Cksum;\r
}\r
\r
/**\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
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                                 Status;\r
  EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL  *NiiProtocol;\r
  PXE_UNDI                                   *Pxe;\r
\r
  Status = gBS->OpenProtocol (\r
                  Controller,\r
                  &gEfiDevicePathProtocolGuid,\r
                  NULL,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_TEST_PROTOCOL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  Controller,\r
                  &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
                  (VOID **)&NiiProtocol,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    if (Status == EFI_ALREADY_STARTED) {\r
      DEBUG ((DEBUG_INFO, "Support(): Already Started. on handle %p\n", Controller));\r
    }\r
\r
    return Status;\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "Support(): UNDI3.1 found on handle %p\n", Controller));\r
\r
  //\r
  // check the version, we don't want to connect to the undi16\r
  //\r
  if (NiiProtocol->Type != EfiNetworkInterfaceUndi) {\r
    Status = EFI_UNSUPPORTED;\r
    goto Done;\r
  }\r
\r
  //\r
  // Check to see if !PXE structure is valid. Paragraph alignment of !PXE structure is required.\r
  //\r
  if ((NiiProtocol->Id & 0x0F) != 0) {\r
    DEBUG ((DEBUG_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
\r
  //\r
  //  Verify !PXE revisions.\r
  //\r
  if (Pxe->hw.Signature != PXE_ROMID_SIGNATURE) {\r
    DEBUG ((DEBUG_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
    DEBUG ((DEBUG_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
    DEBUG ((DEBUG_NET, "\n!PXE.MajorVer is not supported.\n"));\r
    Status = EFI_UNSUPPORTED;\r
    goto Done;\r
  } else if ((Pxe->hw.MajorVer == PXE_ROMID_MAJORVER) && (Pxe->hw.MinorVer < PXE_ROMID_MINORVER)) {\r
    DEBUG ((DEBUG_NET, "\n!PXE.MinorVer is not supported."));\r
    Status = EFI_UNSUPPORTED;\r
    goto Done;\r
  }\r
\r
  //\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
      DEBUG ((DEBUG_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
      DEBUG ((DEBUG_NET, "\n!PXE.BusCnt is zero."));\r
      Status = EFI_UNSUPPORTED;\r
      goto Done;\r
    }\r
  }\r
\r
  Status = EFI_SUCCESS;\r
  DEBUG ((DEBUG_INFO, "Support(): supported on %p\n", Controller));\r
\r
Done:\r
  gBS->CloseProtocol (\r
         Controller,\r
         &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
         This->DriverBindingHandle,\r
         Controller\r
         );\r
\r
  return Status;\r
}\r
\r
/**\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  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 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
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_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                                       *Address;\r
  EFI_HANDLE                                 Handle;\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 ((DEBUG_NET, "\nSnpNotifyNetworkInterfaceIdentifier()  "));\r
\r
  Status = gBS->OpenProtocol (\r
                  Controller,\r
                  &gEfiDevicePathProtocolGuid,\r
                  (VOID **)&NiiDevicePath,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = gBS->LocateDevicePath (\r
                  &gEfiPciIoProtocolGuid,\r
                  &NiiDevicePath,\r
                  &Handle\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  Status = gBS->OpenProtocol (\r
                  Handle,\r
                  &gEfiPciIoProtocolGuid,\r
                  (VOID **)&PciIo,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Get the NII interface.\r
  //\r
  Status = gBS->OpenProtocol (\r
                  Controller,\r
                  &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
                  (VOID **)&Nii,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_BY_DRIVER\r
                  );\r
  if (EFI_ERROR (Status)) {\r
    gBS->CloseProtocol (\r
           Controller,\r
           &gEfiDevicePathProtocolGuid,\r
           This->DriverBindingHandle,\r
           Controller\r
           );\r
    return Status;\r
  }\r
\r
  DEBUG ((DEBUG_INFO, "Start(): UNDI3.1 found\n"));\r
\r
  Pxe = (PXE_UNDI *)(UINTN)(Nii->Id);\r
\r
  if (Calc8BitCksum (Pxe, Pxe->hw.Len) != 0) {\r
    DEBUG ((DEBUG_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
    //\r
    //  We can get any packets.\r
    //\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
    //  broadcast support or we cannot do DHCP!\r
    //\r
  } else {\r
    DEBUG ((DEBUG_NET, "\nUNDI does not have promiscuous or broadcast support."));\r
    goto NiiError;\r
  }\r
\r
  //\r
  // 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 = 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 ((DEBUG_NET, "\nCould not allocate SNP_DRIVER structure.\n"));\r
    goto NiiError;\r
  }\r
\r
  Snp = (SNP_DRIVER *)(UINTN)Address;\r
\r
  ZeroMem (Snp, sizeof (SNP_DRIVER));\r
\r
  Snp->PciIo     = PciIo;\r
  Snp->Signature = SNP_DRIVER_SIGNATURE;\r
\r
  EfiInitializeLock (&Snp->Lock, TPL_NOTIFY);\r
\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.Mode = &Snp->Mode;\r
\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
\r
  Snp->MaxRecycledTxBuf   = SNP_TX_BUFFER_INCREASEMENT;\r
  Snp->RecycledTxBufCount = 0;\r
\r
  if (Nii->Revision >= EFI_NETWORK_INTERFACE_IDENTIFIER_PROTOCOL_REVISION) {\r
    Snp->IfNum = 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->IsSwUndi           = FALSE;\r
    Snp->IssueUndi32Command = &IssueHwUndiCommand;\r
  } else {\r
    Snp->IsSwUndi = TRUE;\r
\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->IssueUndi32Command = (ISSUE_UNDI32_COMMAND)(UINTN)((UINT8)(UINTN)Pxe + Pxe->sw.EntryPoint);\r
    }\r
  }\r
\r
  //\r
  // Allocate a global CPB and DB buffer for this UNDI interface.\r
  // we do this because:\r
  //\r
  // -UNDI 3.0 wants all the addresses passed to it (even the cpb and db) to be\r
  // within 2GB limit, create them here and map them so that when undi calls\r
  // v2p callback to check if the physical address is < 2gb, we will pass.\r
  //\r
  // -This is not a requirement for 3.1 or later UNDIs but the code looks\r
  // simpler if we use the same cpb, db variables for both old and new undi\r
  // interfaces from all the SNP interface calls (we don't map the buffers\r
  // for the newer undi interfaces though)\r
  // .\r
  // -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 = 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 ((DEBUG_NET, "\nCould not allocate CPB and DB structures.\n"));\r
    goto Error_DeleteSNP;\r
  }\r
\r
  Snp->Cpb = (VOID *)(UINTN)Address;\r
  Snp->Db  = (VOID *)((UINTN)Address + 2048);\r
\r
  //\r
  // Find the correct BAR to do IO.\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
  Snp->MemoryBarIndex = PCI_MAX_BAR;\r
  Snp->IoBarIndex     = PCI_MAX_BAR;\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
    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
    FreePool (BarDesc);\r
\r
    if (FoundMemoryBar && FoundIoBar) {\r
      break;\r
    }\r
  }\r
\r
  Status = PxeStart (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 = (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
\r
  Snp->Cdb.IFnum   = Snp->IfNum;\r
  Snp->Cdb.Control = PXE_CONTROL_LAST_CDB_IN_LIST;\r
\r
  DEBUG ((DEBUG_NET, "\nSnp->undi.get_init_info()  "));\r
\r
  (*Snp->IssueUndi32Command)((UINT64)(UINTN)&Snp->Cdb);\r
\r
  //\r
  // Save the INIT Stat Code...\r
  //\r
  InitStatFlags = Snp->Cdb.StatFlags;\r
\r
  if (Snp->Cdb.StatCode != PXE_STATCODE_SUCCESS) {\r
    DEBUG ((DEBUG_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->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->CableDetectSupported = TRUE;\r
      break;\r
\r
    case PXE_STATFLAGS_CABLE_DETECT_NOT_SUPPORTED:\r
    default:\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
  } else {\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
  } else {\r
    Snp->Mode.MultipleTxSupported = FALSE;\r
  }\r
\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
  }\r
\r
  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_PROMISCUOUS_RX_SUPPORTED) != 0) {\r
    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_PROMISCUOUS;\r
  }\r
\r
  if ((Pxe->hw.Implementation & PXE_ROMID_IMP_BROADCAST_RX_SUPPORTED) != 0) {\r
    Snp->Mode.ReceiveFilterMask |= EFI_SIMPLE_NETWORK_RECEIVE_BROADCAST;\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
  }\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
  }\r
\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->TxRxBufferSize = Snp->InitInfo.MemoryRequired;\r
  Status              = PxeInit (Snp, PXE_OPFLAGS_INITIALIZE_DO_NOT_DETECT_CABLE);\r
\r
  if (EFI_ERROR (Status)) {\r
    PxeStop (Snp);\r
    goto Error_DeleteSNP;\r
  }\r
\r
  Status = PxeGetStnAddr (Snp);\r
\r
  if (Status != EFI_SUCCESS) {\r
    DEBUG ((DEBUG_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
\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
  // 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
  PxeShutdown (Snp);\r
  PxeStop (Snp);\r
\r
  if (PcdGetBool (PcdSnpCreateExitBootServicesEvent)) {\r
    //\r
    // Create EXIT_BOOT_SERIVES Event\r
    //\r
    Status = gBS->CreateEventEx (\r
                    EVT_NOTIFY_SIGNAL,\r
                    TPL_CALLBACK,\r
                    SnpNotifyExitBootServices,\r
                    Snp,\r
                    &gEfiEventExitBootServicesGuid,\r
                    &Snp->ExitBootServicesEvent\r
                    );\r
    if (EFI_ERROR (Status)) {\r
      goto Error_DeleteSNP;\r
    }\r
  }\r
\r
  //\r
  //  add SNP to the undi handle\r
  //\r
  Status = gBS->InstallProtocolInterface (\r
                  &Controller,\r
                  &gEfiSimpleNetworkProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  &(Snp->Snp)\r
                  );\r
\r
  if (!EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  PciIo->FreeBuffer (\r
           PciIo,\r
           SNP_MEM_PAGES (4096),\r
           Snp->Cpb\r
           );\r
\r
Error_DeleteSNP:\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
         &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
         This->DriverBindingHandle,\r
         Controller\r
         );\r
\r
  gBS->CloseProtocol (\r
         Controller,\r
         &gEfiDevicePathProtocolGuid,\r
         This->DriverBindingHandle,\r
         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
  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
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
  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
  //\r
  Status = gBS->OpenProtocol (\r
                  Controller,\r
                  &gEfiSimpleNetworkProtocolGuid,\r
                  (VOID **)&SnpProtocol,\r
                  This->DriverBindingHandle,\r
                  Controller,\r
                  EFI_OPEN_PROTOCOL_GET_PROTOCOL\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    return EFI_UNSUPPORTED;\r
  }\r
\r
  Snp = EFI_SIMPLE_NETWORK_DEV_FROM_THIS (SnpProtocol);\r
\r
  Status = gBS->UninstallProtocolInterface (\r
                  Controller,\r
                  &gEfiSimpleNetworkProtocolGuid,\r
                  &Snp->Snp\r
                  );\r
\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (PcdGetBool (PcdSnpCreateExitBootServicesEvent)) {\r
    //\r
    // Close EXIT_BOOT_SERVICES Event\r
    //\r
    gBS->CloseEvent (Snp->ExitBootServicesEvent);\r
  }\r
\r
  Status = gBS->CloseProtocol (\r
                  Controller,\r
                  &gEfiNetworkInterfaceIdentifierProtocolGuid_31,\r
                  This->DriverBindingHandle,\r
                  Controller\r
                  );\r
\r
  Status = gBS->CloseProtocol (\r
                  Controller,\r
                  &gEfiDevicePathProtocolGuid,\r
                  This->DriverBindingHandle,\r
                  Controller\r
                  );\r
\r
  PxeShutdown (Snp);\r
  PxeStop (Snp);\r
\r
  FreePool (Snp->RecycledTxBuf);\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
  The SNP driver entry point.\r
\r
  @param ImageHandle       The driver image handle.\r
  @param SystemTable       The system table.\r
\r
  @retval EFI_SUCCESS      Initialization routine has found UNDI hardware,\r
                           loaded it's ROM, and installed a notify event for\r
                           the Network Identifier Interface Protocol\r
                           successfully.\r
  @retval Other            Return value from HandleProtocol for\r
                           DeviceIoProtocol or LoadedImageProtocol\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
InitializeSnpNiiDriver (\r
  IN EFI_HANDLE        ImageHandle,\r
  IN EFI_SYSTEM_TABLE  *SystemTable\r
  )\r
{\r
  return EfiLibInstallDriverBindingComponentName2 (\r
           ImageHandle,\r
           SystemTable,\r
           &gSimpleNetworkDriverBinding,\r
           ImageHandle,\r
           &gSimpleNetworkComponentName,\r
           &gSimpleNetworkComponentName2\r
           );\r
}\r