Add switches and apis in safebrowsing code to allow tests.

chrome/browser/safe_browsing/safe_browsing_service.cc

 // Copyright (c) 2009 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.
 //
 
 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
 
 #include "base/callback.h"
+#include "base/command_line.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/metrics/metrics_service.h"
 #include "chrome/browser/pref_service.h"
 #include "chrome/browser/profile_manager.h"
 #include "chrome/browser/safe_browsing/protocol_manager.h"
 #include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h"
 #include "chrome/browser/safe_browsing/safe_browsing_database.h"
 #include "chrome/browser/tab_contents/tab_util.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/chrome_thread.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/common/net/url_request_context_getter.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "net/base/registry_controlled_domain.h"
 
 #if defined(OS_WIN)
 #include "chrome/installer/util/browser_distribution.h"
 #endif
 
 using base::Time;
 using base::TimeDelta;
 
+// The default URL prefix where browser fetches chunk updates, hashes,
+// and reports malware.
+static const char* const kSbDefaultInfoURLPrefix =
+    "http://safebrowsing.clients.google.com/safebrowsing";
+
+// The default URL prefix where browser fetches MAC client key.
+static const char* const kSbDefaultMacKeyURLPrefix =
+    "https://sb-ssl.google.com/safebrowsing";
+
 static Profile* GetDefaultProfile() {
   FilePath user_data_dir;
   PathService::Get(chrome::DIR_USER_DATA, &user_data_dir);
   ProfileManager* profile_manager = g_browser_process->profile_manager();
   return profile_manager->GetDefaultProfile(user_data_dir);
 }
 
 SafeBrowsingService::SafeBrowsingService()
     : database_(NULL),
       protocol_manager_(NULL),
@@ skipping 181 matching lines @@
   DCHECK(enabled_);
   if (update_in_progress_) {
     update_in_progress_ = false;
     safe_browsing_thread_->message_loop()->PostTask(FROM_HERE,
         NewRunnableMethod(this,
                           &SafeBrowsingService::DatabaseUpdateFinished,
                           update_succeeded));
   }
 }
 
+bool SafeBrowsingService::IsUpdateInProgress() const {
+  DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO));
+  return update_in_progress_;
+}
+
 void SafeBrowsingService::OnBlockingPageDone(
     const std::vector<UnsafeResource>& resources,
     bool proceed) {
   for (std::vector<UnsafeResource>::const_iterator iter = resources.begin();
        iter != resources.end(); ++iter) {
     const UnsafeResource& resource = *iter;
     NotifyClientBlockingComplete(resource.client, proceed);
 
     if (proceed) {
       // Whitelist this domain and warning type for the given tab.
@@ skipping 92 matching lines @@
 #if defined(OS_WIN)
   BrowserDistribution* dist = BrowserDistribution::GetDistribution();
   std::string client_name(dist->GetSafeBrowsingName());
 #else
 #if defined(GOOGLE_CHROME_BUILD)
   std::string client_name("googlechrome");
 #else
   std::string client_name("chromium");
 #endif
 #endif
+  CommandLine* cmdline = CommandLine::ForCurrentProcess();
+  bool disable_auto_update = cmdline->HasSwitch(switches::kSbDisableAutoUpdate);
+  std::string info_url_prefix =
+      cmdline->HasSwitch(switches::kSbInfoURLPrefix) ?
+      cmdline->GetSwitchValueASCII(switches::kSbInfoURLPrefix) :
+      kSbDefaultInfoURLPrefix;
+  std::string mackey_url_prefix =
+      cmdline->HasSwitch(switches::kSbMacKeyURLPrefix) ?
+      cmdline->GetSwitchValueASCII(switches::kSbMacKeyURLPrefix) :
+      kSbDefaultMacKeyURLPrefix;
 
   protocol_manager_ = new SafeBrowsingProtocolManager(this,
                                                       client_name,
                                                       client_key,
                                                       wrapped_key,
-                                                      request_context_getter);
+                                                      request_context_getter,
+                                                      info_url_prefix,
+                                                      mackey_url_prefix,
+                                                      disable_auto_update);
 
   // Balance the reference added by Start().
   request_context_getter->Release();
 
   protocol_manager_->Initialize();
 }
 
 void SafeBrowsingService::OnIOShutdown() {
   DCHECK(ChromeThread::CurrentlyOn(ChromeThread::IO));
   if (!enabled_)
@@ skipping 391 matching lines @@
     std::vector<SBPrefix> prefix_hits;
     std::vector<SBFullHashResult> full_hits;
     database_->ContainsUrl(page_url, &list, &prefix_hits, &full_hits,
                            protocol_manager_->last_update());
     if (!full_hits.empty())
       return;
   }
 
   protocol_manager_->ReportMalware(malware_url, page_url, referrer_url);
 }