//
//
// Copyright 2017 gRPC authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//

#include "src/core/tsi/ssl_transport_security.h"

#include <grpc/grpc.h>
#include <grpc/support/alloc.h>
#include <grpc/support/string_util.h>
#include <openssl/crypto.h>
#include <openssl/err.h>
#include <openssl/pem.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>

#include <string>
#include <thread>
#include <vector>

#include "src/core/tsi/transport_security.h"
#include "src/core/tsi/transport_security_interface.h"
#include "src/core/util/memory.h"
#include "test/core/test_util/build.h"
#include "test/core/test_util/test_config.h"
#include "test/core/test_util/tls_utils.h"
#include "test/core/tsi/transport_security_test_lib.h"
#include "gtest/gtest.h"
#include "absl/log/log.h"
#include "absl/strings/str_cat.h"

#define SSL_TSI_TEST_ALPN1 "foo"
#define SSL_TSI_TEST_ALPN2 "toto"
#define SSL_TSI_TEST_ALPN3 "baz"
#define SSL_TSI_TEST_ALPN4 "bar"
#define SSL_TSI_TEST_ALPN_NUM 2
#define SSL_TSI_TEST_SERVER_KEY_CERT_PAIRS_NUM 2
#define SSL_TSI_TEST_BAD_SERVER_KEY_CERT_PAIRS_NUM 1
#define SSL_TSI_TEST_LEAF_SIGNED_BY_INTERMEDIATE_KEY_CERT_PAIRS_NUM 1
#define SSL_TSI_TEST_CREDENTIALS_DIR "src/core/tsi/test_creds/"
#define SSL_TSI_TEST_WRONG_SNI "test.google.cn"
#define SSL_TSI_TEST_INVALID_SNI "1.2.3.4"

// OpenSSL 1.1 uses AES256 for encryption session ticket by default so specify
// different STEK size.
#if OPENSSL_VERSION_NUMBER >= 0x10100000 && !defined(OPENSSL_IS_BORINGSSL)
const size_t kSessionTicketEncryptionKeySize = 80;
#else
const size_t kSessionTicketEncryptionKeySize = 48;
#endif

constexpr static size_t kTls13FrameOverhead = 22;

typedef enum AlpnMode {
  NO_ALPN,
  ALPN_CLIENT_NO_SERVER,
  ALPN_SERVER_NO_CLIENT,
  ALPN_CLIENT_SERVER_OK,
  ALPN_CLIENT_SERVER_MISMATCH
} AlpnMode;

typedef struct ssl_alpn_lib {
  AlpnMode alpn_mode;
  const char** server_alpn_protocols;
  const char** client_alpn_protocols;
  uint16_t num_server_alpn_protocols;
  uint16_t num_client_alpn_protocols;
} ssl_alpn_lib;

typedef struct ssl_key_cert_lib {
  bool use_bad_server_cert;
  bool use_bad_client_cert;
  bool use_root_store;
  bool use_pem_root_certs;
  bool use_cert_signed_by_intermediate_ca;
  bool skip_server_certificate_verification;
  char* root_cert;
  tsi_ssl_root_certs_store* root_store;
  std::vector<tsi_ssl_pem_key_cert_pair> server_pem_key_cert_pairs;
  std::vector<tsi_ssl_pem_key_cert_pair> bad_server_pem_key_cert_pairs;
  std::vector<tsi_ssl_pem_key_cert_pair>
      leaf_signed_by_intermediate_key_cert_pairs;
  tsi_ssl_pem_key_cert_pair client_pem_key_cert_pair;
  tsi_ssl_pem_key_cert_pair bad_client_pem_key_cert_pair;
  uint16_t server_num_key_cert_pairs;
  uint16_t bad_server_num_key_cert_pairs;
  uint16_t leaf_signed_by_intermediate_num_key_cert_pairs;
} ssl_key_cert_lib;

static bool check_property(tsi_peer* peer, const char* property_name,
                           const char* property_value) {
  for (size_t i = 0; i < peer->property_count; i++) {
    const tsi_peer_property* prop = &peer->properties[i];
    if (strcmp(prop->name, property_name) == 0) {
      if (strlen(property_value) == prop->value.length &&
          memcmp(prop->value.data, property_value, prop->value.length) == 0) {
        return true;
      }
    }
  }
  return false;
}

static char* load_file(const std::string& path) {
  std::string data = grpc_core::testing::GetFileContents(path);
  return gpr_strdup(data.c_str());
}

static bool is_slow_build() {
#if defined(GPR_ARCH_32) || defined(__APPLE__)
  return true;
#else
  return BuiltUnderMsan() || BuiltUnderTsan() || BuiltUnderUbsan();
#endif
}

static std::string GenerateTrustBundle() {
  // Create a trust bundle, consisting of 200 self-signed certs. The self-signed
  // certs have subject DNs that are sufficiently big and complex that they
  // substantially increase the server handshake message size.
  std::string trust_bundle;
  int trust_bundle_size = is_slow_build() ? 20 : 200;
  for (int i = 0; i < trust_bundle_size; ++i) {
    SelfSignedCertificateOptions options;
    options.common_name =
        absl::StrCat("{46f0eaed-6e05-43f5-9289-379104612fc", i, "}");
    options.organization = absl::StrCat("organization-", i);
    options.organizational_unit = absl::StrCat("organizational-unit-", i);
    std::string self_signed_cert = GenerateSelfSignedCertificate(options);
    trust_bundle.append(self_signed_cert);
  }
  return trust_bundle;
}

class SslTransportSecurityTest
    : public ::testing::TestWithParam<std::tuple<tsi_tls_version, bool>> {
 protected:
  // A tsi_test_fixture implementation
  class SslTsiTestFixture {
   public:
    SslTsiTestFixture(tsi_tls_version tls_version, bool send_client_ca_list) {
      tsi_test_fixture_init(&base_);
      verify_root_cert_subject_ = true;
      base_.test_unused_bytes = true;
      base_.vtable = &kVtable;
      // Create ssl_key_cert_lib.
      key_cert_lib_ = new ssl_key_cert_lib();
      key_cert_lib_->use_bad_server_cert = false;
      key_cert_lib_->use_bad_client_cert = false;
      key_cert_lib_->use_root_store = false;
      key_cert_lib_->use_pem_root_certs = true;
      key_cert_lib_->skip_server_certificate_verification = false;
      key_cert_lib_->server_num_key_cert_pairs =
          SSL_TSI_TEST_SERVER_KEY_CERT_PAIRS_NUM;
      key_cert_lib_->bad_server_num_key_cert_pairs =
          SSL_TSI_TEST_BAD_SERVER_KEY_CERT_PAIRS_NUM;
      key_cert_lib_->leaf_signed_by_intermediate_num_key_cert_pairs =
          SSL_TSI_TEST_LEAF_SIGNED_BY_INTERMEDIATE_KEY_CERT_PAIRS_NUM;
      key_cert_lib_->server_pem_key_cert_pairs.emplace_back(
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "server0.key"),
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "server0.pem"));
      key_cert_lib_->server_pem_key_cert_pairs.emplace_back(
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "server1.key"),
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "server1.pem"));
      key_cert_lib_->bad_server_pem_key_cert_pairs.emplace_back(
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "badserver.key"),
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "badserver.pem"));
      key_cert_lib_->client_pem_key_cert_pair.private_key =
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "client.key");
      key_cert_lib_->client_pem_key_cert_pair.cert_chain =
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "client.pem");
      key_cert_lib_->bad_client_pem_key_cert_pair.private_key =
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "badclient.key");
      key_cert_lib_->bad_client_pem_key_cert_pair.cert_chain =
          grpc_core::testing::GetFileContents(SSL_TSI_TEST_CREDENTIALS_DIR
                                              "badclient.pem");
      key_cert_lib_->bad_server_pem_key_cert_pairs.emplace_back(
          grpc_core::testing::GetFileContents(
              SSL_TSI_TEST_CREDENTIALS_DIR "leaf_signed_by_intermediate.key"),
          grpc_core::testing::GetFileContents(
              SSL_TSI_TEST_CREDENTIALS_DIR "leaf_and_intermediate_chain.pem"));
      key_cert_lib_->root_cert =
          load_file(SSL_TSI_TEST_CREDENTIALS_DIR "ca.pem");
      key_cert_lib_->root_store =
          tsi_ssl_root_certs_store_create(key_cert_lib_->root_cert);
      EXPECT_NE(key_cert_lib_->root_store, nullptr);
      // Create ssl_alpn_lib.
      alpn_lib_ = grpc_core::Zalloc<ssl_alpn_lib>();
      alpn_lib_->server_alpn_protocols = static_cast<const char**>(
          gpr_zalloc(sizeof(char*) * SSL_TSI_TEST_ALPN_NUM));
      alpn_lib_->client_alpn_protocols = static_cast<const char**>(
          gpr_zalloc(sizeof(char*) * SSL_TSI_TEST_ALPN_NUM));
      alpn_lib_->server_alpn_protocols[0] = gpr_strdup(SSL_TSI_TEST_ALPN1);
      alpn_lib_->server_alpn_protocols[1] = gpr_strdup(SSL_TSI_TEST_ALPN3);
      alpn_lib_->client_alpn_protocols[0] = gpr_strdup(SSL_TSI_TEST_ALPN2);
      alpn_lib_->client_alpn_protocols[1] = gpr_strdup(SSL_TSI_TEST_ALPN3);
      alpn_lib_->num_server_alpn_protocols = SSL_TSI_TEST_ALPN_NUM;
      alpn_lib_->num_client_alpn_protocols = SSL_TSI_TEST_ALPN_NUM;
      alpn_lib_->alpn_mode = NO_ALPN;
      server_name_indication_ = nullptr;
      session_reused_ = false;
      session_ticket_key_ = nullptr;
      session_ticket_key_size_ = 0;
      force_client_auth_ = false;
      network_bio_buf_size_ = 0;
      ssl_bio_buf_size_ = 0;
      send_client_ca_list_ = send_client_ca_list;
      tls_version_ = tls_version;
    }

    ~SslTsiTestFixture() {
      // Destroy ssl_alpn_lib.
      for (size_t i = 0; i < alpn_lib_->num_server_alpn_protocols; i++) {
        gpr_free(const_cast<char*>(alpn_lib_->server_alpn_protocols[i]));
      }
      gpr_free(alpn_lib_->server_alpn_protocols);
      for (size_t i = 0; i < alpn_lib_->num_client_alpn_protocols; i++) {
        gpr_free(const_cast<char*>(alpn_lib_->client_alpn_protocols[i]));
      }
      gpr_free(alpn_lib_->client_alpn_protocols);
      gpr_free(alpn_lib_);
      gpr_free(key_cert_lib_->root_cert);
      tsi_ssl_root_certs_store_destroy(key_cert_lib_->root_store);
      delete key_cert_lib_;
      if (session_cache_ != nullptr) {
        tsi_ssl_session_cache_unref(session_cache_);
      }
      // Unreference others.
      tsi_ssl_server_handshaker_factory_unref(server_handshaker_factory_);
      tsi_ssl_client_handshaker_factory_unref(client_handshaker_factory_);
    }

    tsi_test_fixture* GetBaseFixture() { return &base_; }

    void SetVerifyRootCertSubject(bool verify_root_cert_subject) {
      verify_root_cert_subject_ = verify_root_cert_subject;
    }

    ssl_key_cert_lib* MutableKeyCertLib() { return key_cert_lib_; }

    void SetForceClientAuth(bool force_client_auth) {
      force_client_auth_ = force_client_auth;
    }

    void SetAlpnMode(AlpnMode alpn_mode) { alpn_lib_->alpn_mode = alpn_mode; }

    void SetServerNameIndication(char* server_name_indication) {
      server_name_indication_ = server_name_indication;
    }

    void SetSessionCache(tsi_ssl_session_cache* cache) {
      session_cache_ = cache;
    }

    void SetSessionReused(bool session_reused) {
      session_reused_ = session_reused;
    }

    void SetSessionTicketKey(const char* session_ticket_key,
                             size_t session_ticket_key_size) {
      session_ticket_key_ = session_ticket_key;
      session_ticket_key_size_ = session_ticket_key_size;
    }

    void SetBioBufSizes(size_t network_bio_buf_size, size_t ssl_bio_buf_size) {
      network_bio_buf_size_ = network_bio_buf_size;
      ssl_bio_buf_size_ = ssl_bio_buf_size;
    }

    void OverrideHanshakerAlpnClientProtocols(
        std::optional<std::string> alpn_client_protocols) {
      alpn_client_overriden_protocols_ = alpn_client_protocols;
    }

    void OverrideHanshakerAlpnServerProtocols(
        std::optional<std::string> alpn_server_protocols) {
      alpn_server_overriden_protocols_ = alpn_server_protocols;
    }

    void SetExpectedAlpnNegotiatedProtocol(
        std::optional<std::string> expected_alpn_negotiated_protocol) {
      expected_alpn_negotiated_protocol_ = expected_alpn_negotiated_protocol;
    }

   private:
    static void SetupHandshakers(tsi_test_fixture* fixture) {
      SslTsiTestFixture* ssl_fixture =
          reinterpret_cast<SslTsiTestFixture*>(fixture);
      ASSERT_NE(ssl_fixture, nullptr);
      ASSERT_NE(ssl_fixture->key_cert_lib_, nullptr);
      ASSERT_NE(ssl_fixture->alpn_lib_, nullptr);
      ssl_key_cert_lib* key_cert_lib = ssl_fixture->key_cert_lib_;
      ssl_alpn_lib* alpn_lib = ssl_fixture->alpn_lib_;
      // Create client handshaker factory.
      tsi_ssl_client_handshaker_options client_options;
      if (key_cert_lib->use_pem_root_certs &&
          key_cert_lib->root_cert != nullptr) {
        client_options.root_cert_info =
            std::make_shared<tsi::RootCertInfo>(key_cert_lib->root_cert);
      }
      if (ssl_fixture->force_client_auth_) {
        client_options.pem_key_cert_pair =
            key_cert_lib->use_bad_client_cert
                ? &key_cert_lib->bad_client_pem_key_cert_pair
                : &key_cert_lib->client_pem_key_cert_pair;
      }
      if (alpn_lib->alpn_mode == ALPN_CLIENT_NO_SERVER ||
          alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_OK ||
          alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_MISMATCH) {
        client_options.alpn_protocols = alpn_lib->client_alpn_protocols;
        client_options.num_alpn_protocols = alpn_lib->num_client_alpn_protocols;
      }
      client_options.root_store =
          key_cert_lib->use_root_store ? key_cert_lib->root_store : nullptr;
      if (ssl_fixture->session_cache_ != nullptr) {
        client_options.session_cache = ssl_fixture->session_cache_;
      }
      client_options.min_tls_version = ssl_fixture->tls_version_;
      client_options.max_tls_version = ssl_fixture->tls_version_;
      client_options.skip_server_certificate_verification =
          key_cert_lib->skip_server_certificate_verification;
      ASSERT_EQ(tsi_create_ssl_client_handshaker_factory_with_options(
                    &client_options, &ssl_fixture->client_handshaker_factory_),
                TSI_OK);
      // Create server handshaker factory.
      tsi_ssl_server_handshaker_options server_options;
      if (alpn_lib->alpn_mode == ALPN_SERVER_NO_CLIENT ||
          alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_OK ||
          alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_MISMATCH) {
        if (ssl_fixture->alpn_server_overriden_protocols_.has_value()) {
          size_t num_parsed_protocols = 0;
          server_options.alpn_protocols = ParseAlpnStringIntoArray(
              ssl_fixture->alpn_server_overriden_protocols_.value(),
              static_cast<size_t*>(&num_parsed_protocols));
          server_options.num_alpn_protocols = num_parsed_protocols;
        } else {
          server_options.alpn_protocols = alpn_lib->server_alpn_protocols;
          server_options.num_alpn_protocols =
              alpn_lib->num_server_alpn_protocols;
        }
        if (alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_MISMATCH) {
          server_options.num_alpn_protocols--;
        }
      }
      if (key_cert_lib->use_cert_signed_by_intermediate_ca) {
        server_options.pem_key_cert_pairs =
            key_cert_lib->leaf_signed_by_intermediate_key_cert_pairs;
      } else {
        server_options.pem_key_cert_pairs =
            key_cert_lib->use_bad_server_cert
                ? key_cert_lib->bad_server_pem_key_cert_pairs
                : key_cert_lib->server_pem_key_cert_pairs;
      }
      if (key_cert_lib->root_cert != nullptr) {
        server_options.root_cert_info =
            std::make_shared<tsi::RootCertInfo>(key_cert_lib->root_cert);
      }
      if (ssl_fixture->force_client_auth_) {
        server_options.client_certificate_request =
            TSI_REQUEST_AND_REQUIRE_CLIENT_CERTIFICATE_AND_VERIFY;
      } else {
        server_options.client_certificate_request =
            TSI_DONT_REQUEST_CLIENT_CERTIFICATE;
      }
      server_options.send_client_ca_list = ssl_fixture->send_client_ca_list_;
      server_options.session_ticket_key = ssl_fixture->session_ticket_key_;
      server_options.session_ticket_key_size =
          ssl_fixture->session_ticket_key_size_;
      server_options.min_tls_version = ssl_fixture->tls_version_;
      server_options.max_tls_version = ssl_fixture->tls_version_;
      ASSERT_EQ(tsi_create_ssl_server_handshaker_factory_with_options(
                    &server_options, &ssl_fixture->server_handshaker_factory_),
                TSI_OK);
      if (ssl_fixture->alpn_server_overriden_protocols_.has_value()) {
        // Free memory from creating the server protocol list if
        // ParseAlpnStringIntoArray was used.
        for (size_t i = 0; i < server_options.num_alpn_protocols; i++) {
          gpr_free(const_cast<char*>(server_options.alpn_protocols[i]));
        }
        gpr_free(server_options.alpn_protocols);
      }
      // Create server and client handshakers.
      ASSERT_EQ(tsi_ssl_client_handshaker_factory_create_handshaker(
                    ssl_fixture->client_handshaker_factory_,
                    ssl_fixture->server_name_indication_,
                    ssl_fixture->network_bio_buf_size_,
                    ssl_fixture->ssl_bio_buf_size_,
                    ssl_fixture->alpn_client_overriden_protocols_,
                    &ssl_fixture->base_.client_handshaker),
                TSI_OK);
      ASSERT_EQ(tsi_ssl_server_handshaker_factory_create_handshaker(
                    ssl_fixture->server_handshaker_factory_,
                    ssl_fixture->network_bio_buf_size_,
                    ssl_fixture->ssl_bio_buf_size_,
                    &ssl_fixture->base_.server_handshaker),
                TSI_OK);
    }

    static void CheckAlpn(SslTsiTestFixture* ssl_fixture,
                          const tsi_peer* peer) {
      ASSERT_NE(ssl_fixture, nullptr);
      ASSERT_NE(ssl_fixture->alpn_lib_, nullptr);
      ssl_alpn_lib* alpn_lib = ssl_fixture->alpn_lib_;
      const tsi_peer_property* alpn_property =
          tsi_peer_get_property_by_name(peer, TSI_SSL_ALPN_SELECTED_PROTOCOL);
      if (alpn_lib->alpn_mode != ALPN_CLIENT_SERVER_OK) {
        ASSERT_EQ(alpn_property, nullptr);
      } else {
        ASSERT_NE(alpn_property, nullptr);
        std::string expected_match;
        if (ssl_fixture->expected_alpn_negotiated_protocol_.has_value()) {
          expected_match =
              ssl_fixture->expected_alpn_negotiated_protocol_.value();
        } else {
          expected_match = "baz";
        }
        ASSERT_EQ(
            std::string(alpn_property->value.data, alpn_property->value.length),
            expected_match);
      }
    }

    static void CheckSecurityLevel(const tsi_peer* peer) {
      const tsi_peer_property* security_level =
          tsi_peer_get_property_by_name(peer, TSI_SECURITY_LEVEL_PEER_PROPERTY);
      ASSERT_NE(security_level, nullptr);
      const char* expected_match = "TSI_PRIVACY_AND_INTEGRITY";
      ASSERT_EQ(memcmp(security_level->value.data, expected_match,
                       security_level->value.length),
                0);
    }

    static const tsi_peer_property* CheckBasicAuthenticatedPeerAndGetCommonName(
        const tsi_peer* peer) {
      const tsi_peer_property* cert_type_property =
          tsi_peer_get_property_by_name(peer,
                                        TSI_CERTIFICATE_TYPE_PEER_PROPERTY);
      EXPECT_NE(cert_type_property, nullptr);
      EXPECT_EQ(
          memcmp(cert_type_property->value.data, TSI_X509_CERTIFICATE_TYPE,
                 cert_type_property->value.length),
          0);
      const tsi_peer_property* property = tsi_peer_get_property_by_name(
          peer, TSI_X509_SUBJECT_COMMON_NAME_PEER_PROPERTY);
      EXPECT_NE(property, nullptr);
      return property;
    }

    static void CheckSessionReusage(SslTsiTestFixture* ssl_fixture,
                                    tsi_peer* peer) {
      const tsi_peer_property* session_reused = tsi_peer_get_property_by_name(
          peer, TSI_SSL_SESSION_REUSED_PEER_PROPERTY);
      ASSERT_NE(session_reused, nullptr);
      if (ssl_fixture->session_reused_) {
        ASSERT_EQ(strncmp(session_reused->value.data, "true",
                          session_reused->value.length),
                  0);
      } else {
        ASSERT_EQ(strncmp(session_reused->value.data, "false",
                          session_reused->value.length),
                  0);
      }
    }

    // This is tied specifically to server0.pem loaded by the fixture.
    static void CheckServer0Peer(tsi_peer* peer) {
      const tsi_peer_property* property =
          CheckBasicAuthenticatedPeerAndGetCommonName(peer);
      const char* expected_match = "*.test.google.com.au";
      ASSERT_EQ(
          memcmp(property->value.data, expected_match, property->value.length),
          0);
      ASSERT_EQ(tsi_peer_get_property_by_name(
                    peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY),
                nullptr);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "foo.test.google.com.au"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "bar.test.google.com.au"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "BAR.TEST.GOOGLE.COM.AU"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "Bar.Test.Google.Com.Au"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "bAr.TeST.gOOgle.cOm.AU"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "bar.test.google.blah"), 0);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "foo.bar.test.google.com.au"),
                0);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "test.google.com.au"), 0);
      tsi_peer_destruct(peer);
    }

    // This is tied specifically to server1.pem loaded by the fixture.
    static void CheckServer1Peer(tsi_peer* peer) {
      const tsi_peer_property* property =
          CheckBasicAuthenticatedPeerAndGetCommonName(peer);
      const char* expected_match = "*.test.google.com";
      ASSERT_EQ(
          memcmp(property->value.data, expected_match, property->value.length),
          0);
      ASSERT_EQ(
          check_property(peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                         "*.test.google.fr"),
          1);
      ASSERT_EQ(
          check_property(peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                         "waterzooi.test.google.be"),
          1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "foo.test.google.fr"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "bar.test.google.fr"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "waterzooi.test.google.be"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "foo.test.youtube.com"), 1);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "bar.foo.test.google.com"), 0);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "test.google.fr"), 0);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "tartines.test.google.be"), 0);
      ASSERT_EQ(tsi_ssl_peer_matches_name(peer, "tartines.youtube.com"), 0);
      tsi_peer_destruct(peer);
    }

    static void CheckClientPeer(SslTsiTestFixture* ssl_fixture,
                                tsi_peer* peer) {
      ASSERT_NE(ssl_fixture, nullptr);
      ASSERT_NE(ssl_fixture->alpn_lib_, nullptr);
      ssl_alpn_lib* alpn_lib = ssl_fixture->alpn_lib_;
      if (!ssl_fixture->force_client_auth_) {
        ASSERT_EQ(peer->property_count,
                  (alpn_lib->alpn_mode == ALPN_CLIENT_SERVER_OK ? 3 : 2));
      } else {
        const tsi_peer_property* property =
            CheckBasicAuthenticatedPeerAndGetCommonName(peer);
        const char* expected_match = "testclient";
        ASSERT_EQ(memcmp(property->value.data, expected_match,
                         property->value.length),
                  0);
      }
      tsi_peer_destruct(peer);
    }

    static void CheckVerifiedRootCertSubject(const tsi_peer* peer) {
      const tsi_peer_property* verified_root_cert_subject =
          tsi_peer_get_property_by_name(
              peer, TSI_X509_VERIFIED_ROOT_CERT_SUBECT_PEER_PROPERTY);
      ASSERT_NE(verified_root_cert_subject, nullptr);
      const char* expected_match =
          "CN=testca,O=Internet Widgits Pty Ltd,ST=Some-State,C=AU";
      ASSERT_EQ(memcmp(verified_root_cert_subject->value.data, expected_match,
                       verified_root_cert_subject->value.length),
                0);
    }

    static void CheckVerifiedRootCertSubjectUnset(const tsi_peer* peer) {
      const tsi_peer_property* verified_root_cert_subject =
          tsi_peer_get_property_by_name(
              peer, TSI_X509_VERIFIED_ROOT_CERT_SUBECT_PEER_PROPERTY);
      ASSERT_EQ(verified_root_cert_subject, nullptr);
    }

    static void CheckHandshakerPeers(tsi_test_fixture* fixture) {
      SslTsiTestFixture* ssl_fixture =
          reinterpret_cast<SslTsiTestFixture*>(fixture);
      ASSERT_NE(ssl_fixture, nullptr);
      ASSERT_NE(ssl_fixture->key_cert_lib_, nullptr);
      ssl_key_cert_lib* key_cert_lib = ssl_fixture->key_cert_lib_;
      tsi_peer peer;
      // In TLS 1.3, the client-side handshake succeeds even if the client sends
      // a bad certificate. In such a case, the server would fail the TLS
      // handshake and send an alert to the client as the first application data
      // message. In TLS 1.2, the client-side handshake will fail if the client
      // sends a bad certificate.
      //
      // For OpenSSL versions < 1.1, TLS 1.3 is not supported, so the
      // client-side handshake should succeed precisely when the server-side
      // handshake succeeds.
      bool expect_server_success = !(key_cert_lib->use_bad_server_cert ||
                                     (key_cert_lib->use_bad_client_cert &&
                                      ssl_fixture->force_client_auth_));
#if OPENSSL_VERSION_NUMBER >= 0x10100000
      bool expect_client_success =
          ssl_fixture->tls_version_ == tsi_tls_version::TSI_TLS1_2
              ? expect_server_success
              : !key_cert_lib->use_bad_server_cert;
#else
      bool expect_client_success = expect_server_success;
#endif
      if (expect_client_success) {
        ASSERT_EQ(tsi_handshaker_result_extract_peer(
                      ssl_fixture->base_.client_result, &peer),
                  TSI_OK);
        CheckSessionReusage(ssl_fixture, &peer);
        CheckAlpn(ssl_fixture, &peer);
        CheckSecurityLevel(&peer);
        if (ssl_fixture->verify_root_cert_subject_) {
          if (!ssl_fixture->session_reused_) {
            CheckVerifiedRootCertSubject(&peer);
          } else {
            CheckVerifiedRootCertSubjectUnset(&peer);
          }
        }
        if (ssl_fixture->server_name_indication_ == nullptr ||
            strcmp(ssl_fixture->server_name_indication_,
                   SSL_TSI_TEST_WRONG_SNI) == 0 ||
            strcmp(ssl_fixture->server_name_indication_,
                   SSL_TSI_TEST_INVALID_SNI) == 0) {
          // Expect server to use default server0.pem.
          CheckServer0Peer(&peer);
        } else {
          // Expect server to use server1.pem.
          CheckServer1Peer(&peer);
        }
      } else {
        ASSERT_EQ(ssl_fixture->base_.client_result, nullptr);
      }
      if (expect_server_success) {
        ASSERT_EQ(tsi_handshaker_result_extract_peer(
                      ssl_fixture->base_.server_result, &peer),
                  TSI_OK);
        CheckSessionReusage(ssl_fixture, &peer);
        CheckAlpn(ssl_fixture, &peer);
        CheckSecurityLevel(&peer);
        if (ssl_fixture->force_client_auth_ && !ssl_fixture->session_reused_) {
          CheckVerifiedRootCertSubject(&peer);
        } else {
          CheckVerifiedRootCertSubjectUnset(&peer);
        }
        CheckClientPeer(ssl_fixture, &peer);
      } else {
        ASSERT_EQ(ssl_fixture->base_.server_result, nullptr);
      }
    }

    static void Destruct(tsi_test_fixture* fixture) {
      auto* self = reinterpret_cast<SslTsiTestFixture*>(fixture);
      delete self;
    }

    static tsi_test_fixture_vtable kVtable;

    tsi_test_fixture base_;
    ssl_key_cert_lib* key_cert_lib_ = nullptr;
    ssl_alpn_lib* alpn_lib_;
    bool force_client_auth_;
    char* server_name_indication_ = nullptr;
    tsi_ssl_session_cache* session_cache_ = nullptr;
    bool session_reused_;
    const char* session_ticket_key_ = nullptr;
    size_t session_ticket_key_size_;
    size_t network_bio_buf_size_;
    size_t ssl_bio_buf_size_;
    bool verify_root_cert_subject_;
    tsi_tls_version tls_version_;
    bool send_client_ca_list_;
    std::optional<std::string> alpn_client_overriden_protocols_ = std::nullopt;
    std::optional<std::string> alpn_server_overriden_protocols_ = std::nullopt;
    std::optional<std::string> expected_alpn_negotiated_protocol_ =
        std::nullopt;
    tsi_ssl_server_handshaker_factory* server_handshaker_factory_ = nullptr;
    tsi_ssl_client_handshaker_factory* client_handshaker_factory_ = nullptr;
  };

  SslTransportSecurityTest() { grpc_init(); }

  ~SslTransportSecurityTest() override {
    DestroyFixture();
    grpc_shutdown();
  }

  void SetUpSslFixture(tsi_tls_version tls_version, bool send_client_ca_list) {
    ssl_fixture_ = new SslTsiTestFixture(tls_version, send_client_ca_list);
    ssl_tsi_test_fixture_ = ssl_fixture_->GetBaseFixture();
    fixture_destroyed = false;
  }

  void DoHandshake() { tsi_test_do_handshake(ssl_tsi_test_fixture_); }

  void DoRoundTrip() { tsi_test_do_round_trip(ssl_tsi_test_fixture_); }

  void DestroyFixture() {
    if (fixture_destroyed) {
      return;
    }
    tsi_test_fixture_destroy(ssl_tsi_test_fixture_);
    fixture_destroyed = true;
  }

  tsi_test_fixture* ssl_tsi_test_fixture_;
  SslTsiTestFixture* ssl_fixture_;
  bool fixture_destroyed = true;
};

tsi_test_fixture_vtable SslTransportSecurityTest::SslTsiTestFixture::kVtable = {
    SslTransportSecurityTest::SslTsiTestFixture::SetupHandshakers,
    SslTransportSecurityTest::SslTsiTestFixture::CheckHandshakerPeers,
    SslTransportSecurityTest::SslTsiTestFixture::Destruct};

INSTANTIATE_TEST_SUITE_P(
    SslTestGroup, SslTransportSecurityTest,
    testing::Combine(testing::Values(TSI_TLS1_2, TSI_TLS1_3),
                     testing::Values(true, false)),
    [](const testing::TestParamInfo<SslTransportSecurityTest::ParamType>&
           info) {
      return absl::StrCat(
          std::get<0>(info.param) == TSI_TLS1_2 ? "tls_12" : "tls_13", "_",
          std::get<1>(info.param) ? "send_client_ca_list"
                                  : "no_send_client_ca_list");
    });

TEST_P(SslTransportSecurityTest, DoHandshakeTinyHandshakeBuffer) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_tiny_handshake_buffer";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_tsi_test_fixture_->handshake_buffer_size =
      TSI_TEST_TINY_HANDSHAKE_BUFFER_SIZE;
  // Handshake buffer is too small to hold both handshake messages and the
  // unused bytes.
  ssl_tsi_test_fixture_->test_unused_bytes = false;
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeSmallHandshakeBuffer) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_small_handshake_buffer";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_tsi_test_fixture_->handshake_buffer_size =
      TSI_TEST_SMALL_HANDSHAKE_BUFFER_SIZE;
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshake) {
  LOG(INFO) << "ssl_tsi_test_do_handshake";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithRootStore) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_root_store";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->MutableKeyCertLib()->use_root_store = true;
  DoHandshake();
}

// TODO(gregorycooke) - failing with OpenSSL1.0.2
#if OPENSSL_VERSION_NUMBER >= 0x10100000
TEST_P(SslTransportSecurityTest,
       DoHandshakeSkippingServerCertificateVerification) {
  LOG(INFO)
      << "ssl_tsi_test_do_handshake_skipping_server_certificate_verification";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetVerifyRootCertSubject(false);
  ssl_key_cert_lib* key_cert_lib = ssl_fixture_->MutableKeyCertLib();
  key_cert_lib->use_root_store = false;
  key_cert_lib->use_pem_root_certs = false;
  key_cert_lib->skip_server_certificate_verification = true;
  DoHandshake();
}
#endif

TEST_P(SslTransportSecurityTest, DoHandshakeWithLargeServerHandshakeMessages) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_large_server_handshake_messages";
  std::string trust_bundle = GenerateTrustBundle();
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // Force the test to read more handshake bytes from the peer than we have room
  // for in the BIO. The default BIO buffer size is 17kB.
  ssl_tsi_test_fixture_->handshake_buffer_size = 18000;
  // Make a copy of the root cert and free the original.
  ssl_key_cert_lib* key_cert_lib = ssl_fixture_->MutableKeyCertLib();
  std::string root_cert(key_cert_lib->root_cert);
  gpr_free(key_cert_lib->root_cert);
  key_cert_lib->root_cert = nullptr;
  // Create a new root store, consisting of the root cert that is actually
  // needed and 200 self-signed certs.
  std::string effective_trust_bundle = absl::StrCat(root_cert, trust_bundle);
  tsi_ssl_root_certs_store_destroy(key_cert_lib->root_store);
  key_cert_lib->root_cert = const_cast<char*>(effective_trust_bundle.c_str());
  key_cert_lib->root_store =
      tsi_ssl_root_certs_store_create(effective_trust_bundle.c_str());
  key_cert_lib->use_root_store = true;
  ssl_fixture_->SetForceClientAuth(true);
  DoHandshake();
  // Overwrite the root_cert pointer so that tsi_test_fixture_destroy does not
  // try to gpr_free it.
  key_cert_lib->root_cert = nullptr;
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithClientAuthentication) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_client_authentication";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetForceClientAuth(true);
  DoHandshake();
}

TEST_P(SslTransportSecurityTest,
       DoHandshakeWithClientAuthenticationAndRootStore) {
  LOG(INFO)
      << "ssl_tsi_test_do_handshake_with_client_authentication_and_root_store";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetForceClientAuth(true);
  ssl_fixture_->MutableKeyCertLib()->use_root_store = true;
  DoHandshake();
}

TEST_P(SslTransportSecurityTest,
       DoHandshakeWithServerNameIndicationExactDomain) {
  LOG(INFO)
      << "ssl_tsi_test_do_handshake_with_server_name_indication_exact_domain";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // server1 cert contains "waterzooi.test.google.be" in SAN.
  ssl_fixture_->SetServerNameIndication(
      const_cast<char*>("waterzooi.test.google.be"));
  DoHandshake();
}

TEST_P(SslTransportSecurityTest,
       DoHandshakeWithServerNameIndicationWildStarDomain) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_server_name_indication_wild_"
               "star_domain";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // server1 cert contains "*.test.google.fr" in SAN.
  ssl_fixture_->SetServerNameIndication(
      const_cast<char*>("juju.test.google.fr"));
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithWrongServerNameIndication) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_wrong_server_name_indication";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // server certs do not contain "test.google.cn".
  ssl_fixture_->SetServerNameIndication(
      const_cast<char*>(SSL_TSI_TEST_WRONG_SNI));
  DoHandshake();
}

TEST_P(SslTransportSecurityTest,
       DoHandshakeWithInvalidAndIgnoredServerNameIndication) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_invalid_and_ignored_server_name_"
               "indication";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // SNI that's an IP address will be ignored.
  ssl_fixture_->SetServerNameIndication(
      const_cast<char*>(SSL_TSI_TEST_INVALID_SNI));
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithBadServerCert) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_bad_server_cert";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->MutableKeyCertLib()->use_bad_server_cert = true;
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithBadClientCert) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_bad_client_cert";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->MutableKeyCertLib()->use_bad_client_cert = true;
  ssl_fixture_->SetForceClientAuth(true);
  DoHandshake();
}

#ifdef OPENSSL_IS_BORINGSSL
// BoringSSL and OpenSSL have different behaviors on mismatched ALPN.
TEST_P(SslTransportSecurityTest, DoHandshakeAlpnClientNoServer) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_alpn_client_no_server";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_NO_SERVER);
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeAlpnClientServerMismatch) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_alpn_client_server_mismatch";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_SERVER_MISMATCH);
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoRoundTripForAllConfigs) {
  LOG(INFO) << "ssl_tsi_test_do_round_trip_for_all_configs";
  unsigned int* bit_array = static_cast<unsigned int*>(
      gpr_zalloc(sizeof(unsigned int) * TSI_TEST_NUM_OF_ARGUMENTS));
  const unsigned int mask = 1U << (TSI_TEST_NUM_OF_ARGUMENTS - 1);
  for (unsigned int val = 0; val < TSI_TEST_NUM_OF_COMBINATIONS; val++) {
    unsigned int v = val;
    for (unsigned int ind = 0; ind < TSI_TEST_NUM_OF_ARGUMENTS; ind++) {
      bit_array[ind] = (v & mask) ? 1 : 0;
      v <<= 1;
    }
    SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                    /*send_client_ca_list=*/std::get<1>(GetParam()));
    tsi_test_frame_protector_config_destroy(ssl_tsi_test_fixture_->config);
    ssl_tsi_test_fixture_->config = tsi_test_frame_protector_config_create(
        bit_array[0], bit_array[1], bit_array[2], bit_array[3], bit_array[4],
        bit_array[5], bit_array[6]);
    DoRoundTrip();
    DestroyFixture();
  }
  gpr_free(bit_array);
}

TEST_P(SslTransportSecurityTest, DoRoundTripWithErrorOnStack) {
  LOG(INFO) << "ssl_tsi_test_do_round_trip_with_error_on_stack";
  // Invoke an SSL function that causes an error, and ensure the error
  // makes it to the stack.
  ASSERT_FALSE(EC_KEY_new_by_curve_name(NID_rsa));
  ASSERT_NE(ERR_peek_error(), 0);
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  DoRoundTrip();
}

TEST_P(SslTransportSecurityTest, DoRoundTripOddBufferSize) {
  LOG(INFO) << "ssl_tsi_test_do_round_trip_odd_buffer_size";
  const size_t odd_sizes[] = {1025, 2051, 4103, 8207, 16409};
  size_t size = sizeof(odd_sizes) / sizeof(size_t);
  // 1. This test is extremely slow under MSAN and TSAN.
  // 2. On 32-bit, the test is much slower (probably due to lack of boringssl
  // asm optimizations) so we only run a subset of tests to avoid timeout.
  // 3. On Mac OS, we have slower testing machines so we only run a subset
  // of tests to avoid timeout.
  if (is_slow_build()) {
    size = 1;
  }
  for (size_t ind1 = 0; ind1 < size; ind1++) {
    for (size_t ind2 = 0; ind2 < size; ind2++) {
      for (size_t ind3 = 0; ind3 < size; ind3++) {
        for (size_t ind4 = 0; ind4 < size; ind4++) {
          for (size_t ind5 = 0; ind5 < size; ind5++) {
            SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                            /*send_client_ca_list=*/std::get<1>(GetParam()));
            tsi_test_frame_protector_config_set_buffer_size(
                ssl_fixture_->GetBaseFixture()->config, odd_sizes[ind1],
                odd_sizes[ind2], odd_sizes[ind3], odd_sizes[ind4],
                odd_sizes[ind5]);
            DoRoundTrip();
            DestroyFixture();
          }
        }
      }
    }
  }
}

TEST_P(SslTransportSecurityTest, DoHandshakeSessionCache) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_session_cache";
  tsi_ssl_session_cache* session_cache = tsi_ssl_session_cache_create_lru(16);
  char session_ticket_key[kSessionTicketEncryptionKeySize];
  auto do_handshake = [this, &session_ticket_key,
                       &session_cache](bool session_reused) {
    SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                    /*send_client_ca_list=*/std::get<1>(GetParam()));
    ssl_fixture_->SetServerNameIndication(
        const_cast<char*>("waterzooi.test.google.be"));
    ssl_fixture_->SetSessionTicketKey(session_ticket_key,
                                      sizeof(session_ticket_key));
    tsi_ssl_session_cache_ref(session_cache);
    ssl_fixture_->SetSessionCache(session_cache);
    ssl_fixture_->SetSessionReused(session_reused);
    DoRoundTrip();
    DestroyFixture();
  };
  memset(session_ticket_key, 'a', sizeof(session_ticket_key));
  do_handshake(false);
  do_handshake(true);
  do_handshake(true);
  // Changing session_ticket_key on server invalidates ticket.
  memset(session_ticket_key, 'b', sizeof(session_ticket_key));
  do_handshake(false);
  do_handshake(true);
  memset(session_ticket_key, 'c', sizeof(session_ticket_key));
  do_handshake(false);
  do_handshake(true);
  tsi_ssl_session_cache_unref(session_cache);
}
#endif  // OPENSSL_IS_BORINGSSL

TEST_P(SslTransportSecurityTest, DoHandshakeAlpnServerNoClient) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_alpn_server_no_client";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetAlpnMode(ALPN_SERVER_NO_CLIENT);
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeAlpnClientServerOk) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_alpn_client_server_ok";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_SERVER_OK);
  DoHandshake();
}

TEST_P(SslTransportSecurityTest, DoHandshakeWithCustomBioPair) {
  LOG(INFO) << "ssl_tsi_test_do_handshake_with_custom_bio_pair";
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
#if OPENSSL_VERSION_NUMBER >= 0x10100000
  ssl_fixture_->SetBioBufSizes(
      /*network_bio_buf_size=*/TSI_TEST_DEFAULT_BUFFER_SIZE,
      /*ssl_bio_buf_size=*/256);
#endif
  ssl_fixture_->SetForceClientAuth(true);
  DoHandshake();
}

// TODO(matthewstevenson88): Make tests below compatible with OpenSSL.
#if defined(OPENSSL_IS_BORINGSSL)
TEST_P(SslTransportSecurityTest, Protect) {
  LOG(INFO) << "ssl_tsi_test_protect";
  SetUpSslFixture(tsi_tls_version::TSI_TLS1_3, /*send_client_ca_list=*/false);
  DoHandshake();
  tsi_frame_protector* protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->client_result,
                /*max_output_protected_frame_size=*/nullptr, &protector),
            TSI_OK);
  ASSERT_NE(protector, nullptr);
  std::string buffer(1024, 'a');
  std::string protected_bytes = Protect(protector, buffer);
  EXPECT_EQ(protected_bytes.size(), buffer.size() + kTls13FrameOverhead);
  tsi_frame_protector_destroy(protector);
}

TEST_P(SslTransportSecurityTest, ProtectAndUnprotect) {
  LOG(INFO) << "ssl_tsi_test_protect_and_unprotect";
  SetUpSslFixture(tsi_tls_version::TSI_TLS1_3, /*send_client_ca_list=*/false);
  DoHandshake();
  tsi_frame_protector* client_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->client_result,
                /*max_output_protected_frame_size=*/nullptr, &client_protector),
            TSI_OK);
  ASSERT_NE(client_protector, nullptr);
  tsi_frame_protector* server_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->server_result,
                /*max_output_protected_frame_size=*/nullptr, &server_protector),
            TSI_OK);
  ASSERT_NE(server_protector, nullptr);
  std::string buffer(1024, 'a');
  std::string protected_bytes = Protect(client_protector, buffer);
  std::string unprotected_bytes = Unprotect(server_protector, protected_bytes);
  EXPECT_EQ(unprotected_bytes, buffer);
  tsi_frame_protector_destroy(client_protector);
  tsi_frame_protector_destroy(server_protector);
}

TEST_P(SslTransportSecurityTest, ConcurrentlyProtectAndUnprotectOnClient) {
  LOG(INFO) << "ssl_tsi_test_protect_and_unprotect";
  SetUpSslFixture(tsi_tls_version::TSI_TLS1_3, /*send_client_ca_list=*/false);
  DoHandshake();
  tsi_frame_protector* client_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->client_result,
                /*max_output_protected_frame_size=*/nullptr, &client_protector),
            TSI_OK);
  ASSERT_NE(client_protector, nullptr);
  tsi_frame_protector* server_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->server_result,
                /*max_output_protected_frame_size=*/nullptr, &server_protector),
            TSI_OK);
  ASSERT_NE(server_protector, nullptr);
  std::string buffer(1024, 'a');
  std::string protected_bytes = Protect(server_protector, buffer);
  std::thread protect_thread([&client_protector]() {
    std::string second_buffer(2048, 'b');
    std::string second_protected_buffer =
        Protect(client_protector, second_buffer);
    EXPECT_EQ(second_protected_buffer.size(),
              second_buffer.size() + kTls13FrameOverhead);
  });
  std::thread unprotect_thread([&client_protector, protected_bytes, buffer]() {
    std::string unprotected_bytes =
        Unprotect(client_protector, protected_bytes);
    EXPECT_EQ(unprotected_bytes, buffer);
  });
  protect_thread.join();
  unprotect_thread.join();
  tsi_frame_protector_destroy(client_protector);
  tsi_frame_protector_destroy(server_protector);
}

TEST_P(SslTransportSecurityTest, ConcurrentlyProtectAndUnprotectOnServer) {
  LOG(INFO) << "ssl_tsi_test_protect_and_unprotect";
  SetUpSslFixture(tsi_tls_version::TSI_TLS1_3, /*send_client_ca_list=*/false);
  DoHandshake();
  tsi_frame_protector* client_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->client_result,
                /*max_output_protected_frame_size=*/nullptr, &client_protector),
            TSI_OK);
  ASSERT_NE(client_protector, nullptr);
  tsi_frame_protector* server_protector;
  EXPECT_EQ(tsi_handshaker_result_create_frame_protector(
                ssl_tsi_test_fixture_->server_result,
                /*max_output_protected_frame_size=*/nullptr, &server_protector),
            TSI_OK);
  ASSERT_NE(server_protector, nullptr);
  std::string buffer(1024, 'a');
  std::string protected_bytes = Protect(client_protector, buffer);
  std::thread protect_thread([&server_protector]() {
    std::string second_buffer(20, 'b');
    std::string second_protected_buffer =
        Protect(server_protector, second_buffer);
    // The protected bytes may be prefixed by a TLS record containing session
    // tickets, so the total number of protected bytes is greater or equal to
    // the expected bytes, based on the length of the plaintext.
    EXPECT_GE(second_protected_buffer.size(),
              second_buffer.size() + kTls13FrameOverhead);
  });
  std::thread unprotect_thread([&server_protector, protected_bytes, buffer]() {
    std::string unprotected_bytes =
        Unprotect(server_protector, protected_bytes);
    EXPECT_GE(unprotected_bytes, buffer);
  });
  protect_thread.join();
  unprotect_thread.join();
  tsi_frame_protector_destroy(client_protector);
  tsi_frame_protector_destroy(server_protector);
}
#endif  // defined(OPENSSL_IS_BORINGSSL)

static const tsi_ssl_handshaker_factory_vtable* original_vtable;
static bool handshaker_factory_destructor_called;

static void ssl_tsi_test_handshaker_factory_destructor(
    tsi_ssl_handshaker_factory* factory) {
  ASSERT_NE(factory, nullptr);
  handshaker_factory_destructor_called = true;
  if (original_vtable != nullptr && original_vtable->destroy != nullptr) {
    original_vtable->destroy(factory);
  }
}

static tsi_ssl_handshaker_factory_vtable test_handshaker_factory_vtable = {
    ssl_tsi_test_handshaker_factory_destructor};

TEST(SslTransportSecurityTest, TestClientHandshakerFactoryRefcounting) {
  int i;
  char* cert_chain = load_file(SSL_TSI_TEST_CREDENTIALS_DIR "client.pem");

  tsi_ssl_client_handshaker_options options;
  if (cert_chain != nullptr) {
    options.root_cert_info = std::make_shared<tsi::RootCertInfo>(cert_chain);
  }
  tsi_ssl_client_handshaker_factory* client_handshaker_factory;
  ASSERT_EQ(tsi_create_ssl_client_handshaker_factory_with_options(
                &options, &client_handshaker_factory),
            TSI_OK);

  handshaker_factory_destructor_called = false;
  original_vtable = tsi_ssl_handshaker_factory_swap_vtable(
      reinterpret_cast<tsi_ssl_handshaker_factory*>(client_handshaker_factory),
      &test_handshaker_factory_vtable);

  tsi_handshaker* handshaker[3];

  for (i = 0; i < 3; ++i) {
    ASSERT_EQ(
        tsi_ssl_client_handshaker_factory_create_handshaker(
            client_handshaker_factory, "google.com", 0, 0,
            /*alpn_preferred_protocol_list=*/std::nullopt, &handshaker[i]),
        TSI_OK);
  }

  client_handshaker_factory =
      tsi_ssl_client_handshaker_factory_ref(client_handshaker_factory);

  tsi_handshaker_destroy(handshaker[1]);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_handshaker_destroy(handshaker[0]);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_ssl_client_handshaker_factory_unref(client_handshaker_factory);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_handshaker_destroy(handshaker[2]);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_ssl_client_handshaker_factory_unref(client_handshaker_factory);
  ASSERT_TRUE(handshaker_factory_destructor_called);
  gpr_free(cert_chain);
}

TEST(SslTransportSecurityTest, TestServerHandshakerFactoryRefcounting) {
  int i;
  tsi_ssl_server_handshaker_factory* server_handshaker_factory;
  tsi_handshaker* handshaker[3];
  std::string cert_chain = grpc_core::testing::GetFileContents(
      SSL_TSI_TEST_CREDENTIALS_DIR "server0.pem");
  tsi_ssl_pem_key_cert_pair cert_pair;

  cert_pair.cert_chain = cert_chain;
  cert_pair.private_key = grpc_core::testing::GetFileContents(
      SSL_TSI_TEST_CREDENTIALS_DIR "server0.key");
  tsi_ssl_server_handshaker_options options;
  options.pem_key_cert_pairs = {cert_pair};
  if (!cert_chain.empty()) {
    options.root_cert_info = std::make_shared<tsi::RootCertInfo>(cert_chain);
  }

  ASSERT_EQ(tsi_create_ssl_server_handshaker_factory_with_options(
                &options, &server_handshaker_factory),
            TSI_OK);

  handshaker_factory_destructor_called = false;
  original_vtable = tsi_ssl_handshaker_factory_swap_vtable(
      reinterpret_cast<tsi_ssl_handshaker_factory*>(server_handshaker_factory),
      &test_handshaker_factory_vtable);

  for (i = 0; i < 3; ++i) {
    ASSERT_EQ(tsi_ssl_server_handshaker_factory_create_handshaker(
                  server_handshaker_factory, 0, 0, &handshaker[i]),
              TSI_OK);
  }

  tsi_handshaker_destroy(handshaker[1]);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_handshaker_destroy(handshaker[0]);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_ssl_server_handshaker_factory_unref(server_handshaker_factory);
  ASSERT_FALSE(handshaker_factory_destructor_called);

  tsi_handshaker_destroy(handshaker[2]);
  ASSERT_TRUE(handshaker_factory_destructor_called);
}

// Attempting to create a handshaker factory with invalid parameters should fail
// but not crash.
TEST(SslTransportSecurityTest, TestClientHandshakerFactoryBadParams) {
  const char* cert_chain = "This is not a valid PEM file.";

  tsi_ssl_client_handshaker_factory* client_handshaker_factory;
  tsi_ssl_client_handshaker_options options;
  if (cert_chain != nullptr) {
    options.root_cert_info = std::make_shared<tsi::RootCertInfo>(cert_chain);
  }
  ASSERT_EQ(tsi_create_ssl_client_handshaker_factory_with_options(
                &options, &client_handshaker_factory),
            TSI_INVALID_ARGUMENT);
  tsi_ssl_client_handshaker_factory_unref(client_handshaker_factory);
}

TEST(SslTransportSecurityTest, DuplicateRootCertificates) {
  LOG(INFO) << "ssl_tsi_test_duplicate_root_certificates";
  char* root_cert = load_file(SSL_TSI_TEST_CREDENTIALS_DIR "ca.pem");
  char* dup_root_cert = static_cast<char*>(
      gpr_zalloc(sizeof(char) * (strlen(root_cert) * 2 + 1)));
  memcpy(dup_root_cert, root_cert, strlen(root_cert));
  memcpy(dup_root_cert + strlen(root_cert), root_cert, strlen(root_cert));
  tsi_ssl_root_certs_store* root_store =
      tsi_ssl_root_certs_store_create(dup_root_cert);
  ASSERT_NE(root_store, nullptr);
  // Free memory.
  tsi_ssl_root_certs_store_destroy(root_store);
  gpr_free(root_cert);
  gpr_free(dup_root_cert);
}

TEST(SslTransportSecurityTest, ExtractX509SubjectNames) {
  LOG(INFO) << "ssl_tsi_test_extract_x509_subject_names";
  char* cert = load_file(SSL_TSI_TEST_CREDENTIALS_DIR "multi-domain.pem");
  tsi_peer peer;
  ASSERT_EQ(tsi_ssl_extract_x509_subject_names_from_pem_cert(cert, &peer),
            TSI_OK);
  // tsi_peer should include one subject, one common name, one certificate, one
  // security level, ten SAN fields, two DNS SAN fields, three URI fields, two
  // email addresses and two IP addresses.
  size_t expected_property_count = 22;
  ASSERT_EQ(peer.property_count, expected_property_count);
  // Check subject
  const char* expected_subject = "CN=xpigors,OU=Google,L=SF,ST=CA,C=US";
  const tsi_peer_property* property =
      tsi_peer_get_property_by_name(&peer, TSI_X509_SUBJECT_PEER_PROPERTY);
  ASSERT_NE(property, nullptr);
  ASSERT_EQ(
      memcmp(property->value.data, expected_subject, property->value.length),
      0);
  // Check common name
  const char* expected_cn = "xpigors";
  property = tsi_peer_get_property_by_name(
      &peer, TSI_X509_SUBJECT_COMMON_NAME_PEER_PROPERTY);
  ASSERT_NE(property, nullptr);
  ASSERT_EQ(memcmp(property->value.data, expected_cn, property->value.length),
            0);
  // Check certificate data
  property = tsi_peer_get_property_by_name(&peer, TSI_X509_PEM_CERT_PROPERTY);
  ASSERT_NE(property, nullptr);
  ASSERT_EQ(memcmp(property->value.data, cert, property->value.length), 0);
  // Check DNS
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "foo.test.domain.com"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "bar.test.domain.com"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_DNS_PEER_PROPERTY, "foo.test.domain.com"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_DNS_PEER_PROPERTY, "bar.test.domain.com"),
      1);
  // Check URI
  // Note that a valid SPIFFE certificate should only have one URI.
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "spiffe://foo.com/bar/baz"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "https://foo.test.domain.com/test"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "https://bar.test.domain.com/test"),
      1);
  ASSERT_EQ(check_property(&peer, TSI_X509_URI_PEER_PROPERTY,
                           "spiffe://foo.com/bar/baz"),
            1);
  ASSERT_EQ(check_property(&peer, TSI_X509_URI_PEER_PROPERTY,
                           "https://foo.test.domain.com/test"),
            1);
  ASSERT_EQ(check_property(&peer, TSI_X509_URI_PEER_PROPERTY,
                           "https://bar.test.domain.com/test"),
            1);
  // Check email address
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "foo@test.domain.com"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "bar@test.domain.com"),
      1);
  ASSERT_EQ(check_property(&peer, TSI_X509_EMAIL_PEER_PROPERTY,
                           "foo@test.domain.com"),
            1);
  ASSERT_EQ(check_property(&peer, TSI_X509_EMAIL_PEER_PROPERTY,
                           "bar@test.domain.com"),
            1);
  // Check ip address
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "192.168.7.1"),
      1);
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "13::17"),
      1);
  ASSERT_EQ(check_property(&peer, TSI_X509_IP_PEER_PROPERTY, "192.168.7.1"), 1);
  ASSERT_EQ(check_property(&peer, TSI_X509_IP_PEER_PROPERTY, "13::17"), 1);
  // Check other fields
  ASSERT_EQ(
      check_property(&peer, TSI_X509_SUBJECT_ALTERNATIVE_NAME_PEER_PROPERTY,
                     "other types of SAN"),
      1);
  // Free memory
  gpr_free(cert);
  tsi_peer_destruct(&peer);
}

TEST(SslTransportSecurityTest, ExtractCertChain) {
  LOG(INFO) << "ssl_tsi_test_extract_cert_chain";
  char* cert = load_file(SSL_TSI_TEST_CREDENTIALS_DIR "server1.pem");
  char* ca = load_file(SSL_TSI_TEST_CREDENTIALS_DIR "ca.pem");
  char* chain = static_cast<char*>(
      gpr_zalloc(sizeof(char) * (strlen(cert) + strlen(ca) + 1)));
  memcpy(chain, cert, strlen(cert));
  memcpy(chain + strlen(cert), ca, strlen(ca));
  STACK_OF(X509)* cert_chain = sk_X509_new_null();
  ASSERT_NE(cert_chain, nullptr);
  BIO* bio = BIO_new_mem_buf(chain, strlen(chain));
  ASSERT_NE(bio, nullptr);
  STACK_OF(X509_INFO)* certInfos =
      PEM_X509_INFO_read_bio(bio, nullptr, nullptr, nullptr);
  ASSERT_NE(certInfos, nullptr);
  for (size_t i = 0; i < sk_X509_INFO_num(certInfos); i++) {
    X509_INFO* certInfo = sk_X509_INFO_value(certInfos, i);
    if (certInfo->x509 != nullptr) {
      ASSERT_NE(sk_X509_push(cert_chain, certInfo->x509), 0);
#if OPENSSL_VERSION_NUMBER >= 0x10100000
      X509_up_ref(certInfo->x509);
#else
      certInfo->x509->references += 1;
#endif
    }
  }
  tsi_peer_property chain_property;
  ASSERT_EQ(tsi_ssl_get_cert_chain_contents(cert_chain, &chain_property),
            TSI_OK);
  ASSERT_EQ(
      memcmp(chain, chain_property.value.data, chain_property.value.length), 0);
  BIO_free(bio);
  gpr_free(chain);
  gpr_free(cert);
  gpr_free(ca);
  tsi_peer_property_destruct(&chain_property);
  sk_X509_INFO_pop_free(certInfos, X509_INFO_free);
  sk_X509_pop_free(cert_chain, X509_free);
}

// Attempt to perform a handshake between a client and server with ALPN enabled
// and overriding the preferred protocols on the handshaker factory.
TEST_P(SslTransportSecurityTest, TestClientAndServerHandshakerOverrideALPN) {
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->OverrideHanshakerAlpnClientProtocols("toto,bar");
  ssl_fixture_->OverrideHanshakerAlpnServerProtocols("bar,foo,toto");
  ssl_fixture_->SetExpectedAlpnNegotiatedProtocol("bar");
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_SERVER_OK);
  DoHandshake();
}

// Attempt to perform a handshake between a client and server with ALPN enabled
// and overriding the preferred protocols on the handshaker factory.
TEST_P(SslTransportSecurityTest, TestClientHandshakerOverrideALPN) {
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  ssl_fixture_->OverrideHanshakerAlpnClientProtocols("toto,baz");
  // The server default protocols set by the handshaker factory are [foo, baz].
  ssl_fixture_->SetExpectedAlpnNegotiatedProtocol("baz");
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_SERVER_OK);
  DoHandshake();
}

// Attempt to perform a handshake between a client and server with ALPN enabled
// and overriding the preferred protocols on the handshaker factory.
TEST_P(SslTransportSecurityTest, TestServerHandshakerOverrideALPN) {
  SetUpSslFixture(/*tls_version=*/std::get<0>(GetParam()),
                  /*send_client_ca_list=*/std::get<1>(GetParam()));
  // The client default protocols set by the handshaker factory are [toto, baz].
  ssl_fixture_->OverrideHanshakerAlpnServerProtocols("bar,foo,toto");
  ssl_fixture_->SetExpectedAlpnNegotiatedProtocol("toto");
  ssl_fixture_->SetAlpnMode(ALPN_CLIENT_SERVER_OK);
  DoHandshake();
}

int main(int argc, char** argv) {
  grpc::testing::TestEnvironment env(&argc, argv);
  ::testing::InitGoogleTest(&argc, argv);
  grpc::testing::TestGrpcScope grpc_scope;
  return RUN_ALL_TESTS();
}
