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.
5 #ifndef NET_URL_REQUEST_URL_REQUEST_JOB_H_
6 #define NET_URL_REQUEST_URL_REQUEST_JOB_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/system_monitor/system_monitor.h"
16 #include "base/time.h"
17 #include "googleurl/src/gurl.h"
18 #include "net/base/filter.h"
19 #include "net/base/host_port_pair.h"
20 #include "net/base/load_states.h"
21 #include "net/base/net_export.h"
25 class AuthChallengeInfo
;
26 class AuthCredentials
;
29 class HttpRequestHeaders
;
30 class HttpResponseInfo
;
32 class SSLCertRequestInfo
;
36 class URLRequestStatus
;
37 class X509Certificate
;
39 class NET_EXPORT URLRequestJob
: public base::RefCounted
<URLRequestJob
>,
40 public base::SystemMonitor::PowerObserver
{
42 explicit URLRequestJob(URLRequest
* request
);
44 // Returns the request that owns this job. THIS POINTER MAY BE NULL if the
45 // request was destroyed.
46 URLRequest
* request() const {
50 // Sets the upload data, most requests have no upload data, so this is a NOP.
51 // Job types supporting upload data will override this.
52 virtual void SetUpload(UploadData
* upload
);
54 // Sets extra request headers for Job types that support request headers.
55 virtual void SetExtraRequestHeaders(const HttpRequestHeaders
& headers
);
57 // If any error occurs while starting the Job, NotifyStartError should be
59 // This helps ensure that all errors follow more similar notification code
60 // paths, which should simplify testing.
61 virtual void Start() = 0;
63 // This function MUST somehow call NotifyDone/NotifyCanceled or some requests
64 // will get leaked. Certain callers use that message to know when they can
65 // delete their URLRequest object, even when doing a cancel. The default
66 // Kill implementation calls NotifyCanceled, so it is recommended that
67 // subclasses call URLRequestJob::Kill() after doing any additional work.
69 // The job should endeavor to stop working as soon as is convenient, but must
70 // not send and complete notifications from inside this function. Instead,
71 // complete notifications (including "canceled") should be sent from a
72 // callback run from the message loop.
74 // The job is not obliged to immediately stop sending data in response to
75 // this call, nor is it obliged to fail with "canceled" unless not all data
76 // was sent as a result. A typical case would be where the job is almost
77 // complete and can succeed before the canceled notification can be
78 // dispatched (from the message loop).
80 // The job should be prepared to receive multiple calls to kill it, but only
81 // one notification must be issued.
84 // Called to detach the request from this Job. Results in the Job being
85 // killed off eventually. The job must not use the request pointer any more.
88 // Called to read post-filtered data from this Job, returning the number of
89 // bytes read, 0 when there is no more data, or -1 if there was an error.
90 // This is just the backend for URLRequest::Read, see that function for
92 bool Read(IOBuffer
* buf
, int buf_size
, int* bytes_read
);
94 // Stops further caching of this request, if any. For more info, see
95 // URLRequest::StopCaching().
96 virtual void StopCaching();
98 // Called to fetch the current load state for the job.
99 virtual LoadState
GetLoadState() const;
101 // Called to get the upload progress in bytes.
102 virtual uint64
GetUploadProgress() const;
104 // Called to fetch the charset for this request. Only makes sense for some
105 // types of requests. Returns true on success. Calling this on a type that
106 // doesn't have a charset will return false.
107 virtual bool GetCharset(std::string
* charset
);
109 // Called to get response info.
110 virtual void GetResponseInfo(HttpResponseInfo
* info
);
112 // Returns the cookie values included in the response, if applicable.
113 // Returns true if applicable.
114 // NOTE: This removes the cookies from the job, so it will only return
115 // useful results once per job.
116 virtual bool GetResponseCookies(std::vector
<std::string
>* cookies
);
118 // Called to setup a stream filter for this request. An example of filter is
119 // content encoding/decoding.
120 // Subclasses should return the appropriate Filter, or NULL for no Filter.
121 // This class takes ownership of the returned Filter.
123 // The default implementation returns NULL.
124 virtual Filter
* SetupFilter() const;
126 // Called to determine if this response is a redirect. Only makes sense
127 // for some types of requests. This method returns true if the response
128 // is a redirect, and fills in the location param with the URL of the
129 // redirect. The HTTP status code (e.g., 302) is filled into
130 // |*http_status_code| to signify the type of redirect.
132 // The caller is responsible for following the redirect by setting up an
133 // appropriate replacement Job. Note that the redirected location may be
134 // invalid, the caller should be sure it can handle this.
136 // The default implementation inspects the response_info_.
137 virtual bool IsRedirectResponse(GURL
* location
, int* http_status_code
);
139 // Called to determine if it is okay to redirect this job to the specified
140 // location. This may be used to implement protocol-specific restrictions.
141 // If this function returns false, then the URLRequest will fail
142 // reporting ERR_UNSAFE_REDIRECT.
143 virtual bool IsSafeRedirect(const GURL
& location
);
145 // Called to determine if this response is asking for authentication. Only
146 // makes sense for some types of requests. The caller is responsible for
147 // obtaining the credentials passing them to SetAuth.
148 virtual bool NeedsAuth();
150 // Fills the authentication info with the server's response.
151 virtual void GetAuthChallengeInfo(
152 scoped_refptr
<AuthChallengeInfo
>* auth_info
);
154 // Resend the request with authentication credentials.
155 virtual void SetAuth(const AuthCredentials
& credentials
);
157 // Display the error page without asking for credentials again.
158 virtual void CancelAuth();
160 virtual void ContinueWithCertificate(X509Certificate
* client_cert
);
162 // Continue processing the request ignoring the last error.
163 virtual void ContinueDespiteLastError();
165 void FollowDeferredRedirect();
167 // Returns true if the Job is done producing response data and has called
168 // NotifyDone on the request.
169 bool is_done() const { return done_
; }
171 // Get/Set expected content size
172 int64
expected_content_size() const { return expected_content_size_
; }
173 void set_expected_content_size(const int64
& size
) {
174 expected_content_size_
= size
;
177 // Whether we have processed the response for that request yet.
178 bool has_response_started() const { return has_handled_response_
; }
180 // These methods are not applicable to all connections.
181 virtual bool GetMimeType(std::string
* mime_type
) const;
182 virtual int GetResponseCode() const;
184 // Returns the socket address for the connection.
185 // See url_request.h for details.
186 virtual HostPortPair
GetSocketAddress() const;
188 // base::SystemMonitor::PowerObserver methods:
189 // We invoke URLRequestJob::Kill on suspend (crbug.com/4606).
190 virtual void OnSuspend() OVERRIDE
;
192 // Called after a NetworkDelegate has been informed that the URLRequest
193 // will be destroyed. This is used to track that no pending callbacks
194 // exist at destruction time of the URLRequestJob, unless they have been
195 // canceled by an explicit NetworkDelegate::NotifyURLRequestDestroyed() call.
196 virtual void NotifyURLRequestDestroyed();
199 friend class base::RefCounted
<URLRequestJob
>;
200 virtual ~URLRequestJob();
202 // Notifies the job that a certificate is requested.
203 void NotifyCertificateRequested(SSLCertRequestInfo
* cert_request_info
);
205 // Notifies the job about an SSL certificate error.
206 void NotifySSLCertificateError(const SSLInfo
& ssl_info
, bool fatal
);
208 // Delegates to URLRequest::Delegate.
209 bool CanGetCookies(const CookieList
& cookie_list
) const;
211 // Delegates to URLRequest::Delegate.
212 bool CanSetCookie(const std::string
& cookie_line
,
213 CookieOptions
* options
) const;
215 // Notifies the job that headers have been received.
216 void NotifyHeadersComplete();
218 // Notifies the request that the job has completed a Read operation.
219 void NotifyReadComplete(int bytes_read
);
221 // Notifies the request that a start error has occurred.
222 void NotifyStartError(const URLRequestStatus
& status
);
224 // NotifyDone marks when we are done with a request. It is really
225 // a glorified set_status, but also does internal state checking and
226 // job tracking. It should be called once per request, when the job is
227 // finished doing all IO.
228 void NotifyDone(const URLRequestStatus
& status
);
230 // Some work performed by NotifyDone must be completed on a separate task
231 // so as to avoid re-entering the delegate. This method exists to perform
233 void CompleteNotifyDone();
235 // Used as an asynchronous callback for Kill to notify the URLRequest
236 // that we were canceled.
237 void NotifyCanceled();
239 // Notifies the job the request should be restarted.
240 // Should only be called if the job has not started a resposne.
241 void NotifyRestartRequired();
243 // Called when the delegate blocks or unblocks this request when intercepting
245 void SetBlockedOnDelegate();
246 void SetUnblockedOnDelegate();
248 // Called to read raw (pre-filtered) data from this Job.
249 // If returning true, data was read from the job. buf will contain
250 // the data, and bytes_read will receive the number of bytes read.
251 // If returning true, and bytes_read is returned as 0, there is no
252 // additional data to be read.
253 // If returning false, an error occurred or an async IO is now pending.
254 // If async IO is pending, the status of the request will be
255 // URLRequestStatus::IO_PENDING, and buf must remain available until the
256 // operation is completed. See comments on URLRequest::Read for more
258 virtual bool ReadRawData(IOBuffer
* buf
, int buf_size
, int *bytes_read
);
260 // Called to tell the job that a filter has successfully reached the end of
262 virtual void DoneReading();
264 // Informs the filter that data has been read into its buffer
265 void FilteredDataRead(int bytes_read
);
267 // Reads filtered data from the request. Returns true if successful,
268 // false otherwise. Note, if there is not enough data received to
269 // return data, this call can issue a new async IO request under
271 bool ReadFilteredData(int *bytes_read
);
273 // Whether the response is being filtered in this job.
274 // Only valid after NotifyHeadersComplete() has been called.
275 bool HasFilter() { return filter_
!= NULL
; }
277 // At or near destruction time, a derived class may request that the filters
278 // be destroyed so that statistics can be gathered while the derived class is
279 // still present to assist in calculations. This is used by URLRequestHttpJob
280 // to get SDCH to emit stats.
281 void DestroyFilters() { filter_
.reset(); }
283 // The status of the job.
284 const URLRequestStatus
GetStatus();
286 // Set the status of the job.
287 void SetStatus(const URLRequestStatus
& status
);
289 // The number of bytes read before passing to the filter.
290 int prefilter_bytes_read() const { return prefilter_bytes_read_
; }
292 // The number of bytes read after passing through the filter.
293 int postfilter_bytes_read() const { return postfilter_bytes_read_
; }
295 // Total number of bytes read from network (or cache) and typically handed
296 // to filter to process. Used to histogram compression ratios, and error
297 // recovery scenarios in filters.
298 int64
filter_input_byte_count() const { return filter_input_byte_count_
; }
300 // The request that initiated this job. This value MAY BE NULL if the
301 // request was released by DetachRequest().
302 URLRequest
* request_
;
305 // When data filtering is enabled, this function is used to read data
306 // for the filter. Returns true if raw data was read. Returns false if
307 // an error occurred (or we are waiting for IO to complete).
308 bool ReadRawDataForFilter(int *bytes_read
);
310 // Invokes ReadRawData and records bytes read if the read completes
312 bool ReadRawDataHelper(IOBuffer
* buf
, int buf_size
, int* bytes_read
);
314 // Called in response to a redirect that was not canceled to follow the
315 // redirect. The current job will be replaced with a new job loading the
316 // given redirect destination.
317 void FollowRedirect(const GURL
& location
, int http_status_code
);
319 // Called after every raw read. If |bytes_read| is > 0, this indicates
320 // a successful read of |bytes_read| unfiltered bytes. If |bytes_read|
321 // is 0, this indicates that there is no additional data to read. If
322 // |bytes_read| is < 0, an error occurred and no bytes were read.
323 void OnRawReadComplete(int bytes_read
);
325 // Updates the profiling info and notifies observers that an additional
326 // |bytes_read| unfiltered bytes have been read for this job.
327 void RecordBytesRead(int bytes_read
);
329 // Called to query whether there is data available in the filter to be read
331 bool FilterHasData();
333 // Subclasses may implement this method to record packet arrival times.
334 // The default implementation does nothing.
335 virtual void UpdatePacketReadTimes();
337 // Indicates that the job is done producing data, either it has completed
338 // all the data or an error has been encountered. Set exclusively by
339 // NotifyDone so that it is kept in sync with the request.
342 int prefilter_bytes_read_
;
343 int postfilter_bytes_read_
;
344 int64 filter_input_byte_count_
;
346 // The data stream filter which is enabled on demand.
347 scoped_ptr
<Filter
> filter_
;
349 // If the filter filled its output buffer, then there is a change that it
350 // still has internal data to emit, and this flag is set.
351 bool filter_needs_more_output_space_
;
353 // When we filter data, we receive data into the filter buffers. After
354 // processing the filtered data, we return the data in the caller's buffer.
355 // While the async IO is in progress, we save the user buffer here, and
356 // when the IO completes, we fill this in.
357 scoped_refptr
<IOBuffer
> filtered_read_buffer_
;
358 int filtered_read_buffer_len_
;
360 // We keep a pointer to the read buffer while asynchronous reads are
361 // in progress, so we are able to pass those bytes to job observers.
362 scoped_refptr
<IOBuffer
> raw_read_buffer_
;
364 // Used by HandleResponseIfNecessary to track whether we've sent the
365 // OnResponseStarted callback and potentially redirect callbacks as well.
366 bool has_handled_response_
;
368 // Expected content size
369 int64 expected_content_size_
;
371 // Set when a redirect is deferred.
372 GURL deferred_redirect_url_
;
373 int deferred_redirect_status_code_
;
375 base::WeakPtrFactory
<URLRequestJob
> weak_factory_
;
377 DISALLOW_COPY_AND_ASSIGN(URLRequestJob
);
382 #endif // NET_URL_REQUEST_URL_REQUEST_JOB_H_