Home | History | Annotate | Download | only in extensions 
      1 // Copyright 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_EXTENSIONS_BLACKLIST_STATE_FETCHER_H_
      6 #define CHROME_BROWSER_EXTENSIONS_BLACKLIST_STATE_FETCHER_H_
      7 
      8 #include <set>
      9 #include <string>
     10 
     11 #include "base/callback.h"
     12 #include "base/memory/scoped_ptr.h"
     13 #include "base/memory/weak_ptr.h"
     14 #include "chrome/browser/safe_browsing/protocol_manager_helper.h"
     15 #include "extensions/browser/blacklist_state.h"
     16 #include "net/url_request/url_fetcher.h"
     17 #include "net/url_request/url_fetcher_delegate.h"
     18 
     19 namespace net {
     20 class URLRequestContextGetter;
     21 }
     22 
     23 namespace extensions {
     24 
     25 class BlacklistStateFetcher : public net::URLFetcherDelegate {
     26  public:
     27   typedef base::Callback<void(BlacklistState)> RequestCallback;
     28 
     29   BlacklistStateFetcher();
     30 
     31   virtual ~BlacklistStateFetcher();
     32 
     33   virtual void Request(const std::string& id, const RequestCallback& callback);
     34 
     35   void SetSafeBrowsingConfig(const SafeBrowsingProtocolConfig& config);
     36 
     37   void SetURLRequestContextForTest(
     38       net::URLRequestContextGetter* parent_request_context);
     39 
     40  protected:
     41   // net::URLFetcherDelegate interface.
     42   virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;
     43 
     44  private:
     45   typedef std::multimap<std::string, RequestCallback> CallbackMultiMap;
     46 
     47   GURL RequestUrl() const;
     48 
     49   void SaveRequestContext(
     50     const std::string& id,
     51     scoped_refptr<net::URLRequestContextGetter> request_context_getter);
     52 
     53   void SendRequest(const std::string& id);
     54 
     55   // ID for URLFetchers for testing.
     56   int url_fetcher_id_;
     57 
     58   scoped_ptr<SafeBrowsingProtocolConfig> safe_browsing_config_;
     59   scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
     60   scoped_refptr<net::URLRequestContextGetter> parent_request_context_for_test_;
     61 
     62   // Extension id by URLFetcher.
     63   std::map<const net::URLFetcher*, std::string> requests_;
     64 
     65   // Callbacks by extension ID.
     66   CallbackMultiMap callbacks_;
     67 
     68   base::WeakPtrFactory<BlacklistStateFetcher> weak_ptr_factory_;
     69 
     70   DISALLOW_COPY_AND_ASSIGN(BlacklistStateFetcher);
     71 };
     72 
     73 }  // namespace extensions
     74 
     75 #endif  // CHROME_BROWSER_EXTENSIONS_BLACKLIST_STATE_FETCHER_H_
     76 
     77