gnutls.c
上传用户:kjfoods
上传日期:2020-07-06
资源大小:29949k
文件大小:32k
源码类别:

midi

开发平台:

Unix_Linux

  1. /*****************************************************************************
  2.  * gnutls.c
  3.  *****************************************************************************
  4.  * Copyright (C) 2004-2006 Rémi Denis-Courmont
  5.  * $Id: a718278cb742a675042461c613c95b9ca5e1447e $
  6.  *
  7.  * Authors: Rémi Denis-Courmont <rem # videolan.org>
  8.  *
  9.  * This program is free software; you can redistribute it and/or modify
  10.  * it under the terms of the GNU General Public License as published by
  11.  * the Free Software Foundation; either version 2 of the License, or
  12.  * (at your option) any later version.
  13.  *
  14.  * This program is distributed in the hope that it will be useful,
  15.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17.  * GNU General Public License for more details.
  18.  *
  19.  * You should have received a copy of the GNU General Public License
  20.  * along with this program; if not, write to the Free Software
  21.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22.  *****************************************************************************/
  23. /*****************************************************************************
  24.  * Preamble
  25.  *****************************************************************************/
  26. #ifdef HAVE_CONFIG_H
  27. # include "config.h"
  28. #endif
  29. #include <vlc_common.h>
  30. #include <vlc_plugin.h>
  31. #include <errno.h>
  32. #include <time.h>
  33. #include <sys/types.h>
  34. #include <errno.h>
  35. #ifdef HAVE_DIRENT_H
  36. # include <dirent.h>
  37. #endif
  38. #ifdef HAVE_SYS_STAT_H
  39. # include <sys/stat.h>
  40. #endif
  41. #ifdef WIN32
  42. # include <io.h>
  43. #else
  44. # include <unistd.h>
  45. #endif
  46. # include <fcntl.h>
  47. #include <vlc_tls.h>
  48. #include <vlc_charset.h>
  49. #include <vlc_block.h>
  50. #include <gcrypt.h>
  51. #include <gnutls/gnutls.h>
  52. #include <gnutls/x509.h>
  53. #include <vlc_gcrypt.h>
  54. #define CACHE_TIMEOUT     3600
  55. #define CACHE_SIZE          64
  56. #include "dhparams.h"
  57. #include <assert.h>
  58. /*****************************************************************************
  59.  * Module descriptor
  60.  *****************************************************************************/
  61. static int  OpenClient  (vlc_object_t *);
  62. static void CloseClient (vlc_object_t *);
  63. static int  OpenServer  (vlc_object_t *);
  64. static void CloseServer (vlc_object_t *);
  65. #define CACHE_TIMEOUT_TEXT N_("Expiration time for resumed TLS sessions")
  66. #define CACHE_TIMEOUT_LONGTEXT N_( 
  67.     "It is possible to cache the resumed TLS sessions. This is the expiration "
  68.     "time of the sessions stored in this cache, in seconds." )
  69. #define CACHE_SIZE_TEXT N_("Number of resumed TLS sessions")
  70. #define CACHE_SIZE_LONGTEXT N_( 
  71.     "This is the maximum number of resumed TLS sessions that " 
  72.     "the cache will hold." )
  73. vlc_module_begin ()
  74.     set_shortname( "GnuTLS" )
  75.     set_description( N_("GnuTLS transport layer security") )
  76.     set_capability( "tls client", 1 )
  77.     set_callbacks( OpenClient, CloseClient )
  78.     set_category( CAT_ADVANCED )
  79.     set_subcategory( SUBCAT_ADVANCED_MISC )
  80.     add_obsolete_bool( "tls-check-cert" )
  81.     add_obsolete_bool( "tls-check-hostname" )
  82.     add_submodule ()
  83.         set_description( N_("GnuTLS server") )
  84.         set_capability( "tls server", 1 )
  85.         set_category( CAT_ADVANCED )
  86.         set_subcategory( SUBCAT_ADVANCED_MISC )
  87.         set_callbacks( OpenServer, CloseServer )
  88.         add_obsolete_integer( "gnutls-dh-bits" )
  89.         add_integer( "gnutls-cache-timeout", CACHE_TIMEOUT, NULL,
  90.                     CACHE_TIMEOUT_TEXT, CACHE_TIMEOUT_LONGTEXT, true )
  91.         add_integer( "gnutls-cache-size", CACHE_SIZE, NULL, CACHE_SIZE_TEXT,
  92.                     CACHE_SIZE_LONGTEXT, true )
  93. vlc_module_end ()
  94. static vlc_mutex_t gnutls_mutex = VLC_STATIC_MUTEX;
  95. /**
  96.  * Initializes GnuTLS with proper locking.
  97.  * @return VLC_SUCCESS on success, a VLC error code otherwise.
  98.  */
  99. static int gnutls_Init (vlc_object_t *p_this)
  100. {
  101.     int ret = VLC_EGENERIC;
  102.     vlc_gcrypt_init (); /* GnuTLS depends on gcrypt */
  103.     vlc_mutex_lock (&gnutls_mutex);
  104.     if (gnutls_global_init ())
  105.     {
  106.         msg_Err (p_this, "cannot initialize GnuTLS");
  107.         goto error;
  108.     }
  109.     const char *psz_version = gnutls_check_version ("1.3.3");
  110.     if (psz_version == NULL)
  111.     {
  112.         msg_Err (p_this, "unsupported GnuTLS version");
  113.         gnutls_global_deinit ();
  114.         goto error;
  115.     }
  116.     msg_Dbg (p_this, "GnuTLS v%s initialized", psz_version);
  117.     ret = VLC_SUCCESS;
  118. error:
  119.     vlc_mutex_unlock (&gnutls_mutex);
  120.     return ret;
  121. }
  122. /**
  123.  * Deinitializes GnuTLS.
  124.  */
  125. static void gnutls_Deinit (vlc_object_t *p_this)
  126. {
  127.     vlc_mutex_lock (&gnutls_mutex);
  128.     gnutls_global_deinit ();
  129.     msg_Dbg (p_this, "GnuTLS deinitialized");
  130.     vlc_mutex_unlock (&gnutls_mutex);
  131. }
  132. static int gnutls_Error (vlc_object_t *obj, int val)
  133. {
  134.     switch (val)
  135.     {
  136.         case GNUTLS_E_AGAIN:
  137. #ifndef WIN32
  138.             errno = EAGAIN;
  139.             break;
  140. #endif
  141.             /* WinSock does not return EAGAIN, return EINTR instead */
  142.         case GNUTLS_E_INTERRUPTED:
  143. #ifdef WIN32
  144.             WSASetLastError (WSAEINTR);
  145. #else
  146.             errno = EINTR;
  147. #endif
  148.             break;
  149.         default:
  150.             msg_Err (obj, "%s", gnutls_strerror (val));
  151. #ifndef NDEBUG
  152.             if (!gnutls_error_is_fatal (val))
  153.                 msg_Err (obj, "Error above should be handled");
  154. #endif
  155. #ifdef WIN32
  156.             WSASetLastError (WSAECONNRESET);
  157. #else
  158.             errno = ECONNRESET;
  159. #endif
  160.     }
  161.     return -1;
  162. }
  163. struct tls_session_sys_t
  164. {
  165.     gnutls_session_t session;
  166.     char            *psz_hostname;
  167.     bool       b_handshaked;
  168. };
  169. /**
  170.  * Sends data through a TLS session.
  171.  */
  172. static int
  173. gnutls_Send( void *p_session, const void *buf, int i_length )
  174. {
  175.     int val;
  176.     tls_session_sys_t *p_sys;
  177.     p_sys = (tls_session_sys_t *)(((tls_session_t *)p_session)->p_sys);
  178.     val = gnutls_record_send( p_sys->session, buf, i_length );
  179.     return (val < 0) ? gnutls_Error ((vlc_object_t *)p_session, val) : val;
  180. }
  181. /**
  182.  * Receives data through a TLS session.
  183.  */
  184. static int
  185. gnutls_Recv( void *p_session, void *buf, int i_length )
  186. {
  187.     int val;
  188.     tls_session_sys_t *p_sys;
  189.     p_sys = (tls_session_sys_t *)(((tls_session_t *)p_session)->p_sys);
  190.     val = gnutls_record_recv( p_sys->session, buf, i_length );
  191.     return (val < 0) ? gnutls_Error ((vlc_object_t *)p_session, val) : val;
  192. }
  193. /**
  194.  * Starts or continues the TLS handshake.
  195.  *
  196.  * @return -1 on fatal error, 0 on succesful handshake completion,
  197.  * 1 if more would-be blocking recv is needed,
  198.  * 2 if more would-be blocking send is required.
  199.  */
  200. static int
  201. gnutls_ContinueHandshake (tls_session_t *p_session)
  202. {
  203.     tls_session_sys_t *p_sys = p_session->p_sys;
  204.     int val;
  205. #ifdef WIN32
  206.     WSASetLastError( 0 );
  207. #endif
  208.     val = gnutls_handshake( p_sys->session );
  209.     if( ( val == GNUTLS_E_AGAIN ) || ( val == GNUTLS_E_INTERRUPTED ) )
  210.         return 1 + gnutls_record_get_direction( p_sys->session );
  211.     if( val < 0 )
  212.     {
  213. #ifdef WIN32
  214.         msg_Dbg( p_session, "Winsock error %d", WSAGetLastError( ) );
  215. #endif
  216.         msg_Err( p_session, "TLS handshake error: %s",
  217.                  gnutls_strerror( val ) );
  218.         return -1;
  219.     }
  220.     p_sys->b_handshaked = true;
  221.     return 0;
  222. }
  223. typedef struct
  224. {
  225.     int flag;
  226.     const char *msg;
  227. } error_msg_t;
  228. static const error_msg_t cert_errors[] =
  229. {
  230.     { GNUTLS_CERT_INVALID,
  231.         "Certificate could not be verified" },
  232.     { GNUTLS_CERT_REVOKED,
  233.         "Certificate was revoked" },
  234.     { GNUTLS_CERT_SIGNER_NOT_FOUND,
  235.         "Certificate's signer was not found" },
  236.     { GNUTLS_CERT_SIGNER_NOT_CA,
  237.         "Certificate's signer is not a CA" },
  238.     { GNUTLS_CERT_INSECURE_ALGORITHM,
  239.         "Insecure certificate signature algorithm" },
  240.     { 0, NULL }
  241. };
  242. static int
  243. gnutls_HandshakeAndValidate( tls_session_t *session )
  244. {
  245.     int val = gnutls_ContinueHandshake( session );
  246.     if( val )
  247.         return val;
  248.     tls_session_sys_t *p_sys = (tls_session_sys_t *)(session->p_sys);
  249.     /* certificates chain verification */
  250.     unsigned status;
  251.     val = gnutls_certificate_verify_peers2( p_sys->session, &status );
  252.     if( val )
  253.     {
  254.         msg_Err( session, "Certificate verification failed: %s",
  255.                  gnutls_strerror( val ) );
  256.         return -1;
  257.     }
  258.     if( status )
  259.     {
  260.         msg_Err( session, "TLS session: access denied" );
  261.         for( const error_msg_t *e = cert_errors; e->flag; e++ )
  262.         {
  263.             if( status & e->flag )
  264.             {
  265.                 msg_Err( session, "%s", e->msg );
  266.                 status &= ~e->flag;
  267.             }
  268.         }
  269.         if( status )
  270.             msg_Err( session,
  271.                      "unknown certificate error (you found a bug in VLC)" );
  272.         return -1;
  273.     }
  274.     /* certificate (host)name verification */
  275.     const gnutls_datum_t *data;
  276.     data = gnutls_certificate_get_peers (p_sys->session, &(unsigned){0});
  277.     if( data == NULL )
  278.     {
  279.         msg_Err( session, "Peer certificate not available" );
  280.         return -1;
  281.     }
  282.     gnutls_x509_crt_t cert;
  283.     val = gnutls_x509_crt_init( &cert );
  284.     if( val )
  285.     {
  286.         msg_Err( session, "x509 fatal error: %s", gnutls_strerror( val ) );
  287.         return -1;
  288.     }
  289.     val = gnutls_x509_crt_import( cert, data, GNUTLS_X509_FMT_DER );
  290.     if( val )
  291.     {
  292.         msg_Err( session, "Certificate import error: %s",
  293.                  gnutls_strerror( val ) );
  294.         goto error;
  295.     }
  296.     assert( p_sys->psz_hostname != NULL );
  297.     if ( !gnutls_x509_crt_check_hostname( cert, p_sys->psz_hostname ) )
  298.     {
  299.         msg_Err( session, "Certificate does not match "%s"",
  300.                  p_sys->psz_hostname );
  301.         goto error;
  302.     }
  303.     if( gnutls_x509_crt_get_expiration_time( cert ) < time( NULL ) )
  304.     {
  305.         msg_Err( session, "Certificate expired" );
  306.         goto error;
  307.     }
  308.     if( gnutls_x509_crt_get_activation_time( cert ) > time ( NULL ) )
  309.     {
  310.         msg_Err( session, "Certificate not yet valid" );
  311.         goto error;
  312.     }
  313.     gnutls_x509_crt_deinit( cert );
  314.     msg_Dbg( session, "TLS/x509 certificate verified" );
  315.     return 0;
  316. error:
  317.     gnutls_x509_crt_deinit( cert );
  318.     return -1;
  319. }
  320. /**
  321.  * Sets the operating system file descriptor backend for the TLS sesison.
  322.  *
  323.  * @param fd stream socket already connected with the peer.
  324.  */
  325. static void
  326. gnutls_SetFD (tls_session_t *p_session, int fd)
  327. {
  328.     gnutls_transport_set_ptr (p_session->p_sys->session,
  329.                               (gnutls_transport_ptr_t)(intptr_t)fd);
  330. }
  331. typedef int (*tls_prio_func) (gnutls_session_t, const int *);
  332. static int
  333. gnutls_SetPriority (vlc_object_t *restrict obj, const char *restrict name,
  334.                     tls_prio_func func, gnutls_session_t session,
  335.                     const int *restrict values)
  336. {
  337.     int val = func (session, values);
  338.     if (val < 0)
  339.     {
  340.         msg_Err (obj, "cannot set %s priorities: %s", name,
  341.                  gnutls_strerror (val));
  342.         return VLC_EGENERIC;
  343.     }
  344.     return VLC_SUCCESS;
  345. }
  346. static int
  347. gnutls_SessionPrioritize (vlc_object_t *obj, gnutls_session_t session)
  348. {
  349.     /* Note that ordering matters (on the client side) */
  350.     static const int protos[] =
  351.     {
  352.         /*GNUTLS_TLS1_2, as of GnuTLS 2.6.5, still not ratified */
  353.         GNUTLS_TLS1_1,
  354.         GNUTLS_TLS1_0,
  355.         GNUTLS_SSL3,
  356.         0
  357.     };
  358.     static const int comps[] =
  359.     {
  360.         GNUTLS_COMP_DEFLATE,
  361.         GNUTLS_COMP_NULL,
  362.         0
  363.     };
  364.     static const int macs[] =
  365.     {
  366.         GNUTLS_MAC_SHA512,
  367.         GNUTLS_MAC_SHA384,
  368.         GNUTLS_MAC_SHA256,
  369.         GNUTLS_MAC_SHA1,
  370.         GNUTLS_MAC_RMD160, // RIPEMD
  371.         GNUTLS_MAC_MD5,
  372.         //GNUTLS_MAC_MD2,
  373.         //GNUTLS_MAC_NULL,
  374.         0
  375.     };
  376.     static const int ciphers[] =
  377.     {
  378.         GNUTLS_CIPHER_AES_256_CBC,
  379.         GNUTLS_CIPHER_AES_128_CBC,
  380.         GNUTLS_CIPHER_3DES_CBC,
  381.         GNUTLS_CIPHER_ARCFOUR_128,
  382.         // TODO? Camellia ciphers?
  383.         //GNUTLS_CIPHER_DES_CBC,
  384.         //GNUTLS_CIPHER_ARCFOUR_40,
  385.         //GNUTLS_CIPHER_RC2_40_CBC,
  386.         //GNUTLS_CIPHER_NULL,
  387.         0
  388.     };
  389.     static const int kx[] =
  390.     {
  391.         GNUTLS_KX_DHE_RSA,
  392.         GNUTLS_KX_DHE_DSS,
  393.         GNUTLS_KX_RSA,
  394.         //GNUTLS_KX_RSA_EXPORT,
  395.         //GNUTLS_KX_DHE_PSK, TODO
  396.         //GNUTLS_KX_PSK,     TODO
  397.         //GNUTLS_KX_SRP_RSA, TODO
  398.         //GNUTLS_KX_SRP_DSS, TODO
  399.         //GNUTLS_KX_SRP,     TODO
  400.         //GNUTLS_KX_ANON_DH,
  401.         0
  402.     };
  403.     static const int cert_types[] =
  404.     {
  405.         GNUTLS_CRT_X509,
  406.         //GNUTLS_CRT_OPENPGP, TODO
  407.         0
  408.     };
  409.     int val = gnutls_set_default_priority (session);
  410.     if (val < 0)
  411.     {
  412.         msg_Err (obj, "cannot set default TLS priorities: %s",
  413.                  gnutls_strerror (val));
  414.         return VLC_EGENERIC;
  415.     }
  416.     if (gnutls_SetPriority (obj, "protocols",
  417.                             gnutls_protocol_set_priority, session, protos)
  418.      || gnutls_SetPriority (obj, "compression algorithms",
  419.                             gnutls_compression_set_priority, session, comps)
  420.      || gnutls_SetPriority (obj, "MAC algorithms",
  421.                             gnutls_mac_set_priority, session, macs)
  422.      || gnutls_SetPriority (obj, "ciphers",
  423.                             gnutls_cipher_set_priority, session, ciphers)
  424.      || gnutls_SetPriority (obj, "key exchange algorithms",
  425.                             gnutls_kx_set_priority, session, kx)
  426.      || gnutls_SetPriority (obj, "certificate types",
  427.                             gnutls_certificate_type_set_priority, session,
  428.                             cert_types))
  429.         return VLC_EGENERIC;
  430.     return VLC_SUCCESS;
  431. }
  432. static int
  433. gnutls_Addx509File( vlc_object_t *p_this,
  434.                     gnutls_certificate_credentials_t cred,
  435.                     const char *psz_path, bool b_priv );
  436. static int
  437. gnutls_Addx509Directory( vlc_object_t *p_this,
  438.                          gnutls_certificate_credentials_t cred,
  439.                          const char *psz_dirname,
  440.                          bool b_priv )
  441. {
  442.     DIR* dir;
  443.     if( *psz_dirname == '' )
  444.         psz_dirname = ".";
  445.     dir = utf8_opendir( psz_dirname );
  446.     if( dir == NULL )
  447.     {
  448.         if (errno != ENOENT)
  449.         {
  450.             msg_Err (p_this, "cannot open directory (%s): %m", psz_dirname);
  451.             return VLC_EGENERIC;
  452.         }
  453.         msg_Dbg (p_this, "creating empty certificate directory: %s",
  454.                  psz_dirname);
  455.         utf8_mkdir (psz_dirname, b_priv ? 0700 : 0755);
  456.         return VLC_SUCCESS;
  457.     }
  458. #ifdef S_ISLNK
  459.     else
  460.     {
  461.         struct stat st1, st2;
  462.         int fd = dirfd( dir );
  463.         /*
  464.          * Gets stats for the directory path, checks that it is not a
  465.          * symbolic link (to avoid possibly infinite recursion), and verifies
  466.          * that the inode is still the same, to avoid TOCTOU race condition.
  467.          */
  468.         if( ( fd == -1)
  469.          || fstat( fd, &st1 ) || utf8_lstat( psz_dirname, &st2 )
  470.          || S_ISLNK( st2.st_mode ) || ( st1.st_ino != st2.st_ino ) )
  471.         {
  472.             closedir( dir );
  473.             return VLC_EGENERIC;
  474.         }
  475.     }
  476. #endif
  477.     for (;;)
  478.     {
  479.         char *ent = utf8_readdir (dir);
  480.         if (ent == NULL)
  481.             break;
  482.         if ((strcmp (ent, ".") == 0) || (strcmp (ent, "..") == 0))
  483.         {
  484.             free( ent );
  485.             continue;
  486.         }
  487.         char path[strlen (psz_dirname) + strlen (ent) + 2];
  488.         sprintf (path, "%s"DIR_SEP"%s", psz_dirname, ent);
  489.         free (ent);
  490.         gnutls_Addx509File( p_this, cred, path, b_priv );
  491.     }
  492.     closedir( dir );
  493.     return VLC_SUCCESS;
  494. }
  495. static int
  496. gnutls_Addx509File( vlc_object_t *p_this,
  497.                     gnutls_certificate_credentials cred,
  498.                     const char *psz_path, bool b_priv )
  499. {
  500.     struct stat st;
  501.     int fd = utf8_open (psz_path, O_RDONLY, 0);
  502.     if (fd == -1)
  503.         goto error;
  504.     block_t *block = block_File (fd);
  505.     if (block != NULL)
  506.     {
  507.         close (fd);
  508.         gnutls_datum data = {
  509.             .data = block->p_buffer,
  510.             .size = block->i_buffer,
  511.         };
  512.         int res = b_priv
  513.             ? gnutls_certificate_set_x509_key_mem (cred, &data, &data,
  514.                                                    GNUTLS_X509_FMT_PEM)
  515.             : gnutls_certificate_set_x509_trust_mem (cred, &data,
  516.                                                      GNUTLS_X509_FMT_PEM);
  517.         block_Release (block);
  518.         if (res < 0)
  519.         {
  520.             msg_Warn (p_this, "cannot add x509 credentials (%s): %s",
  521.                       psz_path, gnutls_strerror (res));
  522.             return VLC_EGENERIC;
  523.         }
  524.         msg_Dbg (p_this, "added x509 credentials (%s)", psz_path);
  525.         return VLC_SUCCESS;
  526.     }
  527.     if (!fstat (fd, &st) && S_ISDIR (st.st_mode))
  528.     {
  529.         close (fd);
  530.         msg_Dbg (p_this, "looking recursively for x509 credentials in %s",
  531.                  psz_path);
  532.         return gnutls_Addx509Directory (p_this, cred, psz_path, b_priv);
  533.     }
  534. error:
  535.     msg_Warn (p_this, "cannot add x509 credentials (%s): %m", psz_path);
  536.     if (fd != -1)
  537.         close (fd);
  538.     return VLC_EGENERIC;
  539. }
  540. /** TLS client session data */
  541. typedef struct tls_client_sys_t
  542. {
  543.     struct tls_session_sys_t         session;
  544.     gnutls_certificate_credentials_t x509_cred;
  545. } tls_client_sys_t;
  546. /**
  547.  * Initializes a client-side TLS session.
  548.  */
  549. static int OpenClient (vlc_object_t *obj)
  550. {
  551.     tls_session_t *p_session = (tls_session_t *)obj;
  552.     int i_val;
  553.     if (gnutls_Init (obj))
  554.         return VLC_EGENERIC;
  555.     tls_client_sys_t *p_sys = malloc (sizeof (*p_sys));
  556.     if (p_sys == NULL)
  557.     {
  558.         gnutls_Deinit (obj);
  559.         return VLC_ENOMEM;
  560.     }
  561.     p_session->p_sys = &p_sys->session;
  562.     p_session->sock.p_sys = p_session;
  563.     p_session->sock.pf_send = gnutls_Send;
  564.     p_session->sock.pf_recv = gnutls_Recv;
  565.     p_session->pf_set_fd = gnutls_SetFD;
  566.     p_sys->session.b_handshaked = false;
  567.     i_val = gnutls_certificate_allocate_credentials (&p_sys->x509_cred);
  568.     if (i_val != 0)
  569.     {
  570.         msg_Err (obj, "cannot allocate X509 credentials: %s",
  571.                  gnutls_strerror (i_val));
  572.         goto error;
  573.     }
  574.     char *userdir = config_GetUserDataDir ();
  575.     if (userdir != NULL)
  576.     {
  577.         char path[strlen (userdir) + sizeof ("/ssl/private")];
  578.         sprintf (path, "%s/ssl", userdir);
  579.         utf8_mkdir (path, 0755);
  580.         sprintf (path, "%s/ssl/certs", userdir);
  581.         gnutls_Addx509Directory (VLC_OBJECT (p_session),
  582.                                  p_sys->x509_cred, path, false);
  583.         sprintf (path, "%s/ssl/private", userdir);
  584.         gnutls_Addx509Directory (VLC_OBJECT (p_session), p_sys->x509_cred,
  585.                                  path, true);
  586.         free (userdir);
  587.     }
  588.     const char *confdir = config_GetConfDir ();
  589.     {
  590.         char path[strlen (confdir)
  591.                    + sizeof ("/ssl/certs/ca-certificates.crt")];
  592.         sprintf (path, "%s/ssl/certs/ca-certificates.crt", confdir);
  593.         gnutls_Addx509File (VLC_OBJECT (p_session),
  594.                             p_sys->x509_cred, path, false);
  595.     }
  596.     p_session->pf_handshake = gnutls_HandshakeAndValidate;
  597.     /*p_session->pf_handshake = gnutls_ContinueHandshake;*/
  598.     i_val = gnutls_init (&p_sys->session.session, GNUTLS_CLIENT);
  599.     if (i_val != 0)
  600.     {
  601.         msg_Err (obj, "cannot initialize TLS session: %s",
  602.                  gnutls_strerror (i_val));
  603.         gnutls_certificate_free_credentials (p_sys->x509_cred);
  604.         goto error;
  605.     }
  606.     if (gnutls_SessionPrioritize (VLC_OBJECT (p_session),
  607.                                   p_sys->session.session))
  608.         goto s_error;
  609.     /* minimum DH prime bits */
  610.     gnutls_dh_set_prime_bits (p_sys->session.session, 1024);
  611.     i_val = gnutls_credentials_set (p_sys->session.session,
  612.                                     GNUTLS_CRD_CERTIFICATE,
  613.                                     p_sys->x509_cred);
  614.     if (i_val < 0)
  615.     {
  616.         msg_Err (obj, "cannot set TLS session credentials: %s",
  617.                  gnutls_strerror (i_val));
  618.         goto s_error;
  619.     }
  620.     char *servername = var_GetNonEmptyString (p_session, "tls-server-name");
  621.     if (servername == NULL )
  622.         msg_Err (p_session, "server name missing for TLS session");
  623.     else
  624.         gnutls_server_name_set (p_sys->session.session, GNUTLS_NAME_DNS,
  625.                                 servername, strlen (servername));
  626.     p_sys->session.psz_hostname = servername;
  627.     return VLC_SUCCESS;
  628. s_error:
  629.     gnutls_deinit (p_sys->session.session);
  630.     gnutls_certificate_free_credentials (p_sys->x509_cred);
  631. error:
  632.     gnutls_Deinit (obj);
  633.     free (p_sys);
  634.     return VLC_EGENERIC;
  635. }
  636. static void CloseClient (vlc_object_t *obj)
  637. {
  638.     tls_session_t *client = (tls_session_t *)obj;
  639.     tls_client_sys_t *p_sys = (tls_client_sys_t *)(client->p_sys);
  640.     if (p_sys->session.b_handshaked == true)
  641.         gnutls_bye (p_sys->session.session, GNUTLS_SHUT_WR);
  642.     gnutls_deinit (p_sys->session.session);
  643.     /* credentials must be free'd *after* gnutls_deinit() */
  644.     gnutls_certificate_free_credentials (p_sys->x509_cred);
  645.     gnutls_Deinit (obj);
  646.     free (p_sys->session.psz_hostname);
  647.     free (p_sys);
  648. }
  649. /**
  650.  * Server-side TLS
  651.  */
  652. struct tls_server_sys_t
  653. {
  654.     gnutls_certificate_credentials_t x509_cred;
  655.     gnutls_dh_params_t               dh_params;
  656.     struct saved_session_t          *p_cache;
  657.     struct saved_session_t          *p_store;
  658.     int                              i_cache_size;
  659.     vlc_mutex_t                      cache_lock;
  660.     int                            (*pf_handshake) (tls_session_t *);
  661. };
  662. /**
  663.  * TLS session resumption callbacks (server-side)
  664.  */
  665. #define MAX_SESSION_ID    32
  666. #define MAX_SESSION_DATA  1024
  667. typedef struct saved_session_t
  668. {
  669.     char id[MAX_SESSION_ID];
  670.     char data[MAX_SESSION_DATA];
  671.     unsigned i_idlen;
  672.     unsigned i_datalen;
  673. } saved_session_t;
  674. static int cb_store( void *p_server, gnutls_datum key, gnutls_datum data )
  675. {
  676.     tls_server_sys_t *p_sys = ((tls_server_t *)p_server)->p_sys;
  677.     if( ( p_sys->i_cache_size == 0 )
  678.      || ( key.size > MAX_SESSION_ID )
  679.      || ( data.size > MAX_SESSION_DATA ) )
  680.         return -1;
  681.     vlc_mutex_lock( &p_sys->cache_lock );
  682.     memcpy( p_sys->p_store->id, key.data, key.size);
  683.     memcpy( p_sys->p_store->data, data.data, data.size );
  684.     p_sys->p_store->i_idlen = key.size;
  685.     p_sys->p_store->i_datalen = data.size;
  686.     p_sys->p_store++;
  687.     if( ( p_sys->p_store - p_sys->p_cache ) == p_sys->i_cache_size )
  688.         p_sys->p_store = p_sys->p_cache;
  689.     vlc_mutex_unlock( &p_sys->cache_lock );
  690.     return 0;
  691. }
  692. static gnutls_datum cb_fetch( void *p_server, gnutls_datum key )
  693. {
  694.     static const gnutls_datum_t err_datum = { NULL, 0 };
  695.     tls_server_sys_t *p_sys = ((tls_server_t *)p_server)->p_sys;
  696.     saved_session_t *p_session, *p_end;
  697.     p_session = p_sys->p_cache;
  698.     p_end = p_session + p_sys->i_cache_size;
  699.     vlc_mutex_lock( &p_sys->cache_lock );
  700.     while( p_session < p_end )
  701.     {
  702.         if( ( p_session->i_idlen == key.size )
  703.          && !memcmp( p_session->id, key.data, key.size ) )
  704.         {
  705.             gnutls_datum_t data;
  706.             data.size = p_session->i_datalen;
  707.             data.data = gnutls_malloc( data.size );
  708.             if( data.data == NULL )
  709.             {
  710.                 vlc_mutex_unlock( &p_sys->cache_lock );
  711.                 return err_datum;
  712.             }
  713.             memcpy( data.data, p_session->data, data.size );
  714.             vlc_mutex_unlock( &p_sys->cache_lock );
  715.             return data;
  716.         }
  717.         p_session++;
  718.     }
  719.     vlc_mutex_unlock( &p_sys->cache_lock );
  720.     return err_datum;
  721. }
  722. static int cb_delete( void *p_server, gnutls_datum key )
  723. {
  724.     tls_server_sys_t *p_sys = ((tls_server_t *)p_server)->p_sys;
  725.     saved_session_t *p_session, *p_end;
  726.     p_session = p_sys->p_cache;
  727.     p_end = p_session + p_sys->i_cache_size;
  728.     vlc_mutex_lock( &p_sys->cache_lock );
  729.     while( p_session < p_end )
  730.     {
  731.         if( ( p_session->i_idlen == key.size )
  732.          && !memcmp( p_session->id, key.data, key.size ) )
  733.         {
  734.             p_session->i_datalen = p_session->i_idlen = 0;
  735.             vlc_mutex_unlock( &p_sys->cache_lock );
  736.             return 0;
  737.         }
  738.         p_session++;
  739.     }
  740.     vlc_mutex_unlock( &p_sys->cache_lock );
  741.     return -1;
  742. }
  743. /**
  744.  * Terminates TLS session and releases session data.
  745.  * You still have to close the socket yourself.
  746.  */
  747. static void
  748. gnutls_SessionClose (tls_server_t *p_server, tls_session_t *p_session)
  749. {
  750.     tls_session_sys_t *p_sys = p_session->p_sys;
  751.     (void)p_server;
  752.     if( p_sys->b_handshaked == true )
  753.         gnutls_bye( p_sys->session, GNUTLS_SHUT_WR );
  754.     gnutls_deinit( p_sys->session );
  755.     vlc_object_detach( p_session );
  756.     vlc_object_release( p_session );
  757.     free( p_sys );
  758. }
  759. /**
  760.  * Initializes a server-side TLS session.
  761.  */
  762. static tls_session_t *
  763. gnutls_ServerSessionPrepare( tls_server_t *p_server )
  764. {
  765.     tls_session_t *p_session;
  766.     tls_server_sys_t *p_server_sys;
  767.     gnutls_session_t session;
  768.     int i_val;
  769.     p_session = vlc_object_create( p_server, sizeof (struct tls_session_t) );
  770.     if( p_session == NULL )
  771.         return NULL;
  772.     p_session->p_sys = malloc( sizeof(struct tls_session_sys_t) );
  773.     if( p_session->p_sys == NULL )
  774.     {
  775.         vlc_object_release( p_session );
  776.         return NULL;
  777.     }
  778.     p_server_sys = p_server->p_sys;
  779.     p_session->sock.p_sys = p_session;
  780.     p_session->sock.pf_send = gnutls_Send;
  781.     p_session->sock.pf_recv = gnutls_Recv;
  782.     p_session->pf_set_fd = gnutls_SetFD;
  783.     p_session->pf_handshake = p_server_sys->pf_handshake;
  784.     p_session->p_sys->b_handshaked = false;
  785.     p_session->p_sys->psz_hostname = NULL;
  786.     i_val = gnutls_init( &session, GNUTLS_SERVER );
  787.     if( i_val != 0 )
  788.     {
  789.         msg_Err( p_server, "cannot initialize TLS session: %s",
  790.                  gnutls_strerror( i_val ) );
  791.         goto error;
  792.     }
  793.     p_session->p_sys->session = session;
  794.     if (gnutls_SessionPrioritize (VLC_OBJECT (p_session), session))
  795.     {
  796.         gnutls_deinit( session );
  797.         goto error;
  798.     }
  799.     i_val = gnutls_credentials_set( session, GNUTLS_CRD_CERTIFICATE,
  800.                                     p_server_sys->x509_cred );
  801.     if( i_val < 0 )
  802.     {
  803.         msg_Err( p_server, "cannot set TLS session credentials: %s",
  804.                  gnutls_strerror( i_val ) );
  805.         gnutls_deinit( session );
  806.         goto error;
  807.     }
  808.     if (p_session->pf_handshake == gnutls_HandshakeAndValidate)
  809.         gnutls_certificate_server_set_request (session, GNUTLS_CERT_REQUIRE);
  810.     /* Session resumption support */
  811.     i_val = config_GetInt (p_server, "gnutls-cache-timeout");
  812.     if (i_val >= 0)
  813.         gnutls_db_set_cache_expiration (session, i_val);
  814.     gnutls_db_set_retrieve_function( session, cb_fetch );
  815.     gnutls_db_set_remove_function( session, cb_delete );
  816.     gnutls_db_set_store_function( session, cb_store );
  817.     gnutls_db_set_ptr( session, p_server );
  818.     return p_session;
  819. error:
  820.     free( p_session->p_sys );
  821.     vlc_object_detach( p_session );
  822.     vlc_object_release( p_session );
  823.     return NULL;
  824. }
  825. /**
  826.  * Adds one or more certificate authorities.
  827.  *
  828.  * @param psz_ca_path (Unicode) path to an x509 certificates list.
  829.  *
  830.  * @return -1 on error, 0 on success.
  831.  */
  832. static int
  833. gnutls_ServerAddCA( tls_server_t *p_server, const char *psz_ca_path )
  834. {
  835.     tls_server_sys_t *p_sys;
  836.     char *psz_local_path;
  837.     int val;
  838.     p_sys = (tls_server_sys_t *)(p_server->p_sys);
  839.     psz_local_path = ToLocale( psz_ca_path );
  840.     val = gnutls_certificate_set_x509_trust_file( p_sys->x509_cred,
  841.                                                   psz_local_path,
  842.                                                   GNUTLS_X509_FMT_PEM );
  843.     LocaleFree( psz_local_path );
  844.     if( val < 0 )
  845.     {
  846.         msg_Err( p_server, "cannot add trusted CA (%s): %s", psz_ca_path,
  847.                  gnutls_strerror( val ) );
  848.         return VLC_EGENERIC;
  849.     }
  850.     msg_Dbg( p_server, " %d trusted CA added (%s)", val, psz_ca_path );
  851.     /* enables peer's certificate verification */
  852.     p_sys->pf_handshake = gnutls_HandshakeAndValidate;
  853.     return VLC_SUCCESS;
  854. }
  855. /**
  856.  * Adds a certificates revocation list to be sent to TLS clients.
  857.  *
  858.  * @param psz_crl_path (Unicode) path of the CRL file.
  859.  *
  860.  * @return -1 on error, 0 on success.
  861.  */
  862. static int
  863. gnutls_ServerAddCRL( tls_server_t *p_server, const char *psz_crl_path )
  864. {
  865.     int val;
  866.     char *psz_local_path = ToLocale( psz_crl_path );
  867.     val = gnutls_certificate_set_x509_crl_file( ((tls_server_sys_t *)
  868.                                                 (p_server->p_sys))->x509_cred,
  869.                                                 psz_local_path,
  870.                                                 GNUTLS_X509_FMT_PEM );
  871.     LocaleFree( psz_crl_path );
  872.     if( val < 0 )
  873.     {
  874.         msg_Err( p_server, "cannot add CRL (%s): %s", psz_crl_path,
  875.                  gnutls_strerror( val ) );
  876.         return VLC_EGENERIC;
  877.     }
  878.     msg_Dbg( p_server, "%d CRL added (%s)", val, psz_crl_path );
  879.     return VLC_SUCCESS;
  880. }
  881. /**
  882.  * Allocates a whole server's TLS credentials.
  883.  */
  884. static int OpenServer (vlc_object_t *obj)
  885. {
  886.     tls_server_t *p_server = (tls_server_t *)obj;
  887.     tls_server_sys_t *p_sys;
  888.     int val;
  889.     if (gnutls_Init (obj))
  890.         return VLC_EGENERIC;
  891.     msg_Dbg (obj, "creating TLS server");
  892.     p_sys = (tls_server_sys_t *)malloc( sizeof(struct tls_server_sys_t) );
  893.     if( p_sys == NULL )
  894.         return VLC_ENOMEM;
  895.     p_sys->i_cache_size = config_GetInt (obj, "gnutls-cache-size");
  896.     if (p_sys->i_cache_size == -1) /* Duh, config subsystem exploded?! */
  897.         p_sys->i_cache_size = 0;
  898.     p_sys->p_cache = calloc (p_sys->i_cache_size,
  899.                              sizeof (struct saved_session_t));
  900.     if (p_sys->p_cache == NULL)
  901.     {
  902.         free (p_sys);
  903.         return VLC_ENOMEM;
  904.     }
  905.     p_sys->p_store = p_sys->p_cache;
  906.     p_server->p_sys = p_sys;
  907.     p_server->pf_add_CA  = gnutls_ServerAddCA;
  908.     p_server->pf_add_CRL = gnutls_ServerAddCRL;
  909.     p_server->pf_open    = gnutls_ServerSessionPrepare;
  910.     p_server->pf_close   = gnutls_SessionClose;
  911.     /* No certificate validation by default */
  912.     p_sys->pf_handshake  = gnutls_ContinueHandshake;
  913.     vlc_mutex_init( &p_sys->cache_lock );
  914.     /* Sets server's credentials */
  915.     val = gnutls_certificate_allocate_credentials( &p_sys->x509_cred );
  916.     if( val != 0 )
  917.     {
  918.         msg_Err( p_server, "cannot allocate X509 credentials: %s",
  919.                  gnutls_strerror( val ) );
  920.         goto error;
  921.     }
  922.     char *psz_cert_path = var_GetNonEmptyString (obj, "tls-x509-cert");
  923.     char *psz_key_path = var_GetNonEmptyString (obj, "tls-x509-key");
  924.     const char *psz_local_cert = ToLocale (psz_cert_path);
  925.     const char *psz_local_key = ToLocale (psz_key_path);
  926.     val = gnutls_certificate_set_x509_key_file (p_sys->x509_cred,
  927.                                                 psz_local_cert, psz_local_key,
  928.                                                 GNUTLS_X509_FMT_PEM );
  929.     LocaleFree (psz_local_key);
  930.     free (psz_key_path);
  931.     LocaleFree (psz_local_cert);
  932.     free (psz_cert_path);
  933.     if( val < 0 )
  934.     {
  935.         msg_Err( p_server, "cannot set certificate chain or private key: %s",
  936.                  gnutls_strerror( val ) );
  937.         gnutls_certificate_free_credentials( p_sys->x509_cred );
  938.         goto error;
  939.     }
  940.     /* FIXME:
  941.      * - support other ciper suites
  942.      */
  943.     val = gnutls_dh_params_init (&p_sys->dh_params);
  944.     if (val >= 0)
  945.     {
  946.         const gnutls_datum_t data = {
  947.             .data = (unsigned char *)dh_params,
  948.             .size = sizeof (dh_params) - 1,
  949.         };
  950.         val = gnutls_dh_params_import_pkcs3 (p_sys->dh_params, &data,
  951.                                              GNUTLS_X509_FMT_PEM);
  952.         if (val == 0)
  953.             gnutls_certificate_set_dh_params (p_sys->x509_cred,
  954.                                               p_sys->dh_params);
  955.     }
  956.     if (val < 0)
  957.     {
  958.         msg_Err (p_server, "cannot initialize DHE cipher suites: %s",
  959.                  gnutls_strerror (val));
  960.     }
  961.     return VLC_SUCCESS;
  962. error:
  963.     vlc_mutex_destroy (&p_sys->cache_lock);
  964.     free (p_sys->p_cache);
  965.     free (p_sys);
  966.     return VLC_EGENERIC;
  967. }
  968. /**
  969.  * Destroys a TLS server object.
  970.  */
  971. static void CloseServer (vlc_object_t *p_server)
  972. {
  973.     tls_server_sys_t *p_sys = ((tls_server_t *)p_server)->p_sys;
  974.     vlc_mutex_destroy (&p_sys->cache_lock);
  975.     free (p_sys->p_cache);
  976.     /* all sessions depending on the server are now deinitialized */
  977.     gnutls_certificate_free_credentials (p_sys->x509_cred);
  978.     gnutls_dh_params_deinit (p_sys->dh_params);
  979.     free (p_sys);
  980.     gnutls_Deinit (p_server);
  981. }