Receive.c

Go to the documentation of this file.

 
#include "Snp.h"
 
EFI_STATUS
PxeReceive (
  SNP_DRIVER       *Snp,
  VOID             *Buffer,
  UINTN            *BufferSize,
  UINTN            *HeaderSize,
  EFI_MAC_ADDRESS  *SrcAddr,
  EFI_MAC_ADDRESS  *DestAddr,
  UINT16           *Protocol
  )
{
  PXE_CPB_RECEIVE  *Cpb;
  PXE_DB_RECEIVE   *Db;
  UINTN            BuffSize;
 
  Cpb      = Snp->Cpb;
  Db       = Snp->Db;
  BuffSize = *BufferSize;
 
  Cpb->BufferAddr = (UINT64)(UINTN)Buffer;
  Cpb->BufferLen  = (UINT32)*BufferSize;
 
  Cpb->reserved = 0;
 
  Snp->Cdb.OpCode  = PXE_OPCODE_RECEIVE;
  Snp->Cdb.OpFlags = PXE_OPFLAGS_NOT_USED;
 
  Snp->Cdb.CPBsize = (UINT16)sizeof (PXE_CPB_RECEIVE);
  Snp->Cdb.CPBaddr = (UINT64)(UINTN)Cpb;
 
  Snp->Cdb.DBsize = (UINT16)sizeof (PXE_DB_RECEIVE);
  Snp->Cdb.DBaddr = (UINT64)(UINTN)Db;
 
  Snp->Cdb.StatCode  = PXE_STATCODE_INITIALIZE;
  Snp->Cdb.StatFlags = PXE_STATFLAGS_INITIALIZE;
  Snp->Cdb.IFnum     = Snp->IfNum;
  Snp->Cdb.Control   = PXE_CONTROL_LAST_CDB_IN_LIST;
 
  //
  // Issue UNDI command and check result.
  //
  DEBUG ((DEBUG_NET, "\nsnp->undi.receive ()  "));
 
  (*Snp->IssueUndi32Command)((UINT64)(UINTN)&Snp->Cdb);
 
  switch (Snp->Cdb.StatCode) {
    case PXE_STATCODE_SUCCESS:
      break;
 
    case PXE_STATCODE_NO_DATA:
      DEBUG (
        (DEBUG_NET,
         "\nsnp->undi.receive ()  %xh:%xh\n",
         Snp->Cdb.StatFlags,
         Snp->Cdb.StatCode)
        );
 
      return EFI_NOT_READY;
 
    default:
      DEBUG (
        (DEBUG_ERROR,
         "\nsnp->undi.receive()  %xh:%xh\n",
         Snp->Cdb.StatFlags,
         Snp->Cdb.StatCode)
        );
 
      return EFI_DEVICE_ERROR;
  }
 
  *BufferSize = Db->FrameLen;
 
  if (HeaderSize != NULL) {
    *HeaderSize = Db->MediaHeaderLen;
  }
 
  if (SrcAddr != NULL) {
    CopyMem (SrcAddr, &Db->SrcAddr, Snp->Mode.HwAddressSize);
  }
 
  if (DestAddr != NULL) {
    CopyMem (DestAddr, &Db->DestAddr, Snp->Mode.HwAddressSize);
  }
 
  if (Protocol != NULL) {
    //
    // We need to do the byte swapping
    //
    *Protocol = (UINT16)PXE_SWAP_UINT16 (Db->Protocol);
  }
 
  //
  // We have received a packet from network interface, which implies that the
  // network cable should be present. While, some UNDI driver may not report
  // correct media status during Snp->Initialize(). So, we need ensure
  // MediaPresent in SNP mode data is set to correct value.
  //
  if (Snp->Mode.MediaPresentSupported && !Snp->Mode.MediaPresent) {
    Snp->Mode.MediaPresent = TRUE;
  }
 
  return (*BufferSize <= BuffSize) ? EFI_SUCCESS : EFI_BUFFER_TOO_SMALL;
}
 
EFI_STATUS
EFIAPI
SnpUndi32Receive (
  IN EFI_SIMPLE_NETWORK_PROTOCOL  *This,
  OUT UINTN                       *HeaderSize OPTIONAL,
  IN OUT UINTN                    *BufferSize,
  OUT VOID                        *Buffer,
  OUT EFI_MAC_ADDRESS             *SrcAddr OPTIONAL,
  OUT EFI_MAC_ADDRESS             *DestAddr OPTIONAL,
  OUT UINT16                      *Protocol OPTIONAL
  )
{
  SNP_DRIVER  *Snp;
  EFI_TPL     OldTpl;
  EFI_STATUS  Status;
 
  if (This == NULL) {
    return EFI_INVALID_PARAMETER;
  }
 
  Snp = EFI_SIMPLE_NETWORK_DEV_FROM_THIS (This);
 
  OldTpl = gBS->RaiseTPL (TPL_CALLBACK);
 
  switch (Snp->Mode.State) {
    case EfiSimpleNetworkInitialized:
      break;
 
    case EfiSimpleNetworkStopped:
      Status = EFI_NOT_STARTED;
      goto ON_EXIT;
 
    default:
      Status = EFI_DEVICE_ERROR;
      goto ON_EXIT;
  }
 
  if ((BufferSize == NULL) || (Buffer == NULL)) {
    Status = EFI_INVALID_PARAMETER;
    goto ON_EXIT;
  }
 
  if (Snp->Mode.ReceiveFilterSetting == 0) {
    Status = EFI_DEVICE_ERROR;
    goto ON_EXIT;
  }
 
  Status = PxeReceive (
             Snp,
             Buffer,
             BufferSize,
             HeaderSize,
             SrcAddr,
             DestAddr,
             Protocol
             );
 
ON_EXIT:
  gBS->RestoreTPL (OldTpl);
 
  return Status;
}