Make SDCH dictionaries persistent across browser restart.

net/url_request/sdch_dictionary_fetcher.h

 // Copyright 2014 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.
 
 // TODO(rdsmith): This class needs to delegate URLRequest::Delegate methods
 // to the net/ embedder for correct implementation of authentication.
 // Specifically, this class needs the embedder to provide functionality
 // corresponding to
 // URLRequest::Delegate::{OnAuthRequired,OnCertificateRequested}.
 
@@ skipping 27 matching lines @@
 // attempting to fetch each dictionary once.
 class NET_EXPORT SdchDictionaryFetcher : public URLRequest::Delegate,
                                          public base::NonThreadSafe {
  public:
   typedef base::Callback<void(const std::string& dictionary_text,
                               const GURL& dictionary_url,
                               const BoundNetLog& net_log)>
       OnDictionaryFetchedCallback;
 
   // The consumer must guarantee that |*context| outlives this object.
-  // |callback| will be called on successful dictionary fetch
-  // requested through Schedule().  |callback| will not be called
-  // after object destruction.
-  SdchDictionaryFetcher(URLRequestContext* context,
-                        const OnDictionaryFetchedCallback& callback);
+  explicit SdchDictionaryFetcher(URLRequestContext* context);
   ~SdchDictionaryFetcher() override;
 
-  // Request a new dictionary fetch.
-  void Schedule(const GURL& dictionary_url);
+  // Request a new dictionary fetch.  The callback will be called
+  // only if the dictionary is successfully fetched. Returns true if a
+  // request for |dictionary_url| has been scheduled, and false otherwise.
+  virtual bool Schedule(const GURL& dictionary_url,
+                        const OnDictionaryFetchedCallback& callback);
+
+  // Request a dictionary fetch from cache only.  The callback will be called
+  // only if the dictionary is successfully fetched. Returns true if a request
+  // for |dictionary_url| has been scheduled, and false otherwise.
+  virtual bool ScheduleReload(const GURL& dictionary_url,
+                              const OnDictionaryFetchedCallback& callback);
 
   // Cancel any in-progress requests.
-  void Cancel();
+  virtual void Cancel();
 
   // Implementation of URLRequest::Delegate methods.
   void OnResponseStarted(URLRequest* request) override;
   void OnReadCompleted(URLRequest* request, int bytes_read) override;
 
  private:
   enum State {
     STATE_NONE,
     STATE_SEND_REQUEST,
     STATE_SEND_REQUEST_COMPLETE,
     STATE_READ_BODY,
     STATE_READ_BODY_COMPLETE,
     STATE_REQUEST_COMPLETE,
   };
 
+  // Represents an entry in the queue of SDCH dictionaries to fetch. |url| is
+  // the URL to fetch from, |download_only_from_cache| is true if the fetch
+  // should only be served from cache, and |callback| is called when this queued
+  // fetch is complete.
+  struct QueuedFetchInfo {
+    QueuedFetchInfo();
+    QueuedFetchInfo(const GURL& url,
+                    bool download_only_from_cache,
+                    const OnDictionaryFetchedCallback& callback);
+    QueuedFetchInfo(const QueuedFetchInfo& rhs);
+    QueuedFetchInfo& operator=(const QueuedFetchInfo& rhs);
+
+    ~QueuedFetchInfo();
+
+    GURL url;
+    bool download_only_from_cache;
+    OnDictionaryFetchedCallback callback;
+  };
+
+  // Schedule implementation. Returns true if a request for |dictionary_url| has
+  // been added to the queue, and false otherwise.
+  bool ScheduleInternal(const GURL& dictionary_url,
+                        bool reload,
+                        const OnDictionaryFetchedCallback& callback);
+
+  // Null out the current request and push the state machine to the
+  // next request, if any.
+  void ResetRequest();
+
   // State machine implementation.
   int DoLoop(int rv);
   int DoSendRequest(int rv);
   int DoSendRequestComplete(int rv);
   int DoReadBody(int rv);
   int DoReadBodyComplete(int rv);
   int DoCompleteRequest(int rv);
 
   State next_state_;
   bool in_loop_;
 
   // A queue of URLs that are being used to download dictionaries.
-  std::queue<GURL> fetch_queue_;
+  std::queue<QueuedFetchInfo> fetch_queue_;
 
-  // The request and buffer used for getting the current dictionary
-  // Both are null when a fetch is not in progress.
+  // The request, buffer, and consumer supplied data used for getting
+  // the current dictionary.  All are null when a fetch is not in progress.
   scoped_ptr<URLRequest> current_request_;
   scoped_refptr<IOBuffer> buffer_;
+  OnDictionaryFetchedCallback current_callback_;
 
   // The currently accumulating dictionary.
   std::string dictionary_;
 
   // Althought the SDCH spec does not preclude a server from using a single URL
   // to load several distinct dictionaries (by telling a client to load a
   // dictionary from an URL several times), current implementations seem to have
   // that 1-1 relationship (i.e., each URL points at a single dictionary, and
   // the dictionary content does not change over time, and hence is not worth
   // trying to load more than once). In addition, some dictionaries prove
   // unloadable only after downloading them (because they are too large?  ...or
   // malformed?). As a protective element, Chromium will *only* load a
   // dictionary at most once from a given URL (so that it doesn't waste
   // bandwidth trying repeatedly).
   // The following set lists all the dictionary URLs that we've tried to load,
   // so that we won't try to load from an URL more than once.
   // TODO(jar): Try to augment the SDCH proposal to include this restiction.
   std::set<GURL> attempted_load_;
 
   // Store the URLRequestContext associated with the owning SdchManager for
   // use while fetching.
   URLRequestContext* const context_;
 
-  const OnDictionaryFetchedCallback dictionary_fetched_callback_;
-
   base::WeakPtrFactory<SdchDictionaryFetcher> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(SdchDictionaryFetcher);
 };
 
 }  // namespace net
 
 #endif  // NET_URL_REQUEST_SDCH_DICTIONARY_FETCHER_H_