ssl.c

   1 /*********************************************************************
   2  *
   3  * File        :  $Source: /cvsroot/ijbswa/current/ssl.c,v $
   4  *
   5  * Purpose     :  File with TLS/SSL extension. Contains methods for
   6  *                creating, using and closing TLS/SSL connections.
   7  *
   8  * Copyright   :  Written by and Copyright (c) 2017 Vaclav Svec. FIT CVUT.
   9  *                Copyright (C) 2018-2020 by Fabian Keil <fk@fabiankeil.de>
  10  *
  11  *                This program is free software; you can redistribute it
  12  *                and/or modify it under the terms of the GNU General
  13  *                Public License as published by the Free Software
  14  *                Foundation; either version 2 of the License, or (at
  15  *                your option) any later version.
  16  *
  17  *                This program is distributed in the hope that it will
  18  *                be useful, but WITHOUT ANY WARRANTY; without even the
  19  *                implied warranty of MERCHANTABILITY or FITNESS FOR A
  20  *                PARTICULAR PURPOSE.  See the GNU General Public
  21  *                License for more details.
  22  *
  23  *                The GNU General Public License should be included with
  24  *                this file.  If not, you can view it at
  25  *                http://www.gnu.org/copyleft/gpl.html
  26  *                or write to the Free Software Foundation, Inc., 59
  27  *                Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  28  *
  29  *********************************************************************/
  30
  31 #include <string.h>
  32 #include <unistd.h>
  33
  34 #if !defined(MBEDTLS_CONFIG_FILE)
  35 #  include "mbedtls/config.h"
  36 #else
  37 #  include MBEDTLS_CONFIG_FILE
  38 #endif
  39
  40 #include "mbedtls/md5.h"
  41 #include "mbedtls/pem.h"
  42 #include "mbedtls/base64.h"
  43 #include "mbedtls/error.h"
  44
  45 #include "config.h"
  46 #include "project.h"
  47 #include "miscutil.h"
  48 #include "errlog.h"
  49 #include "jcc.h"
  50 #include "ssl.h"
  51
  52
  53 /*
  54  * Macros for searching begin and end of certificates.
  55  * Necessary to convert structure mbedtls_x509_crt to crt file.
  56  */
  57 #define PEM_BEGIN_CRT     "-----BEGIN CERTIFICATE-----\n"
  58 #define PEM_END_CRT       "-----END CERTIFICATE-----\n"
  59
  60 /*
  61  * Macros for ssl.c
  62  */
  63 #define ERROR_BUF_SIZE                   1024              /* Size of buffer for error messages */
  64 #define CERTIFICATE_BUF_SIZE             16384             /* Size of buffer to save certificate. Value 4096 is mbedtls library buffer size for certificate in DER form */
  65 #define PRIVATE_KEY_BUF_SIZE             16000             /* Size of buffer to save private key. Value 16000 is taken from mbed TLS library examples. */
  66 #define RSA_KEY_PUBLIC_EXPONENT          65537             /* Public exponent for RSA private key generating */
  67 #define RSA_KEYSIZE                      2048              /* Size of generated RSA keys */
  68 #define GENERATED_CERT_VALID_FROM        "20100101000000"  /* Date and time, which will be set in generated certificates as parameter valid from */
  69 #define GENERATED_CERT_VALID_TO          "20401231235959"  /* Date and time, which will be set in generated certificates as parameter valid to */
  70 #define CERT_SIGNATURE_ALGORITHM         MBEDTLS_MD_SHA256 /* The MD algorithm to use for the signature */
  71 #define CERT_SERIAL_NUM_LENGTH           4                 /* Bytes of hash to be used for creating serial number of certificate. Min=2 and max=16 */
  72 #define INVALID_CERT_INFO_BUF_SIZE       2048              /* Size of buffer for message with information about reason of certificate invalidity. Data after the end of buffer will not be saved */
  73 #define CERT_PARAM_COMMON_NAME           "CN="
  74 #define CERT_PARAM_ORGANIZATION          ",O="
  75 #define CERT_PARAM_ORG_UNIT              ",OU="
  76 #define CERT_PARAM_COUNTRY               ",C=CZ"
  77 #define KEY_FILE_TYPE                    ".pem"
  78 #define CERT_FILE_TYPE                   ".crt"
  79 #define CERT_SUBJECT_PASSWORD            ""
  80 #define CERT_INFO_PREFIX                 ""
  81
  82 /*
  83  * Properties of cert for generating
  84  */
  85 typedef struct {
  86    char       *issuer_crt;                         /* filename of the issuer certificate       */
  87    char       *subject_key;                        /* filename of the subject key file         */
  88    char       *issuer_key;                         /* filename of the issuer key file          */
  89    const char *subject_pwd;                        /* password for the subject key file        */
  90    const char *issuer_pwd;                         /* password for the issuer key file         */
  91    char       *output_file;                        /* where to store the constructed key file  */
  92    const char *subject_name;                       /* subject name for certificate             */
  93    char       issuer_name[ISSUER_NAME_BUF_SIZE];   /* issuer name for certificate              */
  94    const char *not_before;                         /* validity period not before               */
  95    const char *not_after;                          /* validity period not after                */
  96    const char *serial;                             /* serial number string                     */
  97    int        is_ca;                               /* is a CA certificate                      */
  98    int        max_pathlen;                         /* maximum CA path length                   */
  99 } cert_options;
 100
 101 /*
 102  * Properties of key for generating
 103  */
 104 typedef struct {
 105    mbedtls_pk_type_t type;   /* type of key to generate  */
 106    int  rsa_keysize;         /* length of key in bits    */
 107    char *key_file_path;      /* filename of the key file */
 108 } key_options;
 109
 110 static int generate_webpage_certificate(struct client_state *csp);
 111 static char *make_certs_path(const char *conf_dir, const char *file_name, const char *suffix);
 112 static int file_exists(const char *path);
 113 static int host_to_hash(struct client_state *csp);
 114 static int ssl_verify_callback(void *data, mbedtls_x509_crt *crt, int depth, uint32_t *flags);
 115 static void free_certificate_chain(struct client_state *csp);
 116 static unsigned long  get_certificate_serial(struct client_state *csp);
 117 static void free_client_ssl_structures(struct client_state *csp);
 118 static void free_server_ssl_structures(struct client_state *csp);
 119 static int seed_rng(struct client_state *csp);
 120
 121 /*********************************************************************
 122  *
 123  * Function    :  client_use_ssl
 124  *
 125  * Description :  Tests if client in current client state structure
 126  *                should use SSL connection or standard connection.
 127  *
 128  * Parameters  :
 129  *          1  :  csp = Current client state (buffers, headers, etc...)
 130  *
 131  * Returns     :  If client should use TLS/SSL connection, 1 is returned.
 132  *                Otherwise 0 is returned.
 133  *
 134  *********************************************************************/
 135 extern int client_use_ssl(const struct client_state *csp)
 136 {
 137    return csp->http->client_ssl;
 138 }
 139
 140
 141 /*********************************************************************
 142  *
 143  * Function    :  server_use_ssl
 144  *
 145  * Description :  Tests if server in current client state structure
 146  *                should use SSL connection or standard connection.
 147  *
 148  * Parameters  :
 149  *          1  :  csp = Current client state (buffers, headers, etc...)
 150  *
 151  * Returns     :  If server should use TLS/SSL connection, 1 is returned.
 152  *                Otherwise 0 is returned.
 153  *
 154  *********************************************************************/
 155 extern int server_use_ssl(const struct client_state *csp)
 156 {
 157    return csp->http->server_ssl;
 158 }
 159
 160
 161 /*********************************************************************
 162  *
 163  * Function    :  is_ssl_pending
 164  *
 165  * Description :  Tests if there are some waiting data on ssl connection
 166  *
 167  * Parameters  :
 168  *          1  :  ssl = SSL context to test
 169  *
 170  * Returns     :   0 => No data are pending
 171  *                >0 => Pending data length
 172  *
 173  *********************************************************************/
 174 extern size_t is_ssl_pending(mbedtls_ssl_context *ssl)
 175 {
 176    if (ssl == NULL)
 177    {
 178       return 0;
 179    }
 180
 181    return mbedtls_ssl_get_bytes_avail(ssl);
 182 }
 183
 184
 185 /*********************************************************************
 186  *
 187  * Function    :  ssl_send_data
 188  *
 189  * Description :  Sends the content of buf (for n bytes) to given SSL
 190  *                connection context.
 191  *
 192  * Parameters  :
 193  *          1  :  ssl = SSL context to send data to
 194  *          2  :  buf = Pointer to data to be sent
 195  *          3  :  len = Length of data to be sent to the SSL context
 196  *
 197  * Returns     :  Length of sent data or negative value on error.
 198  *
 199  *********************************************************************/
 200 extern int ssl_send_data(mbedtls_ssl_context *ssl, const unsigned char *buf, size_t len)
 201 {
 202    int ret = 0;
 203    size_t max_fragment_size = 0;  /* Maximal length of data in one SSL fragment*/
 204    int send_len             = 0;  /* length of one data part to send */
 205    int pos                  = 0;  /* Position of unsent part in buffer */
 206
 207    if (len == 0)
 208    {
 209       return 0;
 210    }
 211
 212    /* Getting maximal length of data sent in one fragment */
 213    max_fragment_size = mbedtls_ssl_get_max_frag_len(ssl);
 214
 215    /*
 216     * Whole buffer must be sent in many fragments, because each fragment
 217     * has its maximal length.
 218     */
 219    while (pos < len)
 220    {
 221       /* Compute length of data, that can be send in next fragment */
 222       if ((pos + (int)max_fragment_size) > len)
 223       {
 224          send_len = (int)len - pos;
 225       }
 226       else
 227       {
 228          send_len = (int)max_fragment_size;
 229       }
 230
 231       log_error(LOG_LEVEL_WRITING, "TLS: %N", send_len, buf+pos);
 232
 233       /*
 234        * Sending one part of the buffer
 235        */
 236       while ((ret = mbedtls_ssl_write(ssl,
 237          (const unsigned char *)(buf + pos),
 238          (size_t)send_len)) < 0)
 239       {
 240          if (ret != MBEDTLS_ERR_SSL_WANT_READ &&
 241              ret != MBEDTLS_ERR_SSL_WANT_WRITE)
 242          {
 243             char err_buf[ERROR_BUF_SIZE];
 244
 245             mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 246             log_error(LOG_LEVEL_ERROR,
 247                "Sending data over TLS/SSL failed: %s", err_buf);
 248             return -1;
 249          }
 250       }
 251       /* Adding count of sent bytes to position in buffer */
 252       pos = pos + send_len;
 253    }
 254
 255    return (int)len;
 256 }
 257
 258
 259 /*********************************************************************
 260  *
 261  * Function    :  ssl_recv_data
 262  *
 263  * Description :  Receives data from given SSL context and puts
 264  *                it into buffer.
 265  *
 266  * Parameters  :
 267  *          1  :  ssl = SSL context to receive data from
 268  *          2  :  buf = Pointer to buffer where data will be written
 269  *          3  :  max_length = Maximum number of bytes to read
 270  *
 271  * Returns     :  Number of bytes read, 0 for EOF, or -1
 272  *                on error.
 273  *
 274  *********************************************************************/
 275 extern int ssl_recv_data(mbedtls_ssl_context *ssl, unsigned char *buf, size_t max_length)
 276 {
 277    int ret = 0;
 278    memset(buf, 0, max_length);
 279
 280    /*
 281     * Receiving data from SSL context into buffer
 282     */
 283    do
 284    {
 285       ret = mbedtls_ssl_read(ssl, buf, max_length);
 286    } while (ret == MBEDTLS_ERR_SSL_WANT_READ
 287       || ret == MBEDTLS_ERR_SSL_WANT_WRITE);
 288
 289    if (ret < 0)
 290    {
 291       char err_buf[ERROR_BUF_SIZE];
 292
 293       if (ret == MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY)
 294       {
 295          log_error(LOG_LEVEL_CONNECT,
 296             "The peer notified us that the connection is going to be closed");
 297          return 0;
 298       }
 299       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 300       log_error(LOG_LEVEL_ERROR,
 301          "Receiving data over TLS/SSL failed: %s", err_buf);
 302
 303       return -1;
 304    }
 305
 306    log_error(LOG_LEVEL_RECEIVED, "TLS: %N", ret, buf);
 307
 308    return ret;
 309 }
 310
 311
 312 /*********************************************************************
 313  *
 314  * Function    :  ssl_flush_socket
 315  *
 316  * Description :  Send any pending "buffered" content with given
 317  *                SSL connection. Alternative to function flush_socket.
 318  *
 319  * Parameters  :
 320  *          1  :  ssl = SSL context to send buffer to
 321  *          2  :  iob = The I/O buffer to flush, usually csp->iob.
 322  *
 323  * Returns     :  On success, the number of bytes send are returned (zero
 324  *                indicates nothing was sent).  On error, -1 is returned.
 325  *
 326  *********************************************************************/
 327 extern long ssl_flush_socket(mbedtls_ssl_context *ssl, struct iob *iob)
 328 {
 329    /* Computing length of buffer part to send */
 330    long len = iob->eod - iob->cur;
 331
 332    if (len <= 0)
 333    {
 334       return(0);
 335    }
 336
 337    /* Sending data to given SSl context */
 338    if (ssl_send_data(ssl, (const unsigned char *)iob->cur, (size_t)len) < 0)
 339    {
 340       return -1;
 341    }
 342    iob->eod = iob->cur = iob->buf;
 343    return(len);
 344 }
 345
 346
 347 /*********************************************************************
 348  *
 349  * Function    :  ssl_debug_callback
 350  *
 351  * Description :  Debug callback function for mbedtls library.
 352  *                Prints info into log file.
 353  *
 354  * Parameters  :
 355  *          1  :  ctx   = File to save log in
 356  *          2  :  level = Debug level
 357  *          3  :  file  = File calling debug message
 358  *          4  :  line  = Line calling debug message
 359  *          5  :  str   = Debug message
 360  *
 361  * Returns     :  N/A
 362  *
 363  *********************************************************************/
 364 static void ssl_debug_callback(void *ctx, int level, const char *file, int line, const char *str)
 365 {
 366    /*
 367    ((void)level);
 368    fprintf((FILE *)ctx, "%s:%04d: %s", file, line, str);
 369    fflush((FILE *)ctx);
 370    log_error(LOG_LEVEL_INFO, "SSL debug message: %s:%04d: %s", file, line, str);
 371    */
 372 }
 373
 374
 375 /*********************************************************************
 376  *
 377  * Function    :  create_client_ssl_connection
 378  *
 379  * Description :  Creates TLS/SSL secured connection with client
 380  *
 381  * Parameters  :
 382  *          1  :  csp = Current client state (buffers, headers, etc...)
 383  *
 384  * Returns     :  0 on success, negative value if connection wasn't created
 385  *                successfully.
 386  *
 387  *********************************************************************/
 388 extern int create_client_ssl_connection(struct client_state *csp)
 389 {
 390    /* Paths to certificates file and key file */
 391    char *key_file  = NULL;
 392    char *ca_file   = NULL;
 393    char *cert_file = NULL;
 394    int ret = 0;
 395    char err_buf[ERROR_BUF_SIZE];
 396
 397    /*
 398     * Initializing mbedtls structures for TLS/SSL connection
 399     */
 400    mbedtls_net_init(&(csp->mbedtls_client_attr.socket_fd));
 401    mbedtls_ssl_init(&(csp->mbedtls_client_attr.ssl));
 402    mbedtls_ssl_config_init(&(csp->mbedtls_client_attr.conf));
 403    mbedtls_x509_crt_init(&(csp->mbedtls_client_attr.server_cert));
 404    mbedtls_pk_init(&(csp->mbedtls_client_attr.prim_key));
 405 #if defined(MBEDTLS_SSL_CACHE_C)
 406    mbedtls_ssl_cache_init(&(csp->mbedtls_client_attr.cache));
 407 #endif
 408
 409    /*
 410     * Preparing hash of host for creating certificates
 411     */
 412    ret = host_to_hash(csp);
 413    if (ret != 0)
 414    {
 415       log_error(LOG_LEVEL_ERROR, "Generating hash of host failed: %d", ret);
 416       ret = -1;
 417       goto exit;
 418    }
 419
 420    /*
 421     * Preparing paths to certificates files and key file
 422     */
 423    ca_file   = csp->config->ca_cert_file;
 424    cert_file = make_certs_path(csp->config->certificate_directory,
 425       (const char *)csp->http->hash_of_host_hex, CERT_FILE_TYPE);
 426    key_file  = make_certs_path(csp->config->certificate_directory,
 427       (const char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
 428
 429    if (cert_file == NULL || key_file == NULL)
 430    {
 431       ret = -1;
 432       goto exit;
 433    }
 434
 435    /*
 436     * Generating certificate for requested host. Mutex to prevent
 437     * certificate and key inconsistence must be locked.
 438     */
 439    privoxy_mutex_lock(&certificate_mutex);
 440
 441    ret = generate_webpage_certificate(csp);
 442    if (ret < 0)
 443    {
 444       log_error(LOG_LEVEL_ERROR,
 445          "Generate_webpage_certificate failed: %d", ret);
 446       privoxy_mutex_unlock(&certificate_mutex);
 447       ret = -1;
 448       goto exit;
 449    }
 450    privoxy_mutex_unlock(&certificate_mutex);
 451
 452    /*
 453     * Seed the RNG
 454     */
 455    ret = seed_rng(csp);
 456    if (ret != 0)
 457    {
 458       ret = -1;
 459       goto exit;
 460    }
 461
 462    /*
 463     * Loading CA file, webpage certificate and key files
 464     */
 465    ret = mbedtls_x509_crt_parse_file(&(csp->mbedtls_client_attr.server_cert),
 466       cert_file);
 467    if (ret != 0)
 468    {
 469       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 470       log_error(LOG_LEVEL_ERROR,
 471          "Loading webpage certificate %s failed: %s", cert_file, err_buf);
 472       ret = -1;
 473       goto exit;
 474    }
 475
 476    ret = mbedtls_x509_crt_parse_file(&(csp->mbedtls_client_attr.server_cert),
 477       ca_file);
 478    if (ret != 0)
 479    {
 480       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 481       log_error(LOG_LEVEL_ERROR,
 482          "Loading CA certificate %s failed: %s", ca_file, err_buf);
 483       ret = -1;
 484       goto exit;
 485    }
 486
 487    ret = mbedtls_pk_parse_keyfile(&(csp->mbedtls_client_attr.prim_key),
 488       key_file, NULL);
 489    if (ret != 0)
 490    {
 491       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 492       log_error(LOG_LEVEL_ERROR,
 493          "Loading and parsing webpage certificate private key %s failed: %s",
 494          key_file, err_buf);
 495       ret = -1;
 496       goto exit;
 497    }
 498
 499    /*
 500     * Setting SSL parameters
 501     */
 502    ret = mbedtls_ssl_config_defaults(&(csp->mbedtls_client_attr.conf),
 503       MBEDTLS_SSL_IS_SERVER, MBEDTLS_SSL_TRANSPORT_STREAM,
 504       MBEDTLS_SSL_PRESET_DEFAULT);
 505    if (ret != 0)
 506    {
 507       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 508       log_error(LOG_LEVEL_ERROR,
 509          "mbedtls_ssl_config_defaults failed: %s", err_buf);
 510       ret = -1;
 511       goto exit;
 512    }
 513
 514    mbedtls_ssl_conf_rng(&(csp->mbedtls_client_attr.conf),
 515       mbedtls_ctr_drbg_random, &ctr_drbg);
 516    mbedtls_ssl_conf_dbg(&(csp->mbedtls_client_attr.conf),
 517       ssl_debug_callback, stdout);
 518
 519 #if defined(MBEDTLS_SSL_CACHE_C)
 520    mbedtls_ssl_conf_session_cache(&(csp->mbedtls_client_attr.conf),
 521       &(csp->mbedtls_client_attr.cache), mbedtls_ssl_cache_get,
 522       mbedtls_ssl_cache_set);
 523 #endif
 524
 525    /*
 526     * Setting certificates
 527     */
 528    ret = mbedtls_ssl_conf_own_cert(&(csp->mbedtls_client_attr.conf),
 529       &(csp->mbedtls_client_attr.server_cert),
 530       &(csp->mbedtls_client_attr.prim_key));
 531    if (ret != 0)
 532    {
 533       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 534       log_error(LOG_LEVEL_ERROR,
 535          "mbedtls_ssl_conf_own_cert failed: %s", err_buf);
 536       ret = -1;
 537       goto exit;
 538    }
 539
 540    ret = mbedtls_ssl_setup(&(csp->mbedtls_client_attr.ssl),
 541       &(csp->mbedtls_client_attr.conf));
 542    if (ret != 0)
 543    {
 544       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 545       log_error(LOG_LEVEL_ERROR, "mbedtls_ssl_setup failed: %s", err_buf);
 546       ret = -1;
 547       goto exit;
 548    }
 549
 550    mbedtls_ssl_set_bio(&(csp->mbedtls_client_attr.ssl),
 551       &(csp->mbedtls_client_attr.socket_fd), mbedtls_net_send,
 552       mbedtls_net_recv, NULL);
 553    mbedtls_ssl_session_reset(&(csp->mbedtls_client_attr.ssl));
 554
 555    /*
 556     * Setting socket fd in mbedtls_net_context structure. This structure
 557     * can't be set by mbedtls functions, because we already have created
 558     * a TCP connection when this function is called.
 559     */
 560    csp->mbedtls_client_attr.socket_fd.fd = csp->cfd;
 561
 562    /*
 563     *  Handshake with client
 564     */
 565    log_error(LOG_LEVEL_CONNECT,
 566       "Performing the TLS/SSL handshake with client. Hash of host: %s",
 567       csp->http->hash_of_host_hex);
 568    while ((ret = mbedtls_ssl_handshake(&(csp->mbedtls_client_attr.ssl))) != 0)
 569    {
 570       if (ret != MBEDTLS_ERR_SSL_WANT_READ &&
 571           ret != MBEDTLS_ERR_SSL_WANT_WRITE)
 572       {
 573          mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 574          log_error(LOG_LEVEL_ERROR,
 575             "medtls_ssl_handshake with client failed: %s", err_buf);
 576          ret = -1;
 577          goto exit;
 578       }
 579    }
 580
 581    log_error(LOG_LEVEL_CONNECT, "Client successfully connected over TLS/SSL");
 582    csp->ssl_with_client_is_opened = 1;
 583
 584 exit:
 585    /*
 586     * Freeing allocated paths to files
 587     */
 588    freez(cert_file);
 589    freez(key_file);
 590
 591    /* Freeing structures if connection wasn't created successfully */
 592    if (ret < 0)
 593    {
 594       free_client_ssl_structures(csp);
 595    }
 596    return ret;
 597 }
 598
 599
 600 /*********************************************************************
 601  *
 602  * Function    :  close_client_ssl_connection
 603  *
 604  * Description :  Closes TLS/SSL connection with client. This function
 605  *                checks if this connection is already created.
 606  *
 607  * Parameters  :
 608  *          1  :  csp = Current client state (buffers, headers, etc...)
 609  *
 610  * Returns     :  N/A
 611  *
 612  *********************************************************************/
 613 extern void close_client_ssl_connection(struct client_state *csp)
 614 {
 615    int ret = 0;
 616
 617    if (csp->ssl_with_client_is_opened == 0)
 618    {
 619       return;
 620    }
 621
 622    /*
 623     * Notifying the peer that the connection is being closed.
 624     */
 625    do {
 626       ret = mbedtls_ssl_close_notify(&(csp->mbedtls_client_attr.ssl));
 627    } while (ret == MBEDTLS_ERR_SSL_WANT_WRITE);
 628
 629    free_client_ssl_structures(csp);
 630    csp->ssl_with_client_is_opened = 0;
 631 }
 632
 633
 634 /*********************************************************************
 635  *
 636  * Function    :  free_client_ssl_structures
 637  *
 638  * Description :  Frees structures used for SSL communication with
 639  *                client.
 640  *
 641  * Parameters  :
 642  *          1  :  csp = Current client state (buffers, headers, etc...)
 643  *
 644  * Returns     :  N/A
 645  *
 646  *********************************************************************/
 647 static void free_client_ssl_structures(struct client_state *csp)
 648 {
 649    /*
 650    * We can't use function mbedtls_net_free, because this function
 651    * inter alia close TCP connection on setted fd. Instead of this
 652    * function, we change fd to -1, which is the same what does
 653    * rest of mbedtls_net_free function.
 654    */
 655    csp->mbedtls_client_attr.socket_fd.fd = -1;
 656
 657    /* Freeing mbedtls structures */
 658    mbedtls_x509_crt_free(&(csp->mbedtls_client_attr.server_cert));
 659    mbedtls_pk_free(&(csp->mbedtls_client_attr.prim_key));
 660    mbedtls_ssl_free(&(csp->mbedtls_client_attr.ssl));
 661    mbedtls_ssl_config_free(&(csp->mbedtls_client_attr.conf));
 662 #if defined(MBEDTLS_SSL_CACHE_C)
 663    mbedtls_ssl_cache_free(&(csp->mbedtls_client_attr.cache));
 664 #endif
 665 }
 666
 667
 668 /*********************************************************************
 669  *
 670  * Function    :  create_server_ssl_connection
 671  *
 672  * Description :  Creates TLS/SSL secured connection with server.
 673  *
 674  * Parameters  :
 675  *          1  :  csp = Current client state (buffers, headers, etc...)
 676  *
 677  * Returns     :  0 on success, negative value if connection wasn't created
 678  *                successfully.
 679  *
 680  *********************************************************************/
 681 extern int create_server_ssl_connection(struct client_state *csp)
 682 {
 683    int ret = 0;
 684    char err_buf[ERROR_BUF_SIZE];
 685    char *trusted_cas_file = NULL;
 686    int auth_mode = MBEDTLS_SSL_VERIFY_REQUIRED;
 687
 688    csp->server_cert_verification_result = SSL_CERT_NOT_VERIFIED;
 689    csp->server_certs_chain.next = NULL;
 690
 691    /* Setting path to file with trusted CAs */
 692    trusted_cas_file = csp->config->trusted_cas_file;
 693
 694    /*
 695     * Initializing mbedtls structures for TLS/SSL connection
 696     */
 697    mbedtls_net_init(&(csp->mbedtls_server_attr.socket_fd));
 698    mbedtls_ssl_init(&(csp->mbedtls_server_attr.ssl));
 699    mbedtls_ssl_config_init(&(csp->mbedtls_server_attr.conf));
 700    mbedtls_x509_crt_init(&(csp->mbedtls_server_attr.ca_cert));
 701
 702    /*
 703    * Setting socket fd in mbedtls_net_context structure. This structure
 704    * can't be set by mbedtls functions, because we already have created
 705    * TCP connection when calling this function.
 706    */
 707    csp->mbedtls_server_attr.socket_fd.fd = csp->server_connection.sfd;
 708
 709    /*
 710     * Seed the RNG
 711     */
 712    ret = seed_rng(csp);
 713    if (ret != 0)
 714    {
 715       ret = -1;
 716       goto exit;
 717    }
 718
 719    /*
 720     * Loading file with trusted CAs
 721     */
 722    ret = mbedtls_x509_crt_parse_file(&(csp->mbedtls_server_attr.ca_cert),
 723       trusted_cas_file);
 724    if (ret < 0)
 725    {
 726       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 727       log_error(LOG_LEVEL_ERROR, "Loading trusted CAs file %s failed: %s",
 728          trusted_cas_file, err_buf);
 729       ret = -1;
 730       goto exit;
 731    }
 732
 733    /*
 734     * Set TLS/SSL options
 735     */
 736    ret = mbedtls_ssl_config_defaults(&(csp->mbedtls_server_attr.conf),
 737       MBEDTLS_SSL_IS_CLIENT,
 738       MBEDTLS_SSL_TRANSPORT_STREAM,
 739       MBEDTLS_SSL_PRESET_DEFAULT);
 740    if (ret != 0)
 741    {
 742       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 743       log_error(LOG_LEVEL_ERROR, "mbedtls_ssl_config_defaults failed: %s",
 744          err_buf);
 745       ret = -1;
 746       goto exit;
 747    }
 748
 749    /*
 750     * Setting how strict should certificate verification be and other
 751     * parameters for certificate verification
 752     */
 753    if (csp->dont_verify_certificate)
 754    {
 755       auth_mode = MBEDTLS_SSL_VERIFY_NONE;
 756    }
 757
 758    mbedtls_ssl_conf_authmode(&(csp->mbedtls_server_attr.conf), auth_mode);
 759    mbedtls_ssl_conf_ca_chain(&(csp->mbedtls_server_attr.conf),
 760       &(csp->mbedtls_server_attr.ca_cert), NULL);
 761
 762    /* Setting callback function for certificates verification */
 763    mbedtls_ssl_conf_verify(&(csp->mbedtls_server_attr.conf),
 764       ssl_verify_callback, (void *)csp);
 765
 766    mbedtls_ssl_conf_rng(&(csp->mbedtls_server_attr.conf),
 767       mbedtls_ctr_drbg_random, &ctr_drbg);
 768    mbedtls_ssl_conf_dbg(&(csp->mbedtls_server_attr.conf),
 769       ssl_debug_callback, stdout);
 770
 771    ret = mbedtls_ssl_setup(&(csp->mbedtls_server_attr.ssl),
 772       &(csp->mbedtls_server_attr.conf));
 773    if (ret != 0)
 774    {
 775       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 776       log_error(LOG_LEVEL_ERROR, "mbedtls_ssl_setup failed: %s", err_buf);
 777       ret = -1;
 778       goto exit;
 779    }
 780
 781    /*
 782     * Set the hostname to check against the received server certificate
 783     */
 784    ret = mbedtls_ssl_set_hostname(&(csp->mbedtls_server_attr.ssl),
 785       csp->http->host);
 786    if (ret != 0)
 787    {
 788       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 789       log_error(LOG_LEVEL_ERROR, "mbedtls_ssl_set_hostname failed: %s",
 790          err_buf);
 791       ret = -1;
 792       goto exit;
 793    }
 794
 795    mbedtls_ssl_set_bio(&(csp->mbedtls_server_attr.ssl),
 796       &(csp->mbedtls_server_attr.socket_fd), mbedtls_net_send,
 797       mbedtls_net_recv, NULL);
 798
 799    /*
 800     * Handshake with server
 801     */
 802    log_error(LOG_LEVEL_CONNECT,
 803       "Performing the TLS/SSL handshake with the server");
 804
 805    while ((ret = mbedtls_ssl_handshake(&(csp->mbedtls_server_attr.ssl))) != 0)
 806    {
 807       if (ret != MBEDTLS_ERR_SSL_WANT_READ
 808        && ret != MBEDTLS_ERR_SSL_WANT_WRITE)
 809       {
 810          mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 811
 812          if (ret == MBEDTLS_ERR_X509_CERT_VERIFY_FAILED)
 813          {
 814             char reason[INVALID_CERT_INFO_BUF_SIZE];
 815
 816             csp->server_cert_verification_result =
 817                mbedtls_ssl_get_verify_result(&(csp->mbedtls_server_attr.ssl));
 818             mbedtls_x509_crt_verify_info(reason, sizeof(reason), "",
 819                csp->server_cert_verification_result);
 820
 821             /* Log the reason without the trailing new line */
 822             log_error(LOG_LEVEL_ERROR,
 823                "X509 certificate verification for %s failed: %N",
 824                csp->http->hostport, strlen(reason)-1, reason);
 825             ret = -1;
 826          }
 827          else
 828          {
 829             log_error(LOG_LEVEL_ERROR,
 830                "mbedtls_ssl_handshake with server failed: %s", err_buf);
 831             ret = -1;
 832          }
 833          goto exit;
 834       }
 835    }
 836
 837    log_error(LOG_LEVEL_CONNECT, "Server successfully connected over TLS/SSL");
 838
 839    /*
 840     * Server certificate chain is valid, so we can clean
 841     * chain, because we will not send it to client.
 842     */
 843    free_certificate_chain(csp);
 844
 845    csp->ssl_with_server_is_opened = 1;
 846    csp->server_cert_verification_result =
 847       mbedtls_ssl_get_verify_result(&(csp->mbedtls_server_attr.ssl));
 848
 849 exit:
 850    /* Freeing structures if connection wasn't created successfully */
 851    if (ret < 0)
 852    {
 853       free_server_ssl_structures(csp);
 854    }
 855
 856    return ret;
 857 }
 858
 859
 860 /*********************************************************************
 861  *
 862  * Function    :  close_server_ssl_connection
 863  *
 864  * Description :  Closes TLS/SSL connection with server. This function
 865  *                checks if this connection is already opened.
 866  *
 867  * Parameters  :
 868  *          1  :  csp = Current client state (buffers, headers, etc...)
 869  *
 870  * Returns     :  N/A
 871  *
 872  *********************************************************************/
 873 static void close_server_ssl_connection(struct client_state *csp)
 874 {
 875    int ret = 0;
 876
 877    if (csp->ssl_with_server_is_opened == 0)
 878    {
 879       return;
 880    }
 881
 882    /*
 883    * Notifying the peer that the connection is being closed.
 884    */
 885    do {
 886       ret = mbedtls_ssl_close_notify(&(csp->mbedtls_server_attr.ssl));
 887    } while (ret == MBEDTLS_ERR_SSL_WANT_WRITE);
 888
 889    free_server_ssl_structures(csp);
 890    csp->ssl_with_server_is_opened = 0;
 891 }
 892
 893
 894 /*********************************************************************
 895  *
 896  * Function    :  free_server_ssl_structures
 897  *
 898  * Description :  Frees structures used for SSL communication with server
 899  *
 900  * Parameters  :
 901  *          1  :  csp = Current client state (buffers, headers, etc...)
 902  *
 903  * Returns     :  N/A
 904  *
 905  *********************************************************************/
 906 static void free_server_ssl_structures(struct client_state *csp)
 907 {
 908    /*
 909    * We can't use function mbedtls_net_free, because this function
 910    * inter alia close TCP connection on setted fd. Instead of this
 911    * function, we change fd to -1, which is the same what does
 912    * rest of mbedtls_net_free function.
 913    */
 914    csp->mbedtls_server_attr.socket_fd.fd = -1;
 915
 916    mbedtls_x509_crt_free(&(csp->mbedtls_server_attr.ca_cert));
 917    mbedtls_ssl_free(&(csp->mbedtls_server_attr.ssl));
 918    mbedtls_ssl_config_free(&(csp->mbedtls_server_attr.conf));
 919 }
 920
 921
 922 /*********************************************************************
 923  *
 924  * Function    :  close_client_and_server_ssl_connections
 925  *
 926  * Description :  Checks if client or server should use secured
 927  *                connection over SSL and if so, closes all of them.
 928  *
 929  * Parameters  :
 930  *          1  :  csp = Current client state (buffers, headers, etc...)
 931  *
 932  * Returns     :  N/A
 933  *
 934  *********************************************************************/
 935 extern void close_client_and_server_ssl_connections(struct client_state *csp)
 936 {
 937    if (client_use_ssl(csp) == 1)
 938    {
 939       close_client_ssl_connection(csp);
 940    }
 941    if (server_use_ssl(csp) == 1)
 942    {
 943       close_server_ssl_connection(csp);
 944    }
 945 }
 946
 947 /*====================== Certificates ======================*/
 948
 949 /*********************************************************************
 950  *
 951  * Function    :  write_certificate
 952  *
 953  * Description :  Writes certificate into file.
 954  *
 955  * Parameters  :
 956  *          1  :  crt = certificate to write into file
 957  *          2  :  output_file = path to save certificate file
 958  *          3  :  f_rng = mbedtls_ctr_drbg_random
 959  *          4  :  p_rng = mbedtls_ctr_drbg_context
 960  *
 961  * Returns     :  Length of written certificate on success or negative value
 962  *                on error
 963  *
 964  *********************************************************************/
 965 static int write_certificate(mbedtls_x509write_cert *crt, const char *output_file,
 966    int(*f_rng)(void *, unsigned char *, size_t), void *p_rng)
 967 {
 968    FILE *f = NULL;
 969    size_t len = 0;
 970    unsigned char cert_buf[CERTIFICATE_BUF_SIZE + 1]; /* Buffer for certificate in PEM format + terminating NULL */
 971    int ret = 0;
 972    char err_buf[ERROR_BUF_SIZE];
 973
 974    memset(cert_buf, 0, sizeof(cert_buf));
 975
 976    /*
 977     * Writing certificate into PEM string. If buffer is too small, function
 978     * returns specific error and no buffer overflow can happen.
 979     */
 980    if ((ret = mbedtls_x509write_crt_pem(crt, cert_buf,
 981       sizeof(cert_buf) - 1, f_rng, p_rng)) != 0)
 982    {
 983       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
 984       log_error(LOG_LEVEL_ERROR,
 985          "Writing certificate into buffer failed: %s", err_buf);
 986       return -1;
 987    }
 988
 989    len = strlen((char *)cert_buf);
 990
 991    /*
 992     * Saving certificate into file
 993     */
 994    if ((f = fopen(output_file, "w")) == NULL)
 995    {
 996       log_error(LOG_LEVEL_ERROR, "Opening file %s to save certificate failed",
 997          output_file);
 998       return -1;
 999    }
1000
1001    if (fwrite(cert_buf, 1, len, f) != len)
1002    {
1003       log_error(LOG_LEVEL_ERROR,
1004          "Writing certificate into file %s failed", output_file);
1005       fclose(f);
1006       return -1;
1007    }
1008
1009    fclose(f);
1010
1011    return (int)len;
1012 }
1013
1014
1015 /*********************************************************************
1016  *
1017  * Function    :  write_private_key
1018  *
1019  * Description :  Writes private key into file and copies saved
1020  *                content into given pointer to string. If function
1021  *                returns 0 for success, this copy must be freed by
1022  *                caller.
1023  *
1024  * Parameters  :
1025  *          1  :  key = key to write into file
1026  *          2  :  ret_buf = pointer to string with created key file content
1027  *          3  :  key_file_path = path where to save key file
1028  *
1029  * Returns     :  Length of written private key on success or negative value
1030  *                on error
1031  *
1032  *********************************************************************/
1033 static int write_private_key(mbedtls_pk_context *key, unsigned char **ret_buf,
1034    const char *key_file_path)
1035 {
1036    size_t len = 0;                /* Length of created key    */
1037    FILE *f = NULL;                /* File to save certificate */
1038    int ret = 0;
1039    char err_buf[ERROR_BUF_SIZE];
1040
1041    /* Initializing buffer for key file content */
1042    *ret_buf = zalloc_or_die(PRIVATE_KEY_BUF_SIZE + 1);
1043
1044    /*
1045     * Writing private key into PEM string
1046     */
1047    if ((ret = mbedtls_pk_write_key_pem(key, *ret_buf, PRIVATE_KEY_BUF_SIZE)) != 0)
1048    {
1049       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1050       log_error(LOG_LEVEL_ERROR,
1051          "Writing private key into PEM string failed: %s", err_buf);
1052       ret = -1;
1053       goto exit;
1054    }
1055    len = strlen((char *)*ret_buf);
1056
1057    /*
1058     * Saving key into file
1059     */
1060    if ((f = fopen(key_file_path, "wb")) == NULL)
1061    {
1062       log_error(LOG_LEVEL_ERROR,
1063          "Opening file %s to save private key failed: %E",
1064          key_file_path);
1065       ret = -1;
1066       goto exit;
1067    }
1068
1069    if (fwrite(*ret_buf, 1, len, f) != len)
1070    {
1071       fclose(f);
1072       log_error(LOG_LEVEL_ERROR,
1073          "Writing private key into file %s failed",
1074          key_file_path);
1075       ret = -1;
1076       goto exit;
1077    }
1078
1079    fclose(f);
1080
1081 exit:
1082    if (ret < 0)
1083    {
1084       freez(*ret_buf);
1085       *ret_buf = NULL;
1086       return ret;
1087    }
1088    return (int)len;
1089 }
1090
1091
1092 /*********************************************************************
1093  *
1094  * Function    :  generate_key
1095  *
1096  * Description : Tests if private key for host saved in csp already
1097  *               exists.  If this file doesn't exists, a new key is
1098  *               generated and saved in a file. The generated key is also
1099  *               copied into given parameter key_buf, which must be then
1100  *               freed by caller. If file with key exists, key_buf
1101  *               contain NULL and no private key is generated.
1102  *
1103  * Parameters  :
1104  *          1  :  csp = Current client state (buffers, headers, etc...)
1105  *          2  :  key_buf = buffer to save new generated key
1106  *
1107  * Returns     :  -1 => Error while generating private key
1108  *                 0 => Key already exists
1109  *                >0 => Length of generated private key
1110  *
1111  *********************************************************************/
1112 static int generate_key(struct client_state *csp, unsigned char **key_buf)
1113 {
1114    mbedtls_pk_context key;
1115    key_options key_opt;
1116    int ret = 0;
1117    char err_buf[ERROR_BUF_SIZE];
1118
1119    key_opt.key_file_path = NULL;
1120
1121    /*
1122     * Initializing structures for key generating
1123     */
1124    mbedtls_pk_init(&key);
1125
1126    /*
1127     * Preparing path for key file and other properties for generating key
1128     */
1129    key_opt.type        = MBEDTLS_PK_RSA;
1130    key_opt.rsa_keysize = RSA_KEYSIZE;
1131
1132    key_opt.key_file_path = make_certs_path(csp->config->certificate_directory,
1133       (char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
1134    if (key_opt.key_file_path == NULL)
1135    {
1136       ret = -1;
1137       goto exit;
1138    }
1139
1140    /*
1141     * Test if key already exists. If so, we don't have to create it again.
1142     */
1143    if (file_exists(key_opt.key_file_path) == 1)
1144    {
1145       ret = 0;
1146       goto exit;
1147    }
1148
1149    /*
1150     * Seed the RNG
1151     */
1152    ret = seed_rng(csp);
1153    if (ret != 0)
1154    {
1155       ret = -1;
1156       goto exit;
1157    }
1158
1159    /*
1160     * Setting attributes of private key and generating it
1161     */
1162    if ((ret = mbedtls_pk_setup(&key,
1163       mbedtls_pk_info_from_type(key_opt.type))) != 0)
1164    {
1165       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1166       log_error(LOG_LEVEL_ERROR, "mbedtls_pk_setup failed: %s", err_buf);
1167       ret = -1;
1168       goto exit;
1169    }
1170
1171    ret = mbedtls_rsa_gen_key(mbedtls_pk_rsa(key), mbedtls_ctr_drbg_random,
1172       &ctr_drbg, (unsigned)key_opt.rsa_keysize, RSA_KEY_PUBLIC_EXPONENT);
1173    if (ret != 0)
1174    {
1175       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1176       log_error(LOG_LEVEL_ERROR, "Key generating failed: %s", err_buf);
1177       ret = -1;
1178       goto exit;
1179    }
1180
1181    /*
1182     * Exporting private key into file
1183     */
1184    if ((ret = write_private_key(&key, key_buf, key_opt.key_file_path)) < 0)
1185    {
1186       log_error(LOG_LEVEL_ERROR,
1187          "Writing private key into file %s failed", key_opt.key_file_path);
1188       ret = -1;
1189       goto exit;
1190    }
1191
1192 exit:
1193    /*
1194     * Freeing used variables
1195     */
1196    freez(key_opt.key_file_path);
1197
1198    mbedtls_pk_free(&key);
1199
1200    return ret;
1201 }
1202
1203
1204 /*********************************************************************
1205  *
1206  * Function    :  ssl_certificate_is_invalid
1207  *
1208  * Description :  Checks whether or not a certificate is valid.
1209  *                Currently only checks that the certificate can be
1210  *                parsed and that the "valid to" date is in the future.
1211  *
1212  * Parameters  :
1213  *          1  :  cert_file = The certificate to check
1214  *
1215  * Returns     :   0 => The certificate is valid.
1216  *                 1 => The certificate is invalid
1217  *
1218  *********************************************************************/
1219 static int ssl_certificate_is_invalid(const char *cert_file)
1220 {
1221    mbedtls_x509_crt cert;
1222    int ret;
1223
1224    mbedtls_x509_crt_init(&cert);
1225
1226    ret = mbedtls_x509_crt_parse_file(&cert, cert_file);
1227    if (ret != 0)
1228    {
1229       char err_buf[ERROR_BUF_SIZE];
1230
1231       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1232       log_error(LOG_LEVEL_ERROR,
1233          "Loading certificate %s to check validity failed: %s",
1234          cert_file, err_buf);
1235       mbedtls_x509_crt_free(&cert);
1236
1237       return 1;
1238    }
1239    if (mbedtls_x509_time_is_past(&cert.valid_to))
1240    {
1241       mbedtls_x509_crt_free(&cert);
1242
1243       return 1;
1244    }
1245
1246    mbedtls_x509_crt_free(&cert);
1247
1248    return 0;
1249
1250 }
1251
1252
1253 /*********************************************************************
1254  *
1255  * Function    :  generate_webpage_certificate
1256  *
1257  * Description :  Creates certificate file in presetted directory.
1258  *                If certificate already exists, no other certificate
1259  *                will be created. Subject of certificate is named
1260  *                by csp->http->host from parameter. This function also
1261  *                triggers generating of private key for new certificate.
1262  *
1263  * Parameters  :
1264  *          1  :  csp = Current client state (buffers, headers, etc...)
1265  *
1266  * Returns     :  -1 => Error while creating certificate.
1267  *                 0 => Certificate already exists.
1268  *                >0 => Length of created certificate.
1269  *
1270  *********************************************************************/
1271 static int generate_webpage_certificate(struct client_state *csp)
1272 {
1273    mbedtls_x509_crt issuer_cert;
1274    mbedtls_pk_context loaded_issuer_key, loaded_subject_key;
1275    mbedtls_pk_context *issuer_key  = &loaded_issuer_key;
1276    mbedtls_pk_context *subject_key = &loaded_subject_key;
1277    mbedtls_x509write_cert cert;
1278    mbedtls_mpi serial;
1279
1280    unsigned char *key_buf = NULL;    /* Buffer for created key */
1281
1282    int ret = 0;
1283    char err_buf[ERROR_BUF_SIZE];
1284    cert_options cert_opt;
1285
1286    /* Paths to keys and certificates needed to create certificate */
1287    cert_opt.issuer_key  = NULL;
1288    cert_opt.subject_key = NULL;
1289    cert_opt.issuer_crt  = NULL;
1290    cert_opt.output_file = NULL;
1291
1292    /*
1293     * Create key for requested host
1294     */
1295    int subject_key_len = generate_key(csp, &key_buf);
1296    if (subject_key_len < 0)
1297    {
1298       log_error(LOG_LEVEL_ERROR, "Key generating failed");
1299       return -1;
1300    }
1301
1302    /*
1303     * Initializing structures for certificate generating
1304     */
1305    mbedtls_x509write_crt_init(&cert);
1306    mbedtls_x509write_crt_set_md_alg(&cert, CERT_SIGNATURE_ALGORITHM);
1307    mbedtls_pk_init(&loaded_issuer_key);
1308    mbedtls_pk_init(&loaded_subject_key);
1309    mbedtls_mpi_init(&serial);
1310    mbedtls_x509_crt_init(&issuer_cert);
1311
1312    /*
1313     * Presetting parameters for certificate. We must compute total length
1314     * of parameters.
1315     */
1316    size_t cert_params_len = strlen(CERT_PARAM_COMMON_NAME) +
1317       strlen(CERT_PARAM_ORGANIZATION) + strlen(CERT_PARAM_COUNTRY) +
1318       strlen(CERT_PARAM_ORG_UNIT) +
1319       3 * strlen(csp->http->host) + 1;
1320    char cert_params[cert_params_len];
1321    memset(cert_params, 0, cert_params_len);
1322
1323    /*
1324     * Converting unsigned long serial number to char * serial number.
1325     * We must compute length of serial number in string + terminating null.
1326     */
1327    unsigned long certificate_serial = get_certificate_serial(csp);
1328    int serial_num_size = snprintf(NULL, 0, "%lu", certificate_serial) + 1;
1329    if (serial_num_size <= 0)
1330    {
1331       serial_num_size = 1;
1332    }
1333
1334    char serial_num_text[serial_num_size];  /* Buffer for serial number */
1335    ret = snprintf(serial_num_text, (size_t)serial_num_size, "%lu", certificate_serial);
1336    if (ret < 0 || ret >= serial_num_size)
1337    {
1338       log_error(LOG_LEVEL_ERROR,
1339          "Converting certificate serial number into string failed");
1340       ret = -1;
1341       goto exit;
1342    }
1343
1344    /*
1345     * Preparing parameters for certificate
1346     */
1347    strlcpy(cert_params, CERT_PARAM_COMMON_NAME,  cert_params_len);
1348    strlcat(cert_params, csp->http->host,         cert_params_len);
1349    strlcat(cert_params, CERT_PARAM_ORGANIZATION, cert_params_len);
1350    strlcat(cert_params, csp->http->host,         cert_params_len);
1351    strlcat(cert_params, CERT_PARAM_ORG_UNIT,     cert_params_len);
1352    strlcat(cert_params, csp->http->host,         cert_params_len);
1353    strlcat(cert_params, CERT_PARAM_COUNTRY,      cert_params_len);
1354
1355    cert_opt.issuer_crt = csp->config->ca_cert_file;
1356    cert_opt.issuer_key = csp->config->ca_key_file;
1357    cert_opt.subject_key = make_certs_path(csp->config->certificate_directory,
1358       (const char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
1359    cert_opt.output_file = make_certs_path(csp->config->certificate_directory,
1360       (const char *)csp->http->hash_of_host_hex, CERT_FILE_TYPE);
1361
1362    if (cert_opt.subject_key == NULL || cert_opt.output_file == NULL)
1363    {
1364       ret = -1;
1365       goto exit;
1366    }
1367
1368    cert_opt.subject_pwd   = CERT_SUBJECT_PASSWORD;
1369    cert_opt.issuer_pwd    = csp->config->ca_password;
1370    cert_opt.subject_name  = cert_params;
1371    cert_opt.not_before    = GENERATED_CERT_VALID_FROM;
1372    cert_opt.not_after     = GENERATED_CERT_VALID_TO;
1373    cert_opt.serial        = serial_num_text;
1374    cert_opt.is_ca         = 0;
1375    cert_opt.max_pathlen   = -1;
1376
1377    /*
1378     * Test if certificate exists and private key was already created
1379     */
1380    if (file_exists(cert_opt.output_file) == 1 && subject_key_len == 0)
1381    {
1382       /* The file exists, but is it valid */
1383       if (ssl_certificate_is_invalid(cert_opt.output_file))
1384       {
1385          log_error(LOG_LEVEL_CONNECT,
1386             "Certificate %s is no longer valid. Removing.",
1387             cert_opt.output_file);
1388          if (unlink(cert_opt.output_file))
1389          {
1390             log_error(LOG_LEVEL_ERROR, "Failed to unlink %s: %E",
1391                cert_opt.output_file);
1392             ret = -1;
1393             goto exit;
1394          }
1395       }
1396       else
1397       {
1398          ret = 0;
1399          goto exit;
1400       }
1401    }
1402
1403    /*
1404     * Seed the PRNG
1405     */
1406    ret = seed_rng(csp);
1407    if (ret != 0)
1408    {
1409       ret = -1;
1410       goto exit;
1411    }
1412
1413    /*
1414     * Parse serial to MPI
1415     */
1416    ret = mbedtls_mpi_read_string(&serial, 10, cert_opt.serial);
1417    if (ret != 0)
1418    {
1419       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1420       log_error(LOG_LEVEL_ERROR,
1421          "mbedtls_mpi_read_string failed: %s", err_buf);
1422       ret = -1;
1423       goto exit;
1424    }
1425
1426    /*
1427     * Loading certificates
1428     */
1429    ret = mbedtls_x509_crt_parse_file(&issuer_cert, cert_opt.issuer_crt);
1430    if (ret != 0)
1431    {
1432       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1433       log_error(LOG_LEVEL_ERROR, "Loading issuer certificate %s failed: %s",
1434          cert_opt.issuer_crt, err_buf);
1435       ret = -1;
1436       goto exit;
1437    }
1438
1439    ret = mbedtls_x509_dn_gets(cert_opt.issuer_name,
1440       sizeof(cert_opt.issuer_name), &issuer_cert.subject);
1441    if (ret < 0)
1442    {
1443       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1444       log_error(LOG_LEVEL_ERROR, "mbedtls_x509_dn_gets failed: %s", err_buf);
1445       ret = -1;
1446       goto exit;
1447    }
1448
1449    /*
1450     * Loading keys from file or from buffer
1451     */
1452    if (key_buf != NULL && subject_key_len > 0)
1453    {
1454       /* Key was created in this function and is stored in buffer */
1455       ret = mbedtls_pk_parse_key(&loaded_subject_key, key_buf,
1456          (size_t)(subject_key_len + 1), (unsigned const char *)
1457          cert_opt.subject_pwd, strlen(cert_opt.subject_pwd));
1458    }
1459    else
1460    {
1461       /* Key wasn't created in this function, because it already existed */
1462       ret = mbedtls_pk_parse_keyfile(&loaded_subject_key,
1463          cert_opt.subject_key, cert_opt.subject_pwd);
1464    }
1465
1466    if (ret != 0)
1467    {
1468       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1469       log_error(LOG_LEVEL_ERROR, "Parsing subject key %s failed: %s",
1470          cert_opt.subject_key, err_buf);
1471       ret = -1;
1472       goto exit;
1473    }
1474
1475    ret = mbedtls_pk_parse_keyfile(&loaded_issuer_key, cert_opt.issuer_key,
1476       cert_opt.issuer_pwd);
1477    if (ret != 0)
1478    {
1479       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1480       log_error(LOG_LEVEL_ERROR,
1481          "Parsing issuer key %s failed: %s", cert_opt.issuer_key, err_buf);
1482       ret = -1;
1483       goto exit;
1484    }
1485
1486    /*
1487     * Check if key and issuer certificate match
1488     */
1489    if (!mbedtls_pk_can_do(&issuer_cert.pk, MBEDTLS_PK_RSA) ||
1490       mbedtls_mpi_cmp_mpi(&mbedtls_pk_rsa(issuer_cert.pk)->N,
1491          &mbedtls_pk_rsa(*issuer_key)->N) != 0 ||
1492       mbedtls_mpi_cmp_mpi(&mbedtls_pk_rsa(issuer_cert.pk)->E,
1493          &mbedtls_pk_rsa(*issuer_key)->E) != 0)
1494    {
1495       log_error(LOG_LEVEL_ERROR,
1496          "Issuer key doesn't match issuer certificate");
1497       ret = -1;
1498       goto exit;
1499    }
1500
1501    mbedtls_x509write_crt_set_subject_key(&cert, subject_key);
1502    mbedtls_x509write_crt_set_issuer_key(&cert, issuer_key);
1503
1504    /*
1505     * Setting parameters of signed certificate
1506     */
1507    ret = mbedtls_x509write_crt_set_subject_name(&cert, cert_opt.subject_name);
1508    if (ret != 0)
1509    {
1510       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1511       log_error(LOG_LEVEL_ERROR,
1512          "Setting subject name in signed certificate failed: %s", err_buf);
1513       ret = -1;
1514       goto exit;
1515    }
1516
1517    ret = mbedtls_x509write_crt_set_issuer_name(&cert, cert_opt.issuer_name);
1518    if (ret != 0)
1519    {
1520       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1521       log_error(LOG_LEVEL_ERROR,
1522          "Setting issuer name in signed certificate failed: %s", err_buf);
1523       ret = -1;
1524       goto exit;
1525    }
1526
1527    ret = mbedtls_x509write_crt_set_serial(&cert, &serial);
1528    if (ret != 0)
1529    {
1530       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1531       log_error(LOG_LEVEL_ERROR,
1532          "Setting serial number in signed certificate failed: %s", err_buf);
1533       ret = -1;
1534       goto exit;
1535    }
1536
1537    ret = mbedtls_x509write_crt_set_validity(&cert, cert_opt.not_before,
1538       cert_opt.not_after);
1539    if (ret != 0)
1540    {
1541       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1542       log_error(LOG_LEVEL_ERROR,
1543          "Setting validity in signed certificate failed: %s", err_buf);
1544       ret = -1;
1545       goto exit;
1546    }
1547
1548    /*
1549     * Setting the basicConstraints extension for certificate
1550     */
1551    ret = mbedtls_x509write_crt_set_basic_constraints(&cert, cert_opt.is_ca,
1552       cert_opt.max_pathlen);
1553    if (ret != 0)
1554    {
1555       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1556       log_error(LOG_LEVEL_ERROR, "Setting the basicConstraints extension "
1557          "in signed certificate failed: %s", err_buf);
1558       ret = -1;
1559       goto exit;
1560    }
1561
1562 #if defined(MBEDTLS_SHA1_C)
1563    /* Setting the subjectKeyIdentifier extension for certificate */
1564    ret = mbedtls_x509write_crt_set_subject_key_identifier(&cert);
1565    if (ret != 0)
1566    {
1567       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1568       log_error(LOG_LEVEL_ERROR, "mbedtls_x509write_crt_set_subject_key_"
1569          "identifier failed: %s", err_buf);
1570       ret = -1;
1571       goto exit;
1572    }
1573
1574    /* Setting the authorityKeyIdentifier extension for certificate */
1575    ret = mbedtls_x509write_crt_set_authority_key_identifier(&cert);
1576    if (ret != 0)
1577    {
1578       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1579       log_error(LOG_LEVEL_ERROR, "mbedtls_x509write_crt_set_authority_key_"
1580          "identifier failed: %s", err_buf);
1581       ret = -1;
1582       goto exit;
1583    }
1584 #endif /* MBEDTLS_SHA1_C */
1585
1586    /*
1587     * Writing certificate into file
1588     */
1589    ret = write_certificate(&cert, cert_opt.output_file,
1590       mbedtls_ctr_drbg_random, &ctr_drbg);
1591    if (ret < 0)
1592    {
1593       log_error(LOG_LEVEL_ERROR, "Writing certificate into file failed");
1594       goto exit;
1595    }
1596
1597 exit:
1598    /*
1599     * Freeing used structures
1600     */
1601    mbedtls_x509write_crt_free(&cert);
1602    mbedtls_pk_free(&loaded_subject_key);
1603    mbedtls_pk_free(&loaded_issuer_key);
1604    mbedtls_mpi_free(&serial);
1605    mbedtls_x509_crt_free(&issuer_cert);
1606
1607    freez(cert_opt.subject_key);
1608    freez(cert_opt.output_file);
1609    freez(key_buf);
1610
1611    return ret;
1612 }
1613
1614
1615 /*********************************************************************
1616  *
1617  * Function    :  make_certs_path
1618  *
1619  * Description : Creates path to file from three pieces. This function
1620  *               takes parameters and puts them in one new mallocated
1621  *               char * in correct order. Returned variable must be freed
1622  *               by caller. This function is mainly used for creating
1623  *               paths of certificates and keys files.
1624  *
1625  * Parameters  :
1626  *          1  :  conf_dir  = Name/path of directory where is the file.
1627  *                            '.' can be used for current directory.
1628  *          2  :  file_name = Name of file in conf_dir without suffix.
1629  *          3  :  suffix    = Suffix of given file_name.
1630  *
1631  * Returns     :  path => Path was built up successfully
1632  *                NULL => Path can't be built up
1633  *
1634  *********************************************************************/
1635 static char *make_certs_path(const char *conf_dir, const char *file_name,
1636    const char *suffix)
1637 {
1638    /* Test if all given parameters are valid */
1639    if (conf_dir == NULL || *conf_dir == '\0' || file_name == NULL ||
1640       *file_name == '\0' || suffix == NULL || *suffix == '\0')
1641    {
1642       log_error(LOG_LEVEL_ERROR,
1643          "make_certs_path failed: bad input parameters");
1644       return NULL;
1645    }
1646
1647    char *path = NULL;
1648    size_t path_size = strlen(conf_dir)
1649       + strlen(file_name) + strlen(suffix) + 2;
1650
1651    /* Setting delimiter and editing path length */
1652 #if defined(_WIN32) || defined(__OS2__)
1653    char delim[] = "\\";
1654    path_size += 1;
1655 #else /* ifndef _WIN32 || __OS2__ */
1656    char delim[] = "/";
1657 #endif /* ifndef _WIN32 || __OS2__ */
1658
1659    /*
1660     * Building up path from many parts
1661     */
1662 #if defined(unix)
1663    if (*conf_dir != '/' && basedir && *basedir)
1664    {
1665       /*
1666        * Replacing conf_dir with basedir. This new variable contains
1667        * absolute path to cwd.
1668        */
1669       path_size += strlen(basedir) + 2;
1670       path = zalloc_or_die(path_size);
1671
1672       strlcpy(path, basedir,   path_size);
1673       strlcat(path, delim,     path_size);
1674       strlcat(path, conf_dir,  path_size);
1675       strlcat(path, delim,     path_size);
1676       strlcat(path, file_name, path_size);
1677       strlcat(path, suffix,    path_size);
1678    }
1679    else
1680 #endif /* defined unix */
1681    {
1682       path = zalloc_or_die(path_size);
1683
1684       strlcpy(path, conf_dir,  path_size);
1685       strlcat(path, delim,     path_size);
1686       strlcat(path, file_name, path_size);
1687       strlcat(path, suffix,    path_size);
1688    }
1689
1690    return path;
1691 }
1692
1693
1694 /*********************************************************************
1695  *
1696  * Function    :  get_certificate_serial
1697  *
1698  * Description :  Computes serial number for new certificate from host
1699  *                name hash. This hash must be already saved in csp
1700  *                structure.
1701  *
1702  * Parameters  :
1703  *          1  :  csp = Current client state (buffers, headers, etc...)
1704  *
1705  * Returns     :  Serial number for new certificate
1706  *
1707  *********************************************************************/
1708 static unsigned long get_certificate_serial(struct client_state *csp)
1709 {
1710    unsigned long exp    = 1;
1711    unsigned long serial = 0;
1712
1713    int i = CERT_SERIAL_NUM_LENGTH;
1714    /* Length of hash is 16 bytes, we must avoid to read next chars */
1715    if (i > 16)
1716    {
1717       i = 16;
1718    }
1719    if (i < 2)
1720    {
1721       i = 2;
1722    }
1723
1724    for (; i >= 0; i--)
1725    {
1726       serial += exp * (unsigned)csp->http->hash_of_host[i];
1727       exp *= 256;
1728    }
1729    return serial;
1730 }
1731
1732
1733 /*********************************************************************
1734  *
1735  * Function    :  ssl_send_certificate_error
1736  *
1737  * Description :  Sends info about invalid server certificate to client.
1738  *                Sent message is including all trusted chain certificates,
1739  *                that can be downloaded in web browser.
1740  *
1741  * Parameters  :
1742  *          1  :  csp = Current client state (buffers, headers, etc...)
1743  *
1744  * Returns     :  N/A
1745  *
1746  *********************************************************************/
1747 extern void ssl_send_certificate_error(struct client_state *csp)
1748 {
1749    size_t message_len = 0;
1750    int ret = 0;
1751    struct certs_chain *cert = NULL;
1752
1753    /* Header of message with certificate informations */
1754    const char message_begin[] =
1755       "HTTP/1.1 200 OK\r\n"
1756       "Content-Type: text/html\r\n"
1757       "Connection: close\r\n\r\n"
1758       "<html><body><h1>Server certificate verification failed</h1><p>Reason: ";
1759    const char message_end[] = "</body></html>\r\n\r\n";
1760    char reason[INVALID_CERT_INFO_BUF_SIZE];
1761    memset(reason, 0, sizeof(reason));
1762
1763    /* Get verification message from verification return code */
1764    mbedtls_x509_crt_verify_info(reason, sizeof(reason), " ",
1765       csp->server_cert_verification_result);
1766
1767    /*
1768     * Computing total length of message with all certificates inside
1769     */
1770    message_len = strlen(message_begin) + strlen(message_end)
1771                  + strlen(reason) + strlen("</p>") + 1;
1772
1773    cert = &(csp->server_certs_chain);
1774    while (cert->next != NULL)
1775    {
1776       size_t base64_len = 4 * ((strlen(cert->file_buf) + 2) / 3) + 1;
1777
1778       message_len += strlen(cert->text_buf) + strlen("<pre></pre>\n")
1779                      +  base64_len + strlen("<a href=\"data:application"
1780                         "/x-x509-ca-cert;base64,\">Download certificate</a>");
1781       cert = cert->next;
1782    }
1783
1784    /*
1785     * Joining all blocks in one long message
1786     */
1787    char message[message_len];
1788    memset(message, 0, message_len);
1789
1790    strlcpy(message, message_begin, message_len);
1791    strlcat(message, reason       , message_len);
1792    strlcat(message, "</p>"       , message_len);
1793
1794    cert = &(csp->server_certs_chain);
1795    while (cert->next != NULL)
1796    {
1797       size_t olen = 0;
1798       size_t base64_len = 4 * ((strlen(cert->file_buf) + 2) / 3) + 1; /* +1 for terminating null*/
1799       char base64_buf[base64_len];
1800       memset(base64_buf, 0, base64_len);
1801
1802       /* Encoding certificate into base64 code */
1803       ret = mbedtls_base64_encode((unsigned char*)base64_buf,
1804                base64_len, &olen, (const unsigned char*)cert->file_buf,
1805                strlen(cert->file_buf));
1806       if (ret != 0)
1807       {
1808          log_error(LOG_LEVEL_ERROR,
1809             "Encoding to base64 failed, buffer is to small");
1810       }
1811
1812       strlcat(message, "<pre>",        message_len);
1813       strlcat(message, cert->text_buf, message_len);
1814       strlcat(message, "</pre>\n",     message_len);
1815
1816       if (ret == 0)
1817       {
1818          strlcat(message, "<a href=\"data:application/x-x509-ca-cert;base64,",
1819             message_len);
1820          strlcat(message, base64_buf, message_len);
1821          strlcat(message, "\">Download certificate</a>", message_len);
1822       }
1823
1824       cert = cert->next;
1825    }
1826    strlcat(message, message_end, message_len);
1827
1828    /*
1829     * Sending final message to client
1830     */
1831    ssl_send_data(&(csp->mbedtls_client_attr.ssl),
1832       (const unsigned char *)message, strlen(message));
1833
1834    free_certificate_chain(csp);
1835 }
1836
1837
1838 /*********************************************************************
1839  *
1840  * Function    :  ssl_verify_callback
1841  *
1842  * Description :  This is a callback function for certificate verification.
1843  *                It's called for all certificates in server certificate
1844  *                trusted chain and it's preparing information about this
1845  *                certificates. Prepared informations can be used to inform
1846  *                user about invalid certificates.
1847  *
1848  * Parameters  :
1849  *          1  :  csp_void = Current client state (buffers, headers, etc...)
1850  *          2  :  crt   = certificate from trusted chain
1851  *          3  :  depth = depth in trusted chain
1852  *          4  :  flags = certificate flags
1853  *
1854  * Returns     :  0 on success and negative value on error
1855  *
1856  *********************************************************************/
1857 static int ssl_verify_callback(void *csp_void, mbedtls_x509_crt *crt,
1858    int depth, uint32_t *flags)
1859 {
1860    struct client_state *csp  = (struct client_state *)csp_void;
1861    struct certs_chain  *last = &(csp->server_certs_chain);
1862    size_t olen = 0;
1863    int ret = 0;
1864
1865    /*
1866     * Searching for last item in certificates linked list
1867     */
1868    while (last->next != NULL)
1869    {
1870       last = last->next;
1871    }
1872
1873    /*
1874     * Preparing next item in linked list for next certificate
1875     */
1876    last->next = malloc_or_die(sizeof(struct certs_chain));
1877    last->next->next = NULL;
1878    memset(last->next->text_buf, 0, sizeof(last->next->text_buf));
1879    memset(last->next->file_buf, 0, sizeof(last->next->file_buf));
1880
1881    /*
1882     * Saving certificate file into buffer
1883     */
1884    if ((ret = mbedtls_pem_write_buffer(PEM_BEGIN_CRT, PEM_END_CRT,
1885       crt->raw.p, crt->raw.len, (unsigned char *)last->file_buf,
1886       sizeof(last->file_buf)-1, &olen)) != 0)
1887    {
1888       return(ret);
1889    }
1890
1891    /*
1892     * Saving certificate information into buffer
1893     */
1894    mbedtls_x509_crt_info(last->text_buf, sizeof(last->text_buf) - 1,
1895       CERT_INFO_PREFIX, crt);
1896
1897    return 0;
1898 }
1899
1900
1901 /*********************************************************************
1902  *
1903  * Function    :  free_certificate_chain
1904  *
1905  * Description :  Frees certificates linked list. This linked list is
1906  *                used to save informations about certificates in
1907  *                trusted chain.
1908  *
1909  * Parameters  :
1910  *          1  :  csp = Current client state (buffers, headers, etc...)
1911  *
1912  * Returns     :  N/A
1913  *
1914  *********************************************************************/
1915 static void free_certificate_chain(struct client_state *csp)
1916 {
1917    struct certs_chain *cert = csp->server_certs_chain.next;
1918
1919    /* Cleaning buffers */
1920    memset(csp->server_certs_chain.text_buf, 0,
1921       sizeof(csp->server_certs_chain.text_buf));
1922    memset(csp->server_certs_chain.file_buf, 0,
1923       sizeof(csp->server_certs_chain.file_buf));
1924    csp->server_certs_chain.next = NULL;
1925
1926    /* Freeing memory in whole linked list */
1927    if (cert != NULL)
1928    {
1929       do
1930       {
1931          struct certs_chain *cert_for_free = cert;
1932          cert = cert->next;
1933          freez(cert_for_free);
1934       } while (cert != NULL);
1935    }
1936 }
1937
1938
1939 /*********************************************************************
1940  *
1941  * Function    :  file_exists
1942  *
1943  * Description :  Tests if file exists and is readable.
1944  *
1945  * Parameters  :
1946  *          1  :  path = Path to tested file.
1947  *
1948  * Returns     :  1 => File exists and is readable.
1949  *                0 => File doesn't exist or is not readable.
1950  *
1951  *********************************************************************/
1952 static int file_exists(const char *path)
1953 {
1954    FILE *f;
1955    if ((f = fopen(path, "r")) != NULL)
1956    {
1957       fclose(f);
1958       return 1;
1959    }
1960
1961    return 0;
1962 }
1963
1964
1965 /*********************************************************************
1966  *
1967  * Function    :  host_to_hash
1968  *
1969  * Description :  Creates MD5 hash from host name. Host name is loaded
1970  *                from structure csp and saved again into it.
1971  *
1972  * Parameters  :
1973  *          1  :  csp = Current client state (buffers, headers, etc...)
1974  *
1975  * Returns     :  1 => Error while creating hash
1976  *                0 => Hash created successfully
1977  *
1978  *********************************************************************/
1979 static int host_to_hash(struct client_state *csp)
1980 {
1981    int ret = 0;
1982
1983 #if !defined(MBEDTLS_MD5_C)
1984 #error mbedTLS needs to be compiled with md5 support
1985 #else
1986    memset(csp->http->hash_of_host, 0, sizeof(csp->http->hash_of_host));
1987    mbedtls_md5((unsigned char *)csp->http->host, strlen(csp->http->host),
1988       csp->http->hash_of_host);
1989
1990    /* Converting hash into string with hex */
1991    size_t i = 0;
1992    for (; i < 16; i++)
1993    {
1994       if ((ret = sprintf((char *)csp->http->hash_of_host_hex + 2 * i, "%02x",
1995          csp->http->hash_of_host[i])) < 0)
1996       {
1997          log_error(LOG_LEVEL_ERROR, "Sprintf return value: %d", ret);
1998          return -1;
1999       }
2000    }
2001
2002    return 0;
2003 #endif /* MBEDTLS_MD5_C */
2004 }
2005
2006
2007 /*********************************************************************
2008  *
2009  * Function    :  tunnel_established_successfully
2010  *
2011  * Description :  Check if parent proxy server response contains
2012  *                informations about successfully created connection with
2013  *                destination server. (HTTP/... 2xx ...)
2014  *
2015  * Parameters  :
2016  *          1  :  server_response = Buffer with parent proxy server response
2017  *          2  :  response_len = Length of server_response
2018  *
2019  * Returns     :  1 => Connection created successfully
2020  *                0 => Connection wasn't created successfully
2021  *
2022  *********************************************************************/
2023 extern int tunnel_established_successfully(const char *server_response,
2024    unsigned int response_len)
2025 {
2026    unsigned int pos = 0;
2027
2028    if (server_response == NULL)
2029    {
2030       return 0;
2031    }
2032
2033    /* Tests if "HTTP/" string is at the begin of received response */
2034    if (strncmp(server_response, "HTTP/", 5) != 0)
2035    {
2036       return 0;
2037    }
2038
2039    for (pos = 0; pos < response_len; pos++)
2040    {
2041       if (server_response[pos] == ' ')
2042       {
2043          break;
2044       }
2045    }
2046
2047    /*
2048     * response_len -3 because of buffer end, response structure and 200 code.
2049     * There must be at least 3 chars after space.
2050     * End of buffer: ... 2xx'\0'
2051     *             pos = |
2052     */
2053    if (pos >= (response_len - 3))
2054    {
2055       return 0;
2056    }
2057
2058    /* Test HTTP status code */
2059    if (server_response[pos + 1] != '2')
2060    {
2061       return 0;
2062    }
2063
2064    return 1;
2065 }
2066
2067
2068 /*********************************************************************
2069  *
2070  * Function    :  seed_rng
2071  *
2072  * Description :  Seeding the RNG for all SSL uses
2073  *
2074  * Parameters  :
2075  *          1  :  csp = Current client state (buffers, headers, etc...)
2076  *
2077  * Returns     : -1 => RNG wasn't seed successfully
2078  *                0 => RNG is seeded successfully
2079  *
2080  *********************************************************************/
2081 static int seed_rng(struct client_state *csp)
2082 {
2083    int ret = 0;
2084    char err_buf[ERROR_BUF_SIZE];
2085
2086    if (rng_seeded == 0)
2087    {
2088       privoxy_mutex_lock(&rng_mutex);
2089       if (rng_seeded == 0)
2090       {
2091          mbedtls_ctr_drbg_init(&ctr_drbg);
2092          mbedtls_entropy_init(&entropy);
2093          ret = mbedtls_ctr_drbg_seed(&ctr_drbg, mbedtls_entropy_func,
2094             &entropy, NULL, 0);
2095          if (ret != 0)
2096          {
2097             mbedtls_strerror(ret, err_buf, sizeof(err_buf));
2098             log_error(LOG_LEVEL_ERROR,
2099                "mbedtls_ctr_drbg_seed failed: %s", err_buf);
2100             privoxy_mutex_unlock(&rng_mutex);
2101             return -1;
2102          }
2103          rng_seeded = 1;
2104       }
2105       privoxy_mutex_unlock(&rng_mutex);
2106    }
2107    return 0;
2108 }