| // Copyright (c) 2014 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. |
| #ifndef COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_ |
| #define COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_ |
| |
| #include "base/macros.h" |
| #include "base/process/process.h" |
| #include "base/strings/string16.h" |
| #include "base/time/time.h" |
| #include "base/win/scoped_handle.h" |
| |
| namespace base { |
| class CommandLine; |
| } |
| |
| namespace browser_watcher { |
| |
| // Watches for the exit code of a process and records it in a given registry |
| // location. |
| class ExitCodeWatcher { |
| public: |
| // Name of the switch used for the parent process handle. |
| static const char kParenthHandleSwitch[]; |
| |
| // Initialize the watcher with a registry path. |
| explicit ExitCodeWatcher(const base::char16* registry_path); |
| ~ExitCodeWatcher(); |
| |
| // Initializes from arguments on |cmd_line|, returns true on success. |
| // This function expects the process handle indicated by kParentHandleSwitch |
| // in |cmd_line| to be open with sufficient privilege to wait and retrieve |
| // the process exit code. |
| // It checks the handle for validity and takes ownership of it. |
| // The intent is for this handle to be inherited into the watcher process |
| // hosting the instance of this class. |
| bool ParseArguments(const base::CommandLine& cmd_line); |
| |
| // Waits for the process to exit and records its exit code in registry. |
| // This is a blocking call. |
| void WaitForExit(); |
| |
| const base::Process& process() const { return process_; } |
| |
| private: |
| // Writes |exit_code| to registry, returns true on success. |
| bool WriteProcessExitCode(int exit_code); |
| |
| // The registry path the exit codes are written to. |
| base::string16 registry_path_; |
| |
| // Handle, PID, and creation time of the watched process. |
| base::Process process_; |
| base::Time process_creation_time_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ExitCodeWatcher); |
| }; |
| |
| } // namespace browser_watcher |
| |
| #endif // COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_ |