9dc14ab57892787a744f73ee0c23aca9639f07fd
[privoxy.git] / 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 int get_certificate_mutex_id(struct client_state *csp);
117 static unsigned long  get_certificate_serial(struct client_state *csp);
118 static void free_client_ssl_structures(struct client_state *csp);
119 static void free_server_ssl_structures(struct client_state *csp);
120 static int seed_rng(struct client_state *csp);
121
122 /*********************************************************************
123  *
124  * Function    :  client_use_ssl
125  *
126  * Description :  Tests if client in current client state structure
127  *                should use SSL connection or standard connection.
128  *
129  * Parameters  :
130  *          1  :  csp = Current client state (buffers, headers, etc...)
131  *
132  * Returns     :  If client should use TLS/SSL connection, 1 is returned.
133  *                Otherwise 0 is returned.
134  *
135  *********************************************************************/
136 extern int client_use_ssl(const struct client_state *csp)
137 {
138    return csp->http->client_ssl;
139 }
140
141
142 /*********************************************************************
143  *
144  * Function    :  server_use_ssl
145  *
146  * Description :  Tests if server in current client state structure
147  *                should use SSL connection or standard connection.
148  *
149  * Parameters  :
150  *          1  :  csp = Current client state (buffers, headers, etc...)
151  *
152  * Returns     :  If server should use TLS/SSL connection, 1 is returned.
153  *                Otherwise 0 is returned.
154  *
155  *********************************************************************/
156 extern int server_use_ssl(const struct client_state *csp)
157 {
158    return csp->http->server_ssl;
159 }
160
161
162 /*********************************************************************
163  *
164  * Function    :  is_ssl_pending
165  *
166  * Description :  Tests if there are some waiting data on ssl connection
167  *
168  * Parameters  :
169  *          1  :  ssl = SSL context to test
170  *
171  * Returns     :   0 => No data are pending
172  *                >0 => Pending data length
173  *
174  *********************************************************************/
175 extern size_t is_ssl_pending(mbedtls_ssl_context *ssl)
176 {
177    if (ssl == NULL)
178    {
179       return 0;
180    }
181
182    return mbedtls_ssl_get_bytes_avail(ssl);
183 }
184
185
186 /*********************************************************************
187  *
188  * Function    :  ssl_send_data
189  *
190  * Description :  Sends the content of buf (for n bytes) to given SSL
191  *                connection context.
192  *
193  * Parameters  :
194  *          1  :  ssl = SSL context to send data to
195  *          2  :  buf = Pointer to data to be sent
196  *          3  :  len = Length of data to be sent to the SSL context
197  *
198  * Returns     :  Length of sent data or negative value on error.
199  *
200  *********************************************************************/
201 extern int ssl_send_data(mbedtls_ssl_context *ssl, const unsigned char *buf, size_t len)
202 {
203    int ret = 0;
204    size_t max_fragment_size = 0;  /* Maximal length of data in one SSL fragment*/
205    int send_len             = 0;  /* length of one data part to send */
206    int pos                  = 0;  /* Position of unsent part in buffer */
207
208    if (len == 0)
209    {
210       return 0;
211    }
212
213    /* Getting maximal length of data sent in one fragment */
214    max_fragment_size = mbedtls_ssl_get_max_frag_len(ssl);
215
216    /*
217     * Whole buffer must be sent in many fragments, because each fragment
218     * has its maximal length.
219     */
220    while (pos < len)
221    {
222       /* Compute length of data, that can be send in next fragment */
223       if ((pos + (int)max_fragment_size) > len)
224       {
225          send_len = (int)len - pos;
226       }
227       else
228       {
229          send_len = (int)max_fragment_size;
230       }
231
232       /*
233        * Sending one part of the buffer
234        */
235       while ((ret = mbedtls_ssl_write(ssl,
236          (const unsigned char *)(buf + pos),
237          (size_t)send_len)) < 0)
238       {
239          if (ret != MBEDTLS_ERR_SSL_WANT_READ &&
240              ret != MBEDTLS_ERR_SSL_WANT_WRITE)
241          {
242             char err_buf[ERROR_BUF_SIZE];
243
244             mbedtls_strerror(ret, err_buf, sizeof(err_buf));
245             log_error(LOG_LEVEL_ERROR,
246                "Sending data over TLS/SSL failed: %s", err_buf);
247             return -1;
248          }
249       }
250       /* Adding count of sent bytes to position in buffer */
251       pos = pos + send_len;
252    }
253
254    return (int)len;
255 }
256
257
258 /*********************************************************************
259  *
260  * Function    :  ssl_recv_data
261  *
262  * Description :  Receives data from given SSL context and puts
263  *                it into buffer.
264  *
265  * Parameters  :
266  *          1  :  ssl = SSL context to receive data from
267  *          2  :  buf = Pointer to buffer where data will be written
268  *          3  :  max_length = Maximum number of bytes to read
269  *
270  * Returns     :  Number of bytes read, 0 for EOF, or -1
271  *                on error.
272  *
273  *********************************************************************/
274 extern int ssl_recv_data(mbedtls_ssl_context *ssl, unsigned char *buf, size_t max_length)
275 {
276    int ret = 0;
277    memset(buf, 0, max_length);
278
279    /*
280     * Receiving data from SSL context into buffer
281     */
282    do
283    {
284       ret = mbedtls_ssl_read(ssl, buf, max_length);
285    } while (ret == MBEDTLS_ERR_SSL_WANT_READ
286       || ret == MBEDTLS_ERR_SSL_WANT_WRITE);
287
288    if (ret < 0)
289    {
290       char err_buf[ERROR_BUF_SIZE];
291
292       if (ret == MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY)
293       {
294          log_error(LOG_LEVEL_CONNECT,
295             "The peer notified us that the connection is going to be closed");
296          return 0;
297       }
298       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
299       log_error(LOG_LEVEL_ERROR,
300          "Receiving data over TLS/SSL failed: %s", err_buf);
301
302       return -1;
303    }
304
305    log_error(LOG_LEVEL_RECEIVED, "TLS: %N", ret, buf);
306
307    return ret;
308 }
309
310
311 /*********************************************************************
312  *
313  * Function    :  ssl_flush_socket
314  *
315  * Description :  Send any pending "buffered" content with given
316  *                SSL connection. Alternative to function flush_socket.
317  *
318  * Parameters  :
319  *          1  :  ssl = SSL context to send buffer to
320  *          2  :  iob = The I/O buffer to flush, usually csp->iob.
321  *
322  * Returns     :  On success, the number of bytes send are returned (zero
323  *                indicates nothing was sent).  On error, -1 is returned.
324  *
325  *********************************************************************/
326 extern long ssl_flush_socket(mbedtls_ssl_context *ssl, struct iob *iob)
327 {
328    /* Computing length of buffer part to send */
329    long len = iob->eod - iob->cur;
330
331    if (len <= 0)
332    {
333       return(0);
334    }
335
336    /* Sending data to given SSl context */
337    if (ssl_send_data(ssl, (const unsigned char *)iob->cur, (size_t)len) < 0)
338    {
339       return -1;
340    }
341    iob->eod = iob->cur = iob->buf;
342    return(len);
343 }
344
345
346 /*********************************************************************
347  *
348  * Function    :  ssl_debug_callback
349  *
350  * Description :  Debug callback function for mbedtls library.
351  *                Prints info into log file.
352  *
353  * Parameters  :
354  *          1  :  ctx   = File to save log in
355  *          2  :  level = Debug level
356  *          3  :  file  = File calling debug message
357  *          4  :  line  = Line calling debug message
358  *          5  :  str   = Debug message
359  *
360  * Returns     :  N/A
361  *
362  *********************************************************************/
363 static void ssl_debug_callback(void *ctx, int level, const char *file, int line, const char *str)
364 {
365    /*
366    ((void)level);
367    fprintf((FILE *)ctx, "%s:%04d: %s", file, line, str);
368    fflush((FILE *)ctx);
369    log_error(LOG_LEVEL_INFO, "SSL debug message: %s:%04d: %s", file, line, str);
370    */
371 }
372
373
374 /*********************************************************************
375  *
376  * Function    :  create_client_ssl_connection
377  *
378  * Description :  Creates TLS/SSL secured connection with client
379  *
380  * Parameters  :
381  *          1  :  csp = Current client state (buffers, headers, etc...)
382  *
383  * Returns     :  0 on success, negative value if connection wasn't created
384  *                successfully.
385  *
386  *********************************************************************/
387 extern int create_client_ssl_connection(struct client_state *csp)
388 {
389    /* Paths to certificates file and key file */
390    char *key_file  = NULL;
391    char *ca_file   = NULL;
392    char *cert_file = NULL;
393    int ret = 0;
394    char err_buf[ERROR_BUF_SIZE];
395
396    /*
397     * Initializing mbedtls structures for TLS/SSL connection
398     */
399    mbedtls_net_init(&(csp->mbedtls_client_attr.socket_fd));
400    mbedtls_ssl_init(&(csp->mbedtls_client_attr.ssl));
401    mbedtls_ssl_config_init(&(csp->mbedtls_client_attr.conf));
402    mbedtls_x509_crt_init(&(csp->mbedtls_client_attr.server_cert));
403    mbedtls_pk_init(&(csp->mbedtls_client_attr.prim_key));
404 #if defined(MBEDTLS_SSL_CACHE_C)
405    mbedtls_ssl_cache_init(&(csp->mbedtls_client_attr.cache));
406 #endif
407
408    /*
409     * Preparing hash of host for creating certificates
410     */
411    ret = host_to_hash(csp);
412    if (ret != 0)
413    {
414       log_error(LOG_LEVEL_ERROR, "Generating hash of host failed: %d", ret);
415       ret = -1;
416       goto exit;
417    }
418
419    /*
420     * Preparing paths to certificates files and key file
421     */
422    ca_file   = csp->config->ca_cert_file;
423    cert_file = make_certs_path(csp->config->certificate_directory,
424       (const char *)csp->http->hash_of_host_hex, CERT_FILE_TYPE);
425    key_file  = make_certs_path(csp->config->certificate_directory,
426       (const char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
427
428    if (cert_file == NULL || key_file == NULL)
429    {
430       ret = -1;
431       goto exit;
432    }
433
434    /*
435     * Generating certificate for requested host. Mutex to prevent
436     * certificate and key inconsistence must be locked.
437     */
438    unsigned int cert_mutex_id = get_certificate_mutex_id(csp);
439    privoxy_mutex_lock(&(certificates_mutexes[cert_mutex_id]));
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(&(certificates_mutexes[cert_mutex_id]));
447       ret = -1;
448       goto exit;
449    }
450    privoxy_mutex_unlock(&(certificates_mutexes[cert_mutex_id]));
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 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             log_error(LOG_LEVEL_ERROR,
815                "Server certificate verification failed: %s", err_buf);
816             csp->server_cert_verification_result =
817                mbedtls_ssl_get_verify_result(&(csp->mbedtls_server_attr.ssl));
818
819             ret = -1;
820          }
821          else
822          {
823             log_error(LOG_LEVEL_ERROR,
824                "mbedtls_ssl_handshake with server failed: %s", err_buf);
825             ret = -1;
826          }
827          goto exit;
828       }
829    }
830
831    log_error(LOG_LEVEL_CONNECT, "Server successfully connected over TLS/SSL");
832
833    /*
834     * Server certificate chain is valid, so we can clean
835     * chain, because we will not send it to client.
836     */
837    free_certificate_chain(csp);
838
839    csp->ssl_with_server_is_opened = 1;
840    csp->server_cert_verification_result =
841       mbedtls_ssl_get_verify_result(&(csp->mbedtls_server_attr.ssl));
842
843 exit:
844    /* Freeing structures if connection wasn't created successfully */
845    if (ret < 0)
846    {
847       free_server_ssl_structures(csp);
848    }
849
850    return ret;
851 }
852
853
854 /*********************************************************************
855  *
856  * Function    :  close_server_ssl_connection
857  *
858  * Description :  Closes TLS/SSL connection with server. This function
859  *                checks if this connection is already opened.
860  *
861  * Parameters  :
862  *          1  :  csp = Current client state (buffers, headers, etc...)
863  *
864  * Returns     :  N/A
865  *
866  *********************************************************************/
867 static void close_server_ssl_connection(struct client_state *csp)
868 {
869    int ret = 0;
870
871    if (csp->ssl_with_server_is_opened == 0)
872    {
873       return;
874    }
875
876    /*
877    * Notifying the peer that the connection is being closed.
878    */
879    do {
880       ret = mbedtls_ssl_close_notify(&(csp->mbedtls_server_attr.ssl));
881    } while (ret == MBEDTLS_ERR_SSL_WANT_WRITE);
882
883    free_server_ssl_structures(csp);
884    csp->ssl_with_server_is_opened = 0;
885 }
886
887
888 /*********************************************************************
889  *
890  * Function    :  free_server_ssl_structures
891  *
892  * Description :  Frees structures used for SSL communication with server
893  *
894  * Parameters  :
895  *          1  :  csp = Current client state (buffers, headers, etc...)
896  *
897  * Returns     :  N/A
898  *
899  *********************************************************************/
900 static void free_server_ssl_structures(struct client_state *csp)
901 {
902    /*
903    * We can't use function mbedtls_net_free, because this function
904    * inter alia close TCP connection on setted fd. Instead of this
905    * function, we change fd to -1, which is the same what does
906    * rest of mbedtls_net_free function.
907    */
908    csp->mbedtls_client_attr.socket_fd.fd = -1;
909
910    mbedtls_x509_crt_free(&(csp->mbedtls_server_attr.ca_cert));
911    mbedtls_ssl_free(&(csp->mbedtls_server_attr.ssl));
912    mbedtls_ssl_config_free(&(csp->mbedtls_server_attr.conf));
913 }
914
915
916 /*********************************************************************
917  *
918  * Function    :  close_client_and_server_ssl_connections
919  *
920  * Description :  Checks if client or server should use secured
921  *                connection over SSL and if so, closes all of them.
922  *
923  * Parameters  :
924  *          1  :  csp = Current client state (buffers, headers, etc...)
925  *
926  * Returns     :  N/A
927  *
928  *********************************************************************/
929 extern void close_client_and_server_ssl_connections(struct client_state *csp)
930 {
931    if (client_use_ssl(csp) == 1)
932    {
933       close_client_ssl_connection(csp);
934    }
935    if (server_use_ssl(csp) == 1)
936    {
937       close_server_ssl_connection(csp);
938    }
939 }
940
941 /*====================== Certificates ======================*/
942
943 /*********************************************************************
944  *
945  * Function    :  write_certificate
946  *
947  * Description :  Writes certificate into file.
948  *
949  * Parameters  :
950  *          1  :  crt = certificate to write into file
951  *          2  :  output_file = path to save certificate file
952  *          3  :  f_rng = mbedtls_ctr_drbg_random
953  *          4  :  p_rng = mbedtls_ctr_drbg_context
954  *
955  * Returns     :  Length of written certificate on success or negative value
956  *                on error
957  *
958  *********************************************************************/
959 static int write_certificate(mbedtls_x509write_cert *crt, const char *output_file,
960    int(*f_rng)(void *, unsigned char *, size_t), void *p_rng)
961 {
962    FILE *f = NULL;
963    size_t len = 0;
964    unsigned char cert_buf[CERTIFICATE_BUF_SIZE + 1]; /* Buffer for certificate in PEM format + terminating NULL */
965    int ret = 0;
966    char err_buf[ERROR_BUF_SIZE];
967
968    memset(cert_buf, 0, sizeof(cert_buf));
969
970    /*
971     * Writing certificate into PEM string. If buffer is too small, function
972     * returns specific error and no buffer overflow can happen.
973     */
974    if ((ret = mbedtls_x509write_crt_pem(crt, cert_buf,
975       sizeof(cert_buf) - 1, f_rng, p_rng)) != 0)
976    {
977       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
978       log_error(LOG_LEVEL_ERROR,
979          "Writing certificate into buffer failed: %s", err_buf);
980       return -1;
981    }
982
983    len = strlen((char *)cert_buf);
984
985    /*
986     * Saving certificate into file
987     */
988    if ((f = fopen(output_file, "w")) == NULL)
989    {
990       log_error(LOG_LEVEL_ERROR, "Opening file %s to save certificate failed",
991          output_file);
992       return -1;
993    }
994
995    if (fwrite(cert_buf, 1, len, f) != len)
996    {
997       log_error(LOG_LEVEL_ERROR,
998          "Writing certificate into file %s failed", output_file);
999       fclose(f);
1000       return -1;
1001    }
1002
1003    fclose(f);
1004
1005    return (int)len;
1006 }
1007
1008
1009 /*********************************************************************
1010  *
1011  * Function    :  write_private_key
1012  *
1013  * Description :  Writes private key into file and copies saved
1014  *                content into given pointer to string. If function
1015  *                returns 0 for success, this copy must be freed by
1016  *                caller.
1017  *
1018  * Parameters  :
1019  *          1  :  key = key to write into file
1020  *          2  :  ret_buf = pointer to string with created key file content
1021  *          3  :  key_file_path = path where to save key file
1022  *
1023  * Returns     :  Length of written private key on success or negative value
1024  *                on error
1025  *
1026  *********************************************************************/
1027 static int write_private_key(mbedtls_pk_context *key, unsigned char **ret_buf,
1028    const char *key_file_path)
1029 {
1030    size_t len = 0;                /* Length of created key    */
1031    FILE *f = NULL;                /* File to save certificate */
1032    int ret = 0;
1033    char err_buf[ERROR_BUF_SIZE];
1034
1035    /* Initializing buffer for key file content */
1036    *ret_buf = zalloc_or_die(PRIVATE_KEY_BUF_SIZE + 1);
1037
1038    /*
1039     * Writing private key into PEM string
1040     */
1041    if ((ret = mbedtls_pk_write_key_pem(key, *ret_buf, PRIVATE_KEY_BUF_SIZE)) != 0)
1042    {
1043       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1044       log_error(LOG_LEVEL_ERROR,
1045          "Writing private key into PEM string failed: %s", err_buf);
1046       ret = -1;
1047       goto exit;
1048    }
1049    len = strlen((char *)*ret_buf);
1050
1051    /*
1052     * Saving key into file
1053     */
1054    if ((f = fopen(key_file_path, "wb")) == NULL)
1055    {
1056       log_error(LOG_LEVEL_ERROR,
1057          "Opening file %s to save private key failed: %E",
1058          key_file_path);
1059       ret = -1;
1060       goto exit;
1061    }
1062
1063    if (fwrite(*ret_buf, 1, len, f) != len)
1064    {
1065       fclose(f);
1066       log_error(LOG_LEVEL_ERROR,
1067          "Writing private key into file %s failed",
1068          key_file_path);
1069       ret = -1;
1070       goto exit;
1071    }
1072
1073    fclose(f);
1074
1075 exit:
1076    if (ret < 0)
1077    {
1078       freez(*ret_buf);
1079       *ret_buf = NULL;
1080       return ret;
1081    }
1082    return (int)len;
1083 }
1084
1085
1086 /*********************************************************************
1087  *
1088  * Function    :  generate_key
1089  *
1090  * Description : Tests if private key for host saved in csp already
1091  *               exists.  If this file doesn't exists, a new key is
1092  *               generated and saved in a file. The generated key is also
1093  *               copied into given parameter key_buf, which must be then
1094  *               freed by caller. If file with key exists, key_buf
1095  *               contain NULL and no private key is generated.
1096  *
1097  * Parameters  :
1098  *          1  :  csp = Current client state (buffers, headers, etc...)
1099  *          2  :  key_buf = buffer to save new generated key
1100  *
1101  * Returns     :  -1 => Error while generating private key
1102  *                 0 => Key already exists
1103  *                >0 => Length of generated private key
1104  *
1105  *********************************************************************/
1106 static int generate_key(struct client_state *csp, unsigned char **key_buf)
1107 {
1108    mbedtls_pk_context key;
1109    key_options key_opt;
1110    int ret = 0;
1111    char err_buf[ERROR_BUF_SIZE];
1112
1113    key_opt.key_file_path = NULL;
1114
1115    /*
1116     * Initializing structures for key generating
1117     */
1118    mbedtls_pk_init(&key);
1119
1120    /*
1121     * Preparing path for key file and other properties for generating key
1122     */
1123    key_opt.type        = MBEDTLS_PK_RSA;
1124    key_opt.rsa_keysize = RSA_KEYSIZE;
1125
1126    key_opt.key_file_path = make_certs_path(csp->config->certificate_directory,
1127       (char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
1128    if (key_opt.key_file_path == NULL)
1129    {
1130       ret = -1;
1131       goto exit;
1132    }
1133
1134    /*
1135     * Test if key already exists. If so, we don't have to create it again.
1136     */
1137    if (file_exists(key_opt.key_file_path) == 1)
1138    {
1139       ret = 0;
1140       goto exit;
1141    }
1142
1143    /*
1144     * Seed the RNG
1145     */
1146    ret = seed_rng(csp);
1147    if (ret != 0)
1148    {
1149       ret = -1;
1150       goto exit;
1151    }
1152
1153    /*
1154     * Setting attributes of private key and generating it
1155     */
1156    if ((ret = mbedtls_pk_setup(&key,
1157       mbedtls_pk_info_from_type(key_opt.type))) != 0)
1158    {
1159       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1160       log_error(LOG_LEVEL_ERROR, "mbedtls_pk_setup failed: %s", err_buf);
1161       ret = -1;
1162       goto exit;
1163    }
1164
1165    ret = mbedtls_rsa_gen_key(mbedtls_pk_rsa(key), mbedtls_ctr_drbg_random,
1166       &ctr_drbg, (unsigned)key_opt.rsa_keysize, RSA_KEY_PUBLIC_EXPONENT);
1167    if (ret != 0)
1168    {
1169       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1170       log_error(LOG_LEVEL_ERROR, "Key generating failed: %s", err_buf);
1171       ret = -1;
1172       goto exit;
1173    }
1174
1175    /*
1176     * Exporting private key into file
1177     */
1178    if ((ret = write_private_key(&key, key_buf, key_opt.key_file_path)) < 0)
1179    {
1180       log_error(LOG_LEVEL_ERROR,
1181          "Writing private key into file %s failed", key_opt.key_file_path);
1182       ret = -1;
1183       goto exit;
1184    }
1185
1186 exit:
1187    /*
1188     * Freeing used variables
1189     */
1190    freez(key_opt.key_file_path);
1191
1192    mbedtls_pk_free(&key);
1193
1194    return ret;
1195 }
1196
1197
1198 /*********************************************************************
1199  *
1200  * Function    :  generate_webpage_certificate
1201  *
1202  * Description :  Creates certificate file in presetted directory.
1203  *                If certificate already exists, no other certificate
1204  *                will be created. Subject of certificate is named
1205  *                by csp->http->host from parameter. This function also
1206  *                triggers generating of private key for new certificate.
1207  *
1208  * Parameters  :
1209  *          1  :  csp = Current client state (buffers, headers, etc...)
1210  *
1211  * Returns     :  -1 => Error while creating certificate.
1212  *                 0 => Certificate already exists.
1213  *                >0 => Length of created certificate.
1214  *
1215  *********************************************************************/
1216 static int generate_webpage_certificate(struct client_state *csp)
1217 {
1218    mbedtls_x509_crt issuer_cert;
1219    mbedtls_pk_context loaded_issuer_key, loaded_subject_key;
1220    mbedtls_pk_context *issuer_key  = &loaded_issuer_key;
1221    mbedtls_pk_context *subject_key = &loaded_subject_key;
1222    mbedtls_x509write_cert cert;
1223    mbedtls_mpi serial;
1224
1225    unsigned char *key_buf = NULL;    /* Buffer for created key */
1226
1227    int ret = 0;
1228    char err_buf[ERROR_BUF_SIZE];
1229    cert_options cert_opt;
1230
1231    /* Paths to keys and certificates needed to create certificate */
1232    cert_opt.issuer_key  = NULL;
1233    cert_opt.subject_key = NULL;
1234    cert_opt.issuer_crt  = NULL;
1235    cert_opt.output_file = NULL;
1236
1237    /*
1238     * Create key for requested host
1239     */
1240    int subject_key_len = generate_key(csp, &key_buf);
1241    if (subject_key_len < 0)
1242    {
1243       log_error(LOG_LEVEL_ERROR, "Key generating failed");
1244       return -1;
1245    }
1246
1247    /*
1248     * Initializing structures for certificate generating
1249     */
1250    mbedtls_x509write_crt_init(&cert);
1251    mbedtls_x509write_crt_set_md_alg(&cert, CERT_SIGNATURE_ALGORITHM);
1252    mbedtls_pk_init(&loaded_issuer_key);
1253    mbedtls_pk_init(&loaded_subject_key);
1254    mbedtls_mpi_init(&serial);
1255    mbedtls_x509_crt_init(&issuer_cert);
1256
1257    /*
1258     * Presetting parameters for certificate. We must compute total length
1259     * of parameters.
1260     */
1261    size_t cert_params_len = strlen(CERT_PARAM_COMMON_NAME) +
1262       strlen(CERT_PARAM_ORGANIZATION) + strlen(CERT_PARAM_COUNTRY) +
1263       strlen(CERT_PARAM_ORG_UNIT) +
1264       3 * strlen(csp->http->host) + 1;
1265    char cert_params[cert_params_len];
1266    memset(cert_params, 0, cert_params_len);
1267
1268    /*
1269     * Converting unsigned long serial number to char * serial number.
1270     * We must compute length of serial number in string + terminating null.
1271     */
1272    unsigned long certificate_serial = get_certificate_serial(csp);
1273    int serial_num_size = snprintf(NULL, 0, "%lu", certificate_serial) + 1;
1274    if (serial_num_size <= 0)
1275    {
1276       serial_num_size = 1;
1277    }
1278
1279    char serial_num_text[serial_num_size];  /* Buffer for serial number */
1280    ret = snprintf(serial_num_text, (size_t)serial_num_size, "%lu", certificate_serial);
1281    if (ret < 0 || ret >= serial_num_size)
1282    {
1283       log_error(LOG_LEVEL_ERROR,
1284          "Converting certificate serial number into string failed");
1285       ret = -1;
1286       goto exit;
1287    }
1288
1289    /*
1290     * Preparing parameters for certificate
1291     */
1292    strlcpy(cert_params, CERT_PARAM_COMMON_NAME,  cert_params_len);
1293    strlcat(cert_params, csp->http->host,         cert_params_len);
1294    strlcat(cert_params, CERT_PARAM_ORGANIZATION, cert_params_len);
1295    strlcat(cert_params, csp->http->host,         cert_params_len);
1296    strlcat(cert_params, CERT_PARAM_ORG_UNIT,     cert_params_len);
1297    strlcat(cert_params, csp->http->host,         cert_params_len);
1298    strlcat(cert_params, CERT_PARAM_COUNTRY,      cert_params_len);
1299
1300    cert_opt.issuer_crt = csp->config->ca_cert_file;
1301    cert_opt.issuer_key = csp->config->ca_key_file;
1302    cert_opt.subject_key = make_certs_path(csp->config->certificate_directory,
1303       (const char *)csp->http->hash_of_host_hex, KEY_FILE_TYPE);
1304    cert_opt.output_file = make_certs_path(csp->config->certificate_directory,
1305       (const char *)csp->http->hash_of_host_hex, CERT_FILE_TYPE);
1306
1307    if (cert_opt.subject_key == NULL || cert_opt.output_file == NULL)
1308    {
1309       ret = -1;
1310       goto exit;
1311    }
1312
1313    cert_opt.subject_pwd   = CERT_SUBJECT_PASSWORD;
1314    cert_opt.issuer_pwd    = csp->config->ca_password;
1315    cert_opt.subject_name  = cert_params;
1316    cert_opt.not_before    = GENERATED_CERT_VALID_FROM;
1317    cert_opt.not_after     = GENERATED_CERT_VALID_TO;
1318    cert_opt.serial        = serial_num_text;
1319    cert_opt.is_ca         = 0;
1320    cert_opt.max_pathlen   = -1;
1321
1322    /*
1323     * Test if certificate exists and private key was already created
1324     */
1325    if (file_exists(cert_opt.output_file) == 1 && subject_key_len == 0)
1326    {
1327       ret = 0;
1328       goto exit;
1329    }
1330
1331    /*
1332     * Seed the PRNG
1333     */
1334    ret = seed_rng(csp);
1335    if (ret != 0)
1336    {
1337       ret = -1;
1338       goto exit;
1339    }
1340
1341    /*
1342     * Parse serial to MPI
1343     */
1344    ret = mbedtls_mpi_read_string(&serial, 10, cert_opt.serial);
1345    if (ret != 0)
1346    {
1347       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1348       log_error(LOG_LEVEL_ERROR,
1349          "mbedtls_mpi_read_string failed: %s", err_buf);
1350       ret = -1;
1351       goto exit;
1352    }
1353
1354    /*
1355     * Loading certificates
1356     */
1357    ret = mbedtls_x509_crt_parse_file(&issuer_cert, cert_opt.issuer_crt);
1358    if (ret != 0)
1359    {
1360       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1361       log_error(LOG_LEVEL_ERROR, "Loading issuer certificate %s failed: %s",
1362          cert_opt.issuer_crt, err_buf);
1363       ret = -1;
1364       goto exit;
1365    }
1366
1367    ret = mbedtls_x509_dn_gets(cert_opt.issuer_name,
1368       sizeof(cert_opt.issuer_name), &issuer_cert.subject);
1369    if (ret < 0)
1370    {
1371       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1372       log_error(LOG_LEVEL_ERROR, "mbedtls_x509_dn_gets failed: %s", err_buf);
1373       ret = -1;
1374       goto exit;
1375    }
1376
1377    /*
1378     * Loading keys from file or from buffer
1379     */
1380    if (key_buf != NULL && subject_key_len > 0)
1381    {
1382       /* Key was created in this function and is stored in buffer */
1383       ret = mbedtls_pk_parse_key(&loaded_subject_key, key_buf,
1384          (size_t)(subject_key_len + 1), (unsigned const char *)
1385          cert_opt.subject_pwd, strlen(cert_opt.subject_pwd));
1386    }
1387    else
1388    {
1389       /* Key wasn't created in this function, because it already existed */
1390       ret = mbedtls_pk_parse_keyfile(&loaded_subject_key,
1391          cert_opt.subject_key, cert_opt.subject_pwd);
1392    }
1393
1394    if (ret != 0)
1395    {
1396       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1397       log_error(LOG_LEVEL_ERROR, "Parsing subject key %s failed: %s",
1398          cert_opt.subject_key, err_buf);
1399       ret = -1;
1400       goto exit;
1401    }
1402
1403    ret = mbedtls_pk_parse_keyfile(&loaded_issuer_key, cert_opt.issuer_key,
1404       cert_opt.issuer_pwd);
1405    if (ret != 0)
1406    {
1407       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1408       log_error(LOG_LEVEL_ERROR,
1409          "Parsing issuer key %s failed: %s", cert_opt.issuer_key, err_buf);
1410       ret = -1;
1411       goto exit;
1412    }
1413
1414    /*
1415     * Check if key and issuer certificate match
1416     */
1417    if (!mbedtls_pk_can_do(&issuer_cert.pk, MBEDTLS_PK_RSA) ||
1418       mbedtls_mpi_cmp_mpi(&mbedtls_pk_rsa(issuer_cert.pk)->N,
1419          &mbedtls_pk_rsa(*issuer_key)->N) != 0 ||
1420       mbedtls_mpi_cmp_mpi(&mbedtls_pk_rsa(issuer_cert.pk)->E,
1421          &mbedtls_pk_rsa(*issuer_key)->E) != 0)
1422    {
1423       log_error(LOG_LEVEL_ERROR,
1424          "Issuer key doesn't match issuer certificate");
1425       ret = -1;
1426       goto exit;
1427    }
1428
1429    mbedtls_x509write_crt_set_subject_key(&cert, subject_key);
1430    mbedtls_x509write_crt_set_issuer_key(&cert, issuer_key);
1431
1432    /*
1433     * Setting parameters of signed certificate
1434     */
1435    ret = mbedtls_x509write_crt_set_subject_name(&cert, cert_opt.subject_name);
1436    if (ret != 0)
1437    {
1438       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1439       log_error(LOG_LEVEL_ERROR,
1440          "Setting subject name in signed certificate failed: %s", err_buf);
1441       ret = -1;
1442       goto exit;
1443    }
1444
1445    ret = mbedtls_x509write_crt_set_issuer_name(&cert, cert_opt.issuer_name);
1446    if (ret != 0)
1447    {
1448       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1449       log_error(LOG_LEVEL_ERROR,
1450          "Setting issuer name in signed certificate failed: %s", err_buf);
1451       ret = -1;
1452       goto exit;
1453    }
1454
1455    ret = mbedtls_x509write_crt_set_serial(&cert, &serial);
1456    if (ret != 0)
1457    {
1458       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1459       log_error(LOG_LEVEL_ERROR,
1460          "Setting serial number in signed certificate failed: %s", err_buf);
1461       ret = -1;
1462       goto exit;
1463    }
1464
1465    ret = mbedtls_x509write_crt_set_validity(&cert, cert_opt.not_before,
1466       cert_opt.not_after);
1467    if (ret != 0)
1468    {
1469       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1470       log_error(LOG_LEVEL_ERROR,
1471          "Setting validity in signed certificate failed: %s", err_buf);
1472       ret = -1;
1473       goto exit;
1474    }
1475
1476    /*
1477     * Setting the basicConstraints extension for certificate
1478     */
1479    ret = mbedtls_x509write_crt_set_basic_constraints(&cert, cert_opt.is_ca,
1480       cert_opt.max_pathlen);
1481    if (ret != 0)
1482    {
1483       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1484       log_error(LOG_LEVEL_ERROR, "Setting the basicConstraints extension "
1485          "in signed certificate failed: %s", err_buf);
1486       ret = -1;
1487       goto exit;
1488    }
1489
1490 #if defined(MBEDTLS_SHA1_C)
1491    /* Setting the subjectKeyIdentifier extension for certificate */
1492    ret = mbedtls_x509write_crt_set_subject_key_identifier(&cert);
1493    if (ret != 0)
1494    {
1495       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1496       log_error(LOG_LEVEL_ERROR, "mbedtls_x509write_crt_set_subject_key_"
1497          "identifier failed: %s", err_buf);
1498       ret = -1;
1499       goto exit;
1500    }
1501
1502    /* Setting the authorityKeyIdentifier extension for certificate */
1503    ret = mbedtls_x509write_crt_set_authority_key_identifier(&cert);
1504    if (ret != 0)
1505    {
1506       mbedtls_strerror(ret, err_buf, sizeof(err_buf));
1507       log_error(LOG_LEVEL_ERROR, "mbedtls_x509write_crt_set_authority_key_"
1508          "identifier failed: %s", err_buf);
1509       ret = -1;
1510       goto exit;
1511    }
1512 #endif /* MBEDTLS_SHA1_C */
1513
1514    /*
1515     * Writing certificate into file
1516     */
1517    ret = write_certificate(&cert, cert_opt.output_file,
1518       mbedtls_ctr_drbg_random, &ctr_drbg);
1519    if (ret < 0)
1520    {
1521       log_error(LOG_LEVEL_ERROR, "Writing certificate into file failed");
1522       goto exit;
1523    }
1524
1525 exit:
1526    /*
1527     * Freeing used structures
1528     */
1529    mbedtls_x509write_crt_free(&cert);
1530    mbedtls_pk_free(&loaded_subject_key);
1531    mbedtls_pk_free(&loaded_issuer_key);
1532    mbedtls_mpi_free(&serial);
1533    mbedtls_x509_crt_free(&issuer_cert);
1534
1535    freez(cert_opt.subject_key);
1536    freez(cert_opt.output_file);
1537    freez(key_buf);
1538
1539    return ret;
1540 }
1541
1542
1543 /*********************************************************************
1544  *
1545  * Function    :  make_certs_path
1546  *
1547  * Description : Creates path to file from three pieces. This function
1548  *               takes parameters and puts them in one new mallocated
1549  *               char * in correct order. Returned variable must be freed
1550  *               by caller. This function is mainly used for creating
1551  *               paths of certificates and keys files.
1552  *
1553  * Parameters  :
1554  *          1  :  conf_dir  = Name/path of directory where is the file.
1555  *                            '.' can be used for current directory.
1556  *          2  :  file_name = Name of file in conf_dir without suffix.
1557  *          3  :  suffix    = Suffix of given file_name.
1558  *
1559  * Returns     :  path => Path was built up successfully
1560  *                NULL => Path can't be built up
1561  *
1562  *********************************************************************/
1563 static char *make_certs_path(const char *conf_dir, const char *file_name,
1564    const char *suffix)
1565 {
1566    /* Test if all given parameters are valid */
1567    if (conf_dir == NULL || *conf_dir == '\0' || file_name == NULL ||
1568       *file_name == '\0' || suffix == NULL || *suffix == '\0')
1569    {
1570       log_error(LOG_LEVEL_ERROR,
1571          "make_certs_path failed: bad input parameters");
1572       return NULL;
1573    }
1574
1575    char *path = NULL;
1576    size_t path_size = strlen(conf_dir)
1577       + strlen(file_name) + strlen(suffix) + 2;
1578
1579    /* Setting delimiter and editing path length */
1580 #if defined(_WIN32) || defined(__OS2__)
1581    char delim[] = "\\";
1582    path_size += 1;
1583 #else /* ifndef _WIN32 || __OS2__ */
1584    char delim[] = "/";
1585 #endif /* ifndef _WIN32 || __OS2__ */
1586
1587    /*
1588     * Building up path from many parts
1589     */
1590 #if defined(unix)
1591    if (*conf_dir != '/' && basedir && *basedir)
1592    {
1593       /*
1594        * Replacing conf_dir with basedir. This new variable contains
1595        * absolute path to cwd.
1596        */
1597       path_size += strlen(basedir) + 2;
1598       path = zalloc_or_die(path_size);
1599
1600       strlcpy(path, basedir,   path_size);
1601       strlcat(path, delim,     path_size);
1602       strlcat(path, conf_dir,  path_size);
1603       strlcat(path, delim,     path_size);
1604       strlcat(path, file_name, path_size);
1605       strlcat(path, suffix,    path_size);
1606    }
1607    else
1608 #endif /* defined unix */
1609    {
1610       path = zalloc_or_die(path_size);
1611
1612       strlcpy(path, conf_dir,  path_size);
1613       strlcat(path, delim,     path_size);
1614       strlcat(path, file_name, path_size);
1615       strlcat(path, suffix,    path_size);
1616    }
1617
1618    return path;
1619 }
1620
1621
1622 /*********************************************************************
1623  *
1624  * Function    :  get_certificate_mutex_id
1625  *
1626  * Description :  Computes mutex id from host name hash. This hash must
1627  *                be already saved in csp structure
1628  *
1629  * Parameters  :
1630  *          1  :  csp = Current client state (buffers, headers, etc...)
1631  *
1632  * Returns     :  Mutex id for given host name
1633  *
1634  *********************************************************************/
1635 static unsigned int get_certificate_mutex_id(struct client_state *csp) {
1636 #ifdef LIMIT_MUTEX_NUMBER
1637    return (unsigned int)(csp->http->hash_of_host[0] % 32);
1638 #else
1639    return (unsigned int)(csp->http->hash_of_host[1]
1640       + 256 * (int)csp->http->hash_of_host[0]);
1641 #endif /* LIMIT_MUTEX_NUMBER */
1642 }
1643
1644
1645 /*********************************************************************
1646  *
1647  * Function    :  get_certificate_serial
1648  *
1649  * Description :  Computes serial number for new certificate from host
1650  *                name hash. This hash must be already saved in csp
1651  *                structure.
1652  *
1653  * Parameters  :
1654  *          1  :  csp = Current client state (buffers, headers, etc...)
1655  *
1656  * Returns     :  Serial number for new certificate
1657  *
1658  *********************************************************************/
1659 static unsigned long  get_certificate_serial(struct client_state *csp) {
1660    unsigned long exp    = 1;
1661    unsigned long serial = 0;
1662
1663    int i = CERT_SERIAL_NUM_LENGTH;
1664    /* Length of hash is 16 bytes, we must avoid to read next chars */
1665    if (i > 16)
1666    {
1667       i = 16;
1668    }
1669    if (i < 2)
1670    {
1671       i = 2;
1672    }
1673
1674    for (; i >= 0; i--)
1675    {
1676       serial += exp * (unsigned)csp->http->hash_of_host[i];
1677       exp *= 256;
1678    }
1679    return serial;
1680 }
1681
1682
1683 /*********************************************************************
1684  *
1685  * Function    :  ssl_send_certificate_error
1686  *
1687  * Description :  Sends info about invalid server certificate to client.
1688  *                Sent message is including all trusted chain certificates,
1689  *                that can be downloaded in web browser.
1690  *
1691  * Parameters  :
1692  *          1  :  csp = Current client state (buffers, headers, etc...)
1693  *
1694  * Returns     :  N/A
1695  *
1696  *********************************************************************/
1697 extern void ssl_send_certificate_error(struct client_state *csp)
1698 {
1699    size_t message_len = 0;
1700    int ret = 0;
1701    struct certs_chain *cert = NULL;
1702
1703    /* Header of message with certificate informations */
1704    const char message_begin[] =
1705       "HTTP/1.1 200 OK\r\n"
1706       "Content-Type: text/html\r\n"
1707       "Connection: close\r\n\r\n"
1708       "<html><body><h1>Invalid server certificate</h1><p>Reason: ";
1709    const char message_end[] = "</body></html>\r\n\r\n";
1710    char reason[INVALID_CERT_INFO_BUF_SIZE];
1711    memset(reason, 0, sizeof(reason));
1712
1713    /* Get verification message from verification return code */
1714    mbedtls_x509_crt_verify_info(reason, sizeof(reason), " ",
1715       csp->server_cert_verification_result);
1716
1717    /*
1718     * Computing total length of message with all certificates inside
1719     */
1720    message_len = strlen(message_begin) + strlen(message_end)
1721                  + strlen(reason) + strlen("</p>") + 1;
1722
1723    cert = &(csp->server_certs_chain);
1724    while (cert->next != NULL)
1725    {
1726       size_t base64_len = 4 * ((strlen(cert->file_buf) + 2) / 3) + 1;
1727
1728       message_len += strlen(cert->text_buf) + strlen("<pre></pre>\n")
1729                      +  base64_len + strlen("<a href=\"data:application"
1730                         "/x-x509-ca-cert;base64,\">Download certificate</a>");
1731       cert = cert->next;
1732    }
1733
1734    /*
1735     * Joining all blocks in one long message
1736     */
1737    char message[message_len];
1738    memset(message, 0, message_len);
1739
1740    strlcpy(message, message_begin, message_len);
1741    strlcat(message, reason       , message_len);
1742    strlcat(message, "</p>"       , message_len);
1743
1744    cert = &(csp->server_certs_chain);
1745    while (cert->next != NULL)
1746    {
1747       size_t olen = 0;
1748       size_t base64_len = 4 * ((strlen(cert->file_buf) + 2) / 3) + 1; /* +1 for terminating null*/
1749       char base64_buf[base64_len];
1750       memset(base64_buf, 0, base64_len);
1751
1752       /* Encoding certificate into base64 code */
1753       ret = mbedtls_base64_encode((unsigned char*)base64_buf,
1754                base64_len, &olen, (const unsigned char*)cert->file_buf,
1755                strlen(cert->file_buf));
1756       if (ret != 0)
1757       {
1758          log_error(LOG_LEVEL_ERROR,
1759             "Encoding to base64 failed, buffer is to small");
1760       }
1761
1762       strlcat(message, "<pre>",        message_len);
1763       strlcat(message, cert->text_buf, message_len);
1764       strlcat(message, "</pre>\n",     message_len);
1765
1766       if (ret == 0)
1767       {
1768          strlcat(message, "<a href=\"data:application/x-x509-ca-cert;base64,",
1769             message_len);
1770          strlcat(message, base64_buf, message_len);
1771          strlcat(message, "\">Download certificate</a>", message_len);
1772       }
1773
1774       cert = cert->next;
1775    }
1776    strlcat(message, message_end, message_len);
1777
1778    /*
1779     * Sending final message to client
1780     */
1781    ssl_send_data(&(csp->mbedtls_client_attr.ssl),
1782       (const unsigned char *)message, strlen(message));
1783    /*
1784     * Waiting before closing connection. Some browsers don't show received
1785     * message if there isn't this delay.
1786     */
1787    sleep(1);
1788
1789    free_certificate_chain(csp);
1790 }
1791
1792
1793 /*********************************************************************
1794  *
1795  * Function    :  ssl_verify_callback
1796  *
1797  * Description :  This is a callback function for certificate verification.
1798  *                It's called for all certificates in server certificate
1799  *                trusted chain and it's preparing information about this
1800  *                certificates. Prepared informations can be used to inform
1801  *                user about invalid certificates.
1802  *
1803  * Parameters  :
1804  *          1  :  csp_void = Current client state (buffers, headers, etc...)
1805  *          2  :  crt   = certificate from trusted chain
1806  *          3  :  depth = depth in trusted chain
1807  *          4  :  flags = certificate flags
1808  *
1809  * Returns     :  0 on success and negative value on error
1810  *
1811  *********************************************************************/
1812 static int ssl_verify_callback(void *csp_void, mbedtls_x509_crt *crt,
1813    int depth, uint32_t *flags)
1814 {
1815    struct client_state *csp  = (struct client_state *)csp_void;
1816    struct certs_chain  *last = &(csp->server_certs_chain);
1817    size_t olen = 0;
1818    int ret = 0;
1819
1820    /*
1821     * Searching for last item in certificates linked list
1822     */
1823    while (last->next != NULL)
1824    {
1825       last = last->next;
1826    }
1827
1828    /*
1829     * Preparing next item in linked list for next certificate
1830     */
1831    last->next = malloc_or_die(sizeof(struct certs_chain));
1832    last->next->next = NULL;
1833    memset(last->next->text_buf, 0, sizeof(last->next->text_buf));
1834    memset(last->next->file_buf, 0, sizeof(last->next->file_buf));
1835
1836    /*
1837     * Saving certificate file into buffer
1838     */
1839    if ((ret = mbedtls_pem_write_buffer(PEM_BEGIN_CRT, PEM_END_CRT,
1840       crt->raw.p, crt->raw.len, (unsigned char *)last->file_buf,
1841       sizeof(last->file_buf)-1, &olen)) != 0)
1842    {
1843       return(ret);
1844    }
1845
1846    /*
1847     * Saving certificate information into buffer
1848     */
1849    mbedtls_x509_crt_info(last->text_buf, sizeof(last->text_buf) - 1,
1850       CERT_INFO_PREFIX, crt);
1851
1852    return 0;
1853 }
1854
1855
1856 /*********************************************************************
1857  *
1858  * Function    :  free_certificate_chain
1859  *
1860  * Description :  Frees certificates linked list. This linked list is
1861  *                used to save informations about certificates in
1862  *                trusted chain.
1863  *
1864  * Parameters  :
1865  *          1  :  csp = Current client state (buffers, headers, etc...)
1866  *
1867  * Returns     :  N/A
1868  *
1869  *********************************************************************/
1870 static void free_certificate_chain(struct client_state *csp)
1871 {
1872    struct certs_chain *cert = csp->server_certs_chain.next;
1873
1874    /* Cleaning buffers */
1875    memset(csp->server_certs_chain.text_buf, 0,
1876       sizeof(csp->server_certs_chain.text_buf));
1877    memset(csp->server_certs_chain.file_buf, 0,
1878       sizeof(csp->server_certs_chain.file_buf));
1879    csp->server_certs_chain.next = NULL;
1880
1881    /* Freeing memory in whole linked list */
1882    if (cert != NULL)
1883    {
1884       do
1885       {
1886          struct certs_chain *cert_for_free = cert;
1887          cert = cert->next;
1888          freez(cert_for_free);
1889       } while (cert != NULL);
1890    }
1891 }
1892
1893
1894 /*********************************************************************
1895  *
1896  * Function    :  file_exists
1897  *
1898  * Description :  Tests if file exists and is readable.
1899  *
1900  * Parameters  :
1901  *          1  :  path = Path to tested file.
1902  *
1903  * Returns     :  1 => File exists and is readable.
1904  *                0 => File doesn't exist or is not readable.
1905  *
1906  *********************************************************************/
1907 static int file_exists(const char *path)
1908 {
1909    FILE *f;
1910    if ((f = fopen(path, "r")) != NULL)
1911    {
1912       fclose(f);
1913       return 1;
1914    }
1915
1916    return 0;
1917 }
1918
1919
1920 /*********************************************************************
1921  *
1922  * Function    :  host_to_hash
1923  *
1924  * Description :  Creates MD5 hash from host name. Host name is loaded
1925  *                from structure csp and saved again into it.
1926  *
1927  * Parameters  :
1928  *          1  :  csp = Current client state (buffers, headers, etc...)
1929  *
1930  * Returns     :  1 => Error while creating hash
1931  *                0 => Hash created successfully
1932  *
1933  *********************************************************************/
1934 static int host_to_hash(struct client_state *csp)
1935 {
1936    int ret = 0;
1937
1938 #if !defined(MBEDTLS_MD5_C)
1939    log_error(LOG_LEVEL_ERROR, "MBEDTLS_MD5_C is not defined. Can't create"
1940       "MD5 hash for certificate and key name.");
1941    return -1;
1942 #else
1943    memset(csp->http->hash_of_host, 0, sizeof(csp->http->hash_of_host));
1944    mbedtls_md5((unsigned char *)csp->http->host, strlen(csp->http->host),
1945       csp->http->hash_of_host);
1946
1947    /* Converting hash into string with hex */
1948    size_t i = 0;
1949    for (; i < 16; i++)
1950    {
1951       if ((ret = sprintf((char *)csp->http->hash_of_host_hex + 2 * i, "%02x",
1952          csp->http->hash_of_host[i])) < 0)
1953       {
1954          log_error(LOG_LEVEL_ERROR, "Sprintf return value: %d", ret);
1955          return -1;
1956       }
1957    }
1958
1959    return 0;
1960 #endif /* MBEDTLS_MD5_C */
1961 }
1962
1963
1964 /*********************************************************************
1965  *
1966  * Function    :  tunnel_established_successfully
1967  *
1968  * Description :  Check if parent proxy server response contains
1969  *                informations about successfully created connection with
1970  *                destination server. (HTTP/... 2xx ...)
1971  *
1972  * Parameters  :
1973  *          1  :  server_response = Buffer with parent proxy server response
1974  *          2  :  response_len = Length of server_response
1975  *
1976  * Returns     :  1 => Connection created successfully
1977  *                0 => Connection wasn't created successfully
1978  *
1979  *********************************************************************/
1980 extern int tunnel_established_successfully(const char *server_response,
1981    unsigned int response_len)
1982 {
1983    unsigned int pos = 0;
1984
1985    if (server_response == NULL)
1986    {
1987       return 0;
1988    }
1989
1990    /* Tests if "HTTP/" string is at the begin of received response */
1991    if (strncmp(server_response, "HTTP/", 5) != 0)
1992    {
1993       return 0;
1994    }
1995
1996    for (pos = 0; pos < response_len; pos++)
1997    {
1998       if (server_response[pos] == ' ')
1999       {
2000          break;
2001       }
2002    }
2003
2004    /*
2005     * response_len -3 because of buffer end, response structure and 200 code.
2006     * There must be at least 3 chars after space.
2007     * End of buffer: ... 2xx'\0'
2008     *             pos = |
2009     */
2010    if (pos >= (response_len - 3))
2011    {
2012       return 0;
2013    }
2014
2015    /* Test HTTP status code */
2016    if (server_response[pos + 1] != '2')
2017    {
2018       return 0;
2019    }
2020
2021    return 1;
2022 }
2023
2024
2025 /*********************************************************************
2026  *
2027  * Function    :  seed_rng
2028  *
2029  * Description :  Seeding the RNG for all SSL uses
2030  *
2031  * Parameters  :
2032  *          1  :  csp = Current client state (buffers, headers, etc...)
2033  *
2034  * Returns     : -1 => RNG wasn't seed successfully
2035  *                0 => RNG is seeded successfully
2036  *
2037  *********************************************************************/
2038 static int seed_rng(struct client_state *csp)
2039 {
2040    int ret = 0;
2041    char err_buf[ERROR_BUF_SIZE];
2042
2043    if (rng_seeded == 0)
2044    {
2045       privoxy_mutex_lock(&rng_mutex);
2046       if (rng_seeded == 0)
2047       {
2048          mbedtls_ctr_drbg_init(&ctr_drbg);
2049          mbedtls_entropy_init(&entropy);
2050          ret = mbedtls_ctr_drbg_seed(&ctr_drbg, mbedtls_entropy_func,
2051             &entropy, NULL, 0);
2052          if (ret != 0)
2053          {
2054             mbedtls_strerror(ret, err_buf, sizeof(err_buf));
2055             log_error(LOG_LEVEL_ERROR,
2056                "mbedtls_ctr_drbg_seed failed: %s", err_buf);
2057             privoxy_mutex_unlock(&rng_mutex);
2058             return -1;
2059          }
2060          rng_seeded = 1;
2061       }
2062       privoxy_mutex_unlock(&rng_mutex);
2063    }
2064    return 0;
2065 }