diff options
Diffstat (limited to 'release/src/router/cyassl/src/cyassl_io.c')
| -rw-r--r-- | release/src/router/cyassl/src/cyassl_io.c | 190 |
1 files changed, 190 insertions, 0 deletions
diff --git a/release/src/router/cyassl/src/cyassl_io.c b/release/src/router/cyassl/src/cyassl_io.c new file mode 100644 index 00000000..a5f5283f --- /dev/null +++ b/release/src/router/cyassl/src/cyassl_io.c @@ -0,0 +1,190 @@ +/* cyassl_io.c + * + * Copyright (C) 2006-2011 Sawtooth Consulting Ltd. + * + * This file is part of CyaSSL. + * + * CyaSSL is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * CyaSSL is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA + */ + + +#ifdef _WIN32_WCE + /* On WinCE winsock2.h must be included before windows.h for socket stuff */ + #include <winsock2.h> +#endif + +#include "cyassl_int.h" + +/* if user writes own I/O callbacks they can define CYASSL_USER_IO to remove + automatic setting of default I/O functions EmbedSend() and EmbedReceive() + but they'll still nedd SetCallback xxx() at end of file +*/ +#ifndef CYASSL_USER_IO + +#ifdef HAVE_LIBZ + #include "zlib.h" +#endif + +#ifndef USE_WINDOWS_API + #include <sys/types.h> + #include <errno.h> + #include <unistd.h> + #include <fcntl.h> + #if !(defined(DEVKITPRO) || defined(THREADX)) + #include <sys/socket.h> + #include <arpa/inet.h> + #include <netinet/in.h> + #include <netdb.h> + #include <sys/ioctl.h> + #endif + #ifdef THREADX + #include <socket.h> + #endif +#endif /* USE_WINDOWS_API */ + +#ifdef __sun + #include <sys/filio.h> +#endif + +#ifdef USE_WINDOWS_API + /* no epipe yet */ + #ifndef WSAEPIPE + #define WSAEPIPE -12345 + #endif + #define SOCKET_EWOULDBLOCK WSAEWOULDBLOCK + #define SOCKET_EAGAIN WSAEWOULDBLOCK + #define SOCKET_ECONNRESET WSAECONNRESET + #define SOCKET_EINTR WSAEINTR + #define SOCKET_EPIPE WSAEPIPE +#else + #define SOCKET_EWOULDBLOCK EWOULDBLOCK + #define SOCKET_EAGAIN EAGAIN + #define SOCKET_ECONNRESET ECONNRESET + #define SOCKET_EINTR EINTR + #define SOCKET_EPIPE EPIPE +#endif /* USE_WINDOWS_API */ + + +#ifdef DEVKITPRO + /* from network.h */ + int net_send(int, const void*, int, unsigned int); + int net_recv(int, void*, int, unsigned int); + #define SEND_FUNCTION net_send + #define RECV_FUNCTION net_recv +#else + #define SEND_FUNCTION send + #define RECV_FUNCTION recv +#endif + + +static INLINE int LastError(void) +{ +#ifdef USE_WINDOWS_API + return WSAGetLastError(); +#else + return errno; +#endif +} + +/* The receive embedded callback + * return : nb bytes read, or error + */ +int EmbedReceive(char *buf, int sz, void *ctx) +{ + int recvd; + int err; + int socket = *(int*)ctx; + + recvd = RECV_FUNCTION(socket, (char *)buf, sz, 0); + + if (recvd == -1) { + err = LastError(); + if (err == SOCKET_EWOULDBLOCK || + err == SOCKET_EAGAIN) + return IO_ERR_WANT_READ; + + else if (err == SOCKET_ECONNRESET) + return IO_ERR_CONN_RST; + + else if (err == SOCKET_EINTR) + return IO_ERR_ISR; + + else + return IO_ERR_GENERAL; + } + else if (recvd == 0) + return IO_ERR_CONN_CLOSE; + + return recvd; +} + +/* The send embedded callback + * return : nb bytes sent, or error + */ +int EmbedSend(char *buf, int sz, void *ctx) +{ + int socket = *(int*)ctx; + int sent; + int len = sz; + + sent = SEND_FUNCTION(socket, &buf[sz - len], len, 0); + + if (sent == -1) { + if (LastError() == SOCKET_EWOULDBLOCK || + LastError() == SOCKET_EAGAIN) + return IO_ERR_WANT_WRITE; + + else if (LastError() == SOCKET_ECONNRESET) + return IO_ERR_CONN_RST; + + else if (LastError() == SOCKET_EINTR) + return IO_ERR_ISR; + + else if (LastError() == SOCKET_EPIPE) + return IO_ERR_CONN_CLOSE; + + else + return IO_ERR_GENERAL; + } + + return sent; +} + + +#endif /* CYASSL_USER_IO */ + +void CyaSSL_SetIORecv(SSL_CTX *ctx, CallbackIORecv CBIORecv) +{ + ctx->CBIORecv = CBIORecv; +} + + +void CyaSSL_SetIOSend(SSL_CTX *ctx, CallbackIOSend CBIOSend) +{ + ctx->CBIOSend = CBIOSend; +} + + +void CyaSSL_SetIOReadCtx(SSL* ssl, void *rctx) +{ + ssl->IOCB_ReadCtx = rctx; +} + + +void CyaSSL_SetIOWriteCtx(SSL* ssl, void *wctx) +{ + ssl->IOCB_WriteCtx = wctx; +} + |