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 3 // found in the LICENSE file. 4 5 #ifndef CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_ 6 #define CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_ 7 8 #include <string> 9 10 #include "base/memory/ref_counted.h" 11 #include "content/browser/loader/resource_handler.h" 12 #include "url/gurl.h" 13 14 namespace net { 15 class URLRequest; 16 } 17 18 namespace content { 19 class SaveFileManager; 20 21 // Forwards data to the save thread. 22 class SaveFileResourceHandler : public ResourceHandler { 23 public: 24 SaveFileResourceHandler(net::URLRequest* request, 25 int render_process_host_id, 26 int render_view_id, 27 const GURL& url, 28 SaveFileManager* manager); 29 virtual ~SaveFileResourceHandler(); 30 31 // ResourceHandler Implementation: 32 virtual bool OnUploadProgress(uint64 position, uint64 size) OVERRIDE; 33 34 // Saves the redirected URL to final_url_, we need to use the original 35 // URL to match original request. 36 virtual bool OnRequestRedirected(const net::RedirectInfo& redirect_info, 37 ResourceResponse* response, 38 bool* defer) OVERRIDE; 39 40 // Sends the download creation information to the download thread. 41 virtual bool OnResponseStarted(ResourceResponse* response, 42 bool* defer) OVERRIDE; 43 44 // Pass-through implementation. 45 virtual bool OnWillStart(const GURL& url, bool* defer) OVERRIDE; 46 47 // Pass-through implementation. 48 virtual bool OnBeforeNetworkStart(const GURL& url, bool* defer) OVERRIDE; 49 50 // Creates a new buffer, which will be handed to the download thread for file 51 // writing and deletion. 52 virtual bool OnWillRead(scoped_refptr<net::IOBuffer>* buf, 53 int* buf_size, 54 int min_size) OVERRIDE; 55 56 // Passes the buffer to the download file writer. 57 virtual bool OnReadCompleted(int bytes_read, bool* defer) OVERRIDE; 58 59 virtual void OnResponseCompleted(const net::URLRequestStatus& status, 60 const std::string& security_info, 61 bool* defer) OVERRIDE; 62 63 // N/A to this flavor of SaveFileResourceHandler. 64 virtual void OnDataDownloaded(int bytes_downloaded) OVERRIDE; 65 66 // If the content-length header is not present (or contains something other 67 // than numbers), StringToInt64 returns 0, which indicates 'unknown size' and 68 // is handled correctly by the SaveManager. 69 void set_content_length(const std::string& content_length); 70 set_content_disposition(const std::string & content_disposition)71 void set_content_disposition(const std::string& content_disposition) { 72 content_disposition_ = content_disposition; 73 } 74 75 private: 76 int save_id_; 77 int render_process_id_; 78 int render_view_id_; 79 scoped_refptr<net::IOBuffer> read_buffer_; 80 std::string content_disposition_; 81 GURL url_; 82 GURL final_url_; 83 int64 content_length_; 84 SaveFileManager* save_manager_; 85 86 static const int kReadBufSize = 32768; // bytes 87 88 DISALLOW_COPY_AND_ASSIGN(SaveFileResourceHandler); 89 }; 90 91 } // namespace content 92 93 #endif // CONTENT_BROWSER_DOWNLOAD_SAVE_FILE_RESOURCE_HANDLER_H_ 94