Home | History | Annotate | Download | only in safe_browsing 
      1 // Copyright (c) 2012 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_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
      6 #define CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
      7 
      8 // This class gets redirect chain for urls from the history service.
      9 
     10 #include <string>
     11 #include <vector>
     12 
     13 #include "base/callback.h"
     14 #include "base/containers/hash_tables.h"
     15 #include "base/memory/linked_ptr.h"
     16 #include "base/memory/ref_counted.h"
     17 #include "base/sequenced_task_runner_helpers.h"
     18 #include "chrome/browser/history/history_service.h"
     19 #include "content/public/browser/browser_thread.h"
     20 #include "content/public/browser/notification_observer.h"
     21 #include "content/public/browser/notification_registrar.h"
     22 #include "net/base/completion_callback.h"
     23 
     24 namespace safe_browsing {
     25 typedef std::vector<GURL> RedirectChain;
     26 }
     27 
     28 class MalwareDetailsRedirectsCollector
     29     : public base::RefCountedThreadSafe<
     30           MalwareDetailsRedirectsCollector,
     31           content::BrowserThread::DeleteOnUIThread>,
     32       public content::NotificationObserver {
     33  public:
     34   explicit MalwareDetailsRedirectsCollector(Profile* profile);
     35 
     36   // Collects urls' redirects chain information from the history service.
     37   // We get access to history service via web_contents in UI thread.
     38   // Notice the callback will be posted to the IO thread.
     39   void StartHistoryCollection(const std::vector<GURL>& urls,
     40                               const base::Closure& callback);
     41 
     42   // Returns whether or not StartCacheCollection has been called.
     43   bool HasStarted() const;
     44 
     45   // Returns the redirect urls we get from history service
     46   const std::vector<safe_browsing::RedirectChain>& GetCollectedUrls() const;
     47 
     48   // content::NotificationObserver
     49   virtual void Observe(int type,
     50                        const content::NotificationSource& source,
     51                        const content::NotificationDetails& details) OVERRIDE;
     52 
     53  private:
     54   friend struct content::BrowserThread::DeleteOnThread<
     55       content::BrowserThread::UI>;
     56   friend class base::DeleteHelper<MalwareDetailsRedirectsCollector>;
     57 
     58   virtual ~MalwareDetailsRedirectsCollector();
     59 
     60   void StartGetRedirects(const std::vector<GURL>& urls);
     61   void GetRedirects(const GURL& url);
     62   void OnGotQueryRedirectsTo(HistoryService::Handle handle,
     63                              GURL url,
     64                              bool success,
     65                              history::RedirectList* redirect_list);
     66 
     67   // Posts the callback method back to IO thread when redirects collecting
     68   // is all done.
     69   void AllDone();
     70 
     71   Profile* profile_;
     72   CancelableRequestConsumer request_consumer_;
     73 
     74   // Method we call when we are done. The caller must be alive for the
     75   // whole time, we are modifying its state (see above).
     76   base::Closure callback_;
     77 
     78   // Sets to true once StartHistoryCollection is called
     79   bool has_started_;
     80 
     81   // The urls we need to get redirects for
     82   std::vector<GURL> urls_;
     83   // The iterator goes over urls_
     84   std::vector<GURL>::iterator urls_it_;
     85   // The collected directs from history service
     86   std::vector<safe_browsing::RedirectChain> redirects_urls_;
     87 
     88   content::NotificationRegistrar registrar_;
     89 
     90   DISALLOW_COPY_AND_ASSIGN(MalwareDetailsRedirectsCollector);
     91 };
     92 
     93 #endif  // CHROME_BROWSER_SAFE_BROWSING_MALWARE_DETAILS_HISTORY_H_
     94