src/content/renderer/fetchers/resource_fetcher_impl.h

   1 // Copyright (c) 2011 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 CONTENT_RENDERER_FETCHERS_RESOURCE_FETCHER_H_
   6 #define CONTENT_RENDERER_FETCHERS_RESOURCE_FETCHER_H_
   7
   8 #include <string>
   9
  10 #include "base/basictypes.h"
  11 #include "base/callback.h"
  12 #include "base/compiler_specific.h"
  13 #include "base/memory/scoped_ptr.h"
  14 #include "base/timer/timer.h"
  15 #include "content/public/renderer/resource_fetcher.h"
  16 #include "third_party/WebKit/public/platform/WebURLLoaderClient.h"
  17 #include "third_party/WebKit/public/platform/WebURLRequest.h"
  18 #include "third_party/WebKit/public/platform/WebURLResponse.h"
  19
  20 class GURL;
  21
  22 namespace blink {
  23 class WebFrame;
  24 class WebURLLoader;
  25 struct WebURLError;
  26 }
  27
  28 namespace content {
  29
  30 class ResourceFetcherImpl : public ResourceFetcher,
  31                             public blink::WebURLLoaderClient {
  32  public:
  33   // ResourceFetcher implementation:
  34   virtual void SetMethod(const std::string& method) OVERRIDE;
  35   virtual void SetBody(const std::string& body) OVERRIDE;
  36   virtual void SetHeader(const std::string& header,
  37                          const std::string& value) OVERRIDE;
  38   virtual void Start(blink::WebFrame* frame,
  39                      blink::WebURLRequest::RequestContext request_context,
  40                      blink::WebURLRequest::FrameType frame_type,
  41                      const Callback& callback) OVERRIDE;
  42   virtual void SetTimeout(const base::TimeDelta& timeout) OVERRIDE;
  43
  44  private:
  45   friend class ResourceFetcher;
  46
  47   explicit ResourceFetcherImpl(const GURL& url);
  48
  49   virtual ~ResourceFetcherImpl();
  50
  51   void RunCallback(const blink::WebURLResponse& response,
  52                    const std::string& data);
  53
  54   // Callback for timer that limits how long we wait for the server.  If this
  55   // timer fires and the request hasn't completed, we kill the request.
  56   void TimeoutFired();
  57
  58   // WebURLLoaderClient methods:
  59   virtual void willSendRequest(
  60       blink::WebURLLoader* loader, blink::WebURLRequest& new_request,
  61       const blink::WebURLResponse& redirect_response);
  62   virtual void didSendData(
  63       blink::WebURLLoader* loader, unsigned long long bytes_sent,
  64       unsigned long long total_bytes_to_be_sent);
  65   virtual void didReceiveResponse(
  66       blink::WebURLLoader* loader, const blink::WebURLResponse& response);
  67   virtual void didReceiveCachedMetadata(
  68       blink::WebURLLoader* loader, const char* data, int data_length);
  69
  70   virtual void didReceiveData(
  71       blink::WebURLLoader* loader, const char* data, int data_length,
  72       int encoded_data_length);
  73   virtual void didFinishLoading(
  74       blink::WebURLLoader* loader, double finishTime,
  75       int64_t total_encoded_data_length);
  76   virtual void didFail(
  77       blink::WebURLLoader* loader, const blink::WebURLError& error);
  78
  79   scoped_ptr<blink::WebURLLoader> loader_;
  80
  81   // Request to send.  Released once Start() is called.
  82   blink::WebURLRequest request_;
  83
  84   // Set to true once the request is complete.
  85   bool completed_;
  86
  87   // Buffer to hold the content from the server.
  88   std::string data_;
  89
  90   // A copy of the original resource response.
  91   blink::WebURLResponse response_;
  92
  93   // Callback when we're done.
  94   Callback callback_;
  95
  96   // Buffer to hold metadata from the cache.
  97   std::string metadata_;
  98
  99   // Limit how long to wait for the server.
 100   base::OneShotTimer<ResourceFetcherImpl> timeout_timer_;
 101
 102   DISALLOW_COPY_AND_ASSIGN(ResourceFetcherImpl);
 103 };
 104
 105 }  // namespace content
 106
 107 #endif  // CONTENT_RENDERER_FETCHERS_RESOURCE_FETCHER_H_