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

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