run_testserver.cc 5.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. // Copyright 2013 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include <stdio.h>
  5. #include <memory>
  6. #include "base/at_exit.h"
  7. #include "base/command_line.h"
  8. #include "base/files/file_path.h"
  9. #include "base/files/file_util.h"
  10. #include "base/logging.h"
  11. #include "base/message_loop/message_pump_type.h"
  12. #include "base/run_loop.h"
  13. #include "base/strings/utf_string_conversions.h"
  14. #include "base/task/single_thread_task_executor.h"
  15. #include "base/test/test_timeouts.h"
  16. #include "net/test/embedded_test_server/embedded_test_server.h"
  17. #include "net/test/spawned_test_server/spawned_test_server.h"
  18. static void PrintUsage() {
  19. printf(
  20. "run_testserver --doc-root=relpath\n"
  21. " [--http|--https|--ws|--wss]\n"
  22. " [--ssl-cert=ok|mismatched-name|expired]\n");
  23. printf("(NOTE: relpath should be relative to the 'src' directory.\n");
  24. }
  25. int main(int argc, const char* argv[]) {
  26. base::AtExitManager at_exit_manager;
  27. base::SingleThreadTaskExecutor io_task_executor(base::MessagePumpType::IO);
  28. // Process command line
  29. base::CommandLine::Init(argc, argv);
  30. base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  31. logging::LoggingSettings settings;
  32. settings.logging_dest = logging::LOG_TO_ALL;
  33. settings.log_file_path = FILE_PATH_LITERAL("testserver.log");
  34. if (!logging::InitLogging(settings)) {
  35. printf("Error: could not initialize logging. Exiting.\n");
  36. return -1;
  37. }
  38. TestTimeouts::Initialize();
  39. if (command_line->GetSwitches().empty() ||
  40. command_line->HasSwitch("help")) {
  41. PrintUsage();
  42. return -1;
  43. }
  44. // If populated, EmbeddedTestServer is used instead of the SpawnedTestServer.
  45. absl::optional<net::EmbeddedTestServer::Type> embedded_test_server_type;
  46. net::SpawnedTestServer::Type server_type;
  47. if (command_line->HasSwitch("http")) {
  48. embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTP;
  49. } else if (command_line->HasSwitch("https")) {
  50. embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTPS;
  51. } else if (command_line->HasSwitch("ws")) {
  52. server_type = net::SpawnedTestServer::TYPE_WS;
  53. } else if (command_line->HasSwitch("wss")) {
  54. server_type = net::SpawnedTestServer::TYPE_WSS;
  55. } else {
  56. // If no scheme switch is specified, select http or https scheme.
  57. // TODO(toyoshim): Remove this estimation.
  58. if (command_line->HasSwitch("ssl-cert")) {
  59. embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTPS;
  60. } else {
  61. embedded_test_server_type = net::EmbeddedTestServer::TYPE_HTTP;
  62. }
  63. }
  64. net::SpawnedTestServer::SSLOptions ssl_options;
  65. net::EmbeddedTestServer::ServerCertificate server_certificate;
  66. if (command_line->HasSwitch("ssl-cert")) {
  67. if ((embedded_test_server_type.has_value() &&
  68. *embedded_test_server_type != net::EmbeddedTestServer::TYPE_HTTPS) ||
  69. (!embedded_test_server_type.has_value() &&
  70. !net::SpawnedTestServer::UsingSSL(server_type))) {
  71. printf("Error: --ssl-cert is specified on non-secure scheme\n");
  72. PrintUsage();
  73. return -1;
  74. }
  75. std::string cert_option = command_line->GetSwitchValueASCII("ssl-cert");
  76. if (cert_option == "ok") {
  77. ssl_options.server_certificate =
  78. net::SpawnedTestServer::SSLOptions::CERT_OK;
  79. server_certificate = net::EmbeddedTestServer::CERT_OK;
  80. } else if (cert_option == "mismatched-name") {
  81. ssl_options.server_certificate =
  82. net::SpawnedTestServer::SSLOptions::CERT_MISMATCHED_NAME;
  83. server_certificate = net::EmbeddedTestServer::CERT_MISMATCHED_NAME;
  84. } else if (cert_option == "expired") {
  85. ssl_options.server_certificate =
  86. net::SpawnedTestServer::SSLOptions::CERT_EXPIRED;
  87. server_certificate = net::EmbeddedTestServer::CERT_EXPIRED;
  88. } else {
  89. printf("Error: --ssl-cert has invalid value %s\n", cert_option.c_str());
  90. PrintUsage();
  91. return -1;
  92. }
  93. }
  94. base::FilePath doc_root = command_line->GetSwitchValuePath("doc-root");
  95. if (doc_root.empty()) {
  96. printf("Error: --doc-root must be specified\n");
  97. PrintUsage();
  98. return -1;
  99. }
  100. base::FilePath full_path =
  101. net::test_server::EmbeddedTestServer::GetFullPathFromSourceDirectory(
  102. doc_root);
  103. if (!base::DirectoryExists(full_path)) {
  104. printf("Error: invalid doc root: \"%s\" does not exist!\n",
  105. base::UTF16ToUTF8(full_path.LossyDisplayName()).c_str());
  106. return -1;
  107. }
  108. // Use EmbeddedTestServer, if it supports the provided configuration.
  109. if (embedded_test_server_type.has_value()) {
  110. net::EmbeddedTestServer embedded_test_server(*embedded_test_server_type);
  111. if (*embedded_test_server_type == net::EmbeddedTestServer::TYPE_HTTPS) {
  112. embedded_test_server.SetSSLConfig(server_certificate);
  113. }
  114. embedded_test_server.AddDefaultHandlers(doc_root);
  115. if (!embedded_test_server.Start()) {
  116. printf("Error: failed to start embedded test server. Exiting.\n");
  117. return -1;
  118. }
  119. printf("Embedded test server running at %s (type ctrl+c to exit)\n",
  120. embedded_test_server.host_port_pair().ToString().c_str());
  121. base::RunLoop().Run();
  122. return 0;
  123. }
  124. // Otherwise, use the SpawnedTestServer.
  125. std::unique_ptr<net::SpawnedTestServer> test_server;
  126. if (net::SpawnedTestServer::UsingSSL(server_type)) {
  127. test_server = std::make_unique<net::SpawnedTestServer>(
  128. server_type, ssl_options, doc_root);
  129. } else {
  130. test_server =
  131. std::make_unique<net::SpawnedTestServer>(server_type, doc_root);
  132. }
  133. if (!test_server->Start()) {
  134. printf("Error: failed to start test server. Exiting.\n");
  135. return -1;
  136. }
  137. printf("testserver running at %s (type ctrl+c to exit)\n",
  138. test_server->host_port_pair().ToString().c_str());
  139. base::RunLoop().Run();
  140. }