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