[mirror_edk2.git] / NetworkPkg / UefiPxeBcDxe / PxeBcMtftp.c

/** @file\r
  Functions implementation related with Mtftp for UefiPxeBc Driver.\r
\r
  Copyright (c) 2007 - 2018, Intel Corporation. All rights reserved.<BR>\r
\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include "PxeBcImpl.h"\r
\r
CHAR8  *mMtftpOptions[PXE_MTFTP_OPTION_MAXIMUM_INDEX] = {\r
  "blksize",\r
  "timeout",\r
  "tsize",\r
  "multicast",\r
  "windowsize"\r
};\r
\r
/**\r
  This is a callback function when packets are received or transmitted in Mtftp driver.\r
\r
  A callback function that is provided by the caller to intercept\r
  the EFI_MTFTP6_OPCODE_DATA or EFI_MTFTP6_OPCODE_DATA8 packets processed in the\r
  EFI_MTFTP6_PROTOCOL.ReadFile() function, and alternatively to intercept\r
  EFI_MTFTP6_OPCODE_OACK or EFI_MTFTP6_OPCODE_ERROR packets during a call to\r
  EFI_MTFTP6_PROTOCOL.ReadFile(), WriteFile() or ReadDirectory().\r
\r
  @param[in]  This           Pointer to EFI_MTFTP6_PROTOCOL.\r
  @param[in]  Token          Pointer to EFI_MTFTP6_TOKEN.\r
  @param[in]  PacketLen      Length of EFI_MTFTP6_PACKET.\r
  @param[in]  Packet         Pointer to EFI_MTFTP6_PACKET to be checked.\r
\r
  @retval EFI_SUCCESS    The current operation succeeded.\r
  @retval EFI_ABORTED    Abort the current transfer process.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PxeBcMtftp6CheckPacket (\r
  IN EFI_MTFTP6_PROTOCOL  *This,\r
  IN EFI_MTFTP6_TOKEN     *Token,\r
  IN UINT16               PacketLen,\r
  IN EFI_MTFTP6_PACKET    *Packet\r
  )\r
{\r
  PXEBC_PRIVATE_DATA                   *Private;\r
  EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL  *Callback;\r
  EFI_STATUS                           Status;\r
\r
  Private  = (PXEBC_PRIVATE_DATA *)Token->Context;\r
  Callback = Private->PxeBcCallback;\r
  Status   = EFI_SUCCESS;\r
\r
  if (Packet->OpCode == EFI_MTFTP6_OPCODE_ERROR) {\r
    //\r
    // Store the tftp error message into mode data and set the received flag.\r
    //\r
    Private->Mode.TftpErrorReceived   = TRUE;\r
    Private->Mode.TftpError.ErrorCode = (UINT8)Packet->Error.ErrorCode;\r
    AsciiStrnCpyS (\r
      Private->Mode.TftpError.ErrorString,\r
      PXE_MTFTP_ERROR_STRING_LENGTH,\r
      (CHAR8 *)Packet->Error.ErrorMessage,\r
      PXE_MTFTP_ERROR_STRING_LENGTH - 1\r
      );\r
    Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';\r
  }\r
\r
  if (Callback != NULL) {\r
    //\r
    // Callback to user if has when received any tftp packet.\r
    //\r
    Status = Callback->Callback (\r
                         Callback,\r
                         Private->Function,\r
                         TRUE,\r
                         PacketLen,\r
                         (EFI_PXE_BASE_CODE_PACKET *)Packet\r
                         );\r
    if (Status != EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE) {\r
      //\r
      // User wants to abort current process if not EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE.\r
      //\r
      Status = EFI_ABORTED;\r
    } else {\r
      //\r
      // User wants to continue current process if EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE.\r
      //\r
      Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to get the size of a file using Tftp.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to EFI_MTFTP6_CONFIG_DATA.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the size of file.\r
  @retval EFI_NOT_FOUND      Parse the tftp options failed.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Has not obtained the size of the file.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp6GetFileSize (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP6_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN OUT UINT64                  *BufferSize\r
  )\r
{\r
  EFI_MTFTP6_PROTOCOL  *Mtftp6;\r
  EFI_MTFTP6_OPTION    ReqOpt[3];\r
  EFI_MTFTP6_PACKET    *Packet;\r
  EFI_MTFTP6_OPTION    *Option;\r
  UINT32               PktLen;\r
  UINT8                OptBuf[PXE_MTFTP_OPTBUF_MAXNUM_INDEX];\r
  UINTN                OptBufSize;\r
  UINT32               OptCnt;\r
  EFI_STATUS           Status;\r
\r
  *BufferSize               = 0;\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp6                    = Private->Mtftp6;\r
  Packet                    = NULL;\r
  Option                    = NULL;\r
  PktLen                    = 0;\r
  OptBufSize                = PXE_MTFTP_OPTBUF_MAXNUM_INDEX;\r
  OptCnt                    = 1;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp6->Configure (Mtftp6, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Build the required options for get info.\r
  //\r
  ReqOpt[0].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_TSIZE_INDEX];\r
  PxeBcUintnToAscDec (0, OptBuf, OptBufSize);\r
  ReqOpt[0].ValueStr = OptBuf;\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = (UINT8 *)(ReqOpt[OptCnt-1].ValueStr + AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    OptBufSize              -= (AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, OptBufSize);\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = (UINT8 *)(ReqOpt[OptCnt-1].ValueStr + AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    OptBufSize              -= (AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, OptBufSize);\r
    OptCnt++;\r
  }\r
\r
  Status = Mtftp6->GetInfo (\r
                     Mtftp6,\r
                     NULL,\r
                     Filename,\r
                     NULL,\r
                     (UINT8)OptCnt,\r
                     ReqOpt,\r
                     &PktLen,\r
                     &Packet\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    if (Status == EFI_TFTP_ERROR) {\r
      //\r
      // Store the tftp error message into mode data and set the received flag.\r
      //\r
      Private->Mode.TftpErrorReceived   = TRUE;\r
      Private->Mode.TftpError.ErrorCode = (UINT8)Packet->Error.ErrorCode;\r
      AsciiStrnCpyS (\r
        Private->Mode.TftpError.ErrorString,\r
        PXE_MTFTP_ERROR_STRING_LENGTH,\r
        (CHAR8 *)Packet->Error.ErrorMessage,\r
        PXE_MTFTP_ERROR_STRING_LENGTH - 1\r
        );\r
      Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';\r
    }\r
\r
    goto ON_ERROR;\r
  }\r
\r
  //\r
  // Parse the options in the reply packet.\r
  //\r
  OptCnt = 0;\r
  Status = Mtftp6->ParseOptions (\r
                     Mtftp6,\r
                     PktLen,\r
                     Packet,\r
                     (UINT32 *)&OptCnt,\r
                     &Option\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_ERROR;\r
  }\r
\r
  //\r
  // Parse out the value of "tsize" option.\r
  //\r
  Status = EFI_NOT_FOUND;\r
  while (OptCnt != 0) {\r
    if (AsciiStrnCmp ((CHAR8 *)Option[OptCnt - 1].OptionStr, "tsize", 5) == 0) {\r
      *BufferSize = AsciiStrDecimalToUint64 ((CHAR8 *)(Option[OptCnt - 1].ValueStr));\r
      Status      = EFI_SUCCESS;\r
    }\r
\r
    OptCnt--;\r
  }\r
\r
  FreePool (Option);\r
\r
ON_ERROR:\r
  if (Packet != NULL) {\r
    FreePool (Packet);\r
  }\r
\r
  Mtftp6->Configure (Mtftp6, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to get data of a file using Tftp.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to EFI_MTFTP6_CONFIG_DATA.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in]      BufferPtr      Pointer to buffer.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
  @param[in]      DontUseBuffer  Indicates whether with a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully read the data from the special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Read data from file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp6ReadFile (\r
  IN    PXEBC_PRIVATE_DATA       *Private,\r
  IN     EFI_MTFTP6_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize,\r
  IN     BOOLEAN                 DontUseBuffer\r
  )\r
{\r
  EFI_MTFTP6_PROTOCOL  *Mtftp6;\r
  EFI_MTFTP6_TOKEN     Token;\r
  EFI_MTFTP6_OPTION    ReqOpt[2];\r
  UINT32               OptCnt;\r
  UINT8                BlksizeBuf[10];\r
  UINT8                WindowsizeBuf[10];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp6                    = Private->Mtftp6;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp6->Configure (Mtftp6, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = BlksizeBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, sizeof (BlksizeBuf));\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = WindowsizeBuf;\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, sizeof (WindowsizeBuf));\r
    OptCnt++;\r
  }\r
\r
  Token.Event        = NULL;\r
  Token.OverrideData = NULL;\r
  Token.Filename     = Filename;\r
  Token.ModeStr      = NULL;\r
  Token.OptionCount  = OptCnt;\r
  Token.OptionList   = ReqOpt;\r
  Token.Context      = Private;\r
\r
  if (DontUseBuffer) {\r
    Token.BufferSize = 0;\r
    Token.Buffer     = NULL;\r
  } else {\r
    Token.BufferSize = *BufferSize;\r
    Token.Buffer     = BufferPtr;\r
  }\r
\r
  Token.CheckPacket     = PxeBcMtftp6CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp6->ReadFile (Mtftp6, &Token);\r
  //\r
  // Get the real size of received buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp6->Configure (Mtftp6, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is used to write the data of a file using Tftp.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to EFI_MTFTP6_CONFIG_DATA.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       Overwrite      Indicate whether with overwrite attribute.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully wrote the data into a special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval other              Write data into file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp6WriteFile (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP6_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     BOOLEAN                 Overwrite,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize\r
  )\r
{\r
  EFI_MTFTP6_PROTOCOL  *Mtftp6;\r
  EFI_MTFTP6_TOKEN     Token;\r
  EFI_MTFTP6_OPTION    ReqOpt[1];\r
  UINT32               OptCnt;\r
  UINT8                OptBuf[128];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp6                    = Private->Mtftp6;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp6->Configure (Mtftp6, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[0].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[0].ValueStr  = OptBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[0].ValueStr, PXE_MTFTP_OPTBUF_MAXNUM_INDEX);\r
    OptCnt++;\r
  }\r
\r
  Token.Event           = NULL;\r
  Token.OverrideData    = NULL;\r
  Token.Filename        = Filename;\r
  Token.ModeStr         = NULL;\r
  Token.OptionCount     = OptCnt;\r
  Token.OptionList      = ReqOpt;\r
  Token.BufferSize      = *BufferSize;\r
  Token.Buffer          = BufferPtr;\r
  Token.CheckPacket     = PxeBcMtftp6CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp6->WriteFile (Mtftp6, &Token);\r
  //\r
  // Get the real size of transmitted buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp6->Configure (Mtftp6, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to read the data (file) from a directory using Tftp.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to EFI_MTFTP6_CONFIG_DATA.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       WindowSize     Pointer to required window size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
  @param[in]       DontUseBuffer  Indicates whether to use a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the data from the file included in directory.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Operation failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp6ReadDirectory (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP6_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize,\r
  IN     BOOLEAN                 DontUseBuffer\r
  )\r
{\r
  EFI_MTFTP6_PROTOCOL  *Mtftp6;\r
  EFI_MTFTP6_TOKEN     Token;\r
  EFI_MTFTP6_OPTION    ReqOpt[2];\r
  UINT32               OptCnt;\r
  UINT8                BlksizeBuf[10];\r
  UINT8                WindowsizeBuf[10];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp6                    = Private->Mtftp6;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp6->Configure (Mtftp6, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = BlksizeBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, sizeof (BlksizeBuf));\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = WindowsizeBuf;\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, sizeof (WindowsizeBuf));\r
    OptCnt++;\r
  }\r
\r
  Token.Event        = NULL;\r
  Token.OverrideData = NULL;\r
  Token.Filename     = Filename;\r
  Token.ModeStr      = NULL;\r
  Token.OptionCount  = OptCnt;\r
  Token.OptionList   = ReqOpt;\r
  Token.Context      = Private;\r
\r
  if (DontUseBuffer) {\r
    Token.BufferSize = 0;\r
    Token.Buffer     = NULL;\r
  } else {\r
    Token.BufferSize = *BufferSize;\r
    Token.Buffer     = BufferPtr;\r
  }\r
\r
  Token.CheckPacket     = PxeBcMtftp6CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp6->ReadDirectory (Mtftp6, &Token);\r
  //\r
  // Get the real size of received buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp6->Configure (Mtftp6, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This is a callback function when packets are received or transmitted in Mtftp driver.\r
\r
  A callback function that is provided by the caller to intercept\r
  the EFI_MTFTP6_OPCODE_DATA or EFI_MTFTP4_OPCODE_DATA8 packets processed in the\r
  EFI_MTFTP4_PROTOCOL.ReadFile() function, and alternatively to intercept\r
  EFI_MTFTP4_OPCODE_OACK or EFI_MTFTP4_OPCODE_ERROR packets during a call to\r
  EFI_MTFTP4_PROTOCOL.ReadFile(), WriteFile() or ReadDirectory().\r
\r
  @param[in]  This           Pointer to EFI_MTFTP4_PROTOCOL.\r
  @param[in]  Token          Pointer to EFI_MTFTP4_TOKEN.\r
  @param[in]  PacketLen      Length of EFI_MTFTP4_PACKET.\r
  @param[in]  Packet         Pointer to EFI_MTFTP4_PACKET to be checked.\r
\r
  @retval EFI_SUCCESS    The current operation succeeded.\r
  @retval EFI_ABORTED    Abort the current transfer process.\r
\r
**/\r
EFI_STATUS\r
EFIAPI\r
PxeBcMtftp4CheckPacket (\r
  IN EFI_MTFTP4_PROTOCOL  *This,\r
  IN EFI_MTFTP4_TOKEN     *Token,\r
  IN UINT16               PacketLen,\r
  IN EFI_MTFTP4_PACKET    *Packet\r
  )\r
{\r
  PXEBC_PRIVATE_DATA                   *Private;\r
  EFI_PXE_BASE_CODE_CALLBACK_PROTOCOL  *Callback;\r
  EFI_STATUS                           Status;\r
\r
  Private  = (PXEBC_PRIVATE_DATA *)Token->Context;\r
  Callback = Private->PxeBcCallback;\r
  Status   = EFI_SUCCESS;\r
\r
  if (Packet->OpCode == EFI_MTFTP4_OPCODE_ERROR) {\r
    //\r
    // Store the tftp error message into mode data and set the received flag.\r
    //\r
    Private->Mode.TftpErrorReceived   = TRUE;\r
    Private->Mode.TftpError.ErrorCode = (UINT8)Packet->Error.ErrorCode;\r
    AsciiStrnCpyS (\r
      Private->Mode.TftpError.ErrorString,\r
      PXE_MTFTP_ERROR_STRING_LENGTH,\r
      (CHAR8 *)Packet->Error.ErrorMessage,\r
      PXE_MTFTP_ERROR_STRING_LENGTH - 1\r
      );\r
    Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';\r
  }\r
\r
  if (Callback != NULL) {\r
    //\r
    // Callback to user if has when received any tftp packet.\r
    //\r
    Status = Callback->Callback (\r
                         Callback,\r
                         Private->Function,\r
                         TRUE,\r
                         PacketLen,\r
                         (EFI_PXE_BASE_CODE_PACKET *)Packet\r
                         );\r
    if (Status != EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE) {\r
      //\r
      // User wants to abort current process if not EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE.\r
      //\r
      Status = EFI_ABORTED;\r
    } else {\r
      //\r
      // User wants to continue current process if EFI_PXE_BASE_CODE_CALLBACK_STATUS_CONTINUE.\r
      //\r
      Status = EFI_SUCCESS;\r
    }\r
  }\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to get size of a file using Tftp.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to EFI_MTFTP4_CONFIG_DATA.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the size of file.\r
  @retval EFI_NOT_FOUND      Parse the tftp options failed.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Did not obtain the size of the file.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp4GetFileSize (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP4_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN OUT UINT64                  *BufferSize\r
  )\r
{\r
  EFI_MTFTP4_PROTOCOL  *Mtftp4;\r
  EFI_MTFTP4_OPTION    ReqOpt[3];\r
  EFI_MTFTP4_PACKET    *Packet;\r
  EFI_MTFTP4_OPTION    *Option;\r
  UINT32               PktLen;\r
  UINT8                OptBuf[PXE_MTFTP_OPTBUF_MAXNUM_INDEX];\r
  UINTN                OptBufSize;\r
  UINT32               OptCnt;\r
  EFI_STATUS           Status;\r
\r
  *BufferSize               = 0;\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp4                    = Private->Mtftp4;\r
  Packet                    = NULL;\r
  Option                    = NULL;\r
  PktLen                    = 0;\r
  OptBufSize                = PXE_MTFTP_OPTBUF_MAXNUM_INDEX;\r
  OptCnt                    = 1;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp4->Configure (Mtftp4, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  //\r
  // Build the required options for get info.\r
  //\r
  ReqOpt[0].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_TSIZE_INDEX];\r
  PxeBcUintnToAscDec (0, OptBuf, OptBufSize);\r
  ReqOpt[0].ValueStr = OptBuf;\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = (UINT8 *)(ReqOpt[OptCnt-1].ValueStr + AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    OptBufSize              -= (AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, OptBufSize);\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = (UINT8 *)(ReqOpt[OptCnt-1].ValueStr + AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    OptBufSize              -= (AsciiStrLen ((CHAR8 *)ReqOpt[OptCnt-1].ValueStr) + 1);\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, OptBufSize);\r
    OptCnt++;\r
  }\r
\r
  Status = Mtftp4->GetInfo (\r
                     Mtftp4,\r
                     NULL,\r
                     Filename,\r
                     NULL,\r
                     (UINT8)OptCnt,\r
                     ReqOpt,\r
                     &PktLen,\r
                     &Packet\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    if (Status == EFI_TFTP_ERROR) {\r
      //\r
      // Store the tftp error message into mode data and set the received flag.\r
      //\r
      Private->Mode.TftpErrorReceived   = TRUE;\r
      Private->Mode.TftpError.ErrorCode = (UINT8)Packet->Error.ErrorCode;\r
      AsciiStrnCpyS (\r
        Private->Mode.TftpError.ErrorString,\r
        PXE_MTFTP_ERROR_STRING_LENGTH,\r
        (CHAR8 *)Packet->Error.ErrorMessage,\r
        PXE_MTFTP_ERROR_STRING_LENGTH - 1\r
        );\r
      Private->Mode.TftpError.ErrorString[PXE_MTFTP_ERROR_STRING_LENGTH - 1] = '\0';\r
    }\r
\r
    goto ON_ERROR;\r
  }\r
\r
  //\r
  // Parse the options in the reply packet.\r
  //\r
  OptCnt = 0;\r
  Status = Mtftp4->ParseOptions (\r
                     Mtftp4,\r
                     PktLen,\r
                     Packet,\r
                     (UINT32 *)&OptCnt,\r
                     &Option\r
                     );\r
  if (EFI_ERROR (Status)) {\r
    goto ON_ERROR;\r
  }\r
\r
  //\r
  // Parse out the value of "tsize" option.\r
  //\r
  Status = EFI_NOT_FOUND;\r
  while (OptCnt != 0) {\r
    if (AsciiStrnCmp ((CHAR8 *)Option[OptCnt - 1].OptionStr, "tsize", 5) == 0) {\r
      *BufferSize = AsciiStrDecimalToUint64 ((CHAR8 *)(Option[OptCnt - 1].ValueStr));\r
      Status      = EFI_SUCCESS;\r
    }\r
\r
    OptCnt--;\r
  }\r
\r
  FreePool (Option);\r
\r
ON_ERROR:\r
  if (Packet != NULL) {\r
    FreePool (Packet);\r
  }\r
\r
  Mtftp4->Configure (Mtftp4, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to read the data of a file using Tftp.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to EFI_MTFTP4_CONFIG_DATA.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in]      BufferPtr      Pointer to buffer.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
  @param[in]      DontUseBuffer  Indicates whether to use a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully read the data from the special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Read data from file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp4ReadFile (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP4_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize,\r
  IN     BOOLEAN                 DontUseBuffer\r
  )\r
{\r
  EFI_MTFTP4_PROTOCOL  *Mtftp4;\r
  EFI_MTFTP4_TOKEN     Token;\r
  EFI_MTFTP4_OPTION    ReqOpt[2];\r
  UINT32               OptCnt;\r
  UINT8                BlksizeBuf[10];\r
  UINT8                WindowsizeBuf[10];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp4                    = Private->Mtftp4;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp4->Configure (Mtftp4, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = BlksizeBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, sizeof (BlksizeBuf));\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = WindowsizeBuf;\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, sizeof (WindowsizeBuf));\r
    OptCnt++;\r
  }\r
\r
  Token.Event        = NULL;\r
  Token.OverrideData = NULL;\r
  Token.Filename     = Filename;\r
  Token.ModeStr      = NULL;\r
  Token.OptionCount  = OptCnt;\r
  Token.OptionList   = ReqOpt;\r
  Token.Context      = Private;\r
\r
  if (DontUseBuffer) {\r
    Token.BufferSize = 0;\r
    Token.Buffer     = NULL;\r
  } else {\r
    Token.BufferSize = *BufferSize;\r
    Token.Buffer     = BufferPtr;\r
  }\r
\r
  Token.CheckPacket     = PxeBcMtftp4CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp4->ReadFile (Mtftp4, &Token);\r
  //\r
  // Get the real size of received buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp4->Configure (Mtftp4, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to write the data of a file using Tftp.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to EFI_MTFTP4_CONFIG_DATA.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       Overwrite      Indicates whether to use the overwrite attribute.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully write the data  into the special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval other              Write data into file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp4WriteFile (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP4_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     BOOLEAN                 Overwrite,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize\r
  )\r
{\r
  EFI_MTFTP4_PROTOCOL  *Mtftp4;\r
  EFI_MTFTP4_TOKEN     Token;\r
  EFI_MTFTP4_OPTION    ReqOpt[1];\r
  UINT32               OptCnt;\r
  UINT8                OptBuf[128];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp4                    = Private->Mtftp4;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp4->Configure (Mtftp4, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[0].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[0].ValueStr  = OptBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[0].ValueStr, PXE_MTFTP_OPTBUF_MAXNUM_INDEX);\r
    OptCnt++;\r
  }\r
\r
  Token.Event           = NULL;\r
  Token.OverrideData    = NULL;\r
  Token.Filename        = Filename;\r
  Token.ModeStr         = NULL;\r
  Token.OptionCount     = OptCnt;\r
  Token.OptionList      = ReqOpt;\r
  Token.BufferSize      = *BufferSize;\r
  Token.Buffer          = BufferPtr;\r
  Token.CheckPacket     = PxeBcMtftp4CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp4->WriteFile (Mtftp4, &Token);\r
  //\r
  // Get the real size of transmitted buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp4->Configure (Mtftp4, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is to get data (file) from a directory using Tftp.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to EFI_MTFTP4_CONFIG_DATA.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       WindowSize     Pointer to required window size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
  @param[in]       DontUseBuffer  Indicates whether to use a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the data from the file included in the directory.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Operation failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcMtftp4ReadDirectory (\r
  IN     PXEBC_PRIVATE_DATA      *Private,\r
  IN     EFI_MTFTP4_CONFIG_DATA  *Config,\r
  IN     UINT8                   *Filename,\r
  IN     UINTN                   *BlockSize,\r
  IN     UINTN                   *WindowSize,\r
  IN     UINT8                   *BufferPtr,\r
  IN OUT UINT64                  *BufferSize,\r
  IN     BOOLEAN                 DontUseBuffer\r
  )\r
{\r
  EFI_MTFTP4_PROTOCOL  *Mtftp4;\r
  EFI_MTFTP4_TOKEN     Token;\r
  EFI_MTFTP4_OPTION    ReqOpt[2];\r
  UINT32               OptCnt;\r
  UINT8                BlksizeBuf[10];\r
  UINT8                WindowsizeBuf[10];\r
  EFI_STATUS           Status;\r
\r
  Status                    = EFI_DEVICE_ERROR;\r
  Mtftp4                    = Private->Mtftp4;\r
  OptCnt                    = 0;\r
  Config->InitialServerPort = PXEBC_BS_DOWNLOAD_PORT;\r
\r
  Status = Mtftp4->Configure (Mtftp4, Config);\r
  if (EFI_ERROR (Status)) {\r
    return Status;\r
  }\r
\r
  if (BlockSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_BLKSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = BlksizeBuf;\r
    PxeBcUintnToAscDec (*BlockSize, ReqOpt[OptCnt].ValueStr, sizeof (BlksizeBuf));\r
    OptCnt++;\r
  }\r
\r
  if (WindowSize != NULL) {\r
    ReqOpt[OptCnt].OptionStr = (UINT8 *)mMtftpOptions[PXE_MTFTP_OPTION_WINDOWSIZE_INDEX];\r
    ReqOpt[OptCnt].ValueStr  = WindowsizeBuf;\r
    PxeBcUintnToAscDec (*WindowSize, ReqOpt[OptCnt].ValueStr, sizeof (WindowsizeBuf));\r
    OptCnt++;\r
  }\r
\r
  Token.Event        = NULL;\r
  Token.OverrideData = NULL;\r
  Token.Filename     = Filename;\r
  Token.ModeStr      = NULL;\r
  Token.OptionCount  = OptCnt;\r
  Token.OptionList   = ReqOpt;\r
  Token.Context      = Private;\r
\r
  if (DontUseBuffer) {\r
    Token.BufferSize = 0;\r
    Token.Buffer     = NULL;\r
  } else {\r
    Token.BufferSize = *BufferSize;\r
    Token.Buffer     = BufferPtr;\r
  }\r
\r
  Token.CheckPacket     = PxeBcMtftp4CheckPacket;\r
  Token.TimeoutCallback = NULL;\r
  Token.PacketNeeded    = NULL;\r
\r
  Status = Mtftp4->ReadDirectory (Mtftp4, &Token);\r
  //\r
  // Get the real size of received buffer.\r
  //\r
  *BufferSize = Token.BufferSize;\r
\r
  Mtftp4->Configure (Mtftp4, NULL);\r
\r
  return Status;\r
}\r
\r
/**\r
  This function is wrapper to get the file size using TFTP.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to configure data.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the size of file.\r
  @retval EFI_NOT_FOUND      Parse the tftp options failed.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Did not obtain the size of the file.\r
\r
**/\r
EFI_STATUS\r
PxeBcTftpGetFileSize (\r
  IN     PXEBC_PRIVATE_DATA  *Private,\r
  IN     VOID                *Config,\r
  IN     UINT8               *Filename,\r
  IN     UINTN               *BlockSize,\r
  IN     UINTN               *WindowSize,\r
  IN OUT UINT64              *BufferSize\r
  )\r
{\r
  if (Private->PxeBc.Mode->UsingIpv6) {\r
    return PxeBcMtftp6GetFileSize (\r
             Private,\r
             (EFI_MTFTP6_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferSize\r
             );\r
  } else {\r
    return PxeBcMtftp4GetFileSize (\r
             Private,\r
             (EFI_MTFTP4_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferSize\r
             );\r
  }\r
}\r
\r
/**\r
  This function is a wrapper to get file using TFTP.\r
\r
  @param[in]      Private        Pointer to PxeBc private data.\r
  @param[in]      Config         Pointer to config data.\r
  @param[in]      Filename       Pointer to boot file name.\r
  @param[in]      BlockSize      Pointer to required block size.\r
  @param[in]      WindowSize     Pointer to required window size.\r
  @param[in]      BufferPtr      Pointer to buffer.\r
  @param[in, out] BufferSize     Pointer to buffer size.\r
  @param[in]      DontUseBuffer  Indicates whether to use a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully read the data from the special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Read data from file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcTftpReadFile (\r
  IN     PXEBC_PRIVATE_DATA  *Private,\r
  IN     VOID                *Config,\r
  IN     UINT8               *Filename,\r
  IN     UINTN               *BlockSize,\r
  IN     UINTN               *WindowSize,\r
  IN     UINT8               *BufferPtr,\r
  IN OUT UINT64              *BufferSize,\r
  IN     BOOLEAN             DontUseBuffer\r
  )\r
{\r
  if (Private->PxeBc.Mode->UsingIpv6) {\r
    return PxeBcMtftp6ReadFile (\r
             Private,\r
             (EFI_MTFTP6_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferPtr,\r
             BufferSize,\r
             DontUseBuffer\r
             );\r
  } else {\r
    return PxeBcMtftp4ReadFile (\r
             Private,\r
             (EFI_MTFTP4_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferPtr,\r
             BufferSize,\r
             DontUseBuffer\r
             );\r
  }\r
}\r
\r
/**\r
  This function is a wrapper to write file using TFTP.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to config data.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       Overwrite      Indicate whether with overwrite attribute.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
\r
  @retval EFI_SUCCESS        Successfully wrote the data into a special file.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval other              Write data into file failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcTftpWriteFile (\r
  IN     PXEBC_PRIVATE_DATA  *Private,\r
  IN     VOID                *Config,\r
  IN     UINT8               *Filename,\r
  IN     BOOLEAN             Overwrite,\r
  IN     UINTN               *BlockSize,\r
  IN     UINT8               *BufferPtr,\r
  IN OUT UINT64              *BufferSize\r
  )\r
{\r
  if (Private->PxeBc.Mode->UsingIpv6) {\r
    return PxeBcMtftp6WriteFile (\r
             Private,\r
             (EFI_MTFTP6_CONFIG_DATA *)Config,\r
             Filename,\r
             Overwrite,\r
             BlockSize,\r
             BufferPtr,\r
             BufferSize\r
             );\r
  } else {\r
    return PxeBcMtftp4WriteFile (\r
             Private,\r
             (EFI_MTFTP4_CONFIG_DATA *)Config,\r
             Filename,\r
             Overwrite,\r
             BlockSize,\r
             BufferPtr,\r
             BufferSize\r
             );\r
  }\r
}\r
\r
/**\r
  This function is a wrapper to get the data (file) from a directory using TFTP.\r
\r
  @param[in]       Private        Pointer to PxeBc private data.\r
  @param[in]       Config         Pointer to config data.\r
  @param[in]       Filename       Pointer to boot file name.\r
  @param[in]       BlockSize      Pointer to required block size.\r
  @param[in]       WindowSize     Pointer to required window size.\r
  @param[in]       BufferPtr      Pointer to buffer.\r
  @param[in, out]  BufferSize     Pointer to buffer size.\r
  @param[in]       DontUseBuffer  Indicatse whether to use a receive buffer.\r
\r
  @retval EFI_SUCCESS        Successfully obtained the data from the file included in the directory.\r
  @retval EFI_DEVICE_ERROR   The network device encountered an error during this operation.\r
  @retval Others             Operation failed.\r
\r
**/\r
EFI_STATUS\r
PxeBcTftpReadDirectory (\r
  IN     PXEBC_PRIVATE_DATA  *Private,\r
  IN     VOID                *Config,\r
  IN     UINT8               *Filename,\r
  IN     UINTN               *BlockSize,\r
  IN     UINTN               *WindowSize,\r
  IN     UINT8               *BufferPtr,\r
  IN OUT UINT64              *BufferSize,\r
  IN     BOOLEAN             DontUseBuffer\r
  )\r
{\r
  if (Private->PxeBc.Mode->UsingIpv6) {\r
    return PxeBcMtftp6ReadDirectory (\r
             Private,\r
             (EFI_MTFTP6_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferPtr,\r
             BufferSize,\r
             DontUseBuffer\r
             );\r
  } else {\r
    return PxeBcMtftp4ReadDirectory (\r
             Private,\r
             (EFI_MTFTP4_CONFIG_DATA *)Config,\r
             Filename,\r
             BlockSize,\r
             WindowSize,\r
             BufferPtr,\r
             BufferSize,\r
             DontUseBuffer\r
             );\r
  }\r
}\r