chrome/browser/profile_resetter/profile_resetter.h

   1 // Copyright (c) 2013 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
   5 #ifndef CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_
   6 #define CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_
   7
   8 #include <utility>
   9 #include <vector>
  10
  11 #include "base/basictypes.h"
  12 #include "base/callback.h"
  13 #include "base/files/file_path.h"
  14 #include "base/memory/ref_counted.h"
  15 #include "base/memory/scoped_ptr.h"
  16 #include "base/memory/weak_ptr.h"
  17 #include "base/strings/string16.h"
  18 #include "base/threading/non_thread_safe.h"
  19 #include "chrome/browser/browsing_data/browsing_data_remover.h"
  20 #include "chrome/browser/profile_resetter/brandcoded_default_settings.h"
  21 #include "components/search_engines/template_url_service.h"
  22
  23 class Profile;
  24
  25 namespace base {
  26 class CancellationFlag;
  27 }
  28
  29 // This class allows resetting certain aspects of a profile to default values.
  30 // It is used in case the profile has been damaged due to malware or bad user
  31 // settings.
  32 class ProfileResetter : public base::NonThreadSafe,
  33                         public BrowsingDataRemover::Observer {
  34  public:
  35   // Flags indicating what aspects of a profile shall be reset.
  36   enum Resettable {
  37     DEFAULT_SEARCH_ENGINE = 1 << 0,
  38     HOMEPAGE = 1 << 1,
  39     CONTENT_SETTINGS = 1 << 2,
  40     COOKIES_AND_SITE_DATA = 1 << 3,
  41     EXTENSIONS = 1 << 4,
  42     STARTUP_PAGES = 1 << 5,
  43     PINNED_TABS = 1 << 6,
  44     SHORTCUTS = 1 << 7,
  45     // Update ALL if you add new values and check whether the type of
  46     // ResettableFlags needs to be enlarged.
  47     ALL = DEFAULT_SEARCH_ENGINE | HOMEPAGE | CONTENT_SETTINGS |
  48           COOKIES_AND_SITE_DATA | EXTENSIONS | STARTUP_PAGES | PINNED_TABS |
  49           SHORTCUTS
  50   };
  51
  52   // Bit vector for Resettable enum.
  53   typedef uint32 ResettableFlags;
  54
  55   static_assert(sizeof(ResettableFlags) == sizeof(Resettable),
  56                 "ResettableFlags should be the same size as Resettable");
  57
  58   explicit ProfileResetter(Profile* profile);
  59   ~ProfileResetter() override;
  60
  61   // Resets |resettable_flags| and calls |callback| on the UI thread on
  62   // completion. |default_settings| allows the caller to specify some default
  63   // settings. |default_settings| shouldn't be NULL. |accepted_send_feedback|
  64   // identifies whether the user accepted to send feedback or not.
  65   void Reset(ResettableFlags resettable_flags,
  66              scoped_ptr<BrandcodedDefaultSettings> master_settings,
  67              bool accepted_send_feedback,
  68              const base::Closure& callback);
  69
  70   bool IsActive() const;
  71
  72  private:
  73   // Marks |resettable| as done and triggers |callback_| if all pending jobs
  74   // have completed.
  75   void MarkAsDone(Resettable resettable);
  76
  77   void ResetDefaultSearchEngine();
  78   void ResetHomepage();
  79   void ResetContentSettings();
  80   void ResetCookiesAndSiteData();
  81   void ResetExtensions();
  82   void ResetStartupPages();
  83   void ResetPinnedTabs();
  84   void ResetShortcuts();
  85
  86   // BrowsingDataRemover::Observer:
  87   void OnBrowsingDataRemoverDone() override;
  88
  89   // Callback for when TemplateURLService has loaded.
  90   void OnTemplateURLServiceLoaded();
  91
  92   Profile* const profile_;
  93   scoped_ptr<BrandcodedDefaultSettings> master_settings_;
  94   TemplateURLService* template_url_service_;
  95
  96   // Flags of a Resetable indicating which reset operations we are still waiting
  97   // for.
  98   ResettableFlags pending_reset_flags_;
  99
 100   // Called on UI thread when reset has been completed.
 101   base::Closure callback_;
 102
 103   // If non-null it means removal is in progress. BrowsingDataRemover takes care
 104   // of deleting itself when done.
 105   BrowsingDataRemover* cookies_remover_;
 106
 107   scoped_ptr<TemplateURLService::Subscription> template_url_service_sub_;
 108
 109   base::WeakPtrFactory<ProfileResetter> weak_ptr_factory_;
 110
 111   DISALLOW_COPY_AND_ASSIGN(ProfileResetter);
 112 };
 113
 114 // Path to shortcut and command line arguments.
 115 typedef std::pair<base::FilePath, base::string16> ShortcutCommand;
 116
 117 typedef base::RefCountedData<base::CancellationFlag> SharedCancellationFlag;
 118
 119 // On Windows returns all the shortcuts which launch Chrome and corresponding
 120 // arguments. |cancel| can be passed to abort the operation earlier.
 121 // Call on FILE thread.
 122 std::vector<ShortcutCommand> GetChromeLaunchShortcuts(
 123     const scoped_refptr<SharedCancellationFlag>& cancel);
 124
 125 #endif  // CHROME_BROWSER_PROFILE_RESETTER_PROFILE_RESETTER_H_