1 #ifndef __RTMP_SYS_H__ 2 #define __RTMP_SYS_H__ 3 /* 4 * Copyright (C) 2010 Howard Chu 5 * 6 * This file is part of librtmp. 7 * 8 * librtmp is free software; you can redistribute it and/or modify 9 * it under the terms of the GNU Lesser General Public License as 10 * published by the Free Software Foundation; either version 2.1, 11 * or (at your option) any later version. 12 * 13 * librtmp is distributed in the hope that it will be useful, 14 * but WITHOUT ANY WARRANTY; without even the implied warranty of 15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 * GNU General Public License for more details. 17 * 18 * You should have received a copy of the GNU Lesser General Public License 19 * along with librtmp see the file COPYING. If not, write to 20 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, 21 * Boston, MA 02110-1301, USA. 22 * http://www.gnu.org/copyleft/lgpl.html 23 */ 24 25 #ifdef _WIN32 26 27 #include <winsock2.h> 28 #include <ws2tcpip.h> 29 30 #ifdef _MSC_VER /* MSVC */ 31 #define snprintf _snprintf 32 #define strcasecmp stricmp 33 #define strncasecmp strnicmp 34 #define vsnprintf _vsnprintf 35 #endif 36 37 #define GetSockError() WSAGetLastError() 38 #define SetSockError(e) WSASetLastError(e) 39 #define setsockopt(a,b,c,d,e) (setsockopt)(a,b,c,(const char *)d,(int)e) 40 #define EWOULDBLOCK WSAETIMEDOUT /* we don't use nonblocking, but we do use timeouts */ 41 #define sleep(n) Sleep(n*1000) 42 #define msleep(n) Sleep(n) 43 #define SET_RCVTIMEO(tv,s) int tv = s*1000 44 #else /* !_WIN32 */ 45 #include <sys/types.h> 46 #include <sys/socket.h> 47 #include <sys/times.h> 48 #include <netdb.h> 49 #include <unistd.h> 50 #include <netinet/in.h> 51 #include <netinet/tcp.h> 52 #include <arpa/inet.h> 53 #define GetSockError() errno 54 #define SetSockError(e) errno = e 55 #undef closesocket 56 #define closesocket(s) close(s) 57 #define msleep(n) usleep(n*1000) 58 #define SET_RCVTIMEO(tv,s) struct timeval tv = {s,0} 59 #endif 60 61 #include "rtmp.h" 62 63 #ifdef USE_POLARSSL 64 #include <polarssl/version.h> 65 #include <polarssl/net.h> 66 #include <polarssl/ssl.h> 67 #include <polarssl/havege.h> 68 #if POLARSSL_VERSION_NUMBER < 0x01010000 69 #define havege_random havege_rand 70 #endif 71 #if POLARSSL_VERSION_NUMBER >= 0x01020000 72 #define SSL_SET_SESSION(S,resume,timeout,ctx) ssl_set_session(S,ctx) 73 #else 74 #define SSL_SET_SESSION(S,resume,timeout,ctx) ssl_set_session(S,resume,timeout,ctx) 75 #endif 76 typedef struct tls_ctx { 77 havege_state hs; 78 ssl_session ssn; 79 } tls_ctx; 80 typedef struct tls_server_ctx { 81 havege_state *hs; 82 x509_cert cert; 83 rsa_context key; 84 ssl_session ssn; 85 const char *dhm_P, *dhm_G; 86 } tls_server_ctx; 87 88 #define TLS_CTX tls_ctx * 89 #define TLS_client(ctx,s) s = malloc(sizeof(ssl_context)); ssl_init(s);\ 90 ssl_set_endpoint(s, SSL_IS_CLIENT); ssl_set_authmode(s, SSL_VERIFY_NONE);\ 91 ssl_set_rng(s, havege_random, &ctx->hs);\ 92 ssl_set_ciphersuites(s, ssl_default_ciphersuites);\ 93 SSL_SET_SESSION(s, 1, 600, &ctx->ssn) 94 #define TLS_server(ctx,s) s = malloc(sizeof(ssl_context)); ssl_init(s);\ 95 ssl_set_endpoint(s, SSL_IS_SERVER); ssl_set_authmode(s, SSL_VERIFY_NONE);\ 96 ssl_set_rng(s, havege_random, ((tls_server_ctx*)ctx)->hs);\ 97 ssl_set_ciphersuites(s, ssl_default_ciphersuites);\ 98 SSL_SET_SESSION(s, 1, 600, &((tls_server_ctx*)ctx)->ssn);\ 99 ssl_set_own_cert(s, &((tls_server_ctx*)ctx)->cert, &((tls_server_ctx*)ctx)->key);\ 100 ssl_set_dh_param(s, ((tls_server_ctx*)ctx)->dhm_P, ((tls_server_ctx*)ctx)->dhm_G) 101 #define TLS_setfd(s,fd) ssl_set_bio(s, net_recv, &fd, net_send, &fd) 102 #define TLS_connect(s) ssl_handshake(s) 103 #define TLS_accept(s) ssl_handshake(s) 104 #define TLS_read(s,b,l) ssl_read(s,(unsigned char *)b,l) 105 #define TLS_write(s,b,l) ssl_write(s,(unsigned char *)b,l) 106 #define TLS_shutdown(s) ssl_close_notify(s) 107 #define TLS_close(s) ssl_free(s); free(s) 108 109 #elif defined(USE_GNUTLS) 110 #include <gnutls/gnutls.h> 111 typedef struct tls_ctx { 112 gnutls_certificate_credentials_t cred; 113 gnutls_priority_t prios; 114 } tls_ctx; 115 #define TLS_CTX tls_ctx * 116 #define TLS_client(ctx,s) gnutls_init((gnutls_session_t *)(&s), GNUTLS_CLIENT); gnutls_priority_set(s, ctx->prios); gnutls_credentials_set(s, GNUTLS_CRD_CERTIFICATE, ctx->cred) 117 #define TLS_server(ctx,s) gnutls_init((gnutls_session_t *)(&s), GNUTLS_SERVER); gnutls_priority_set_direct(s, "NORMAL", NULL); gnutls_credentials_set(s, GNUTLS_CRD_CERTIFICATE, ctx) 118 #define TLS_setfd(s,fd) gnutls_transport_set_ptr(s, (gnutls_transport_ptr_t)(long)fd) 119 #define TLS_connect(s) gnutls_handshake(s) 120 #define TLS_accept(s) gnutls_handshake(s) 121 #define TLS_read(s,b,l) gnutls_record_recv(s,b,l) 122 #define TLS_write(s,b,l) gnutls_record_send(s,b,l) 123 #define TLS_shutdown(s) gnutls_bye(s, GNUTLS_SHUT_RDWR) 124 #define TLS_close(s) gnutls_deinit(s) 125 126 #else /* USE_OPENSSL */ 127 #define TLS_CTX SSL_CTX * 128 #define TLS_client(ctx,s) s = SSL_new(ctx) 129 #define TLS_server(ctx,s) s = SSL_new(ctx) 130 #define TLS_setfd(s,fd) SSL_set_fd(s,fd) 131 #define TLS_connect(s) SSL_connect(s) 132 #define TLS_accept(s) SSL_accept(s) 133 #define TLS_read(s,b,l) SSL_read(s,b,l) 134 #define TLS_write(s,b,l) SSL_write(s,b,l) 135 #define TLS_shutdown(s) SSL_shutdown(s) 136 #define TLS_close(s) SSL_free(s) 137 138 #endif 139 #endif 140