diff --git a/MdeModulePkg/Include/Library/IpIoLib.h b/MdeModulePkg/Include/Library/IpIoLib.h index 2f6f1e8f53..14f6884d15 100644 --- a/MdeModulePkg/Include/Library/IpIoLib.h +++ b/MdeModulePkg/Include/Library/IpIoLib.h @@ -81,41 +81,36 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. extern EFI_IP4_CONFIG_DATA mIp4IoDefaultIpConfigData; extern EFI_IP6_CONFIG_DATA mIp6IoDefaultIpConfigData; -typedef enum { - IP_VERSION_4 = 4, - IP_VERSION_6 = 6 -} IP_VERSION; +#define IP_VERSION_4 4 +#define IP_VERSION_6 6 /// /// This error will be delivered to the /// listening transportation layer protocol /// that consumes IpIO. /// -typedef enum { - ICMP_ERR_UNREACH_NET = 0, - ICMP_ERR_UNREACH_HOST, - ICMP_ERR_UNREACH_PROTOCOL, - ICMP_ERR_UNREACH_PORT, - ICMP_ERR_MSGSIZE, - ICMP_ERR_UNREACH_SRCFAIL, - ICMP_ERR_TIMXCEED_INTRANS, - ICMP_ERR_TIMXCEED_REASS, - ICMP_ERR_QUENCH, - ICMP_ERR_PARAMPROB -} ICMP_ERROR; -typedef enum { - ICMP6_ERR_UNREACH_NET = 0, - ICMP6_ERR_UNREACH_HOST, - ICMP6_ERR_UNREACH_PROTOCOL, - ICMP6_ERR_UNREACH_PORT, - ICMP6_ERR_PACKAGE_TOOBIG, - ICMP6_ERR_TIMXCEED_HOPLIMIT, - ICMP6_ERR_TIMXCEED_REASS, - ICMP6_ERR_PARAMPROB_HEADER, - ICMP6_ERR_PARAMPROB_NEXHEADER, - ICMP6_ERR_PARAMPROB_IPV6OPTION -} ICMP6_ERROR; +#define ICMP_ERR_UNREACH_NET 0 +#define ICMP_ERR_UNREACH_HOST 1 +#define ICMP_ERR_UNREACH_PROTOCOL 2 +#define ICMP_ERR_UNREACH_PORT 3 +#define ICMP_ERR_MSGSIZE 4 +#define ICMP_ERR_UNREACH_SRCFAIL 5 +#define ICMP_ERR_TIMXCEED_INTRANS 6 +#define ICMP_ERR_TIMXCEED_REASS 7 +#define ICMP_ERR_QUENCH 8 +#define ICMP_ERR_PARAMPROB 9 + +#define ICMP6_ERR_UNREACH_NET 0 +#define ICMP6_ERR_UNREACH_HOST 1 +#define ICMP6_ERR_UNREACH_PROTOCOL 2 +#define ICMP6_ERR_UNREACH_PORT 3 +#define ICMP6_ERR_PACKAGE_TOOBIG 4 +#define ICMP6_ERR_TIMXCEED_HOPLIMIT 5 +#define ICMP6_ERR_TIMXCEED_REASS 6 +#define ICMP6_ERR_PARAMPROB_HEADER 7 +#define ICMP6_ERR_PARAMPROB_NEXHEADER 8 +#define ICMP6_ERR_PARAMPROB_IPV6OPTION 9 /// /// The helper struct for IpIoGetIcmpErrStatus(). It is for internal use only. @@ -150,17 +145,24 @@ typedef union { EFI_IP6_CONFIG_DATA Ip6CfgData; } IP_IO_IP_CONFIG_DATA; +typedef union { + EFI_IP4_HEADER *Ip4Hdr; + EFI_IP6_HEADER *Ip6Hdr; +} IP_IO_IP_HEADER; + +typedef union { + IP4_ADDR SubnetMask; + UINT8 PrefixLength; +} IP_IO_IP_MASK; + /// /// The IP session for an IP receive packet. /// typedef struct _EFI_NET_SESSION_DATA { EFI_IP_ADDRESS Source; ///< Source IP of the received packet EFI_IP_ADDRESS Dest; ///< Destination IP of the received packet - union { - EFI_IP4_HEADER *Ip4Hdr; ///< IP4 header of the received packet - EFI_IP6_HEADER *Ip6Hdr; ///< IP6 header of the received packet - } IpHdr; - IP_VERSION IpVersion; ///< The IP version of the received packet + IP_IO_IP_HEADER IpHdr; ///< IP header of the received packet + UINT8 IpVersion; ///< The IP version of the received packet } EFI_NET_SESSION_DATA; /** @@ -178,7 +180,7 @@ typedef VOID (*PKT_RCVD_NOTIFY) ( IN EFI_STATUS Status, - IN ICMP_ERROR IcmpErr, + IN UINT8 IcmpErr, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Pkt, IN VOID *Context @@ -249,10 +251,7 @@ typedef struct _IP_IO { VOID *SndContext; ///< See IP_IO_OPEN_DATA::SndContext PKT_RCVD_NOTIFY PktRcvdNotify; ///< See IP_IO_OPEN_DATA::PktRcvdNotify PKT_SENT_NOTIFY PktSentNotify; ///< See IP_IO_OPEN_DATA::PktSentNotify - // - // Ip Version - // - IP_VERSION IpVersion; + UINT8 IpVersion; } IP_IO; /// @@ -289,16 +288,13 @@ typedef struct _IP_IO_SEND_ENTRY { /// typedef struct _IP_IO_IP_INFO { EFI_IP_ADDRESS Addr; - union { - IP4_ADDR SubnetMask; - UINT8 PrefixLength; - } PreMask; + IP_IO_IP_MASK PreMask; LIST_ENTRY Entry; EFI_HANDLE ChildHandle; VOID *Ip; IP_IO_IP_COMPLETION_TOKEN DummyRcvToken; INTN RefCnt; - IP_VERSION IpVersion; + UINT8 IpVersion; } IP_IO_IP_INFO; /** @@ -322,7 +318,7 @@ EFIAPI IpIoCreate ( IN EFI_HANDLE Image, IN EFI_HANDLE Controller, - IN IP_VERSION IpVersion + IN UINT8 IpVersion ); /** @@ -518,7 +514,7 @@ IP_IO_IP_INFO * EFIAPI IpIoFindSender ( IN OUT IP_IO **IpIo, - IN IP_VERSION IpVersion, + IN UINT8 IpVersion, IN EFI_IP_ADDRESS *Src ); @@ -541,8 +537,8 @@ IpIoFindSender ( EFI_STATUS EFIAPI IpIoGetIcmpErrStatus ( - IN ICMP_ERROR IcmpError, - IN IP_VERSION IpVersion, + IN UINT8 IcmpError, + IN UINT8 IpVersion, OUT BOOLEAN *IsHard OPTIONAL, OUT BOOLEAN *Notify OPTIONAL ); diff --git a/MdeModulePkg/Include/Library/NetLib.h b/MdeModulePkg/Include/Library/NetLib.h index 61102a788b..1a0507b9e6 100644 --- a/MdeModulePkg/Include/Library/NetLib.h +++ b/MdeModulePkg/Include/Library/NetLib.h @@ -22,34 +22,33 @@ typedef UINT32 IP4_ADDR; typedef UINT32 TCP_SEQNO; typedef UINT16 TCP_PORTNO; -typedef enum { - NET_ETHER_ADDR_LEN = 6, - NET_IFTYPE_ETHERNET = 0x01, - EFI_IP_PROTO_UDP = 0x11, - EFI_IP_PROTO_TCP = 0x06, - EFI_IP_PROTO_ICMP = 0x01, +#define NET_ETHER_ADDR_LEN 6 +#define NET_IFTYPE_ETHERNET 0x01 - // - // The address classification - // - IP4_ADDR_CLASSA = 1, - IP4_ADDR_CLASSB, - IP4_ADDR_CLASSC, - IP4_ADDR_CLASSD, - IP4_ADDR_CLASSE, +#define EFI_IP_PROTO_UDP 0x11 +#define EFI_IP_PROTO_TCP 0x06 +#define EFI_IP_PROTO_ICMP 0x01 - IP4_MASK_NUM = 33 -} IP4_CLASS_TYPE; +// +// The address classification +// +#define IP4_ADDR_CLASSA 1 +#define IP4_ADDR_CLASSB 2 +#define IP4_ADDR_CLASSC 3 +#define IP4_ADDR_CLASSD 4 +#define IP4_ADDR_CLASSE 5 + +#define IP4_MASK_NUM 33 + + +#define IP6_HOP_BY_HOP 0 +#define IP6_DESTINATION 60 +#define IP6_FRAGMENT 44 +#define IP6_AH 51 +#define IP6_ESP 50 +#define IP6_NO_NEXT_HEADER 59 -typedef enum { - IP6_HOP_BY_HOP = 0, - IP6_DESTINATION = 60, - IP6_FRAGMENT = 44, - IP6_AH = 51, - IP6_ESP = 50, - IP6_NO_NEXT_HEADER = 59 -} IP6_EXTENSION_HEADER_TYPE; #pragma pack(1) @@ -275,6 +274,23 @@ Ip6IsValidUnicast ( IN EFI_IPv6_ADDRESS *Ip6 ); +/** + Switches the endianess of an IPv6 address + + This function swaps the bytes in a 128-bit IPv6 address to switch the value + from little endian to big endian or vice versa. The byte swapped value is + returned. + + @param Ip6 Points to an IPv6 address + + @return The byte swapped IPv6 address. + +**/ +EFI_IPv6_ADDRESS * +Ip6Swap128 ( + EFI_IPv6_ADDRESS *Ip6 + ); + extern IP4_ADDR gIp4AllMasks[IP4_MASK_NUM]; @@ -886,20 +902,19 @@ NetLibDefaultUnload ( IN EFI_HANDLE ImageHandle ); -typedef enum { - // - //Various signatures - // - NET_BUF_SIGNATURE = SIGNATURE_32 ('n', 'b', 'u', 'f'), - NET_VECTOR_SIGNATURE = SIGNATURE_32 ('n', 'v', 'e', 'c'), - NET_QUE_SIGNATURE = SIGNATURE_32 ('n', 'b', 'q', 'u'), + +// +//Various signatures +// +#define NET_BUF_SIGNATURE SIGNATURE_32 ('n', 'b', 'u', 'f') +#define NET_VECTOR_SIGNATURE SIGNATURE_32 ('n', 'v', 'e', 'c') +#define NET_QUE_SIGNATURE SIGNATURE_32 ('n', 'b', 'q', 'u') - NET_PROTO_DATA = 64, // Opaque buffer for protocols - NET_BUF_HEAD = 1, // Trim or allocate space from head - NET_BUF_TAIL = 0, // Trim or allocate space from tail - NET_VECTOR_OWN_FIRST = 0x01 // We allocated the 1st block in the vector -} NET_SIGNATURE_TYPE; +#define NET_PROTO_DATA 64 // Opaque buffer for protocols +#define NET_BUF_HEAD 1 // Trim or allocate space from head +#define NET_BUF_TAIL 0 // Trim or allocate space from tail +#define NET_VECTOR_OWN_FIRST 0x01 // We allocated the 1st block in the vector #define NET_CHECK_SIGNATURE(PData, SIGNATURE) \ ASSERT (((PData) != NULL) && ((PData)->Signature == (SIGNATURE))) diff --git a/MdeModulePkg/Include/Library/UdpIoLib.h b/MdeModulePkg/Include/Library/UdpIoLib.h index 30a46fab04..c93fb12085 100644 --- a/MdeModulePkg/Include/Library/UdpIoLib.h +++ b/MdeModulePkg/Include/Library/UdpIoLib.h @@ -2,7 +2,7 @@ This library is used to share code between UEFI network stack modules. It provides the helper routines to access UDP service. It is used by both DHCP and MTFTP. -Copyright (c) 2006 - 2008, Intel Corporation.
+Copyright (c) 2006 - 2009, 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
@@ -13,33 +13,36 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. **/ -#ifndef _UDP4IO_H_ -#define _UDP4IO_H_ +#ifndef _UDP_IO_H_ +#define _UDP_IO_H_ #include +#include #include -typedef struct _UDP_IO_PORT UDP_IO_PORT; +typedef struct _UDP_IO UDP_IO; /// /// Signatures used by UdpIo Library. /// -typedef enum { - UDP_IO_RX_SIGNATURE = SIGNATURE_32 ('U', 'D', 'P', 'R'), - UDP_IO_TX_SIGNATURE = SIGNATURE_32 ('U', 'D', 'P', 'T'), - UDP_IO_SIGNATURE = SIGNATURE_32 ('U', 'D', 'P', 'I') -} UDP_IO_SIGNATURE_TYPE; + +#define UDP_IO_RX_SIGNATURE SIGNATURE_32 ('U', 'D', 'P', 'R') +#define UDP_IO_TX_SIGNATURE SIGNATURE_32 ('U', 'D', 'P', 'T') +#define UDP_IO_SIGNATURE SIGNATURE_32 ('U', 'D', 'P', 'I') + +#define UDP_IO_UDP4_VERSION 4 +#define UDP_IO_UDP6_VERSION 6 /// -/// The Udp4 address pair. +/// The UDP address pair. /// typedef struct { - IP4_ADDR LocalAddr; + EFI_IP_ADDRESS LocalAddr; UINT16 LocalPort; - IP4_ADDR RemoteAddr; + EFI_IP_ADDRESS RemoteAddr; UINT16 RemotePort; -} UDP_POINTS; +} UDP_END_POINT; /** Prototype called when receiving or sending packets to or from a UDP point. @@ -50,7 +53,7 @@ typedef struct { provided to the callback as a reference. @param[in] Packet Packet received or sent - @param[in] Points The Udp4 address pair corresponds to the Udp4 IO + @param[in] EndPoint The UDP address pair corresponds to the UDP IO @param[in] IoStatus Packet receiving or sending status @param[in] Context User-defined data when calling UdpIoRecvDatagram() or UdpIoSendDatagram() @@ -59,7 +62,7 @@ typedef VOID (*UDP_IO_CALLBACK) ( IN NET_BUF *Packet, - IN UDP_POINTS *Points, + IN UDP_END_POINT *EndPoint, IN EFI_STATUS IoStatus, IN VOID *Context ); @@ -72,17 +75,24 @@ VOID /// time. HeadLen gives the length of the application's header. UDP_IO will /// make the application's header continuous before delivering up. /// +typedef union { + EFI_UDP4_COMPLETION_TOKEN Udp4; + EFI_UDP6_COMPLETION_TOKEN Udp6; +} UDP_COMPLETION_TOKEN; + typedef struct { - UINT32 Signature; - UDP_IO_PORT *UdpIo; + UINT32 Signature; + UDP_IO *UdpIo; - UDP_IO_CALLBACK CallBack; - VOID *Context; + UDP_IO_CALLBACK CallBack; + VOID *Context; + UINT32 HeadLen; - UINT32 HeadLen; - EFI_UDP4_COMPLETION_TOKEN UdpToken; + UDP_COMPLETION_TOKEN Token; } UDP_RX_TOKEN; + + /// /// This structure is used internally by UdpIo Library. /// @@ -90,32 +100,41 @@ typedef struct { /// the CallBack will be called. There can be several transmit requests. All transmit requests /// are linked in a list. /// + +typedef union { + EFI_UDP4_SESSION_DATA Udp4; + EFI_UDP6_SESSION_DATA Udp6; +} UDP_SESSION_DATA; + +typedef union { + EFI_UDP4_TRANSMIT_DATA Udp4; + EFI_UDP6_TRANSMIT_DATA Udp6; +} UDP_TRANSMIT_DATA; + typedef struct { - UINT32 Signature; - LIST_ENTRY Link; - UDP_IO_PORT *UdpIo; - - UDP_IO_CALLBACK CallBack; - NET_BUF *Packet; - VOID *Context; - - EFI_UDP4_SESSION_DATA UdpSession; - EFI_IPv4_ADDRESS Gateway; - - EFI_UDP4_COMPLETION_TOKEN UdpToken; - EFI_UDP4_TRANSMIT_DATA UdpTxData; + UINT32 Signature; + LIST_ENTRY Link; + UDP_IO *UdpIo; + UDP_IO_CALLBACK CallBack; + NET_BUF *Packet; + VOID *Context; + EFI_IPv4_ADDRESS Gateway; + UDP_SESSION_DATA Session; + UDP_COMPLETION_TOKEN Token; + UDP_TRANSMIT_DATA Data; } UDP_TX_TOKEN; /// -/// Type defined as UDP_IO_PORT. +/// Type defined as UDP_IO. /// -/// This data structure wraps the Udp4 instance and configuration. -/// UdpIo Library uses this structure for all Udp4 operations. +/// This data structure wraps the UDP instance and configuration. +/// UdpIo Library uses this structure for all Udp4 or Udp6 operations. /// -struct _UDP_IO_PORT { +struct _UDP_IO { UINT32 Signature; LIST_ENTRY Link; INTN RefCnt; + UINT8 UdpVersion; // // Handle used to create/destory UDP child @@ -124,30 +143,38 @@ struct _UDP_IO_PORT { EFI_HANDLE Image; EFI_HANDLE UdpHandle; - EFI_UDP4_PROTOCOL *Udp; ///< The wrapped Udp4 instance. - EFI_UDP4_CONFIG_DATA UdpConfig; EFI_SIMPLE_NETWORK_MODE SnpMode; LIST_ENTRY SentDatagram; ///< A list of UDP_TX_TOKEN. UDP_RX_TOKEN *RecvRequest; + + union { + EFI_UDP4_PROTOCOL *Udp4; + EFI_UDP6_PROTOCOL *Udp6; + } Protocol; + + union { + EFI_UDP4_CONFIG_DATA Udp4; + EFI_UDP6_CONFIG_DATA Udp6; + } Config; }; /** - Prototype called when UdpIo Library configures a Udp4 instance. + Prototype called when UdpIo Library configures a UDP instance. - The prototype is set and called when creating a UDP_IO_PORT in UdpIoCreatePort(). + The prototype is set and called when creating a UDP_IO in UdpIoCreatePort(). - @param[in] UdpIo The UDP_IO_PORT to configure + @param[in] UdpIo The UDP_IO to configure @param[in] Context User-defined data when calling UdpIoCreatePort() @retval EFI_SUCCESS The configuration succeeded - @retval Others The UDP_IO_PORT fails to configure indicating + @retval Others The UDP_IO fails to configure indicating UdpIoCreatePort() should fail **/ typedef EFI_STATUS (*UDP_IO_CONFIG) ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN VOID *Context ); @@ -172,7 +199,7 @@ BOOLEAN Cancel all sent datagrams selected by the parameter ToCancel. If ToCancel is NULL, all the datagrams are cancelled. - @param[in] UdpIo The UDP_IO_PORT to cancel packet. + @param[in] UdpIo The UDP_IO to cancel packet. @param[in] IoStatus The IoStatus to return to the packet owners. @param[in] ToCancel Sets the criteria for canceling a packet. @param[in] Context The opaque parameter to the ToCancel. @@ -181,14 +208,14 @@ BOOLEAN VOID EFIAPI UdpIoCancelDgrams ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN EFI_STATUS IoStatus, - IN UDP_IO_TO_CANCEL ToCancel, OPTIONAL + IN UDP_IO_TO_CANCEL ToCancel OPTIONAL, IN VOID *Context ); /** - Creates a UDP_IO_PORT to access the UDP service. It creates and configures + Creates a UDP_IO to access the UDP service. It creates and configures a UDP child. This function: @@ -199,65 +226,67 @@ UdpIoCancelDgrams ( @param[in] Controller The controller that has the UDP service binding. protocol installed. - @param[in] Image The image handle for the driver. + @param[in] ImageHandle The image handle for the driver. @param[in] Configure The function to configure the created UDP child. + @param[in] UdpVersion The UDP protocol version, UDP4 or UDP6. @param[in] Context The opaque parameter for the Configure funtion. - @return Newly-created UDP_IO_PORT or NULL if failed. + @return Newly-created UDP_IO or NULL if failed. **/ -UDP_IO_PORT * +UDP_IO * EFIAPI -UdpIoCreatePort ( +UdpIoCreateIo ( IN EFI_HANDLE Controller, - IN EFI_HANDLE Image, + IN EFI_HANDLE ImageHandle, IN UDP_IO_CONFIG Configure, + IN UINT8 UdpVersion, IN VOID *Context ); /** - Free the UDP_IO_PORT and all its related resources. + Free the UDP_IO and all its related resources. The function cancels all sent datagrams and receive requests. - @param[in] UdpIo The UDP_IO_PORT to free. + @param[in] UdpIo The UDP_IO to free. - @retval EFI_SUCCESS The UDP_IO_PORT is freed. + @retval EFI_SUCCESS The UDP_IO is freed. **/ EFI_STATUS EFIAPI -UdpIoFreePort ( - IN UDP_IO_PORT *UdpIo +UdpIoFreeIo ( + IN UDP_IO *UdpIo ); /** - Cleans up the UDP_IO_PORT without freeing it. Call this function - if you intend to later re-use the UDP_IO_PORT. + Cleans up the UDP_IO without freeing it. Call this function + if you intend to later re-use the UDP_IO. This function releases all transmitted datagrams and receive requests and configures NULL for the UDP instance. - @param[in] UdpIo The UDP_IO_PORT to clean up. + @param[in] UdpIo The UDP_IO to clean up. **/ VOID EFIAPI -UdpIoCleanPort ( - IN UDP_IO_PORT *UdpIo +UdpIoCleanIo ( + IN UDP_IO *UdpIo ); /** - Sends a packet through the UDP_IO_PORT. + Send a packet through the UDP_IO. - The packet will be wrapped in UDP_TX_TOKEN. The function specific in the CallBack parameter will be called - when the packet is sent. If specified, the optional parameter EndPoint overrides the default - address pair. + The packet will be wrapped in UDP_TX_TOKEN. Function Callback will be called + when the packet is sent. The optional parameter EndPoint overrides the default + address pair if specified. - @param[in] UdpIo The UDP_IO_PORT to send the packet through. + @param[in] UdpIo The UDP_IO to send the packet through. @param[in] Packet The packet to send. - @param[in] EndPoint The local and remote access point. Overrides the + @param[in] EndPoint The local and remote access point. Override the default address pair set during configuration. - @param[in] Gateway The gateway to use. + @param[in] Gateway The gateway to use. @param[in] CallBack The function being called when packet is transmitted or failed. @param[in] Context The opaque parameter passed to CallBack. @@ -270,10 +299,10 @@ UdpIoCleanPort ( EFI_STATUS EFIAPI UdpIoSendDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN NET_BUF *Packet, - IN UDP_POINTS *EndPoint, OPTIONAL - IN IP4_ADDR Gateway, + IN UDP_END_POINT *EndPoint OPTIONAL, + IN EFI_IP_ADDRESS *Gateway OPTIONAL, IN UDP_IO_CALLBACK CallBack, IN VOID *Context ); @@ -281,43 +310,46 @@ UdpIoSendDatagram ( /** Cancel a single sent datagram. - @param[in] UdpIo The UDP_IO_PORT from which to cancel the packet + @param[in] UdpIo The UDP_IO from which to cancel the packet @param[in] Packet The packet to cancel **/ VOID EFIAPI UdpIoCancelSentDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN NET_BUF *Packet ); /** - Issue a receive request to the UDP_IO_PORT. + Issue a receive request to the UDP_IO. This function is called when upper-layer needs packet from UDP for processing. Only one receive request is acceptable at a time. Therefore, one common usage model is to invoke this function inside its Callback function when the former packet is processed. - @param[in] UdpIo The UDP_IO_PORT to receive the packet from. + @param[in] UdpIo The UDP_IO to receive the packet from. @param[in] CallBack The call back function to execute when the packet is received. @param[in] Context The opaque context passed to Callback. - @param[in] HeadLen The length of the upper-layer's protocol header. + @param[in] HeadLen The length of the upper-layer's protocol header. @retval EFI_ALREADY_STARTED There is already a pending receive request. Only one receive request is supported at a time. @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. @retval EFI_SUCCESS The receive request is issued successfully. + @retval EFI_UNSUPPORTED The UDP version in UDP_IO is not supported. **/ EFI_STATUS EFIAPI UdpIoRecvDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN UDP_IO_CALLBACK CallBack, IN VOID *Context, IN UINT32 HeadLen ); + #endif + diff --git a/MdeModulePkg/Library/DxeIpIoLib/DxeIpIoLib.c b/MdeModulePkg/Library/DxeIpIoLib/DxeIpIoLib.c index 5aaef6d743..032545b32b 100644 --- a/MdeModulePkg/Library/DxeIpIoLib/DxeIpIoLib.c +++ b/MdeModulePkg/Library/DxeIpIoLib/DxeIpIoLib.c @@ -136,7 +136,7 @@ IpIoCreateIpChildOpenProtocol ( IN EFI_HANDLE ControllerHandle, IN EFI_HANDLE ImageHandle, IN EFI_HANDLE *ChildHandle, - IN IP_VERSION IpVersion, + IN UINT8 IpVersion, OUT VOID **Interface ) { @@ -213,7 +213,7 @@ IpIoCloseProtocolDestroyIpChild ( IN EFI_HANDLE ControllerHandle, IN EFI_HANDLE ImageHandle, IN EFI_HANDLE ChildHandle, - IN IP_VERSION IpVersion + IN UINT8 IpVersion ) { EFI_STATUS Status; @@ -274,7 +274,7 @@ IpIoIcmpv4Handler ( { IP4_ICMP_ERROR_HEAD *IcmpHdr; EFI_IP4_HEADER *IpHdr; - ICMP_ERROR IcmpErr; + UINT8 IcmpErr; UINT8 *PayLoadHdr; UINT8 Type; UINT8 Code; @@ -307,7 +307,7 @@ IpIoIcmpv4Handler ( case ICMP_CODE_UNREACH_PROTOCOL: case ICMP_CODE_UNREACH_PORT: case ICMP_CODE_UNREACH_SRCFAIL: - IcmpErr = (ICMP_ERROR) (ICMP_ERR_UNREACH_NET + Code); + IcmpErr = (UINT8) (ICMP_ERR_UNREACH_NET + Code); break; @@ -342,7 +342,7 @@ IpIoIcmpv4Handler ( return EFI_ABORTED; } - IcmpErr = (ICMP_ERROR) (Code + ICMP_ERR_TIMXCEED_INTRANS); + IcmpErr = (UINT8) (Code + ICMP_ERR_TIMXCEED_INTRANS); break; @@ -403,7 +403,7 @@ IpIoIcmpv6Handler ( { IP6_ICMP_ERROR_HEAD *IcmpHdr; EFI_IP6_HEADER *IpHdr; - ICMP6_ERROR IcmpErr; + UINT8 IcmpErr; UINT8 *PayLoadHdr; UINT8 Type; UINT8 Code; @@ -470,7 +470,7 @@ IpIoIcmpv6Handler ( return EFI_ABORTED; } - IcmpErr = (ICMP6_ERROR) (ICMP6_ERR_TIMXCEED_HOPLIMIT + Code); + IcmpErr = (UINT8) (ICMP6_ERR_TIMXCEED_HOPLIMIT + Code); break; @@ -479,7 +479,7 @@ IpIoIcmpv6Handler ( return EFI_ABORTED; } - IcmpErr = (ICMP6_ERROR) (ICMP6_ERR_PARAMPROB_HEADER + Code); + IcmpErr = (UINT8) (ICMP6_ERR_PARAMPROB_HEADER + Code); break; @@ -538,7 +538,7 @@ IpIoIcmpv6Handler ( NetbufTrim (Pkt, TrimBytes, TRUE); - IpIo->PktRcvdNotify (EFI_ICMP_ERROR, (ICMP_ERROR) IcmpErr, Session, Pkt, IpIo->RcvdContext); + IpIo->PktRcvdNotify (EFI_ICMP_ERROR, IcmpErr, Session, Pkt, IpIo->RcvdContext); return EFI_SUCCESS; } @@ -1098,7 +1098,7 @@ IpIoListenHandlerDpc ( if (EFI_SUCCESS == Status) { - IpIo->PktRcvdNotify (EFI_SUCCESS, (ICMP_ERROR) 0, &Session, Pkt, IpIo->RcvdContext); + IpIo->PktRcvdNotify (EFI_SUCCESS, 0, &Session, Pkt, IpIo->RcvdContext); } else { // // Status is EFI_ICMP_ERROR @@ -1170,7 +1170,7 @@ EFIAPI IpIoCreate ( IN EFI_HANDLE Image, IN EFI_HANDLE Controller, - IN IP_VERSION IpVersion + IN UINT8 IpVersion ) { EFI_STATUS Status; @@ -1264,7 +1264,7 @@ IpIoOpen ( { EFI_STATUS Status; VOID *Ip; - IP_VERSION IpVersion; + UINT8 IpVersion; if (IpIo->IsConfigured) { return EFI_ACCESS_DENIED; @@ -1379,7 +1379,7 @@ IpIoStop ( EFI_STATUS Status; VOID *Ip; IP_IO_IP_INFO *IpInfo; - IP_VERSION IpVersion; + UINT8 IpVersion; if (!IpIo->IsConfigured) { return EFI_SUCCESS; @@ -1728,7 +1728,7 @@ IpIoConfigIp ( { EFI_STATUS Status; VOID *Ip; - IP_VERSION IpVersion; + UINT8 IpVersion; EFI_IP4_MODE_DATA Ip4ModeData; EFI_IP6_MODE_DATA Ip6ModeData; @@ -1885,7 +1885,7 @@ IpIoRemoveIp ( ) { - IP_VERSION IpVersion; + UINT8 IpVersion; ASSERT (IpInfo->RefCnt > 0); @@ -1957,7 +1957,7 @@ IP_IO_IP_INFO * EFIAPI IpIoFindSender ( IN OUT IP_IO **IpIo, - IN IP_VERSION IpVersion, + IN UINT8 IpVersion, IN EFI_IP_ADDRESS *Src ) { @@ -2021,14 +2021,14 @@ IpIoFindSender ( EFI_STATUS EFIAPI IpIoGetIcmpErrStatus ( - IN ICMP_ERROR IcmpError, - IN IP_VERSION IpVersion, + IN UINT8 IcmpError, + IN UINT8 IpVersion, OUT BOOLEAN *IsHard OPTIONAL, OUT BOOLEAN *Notify OPTIONAL ) { if (IpVersion == IP_VERSION_4 ) { - ASSERT ((IcmpError >= ICMP_ERR_UNREACH_NET) && (IcmpError <= ICMP_ERR_PARAMPROB)); + ASSERT (IcmpError <= ICMP_ERR_PARAMPROB); if (IsHard != NULL) { *IsHard = mIcmpErrMap[IcmpError].IsHard; @@ -2066,7 +2066,7 @@ IpIoGetIcmpErrStatus ( } else if (IpVersion == IP_VERSION_6) { - ASSERT ((IcmpError >= ICMP6_ERR_UNREACH_NET) && (IcmpError <= ICMP6_ERR_PARAMPROB_IPV6OPTION)); + ASSERT (IcmpError <= ICMP6_ERR_PARAMPROB_IPV6OPTION); if (IsHard != NULL) { *IsHard = mIcmp6ErrMap[IcmpError].IsHard; diff --git a/MdeModulePkg/Library/DxeNetLib/DxeNetLib.c b/MdeModulePkg/Library/DxeNetLib/DxeNetLib.c index 4c7414acfb..fb296e5f59 100644 --- a/MdeModulePkg/Library/DxeNetLib/DxeNetLib.c +++ b/MdeModulePkg/Library/DxeNetLib/DxeNetLib.c @@ -222,28 +222,60 @@ Ip6IsValidUnicast ( IN EFI_IPv6_ADDRESS *Ip6 ) { - UINT8 t; - UINT8 i; + UINT8 Byte; + UINT8 Index; if (Ip6->Addr[0] == 0xFF) { return FALSE; } - for (i = 0; i < 15; i++) { - if (Ip6->Addr[i] != 0) { + for (Index = 0; Index < 15; Index++) { + if (Ip6->Addr[Index] != 0) { return TRUE; } } - t = Ip6->Addr[i]; + Byte = Ip6->Addr[Index]; - if (t == 0x0 || t == 0x1) { + if (Byte == 0x0 || Byte == 0x1) { return FALSE; } return TRUE; } +/** + Switches the endianess of an IPv6 address + + This function swaps the bytes in a 128-bit IPv6 address to switch the value + from little endian to big endian or vice versa. The byte swapped value is + returned. + + @param Ip6 Points to an IPv6 address + + @return The byte swapped IPv6 address. + +**/ +EFI_IPv6_ADDRESS * +Ip6Swap128 ( + EFI_IPv6_ADDRESS *Ip6 + ) +{ + UINT64 High; + UINT64 Low; + + CopyMem (&High, Ip6, sizeof (UINT64)); + CopyMem (&Low, &Ip6->Addr[8], sizeof (UINT64)); + + High = SwapBytes64 (High); + Low = SwapBytes64 (Low); + + CopyMem (Ip6, &Low, sizeof (UINT64)); + CopyMem (&Ip6->Addr[8], &High, sizeof (UINT64)); + + return Ip6; +} + /** Initialize a random seed using current time. diff --git a/MdeModulePkg/Library/DxeUdpIoLib/DxeUdpIoLib.c b/MdeModulePkg/Library/DxeUdpIoLib/DxeUdpIoLib.c index 91e211a9bd..bb85aebc28 100644 --- a/MdeModulePkg/Library/DxeUdpIoLib/DxeUdpIoLib.c +++ b/MdeModulePkg/Library/DxeUdpIoLib/DxeUdpIoLib.c @@ -1,7 +1,7 @@ /** @file Help functions to access UDP service, it is used by both the DHCP and MTFTP. -Copyright (c) 2005 - 2007, Intel Corporation.
+Copyright (c) 2005 - 2009, 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
@@ -14,6 +14,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. #include #include +#include #include #include @@ -27,31 +28,46 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. /** Free a UDP_TX_TOKEN. The TX event is closed. - @param[in] Token The UDP_TX_TOKEN to release. + @param[in] TxToken The UDP_TX_TOKEN to release. **/ VOID UdpIoFreeTxToken ( - IN UDP_TX_TOKEN *Token + IN UDP_TX_TOKEN *TxToken ) { - gBS->CloseEvent (Token->UdpToken.Event); - gBS->FreePool (Token); + + if (TxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + gBS->CloseEvent (TxToken->Token.Udp4.Event); + } else if (TxToken->UdpIo->UdpVersion == UDP_IO_UDP6_VERSION) { + gBS->CloseEvent (TxToken->Token.Udp6.Event); + } else { + ASSERT (FALSE); + } + + FreePool (TxToken); } /** Free a UDP_RX_TOKEN. The RX event is closed. - @param[in] Token The UDP_RX_TOKEN to release. + @param[in] RxToken The UDP_RX_TOKEN to release. **/ VOID UdpIoFreeRxToken ( - IN UDP_RX_TOKEN *Token + IN UDP_RX_TOKEN *RxToken ) { - gBS->CloseEvent (Token->UdpToken.Event); - gBS->FreePool (Token); + if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + gBS->CloseEvent (RxToken->Token.Udp4.Event); + } else if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP6_VERSION) { + gBS->CloseEvent (RxToken->Token.Udp6.Event); + } else { + ASSERT (FALSE); + } + + FreePool (RxToken); } /** @@ -69,15 +85,22 @@ UdpIoOnDgramSentDpc ( IN VOID *Context ) { - UDP_TX_TOKEN *Token; + UDP_TX_TOKEN *TxToken; - Token = (UDP_TX_TOKEN *) Context; - ASSERT (Token->Signature == UDP_IO_TX_SIGNATURE); + TxToken = (UDP_TX_TOKEN *) Context; + ASSERT (TxToken->Signature == UDP_IO_TX_SIGNATURE); + ASSERT ((TxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (TxToken->UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); + + RemoveEntryList (&TxToken->Link); - RemoveEntryList (&Token->Link); - Token->CallBack (Token->Packet, NULL, Token->UdpToken.Status, Token->Context); + if (TxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + TxToken->CallBack (TxToken->Packet, NULL, TxToken->Token.Udp4.Status, TxToken->Context); + } else { + TxToken->CallBack (TxToken->Packet, NULL, TxToken->Token.Udp6.Status, TxToken->Context); + } - UdpIoFreeTxToken (Token); + UdpIoFreeTxToken (TxToken); } /** @@ -110,12 +133,20 @@ VOID UdpIoRecycleDgram ( IN VOID *Context ) -{ - UDP_RX_TOKEN *Token; +{ + UDP_RX_TOKEN *RxToken; - Token = (UDP_RX_TOKEN *) Context; - gBS->SignalEvent (Token->UdpToken.Packet.RxData->RecycleSignal); - UdpIoFreeRxToken (Token); + RxToken = (UDP_RX_TOKEN *) Context; + + if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + gBS->SignalEvent (RxToken->Token.Udp4.Packet.RxData->RecycleSignal); + } else if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP6_VERSION) { + gBS->SignalEvent (RxToken->Token.Udp6.Packet.RxData->RecycleSignal); + } else { + ASSERT (FALSE); + } + + UdpIoFreeRxToken (RxToken); } /** @@ -133,69 +164,132 @@ UdpIoOnDgramRcvdDpc ( IN VOID *Context ) { - EFI_UDP4_COMPLETION_TOKEN *UdpToken; - EFI_UDP4_RECEIVE_DATA *UdpRxData; - EFI_UDP4_SESSION_DATA *UdpSession; - UDP_RX_TOKEN *Token; - UDP_POINTS Points; + EFI_STATUS Status; + VOID *Token; + VOID *RxData; + VOID *Session; + UDP_RX_TOKEN *RxToken; + UDP_END_POINT EndPoint; NET_BUF *Netbuf; - Token = (UDP_RX_TOKEN *) Context; + RxToken = (UDP_RX_TOKEN *) Context; - ASSERT ((Token->Signature == UDP_IO_RX_SIGNATURE) && - (Token == Token->UdpIo->RecvRequest)); + ZeroMem (&EndPoint, sizeof(UDP_END_POINT)); + + ASSERT ((RxToken->Signature == UDP_IO_RX_SIGNATURE) && + (RxToken == RxToken->UdpIo->RecvRequest)); + + ASSERT ((RxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (RxToken->UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); // // Clear the receive request first in case that the caller // wants to restart the receive in the callback. // - Token->UdpIo->RecvRequest = NULL; + RxToken->UdpIo->RecvRequest = NULL; - UdpToken = &Token->UdpToken; - UdpRxData = UdpToken->Packet.RxData; + if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + Token = &RxToken->Token.Udp4; + RxData = ((EFI_UDP4_COMPLETION_TOKEN *) Token)->Packet.RxData; + Status = ((EFI_UDP4_COMPLETION_TOKEN *) Token)->Status; + } else { + Token = &RxToken->Token.Udp6; + RxData = ((EFI_UDP6_COMPLETION_TOKEN *) Token)->Packet.RxData; + Status = ((EFI_UDP6_COMPLETION_TOKEN *) Token)->Status; + } - if (EFI_ERROR (UdpToken->Status) || (UdpRxData == NULL)) { - if (UdpToken->Status != EFI_ABORTED) { + if (EFI_ERROR (Status) || RxData == NULL) { + if (Status != EFI_ABORTED) { // // Invoke the CallBack only if the reception is not actively aborted. // - Token->CallBack (NULL, NULL, UdpToken->Status, Token->Context); + RxToken->CallBack (NULL, NULL, Status, RxToken->Context); } - UdpIoFreeRxToken (Token); + UdpIoFreeRxToken (RxToken); return; } // // Build a NET_BUF from the UDP receive data, then deliver it up. // - Netbuf = NetbufFromExt ( - (NET_FRAGMENT *) UdpRxData->FragmentTable, - UdpRxData->FragmentCount, - 0, - (UINT32) Token->HeadLen, - UdpIoRecycleDgram, - Token - ); + if (RxToken->UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + + Netbuf = NetbufFromExt ( + (NET_FRAGMENT *)((EFI_UDP4_RECEIVE_DATA *) RxData)->FragmentTable, + ((EFI_UDP4_RECEIVE_DATA *) RxData)->FragmentCount, + 0, + (UINT32) RxToken->HeadLen, + UdpIoRecycleDgram, + RxToken + ); - if (Netbuf == NULL) { - gBS->SignalEvent (UdpRxData->RecycleSignal); - Token->CallBack (NULL, NULL, EFI_OUT_OF_RESOURCES, Token->Context); + if (Netbuf == NULL) { + gBS->SignalEvent (((EFI_UDP4_RECEIVE_DATA *) RxData)->RecycleSignal); + RxToken->CallBack (NULL, NULL, EFI_OUT_OF_RESOURCES, RxToken->Context); - UdpIoFreeRxToken (Token); - return; + UdpIoFreeRxToken (RxToken); + return; + } + + Session = &((EFI_UDP4_RECEIVE_DATA *) RxData)->UdpSession; + EndPoint.LocalPort = ((EFI_UDP4_SESSION_DATA *) Session)->DestinationPort; + EndPoint.RemotePort = ((EFI_UDP4_SESSION_DATA *) Session)->SourcePort; + + CopyMem ( + &EndPoint.LocalAddr, + &((EFI_UDP4_SESSION_DATA *) Session)->DestinationAddress, + sizeof (EFI_IPv4_ADDRESS) + ); + + CopyMem ( + &EndPoint.RemoteAddr, + &((EFI_UDP4_SESSION_DATA *) Session)->SourceAddress, + sizeof (EFI_IPv4_ADDRESS) + ); + + EndPoint.LocalAddr.Addr[0] = NTOHL (EndPoint.LocalAddr.Addr[0]); + EndPoint.RemoteAddr.Addr[0] = NTOHL (EndPoint.RemoteAddr.Addr[0]); + } else { + + Netbuf = NetbufFromExt ( + (NET_FRAGMENT *)((EFI_UDP6_RECEIVE_DATA *) RxData)->FragmentTable, + ((EFI_UDP6_RECEIVE_DATA *) RxData)->FragmentCount, + 0, + (UINT32) RxToken->HeadLen, + UdpIoRecycleDgram, + RxToken + ); + + if (Netbuf == NULL) { + gBS->SignalEvent (((EFI_UDP6_RECEIVE_DATA *) RxData)->RecycleSignal); + RxToken->CallBack (NULL, NULL, EFI_OUT_OF_RESOURCES, RxToken->Context); + + UdpIoFreeRxToken (RxToken); + return; + } + + Session = &((EFI_UDP6_RECEIVE_DATA *) RxData)->UdpSession; + EndPoint.LocalPort = ((EFI_UDP6_SESSION_DATA *) Session)->DestinationPort; + EndPoint.RemotePort = ((EFI_UDP6_SESSION_DATA *) Session)->SourcePort; + + CopyMem ( + &EndPoint.LocalAddr, + &((EFI_UDP6_SESSION_DATA *) Session)->DestinationAddress, + sizeof (EFI_IPv6_ADDRESS) + ); + + CopyMem ( + &EndPoint.RemoteAddr, + &((EFI_UDP6_SESSION_DATA *) Session)->SourceAddress, + sizeof (EFI_IPv6_ADDRESS) + ); + + Ip6Swap128 (&EndPoint.LocalAddr.v6); + Ip6Swap128 (&EndPoint.RemoteAddr.v6); } - UdpSession = &UdpRxData->UdpSession; - Points.LocalPort = UdpSession->DestinationPort; - Points.RemotePort = UdpSession->SourcePort; - - CopyMem (&Points.LocalAddr, &UdpSession->DestinationAddress, sizeof (IP4_ADDR)); - CopyMem (&Points.RemoteAddr, &UdpSession->SourceAddress, sizeof (IP4_ADDR)); - Points.LocalAddr = NTOHL (Points.LocalAddr); - Points.RemoteAddr = NTOHL (Points.RemoteAddr); - - Token->CallBack (Netbuf, &Points, EFI_SUCCESS, Token->Context); + RxToken->CallBack (Netbuf, &EndPoint, EFI_SUCCESS, RxToken->Context); } /** @@ -231,7 +325,7 @@ UdpIoOnDgramRcvd ( **/ UDP_RX_TOKEN * UdpIoCreateRxToken ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN UDP_IO_CALLBACK CallBack, IN VOID *Context, IN UINT32 HeadLen @@ -240,6 +334,9 @@ UdpIoCreateRxToken ( UDP_RX_TOKEN *Token; EFI_STATUS Status; + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); + Token = AllocatePool (sizeof (UDP_RX_TOKEN)); if (Token == NULL) { @@ -252,19 +349,35 @@ UdpIoCreateRxToken ( Token->Context = Context; Token->HeadLen = HeadLen; - Token->UdpToken.Status = EFI_NOT_READY; - Token->UdpToken.Packet.RxData = NULL; + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + + Token->Token.Udp4.Status = EFI_NOT_READY; + Token->Token.Udp4.Packet.RxData = NULL; + + Status = gBS->CreateEvent ( + EVT_NOTIFY_SIGNAL, + TPL_NOTIFY, + UdpIoOnDgramRcvd, + Token, + &Token->Token.Udp4.Event + ); + } else { + + Token->Token.Udp6.Status = EFI_NOT_READY; + Token->Token.Udp6.Packet.RxData = NULL; + + Status = gBS->CreateEvent ( + EVT_NOTIFY_SIGNAL, + TPL_NOTIFY, + UdpIoOnDgramRcvd, + Token, + &Token->Token.Udp6.Event + ); + } - Status = gBS->CreateEvent ( - EVT_NOTIFY_SIGNAL, - TPL_NOTIFY, - UdpIoOnDgramRcvd, - Token, - &Token->UdpToken.Event - ); if (EFI_ERROR (Status)) { - gBS->FreePool (Token); + FreePool (Token); return NULL; } @@ -272,9 +385,9 @@ UdpIoCreateRxToken ( } /** - Wrap a transmit request into a UDP_TX_TOKEN. + Wrap a transmit request into a new created UDP_TX_TOKEN. - @param[in] UdpIo The UdpIo port to send packet to. + @param[in] UdpIo The UdpIo to send packet to. @param[in] Packet The user's packet. @param[in] EndPoint The local and remote access point. @param[in] Gateway The overrided next hop. @@ -286,188 +399,329 @@ UdpIoCreateRxToken ( **/ UDP_TX_TOKEN * -UdpIoWrapTx ( - IN UDP_IO_PORT *UdpIo, +UdpIoCreateTxToken ( + IN UDP_IO *UdpIo, IN NET_BUF *Packet, - IN UDP_POINTS *EndPoint OPTIONAL, - IN IP4_ADDR Gateway, + IN UDP_END_POINT *EndPoint OPTIONAL, + IN EFI_IP_ADDRESS *Gateway OPTIONAL, IN UDP_IO_CALLBACK CallBack, IN VOID *Context ) { - UDP_TX_TOKEN *Token; - EFI_UDP4_COMPLETION_TOKEN *UdpToken; - EFI_UDP4_TRANSMIT_DATA *UdpTxData; + UDP_TX_TOKEN *TxToken; + VOID *Token; + VOID *Data; EFI_STATUS Status; UINT32 Count; - IP4_ADDR Ip; + UINTN Size; + IP4_ADDR Ip; - Token = AllocatePool (sizeof (UDP_TX_TOKEN) + - sizeof (EFI_UDP4_FRAGMENT_DATA) * (Packet->BlockOpNum - 1)); + ASSERT (Packet != NULL); + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); - if (Token == NULL) { + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + Size = sizeof (UDP_TX_TOKEN) + sizeof (EFI_UDP4_FRAGMENT_DATA) * (Packet->BlockOpNum - 1); + } else { + Size = sizeof (UDP_TX_TOKEN) + sizeof (EFI_UDP6_FRAGMENT_DATA) * (Packet->BlockOpNum - 1); + } + + TxToken = AllocatePool (Size); + + if (TxToken == NULL) { return NULL; } - Token->Signature = UDP_IO_TX_SIGNATURE; - InitializeListHead (&Token->Link); + TxToken->Signature = UDP_IO_TX_SIGNATURE; + InitializeListHead (&TxToken->Link); - Token->UdpIo = UdpIo; - Token->CallBack = CallBack; - Token->Packet = Packet; - Token->Context = Context; + TxToken->UdpIo = UdpIo; + TxToken->CallBack = CallBack; + TxToken->Packet = Packet; + TxToken->Context = Context; - UdpToken = &(Token->UdpToken); - UdpToken->Status = EFI_NOT_READY; + Token = &(TxToken->Token); + Count = Packet->BlockOpNum; - Status = gBS->CreateEvent ( - EVT_NOTIFY_SIGNAL, - TPL_NOTIFY, - UdpIoOnDgramSent, - Token, - &UdpToken->Event - ); + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { - if (EFI_ERROR (Status)) { - gBS->FreePool (Token); - return NULL; + ((EFI_UDP4_COMPLETION_TOKEN *) Token)->Status = EFI_NOT_READY; + + Status = gBS->CreateEvent ( + EVT_NOTIFY_SIGNAL, + TPL_NOTIFY, + UdpIoOnDgramSent, + TxToken, + &((EFI_UDP4_COMPLETION_TOKEN *) Token)->Event + ); + + if (EFI_ERROR (Status)) { + FreePool (TxToken); + return NULL; + } + + Data = &(TxToken->Data.Udp4); + ((EFI_UDP4_COMPLETION_TOKEN *) Token)->Packet.TxData = Data; + + ((EFI_UDP4_TRANSMIT_DATA *) Data)->UdpSessionData = NULL; + ((EFI_UDP4_TRANSMIT_DATA *) Data)->GatewayAddress = NULL; + ((EFI_UDP4_TRANSMIT_DATA *) Data)->DataLength = Packet->TotalSize; + + NetbufBuildExt ( + Packet, + (NET_FRAGMENT *)((EFI_UDP4_TRANSMIT_DATA *) Data)->FragmentTable, + &Count + ); + + ((EFI_UDP4_TRANSMIT_DATA *) Data)->FragmentCount = Count; + + if (EndPoint != NULL) { + Ip = HTONL (EndPoint->LocalAddr.Addr[0]); + CopyMem ( + &TxToken->Session.Udp4.SourceAddress, + &Ip, + sizeof (EFI_IPv4_ADDRESS) + ); + + Ip = HTONL (EndPoint->RemoteAddr.Addr[0]); + CopyMem ( + &TxToken->Session.Udp4.DestinationAddress, + &Ip, + sizeof (EFI_IPv4_ADDRESS) + ); + + TxToken->Session.Udp4.SourcePort = EndPoint->LocalPort; + TxToken->Session.Udp4.DestinationPort = EndPoint->RemotePort; + ((EFI_UDP4_TRANSMIT_DATA *) Data)->UdpSessionData = &(TxToken->Session.Udp4); + } + + if (Gateway != NULL && (Gateway->Addr[0] != 0)) { + Ip = HTONL (Gateway->Addr[0]); + CopyMem (&TxToken->Gateway, &Ip, sizeof (EFI_IPv4_ADDRESS)); + ((EFI_UDP4_TRANSMIT_DATA *) Data)->GatewayAddress = &TxToken->Gateway; + } + + } else { + + ((EFI_UDP6_COMPLETION_TOKEN *) Token)->Status = EFI_NOT_READY; + + Status = gBS->CreateEvent ( + EVT_NOTIFY_SIGNAL, + TPL_NOTIFY, + UdpIoOnDgramSent, + TxToken, + &((EFI_UDP6_COMPLETION_TOKEN *) Token)->Event + ); + + if (EFI_ERROR (Status)) { + FreePool (TxToken); + return NULL; + } + + Data = &(TxToken->Data.Udp6); + ((EFI_UDP6_COMPLETION_TOKEN *) Token)->Packet.TxData = Data; + ((EFI_UDP6_TRANSMIT_DATA *) Data)->UdpSessionData = NULL; + ((EFI_UDP6_TRANSMIT_DATA *) Data)->DataLength = Packet->TotalSize; + + NetbufBuildExt ( + Packet, + (NET_FRAGMENT *)((EFI_UDP6_TRANSMIT_DATA *) Data)->FragmentTable, + &Count + ); + + ((EFI_UDP6_TRANSMIT_DATA *) Data)->FragmentCount = Count; + + if (EndPoint != NULL) { + CopyMem ( + &TxToken->Session.Udp6.SourceAddress, + &EndPoint->LocalAddr.v6, + sizeof(EFI_IPv6_ADDRESS) + ); + + CopyMem ( + &TxToken->Session.Udp6.DestinationAddress, + &EndPoint->RemoteAddr.v6, + sizeof(EFI_IPv6_ADDRESS) + ); + + TxToken->Session.Udp6.SourcePort = EndPoint->LocalPort; + TxToken->Session.Udp6.DestinationPort = EndPoint->RemotePort; + ((EFI_UDP6_TRANSMIT_DATA *) Data)->UdpSessionData = &(TxToken->Session.Udp6); + } } - UdpTxData = &Token->UdpTxData; - UdpToken->Packet.TxData = UdpTxData; - - UdpTxData->UdpSessionData = NULL; - UdpTxData->GatewayAddress = NULL; - - if (EndPoint != NULL) { - Ip = HTONL (EndPoint->LocalAddr); - CopyMem (&Token->UdpSession.SourceAddress, &Ip, sizeof (EFI_IPv4_ADDRESS)); - - Ip = HTONL (EndPoint->RemoteAddr); - CopyMem (&Token->UdpSession.DestinationAddress, &Ip, sizeof (EFI_IPv4_ADDRESS)); - - Token->UdpSession.SourcePort = EndPoint->LocalPort; - Token->UdpSession.DestinationPort = EndPoint->RemotePort; - UdpTxData->UdpSessionData = &Token->UdpSession; - } - - if (Gateway != 0) { - Ip = HTONL (Gateway); - CopyMem (&Token->Gateway, &Ip, sizeof (EFI_IPv4_ADDRESS)); - - UdpTxData->GatewayAddress = &Token->Gateway; - } - - UdpTxData->DataLength = Packet->TotalSize; - Count = Packet->BlockOpNum; - NetbufBuildExt (Packet, (NET_FRAGMENT *) UdpTxData->FragmentTable, &Count); - UdpTxData->FragmentCount = Count; - - return Token; + return TxToken; } /** - Create a UDP_IO_PORT to access the UDP service. It will create and configure + Creates a UDP_IO to access the UDP service. It creates and configures a UDP child. - The function will locate the UDP service binding prototype on the Controller - parameter and use it to create a UDP child (aka Udp instance). Then the UDP - child will be configured by calling Configure function prototype. Any failures - in creating or configure the UDP child will lead to the failure of UDP_IO_PORT - creation. + This function: + # locates the UDP service binding prototype on the Controller parameter + # uses the UDP service binding prototype to create a UDP child (also known as a UDP instance) + # configures the UDP child by calling Configure function prototype. + Any failures in creating or configuring the UDP child return NULL for failure. @param[in] Controller The controller that has the UDP service binding. protocol installed. - @param[in] Image The image handle for the driver. + @param[in] ImageHandle The image handle for the driver. @param[in] Configure The function to configure the created UDP child. + @param[in] UdpVersion The UDP protocol version, UDP4 or UDP6. @param[in] Context The opaque parameter for the Configure funtion. - @return Newly-created UDP_IO_PORT or NULL if failed. + @return Newly-created UDP_IO or NULL if failed. **/ -UDP_IO_PORT * +UDP_IO * EFIAPI -UdpIoCreatePort ( +UdpIoCreateIo ( IN EFI_HANDLE Controller, - IN EFI_HANDLE Image, + IN EFI_HANDLE ImageHandle, IN UDP_IO_CONFIG Configure, + IN UINT8 UdpVersion, IN VOID *Context ) { - UDP_IO_PORT *UdpIo; + UDP_IO *UdpIo; EFI_STATUS Status; ASSERT (Configure != NULL); + ASSERT ((UdpVersion == UDP_IO_UDP4_VERSION) || (UdpVersion == UDP_IO_UDP6_VERSION)); - UdpIo = AllocatePool (sizeof (UDP_IO_PORT)); + UdpIo = AllocatePool (sizeof (UDP_IO)); if (UdpIo == NULL) { return NULL; } + UdpIo->UdpVersion = UdpVersion; UdpIo->Signature = UDP_IO_SIGNATURE; InitializeListHead (&UdpIo->Link); UdpIo->RefCnt = 1; UdpIo->Controller = Controller; - UdpIo->Image = Image; + UdpIo->Image = ImageHandle; InitializeListHead (&UdpIo->SentDatagram); UdpIo->RecvRequest = NULL; UdpIo->UdpHandle = NULL; - // - // Create a UDP child then open and configure it - // - Status = NetLibCreateServiceChild ( - Controller, - Image, - &gEfiUdp4ServiceBindingProtocolGuid, - &UdpIo->UdpHandle - ); + if (UdpVersion == UDP_IO_UDP4_VERSION) { + // + // Create a UDP child then open and configure it + // + Status = NetLibCreateServiceChild ( + Controller, + ImageHandle, + &gEfiUdp4ServiceBindingProtocolGuid, + &UdpIo->UdpHandle + ); + + if (EFI_ERROR (Status)) { + goto FREE_MEM; + } - if (EFI_ERROR (Status)) { - goto FREE_MEM; - } + Status = gBS->OpenProtocol ( + UdpIo->UdpHandle, + &gEfiUdp4ProtocolGuid, + (VOID **) &UdpIo->Protocol.Udp4, + ImageHandle, + Controller, + EFI_OPEN_PROTOCOL_BY_DRIVER + ); - Status = gBS->OpenProtocol ( - UdpIo->UdpHandle, - &gEfiUdp4ProtocolGuid, - (VOID **) &UdpIo->Udp, - Image, - Controller, - EFI_OPEN_PROTOCOL_BY_DRIVER - ); + if (EFI_ERROR (Status)) { + goto FREE_CHILD; + } - if (EFI_ERROR (Status)) { - goto FREE_CHILD; - } + if (EFI_ERROR (Configure (UdpIo, Context))) { + goto CLOSE_PROTOCOL; + } + + Status = UdpIo->Protocol.Udp4->GetModeData ( + UdpIo->Protocol.Udp4, + NULL, + NULL, + NULL, + &UdpIo->SnpMode + ); + + if (EFI_ERROR (Status)) { + goto CLOSE_PROTOCOL; + } - if (EFI_ERROR (Configure (UdpIo, Context))) { - goto CLOSE_PROTOCOL; - } - - Status = UdpIo->Udp->GetModeData (UdpIo->Udp, NULL, NULL, NULL, &UdpIo->SnpMode); - - if (EFI_ERROR (Status)) { - goto CLOSE_PROTOCOL; + } else { + + Status = NetLibCreateServiceChild ( + Controller, + ImageHandle, + &gEfiUdp6ServiceBindingProtocolGuid, + &UdpIo->UdpHandle + ); + + if (EFI_ERROR (Status)) { + goto FREE_MEM; + } + + Status = gBS->OpenProtocol ( + UdpIo->UdpHandle, + &gEfiUdp6ProtocolGuid, + (VOID **) &UdpIo->Protocol.Udp6, + ImageHandle, + Controller, + EFI_OPEN_PROTOCOL_BY_DRIVER + ); + + if (EFI_ERROR (Status)) { + goto FREE_CHILD; + } + + if (EFI_ERROR (Configure (UdpIo, Context))) { + goto CLOSE_PROTOCOL; + } + + Status = UdpIo->Protocol.Udp6->GetModeData ( + UdpIo->Protocol.Udp6, + NULL, + NULL, + NULL, + &UdpIo->SnpMode + ); + + if (EFI_ERROR (Status)) { + goto CLOSE_PROTOCOL; + } } return UdpIo; CLOSE_PROTOCOL: - gBS->CloseProtocol (UdpIo->UdpHandle, &gEfiUdp4ProtocolGuid, Image, Controller); + if (UdpVersion == UDP_IO_UDP4_VERSION) { + gBS->CloseProtocol (UdpIo->UdpHandle, &gEfiUdp4ProtocolGuid, ImageHandle, Controller); + } else { + gBS->CloseProtocol (UdpIo->UdpHandle, &gEfiUdp6ProtocolGuid, ImageHandle, Controller); + } FREE_CHILD: - NetLibDestroyServiceChild ( - Controller, - Image, - &gEfiUdp4ServiceBindingProtocolGuid, - UdpIo->UdpHandle - ); + if (UdpVersion == UDP_IO_UDP4_VERSION) { + NetLibDestroyServiceChild ( + Controller, + ImageHandle, + &gEfiUdp4ServiceBindingProtocolGuid, + UdpIo->UdpHandle + ); + } else { + NetLibDestroyServiceChild ( + Controller, + ImageHandle, + &gEfiUdp6ServiceBindingProtocolGuid, + UdpIo->UdpHandle + ); + } FREE_MEM: - gBS->FreePool (UdpIo); + FreePool (UdpIo); return NULL; } @@ -475,7 +729,7 @@ FREE_MEM: Cancel all the sent datagram that pass the selection criteria of ToCancel. If ToCancel is NULL, all the datagrams are cancelled. - @param[in] UdpIo The UDP_IO_PORT to cancel packet. + @param[in] UdpIo The UDP_IO to cancel packet. @param[in] IoStatus The IoStatus to return to the packet owners. @param[in] ToCancel The select funtion to test whether to cancel this packet or not. @@ -485,7 +739,7 @@ FREE_MEM: VOID EFIAPI UdpIoCancelDgrams ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN EFI_STATUS IoStatus, IN UDP_IO_TO_CANCEL ToCancel, OPTIONAL IN VOID *Context @@ -493,34 +747,45 @@ UdpIoCancelDgrams ( { LIST_ENTRY *Entry; LIST_ENTRY *Next; - UDP_TX_TOKEN *Token; + UDP_TX_TOKEN *TxToken; + + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); NET_LIST_FOR_EACH_SAFE (Entry, Next, &UdpIo->SentDatagram) { - Token = NET_LIST_USER_STRUCT (Entry, UDP_TX_TOKEN, Link); + TxToken = NET_LIST_USER_STRUCT (Entry, UDP_TX_TOKEN, Link); - if ((ToCancel == NULL) || (ToCancel (Token, Context))) { - UdpIo->Udp->Cancel (UdpIo->Udp, &Token->UdpToken); + if ((ToCancel == NULL) || (ToCancel (TxToken, Context))) { + + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + UdpIo->Protocol.Udp4->Cancel (UdpIo->Protocol.Udp4, &TxToken->Token.Udp4); + } else { + UdpIo->Protocol.Udp6->Cancel (UdpIo->Protocol.Udp6, &TxToken->Token.Udp6); + } } - } + } } /** - Free the UDP_IO_PORT and all its related resources. + Free the UDP_IO and all its related resources. The function will cancel all sent datagram and receive request. - @param[in] UdpIo The UDP_IO_PORT to free. + @param[in] UdpIo The UDP_IO to free. - @retval EFI_SUCCESS The UDP_IO_PORT is freed. + @retval EFI_SUCCESS The UDP_IO is freed. **/ EFI_STATUS EFIAPI -UdpIoFreePort ( - IN UDP_IO_PORT *UdpIo +UdpIoFreeIo ( + IN UDP_IO *UdpIo ) { - UDP_RX_TOKEN *RxToken; + UDP_RX_TOKEN *RxToken; + + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); // // Cancel all the sent datagram and receive requests. The @@ -532,78 +797,116 @@ UdpIoFreePort ( // UdpIoCancelDgrams (UdpIo, EFI_ABORTED, NULL, NULL); - if ((RxToken = UdpIo->RecvRequest) != NULL) { - UdpIo->Udp->Cancel (UdpIo->Udp, &RxToken->UdpToken); - } + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { - // - // Close then destory the UDP child - // - gBS->CloseProtocol ( - UdpIo->UdpHandle, - &gEfiUdp4ProtocolGuid, - UdpIo->Image, - UdpIo->Controller - ); + if ((RxToken = UdpIo->RecvRequest) != NULL) { + UdpIo->Protocol.Udp4->Cancel (UdpIo->Protocol.Udp4, &RxToken->Token.Udp4); + } - NetLibDestroyServiceChild ( - UdpIo->Controller, - UdpIo->Image, - &gEfiUdp4ServiceBindingProtocolGuid, - UdpIo->UdpHandle - ); + // + // Close then destory the Udp4 child + // + gBS->CloseProtocol ( + UdpIo->UdpHandle, + &gEfiUdp4ProtocolGuid, + UdpIo->Image, + UdpIo->Controller + ); + + NetLibDestroyServiceChild ( + UdpIo->Controller, + UdpIo->Image, + &gEfiUdp4ServiceBindingProtocolGuid, + UdpIo->UdpHandle + ); + + } else { + + if ((RxToken = UdpIo->RecvRequest) != NULL) { + UdpIo->Protocol.Udp6->Cancel (UdpIo->Protocol.Udp6, &RxToken->Token.Udp6); + } + + // + // Close then destory the Udp6 child + // + gBS->CloseProtocol ( + UdpIo->UdpHandle, + &gEfiUdp6ProtocolGuid, + UdpIo->Image, + UdpIo->Controller + ); + + NetLibDestroyServiceChild ( + UdpIo->Controller, + UdpIo->Image, + &gEfiUdp6ServiceBindingProtocolGuid, + UdpIo->UdpHandle + ); + } if (!IsListEmpty(&UdpIo->Link)) { RemoveEntryList (&UdpIo->Link); } - gBS->FreePool (UdpIo); + FreePool (UdpIo); return EFI_SUCCESS; } /** - Clean up the UDP_IO_PORT without freeing it. The function is called when - user wants to re-use the UDP_IO_PORT later. + Clean up the UDP_IO without freeing it. The function is called when + user wants to re-use the UDP_IO later. It will release all the transmitted datagrams and receive request. It will also configure NULL for the UDP instance. - @param[in] UdpIo The UDP_IO_PORT to clean up. + @param[in] UdpIo The UDP_IO to clean up. **/ VOID EFIAPI -UdpIoCleanPort ( - IN UDP_IO_PORT *UdpIo +UdpIoCleanIo ( + IN UDP_IO *UdpIo ) { UDP_RX_TOKEN *RxToken; + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); + // // Cancel all the sent datagram and receive requests. // UdpIoCancelDgrams (UdpIo, EFI_ABORTED, NULL, NULL); - if ((RxToken = UdpIo->RecvRequest) != NULL) { - UdpIo->Udp->Cancel (UdpIo->Udp, &RxToken->UdpToken); - } + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + if ((RxToken = UdpIo->RecvRequest) != NULL) { + UdpIo->Protocol.Udp4->Cancel (UdpIo->Protocol.Udp4, &RxToken->Token.Udp4); + } - UdpIo->Udp->Configure (UdpIo->Udp, NULL); + UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, NULL); + + } else { + if ((RxToken = UdpIo->RecvRequest) != NULL) { + UdpIo->Protocol.Udp6->Cancel (UdpIo->Protocol.Udp6, &RxToken->Token.Udp6); + } + + UdpIo->Protocol.Udp6->Configure (UdpIo->Protocol.Udp6, NULL); + } } /** - Send a packet through the UDP_IO_PORT. + Send a packet through the UDP_IO. The packet will be wrapped in UDP_TX_TOKEN. Function Callback will be called when the packet is sent. The optional parameter EndPoint overrides the default address pair if specified. - @param[in] UdpIo The UDP_IO_PORT to send the packet through. + @param[in] UdpIo The UDP_IO to send the packet through. @param[in] Packet The packet to send. @param[in] EndPoint The local and remote access point. Override the default address pair set during configuration. - @param[in] Gateway The gateway to use. + @param[in] Gateway The gateway to use. @param[in] CallBack The function being called when packet is transmitted or failed. @param[in] Context The opaque parameter passed to CallBack. @@ -616,20 +919,23 @@ UdpIoCleanPort ( EFI_STATUS EFIAPI UdpIoSendDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN NET_BUF *Packet, - IN UDP_POINTS *EndPoint, OPTIONAL - IN IP4_ADDR Gateway, + IN UDP_END_POINT *EndPoint OPTIONAL, + IN EFI_IP_ADDRESS *Gateway OPTIONAL, IN UDP_IO_CALLBACK CallBack, IN VOID *Context ) { - UDP_TX_TOKEN *Token; + UDP_TX_TOKEN *TxToken; EFI_STATUS Status; - Token = UdpIoWrapTx (UdpIo, Packet, EndPoint, Gateway, CallBack, Context); + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); - if (Token == NULL) { + TxToken = UdpIoCreateTxToken (UdpIo, Packet, EndPoint, Gateway, CallBack, Context); + + if (TxToken == NULL) { return EFI_OUT_OF_RESOURCES; } @@ -637,11 +943,17 @@ UdpIoSendDatagram ( // Insert the tx token into SendDatagram list before transmitting it. Remove // it from the list if the returned status is not EFI_SUCCESS. // - InsertHeadList (&UdpIo->SentDatagram, &Token->Link); - Status = UdpIo->Udp->Transmit (UdpIo->Udp, &Token->UdpToken); + InsertHeadList (&UdpIo->SentDatagram, &TxToken->Link); + + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + Status = UdpIo->Protocol.Udp4->Transmit (UdpIo->Protocol.Udp4, &TxToken->Token.Udp4); + } else { + Status = UdpIo->Protocol.Udp6->Transmit (UdpIo->Protocol.Udp6, &TxToken->Token.Udp6); + } + if (EFI_ERROR (Status)) { - RemoveEntryList (&Token->Link); - UdpIoFreeTxToken (Token); + RemoveEntryList (&TxToken->Link); + UdpIoFreeTxToken (TxToken); return Status; } @@ -678,14 +990,14 @@ UdpIoCancelSingleDgram ( /** Cancel a single sent datagram. - @param[in] UdpIo The UDP_IO_PORT to cancel the packet from + @param[in] UdpIo The UDP_IO to cancel the packet from @param[in] Packet The packet to cancel **/ VOID EFIAPI UdpIoCancelSentDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN NET_BUF *Packet ) { @@ -693,53 +1005,61 @@ UdpIoCancelSentDatagram ( } /** - Issue a receive request to the UDP_IO_PORT. + Issue a receive request to the UDP_IO. This function is called when upper-layer needs packet from UDP for processing. Only one receive request is acceptable at a time so a common usage model is to invoke this function inside its Callback function when the former packet is processed. - @param[in] UdpIo The UDP_IO_PORT to receive the packet from. + @param[in] UdpIo The UDP_IO to receive the packet from. @param[in] CallBack The call back function to execute when the packet is received. @param[in] Context The opaque context passed to Callback. - @param[in] HeadLen The length of the upper-layer's protocol header. + @param[in] HeadLen The length of the upper-layer's protocol header. @retval EFI_ALREADY_STARTED There is already a pending receive request. Only one receive request is supported at a time. @retval EFI_OUT_OF_RESOURCES Failed to allocate needed resources. @retval EFI_SUCCESS The receive request is issued successfully. + @retval EFI_UNSUPPORTED The UDP version in UDP_IO is not supported. **/ EFI_STATUS EFIAPI UdpIoRecvDatagram ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN UDP_IO_CALLBACK CallBack, IN VOID *Context, IN UINT32 HeadLen ) { - UDP_RX_TOKEN *Token; + UDP_RX_TOKEN *RxToken; EFI_STATUS Status; + ASSERT ((UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) || + (UdpIo->UdpVersion == UDP_IO_UDP6_VERSION)); + if (UdpIo->RecvRequest != NULL) { return EFI_ALREADY_STARTED; } - Token = UdpIoCreateRxToken (UdpIo, CallBack, Context, HeadLen); + RxToken = UdpIoCreateRxToken (UdpIo, CallBack, Context, HeadLen); - if (Token == NULL) { + if (RxToken == NULL) { return EFI_OUT_OF_RESOURCES; } - UdpIo->RecvRequest = Token; - Status = UdpIo->Udp->Receive (UdpIo->Udp, &Token->UdpToken); + UdpIo->RecvRequest = RxToken; + if (UdpIo->UdpVersion == UDP_IO_UDP4_VERSION) { + Status = UdpIo->Protocol.Udp4->Receive (UdpIo->Protocol.Udp4, &RxToken->Token.Udp4); + } else { + Status = UdpIo->Protocol.Udp6->Receive (UdpIo->Protocol.Udp6, &RxToken->Token.Udp6); + } if (EFI_ERROR (Status)) { UdpIo->RecvRequest = NULL; - UdpIoFreeRxToken (Token); + UdpIoFreeRxToken (RxToken); } return Status; diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Driver.c b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Driver.c index 1ca0589b90..76c55c4478 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Driver.c +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Driver.c @@ -106,16 +106,16 @@ Dhcp4DriverBindingSupported ( Configure the default UDP child to receive all the DHCP traffics on this network interface. - @param[in] UdpIo The UDP IO port to configure + @param[in] UdpIo The UDP IO to configure @param[in] Context The context to the function - @retval EFI_SUCCESS The UDP IO port is successfully configured. + @retval EFI_SUCCESS The UDP IO is successfully configured. @retval Others Failed to configure the UDP child. **/ EFI_STATUS DhcpConfigUdpIo ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN VOID *Context ) { @@ -139,7 +139,7 @@ DhcpConfigUdpIo ( ZeroMem (&UdpConfigData.SubnetMask, sizeof (EFI_IPv4_ADDRESS)); ZeroMem (&UdpConfigData.RemoteAddress, sizeof (EFI_IPv4_ADDRESS)); - return UdpIo->Udp->Configure (UdpIo->Udp, &UdpConfigData);; + return UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, &UdpConfigData);; } @@ -162,7 +162,7 @@ Dhcp4CloseService ( DhcpCleanLease (DhcpSb); if (DhcpSb->UdpIo != NULL) { - UdpIoFreePort (DhcpSb->UdpIo); + UdpIoFreeIo (DhcpSb->UdpIo); DhcpSb->UdpIo = NULL; } @@ -237,7 +237,13 @@ Dhcp4CreateService ( goto ON_ERROR; } - DhcpSb->UdpIo = UdpIoCreatePort (Controller, ImageHandle, DhcpConfigUdpIo, NULL); + DhcpSb->UdpIo = UdpIoCreateIo ( + Controller, + ImageHandle, + DhcpConfigUdpIo, + UDP_IO_UDP4_VERSION, + NULL + ); if (DhcpSb->UdpIo == NULL) { Status = EFI_OUT_OF_RESOURCES; diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Dxe.inf b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Dxe.inf index f95dfed5ac..44c9d84bdb 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Dxe.inf +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Dxe.inf @@ -1,7 +1,7 @@ #/** @file # Component name for module Dhcp4 # -# Copyright (c) 2007, Intel Corporation +# Copyright (c) 2007 - 2009, 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 @@ -63,4 +63,6 @@ gEfiDhcp4ServiceBindingProtocolGuid # PROTOCOL ALWAYS_CONSUMED gEfiUdp4ProtocolGuid # PROTOCOL ALWAYS_CONSUMED gEfiDhcp4ProtocolGuid # PROTOCOL ALWAYS_CONSUMED + gEfiUdp6ServiceBindingProtocolGuid # PROTOCOL ALWAYS_CONSUMED + gEfiUdp6ProtocolGuid # PROTOCOL ALWAYS_CONSUMED diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.c b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.c index a4b8fa8137..acacec7bb1 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.c +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.c @@ -826,7 +826,7 @@ EfiDhcp4Start ( if (CompletionEvent == NULL) { while (DhcpSb->IoStatus == EFI_ALREADY_STARTED) { - DhcpSb->UdpIo->Udp->Poll (DhcpSb->UdpIo->Udp); + DhcpSb->UdpIo->Protocol.Udp4->Poll (DhcpSb->UdpIo->Protocol.Udp4); } return DhcpSb->IoStatus; @@ -951,7 +951,8 @@ EfiDhcp4RenewRebind ( if (CompletionEvent == NULL) { while (DhcpSb->IoStatus == EFI_ALREADY_STARTED) { - DhcpSb->UdpIo->Udp->Poll (DhcpSb->UdpIo->Udp); + DhcpSb->UdpIo->Protocol.Udp4->Poll (DhcpSb->UdpIo->Protocol.Udp4); + } return DhcpSb->IoStatus; @@ -1173,7 +1174,7 @@ EfiDhcp4Build ( **/ EFI_STATUS Dhcp4InstanceConfigUdpIo ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN VOID *Context ) { @@ -1206,7 +1207,7 @@ Dhcp4InstanceConfigUdpIo ( UdpConfigData.StationPort = Token->ListenPoints[0].ListenPort; } - return UdpIo->Udp->Configure (UdpIo->Udp, &UdpConfigData); + return UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, &UdpConfigData); } /** @@ -1228,7 +1229,13 @@ Dhcp4InstanceCreateUdpIo ( ASSERT (Instance->Token != NULL); DhcpSb = Instance->Service; - Instance->UdpIo = UdpIoCreatePort (DhcpSb->Controller, DhcpSb->Image, Dhcp4InstanceConfigUdpIo, Instance); + Instance->UdpIo = UdpIoCreateIo ( + DhcpSb->Controller, + DhcpSb->Image, + Dhcp4InstanceConfigUdpIo, + UDP_IO_UDP4_VERSION, + Instance + ); if (Instance->UdpIo == NULL) { return EFI_OUT_OF_RESOURCES; } else { @@ -1256,7 +1263,7 @@ DhcpDummyExtFree ( sent out. The packet will be queued to the response queue. @param UdpPacket The Dhcp4 packet. - @param Points Udp4 address pair. + @param EndPoint Udp4 address pair. @param IoStatus Status of the input. @param Context Extra info for the input. @@ -1264,7 +1271,7 @@ DhcpDummyExtFree ( VOID PxeDhcpInput ( NET_BUF *UdpPacket, - UDP_POINTS *Points, + UDP_END_POINT *EndPoint, EFI_STATUS IoStatus, VOID *Context ) @@ -1397,8 +1404,8 @@ SIGNAL_USER: // Clean up the resources dedicated for this transmit receive transaction. // NetbufQueFlush (&Instance->ResponseQueue); - UdpIoCleanPort (Instance->UdpIo); - UdpIoFreePort (Instance->UdpIo); + UdpIoCleanIo (Instance->UdpIo); + UdpIoFreeIo (Instance->UdpIo); Instance->UdpIo = NULL; Instance->Token = NULL; @@ -1439,10 +1446,10 @@ EfiDhcp4TransmitReceive ( EFI_STATUS Status; NET_FRAGMENT Frag; NET_BUF *Wrap; - UDP_POINTS EndPoint; + UDP_END_POINT EndPoint; IP4_ADDR Ip; DHCP_SERVICE *DhcpSb; - IP4_ADDR Gateway; + EFI_IP_ADDRESS Gateway; IP4_ADDR SubnetMask; if ((This == NULL) || (Token == NULL) || (Token->Packet == NULL)) { @@ -1516,16 +1523,15 @@ EfiDhcp4TransmitReceive ( } // - // Set the local address and local port. + // Set the local address and local port to ZERO. // - EndPoint.LocalAddr = 0; - EndPoint.LocalPort = 0; + ZeroMem (&EndPoint, sizeof (UDP_END_POINT)); // // Set the destination address and destination port. // CopyMem (&Ip, &Token->RemoteAddress, sizeof (EFI_IPv4_ADDRESS)); - EndPoint.RemoteAddr = NTOHL (Ip); + EndPoint.RemoteAddr.Addr[0] = NTOHL (Ip); if (Token->RemotePort == 0) { EndPoint.RemotePort = DHCP_SERVER_PORT; @@ -1537,16 +1543,16 @@ EfiDhcp4TransmitReceive ( // Get the gateway. // SubnetMask = DhcpSb->Netmask; - Gateway = 0; - if (!IP4_NET_EQUAL (DhcpSb->ClientAddr, EndPoint.RemoteAddr, SubnetMask)) { - CopyMem (&Gateway, &Token->GatewayAddress, sizeof (EFI_IPv4_ADDRESS)); - Gateway = NTOHL (Gateway); + ZeroMem (&Gateway, sizeof (Gateway)); + if (!IP4_NET_EQUAL (DhcpSb->ClientAddr, EndPoint.RemoteAddr.Addr[0], SubnetMask)) { + CopyMem (&Gateway.v4, &Token->GatewayAddress, sizeof (EFI_IPv4_ADDRESS)); + Gateway.Addr[0] = NTOHL (Gateway.Addr[0]); } // // Transmit the DHCP packet. // - Status = UdpIoSendDatagram (Instance->UdpIo, Wrap, &EndPoint, Gateway, DhcpOnPacketSent, NULL); + Status = UdpIoSendDatagram (Instance->UdpIo, Wrap, &EndPoint, &Gateway, DhcpOnPacketSent, NULL); if (EFI_ERROR (Status)) { NetbufFree (Wrap); goto ON_ERROR; @@ -1563,8 +1569,8 @@ EfiDhcp4TransmitReceive ( ON_ERROR: if (EFI_ERROR (Status) && (Instance->UdpIo != NULL)) { - UdpIoCleanPort (Instance->UdpIo); - UdpIoFreePort (Instance->UdpIo); + UdpIoCleanIo (Instance->UdpIo); + UdpIoFreeIo (Instance->UdpIo); Instance->UdpIo = NULL; Instance->Token = NULL; } @@ -1583,7 +1589,7 @@ ON_ERROR: // free it when timeout. // if (Instance->Timeout > 0) { - Instance->UdpIo->Udp->Poll (Instance->UdpIo->Udp); + Instance->UdpIo->Protocol.Udp4->Poll (Instance->UdpIo->Protocol.Udp4); gBS->RestoreTPL (OldTpl); } else { gBS->RestoreTPL (OldTpl); diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.h b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.h index 36f3ecb819..7d15b846c0 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.h +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Impl.h @@ -69,7 +69,7 @@ struct _DHCP_PROTOCOL { EFI_EVENT RenewRebindEvent; EFI_DHCP4_TRANSMIT_RECEIVE_TOKEN *Token; - UDP_IO_PORT *UdpIo; // The UDP IO used for TransmitReceive. + UDP_IO *UdpIo; // The UDP IO used for TransmitReceive. UINT32 Timeout; NET_BUF_QUEUE ResponseQueue; }; @@ -108,8 +108,8 @@ struct _DHCP_SERVICE { UINT32 T2; INTN ExtraRefresh; // This refresh is reqested by user - UDP_IO_PORT *UdpIo; // Udp child receiving all DHCP message - UDP_IO_PORT *LeaseIoPort; // Udp child with lease IP + UDP_IO *UdpIo; // Udp child receiving all DHCP message + UDP_IO *LeaseIoPort; // Udp child with lease IP EFI_DHCP4_PACKET *LastPacket; // The last sent packet for retransmission EFI_MAC_ADDRESS Mac; UINT8 HwType; diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.c b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.c index be8ceb0450..7c1fa4746a 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.c +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.c @@ -305,7 +305,7 @@ DhcpComputeLease ( DHCP driver needs this port to unicast packet to the server such as DHCP release. - @param[in] UdpIo The UDP IO port to configure + @param[in] UdpIo The UDP IO to configure @param[in] Context Dhcp service instance. @retval EFI_SUCCESS The UDP IO port is successfully configured. @@ -314,7 +314,7 @@ DhcpComputeLease ( **/ EFI_STATUS DhcpConfigLeaseIoPort ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN VOID *Context ) { @@ -349,7 +349,7 @@ DhcpConfigLeaseIoPort ( ZeroMem (&UdpConfigData.RemoteAddress, sizeof (EFI_IPv4_ADDRESS)); - Status = UdpIo->Udp->Configure (UdpIo->Udp, &UdpConfigData); + Status = UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, &UdpConfigData); if (EFI_ERROR (Status)) { return Status; @@ -364,7 +364,7 @@ DhcpConfigLeaseIoPort ( Ip = HTONL (DhcpSb->Para->Router); CopyMem (&Gateway, &Ip, sizeof (EFI_IPv4_ADDRESS)); - UdpIo->Udp->Routes (UdpIo->Udp, FALSE, &Subnet, &Subnet, &Gateway); + UdpIo->Protocol.Udp4->Routes (UdpIo->Protocol.Udp4, FALSE, &Subnet, &Subnet, &Gateway); } return EFI_SUCCESS; @@ -402,7 +402,7 @@ DhcpLeaseAcquired ( } if (DhcpSb->LeaseIoPort != NULL) { - UdpIoFreePort (DhcpSb->LeaseIoPort); + UdpIoFreeIo (DhcpSb->LeaseIoPort); } // @@ -410,10 +410,11 @@ DhcpLeaseAcquired ( // and transmit unicast packet with it as source address. Don't // start receive on this port, the queued packet will be timeout. // - DhcpSb->LeaseIoPort = UdpIoCreatePort ( + DhcpSb->LeaseIoPort = UdpIoCreateIo ( DhcpSb->Controller, DhcpSb->Image, DhcpConfigLeaseIoPort, + UDP_IO_UDP4_VERSION, DhcpSb ); @@ -467,7 +468,7 @@ DhcpCleanLease ( DhcpSb->ExtraRefresh = FALSE; if (DhcpSb->LeaseIoPort != NULL) { - UdpIoFreePort (DhcpSb->LeaseIoPort); + UdpIoFreeIo (DhcpSb->LeaseIoPort); DhcpSb->LeaseIoPort = NULL; } @@ -935,7 +936,7 @@ ON_EXIT: state machine. @param UdpPacket The UDP packets received. - @param Points The local/remote UDP access points + @param EndPoint The local/remote UDP access point @param IoStatus The status of the UDP receive @param Context The opaque parameter to the function. @@ -943,7 +944,7 @@ ON_EXIT: VOID DhcpInput ( NET_BUF *UdpPacket, - UDP_POINTS *Points, + UDP_END_POINT *EndPoint, EFI_STATUS IoStatus, VOID *Context ) @@ -1106,7 +1107,7 @@ DhcpReleasePacket ( Release the net buffer when packet is sent. @param UdpPacket The UDP packets received. - @param Points The local/remote UDP access points + @param EndPoint The local/remote UDP access point @param IoStatus The status of the UDP receive @param Context The opaque parameter to the function. @@ -1114,7 +1115,7 @@ DhcpReleasePacket ( VOID DhcpOnPacketSent ( NET_BUF *Packet, - UDP_POINTS *Points, + UDP_END_POINT *EndPoint, EFI_STATUS IoStatus, VOID *Context ) @@ -1157,8 +1158,8 @@ DhcpSendMessage ( EFI_DHCP4_PACKET *NewPacket; EFI_DHCP4_HEADER *Head; EFI_DHCP4_HEADER *SeedHead; - UDP_IO_PORT *UdpIo; - UDP_POINTS EndPoint; + UDP_IO *UdpIo; + UDP_END_POINT EndPoint; NET_BUF *Wrap; NET_FRAGMENT Frag; EFI_STATUS Status; @@ -1363,16 +1364,16 @@ DhcpSendMessage ( // Broadcast the message, unless we know the server address. // Use the lease UdpIo port to send the unicast packet. // - EndPoint.RemoteAddr = 0xffffffff; - EndPoint.LocalAddr = 0; - EndPoint.RemotePort = DHCP_SERVER_PORT; - EndPoint.LocalPort = DHCP_CLIENT_PORT; - UdpIo = DhcpSb->UdpIo; + EndPoint.RemoteAddr.Addr[0] = 0xffffffff; + EndPoint.LocalAddr.Addr[0] = 0; + EndPoint.RemotePort = DHCP_SERVER_PORT; + EndPoint.LocalPort = DHCP_CLIENT_PORT; + UdpIo = DhcpSb->UdpIo; if ((DhcpSb->DhcpState == Dhcp4Renewing) || (Type == DHCP_MSG_RELEASE)) { - EndPoint.RemoteAddr = DhcpSb->ServerAddr; - EndPoint.LocalAddr = DhcpSb->ClientAddr; - UdpIo = DhcpSb->LeaseIoPort; + EndPoint.RemoteAddr.Addr[0] = DhcpSb->ServerAddr; + EndPoint.LocalAddr.Addr[0] = DhcpSb->ClientAddr; + UdpIo = DhcpSb->LeaseIoPort; } ASSERT (UdpIo != NULL); @@ -1382,7 +1383,7 @@ DhcpSendMessage ( UdpIo, Wrap, &EndPoint, - 0, + NULL, DhcpOnPacketSent, DhcpSb ); @@ -1411,8 +1412,8 @@ DhcpRetransmit ( IN DHCP_SERVICE *DhcpSb ) { - UDP_IO_PORT *UdpIo; - UDP_POINTS EndPoint; + UDP_IO *UdpIo; + UDP_END_POINT EndPoint; NET_BUF *Wrap; NET_FRAGMENT Frag; EFI_STATUS Status; @@ -1435,16 +1436,16 @@ DhcpRetransmit ( // // Broadcast the message, unless we know the server address. // - EndPoint.RemotePort = DHCP_SERVER_PORT; - EndPoint.LocalPort = DHCP_CLIENT_PORT; - EndPoint.RemoteAddr = 0xffffffff; - EndPoint.LocalAddr = 0; - UdpIo = DhcpSb->UdpIo; + EndPoint.RemotePort = DHCP_SERVER_PORT; + EndPoint.LocalPort = DHCP_CLIENT_PORT; + EndPoint.RemoteAddr.Addr[0] = 0xffffffff; + EndPoint.LocalAddr.Addr[0] = 0; + UdpIo = DhcpSb->UdpIo; if (DhcpSb->DhcpState == Dhcp4Renewing) { - EndPoint.RemoteAddr = DhcpSb->ServerAddr; - EndPoint.LocalAddr = DhcpSb->ClientAddr; - UdpIo = DhcpSb->LeaseIoPort; + EndPoint.RemoteAddr.Addr[0] = DhcpSb->ServerAddr; + EndPoint.LocalAddr.Addr[0] = DhcpSb->ClientAddr; + UdpIo = DhcpSb->LeaseIoPort; } ASSERT (UdpIo != NULL); @@ -1454,7 +1455,7 @@ DhcpRetransmit ( UdpIo, Wrap, &EndPoint, - 0, + NULL, DhcpOnPacketSent, DhcpSb ); diff --git a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.h b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.h index d9298ada2f..bc7ef689e1 100644 --- a/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.h +++ b/MdeModulePkg/Universal/Network/Dhcp4Dxe/Dhcp4Io.h @@ -1,7 +1,7 @@ /** @file The DHCP4 protocol implementation. -Copyright (c) 2006 - 2008, Intel Corporation.
+Copyright (c) 2006 - 2009, 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 @@ -140,7 +140,7 @@ DhcpOnTimerTick ( state machine. @param UdpPacket The UDP packets received. - @param Points The local/remote UDP access points + @param EndPoint The local/remote UDP access point @param IoStatus The status of the UDP receive @param Context The opaque parameter to the function. @@ -148,7 +148,7 @@ DhcpOnTimerTick ( VOID DhcpInput ( NET_BUF *UdpPacket, - UDP_POINTS *Points, + UDP_END_POINT *EndPoint, EFI_STATUS IoStatus, VOID *Context ); @@ -183,7 +183,7 @@ DhcpCleanLease ( Release the net buffer when packet is sent. @param UdpPacket The UDP packets received. - @param Points The local/remote UDP access points + @param EndPoint The local/remote UDP access point @param IoStatus The status of the UDP receive @param Context The opaque parameter to the function. @@ -191,7 +191,7 @@ DhcpCleanLease ( VOID DhcpOnPacketSent ( NET_BUF *Packet, - UDP_POINTS *Points, + UDP_END_POINT *EndPoint, EFI_STATUS IoStatus, VOID *Context ); diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Driver.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Driver.c index 56c993eee7..63dcc6c5df 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Driver.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Driver.c @@ -1,7 +1,7 @@ /** @file Implementation of Mtftp drivers. -Copyright (c) 2006 - 2007, Intel Corporation
+Copyright (c) 2006 - 2009, 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 @@ -107,7 +107,7 @@ Mtftp4DriverBindingSupported ( Just leave the Udp child unconfigured. When UDP is unloaded, MTFTP will be informed with DriverBinding Stop. - @param UdpIo The UDP port to configure + @param UdpIo The UDP_IO to configure @param Context The opaque parameter to the callback @retval EFI_SUCCESS It always return EFI_SUCCESS directly. @@ -115,7 +115,7 @@ Mtftp4DriverBindingSupported ( **/ EFI_STATUS Mtftp4ConfigNullUdp ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN VOID *Context ) { @@ -201,7 +201,13 @@ Mtftp4CreateService ( return Status; } - MtftpSb->ConnectUdp = UdpIoCreatePort (Controller, Image, Mtftp4ConfigNullUdp, NULL); + MtftpSb->ConnectUdp = UdpIoCreateIo ( + Controller, + Image, + Mtftp4ConfigNullUdp, + UDP_IO_UDP4_VERSION, + NULL + ); if (MtftpSb->ConnectUdp == NULL) { gBS->CloseEvent (MtftpSb->TimerToGetMap); @@ -226,7 +232,7 @@ Mtftp4CleanService ( IN MTFTP4_SERVICE *MtftpSb ) { - UdpIoFreePort (MtftpSb->ConnectUdp); + UdpIoFreeIo (MtftpSb->ConnectUdp); gBS->CloseEvent (MtftpSb->TimerToGetMap); gBS->CloseEvent (MtftpSb->Timer); } @@ -467,10 +473,11 @@ Mtftp4ServiceBindingCreateChild ( Mtftp4InitProtocol (MtftpSb, Instance); - Instance->UnicastPort = UdpIoCreatePort ( + Instance->UnicastPort = UdpIoCreateIo ( MtftpSb->Controller, MtftpSb->Image, Mtftp4ConfigNullUdp, + UDP_IO_UDP4_VERSION, Instance ); @@ -530,7 +537,7 @@ Mtftp4ServiceBindingCreateChild ( ON_ERROR: if (EFI_ERROR (Status)) { - UdpIoFreePort (Instance->UnicastPort); + UdpIoFreeIo (Instance->UnicastPort); gBS->FreePool (Instance); } @@ -623,7 +630,7 @@ Mtftp4ServiceBindingDestroyChild ( OldTpl = gBS->RaiseTPL (TPL_CALLBACK); Mtftp4CleanOperation (Instance, EFI_DEVICE_ERROR); - UdpIoFreePort (Instance->UnicastPort); + UdpIoFreeIo (Instance->UnicastPort); RemoveEntryList (&Instance->Link); MtftpSb->ChildrenNum--; diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Dxe.inf b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Dxe.inf index b45914d3af..feb1594b25 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Dxe.inf +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Dxe.inf @@ -1,7 +1,7 @@ #/** @file # Component name for module Mtftp4 # -# Copyright (c) 2007, Intel Corporation +# Copyright (c) 2007 - 2009, 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 @@ -67,4 +67,6 @@ gEfiMtftp4ProtocolGuid # PROTOCOL ALWAYS_CONSUMED gEfiUdp4ServiceBindingProtocolGuid # PROTOCOL ALWAYS_CONSUMED gEfiUdp4ProtocolGuid # PROTOCOL ALWAYS_CONSUMED + gEfiUdp6ServiceBindingProtocolGuid # PROTOCOL ALWAYS_CONSUMED + gEfiUdp6ProtocolGuid # PROTOCOL ALWAYS_CONSUMED diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c index 4c459b22b8..06f0231e8b 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.c @@ -50,7 +50,7 @@ Mtftp4CleanOperation ( } ASSERT (Instance->UnicastPort != NULL); - UdpIoCleanPort (Instance->UnicastPort); + UdpIoCleanIo (Instance->UnicastPort); if (Instance->LastPacket != NULL) { NetbufFree (Instance->LastPacket); @@ -58,7 +58,7 @@ Mtftp4CleanOperation ( } if (Instance->McastUdpPort != NULL) { - UdpIoFreePort (Instance->McastUdpPort); + UdpIoFreeIo (Instance->McastUdpPort); Instance->McastUdpPort = NULL; } @@ -211,9 +211,8 @@ Mtftp4OverrideValid ( the UDP is reconfigured. @param Instance The Mtftp instance - @param UdpPort The UDP port to poll - @param UdpCfgData The UDP configure data to reconfigure the UDP - port. + @param UdpIo The UDP_IO to poll + @param UdpCfgData The UDP configure data to reconfigure the UDP_IO @retval TRUE The default address is retrieved and UDP is reconfigured. @retval FALSE Some error occured. @@ -222,7 +221,7 @@ Mtftp4OverrideValid ( BOOLEAN Mtftp4GetMapping ( IN MTFTP4_PROTOCOL *Instance, - IN UDP_IO_PORT *UdpPort, + IN UDP_IO *UdpIo, IN EFI_UDP4_CONFIG_DATA *UdpCfgData ) { @@ -234,7 +233,7 @@ Mtftp4GetMapping ( ASSERT (Instance->Config.UseDefaultSetting); Service = Instance->Service; - Udp = UdpPort->Udp; + Udp = UdpIo->Protocol.Udp4; Status = gBS->SetTimer ( Service->TimerToGetMap, @@ -263,7 +262,7 @@ Mtftp4GetMapping ( /** Configure the UDP port for unicast receiving. - @param UdpIo The UDP port + @param UdpIo The UDP_IO instance @param Instance The MTFTP session @retval EFI_SUCCESS The UDP port is successfully configured for the @@ -272,7 +271,7 @@ Mtftp4GetMapping ( **/ EFI_STATUS Mtftp4ConfigUnicastPort ( - IN UDP_IO_PORT *UdpIo, + IN UDP_IO *UdpIo, IN MTFTP4_PROTOCOL *Instance ) { @@ -301,7 +300,7 @@ Mtftp4ConfigUnicastPort ( Ip = HTONL (Instance->ServerIp); CopyMem (&UdpConfig.RemoteAddress, &Ip, sizeof (EFI_IPv4_ADDRESS)); - Status = UdpIo->Udp->Configure (UdpIo->Udp, &UdpConfig); + Status = UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, &UdpConfig); if ((Status == EFI_NO_MAPPING) && Mtftp4GetMapping (Instance, UdpIo, &UdpConfig)) { return EFI_SUCCESS; @@ -312,9 +311,15 @@ Mtftp4ConfigUnicastPort ( // The station IP address is manually configured and the Gateway IP is not 0. // Add the default route for this UDP instance. // - Status = UdpIo->Udp->Routes (UdpIo->Udp, FALSE, &mZeroIp4Addr, &mZeroIp4Addr, &Config->GatewayIp); + Status = UdpIo->Protocol.Udp4->Routes ( + UdpIo->Protocol.Udp4, + FALSE, + &mZeroIp4Addr, + &mZeroIp4Addr, + &Config->GatewayIp + ); if (EFI_ERROR (Status)) { - UdpIo->Udp->Configure (UdpIo->Udp, NULL); + UdpIo->Protocol.Udp4->Configure (UdpIo->Protocol.Udp4, NULL); } } return Status; @@ -1070,7 +1075,7 @@ EfiMtftp4Poll ( return EFI_DEVICE_ERROR; } - Udp = Instance->UnicastPort->Udp; + Udp = Instance->UnicastPort->Protocol.Udp4; return Udp->Poll (Udp); } diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.h b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.h index 8f39df3c4f..d4644c7084 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.h +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Impl.h @@ -85,7 +85,7 @@ struct _MTFTP4_SERVICE { // This UDP child is used to keep the connection between the UDP // and MTFTP, so MTFTP will be notified when UDP is uninstalled. // - UDP_IO_PORT *ConnectUdp; + UDP_IO *ConnectUdp; }; @@ -131,7 +131,7 @@ struct _MTFTP4_PROTOCOL { UINT16 ListeningPort; UINT16 ConnectedPort; IP4_ADDR Gateway; - UDP_IO_PORT *UnicastPort; + UDP_IO *UnicastPort; // // Timeout and retransmit status @@ -148,7 +148,7 @@ struct _MTFTP4_PROTOCOL { IP4_ADDR McastIp; UINT16 McastPort; BOOLEAN Master; - UDP_IO_PORT *McastUdpPort; + UDP_IO *McastUdpPort; }; /** diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Rrq.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Rrq.c index f5b6bbc887..524db8ff63 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Rrq.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Rrq.c @@ -20,7 +20,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. The packet process callback for MTFTP download. @param UdpPacket The packet received - @param Points The local/remote access point of the packet + @param EndPoint The local/remote access point of the packet @param IoStatus The status of the receiving @param Context Opaque parameter, which is the MTFTP session @@ -28,7 +28,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. VOID Mtftp4RrqInput ( IN NET_BUF *UdpPacket, - IN UDP_POINTS *Points, + IN UDP_END_POINT *EndPoint, IN EFI_STATUS IoStatus, IN VOID *Context ); @@ -370,7 +370,7 @@ Mtftp4RrqOackValid ( /** Configure a UDP IO port to receive the multicast. - @param McastIo The UDP IO port to configure + @param McastIo The UDP IO to configure @param Context The opaque parameter to the function which is the MTFTP session. @@ -380,7 +380,7 @@ Mtftp4RrqOackValid ( **/ EFI_STATUS Mtftp4RrqConfigMcastPort ( - IN UDP_IO_PORT *McastIo, + IN UDP_IO *McastIo, IN VOID *Context ) { @@ -412,7 +412,7 @@ Mtftp4RrqConfigMcastPort ( Ip = HTONL (Instance->ServerIp); CopyMem (&UdpConfig.RemoteAddress, &Ip, sizeof (EFI_IPv4_ADDRESS)); - Status = McastIo->Udp->Configure (McastIo->Udp, &UdpConfig); + Status = McastIo->Protocol.Udp4->Configure (McastIo->Protocol.Udp4, &UdpConfig); if (EFI_ERROR (Status)) { return Status; @@ -424,16 +424,16 @@ Mtftp4RrqConfigMcastPort ( // The station IP address is manually configured and the Gateway IP is not 0. // Add the default route for this UDP instance. // - Status = McastIo->Udp->Routes ( - McastIo->Udp, - FALSE, - &mZeroIp4Addr, - &mZeroIp4Addr, - &Config->GatewayIp - ); + Status = McastIo->Protocol.Udp4->Routes ( + McastIo->Protocol.Udp4, + FALSE, + &mZeroIp4Addr, + &mZeroIp4Addr, + &Config->GatewayIp + ); if (EFI_ERROR (Status)) { - McastIo->Udp->Configure (McastIo->Udp, NULL); + McastIo->Protocol.Udp4->Configure (McastIo->Protocol.Udp4, NULL); return Status; } } @@ -444,7 +444,7 @@ Mtftp4RrqConfigMcastPort ( Ip = HTONL (Instance->McastIp); CopyMem (&Group, &Ip, sizeof (EFI_IPv4_ADDRESS)); - return McastIo->Udp->Groups (McastIo->Udp, TRUE, &Group); + return McastIo->Protocol.Udp4->Groups (McastIo->Protocol.Udp4, TRUE, &Group); } @@ -539,10 +539,11 @@ Mtftp4RrqHandleOack ( // Instance->McastIp = Reply.McastIp; Instance->McastPort = Reply.McastPort; - Instance->McastUdpPort = UdpIoCreatePort ( + Instance->McastUdpPort = UdpIoCreateIo ( Instance->Service->Controller, Instance->Service->Image, Mtftp4RrqConfigMcastPort, + UDP_IO_UDP4_VERSION, Instance ); @@ -598,7 +599,7 @@ Mtftp4RrqHandleOack ( The packet process callback for MTFTP download. @param UdpPacket The packet received - @param Points The local/remote access point of the packet + @param EndPoint The local/remote access point of the packet @param IoStatus The status of the receiving @param Context Opaque parameter, which is the MTFTP session @@ -606,7 +607,7 @@ Mtftp4RrqHandleOack ( VOID Mtftp4RrqInput ( IN NET_BUF *UdpPacket, - IN UDP_POINTS *Points, + IN UDP_END_POINT *EndPoint, IN EFI_STATUS IoStatus, IN VOID *Context ) @@ -637,7 +638,7 @@ Mtftp4RrqInput ( // // Find the port this packet is from to restart receive correctly. // - Multicast = (BOOLEAN) (Points->LocalAddr == Instance->McastIp); + Multicast = (BOOLEAN) (EndPoint->LocalAddr.Addr[0] == Instance->McastIp); if (UdpPacket->TotalSize < MTFTP4_OPCODE_LEN) { goto ON_EXIT; @@ -649,11 +650,11 @@ Mtftp4RrqInput ( // is required to use the same port as RemotePort to multicast the // data. // - if (Points->RemotePort != Instance->ConnectedPort) { + if (EndPoint->RemotePort != Instance->ConnectedPort) { if (Instance->ConnectedPort != 0) { goto ON_EXIT; } else { - Instance->ConnectedPort = Points->RemotePort; + Instance->ConnectedPort = EndPoint->RemotePort; } } diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c index 5c8516f630..a758f4bb0c 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Support.c @@ -359,7 +359,7 @@ Mtftp4SendError ( It simply frees the packet. @param Packet The transmitted (or failed to) packet - @param Points The local and remote UDP access point + @param EndPoint The local and remote UDP access point @param IoStatus The result of the transmission @param Context Opaque parameter to the callback @@ -367,7 +367,7 @@ Mtftp4SendError ( VOID Mtftp4OnPacketSent ( IN NET_BUF *Packet, - IN UDP_POINTS *Points, + IN UDP_END_POINT *EndPoint, IN EFI_STATUS IoStatus, IN VOID *Context ) @@ -415,7 +415,7 @@ Mtftp4SendPacket ( IN OUT NET_BUF *Packet ) { - UDP_POINTS UdpPoint; + UDP_END_POINT UdpPoint; EFI_STATUS Status; UINT16 OpCode; UINT16 Value; @@ -427,14 +427,13 @@ Mtftp4SendPacket ( NetbufFree (Instance->LastPacket); } - Instance->LastPacket = Packet; + Instance->LastPacket = Packet; - Instance->CurRetry = 0; + Instance->CurRetry = 0; Mtftp4SetTimeout (Instance); - UdpPoint.LocalAddr = 0; - UdpPoint.LocalPort = 0; - UdpPoint.RemoteAddr = Instance->ServerIp; + ZeroMem (&UdpPoint, sizeof (UdpPoint)); + UdpPoint.RemoteAddr.Addr[0] = Instance->ServerIp; // // Send the requests to the listening port, other packets @@ -457,7 +456,7 @@ Mtftp4SendPacket ( Instance->UnicastPort, Packet, &UdpPoint, - 0, + NULL, Mtftp4OnPacketSent, Instance ); @@ -484,16 +483,15 @@ Mtftp4Retransmit ( IN MTFTP4_PROTOCOL *Instance ) { - UDP_POINTS UdpPoint; + UDP_END_POINT UdpPoint; EFI_STATUS Status; UINT16 OpCode; UINT16 Value; ASSERT (Instance->LastPacket != NULL); - UdpPoint.LocalAddr = 0; - UdpPoint.LocalPort = 0; - UdpPoint.RemoteAddr = Instance->ServerIp; + ZeroMem (&UdpPoint, sizeof (UdpPoint)); + UdpPoint.RemoteAddr.Addr[0] = Instance->ServerIp; // // Set the requests to the listening port, other packets to the connected port @@ -514,7 +512,7 @@ Mtftp4Retransmit ( Instance->UnicastPort, Instance->LastPacket, &UdpPoint, - 0, + NULL, Mtftp4OnPacketSent, Instance ); diff --git a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Wrq.c b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Wrq.c index 872b2953c2..87ec0c1d64 100644 --- a/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Wrq.c +++ b/MdeModulePkg/Universal/Network/Mtftp4Dxe/Mtftp4Wrq.c @@ -329,7 +329,7 @@ Mtftp4WrqHandleOack ( The input process routine for MTFTP upload. @param UdpPacket The received MTFTP packet. - @param Points The local/remote access point + @param EndPoint The local/remote access point @param IoStatus The result of the packet receiving @param Context Opaque parameter for the callback, which is the MTFTP session. @@ -337,7 +337,7 @@ Mtftp4WrqHandleOack ( VOID Mtftp4WrqInput ( IN NET_BUF *UdpPacket, - IN UDP_POINTS *Points, + IN UDP_END_POINT *EndPoint, IN EFI_STATUS IoStatus, IN VOID *Context ) @@ -371,11 +371,11 @@ Mtftp4WrqInput ( // Client send initial request to server's listening port. Server // will select a UDP port to communicate with the client. // - if (Points->RemotePort != Instance->ConnectedPort) { + if (EndPoint->RemotePort != Instance->ConnectedPort) { if (Instance->ConnectedPort != 0) { goto ON_EXIT; } else { - Instance->ConnectedPort = Points->RemotePort; + Instance->ConnectedPort = EndPoint->RemotePort; } } diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.h b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.h index 09388b5d83..f43043758e 100644 --- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.h +++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Driver.h @@ -1,7 +1,7 @@ /** @file Tcp driver function header. -Copyright (c) 2005 - 2006, Intel Corporation
+Copyright (c) 2005 - 2009, 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 @@ -78,7 +78,7 @@ typedef struct _TCP4_PROTO_DATA { VOID Tcp4RxCallback ( IN EFI_STATUS Status, - IN ICMP_ERROR IcmpErr, + IN UINT8 IcmpErr, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Pkt, IN VOID *Context OPTIONAL diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Func.h b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Func.h index fc14b3b4a6..2c6054b5b3 100644 --- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Func.h +++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Func.h @@ -1,7 +1,7 @@ /** @file Tcp function header file. -Copyright (c) 2005 - 2006, Intel Corporation
+Copyright (c) 2005 - 2009, 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 @@ -433,7 +433,7 @@ TcpGetMaxSndNxt ( VOID TcpIcmpInput ( IN NET_BUF *Nbuf, - IN ICMP_ERROR IcmpErr, + IN UINT8 IcmpErr, IN UINT32 Src, IN UINT32 Dst ); diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c index a2cf20edf1..b29e6cb672 100644 --- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c +++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Input.c @@ -1407,7 +1407,7 @@ DISCARD: VOID TcpIcmpInput ( IN NET_BUF *Nbuf, - IN ICMP_ERROR IcmpErr, + IN UINT8 IcmpErr, IN UINT32 Src, IN UINT32 Dst ) diff --git a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Io.c b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Io.c index 39e8fbe9df..3adf356e0b 100644 --- a/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Io.c +++ b/MdeModulePkg/Universal/Network/Tcp4Dxe/Tcp4Io.c @@ -30,7 +30,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED. VOID Tcp4RxCallback ( IN EFI_STATUS Status, - IN ICMP_ERROR IcmpErr, + IN UINT8 IcmpErr, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Pkt, IN VOID *Context OPTIONAL diff --git a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c index b0bc0ae43d..c923117a1c 100644 --- a/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c +++ b/MdeModulePkg/Universal/Network/Udp4Dxe/Udp4Impl.c @@ -86,7 +86,7 @@ Udp4DgramSent ( VOID Udp4DgramRcvd ( IN EFI_STATUS Status, - IN ICMP_ERROR IcmpError, + IN UINT8 IcmpError, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Packet, IN VOID *Context @@ -227,7 +227,7 @@ Udp4Demultiplex ( VOID Udp4IcmpHandler ( IN UDP4_SERVICE_DATA *Udp4Service, - IN ICMP_ERROR IcmpError, + IN UINT8 IcmpError, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Packet ); @@ -1029,7 +1029,7 @@ Udp4DgramSent ( VOID Udp4DgramRcvd ( IN EFI_STATUS Status, - IN ICMP_ERROR IcmpError, + IN UINT8 IcmpError, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Packet, IN VOID *Context @@ -1780,7 +1780,7 @@ Udp4SendPortUnreach ( VOID Udp4IcmpHandler ( IN UDP4_SERVICE_DATA *Udp4Service, - IN ICMP_ERROR IcmpError, + IN UINT8 IcmpError, IN EFI_NET_SESSION_DATA *NetSession, IN NET_BUF *Packet )