OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/download/download_throttling_resource_handler.h" | 5 #include "chrome/browser/download/download_throttling_resource_handler.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "content/browser/download/download_id.h" | 8 #include "content/browser/download/download_id.h" |
9 #include "content/browser/download/download_resource_handler.h" | 9 #include "content/browser/download/download_resource_handler.h" |
10 #include "content/browser/download/download_stats.h" | 10 #include "content/browser/download/download_stats.h" |
11 #include "content/browser/renderer_host/resource_dispatcher_host.h" | 11 #include "content/browser/renderer_host/resource_dispatcher_host.h" |
12 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" | 12 #include "content/browser/renderer_host/resource_dispatcher_host_request_info.h" |
13 #include "content/browser/resource_context.h" | 13 #include "content/browser/resource_context.h" |
14 #include "content/common/resource_response.h" | 14 #include "content/public/common/resource_response.h" |
15 #include "net/base/io_buffer.h" | 15 #include "net/base/io_buffer.h" |
16 #include "net/base/mime_sniffer.h" | 16 #include "net/base/mime_sniffer.h" |
17 | 17 |
18 DownloadThrottlingResourceHandler::DownloadThrottlingResourceHandler( | 18 DownloadThrottlingResourceHandler::DownloadThrottlingResourceHandler( |
19 ResourceHandler* next_handler, | 19 ResourceHandler* next_handler, |
20 ResourceDispatcherHost* host, | 20 ResourceDispatcherHost* host, |
21 DownloadRequestLimiter* limiter, | 21 DownloadRequestLimiter* limiter, |
22 net::URLRequest* request, | 22 net::URLRequest* request, |
23 const GURL& url, | 23 const GURL& url, |
24 int render_process_host_id, | 24 int render_process_host_id, |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
58 uint64 size) { | 58 uint64 size) { |
59 DCHECK(!request_closed_); | 59 DCHECK(!request_closed_); |
60 if (request_allowed_) | 60 if (request_allowed_) |
61 return next_handler_->OnUploadProgress(request_id, position, size); | 61 return next_handler_->OnUploadProgress(request_id, position, size); |
62 return true; | 62 return true; |
63 } | 63 } |
64 | 64 |
65 bool DownloadThrottlingResourceHandler::OnRequestRedirected( | 65 bool DownloadThrottlingResourceHandler::OnRequestRedirected( |
66 int request_id, | 66 int request_id, |
67 const GURL& url, | 67 const GURL& url, |
68 ResourceResponse* response, | 68 content::ResourceResponse* response, |
69 bool* defer) { | 69 bool* defer) { |
70 DCHECK(!request_closed_); | 70 DCHECK(!request_closed_); |
71 if (request_allowed_) { | 71 if (request_allowed_) { |
72 return next_handler_->OnRequestRedirected(request_id, url, response, defer); | 72 return next_handler_->OnRequestRedirected(request_id, url, response, defer); |
73 } | 73 } |
74 url_ = url; | 74 url_ = url; |
75 return true; | 75 return true; |
76 } | 76 } |
77 | 77 |
78 bool DownloadThrottlingResourceHandler::OnResponseStarted( | 78 bool DownloadThrottlingResourceHandler::OnResponseStarted( |
79 int request_id, | 79 int request_id, |
80 ResourceResponse* response) { | 80 content::ResourceResponse* response) { |
81 DCHECK(!request_closed_); | 81 DCHECK(!request_closed_); |
82 if (request_allowed_) | 82 if (request_allowed_) |
83 return next_handler_->OnResponseStarted(request_id, response); | 83 return next_handler_->OnResponseStarted(request_id, response); |
84 response_ = response; | 84 response_ = response; |
85 return true; | 85 return true; |
86 } | 86 } |
87 | 87 |
88 bool DownloadThrottlingResourceHandler::OnWillStart(int request_id, | 88 bool DownloadThrottlingResourceHandler::OnWillStart(int request_id, |
89 const GURL& url, | 89 const GURL& url, |
90 bool* defer) { | 90 bool* defer) { |
(...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
187 int buf_size; | 187 int buf_size; |
188 if (next_handler_->OnWillRead(request_id_, &buffer, &buf_size, | 188 if (next_handler_->OnWillRead(request_id_, &buffer, &buf_size, |
189 tmp_buffer_length_)) { | 189 tmp_buffer_length_)) { |
190 CHECK(buf_size >= tmp_buffer_length_); | 190 CHECK(buf_size >= tmp_buffer_length_); |
191 memcpy(buffer->data(), tmp_buffer_->data(), tmp_buffer_length_); | 191 memcpy(buffer->data(), tmp_buffer_->data(), tmp_buffer_length_); |
192 next_handler_->OnReadCompleted(request_id_, &tmp_buffer_length_); | 192 next_handler_->OnReadCompleted(request_id_, &tmp_buffer_length_); |
193 } | 193 } |
194 tmp_buffer_length_ = 0; | 194 tmp_buffer_length_ = 0; |
195 tmp_buffer_ = NULL; | 195 tmp_buffer_ = NULL; |
196 } | 196 } |
OLD | NEW |