[mirror_edk2.git] / AppPkg / Applications / Sockets / TftpServer / TftpServer.c

/*++\r
  This file contains an 'Intel UEFI Application' and is\r
  licensed for Intel CPUs and chipsets under the terms of your\r
  license agreement with Intel or your vendor.  This file may\r
  be modified by the user, subject to additional terms of the\r
  license agreement\r
--*/\r
/*++\r
\r
Copyright (c)  2011 Intel Corporation. All rights reserved\r
This software and associated documentation (if any) is furnished\r
under a license and may only be used or copied in accordance\r
with the terms of the license. Except as permitted by such\r
license, no part of this software or documentation may be\r
reproduced, stored in a retrieval system, or transmitted in any\r
form or by any means without the express written consent of\r
Intel Corporation.\r
\r
--*/\r
\r
/** @file\r
  This is a simple TFTP server application\r
\r
**/\r
\r
#include <TftpServer.h>\r
\r
TSDT_TFTP_SERVER mTftpServer; ///<  TFTP server's control structure\r
\r
\r
/**\r
  Add a connection context to the list of connection contexts.\r
\r
  @param [in] pTftpServer   The TFTP server control structure address.\r
\r
  @retval Context structure address, NULL if allocation fails\r
\r
**/\r
TSDT_CONNECTION_CONTEXT *\r
ContextAdd (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  size_t LengthInBytes;\r
  TSDT_CONNECTION_CONTEXT * pContext;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Use for/break instead of goto\r
  //\r
  for ( ; ; ) {\r
    //\r
    //  Allocate a new context\r
    //\r
    LengthInBytes = sizeof ( *pContext );\r
    Status = gBS->AllocatePool ( EfiRuntimeServicesData,\r
                                 LengthInBytes,\r
                                 (VOID **)&pContext );\r
    if ( EFI_ERROR ( Status )) {\r
      DEBUG (( DEBUG_ERROR | DEBUG_POOL,\r
                "ERROR - Failed to allocate the context, Status: %r\r\n",\r
                Status ));\r
      pContext = NULL;\r
      break;\r
    }\r
\r
    //\r
    //  Initialize the context\r
    //\r
    ZeroMem ( pContext, LengthInBytes );\r
    CopyMem ( &pContext->RemoteAddress,\r
              &pTftpServer->RemoteAddress,\r
              sizeof ( pContext->RemoteAddress ));\r
    pContext->BlockSize = TFTP_MAX_BLOCK_SIZE;\r
    pContext->pBuffer = &pContext->FileData[0];\r
    pContext->pEnd = &pContext->pBuffer[sizeof ( pContext->pBuffer )];\r
    pContext->MaxTransferSize = 0;\r
    pContext->MaxTransferSize -= 1;\r
\r
    //\r
    //  Display the new context\r
    //\r
    DEBUG (( DEBUG_PORT_WORK | DEBUG_INFO,\r
              "0x%08x: Context for %d.%d.%d.%d:%d\r\n",\r
              pContext,\r
              (UINT8)pContext->RemoteAddress.sin_addr.s_addr,\r
              (UINT8)( pContext->RemoteAddress.sin_addr.s_addr >> 8 ),\r
              (UINT8)( pContext->RemoteAddress.sin_addr.s_addr >> 16 ),\r
              (UINT8)( pContext->RemoteAddress.sin_addr.s_addr >> 24 ),\r
              htons ( pContext->RemoteAddress.sin_port )));\r
\r
    //\r
    //  Add the context to the context list\r
    //\r
    pContext->pNext = pTftpServer->pContextList;\r
    pTftpServer->pContextList = pContext;\r
\r
    //\r
    //  All done\r
    //\r
    break;\r
  }\r
\r
  //\r
  //  Return the connection context\r
  //\r
  DBG_EXIT_STATUS ( pContext );\r
  return pContext;\r
}\r
\r
\r
/**\r
  Locate a remote connection context.\r
\r
  @param [in] pTftpServer   The TFTP server control structure address.\r
\r
  @param [in] pIpAddress    The start of the remote IP address in network order\r
\r
  @param [in] Port          The remote port number\r
\r
  @retval Context structure address, NULL if not found\r
\r
**/\r
TSDT_CONNECTION_CONTEXT *\r
ContextFind (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  TSDT_CONNECTION_CONTEXT * pContext;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Walk the list of connection contexts\r
  //\r
  pContext = pTftpServer->pContextList;\r
  while ( NULL != pContext ) {\r
    //\r
    //  Attempt to locate the remote network connection\r
    //\r
    if (( pTftpServer->RemoteAddress.sin_addr.s_addr == pContext->RemoteAddress.sin_addr.s_addr )\r
      && ( pTftpServer->RemoteAddress.sin_port == pContext->RemoteAddress.sin_port )) {\r
      //\r
      //  The connection was found\r
      //\r
      DEBUG (( DEBUG_TFTP_REQUEST,\r
                "0x%08x: pContext found\r\n",\r
                pContext ));\r
      break;\r
    }\r
\r
    //\r
    //  Set the next context\r
    //\r
    pContext = pContext->pNext;\r
  }\r
\r
  //\r
  //  Return the connection context structure address\r
  //\r
  DBG_EXIT_HEX ( pContext );\r
  return pContext;\r
}\r
\r
\r
/**\r
  Remove a context from the list.\r
\r
  @param [in] pTftpServer    The TFTP server control structure address.\r
\r
  @param [in] pContext       The context structure address.\r
\r
**/\r
VOID\r
ContextRemove (\r
  IN TSDT_TFTP_SERVER * pTftpServer,\r
  IN TSDT_CONNECTION_CONTEXT * pContext\r
  )\r
{\r
  TSDT_CONNECTION_CONTEXT * pNextContext;\r
  TSDT_CONNECTION_CONTEXT * pPreviousContext;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Attempt to locate the context in the list\r
  //\r
  pPreviousContext = NULL;\r
  pNextContext = pTftpServer->pContextList;\r
  while ( NULL != pNextContext ) {\r
    //\r
    //  Determine if the context was found\r
    //\r
    if ( pNextContext == pContext ) {\r
      //\r
      //  Remove the context from the list\r
      //\r
      if ( NULL == pPreviousContext ) {\r
        pTftpServer->pContextList = pContext->pNext;\r
      }\r
      else {\r
        pPreviousContext->pNext = pContext->pNext;\r
      }\r
      break;\r
    }\r
\r
    //\r
    //  Set the next context\r
    //\r
    pPreviousContext = pNextContext;\r
    pNextContext = pNextContext->pNext;\r
  }\r
\r
  //\r
  //  Determine if the context was found\r
  //\r
  if ( NULL != pContext ) {\r
    //\r
    //  Return the resources\r
    //\r
    gBS->FreePool ( pContext );\r
  }\r
\r
  DBG_EXIT ( );\r
}\r
\r
\r
/**\r
  Process the work for the sockets.\r
\r
  @param [in] pTftpServer   The TFTP server control structure address.\r
\r
**/\r
VOID\r
PortWork (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  TSDT_CONNECTION_CONTEXT * pContext;\r
  socklen_t RemoteAddressLength;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Handle input events\r
  //\r
  if ( 0 != ( pTftpServer->TftpPort.revents & POLLRDNORM )) {\r
    //\r
    //  Receive the message from the remote system\r
    //\r
    RemoteAddressLength = sizeof ( pTftpServer->RemoteAddress );\r
    pTftpServer->RxBytes = recvfrom ( pTftpServer->TftpPort.fd,\r
                                      &pTftpServer->RxBuffer[0],\r
                                      sizeof ( pTftpServer->RxBuffer ),\r
                                      0,\r
                                      (struct sockaddr *) &pTftpServer->RemoteAddress,\r
                                      &RemoteAddressLength );\r
    if ( -1 != pTftpServer->RxBytes ) {\r
      pTftpServer->RemoteAddress.sin_len = (UINT8) RemoteAddressLength;\r
      DEBUG (( DEBUG_TFTP_PORT,\r
                 "Received %d bytes from %d.%d.%d.%d:%d\r\n",\r
                 pTftpServer->RxBytes,\r
                 pTftpServer->RemoteAddress.sin_addr.s_addr & 0xff,\r
                 ( pTftpServer->RemoteAddress.sin_addr.s_addr >> 8 ) & 0xff,\r
                 ( pTftpServer->RemoteAddress.sin_addr.s_addr >> 16 ) & 0xff,\r
                 ( pTftpServer->RemoteAddress.sin_addr.s_addr >> 24 ) & 0xff,\r
                 htons ( pTftpServer->RemoteAddress.sin_port )));\r
\r
      //\r
      //  Lookup connection context using the remote system address and port\r
      //  to determine if an existing connection to this remote\r
      //  system exists\r
      //\r
      pContext = ContextFind ( pTftpServer );\r
\r
      //\r
      //  Process the received message\r
      //\r
      TftpProcessRequest ( pTftpServer, pContext );\r
    }\r
    else {\r
      //\r
      //  Receive error on the TFTP server port\r
      //  Close the server socket\r
      //\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - Failed receive on TFTP server port, errno: 0x%08x\r\n",\r
                errno ));\r
      pTftpServer->TftpPort.revents |= POLLHUP;\r
    }\r
  }\r
\r
  //\r
  //  Handle the close event\r
  //\r
  if ( 0 != ( pTftpServer->TftpPort.revents & POLLHUP )) {\r
    //\r
    //  Close the port\r
    //\r
    close ( pTftpServer->TftpPort.fd );\r
    pTftpServer->TftpPort.fd = -1;\r
  }\r
\r
  DBG_EXIT ( );\r
}\r
\r
\r
/**\r
  Scan the list of sockets and process any pending work\r
\r
  @param [in] pTftpServer   The TFTP server control structure address.\r
\r
**/\r
VOID\r
SocketPoll (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  int FDCount;\r
\r
  DEBUG (( DEBUG_SOCKET_POLL, "Entering SocketPoll\r\n" ));\r
\r
  //\r
  //  Determine if any ports are active\r
  //\r
  FDCount = poll ( &pTftpServer->TftpPort,\r
                   1,\r
                   CLIENT_POLL_DELAY );\r
  if ( -1 == FDCount ) {\r
    DEBUG (( DEBUG_ERROR | DEBUG_SOCKET_POLL,\r
              "ERROR - errno: %d\r\n",\r
              errno ));\r
  }\r
\r
  if ( 0 < FDCount ) {\r
    //\r
    //  Process this port\r
    //\r
    PortWork ( pTftpServer );\r
    pTftpServer->TftpPort.revents = 0;\r
  }\r
\r
  DEBUG (( DEBUG_SOCKET_POLL, "Exiting SocketPoll\r\n" ));\r
}\r
\r
\r
/**\r
  Convert a character to lower case\r
\r
  @param [in] Character The character to convert\r
\r
  @return   The lower case equivalent of the character\r
\r
**/\r
int\r
tolower (\r
  int Character\r
  )\r
{\r
  //\r
  //  Determine if the character is upper case\r
  //\r
  if (( 'A' <= Character ) && ( 'Z' >= Character )) {\r
    //\r
    //  Convert the character to lower caes\r
    //\r
    Character += 'a' - 'A';\r
  }\r
\r
  //\r
  //  Return the converted character\r
  //\r
  return Character;\r
}\r
\r
\r
/**\r
  Case independent string comparison\r
\r
  @param [in] pString1  Zero terminated string address\r
  @param [in] pString2  Zero terminated string address\r
\r
  @return     Returns the first character difference between string 1\r
              and string 2.\r
\r
**/\r
int\r
stricmp (\r
  char * pString1,\r
  char * pString2\r
  )\r
{\r
  int Char1;\r
  int Char2;\r
  int Difference;\r
\r
  //\r
  //  Walk the length of the strings\r
  //\r
  do {\r
    //\r
    //  Get the next characters\r
    //\r
    Char1 = (UINT8)*pString1++;\r
    Char2 = (UINT8)*pString2++;\r
\r
    //\r
    //  Convert them to lower case\r
    //\r
    Char1 = tolower ( Char1 );\r
    Char2 = tolower ( Char2 );\r
\r
    //\r
    //  Done when the characters differ\r
    //\r
    Difference = Char1 - Char2;\r
    if ( 0 != Difference ) {\r
      break;\r
    }\r
\r
    //\r
    //  Done at the end of the string\r
    //\r
  } while ( 0 != Char1 );\r
\r
  //\r
  //  Return the difference\r
  //\r
  return Difference;\r
}\r
\r
\r
/**\r
  Get the next TFTP option\r
\r
  @param [in] pOption       Address of a zero terminated option string\r
  @param [in] pEnd          End of buffer address\r
  @param [in] ppNextOption  Address to receive the address of the next\r
                            zero terminated option string\r
\r
  @retval EFI_SUCCESS   Message processed successfully\r
\r
**/\r
EFI_STATUS\r
TftpOptionGet (\r
  IN UINT8 * pOption,\r
  IN UINT8 * pEnd,\r
  IN UINT8 ** ppNextOption\r
  )\r
{\r
  UINT8 * pNextOption;\r
  EFI_STATUS Status;\r
\r
  //\r
  //  Locate the end of the option\r
  //\r
  pNextOption = pOption;\r
  while (( pEnd > pNextOption ) && ( 0 != *pNextOption )) {\r
    pNextOption += 1;\r
  }\r
  if ( pEnd <= pNextOption ) {\r
    //\r
    //  Error - end of buffer reached\r
    //\r
    DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r
              "ERROR - Option without zero termination received!\r\n" ));\r
    Status = EFI_INVALID_PARAMETER;\r
  }\r
  else {\r
    //\r
    //  Zero terminated option found\r
    //\r
    pNextOption += 1;\r
\r
    //\r
    //  Display the zero terminated ASCII option string\r
    //\r
    DEBUG (( DEBUG_TFTP_REQUEST,\r
              "Option: %a\r\n",\r
              pOption ));\r
    Status = EFI_SUCCESS;\r
  }\r
\r
  //\r
  //  Return the next option address\r
  //\r
  *ppNextOption = pNextOption;\r
\r
  //\r
  //  Return the operation status\r
  //\r
  return Status;\r
}\r
\r
\r
/**\r
  Place an option value into the option acknowledgement\r
\r
  @param [in] pOack     Option acknowledgement address\r
  @param [in] Value     Value to translate into ASCII decimal\r
\r
  @return               Option acknowledgement address\r
\r
**/\r
UINT8 *\r
TftpOptionSet (\r
  IN UINT8 * pOack,\r
  IN UINT64 Value\r
  )\r
{\r
  UINT64 NextValue;\r
\r
  //\r
  //  Determine the next value\r
  //\r
  NextValue = Value / 10;\r
\r
  //\r
  //  Supress leading zeros\r
  //\r
  if ( 0 != NextValue ) {\r
    pOack = TftpOptionSet ( pOack, NextValue );\r
  }\r
\r
  //\r
  //  Output this digit\r
  //\r
  *pOack++ = (UINT8)( Value - ( NextValue * 10 ) + '0' );\r
\r
  //\r
  //  Return the next option acknowledgement location\r
  //\r
  return pOack;\r
}\r
\r
\r
/**\r
  Process the TFTP request\r
\r
  @param [in] pContext  The context structure address.\r
  @param [in] pOption   Address of the first zero terminated option string\r
  @param [in] pEnd      End of buffer address\r
\r
**/\r
VOID\r
TftpOptions (\r
  IN TSDT_CONNECTION_CONTEXT * pContext,\r
  IN UINT8 * pOption,\r
  IN UINT8 * pEnd\r
  )\r
{\r
  UINT8 * pNextOption;\r
  UINT8 * pOack;\r
  UINT8 * pTemp;\r
  UINT8 * pValue;\r
  EFI_STATUS Status;\r
  INT32 Value;\r
\r
  //\r
  //  Start the OACK packet\r
  //  Let the OACK handle the parsing errors\r
  //  See http://tools.ietf.org/html/rfc2347\r
  //\r
  pOack = &pContext->TxBuffer[0];\r
  *pOack++ = 0;\r
  *pOack++ = TFTP_OP_OACK;\r
  pContext->TxBytes = 2;\r
\r
  //\r
  //  Walk the list of options\r
  //\r
  do {\r
    //\r
    //  Get the next option, skip junk at end of message\r
    //\r
    Status = TftpOptionGet ( pOption, pEnd, &pNextOption );\r
    if ( !EFI_ERROR ( Status )) {\r
      //\r
      //  Process the option\r
      //\r
\r
      //\r
      //  blksize - See http://tools.ietf.org/html/rfc2348\r
      //\r
      pValue = pNextOption;\r
      if ( 0 == stricmp ((char *)pOption, "blksize" )) {\r
        //\r
        //  Get the value\r
        //\r
        Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r
        if ( !EFI_ERROR ( Status )) {\r
          //\r
          //  Validate the block size, skip non-numeric block sizes\r
          //\r
          Status = TftpOptionValue ( pValue, &Value );\r
          if ( !EFI_ERROR ( Status )) {\r
            //\r
            //  Propose a smaller block size if necessary\r
            //\r
            if ( Value > TFTP_MAX_BLOCK_SIZE ) {\r
              Value = TFTP_MAX_BLOCK_SIZE;\r
            }\r
\r
            //\r
            //  Set the new block size\r
            //\r
            pContext->BlockSize = Value;\r
            DEBUG (( DEBUG_TFTP_REQUEST,\r
                      "Using block size of %d bytes\r\n",\r
                      pContext->BlockSize ));\r
\r
            //\r
            //  Update the OACK\r
            //\r
            pTemp = pOack;\r
            *pOack++ = 'b';\r
            *pOack++ = 'l';\r
            *pOack++ = 'k';\r
            *pOack++ = 's';\r
            *pOack++ = 'i';\r
            *pOack++ = 'z';\r
            *pOack++ = 'e';\r
            *pOack++ = 0;\r
            pOack = TftpOptionSet ( pOack, pContext->BlockSize );\r
            *pOack++ = 0;\r
            pContext->TxBytes += pOack - pTemp;\r
          }\r
        }\r
      }\r
\r
      //\r
      //  timeout - See http://tools.ietf.org/html/rfc2349\r
      //\r
      else if ( 0 == stricmp ((char *)pOption, "timeout" )) {\r
        //\r
        //  Get the value\r
        //\r
        Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r
        if ( !EFI_ERROR ( Status )) {\r
          Status = TftpOptionValue ( pValue, &Value );\r
          if ( !EFI_ERROR ( Status )) {\r
            //\r
            //  Set the timeout value\r
            //\r
            pContext->Timeout = Value;\r
            DEBUG (( DEBUG_TFTP_REQUEST,\r
                      "Using timeout of %d seconds\r\n",\r
                      pContext->Timeout ));\r
\r
            //\r
            //  Update the OACK\r
            //\r
            pTemp = pOack;\r
            *pOack++ = 't';\r
            *pOack++ = 'i';\r
            *pOack++ = 'm';\r
            *pOack++ = 'e';\r
            *pOack++ = 'o';\r
            *pOack++ = 'u';\r
            *pOack++ = 't';\r
            *pOack++ = 0;\r
            pOack = TftpOptionSet ( pOack, pContext->Timeout );\r
            *pOack++ = 0;\r
            pContext->TxBytes += pOack - pTemp;\r
          }\r
        }\r
      }\r
\r
      //\r
      //  tsize - See http://tools.ietf.org/html/rfc2349\r
      //\r
      else if ( 0 == stricmp ((char *)pOption, "tsize" )) {\r
        //\r
        //  Get the value\r
        //\r
        Status = TftpOptionGet ( pValue, pEnd, &pNextOption );\r
        if ( !EFI_ERROR ( Status )) {\r
          Status = TftpOptionValue ( pValue, &Value );\r
          if ( !EFI_ERROR ( Status )) {\r
            //\r
            //  Return the file size\r
            //\r
            DEBUG (( DEBUG_TFTP_REQUEST,\r
                      "Returning file size of %Ld bytes\r\n",\r
                      pContext->LengthInBytes ));\r
\r
            //\r
            //  Update the OACK\r
            //\r
            pTemp = pOack;\r
            *pOack++ = 't';\r
            *pOack++ = 's';\r
            *pOack++ = 'i';\r
            *pOack++ = 'z';\r
            *pOack++ = 'e';\r
            *pOack++ = 0;\r
            pOack = TftpOptionSet ( pOack, pContext->LengthInBytes );\r
            *pOack++ = 0;\r
            pContext->TxBytes += pOack - pTemp;\r
          }\r
        }\r
      }\r
      else {\r
        //\r
        //  Unknown option - Ignore it\r
        //\r
        DEBUG (( DEBUG_WARN | DEBUG_TFTP_REQUEST,\r
                  "WARNING - Skipping unknown option: %a\r\n",\r
                  pOption ));\r
      }\r
    }\r
\r
    //\r
    //  Set the next option\r
    //\r
    pOption = pNextOption;\r
  } while ( pEnd > pOption );\r
}\r
\r
\r
/**\r
  Process the TFTP request\r
\r
  @param [in] pOption   Address of the first zero terminated option string\r
  @param [in] pValue    Address to receive the value\r
\r
  @retval EFI_SUCCESS   Option translated into a value\r
\r
**/\r
EFI_STATUS\r
TftpOptionValue (\r
  IN UINT8 * pOption,\r
  IN INT32 * pValue\r
  )\r
{\r
  UINT8 Digit;\r
  EFI_STATUS Status;\r
  INT32 Value;\r
\r
  //\r
  //  Assume success\r
  //\r
  Status = EFI_SUCCESS;\r
\r
  //\r
  //  Walk the characters in the option\r
  //\r
  Value = 0;\r
  while ( 0 != *pOption ) {\r
    //\r
    //  Convert the next digit to binary\r
    //\r
    Digit = *pOption++;\r
    if (( '0' <= Digit ) && ( '9' >= Digit )) {\r
      Value *= 10;\r
      Value += Digit - '0';\r
    }\r
    else {\r
      DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r
                "ERROR - Invalid character '0x%02x' in the value\r\n",\r
                Digit ));\r
      Status = EFI_INVALID_PARAMETER;\r
      break;\r
    }\r
  }\r
\r
  //\r
  //  Return the value\r
  //\r
  *pValue = Value;\r
\r
  //\r
  //  Return the conversion status\r
  //\r
  return Status;\r
}\r
\r
\r
/**\r
  Process the TFTP request\r
\r
  @param [in] pTftpServer The TFTP server control structure address.\r
  @param [in] pContext    Connection context structure address\r
\r
**/\r
VOID\r
TftpProcessRequest (\r
  IN TSDT_TFTP_SERVER * pTftpServer,\r
  IN TSDT_CONNECTION_CONTEXT * pContext\r
  )\r
{\r
  BOOLEAN bCloseContext;\r
  BOOLEAN bIgnorePacket;\r
  UINT16 BlockNumber;\r
  UINT16 Opcode;\r
  UINT8 * pBuffer;\r
  UINT8 * pEnd;\r
  UINT8 * pFileName;\r
  UINT8 * pMode;\r
  UINT8 * pOption;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Get the opcode\r
  //\r
  pBuffer = &pTftpServer->RxBuffer[0];\r
  Opcode = HTONS ( *(UINT16 *)&pBuffer[0]);\r
Print ( L"TFTP Opcode: 0x%08x\r\n", Opcode );\r
\r
  //\r
  //  Validate the parameters\r
  //\r
  bCloseContext = FALSE;\r
  bIgnorePacket = FALSE;\r
  switch ( Opcode ) {\r
  default:\r
    DEBUG (( DEBUG_TFTP_REQUEST,\r
              "ERROR - Unknown TFTP opcode: %d\r\n",\r
              Opcode ));\r
    bIgnorePacket = TRUE;\r
    break;\r
\r
  case TFTP_OP_READ_REQUEST:\r
    break;\r
\r
  case TFTP_OP_DATA:\r
    if ( NULL == pContext ) {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r
                (UINT8)pTftpServer->RemoteAddress.sin_addr.s_addr,\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 8 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 16 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 24 ),\r
                htons ( pTftpServer->RemoteAddress.sin_port )));\r
      bIgnorePacket = TRUE;\r
      break;\r
    }\r
    if ( pContext->bExpectAck ) {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - Expecting ACKs not data for pContext 0x%08x\r\n",\r
                pContext ));\r
      bIgnorePacket = TRUE;\r
      break;\r
    }\r
    if ( pTftpServer->RxBytes > (ssize_t)( pContext->BlockSize + 2 + 2 ))\r
    {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - Receive data length of %d > %d bytes (maximum block size) for pContext 0x%08x\r\n",\r
                pTftpServer->RxBytes - 2 - 2,\r
                pContext->BlockSize,\r
                pContext ));\r
      bIgnorePacket = TRUE;\r
      break;\r
    }\r
    break;\r
\r
  case TFTP_OP_ACK:\r
    if ( NULL == pContext ) {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r
                (UINT8)pTftpServer->RemoteAddress.sin_addr.s_addr,\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 8 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 16 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 24 ),\r
                htons ( pTftpServer->RemoteAddress.sin_port )));\r
      bIgnorePacket = TRUE;\r
    }\r
    if ( !pContext->bExpectAck ) {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - Expecting data not ACKs for pContext 0x%08x\r\n",\r
                pContext ));\r
      bIgnorePacket = TRUE;\r
      break;\r
    }\r
    break;\r
\r
  case TFTP_OP_ERROR:\r
    if ( NULL == pContext ) {\r
      DEBUG (( DEBUG_ERROR,\r
                "ERROR - File not open for %d.%d.%d.%d:%d\r\n",\r
                (UINT8)pTftpServer->RemoteAddress.sin_addr.s_addr,\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 8 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 16 ),\r
                (UINT8)( pTftpServer->RemoteAddress.sin_addr.s_addr >> 24 ),\r
                htons ( pTftpServer->RemoteAddress.sin_port )));\r
      bIgnorePacket = TRUE;\r
    }\r
    break;\r
  }\r
  if ( !bIgnorePacket ) {\r
    //\r
    //  Process the request\r
    //\r
    switch ( Opcode ) {\r
    default:\r
      DEBUG (( DEBUG_TFTP_REQUEST,\r
                "ERROR - Unable to process TFTP opcode: %d\r\n",\r
                Opcode ));\r
      break;\r
\r
    case TFTP_OP_READ_REQUEST:\r
\r
      //\r
      //  Close the context if necessary\r
      //\r
      if ( NULL != pContext ) {\r
        ContextRemove ( pTftpServer, pContext );\r
      }\r
\r
      //\r
      //  Create the connection context\r
      //\r
      pContext = ContextAdd ( pTftpServer );\r
      if ( NULL == pContext ) {\r
        break;\r
      }\r
\r
      //\r
      //  Locate the mode\r
      //\r
      pFileName = &pBuffer[2];\r
      pEnd = &pBuffer[pTftpServer->RxBytes];\r
      pMode = pFileName;\r
      while (( pEnd > pMode ) && ( 0 != *pMode )) {\r
        pMode += 1;\r
      }\r
      if ( pEnd <= pMode ) {\r
        //\r
        //  Mode not found\r
        //\r
        DEBUG (( DEBUG_ERROR | DEBUG_RX,\r
                  "ERROR - File mode not found\r\n" ));\r
        //\r
        //  Tell the client of the error\r
        //\r
        TftpSendError ( pTftpServer,\r
                        pContext,\r
                        0,\r
                        (UINT8 *)"File open mode not found" );\r
        break;\r
      }\r
      pMode += 1;\r
      DEBUG (( DEBUG_TFTP_REQUEST,\r
                "TFTP - FileName: %a\n",\r
                pFileName ));\r
\r
      //\r
      //  Locate the options\r
      //\r
      pOption = pMode;\r
      while (( pEnd > pOption ) && ( 0 != *pOption )) {\r
        pOption += 1;\r
      }\r
      if ( pEnd <= pOption ) {\r
        //\r
        //  End of mode not found\r
        //\r
        DEBUG (( DEBUG_ERROR | DEBUG_RX,\r
                  "ERROR - File mode not valid\r\n" ));\r
        //\r
        //  Tell the client of the error\r
        //\r
        TftpSendError ( pTftpServer,\r
                        pContext,\r
                        0,\r
                        (UINT8 *)"File open mode not valid" );\r
        break;\r
      }\r
      pOption += 1;\r
      DEBUG (( DEBUG_TFTP_REQUEST,\r
                "TFTP - Mode: %a\r\n",\r
                pMode ));\r
\r
      //\r
      //  Verify the mode is supported\r
      //\r
      if ( 0 != stricmp ((char *)pMode, "octet" )) {\r
        //\r
        //  File access mode not supported\r
        //\r
        DEBUG (( DEBUG_ERROR | DEBUG_TFTP_REQUEST,\r
                  "ERROR - File mode %a not supported\r\n",\r
                  pMode ));\r
\r
        //\r
        //  Tell the client of the error\r
        //\r
        TftpSendError ( pTftpServer,\r
                        pContext,\r
                        0,\r
                        (UINT8 *)"File open mode not supported" );\r
        break;\r
      }\r
\r
      //\r
      //  Open the file, close the context on error\r
      //\r
// TODO: Remove the following line\r
pContext->File = (EFI_HANDLE)1;\r
\r
      //\r
      //  Determine the file length\r
      //\r
//fstat\r
\r
      //\r
      //  Process the options\r
      //\r
      TftpOptions ( pContext, pOption, pEnd );\r
\r
      //\r
      //  Read in the first portion of the file\r
      //\r
\r
      //\r
      //  Send the first block\r
      //\r
      pContext->bExpectAck = TRUE;\r
      if ( 2 < pContext->TxBytes ) {\r
        //\r
        //  Send the OACK\r
        //\r
        Status = TftpTxPacket ( pTftpServer, pContext );\r
      }\r
      else {\r
        //\r
        //  Send the first block of data\r
        //\r
        Status = TftpSendNextBlock ( pTftpServer, pContext );\r
      }\r
      break;\r
\r
    case TFTP_OP_ACK:\r
      //\r
      //  Get the block number that is being ACKed\r
      //\r
      BlockNumber = pTftpServer->RxBuffer[2];\r
      BlockNumber <<= 8;\r
      BlockNumber |= pTftpServer->RxBuffer[3];\r
\r
      //\r
      //  Determine if this is the correct ACK\r
      //\r
      DEBUG (( DEBUG_TFTP_ACK,\r
                "ACK for block 0x%04x received\r\n",\r
                BlockNumber ));\r
      if (( !pContext->bExpectAck )\r
        || ( BlockNumber != pContext->AckNext ))\r
      {\r
        DEBUG (( DEBUG_WARN | DEBUG_TFTP_ACK,\r
                  "WARNING - Expecting ACK 0x%0x4 not received ACK 0x%08x\r\n",\r
                  pContext->AckNext,\r
                  BlockNumber ));\r
      }\r
      else {\r
        //\r
        //  Process the expected ACK\r
        //\r
        if ( pContext->bEofSent ) {\r
          bCloseContext = TRUE;\r
        }\r
        else {\r
          //\r
          //  Set the next expected ACK\r
          //\r
          pContext->AckNext += 1;\r
\r
          //\r
          //  Send the next packet of data\r
          //\r
          Status = TftpSendNextBlock ( pTftpServer, pContext );\r
        }\r
      }\r
      break;\r
    }\r
  }\r
\r
  //\r
  //  Determine if the context should be closed\r
  //\r
  if ( bCloseContext ) {\r
    ContextRemove ( pTftpServer, pContext );\r
  }\r
\r
  DBG_EXIT ( );\r
}\r
\r
\r
/**\r
  Build and send an error packet\r
\r
  @param [in] pTftpServer The TFTP server control structure address.\r
  @param [in] pContext    The context structure address.\r
  @param [in] Error       Error number for the packet\r
  @param [in] pError      Zero terminated error string address\r
\r
  @retval EFI_SUCCESS     Message processed successfully\r
\r
**/\r
EFI_STATUS\r
TftpSendError (\r
  IN TSDT_TFTP_SERVER * pTftpServer,\r
  IN TSDT_CONNECTION_CONTEXT * pContext,\r
  IN UINT16 Error,\r
  IN UINT8 * pError\r
  )\r
{\r
  UINT8 Character;\r
  UINT8 * pBuffer;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Build the error packet\r
  //\r
  pBuffer = &pContext->TxBuffer[0];\r
  pBuffer[0] = 0;\r
  pBuffer[1] = TFTP_OP_ERROR;\r
  pBuffer[2] = (UINT8)( Error >> 8 );\r
  pBuffer[3] = (UINT8)Error;\r
\r
  //\r
  //  Copy the zero terminated string into the buffer\r
  //\r
  pBuffer += 4;\r
  do {\r
    Character = *pError++;\r
    *pBuffer++ = Character;\r
  } while ( 0 != Character );\r
\r
  //\r
  //  Send the error message\r
  //\r
  pContext->TxBytes = pBuffer - &pContext->TxBuffer[0];\r
  Status = TftpTxPacket ( pTftpServer, pContext );\r
\r
  //\r
  //  Return the operation status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
\r
/**\r
  Send the next block of file system data\r
\r
  @param [in] pTftpServer The TFTP server control structure address.\r
  @param [in] pContext    The context structure address.\r
\r
  @retval EFI_SUCCESS   Message processed successfully\r
\r
**/\r
EFI_STATUS\r
TftpSendNextBlock (\r
  IN TSDT_TFTP_SERVER * pTftpServer,\r
  IN TSDT_CONNECTION_CONTEXT * pContext\r
  )\r
{\r
  ssize_t LengthInBytes;\r
  UINT8 * pBuffer;\r
  EFI_STATUS Status;\r
\r
  //\r
  //  Determine how much data needs to be sent\r
  //\r
  LengthInBytes = pContext->BlockSize;\r
  if (( pContext->LengthInBytes < TFTP_MAX_BLOCK_SIZE )\r
    || ( LengthInBytes > (ssize_t)pContext->LengthInBytes )) {\r
    LengthInBytes = (ssize_t)pContext->LengthInBytes;\r
    pContext->bEofSent = TRUE;\r
  }\r
\r
  //\r
  //  Set the TFTP opcode and block number\r
  //\r
  pBuffer = &pContext->TxBuffer[0];\r
  *pBuffer++ = 0;\r
  *pBuffer++ = TFTP_OP_DATA;\r
  *pBuffer++ = (UINT8)( pContext->AckNext >> 8 );\r
  *pBuffer++ = (UINT8)pContext->AckNext;\r
\r
  //\r
  //  Copy the file data into the transmit buffer\r
  //\r
  pContext->TxBytes = 2 + 2 + LengthInBytes;\r
  if ( 0 < LengthInBytes ) {\r
    CopyMem ( &pBuffer,\r
              pContext->pBuffer,\r
              LengthInBytes );\r
  }\r
\r
  //\r
  //  Send the next block\r
  //\r
  Status = TftpTxPacket ( pTftpServer, pContext );\r
\r
  //\r
  //  Return the operation status\r
  //\r
  return Status;\r
}\r
\r
\r
/**\r
  Create the port for the TFTP server\r
\r
  This routine polls the network layer to create the TFTP port for the\r
  TFTP server.  More than one attempt may be necessary since it may take\r
  some time to get the IP address and initialize the upper layers of\r
  the network stack.\r
\r
  @param [in] pTftpServer  The TFTP server control structure address.\r
\r
**/\r
VOID\r
TftpServerTimer (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  UINT16 TftpPort;\r
  int SocketStatus;\r
  EFI_STATUS Status;\r
\r
  DEBUG (( DEBUG_SERVER_TIMER, "Entering TftpServerTimer\r\n" ));\r
\r
  //\r
  //  Open the TFTP port on the server\r
  //\r
  do {\r
    do {\r
      //\r
      //  Wait for a while\r
      //\r
      Status = gBS->CheckEvent ( pTftpServer->TimerEvent );\r
    } while ( EFI_SUCCESS != Status );\r
\r
    //\r
    //  Attempt to create the socket for the TFTP server\r
    //\r
    pTftpServer->TftpPort.events = POLLRDNORM | POLLHUP;\r
    pTftpServer->TftpPort.revents = 0;\r
    pTftpServer->TftpPort.fd = socket ( AF_INET,\r
                                        SOCK_DGRAM,\r
                                        IPPROTO_UDP );\r
    if ( -1 != pTftpServer->TftpPort.fd )\r
    {\r
      //\r
      //  Set the socket address\r
      //\r
      ZeroMem ( &pTftpServer->TftpServerAddress,\r
                sizeof ( pTftpServer->TftpServerAddress ));\r
      TftpPort = 69;\r
      DEBUG (( DEBUG_TFTP_PORT,\r
                "TFTP Port: %d\r\n",\r
                TftpPort ));\r
      pTftpServer->TftpServerAddress.sin_len = sizeof ( pTftpServer->TftpServerAddress );\r
      pTftpServer->TftpServerAddress.sin_family = AF_INET;\r
      pTftpServer->TftpServerAddress.sin_addr.s_addr = INADDR_ANY;\r
      pTftpServer->TftpServerAddress.sin_port = htons ( TftpPort );\r
\r
      //\r
      //  Bind the socket to the TFTP port\r
      //\r
      SocketStatus = bind ( pTftpServer->TftpPort.fd,\r
                            (struct sockaddr *) &pTftpServer->TftpServerAddress,\r
                            pTftpServer->TftpServerAddress.sin_len );\r
      if ( -1 != SocketStatus ) {\r
        DEBUG (( DEBUG_TFTP_PORT,\r
                  "0x%08x: Socket bound to port %d\r\n",\r
                  pTftpServer->TftpPort.fd,\r
                  TftpPort ));\r
      }\r
\r
      //\r
      //  Release the socket if necessary\r
      //\r
      if ( -1 == SocketStatus ) {\r
        close ( pTftpServer->TftpPort.fd );\r
        pTftpServer->TftpPort.fd = -1;\r
      }\r
    }\r
\r
    //\r
    //  Wait until the socket is open\r
    //\r
  }while ( -1 == pTftpServer->TftpPort.fd );\r
\r
  DEBUG (( DEBUG_SERVER_TIMER, "Exiting TftpServerTimer\r\n" ));\r
}\r
\r
\r
/**\r
  Start the TFTP server port creation timer\r
\r
  @param [in] pTftpServer The TFTP server control structure address.\r
\r
  @retval EFI_SUCCESS         The timer was successfully started.\r
  @retval EFI_ALREADY_STARTED The timer is already running.\r
  @retval Other               The timer failed to start.\r
\r
**/\r
EFI_STATUS\r
TftpServerTimerStart (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  EFI_STATUS Status;\r
  UINT64 TriggerTime;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Assume the timer is already running\r
  //\r
  Status = EFI_ALREADY_STARTED;\r
  if ( !pTftpServer->bTimerRunning ) {\r
    //\r
    //  Compute the poll interval\r
    //\r
    TriggerTime = TFTP_PORT_POLL_DELAY * ( 1000 * 10 );\r
    Status = gBS->SetTimer ( pTftpServer->TimerEvent,\r
                             TimerPeriodic,\r
                             TriggerTime );\r
    if ( !EFI_ERROR ( Status )) {\r
      DEBUG (( DEBUG_TFTP_PORT, "TFTP port timer started\r\n" ));\r
\r
      //\r
      //  Mark the timer running\r
      //\r
      pTftpServer->bTimerRunning = TRUE;\r
    }\r
    else {\r
      DEBUG (( DEBUG_ERROR | DEBUG_TFTP_PORT,\r
                "ERROR - Failed to start TFTP port timer, Status: %r\r\n",\r
                Status ));\r
    }\r
  }\r
\r
  //\r
  //  Return the operation status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
\r
/**\r
  Stop the TFTP server port creation timer\r
\r
  @param [in] pTftpServer The TFTP server control structure address.\r
\r
  @retval EFI_SUCCESS   The TFTP port timer is stopped\r
  @retval Other         Failed to stop the TFTP port timer\r
\r
**/\r
EFI_STATUS\r
TftpServerTimerStop (\r
  IN TSDT_TFTP_SERVER * pTftpServer\r
  )\r
{\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Assume the timer is stopped\r
  //\r
  Status = EFI_SUCCESS;\r
  if ( pTftpServer->bTimerRunning ) {\r
    //\r
    //  Stop the port creation polling\r
    //\r
    Status = gBS->SetTimer ( pTftpServer->TimerEvent,\r
                             TimerCancel,\r
                             0 );\r
    if ( !EFI_ERROR ( Status )) {\r
      DEBUG (( DEBUG_TFTP_PORT, "TFT[ port timer stopped\r\n" ));\r
\r
      //\r
      //  Mark the timer stopped\r
      //\r
      pTftpServer->bTimerRunning = FALSE;\r
    }\r
    else {\r
      DEBUG (( DEBUG_ERROR | DEBUG_TFTP_PORT,\r
                "ERROR - Failed to stop TFT[ port timer, Status: %r\r\n",\r
                Status ));\r
    }\r
  }\r
\r
  //\r
  //  Return the operation status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
/**\r
  Send the next TFTP packet\r
\r
  @param [in] pTftpServer   The TFTP server control structure address.\r
  @param [in] pContext      The context structure address.\r
\r
  @retval EFI_SUCCESS   Message processed successfully\r
\r
**/\r
EFI_STATUS\r
TftpTxPacket (\r
  IN TSDT_TFTP_SERVER * pTftpServer,\r
  IN TSDT_CONNECTION_CONTEXT * pContext\r
  )\r
{\r
  ssize_t LengthInBytes;\r
  EFI_STATUS Status;\r
\r
  DBG_ENTER ( );\r
\r
  //\r
  //  Assume success\r
  //\r
  Status = EFI_SUCCESS;\r
\r
  //\r
  //  Send the TFTP packet\r
  //\r
  DEBUG (( DEBUG_TX,\r
            "0x%08x: pContext sending 0x%08x bytes\r\n",\r
            pContext,\r
            pContext->TxBytes ));\r
  LengthInBytes = sendto ( pTftpServer->TftpPort.fd,\r
                           &pContext->TxBuffer[0],\r
                           pContext->TxBytes,\r
                           0,\r
                           (struct sockaddr *)&pContext->RemoteAddress,\r
                           pContext->RemoteAddress.sin_len );\r
  if ( -1 == LengthInBytes ) {\r
    DEBUG (( DEBUG_ERROR | DEBUG_TX,\r
              "ERROR - Transmit failure, errno: 0x%08x\r\n",\r
              errno ));\r
    Status = EFI_DEVICE_ERROR;\r
  }\r
\r
  //\r
  //  Return the operation status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r
\r
\r
/**\r
  Entry point for the TFTP server application.\r
\r
  @param [in] Argc  The number of arguments\r
  @param [in] Argv  The argument value array\r
\r
  @retval  0        The application exited normally.\r
  @retval  Other    An error occurred.\r
**/\r
int\r
main (\r
  IN int Argc,\r
  IN char **Argv\r
  )\r
{\r
  TSDT_TFTP_SERVER * pTftpServer;\r
  EFI_STATUS Status;\r
\r
  //\r
  //  Create a timer event to start TFTP port\r
  //\r
  pTftpServer = &mTftpServer;\r
  Status = gBS->CreateEvent ( EVT_TIMER,\r
                              TPL_TFTP_SERVER,\r
                              NULL,\r
                              NULL,\r
                              &pTftpServer->TimerEvent );\r
  if ( !EFI_ERROR ( Status )) {\r
    Status = TftpServerTimerStart ( pTftpServer );\r
    if ( !EFI_ERROR ( Status )) {\r
      //\r
      //  Run the TFTP server forever\r
      //\r
      for ( ; ; ) {\r
        //\r
        //  Poll the network layer to create the TFTP port\r
        //  for the tftp server.  More than one attempt may\r
        //  be necessary since it may take some time to get\r
        //  the IP address and initialize the upper layers\r
        //  of the network stack.\r
        //\r
        TftpServerTimer ( pTftpServer );\r
\r
        //\r
        //  Poll the socket for activity\r
        //\r
        do {\r
          SocketPoll ( pTftpServer );\r
        } while ( -1 != pTftpServer->TftpPort.fd );\r
\r
//\r
// TODO: Remove the following test code\r
//  Exit when the network connection is broken\r
//\r
break;\r
      }\r
\r
      //\r
      //  Done with the timer event\r
      //\r
      TftpServerTimerStop ( pTftpServer );\r
      Status = gBS->CloseEvent ( pTftpServer->TimerEvent );\r
    }\r
  }\r
\r
  //\r
  //  Return the final status\r
  //\r
  DBG_EXIT_STATUS ( Status );\r
  return Status;\r
}\r