Index: chrome/browser/component_updater/component_patcher_win.cc |
diff --git a/chrome/browser/component_updater/component_patcher_win.cc b/chrome/browser/component_updater/component_patcher_win.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8935f421329521f4f884c69072f0c37bafbeca2d |
--- /dev/null |
+++ b/chrome/browser/component_updater/component_patcher_win.cc |
@@ -0,0 +1,118 @@ |
+// 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 "chrome/browser/component_updater/component_patcher_win.h" |
+ |
+#include <string> |
+ |
+#include "base/base_paths.h" |
+#include "base/command_line.h" |
+#include "base/file_util.h" |
+#include "base/path_service.h" |
+#include "base/process/kill.h" |
+#include "base/process/launch.h" |
+#include "base/strings/string_util.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "base/win/scoped_handle.h" |
+#include "chrome/installer/util/util_constants.h" |
+ |
+namespace component_updater { |
+ |
+namespace { |
+ |
+std::string PatchTypeToCommandLineSwitch( |
+ ComponentPatcher::PatchType patch_type) { |
+ if (patch_type == ComponentPatcher::kPatchTypeCourgette) |
+ return std::string(installer::kCourgette); |
+ else if (patch_type == ComponentPatcher::kPatchTypeBsdiff) |
+ return std::string(installer::kBsdiff); |
+ else |
+ return std::string(); |
+} |
+ |
+// Finds the path to the setup.exe. First, it looks for the program in the |
+// "installer" directory. If the program is not found there, it tries to find it |
+// in the directory where chrome.dll lives. Returns the path to the setup.exe, |
+// if the path exists, otherwise it returns an an empty path. |
+base::FilePath FindSetupProgram() { |
+ base::FilePath exe_dir; |
+ if (!PathService::Get(base::DIR_MODULE, &exe_dir)) |
+ return base::FilePath(); |
+ |
+ const std::string installer_dir(base::UTF16ToASCII(installer::kInstallerDir)); |
+ const std::string setup_exe(base::UTF16ToASCII(installer::kSetupExe)); |
+ |
+ base::FilePath setup_path = exe_dir; |
+ setup_path = setup_path.AppendASCII(installer_dir); |
+ setup_path = setup_path.AppendASCII(setup_exe); |
+ if (base::PathExists(setup_path)) |
+ return setup_path; |
+ |
+ setup_path = exe_dir; |
+ setup_path = setup_path.AppendASCII(setup_exe); |
+ if (base::PathExists(setup_path)) |
+ return setup_path; |
+ |
+ return base::FilePath(); |
+} |
+ |
+} // namespace |
+ |
+// Applies the patch to the input file. Returns kNone if the patch was |
+// successfully applied, kDeltaOperationFailure if the patch operation |
+// encountered errors, and kDeltaPatchProcessFailure if there was an error |
+// when running the patch code out of process. In the error case, detailed error |
+// information could be returned in the error parameter. |
+ComponentUnpacker::Error ComponentPatcherWin::Patch( |
+ PatchType patch_type, |
+ const base::FilePath& input_file, |
+ const base::FilePath& patch_file, |
+ const base::FilePath& output_file, |
+ int* error) { |
+ *error = 0; |
+ |
+ const base::FilePath exe_path = FindSetupProgram(); |
+ if (exe_path.empty()) |
+ return ComponentUnpacker::kDeltaPatchProcessFailure; |
+ |
+ const std::string patch_type_str(PatchTypeToCommandLineSwitch(patch_type)); |
+ |
+ CommandLine cl(CommandLine::NO_PROGRAM); |
+ cl.AppendSwitchASCII(installer::switches::kPatch, patch_type_str.c_str()); |
+ cl.AppendSwitchPath(installer::switches::kInputFile, input_file); |
+ cl.AppendSwitchPath(installer::switches::kPatchFile, patch_file); |
+ cl.AppendSwitchPath(installer::switches::kOutputFile, output_file); |
+ |
+ // Create the child process in a job object. The job object prevents leaving |
+ // child processes around when the parent process exits, either gracefully or |
+ // accidentally. |
+ base::win::ScopedHandle job(CreateJobObject(NULL, NULL)); |
+ if (!job || |
+ !base::SetJobObjectLimitFlags(job, JOB_OBJECT_LIMIT_KILL_ON_JOB_CLOSE)) { |
+ *error = GetLastError(); |
+ return ComponentUnpacker::kDeltaPatchProcessFailure; |
+ } |
+ |
+ base::LaunchOptions launch_options; |
+ launch_options.wait = true; |
+ launch_options.job_handle = job; |
+ launch_options.start_hidden = true; |
+ CommandLine setup_path(exe_path); |
+ setup_path.AppendArguments(cl, false); |
+ |
+ // |ph| is closed by WaitForExitCode. |
+ base::ProcessHandle ph = base::kNullProcessHandle; |
+ int exit_code = 0; |
+ if (!base::LaunchProcess(setup_path, launch_options, &ph) || |
+ !base::WaitForExitCode(ph, &exit_code)) { |
+ *error = GetLastError(); |
+ return ComponentUnpacker::kDeltaPatchProcessFailure; |
+ } |
+ |
+ *error = exit_code; |
+ return *error ? ComponentUnpacker::kDeltaOperationFailure : |
+ ComponentUnpacker::kNone; |
+} |
+ |
+} // namespace component_updater |