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 #include "net/disk_cache/blockfile/file.h" | 5 #include "net/disk_cache/blockfile/file.h" |
6 | 6 |
7 #include "base/files/file_path.h" | 7 #include "base/files/file_path.h" |
8 #include "base/lazy_instance.h" | 8 #include "base/lazy_instance.h" |
9 #include "base/message_loop/message_loop.h" | 9 #include "base/message_loop/message_loop.h" |
10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
11 #include "net/disk_cache/disk_cache.h" | 11 #include "net/disk_cache/disk_cache.h" |
12 | 12 |
13 namespace { | 13 namespace { |
14 | 14 |
15 // Structure used for asynchronous operations. | 15 // Structure used for asynchronous operations. |
16 struct MyOverlapped { | 16 struct MyOverlapped { |
17 MyOverlapped(disk_cache::File* file, size_t offset, | 17 MyOverlapped(disk_cache::File* file, |
| 18 size_t offset, |
18 disk_cache::FileIOCallback* callback); | 19 disk_cache::FileIOCallback* callback); |
19 ~MyOverlapped() {} | 20 ~MyOverlapped() {} |
20 OVERLAPPED* overlapped() { | 21 OVERLAPPED* overlapped() { return &context_.overlapped; } |
21 return &context_.overlapped; | |
22 } | |
23 | 22 |
24 base::MessageLoopForIO::IOContext context_; | 23 base::MessageLoopForIO::IOContext context_; |
25 scoped_refptr<disk_cache::File> file_; | 24 scoped_refptr<disk_cache::File> file_; |
26 disk_cache::FileIOCallback* callback_; | 25 disk_cache::FileIOCallback* callback_; |
27 }; | 26 }; |
28 | 27 |
29 COMPILE_ASSERT(!offsetof(MyOverlapped, context_), starts_with_overlapped); | 28 COMPILE_ASSERT(!offsetof(MyOverlapped, context_), starts_with_overlapped); |
30 | 29 |
31 // Helper class to handle the IO completion notifications from the message loop. | 30 // Helper class to handle the IO completion notifications from the message loop. |
32 class CompletionHandler : public base::MessageLoopForIO::IOHandler { | 31 class CompletionHandler : public base::MessageLoopForIO::IOHandler { |
(...skipping 16 matching lines...) Expand all Loading... |
49 actual_bytes = static_cast<DWORD>(net::ERR_CACHE_READ_FAILURE); | 48 actual_bytes = static_cast<DWORD>(net::ERR_CACHE_READ_FAILURE); |
50 NOTREACHED(); | 49 NOTREACHED(); |
51 } | 50 } |
52 | 51 |
53 if (data->callback_) | 52 if (data->callback_) |
54 data->callback_->OnFileIOComplete(static_cast<int>(actual_bytes)); | 53 data->callback_->OnFileIOComplete(static_cast<int>(actual_bytes)); |
55 | 54 |
56 delete data; | 55 delete data; |
57 } | 56 } |
58 | 57 |
59 MyOverlapped::MyOverlapped(disk_cache::File* file, size_t offset, | 58 MyOverlapped::MyOverlapped(disk_cache::File* file, |
| 59 size_t offset, |
60 disk_cache::FileIOCallback* callback) { | 60 disk_cache::FileIOCallback* callback) { |
61 memset(this, 0, sizeof(*this)); | 61 memset(this, 0, sizeof(*this)); |
62 context_.handler = g_completion_handler.Pointer(); | 62 context_.handler = g_completion_handler.Pointer(); |
63 context_.overlapped.Offset = static_cast<DWORD>(offset); | 63 context_.overlapped.Offset = static_cast<DWORD>(offset); |
64 file_ = file; | 64 file_ = file; |
65 callback_ = callback; | 65 callback_ = callback; |
66 } | 66 } |
67 | 67 |
68 } // namespace | 68 } // namespace |
69 | 69 |
70 namespace disk_cache { | 70 namespace disk_cache { |
71 | 71 |
72 File::File(base::File file) | 72 File::File(base::File file) |
73 : init_(true), | 73 : init_(true), mixed_(true), sync_base_file_(file.Pass()) { |
74 mixed_(true), | |
75 sync_base_file_(file.Pass()) { | |
76 } | 74 } |
77 | 75 |
78 bool File::Init(const base::FilePath& name) { | 76 bool File::Init(const base::FilePath& name) { |
79 DCHECK(!init_); | 77 DCHECK(!init_); |
80 if (init_) | 78 if (init_) |
81 return false; | 79 return false; |
82 | 80 |
83 DWORD sharing = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE; | 81 DWORD sharing = FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE; |
84 DWORD access = GENERIC_READ | GENERIC_WRITE | DELETE; | 82 DWORD access = GENERIC_READ | GENERIC_WRITE | DELETE; |
85 base_file_ = | 83 base_file_ = base::File(CreateFile(name.value().c_str(), |
86 base::File(CreateFile(name.value().c_str(), access, sharing, NULL, | 84 access, |
87 OPEN_EXISTING, FILE_FLAG_OVERLAPPED, NULL)); | 85 sharing, |
| 86 NULL, |
| 87 OPEN_EXISTING, |
| 88 FILE_FLAG_OVERLAPPED, |
| 89 NULL)); |
88 | 90 |
89 if (!base_file_.IsValid()) | 91 if (!base_file_.IsValid()) |
90 return false; | 92 return false; |
91 | 93 |
92 base::MessageLoopForIO::current()->RegisterIOHandler( | 94 base::MessageLoopForIO::current()->RegisterIOHandler( |
93 base_file_.GetPlatformFile(), g_completion_handler.Pointer()); | 95 base_file_.GetPlatformFile(), g_completion_handler.Pointer()); |
94 | 96 |
95 init_ = true; | 97 init_ = true; |
96 sync_base_file_ = | 98 sync_base_file_ = base::File(CreateFile( |
97 base::File(CreateFile(name.value().c_str(), access, sharing, NULL, | 99 name.value().c_str(), access, sharing, NULL, OPEN_EXISTING, 0, NULL)); |
98 OPEN_EXISTING, 0, NULL)); | |
99 | 100 |
100 if (!sync_base_file_.IsValid()) | 101 if (!sync_base_file_.IsValid()) |
101 return false; | 102 return false; |
102 | 103 |
103 return true; | 104 return true; |
104 } | 105 } |
105 | 106 |
106 bool File::IsValid() const { | 107 bool File::IsValid() const { |
107 if (!init_) | 108 if (!init_) |
108 return false; | 109 return false; |
109 return base_file_.IsValid() || sync_base_file_.IsValid(); | 110 return base_file_.IsValid() || sync_base_file_.IsValid(); |
110 } | 111 } |
111 | 112 |
112 bool File::Read(void* buffer, size_t buffer_len, size_t offset) { | 113 bool File::Read(void* buffer, size_t buffer_len, size_t offset) { |
113 DCHECK(init_); | 114 DCHECK(init_); |
114 if (buffer_len > ULONG_MAX || offset > LONG_MAX) | 115 if (buffer_len > ULONG_MAX || offset > LONG_MAX) |
115 return false; | 116 return false; |
116 | 117 |
117 int ret = sync_base_file_.Read(offset, static_cast<char*>(buffer), | 118 int ret = |
118 buffer_len); | 119 sync_base_file_.Read(offset, static_cast<char*>(buffer), buffer_len); |
119 return static_cast<int>(buffer_len) == ret; | 120 return static_cast<int>(buffer_len) == ret; |
120 } | 121 } |
121 | 122 |
122 bool File::Write(const void* buffer, size_t buffer_len, size_t offset) { | 123 bool File::Write(const void* buffer, size_t buffer_len, size_t offset) { |
123 DCHECK(init_); | 124 DCHECK(init_); |
124 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) | 125 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) |
125 return false; | 126 return false; |
126 | 127 |
127 int ret = sync_base_file_.Write(offset, static_cast<const char*>(buffer), | 128 int ret = sync_base_file_.Write( |
128 buffer_len); | 129 offset, static_cast<const char*>(buffer), buffer_len); |
129 return static_cast<int>(buffer_len) == ret; | 130 return static_cast<int>(buffer_len) == ret; |
130 } | 131 } |
131 | 132 |
132 // We have to increase the ref counter of the file before performing the IO to | 133 // We have to increase the ref counter of the file before performing the IO to |
133 // prevent the completion to happen with an invalid handle (if the file is | 134 // prevent the completion to happen with an invalid handle (if the file is |
134 // closed while the IO is in flight). | 135 // closed while the IO is in flight). |
135 bool File::Read(void* buffer, size_t buffer_len, size_t offset, | 136 bool File::Read(void* buffer, |
136 FileIOCallback* callback, bool* completed) { | 137 size_t buffer_len, |
| 138 size_t offset, |
| 139 FileIOCallback* callback, |
| 140 bool* completed) { |
137 DCHECK(init_); | 141 DCHECK(init_); |
138 if (!callback) { | 142 if (!callback) { |
139 if (completed) | 143 if (completed) |
140 *completed = true; | 144 *completed = true; |
141 return Read(buffer, buffer_len, offset); | 145 return Read(buffer, buffer_len, offset); |
142 } | 146 } |
143 | 147 |
144 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) | 148 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) |
145 return false; | 149 return false; |
146 | 150 |
147 MyOverlapped* data = new MyOverlapped(this, offset, callback); | 151 MyOverlapped* data = new MyOverlapped(this, offset, callback); |
148 DWORD size = static_cast<DWORD>(buffer_len); | 152 DWORD size = static_cast<DWORD>(buffer_len); |
149 | 153 |
150 DWORD actual; | 154 DWORD actual; |
151 if (!ReadFile(base_file_.GetPlatformFile(), buffer, size, &actual, | 155 if (!ReadFile(base_file_.GetPlatformFile(), |
| 156 buffer, |
| 157 size, |
| 158 &actual, |
152 data->overlapped())) { | 159 data->overlapped())) { |
153 *completed = false; | 160 *completed = false; |
154 if (GetLastError() == ERROR_IO_PENDING) | 161 if (GetLastError() == ERROR_IO_PENDING) |
155 return true; | 162 return true; |
156 delete data; | 163 delete data; |
157 return false; | 164 return false; |
158 } | 165 } |
159 | 166 |
160 // The operation completed already. We'll be called back anyway. | 167 // The operation completed already. We'll be called back anyway. |
161 *completed = (actual == size); | 168 *completed = (actual == size); |
162 DCHECK_EQ(size, actual); | 169 DCHECK_EQ(size, actual); |
163 data->callback_ = NULL; | 170 data->callback_ = NULL; |
164 data->file_ = NULL; // There is no reason to hold on to this anymore. | 171 data->file_ = NULL; // There is no reason to hold on to this anymore. |
165 return *completed; | 172 return *completed; |
166 } | 173 } |
167 | 174 |
168 bool File::Write(const void* buffer, size_t buffer_len, size_t offset, | 175 bool File::Write(const void* buffer, |
169 FileIOCallback* callback, bool* completed) { | 176 size_t buffer_len, |
| 177 size_t offset, |
| 178 FileIOCallback* callback, |
| 179 bool* completed) { |
170 DCHECK(init_); | 180 DCHECK(init_); |
171 if (!callback) { | 181 if (!callback) { |
172 if (completed) | 182 if (completed) |
173 *completed = true; | 183 *completed = true; |
174 return Write(buffer, buffer_len, offset); | 184 return Write(buffer, buffer_len, offset); |
175 } | 185 } |
176 | 186 |
177 return AsyncWrite(buffer, buffer_len, offset, callback, completed); | 187 return AsyncWrite(buffer, buffer_len, offset, callback, completed); |
178 } | 188 } |
179 | 189 |
180 File::~File() { | 190 File::~File() { |
181 } | 191 } |
182 | 192 |
183 base::PlatformFile File::platform_file() const { | 193 base::PlatformFile File::platform_file() const { |
184 DCHECK(init_); | 194 DCHECK(init_); |
185 return base_file_.IsValid() ? base_file_.GetPlatformFile() : | 195 return base_file_.IsValid() ? base_file_.GetPlatformFile() |
186 sync_base_file_.GetPlatformFile(); | 196 : sync_base_file_.GetPlatformFile(); |
187 } | 197 } |
188 | 198 |
189 bool File::AsyncWrite(const void* buffer, size_t buffer_len, size_t offset, | 199 bool File::AsyncWrite(const void* buffer, |
190 FileIOCallback* callback, bool* completed) { | 200 size_t buffer_len, |
| 201 size_t offset, |
| 202 FileIOCallback* callback, |
| 203 bool* completed) { |
191 DCHECK(init_); | 204 DCHECK(init_); |
192 DCHECK(callback); | 205 DCHECK(callback); |
193 DCHECK(completed); | 206 DCHECK(completed); |
194 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) | 207 if (buffer_len > ULONG_MAX || offset > ULONG_MAX) |
195 return false; | 208 return false; |
196 | 209 |
197 MyOverlapped* data = new MyOverlapped(this, offset, callback); | 210 MyOverlapped* data = new MyOverlapped(this, offset, callback); |
198 DWORD size = static_cast<DWORD>(buffer_len); | 211 DWORD size = static_cast<DWORD>(buffer_len); |
199 | 212 |
200 DWORD actual; | 213 DWORD actual; |
201 if (!WriteFile(base_file_.GetPlatformFile(), buffer, size, &actual, | 214 if (!WriteFile(base_file_.GetPlatformFile(), |
| 215 buffer, |
| 216 size, |
| 217 &actual, |
202 data->overlapped())) { | 218 data->overlapped())) { |
203 *completed = false; | 219 *completed = false; |
204 if (GetLastError() == ERROR_IO_PENDING) | 220 if (GetLastError() == ERROR_IO_PENDING) |
205 return true; | 221 return true; |
206 delete data; | 222 delete data; |
207 return false; | 223 return false; |
208 } | 224 } |
209 | 225 |
210 // The operation completed already. We'll be called back anyway. | 226 // The operation completed already. We'll be called back anyway. |
211 *completed = (actual == size); | 227 *completed = (actual == size); |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
248 base::MessageLoopForIO::IOHandler* handler = g_completion_handler.Pointer(); | 264 base::MessageLoopForIO::IOHandler* handler = g_completion_handler.Pointer(); |
249 base::MessageLoopForIO::current()->WaitForIOCompletion(100, handler); | 265 base::MessageLoopForIO::current()->WaitForIOCompletion(100, handler); |
250 } | 266 } |
251 } | 267 } |
252 | 268 |
253 // Static. | 269 // Static. |
254 void File::DropPendingIO() { | 270 void File::DropPendingIO() { |
255 } | 271 } |
256 | 272 |
257 } // namespace disk_cache | 273 } // namespace disk_cache |
OLD | NEW |