2. apply for doxgen format for file header. git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@5043 6f19259b-4bc3-4df7-8a09-765794883524
		
			
				
	
	
		
			1413 lines
		
	
	
		
			31 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1413 lines
		
	
	
		
			31 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/** @file
 | 
						|
 | 
						|
Copyright (c) 2005 - 2007, Intel Corporation
 | 
						|
All rights reserved. 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
 | 
						|
http://opensource.org/licenses/bsd-license.php
 | 
						|
 | 
						|
THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 | 
						|
 | 
						|
Module Name:
 | 
						|
 | 
						|
  IpIo.c
 | 
						|
 | 
						|
Abstract:
 | 
						|
 | 
						|
  The implementation of the IpIo layer.
 | 
						|
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
#include <PiDxe.h>
 | 
						|
 | 
						|
#include <Protocol/Udp4.h>
 | 
						|
 | 
						|
#include <Library/IpIoLib.h>
 | 
						|
#include <Library/BaseLib.h>
 | 
						|
#include <Library/DebugLib.h>
 | 
						|
#include <Library/UefiBootServicesTableLib.h>
 | 
						|
#include <Library/MemoryAllocationLib.h>
 | 
						|
#include <Library/BaseMemoryLib.h>
 | 
						|
 | 
						|
 | 
						|
#define NET_PROTO_HDR(Buf, Type)  ((Type *) ((Buf)->BlockOp[0].Head))
 | 
						|
#define ICMP_ERRLEN(IpHdr) \
 | 
						|
  (sizeof(IP4_ICMP_HEAD) + EFI_IP4_HEADER_LEN(IpHdr) + 8)
 | 
						|
 | 
						|
LIST_ENTRY  mActiveIpIoList = {
 | 
						|
  &mActiveIpIoList,
 | 
						|
  &mActiveIpIoList
 | 
						|
};
 | 
						|
 | 
						|
EFI_IP4_CONFIG_DATA  mIpIoDefaultIpConfigData = {
 | 
						|
  EFI_IP_PROTO_UDP,
 | 
						|
  FALSE,
 | 
						|
  TRUE,
 | 
						|
  FALSE,
 | 
						|
  FALSE,
 | 
						|
  FALSE,
 | 
						|
  {{0, 0, 0, 0}},
 | 
						|
  {{0, 0, 0, 0}},
 | 
						|
  0,
 | 
						|
  255,
 | 
						|
  FALSE,
 | 
						|
  FALSE,
 | 
						|
  0,
 | 
						|
  0
 | 
						|
};
 | 
						|
 | 
						|
STATIC ICMP_ERROR_INFO  mIcmpErrMap[10] = {
 | 
						|
  {FALSE, TRUE},
 | 
						|
  {FALSE, TRUE},
 | 
						|
  {TRUE, TRUE},
 | 
						|
  {TRUE, TRUE},
 | 
						|
  {TRUE, TRUE},
 | 
						|
  {FALSE, TRUE},
 | 
						|
  {FALSE, TRUE},
 | 
						|
  {FALSE, TRUE},
 | 
						|
  {FALSE, FALSE},
 | 
						|
  {FALSE, TRUE}
 | 
						|
};
 | 
						|
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoTransmitHandlerDpc (
 | 
						|
  IN VOID      *Context
 | 
						|
  );
 | 
						|
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoTransmitHandler (
 | 
						|
  IN EFI_EVENT Event,
 | 
						|
  IN VOID      *Context
 | 
						|
  );
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function create an ip child ,open the IP protocol, return the opened
 | 
						|
  Ip protocol to Interface.
 | 
						|
 | 
						|
  @param  ControllerHandle      The controller handle.
 | 
						|
  @param  ImageHandle           The image handle.
 | 
						|
  @param  ChildHandle           Pointer to the buffer to save the ip child handle.
 | 
						|
  @param  Interface             Pointer used to get the ip protocol interface.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The ip child is created and the ip protocol
 | 
						|
                                interface is retrieved.
 | 
						|
  @retval other                 The required operation failed.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
EFI_STATUS
 | 
						|
IpIoCreateIpChildOpenProtocol (
 | 
						|
  IN  EFI_HANDLE  ControllerHandle,
 | 
						|
  IN  EFI_HANDLE  ImageHandle,
 | 
						|
  IN  EFI_HANDLE  *ChildHandle,
 | 
						|
  OUT VOID        **Interface
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
 | 
						|
  //
 | 
						|
  // Create an ip child.
 | 
						|
  //
 | 
						|
  Status = NetLibCreateServiceChild (
 | 
						|
             ControllerHandle,
 | 
						|
             ImageHandle,
 | 
						|
             &gEfiIp4ServiceBindingProtocolGuid,
 | 
						|
             ChildHandle
 | 
						|
             );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Open the ip protocol installed on the *ChildHandle.
 | 
						|
  //
 | 
						|
  Status = gBS->OpenProtocol (
 | 
						|
                  *ChildHandle,
 | 
						|
                  &gEfiIp4ProtocolGuid,
 | 
						|
                  Interface,
 | 
						|
                  ImageHandle,
 | 
						|
                  ControllerHandle,
 | 
						|
                  EFI_OPEN_PROTOCOL_BY_DRIVER
 | 
						|
                  );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    //
 | 
						|
    // On failure, destroy the ip child.
 | 
						|
    //
 | 
						|
    NetLibDestroyServiceChild (
 | 
						|
      ControllerHandle,
 | 
						|
      ImageHandle,
 | 
						|
      &gEfiIp4ServiceBindingProtocolGuid,
 | 
						|
      *ChildHandle
 | 
						|
      );
 | 
						|
  }
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  This function close the previously openned ip protocol and destroy the ip child.
 | 
						|
 | 
						|
  @param  ControllerHandle      The controller handle.
 | 
						|
  @param  ImageHandle           the image handle.
 | 
						|
  @param  ChildHandle           The child handle of the ip child.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The ip protocol is closed and the relevant ip child
 | 
						|
                                is destroyed.
 | 
						|
  @retval other                 The required operation failed.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
EFI_STATUS
 | 
						|
IpIoCloseProtocolDestroyIpChild (
 | 
						|
  IN EFI_HANDLE  ControllerHandle,
 | 
						|
  IN EFI_HANDLE  ImageHandle,
 | 
						|
  IN EFI_HANDLE  ChildHandle
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
 | 
						|
  //
 | 
						|
  // Close the previously openned ip protocol.
 | 
						|
  //
 | 
						|
  gBS->CloseProtocol (
 | 
						|
         ChildHandle,
 | 
						|
         &gEfiIp4ProtocolGuid,
 | 
						|
         ImageHandle,
 | 
						|
         ControllerHandle
 | 
						|
         );
 | 
						|
 | 
						|
  //
 | 
						|
  // Destroy the ip child.
 | 
						|
  //
 | 
						|
  Status = NetLibDestroyServiceChild (
 | 
						|
             ControllerHandle,
 | 
						|
             ImageHandle,
 | 
						|
             &gEfiIp4ServiceBindingProtocolGuid,
 | 
						|
             ChildHandle
 | 
						|
             );
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Handle ICMP packets.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance.
 | 
						|
  @param  Pkt                   Pointer to the ICMP packet.
 | 
						|
  @param  Session               Pointer to the net session of this ICMP packet.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The ICMP packet is handled successfully.
 | 
						|
  @retval EFI_ABORTED           This type of ICMP packet is not supported.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
EFI_STATUS
 | 
						|
IpIoIcmpHandler (
 | 
						|
  IN IP_IO                *IpIo,
 | 
						|
  IN NET_BUF              *Pkt,
 | 
						|
  IN EFI_NET_SESSION_DATA *Session
 | 
						|
  )
 | 
						|
{
 | 
						|
  IP4_ICMP_ERROR_HEAD  *IcmpHdr;
 | 
						|
  EFI_IP4_HEADER       *IpHdr;
 | 
						|
  ICMP_ERROR           IcmpErr;
 | 
						|
  UINT8                *PayLoadHdr;
 | 
						|
  UINT8                Type;
 | 
						|
  UINT8                Code;
 | 
						|
  UINT32               TrimBytes;
 | 
						|
 | 
						|
  IcmpHdr = NET_PROTO_HDR (Pkt, IP4_ICMP_ERROR_HEAD);
 | 
						|
  IpHdr   = (EFI_IP4_HEADER *) (&IcmpHdr->IpHead);
 | 
						|
 | 
						|
  //
 | 
						|
  // Check the ICMP packet length.
 | 
						|
  //
 | 
						|
  if (Pkt->TotalSize < ICMP_ERRLEN (IpHdr)) {
 | 
						|
 | 
						|
    return EFI_ABORTED;
 | 
						|
  }
 | 
						|
 | 
						|
  Type = IcmpHdr->Head.Type;
 | 
						|
  Code = IcmpHdr->Head.Code;
 | 
						|
 | 
						|
  //
 | 
						|
  // Analyze the ICMP Error in this ICMP pkt
 | 
						|
  //
 | 
						|
  switch (Type) {
 | 
						|
  case ICMP_TYPE_UNREACH:
 | 
						|
    switch (Code) {
 | 
						|
    case ICMP_CODE_UNREACH_NET:
 | 
						|
    case ICMP_CODE_UNREACH_HOST:
 | 
						|
    case ICMP_CODE_UNREACH_PROTOCOL:
 | 
						|
    case ICMP_CODE_UNREACH_PORT:
 | 
						|
    case ICMP_CODE_UNREACH_SRCFAIL:
 | 
						|
      IcmpErr = (ICMP_ERROR) (ICMP_ERR_UNREACH_NET + Code);
 | 
						|
 | 
						|
      break;
 | 
						|
 | 
						|
    case ICMP_CODE_UNREACH_NEEDFRAG:
 | 
						|
      IcmpErr = ICMP_ERR_MSGSIZE;
 | 
						|
 | 
						|
      break;
 | 
						|
 | 
						|
    case ICMP_CODE_UNREACH_NET_UNKNOWN:
 | 
						|
    case ICMP_CODE_UNREACH_NET_PROHIB:
 | 
						|
    case ICMP_CODE_UNREACH_TOSNET:
 | 
						|
      IcmpErr = ICMP_ERR_UNREACH_NET;
 | 
						|
 | 
						|
      break;
 | 
						|
 | 
						|
    case ICMP_CODE_UNREACH_HOST_UNKNOWN:
 | 
						|
    case ICMP_CODE_UNREACH_ISOLATED:
 | 
						|
    case ICMP_CODE_UNREACH_HOST_PROHIB:
 | 
						|
    case ICMP_CODE_UNREACH_TOSHOST:
 | 
						|
      IcmpErr = ICMP_ERR_UNREACH_HOST;
 | 
						|
 | 
						|
      break;
 | 
						|
 | 
						|
    default:
 | 
						|
      return EFI_ABORTED;
 | 
						|
 | 
						|
      break;
 | 
						|
    }
 | 
						|
 | 
						|
    break;
 | 
						|
 | 
						|
  case ICMP_TYPE_TIMXCEED:
 | 
						|
    if (Code > 1) {
 | 
						|
      return EFI_ABORTED;
 | 
						|
    }
 | 
						|
 | 
						|
    IcmpErr = (ICMP_ERROR) (Code + ICMP_ERR_TIMXCEED_INTRANS);
 | 
						|
 | 
						|
    break;
 | 
						|
 | 
						|
  case ICMP_TYPE_PARAMPROB:
 | 
						|
    if (Code > 1) {
 | 
						|
      return EFI_ABORTED;
 | 
						|
    }
 | 
						|
 | 
						|
    IcmpErr = ICMP_ERR_PARAMPROB;
 | 
						|
 | 
						|
    break;
 | 
						|
 | 
						|
  case ICMP_TYPE_SOURCEQUENCH:
 | 
						|
    if (Code != 0) {
 | 
						|
      return EFI_ABORTED;
 | 
						|
    }
 | 
						|
 | 
						|
    IcmpErr = ICMP_ERR_QUENCH;
 | 
						|
 | 
						|
    break;
 | 
						|
 | 
						|
  default:
 | 
						|
    return EFI_ABORTED;
 | 
						|
 | 
						|
    break;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Notify user the ICMP pkt only containing payload except
 | 
						|
  // IP and ICMP header
 | 
						|
  //
 | 
						|
  PayLoadHdr = (UINT8 *) ((UINT8 *) IpHdr + EFI_IP4_HEADER_LEN (IpHdr));
 | 
						|
  TrimBytes  = (UINT32) (PayLoadHdr - (UINT8 *) IcmpHdr);
 | 
						|
 | 
						|
  NetbufTrim (Pkt, TrimBytes, TRUE);
 | 
						|
 | 
						|
  IpIo->PktRcvdNotify (EFI_ICMP_ERROR, IcmpErr, Session, Pkt, IpIo->RcvdContext);
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Ext free function for net buffer. This function is
 | 
						|
  called when the net buffer is freed. It is used to
 | 
						|
  signal the recycle event to notify IP to recycle the
 | 
						|
  data buffer.
 | 
						|
 | 
						|
  @param  Event                 The event to be signaled.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
IpIoExtFree (
 | 
						|
  IN VOID  *Event
 | 
						|
  )
 | 
						|
{
 | 
						|
  gBS->SignalEvent ((EFI_EVENT) Event);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Create a send entry to wrap a packet before sending
 | 
						|
  out it through IP.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance.
 | 
						|
  @param  Pkt                   Pointer to the packet.
 | 
						|
  @param  Sender                Pointer to the IP sender.
 | 
						|
  @param  NotifyData            Pointer to the notify data.
 | 
						|
  @param  Dest                  Pointer to the destination IP address.
 | 
						|
  @param  Override              Pointer to the overriden IP_IO data.
 | 
						|
 | 
						|
  @return Pointer to the data structure created to wrap the packet. If NULL,
 | 
						|
  @return resource limit occurred.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
IP_IO_SEND_ENTRY *
 | 
						|
IpIoCreateSndEntry (
 | 
						|
  IN IP_IO             *IpIo,
 | 
						|
  IN NET_BUF           *Pkt,
 | 
						|
  IN EFI_IP4_PROTOCOL  *Sender,
 | 
						|
  IN VOID              *Context    OPTIONAL,
 | 
						|
  IN VOID              *NotifyData OPTIONAL,
 | 
						|
  IN IP4_ADDR          Dest,
 | 
						|
  IN IP_IO_OVERRIDE    *Override
 | 
						|
  )
 | 
						|
{
 | 
						|
  IP_IO_SEND_ENTRY          *SndEntry;
 | 
						|
  EFI_IP4_COMPLETION_TOKEN  *SndToken;
 | 
						|
  EFI_IP4_TRANSMIT_DATA     *TxData;
 | 
						|
  EFI_STATUS                Status;
 | 
						|
  EFI_IP4_OVERRIDE_DATA     *OverrideData;
 | 
						|
  volatile UINT32           Index;
 | 
						|
 | 
						|
  //
 | 
						|
  // Allocate resource for SndEntry
 | 
						|
  //
 | 
						|
  SndEntry = AllocatePool (sizeof (IP_IO_SEND_ENTRY));
 | 
						|
  if (NULL == SndEntry) {
 | 
						|
    return NULL;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Allocate resource for SndToken
 | 
						|
  //
 | 
						|
  SndToken = AllocatePool (sizeof (EFI_IP4_COMPLETION_TOKEN));
 | 
						|
  if (NULL == SndToken) {
 | 
						|
    goto ReleaseSndEntry;
 | 
						|
  }
 | 
						|
 | 
						|
  Status = gBS->CreateEvent (
 | 
						|
                  EVT_NOTIFY_SIGNAL,
 | 
						|
                  TPL_NOTIFY,
 | 
						|
                  IpIoTransmitHandler,
 | 
						|
                  SndEntry,
 | 
						|
                  &(SndToken->Event)
 | 
						|
                  );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto ReleaseSndToken;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Allocate resource for TxData
 | 
						|
  //
 | 
						|
  TxData = AllocatePool (
 | 
						|
    sizeof (EFI_IP4_TRANSMIT_DATA) +
 | 
						|
    sizeof (EFI_IP4_FRAGMENT_DATA) * (Pkt->BlockOpNum - 1)
 | 
						|
    );
 | 
						|
 | 
						|
  if (NULL == TxData) {
 | 
						|
    goto ReleaseEvent;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Allocate resource for OverrideData if needed
 | 
						|
  //
 | 
						|
  OverrideData = NULL;
 | 
						|
  if (NULL != Override) {
 | 
						|
 | 
						|
    OverrideData = AllocatePool (sizeof (EFI_IP4_OVERRIDE_DATA));
 | 
						|
    if (NULL == OverrideData) {
 | 
						|
      goto ReleaseResource;
 | 
						|
    }
 | 
						|
    //
 | 
						|
    // Set the fields of OverrideData
 | 
						|
    //
 | 
						|
    CopyMem (OverrideData, Override, sizeof (*OverrideData));
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Set the fields of TxData
 | 
						|
  //
 | 
						|
  CopyMem (&TxData->DestinationAddress, &Dest, sizeof (EFI_IPv4_ADDRESS));
 | 
						|
  TxData->OverrideData                  = OverrideData;
 | 
						|
  TxData->OptionsLength                 = 0;
 | 
						|
  TxData->OptionsBuffer                 = NULL;
 | 
						|
  TxData->TotalDataLength               = Pkt->TotalSize;
 | 
						|
  TxData->FragmentCount                 = Pkt->BlockOpNum;
 | 
						|
 | 
						|
 | 
						|
  for (Index = 0; Index < Pkt->BlockOpNum; Index++) {
 | 
						|
    TxData->FragmentTable[Index].FragmentBuffer = Pkt->BlockOp[Index].Head;
 | 
						|
    TxData->FragmentTable[Index].FragmentLength = Pkt->BlockOp[Index].Size;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Set the fields of SndToken
 | 
						|
  //
 | 
						|
  SndToken->Packet.TxData = TxData;
 | 
						|
 | 
						|
  //
 | 
						|
  // Set the fields of SndEntry
 | 
						|
  //
 | 
						|
  SndEntry->IpIo        = IpIo;
 | 
						|
  SndEntry->Ip      = Sender;
 | 
						|
  SndEntry->Context     = Context;
 | 
						|
  SndEntry->NotifyData  = NotifyData;
 | 
						|
 | 
						|
  SndEntry->Pkt         = Pkt;
 | 
						|
  NET_GET_REF (Pkt);
 | 
						|
 | 
						|
  SndEntry->SndToken = SndToken;
 | 
						|
 | 
						|
  InsertTailList (&IpIo->PendingSndList, &SndEntry->Entry);
 | 
						|
 | 
						|
  return SndEntry;
 | 
						|
 | 
						|
ReleaseResource:
 | 
						|
  gBS->FreePool (TxData);
 | 
						|
 | 
						|
ReleaseEvent:
 | 
						|
  gBS->CloseEvent (SndToken->Event);
 | 
						|
 | 
						|
ReleaseSndToken:
 | 
						|
  gBS->FreePool (SndToken);
 | 
						|
 | 
						|
ReleaseSndEntry:
 | 
						|
  gBS->FreePool (SndEntry);
 | 
						|
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Destroy the SndEntry.
 | 
						|
 | 
						|
  @param  SndEntry              Pointer to the send entry to be destroyed.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
IpIoDestroySndEntry (
 | 
						|
  IN IP_IO_SEND_ENTRY  *SndEntry
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_IP4_TRANSMIT_DATA  *TxData;
 | 
						|
 | 
						|
  TxData = SndEntry->SndToken->Packet.TxData;
 | 
						|
 | 
						|
  if (NULL != TxData->OverrideData) {
 | 
						|
    gBS->FreePool (TxData->OverrideData);
 | 
						|
  }
 | 
						|
 | 
						|
  gBS->FreePool (TxData);
 | 
						|
  NetbufFree (SndEntry->Pkt);
 | 
						|
  gBS->CloseEvent (SndEntry->SndToken->Event);
 | 
						|
 | 
						|
  gBS->FreePool (SndEntry->SndToken);
 | 
						|
  RemoveEntryList (&SndEntry->Entry);
 | 
						|
 | 
						|
  gBS->FreePool (SndEntry);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Notify function for IP transmit token.
 | 
						|
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoTransmitHandlerDpc (
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  IP_IO             *IpIo;
 | 
						|
  IP_IO_SEND_ENTRY  *SndEntry;
 | 
						|
 | 
						|
  SndEntry  = (IP_IO_SEND_ENTRY *) Context;
 | 
						|
 | 
						|
  IpIo      = SndEntry->IpIo;
 | 
						|
 | 
						|
  if (IpIo->PktSentNotify && SndEntry->NotifyData) {
 | 
						|
    IpIo->PktSentNotify (
 | 
						|
            SndEntry->SndToken->Status,
 | 
						|
            SndEntry->Context,
 | 
						|
            SndEntry->Ip,
 | 
						|
            SndEntry->NotifyData
 | 
						|
            );
 | 
						|
  }
 | 
						|
 | 
						|
  IpIoDestroySndEntry (SndEntry);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
  Notify function for IP transmit token.
 | 
						|
 | 
						|
  @param  Event                 The event signaled.
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoTransmitHandler (
 | 
						|
  IN EFI_EVENT Event,
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  //
 | 
						|
  // Request IpIoTransmitHandlerDpc as a DPC at TPL_CALLBACK
 | 
						|
  //
 | 
						|
  NetLibQueueDpc (TPL_CALLBACK, IpIoTransmitHandlerDpc, Context);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  The dummy handler for the dummy IP receive token.
 | 
						|
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoDummyHandlerDpc (
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  IP_IO_IP_INFO             *IpInfo;
 | 
						|
  EFI_IP4_COMPLETION_TOKEN  *DummyToken;
 | 
						|
 | 
						|
  IpInfo      = (IP_IO_IP_INFO *) Context;
 | 
						|
  DummyToken  = &(IpInfo->DummyRcvToken);
 | 
						|
 | 
						|
  if (EFI_ABORTED == DummyToken->Status) {
 | 
						|
    //
 | 
						|
    // The reception is actively aborted by the consumer, directly return.
 | 
						|
    //
 | 
						|
    return;
 | 
						|
  } else if (EFI_SUCCESS == DummyToken->Status) {
 | 
						|
    ASSERT (DummyToken->Packet.RxData);
 | 
						|
 | 
						|
    gBS->SignalEvent (DummyToken->Packet.RxData->RecycleSignal);
 | 
						|
  }
 | 
						|
 | 
						|
  IpInfo->Ip->Receive (IpInfo->Ip, DummyToken);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Request IpIoDummyHandlerDpc as a DPC at TPL_CALLBACK.
 | 
						|
 | 
						|
  @param  Event                 The event signaled.
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoDummyHandler (
 | 
						|
  IN EFI_EVENT Event,
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  //
 | 
						|
  // Request IpIoDummyHandlerDpc as a DPC at TPL_CALLBACK
 | 
						|
  //
 | 
						|
  NetLibQueueDpc (TPL_CALLBACK, IpIoDummyHandlerDpc, Context);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Notify function for the IP receive token, used to process
 | 
						|
  the received IP packets.
 | 
						|
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoListenHandlerDpc (
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  IP_IO                 *IpIo;
 | 
						|
  EFI_STATUS            Status;
 | 
						|
  EFI_IP4_RECEIVE_DATA  *RxData;
 | 
						|
  EFI_IP4_PROTOCOL      *Ip;
 | 
						|
  EFI_NET_SESSION_DATA  Session;
 | 
						|
  NET_BUF               *Pkt;
 | 
						|
 | 
						|
  IpIo    = (IP_IO *) Context;
 | 
						|
 | 
						|
  Ip      = IpIo->Ip;
 | 
						|
  Status  = IpIo->RcvToken.Status;
 | 
						|
  RxData  = IpIo->RcvToken.Packet.RxData;
 | 
						|
 | 
						|
  if (EFI_ABORTED == Status) {
 | 
						|
    //
 | 
						|
    // The reception is actively aborted by the consumer, directly return.
 | 
						|
    //
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  if (((EFI_SUCCESS != Status) && (EFI_ICMP_ERROR != Status)) || (NULL == RxData)) {
 | 
						|
    //
 | 
						|
    // Only process the normal packets and the icmp error packets, if RxData is NULL
 | 
						|
    // with Status == EFI_SUCCESS or EFI_ICMP_ERROR, just resume the receive although
 | 
						|
    // this should be a bug of the low layer (IP).
 | 
						|
    //
 | 
						|
    goto Resume;
 | 
						|
  }
 | 
						|
 | 
						|
  if (NULL == IpIo->PktRcvdNotify) {
 | 
						|
    goto CleanUp;
 | 
						|
  }
 | 
						|
 | 
						|
  if ((EFI_IP4 (RxData->Header->SourceAddress) != 0) &&
 | 
						|
    !Ip4IsUnicast (EFI_NTOHL (RxData->Header->SourceAddress), 0)) {
 | 
						|
    //
 | 
						|
    // The source address is not zero and it's not a unicast IP address, discard it.
 | 
						|
    //
 | 
						|
    goto CleanUp;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Create a netbuffer representing packet
 | 
						|
  //
 | 
						|
  Pkt = NetbufFromExt (
 | 
						|
          (NET_FRAGMENT *) RxData->FragmentTable,
 | 
						|
          RxData->FragmentCount,
 | 
						|
          0,
 | 
						|
          0,
 | 
						|
          IpIoExtFree,
 | 
						|
          RxData->RecycleSignal
 | 
						|
          );
 | 
						|
  if (NULL == Pkt) {
 | 
						|
    goto CleanUp;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Create a net session
 | 
						|
  //
 | 
						|
  Session.Source = EFI_IP4 (RxData->Header->SourceAddress);
 | 
						|
  Session.Dest   = EFI_IP4 (RxData->Header->DestinationAddress);
 | 
						|
  Session.IpHdr  = RxData->Header;
 | 
						|
 | 
						|
  if (EFI_SUCCESS == Status) {
 | 
						|
 | 
						|
    IpIo->PktRcvdNotify (EFI_SUCCESS, (ICMP_ERROR) 0, &Session, Pkt, IpIo->RcvdContext);
 | 
						|
  } else {
 | 
						|
    //
 | 
						|
    // Status is EFI_ICMP_ERROR
 | 
						|
    //
 | 
						|
    Status = IpIoIcmpHandler (IpIo, Pkt, &Session);
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      NetbufFree (Pkt);
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  goto Resume;
 | 
						|
 | 
						|
CleanUp:
 | 
						|
  gBS->SignalEvent (RxData->RecycleSignal);
 | 
						|
 | 
						|
Resume:
 | 
						|
  Ip->Receive (Ip, &(IpIo->RcvToken));
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Request IpIoListenHandlerDpc as a DPC at TPL_CALLBACK
 | 
						|
 | 
						|
  @param  Event                 The event signaled.
 | 
						|
  @param  Context               The context passed in by the event notifier.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
STATIC
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoListenHandler (
 | 
						|
  IN EFI_EVENT Event,
 | 
						|
  IN VOID      *Context
 | 
						|
  )
 | 
						|
{
 | 
						|
  //
 | 
						|
  // Request IpIoListenHandlerDpc as a DPC at TPL_CALLBACK
 | 
						|
  //
 | 
						|
  NetLibQueueDpc (TPL_CALLBACK, IpIoListenHandlerDpc, Context);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Create a new IP_IO instance.
 | 
						|
 | 
						|
  @param  Image                 The image handle of an IP_IO consumer protocol.
 | 
						|
  @param  Controller            The controller handle of an IP_IO consumer protocol
 | 
						|
                                installed on.
 | 
						|
 | 
						|
  @return Pointer to a newly created IP_IO instance.
 | 
						|
 | 
						|
**/
 | 
						|
IP_IO *
 | 
						|
EFIAPI
 | 
						|
IpIoCreate (
 | 
						|
  IN EFI_HANDLE Image,
 | 
						|
  IN EFI_HANDLE Controller
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS  Status;
 | 
						|
  IP_IO       *IpIo;
 | 
						|
 | 
						|
  IpIo = AllocateZeroPool (sizeof (IP_IO));
 | 
						|
  if (NULL == IpIo) {
 | 
						|
    return NULL;
 | 
						|
  }
 | 
						|
 | 
						|
  InitializeListHead (&(IpIo->PendingSndList));
 | 
						|
  InitializeListHead (&(IpIo->IpList));
 | 
						|
  IpIo->Controller  = Controller;
 | 
						|
  IpIo->Image       = Image;
 | 
						|
 | 
						|
  Status = gBS->CreateEvent (
 | 
						|
                  EVT_NOTIFY_SIGNAL,
 | 
						|
                  TPL_NOTIFY,
 | 
						|
                  IpIoListenHandler,
 | 
						|
                  IpIo,
 | 
						|
                  &(IpIo->RcvToken.Event)
 | 
						|
                  );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto ReleaseIpIo;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Create an IP child and open IP protocol
 | 
						|
  //
 | 
						|
  Status = IpIoCreateIpChildOpenProtocol (
 | 
						|
             Controller,
 | 
						|
             Image,
 | 
						|
             &IpIo->ChildHandle,
 | 
						|
             (VOID **)&(IpIo->Ip)
 | 
						|
             );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto ReleaseIpIo;
 | 
						|
  }
 | 
						|
 | 
						|
  return IpIo;
 | 
						|
 | 
						|
ReleaseIpIo:
 | 
						|
 | 
						|
  if (NULL != IpIo->RcvToken.Event) {
 | 
						|
    gBS->CloseEvent (IpIo->RcvToken.Event);
 | 
						|
  }
 | 
						|
 | 
						|
  gBS->FreePool (IpIo);
 | 
						|
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Open an IP_IO instance for use.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to an IP_IO instance that needs to open.
 | 
						|
  @param  OpenData              The configuration data for the IP_IO instance.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The IP_IO instance opened with OpenData
 | 
						|
                                successfully.
 | 
						|
  @retval other                 Error condition occurred.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
IpIoOpen (
 | 
						|
  IN IP_IO           *IpIo,
 | 
						|
  IN IP_IO_OPEN_DATA *OpenData
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS        Status;
 | 
						|
  EFI_IP4_PROTOCOL  *Ip;
 | 
						|
 | 
						|
  if (IpIo->IsConfigured) {
 | 
						|
    return EFI_ACCESS_DENIED;
 | 
						|
  }
 | 
						|
 | 
						|
  Ip = IpIo->Ip;
 | 
						|
 | 
						|
  //
 | 
						|
  // configure ip
 | 
						|
  //
 | 
						|
  Status = Ip->Configure (Ip, &OpenData->IpConfigData);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // bugbug: to delete the default route entry in this Ip, if it is:
 | 
						|
  // (0.0.0.0, 0.0.0.0, 0.0.0.0). Delete this statement if Ip modified
 | 
						|
  // its code
 | 
						|
  //
 | 
						|
  Status = Ip->Routes (Ip, TRUE, &mZeroIp4Addr, &mZeroIp4Addr, &mZeroIp4Addr);
 | 
						|
 | 
						|
  if (EFI_ERROR (Status) && (EFI_NOT_FOUND != Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  IpIo->PktRcvdNotify = OpenData->PktRcvdNotify;
 | 
						|
  IpIo->PktSentNotify = OpenData->PktSentNotify;
 | 
						|
 | 
						|
  IpIo->RcvdContext   = OpenData->RcvdContext;
 | 
						|
  IpIo->SndContext    = OpenData->SndContext;
 | 
						|
 | 
						|
  IpIo->Protocol      = OpenData->IpConfigData.DefaultProtocol;
 | 
						|
 | 
						|
  //
 | 
						|
  // start to listen incoming packet
 | 
						|
  //
 | 
						|
  Status = Ip->Receive (Ip, &(IpIo->RcvToken));
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    Ip->Configure (Ip, NULL);
 | 
						|
    goto ErrorExit;
 | 
						|
  }
 | 
						|
 | 
						|
  IpIo->IsConfigured = TRUE;
 | 
						|
  InsertTailList (&mActiveIpIoList, &IpIo->Entry);
 | 
						|
 | 
						|
ErrorExit:
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Stop an IP_IO instance.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance that needs to stop.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The IP_IO instance stopped successfully.
 | 
						|
  @retval other                 Error condition occurred.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
IpIoStop (
 | 
						|
  IN IP_IO *IpIo
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS        Status;
 | 
						|
  EFI_IP4_PROTOCOL  *Ip;
 | 
						|
  IP_IO_IP_INFO     *IpInfo;
 | 
						|
 | 
						|
  if (!IpIo->IsConfigured) {
 | 
						|
    return EFI_SUCCESS;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Remove the IpIo from the active IpIo list.
 | 
						|
  //
 | 
						|
  RemoveEntryList (&IpIo->Entry);
 | 
						|
 | 
						|
  Ip = IpIo->Ip;
 | 
						|
 | 
						|
  //
 | 
						|
  // Configure NULL Ip
 | 
						|
  //
 | 
						|
  Status = Ip->Configure (Ip, NULL);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    return Status;
 | 
						|
  }
 | 
						|
 | 
						|
  IpIo->IsConfigured = FALSE;
 | 
						|
 | 
						|
  //
 | 
						|
  // Detroy the Ip List used by IpIo
 | 
						|
  //
 | 
						|
 | 
						|
  while (!IsListEmpty (&(IpIo->IpList))) {
 | 
						|
    IpInfo = NET_LIST_HEAD (&(IpIo->IpList), IP_IO_IP_INFO, Entry);
 | 
						|
 | 
						|
    IpIoRemoveIp (IpIo, IpInfo);
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // All pending snd tokens should be flushed by reseting the IP instances.
 | 
						|
  //
 | 
						|
  ASSERT (IsListEmpty (&IpIo->PendingSndList));
 | 
						|
 | 
						|
  //
 | 
						|
  // Close the receive event.
 | 
						|
  //
 | 
						|
  gBS->CloseEvent (IpIo->RcvToken.Event);
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Destroy an IP_IO instance.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance that needs to
 | 
						|
                                destroy.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The IP_IO instance destroyed successfully.
 | 
						|
  @retval other                 Error condition occurred.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
IpIoDestroy (
 | 
						|
  IN IP_IO *IpIo
 | 
						|
  )
 | 
						|
{
 | 
						|
  //
 | 
						|
  // Stop the IpIo.
 | 
						|
  //
 | 
						|
  IpIoStop (IpIo);
 | 
						|
 | 
						|
  //
 | 
						|
  // Close the IP protocol and destroy the child.
 | 
						|
  //
 | 
						|
  IpIoCloseProtocolDestroyIpChild (IpIo->Controller, IpIo->Image, IpIo->ChildHandle);
 | 
						|
 | 
						|
  gBS->FreePool (IpIo);
 | 
						|
 | 
						|
  return EFI_SUCCESS;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Send out an IP packet.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to an IP_IO instance used for sending IP
 | 
						|
                                packet.
 | 
						|
  @param  Pkt                   Pointer to the IP packet to be sent.
 | 
						|
  @param  Sender                The IP protocol instance used for sending.
 | 
						|
  @param  NotifyData
 | 
						|
  @param  Dest                  The destination IP address to send this packet to.
 | 
						|
  @param  OverrideData          The data to override some configuration of the IP
 | 
						|
                                instance used for sending.
 | 
						|
 | 
						|
  @retval EFI_SUCCESS           The operation is completed successfully.
 | 
						|
  @retval EFI_NOT_STARTED       The IpIo is not configured.
 | 
						|
  @retval EFI_OUT_OF_RESOURCES  Failed due to resource limit.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
IpIoSend (
 | 
						|
  IN IP_IO           *IpIo,
 | 
						|
  IN NET_BUF         *Pkt,
 | 
						|
  IN IP_IO_IP_INFO   *Sender,
 | 
						|
  IN VOID            *Context    OPTIONAL,
 | 
						|
  IN VOID            *NotifyData OPTIONAL,
 | 
						|
  IN IP4_ADDR        Dest,
 | 
						|
  IN IP_IO_OVERRIDE  *OverrideData
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS        Status;
 | 
						|
  EFI_IP4_PROTOCOL  *Ip;
 | 
						|
  IP_IO_SEND_ENTRY  *SndEntry;
 | 
						|
 | 
						|
  if (!IpIo->IsConfigured) {
 | 
						|
    return EFI_NOT_STARTED;
 | 
						|
  }
 | 
						|
 | 
						|
  Ip = (NULL == Sender) ? IpIo->Ip : Sender->Ip;
 | 
						|
 | 
						|
  //
 | 
						|
  // create a new SndEntry
 | 
						|
  //
 | 
						|
  SndEntry = IpIoCreateSndEntry (IpIo, Pkt, Ip, Context, NotifyData, Dest, OverrideData);
 | 
						|
  if (NULL == SndEntry) {
 | 
						|
    return EFI_OUT_OF_RESOURCES;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Send this Packet
 | 
						|
  //
 | 
						|
  Status = Ip->Transmit (Ip, SndEntry->SndToken);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    IpIoDestroySndEntry (SndEntry);
 | 
						|
  }
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Cancel the IP transmit token which wraps this Packet.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance.
 | 
						|
  @param  Packet                Pointer to the packet to cancel.
 | 
						|
 | 
						|
  @return N/A.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoCancelTxToken (
 | 
						|
  IN IP_IO  *IpIo,
 | 
						|
  IN VOID   *Packet
 | 
						|
  )
 | 
						|
{
 | 
						|
  LIST_ENTRY        *Node;
 | 
						|
  IP_IO_SEND_ENTRY  *SndEntry;
 | 
						|
  EFI_IP4_PROTOCOL  *Ip;
 | 
						|
 | 
						|
  ASSERT (IpIo && Packet);
 | 
						|
 | 
						|
  NET_LIST_FOR_EACH (Node, &IpIo->PendingSndList) {
 | 
						|
 | 
						|
    SndEntry = NET_LIST_USER_STRUCT (Node, IP_IO_SEND_ENTRY, Entry);
 | 
						|
 | 
						|
    if (SndEntry->Pkt == Packet) {
 | 
						|
 | 
						|
      Ip = SndEntry->Ip;
 | 
						|
      Ip->Cancel (Ip, SndEntry->SndToken);
 | 
						|
 | 
						|
      break;
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Add a new IP instance for sending data.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to a IP_IO instance to add a new IP
 | 
						|
                                instance for sending purpose.
 | 
						|
 | 
						|
  @return Pointer to the created IP_IO_IP_INFO structure, NULL is failed.
 | 
						|
 | 
						|
**/
 | 
						|
IP_IO_IP_INFO *
 | 
						|
EFIAPI
 | 
						|
IpIoAddIp (
 | 
						|
  IN IP_IO  *IpIo
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS     Status;
 | 
						|
  IP_IO_IP_INFO  *IpInfo;
 | 
						|
 | 
						|
  ASSERT (IpIo);
 | 
						|
 | 
						|
  IpInfo = AllocatePool (sizeof (IP_IO_IP_INFO));
 | 
						|
  if (IpInfo == NULL) {
 | 
						|
    return IpInfo;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Init this IpInfo, set the Addr and SubnetMask to 0 before we configure the IP
 | 
						|
  // instance.
 | 
						|
  //
 | 
						|
  InitializeListHead (&IpInfo->Entry);
 | 
						|
  IpInfo->ChildHandle = NULL;
 | 
						|
  IpInfo->Addr        = 0;
 | 
						|
  IpInfo->SubnetMask  = 0;
 | 
						|
  IpInfo->RefCnt      = 1;
 | 
						|
 | 
						|
  //
 | 
						|
  // Create the IP instance and open the Ip4 protocol.
 | 
						|
  //
 | 
						|
  Status = IpIoCreateIpChildOpenProtocol (
 | 
						|
             IpIo->Controller,
 | 
						|
             IpIo->Image,
 | 
						|
             &IpInfo->ChildHandle,
 | 
						|
             (VOID **) &IpInfo->Ip
 | 
						|
             );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto ReleaseIpInfo;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Create the event for the DummyRcvToken.
 | 
						|
  //
 | 
						|
  Status = gBS->CreateEvent (
 | 
						|
                  EVT_NOTIFY_SIGNAL,
 | 
						|
                  TPL_NOTIFY,
 | 
						|
                  IpIoDummyHandler,
 | 
						|
                  IpInfo,
 | 
						|
                  &IpInfo->DummyRcvToken.Event
 | 
						|
                  );
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto ReleaseIpChild;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // Link this IpInfo into the IpIo.
 | 
						|
  //
 | 
						|
  InsertTailList (&IpIo->IpList, &IpInfo->Entry);
 | 
						|
 | 
						|
  return IpInfo;
 | 
						|
 | 
						|
ReleaseIpChild:
 | 
						|
 | 
						|
  IpIoCloseProtocolDestroyIpChild (
 | 
						|
    IpIo->Controller,
 | 
						|
    IpIo->Image,
 | 
						|
    IpInfo->ChildHandle
 | 
						|
    );
 | 
						|
 | 
						|
ReleaseIpInfo:
 | 
						|
 | 
						|
  gBS->FreePool (IpInfo);
 | 
						|
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Configure the IP instance of this IpInfo and start the receiving if Ip4ConfigData
 | 
						|
  is not NULL.
 | 
						|
 | 
						|
  @param  IpInfo                Pointer to the IP_IO_IP_INFO instance.
 | 
						|
  @param  Ip4ConfigData         The IP4 configure data used to configure the ip
 | 
						|
                                instance, if NULL the ip instance is reseted. If
 | 
						|
                                UseDefaultAddress is set to TRUE, and the configure
 | 
						|
                                operation succeeds, the default address information
 | 
						|
                                is written back in this Ip4ConfigData.
 | 
						|
 | 
						|
  @retval EFI_STATUS            The status returned by IP4->Configure or
 | 
						|
                                IP4->Receive.
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
IpIoConfigIp (
 | 
						|
  IN     IP_IO_IP_INFO        *IpInfo,
 | 
						|
  IN OUT EFI_IP4_CONFIG_DATA  *Ip4ConfigData OPTIONAL
 | 
						|
  )
 | 
						|
{
 | 
						|
  EFI_STATUS         Status;
 | 
						|
  EFI_IP4_PROTOCOL   *Ip;
 | 
						|
  EFI_IP4_MODE_DATA  Ip4ModeData;
 | 
						|
 | 
						|
  ASSERT (IpInfo);
 | 
						|
 | 
						|
  if (IpInfo->RefCnt > 1) {
 | 
						|
    //
 | 
						|
    // This IP instance is shared, don't reconfigure it until it has only one
 | 
						|
    // consumer. Currently, only the tcp children cloned from their passive parent
 | 
						|
    // will share the same IP. So this cases only happens while Ip4ConfigData is NULL,
 | 
						|
    // let the last consumer clean the IP instance.
 | 
						|
    //
 | 
						|
    return EFI_SUCCESS;
 | 
						|
  }
 | 
						|
 | 
						|
  Ip = IpInfo->Ip;
 | 
						|
 | 
						|
  Status = Ip->Configure (Ip, Ip4ConfigData);
 | 
						|
  if (EFI_ERROR (Status)) {
 | 
						|
    goto OnExit;
 | 
						|
  }
 | 
						|
 | 
						|
  if (Ip4ConfigData != NULL) {
 | 
						|
 | 
						|
    if (Ip4ConfigData->UseDefaultAddress) {
 | 
						|
      Ip->GetModeData (Ip, &Ip4ModeData, NULL, NULL);
 | 
						|
 | 
						|
      Ip4ConfigData->StationAddress = Ip4ModeData.ConfigData.StationAddress;
 | 
						|
      Ip4ConfigData->SubnetMask     = Ip4ModeData.ConfigData.SubnetMask;
 | 
						|
    }
 | 
						|
 | 
						|
    CopyMem (&IpInfo->Addr, &Ip4ConfigData->StationAddress, sizeof (IP4_ADDR));
 | 
						|
    CopyMem (&IpInfo->SubnetMask, &Ip4ConfigData->SubnetMask, sizeof (IP4_ADDR));
 | 
						|
 | 
						|
    Status = Ip->Receive (Ip, &IpInfo->DummyRcvToken);
 | 
						|
    if (EFI_ERROR (Status)) {
 | 
						|
      Ip->Configure (Ip, NULL);
 | 
						|
    }
 | 
						|
  } else {
 | 
						|
 | 
						|
    //
 | 
						|
    // The IP instance is reseted, set the stored Addr and SubnetMask to zero.
 | 
						|
    //
 | 
						|
    IpInfo->Addr       = 0;
 | 
						|
    IpInfo->SubnetMask =0;
 | 
						|
  }
 | 
						|
 | 
						|
OnExit:
 | 
						|
 | 
						|
  return Status;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Destroy an IP instance maintained in IpIo->IpList for
 | 
						|
  sending purpose.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the IP_IO instance.
 | 
						|
  @param  IpInfo                Pointer to the IpInfo to be removed.
 | 
						|
 | 
						|
  @return None.
 | 
						|
 | 
						|
**/
 | 
						|
VOID
 | 
						|
EFIAPI
 | 
						|
IpIoRemoveIp (
 | 
						|
  IN IP_IO          *IpIo,
 | 
						|
  IN IP_IO_IP_INFO  *IpInfo
 | 
						|
  )
 | 
						|
{
 | 
						|
  ASSERT (IpInfo->RefCnt > 0);
 | 
						|
 | 
						|
  NET_PUT_REF (IpInfo);
 | 
						|
 | 
						|
  if (IpInfo->RefCnt > 0) {
 | 
						|
 | 
						|
    return;
 | 
						|
  }
 | 
						|
 | 
						|
  RemoveEntryList (&IpInfo->Entry);
 | 
						|
 | 
						|
  IpInfo->Ip->Configure (IpInfo->Ip, NULL);
 | 
						|
 | 
						|
  IpIoCloseProtocolDestroyIpChild (IpIo->Controller, IpIo->Image, IpInfo->ChildHandle);
 | 
						|
 | 
						|
  gBS->CloseEvent (IpInfo->DummyRcvToken.Event);
 | 
						|
 | 
						|
  gBS->FreePool (IpInfo);
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Find the first IP protocol maintained in IpIo whose local
 | 
						|
  address is the same with Src.
 | 
						|
 | 
						|
  @param  IpIo                  Pointer to the pointer of the IP_IO instance.
 | 
						|
  @param  Src                   The local IP address.
 | 
						|
 | 
						|
  @return Pointer to the IP protocol can be used for sending purpose and its local
 | 
						|
  @return address is the same with Src.
 | 
						|
 | 
						|
**/
 | 
						|
IP_IO_IP_INFO *
 | 
						|
EFIAPI
 | 
						|
IpIoFindSender (
 | 
						|
  IN OUT IP_IO     **IpIo,
 | 
						|
  IN     IP4_ADDR  Src
 | 
						|
  )
 | 
						|
{
 | 
						|
  LIST_ENTRY      *IpIoEntry;
 | 
						|
  IP_IO           *IpIoPtr;
 | 
						|
  LIST_ENTRY      *IpInfoEntry;
 | 
						|
  IP_IO_IP_INFO   *IpInfo;
 | 
						|
 | 
						|
  NET_LIST_FOR_EACH (IpIoEntry, &mActiveIpIoList) {
 | 
						|
    IpIoPtr = NET_LIST_USER_STRUCT (IpIoEntry, IP_IO, Entry);
 | 
						|
 | 
						|
    if ((*IpIo != NULL) && (*IpIo != IpIoPtr)) {
 | 
						|
      continue;
 | 
						|
    }
 | 
						|
 | 
						|
    NET_LIST_FOR_EACH (IpInfoEntry, &IpIoPtr->IpList) {
 | 
						|
      IpInfo = NET_LIST_USER_STRUCT (IpInfoEntry, IP_IO_IP_INFO, Entry);
 | 
						|
 | 
						|
      if (IpInfo->Addr == Src) {
 | 
						|
        *IpIo = IpIoPtr;
 | 
						|
        return IpInfo;
 | 
						|
      }
 | 
						|
    }
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // No match.
 | 
						|
  //
 | 
						|
  return NULL;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
  Get the ICMP error map information, the ErrorStatus will be returned.
 | 
						|
  The IsHard and Notify are optional. If they are not NULL, this rouine will
 | 
						|
  fill them.
 | 
						|
  We move IcmpErrMap[] to local variable to enable EBC build.
 | 
						|
 | 
						|
  @param  IcmpError             IcmpError Type
 | 
						|
  @param  IsHard                Whether it is a hard error
 | 
						|
  @param  Notify                Whether it need to notify SockError
 | 
						|
 | 
						|
  @return ICMP Error Status
 | 
						|
 | 
						|
**/
 | 
						|
EFI_STATUS
 | 
						|
EFIAPI
 | 
						|
IpIoGetIcmpErrStatus (
 | 
						|
  IN  ICMP_ERROR  IcmpError,
 | 
						|
  OUT BOOLEAN     *IsHard, OPTIONAL
 | 
						|
  OUT BOOLEAN     *Notify OPTIONAL
 | 
						|
  )
 | 
						|
{
 | 
						|
  ASSERT ((IcmpError >= ICMP_ERR_UNREACH_NET) && (IcmpError <= ICMP_ERR_PARAMPROB));
 | 
						|
 | 
						|
  if (IsHard != NULL) {
 | 
						|
    *IsHard = mIcmpErrMap[IcmpError].IsHard;
 | 
						|
  }
 | 
						|
 | 
						|
  if (Notify != NULL) {
 | 
						|
    *Notify = mIcmpErrMap[IcmpError].Notify;
 | 
						|
  }
 | 
						|
 | 
						|
  switch (IcmpError) {
 | 
						|
  case ICMP_ERR_UNREACH_NET:
 | 
						|
    return  EFI_NETWORK_UNREACHABLE;
 | 
						|
 | 
						|
  case ICMP_ERR_TIMXCEED_INTRANS:
 | 
						|
  case ICMP_ERR_TIMXCEED_REASS:
 | 
						|
  case ICMP_ERR_UNREACH_HOST:
 | 
						|
    return  EFI_HOST_UNREACHABLE;
 | 
						|
 | 
						|
  case ICMP_ERR_UNREACH_PROTOCOL:
 | 
						|
    return  EFI_PROTOCOL_UNREACHABLE;
 | 
						|
 | 
						|
  case ICMP_ERR_UNREACH_PORT:
 | 
						|
    return  EFI_PORT_UNREACHABLE;
 | 
						|
 | 
						|
  case ICMP_ERR_MSGSIZE:
 | 
						|
  case ICMP_ERR_UNREACH_SRCFAIL:
 | 
						|
  case ICMP_ERR_QUENCH:
 | 
						|
  case ICMP_ERR_PARAMPROB:
 | 
						|
    return  EFI_ICMP_ERROR;
 | 
						|
  }
 | 
						|
 | 
						|
  //
 | 
						|
  // will never run here!
 | 
						|
  //
 | 
						|
  ASSERT (FALSE);
 | 
						|
  return EFI_UNSUPPORTED;
 | 
						|
}
 | 
						|
 |