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 |