lib/vtls/mbedtls.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 2010 - 2011, Hoi-Ho Chan, <hoiho.chan@gmail.com>
   9  * Copyright (C) 2012 - 2016, Daniel Stenberg, <daniel@haxx.se>, et al.
  10  *
  11  * This software is licensed as described in the file COPYING, which
  12  * you should have received as part of this distribution. The terms
  13  * are also available at https://curl.haxx.se/docs/copyright.html.
  14  *
  15  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  16  * copies of the Software, and permit persons to whom the Software is
  17  * furnished to do so, under the terms of the COPYING file.
  18  *
  19  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  20  * KIND, either express or implied.
  21  *
  22  ***************************************************************************/
  23
  24 /*
  25  * Source file for all mbedTSL-specific code for the TLS/SSL layer. No code
  26  * but vtls.c should ever call or use these functions.
  27  *
  28  */
  29
  30 #include "curl_setup.h"
  31
  32 #ifdef USE_MBEDTLS
  33
  34 #include <mbedtls/net.h>
  35 #include <mbedtls/ssl.h>
  36 #include <mbedtls/certs.h>
  37 #include <mbedtls/x509.h>
  38 #include <mbedtls/version.h>
  39
  40 #include <mbedtls/error.h>
  41 #include <mbedtls/entropy.h>
  42 #include <mbedtls/ctr_drbg.h>
  43 #include <mbedtls/sha256.h>
  44
  45 #include "urldata.h"
  46 #include "sendf.h"
  47 #include "inet_pton.h"
  48 #include "mbedtls.h"
  49 #include "vtls.h"
  50 #include "parsedate.h"
  51 #include "connect.h" /* for the connect timeout */
  52 #include "select.h"
  53 #include "rawstr.h"
  54 #include "polarssl_threadlock.h"
  55
  56 #define _MPRINTF_REPLACE /* use our functions only */
  57 #include <curl/mprintf.h>
  58 #include "curl_memory.h"
  59 /* The last #include file should be: */
  60 #include "memdebug.h"
  61
  62 /* apply threading? */
  63 #if defined(USE_THREADS_POSIX) || defined(USE_THREADS_WIN32)
  64 #define THREADING_SUPPORT
  65 #endif
  66
  67 #if defined(THREADING_SUPPORT)
  68 static mbedtls_entropy_context entropy;
  69
  70 static int  entropy_init_initialized  = 0;
  71
  72 /* start of entropy_init_mutex() */
  73 static void entropy_init_mutex(mbedtls_entropy_context *ctx)
  74 {
  75   /* lock 0 = entropy_init_mutex() */
  76   Curl_polarsslthreadlock_lock_function(0);
  77   if(entropy_init_initialized == 0) {
  78     mbedtls_entropy_init(ctx);
  79     entropy_init_initialized = 1;
  80   }
  81   Curl_polarsslthreadlock_unlock_function(0);
  82 }
  83 /* end of entropy_init_mutex() */
  84
  85 /* start of entropy_func_mutex() */
  86 static int entropy_func_mutex(void *data, unsigned char *output, size_t len)
  87 {
  88   int ret;
  89   /* lock 1 = entropy_func_mutex() */
  90   Curl_polarsslthreadlock_lock_function(1);
  91   ret = mbedtls_entropy_func(data, output, len);
  92   Curl_polarsslthreadlock_unlock_function(1);
  93
  94   return ret;
  95 }
  96 /* end of entropy_func_mutex() */
  97
  98 #endif /* THREADING_SUPPORT */
  99
 100 /* Define this to enable lots of debugging for mbedTLS */
 101 #undef MBEDTLS_DEBUG
 102
 103 #ifdef MBEDTLS_DEBUG
 104 static void mbedtls_debug(void *context, int level, const char *line)
 105 {
 106   struct SessionHandle *data = NULL;
 107
 108   if(!context)
 109     return;
 110
 111   data = (struct SessionHandle *)context;
 112
 113   infof(data, "%s", line);
 114   (void) level;
 115 }
 116 #else
 117 #endif
 118
 119 /* ALPN for http2? */
 120 #ifdef USE_NGHTTP2
 121 #  undef HAS_ALPN
 122 #  ifdef MBEDTLS_SSL_ALPN
 123 #    define HAS_ALPN
 124 #  endif
 125 #endif
 126
 127
 128 /*
 129  *  profile
 130  */
 131 const mbedtls_x509_crt_profile mbedtls_x509_crt_profile_fr =
 132 {
 133   /* Hashes from SHA-1 and above */
 134   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_SHA1 ) |
 135   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_RIPEMD160 ) |
 136   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_SHA224 ) |
 137   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_SHA256 ) |
 138   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_SHA384 ) |
 139   MBEDTLS_X509_ID_FLAG( MBEDTLS_MD_SHA512 ),
 140   0xFFFFFFF, /* Any PK alg    */
 141   0xFFFFFFF, /* Any curve     */
 142   1024,      /* RSA min key len */
 143 };
 144
 145 /* See https://tls.mbed.org/discussions/generic/
 146    howto-determine-exact-buffer-len-for-mbedtls_pk_write_pubkey_der
 147 */
 148 #define RSA_PUB_DER_MAX_BYTES   (38 + 2 * MBEDTLS_MPI_MAX_SIZE)
 149 #define ECP_PUB_DER_MAX_BYTES   (30 + 2 * MBEDTLS_ECP_MAX_BYTES)
 150
 151 #define PUB_DER_MAX_BYTES   (RSA_PUB_DER_MAX_BYTES > ECP_PUB_DER_MAX_BYTES ? \
 152                              RSA_PUB_DER_MAX_BYTES : ECP_PUB_DER_MAX_BYTES)
 153
 154 static Curl_recv mbedtls_recv;
 155 static Curl_send mbedtls_send;
 156
 157 static CURLcode
 158 mbedtls_connect_step1(struct connectdata *conn,
 159                       int sockindex)
 160 {
 161   struct SessionHandle *data = conn->data;
 162   struct ssl_connect_data* connssl = &conn->ssl[sockindex];
 163
 164   bool sni = TRUE; /* default is SNI enabled */
 165   int ret = -1;
 166 #ifdef ENABLE_IPV6
 167   struct in6_addr addr;
 168 #else
 169   struct in_addr addr;
 170 #endif
 171   void *old_session = NULL;
 172   size_t old_session_size = 0;
 173   char errorbuf[128];
 174   errorbuf[0]=0;
 175
 176   /* mbedTLS only supports SSLv3 and TLSv1 */
 177   if(data->set.ssl.version == CURL_SSLVERSION_SSLv2) {
 178     failf(data, "mbedTLS does not support SSLv2");
 179     return CURLE_SSL_CONNECT_ERROR;
 180   }
 181   else if(data->set.ssl.version == CURL_SSLVERSION_SSLv3)
 182     sni = FALSE; /* SSLv3 has no SNI */
 183
 184 #ifdef THREADING_SUPPORT
 185   entropy_init_mutex(&entropy);
 186   mbedtls_ctr_drbg_init(&connssl->ctr_drbg);
 187
 188   ret = mbedtls_ctr_drbg_seed(&connssl->ctr_drbg, entropy_func_mutex,
 189                               &entropy, connssl->ssn.id,
 190                               connssl->ssn.id_len);
 191   if(ret) {
 192 #ifdef MBEDTLS_ERROR_C
 193     mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 194 #endif /* MBEDTLS_ERROR_C */
 195     failf(data, "Failed - mbedTLS: ctr_drbg_init returned (-0x%04X) %s\n",
 196           -ret, errorbuf);
 197   }
 198 #else
 199   mbedtls_entropy_init(&connssl->entropy);
 200   mbedtls_ctr_drbg_init(&connssl->ctr_drbg);
 201
 202   ret = mbedtls_ctr_drbg_seed(&connssl->ctr_drbg, mbedtls_entropy_func,
 203                               &connssl->entropy, connssl->ssn.id,
 204                               connssl->ssn.id_len);
 205   if(ret) {
 206 #ifdef MBEDTLS_ERROR_C
 207     mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 208 #endif /* MBEDTLS_ERROR_C */
 209     failf(data, "Failed - mbedTLS: ctr_drbg_init returned (-0x%04X) %s\n",
 210           -ret, errorbuf);
 211   }
 212 #endif /* THREADING_SUPPORT */
 213
 214   /* Load the trusted CA */
 215   mbedtls_x509_crt_init(&connssl->cacert);
 216
 217   if(data->set.str[STRING_SSL_CAFILE]) {
 218     ret = mbedtls_x509_crt_parse_file(&connssl->cacert,
 219                                       data->set.str[STRING_SSL_CAFILE]);
 220
 221     if(ret<0) {
 222 #ifdef MBEDTLS_ERROR_C
 223       mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 224 #endif /* MBEDTLS_ERROR_C */
 225       failf(data, "Error reading ca cert file %s - mbedTLS: (-0x%04X) %s",
 226             data->set.str[STRING_SSL_CAFILE], -ret, errorbuf);
 227
 228       if(data->set.ssl.verifypeer)
 229         return CURLE_SSL_CACERT_BADFILE;
 230     }
 231   }
 232
 233   if(data->set.str[STRING_SSL_CAPATH]) {
 234     ret = mbedtls_x509_crt_parse_path(&connssl->cacert,
 235                                       data->set.str[STRING_SSL_CAPATH]);
 236
 237     if(ret<0) {
 238 #ifdef MBEDTLS_ERROR_C
 239       mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 240 #endif /* MBEDTLS_ERROR_C */
 241       failf(data, "Error reading ca cert path %s - mbedTLS: (-0x%04X) %s",
 242             data->set.str[STRING_SSL_CAPATH], -ret, errorbuf);
 243
 244       if(data->set.ssl.verifypeer)
 245         return CURLE_SSL_CACERT_BADFILE;
 246     }
 247   }
 248
 249   /* Load the client certificate */
 250   mbedtls_x509_crt_init(&connssl->clicert);
 251
 252   if(data->set.str[STRING_CERT]) {
 253     ret = mbedtls_x509_crt_parse_file(&connssl->clicert,
 254                                       data->set.str[STRING_CERT]);
 255
 256     if(ret) {
 257 #ifdef MBEDTLS_ERROR_C
 258       mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 259 #endif /* MBEDTLS_ERROR_C */
 260       failf(data, "Error reading client cert file %s - mbedTLS: (-0x%04X) %s",
 261             data->set.str[STRING_CERT], -ret, errorbuf);
 262
 263       return CURLE_SSL_CERTPROBLEM;
 264     }
 265   }
 266
 267   /* Load the client private key */
 268   mbedtls_pk_init(&connssl->pk);
 269
 270   if(data->set.str[STRING_KEY]) {
 271     ret = mbedtls_pk_parse_keyfile(&connssl->pk, data->set.str[STRING_KEY],
 272                                    data->set.str[STRING_KEY_PASSWD]);
 273     if(ret == 0 && !mbedtls_pk_can_do(&connssl->pk, MBEDTLS_PK_RSA))
 274       ret = MBEDTLS_ERR_PK_TYPE_MISMATCH;
 275
 276     if(ret) {
 277 #ifdef MBEDTLS_ERROR_C
 278       mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 279 #endif /* MBEDTLS_ERROR_C */
 280       failf(data, "Error reading private key %s - mbedTLS: (-0x%04X) %s",
 281             data->set.str[STRING_KEY], -ret, errorbuf);
 282
 283       return CURLE_SSL_CERTPROBLEM;
 284     }
 285   }
 286
 287   /* Load the CRL */
 288   mbedtls_x509_crl_init(&connssl->crl);
 289
 290   if(data->set.str[STRING_SSL_CRLFILE]) {
 291     ret = mbedtls_x509_crl_parse_file(&connssl->crl,
 292                                       data->set.str[STRING_SSL_CRLFILE]);
 293
 294     if(ret) {
 295 #ifdef MBEDTLS_ERROR_C
 296       mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 297 #endif /* MBEDTLS_ERROR_C */
 298       failf(data, "Error reading CRL file %s - mbedTLS: (-0x%04X) %s",
 299             data->set.str[STRING_SSL_CRLFILE], -ret, errorbuf);
 300
 301       return CURLE_SSL_CRL_BADFILE;
 302     }
 303   }
 304
 305   infof(data, "mbedTLS: Connecting to %s:%d\n",
 306         conn->host.name, conn->remote_port);
 307
 308   mbedtls_ssl_config_init(&connssl->config);
 309
 310   mbedtls_ssl_init(&connssl->ssl);
 311   if(mbedtls_ssl_setup(&connssl->ssl, &connssl->config)) {
 312     failf(data, "mbedTLS: ssl_init failed");
 313     return CURLE_SSL_CONNECT_ERROR;
 314   }
 315   ret = mbedtls_ssl_config_defaults(&connssl->config,
 316                                     MBEDTLS_SSL_IS_CLIENT,
 317                                     MBEDTLS_SSL_TRANSPORT_STREAM,
 318                                     MBEDTLS_SSL_PRESET_DEFAULT);
 319   if(ret) {
 320     failf(data, "mbedTLS: ssl_config failed");
 321     return CURLE_SSL_CONNECT_ERROR;
 322   }
 323
 324   /* new profile with RSA min key len = 1024 ... */
 325   mbedtls_ssl_conf_cert_profile( &connssl->config,
 326                                  &mbedtls_x509_crt_profile_fr);
 327
 328   switch(data->set.ssl.version) {
 329   case CURL_SSLVERSION_DEFAULT:
 330   case CURL_SSLVERSION_TLSv1:
 331     mbedtls_ssl_conf_min_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 332                                  MBEDTLS_SSL_MINOR_VERSION_1);
 333     infof(data, "mbedTLS: Set min SSL version to TLS 1.0\n");
 334     break;
 335   case CURL_SSLVERSION_SSLv3:
 336     mbedtls_ssl_conf_min_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 337                                  MBEDTLS_SSL_MINOR_VERSION_0);
 338     mbedtls_ssl_conf_max_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 339                                  MBEDTLS_SSL_MINOR_VERSION_0);
 340     infof(data, "mbedTLS: Set SSL version to SSLv3\n");
 341     break;
 342   case CURL_SSLVERSION_TLSv1_0:
 343     mbedtls_ssl_conf_min_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 344                                  MBEDTLS_SSL_MINOR_VERSION_1);
 345     mbedtls_ssl_conf_max_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 346                                  MBEDTLS_SSL_MINOR_VERSION_1);
 347     infof(data, "mbedTLS: Set SSL version to TLS 1.0\n");
 348     break;
 349   case CURL_SSLVERSION_TLSv1_1:
 350     mbedtls_ssl_conf_min_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 351                                  MBEDTLS_SSL_MINOR_VERSION_2);
 352     mbedtls_ssl_conf_max_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 353                                  MBEDTLS_SSL_MINOR_VERSION_2);
 354     infof(data, "mbedTLS: Set SSL version to TLS 1.1\n");
 355     break;
 356   case CURL_SSLVERSION_TLSv1_2:
 357     mbedtls_ssl_conf_min_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 358                                  MBEDTLS_SSL_MINOR_VERSION_3);
 359     mbedtls_ssl_conf_max_version(&connssl->config, MBEDTLS_SSL_MAJOR_VERSION_3,
 360                                  MBEDTLS_SSL_MINOR_VERSION_3);
 361     infof(data, "mbedTLS: Set SSL version to TLS 1.2\n");
 362     break;
 363   default:
 364     failf(data, "mbedTLS: Unsupported SSL protocol version");
 365     return CURLE_SSL_CONNECT_ERROR;
 366   }
 367
 368   mbedtls_ssl_conf_authmode(&connssl->config, MBEDTLS_SSL_VERIFY_OPTIONAL);
 369
 370   mbedtls_ssl_conf_rng(&connssl->config, mbedtls_ctr_drbg_random,
 371                        &connssl->ctr_drbg);
 372   mbedtls_ssl_set_bio(&connssl->ssl, &conn->sock[sockindex],
 373                       mbedtls_net_send,
 374                       mbedtls_net_recv,
 375                       NULL /*  rev_timeout() */);
 376
 377   mbedtls_ssl_conf_ciphersuites(&connssl->config,
 378                                 mbedtls_ssl_list_ciphersuites());
 379   if(!Curl_ssl_getsessionid(conn, &old_session, &old_session_size)) {
 380     memcpy(&connssl->ssn, old_session, old_session_size);
 381     infof(data, "mbedTLS re-using session\n");
 382   }
 383
 384   mbedtls_ssl_set_session(&connssl->ssl,
 385                           &connssl->ssn);
 386
 387   mbedtls_ssl_conf_ca_chain(&connssl->config,
 388                             &connssl->cacert,
 389                             &connssl->crl);
 390
 391   if(data->set.str[STRING_KEY]) {
 392     mbedtls_ssl_conf_own_cert(&connssl->config,
 393                               &connssl->clicert, &connssl->pk);
 394   }
 395   if(!Curl_inet_pton(AF_INET, conn->host.name, &addr) &&
 396 #ifdef ENABLE_IPV6
 397      !Curl_inet_pton(AF_INET6, conn->host.name, &addr) &&
 398 #endif
 399      sni && mbedtls_ssl_set_hostname(&connssl->ssl, conn->host.name)) {
 400     infof(data, "WARNING: failed to configure "
 401           "server name indication (SNI) TLS extension\n");
 402   }
 403
 404 #ifdef HAS_ALPN
 405   if(data->set.ssl_enable_alpn) {
 406     const char **p = &connssl->protocols[0];
 407 #ifdef USE_NGHTTP2
 408     if(data->set.httpversion >= CURL_HTTP_VERSION_2)
 409       *p++ = NGHTTP2_PROTO_VERSION_ID;
 410 #endif
 411     *p++ = ALPN_HTTP_1_1;
 412     *p = NULL;
 413     /* this function doesn't clone the protocols array, which is why we need
 414        to keep it around */
 415     if(mbedtls_ssl_conf_alpn_protocols(&connssl->config,
 416                                        &connssl->protocols[0])) {
 417       failf(data, "Failed setting ALPN protocols");
 418       return CURLE_SSL_CONNECT_ERROR;
 419     }
 420     for(p = &connssl->protocols[0]; *p; ++p)
 421       infof(data, "ALPN, offering %s\n", *p);
 422   }
 423 #endif
 424
 425 #ifdef MBEDTLS_DEBUG
 426   mbedtls_ssl_conf_dbg(&connssl->ssl, mbedtls_debug, data);
 427 #endif
 428
 429   connssl->connecting_state = ssl_connect_2;
 430
 431   return CURLE_OK;
 432 }
 433
 434 static CURLcode
 435 mbedtls_connect_step2(struct connectdata *conn,
 436                       int sockindex)
 437 {
 438   int ret;
 439   struct SessionHandle *data = conn->data;
 440   struct ssl_connect_data* connssl = &conn->ssl[sockindex];
 441   const mbedtls_x509_crt *peercert;
 442
 443 #ifdef HAS_ALPN
 444   const char* next_protocol;
 445 #endif
 446
 447   char errorbuf[128];
 448   errorbuf[0] = 0;
 449
 450   conn->recv[sockindex] = mbedtls_recv;
 451   conn->send[sockindex] = mbedtls_send;
 452
 453   ret = mbedtls_ssl_handshake(&connssl->ssl);
 454
 455   if(ret == MBEDTLS_ERR_SSL_WANT_READ) {
 456     connssl->connecting_state = ssl_connect_2_reading;
 457     return CURLE_OK;
 458   }
 459   else if(ret == MBEDTLS_ERR_SSL_WANT_WRITE) {
 460     connssl->connecting_state = ssl_connect_2_writing;
 461     return CURLE_OK;
 462   }
 463   else if(ret) {
 464 #ifdef MBEDTLS_ERROR_C
 465     mbedtls_strerror(ret, errorbuf, sizeof(errorbuf));
 466 #endif /* MBEDTLS_ERROR_C */
 467     failf(data, "ssl_handshake returned - mbedTLS: (-0x%04X) %s",
 468           -ret, errorbuf);
 469     return CURLE_SSL_CONNECT_ERROR;
 470   }
 471
 472   infof(data, "mbedTLS: Handshake complete, cipher is %s\n",
 473         mbedtls_ssl_get_ciphersuite(&conn->ssl[sockindex].ssl)
 474     );
 475
 476   ret = mbedtls_ssl_get_verify_result(&conn->ssl[sockindex].ssl);
 477
 478   if(ret && data->set.ssl.verifypeer) {
 479     if(ret & MBEDTLS_X509_BADCERT_EXPIRED)
 480       failf(data, "Cert verify failed: BADCERT_EXPIRED");
 481
 482     if(ret & MBEDTLS_X509_BADCERT_REVOKED) {
 483       failf(data, "Cert verify failed: BADCERT_REVOKED");
 484       return CURLE_SSL_CACERT;
 485     }
 486
 487     if(ret & MBEDTLS_X509_BADCERT_CN_MISMATCH)
 488       failf(data, "Cert verify failed: BADCERT_CN_MISMATCH");
 489
 490     if(ret & MBEDTLS_X509_BADCERT_NOT_TRUSTED)
 491       failf(data, "Cert verify failed: BADCERT_NOT_TRUSTED");
 492
 493     return CURLE_PEER_FAILED_VERIFICATION;
 494   }
 495
 496   peercert = mbedtls_ssl_get_peer_cert(&connssl->ssl);
 497
 498   if(peercert && data->set.verbose) {
 499     const size_t bufsize = 16384;
 500     char *buffer = malloc(bufsize);
 501
 502     if(!buffer)
 503       return CURLE_OUT_OF_MEMORY;
 504
 505     if(mbedtls_x509_crt_info(buffer, bufsize, "* ", peercert) > 0)
 506       infof(data, "Dumping cert info:\n%s\n", buffer);
 507     else
 508       infof(data, "Unable to dump certificate information.\n");
 509
 510     free(buffer);
 511   }
 512
 513   if(data->set.str[STRING_SSL_PINNEDPUBLICKEY]) {
 514     int size;
 515     CURLcode result;
 516     mbedtls_x509_crt *p;
 517     unsigned char pubkey[PUB_DER_MAX_BYTES];
 518
 519     if(!peercert || !peercert->raw.p || !peercert->raw.len) {
 520       failf(data, "Failed due to missing peer certificate");
 521       return CURLE_SSL_PINNEDPUBKEYNOTMATCH;
 522     }
 523
 524     p = calloc(1, sizeof(*p));
 525
 526     if(!p)
 527       return CURLE_OUT_OF_MEMORY;
 528
 529     mbedtls_x509_crt_init(p);
 530
 531     /* Make a copy of our const peercert because mbedtls_pk_write_pubkey_der
 532        needs a non-const key, for now.
 533        https://github.com/ARMmbed/mbedtls/issues/396 */
 534     if(mbedtls_x509_crt_parse_der(p, peercert->raw.p, peercert->raw.len)) {
 535       failf(data, "Failed copying peer certificate");
 536       mbedtls_x509_crt_free(p);
 537       free(p);
 538       return CURLE_SSL_PINNEDPUBKEYNOTMATCH;
 539     }
 540
 541     size = mbedtls_pk_write_pubkey_der(&p->pk, pubkey, PUB_DER_MAX_BYTES);
 542
 543     if(size <= 0) {
 544       failf(data, "Failed copying public key from peer certificate");
 545       mbedtls_x509_crt_free(p);
 546       free(p);
 547       return CURLE_SSL_PINNEDPUBKEYNOTMATCH;
 548     }
 549
 550     /* mbedtls_pk_write_pubkey_der writes data at the end of the buffer. */
 551     result = Curl_pin_peer_pubkey(data,
 552                                   data->set.str[STRING_SSL_PINNEDPUBLICKEY],
 553                                   &pubkey[PUB_DER_MAX_BYTES - size], size);
 554     if(result) {
 555       mbedtls_x509_crt_free(p);
 556       free(p);
 557       return result;
 558     }
 559
 560     mbedtls_x509_crt_free(p);
 561     free(p);
 562   }
 563
 564 #ifdef HAS_ALPN
 565   if(data->set.ssl_enable_alpn) {
 566     next_protocol = mbedtls_ssl_get_alpn_protocol(&connssl->ssl);
 567
 568     if(next_protocol) {
 569       infof(data, "ALPN, server accepted to use %s\n", next_protocol);
 570 #ifdef USE_NGHTTP2
 571       if(!strncmp(next_protocol, NGHTTP2_PROTO_VERSION_ID,
 572                   NGHTTP2_PROTO_VERSION_ID_LEN) &&
 573          !next_protocol[NGHTTP2_PROTO_VERSION_ID_LEN]) {
 574         conn->negnpn = CURL_HTTP_VERSION_2;
 575       }
 576       else
 577 #endif
 578         if(!strncmp(next_protocol, ALPN_HTTP_1_1, ALPN_HTTP_1_1_LENGTH) &&
 579            !next_protocol[ALPN_HTTP_1_1_LENGTH]) {
 580           conn->negnpn = CURL_HTTP_VERSION_1_1;
 581         }
 582     }
 583     else {
 584       infof(data, "ALPN, server did not agree to a protocol\n");
 585     }
 586   }
 587 #endif
 588
 589   connssl->connecting_state = ssl_connect_3;
 590   infof(data, "SSL connected\n");
 591
 592   return CURLE_OK;
 593 }
 594
 595 static CURLcode
 596 mbedtls_connect_step3(struct connectdata *conn,
 597                       int sockindex)
 598 {
 599   CURLcode retcode = CURLE_OK;
 600   struct ssl_connect_data *connssl = &conn->ssl[sockindex];
 601   struct SessionHandle *data = conn->data;
 602   void *old_ssl_sessionid = NULL;
 603   mbedtls_ssl_session *our_ssl_sessionid = &conn->ssl[sockindex].ssn;
 604   int incache;
 605
 606   DEBUGASSERT(ssl_connect_3 == connssl->connecting_state);
 607
 608   /* Save the current session data for possible re-use */
 609   incache = !(Curl_ssl_getsessionid(conn, &old_ssl_sessionid, NULL));
 610   if(incache) {
 611     if(old_ssl_sessionid != our_ssl_sessionid) {
 612       infof(data, "old SSL session ID is stale, removing\n");
 613       Curl_ssl_delsessionid(conn, old_ssl_sessionid);
 614       incache = FALSE;
 615     }
 616   }
 617   if(!incache) {
 618     void *new_session = malloc(sizeof(mbedtls_ssl_session));
 619
 620     if(new_session) {
 621       memcpy(new_session, our_ssl_sessionid,
 622              sizeof(mbedtls_ssl_session));
 623
 624       retcode = Curl_ssl_addsessionid(conn, new_session,
 625                                       sizeof(mbedtls_ssl_session));
 626     }
 627     else {
 628       retcode = CURLE_OUT_OF_MEMORY;
 629     }
 630
 631     if(retcode) {
 632       failf(data, "failed to store ssl session");
 633       return retcode;
 634     }
 635   }
 636
 637   connssl->connecting_state = ssl_connect_done;
 638
 639   return CURLE_OK;
 640 }
 641
 642 static ssize_t mbedtls_send(struct connectdata *conn,
 643                             int sockindex,
 644                             const void *mem,
 645                             size_t len,
 646                             CURLcode *curlcode)
 647 {
 648   int ret = -1;
 649
 650   ret = mbedtls_ssl_write(&conn->ssl[sockindex].ssl,
 651                           (unsigned char *)mem, len);
 652
 653   if(ret < 0) {
 654     *curlcode = (ret == MBEDTLS_ERR_SSL_WANT_WRITE) ?
 655       CURLE_AGAIN : CURLE_SEND_ERROR;
 656     ret = -1;
 657   }
 658
 659   return ret;
 660 }
 661
 662 void Curl_mbedtls_close_all(struct SessionHandle *data)
 663 {
 664   (void)data;
 665 }
 666
 667 void Curl_mbedtls_close(struct connectdata *conn, int sockindex)
 668 {
 669   mbedtls_pk_free(&conn->ssl[sockindex].pk);
 670   mbedtls_x509_crt_free(&conn->ssl[sockindex].clicert);
 671   mbedtls_x509_crt_free(&conn->ssl[sockindex].cacert);
 672   mbedtls_x509_crl_free(&conn->ssl[sockindex].crl);
 673   mbedtls_ssl_config_free(&conn->ssl[sockindex].config);
 674   mbedtls_ssl_free(&conn->ssl[sockindex].ssl);
 675   mbedtls_ctr_drbg_free(&conn->ssl[sockindex].ctr_drbg);
 676 #ifndef THREADING_SUPPORT
 677   mbedtls_entropy_free(&conn->ssl[sockindex].entropy);
 678 #endif /* THREADING_SUPPORT */
 679 }
 680
 681 static ssize_t mbedtls_recv(struct connectdata *conn,
 682                             int num,
 683                             char *buf,
 684                             size_t buffersize,
 685                             CURLcode *curlcode)
 686 {
 687   int ret = -1;
 688   ssize_t len = -1;
 689
 690   memset(buf, 0, buffersize);
 691   ret = mbedtls_ssl_read(&conn->ssl[num].ssl, (unsigned char *)buf,
 692                          buffersize);
 693
 694   if(ret <= 0) {
 695     if(ret == MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY)
 696       return 0;
 697
 698     *curlcode = (ret == MBEDTLS_ERR_SSL_WANT_READ) ?
 699       CURLE_AGAIN : CURLE_RECV_ERROR;
 700     return -1;
 701   }
 702
 703   len = ret;
 704
 705   return len;
 706 }
 707
 708 void Curl_mbedtls_session_free(void *ptr)
 709 {
 710   free(ptr);
 711 }
 712
 713 size_t Curl_mbedtls_version(char *buffer, size_t size)
 714 {
 715   unsigned int version = mbedtls_version_get_number();
 716   return snprintf(buffer, size, "mbedTLS/%d.%d.%d", version>>24,
 717                   (version>>16)&0xff, (version>>8)&0xff);
 718 }
 719
 720 static CURLcode
 721 mbedtls_connect_common(struct connectdata *conn,
 722                        int sockindex,
 723                        bool nonblocking,
 724                        bool *done)
 725 {
 726   CURLcode retcode;
 727   struct SessionHandle *data = conn->data;
 728   struct ssl_connect_data *connssl = &conn->ssl[sockindex];
 729   curl_socket_t sockfd = conn->sock[sockindex];
 730   long timeout_ms;
 731   int what;
 732
 733   /* check if the connection has already been established */
 734   if(ssl_connection_complete == connssl->state) {
 735     *done = TRUE;
 736     return CURLE_OK;
 737   }
 738
 739   if(ssl_connect_1==connssl->connecting_state) {
 740     /* Find out how much more time we're allowed */
 741     timeout_ms = Curl_timeleft(data, NULL, TRUE);
 742
 743     if(timeout_ms < 0) {
 744       /* no need to continue if time already is up */
 745       failf(data, "SSL connection timeout");
 746       return CURLE_OPERATION_TIMEDOUT;
 747     }
 748     retcode = mbedtls_connect_step1(conn, sockindex);
 749     if(retcode)
 750       return retcode;
 751   }
 752
 753   while(ssl_connect_2 == connssl->connecting_state ||
 754         ssl_connect_2_reading == connssl->connecting_state ||
 755         ssl_connect_2_writing == connssl->connecting_state) {
 756
 757     /* check allowed time left */
 758     timeout_ms = Curl_timeleft(data, NULL, TRUE);
 759
 760     if(timeout_ms < 0) {
 761       /* no need to continue if time already is up */
 762       failf(data, "SSL connection timeout");
 763       return CURLE_OPERATION_TIMEDOUT;
 764     }
 765
 766     /* if ssl is expecting something, check if it's available. */
 767     if(connssl->connecting_state == ssl_connect_2_reading
 768        || connssl->connecting_state == ssl_connect_2_writing) {
 769
 770       curl_socket_t writefd = ssl_connect_2_writing==
 771         connssl->connecting_state?sockfd:CURL_SOCKET_BAD;
 772       curl_socket_t readfd = ssl_connect_2_reading==
 773         connssl->connecting_state?sockfd:CURL_SOCKET_BAD;
 774
 775       what = Curl_socket_ready(readfd, writefd, nonblocking ? 0 : timeout_ms);
 776       if(what < 0) {
 777         /* fatal error */
 778         failf(data, "select/poll on SSL socket, errno: %d", SOCKERRNO);
 779         return CURLE_SSL_CONNECT_ERROR;
 780       }
 781       else if(0 == what) {
 782         if(nonblocking) {
 783           *done = FALSE;
 784           return CURLE_OK;
 785         }
 786         else {
 787           /* timeout */
 788           failf(data, "SSL connection timeout");
 789           return CURLE_OPERATION_TIMEDOUT;
 790         }
 791       }
 792       /* socket is readable or writable */
 793     }
 794
 795     /* Run transaction, and return to the caller if it failed or if
 796      * this connection is part of a multi handle and this loop would
 797      * execute again. This permits the owner of a multi handle to
 798      * abort a connection attempt before step2 has completed while
 799      * ensuring that a client using select() or epoll() will always
 800      * have a valid fdset to wait on.
 801      */
 802     retcode = mbedtls_connect_step2(conn, sockindex);
 803     if(retcode || (nonblocking &&
 804                    (ssl_connect_2 == connssl->connecting_state ||
 805                     ssl_connect_2_reading == connssl->connecting_state ||
 806                     ssl_connect_2_writing == connssl->connecting_state)))
 807       return retcode;
 808
 809   } /* repeat step2 until all transactions are done. */
 810
 811   if(ssl_connect_3==connssl->connecting_state) {
 812     retcode = mbedtls_connect_step3(conn, sockindex);
 813     if(retcode)
 814       return retcode;
 815   }
 816
 817   if(ssl_connect_done==connssl->connecting_state) {
 818     connssl->state = ssl_connection_complete;
 819     conn->recv[sockindex] = mbedtls_recv;
 820     conn->send[sockindex] = mbedtls_send;
 821     *done = TRUE;
 822   }
 823   else
 824     *done = FALSE;
 825
 826   /* Reset our connect state machine */
 827   connssl->connecting_state = ssl_connect_1;
 828
 829   return CURLE_OK;
 830 }
 831
 832 CURLcode
 833 Curl_mbedtls_connect_nonblocking(struct connectdata *conn,
 834                                  int sockindex,
 835                                  bool *done)
 836 {
 837   return mbedtls_connect_common(conn, sockindex, TRUE, done);
 838 }
 839
 840
 841 CURLcode
 842 Curl_mbedtls_connect(struct connectdata *conn,
 843                      int sockindex)
 844 {
 845   CURLcode retcode;
 846   bool done = FALSE;
 847
 848   retcode = mbedtls_connect_common(conn, sockindex, FALSE, &done);
 849   if(retcode)
 850     return retcode;
 851
 852   DEBUGASSERT(done);
 853
 854   return CURLE_OK;
 855 }
 856
 857 /*
 858  * return 0 error initializing SSL
 859  * return 1 SSL initialized successfully
 860  */
 861 int mbedtls_init(void)
 862 {
 863   return Curl_polarsslthreadlock_thread_setup();
 864 }
 865
 866 void mbedtls_cleanup(void)
 867 {
 868   (void)Curl_polarsslthreadlock_thread_cleanup();
 869 }
 870
 871 #endif /* USE_MBEDTLS */