Post-cleanup settings reset.

chrome/browser/safe_browsing/chrome_cleaner/chrome_cleaner_controller_win.cc

 // Copyright 2017 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.
 
+// DO NOT SUBMIT. Please ignore changes to this file for the moment.
+
 #include "chrome/browser/safe_browsing/chrome_cleaner/chrome_cleaner_controller_win.h"
 
+#include "base/barrier_closure.h"
 #include "base/command_line.h"
 #include "base/logging.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/safe_browsing/chrome_cleaner/settings_resetter_win.h"
+#include "components/chrome_cleaner/public/constants/constants.h"
 #include "content/public/browser/browser_thread.h"
 
 namespace safe_browsing {
 
 namespace {
 
 // Global instance that is set and unset by ChromeCleanerController's
 // constructor and destructor.
 ChromeCleanerController* g_chrome_cleaner_controller = nullptr;
 
@@ skipping 69 matching lines @@
   DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
 
   observer_list_.RemoveObserver(observer);
 }
 
 void ChromeCleanerController::Cleanup() {
   DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
   if (state_ != State::kInfected)
     return;
 
+  // Tag all loaded profiles for settings reset post-cleanup.
+  for (Profile* profile :
+       g_browser_process->profile_manager()->GetLoadedProfiles()) {
+    TagProfileForResetting(profile);
+  }
+
   state_ = State::kCleaning;
   idle_reason_ = IdleReason::kInitialState;
   NotifyAllObservers();
   SetKeepAlive(true);
 
   // DO NOT SUBMIT
   // Temporary hack to fake scanning and cleaning work.
   one_shot_timer_for_testing_.Start(
       FROM_HERE, base::TimeDelta::FromSeconds(3),
       base::Bind(&ChromeCleanerController::OnCleanupFinished,
@@ skipping 21 matching lines @@
   g_chrome_cleaner_controller = this;
 }
 
 ChromeCleanerController::~ChromeCleanerController() {
   DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
   DCHECK_EQ(this, g_chrome_cleaner_controller);
 
   g_chrome_cleaner_controller = nullptr;
 }
 
+constexpr char kPostRebootResetPrefName[] =
+    "safebrowsing.chrome_cleaner.post_reboot_reset";
+
 void ChromeCleanerController::OnCleanupFinished() {
   DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
 
   // DO NOT SUBMIT
   if (base::CommandLine::ForCurrentProcess()->HasSwitch("reboot-required")) {
     state_ = State::kRebootRequired;
     idle_reason_ = IdleReason::kInitialState;
+    NotifyAllObservers();
+    SetKeepAlive(true);
   } else {
     state_ = State::kIdle;
     idle_reason_ = GetIdleReasonFromCommandLine();
+    ResetPostCleanupSettingsIfTagged(
+        g_browser_process->profile_manager()->GetLoadedProfiles(),
+        base::BindOnce(&ChromeCleanerController::FinalizeCleanup,
+                       base::RetainedRef(this)));
   }
+}
+
+void ChromeCleanerController::FinalizeCleanup() {
+  DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
 
   NotifyAllObservers();
   SetKeepAlive(false);
 }
 
 void ChromeCleanerController::NotifyObserver(Observer* observer) const {
   DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
   switch (state_) {
     case State::kIdle:
       observer->OnIdle(idle_reason_);
@@ skipping 26 matching lines @@
     return;
 
   keep_alive_ = keep_alive;
   if (keep_alive_)
     AddRef();
   else
     Release();
 }
 
 }  // namespace safe_browsing