librtmp/rtmp_sys.h

   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