Mark Radbourne / mbedtls

mbed TLS Build

Dependents:   Encrypt_Decrypt1 mbed_blink_tls encrypt encrypt

library/net_sockets.c@0:cdf462088d13, 2017-01-05 (annotated)

Committer:
markrad
Date:
Thu Jan 05 00:18:44 2017 +0000
Revision:
0:cdf462088d13
Initial commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
markrad 0:cdf462088d13 1 /*
markrad 0:cdf462088d13 2 * TCP/IP or UDP/IP networking functions
markrad 0:cdf462088d13 3 *
markrad 0:cdf462088d13 4 * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
markrad 0:cdf462088d13 5 * SPDX-License-Identifier: Apache-2.0
markrad 0:cdf462088d13 6 *
markrad 0:cdf462088d13 7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
markrad 0:cdf462088d13 8 * not use this file except in compliance with the License.
markrad 0:cdf462088d13 9 * You may obtain a copy of the License at
markrad 0:cdf462088d13 10 *
markrad 0:cdf462088d13 11 * http://www.apache.org/licenses/LICENSE-2.0
markrad 0:cdf462088d13 12 *
markrad 0:cdf462088d13 13 * Unless required by applicable law or agreed to in writing, software
markrad 0:cdf462088d13 14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
markrad 0:cdf462088d13 15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
markrad 0:cdf462088d13 16 * See the License for the specific language governing permissions and
markrad 0:cdf462088d13 17 * limitations under the License.
markrad 0:cdf462088d13 18 *
markrad 0:cdf462088d13 19 * This file is part of mbed TLS (https://tls.mbed.org)
markrad 0:cdf462088d13 20 */
markrad 0:cdf462088d13 21
markrad 0:cdf462088d13 22 #if !defined(MBEDTLS_CONFIG_FILE)
markrad 0:cdf462088d13 23 #include "mbedtls/config.h"
markrad 0:cdf462088d13 24 #else
markrad 0:cdf462088d13 25 #include MBEDTLS_CONFIG_FILE
markrad 0:cdf462088d13 26 #endif
markrad 0:cdf462088d13 27
markrad 0:cdf462088d13 28 #if defined(MBEDTLS_NET_C)
markrad 0:cdf462088d13 29
markrad 0:cdf462088d13 30 #if !defined(unix) && !defined(__unix__) && !defined(__unix) && \
markrad 0:cdf462088d13 31 !defined(__APPLE__) && !defined(_WIN32)
markrad 0:cdf462088d13 32 #error "This module only works on Unix and Windows, see MBEDTLS_NET_C in config.h"
markrad 0:cdf462088d13 33 #endif
markrad 0:cdf462088d13 34
markrad 0:cdf462088d13 35 #if defined(MBEDTLS_PLATFORM_C)
markrad 0:cdf462088d13 36 #include "mbedtls/platform.h"
markrad 0:cdf462088d13 37 #else
markrad 0:cdf462088d13 38 #include <stdlib.h>
markrad 0:cdf462088d13 39 #endif
markrad 0:cdf462088d13 40
markrad 0:cdf462088d13 41 #include "mbedtls/net_sockets.h"
markrad 0:cdf462088d13 42
markrad 0:cdf462088d13 43 #include <string.h>
markrad 0:cdf462088d13 44
markrad 0:cdf462088d13 45 #if (defined(_WIN32) || defined(_WIN32_WCE)) && !defined(EFIX64) && \
markrad 0:cdf462088d13 46 !defined(EFI32)
markrad 0:cdf462088d13 47
markrad 0:cdf462088d13 48 #ifdef _WIN32_WINNT
markrad 0:cdf462088d13 49 #undef _WIN32_WINNT
markrad 0:cdf462088d13 50 #endif
markrad 0:cdf462088d13 51 /* Enables getaddrinfo() & Co */
markrad 0:cdf462088d13 52 #define _WIN32_WINNT 0x0501
markrad 0:cdf462088d13 53 #include <ws2tcpip.h>
markrad 0:cdf462088d13 54
markrad 0:cdf462088d13 55 #include <winsock2.h>
markrad 0:cdf462088d13 56 #include <windows.h>
markrad 0:cdf462088d13 57
markrad 0:cdf462088d13 58 #if defined(_MSC_VER)
markrad 0:cdf462088d13 59 #if defined(_WIN32_WCE)
markrad 0:cdf462088d13 60 #pragma comment( lib, "ws2.lib" )
markrad 0:cdf462088d13 61 #else
markrad 0:cdf462088d13 62 #pragma comment( lib, "ws2_32.lib" )
markrad 0:cdf462088d13 63 #endif
markrad 0:cdf462088d13 64 #endif /* _MSC_VER */
markrad 0:cdf462088d13 65
markrad 0:cdf462088d13 66 #define read(fd,buf,len) recv(fd,(char*)buf,(int) len,0)
markrad 0:cdf462088d13 67 #define write(fd,buf,len) send(fd,(char*)buf,(int) len,0)
markrad 0:cdf462088d13 68 #define close(fd) closesocket(fd)
markrad 0:cdf462088d13 69
markrad 0:cdf462088d13 70 static int wsa_init_done = 0;
markrad 0:cdf462088d13 71
markrad 0:cdf462088d13 72 #else /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */
markrad 0:cdf462088d13 73
markrad 0:cdf462088d13 74 #include <sys/types.h>
markrad 0:cdf462088d13 75 #include <sys/socket.h>
markrad 0:cdf462088d13 76 #include <netinet/in.h>
markrad 0:cdf462088d13 77 #include <arpa/inet.h>
markrad 0:cdf462088d13 78 #include <sys/time.h>
markrad 0:cdf462088d13 79 #include <unistd.h>
markrad 0:cdf462088d13 80 #include <signal.h>
markrad 0:cdf462088d13 81 #include <fcntl.h>
markrad 0:cdf462088d13 82 #include <netdb.h>
markrad 0:cdf462088d13 83 #include <errno.h>
markrad 0:cdf462088d13 84
markrad 0:cdf462088d13 85 #endif /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */
markrad 0:cdf462088d13 86
markrad 0:cdf462088d13 87 /* Some MS functions want int and MSVC warns if we pass size_t,
markrad 0:cdf462088d13 88 * but the standard fucntions use socklen_t, so cast only for MSVC */
markrad 0:cdf462088d13 89 #if defined(_MSC_VER)
markrad 0:cdf462088d13 90 #define MSVC_INT_CAST (int)
markrad 0:cdf462088d13 91 #else
markrad 0:cdf462088d13 92 #define MSVC_INT_CAST
markrad 0:cdf462088d13 93 #endif
markrad 0:cdf462088d13 94
markrad 0:cdf462088d13 95 #include <stdio.h>
markrad 0:cdf462088d13 96
markrad 0:cdf462088d13 97 #include <time.h>
markrad 0:cdf462088d13 98
markrad 0:cdf462088d13 99 #include <stdint.h>
markrad 0:cdf462088d13 100
markrad 0:cdf462088d13 101 /*
markrad 0:cdf462088d13 102 * Prepare for using the sockets interface
markrad 0:cdf462088d13 103 */
markrad 0:cdf462088d13 104 static int net_prepare( void )
markrad 0:cdf462088d13 105 {
markrad 0:cdf462088d13 106 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 107 !defined(EFI32)
markrad 0:cdf462088d13 108 WSADATA wsaData;
markrad 0:cdf462088d13 109
markrad 0:cdf462088d13 110 if( wsa_init_done == 0 )
markrad 0:cdf462088d13 111 {
markrad 0:cdf462088d13 112 if( WSAStartup( MAKEWORD(2,0), &wsaData ) != 0 )
markrad 0:cdf462088d13 113 return( MBEDTLS_ERR_NET_SOCKET_FAILED );
markrad 0:cdf462088d13 114
markrad 0:cdf462088d13 115 wsa_init_done = 1;
markrad 0:cdf462088d13 116 }
markrad 0:cdf462088d13 117 #else
markrad 0:cdf462088d13 118 #if !defined(EFIX64) && !defined(EFI32)
markrad 0:cdf462088d13 119 signal( SIGPIPE, SIG_IGN );
markrad 0:cdf462088d13 120 #endif
markrad 0:cdf462088d13 121 #endif
markrad 0:cdf462088d13 122 return( 0 );
markrad 0:cdf462088d13 123 }
markrad 0:cdf462088d13 124
markrad 0:cdf462088d13 125 /*
markrad 0:cdf462088d13 126 * Initialize a context
markrad 0:cdf462088d13 127 */
markrad 0:cdf462088d13 128 void mbedtls_net_init( mbedtls_net_context *ctx )
markrad 0:cdf462088d13 129 {
markrad 0:cdf462088d13 130 ctx->fd = -1;
markrad 0:cdf462088d13 131 }
markrad 0:cdf462088d13 132
markrad 0:cdf462088d13 133 /*
markrad 0:cdf462088d13 134 * Initiate a TCP connection with host:port and the given protocol
markrad 0:cdf462088d13 135 */
markrad 0:cdf462088d13 136 int mbedtls_net_connect( mbedtls_net_context *ctx, const char *host, const char *port, int proto )
markrad 0:cdf462088d13 137 {
markrad 0:cdf462088d13 138 int ret;
markrad 0:cdf462088d13 139 struct addrinfo hints, *addr_list, *cur;
markrad 0:cdf462088d13 140
markrad 0:cdf462088d13 141 if( ( ret = net_prepare() ) != 0 )
markrad 0:cdf462088d13 142 return( ret );
markrad 0:cdf462088d13 143
markrad 0:cdf462088d13 144 /* Do name resolution with both IPv6 and IPv4 */
markrad 0:cdf462088d13 145 memset( &hints, 0, sizeof( hints ) );
markrad 0:cdf462088d13 146 hints.ai_family = AF_UNSPEC;
markrad 0:cdf462088d13 147 hints.ai_socktype = proto == MBEDTLS_NET_PROTO_UDP ? SOCK_DGRAM : SOCK_STREAM;
markrad 0:cdf462088d13 148 hints.ai_protocol = proto == MBEDTLS_NET_PROTO_UDP ? IPPROTO_UDP : IPPROTO_TCP;
markrad 0:cdf462088d13 149
markrad 0:cdf462088d13 150 if( getaddrinfo( host, port, &hints, &addr_list ) != 0 )
markrad 0:cdf462088d13 151 return( MBEDTLS_ERR_NET_UNKNOWN_HOST );
markrad 0:cdf462088d13 152
markrad 0:cdf462088d13 153 /* Try the sockaddrs until a connection succeeds */
markrad 0:cdf462088d13 154 ret = MBEDTLS_ERR_NET_UNKNOWN_HOST;
markrad 0:cdf462088d13 155 for( cur = addr_list; cur != NULL; cur = cur->ai_next )
markrad 0:cdf462088d13 156 {
markrad 0:cdf462088d13 157 ctx->fd = (int) socket( cur->ai_family, cur->ai_socktype,
markrad 0:cdf462088d13 158 cur->ai_protocol );
markrad 0:cdf462088d13 159 if( ctx->fd < 0 )
markrad 0:cdf462088d13 160 {
markrad 0:cdf462088d13 161 ret = MBEDTLS_ERR_NET_SOCKET_FAILED;
markrad 0:cdf462088d13 162 continue;
markrad 0:cdf462088d13 163 }
markrad 0:cdf462088d13 164
markrad 0:cdf462088d13 165 if( connect( ctx->fd, cur->ai_addr, MSVC_INT_CAST cur->ai_addrlen ) == 0 )
markrad 0:cdf462088d13 166 {
markrad 0:cdf462088d13 167 ret = 0;
markrad 0:cdf462088d13 168 break;
markrad 0:cdf462088d13 169 }
markrad 0:cdf462088d13 170
markrad 0:cdf462088d13 171 close( ctx->fd );
markrad 0:cdf462088d13 172 ret = MBEDTLS_ERR_NET_CONNECT_FAILED;
markrad 0:cdf462088d13 173 }
markrad 0:cdf462088d13 174
markrad 0:cdf462088d13 175 freeaddrinfo( addr_list );
markrad 0:cdf462088d13 176
markrad 0:cdf462088d13 177 return( ret );
markrad 0:cdf462088d13 178 }
markrad 0:cdf462088d13 179
markrad 0:cdf462088d13 180 /*
markrad 0:cdf462088d13 181 * Create a listening socket on bind_ip:port
markrad 0:cdf462088d13 182 */
markrad 0:cdf462088d13 183 int mbedtls_net_bind( mbedtls_net_context *ctx, const char *bind_ip, const char *port, int proto )
markrad 0:cdf462088d13 184 {
markrad 0:cdf462088d13 185 int n, ret;
markrad 0:cdf462088d13 186 struct addrinfo hints, *addr_list, *cur;
markrad 0:cdf462088d13 187
markrad 0:cdf462088d13 188 if( ( ret = net_prepare() ) != 0 )
markrad 0:cdf462088d13 189 return( ret );
markrad 0:cdf462088d13 190
markrad 0:cdf462088d13 191 /* Bind to IPv6 and/or IPv4, but only in the desired protocol */
markrad 0:cdf462088d13 192 memset( &hints, 0, sizeof( hints ) );
markrad 0:cdf462088d13 193 hints.ai_family = AF_UNSPEC;
markrad 0:cdf462088d13 194 hints.ai_socktype = proto == MBEDTLS_NET_PROTO_UDP ? SOCK_DGRAM : SOCK_STREAM;
markrad 0:cdf462088d13 195 hints.ai_protocol = proto == MBEDTLS_NET_PROTO_UDP ? IPPROTO_UDP : IPPROTO_TCP;
markrad 0:cdf462088d13 196 if( bind_ip == NULL )
markrad 0:cdf462088d13 197 hints.ai_flags = AI_PASSIVE;
markrad 0:cdf462088d13 198
markrad 0:cdf462088d13 199 if( getaddrinfo( bind_ip, port, &hints, &addr_list ) != 0 )
markrad 0:cdf462088d13 200 return( MBEDTLS_ERR_NET_UNKNOWN_HOST );
markrad 0:cdf462088d13 201
markrad 0:cdf462088d13 202 /* Try the sockaddrs until a binding succeeds */
markrad 0:cdf462088d13 203 ret = MBEDTLS_ERR_NET_UNKNOWN_HOST;
markrad 0:cdf462088d13 204 for( cur = addr_list; cur != NULL; cur = cur->ai_next )
markrad 0:cdf462088d13 205 {
markrad 0:cdf462088d13 206 ctx->fd = (int) socket( cur->ai_family, cur->ai_socktype,
markrad 0:cdf462088d13 207 cur->ai_protocol );
markrad 0:cdf462088d13 208 if( ctx->fd < 0 )
markrad 0:cdf462088d13 209 {
markrad 0:cdf462088d13 210 ret = MBEDTLS_ERR_NET_SOCKET_FAILED;
markrad 0:cdf462088d13 211 continue;
markrad 0:cdf462088d13 212 }
markrad 0:cdf462088d13 213
markrad 0:cdf462088d13 214 n = 1;
markrad 0:cdf462088d13 215 if( setsockopt( ctx->fd, SOL_SOCKET, SO_REUSEADDR,
markrad 0:cdf462088d13 216 (const char *) &n, sizeof( n ) ) != 0 )
markrad 0:cdf462088d13 217 {
markrad 0:cdf462088d13 218 close( ctx->fd );
markrad 0:cdf462088d13 219 ret = MBEDTLS_ERR_NET_SOCKET_FAILED;
markrad 0:cdf462088d13 220 continue;
markrad 0:cdf462088d13 221 }
markrad 0:cdf462088d13 222
markrad 0:cdf462088d13 223 if( bind( ctx->fd, cur->ai_addr, MSVC_INT_CAST cur->ai_addrlen ) != 0 )
markrad 0:cdf462088d13 224 {
markrad 0:cdf462088d13 225 close( ctx->fd );
markrad 0:cdf462088d13 226 ret = MBEDTLS_ERR_NET_BIND_FAILED;
markrad 0:cdf462088d13 227 continue;
markrad 0:cdf462088d13 228 }
markrad 0:cdf462088d13 229
markrad 0:cdf462088d13 230 /* Listen only makes sense for TCP */
markrad 0:cdf462088d13 231 if( proto == MBEDTLS_NET_PROTO_TCP )
markrad 0:cdf462088d13 232 {
markrad 0:cdf462088d13 233 if( listen( ctx->fd, MBEDTLS_NET_LISTEN_BACKLOG ) != 0 )
markrad 0:cdf462088d13 234 {
markrad 0:cdf462088d13 235 close( ctx->fd );
markrad 0:cdf462088d13 236 ret = MBEDTLS_ERR_NET_LISTEN_FAILED;
markrad 0:cdf462088d13 237 continue;
markrad 0:cdf462088d13 238 }
markrad 0:cdf462088d13 239 }
markrad 0:cdf462088d13 240
markrad 0:cdf462088d13 241 /* Bind was successful */
markrad 0:cdf462088d13 242 ret = 0;
markrad 0:cdf462088d13 243 break;
markrad 0:cdf462088d13 244 }
markrad 0:cdf462088d13 245
markrad 0:cdf462088d13 246 freeaddrinfo( addr_list );
markrad 0:cdf462088d13 247
markrad 0:cdf462088d13 248 return( ret );
markrad 0:cdf462088d13 249
markrad 0:cdf462088d13 250 }
markrad 0:cdf462088d13 251
markrad 0:cdf462088d13 252 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 253 !defined(EFI32)
markrad 0:cdf462088d13 254 /*
markrad 0:cdf462088d13 255 * Check if the requested operation would be blocking on a non-blocking socket
markrad 0:cdf462088d13 256 * and thus 'failed' with a negative return value.
markrad 0:cdf462088d13 257 */
markrad 0:cdf462088d13 258 static int net_would_block( const mbedtls_net_context *ctx )
markrad 0:cdf462088d13 259 {
markrad 0:cdf462088d13 260 ((void) ctx);
markrad 0:cdf462088d13 261 return( WSAGetLastError() == WSAEWOULDBLOCK );
markrad 0:cdf462088d13 262 }
markrad 0:cdf462088d13 263 #else
markrad 0:cdf462088d13 264 /*
markrad 0:cdf462088d13 265 * Check if the requested operation would be blocking on a non-blocking socket
markrad 0:cdf462088d13 266 * and thus 'failed' with a negative return value.
markrad 0:cdf462088d13 267 *
markrad 0:cdf462088d13 268 * Note: on a blocking socket this function always returns 0!
markrad 0:cdf462088d13 269 */
markrad 0:cdf462088d13 270 static int net_would_block( const mbedtls_net_context *ctx )
markrad 0:cdf462088d13 271 {
markrad 0:cdf462088d13 272 /*
markrad 0:cdf462088d13 273 * Never return 'WOULD BLOCK' on a non-blocking socket
markrad 0:cdf462088d13 274 */
markrad 0:cdf462088d13 275 if( ( fcntl( ctx->fd, F_GETFL ) & O_NONBLOCK ) != O_NONBLOCK )
markrad 0:cdf462088d13 276 return( 0 );
markrad 0:cdf462088d13 277
markrad 0:cdf462088d13 278 switch( errno )
markrad 0:cdf462088d13 279 {
markrad 0:cdf462088d13 280 #if defined EAGAIN
markrad 0:cdf462088d13 281 case EAGAIN:
markrad 0:cdf462088d13 282 #endif
markrad 0:cdf462088d13 283 #if defined EWOULDBLOCK && EWOULDBLOCK != EAGAIN
markrad 0:cdf462088d13 284 case EWOULDBLOCK:
markrad 0:cdf462088d13 285 #endif
markrad 0:cdf462088d13 286 return( 1 );
markrad 0:cdf462088d13 287 }
markrad 0:cdf462088d13 288 return( 0 );
markrad 0:cdf462088d13 289 }
markrad 0:cdf462088d13 290 #endif /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */
markrad 0:cdf462088d13 291
markrad 0:cdf462088d13 292 /*
markrad 0:cdf462088d13 293 * Accept a connection from a remote client
markrad 0:cdf462088d13 294 */
markrad 0:cdf462088d13 295 int mbedtls_net_accept( mbedtls_net_context *bind_ctx,
markrad 0:cdf462088d13 296 mbedtls_net_context *client_ctx,
markrad 0:cdf462088d13 297 void *client_ip, size_t buf_size, size_t *ip_len )
markrad 0:cdf462088d13 298 {
markrad 0:cdf462088d13 299 int ret;
markrad 0:cdf462088d13 300 int type;
markrad 0:cdf462088d13 301
markrad 0:cdf462088d13 302 struct sockaddr_storage client_addr;
markrad 0:cdf462088d13 303
markrad 0:cdf462088d13 304 #if defined(__socklen_t_defined) || defined(_SOCKLEN_T) || \
markrad 0:cdf462088d13 305 defined(_SOCKLEN_T_DECLARED) || defined(__DEFINED_socklen_t)
markrad 0:cdf462088d13 306 socklen_t n = (socklen_t) sizeof( client_addr );
markrad 0:cdf462088d13 307 socklen_t type_len = (socklen_t) sizeof( type );
markrad 0:cdf462088d13 308 #else
markrad 0:cdf462088d13 309 int n = (int) sizeof( client_addr );
markrad 0:cdf462088d13 310 int type_len = (int) sizeof( type );
markrad 0:cdf462088d13 311 #endif
markrad 0:cdf462088d13 312
markrad 0:cdf462088d13 313 /* Is this a TCP or UDP socket? */
markrad 0:cdf462088d13 314 if( getsockopt( bind_ctx->fd, SOL_SOCKET, SO_TYPE,
markrad 0:cdf462088d13 315 (void *) &type, &type_len ) != 0 ||
markrad 0:cdf462088d13 316 ( type != SOCK_STREAM && type != SOCK_DGRAM ) )
markrad 0:cdf462088d13 317 {
markrad 0:cdf462088d13 318 return( MBEDTLS_ERR_NET_ACCEPT_FAILED );
markrad 0:cdf462088d13 319 }
markrad 0:cdf462088d13 320
markrad 0:cdf462088d13 321 if( type == SOCK_STREAM )
markrad 0:cdf462088d13 322 {
markrad 0:cdf462088d13 323 /* TCP: actual accept() */
markrad 0:cdf462088d13 324 ret = client_ctx->fd = (int) accept( bind_ctx->fd,
markrad 0:cdf462088d13 325 (struct sockaddr *) &client_addr, &n );
markrad 0:cdf462088d13 326 }
markrad 0:cdf462088d13 327 else
markrad 0:cdf462088d13 328 {
markrad 0:cdf462088d13 329 /* UDP: wait for a message, but keep it in the queue */
markrad 0:cdf462088d13 330 char buf[1] = { 0 };
markrad 0:cdf462088d13 331
markrad 0:cdf462088d13 332 ret = (int) recvfrom( bind_ctx->fd, buf, sizeof( buf ), MSG_PEEK,
markrad 0:cdf462088d13 333 (struct sockaddr *) &client_addr, &n );
markrad 0:cdf462088d13 334
markrad 0:cdf462088d13 335 #if defined(_WIN32)
markrad 0:cdf462088d13 336 if( ret == SOCKET_ERROR &&
markrad 0:cdf462088d13 337 WSAGetLastError() == WSAEMSGSIZE )
markrad 0:cdf462088d13 338 {
markrad 0:cdf462088d13 339 /* We know buf is too small, thanks, just peeking here */
markrad 0:cdf462088d13 340 ret = 0;
markrad 0:cdf462088d13 341 }
markrad 0:cdf462088d13 342 #endif
markrad 0:cdf462088d13 343 }
markrad 0:cdf462088d13 344
markrad 0:cdf462088d13 345 if( ret < 0 )
markrad 0:cdf462088d13 346 {
markrad 0:cdf462088d13 347 if( net_would_block( bind_ctx ) != 0 )
markrad 0:cdf462088d13 348 return( MBEDTLS_ERR_SSL_WANT_READ );
markrad 0:cdf462088d13 349
markrad 0:cdf462088d13 350 return( MBEDTLS_ERR_NET_ACCEPT_FAILED );
markrad 0:cdf462088d13 351 }
markrad 0:cdf462088d13 352
markrad 0:cdf462088d13 353 /* UDP: hijack the listening socket to communicate with the client,
markrad 0:cdf462088d13 354 * then bind a new socket to accept new connections */
markrad 0:cdf462088d13 355 if( type != SOCK_STREAM )
markrad 0:cdf462088d13 356 {
markrad 0:cdf462088d13 357 struct sockaddr_storage local_addr;
markrad 0:cdf462088d13 358 int one = 1;
markrad 0:cdf462088d13 359
markrad 0:cdf462088d13 360 if( connect( bind_ctx->fd, (struct sockaddr *) &client_addr, n ) != 0 )
markrad 0:cdf462088d13 361 return( MBEDTLS_ERR_NET_ACCEPT_FAILED );
markrad 0:cdf462088d13 362
markrad 0:cdf462088d13 363 client_ctx->fd = bind_ctx->fd;
markrad 0:cdf462088d13 364 bind_ctx->fd = -1; /* In case we exit early */
markrad 0:cdf462088d13 365
markrad 0:cdf462088d13 366 n = sizeof( struct sockaddr_storage );
markrad 0:cdf462088d13 367 if( getsockname( client_ctx->fd,
markrad 0:cdf462088d13 368 (struct sockaddr *) &local_addr, &n ) != 0 ||
markrad 0:cdf462088d13 369 ( bind_ctx->fd = (int) socket( local_addr.ss_family,
markrad 0:cdf462088d13 370 SOCK_DGRAM, IPPROTO_UDP ) ) < 0 ||
markrad 0:cdf462088d13 371 setsockopt( bind_ctx->fd, SOL_SOCKET, SO_REUSEADDR,
markrad 0:cdf462088d13 372 (const char *) &one, sizeof( one ) ) != 0 )
markrad 0:cdf462088d13 373 {
markrad 0:cdf462088d13 374 return( MBEDTLS_ERR_NET_SOCKET_FAILED );
markrad 0:cdf462088d13 375 }
markrad 0:cdf462088d13 376
markrad 0:cdf462088d13 377 if( bind( bind_ctx->fd, (struct sockaddr *) &local_addr, n ) != 0 )
markrad 0:cdf462088d13 378 {
markrad 0:cdf462088d13 379 return( MBEDTLS_ERR_NET_BIND_FAILED );
markrad 0:cdf462088d13 380 }
markrad 0:cdf462088d13 381 }
markrad 0:cdf462088d13 382
markrad 0:cdf462088d13 383 if( client_ip != NULL )
markrad 0:cdf462088d13 384 {
markrad 0:cdf462088d13 385 if( client_addr.ss_family == AF_INET )
markrad 0:cdf462088d13 386 {
markrad 0:cdf462088d13 387 struct sockaddr_in *addr4 = (struct sockaddr_in *) &client_addr;
markrad 0:cdf462088d13 388 *ip_len = sizeof( addr4->sin_addr.s_addr );
markrad 0:cdf462088d13 389
markrad 0:cdf462088d13 390 if( buf_size < *ip_len )
markrad 0:cdf462088d13 391 return( MBEDTLS_ERR_NET_BUFFER_TOO_SMALL );
markrad 0:cdf462088d13 392
markrad 0:cdf462088d13 393 memcpy( client_ip, &addr4->sin_addr.s_addr, *ip_len );
markrad 0:cdf462088d13 394 }
markrad 0:cdf462088d13 395 else
markrad 0:cdf462088d13 396 {
markrad 0:cdf462088d13 397 struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *) &client_addr;
markrad 0:cdf462088d13 398 *ip_len = sizeof( addr6->sin6_addr.s6_addr );
markrad 0:cdf462088d13 399
markrad 0:cdf462088d13 400 if( buf_size < *ip_len )
markrad 0:cdf462088d13 401 return( MBEDTLS_ERR_NET_BUFFER_TOO_SMALL );
markrad 0:cdf462088d13 402
markrad 0:cdf462088d13 403 memcpy( client_ip, &addr6->sin6_addr.s6_addr, *ip_len);
markrad 0:cdf462088d13 404 }
markrad 0:cdf462088d13 405 }
markrad 0:cdf462088d13 406
markrad 0:cdf462088d13 407 return( 0 );
markrad 0:cdf462088d13 408 }
markrad 0:cdf462088d13 409
markrad 0:cdf462088d13 410 /*
markrad 0:cdf462088d13 411 * Set the socket blocking or non-blocking
markrad 0:cdf462088d13 412 */
markrad 0:cdf462088d13 413 int mbedtls_net_set_block( mbedtls_net_context *ctx )
markrad 0:cdf462088d13 414 {
markrad 0:cdf462088d13 415 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 416 !defined(EFI32)
markrad 0:cdf462088d13 417 u_long n = 0;
markrad 0:cdf462088d13 418 return( ioctlsocket( ctx->fd, FIONBIO, &n ) );
markrad 0:cdf462088d13 419 #else
markrad 0:cdf462088d13 420 return( fcntl( ctx->fd, F_SETFL, fcntl( ctx->fd, F_GETFL ) & ~O_NONBLOCK ) );
markrad 0:cdf462088d13 421 #endif
markrad 0:cdf462088d13 422 }
markrad 0:cdf462088d13 423
markrad 0:cdf462088d13 424 int mbedtls_net_set_nonblock( mbedtls_net_context *ctx )
markrad 0:cdf462088d13 425 {
markrad 0:cdf462088d13 426 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 427 !defined(EFI32)
markrad 0:cdf462088d13 428 u_long n = 1;
markrad 0:cdf462088d13 429 return( ioctlsocket( ctx->fd, FIONBIO, &n ) );
markrad 0:cdf462088d13 430 #else
markrad 0:cdf462088d13 431 return( fcntl( ctx->fd, F_SETFL, fcntl( ctx->fd, F_GETFL ) | O_NONBLOCK ) );
markrad 0:cdf462088d13 432 #endif
markrad 0:cdf462088d13 433 }
markrad 0:cdf462088d13 434
markrad 0:cdf462088d13 435 /*
markrad 0:cdf462088d13 436 * Portable usleep helper
markrad 0:cdf462088d13 437 */
markrad 0:cdf462088d13 438 void mbedtls_net_usleep( unsigned long usec )
markrad 0:cdf462088d13 439 {
markrad 0:cdf462088d13 440 #if defined(_WIN32)
markrad 0:cdf462088d13 441 Sleep( ( usec + 999 ) / 1000 );
markrad 0:cdf462088d13 442 #else
markrad 0:cdf462088d13 443 struct timeval tv;
markrad 0:cdf462088d13 444 tv.tv_sec = usec / 1000000;
markrad 0:cdf462088d13 445 #if defined(__unix__) || defined(__unix) || \
markrad 0:cdf462088d13 446 ( defined(__APPLE__) && defined(__MACH__) )
markrad 0:cdf462088d13 447 tv.tv_usec = (suseconds_t) usec % 1000000;
markrad 0:cdf462088d13 448 #else
markrad 0:cdf462088d13 449 tv.tv_usec = usec % 1000000;
markrad 0:cdf462088d13 450 #endif
markrad 0:cdf462088d13 451 select( 0, NULL, NULL, NULL, &tv );
markrad 0:cdf462088d13 452 #endif
markrad 0:cdf462088d13 453 }
markrad 0:cdf462088d13 454
markrad 0:cdf462088d13 455 /*
markrad 0:cdf462088d13 456 * Read at most 'len' characters
markrad 0:cdf462088d13 457 */
markrad 0:cdf462088d13 458 int mbedtls_net_recv( void *ctx, unsigned char *buf, size_t len )
markrad 0:cdf462088d13 459 {
markrad 0:cdf462088d13 460 int ret;
markrad 0:cdf462088d13 461 int fd = ((mbedtls_net_context *) ctx)->fd;
markrad 0:cdf462088d13 462
markrad 0:cdf462088d13 463 if( fd < 0 )
markrad 0:cdf462088d13 464 return( MBEDTLS_ERR_NET_INVALID_CONTEXT );
markrad 0:cdf462088d13 465
markrad 0:cdf462088d13 466 ret = (int) read( fd, buf, len );
markrad 0:cdf462088d13 467
markrad 0:cdf462088d13 468 if( ret < 0 )
markrad 0:cdf462088d13 469 {
markrad 0:cdf462088d13 470 if( net_would_block( ctx ) != 0 )
markrad 0:cdf462088d13 471 return( MBEDTLS_ERR_SSL_WANT_READ );
markrad 0:cdf462088d13 472
markrad 0:cdf462088d13 473 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 474 !defined(EFI32)
markrad 0:cdf462088d13 475 if( WSAGetLastError() == WSAECONNRESET )
markrad 0:cdf462088d13 476 return( MBEDTLS_ERR_NET_CONN_RESET );
markrad 0:cdf462088d13 477 #else
markrad 0:cdf462088d13 478 if( errno == EPIPE || errno == ECONNRESET )
markrad 0:cdf462088d13 479 return( MBEDTLS_ERR_NET_CONN_RESET );
markrad 0:cdf462088d13 480
markrad 0:cdf462088d13 481 if( errno == EINTR )
markrad 0:cdf462088d13 482 return( MBEDTLS_ERR_SSL_WANT_READ );
markrad 0:cdf462088d13 483 #endif
markrad 0:cdf462088d13 484
markrad 0:cdf462088d13 485 return( MBEDTLS_ERR_NET_RECV_FAILED );
markrad 0:cdf462088d13 486 }
markrad 0:cdf462088d13 487
markrad 0:cdf462088d13 488 return( ret );
markrad 0:cdf462088d13 489 }
markrad 0:cdf462088d13 490
markrad 0:cdf462088d13 491 /*
markrad 0:cdf462088d13 492 * Read at most 'len' characters, blocking for at most 'timeout' ms
markrad 0:cdf462088d13 493 */
markrad 0:cdf462088d13 494 int mbedtls_net_recv_timeout( void *ctx, unsigned char *buf, size_t len,
markrad 0:cdf462088d13 495 uint32_t timeout )
markrad 0:cdf462088d13 496 {
markrad 0:cdf462088d13 497 int ret;
markrad 0:cdf462088d13 498 struct timeval tv;
markrad 0:cdf462088d13 499 fd_set read_fds;
markrad 0:cdf462088d13 500 int fd = ((mbedtls_net_context *) ctx)->fd;
markrad 0:cdf462088d13 501
markrad 0:cdf462088d13 502 if( fd < 0 )
markrad 0:cdf462088d13 503 return( MBEDTLS_ERR_NET_INVALID_CONTEXT );
markrad 0:cdf462088d13 504
markrad 0:cdf462088d13 505 FD_ZERO( &read_fds );
markrad 0:cdf462088d13 506 FD_SET( fd, &read_fds );
markrad 0:cdf462088d13 507
markrad 0:cdf462088d13 508 tv.tv_sec = timeout / 1000;
markrad 0:cdf462088d13 509 tv.tv_usec = ( timeout % 1000 ) * 1000;
markrad 0:cdf462088d13 510
markrad 0:cdf462088d13 511 ret = select( fd + 1, &read_fds, NULL, NULL, timeout == 0 ? NULL : &tv );
markrad 0:cdf462088d13 512
markrad 0:cdf462088d13 513 /* Zero fds ready means we timed out */
markrad 0:cdf462088d13 514 if( ret == 0 )
markrad 0:cdf462088d13 515 return( MBEDTLS_ERR_SSL_TIMEOUT );
markrad 0:cdf462088d13 516
markrad 0:cdf462088d13 517 if( ret < 0 )
markrad 0:cdf462088d13 518 {
markrad 0:cdf462088d13 519 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 520 !defined(EFI32)
markrad 0:cdf462088d13 521 if( WSAGetLastError() == WSAEINTR )
markrad 0:cdf462088d13 522 return( MBEDTLS_ERR_SSL_WANT_READ );
markrad 0:cdf462088d13 523 #else
markrad 0:cdf462088d13 524 if( errno == EINTR )
markrad 0:cdf462088d13 525 return( MBEDTLS_ERR_SSL_WANT_READ );
markrad 0:cdf462088d13 526 #endif
markrad 0:cdf462088d13 527
markrad 0:cdf462088d13 528 return( MBEDTLS_ERR_NET_RECV_FAILED );
markrad 0:cdf462088d13 529 }
markrad 0:cdf462088d13 530
markrad 0:cdf462088d13 531 /* This call will not block */
markrad 0:cdf462088d13 532 return( mbedtls_net_recv( ctx, buf, len ) );
markrad 0:cdf462088d13 533 }
markrad 0:cdf462088d13 534
markrad 0:cdf462088d13 535 /*
markrad 0:cdf462088d13 536 * Write at most 'len' characters
markrad 0:cdf462088d13 537 */
markrad 0:cdf462088d13 538 int mbedtls_net_send( void *ctx, const unsigned char *buf, size_t len )
markrad 0:cdf462088d13 539 {
markrad 0:cdf462088d13 540 int ret;
markrad 0:cdf462088d13 541 int fd = ((mbedtls_net_context *) ctx)->fd;
markrad 0:cdf462088d13 542
markrad 0:cdf462088d13 543 if( fd < 0 )
markrad 0:cdf462088d13 544 return( MBEDTLS_ERR_NET_INVALID_CONTEXT );
markrad 0:cdf462088d13 545
markrad 0:cdf462088d13 546 ret = (int) write( fd, buf, len );
markrad 0:cdf462088d13 547
markrad 0:cdf462088d13 548 if( ret < 0 )
markrad 0:cdf462088d13 549 {
markrad 0:cdf462088d13 550 if( net_would_block( ctx ) != 0 )
markrad 0:cdf462088d13 551 return( MBEDTLS_ERR_SSL_WANT_WRITE );
markrad 0:cdf462088d13 552
markrad 0:cdf462088d13 553 #if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \
markrad 0:cdf462088d13 554 !defined(EFI32)
markrad 0:cdf462088d13 555 if( WSAGetLastError() == WSAECONNRESET )
markrad 0:cdf462088d13 556 return( MBEDTLS_ERR_NET_CONN_RESET );
markrad 0:cdf462088d13 557 #else
markrad 0:cdf462088d13 558 if( errno == EPIPE || errno == ECONNRESET )
markrad 0:cdf462088d13 559 return( MBEDTLS_ERR_NET_CONN_RESET );
markrad 0:cdf462088d13 560
markrad 0:cdf462088d13 561 if( errno == EINTR )
markrad 0:cdf462088d13 562 return( MBEDTLS_ERR_SSL_WANT_WRITE );
markrad 0:cdf462088d13 563 #endif
markrad 0:cdf462088d13 564
markrad 0:cdf462088d13 565 return( MBEDTLS_ERR_NET_SEND_FAILED );
markrad 0:cdf462088d13 566 }
markrad 0:cdf462088d13 567
markrad 0:cdf462088d13 568 return( ret );
markrad 0:cdf462088d13 569 }
markrad 0:cdf462088d13 570
markrad 0:cdf462088d13 571 /*
markrad 0:cdf462088d13 572 * Gracefully close the connection
markrad 0:cdf462088d13 573 */
markrad 0:cdf462088d13 574 void mbedtls_net_free( mbedtls_net_context *ctx )
markrad 0:cdf462088d13 575 {
markrad 0:cdf462088d13 576 if( ctx->fd == -1 )
markrad 0:cdf462088d13 577 return;
markrad 0:cdf462088d13 578
markrad 0:cdf462088d13 579 shutdown( ctx->fd, 2 );
markrad 0:cdf462088d13 580 close( ctx->fd );
markrad 0:cdf462088d13 581
markrad 0:cdf462088d13 582 ctx->fd = -1;
markrad 0:cdf462088d13 583 }
markrad 0:cdf462088d13 584
markrad 0:cdf462088d13 585 #endif /* MBEDTLS_NET_C */