]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.h
MdeModulePkg/Network: Fix potential ASSERT if NetIp4IsUnicast is called
[mirror_edk2.git] / MdeModulePkg / Universal / Network / Mtftp4Dxe / Mtftp4Impl.h
index 04ca1a1eb38ae876a0e4faf3c3ed7a37a7b388ce..527fd1db10d1723b4c6d75023f3c4f90911595d7 100644 (file)
@@ -1,46 +1,42 @@
 /** @file\r
-\r
-Copyright (c) 2006 - 2007, Intel Corporation\r
-All rights reserved. This program and the accompanying materials\r
+  \r
+  Mtftp4 Implementation.\r
+  \r
+  Mtftp4 Implementation, it supports the following RFCs:\r
+  RFC1350 - THE TFTP PROTOCOL (REVISION 2)\r
+  RFC2090 - TFTP Multicast Option\r
+  RFC2347 - TFTP Option Extension\r
+  RFC2348 - TFTP Blocksize Option\r
+  RFC2349 - TFTP Timeout Interval and Transfer Size Options\r
+  \r
+Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>\r
+This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
-http://opensource.org/licenses/bsd-license.php\r
+http://opensource.org/licenses/bsd-license.php<BR>\r
 \r
 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 \r
-Module Name:\r
-\r
-  Mtftp4Impl.h\r
-\r
-Abstract:\r
-\r
- Mtftp4 Implementation, it supports the following RFCs:\r
-   RFC1350 - THE TFTP PROTOCOL (REVISION 2)\r
-   RFC2090 - TFTP Multicast Option\r
-   RFC2347 - TFTP Option Extension\r
-   RFC2348 - TFTP Blocksize Option\r
-   RFC2349 - TFTP Timeout Interval and Transfer Size Options\r
-\r
-\r
 **/\r
 \r
+\r
 #ifndef __EFI_MTFTP4_IMPL_H__\r
 #define __EFI_MTFTP4_IMPL_H__\r
 \r
-#include <PiDxe.h>\r
+#include <Uefi.h>\r
 \r
 #include <Protocol/Udp4.h>\r
 #include <Protocol/Mtftp4.h>\r
 \r
 #include <Library/DebugLib.h>\r
-#include <Library/UefiDriverEntryPoint.h>\r
+#include <Library/BaseMemoryLib.h>\r
+#include <Library/MemoryAllocationLib.h>\r
 #include <Library/UefiBootServicesTableLib.h>\r
-#include <Library/UefiLib.h>\r
-#include <Library/BaseLib.h>\r
 #include <Library/UdpIoLib.h>\r
-#include <Library/MemoryAllocationLib.h>\r
-#include <Library/BaseMemoryLib.h>\r
+#include <Library/PrintLib.h>\r
+\r
+extern EFI_MTFTP4_PROTOCOL  gMtftp4ProtocolTemplate;\r
 \r
 typedef struct _MTFTP4_SERVICE  MTFTP4_SERVICE;\r
 typedef struct _MTFTP4_PROTOCOL MTFTP4_PROTOCOL;\r
@@ -49,31 +45,34 @@ typedef struct _MTFTP4_PROTOCOL MTFTP4_PROTOCOL;
 #include "Mtftp4Option.h"\r
 #include "Mtftp4Support.h"\r
 \r
-enum {\r
-  MTFTP4_SERVICE_SIGNATURE   = EFI_SIGNATURE_32 ('T', 'F', 'T', 'P'),\r
-  MTFTP4_PROTOCOL_SIGNATURE  = EFI_SIGNATURE_32 ('t', 'f', 't', 'p'),\r
 \r
-  MTFTP4_DEFAULT_SERVER_PORT = 69,\r
-  MTFTP4_DEFAULT_TIMEOUT     = 3,\r
-  MTFTP4_DEFAULT_RETRY       = 5,\r
-  MTFTP4_DEFAULT_BLKSIZE     = 512,\r
-  MTFTP4_TIME_TO_GETMAP      = 5,\r
+///\r
+/// Some constant value of Mtftp service.\r
+///\r
+#define MTFTP4_SERVICE_SIGNATURE    SIGNATURE_32 ('T', 'F', 'T', 'P')\r
+#define MTFTP4_PROTOCOL_SIGNATURE   SIGNATURE_32 ('t', 'f', 't', 'p')\r
 \r
-  MTFTP4_STATE_UNCONFIGED    = 0,\r
-  MTFTP4_STATE_CONFIGED,\r
-  MTFTP4_STATE_DESTORY\r
-};\r
+#define MTFTP4_DEFAULT_SERVER_PORT  69\r
+#define MTFTP4_DEFAULT_TIMEOUT      3\r
+#define MTFTP4_DEFAULT_RETRY        5\r
+#define MTFTP4_DEFAULT_BLKSIZE      512\r
+#define MTFTP4_TIME_TO_GETMAP       5\r
 \r
+#define MTFTP4_STATE_UNCONFIGED     0\r
+#define MTFTP4_STATE_CONFIGED       1\r
+#define MTFTP4_STATE_DESTROY        2\r
+\r
+///\r
+/// Mtftp service block\r
+///\r
 struct _MTFTP4_SERVICE {\r
   UINT32                        Signature;\r
   EFI_SERVICE_BINDING_PROTOCOL  ServiceBinding;\r
 \r
-  BOOLEAN                       InDestory;\r
-\r
   UINT16                        ChildrenNum;\r
   LIST_ENTRY                    Children;\r
 \r
-  EFI_EVENT                     Timer;  // Ticking timer for all the MTFTP clients\r
+  EFI_EVENT                     Timer;  ///< Ticking timer for all the MTFTP clients\r
   EFI_EVENT                     TimerToGetMap;\r
 \r
   EFI_HANDLE                    Controller;\r
@@ -83,9 +82,10 @@ struct _MTFTP4_SERVICE {
   // This UDP child is used to keep the connection between the UDP\r
   // and MTFTP, so MTFTP will be notified when UDP is uninstalled.\r
   //\r
-  UDP_IO_PORT                   *ConnectUdp;\r
+  UDP_IO                        *ConnectUdp;\r
 };\r
 \r
+\r
 typedef struct {\r
   EFI_MTFTP4_PACKET             **Packet;\r
   UINT32                        *PacketLen;\r
@@ -98,7 +98,7 @@ struct _MTFTP4_PROTOCOL {
   EFI_MTFTP4_PROTOCOL           Mtftp4;\r
 \r
   INTN                          State;\r
-  BOOLEAN                       Indestory;\r
+  BOOLEAN                       InDestroy;\r
 \r
   MTFTP4_SERVICE                *Service;\r
   EFI_HANDLE                    Handle;\r
@@ -128,7 +128,7 @@ struct _MTFTP4_PROTOCOL {
   UINT16                        ListeningPort;\r
   UINT16                        ConnectedPort;\r
   IP4_ADDR                      Gateway;\r
-  UDP_IO_PORT                   *UnicastPort;\r
+  UDP_IO                        *UnicastPort;\r
 \r
   //\r
   // Timeout and retransmit status\r
@@ -145,27 +145,66 @@ struct _MTFTP4_PROTOCOL {
   IP4_ADDR                      McastIp;\r
   UINT16                        McastPort;\r
   BOOLEAN                       Master;\r
-  UDP_IO_PORT                   *McastUdpPort;\r
-\r
-  MTFTP4_GETINFO_STATE          GetInfoState;\r
+  UDP_IO                        *McastUdpPort;\r
 };\r
 \r
+typedef struct {\r
+  EFI_SERVICE_BINDING_PROTOCOL  *ServiceBinding;\r
+  UINTN                         NumberOfChildren;\r
+  EFI_HANDLE                    *ChildHandleBuffer;\r
+} MTFTP4_DESTROY_CHILD_IN_HANDLE_BUF_CONTEXT;\r
+\r
+/**\r
+  Clean up the MTFTP session to get ready for new operation.\r
+\r
+  @param  Instance               The MTFTP session to clean up\r
+  @param  Result                 The result to return to the caller who initiated\r
+                                 the operation.\r
+**/\r
 VOID\r
 Mtftp4CleanOperation (\r
-  IN MTFTP4_PROTOCOL            *Instance,\r
-  IN EFI_STATUS                 Result\r
+  IN OUT MTFTP4_PROTOCOL        *Instance,\r
+  IN     EFI_STATUS             Result\r
   );\r
 \r
+/**\r
+  Start the MTFTP session for upload.\r
+  \r
+  It will first init some states, then send the WRQ request packet, \r
+  and start receiving the packet.\r
+\r
+  @param  Instance              The MTFTP session\r
+  @param  Operation             Redundant parameter, which is always\r
+                                EFI_MTFTP4_OPCODE_WRQ here.\r
+\r
+  @retval EFI_SUCCESS           The upload process has been started.\r
+  @retval Others                Failed to start the upload.\r
+\r
+**/\r
 EFI_STATUS\r
 Mtftp4WrqStart (\r
-  IN MTFTP4_PROTOCOL            *Instance,\r
-  IN UINT16                     Operation\r
+  IN MTFTP4_PROTOCOL        *Instance,\r
+  IN UINT16                 Operation\r
   );\r
 \r
+/**\r
+  Start the MTFTP session to download. \r
+  \r
+  It will first initialize some of the internal states then build and send a RRQ \r
+  reqeuest packet, at last, it will start receive for the downloading.\r
+\r
+  @param  Instance              The Mtftp session\r
+  @param  Operation             The MTFTP opcode, it may be a EFI_MTFTP4_OPCODE_RRQ\r
+                                or EFI_MTFTP4_OPCODE_DIR.\r
+\r
+  @retval EFI_SUCCESS           The mtftp download session is started.\r
+  @retval Others                Failed to start downloading.\r
+\r
+**/\r
 EFI_STATUS\r
 Mtftp4RrqStart (\r
-  IN MTFTP4_PROTOCOL            *Instance,\r
-  IN UINT16                     Operation\r
+  IN MTFTP4_PROTOCOL        *Instance,\r
+  IN UINT16                 Operation\r
   );\r
 \r
 #define MTFTP4_SERVICE_FROM_THIS(a)   \\r
@@ -174,5 +213,4 @@ Mtftp4RrqStart (
 #define MTFTP4_PROTOCOL_FROM_THIS(a)  \\r
   CR (a, MTFTP4_PROTOCOL, Mtftp4, MTFTP4_PROTOCOL_SIGNATURE)\r
 \r
-extern EFI_MTFTP4_PROTOCOL  gMtftp4ProtocolTemplate;\r
 #endif\r