summaryrefslogtreecommitdiffstats
path: root/webkit/glue/resource_fetcher.h
blob: 34ced2efca818fe44606b65e51916c892bdc3f09 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
// Copyright (c) 2006-2008 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.
//
// A wrapper around ResourceHandle and ResourceHandleClient that simplifies
// the download of an HTTP object.  The interface is modeled after URLFetcher
// in the /chrome/browser.
//
// ResourceFetcher::Delegate::OnURLFetchComplete will be called async after
// the ResourceFetcher object is created.

#ifndef WEBKIT_GLUE_RESOURCE_FETCHER_H__
#define WEBKIT_GLUE_RESOURCE_FETCHER_H__

#include <string>

#include "base/basictypes.h"
#include "base/scoped_ptr.h"
#include "googleurl/src/gurl.h"

#pragma warning(push, 0)
#include "Frame.h"
#include "Timer.h"
#include "ResourceHandleClient.h"
#include "ResourceResponse.h"
#pragma warning(pop)

class GURL;
class WebCore::ResourceHandle;

class ResourceFetcher : public WebCore::ResourceHandleClient {
 public:
  class Delegate {
   public:
    // This will be called when the URL has been fetched, successfully or not.
    // If there is a failure, response and data will both be empty.
    // |response| and |data| are both valid until the URLFetcher instance is
    // destroyed.
    virtual void OnURLFetchComplete(const WebCore::ResourceResponse& response,
                                    const std::string& data) = 0;
  };

  // We need a frame and frame loader to make requests.
  ResourceFetcher(const GURL& url, WebCore::Frame* frame, Delegate* d);
  ~ResourceFetcher();

  // Stop the request and don't call the callback.
  void Cancel();

  bool completed() { return completed_; }

  // ResourceHandleClient methods
  virtual void didReceiveResponse(WebCore::ResourceHandle* resource_handle,
                                  const WebCore::ResourceResponse& response);

  virtual void didReceiveData(WebCore::ResourceHandle* resource_handle, 
                              const char* data, int length, int total_length);

  virtual void didFinishLoading(WebCore::ResourceHandle* resource_handle);

  virtual void didFail(WebCore::ResourceHandle* resource_handle, 
                       const WebCore::ResourceError& error);

 protected:
  // The parent ResourceHandle
  RefPtr<WebCore::ResourceHandle> loader_;

  // URL we're fetching
  GURL url_;

  // Callback when we're done
  Delegate* delegate_;

  // A copy of the original resource response
  WebCore::ResourceResponse response_;

  // Set to true once the request is compelte.
  bool completed_;

 private:
  // If we fail to start the request, we still want to finish async.
  typedef WebCore::Timer<ResourceFetcher> StartFailedTimer;

  // Start the actual download.
  void Start(WebCore::Frame* frame);

  // Callback function if Start fails.
  void StartFailed(StartFailedTimer* timer);

  // Timer for calling StartFailed async.
  scoped_ptr<StartFailedTimer> start_failed_timer_;

  // Buffer to hold the content from the server.
  std::string data_;
};

/////////////////////////////////////////////////////////////////////////////
// A resource fetcher with a timeout
class ResourceFetcherWithTimeout : public ResourceFetcher {
 public:
  ResourceFetcherWithTimeout(const GURL& url, WebCore::Frame* frame, double
                             timeout_secs, Delegate* d);
  virtual ~ResourceFetcherWithTimeout() {}

 private:
  typedef WebCore::Timer<ResourceFetcherWithTimeout> FetchTimer;

  // Callback for timer that limits how long we wait for the alternate error
  // page server.  If this timer fires and the request hasn't completed, we
  // kill the request.
  void TimeoutFired(FetchTimer* timer);

  // Limit how long we wait for the alternate error page server.
  scoped_ptr<FetchTimer> timeout_timer_;
};

#endif  // WEBKIT_GLUE_RESOURCE_FETCHER_H__