git.sur5r.net Git - freertos/blob - FreeRTOS-Plus/Demo/FreeRTOS_Plus_UDP_and_CLI_Windows_Simulator/DemoTasks/SelectServer.c

   1 /*\r
   2     FreeRTOS V7.5.2 - Copyright (C) 2013 Real Time Engineers Ltd.\r
   3 \r
   4     VISIT http://www.FreeRTOS.org TO ENSURE YOU ARE USING THE LATEST VERSION.\r
   5 \r
   6     ***************************************************************************\r
   7      *                                                                       *\r
   8      *    FreeRTOS provides completely free yet professionally developed,    *\r
   9      *    robust, strictly quality controlled, supported, and cross          *\r
  10      *    platform software that has become a de facto standard.             *\r
  11      *                                                                       *\r
  12      *    Help yourself get started quickly and support the FreeRTOS         *\r
  13      *    project by purchasing a FreeRTOS tutorial book, reference          *\r
  14      *    manual, or both from: http://www.FreeRTOS.org/Documentation        *\r
  15      *                                                                       *\r
  16      *    Thank you!                                                         *\r
  17      *                                                                       *\r
  18     ***************************************************************************\r
  19 \r
  20     This file is part of the FreeRTOS distribution.\r
  21 \r
  22     FreeRTOS is free software; you can redistribute it and/or modify it under\r
  23     the terms of the GNU General Public License (version 2) as published by the\r
  24     Free Software Foundation >>!AND MODIFIED BY!<< the FreeRTOS exception.\r
  25 \r
  26     >>! NOTE: The modification to the GPL is included to allow you to distribute\r
  27     >>! a combined work that includes FreeRTOS without being obliged to provide\r
  28     >>! the source code for proprietary components outside of the FreeRTOS\r
  29     >>! kernel.\r
  30 \r
  31     FreeRTOS is distributed in the hope that it will be useful, but WITHOUT ANY\r
  32     WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS\r
  33     FOR A PARTICULAR PURPOSE.  Full license text is available from the following\r
  34     link: http://www.freertos.org/a00114.html\r
  35 \r
  36     1 tab == 4 spaces!\r
  37 \r
  38     ***************************************************************************\r
  39      *                                                                       *\r
  40      *    Having a problem?  Start by reading the FAQ "My application does   *\r
  41      *    not run, what could be wrong?"                                     *\r
  42      *                                                                       *\r
  43      *    http://www.FreeRTOS.org/FAQHelp.html                               *\r
  44      *                                                                       *\r
  45     ***************************************************************************\r
  46 \r
  47     http://www.FreeRTOS.org - Documentation, books, training, latest versions,\r
  48     license and Real Time Engineers Ltd. contact details.\r
  49 \r
  50     http://www.FreeRTOS.org/plus - A selection of FreeRTOS ecosystem products,\r
  51     including FreeRTOS+Trace - an indispensable productivity tool, a DOS\r
  52     compatible FAT file system, and our tiny thread aware UDP/IP stack.\r
  53 \r
  54     http://www.OpenRTOS.com - Real Time Engineers ltd license FreeRTOS to High\r
  55     Integrity Systems to sell under the OpenRTOS brand.  Low cost OpenRTOS\r
  56     licenses offer ticketed support, indemnification and middleware.\r
  57 \r
  58     http://www.SafeRTOS.com - High Integrity Systems also provide a safety\r
  59     engineered and independently SIL3 certified version for use in safety and\r
  60     mission critical applications that require provable dependability.\r
  61 \r
  62     1 tab == 4 spaces!\r
  63 */\r
  64 \r
  65 /*\r
  66  * A number of sockets are created and added to a set. One task then blocks on\r
  67  * the set while the other task sends data to a (pseudo) random member of the\r
  68  * set.\r
  69  */\r
  70 \r
  71 /* Standard includes. */\r
  72 #include <stdint.h>\r
  73 #include <stdio.h>\r
  74 \r
  75 /* FreeRTOS includes. */\r
  76 #include "FreeRTOS.h"\r
  77 #include "task.h"\r
  78 \r
  79 /* FreeRTOS+UDP includes. */\r
  80 #include "FreeRTOS_UDP_IP.h"\r
  81 #include "FreeRTOS_Sockets.h"\r
  82 \r
  83 #define selTINY_DELAY                           ( ( portTickType ) 2 )\r
  84 #define selNUMBER_OF_SOCKETS            ( 3 )\r
  85 #define selSELECT_QUEUE_SIZE            ( selNUMBER_OF_SOCKETS * 2 )\r
  86 \r
  87 /*\r
  88  * Sends data to multiple different sockets.\r
  89  */\r
  90 static void prvMultipleSocketTxTask( void *pvParameters );\r
  91 \r
  92 /*\r
  93  * Uses the FreeRTOS_select() API function to receive from multiple sockets.\r
  94  */\r
  95 static void prvMultipleSocketRxTask( void *pvParameters );\r
  96 \r
  97 /*-----------------------------------------------------------*/\r
  98 \r
  99 static xSocket_t xRxSockets[ selNUMBER_OF_SOCKETS ] = { 0 };\r
 100 \r
 101 /*-----------------------------------------------------------*/\r
 102 \r
 103 void vStartSelectUDPServerTasks( uint16_t usStackSize, uint32_t ulFirstPortNumber, unsigned portBASE_TYPE uxPriority )\r
 104 {\r
 105         /* Create task that sends to multiple sockets, and the task that uses the\r
 106         FreeRTOS_select() function to receive from multiple sockets.  The first\r
 107         port number to use is passed into both tasks using the task's parameter.\r
 108         Other port numbers are consecutive from the first. */\r
 109         xTaskCreate( prvMultipleSocketTxTask, ( const signed char * const ) "MultiTx", usStackSize, ( void * ) ulFirstPortNumber, uxPriority, NULL );\r
 110         xTaskCreate( prvMultipleSocketRxTask, ( const signed char * const ) "MultiRx", usStackSize, ( void * ) ulFirstPortNumber, uxPriority, NULL );\r
 111 }\r
 112 /*-----------------------------------------------------------*/\r
 113 \r
 114 static void prvMultipleSocketRxTask( void *pvParameters )\r
 115 {\r
 116 xSocketSet_t xFD_Set;\r
 117 xSocket_t xSocket;\r
 118 struct freertos_sockaddr xAddress;\r
 119 uint32_t xClientLength = sizeof( struct freertos_sockaddr ), ulFirstRxPortNumber, x;\r
 120 uint32_t ulReceivedValue = 0, ulExpectedValue = 0UL, ulReceivedCount[ selNUMBER_OF_SOCKETS ] = { 0 };\r
 121 int32_t lBytes;\r
 122 const portTickType xRxBlockTime = 0;\r
 123 \r
 124         /* The number of the port the first Rx socket will be bound to is passed in\r
 125         as the task parameter.  Other port numbers used are consecutive from this. */\r
 126         ulFirstRxPortNumber = ( uint32_t ) pvParameters;\r
 127 \r
 128         /* Create the set of sockets that will be passed into FreeRTOS_select(). */\r
 129         xFD_Set = FreeRTOS_CreateSocketSet( selSELECT_QUEUE_SIZE );\r
 130 \r
 131         for( x = 0; x < selNUMBER_OF_SOCKETS; x++ )\r
 132         {\r
 133                 /* Create the next Rx socket. */\r
 134                 xRxSockets[ x ] = FreeRTOS_socket( FREERTOS_AF_INET, FREERTOS_SOCK_DGRAM, FREERTOS_IPPROTO_UDP );\r
 135                 configASSERT( xRxSockets[ x ] );\r
 136 \r
 137                 /* Bind to the next port number. */\r
 138                 xAddress.sin_port = FreeRTOS_htons( ( uint16_t ) ( ulFirstRxPortNumber + x ) );\r
 139                 FreeRTOS_bind( xRxSockets[ x ], &xAddress, sizeof( struct freertos_sockaddr ) );\r
 140 \r
 141                 /* There should always be data available on the socket returned from\r
 142                 FreeRTOS_select() so blocking on a read should not be necessary. */\r
 143                 FreeRTOS_setsockopt( xRxSockets[ x ], 0, FREERTOS_SO_RCVTIMEO, &xRxBlockTime, sizeof( xRxBlockTime ) );\r
 144 \r
 145                 /* Add the created socket to the set. */\r
 146                 FreeRTOS_FD_SET( xRxSockets[ x ], xFD_Set );\r
 147         }\r
 148 \r
 149         for( ;; )\r
 150         {\r
 151                 /* Wait for a socket from the set to become available for reading. */\r
 152                 xSocket = FreeRTOS_select( xFD_Set, portMAX_DELAY );\r
 153 \r
 154                 /* xSocket should never be NULL because FreeRTOS_select() was called\r
 155                 with an indefinite delay (assuming INCLUDE_vTaskSuspend is set to 1). */\r
 156                 configASSERT( xSocket );\r
 157 \r
 158                 lBytes = FreeRTOS_recvfrom( xSocket, &( ulReceivedValue ), sizeof( uint32_t ), 0, &xAddress, &xClientLength );\r
 159 \r
 160                 /* It is possible that the first value received will not be zero\r
 161                 because the first few transmitted packets may have been dropped to\r
 162                 send an ARP and then wait the ARP reply. */\r
 163                 if( ulExpectedValue == 0 )\r
 164                 {\r
 165                         if( ulExpectedValue != ulReceivedValue )\r
 166                         {\r
 167                                 /* Correct for packets lost to ARP traffic. */\r
 168                                 ulExpectedValue = ulReceivedValue;\r
 169                         }\r
 170                 }\r
 171 \r
 172                 /* Data should always be available even though the block time was set\r
 173                 to zero because the socket was returned from FreeRTOS_select(). */\r
 174                 configASSERT( lBytes == 4 );\r
 175                 configASSERT( ulReceivedValue == ulExpectedValue );\r
 176 \r
 177                 ulExpectedValue++;\r
 178 \r
 179                 /* Keep a record of the number of times each socket has been used so it\r
 180                 can be verified (using the debugger) that they all get used. */\r
 181                 for( x= 0; x < selNUMBER_OF_SOCKETS; x++ )\r
 182                 {\r
 183                         if( xSocket == xRxSockets[ x ] )\r
 184                         {\r
 185                                 ( ulReceivedCount[ x ] )++;\r
 186                                 break;\r
 187                         }\r
 188                 }\r
 189         }\r
 190 }\r
 191 /*-----------------------------------------------------------*/\r
 192 \r
 193 static void prvMultipleSocketTxTask( void *pvParameters )\r
 194 {\r
 195 uint32_t ulTxValue = 0;\r
 196 struct freertos_sockaddr xDestinationAddress;\r
 197 uint32_t ulIPAddress, ulFirstDestinationPortNumber, xPortNumber;\r
 198 xSocket_t xTxSocket;\r
 199 const portTickType xShortDelay = 100 / portTICK_RATE_MS, xSendBlockTime = 500 / portTICK_RATE_MS;\r
 200 \r
 201         srand( ( unsigned int ) &xPortNumber );\r
 202 \r
 203         /* The first destination port number is passed in as the task parameter.\r
 204         Other destination port numbers used are consecutive from this. */\r
 205         ulFirstDestinationPortNumber = ( uint32_t ) pvParameters;\r
 206 \r
 207         /* Create the socket used to send to the sockets created by the Rx task.\r
 208         Let the IP stack select a port to bind to. */\r
 209         xTxSocket = FreeRTOS_socket( FREERTOS_AF_INET, FREERTOS_SOCK_DGRAM, FREERTOS_IPPROTO_UDP );\r
 210         FreeRTOS_bind( xTxSocket, NULL, sizeof( struct freertos_sockaddr ) );\r
 211 \r
 212         /* The Rx and Tx tasks execute at the same priority so it is possible that\r
 213         the Tx task will fill up the send queue - set a Tx block time to ensure\r
 214         flow control is managed if this happens. */\r
 215         FreeRTOS_setsockopt( xTxSocket, 0, FREERTOS_SO_SNDTIMEO, &xSendBlockTime, sizeof( xSendBlockTime ) );\r
 216 \r
 217         /* It is assumed that this task is not created until the network is up,\r
 218         so the IP address can be obtained immediately.  Store the IP address being\r
 219         used in ulIPAddress.  This is done so the socket can send to a different\r
 220         port on the same IP address. */\r
 221         FreeRTOS_GetAddressConfiguration( &ulIPAddress, NULL, NULL, NULL );\r
 222 \r
 223         /* This test sends to itself, so data sent from here is received by a server\r
 224         socket on the same IP address.  Setup the freertos_sockaddr structure with\r
 225         this nodes IP address. */\r
 226         xDestinationAddress.sin_addr = ulIPAddress;\r
 227 \r
 228         /* Block for a short time to ensure the task implemented by the\r
 229         prvMultipleSocketRxTask() function has finished creating the Rx sockets. */\r
 230         vTaskDelay( xShortDelay );\r
 231 \r
 232         for( ;; )\r
 233         {\r
 234                 /* Pseudo randomly select the destination port number from the range of\r
 235                 valid destination port numbers. */\r
 236                 xPortNumber = rand() % selNUMBER_OF_SOCKETS;\r
 237                 xDestinationAddress.sin_port = ( uint16_t ) ( ulFirstDestinationPortNumber + xPortNumber );\r
 238                 xDestinationAddress.sin_port = FreeRTOS_htons( xDestinationAddress.sin_port );\r
 239 \r
 240                 /* Send an incrementing value. */\r
 241                 FreeRTOS_sendto( xTxSocket, &ulTxValue, sizeof( ulTxValue ), 0, &xDestinationAddress, sizeof( xDestinationAddress ) );\r
 242                 ulTxValue++;\r
 243 \r
 244                 /* Delay here because in the Windows simulator the MAC interrupt\r
 245                 simulator delays, so network trafic cannot be received any faster\r
 246                 than this. */\r
 247                 vTaskDelay( configWINDOWS_MAC_INTERRUPT_SIMULATOR_DELAY );\r
 248         }\r
 249 }\r
 250 \r
 251 \r
 252 \r