| // 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 |