| /* | |
| * FreeRTOS+TCP V2.0.11 | |
| * Copyright (C) 2017 Amazon.com, Inc. or its affiliates. All Rights Reserved. | |
| * | |
| * Permission is hereby granted, free of charge, to any person obtaining a copy of | |
| * this software and associated documentation files (the "Software"), to deal in | |
| * the Software without restriction, including without limitation the rights to | |
| * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of | |
| * the Software, and to permit persons to whom the Software is furnished to do so, | |
| * subject to the following conditions: | |
| * | |
| * The above copyright notice and this permission notice shall be included in all | |
| * copies or substantial portions of the Software. | |
| * | |
| * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
| * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS | |
| * FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR | |
| * COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER | |
| * IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN | |
| * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. | |
| * | |
| * http://aws.amazon.com/freertos | |
| * http://www.FreeRTOS.org | |
| */ | |
| /* Standard includes. */ | |
| #include <stdint.h> | |
| #include <stdio.h> | |
| /* FreeRTOS includes. */ | |
| #include "FreeRTOS.h" | |
| #include "task.h" | |
| #include "queue.h" | |
| #include "semphr.h" | |
| /* FreeRTOS+TCP includes. */ | |
| #include "FreeRTOS_IP.h" | |
| #include "FreeRTOS_Sockets.h" | |
| #include "FreeRTOS_IP_Private.h" | |
| #include "FreeRTOS_UDP_IP.h" | |
| #include "FreeRTOS_ARP.h" | |
| #include "FreeRTOS_DHCP.h" | |
| #include "NetworkInterface.h" | |
| #include "NetworkBufferManagement.h" | |
| #if( ipconfigUSE_DNS == 1 ) | |
| #include "FreeRTOS_DNS.h" | |
| #endif | |
| /* The expected IP version and header length coded into the IP header itself. */ | |
| #define ipIP_VERSION_AND_HEADER_LENGTH_BYTE ( ( uint8_t ) 0x45 ) | |
| /* Part of the Ethernet and IP headers are always constant when sending an IPv4 | |
| UDP packet. This array defines the constant parts, allowing this part of the | |
| packet to be filled in using a simple memcpy() instead of individual writes. */ | |
| UDPPacketHeader_t xDefaultPartUDPPacketHeader = | |
| { | |
| /* .ucBytes : */ | |
| { | |
| 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, /* Ethernet source MAC address. */ | |
| 0x08, 0x00, /* Ethernet frame type. */ | |
| ipIP_VERSION_AND_HEADER_LENGTH_BYTE, /* ucVersionHeaderLength. */ | |
| 0x00, /* ucDifferentiatedServicesCode. */ | |
| 0x00, 0x00, /* usLength. */ | |
| 0x00, 0x00, /* usIdentification. */ | |
| 0x00, 0x00, /* usFragmentOffset. */ | |
| ipconfigUDP_TIME_TO_LIVE, /* ucTimeToLive */ | |
| ipPROTOCOL_UDP, /* ucProtocol. */ | |
| 0x00, 0x00, /* usHeaderChecksum. */ | |
| 0x00, 0x00, 0x00, 0x00 /* Source IP address. */ | |
| } | |
| }; | |
| /*-----------------------------------------------------------*/ | |
| void vProcessGeneratedUDPPacket( NetworkBufferDescriptor_t * const pxNetworkBuffer ) | |
| { | |
| UDPPacket_t *pxUDPPacket; | |
| IPHeader_t *pxIPHeader; | |
| eARPLookupResult_t eReturned; | |
| uint32_t ulIPAddress = pxNetworkBuffer->ulIPAddress; | |
| /* Map the UDP packet onto the start of the frame. */ | |
| pxUDPPacket = ( UDPPacket_t * ) pxNetworkBuffer->pucEthernetBuffer; | |
| /* Determine the ARP cache status for the requested IP address. */ | |
| eReturned = eARPGetCacheEntry( &( ulIPAddress ), &( pxUDPPacket->xEthernetHeader.xDestinationAddress ) ); | |
| if( eReturned != eCantSendPacket ) | |
| { | |
| if( eReturned == eARPCacheHit ) | |
| { | |
| #if( ipconfigDRIVER_INCLUDED_TX_IP_CHECKSUM == 0 ) | |
| uint8_t ucSocketOptions; | |
| #endif | |
| iptraceSENDING_UDP_PACKET( pxNetworkBuffer->ulIPAddress ); | |
| /* Create short cuts to the data within the packet. */ | |
| pxIPHeader = &( pxUDPPacket->xIPHeader ); | |
| #if ( ipconfigSUPPORT_OUTGOING_PINGS == 1 ) | |
| /* Is it possible that the packet is not actually a UDP packet | |
| after all, but an ICMP packet. */ | |
| if( pxNetworkBuffer->usPort != ipPACKET_CONTAINS_ICMP_DATA ) | |
| #endif /* ipconfigSUPPORT_OUTGOING_PINGS */ | |
| { | |
| UDPHeader_t *pxUDPHeader; | |
| pxUDPHeader = &( pxUDPPacket->xUDPHeader ); | |
| pxUDPHeader->usDestinationPort = pxNetworkBuffer->usPort; | |
| pxUDPHeader->usSourcePort = pxNetworkBuffer->usBoundPort; | |
| pxUDPHeader->usLength = ( uint16_t ) ( pxNetworkBuffer->xDataLength + sizeof( UDPHeader_t ) ); | |
| pxUDPHeader->usLength = FreeRTOS_htons( pxUDPHeader->usLength ); | |
| pxUDPHeader->usChecksum = 0u; | |
| } | |
| /* memcpy() the constant parts of the header information into | |
| the correct location within the packet. This fills in: | |
| xEthernetHeader.xSourceAddress | |
| xEthernetHeader.usFrameType | |
| xIPHeader.ucVersionHeaderLength | |
| xIPHeader.ucDifferentiatedServicesCode | |
| xIPHeader.usLength | |
| xIPHeader.usIdentification | |
| xIPHeader.usFragmentOffset | |
| xIPHeader.ucTimeToLive | |
| xIPHeader.ucProtocol | |
| and | |
| xIPHeader.usHeaderChecksum | |
| */ | |
| /* Save options now, as they will be overwritten by memcpy */ | |
| #if( ipconfigDRIVER_INCLUDED_TX_IP_CHECKSUM == 0 ) | |
| ucSocketOptions = pxNetworkBuffer->pucEthernetBuffer[ ipSOCKET_OPTIONS_OFFSET ]; | |
| #endif | |
| /* | |
| * Offset the memcpy by the size of a MAC address to start at the packet's | |
| * Ethernet header 'source' MAC address; the preceding 'destination' should not be altered. | |
| */ | |
| char *pxUdpSrcAddrOffset = ( char *) pxUDPPacket + sizeof( MACAddress_t ); | |
| memcpy( pxUdpSrcAddrOffset, xDefaultPartUDPPacketHeader.ucBytes, sizeof( xDefaultPartUDPPacketHeader ) ); | |
| #if ipconfigSUPPORT_OUTGOING_PINGS == 1 | |
| if( pxNetworkBuffer->usPort == ipPACKET_CONTAINS_ICMP_DATA ) | |
| { | |
| pxIPHeader->ucProtocol = ipPROTOCOL_ICMP; | |
| pxIPHeader->usLength = ( uint16_t ) ( pxNetworkBuffer->xDataLength + sizeof( IPHeader_t ) ); | |
| } | |
| else | |
| #endif /* ipconfigSUPPORT_OUTGOING_PINGS */ | |
| { | |
| pxIPHeader->usLength = ( uint16_t ) ( pxNetworkBuffer->xDataLength + sizeof( IPHeader_t ) + sizeof( UDPHeader_t ) ); | |
| } | |
| /* The total transmit size adds on the Ethernet header. */ | |
| pxNetworkBuffer->xDataLength = pxIPHeader->usLength + sizeof( EthernetHeader_t ); | |
| pxIPHeader->usLength = FreeRTOS_htons( pxIPHeader->usLength ); | |
| /* HT:endian: changed back to network endian */ | |
| pxIPHeader->ulDestinationIPAddress = pxNetworkBuffer->ulIPAddress; | |
| #if( ipconfigUSE_LLMNR == 1 ) | |
| { | |
| /* LLMNR messages are typically used on a LAN and they're | |
| * not supposed to cross routers */ | |
| if( pxNetworkBuffer->ulIPAddress == ipLLMNR_IP_ADDR ) | |
| { | |
| pxIPHeader->ucTimeToLive = 0x01; | |
| } | |
| } | |
| #endif | |
| #if( ipconfigDRIVER_INCLUDED_TX_IP_CHECKSUM == 0 ) | |
| { | |
| pxIPHeader->usHeaderChecksum = 0u; | |
| pxIPHeader->usHeaderChecksum = usGenerateChecksum( 0UL, ( uint8_t * ) &( pxIPHeader->ucVersionHeaderLength ), ipSIZE_OF_IPv4_HEADER ); | |
| pxIPHeader->usHeaderChecksum = ~FreeRTOS_htons( pxIPHeader->usHeaderChecksum ); | |
| if( ( ucSocketOptions & ( uint8_t ) FREERTOS_SO_UDPCKSUM_OUT ) != 0u ) | |
| { | |
| usGenerateProtocolChecksum( (uint8_t*)pxUDPPacket, pxNetworkBuffer->xDataLength, pdTRUE ); | |
| } | |
| else | |
| { | |
| pxUDPPacket->xUDPHeader.usChecksum = 0u; | |
| } | |
| } | |
| #endif | |
| } | |
| else if( eReturned == eARPCacheMiss ) | |
| { | |
| /* Add an entry to the ARP table with a null hardware address. | |
| This allows the ARP timer to know that an ARP reply is | |
| outstanding, and perform retransmissions if necessary. */ | |
| vARPRefreshCacheEntry( NULL, ulIPAddress ); | |
| /* Generate an ARP for the required IP address. */ | |
| iptracePACKET_DROPPED_TO_GENERATE_ARP( pxNetworkBuffer->ulIPAddress ); | |
| pxNetworkBuffer->ulIPAddress = ulIPAddress; | |
| vARPGenerateRequestPacket( pxNetworkBuffer ); | |
| } | |
| else | |
| { | |
| /* The lookup indicated that an ARP request has already been | |
| sent out for the queried IP address. */ | |
| eReturned = eCantSendPacket; | |
| } | |
| } | |
| if( eReturned != eCantSendPacket ) | |
| { | |
| /* The network driver is responsible for freeing the network buffer | |
| after the packet has been sent. */ | |
| #if defined( ipconfigETHERNET_MINIMUM_PACKET_BYTES ) | |
| { | |
| if( pxNetworkBuffer->xDataLength < ( size_t ) ipconfigETHERNET_MINIMUM_PACKET_BYTES ) | |
| { | |
| BaseType_t xIndex; | |
| for( xIndex = ( BaseType_t ) pxNetworkBuffer->xDataLength; xIndex < ( BaseType_t ) ipconfigETHERNET_MINIMUM_PACKET_BYTES; xIndex++ ) | |
| { | |
| pxNetworkBuffer->pucEthernetBuffer[ xIndex ] = 0u; | |
| } | |
| pxNetworkBuffer->xDataLength = ( size_t ) ipconfigETHERNET_MINIMUM_PACKET_BYTES; | |
| } | |
| } | |
| #endif | |
| xNetworkInterfaceOutput( pxNetworkBuffer, pdTRUE ); | |
| } | |
| else | |
| { | |
| /* The packet can't be sent (DHCP not completed?). Just drop the | |
| packet. */ | |
| vReleaseNetworkBufferAndDescriptor( pxNetworkBuffer ); | |
| } | |
| } | |
| /*-----------------------------------------------------------*/ | |
| BaseType_t xProcessReceivedUDPPacket( NetworkBufferDescriptor_t *pxNetworkBuffer, uint16_t usPort ) | |
| { | |
| BaseType_t xReturn = pdPASS; | |
| FreeRTOS_Socket_t *pxSocket; | |
| UDPPacket_t *pxUDPPacket = (UDPPacket_t *) pxNetworkBuffer->pucEthernetBuffer; | |
| /* Caller must check for minimum packet size. */ | |
| pxSocket = pxUDPSocketLookup( usPort ); | |
| if( pxSocket ) | |
| { | |
| /* When refreshing the ARP cache with received UDP packets we must be | |
| careful; hundreds of broadcast messages may pass and if we're not | |
| handling them, no use to fill the ARP cache with those IP addresses. */ | |
| vARPRefreshCacheEntry( &( pxUDPPacket->xEthernetHeader.xSourceAddress ), pxUDPPacket->xIPHeader.ulSourceIPAddress ); | |
| #if( ipconfigUSE_CALLBACKS == 1 ) | |
| { | |
| /* Did the owner of this socket register a reception handler ? */ | |
| if( ipconfigIS_VALID_PROG_ADDRESS( pxSocket->u.xUDP.pxHandleReceive ) ) | |
| { | |
| struct freertos_sockaddr xSourceAddress, destinationAddress; | |
| void *pcData = ( void * ) &( pxNetworkBuffer->pucEthernetBuffer[ ipUDP_PAYLOAD_OFFSET_IPv4 ] ); | |
| FOnUDPReceive_t xHandler = ( FOnUDPReceive_t ) pxSocket->u.xUDP.pxHandleReceive; | |
| xSourceAddress.sin_port = pxNetworkBuffer->usPort; | |
| xSourceAddress.sin_addr = pxNetworkBuffer->ulIPAddress; | |
| destinationAddress.sin_port = usPort; | |
| destinationAddress.sin_addr = pxUDPPacket->xIPHeader.ulDestinationIPAddress; | |
| if( xHandler( ( Socket_t * ) pxSocket, ( void* ) pcData, ( size_t ) pxNetworkBuffer->xDataLength, | |
| &xSourceAddress, &destinationAddress ) ) | |
| { | |
| xReturn = pdFAIL; /* FAIL means that we did not consume or release the buffer */ | |
| } | |
| } | |
| } | |
| #endif /* ipconfigUSE_CALLBACKS */ | |
| #if( ipconfigUDP_MAX_RX_PACKETS > 0 ) | |
| { | |
| if( xReturn == pdPASS ) | |
| { | |
| if ( listCURRENT_LIST_LENGTH( &( pxSocket->u.xUDP.xWaitingPacketsList ) ) >= pxSocket->u.xUDP.uxMaxPackets ) | |
| { | |
| FreeRTOS_debug_printf( ( "xProcessReceivedUDPPacket: buffer full %ld >= %ld port %u\n", | |
| listCURRENT_LIST_LENGTH( &( pxSocket->u.xUDP.xWaitingPacketsList ) ), | |
| pxSocket->u.xUDP.uxMaxPackets, pxSocket->usLocalPort ) ); | |
| xReturn = pdFAIL; /* we did not consume or release the buffer */ | |
| } | |
| } | |
| } | |
| #endif | |
| if( xReturn == pdPASS ) | |
| { | |
| vTaskSuspendAll(); | |
| { | |
| if( xReturn == pdPASS ) | |
| { | |
| taskENTER_CRITICAL(); | |
| { | |
| /* Add the network packet to the list of packets to be | |
| processed by the socket. */ | |
| vListInsertEnd( &( pxSocket->u.xUDP.xWaitingPacketsList ), &( pxNetworkBuffer->xBufferListItem ) ); | |
| } | |
| taskEXIT_CRITICAL(); | |
| } | |
| } | |
| xTaskResumeAll(); | |
| /* Set the socket's receive event */ | |
| if( pxSocket->xEventGroup != NULL ) | |
| { | |
| xEventGroupSetBits( pxSocket->xEventGroup, eSOCKET_RECEIVE ); | |
| } | |
| #if( ipconfigSUPPORT_SELECT_FUNCTION == 1 ) | |
| { | |
| if( ( pxSocket->pxSocketSet != NULL ) && ( ( pxSocket->xSelectBits & eSELECT_READ ) != 0 ) ) | |
| { | |
| xEventGroupSetBits( pxSocket->pxSocketSet->xSelectGroup, eSELECT_READ ); | |
| } | |
| } | |
| #endif | |
| #if( ipconfigSOCKET_HAS_USER_SEMAPHORE == 1 ) | |
| { | |
| if( pxSocket->pxUserSemaphore != NULL ) | |
| { | |
| xSemaphoreGive( pxSocket->pxUserSemaphore ); | |
| } | |
| } | |
| #endif | |
| #if( ipconfigUSE_DHCP == 1 ) | |
| { | |
| if( xIsDHCPSocket( pxSocket ) ) | |
| { | |
| xSendEventToIPTask( eDHCPEvent ); | |
| } | |
| } | |
| #endif | |
| } | |
| } | |
| else | |
| { | |
| /* There is no socket listening to the target port, but still it might | |
| be for this node. */ | |
| #if( ipconfigUSE_LLMNR == 1 ) | |
| /* a LLMNR request, check for the destination port. */ | |
| if( ( usPort == FreeRTOS_ntohs( ipLLMNR_PORT ) ) || | |
| ( pxUDPPacket->xUDPHeader.usSourcePort == FreeRTOS_ntohs( ipLLMNR_PORT ) ) ) | |
| { | |
| vARPRefreshCacheEntry( &( pxUDPPacket->xEthernetHeader.xSourceAddress ), pxUDPPacket->xIPHeader.ulSourceIPAddress ); | |
| xReturn = ( BaseType_t )ulDNSHandlePacket( pxNetworkBuffer ); | |
| } | |
| else | |
| #endif /* ipconfigUSE_LLMNR */ | |
| #if( ipconfigUSE_NBNS == 1 ) | |
| /* a NetBIOS request, check for the destination port */ | |
| if( ( usPort == FreeRTOS_ntohs( ipNBNS_PORT ) ) || | |
| ( pxUDPPacket->xUDPHeader.usSourcePort == FreeRTOS_ntohs( ipNBNS_PORT ) ) ) | |
| { | |
| vARPRefreshCacheEntry( &( pxUDPPacket->xEthernetHeader.xSourceAddress ), pxUDPPacket->xIPHeader.ulSourceIPAddress ); | |
| xReturn = ( BaseType_t )ulNBNSHandlePacket( pxNetworkBuffer ); | |
| } | |
| else | |
| #endif /* ipconfigUSE_NBNS */ | |
| { | |
| xReturn = pdFAIL; | |
| } | |
| } | |
| return xReturn; | |
| } | |
| /*-----------------------------------------------------------*/ |