net/proxy/proxy_script_fetcher_impl.h

   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 NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_
   6 #define NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/compiler_specific.h"
  12 #include "base/memory/ref_counted.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/memory/weak_ptr.h"
  15 #include "base/strings/string16.h"
  16 #include "base/time/time.h"
  17 #include "net/proxy/proxy_script_fetcher.h"
  18 #include "net/url_request/url_request.h"
  19
  20 class GURL;
  21
  22 namespace net {
  23
  24 class URLRequestContext;
  25
  26 // Implementation of ProxyScriptFetcher that downloads scripts using the
  27 // specified request context.
  28 class NET_EXPORT ProxyScriptFetcherImpl : public ProxyScriptFetcher,
  29                                           public URLRequest::Delegate {
  30  public:
  31   // Creates a ProxyScriptFetcher that issues requests through
  32   // |url_request_context|. |url_request_context| must remain valid for the
  33   // lifetime of ProxyScriptFetcherImpl.
  34   // Note that while a request is in progress, we will be holding a reference
  35   // to |url_request_context|. Be careful not to create cycles between the
  36   // fetcher and the context; you can break such cycles by calling Cancel().
  37   explicit ProxyScriptFetcherImpl(URLRequestContext* url_request_context);
  38
  39   ~ProxyScriptFetcherImpl() override;
  40
  41   // Used by unit-tests to modify the default limits.
  42   base::TimeDelta SetTimeoutConstraint(base::TimeDelta timeout);
  43   size_t SetSizeConstraint(size_t size_bytes);
  44
  45   void OnResponseCompleted(URLRequest* request);
  46
  47   // ProxyScriptFetcher methods:
  48   int Fetch(const GURL& url,
  49             base::string16* text,
  50             const CompletionCallback& callback) override;
  51   void Cancel() override;
  52   URLRequestContext* GetRequestContext() const override;
  53
  54   // URLRequest::Delegate methods:
  55   void OnAuthRequired(URLRequest* request,
  56                       AuthChallengeInfo* auth_info) override;
  57   void OnSSLCertificateError(URLRequest* request,
  58                              const SSLInfo& ssl_info,
  59                              bool is_hsts_ok) override;
  60   void OnResponseStarted(URLRequest* request) override;
  61   void OnReadCompleted(URLRequest* request, int num_bytes) override;
  62
  63  private:
  64   enum { kBufSize = 4096 };
  65
  66   // Read more bytes from the response.
  67   void ReadBody(URLRequest* request);
  68
  69   // Handles a response from Read(). Returns true if we should continue trying
  70   // to read. |num_bytes| is 0 for EOF, and < 0 on errors.
  71   bool ConsumeBytesRead(URLRequest* request, int num_bytes);
  72
  73   // Called once the request has completed to notify the caller of
  74   // |response_code_| and |response_text_|.
  75   void FetchCompleted();
  76
  77   // Clear out the state for the current request.
  78   void ResetCurRequestState();
  79
  80   // Callback for time-out task of request with id |id|.
  81   void OnTimeout(int id);
  82
  83   // The context used for making network requests.
  84   URLRequestContext* const url_request_context_;
  85
  86   // Buffer that URLRequest writes into.
  87   scoped_refptr<IOBuffer> buf_;
  88
  89   // The next ID to use for |cur_request_| (monotonically increasing).
  90   int next_id_;
  91
  92   // The current (in progress) request, or NULL.
  93   scoped_ptr<URLRequest> cur_request_;
  94
  95   // State for current request (only valid when |cur_request_| is not NULL):
  96
  97   // Unique ID for the current request.
  98   int cur_request_id_;
  99
 100   // Callback to invoke on completion of the fetch.
 101   CompletionCallback callback_;
 102
 103   // Holds the error condition that was hit on the current request, or OK.
 104   int result_code_;
 105
 106   // Holds the bytes read so far. Will not exceed |max_response_bytes|.
 107   std::string bytes_read_so_far_;
 108
 109   // This buffer is owned by the owner of |callback|, and will be filled with
 110   // UTF16 response on completion.
 111   base::string16* result_text_;
 112
 113   // The maximum number of bytes to allow in responses.
 114   size_t max_response_bytes_;
 115
 116   // The maximum amount of time to wait for download to complete.
 117   base::TimeDelta max_duration_;
 118
 119   // The time that the fetch started.
 120   base::Time fetch_start_time_;
 121
 122   // Factory for creating the time-out task. This takes care of revoking
 123   // outstanding tasks when |this| is deleted.
 124   base::WeakPtrFactory<ProxyScriptFetcherImpl> weak_factory_;
 125
 126   DISALLOW_COPY_AND_ASSIGN(ProxyScriptFetcherImpl);
 127 };
 128
 129 }  // namespace net
 130
 131 #endif  // NET_PROXY_PROXY_SCRIPT_FETCHER_IMPL_H_