--- /dev/null
+/*\r
+ * TCP/IP or UDP/IP networking functions\r
+ *\r
+ * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved\r
+ * SPDX-License-Identifier: Apache-2.0\r
+ *\r
+ * Licensed under the Apache License, Version 2.0 (the "License"); you may\r
+ * not use this file except in compliance with the License.\r
+ * You may obtain a copy of the License at\r
+ *\r
+ * http://www.apache.org/licenses/LICENSE-2.0\r
+ *\r
+ * Unless required by applicable law or agreed to in writing, software\r
+ * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT\r
+ * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+ * See the License for the specific language governing permissions and\r
+ * limitations under the License.\r
+ *\r
+ * This file is part of mbed TLS (https://tls.mbed.org)\r
+ */\r
+\r
+/* Enable definition of getaddrinfo() even when compiling with -std=c99. Must\r
+ * be set before config.h, which pulls in glibc's features.h indirectly.\r
+ * Harmless on other platforms. */\r
+#define _POSIX_C_SOURCE 200112L\r
+\r
+#if !defined(MBEDTLS_CONFIG_FILE)\r
+#include "mbedtls/config.h"\r
+#else\r
+#include MBEDTLS_CONFIG_FILE\r
+#endif\r
+\r
+#if defined(MBEDTLS_NET_C)\r
+\r
+#if !defined(unix) && !defined(__unix__) && !defined(__unix) && \\r
+ !defined(__APPLE__) && !defined(_WIN32) && !defined(__QNXNTO__) && \\r
+ !defined(__HAIKU__)\r
+#error "This module only works on Unix and Windows, see MBEDTLS_NET_C in config.h"\r
+#endif\r
+\r
+#if defined(MBEDTLS_PLATFORM_C)\r
+#include "mbedtls/platform.h"\r
+#else\r
+#include <stdlib.h>\r
+#endif\r
+\r
+#include "mbedtls/net_sockets.h"\r
+\r
+#include <string.h>\r
+\r
+#if (defined(_WIN32) || defined(_WIN32_WCE)) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+\r
+#define IS_EINTR( ret ) ( ( ret ) == WSAEINTR )\r
+\r
+#if !defined(_WIN32_WINNT) || (_WIN32_WINNT < 0x0501)\r
+#undef _WIN32_WINNT\r
+/* Enables getaddrinfo() & Co */\r
+#define _WIN32_WINNT 0x0501\r
+#endif\r
+\r
+#include <ws2tcpip.h>\r
+\r
+#include <winsock2.h>\r
+#include <windows.h>\r
+\r
+#if defined(_MSC_VER)\r
+#if defined(_WIN32_WCE)\r
+#pragma comment( lib, "ws2.lib" )\r
+#else\r
+#pragma comment( lib, "ws2_32.lib" )\r
+#endif\r
+#endif /* _MSC_VER */\r
+\r
+#define read(fd,buf,len) recv( fd, (char*)( buf ), (int)( len ), 0 )\r
+#define write(fd,buf,len) send( fd, (char*)( buf ), (int)( len ), 0 )\r
+#define close(fd) closesocket(fd)\r
+\r
+static int wsa_init_done = 0;\r
+\r
+#else /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */\r
+\r
+#include <sys/types.h>\r
+#include <sys/socket.h>\r
+#include <netinet/in.h>\r
+#include <arpa/inet.h>\r
+#include <sys/time.h>\r
+#include <unistd.h>\r
+#include <signal.h>\r
+#include <fcntl.h>\r
+#include <netdb.h>\r
+#include <errno.h>\r
+\r
+#define IS_EINTR( ret ) ( ( ret ) == EINTR )\r
+\r
+#endif /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */\r
+\r
+/* Some MS functions want int and MSVC warns if we pass size_t,\r
+ * but the standard functions use socklen_t, so cast only for MSVC */\r
+#if defined(_MSC_VER)\r
+#define MSVC_INT_CAST (int)\r
+#else\r
+#define MSVC_INT_CAST\r
+#endif\r
+\r
+#include <stdio.h>\r
+\r
+#include <time.h>\r
+\r
+#include <stdint.h>\r
+\r
+/*\r
+ * Prepare for using the sockets interface\r
+ */\r
+static int net_prepare( void )\r
+{\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ WSADATA wsaData;\r
+\r
+ if( wsa_init_done == 0 )\r
+ {\r
+ if( WSAStartup( MAKEWORD(2,0), &wsaData ) != 0 )\r
+ return( MBEDTLS_ERR_NET_SOCKET_FAILED );\r
+\r
+ wsa_init_done = 1;\r
+ }\r
+#else\r
+#if !defined(EFIX64) && !defined(EFI32)\r
+ signal( SIGPIPE, SIG_IGN );\r
+#endif\r
+#endif\r
+ return( 0 );\r
+}\r
+\r
+/*\r
+ * Initialize a context\r
+ */\r
+void mbedtls_net_init( mbedtls_net_context *ctx )\r
+{\r
+ ctx->fd = -1;\r
+}\r
+\r
+/*\r
+ * Initiate a TCP connection with host:port and the given protocol\r
+ */\r
+int mbedtls_net_connect( mbedtls_net_context *ctx, const char *host,\r
+ const char *port, int proto )\r
+{\r
+ int ret;\r
+ struct addrinfo hints, *addr_list, *cur;\r
+\r
+ if( ( ret = net_prepare() ) != 0 )\r
+ return( ret );\r
+\r
+ /* Do name resolution with both IPv6 and IPv4 */\r
+ memset( &hints, 0, sizeof( hints ) );\r
+ hints.ai_family = AF_UNSPEC;\r
+ hints.ai_socktype = proto == MBEDTLS_NET_PROTO_UDP ? SOCK_DGRAM : SOCK_STREAM;\r
+ hints.ai_protocol = proto == MBEDTLS_NET_PROTO_UDP ? IPPROTO_UDP : IPPROTO_TCP;\r
+\r
+ if( getaddrinfo( host, port, &hints, &addr_list ) != 0 )\r
+ return( MBEDTLS_ERR_NET_UNKNOWN_HOST );\r
+\r
+ /* Try the sockaddrs until a connection succeeds */\r
+ ret = MBEDTLS_ERR_NET_UNKNOWN_HOST;\r
+ for( cur = addr_list; cur != NULL; cur = cur->ai_next )\r
+ {\r
+ ctx->fd = (int) socket( cur->ai_family, cur->ai_socktype,\r
+ cur->ai_protocol );\r
+ if( ctx->fd < 0 )\r
+ {\r
+ ret = MBEDTLS_ERR_NET_SOCKET_FAILED;\r
+ continue;\r
+ }\r
+\r
+ if( connect( ctx->fd, cur->ai_addr, MSVC_INT_CAST cur->ai_addrlen ) == 0 )\r
+ {\r
+ ret = 0;\r
+ break;\r
+ }\r
+\r
+ close( ctx->fd );\r
+ ret = MBEDTLS_ERR_NET_CONNECT_FAILED;\r
+ }\r
+\r
+ freeaddrinfo( addr_list );\r
+\r
+ return( ret );\r
+}\r
+\r
+/*\r
+ * Create a listening socket on bind_ip:port\r
+ */\r
+int mbedtls_net_bind( mbedtls_net_context *ctx, const char *bind_ip, const char *port, int proto )\r
+{\r
+ int n, ret;\r
+ struct addrinfo hints, *addr_list, *cur;\r
+\r
+ if( ( ret = net_prepare() ) != 0 )\r
+ return( ret );\r
+\r
+ /* Bind to IPv6 and/or IPv4, but only in the desired protocol */\r
+ memset( &hints, 0, sizeof( hints ) );\r
+ hints.ai_family = AF_UNSPEC;\r
+ hints.ai_socktype = proto == MBEDTLS_NET_PROTO_UDP ? SOCK_DGRAM : SOCK_STREAM;\r
+ hints.ai_protocol = proto == MBEDTLS_NET_PROTO_UDP ? IPPROTO_UDP : IPPROTO_TCP;\r
+ if( bind_ip == NULL )\r
+ hints.ai_flags = AI_PASSIVE;\r
+\r
+ if( getaddrinfo( bind_ip, port, &hints, &addr_list ) != 0 )\r
+ return( MBEDTLS_ERR_NET_UNKNOWN_HOST );\r
+\r
+ /* Try the sockaddrs until a binding succeeds */\r
+ ret = MBEDTLS_ERR_NET_UNKNOWN_HOST;\r
+ for( cur = addr_list; cur != NULL; cur = cur->ai_next )\r
+ {\r
+ ctx->fd = (int) socket( cur->ai_family, cur->ai_socktype,\r
+ cur->ai_protocol );\r
+ if( ctx->fd < 0 )\r
+ {\r
+ ret = MBEDTLS_ERR_NET_SOCKET_FAILED;\r
+ continue;\r
+ }\r
+\r
+ n = 1;\r
+ if( setsockopt( ctx->fd, SOL_SOCKET, SO_REUSEADDR,\r
+ (const char *) &n, sizeof( n ) ) != 0 )\r
+ {\r
+ close( ctx->fd );\r
+ ret = MBEDTLS_ERR_NET_SOCKET_FAILED;\r
+ continue;\r
+ }\r
+\r
+ if( bind( ctx->fd, cur->ai_addr, MSVC_INT_CAST cur->ai_addrlen ) != 0 )\r
+ {\r
+ close( ctx->fd );\r
+ ret = MBEDTLS_ERR_NET_BIND_FAILED;\r
+ continue;\r
+ }\r
+\r
+ /* Listen only makes sense for TCP */\r
+ if( proto == MBEDTLS_NET_PROTO_TCP )\r
+ {\r
+ if( listen( ctx->fd, MBEDTLS_NET_LISTEN_BACKLOG ) != 0 )\r
+ {\r
+ close( ctx->fd );\r
+ ret = MBEDTLS_ERR_NET_LISTEN_FAILED;\r
+ continue;\r
+ }\r
+ }\r
+\r
+ /* Bind was successful */\r
+ ret = 0;\r
+ break;\r
+ }\r
+\r
+ freeaddrinfo( addr_list );\r
+\r
+ return( ret );\r
+\r
+}\r
+\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+/*\r
+ * Check if the requested operation would be blocking on a non-blocking socket\r
+ * and thus 'failed' with a negative return value.\r
+ */\r
+static int net_would_block( const mbedtls_net_context *ctx )\r
+{\r
+ ((void) ctx);\r
+ return( WSAGetLastError() == WSAEWOULDBLOCK );\r
+}\r
+#else\r
+/*\r
+ * Check if the requested operation would be blocking on a non-blocking socket\r
+ * and thus 'failed' with a negative return value.\r
+ *\r
+ * Note: on a blocking socket this function always returns 0!\r
+ */\r
+static int net_would_block( const mbedtls_net_context *ctx )\r
+{\r
+ int err = errno;\r
+\r
+ /*\r
+ * Never return 'WOULD BLOCK' on a non-blocking socket\r
+ */\r
+ if( ( fcntl( ctx->fd, F_GETFL ) & O_NONBLOCK ) != O_NONBLOCK )\r
+ {\r
+ errno = err;\r
+ return( 0 );\r
+ }\r
+\r
+ switch( errno = err )\r
+ {\r
+#if defined EAGAIN\r
+ case EAGAIN:\r
+#endif\r
+#if defined EWOULDBLOCK && EWOULDBLOCK != EAGAIN\r
+ case EWOULDBLOCK:\r
+#endif\r
+ return( 1 );\r
+ }\r
+ return( 0 );\r
+}\r
+#endif /* ( _WIN32 || _WIN32_WCE ) && !EFIX64 && !EFI32 */\r
+\r
+/*\r
+ * Accept a connection from a remote client\r
+ */\r
+int mbedtls_net_accept( mbedtls_net_context *bind_ctx,\r
+ mbedtls_net_context *client_ctx,\r
+ void *client_ip, size_t buf_size, size_t *ip_len )\r
+{\r
+ int ret;\r
+ int type;\r
+\r
+ struct sockaddr_storage client_addr;\r
+\r
+#if defined(__socklen_t_defined) || defined(_SOCKLEN_T) || \\r
+ defined(_SOCKLEN_T_DECLARED) || defined(__DEFINED_socklen_t)\r
+ socklen_t n = (socklen_t) sizeof( client_addr );\r
+ socklen_t type_len = (socklen_t) sizeof( type );\r
+#else\r
+ int n = (int) sizeof( client_addr );\r
+ int type_len = (int) sizeof( type );\r
+#endif\r
+\r
+ /* Is this a TCP or UDP socket? */\r
+ if( getsockopt( bind_ctx->fd, SOL_SOCKET, SO_TYPE,\r
+ (void *) &type, &type_len ) != 0 ||\r
+ ( type != SOCK_STREAM && type != SOCK_DGRAM ) )\r
+ {\r
+ return( MBEDTLS_ERR_NET_ACCEPT_FAILED );\r
+ }\r
+\r
+ if( type == SOCK_STREAM )\r
+ {\r
+ /* TCP: actual accept() */\r
+ ret = client_ctx->fd = (int) accept( bind_ctx->fd,\r
+ (struct sockaddr *) &client_addr, &n );\r
+ }\r
+ else\r
+ {\r
+ /* UDP: wait for a message, but keep it in the queue */\r
+ char buf[1] = { 0 };\r
+\r
+ ret = (int) recvfrom( bind_ctx->fd, buf, sizeof( buf ), MSG_PEEK,\r
+ (struct sockaddr *) &client_addr, &n );\r
+\r
+#if defined(_WIN32)\r
+ if( ret == SOCKET_ERROR &&\r
+ WSAGetLastError() == WSAEMSGSIZE )\r
+ {\r
+ /* We know buf is too small, thanks, just peeking here */\r
+ ret = 0;\r
+ }\r
+#endif\r
+ }\r
+\r
+ if( ret < 0 )\r
+ {\r
+ if( net_would_block( bind_ctx ) != 0 )\r
+ return( MBEDTLS_ERR_SSL_WANT_READ );\r
+\r
+ return( MBEDTLS_ERR_NET_ACCEPT_FAILED );\r
+ }\r
+\r
+ /* UDP: hijack the listening socket to communicate with the client,\r
+ * then bind a new socket to accept new connections */\r
+ if( type != SOCK_STREAM )\r
+ {\r
+ struct sockaddr_storage local_addr;\r
+ int one = 1;\r
+\r
+ if( connect( bind_ctx->fd, (struct sockaddr *) &client_addr, n ) != 0 )\r
+ return( MBEDTLS_ERR_NET_ACCEPT_FAILED );\r
+\r
+ client_ctx->fd = bind_ctx->fd;\r
+ bind_ctx->fd = -1; /* In case we exit early */\r
+\r
+ n = sizeof( struct sockaddr_storage );\r
+ if( getsockname( client_ctx->fd,\r
+ (struct sockaddr *) &local_addr, &n ) != 0 ||\r
+ ( bind_ctx->fd = (int) socket( local_addr.ss_family,\r
+ SOCK_DGRAM, IPPROTO_UDP ) ) < 0 ||\r
+ setsockopt( bind_ctx->fd, SOL_SOCKET, SO_REUSEADDR,\r
+ (const char *) &one, sizeof( one ) ) != 0 )\r
+ {\r
+ return( MBEDTLS_ERR_NET_SOCKET_FAILED );\r
+ }\r
+\r
+ if( bind( bind_ctx->fd, (struct sockaddr *) &local_addr, n ) != 0 )\r
+ {\r
+ return( MBEDTLS_ERR_NET_BIND_FAILED );\r
+ }\r
+ }\r
+\r
+ if( client_ip != NULL )\r
+ {\r
+ if( client_addr.ss_family == AF_INET )\r
+ {\r
+ struct sockaddr_in *addr4 = (struct sockaddr_in *) &client_addr;\r
+ *ip_len = sizeof( addr4->sin_addr.s_addr );\r
+\r
+ if( buf_size < *ip_len )\r
+ return( MBEDTLS_ERR_NET_BUFFER_TOO_SMALL );\r
+\r
+ memcpy( client_ip, &addr4->sin_addr.s_addr, *ip_len );\r
+ }\r
+ else\r
+ {\r
+ struct sockaddr_in6 *addr6 = (struct sockaddr_in6 *) &client_addr;\r
+ *ip_len = sizeof( addr6->sin6_addr.s6_addr );\r
+\r
+ if( buf_size < *ip_len )\r
+ return( MBEDTLS_ERR_NET_BUFFER_TOO_SMALL );\r
+\r
+ memcpy( client_ip, &addr6->sin6_addr.s6_addr, *ip_len);\r
+ }\r
+ }\r
+\r
+ return( 0 );\r
+}\r
+\r
+/*\r
+ * Set the socket blocking or non-blocking\r
+ */\r
+int mbedtls_net_set_block( mbedtls_net_context *ctx )\r
+{\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ u_long n = 0;\r
+ return( ioctlsocket( ctx->fd, FIONBIO, &n ) );\r
+#else\r
+ return( fcntl( ctx->fd, F_SETFL, fcntl( ctx->fd, F_GETFL ) & ~O_NONBLOCK ) );\r
+#endif\r
+}\r
+\r
+int mbedtls_net_set_nonblock( mbedtls_net_context *ctx )\r
+{\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ u_long n = 1;\r
+ return( ioctlsocket( ctx->fd, FIONBIO, &n ) );\r
+#else\r
+ return( fcntl( ctx->fd, F_SETFL, fcntl( ctx->fd, F_GETFL ) | O_NONBLOCK ) );\r
+#endif\r
+}\r
+\r
+/*\r
+ * Check if data is available on the socket\r
+ */\r
+\r
+int mbedtls_net_poll( mbedtls_net_context *ctx, uint32_t rw, uint32_t timeout )\r
+{\r
+ int ret;\r
+ struct timeval tv;\r
+\r
+ fd_set read_fds;\r
+ fd_set write_fds;\r
+\r
+ int fd = ctx->fd;\r
+\r
+ if( fd < 0 )\r
+ return( MBEDTLS_ERR_NET_INVALID_CONTEXT );\r
+\r
+#if defined(__has_feature)\r
+#if __has_feature(memory_sanitizer)\r
+ /* Ensure that memory sanitizers consider read_fds and write_fds as\r
+ * initialized even on platforms such as Glibc/x86_64 where FD_ZERO\r
+ * is implemented in assembly. */\r
+ memset( &read_fds, 0, sizeof( read_fds ) );\r
+ memset( &write_fds, 0, sizeof( write_fds ) );\r
+#endif\r
+#endif\r
+\r
+ FD_ZERO( &read_fds );\r
+ if( rw & MBEDTLS_NET_POLL_READ )\r
+ {\r
+ rw &= ~MBEDTLS_NET_POLL_READ;\r
+ FD_SET( fd, &read_fds );\r
+ }\r
+\r
+ FD_ZERO( &write_fds );\r
+ if( rw & MBEDTLS_NET_POLL_WRITE )\r
+ {\r
+ rw &= ~MBEDTLS_NET_POLL_WRITE;\r
+ FD_SET( fd, &write_fds );\r
+ }\r
+\r
+ if( rw != 0 )\r
+ return( MBEDTLS_ERR_NET_BAD_INPUT_DATA );\r
+\r
+ tv.tv_sec = timeout / 1000;\r
+ tv.tv_usec = ( timeout % 1000 ) * 1000;\r
+\r
+ do\r
+ {\r
+ ret = select( fd + 1, &read_fds, &write_fds, NULL,\r
+ timeout == (uint32_t) -1 ? NULL : &tv );\r
+ }\r
+ while( IS_EINTR( ret ) );\r
+\r
+ if( ret < 0 )\r
+ return( MBEDTLS_ERR_NET_POLL_FAILED );\r
+\r
+ ret = 0;\r
+ if( FD_ISSET( fd, &read_fds ) )\r
+ ret |= MBEDTLS_NET_POLL_READ;\r
+ if( FD_ISSET( fd, &write_fds ) )\r
+ ret |= MBEDTLS_NET_POLL_WRITE;\r
+\r
+ return( ret );\r
+}\r
+\r
+/*\r
+ * Portable usleep helper\r
+ */\r
+void mbedtls_net_usleep( unsigned long usec )\r
+{\r
+#if defined(_WIN32)\r
+ Sleep( ( usec + 999 ) / 1000 );\r
+#else\r
+ struct timeval tv;\r
+ tv.tv_sec = usec / 1000000;\r
+#if defined(__unix__) || defined(__unix) || \\r
+ ( defined(__APPLE__) && defined(__MACH__) )\r
+ tv.tv_usec = (suseconds_t) usec % 1000000;\r
+#else\r
+ tv.tv_usec = usec % 1000000;\r
+#endif\r
+ select( 0, NULL, NULL, NULL, &tv );\r
+#endif\r
+}\r
+\r
+/*\r
+ * Read at most 'len' characters\r
+ */\r
+int mbedtls_net_recv( void *ctx, unsigned char *buf, size_t len )\r
+{\r
+ int ret;\r
+ int fd = ((mbedtls_net_context *) ctx)->fd;\r
+\r
+ if( fd < 0 )\r
+ return( MBEDTLS_ERR_NET_INVALID_CONTEXT );\r
+\r
+ ret = (int) read( fd, buf, len );\r
+\r
+ if( ret < 0 )\r
+ {\r
+ if( net_would_block( ctx ) != 0 )\r
+ return( MBEDTLS_ERR_SSL_WANT_READ );\r
+\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ if( WSAGetLastError() == WSAECONNRESET )\r
+ return( MBEDTLS_ERR_NET_CONN_RESET );\r
+#else\r
+ if( errno == EPIPE || errno == ECONNRESET )\r
+ return( MBEDTLS_ERR_NET_CONN_RESET );\r
+\r
+ if( errno == EINTR )\r
+ return( MBEDTLS_ERR_SSL_WANT_READ );\r
+#endif\r
+\r
+ return( MBEDTLS_ERR_NET_RECV_FAILED );\r
+ }\r
+\r
+ return( ret );\r
+}\r
+\r
+/*\r
+ * Read at most 'len' characters, blocking for at most 'timeout' ms\r
+ */\r
+int mbedtls_net_recv_timeout( void *ctx, unsigned char *buf,\r
+ size_t len, uint32_t timeout )\r
+{\r
+ int ret;\r
+ struct timeval tv;\r
+ fd_set read_fds;\r
+ int fd = ((mbedtls_net_context *) ctx)->fd;\r
+\r
+ if( fd < 0 )\r
+ return( MBEDTLS_ERR_NET_INVALID_CONTEXT );\r
+\r
+ FD_ZERO( &read_fds );\r
+ FD_SET( fd, &read_fds );\r
+\r
+ tv.tv_sec = timeout / 1000;\r
+ tv.tv_usec = ( timeout % 1000 ) * 1000;\r
+\r
+ ret = select( fd + 1, &read_fds, NULL, NULL, timeout == 0 ? NULL : &tv );\r
+\r
+ /* Zero fds ready means we timed out */\r
+ if( ret == 0 )\r
+ return( MBEDTLS_ERR_SSL_TIMEOUT );\r
+\r
+ if( ret < 0 )\r
+ {\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ if( WSAGetLastError() == WSAEINTR )\r
+ return( MBEDTLS_ERR_SSL_WANT_READ );\r
+#else\r
+ if( errno == EINTR )\r
+ return( MBEDTLS_ERR_SSL_WANT_READ );\r
+#endif\r
+\r
+ return( MBEDTLS_ERR_NET_RECV_FAILED );\r
+ }\r
+\r
+ /* This call will not block */\r
+ return( mbedtls_net_recv( ctx, buf, len ) );\r
+}\r
+\r
+/*\r
+ * Write at most 'len' characters\r
+ */\r
+int mbedtls_net_send( void *ctx, const unsigned char *buf, size_t len )\r
+{\r
+ int ret;\r
+ int fd = ((mbedtls_net_context *) ctx)->fd;\r
+\r
+ if( fd < 0 )\r
+ return( MBEDTLS_ERR_NET_INVALID_CONTEXT );\r
+\r
+ ret = (int) write( fd, buf, len );\r
+\r
+ if( ret < 0 )\r
+ {\r
+ if( net_would_block( ctx ) != 0 )\r
+ return( MBEDTLS_ERR_SSL_WANT_WRITE );\r
+\r
+#if ( defined(_WIN32) || defined(_WIN32_WCE) ) && !defined(EFIX64) && \\r
+ !defined(EFI32)\r
+ if( WSAGetLastError() == WSAECONNRESET )\r
+ return( MBEDTLS_ERR_NET_CONN_RESET );\r
+#else\r
+ if( errno == EPIPE || errno == ECONNRESET )\r
+ return( MBEDTLS_ERR_NET_CONN_RESET );\r
+\r
+ if( errno == EINTR )\r
+ return( MBEDTLS_ERR_SSL_WANT_WRITE );\r
+#endif\r
+\r
+ return( MBEDTLS_ERR_NET_SEND_FAILED );\r
+ }\r
+\r
+ return( ret );\r
+}\r
+\r
+/*\r
+ * Gracefully close the connection\r
+ */\r
+void mbedtls_net_free( mbedtls_net_context *ctx )\r
+{\r
+ if( ctx->fd == -1 )\r
+ return;\r
+\r
+ shutdown( ctx->fd, 2 );\r
+ close( ctx->fd );\r
+\r
+ ctx->fd = -1;\r
+}\r
+\r
+#endif /* MBEDTLS_NET_C */\r