blob: 1c3ec136b1d015d7b76bac96fedaae6e3f155dbc [file] [log] [blame] [edit]
// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "quiche/quic/tools/quic_toy_server.h"
#include <memory>
#include <optional>
#include <string>
#include <utility>
#include <vector>
#include "absl/container/flat_hash_set.h"
#include "absl/strings/str_cat.h"
#include "absl/strings/str_split.h"
#include "absl/strings/string_view.h"
#include "quiche/quic/core/quic_server_id.h"
#include "quiche/quic/core/quic_versions.h"
#include "quiche/quic/platform/api/quic_default_proof_providers.h"
#include "quiche/quic/platform/api/quic_socket_address.h"
#include "quiche/quic/tools/connect_server_backend.h"
#include "quiche/quic/tools/quic_memory_cache_backend.h"
#include "quiche/common/platform/api/quiche_command_line_flags.h"
#include "quiche/common/platform/api/quiche_logging.h"
#include "quiche/common/quiche_random.h"
DEFINE_QUICHE_COMMAND_LINE_FLAG(int32_t, port, 6121,
"The port the quic server will listen on.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
std::string, quic_response_cache_dir, "",
"Specifies the directory used during QuicHttpResponseCache "
"construction to seed the cache. Cache directory can be "
"generated using `wget -p --save-headers <url>`");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
bool, generate_dynamic_responses, false,
"If true, then URLs which have a numeric path will send a dynamically "
"generated response of that many bytes.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
std::string, quic_versions, "",
"QUIC versions to enable, e.g. \"h3-25,h3-27\". If not set, then all "
"available versions are enabled.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(bool, enable_webtransport, false,
"If true, WebTransport support is enabled.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
std::string, connect_proxy_destinations, "",
"Specifies a comma-separated list of destinations (\"hostname:port\") to "
"which the QUIC server will allow tunneling via CONNECT.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
std::string, connect_udp_proxy_targets, "",
"Specifies a comma-separated list of target servers (\"hostname:port\") to "
"which the QUIC server will allow tunneling via CONNECT-UDP.");
DEFINE_QUICHE_COMMAND_LINE_FLAG(
std::string, proxy_server_label, "",
"Specifies an identifier to identify the server in proxy error headers, "
"per the requirements of RFC 9209, Section 2. It should uniquely identify "
"the running service between separate running instances of the QUIC toy "
"server binary. If not specified, one will be randomly generated as "
"\"QuicToyServerN\" where N is a random uint64_t.");
namespace quic {
std::unique_ptr<quic::QuicSimpleServerBackend>
QuicToyServer::MemoryCacheBackendFactory::CreateBackend() {
auto memory_cache_backend = std::make_unique<QuicMemoryCacheBackend>();
if (quiche::GetQuicheCommandLineFlag(FLAGS_generate_dynamic_responses)) {
memory_cache_backend->GenerateDynamicResponses();
}
if (!quiche::GetQuicheCommandLineFlag(FLAGS_quic_response_cache_dir)
.empty()) {
memory_cache_backend->InitializeBackend(
quiche::GetQuicheCommandLineFlag(FLAGS_quic_response_cache_dir));
}
if (quiche::GetQuicheCommandLineFlag(FLAGS_enable_webtransport)) {
memory_cache_backend->EnableWebTransport();
}
if (!quiche::GetQuicheCommandLineFlag(FLAGS_connect_proxy_destinations)
.empty() ||
!quiche::GetQuicheCommandLineFlag(FLAGS_connect_udp_proxy_targets)
.empty()) {
absl::flat_hash_set<QuicServerId> connect_proxy_destinations;
for (absl::string_view destination : absl::StrSplit(
quiche::GetQuicheCommandLineFlag(FLAGS_connect_proxy_destinations),
',', absl::SkipEmpty())) {
std::optional<QuicServerId> destination_server_id =
QuicServerId::ParseFromHostPortString(destination);
QUICHE_CHECK(destination_server_id.has_value());
connect_proxy_destinations.insert(*std::move(destination_server_id));
}
absl::flat_hash_set<QuicServerId> connect_udp_proxy_targets;
for (absl::string_view target : absl::StrSplit(
quiche::GetQuicheCommandLineFlag(FLAGS_connect_udp_proxy_targets),
',', absl::SkipEmpty())) {
std::optional<QuicServerId> target_server_id =
QuicServerId::ParseFromHostPortString(target);
QUICHE_CHECK(target_server_id.has_value());
connect_udp_proxy_targets.insert(*std::move(target_server_id));
}
QUICHE_CHECK(!connect_proxy_destinations.empty() ||
!connect_udp_proxy_targets.empty());
std::string proxy_server_label =
quiche::GetQuicheCommandLineFlag(FLAGS_proxy_server_label);
if (proxy_server_label.empty()) {
proxy_server_label = absl::StrCat(
"QuicToyServer",
quiche::QuicheRandom::GetInstance()->InsecureRandUint64());
}
return std::make_unique<ConnectServerBackend>(
std::move(memory_cache_backend), std::move(connect_proxy_destinations),
std::move(connect_udp_proxy_targets), std::move(proxy_server_label));
}
return memory_cache_backend;
}
QuicToyServer::QuicToyServer(BackendFactory* backend_factory,
ServerFactory* server_factory)
: backend_factory_(backend_factory), server_factory_(server_factory) {}
int QuicToyServer::Start() {
ParsedQuicVersionVector supported_versions = AllSupportedVersions();
std::string versions_string =
quiche::GetQuicheCommandLineFlag(FLAGS_quic_versions);
if (!versions_string.empty()) {
supported_versions = ParseQuicVersionVectorString(versions_string);
}
if (supported_versions.empty()) {
return 1;
}
for (const auto& version : supported_versions) {
QuicEnableVersion(version);
}
auto proof_source = quic::CreateDefaultProofSource();
auto backend = backend_factory_->CreateBackend();
auto server = server_factory_->CreateServer(
backend.get(), std::move(proof_source), supported_versions);
if (!server->CreateUDPSocketAndListen(quic::QuicSocketAddress(
quic::QuicIpAddress::Any6(),
quiche::GetQuicheCommandLineFlag(FLAGS_port)))) {
return 1;
}
server->HandleEventsForever();
return 0;
}
} // namespace quic