| OLD | NEW |
|---|
| (Empty) |
| 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. | |
| 4 | |
| 5 #include "chrome/browser/renderer_host/save_file_resource_handler.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "base/message_loop.h" | |
| 9 #include "base/string_number_conversions.h" | |
| 10 #include "chrome/browser/download/save_file_manager.h" | |
| 11 #include "content/browser/browser_thread.h" | |
| 12 #include "net/base/io_buffer.h" | |
| 13 #include "net/url_request/url_request_status.h" | |
| 14 | |
| 15 SaveFileResourceHandler::SaveFileResourceHandler(int render_process_host_id, | |
| 16 int render_view_id, | |
| 17 const GURL& url, | |
| 18 SaveFileManager* manager) | |
| 19 : save_id_(-1), | |
| 20 render_process_id_(render_process_host_id), | |
| 21 render_view_id_(render_view_id), | |
| 22 url_(url), | |
| 23 content_length_(0), | |
| 24 save_manager_(manager) { | |
| 25 } | |
| 26 | |
| 27 bool SaveFileResourceHandler::OnUploadProgress(int request_id, | |
| 28 uint64 position, | |
| 29 uint64 size) { | |
| 30 return true; | |
| 31 } | |
| 32 | |
| 33 bool SaveFileResourceHandler::OnRequestRedirected(int request_id, | |
| 34 const GURL& url, | |
| 35 ResourceResponse* response, | |
| 36 bool* defer) { | |
| 37 final_url_ = url; | |
| 38 return true; | |
| 39 } | |
| 40 | |
| 41 bool SaveFileResourceHandler::OnResponseStarted(int request_id, | |
| 42 ResourceResponse* response) { | |
| 43 save_id_ = save_manager_->GetNextId(); | |
| 44 // |save_manager_| consumes (deletes): | |
| 45 SaveFileCreateInfo* info = new SaveFileCreateInfo; | |
| 46 info->url = url_; | |
| 47 info->final_url = final_url_; | |
| 48 info->total_bytes = content_length_; | |
| 49 info->save_id = save_id_; | |
| 50 info->render_process_id = render_process_id_; | |
| 51 info->render_view_id = render_view_id_; | |
| 52 info->request_id = request_id; | |
| 53 info->content_disposition = content_disposition_; | |
| 54 info->save_source = SaveFileCreateInfo::SAVE_FILE_FROM_NET; | |
| 55 BrowserThread::PostTask( | |
| 56 BrowserThread::FILE, FROM_HERE, | |
| 57 NewRunnableMethod(save_manager_, | |
| 58 &SaveFileManager::StartSave, | |
| 59 info)); | |
| 60 return true; | |
| 61 } | |
| 62 | |
| 63 bool SaveFileResourceHandler::OnWillStart(int request_id, | |
| 64 const GURL& url, | |
| 65 bool* defer) { | |
| 66 return true; | |
| 67 } | |
| 68 | |
| 69 bool SaveFileResourceHandler::OnWillRead(int request_id, net::IOBuffer** buf, | |
| 70 int* buf_size, int min_size) { | |
| 71 DCHECK(buf && buf_size); | |
| 72 if (!read_buffer_) { | |
| 73 *buf_size = min_size < 0 ? kReadBufSize : min_size; | |
| 74 read_buffer_ = new net::IOBuffer(*buf_size); | |
| 75 } | |
| 76 *buf = read_buffer_.get(); | |
| 77 return true; | |
| 78 } | |
| 79 | |
| 80 bool SaveFileResourceHandler::OnReadCompleted(int request_id, int* bytes_read) { | |
| 81 DCHECK(read_buffer_); | |
| 82 // We are passing ownership of this buffer to the save file manager. | |
| 83 scoped_refptr<net::IOBuffer> buffer; | |
| 84 read_buffer_.swap(buffer); | |
| 85 BrowserThread::PostTask( | |
| 86 BrowserThread::FILE, FROM_HERE, | |
| 87 NewRunnableMethod(save_manager_, | |
| 88 &SaveFileManager::UpdateSaveProgress, | |
| 89 save_id_, | |
| 90 buffer, | |
| 91 *bytes_read)); | |
| 92 return true; | |
| 93 } | |
| 94 | |
| 95 bool SaveFileResourceHandler::OnResponseCompleted( | |
| 96 int request_id, | |
| 97 const net::URLRequestStatus& status, | |
| 98 const std::string& security_info) { | |
| 99 BrowserThread::PostTask( | |
| 100 BrowserThread::FILE, FROM_HERE, | |
| 101 NewRunnableMethod(save_manager_, | |
| 102 &SaveFileManager::SaveFinished, | |
| 103 save_id_, | |
| 104 url_, | |
| 105 render_process_id_, | |
| 106 status.is_success() && !status.is_io_pending())); | |
| 107 read_buffer_ = NULL; | |
| 108 return true; | |
| 109 } | |
| 110 | |
| 111 void SaveFileResourceHandler::OnRequestClosed() { | |
| 112 } | |
| 113 | |
| 114 void SaveFileResourceHandler::set_content_length( | |
| 115 const std::string& content_length) { | |
| 116 base::StringToInt64(content_length, &content_length_); | |
| 117 } | |
| 118 | |
| 119 SaveFileResourceHandler::~SaveFileResourceHandler() {} | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 7373004:
Move the save file code from chrome to content. This is just a file move so the DEPS in content\b... (Closed)
Base URL: svn://chrome-svn/chrome/trunk/src/