Reload force-installed extensions on crash/force-close

chrome/browser/background/background_contents_service.h

 // Copyright (c) 2012 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 CHROME_BROWSER_BACKGROUND_BACKGROUND_CONTENTS_SERVICE_H_
 #define CHROME_BROWSER_BACKGROUND_BACKGROUND_CONTENTS_SERVICE_H_
 
 #include <map>
+#include <queue>
+#include <set>
 #include <string>
 #include <vector>
 
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
+#include "base/memory/weak_ptr.h"
+#include "base/time/time.h"
 #include "chrome/browser/tab_contents/background_contents.h"
+#include "chrome/common/extensions/extension.h"
 #include "components/browser_context_keyed_service/browser_context_keyed_service.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/common/window_container_type.h"
 #include "ui/base/window_open_disposition.h"
 #include "url/gurl.h"
 
 class CommandLine;
 class PrefService;
 class Profile;
@@ skipping 20 matching lines @@
 // It is also responsible for tracking the association between
 // BackgroundContents and their parent app, and shutting them down when the
 // parent app is unloaded.
 class BackgroundContentsService : private content::NotificationObserver,
                                   public BackgroundContents::Delegate,
                                   public BrowserContextKeyedService {
  public:
   BackgroundContentsService(Profile* profile, const CommandLine* command_line);
   virtual ~BackgroundContentsService();
 
+  // Allows tests to reduce the time between a force-installed app/extension
+  // crashing and when we reload it, and the amount of time we wait for further
+  // crashes before showing a balloon saying the app/extension is misbehaving.
+  static void SetCrashDelaysForForceInstalledAppsAndExtensionsForTesting(
+      int restart_delay_in_ms, int crash_window_in_ms);
+
   // Returns the BackgroundContents associated with the passed application id,
   // or NULL if none.
   BackgroundContents* GetAppBackgroundContents(const string16& appid);
 
   // Returns true if there's a registered BackgroundContents for this app. It
   // is possible for this routine to return true when GetAppBackgroundContents()
   // returns false, if the BackgroundContents closed due to the render process
   // crashing.
   bool HasRegisteredBackgroundContents(const string16& appid);
 
@@ skipping 49 matching lines @@
                            KillBGContents);
 
   // Registers for various notifications.
   void StartObserving(Profile* profile);
 
   // content::NotificationObserver implementation.
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE;
 
+  // Restarts a force-installed app/extension after a crash. Notifies user if
+  // crash recurs frequently.
+  void RestartForceInstalledExtensionOnCrash(
+      const extensions::Extension* extension, Profile* profile);
+
+  // Reloads the extension with the given id.
+  void ReloadExtension(const std::string& extension_id, Profile* profile);
+
   // Loads all registered BackgroundContents at startup.
   void LoadBackgroundContentsFromPrefs(Profile* profile);
 
   // Load a BackgroundContent; the settings are read from the provided
   // dictionary.
   void LoadBackgroundContentsFromDictionary(
       Profile* profile,
       const std::string& extension_id,
       const base::DictionaryValue* contents);
 
@@ skipping 29 matching lines @@
   void ShutdownAssociatedBackgroundContents(const string16& appid);
 
   // Returns true if this BackgroundContents is in the contents_list_.
   bool IsTracked(BackgroundContents* contents) const;
 
   // Sends out a notification when our association of background contents with
   // apps may have changed (used by BackgroundApplicationListModel to update the
   // set of background apps as new background contents are opened/closed).
   void SendChangeNotification(Profile* profile);
 
+  // Delay (in ms) before restarting a force-installed extension that crashed.
+  static int restart_delay_in_ms_;
+
+  // When a force-installed app/extension crashes, we check if it's in a crash/
+  // reload loop by checking if the number of crashes exceeds a threshold in a
+  // given time window. The duration of that window is given by:
+  // kMisbehaveCrashCountThreshold * (restart_delay_in_ms + crash_window_in_ms)
+  static int crash_window_in_ms_;
+
   // PrefService used to store list of background pages (or NULL if this is
   // running under an incognito profile).
   PrefService* prefs_;
   content::NotificationRegistrar registrar_;
 
   // Information we track about each BackgroundContents.
   struct BackgroundContentsInfo {
     // The BackgroundContents whose information we are tracking.
     BackgroundContents* contents;
     // The name of the top level frame for this BackgroundContents.
     string16 frame_name;
   };
 
   // Map associating currently loaded BackgroundContents with their parent
   // applications.
   // Key: application id
   // Value: BackgroundContentsInfo for the BC associated with that application
   typedef std::map<string16, BackgroundContentsInfo> BackgroundContentsMap;
   BackgroundContentsMap contents_map_;
 
+  // Map associating IDs of force-installed extensions/apps with their most
+  // recent crash timestamps.
+  // Key: app/extension id.
+  // Value: queue containing up to 5 most recent crash timestamps.
+  std::map<std::string, std::queue<base::TimeTicks> > extension_crashlog_map_;
+
+  // Map containing ids of force-installed apps/extensions for which we have
+  // already shown an 'App/Extension is misbehaving' balloon.
+  std::set<std::string> misbehaving_extensions_;
+
+  base::WeakPtrFactory<BackgroundContentsService> weak_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(BackgroundContentsService);
 };
 
 #endif  // CHROME_BROWSER_BACKGROUND_BACKGROUND_CONTENTS_SERVICE_H_