Add a new app API to enable watchdog behavior restarts in kiosk apps

extensions/browser/api/runtime/runtime_api.h

 // Copyright 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 EXTENSIONS_BROWSER_API_RUNTIME_RUNTIME_API_H_
 #define EXTENSIONS_BROWSER_API_RUNTIME_RUNTIME_API_H_
 
 #include <string>
 
 #include "base/macros.h"
 #include "base/scoped_observer.h"
+#include "base/time/time.h"
+#include "base/timer/timer.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "extensions/browser/api/runtime/runtime_api_delegate.h"
 #include "extensions/browser/browser_context_keyed_api_factory.h"
 #include "extensions/browser/extension_function.h"
 #include "extensions/browser/extension_registry_observer.h"
 #include "extensions/browser/process_manager.h"
 #include "extensions/browser/process_manager_observer.h"
 #include "extensions/browser/update_observer.h"
 #include "extensions/common/api/runtime.h"
@@ skipping 36 matching lines @@
   void Observe(int type,
                const content::NotificationSource& source,
                const content::NotificationDetails& details) override;
 
   void ReloadExtension(const std::string& extension_id);
   bool CheckForUpdates(const std::string& extension_id,
                        const RuntimeAPIDelegate::UpdateCheckCallback& callback);
   void OpenURL(const GURL& uninstall_url);
   bool GetPlatformInfo(api::runtime::PlatformInfo* info);
   bool RestartDevice(std::string* error_message);
+
+  using OnWatchdogTimeoutCallback =
+      base::Callback<void(bool success, const std::string& message)>;
+  void RestartDeviceOnWatchdogTimeout(
+      const std::string& extension_id,
+      int seconds_from_now,
+      const OnWatchdogTimeoutCallback& callback);
+
   bool OpenOptionsPage(const Extension* extension);
 
  private:
   friend class BrowserContextKeyedAPIFactory<RuntimeAPI>;
+  friend class RestartOnWatchdogApiTest;
 
   // ExtensionRegistryObserver implementation.
   void OnExtensionLoaded(content::BrowserContext* browser_context,
                          const Extension* extension) override;
   void OnExtensionWillBeInstalled(content::BrowserContext* browser_context,
                                   const Extension* extension,
                                   bool is_update,
                                   const std::string& old_name) override;
   void OnExtensionUninstalled(content::BrowserContext* browser_context,
                               const Extension* extension,
                               UninstallReason reason) override;
 
+  // Cancels any previously scheduled restart request, invoking its associated
+  // JS callback function stored in |current_watchdog_request_callback_|.
+  void MaybeCancelRunningWatchdogTimer();
+
+  void ScheduleDelayedRestart(const std::string& extension_id,
+                              const base::Time& now,
+                              int seconds_from_now,
+                              const OnWatchdogTimeoutCallback& callback,
+                              std::unique_ptr<base::Value> stored_last_restart);
+
+  void OnRestartWatchdogTimeout(const std::string& extension_id);
+
   // BrowserContextKeyedAPI implementation:
   static const char* service_name() { return "RuntimeAPI"; }
   static const bool kServiceRedirectedInIncognito = true;
   static const bool kServiceIsNULLWhileTesting = true;
   void Shutdown() override;
 
   // extensions::UpdateObserver overrides:
   void OnAppUpdateAvailable(const Extension* extension) override;
   void OnChromeUpdateAvailable() override;
 
   // ProcessManagerObserver implementation:
   void OnBackgroundHostStartup(const Extension* extension) override;
 
   // Pref related functions that deals with info about installed extensions that
   // has not been loaded yet.
   // Used to send chrome.runtime.onInstalled event upon loading the extensions.
   bool ReadPendingOnInstallInfoFromPref(const ExtensionId& extension_id,
                                         base::Version* previous_version);
   void RemovePendingOnInstallInfoFromPref(const ExtensionId& extension_id);
   void StorePendingOnInstallInfoToPref(const Extension* extension);
 
+  void SetMinDurationBetweenRestartsForTesting(base::TimeDelta delta) {
+    minimum_duration_between_restarts_ = delta;
+  }
+
   std::unique_ptr<RuntimeAPIDelegate> delegate_;
 
   content::BrowserContext* browser_context_;
 
   // True if we should dispatch the chrome.runtime.onInstalled event with
   // reason "chrome_update" upon loading each extension.
   bool dispatch_chrome_updated_event_;
 
   content::NotificationRegistrar registrar_;
 
   // Listen to extension notifications.
   ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver>
       extension_registry_observer_;
   ScopedObserver<ProcessManager, ProcessManagerObserver>
       process_manager_observer_;
 
+  // The ID of the first extension to call the restartOnWatchdog API. Any other
+  // extensions to call this API after that will fail.
+  std::string schedule_restart_first_extension_id_;
+
+  // The current JS callback for the current scheduled restart which will be
+  // invoked when that restart attempt is executed.
+  OnWatchdogTimeoutCallback current_watchdog_request_callback_;
+
+  // The timer that will trigger a device restart when it times out.
+  base::OneShotTimer watchdog_timer_;
+
+  // The minimum allowed duration between two successive restarts on watchdog
+  // timer.
+  base::TimeDelta minimum_duration_between_restarts_;
+
+  base::WeakPtrFactory<RuntimeAPI> weak_ptr_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(RuntimeAPI);
 };
 
 template <>
 void BrowserContextKeyedAPIFactory<RuntimeAPI>::DeclareFactoryDependencies();
 
 class RuntimeEventRouter {
  public:
   // Dispatches the onStartup event to all currently-loaded extensions.
   static void DispatchOnStartupEvent(content::BrowserContext* context,
@@ skipping 82 matching lines @@
 
 class RuntimeRestartFunction : public UIThreadExtensionFunction {
  public:
   DECLARE_EXTENSION_FUNCTION("runtime.restart", RUNTIME_RESTART)
 
  protected:
   ~RuntimeRestartFunction() override {}
   ResponseAction Run() override;
 };
 
+class RuntimeRestartOnWatchdogFunction : public UIThreadExtensionFunction {
+ public:
+  DECLARE_EXTENSION_FUNCTION("runtime.restartOnWatchdog",
+                             RUNTIME_RESTARTONWATCHDOG)
+
+ protected:
+  ~RuntimeRestartOnWatchdogFunction() override {}
+  ResponseAction Run() override;
+
+ private:
+  void OnWatchdogTimeout(bool success, const std::string& message);
+};
+
 class RuntimeGetPlatformInfoFunction : public UIThreadExtensionFunction {
  public:
   DECLARE_EXTENSION_FUNCTION("runtime.getPlatformInfo",
                              RUNTIME_GETPLATFORMINFO);
 
  protected:
   ~RuntimeGetPlatformInfoFunction() override {}
   ResponseAction Run() override;
 };
 
 class RuntimeGetPackageDirectoryEntryFunction
     : public UIThreadExtensionFunction {
  public:
   DECLARE_EXTENSION_FUNCTION("runtime.getPackageDirectoryEntry",
                              RUNTIME_GETPACKAGEDIRECTORYENTRY)
 
  protected:
   ~RuntimeGetPackageDirectoryEntryFunction() override {}
   ResponseAction Run() override;
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_API_RUNTIME_RUNTIME_API_H_