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