h2_origin_mock_server.cc

Go to the documentation of this file.

/*
 * Copyright (C) Max Dymond, <cmeister2@gmail.com>, et al.
 *
 * SPDX-License-Identifier: curl
 */

 
#include "proto_fuzzer/h2_origin_mock_server.h"
 
#include <utility>
 
namespace proto_fuzzer {
 
namespace {

TlsGroupPolicy GroupPolicyFor(curl::fuzzer::proto::TlsGroupProfile profile) {
  return profile == curl::fuzzer::proto::TLS_GROUP_PROVIDER_DEFAULT ? TlsGroupPolicy::kProviderDefault
                                                                    : TlsGroupPolicy::kX25519;
}
 
}  // namespace

H2OriginMockServer::H2OriginMockServer(curl::fuzzer::proto::TlsCertificateChainProfile certificate_chain,
                                       curl::fuzzer::proto::TlsGroupProfile group_profile)
    : TlsMockServer(TlsApplicationProtocol::kHttp2, certificate_chain, GroupPolicyFor(group_profile)) {}
 
H2OriginMockServer::~H2OriginMockServer() {
  // The multi borrows runtime_ for push callbacks, and connections borrow its
  // client-frame observer. Release both while that member is still alive.
  retained_multi_.reset();
  ResetConnections();
}

void H2OriginMockServer::Install(CURL* easy) {
  TlsMockServer::Install(easy);
  if (group_policy() == TlsGroupPolicy::kX25519) {
    (void)curl_easy_setopt(easy, CURLOPT_SSL_EC_CURVES, "X25519");
  }
  (void)curl_easy_setopt(easy, CURLOPT_HTTP_VERSION, CURL_HTTP_VERSION_2TLS);
  (void)curl_easy_setopt(easy, CURLOPT_UPKEEP_INTERVAL_MS, 0L);
}

std::unique_ptr<MockConnection> H2OriginMockServer::CreateConnection() {
  std::unique_ptr<MockConnection> connection = TlsMockServer::CreateConnection();
  runtime_.AttachConnection(connection.get());
  return connection;
}

void H2OriginMockServer::RunLoop(CURLM* multi, CURL* easy, const curl::fuzzer::proto::Scenario& scenario) {
  runtime_.PrepareMulti(multi, scenario.accept_h2_push());
  if (!scenario.has_http2_plan()) {
    MockServer::RunLoop(multi, easy, scenario);
  } else {
    SetScripts(scenario);
    runtime_.ResetPlan(scenario.http2_plan(), "https");
    runtime_.DrivePlan(multi, *this, kMaxIdleIterations, kMaxDriveIterations, [this, easy] {
      ObserveActiveTransfer(easy);
      ResumeResponseIfRequested(easy);
      runtime_.ApplyPostHeadersTransition(easy);
    });
  }
  runtime_.FinishTransfer(easy, connection());
}

void H2OriginMockServer::HandleDetachedMulti(CurlMultiPtr multi) { retained_multi_ = std::move(multi); }

const H2Runtime& H2OriginMockServer::runtime() const { return runtime_; }

std::size_t H2OriginMockServer::cleaned_push_count() const { return additional_handle_cleanup_count(); }
 
}  // namespace proto_fuzzer