/*

 * libwebsockets - small server side websockets and web server implementation

 *

 * Copyright (C) 2010 - 2019 Andy Green <andy@warmcat.com>

 *

 * Permission is hereby granted, free of charge, to any person obtaining a copy

 * of this software and associated documentation files (the "Software"), to

 * deal in the Software without restriction, including without limitation the

 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or

 * sell copies of the Software, and to permit persons to whom the Software is

 * furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice shall be included in

 * all copies or substantial portions of the Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS

 * IN THE SOFTWARE.

 */

#include "lws_config.h"

#ifdef LWS_HAVE_X509_VERIFY_PARAM_set1_host

/* Before glibc 2.10, strnlen required _GNU_SOURCE */

#if !defined(_GNU_SOURCE)

#define _GNU_SOURCE

#endif

#endif

#include <string.h>

#include "private-lib-core.h"

#include "private-lib-tls-openssl.h"

/*

 * Care: many openssl apis return 1 for success.  These are translated to the

 * lws convention of 0 for success.

 */

int lws_openssl_describe_cipher(struct lws *wsi);

extern int openssl_websocket_private_data_index,

    openssl_SSL_CTX_private_data_index;

#if !defined(USE_WOLFSSL)

#if 0

#if defined(LWS_WITH_TLS_JIT_TRUST)

/*

 * Completion of sync or async JIT trust lookup

 */

int

lws_tls_jit_trust_got_cert_cb(void *got_opaque, const uint8_t *der,

            size_t der_len)

{

  X509 *x = d2i_X509(NULL, &der, (long)der_len);

  /** !!! this is not safe for async atm */

  struct lws *wsi = (struct lws *)got_opaque;

  X509_STORE *xs;

  int ret = 0;

  if (!x) {

    lwsl_err("%s: failed\n", __func__);

    return 1;

  }

  xs = SSL_CTX_get_cert_store(SSL_get_SSL_CTX(wsi->tls.ssl));

  if (xs) {

    if (X509_STORE_add_cert(xs, x) != 1) {

      lwsl_warn("%s: unable to set trusted CA\n", __func__);

      ret = 1;

    } else

      lwsl_notice("%s: added trusted CA to CTX for next time\n",

          __func__);

  } else

    lwsl_warn("%s: couldn't get cert store\n", __func__);

  X509_free(x);

  return ret;

}

#endif

#endif

static int

OpenSSL_client_verify_callback(int preverify_ok, X509_STORE_CTX *x509_ctx)

{

  SSL *ssl;

  int n, err = 0;

  struct lws *wsi;

  /* keep old behaviour accepting self-signed server certs */

  if (!preverify_ok) {

    err = X509_STORE_CTX_get_error(x509_ctx);

    if (err != X509_V_OK) {

      ssl = X509_STORE_CTX_get_ex_data(x509_ctx,

          SSL_get_ex_data_X509_STORE_CTX_idx());

      wsi = SSL_get_ex_data(ssl,

          openssl_websocket_private_data_index);

      if (!wsi) {

        lwsl_err("%s: can't get wsi from ssl privdata\n",

           __func__);

        return 0;

      }

      if ((err == X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT ||

           err == X509_V_ERR_SELF_SIGNED_CERT_IN_CHAIN) &&

           wsi->tls.use_ssl & LCCSCF_ALLOW_SELFSIGNED) {

        lwsl_notice("accepting self-signed "

              "certificate (verify_callback)\n");

        X509_STORE_CTX_set_error(x509_ctx, X509_V_OK);

        return 1; // ok

    } else if ((err == X509_V_ERR_UNABLE_TO_GET_ISSUER_CERT_LOCALLY ||

          err == X509_V_ERR_UNABLE_TO_VERIFY_LEAF_SIGNATURE) &&

          wsi->tls.use_ssl & LCCSCF_ALLOW_INSECURE) {

        lwsl_notice("accepting non-trusted certificate\n");

        X509_STORE_CTX_set_error(x509_ctx, X509_V_OK);

        return 1;  /* ok */

      } else if ((err == X509_V_ERR_CERT_NOT_YET_VALID ||

            err == X509_V_ERR_CERT_HAS_EXPIRED) &&

            wsi->tls.use_ssl & LCCSCF_ALLOW_EXPIRED) {

        if (err == X509_V_ERR_CERT_NOT_YET_VALID)

          lwsl_notice("accepting not yet valid "

                "certificate (verify_"

                "callback)\n");

        else if (err == X509_V_ERR_CERT_HAS_EXPIRED)

          lwsl_notice("accepting expired "

                "certificate (verify_"

                "callback)\n");

        X509_STORE_CTX_set_error(x509_ctx, X509_V_OK);

        return 1; // ok

      }

    }

  }

  ssl = X509_STORE_CTX_get_ex_data(x509_ctx,

           SSL_get_ex_data_X509_STORE_CTX_idx());

  wsi = SSL_get_ex_data(ssl, openssl_websocket_private_data_index);

  if (!wsi) {

    lwsl_err("%s: can't get wsi from ssl privdata\n",  __func__);

    return 0;

  }

#if defined(LWS_WITH_TLS_JIT_TRUST)

  if (err == X509_V_ERR_UNABLE_TO_GET_ISSUER_CERT_LOCALLY) {

    union lws_tls_cert_info_results ci;

    STACK_OF(X509) *x509_stack;

    x509_stack = X509_STORE_CTX_get1_chain(x509_ctx);

    if (x509_stack) {

      for (n = 0; n < OPENSSL_sk_num((const OPENSSL_STACK *)x509_stack) &&

            wsi->tls.kid_chain.count !=

             LWS_ARRAY_SIZE(wsi->tls.kid_chain.akid); n++) {

        X509 *x509 = OPENSSL_sk_value((const OPENSSL_STACK *)x509_stack, n);

        if (!lws_tls_openssl_cert_info(x509,

              LWS_TLS_CERT_INFO_SUBJECT_KEY_ID,

              &ci, 0))

          lws_tls_kid_copy(&ci,

            &wsi->tls.kid_chain.skid[

                 wsi->tls.kid_chain.count]);

        if (!lws_tls_openssl_cert_info(x509,

               LWS_TLS_CERT_INFO_AUTHORITY_KEY_ID,

               &ci, 0))

          lws_tls_kid_copy(&ci,

             &wsi->tls.kid_chain.akid[

                 wsi->tls.kid_chain.count]);

        wsi->tls.kid_chain.count++;

      }

      sk_X509_pop_free(x509_stack, X509_free);

    }

    lws_tls_jit_trust_sort_kids(wsi, &wsi->tls.kid_chain);

  }

#endif

  n = lws_get_context_protocol(wsi->a.context, 0).callback(wsi,

      LWS_CALLBACK_OPENSSL_PERFORM_SERVER_CERT_VERIFICATION,

      x509_ctx, ssl, (unsigned int)preverify_ok);

  /* keep old behaviour if something wrong with server certs */

  /* if ssl error is overruled in callback and cert is ok,

   * X509_STORE_CTX_set_error(x509_ctx, X509_V_OK); must be set and

   * return value is 0 from callback */

  if (!preverify_ok) {

    int err = X509_STORE_CTX_get_error(x509_ctx);

    if (err != X509_V_OK) {

      /* cert validation error was not handled in callback */

      int depth = X509_STORE_CTX_get_error_depth(x509_ctx);

      const char *msg = X509_verify_cert_error_string(err);

      lws_strncpy(wsi->tls.err_helper, msg,

            sizeof(wsi->tls.err_helper));

      lwsl_err("SSL error: %s (preverify_ok=%d;err=%d;"

         "depth=%d)\n", msg, preverify_ok, err, depth);

#if defined(LWS_WITH_SYS_METRICS)

      {

        char buckname[64];

        lws_snprintf(buckname, sizeof(buckname),

               "tls=\"%s\"", msg);

        lws_metrics_hist_bump_describe_wsi(wsi,

          lws_metrics_priv_to_pub(wsi->a.context->mth_conn_failures),

          buckname);

      }

#endif

      return preverify_ok;  // not ok

    }

  }

  /*

   * convert callback return code from 0 = OK to verify callback

   * return value 1 = OK

   */

  return !n;

}

#endif

int

lws_ssl_client_bio_create(struct lws *wsi)

{

  char hostname[128], *p;

#if defined(LWS_HAVE_SSL_set_alpn_protos) && \

    defined(LWS_HAVE_SSL_get0_alpn_selected)

  uint8_t openssl_alpn[40];

  const char *alpn_comma = wsi->a.context->tls.alpn_default;

  int n;

#endif

  if (wsi->stash) {

    lws_strncpy(hostname, wsi->stash->cis[CIS_HOST], sizeof(hostname));

#if defined(LWS_HAVE_SSL_set_alpn_protos) && \

    defined(LWS_HAVE_SSL_get0_alpn_selected)

    alpn_comma = wsi->stash->cis[CIS_ALPN];

#endif

  } else {

#if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)

    if (lws_hdr_copy(wsi, hostname, sizeof(hostname),

         _WSI_TOKEN_CLIENT_HOST) <= 0)

#endif

    {

      lwsl_err("%s: Unable to get hostname\n", __func__);

      return -1;

    }

  }

  /*

   * remove any :port part on the hostname... necessary for network

   * connection but typical certificates do not contain it

   */

  p = hostname;

  while (*p) {

    if (*p == ':') {

      *p = '\0';

      break;

    }

    p++;

  }

  wsi->tls.ssl = SSL_new(wsi->a.vhost->tls.ssl_client_ctx);

  if (!wsi->tls.ssl) {

    const char *es = ERR_error_string(

#if defined(LWS_WITH_BORINGSSL)

  (uint32_t)

#else

  (unsigned long)

#endif

  lws_ssl_get_error(wsi, 0), NULL);

    lwsl_err("SSL_new failed: %s\n", es);

    lws_tls_err_describe_clear();

    return -1;

  }

#if defined(LWS_WITH_TLS_SESSIONS)

  if (!(wsi->a.vhost->options & LWS_SERVER_OPTION_DISABLE_TLS_SESSION_CACHE))

    lws_tls_reuse_session(wsi);

#endif

#if defined (LWS_HAVE_SSL_SET_INFO_CALLBACK)

  if (wsi->a.vhost->tls.ssl_info_event_mask)

    SSL_set_info_callback(wsi->tls.ssl, lws_ssl_info_callback);

#endif

#if defined(LWS_HAVE_X509_VERIFY_PARAM_set1_host)

  if (!(wsi->tls.use_ssl & LCCSCF_SKIP_SERVER_CERT_HOSTNAME_CHECK)) {

#if !defined(USE_WOLFSSL)

    X509_VERIFY_PARAM *param = SSL_get0_param(wsi->tls.ssl);

    /* Enable automatic hostname checks */

    X509_VERIFY_PARAM_set_hostflags(param,

          X509_CHECK_FLAG_NO_PARTIAL_WILDCARDS);

    /* Handle the case where the hostname is an IP address */

    if (!X509_VERIFY_PARAM_set1_ip_asc(param, hostname))

      X509_VERIFY_PARAM_set1_host(param, hostname,

          strnlen(hostname, sizeof(hostname)));

#endif

  }

#else

  if (!(wsi->tls.use_ssl & LCCSCF_SKIP_SERVER_CERT_HOSTNAME_CHECK)) {

    lwsl_err("%s: your tls lib is too old to have "

       "X509_VERIFY_PARAM_set1_host, failing all client tls\n",

       __func__);

    return -1;

  }

#endif

#if !defined(USE_WOLFSSL)

#ifndef USE_OLD_CYASSL

  /* OpenSSL_client_verify_callback will be called @ SSL_connect() */

  SSL_set_verify(wsi->tls.ssl, SSL_VERIFY_PEER,

           OpenSSL_client_verify_callback);

#endif

#endif

#if !defined(USE_WOLFSSL)

  SSL_set_mode(wsi->tls.ssl,  SSL_MODE_ACCEPT_MOVING_WRITE_BUFFER);

#endif

  /*

   * use server name indication (SNI), if supported,

   * when establishing connection

   */

#ifdef USE_WOLFSSL

#ifdef USE_OLD_CYASSL

#ifdef CYASSL_SNI_HOST_NAME

  CyaSSL_UseSNI(wsi->tls.ssl, CYASSL_SNI_HOST_NAME, hostname,

          strlen(hostname));

#endif

#else

#if defined(WOLFSSL_SNI_HOST_NAME) || defined(HAVE_SNI)

  wolfSSL_UseSNI(wsi->tls.ssl, WOLFSSL_SNI_HOST_NAME, hostname,

           (unsigned short)strlen(hostname));

#endif

#endif

#else

#ifdef SSL_CTRL_SET_TLSEXT_HOSTNAME

  SSL_set_tlsext_host_name(wsi->tls.ssl, hostname);

#endif

#endif

#ifdef USE_WOLFSSL

  /*

   * wolfSSL/CyaSSL does certificate verification differently

   * from OpenSSL.

   * If we should ignore the certificate, we need to set

   * this before SSL_new and SSL_connect is called.

   * Otherwise the connect will simply fail with error code -155

   */

#ifdef USE_OLD_CYASSL

  if (wsi->tls.use_ssl & LCCSCF_ALLOW_SELFSIGNED)

    CyaSSL_set_verify(wsi->tls.ssl, SSL_VERIFY_NONE, NULL);

#else

  if (wsi->tls.use_ssl & LCCSCF_ALLOW_SELFSIGNED)

    wolfSSL_set_verify(wsi->tls.ssl, SSL_VERIFY_NONE, NULL);

#endif

#endif /* USE_WOLFSSL */

  wsi->tls.client_bio = BIO_new_socket((int)(lws_intptr_t)wsi->desc.sockfd,

               BIO_NOCLOSE);

  SSL_set_bio(wsi->tls.ssl, wsi->tls.client_bio, wsi->tls.client_bio);

#ifdef USE_WOLFSSL

#ifdef USE_OLD_CYASSL

  CyaSSL_set_using_nonblock(wsi->tls.ssl, 1);

#else

  wolfSSL_set_using_nonblock(wsi->tls.ssl, 1);

#endif

#else

  BIO_set_nbio(wsi->tls.client_bio, 1); /* nonblocking */

#endif

#if defined(LWS_HAVE_SSL_set_alpn_protos) && \

    defined(LWS_HAVE_SSL_get0_alpn_selected)

  if (wsi->a.vhost->tls.alpn)

    alpn_comma = wsi->a.vhost->tls.alpn;

  if (wsi->stash) {

    alpn_comma = wsi->stash->cis[CIS_ALPN];

#if defined(LWS_ROLE_H1) || defined(LWS_ROLE_H2)

  } else {

    if (lws_hdr_copy(wsi, hostname, sizeof(hostname),

         _WSI_TOKEN_CLIENT_ALPN) > 0)

      alpn_comma = hostname;

#endif

  }

  lwsl_info("%s client conn using alpn list '%s'\n", wsi->role_ops->name, alpn_comma);

  n = lws_alpn_comma_to_openssl(alpn_comma, openssl_alpn,

              sizeof(openssl_alpn) - 1);

  SSL_set_alpn_protos(wsi->tls.ssl, openssl_alpn, (unsigned int)n);

#endif

  SSL_set_ex_data(wsi->tls.ssl, openssl_websocket_private_data_index,

      wsi);

  if (wsi->sys_tls_client_cert) {

    lws_system_blob_t *b = lws_system_get_blob(wsi->a.context,

          LWS_SYSBLOB_TYPE_CLIENT_CERT_DER,

          wsi->sys_tls_client_cert - 1);

    const uint8_t *data;

    size_t size;

    if (!b)

      goto no_client_cert;

    /*

     * Set up the per-connection client cert

     */

    size = lws_system_blob_get_size(b);

    if (!size)

      goto no_client_cert;

    if (lws_system_blob_get_single_ptr(b, &data))

      goto no_client_cert;

    if (SSL_use_certificate_ASN1(wsi->tls.ssl,

#if defined(USE_WOLFSSL)

      (unsigned char *)

#endif

          data,

#if defined(LWS_WITH_BORINGSSL)

          (size_t)

#else

          (int)

#endif

          size) != 1) {

      lwsl_err("%s: use_certificate failed\n", __func__);

      lws_tls_err_describe_clear();

      goto no_client_cert;

    }

    b = lws_system_get_blob(wsi->a.context,

          LWS_SYSBLOB_TYPE_CLIENT_KEY_DER,

          wsi->sys_tls_client_cert - 1);

    if (!b)

      goto no_client_cert;

    size = lws_system_blob_get_size(b);

    if (!size)

      goto no_client_cert;

    if (lws_system_blob_get_single_ptr(b, &data))

      goto no_client_cert;

    if (SSL_use_PrivateKey_ASN1(EVP_PKEY_RSA, wsi->tls.ssl,

#if defined(USE_WOLFSSL)

      (unsigned char *)

#endif

              data,

#if defined(LWS_WITH_BORINGSSL)

          (size_t)

#else

          (int)

#endif

              size) != 1 &&

        SSL_use_PrivateKey_ASN1(EVP_PKEY_EC, wsi->tls.ssl,

#if defined(USE_WOLFSSL)

      (unsigned char *)

#endif

              data,

#if defined(LWS_WITH_BORINGSSL)

          (size_t)

#else

          (int)

#endif

              size) != 1) {

      lwsl_err("%s: use_privkey failed\n", __func__);

      lws_tls_err_describe_clear();

      goto no_client_cert;

    }

    if (SSL_check_private_key(wsi->tls.ssl) != 1) {

      lwsl_err("Private SSL key doesn't match cert\n");

      lws_tls_err_describe_clear();

      return 1;

    }

    lwsl_notice("%s: set system client cert %u\n", __func__,

        wsi->sys_tls_client_cert - 1);

  }

  return 0;

no_client_cert:

  lwsl_err("%s: unable to set up system client cert %d\n", __func__,

      wsi->sys_tls_client_cert - 1);

  return 1;

}

enum lws_ssl_capable_status

lws_tls_client_connect(struct lws *wsi, char *errbuf, size_t elen)

{

#if defined(LWS_HAVE_SSL_set_alpn_protos) && \

    defined(LWS_HAVE_SSL_get0_alpn_selected)

  const unsigned char *prot;

  char a[32];

  unsigned int len;

#endif

  int m, n, en;

  unsigned long l;

#if defined(LWS_WITH_TLS_SESSIONS) && defined(LWS_HAVE_SSL_SESSION_set_time)

  SSL_SESSION *sess;

#endif

  errno = 0;

  ERR_clear_error();

  wsi->tls.err_helper[0] = '\0';

  n = SSL_connect(wsi->tls.ssl);

  en = errno;

  m = lws_ssl_get_error(wsi, n);

  if (m == SSL_ERROR_SYSCALL

#if defined(WIN32)

      && en

#endif

  ) {

#if defined(WIN32) || (_LWS_ENABLED_LOGS & LLL_INFO)

    lwsl_info("%s: n %d, m %d, errno %d\n", __func__, n, m, en);

#endif

    lws_snprintf(errbuf, elen, "connect SYSCALL %d", en);

    return LWS_SSL_CAPABLE_ERROR;

  }

  if (m == SSL_ERROR_SSL) {

    l = ERR_get_error();

    n = lws_snprintf(errbuf, elen, "tls: %s", wsi->tls.err_helper);

    if (!wsi->tls.err_helper[0])

      ERR_error_string_n((unsigned int)l, errbuf + n, (elen - (unsigned int)n));

    return LWS_SSL_CAPABLE_ERROR;

  }

#if defined(LWS_WITH_TLS_SESSIONS)

  if (SSL_session_reused(wsi->tls.ssl)) {

#if defined(LWS_HAVE_SSL_SESSION_set_time)

    sess = SSL_get_session(wsi->tls.ssl);

    if (sess) /* should always be true */

#if defined(OPENSSL_IS_BORINGSSL)

      SSL_SESSION_set_time(sess, (uint64_t)time(NULL)); /* extend session lifetime */

#else

      SSL_SESSION_set_time(sess, (long)time(NULL)); /* extend session lifetime */

#endif

#endif

  }

#endif

  if (m == SSL_ERROR_WANT_READ || SSL_want_read(wsi->tls.ssl))

    return LWS_SSL_CAPABLE_MORE_SERVICE_READ;

  if (m == SSL_ERROR_WANT_WRITE || SSL_want_write(wsi->tls.ssl))

    return LWS_SSL_CAPABLE_MORE_SERVICE_WRITE;

  if (n == 1 || m == SSL_ERROR_SYSCALL) {

#if defined(LWS_HAVE_SSL_set_alpn_protos) && \

    defined(LWS_HAVE_SSL_get0_alpn_selected)

    SSL_get0_alpn_selected(wsi->tls.ssl, &prot, &len);

    if (len >= sizeof(a))

      len = sizeof(a) - 1;

    memcpy(a, (const char *)prot, len);

    a[len] = '\0';

    lws_role_call_alpn_negotiated(wsi, (const char *)a);

#endif

#if defined(LWS_TLS_SYNTHESIZE_CB)

    lws_sul_schedule(wsi->a.context, wsi->tsi,

         &wsi->tls.sul_cb_synth,

         lws_sess_cache_synth_cb, 500 * LWS_US_PER_MS);

#endif

    lwsl_info("client connect OK\n");

    lws_openssl_describe_cipher(wsi);

    return LWS_SSL_CAPABLE_DONE;

  }

  if (!n) /* we don't know what he wants, but he says to retry */

    return LWS_SSL_CAPABLE_MORE_SERVICE;

  lws_snprintf(errbuf, elen, "connect unk %d", m);

  return LWS_SSL_CAPABLE_ERROR;

}

int

lws_tls_client_confirm_peer_cert(struct lws *wsi, char *ebuf, size_t ebuf_len)

{

#if !defined(USE_WOLFSSL)

  struct lws_context_per_thread *pt = &wsi->a.context->pt[(int)wsi->tsi];

  char *p = (char *)&pt->serv_buf[0];

  const char *es, *type = "";

  unsigned int avoid = 0;

  char *sb = p;

  long n;

  errno = 0;

  ERR_clear_error();

  n = SSL_get_verify_result(wsi->tls.ssl);

  switch (n) {

  case X509_V_OK:

    return 0;

  case X509_V_ERR_HOSTNAME_MISMATCH:

    type = "tls=hostname";

    avoid = LCCSCF_SKIP_SERVER_CERT_HOSTNAME_CHECK;

    break;

  case X509_V_ERR_INVALID_CA:

  case X509_V_ERR_DEPTH_ZERO_SELF_SIGNED_CERT:

  case X509_V_ERR_SELF_SIGNED_CERT_IN_CHAIN:

    type = "tls=invalidca";

    avoid = LCCSCF_ALLOW_SELFSIGNED;

    break;

  case X509_V_ERR_CERT_NOT_YET_VALID:

    type = "tls=notyetvalid";

    avoid = LCCSCF_ALLOW_EXPIRED;

    break;

  case X509_V_ERR_CERT_HAS_EXPIRED:

    type = "tls=expired";

    avoid = LCCSCF_ALLOW_EXPIRED;

    break;

  }

  lwsl_info("%s: cert problem: %s\n", __func__, type);

#if defined(LWS_WITH_SYS_METRICS)

  lws_metrics_hist_bump_describe_wsi(wsi,

      lws_metrics_priv_to_pub(wsi->a.context->mth_conn_failures), type);

#endif

  if (wsi->tls.use_ssl & avoid) {

    lwsl_info("%s: allowing anyway\n", __func__);

    return 0;

  }

  es = ERR_error_string(

  #if defined(LWS_WITH_BORINGSSL)

           (uint32_t)

  #else

           (unsigned long)

  #endif

           n, sb);

  lws_snprintf(ebuf, ebuf_len,

    "server's cert didn't look good, %s X509_V_ERR = %ld: %s\n",

     type, n, es);

  lwsl_info("%s\n", ebuf);

  lws_tls_err_describe_clear();

  return -1;

#else /* USE_WOLFSSL */

  return 0;

#endif

}

int

lws_tls_client_vhost_extra_cert_mem(struct lws_vhost *vh,

                const uint8_t *der, size_t der_len)

{

  X509_STORE *st;

#if defined(USE_WOLFSSL)

  X509 *x  = d2i_X509(NULL, &der, (int)der_len);

#else

  X509 *x  = d2i_X509(NULL, &der, (long)der_len);

#endif

  int n;

  if (!x) {

    lwsl_err("%s: Failed to load DER\n", __func__);

    lws_tls_err_describe_clear();

    return 1;

  }

  st = SSL_CTX_get_cert_store(vh->tls.ssl_client_ctx);

  if (!st) {

    lwsl_err("%s: failed to get cert store\n", __func__);

    X509_free(x);

    return 1;

  }

  n = X509_STORE_add_cert(st, x);

  if (n != 1)

    lwsl_err("%s: failed to add cert\n", __func__);

  X509_free(x);

  return n != 1;

}

#if defined(LWS_HAVE_SSL_CTX_set_keylog_callback) && \

  defined(LWS_WITH_TLS) && defined(LWS_WITH_CLIENT)

void

lws_klog_dump(const SSL *ssl, const char *line)

{

  struct lws *wsi = SSL_get_ex_data(ssl,

            openssl_websocket_private_data_index);

  char path[128], hdr[128], ts[64];

  size_t w = 0, wx = 0;

  int fd, t;

  if (!wsi || !wsi->a.context->keylog_file[0] || !wsi->a.vhost)

    return;

  lws_snprintf(path, sizeof(path), "%s.%s", wsi->a.context->keylog_file,

      wsi->a.vhost->name);

  fd = open(path, O_CREAT | O_RDWR | O_APPEND, 0600);

  if (fd == -1) {

    lwsl_vhost_warn(wsi->a.vhost, "Failed to append %s", path);

    return;

  }

  /* the first item in the chunk */

  if (!strncmp(line, "SERVER_HANDSHAKE_TRAFFIC_SECRET", 31)) {

    w += (size_t)write(fd, "\n# ", 3);

    wx += 3;

    t = lwsl_timestamp(LLL_WARN, ts, sizeof(ts));

    wx += (size_t)t;

    w += (size_t)write(fd, ts, (size_t)t);

    t = lws_snprintf(hdr, sizeof(hdr), "%s\n", wsi->lc.gutag);

    w += (size_t)write(fd, hdr, (size_t)t);

    wx += (size_t)t;

    lwsl_vhost_warn(wsi->a.vhost, "appended ssl keylog: %s", path);

  }

  wx += strlen(line) + 1;

  w += (size_t)write(fd, line, 

#if defined(WIN32)

      (unsigned int)

#endif

      strlen(line));

  w += (size_t)write(fd, "\n", 1);

  close(fd);

  if (w != wx) {

    lwsl_vhost_warn(wsi->a.vhost, "Failed to write %s", path);

    return;

  }

}

#endif

int

lws_tls_client_create_vhost_context(struct lws_vhost *vh,

            const struct lws_context_creation_info *info,

            const char *cipher_list,

            const char *ca_filepath,

            const void *ca_mem,

            unsigned int ca_mem_len,

            const char *cert_filepath,

            const void *cert_mem,

            unsigned int cert_mem_len,

            const char *private_key_filepath,

          const void *key_mem,

            unsigned int key_mem_len

          )

{

  struct lws_tls_client_reuse *tcr;

  X509_STORE *x509_store;

  unsigned long error;

  SSL_METHOD *method;

  EVP_MD_CTX *mdctx;

  unsigned int len;

  uint8_t hash[32];

  X509 *client_CA;

  char c;

  int n;

  /* basic openssl init already happened in context init */

  /* choose the most recent spin of the api */

#if defined(LWS_HAVE_TLS_CLIENT_METHOD)

  method = (SSL_METHOD *)TLS_client_method();

#elif defined(LWS_HAVE_TLSV1_2_CLIENT_METHOD)

  method = (SSL_METHOD *)TLSv1_2_client_method();

#else

  method = (SSL_METHOD *)SSLv23_client_method();

#endif

  if (!method) {

    const char *es;

    error = ERR_get_error();

    es = ERR_error_string(

    #if defined(LWS_WITH_BORINGSSL)

      (uint32_t)

    #else

      (unsigned long)

    #endif

       error, (char *)vh->context->pt[0].serv_buf);

    lwsl_err("problem creating ssl method %lu: %s\n",

      error, es);

    return 1;

  }

  /*

   * OpenSSL client contexts are quite expensive, because they bring in

   * the system certificate bundle for each one.  So if you have multiple

   * vhosts, each with a client context, it can add up to several

   * megabytes of heap.  In the case the client contexts are configured

   * identically, they could perfectly well have shared just the one.

   *

   * For that reason, use a hash to fingerprint the context configuration

   * and prefer to reuse an existing one with the same fingerprint if

   * possible.

   */

   mdctx = EVP_MD_CTX_create();

   if (!mdctx)

     return 1;

  if (EVP_DigestInit_ex(mdctx, EVP_sha256(), NULL) != 1) {

    EVP_MD_CTX_destroy(mdctx);

    return 1;

  }

  if (info->ssl_client_options_set)

    EVP_DigestUpdate(mdctx, &info->ssl_client_options_set,

         sizeof(info->ssl_client_options_set));

#if (OPENSSL_VERSION_NUMBER >= 0x009080df) && !defined(USE_WOLFSSL)

  if (info->ssl_client_options_clear)

    EVP_DigestUpdate(mdctx, &info->ssl_client_options_clear,

         sizeof(info->ssl_client_options_clear));

#endif

  if (cipher_list)

    EVP_DigestUpdate(mdctx, cipher_list, strlen(cipher_list));

#if defined(LWS_HAVE_SSL_CTX_set_ciphersuites)

  if (info->client_tls_1_3_plus_cipher_list)

    EVP_DigestUpdate(mdctx, info->client_tls_1_3_plus_cipher_list,

         strlen(info->client_tls_1_3_plus_cipher_list));

#endif

  if (!lws_check_opt(vh->options, LWS_SERVER_OPTION_DISABLE_OS_CA_CERTS)) {

    c = 1;

    EVP_DigestUpdate(mdctx, &c, 1);

  }

  if (ca_filepath)

    EVP_DigestUpdate(mdctx, ca_filepath, strlen(ca_filepath));

  if (cert_filepath)

    EVP_DigestUpdate(mdctx, cert_filepath, strlen(cert_filepath));

  if (private_key_filepath)

    EVP_DigestUpdate(mdctx, private_key_filepath,

         strlen(private_key_filepath));

  if (ca_mem && ca_mem_len)

    EVP_DigestUpdate(mdctx, ca_mem, ca_mem_len);

  if (cert_mem && cert_mem_len)

    EVP_DigestUpdate(mdctx, cert_mem, cert_mem_len);

  len = sizeof(hash);

  EVP_DigestFinal_ex(mdctx, hash, &len);

  EVP_MD_CTX_destroy(mdctx);

  /* look for existing client context with same config already */

  lws_start_foreach_dll_safe(struct lws_dll2 *, p, tp,

       lws_dll2_get_head(&vh->context->tls.cc_owner)) {

    tcr = lws_container_of(p, struct lws_tls_client_reuse, cc_list);

    if (!memcmp(hash, tcr->hash, len)) {

      /* it's a match */

      tcr->refcount++;

      vh->tls.ssl_client_ctx = tcr->ssl_client_ctx;

      vh->tls.tcr = tcr;

      lwsl_info("%s: vh %s: reusing client ctx %d: use %d\n",

           __func__, vh->name, tcr->index,

           tcr->refcount);

      return 0;

    }

  } lws_end_foreach_dll_safe(p, tp);

  /* no existing one the same... create new client SSL_CTX */

  errno = 0;

  ERR_clear_error();

  vh->tls.ssl_client_ctx = SSL_CTX_new(method);

  if (!vh->tls.ssl_client_ctx) {

    const char *es;

    error = ERR_get_error();

    es = ERR_error_string(

    #if defined(LWS_WITH_BORINGSSL)

      (uint32_t)

    #else

      (unsigned long)

    #endif

       error, (char *)vh->context->pt[0].serv_buf);

    lwsl_err("problem creating ssl context %lu: %s\n",

      error, es);

    return 1;

  }

  SSL_CTX_set_ex_data(vh->tls.ssl_client_ctx,

        openssl_SSL_CTX_private_data_index,

        (char *)vh->context);

  lws_plat_vhost_tls_client_ctx_init(vh);

  tcr = lws_zalloc(sizeof(*tcr), "client ctx tcr");

  if (!tcr) {

    SSL_CTX_free(vh->tls.ssl_client_ctx);

    return 1;

  }