| // 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 "quic/tools/quic_toy_server.h" |
| |
| #include <utility> |
| #include <vector> |
| |
| #include "quic/core/quic_versions.h" |
| #include "quic/platform/api/quic_default_proof_providers.h" |
| #include "quic/platform/api/quic_flags.h" |
| #include "quic/platform/api/quic_socket_address.h" |
| #include "quic/tools/quic_memory_cache_backend.h" |
| |
| DEFINE_QUIC_COMMAND_LINE_FLAG(int32_t, |
| port, |
| 6121, |
| "The port the quic server will listen on."); |
| |
| DEFINE_QUIC_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_QUIC_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_QUIC_COMMAND_LINE_FLAG(bool, |
| quic_ietf_draft, |
| false, |
| "Only enable IETF draft versions. This also " |
| "enables required internal QUIC flags."); |
| |
| DEFINE_QUIC_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."); |
| |
| namespace quic { |
| |
| std::unique_ptr<quic::QuicSimpleServerBackend> |
| QuicToyServer::MemoryCacheBackendFactory::CreateBackend() { |
| auto memory_cache_backend = std::make_unique<QuicMemoryCacheBackend>(); |
| if (GetQuicFlag(FLAGS_generate_dynamic_responses)) { |
| memory_cache_backend->GenerateDynamicResponses(); |
| } |
| if (!GetQuicFlag(FLAGS_quic_response_cache_dir).empty()) { |
| memory_cache_backend->InitializeBackend( |
| GetQuicFlag(FLAGS_quic_response_cache_dir)); |
| } |
| 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; |
| if (GetQuicFlag(FLAGS_quic_ietf_draft)) { |
| QuicVersionInitializeSupportForIetfDraft(); |
| for (const ParsedQuicVersion& version : AllSupportedVersions()) { |
| // Add all versions that supports IETF QUIC. |
| if (version.HasIetfQuicFrames() && |
| version.handshake_protocol == quic::PROTOCOL_TLS1_3) { |
| supported_versions.push_back(version); |
| } |
| } |
| } else { |
| supported_versions = AllSupportedVersions(); |
| } |
| std::string versions_string = GetQuicFlag(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(), GetQuicFlag(FLAGS_port)))) { |
| return 1; |
| } |
| |
| server->HandleEventsForever(); |
| return 0; |
| } |
| |
| } // namespace quic |