content/test/layout_test_http_server.cc

   1 // Copyright (c) 2012 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
   5 #include "content/test/layout_test_http_server.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/logging.h"
   9 #include "base/path_service.h"
  10 #include "base/process_util.h"
  11 #include "base/string_number_conversions.h"
  12 #include "content/public/common/content_paths.h"
  13 #include "net/test/python_utils.h"
  14
  15 #if defined(OS_WIN)
  16 #include "base/win/windows_version.h"
  17 #endif
  18
  19 namespace content {
  20 namespace {
  21
  22 bool PrepareCommandLine(CommandLine* cmd_line) {
  23   base::FilePath src_path;
  24   if (!PathService::Get(base::DIR_SOURCE_ROOT, &src_path))
  25     return false;
  26
  27   if (!GetPythonCommand(cmd_line))
  28     return false;
  29
  30   base::FilePath script_path(src_path);
  31   script_path = script_path.AppendASCII("third_party");
  32   script_path = script_path.AppendASCII("WebKit");
  33   script_path = script_path.AppendASCII("Tools");
  34   script_path = script_path.AppendASCII("Scripts");
  35   script_path = script_path.AppendASCII("new-run-webkit-httpd");
  36
  37   cmd_line->AppendArgPath(script_path);
  38   return true;
  39 }
  40
  41 }  // namespace
  42
  43 LayoutTestHttpServer::LayoutTestHttpServer(const base::FilePath& root_directory,
  44                                            int port)
  45     : root_directory_(root_directory),
  46       port_(port),
  47       running_(false) {
  48 }
  49
  50 LayoutTestHttpServer::~LayoutTestHttpServer() {
  51   if (running_ && !Stop())
  52     LOG(ERROR) << "LayoutTestHttpServer failed to stop.";
  53 }
  54
  55 bool LayoutTestHttpServer::Start() {
  56   if (running_) {
  57     LOG(ERROR) << "LayoutTestHttpServer already running.";
  58     return false;
  59   }
  60
  61   CommandLine cmd_line(CommandLine::NO_PROGRAM);
  62   if (!PrepareCommandLine(&cmd_line))
  63     return false;
  64   cmd_line.AppendArg("--server=start");
  65   cmd_line.AppendArg("--register_cygwin");
  66   cmd_line.AppendArgNative(FILE_PATH_LITERAL("--root=") +
  67                            root_directory_.value());
  68   cmd_line.AppendArg("--port=" + base::IntToString(port_));
  69
  70   base::FilePath layout_tests_dir;
  71   if (!PathService::Get(DIR_LAYOUT_TESTS, &layout_tests_dir))
  72     return false;
  73   cmd_line.AppendArgNative(FILE_PATH_LITERAL("--layout_tests_dir=") +
  74                            layout_tests_dir.value());
  75
  76 #if defined(OS_WIN)
  77   // For Windows 7, if we start the lighttpd server on the foreground mode,
  78   // it will mess up with the command window and cause conhost.exe to crash. To
  79   // work around this, we start the http server on the background mode.
  80   if (base::win::GetVersion() >= base::win::VERSION_WIN7)
  81     cmd_line.AppendArg("--run_background");
  82
  83   job_handle_.Set(CreateJobObject(NULL, NULL));
  84   if (!job_handle_.IsValid()) {
  85     LOG(ERROR) << "Could not create JobObject.";
  86     return false;
  87   }
  88
  89   if (!base::SetJobObjectAsKillOnJobClose(job_handle_.Get())) {
  90     LOG(ERROR) << "Could not SetInformationJobObject.";
  91     return false;
  92   }
  93 #endif
  94
  95   // The Python script waits for the server to start responding to requests,
  96   // then exits.  So we want to wait for the Python script to exit before
  97   // continuing.
  98   base::LaunchOptions options;
  99   options.wait = true;
 100 #if defined(OS_WIN)
 101   options.job_handle = job_handle_.Get();
 102 #endif
 103   running_ = base::LaunchProcess(cmd_line, options, NULL);
 104   return running_;
 105 }
 106
 107 bool LayoutTestHttpServer::Stop() {
 108   if (!running_) {
 109     LOG(ERROR) << "LayoutTestHttpServer not running.";
 110     return false;
 111   }
 112
 113   CommandLine cmd_line(CommandLine::NO_PROGRAM);
 114   if (!PrepareCommandLine(&cmd_line))
 115     return false;
 116   cmd_line.AppendArg("--server=stop");
 117
 118   base::LaunchOptions options;
 119   options.wait = true;
 120 #if defined(OS_WIN)
 121   options.job_handle = job_handle_.Get();
 122 #endif
 123   bool stopped = base::LaunchProcess(cmd_line, options, NULL);
 124   running_ = !stopped;
 125
 126 #if defined(OS_WIN)
 127   // Close the job object handle now. This should clean up
 128   // any orphaned processes.
 129   job_handle_.Close();
 130 #endif
 131
 132   return stopped;
 133 }
 134
 135 }  // namespace content