123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162 |
- // Copyright 2013 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 <stdio.h>
- #include <memory>
- #include "base/at_exit.h"
- #include "base/command_line.h"
- #include "base/files/file_path.h"
- #include "base/files/file_util.h"
- #include "base/logging.h"
- #include "base/message_loop/message_pump_type.h"
- #include "base/run_loop.h"
- #include "base/strings/utf_string_conversions.h"
- #include "base/task/single_thread_task_executor.h"
- #include "base/test/test_timeouts.h"
- #include "net/test/embedded_test_server/embedded_test_server.h"
- #include "net/test/spawned_test_server/spawned_test_server.h"
- static void PrintUsage() {
- printf(
- "run_testserver --doc-root=relpath\n"
- " [--http|--https|--ws|--wss]\n"
- " [--ssl-cert=ok|mismatched-name|expired]\n");
- printf("(NOTE: relpath should be relative to the 'src' directory.\n");
- }
- int main(int argc, const char* argv[]) {
- base::AtExitManager at_exit_manager;
- base::SingleThreadTaskExecutor io_task_executor(base::MessagePumpType::IO);
- // Process command line
- base::CommandLine::Init(argc, argv);
- base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
- logging::LoggingSettings settings;
- settings.logging_dest = logging::LOG_TO_ALL;
- settings.log_file_path = FILE_PATH_LITERAL("testserver.log");
- if (!logging::InitLogging(settings)) {
- printf("Error: could not initialize logging. Exiting.\n");
- return -1;
- }
- TestTimeouts::Initialize();
- if (command_line->GetSwitches().empty() ||
- command_line->HasSwitch("help")) {
- PrintUsage();
- return -1;
- }
- // If populated, EmbeddedTestServer is used instead of the SpawnedTestServer.
- absl::optional<net::EmbeddedTestServer::Type> embedded_test_server_type;
- net::SpawnedTestServer::Type server_type;
- if (command_line->HasSwitch("http")) {
- embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTP;
- } else if (command_line->HasSwitch("https")) {
- embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTPS;
- } else if (command_line->HasSwitch("ws")) {
- server_type = net::SpawnedTestServer::TYPE_WS;
- } else if (command_line->HasSwitch("wss")) {
- server_type = net::SpawnedTestServer::TYPE_WSS;
- } else {
- // If no scheme switch is specified, select http or https scheme.
- // TODO(toyoshim): Remove this estimation.
- if (command_line->HasSwitch("ssl-cert")) {
- embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTPS;
- } else {
- embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTP;
- }
- }
- net::SpawnedTestServer::SSLOptions ssl_options;
- net::EmbeddedTestServer::ServerCertificate server_certificate;
- if (command_line->HasSwitch("ssl-cert")) {
- if ((embedded_test_server_type.has_value() &&
- *embedded_test_server_type != net::EmbeddedTestServer::TYPE_HTTPS) ||
- (!embedded_test_server_type.has_value() &&
- !net::SpawnedTestServer::UsingSSL(server_type))) {
- printf("Error: --ssl-cert is specified on non-secure scheme\n");
- PrintUsage();
- return -1;
- }
- std::string cert_option = command_line->GetSwitchValueASCII("ssl-cert");
- if (cert_option == "ok") {
- ssl_options.server_certificate =
- net::SpawnedTestServer::SSLOptions::CERT_OK;
- server_certificate = net::EmbeddedTestServer::CERT_OK;
- } else if (cert_option == "mismatched-name") {
- ssl_options.server_certificate =
- net::SpawnedTestServer::SSLOptions::CERT_MISMATCHED_NAME;
- server_certificate = net::EmbeddedTestServer::CERT_MISMATCHED_NAME;
- } else if (cert_option == "expired") {
- ssl_options.server_certificate =
- net::SpawnedTestServer::SSLOptions::CERT_EXPIRED;
- server_certificate = net::EmbeddedTestServer::CERT_EXPIRED;
- } else {
- printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str());
- PrintUsage();
- return -1;
- }
- }
- base::FilePath doc_root = command_line->GetSwitchValuePath("doc-root");
- if (doc_root.empty()) {
- printf("Error: --doc-root must be specified\n");
- PrintUsage();
- return -1;
- }
- base::FilePath full_path =
- net::test_server::EmbeddedTestServer::GetFullPathFromSourceDirectory(
- doc_root);
- if (!base::DirectoryExists(full_path)) {
- printf("Error: invalid doc root: \"%s\" does not exist!\n",
- base::UTF16ToUTF8(full_path.LossyDisplayName()).c_str());
- return -1;
- }
- // Use EmbeddedTestServer, if it supports the provided configuration.
- if (embedded_test_server_type.has_value()) {
- net::EmbeddedTestServer embedded_test_server(*embedded_test_server_type);
- if (*embedded_test_server_type == net::EmbeddedTestServer::TYPE_HTTPS) {
- embedded_test_server.SetSSLConfig(server_certificate);
- }
- embedded_test_server.AddDefaultHandlers(doc_root);
- if (!embedded_test_server.Start()) {
- printf("Error: failed to start embedded test server. Exiting.\n");
- return -1;
- }
- printf("Embedded test server running at %s (type ctrl+c to exit)\n",
- embedded_test_server.host_port_pair().ToString().c_str());
- base::RunLoop().Run();
- return 0;
- }
- // Otherwise, use the SpawnedTestServer.
- std::unique_ptr<net::SpawnedTestServer> test_server;
- if (net::SpawnedTestServer::UsingSSL(server_type)) {
- test_server = std::make_unique<net::SpawnedTestServer>(
- server_type, ssl_options, doc_root);
- } else {
- test_server =
- std::make_unique<net::SpawnedTestServer>(server_type, doc_root);
- }
- if (!test_server->Start()) {
- printf("Error: failed to start test server. Exiting.\n");
- return -1;
- }
- printf("testserver running at %s (type ctrl+c to exit)\n",
- test_server->host_port_pair().ToString().c_str());
- base::RunLoop().Run();
- }
|