| 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 #ifndef NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ | 5 #ifndef NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ |
| 6 #define NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ | 6 #define NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ |
| 7 | 7 |
| 8 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 9 #include "base/files/file.h" |
| 9 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| 10 #include "base/gtest_prod_util.h" | 11 #include "base/gtest_prod_util.h" |
| 11 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 12 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 13 #include "base/memory/weak_ptr.h" | 14 #include "base/memory/weak_ptr.h" |
| 14 #include "base/time/time.h" | 15 #include "base/time/time.h" |
| 15 #include "net/base/upload_element_reader.h" | 16 #include "net/base/upload_element_reader.h" |
| 16 | 17 |
| 17 namespace base { | 18 namespace base { |
| 18 class TaskRunner; | 19 class TaskRunner; |
| (...skipping 24 matching lines...) Expand all Loading... |
| 43 // UploadElementReader overrides: | 44 // UploadElementReader overrides: |
| 44 virtual const UploadFileElementReader* AsFileReader() const OVERRIDE; | 45 virtual const UploadFileElementReader* AsFileReader() const OVERRIDE; |
| 45 virtual int Init(const CompletionCallback& callback) OVERRIDE; | 46 virtual int Init(const CompletionCallback& callback) OVERRIDE; |
| 46 virtual uint64 GetContentLength() const OVERRIDE; | 47 virtual uint64 GetContentLength() const OVERRIDE; |
| 47 virtual uint64 BytesRemaining() const OVERRIDE; | 48 virtual uint64 BytesRemaining() const OVERRIDE; |
| 48 virtual int Read(IOBuffer* buf, | 49 virtual int Read(IOBuffer* buf, |
| 49 int buf_length, | 50 int buf_length, |
| 50 const CompletionCallback& callback) OVERRIDE; | 51 const CompletionCallback& callback) OVERRIDE; |
| 51 | 52 |
| 52 private: | 53 private: |
| 53 // Deletes FileStream with |task_runner| to avoid blocking the IO thread. | |
| 54 // This class is used as a template argument of scoped_ptr. | |
| 55 class FileStreamDeleter { | |
| 56 public: | |
| 57 explicit FileStreamDeleter(base::TaskRunner* task_runner); | |
| 58 ~FileStreamDeleter(); | |
| 59 void operator() (FileStream* file_stream) const; | |
| 60 | |
| 61 private: | |
| 62 scoped_refptr<base::TaskRunner> task_runner_; | |
| 63 }; | |
| 64 | |
| 65 typedef scoped_ptr<FileStream, FileStreamDeleter> ScopedFileStreamPtr; | |
| 66 | |
| 67 FRIEND_TEST_ALL_PREFIXES(UploadDataStreamTest, FileSmallerThanLength); | 54 FRIEND_TEST_ALL_PREFIXES(UploadDataStreamTest, FileSmallerThanLength); |
| 68 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionTest, | 55 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionTest, |
| 69 UploadFileSmallerThanLength); | 56 UploadFileSmallerThanLength); |
| 70 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionSpdy2Test, | 57 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionSpdy2Test, |
| 71 UploadFileSmallerThanLength); | 58 UploadFileSmallerThanLength); |
| 72 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionSpdy3Test, | 59 FRIEND_TEST_ALL_PREFIXES(HttpNetworkTransactionSpdy3Test, |
| 73 UploadFileSmallerThanLength); | 60 UploadFileSmallerThanLength); |
| 74 | 61 |
| 75 // Resets this instance to the uninitialized state. | 62 // Resets this instance to the uninitialized state. |
| 76 void Reset(); | 63 void Reset(); |
| 77 | 64 |
| 78 // This method is used to implement Init(). | 65 // These methods are used to implement Init(). |
| 79 void OnInitCompleted(ScopedFileStreamPtr* file_stream, | 66 void OnOpenCompleted(const CompletionCallback& callback, int result); |
| 80 uint64* content_length, | 67 void OnSeekCompleted(const CompletionCallback& callback, int64 result); |
| 81 const CompletionCallback& callback, | 68 void OnGetFileInfoCompleted(const CompletionCallback& callback, |
| 82 int result); | 69 base::File::Info* file_info, |
| 70 bool result); |
| 83 | 71 |
| 84 // This method is used to implement Read(). | 72 // This method is used to implement Read(). |
| 85 void OnReadCompleted(ScopedFileStreamPtr file_stream, | 73 void OnReadCompleted(const CompletionCallback& callback, int result); |
| 86 const CompletionCallback& callback, | |
| 87 int result); | |
| 88 | 74 |
| 89 // Sets an value to override the result for GetContentLength(). | 75 // Sets an value to override the result for GetContentLength(). |
| 90 // Used for tests. | 76 // Used for tests. |
| 91 struct NET_EXPORT_PRIVATE ScopedOverridingContentLengthForTests { | 77 struct NET_EXPORT_PRIVATE ScopedOverridingContentLengthForTests { |
| 92 ScopedOverridingContentLengthForTests(uint64 value); | 78 ScopedOverridingContentLengthForTests(uint64 value); |
| 93 ~ScopedOverridingContentLengthForTests(); | 79 ~ScopedOverridingContentLengthForTests(); |
| 94 }; | 80 }; |
| 95 | 81 |
| 96 scoped_refptr<base::TaskRunner> task_runner_; | 82 scoped_refptr<base::TaskRunner> task_runner_; |
| 97 const base::FilePath path_; | 83 const base::FilePath path_; |
| 98 const uint64 range_offset_; | 84 const uint64 range_offset_; |
| 99 const uint64 range_length_; | 85 const uint64 range_length_; |
| 100 const base::Time expected_modification_time_; | 86 const base::Time expected_modification_time_; |
| 101 ScopedFileStreamPtr file_stream_; | 87 scoped_ptr<FileStream> file_stream_; |
| 102 uint64 content_length_; | 88 uint64 content_length_; |
| 103 uint64 bytes_remaining_; | 89 uint64 bytes_remaining_; |
| 104 base::WeakPtrFactory<UploadFileElementReader> weak_ptr_factory_; | 90 base::WeakPtrFactory<UploadFileElementReader> weak_ptr_factory_; |
| 105 | 91 |
| 106 DISALLOW_COPY_AND_ASSIGN(UploadFileElementReader); | 92 DISALLOW_COPY_AND_ASSIGN(UploadFileElementReader); |
| 107 }; | 93 }; |
| 108 | 94 |
| 109 } // namespace net | 95 } // namespace net |
| 110 | 96 |
| 111 #endif // NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ | 97 #endif // NET_BASE_UPLOAD_FILE_ELEMENT_READER_H_ |
| OLD | NEW |