| 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 // The DownloadFileManager owns a set of DownloadFile objects, each of which | 5 // The DownloadFileManager owns a set of DownloadFile objects, each of which |
| 6 // represent one in progress download and performs the disk IO for that | 6 // represent one in progress download and performs the disk IO for that |
| 7 // download. The DownloadFileManager itself is a singleton object owned by the | 7 // download. The DownloadFileManager itself is a singleton object owned by the |
| 8 // ResourceDispatcherHostImpl. | 8 // ResourceDispatcherHostImpl. |
| 9 // | 9 // |
| 10 // The DownloadFileManager uses the file_thread for performing file write | 10 // The DownloadFileManager uses the file_thread for performing file write |
| (...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 43 | 43 |
| 44 #include "base/atomic_sequence_num.h" | 44 #include "base/atomic_sequence_num.h" |
| 45 #include "base/basictypes.h" | 45 #include "base/basictypes.h" |
| 46 #include "base/callback_forward.h" | 46 #include "base/callback_forward.h" |
| 47 #include "base/gtest_prod_util.h" | 47 #include "base/gtest_prod_util.h" |
| 48 #include "base/hash_tables.h" | 48 #include "base/hash_tables.h" |
| 49 #include "base/memory/ref_counted.h" | 49 #include "base/memory/ref_counted.h" |
| 50 #include "base/memory/scoped_ptr.h" | 50 #include "base/memory/scoped_ptr.h" |
| 51 #include "base/timer.h" | 51 #include "base/timer.h" |
| 52 #include "content/browser/download/download_file.h" | 52 #include "content/browser/download/download_file.h" |
| 53 #include "content/browser/download/download_file_factory.h" |
| 53 #include "content/common/content_export.h" | 54 #include "content/common/content_export.h" |
| 54 #include "content/public/browser/download_id.h" | 55 #include "content/public/browser/download_id.h" |
| 55 #include "content/public/browser/download_interrupt_reasons.h" | 56 #include "content/public/browser/download_interrupt_reasons.h" |
| 56 #include "net/base/net_errors.h" | 57 #include "net/base/net_errors.h" |
| 57 #include "ui/gfx/native_widget_types.h" | 58 #include "ui/gfx/native_widget_types.h" |
| 58 | 59 |
| 59 struct DownloadCreateInfo; | 60 struct DownloadCreateInfo; |
| 60 class DownloadRequestHandle; | 61 class DownloadRequestHandle; |
| 61 class FilePath; | 62 class FilePath; |
| 62 | 63 |
| (...skipping 15 matching lines...) Expand all Loading... |
| 78 public: | 79 public: |
| 79 // Callback used with CreateDownloadFile(). |reason| will be | 80 // Callback used with CreateDownloadFile(). |reason| will be |
| 80 // DOWNLOAD_INTERRUPT_REASON_NONE on a successful creation. | 81 // DOWNLOAD_INTERRUPT_REASON_NONE on a successful creation. |
| 81 typedef base::Callback<void(content::DownloadInterruptReason reason)> | 82 typedef base::Callback<void(content::DownloadInterruptReason reason)> |
| 82 CreateDownloadFileCallback; | 83 CreateDownloadFileCallback; |
| 83 | 84 |
| 84 // Callback used with RenameDownloadFile(). | 85 // Callback used with RenameDownloadFile(). |
| 85 typedef content::DownloadFile::RenameCompletionCallback | 86 typedef content::DownloadFile::RenameCompletionCallback |
| 86 RenameCompletionCallback; | 87 RenameCompletionCallback; |
| 87 | 88 |
| 88 class DownloadFileFactory { | |
| 89 public: | |
| 90 virtual ~DownloadFileFactory() {} | |
| 91 | |
| 92 virtual content::DownloadFile* CreateFile( | |
| 93 DownloadCreateInfo* info, | |
| 94 scoped_ptr<content::ByteStreamReader> stream, | |
| 95 content::DownloadManager* download_manager, | |
| 96 bool calculate_hash, | |
| 97 const net::BoundNetLog& bound_net_log) = 0; | |
| 98 }; | |
| 99 | |
| 100 // Takes ownership of the factory. | 89 // Takes ownership of the factory. |
| 101 // Passing in a NULL for |factory| will cause a default | 90 // Passing in a NULL for |factory| will cause a default |
| 102 // |DownloadFileFactory| to be used. | 91 // |DownloadFileFactory| to be used. |
| 103 explicit DownloadFileManager(DownloadFileFactory* factory); | 92 explicit DownloadFileManager(content::DownloadFileFactory* factory); |
| 104 | 93 |
| 105 // Create a download file and record it in the download file manager. | 94 // Create a download file and record it in the download file manager. |
| 106 virtual void CreateDownloadFile( | 95 virtual void CreateDownloadFile( |
| 107 scoped_ptr<DownloadCreateInfo> info, | 96 scoped_ptr<DownloadCreateInfo> info, |
| 108 scoped_ptr<content::ByteStreamReader> stream, | 97 scoped_ptr<content::ByteStreamReader> stream, |
| 109 scoped_refptr<content::DownloadManager> download_manager, | 98 scoped_refptr<content::DownloadManager> download_manager, |
| 110 bool hash_needed, | 99 bool hash_needed, |
| 111 const net::BoundNetLog& bound_net_log, | 100 const net::BoundNetLog& bound_net_log, |
| 112 const CreateDownloadFileCallback& callback); | 101 const CreateDownloadFileCallback& callback); |
| 113 | 102 |
| (...skipping 18 matching lines...) Expand all Loading... |
| 132 virtual void RenameDownloadFile( | 121 virtual void RenameDownloadFile( |
| 133 content::DownloadId id, | 122 content::DownloadId id, |
| 134 const FilePath& full_path, | 123 const FilePath& full_path, |
| 135 bool overwrite_existing_file, | 124 bool overwrite_existing_file, |
| 136 const RenameCompletionCallback& callback); | 125 const RenameCompletionCallback& callback); |
| 137 | 126 |
| 138 // The number of downloads currently active on the DownloadFileManager. | 127 // The number of downloads currently active on the DownloadFileManager. |
| 139 // Primarily for testing. | 128 // Primarily for testing. |
| 140 virtual int NumberOfActiveDownloads() const; | 129 virtual int NumberOfActiveDownloads() const; |
| 141 | 130 |
| 142 void SetFileFactoryForTesting(scoped_ptr<DownloadFileFactory> file_factory) { | 131 void SetFileFactoryForTesting( |
| 132 scoped_ptr<content::DownloadFileFactory> file_factory) { |
| 143 download_file_factory_.reset(file_factory.release()); | 133 download_file_factory_.reset(file_factory.release()); |
| 144 } | 134 } |
| 145 | 135 |
| 146 DownloadFileFactory* GetFileFactoryForTesting() const { | 136 content::DownloadFileFactory* GetFileFactoryForTesting() const { |
| 147 return download_file_factory_.get(); // Explicitly NOT a scoped_ptr. | 137 return download_file_factory_.get(); // Explicitly NOT a scoped_ptr. |
| 148 } | 138 } |
| 149 | 139 |
| 150 protected: | 140 protected: |
| 151 virtual ~DownloadFileManager(); | 141 virtual ~DownloadFileManager(); |
| 152 | 142 |
| 153 private: | 143 private: |
| 154 friend class base::RefCountedThreadSafe<DownloadFileManager>; | 144 friend class base::RefCountedThreadSafe<DownloadFileManager>; |
| 155 friend class DownloadFileManagerTest; | 145 friend class DownloadFileManagerTest; |
| 156 friend class DownloadManagerTest; | 146 friend class DownloadManagerTest; |
| 157 FRIEND_TEST_ALL_PREFIXES(DownloadManagerTest, StartDownload); | 147 FRIEND_TEST_ALL_PREFIXES(DownloadManagerTest, StartDownload); |
| 158 | 148 |
| 159 // Clean up helper that runs on the download thread. | 149 // Clean up helper that runs on the download thread. |
| 160 void OnShutdown(); | 150 void OnShutdown(); |
| 161 | 151 |
| 162 // Called only on the download thread. | 152 // Called only on the download thread. |
| 163 content::DownloadFile* GetDownloadFile(content::DownloadId global_id); | 153 content::DownloadFile* GetDownloadFile(content::DownloadId global_id); |
| 164 | 154 |
| 165 // Erases the download file with the given the download |id| and removes | 155 // Erases the download file with the given the download |id| and removes |
| 166 // it from the maps. | 156 // it from the maps. |
| 167 void EraseDownload(content::DownloadId global_id); | 157 void EraseDownload(content::DownloadId global_id); |
| 168 | 158 |
| 169 typedef base::hash_map<content::DownloadId, content::DownloadFile*> | 159 typedef base::hash_map<content::DownloadId, content::DownloadFile*> |
| 170 DownloadFileMap; | 160 DownloadFileMap; |
| 171 | 161 |
| 172 // A map of all in progress downloads. It owns the download files. | 162 // A map of all in progress downloads. It owns the download files. |
| 173 DownloadFileMap downloads_; | 163 DownloadFileMap downloads_; |
| 174 | 164 |
| 175 scoped_ptr<DownloadFileFactory> download_file_factory_; | 165 scoped_ptr<content::DownloadFileFactory> download_file_factory_; |
| 176 | 166 |
| 177 DISALLOW_COPY_AND_ASSIGN(DownloadFileManager); | 167 DISALLOW_COPY_AND_ASSIGN(DownloadFileManager); |
| 178 }; | 168 }; |
| 179 | 169 |
| 180 #endif // CONTENT_BROWSER_DOWNLOAD_DOWNLOAD_FILE_MANAGER_H_ | 170 #endif // CONTENT_BROWSER_DOWNLOAD_DOWNLOAD_FILE_MANAGER_H_ |
| OLD | NEW |