git.sur5r.net Git - freertos/blob - FreeRTOS-Plus/Demo/FreeRTOS_Plus_CyaSSL_Windows_Simulator/SecureTCPClientTask.c

   1 /*\r
   2     FreeRTOS V7.4.2 - Copyright (C) 2013 Real Time Engineers Ltd.\r
   3 \r
   4     FEATURES AND PORTS ARE ADDED TO FREERTOS ALL THE TIME.  PLEASE VISIT\r
   5     http://www.FreeRTOS.org TO ENSURE YOU ARE USING THE LATEST VERSION.\r
   6 \r
   7     ***************************************************************************\r
   8      *                                                                       *\r
   9      *    FreeRTOS tutorial books are available in pdf and paperback.        *\r
  10      *    Complete, revised, and edited pdf reference manuals are also       *\r
  11      *    available.                                                         *\r
  12      *                                                                       *\r
  13      *    Purchasing FreeRTOS documentation will not only help you, by       *\r
  14      *    ensuring you get running as quickly as possible and with an        *\r
  15      *    in-depth knowledge of how to use FreeRTOS, it will also help       *\r
  16      *    the FreeRTOS project to continue with its mission of providing     *\r
  17      *    professional grade, cross platform, de facto standard solutions    *\r
  18      *    for microcontrollers - completely free of charge!                  *\r
  19      *                                                                       *\r
  20      *    >>> See http://www.FreeRTOS.org/Documentation for details. <<<     *\r
  21      *                                                                       *\r
  22      *    Thank you for using FreeRTOS, and thank you for your support!      *\r
  23      *                                                                       *\r
  24     ***************************************************************************\r
  25 \r
  26 \r
  27     This file is part of the FreeRTOS distribution.\r
  28 \r
  29     FreeRTOS is free software; you can redistribute it and/or modify it under\r
  30     the terms of the GNU General Public License (version 2) as published by the\r
  31     Free Software Foundation AND MODIFIED BY the FreeRTOS exception.\r
  32 \r
  33     >>>>>>NOTE<<<<<< The modification to the GPL is included to allow you to\r
  34     distribute a combined work that includes FreeRTOS without being obliged to\r
  35     provide the source code for proprietary components outside of the FreeRTOS\r
  36     kernel.\r
  37 \r
  38     FreeRTOS is distributed in the hope that it will be useful, but WITHOUT ANY\r
  39     WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS\r
  40     FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more\r
  41     details. You should have received a copy of the GNU General Public License\r
  42     and the FreeRTOS license exception along with FreeRTOS; if not it can be\r
  43     viewed here: http://www.freertos.org/a00114.html and also obtained by\r
  44     writing to Real Time Engineers Ltd., contact details for whom are available\r
  45     on the FreeRTOS WEB site.\r
  46 \r
  47     1 tab == 4 spaces!\r
  48 \r
  49     ***************************************************************************\r
  50      *                                                                       *\r
  51      *    Having a problem?  Start by reading the FAQ "My application does   *\r
  52      *    not run, what could be wrong?"                                     *\r
  53      *                                                                       *\r
  54      *    http://www.FreeRTOS.org/FAQHelp.html                               *\r
  55      *                                                                       *\r
  56     ***************************************************************************\r
  57 \r
  58 \r
  59     http://www.FreeRTOS.org - Documentation, books, training, latest versions, \r
  60     license and Real Time Engineers Ltd. contact details.\r
  61 \r
  62     http://www.FreeRTOS.org/plus - A selection of FreeRTOS ecosystem products,\r
  63     including FreeRTOS+Trace - an indispensable productivity tool, and our new\r
  64     fully thread aware and reentrant UDP/IP stack.\r
  65 \r
  66     http://www.OpenRTOS.com - Real Time Engineers ltd license FreeRTOS to High \r
  67     Integrity Systems, who sell the code with commercial support, \r
  68     indemnification and middleware, under the OpenRTOS brand.\r
  69     \r
  70     http://www.SafeRTOS.com - High Integrity Systems also provide a safety \r
  71     engineered and independently SIL3 certified version for use in safety and \r
  72     mission critical applications that require provable dependability.\r
  73 */\r
  74 \r
  75 #pragma comment( lib, "ws2_32.lib" )\r
  76 \r
  77 /* Win32 includes. */\r
  78 #include <WinSock2.h>\r
  79 \r
  80 /* CyaSSL includes. */\r
  81 #include "cyassl/ssl.h"\r
  82 \r
  83 /* Standard includes. */\r
  84 #include <stdint.h>\r
  85 #include <stdio.h>\r
  86 \r
  87 /* FreeRTOS includes. */\r
  88 #include "FreeRTOS.h"\r
  89 #include "task.h"\r
  90 \r
  91 /*-----------------------------------------------------------*/\r
  92 \r
  93 /* The CyaSSL context for the client. */\r
  94 static CYASSL_CTX* xCyaSSL_ClientContext = NULL;\r
  95 \r
  96 /*-----------------------------------------------------------*/\r
  97 \r
  98 /* See the comments at the top of main.c. */\r
  99 void vSecureTCPClientTask( void *pvParameters )\r
 100 {\r
 101 SOCKET xClientSocket;\r
 102 struct sockaddr_in xConnection;\r
 103 CYASSL* xCyaSSL_Object;\r
 104 WORD wVersionRequested;\r
 105 WSADATA xWSAData;\r
 106 uint8_t cString[ 50 ];\r
 107 portBASE_TYPE lReturned;\r
 108 uint32_t ulCount = 0UL;\r
 109 \r
 110         /* Remove compiler warning about unused parameters. */\r
 111         ( void ) pvParameters;\r
 112 \r
 113         /* Prepare to use WinSock. */\r
 114         wVersionRequested = MAKEWORD( 2, 2 );\r
 115         configASSERT( WSAStartup( wVersionRequested, &xWSAData ) == 0 );\r
 116 \r
 117         /* Set family and port for client socket. */\r
 118         memset( ( void * ) &xConnection, 0x00, sizeof( struct sockaddr_in ) );\r
 119         xConnection.sin_family = AF_INET;\r
 120         xConnection.sin_addr.s_addr = inet_addr("127.0.0.1");\r
 121         xConnection.sin_port = htons( configTCP_PORT_NUMBER );\r
 122 \r
 123     /* Attempt to create a context that uses the TLS V1 server protocol. */\r
 124     xCyaSSL_ClientContext = CyaSSL_CTX_new( CyaTLSv1_client_method() );\r
 125         configASSERT( xCyaSSL_ClientContext );\r
 126 \r
 127     /* Load the CA certificate. */\r
 128     lReturned = CyaSSL_CTX_load_verify_locations( xCyaSSL_ClientContext, "ca-cert.pem", 0 );\r
 129         configASSERT( lReturned == SSL_SUCCESS );\r
 130 \r
 131         for( ;; )\r
 132         {\r
 133                 /* Create the socket. */\r
 134                 xClientSocket = socket( AF_INET, SOCK_STREAM, 0 );\r
 135                 configASSERT( xClientSocket != INVALID_SOCKET );\r
 136 \r
 137                 /* Connect to the secure server. */\r
 138                 if( connect( xClientSocket, ( SOCKADDR * ) &xConnection, sizeof( xConnection ) ) == 0 )\r
 139                 {\r
 140                         /* The connect was successful.  Create a CyaSSL object to associate \r
 141                         with this connection. */\r
 142                         xCyaSSL_Object = CyaSSL_new( xCyaSSL_ClientContext );\r
 143         \r
 144                         if( xCyaSSL_Object != NULL )\r
 145                         {\r
 146                                 /* Associate the created CyaSSL object with the connected \r
 147                                 socket. */\r
 148                                 lReturned = CyaSSL_set_fd( xCyaSSL_Object, xClientSocket );\r
 149                                 configASSERT( lReturned == SSL_SUCCESS );\r
 150 \r
 151                                 /* The count is used to differentiate between messages sent to\r
 152                                 the server, and to break out of the do while loop below. */\r
 153                                 ulCount = 0UL;\r
 154 \r
 155                                 do\r
 156                                 {\r
 157                                         /* Create the string that is sent to the secure server. */\r
 158                                         sprintf( ( char * ) cString, "Message number %lu\r\n", ulCount );\r
 159 \r
 160                                         /* The next line is the secure equivalent of the standard \r
 161                                         sockets call:\r
 162                                         lReturned = send( xClientSocket, cString, strlen( cString ) + 1, 0 ); */\r
 163                                         lReturned = CyaSSL_write( xCyaSSL_Object, ( const char * ) cString, strlen( ( const char * ) cString ) + 1 );\r
 164                                         \r
 165                                         \r
 166                                         /* Short delay to prevent the messages streaming up the\r
 167                                         console too quickly. */\r
 168                                         vTaskDelay( 5 );\r
 169                                         ulCount++;\r
 170 \r
 171                                 } while( ( lReturned != SOCKET_ERROR ) && ( ulCount < 10UL ) );\r
 172                         }\r
 173                                                 \r
 174                         CyaSSL_free( xCyaSSL_Object );\r
 175                         closesocket( xClientSocket );\r
 176 \r
 177                         /* Delay for a short time before starting over. */\r
 178                         vTaskDelay( 50 );\r
 179                 }\r
 180         }\r
 181 }\r
 182 /*-----------------------------------------------------------*/\r
 183 \r