| Index: trunk/src/tools/gn/function_exec_script.cc
|
| ===================================================================
|
| --- trunk/src/tools/gn/function_exec_script.cc (revision 214322)
|
| +++ trunk/src/tools/gn/function_exec_script.cc (working copy)
|
| @@ -1,254 +0,0 @@
|
| -// Copyright (c) 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 "base/command_line.h"
|
| -#include "base/file_util.h"
|
| -#include "base/strings/string_number_conversions.h"
|
| -#include "build/build_config.h"
|
| -#include "tools/gn/err.h"
|
| -#include "tools/gn/filesystem_utils.h"
|
| -#include "tools/gn/functions.h"
|
| -#include "tools/gn/input_conversion.h"
|
| -#include "tools/gn/input_file.h"
|
| -#include "tools/gn/parse_tree.h"
|
| -#include "tools/gn/scheduler.h"
|
| -#include "tools/gn/value.h"
|
| -
|
| -#if defined(OS_WIN)
|
| -#include <windows.h>
|
| -
|
| -#include "base/win/scoped_handle.h"
|
| -#include "base/win/scoped_process_information.h"
|
| -#endif
|
| -
|
| -/*
|
| -exec_script: Synchronously run a script and return the output.
|
| -
|
| - exec_script(filename, arguments, input_conversion, [file_dependencies])
|
| -
|
| - Runs the given script, returning the stdout of the script. The build
|
| - generation will fail if the script does not exist or returns a nonzero
|
| - exit code.
|
| -
|
| -Arguments:
|
| -
|
| - filename:
|
| - File name of python script to execute, relative to the build file.
|
| -
|
| - arguments:
|
| - A list of strings to be passed to the scripe as arguments.
|
| -
|
| - input_conversion:
|
| - Controls how the file is read and parsed. See "help input_conversion".
|
| -
|
| - dependencies:
|
| - (Optional) A list of files that this script reads or otherwise depends
|
| - on. These dependencies will be added to the build result such that if
|
| - any of them change, the build will be regenerated and the script will
|
| - be re-run.
|
| -
|
| - The script itself will be an implicit dependency so you do not need to
|
| - list it.
|
| -
|
| -Example:
|
| -
|
| - all_lines = exec_script("myscript.py", [some_input], "list lines",
|
| - ["data_file.txt"])
|
| -*/
|
| -
|
| -namespace {
|
| -
|
| -#if defined(OS_WIN)
|
| -bool ExecProcess(const CommandLine& cmdline,
|
| - const base::FilePath& startup_dir,
|
| - std::string* std_out,
|
| - std::string* std_err,
|
| - int* exit_code) {
|
| - SECURITY_ATTRIBUTES sa_attr;
|
| - // Set the bInheritHandle flag so pipe handles are inherited.
|
| - sa_attr.nLength = sizeof(SECURITY_ATTRIBUTES);
|
| - sa_attr.bInheritHandle = TRUE;
|
| - sa_attr.lpSecurityDescriptor = NULL;
|
| -
|
| - // Create the pipe for the child process's STDOUT.
|
| - HANDLE out_read = NULL;
|
| - HANDLE out_write = NULL;
|
| - if (!CreatePipe(&out_read, &out_write, &sa_attr, 0)) {
|
| - NOTREACHED() << "Failed to create pipe";
|
| - return false;
|
| - }
|
| - base::win::ScopedHandle scoped_out_read(out_read);
|
| - base::win::ScopedHandle scoped_out_write(out_write);
|
| -
|
| - // Create the pipe for the child process's STDERR.
|
| - HANDLE err_read = NULL;
|
| - HANDLE err_write = NULL;
|
| - if (!CreatePipe(&err_read, &err_write, &sa_attr, 0)) {
|
| - NOTREACHED() << "Failed to create pipe";
|
| - return false;
|
| - }
|
| - base::win::ScopedHandle scoped_err_read(err_read);
|
| - base::win::ScopedHandle scoped_err_write(err_write);
|
| -
|
| - // Ensure the read handle to the pipe for STDOUT/STDERR is not inherited.
|
| - if (!SetHandleInformation(out_read, HANDLE_FLAG_INHERIT, 0)) {
|
| - NOTREACHED() << "Failed to disabled pipe inheritance";
|
| - return false;
|
| - }
|
| - if (!SetHandleInformation(err_read, HANDLE_FLAG_INHERIT, 0)) {
|
| - NOTREACHED() << "Failed to disabled pipe inheritance";
|
| - return false;
|
| - }
|
| -
|
| - base::FilePath::StringType cmdline_str(cmdline.GetCommandLineString());
|
| -
|
| - base::win::ScopedProcessInformation proc_info;
|
| - STARTUPINFO start_info = { 0 };
|
| -
|
| - start_info.cb = sizeof(STARTUPINFO);
|
| - start_info.hStdOutput = out_write;
|
| - // Keep the normal stdin.
|
| - start_info.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
|
| - // FIXME(brettw) set stderr here when we actually read it below.
|
| - //start_info.hStdError = err_write;
|
| - start_info.hStdError = GetStdHandle(STD_ERROR_HANDLE);
|
| - start_info.dwFlags |= STARTF_USESTDHANDLES;
|
| -
|
| - // Create the child process.
|
| - if (!CreateProcess(NULL,
|
| - &cmdline_str[0],
|
| - NULL, NULL,
|
| - TRUE, // Handles are inherited.
|
| - 0, NULL,
|
| - startup_dir.value().c_str(),
|
| - &start_info, proc_info.Receive())) {
|
| - return false;
|
| - }
|
| -
|
| - // Close our writing end of pipes now. Otherwise later read would not be able
|
| - // to detect end of child's output.
|
| - scoped_out_write.Close();
|
| - scoped_err_write.Close();
|
| -
|
| - // Read output from the child process's pipe for STDOUT
|
| - const int kBufferSize = 1024;
|
| - char buffer[kBufferSize];
|
| -
|
| - // FIXME(brettw) read from stderr here! This is complicated because we want
|
| - // to read both of them at the same time, probably need overlapped I/O.
|
| - // Also uncomment start_info code above.
|
| - for (;;) {
|
| - DWORD bytes_read = 0;
|
| - BOOL success = ReadFile(out_read, buffer, kBufferSize, &bytes_read, NULL);
|
| - if (!success || bytes_read == 0)
|
| - break;
|
| - std_out->append(buffer, bytes_read);
|
| - }
|
| -
|
| - // Let's wait for the process to finish.
|
| - WaitForSingleObject(proc_info.process_handle(), INFINITE);
|
| -
|
| - DWORD dw_exit_code;
|
| - GetExitCodeProcess(proc_info.process_handle(), &dw_exit_code);
|
| - *exit_code = static_cast<int>(dw_exit_code);
|
| -
|
| - return true;
|
| -}
|
| -#else
|
| -bool ExecProcess(const CommandLine& cmdline,
|
| - const base::FilePath& startup_dir,
|
| - std::string* std_out,
|
| - std::string* std_err,
|
| - int* exit_code) {
|
| - //NOTREACHED() << "Implement me.";
|
| - return false;
|
| -}
|
| -#endif
|
| -
|
| -} // namespace
|
| -
|
| -Value ExecuteExecScript(Scope* scope,
|
| - const FunctionCallNode* function,
|
| - const std::vector<Value>& args,
|
| - Err* err) {
|
| - if (args.size() != 3 && args.size() != 4) {
|
| - *err = Err(function->function(), "Wrong number of args to write_file",
|
| - "I expected three or four arguments.");
|
| - return Value();
|
| - }
|
| -
|
| - const Settings* settings = scope->settings();
|
| - const BuildSettings* build_settings = settings->build_settings();
|
| - const SourceDir& cur_dir = SourceDirForFunctionCall(function);
|
| -
|
| - // Find the python script to run.
|
| - if (!args[0].VerifyTypeIs(Value::STRING, err))
|
| - return Value();
|
| - SourceFile script_source =
|
| - cur_dir.ResolveRelativeFile(args[0].string_value());
|
| - base::FilePath script_path = build_settings->GetFullPath(script_source);
|
| - if (!build_settings->secondary_source_path().empty() &&
|
| - !base::PathExists(script_path)) {
|
| - // Fall back to secondary source root when the file doesn't exist.
|
| - script_path = build_settings->GetFullPathSecondary(script_source);
|
| - }
|
| -
|
| - // Add all dependencies of this script, including the script itself, to the
|
| - // build deps.
|
| - g_scheduler->AddGenDependency(script_source);
|
| - if (args.size() == 4) {
|
| - const Value& deps_value = args[3];
|
| - if (!deps_value.VerifyTypeIs(Value::LIST, err))
|
| - return Value();
|
| -
|
| - for (size_t i = 0; i < deps_value.list_value().size(); i++) {
|
| - if (!deps_value.list_value()[0].VerifyTypeIs(Value::STRING, err))
|
| - return Value();
|
| - g_scheduler->AddGenDependency(cur_dir.ResolveRelativeFile(
|
| - deps_value.list_value()[0].string_value()));
|
| - }
|
| - }
|
| -
|
| - // Make the command line.
|
| - const base::FilePath& python_path = build_settings->python_path();
|
| - CommandLine cmdline(python_path);
|
| - cmdline.AppendArgPath(script_path);
|
| -
|
| - const Value& script_args = args[1];
|
| - if (!script_args.VerifyTypeIs(Value::LIST, err))
|
| - return Value();
|
| - for (size_t i = 0; i < script_args.list_value().size(); i++) {
|
| - if (!script_args.list_value()[i].VerifyTypeIs(Value::STRING, err))
|
| - return Value();
|
| - cmdline.AppendArg(script_args.list_value()[i].string_value());
|
| - }
|
| -
|
| - // Execute the process.
|
| - // TODO(brettw) set the environment block.
|
| - std::string output;
|
| - std::string stderr_output; // TODO(brettw) not hooked up, see above.
|
| - int exit_code = 0;
|
| - if (!ExecProcess(cmdline, build_settings->GetFullPath(cur_dir),
|
| - &output, &stderr_output, &exit_code)) {
|
| - *err = Err(function->function(), "Could not execute python.",
|
| - "I was trying to execute \"" + FilePathToUTF8(python_path) + "\".");
|
| - return Value();
|
| - }
|
| -
|
| - // TODO(brettw) maybe we need stderr also for reasonable stack dumps.
|
| - if (exit_code != 0) {
|
| - std::string msg =
|
| - std::string("I was running \"") + FilePathToUTF8(script_path) + "\"\n"
|
| - "and it returned " + base::IntToString(exit_code);
|
| - if (!output.empty())
|
| - msg += " and printed out:\n\n" + output;
|
| - else
|
| - msg += ".";
|
| - *err = Err(function->function(), "Script returned non-zero exit code.",
|
| - msg);
|
| - return Value();
|
| - }
|
| -
|
| - return ConvertInputToValue(output, function, args[2], err);
|
| -}
|
|
|