[email protected] | 27a112c | 2011-01-06 04:19:30 | [diff] [blame] | 1 | // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 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 CHROME_BROWSER_RENDERER_HOST_DOWNLOAD_RESOURCE_HANDLER_H_ |
| 6 | #define CHROME_BROWSER_RENDERER_HOST_DOWNLOAD_RESOURCE_HANDLER_H_ |
[email protected] | 32b76ef | 2010-07-26 23:08:24 | [diff] [blame] | 7 | #pragma once |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 8 | |
| 9 | #include <string> |
| 10 | |
[email protected] | 92b24c1 | 2009-12-10 20:04:35 | [diff] [blame] | 11 | #include "base/timer.h" |
[email protected] | 8af9d03 | 2010-02-10 00:00:32 | [diff] [blame] | 12 | #include "chrome/browser/download/download_file.h" |
[email protected] | 5de63471 | 2011-03-02 00:20:19 | [diff] [blame^] | 13 | #include "content/browser/renderer_host/global_request_id.h" |
| 14 | #include "content/browser/renderer_host/resource_handler.h" |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 15 | |
[email protected] | 92b24c1 | 2009-12-10 20:04:35 | [diff] [blame] | 16 | class DownloadFileManager; |
| 17 | class ResourceDispatcherHost; |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 18 | struct DownloadBuffer; |
| 19 | |
[email protected] | edfe7fab | 2010-11-28 13:11:52 | [diff] [blame] | 20 | namespace net { |
| 21 | class URLRequest; |
| 22 | } // namespace net |
| 23 | |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 24 | // Forwards data to the download thread. |
[email protected] | 287b86b | 2011-02-26 00:11:35 | [diff] [blame] | 25 | class DownloadResourceHandler : public ResourceHandler { |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 26 | public: |
| 27 | DownloadResourceHandler(ResourceDispatcherHost* rdh, |
| 28 | int render_process_host_id, |
| 29 | int render_view_id, |
| 30 | int request_id, |
[email protected] | f6b4853 | 2009-02-12 01:56:32 | [diff] [blame] | 31 | const GURL& url, |
[email protected] | aa033af | 2010-07-27 18:16:39 | [diff] [blame] | 32 | DownloadFileManager* download_file_manager, |
[email protected] | edfe7fab | 2010-11-28 13:11:52 | [diff] [blame] | 33 | net::URLRequest* request, |
[email protected] | 6aa4a1c0 | 2010-01-15 18:49:58 | [diff] [blame] | 34 | bool save_as, |
[email protected] | 8af9d03 | 2010-02-10 00:00:32 | [diff] [blame] | 35 | const DownloadSaveInfo& save_info); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 36 | |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 37 | virtual bool OnUploadProgress(int request_id, uint64 position, uint64 size); |
[email protected] | afd832c | 2010-03-02 04:53:31 | [diff] [blame] | 38 | |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 39 | // Not needed, as this event handler ought to be the final resource. |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 40 | virtual bool OnRequestRedirected(int request_id, const GURL& url, |
| 41 | ResourceResponse* response, bool* defer); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 42 | |
| 43 | // Send the download creation information to the download thread. |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 44 | virtual bool OnResponseStarted(int request_id, ResourceResponse* response); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 45 | |
[email protected] | afd832c | 2010-03-02 04:53:31 | [diff] [blame] | 46 | // Pass-through implementation. |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 47 | virtual bool OnWillStart(int request_id, const GURL& url, bool* defer); |
[email protected] | afd832c | 2010-03-02 04:53:31 | [diff] [blame] | 48 | |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 49 | // Create a new buffer, which will be handed to the download thread for file |
| 50 | // writing and deletion. |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 51 | virtual bool OnWillRead(int request_id, net::IOBuffer** buf, int* buf_size, |
| 52 | int min_size); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 53 | |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 54 | virtual bool OnReadCompleted(int request_id, int* bytes_read); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 55 | |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 56 | virtual bool OnResponseCompleted(int request_id, |
[email protected] | 27a112c | 2011-01-06 04:19:30 | [diff] [blame] | 57 | const net::URLRequestStatus& status, |
[email protected] | 78994ab0 | 2010-12-08 18:06:44 | [diff] [blame] | 58 | const std::string& security_info); |
| 59 | virtual void OnRequestClosed(); |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 60 | |
| 61 | // If the content-length header is not present (or contains something other |
| 62 | // than numbers), the incoming content_length is -1 (unknown size). |
| 63 | // Set the content length to 0 to indicate unknown size to DownloadManager. |
| 64 | void set_content_length(const int64& content_length); |
| 65 | |
| 66 | void set_content_disposition(const std::string& content_disposition); |
| 67 | |
| 68 | void CheckWriteProgress(); |
| 69 | |
[email protected] | da6e392 | 2010-11-24 21:45:50 | [diff] [blame] | 70 | std::string DebugString() const; |
| 71 | |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 72 | private: |
[email protected] | 601858c0 | 2010-09-01 17:08:20 | [diff] [blame] | 73 | ~DownloadResourceHandler(); |
[email protected] | 84bbb2b5 | 2009-11-07 00:23:14 | [diff] [blame] | 74 | |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 75 | void StartPauseTimer(); |
| 76 | |
| 77 | int download_id_; |
[email protected] | 92b24c1 | 2009-12-10 20:04:35 | [diff] [blame] | 78 | GlobalRequestID global_id_; |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 79 | int render_view_id_; |
[email protected] | 9dea9e1f | 2009-01-29 00:30:47 | [diff] [blame] | 80 | scoped_refptr<net::IOBuffer> read_buffer_; |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 81 | std::string content_disposition_; |
[email protected] | 79558c2a | 2011-01-14 21:46:07 | [diff] [blame] | 82 | GURL url_; // final URL from which we're downloading. |
| 83 | GURL original_url_; // original URL before any redirection by the server. |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 84 | int64 content_length_; |
[email protected] | aa033af | 2010-07-27 18:16:39 | [diff] [blame] | 85 | DownloadFileManager* download_file_manager_; |
[email protected] | edfe7fab | 2010-11-28 13:11:52 | [diff] [blame] | 86 | net::URLRequest* request_; |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 87 | bool save_as_; // Request was initiated via "Save As" by the user. |
[email protected] | 8af9d03 | 2010-02-10 00:00:32 | [diff] [blame] | 88 | DownloadSaveInfo save_info_; |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 89 | DownloadBuffer* buffer_; |
| 90 | ResourceDispatcherHost* rdh_; |
| 91 | bool is_paused_; |
| 92 | base::OneShotTimer<DownloadResourceHandler> pause_timer_; |
[email protected] | 9fc1e6c1 | 2010-12-22 21:07:05 | [diff] [blame] | 93 | base::TimeTicks download_start_time_; // used to collect stats. |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 94 | static const int kReadBufSize = 32768; // bytes |
[email protected] | 3c2fe4e | 2009-01-08 12:07:23 | [diff] [blame] | 95 | static const size_t kLoadsToWrite = 100; // number of data buffers queued |
[email protected] | e3c404b | 2008-12-23 01:07:32 | [diff] [blame] | 96 | static const int kThrottleTimeMs = 200; // milliseconds |
| 97 | |
| 98 | DISALLOW_COPY_AND_ASSIGN(DownloadResourceHandler); |
| 99 | }; |
| 100 | |
| 101 | #endif // CHROME_BROWSER_RENDERER_HOST_DOWNLOAD_RESOURCE_HANDLER_H_ |