blob: 319eee0789e21e15ee491541e08659167424de57 [file] [log] [blame]
// Copyright (c) 2012 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 "net/test/spawned_test_server/local_test_server.h"
#include <windows.h>
#include "base/base_paths.h"
#include "base/bind.h"
#include "base/command_line.h"
#include "base/environment.h"
#include "base/files/file_path.h"
#include "base/logging.h"
#include "base/path_service.h"
#include "base/process/launch.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/win/scoped_handle.h"
#include "net/test/python_utils.h"
namespace {
// Given a file handle, reads into |buffer| until |bytes_max| bytes
// has been read or an error has been encountered. Returns
// true if the read was successful.
bool ReadData(HANDLE read_fd,
HANDLE write_fd,
DWORD bytes_max,
uint8_t* buffer) {
DWORD bytes_read = 0;
while (bytes_read < bytes_max) {
DWORD num_bytes;
if (!ReadFile(read_fd, buffer + bytes_read, bytes_max - bytes_read,
&num_bytes, nullptr)) {
PLOG(ERROR) << "ReadFile failed";
return false;
if (num_bytes <= 0) {
LOG(ERROR) << "ReadFile returned invalid byte count: " << num_bytes;
return false;
bytes_read += num_bytes;
return true;
} // namespace
namespace net {
bool LocalTestServer::LaunchPython(
const base::FilePath& testserver_path,
const std::vector<base::FilePath>& python_path) {
base::CommandLine python_command(base::CommandLine::NO_PROGRAM);
if (!GetPythonCommand(&python_command))
return false;
if (!AddCommandLineArguments(&python_command))
return false;
HANDLE child_read = nullptr;
HANDLE child_write = nullptr;
if (!CreatePipe(&child_read, &child_write, nullptr, 0)) {
PLOG(ERROR) << "Failed to create pipe";
return false;
// Have the child inherit the write half.
if (!::DuplicateHandle(::GetCurrentProcess(), child_write,
::GetCurrentProcess(), &child_write, 0, TRUE,
PLOG(ERROR) << "Failed to enable pipe inheritance";
return false;
// Pass the handle on the command-line. Although HANDLE is a
// pointer, truncating it on 64-bit machines is okay. See
// "64-bit versions of Windows use 32-bit handles for
// interoperability. When sharing a handle between 32-bit and 64-bit
// applications, only the lower 32 bits are significant, so it is
// safe to truncate the handle (when passing it from 64-bit to
// 32-bit) or sign-extend the handle (when passing it from 32-bit to
// 64-bit)."
"--startup-pipe=" +
base::LaunchOptions launch_options;
SetPythonPathInEnvironment(python_path, &launch_options.environment);
// Set CWD to source root.
if (!base::PathService::Get(base::DIR_SOURCE_ROOT,
&launch_options.current_directory)) {
LOG(ERROR) << "Failed to get DIR_SOURCE_ROOT";
return false;
// TODO(brettw) bug 748258: Share only explicit handles.
launch_options.inherit_mode = base::LaunchOptions::Inherit::kAll;
process_ = base::LaunchProcess(python_command, launch_options);
if (!process_.IsValid()) {
LOG(ERROR) << "Failed to launch " << python_command.GetCommandLineString();
return false;
return true;
bool LocalTestServer::WaitToStart() {
base::win::ScopedHandle read_fd(child_read_fd_.Take());
base::win::ScopedHandle write_fd(child_write_fd_.Take());
uint32_t server_data_len = 0;
if (!ReadData(read_fd.Get(), write_fd.Get(), sizeof(server_data_len),
reinterpret_cast<uint8_t*>(&server_data_len))) {
LOG(ERROR) << "Could not read server_data_len";
return false;
std::string server_data(server_data_len, '\0');
if (!ReadData(read_fd.Get(), write_fd.Get(), server_data_len,
reinterpret_cast<uint8_t*>(&server_data[0]))) {
LOG(ERROR) << "Could not read server_data (" << server_data_len
<< " bytes)";
return false;
int port;
if (!SetAndParseServerData(server_data, &port)) {
LOG(ERROR) << "Could not parse server_data: " << server_data;
return false;
return true;
} // namespace net