| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "content/browser/download/save_file_resource_handler.h" | 5 #include "content/browser/download/save_file_resource_handler.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/logging.h" | 8 #include "base/logging.h" |
| 9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
| 10 #include "base/strings/string_number_conversions.h" | 10 #include "base/strings/string_number_conversions.h" |
| 11 #include "content/browser/download/save_file_manager.h" | 11 #include "content/browser/download/save_file_manager.h" |
| 12 #include "content/public/browser/browser_thread.h" | 12 #include "content/public/browser/browser_thread.h" |
| 13 #include "net/base/io_buffer.h" | 13 #include "net/base/io_buffer.h" |
| 14 #include "net/url_request/redirect_info.h" | 14 #include "net/url_request/redirect_info.h" |
| 15 #include "net/url_request/url_request_status.h" | 15 #include "net/url_request/url_request_status.h" |
| 16 | 16 |
| 17 namespace content { | 17 namespace content { |
| 18 | 18 |
| 19 SaveFileResourceHandler::SaveFileResourceHandler(net::URLRequest* request, | 19 SaveFileResourceHandler::SaveFileResourceHandler(net::URLRequest* request, |
| 20 int save_item_id, |
| 20 int save_package_id, | 21 int save_package_id, |
| 21 int render_process_host_id, | 22 int render_process_host_id, |
| 22 int render_frame_routing_id, | 23 int render_frame_routing_id, |
| 23 const GURL& url, | 24 const GURL& url, |
| 24 SaveFileManager* manager) | 25 SaveFileManager* manager) |
| 25 : ResourceHandler(request), | 26 : ResourceHandler(request), |
| 26 save_id_(-1), | 27 save_item_id_(save_item_id), |
| 27 save_package_id_(save_package_id), | 28 save_package_id_(save_package_id), |
| 28 render_process_id_(render_process_host_id), | 29 render_process_id_(render_process_host_id), |
| 29 render_frame_routing_id_(render_frame_routing_id), | 30 render_frame_routing_id_(render_frame_routing_id), |
| 30 url_(url), | 31 url_(url), |
| 31 content_length_(0), | 32 content_length_(0), |
| 32 save_manager_(manager) {} | 33 save_manager_(manager) {} |
| 33 | 34 |
| 34 SaveFileResourceHandler::~SaveFileResourceHandler() { | 35 SaveFileResourceHandler::~SaveFileResourceHandler() { |
| 35 } | 36 } |
| 36 | 37 |
| 37 bool SaveFileResourceHandler::OnRequestRedirected( | 38 bool SaveFileResourceHandler::OnRequestRedirected( |
| 38 const net::RedirectInfo& redirect_info, | 39 const net::RedirectInfo& redirect_info, |
| 39 ResourceResponse* response, | 40 ResourceResponse* response, |
| 40 bool* defer) { | 41 bool* defer) { |
| 41 final_url_ = redirect_info.new_url; | 42 final_url_ = redirect_info.new_url; |
| 42 return true; | 43 return true; |
| 43 } | 44 } |
| 44 | 45 |
| 45 bool SaveFileResourceHandler::OnResponseStarted(ResourceResponse* response, | 46 bool SaveFileResourceHandler::OnResponseStarted(ResourceResponse* response, |
| 46 bool* defer) { | 47 bool* defer) { |
| 47 save_id_ = save_manager_->GetNextId(); | |
| 48 // |save_manager_| consumes (deletes): | 48 // |save_manager_| consumes (deletes): |
| 49 SaveFileCreateInfo* info = new SaveFileCreateInfo; | 49 SaveFileCreateInfo* info = new SaveFileCreateInfo( |
| 50 info->url = url_; | 50 url_, final_url_, save_item_id_, save_package_id_, render_process_id_, |
| 51 info->final_url = final_url_; | 51 render_frame_routing_id_, GetRequestID(), content_disposition_, |
| 52 info->total_bytes = content_length_; | 52 content_length_); |
| 53 info->save_id = save_id_; | |
| 54 info->save_package_id = save_package_id_; | |
| 55 info->render_process_id = render_process_id_; | |
| 56 info->render_frame_routing_id = render_frame_routing_id_; | |
| 57 info->request_id = GetRequestID(); | |
| 58 info->content_disposition = content_disposition_; | |
| 59 info->save_source = SaveFileCreateInfo::SAVE_FILE_FROM_NET; | |
| 60 BrowserThread::PostTask( | 53 BrowserThread::PostTask( |
| 61 BrowserThread::FILE, FROM_HERE, | 54 BrowserThread::FILE, FROM_HERE, |
| 62 base::Bind(&SaveFileManager::StartSave, save_manager_, info)); | 55 base::Bind(&SaveFileManager::StartSave, save_manager_, info)); |
| 63 return true; | 56 return true; |
| 64 } | 57 } |
| 65 | 58 |
| 66 bool SaveFileResourceHandler::OnWillStart(const GURL& url, bool* defer) { | 59 bool SaveFileResourceHandler::OnWillStart(const GURL& url, bool* defer) { |
| 67 return true; | 60 return true; |
| 68 } | 61 } |
| 69 | 62 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 84 return true; | 77 return true; |
| 85 } | 78 } |
| 86 | 79 |
| 87 bool SaveFileResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { | 80 bool SaveFileResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { |
| 88 DCHECK(read_buffer_.get()); | 81 DCHECK(read_buffer_.get()); |
| 89 // We are passing ownership of this buffer to the save file manager. | 82 // We are passing ownership of this buffer to the save file manager. |
| 90 scoped_refptr<net::IOBuffer> buffer; | 83 scoped_refptr<net::IOBuffer> buffer; |
| 91 read_buffer_.swap(buffer); | 84 read_buffer_.swap(buffer); |
| 92 BrowserThread::PostTask( | 85 BrowserThread::PostTask( |
| 93 BrowserThread::FILE, FROM_HERE, | 86 BrowserThread::FILE, FROM_HERE, |
| 94 base::Bind(&SaveFileManager::UpdateSaveProgress, | 87 base::Bind(&SaveFileManager::UpdateSaveProgress, save_manager_, |
| 95 save_manager_, save_id_, buffer, bytes_read)); | 88 save_item_id_, buffer, bytes_read)); |
| 96 return true; | 89 return true; |
| 97 } | 90 } |
| 98 | 91 |
| 99 void SaveFileResourceHandler::OnResponseCompleted( | 92 void SaveFileResourceHandler::OnResponseCompleted( |
| 100 const net::URLRequestStatus& status, | 93 const net::URLRequestStatus& status, |
| 101 const std::string& security_info, | 94 const std::string& security_info, |
| 102 bool* defer) { | 95 bool* defer) { |
| 103 BrowserThread::PostTask( | 96 BrowserThread::PostTask( |
| 104 BrowserThread::FILE, FROM_HERE, | 97 BrowserThread::FILE, FROM_HERE, |
| 105 base::Bind(&SaveFileManager::SaveFinished, save_manager_, save_id_, url_, | 98 base::Bind(&SaveFileManager::SaveFinished, save_manager_, save_item_id_, |
| 106 save_package_id_, | 99 save_package_id_, |
| 107 status.is_success() && !status.is_io_pending())); | 100 status.is_success() && !status.is_io_pending())); |
| 108 read_buffer_ = NULL; | 101 read_buffer_ = NULL; |
| 109 } | 102 } |
| 110 | 103 |
| 111 void SaveFileResourceHandler::OnDataDownloaded(int bytes_downloaded) { | 104 void SaveFileResourceHandler::OnDataDownloaded(int bytes_downloaded) { |
| 112 NOTREACHED(); | 105 NOTREACHED(); |
| 113 } | 106 } |
| 114 | 107 |
| 115 void SaveFileResourceHandler::set_content_length( | 108 void SaveFileResourceHandler::set_content_length( |
| 116 const std::string& content_length) { | 109 const std::string& content_length) { |
| 117 base::StringToInt64(content_length, &content_length_); | 110 base::StringToInt64(content_length, &content_length_); |
| 118 } | 111 } |
| 119 | 112 |
| 120 } // namespace content | 113 } // namespace content |
| OLD | NEW |