// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef REMOTING_HOST_SHUTDOWN_WATCHDOG_H_
#define REMOTING_HOST_SHUTDOWN_WATCHDOG_H_

#include "base/synchronization/lock.h"
#include "base/threading/watchdog.h"

namespace remoting {

// This implements a watchdog timer that ensures the host process eventually
// terminates, even if some threads are blocked or being kept alive for
// some reason. This is not expected to trigger if host shutdown is working
// correctly (on a normally loaded system). The triggering of the alarm
// indicates a sign of trouble, and so the Alarm() method will log some
// diagnostic information before shutting down the process.
class ShutdownWatchdog : public base::Watchdog::Delegate {
 public:
  // Creates a watchdog that waits for |duration| (after the watchdog is
  // armed) before shutting down the process.
  explicit ShutdownWatchdog(const base::TimeDelta& duration);

  ShutdownWatchdog(const ShutdownWatchdog&) = delete;
  ShutdownWatchdog& operator=(const ShutdownWatchdog&) = delete;

  void Arm();

  // This method should be called to set the process's exit-code before arming
  // the watchdog. Otherwise an exit-code of 0 is assumed.
  void SetExitCode(int exit_code);

  void Alarm() override;

 private:
  int exit_code_;

  // Protects |exit_code_|, since Alarm() gets called on a separate thread.
  base::Lock lock_;
  base::Watchdog watchdog_;
};

}  // namespace remoting

#endif  // REMOTING_HOST_SHUTDOWN_WATCHDOG_H_