Add SafeBrowsingURLRequestContext which stores cookies.

chrome/browser/safe_browsing/safe_browsing_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.
 //
 // The Safe Browsing service is responsible for downloading anti-phishing and
 // anti-malware tables and checking urls against them.
 
 #ifndef CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_
 #define CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_
 #pragma once
@@ skipping 17 matching lines @@
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "googleurl/src/gurl.h"
 
 class MalwareDetails;
 class PrefChangeRegistrar;
 class PrefService;
 class SafeBrowsingDatabase;
 class SafeBrowsingProtocolManager;
 class SafeBrowsingServiceFactory;
+class SafeBrowsingURLRequestContextGetter;
 
 namespace base {
 class Thread;
 }
 
 namespace net {
+class URLRequestContext;
 class URLRequestContextGetter;
 }
 
 namespace safe_browsing {
 class ClientSideDetectionService;
 class DownloadProtectionService;
 }
 
 // Construction needs to happen on the main thread.
 class SafeBrowsingService
@@ skipping 280 matching lines @@
   struct QueuedCheck {
     Client* client;
     GURL url;
     base::TimeTicks start;  // When check was queued.
   };
 
   friend struct content::BrowserThread::DeleteOnThread<
       content::BrowserThread::UI>;
   friend class base::DeleteHelper<SafeBrowsingService>;
   friend class SafeBrowsingServiceTest;
+  friend class SafeBrowsingURLRequestContextGetter;
 
-  // Called to initialize objects that are used on the io_thread.
-  void OnIOInitialize(const std::string& client_key,
-                      const std::string& wrapped_key,
-                      net::URLRequestContextGetter* request_context_getter);
+  void InitURLRequestContextOnIOThread(
+      net::URLRequestContextGetter* system_url_request_context_getter);
 
-  // Called to shutdown operations on the io_thread.
-  void OnIOShutdown();
+  void DestroyURLRequestContextOnIOThread();
+
+  // Called to initialize objects that are used on the io_thread.  This may be
+  // called multiple times during the life of the SafeBrowsingService.
+  void StartOnIOThread(const std::string& client_key,
+                      const std::string& wrapped_key);

[willchan no longer on Chromium, 2012/02/04 22:30:20]

indentation off

[mattm, 2012/02/07 01:28:01]

Done.

+
+  // Called to shutdown operations on the io_thread. This may be called multiple
+  // times during the life of the SafeBrowsingService.
+  void StopOnIOThread();
 
   // Returns whether |database_| exists and is accessible.
   bool DatabaseAvailable() const;
 
   // Called on the IO thread.  If the database does not exist, queues up a call
   // on the db thread to create it.  Returns whether the database is available.
   //
   // Note that this is only needed outside the db thread, since functions on the
   // db thread can call GetDatabase() directly.
   bool MakeDatabaseAvailable();
@@ skipping 134 matching lines @@
   // Used for issuing only one GetHash request for a given prefix.
   GetHashRequests gethash_requests_;
 
   // The persistent database.  We don't use a scoped_ptr because it
   // needs to be destructed on a different thread than this object.
   SafeBrowsingDatabase* database_;
 
   // Lock used to prevent possible data races due to compiler optimizations.
   mutable base::Lock database_lock_;
 
+  // The SafeBrowsingURLRequestContextGetter used to access
+  // |url_request_context_|.
+  scoped_refptr<net::URLRequestContextGetter>
+      url_request_context_getter_;
+
+  // The SafeBrowsingURLRequestContext.
+  scoped_refptr<net::URLRequestContext> url_request_context_;
+
   // Handles interaction with SafeBrowsing servers.
   SafeBrowsingProtocolManager* protocol_manager_;
 
   // Only access this whitelist from the UI thread.
   std::vector<WhiteListedEntry> white_listed_entries_;
 
   // Whether the service is running. 'enabled_' is used by SafeBrowsingService
   // on the IO thread during normal operations.
   bool enabled_;
 
@@ skipping 63 matching lines @@
 class SafeBrowsingServiceFactory {
  public:
   SafeBrowsingServiceFactory() { }
   virtual ~SafeBrowsingServiceFactory() { }
   virtual SafeBrowsingService* CreateSafeBrowsingService() = 0;
  private:
   DISALLOW_COPY_AND_ASSIGN(SafeBrowsingServiceFactory);
 };
 
 #endif  // CHROME_BROWSER_SAFE_BROWSING_SAFE_BROWSING_SERVICE_H_