components/browser_watcher/exit_code_watcher_win.h

   1 // Copyright (c) 2014 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 #ifndef COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_
   5 #define COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_
   6
   7 #include "base/macros.h"
   8 #include "base/process/process.h"
   9 #include "base/strings/string16.h"
  10 #include "base/time/time.h"
  11 #include "base/win/scoped_handle.h"
  12
  13 namespace browser_watcher {
  14
  15 // Watches for the exit code of a process and records it in a given registry
  16 // location.
  17 class ExitCodeWatcher {
  18  public:
  19   // Initialize the watcher with a registry path.
  20   explicit ExitCodeWatcher(const base::char16* registry_path);
  21   ~ExitCodeWatcher();
  22
  23   // Initializes from arguments on |cmd_line|, returns true on success.
  24   // This function expects |process| to be open with sufficient privilege to
  25   // wait and retrieve the process exit code.
  26   // It checks the handle for validity and takes ownership of it.
  27   // The intent is for this handle to be inherited into the watcher process
  28   // hosting the instance of this class.
  29   bool Initialize(base::Process process);
  30
  31   // Waits for the process to exit and records its exit code in registry.
  32   // This is a blocking call.
  33   void WaitForExit();
  34
  35   const base::Process& process() const { return process_; }
  36   int exit_code() const { return exit_code_; }
  37
  38  private:
  39   // Writes |exit_code| to registry, returns true on success.
  40   bool WriteProcessExitCode(int exit_code);
  41
  42   // The registry path the exit codes are written to.
  43   base::string16 registry_path_;
  44
  45   // Watched process and its creation time.
  46   base::Process process_;
  47   base::Time process_creation_time_;
  48
  49   // The exit code of the watched process. Valid after WaitForExit.
  50   int exit_code_;
  51
  52   DISALLOW_COPY_AND_ASSIGN(ExitCodeWatcher);
  53 };
  54
  55 }  // namespace browser_watcher
  56
  57 #endif  // COMPONENTS_BROWSER_WATCHER_EXIT_CODE_WATCHER_WIN_H_