Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "content/browser/service_worker/service_worker_cache_writer.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "content/browser/appcache/appcache_response.h" | |
| 10 #include "content/browser/service_worker/service_worker_disk_cache.h" | |
| 11 #include "content/browser/service_worker/service_worker_storage.h" | |
| 12 | |
| 13 namespace { | |
| 14 | |
| 15 const size_t kCopyBufferSize = 16 * 1024; | |
| 16 | |
| 17 enum { | |
| 18 STATE_READ_HEADERS_FOR_COMPARE, | |
| 19 STATE_READ_HEADERS_FOR_COMPARE_DONE, | |
| 20 STATE_READ_DATA_FOR_COMPARE, | |
| 21 STATE_READ_DATA_FOR_COMPARE_DONE, | |
| 22 STATE_READ_HEADERS_FOR_COPY, | |
| 23 STATE_READ_HEADERS_FOR_COPY_DONE, | |
| 24 STATE_READ_DATA_FOR_COPY, | |
| 25 STATE_READ_DATA_FOR_COPY_DONE, | |
| 26 STATE_WRITE_HEADERS_FOR_PASSTHROUGH, | |
| 27 STATE_WRITE_HEADERS_FOR_PASSTHROUGH_DONE, | |
| 28 STATE_WRITE_DATA_FOR_PASSTHROUGH, | |
| 29 STATE_WRITE_DATA_FOR_PASSTHROUGH_DONE, | |
| 30 STATE_WRITE_HEADERS_FOR_COPY, | |
| 31 STATE_WRITE_HEADERS_FOR_COPY_DONE, | |
| 32 STATE_WRITE_DATA_FOR_COPY, | |
| 33 STATE_WRITE_DATA_FOR_COPY_DONE, | |
| 34 }; | |
| 35 | |
| 36 // Shim class used to turn always-async functions into async-or-result | |
| 37 // functions. See the comments below near ReadInfoHelper. | |
| 38 class AsyncOnlyCompletionCallbackAdaptor | |
| 39 : public base::RefCounted<AsyncOnlyCompletionCallbackAdaptor> { | |
| 40 public: | |
| 41 AsyncOnlyCompletionCallbackAdaptor(const net::CompletionCallback& callback) | |
| 42 : done_(false), | |
| 43 async_(false), | |
| 44 result_(net::ERR_IO_PENDING), | |
| 45 callback_(callback) {} | |
| 46 | |
| 47 void set_async(bool async) { async_ = async; } | |
| 48 bool async() { return async_; } | |
| 49 bool done() { return done_; } | |
| 50 int result() { return result_; } | |
| 51 | |
| 52 void WrappedCallback(int result) { | |
| 53 result_ = result; | |
| 54 if (async_) | |
| 55 callback_.Run(result); | |
| 56 else | |
| 57 done_ = true; | |
| 58 } | |
| 59 | |
| 60 private: | |
| 61 friend class base::RefCounted<AsyncOnlyCompletionCallbackAdaptor>; | |
| 62 virtual ~AsyncOnlyCompletionCallbackAdaptor() {} | |
| 63 | |
| 64 bool done_; | |
| 65 bool async_; | |
| 66 int result_; | |
| 67 net::CompletionCallback callback_; | |
| 68 }; | |
| 69 | |
| 70 } // namespace | |
| 71 | |
| 72 namespace content { | |
| 73 | |
| 74 ServiceWorkerCacheWriter::StateMachine::State::State() {} | |
| 75 | |
| 76 ServiceWorkerCacheWriter::StateMachine::State::State( | |
| 77 int id, | |
| 78 const std::string& name, | |
| 79 const StateMachine::Handler& handler) | |
| 80 : id(id), name(name), handler(handler) {} | |
| 81 | |
| 82 ServiceWorkerCacheWriter::StateMachine::State::~State() {} | |
| 83 | |
| 84 ServiceWorkerCacheWriter::StateMachine::StateMachine() : state_(STATE_START) {} | |
| 85 ServiceWorkerCacheWriter::StateMachine::~StateMachine() {} | |
| 86 | |
| 87 void ServiceWorkerCacheWriter::StateMachine::AddState( | |
| 88 int id, | |
| 89 const std::string& name, | |
| 90 const StateMachine::Handler& handler) { | |
| 91 State state(id, name, handler); | |
| 92 states_[id] = state; | |
| 93 } | |
| 94 | |
| 95 void ServiceWorkerCacheWriter::StateMachine::AddTransition(int from_id, | |
| 96 int to_id) { | |
| 97 DCHECK_EQ(1U, states_.count(from_id)); | |
| 98 DCHECK_EQ(1U, states_.count(to_id)); | |
| 99 DCHECK_NE(to_id, STATE_START); | |
| 100 DCHECK_NE(from_id, STATE_DONE); | |
| 101 states_[from_id].allowed_to.insert(to_id); | |
| 102 } | |
| 103 | |
| 104 void ServiceWorkerCacheWriter::StateMachine::CheckValidTransition(int from_id, | |
| 105 int to_id) { | |
| 106 DCHECK_EQ(1U, states_.count(from_id)) << " State " << from_id | |
| 107 << " does not exist."; | |
| 108 DCHECK_EQ(1U, states_.count(to_id)) << " State " << to_id << " (from " | |
| 109 << states_[from_id].name | |
| 110 << ") does not exist."; | |
| 111 if (to_id == STATE_DONE) | |
| 112 return; | |
| 113 DCHECK_EQ(1U, states_[from_id].allowed_to.count(to_id)) | |
| 114 << " Transition from " << states_[from_id].name << " to " | |
| 115 << states_[to_id].name << " is not allowed."; | |
| 116 } | |
| 117 | |
| 118 int ServiceWorkerCacheWriter::StateMachine::Run(int status) { | |
| 119 DCHECK_EQ(1U, states_.count(state_)); | |
| 120 bool pause = false; | |
| 121 VLOG(1) << "Run: initial " << states_[state_].name; | |
| 122 do { | |
| 123 int next_state = -1; | |
| 124 VLOG(1) << " running " << states_[state_].name; | |
| 125 status = states_[state_].handler.Run(&next_state, &pause, status); | |
| 126 VLOG(1) << " -> " << states_[next_state].name << " " << pause << " " | |
| 127 << status; | |
| 128 CheckValidTransition(state_, next_state); | |
| 129 state_ = next_state; | |
| 130 } while (status >= net::OK && state_ != STATE_DONE && !pause); | |
| 131 return status; | |
| 132 } | |
| 133 | |
| 134 ServiceWorkerCacheWriter::ServiceWorkerCacheWriter( | |
| 135 const ResponseReaderCreator& reader_creator, | |
| 136 const ResponseWriterCreator& writer_creator) | |
| 137 : reader_creator_(reader_creator), | |
| 138 writer_creator_(writer_creator), | |
| 139 weak_factory_(this) { | |
| 140 // This macro defines the state |id| to be named "name" and be bound to this | |
| 141 // object's method called name, so: | |
| 142 // DEFINE_STATE(STATE_FOO, Foo) | |
| 143 // Defines a state STATE_FOO, named "Foo", that is bound to this->Foo(). | |
| 144 #define DEFINE_STATE(id, name) \ | |
| 145 state_machine_.AddState( \ | |
| 146 id, #name, \ | |
| 147 base::Bind(&ServiceWorkerCacheWriter::name, base::Unretained(this))) | |
| 148 | |
| 149 DEFINE_STATE(StateMachine::STATE_START, Start); | |
| 150 DEFINE_STATE(StateMachine::STATE_DONE, Done); | |
| 151 | |
| 152 DEFINE_STATE(STATE_WRITE_HEADERS_FOR_PASSTHROUGH, WriteHeadersForPassthrough); | |
| 153 DEFINE_STATE(STATE_WRITE_HEADERS_FOR_PASSTHROUGH_DONE, | |
| 154 WriteHeadersForPassthroughDone); | |
| 155 DEFINE_STATE(STATE_WRITE_DATA_FOR_PASSTHROUGH, WriteDataForPassthrough); | |
| 156 DEFINE_STATE(STATE_WRITE_DATA_FOR_PASSTHROUGH_DONE, | |
| 157 WriteDataForPassthroughDone); | |
| 158 | |
| 159 DEFINE_STATE(STATE_READ_HEADERS_FOR_COMPARE, ReadHeadersForCompare); | |
| 160 DEFINE_STATE(STATE_READ_HEADERS_FOR_COMPARE_DONE, ReadHeadersForCompareDone); | |
| 161 DEFINE_STATE(STATE_READ_DATA_FOR_COMPARE, ReadDataForCompare); | |
| 162 DEFINE_STATE(STATE_READ_DATA_FOR_COMPARE_DONE, ReadDataForCompareDone); | |
| 163 | |
| 164 DEFINE_STATE(STATE_READ_HEADERS_FOR_COPY, ReadHeadersForCopy); | |
| 165 DEFINE_STATE(STATE_READ_HEADERS_FOR_COPY_DONE, ReadHeadersForCopyDone); | |
| 166 DEFINE_STATE(STATE_WRITE_HEADERS_FOR_COPY, WriteHeadersForCopy); | |
| 167 DEFINE_STATE(STATE_WRITE_HEADERS_FOR_COPY_DONE, WriteHeadersForCopyDone); | |
| 168 | |
| 169 DEFINE_STATE(STATE_READ_DATA_FOR_COPY, ReadDataForCopy); | |
| 170 DEFINE_STATE(STATE_READ_DATA_FOR_COPY_DONE, ReadDataForCopyDone); | |
| 171 DEFINE_STATE(STATE_WRITE_DATA_FOR_COPY, WriteDataForCopy); | |
| 172 DEFINE_STATE(STATE_WRITE_DATA_FOR_COPY_DONE, WriteDataForCopyDone); | |
| 173 | |
| 174 #undef DEFINE_STATE | |
| 175 | |
| 176 state_machine_.AddTransition(StateMachine::STATE_START, | |
| 177 STATE_WRITE_HEADERS_FOR_PASSTHROUGH); | |
| 178 state_machine_.AddTransition(StateMachine::STATE_START, | |
| 179 STATE_READ_HEADERS_FOR_COMPARE); | |
| 180 // The passthrough write loop: | |
| 181 // WriteHeaders -> WriteHeadersDone -> WriteData -> WriteDataDone | |
| 182 // -> WriteData | |
| 183 state_machine_.AddTransition(STATE_WRITE_HEADERS_FOR_PASSTHROUGH, | |
| 184 STATE_WRITE_HEADERS_FOR_PASSTHROUGH_DONE); | |
| 185 state_machine_.AddTransition(STATE_WRITE_HEADERS_FOR_PASSTHROUGH_DONE, | |
| 186 STATE_WRITE_DATA_FOR_PASSTHROUGH); | |
| 187 state_machine_.AddTransition(STATE_WRITE_DATA_FOR_PASSTHROUGH, | |
| 188 STATE_WRITE_DATA_FOR_PASSTHROUGH_DONE); | |
| 189 state_machine_.AddTransition(STATE_WRITE_DATA_FOR_PASSTHROUGH_DONE, | |
| 190 STATE_WRITE_DATA_FOR_PASSTHROUGH); | |
| 191 | |
| 192 // The comparator loop: | |
| 193 // ReadHeaders -> ReadHeadersDone -> ReadData -> ReadDataDone -> ReadData | |
| 194 // Exits via ReadHeadersForCopy. | |
| 195 // ReadDataForCompareDone has a self-edge, see the comments in the | |
| 196 // function for details. | |
| 197 state_machine_.AddTransition(STATE_READ_HEADERS_FOR_COMPARE, | |
| 198 STATE_READ_HEADERS_FOR_COMPARE_DONE); | |
| 199 state_machine_.AddTransition(STATE_READ_HEADERS_FOR_COMPARE_DONE, | |
| 200 STATE_READ_DATA_FOR_COMPARE); | |
| 201 state_machine_.AddTransition(STATE_READ_DATA_FOR_COMPARE, | |
| 202 STATE_READ_DATA_FOR_COMPARE_DONE); | |
| 203 state_machine_.AddTransition(STATE_READ_DATA_FOR_COMPARE_DONE, | |
| 204 STATE_READ_DATA_FOR_COMPARE); | |
| 205 state_machine_.AddTransition(STATE_READ_DATA_FOR_COMPARE_DONE, | |
| 206 STATE_READ_DATA_FOR_COMPARE_DONE); | |
| 207 state_machine_.AddTransition(STATE_READ_DATA_FOR_COMPARE_DONE, | |
| 208 STATE_READ_HEADERS_FOR_COPY); | |
| 209 | |
| 210 // ReadHeaders -> ReadHeadersDone -> WriteHeaders -> WriteHeadersDone | |
| 211 // -> ReadData | |
| 212 state_machine_.AddTransition(STATE_READ_HEADERS_FOR_COPY, | |
| 213 STATE_READ_HEADERS_FOR_COPY_DONE); | |
| 214 state_machine_.AddTransition(STATE_READ_HEADERS_FOR_COPY_DONE, | |
| 215 STATE_WRITE_HEADERS_FOR_COPY); | |
| 216 state_machine_.AddTransition(STATE_WRITE_HEADERS_FOR_COPY, | |
| 217 STATE_WRITE_HEADERS_FOR_COPY_DONE); | |
| 218 state_machine_.AddTransition(STATE_WRITE_HEADERS_FOR_COPY_DONE, | |
| 219 STATE_READ_DATA_FOR_COPY); | |
| 220 | |
| 221 // The ReadDataForCopy loop: | |
| 222 // ReadData -> ReadDataDone -> WriteData -> WriteDataDone -> ReadData | |
| 223 // This can bail out via: | |
| 224 // ReadData -> WriteDataForPassthrough | |
| 225 state_machine_.AddTransition(STATE_READ_DATA_FOR_COPY, | |
| 226 STATE_READ_DATA_FOR_COPY_DONE); | |
| 227 state_machine_.AddTransition(STATE_READ_DATA_FOR_COPY_DONE, | |
| 228 STATE_WRITE_DATA_FOR_COPY); | |
| 229 state_machine_.AddTransition(STATE_WRITE_DATA_FOR_COPY, | |
| 230 STATE_WRITE_DATA_FOR_COPY_DONE); | |
| 231 state_machine_.AddTransition(STATE_WRITE_DATA_FOR_COPY_DONE, | |
| 232 STATE_READ_DATA_FOR_COPY); | |
| 233 state_machine_.AddTransition(STATE_READ_DATA_FOR_COPY, | |
| 234 STATE_WRITE_DATA_FOR_PASSTHROUGH); | |
| 235 } | |
| 236 | |
| 237 ServiceWorkerCacheWriter::~ServiceWorkerCacheWriter() {} | |
| 238 | |
| 239 net::Error ServiceWorkerCacheWriter::WriteHeaders( | |
| 240 HttpResponseInfoIOBuffer* headers, | |
| 241 const OnWriteCompleteCallback& callback) { | |
| 242 headers_to_write_ = headers; | |
| 243 pending_callback_ = callback; | |
| 244 int result = state_machine_.Run(net::OK); | |
|
falken
2015/08/28 05:45:24
When trying to trace through this code I found it'
Elly Fong-Jones
2015/08/31 15:03:45
We're actually in STATE_START here. I've added a D
| |
| 245 return result >= 0 ? net::OK : static_cast<net::Error>(result); | |
| 246 } | |
| 247 | |
| 248 net::Error ServiceWorkerCacheWriter::WriteData( | |
| 249 net::IOBuffer* buf, | |
| 250 size_t buf_size, | |
| 251 const OnWriteCompleteCallback& callback) { | |
| 252 data_to_write_ = buf; | |
| 253 len_to_write_ = buf_size; | |
| 254 pending_callback_ = callback; | |
| 255 int result = state_machine_.Run(net::OK); | |
| 256 return result >= 0 ? net::OK : static_cast<net::Error>(result); | |
| 257 } | |
| 258 | |
| 259 size_t ServiceWorkerCacheWriter::BytesWritten() const { | |
| 260 return bytes_written_; | |
| 261 } | |
| 262 | |
| 263 int ServiceWorkerCacheWriter::Start(int* next_state, bool* pause, int result) { | |
| 264 bytes_written_ = 0; | |
| 265 compare_reader_ = reader_creator_.Run(); | |
| 266 if (compare_reader_.get()) | |
| 267 *next_state = STATE_READ_HEADERS_FOR_COMPARE; | |
| 268 else | |
| 269 // No existing reader, just write the headers back directly. | |
| 270 *next_state = STATE_WRITE_HEADERS_FOR_PASSTHROUGH; | |
| 271 return net::OK; | |
| 272 } | |
| 273 | |
| 274 int ServiceWorkerCacheWriter::ReadHeadersForCompare(int* next_state, | |
| 275 bool* pause, | |
| 276 int result) { | |
| 277 DCHECK(headers_to_write_); | |
| 278 | |
| 279 headers_to_read_ = new HttpResponseInfoIOBuffer; | |
| 280 *next_state = STATE_READ_HEADERS_FOR_COMPARE_DONE; | |
| 281 return ReadInfoHelper(compare_reader_, headers_to_read_.get()); | |
| 282 } | |
| 283 | |
| 284 int ServiceWorkerCacheWriter::ReadHeadersForCompareDone(int* next_state, | |
| 285 bool* pause, | |
| 286 int result) { | |
| 287 if (result < 0) { | |
| 288 *next_state = StateMachine::STATE_DONE; | |
| 289 return static_cast<int>(result); | |
| 290 } | |
| 291 cached_length_ = headers_to_read_->response_data_size; | |
| 292 net_length_ = headers_to_write_->response_data_size; | |
| 293 bytes_compared_ = 0; | |
| 294 *pause = true; | |
| 295 *next_state = STATE_READ_DATA_FOR_COMPARE; | |
| 296 return net::OK; | |
| 297 } | |
| 298 | |
| 299 int ServiceWorkerCacheWriter::ReadDataForCompare(int* next_state, | |
| 300 bool* pause, | |
| 301 int result) { | |
| 302 DCHECK(data_to_write_); | |
| 303 | |
| 304 data_to_read_ = new net::IOBuffer(len_to_write_); | |
| 305 len_to_read_ = len_to_write_; | |
| 306 *next_state = STATE_READ_DATA_FOR_COMPARE_DONE; | |
| 307 compare_offset_ = 0; | |
| 308 return ReadDataHelper(compare_reader_, data_to_read_.get(), len_to_read_); | |
| 309 } | |
| 310 | |
| 311 int ServiceWorkerCacheWriter::ReadDataForCompareDone(int* next_state, | |
| 312 bool* pause, | |
| 313 int result) { | |
| 314 DCHECK(data_to_read_); | |
| 315 DCHECK(data_to_write_); | |
| 316 DCHECK_EQ(len_to_read_, len_to_write_); | |
| 317 DCHECK_LE(result + compare_offset_, static_cast<size_t>(len_to_write_)); | |
| 318 | |
| 319 if (result < 0) { | |
| 320 *next_state = StateMachine::STATE_DONE; | |
| 321 return result; | |
| 322 } | |
| 323 | |
| 324 // Premature EOF while reading the AppCache data to compare. Fail the | |
|
falken
2015/08/28 05:45:24
I wouldn't call it AppCache data. Service worker h
Elly Fong-Jones
2015/08/31 15:03:45
Done.
| |
| 325 // comparison. | |
| 326 if (result == 0) { | |
| 327 *next_state = STATE_READ_HEADERS_FOR_COPY; | |
| 328 return net::OK; | |
| 329 } | |
| 330 | |
| 331 // Compare the data from AppCache to the data from the network. | |
| 332 if (memcmp(data_to_read_->data(), data_to_write_->data() + compare_offset_, | |
| 333 result)) { | |
| 334 // Data mismatched. This method already validated that all the bytes through | |
| 335 // |bytes_compared_| were identical, so copy the first |bytes_compared_| | |
| 336 // over, then start writing network data back after the changed point. | |
| 337 // | |
| 338 // Note that the state machine does NOT get paused here, since there is | |
| 339 // still further work to do before this object is ready to handle another | |
| 340 // WriteData call. | |
| 341 *next_state = STATE_READ_HEADERS_FOR_COPY; | |
| 342 return net::OK; | |
| 343 } | |
| 344 | |
| 345 compare_offset_ += result; | |
| 346 | |
| 347 // This is a little bit tricky. It is possible that not enough data was read | |
| 348 // to finish comparing the entire block of data from the network (which is | |
| 349 // kept in len_to_write_), so this method may need to issue another read and | |
| 350 // return to this state. | |
| 351 // | |
| 352 // Compare isn't complete yet. Issue another read for the remaining data. Note | |
| 353 // that this reuses the same IOBuffer. | |
| 354 if (compare_offset_ < static_cast<size_t>(len_to_read_)) { | |
| 355 *next_state = STATE_READ_DATA_FOR_COMPARE_DONE; | |
| 356 return ReadDataHelper(compare_reader_, data_to_read_.get(), | |
| 357 len_to_read_ - compare_offset_); | |
| 358 } | |
| 359 | |
| 360 // Cached entry is longer than the network entry but the prefix matches. Copy | |
| 361 // just the prefix. | |
| 362 if (bytes_compared_ + compare_offset_ >= net_length_ && | |
| 363 net_length_ < cached_length_) { | |
| 364 *next_state = STATE_READ_HEADERS_FOR_COPY; | |
| 365 return net::OK; | |
| 366 } | |
| 367 | |
| 368 // bytes_compared_ only gets incremented when a full block is compared, to | |
| 369 // avoid having to use only parts of the buffered network data. | |
| 370 bytes_compared_ += result; | |
| 371 *next_state = STATE_READ_DATA_FOR_COMPARE; | |
| 372 *pause = true; | |
| 373 return net::OK; | |
| 374 } | |
| 375 | |
| 376 int ServiceWorkerCacheWriter::WriteHeadersForPassthrough(int* next_state, | |
| 377 bool* pause, | |
| 378 int result) { | |
| 379 writer_ = writer_creator_.Run(); | |
| 380 *next_state = STATE_WRITE_HEADERS_FOR_PASSTHROUGH_DONE; | |
| 381 return WriteInfoHelper(writer_, headers_to_write_.get()); | |
| 382 } | |
| 383 | |
| 384 int ServiceWorkerCacheWriter::WriteHeadersForPassthroughDone(int* next_state, | |
| 385 bool* pause, | |
| 386 int result) { | |
| 387 *next_state = STATE_WRITE_DATA_FOR_PASSTHROUGH; | |
| 388 *pause = true; | |
| 389 return net::OK; | |
| 390 } | |
| 391 | |
| 392 int ServiceWorkerCacheWriter::WriteDataForPassthrough(int* next_state, | |
| 393 bool* pause, | |
| 394 int result) { | |
| 395 *next_state = STATE_WRITE_DATA_FOR_PASSTHROUGH_DONE; | |
| 396 return WriteDataHelper(writer_, data_to_write_.get(), len_to_write_); | |
| 397 } | |
| 398 | |
| 399 int ServiceWorkerCacheWriter::WriteDataForPassthroughDone(int* next_state, | |
| 400 bool* pause, | |
| 401 int result) { | |
| 402 if (result < 0) { | |
| 403 *next_state = StateMachine::STATE_DONE; | |
| 404 return static_cast<int>(result); | |
| 405 } | |
| 406 bytes_written_ += result; | |
| 407 *next_state = STATE_WRITE_DATA_FOR_PASSTHROUGH; | |
| 408 *pause = true; | |
| 409 return net::OK; | |
| 410 } | |
| 411 | |
| 412 int ServiceWorkerCacheWriter::ReadHeadersForCopy(int* next_state, | |
| 413 bool* pause, | |
| 414 int result) { | |
| 415 bytes_copied_ = 0; | |
| 416 copy_reader_ = reader_creator_.Run(); | |
| 417 headers_to_read_ = new HttpResponseInfoIOBuffer; | |
| 418 data_to_copy_ = new net::IOBuffer(kCopyBufferSize); | |
| 419 *next_state = STATE_READ_HEADERS_FOR_COPY_DONE; | |
| 420 return ReadInfoHelper(copy_reader_, headers_to_read_.get()); | |
| 421 } | |
| 422 | |
| 423 int ServiceWorkerCacheWriter::ReadHeadersForCopyDone(int* next_state, | |
| 424 bool* pause, | |
| 425 int result) { | |
| 426 if (result < 0) { | |
| 427 *next_state = StateMachine::STATE_DONE; | |
| 428 return static_cast<int>(result); | |
| 429 } | |
| 430 *next_state = STATE_WRITE_HEADERS_FOR_COPY; | |
| 431 return net::OK; | |
| 432 } | |
| 433 | |
| 434 // Write the just-read headers back to the cache. | |
| 435 // Note that this method must create |writer_|, since the only paths to this | |
| 436 // state never create a writer. | |
| 437 // Also note that this *discards* the read headers and replaces them with the | |
| 438 // net headers. | |
| 439 int ServiceWorkerCacheWriter::WriteHeadersForCopy(int* next_state, | |
| 440 bool* pause, | |
| 441 int result) { | |
| 442 DCHECK(!writer_); | |
| 443 writer_ = writer_creator_.Run(); | |
| 444 *next_state = STATE_WRITE_HEADERS_FOR_COPY_DONE; | |
| 445 return WriteInfoHelper(writer_, headers_to_write_.get()); | |
| 446 } | |
| 447 | |
| 448 int ServiceWorkerCacheWriter::WriteHeadersForCopyDone(int* next_state, | |
| 449 bool* pause, | |
| 450 int result) { | |
| 451 if (result < 0) { | |
| 452 *next_state = StateMachine::STATE_DONE; | |
| 453 return static_cast<int>(result); | |
| 454 } | |
| 455 *next_state = STATE_READ_DATA_FOR_COPY; | |
| 456 return net::OK; | |
| 457 } | |
| 458 | |
| 459 int ServiceWorkerCacheWriter::ReadDataForCopy(int* next_state, | |
| 460 bool* pause, | |
| 461 int result) { | |
| 462 size_t to_read = std::min(kCopyBufferSize, bytes_compared_ - bytes_copied_); | |
| 463 // At this point, all compared bytes have been read. Currently | |
| 464 // |data_to_write_| and |len_to_write_| hold the chunk of network input that | |
| 465 // caused the comparison failure, so those need to be written back and this | |
| 466 // object needs to go into passthrough mode. | |
| 467 if (to_read == 0) { | |
| 468 *next_state = STATE_WRITE_DATA_FOR_PASSTHROUGH; | |
| 469 return net::OK; | |
| 470 } | |
| 471 *next_state = STATE_READ_DATA_FOR_COPY_DONE; | |
| 472 return ReadDataHelper(copy_reader_, data_to_copy_.get(), to_read); | |
| 473 } | |
| 474 | |
| 475 int ServiceWorkerCacheWriter::ReadDataForCopyDone(int* next_state, | |
| 476 bool* pause, | |
| 477 int result) { | |
| 478 if (result < 0) { | |
| 479 *next_state = StateMachine::STATE_DONE; | |
| 480 return result; | |
| 481 } | |
| 482 *next_state = STATE_WRITE_DATA_FOR_COPY; | |
| 483 return result; | |
| 484 } | |
| 485 | |
| 486 int ServiceWorkerCacheWriter::WriteDataForCopy(int* next_state, | |
| 487 bool* pause, | |
| 488 int result) { | |
| 489 *next_state = STATE_WRITE_DATA_FOR_COPY_DONE; | |
| 490 DCHECK_GT(result, 0); | |
| 491 return WriteDataHelper(writer_, data_to_copy_.get(), result); | |
| 492 } | |
| 493 | |
| 494 int ServiceWorkerCacheWriter::WriteDataForCopyDone(int* next_state, | |
| 495 bool* pause, | |
| 496 int result) { | |
| 497 if (result < 0) { | |
| 498 *next_state = StateMachine::STATE_DONE; | |
| 499 return result; | |
| 500 } | |
| 501 bytes_written_ += result; | |
| 502 bytes_copied_ += result; | |
| 503 *next_state = STATE_READ_DATA_FOR_COPY; | |
| 504 return result; | |
| 505 } | |
| 506 | |
| 507 int ServiceWorkerCacheWriter::Done(int* next_state, bool* pause, int result) { | |
| 508 *next_state = StateMachine::STATE_DONE; | |
| 509 return net::OK; | |
| 510 } | |
| 511 | |
| 512 // These helpers adapt the AppCache "always use the callback" pattern to the | |
| 513 // //net "only use the callback for async" pattern using | |
| 514 // AsyncCompletionCallbackAdaptor. | |
| 515 // | |
| 516 // Specifically, these methods return result codes directly for synchronous | |
| 517 // completions, and only run their callback (which is AsyncRunStateMachine) for | |
| 518 // asynchronous completions. | |
| 519 | |
| 520 int ServiceWorkerCacheWriter::ReadInfoHelper( | |
| 521 const scoped_ptr<ServiceWorkerResponseReader>& reader, | |
| 522 HttpResponseInfoIOBuffer* buf) { | |
| 523 net::CompletionCallback run_callback = | |
| 524 base::Bind(&ServiceWorkerCacheWriter::AsyncRunStateMachine, | |
| 525 weak_factory_.GetWeakPtr()); | |
| 526 scoped_refptr<AsyncOnlyCompletionCallbackAdaptor> adaptor( | |
| 527 new AsyncOnlyCompletionCallbackAdaptor(run_callback)); | |
| 528 reader->ReadInfo( | |
| 529 buf, base::Bind(&AsyncOnlyCompletionCallbackAdaptor::WrappedCallback, | |
| 530 adaptor)); | |
| 531 adaptor->set_async(true); | |
| 532 return adaptor->result(); | |
| 533 } | |
| 534 | |
| 535 int ServiceWorkerCacheWriter::ReadDataHelper( | |
| 536 const scoped_ptr<ServiceWorkerResponseReader>& reader, | |
| 537 net::IOBuffer* buf, | |
| 538 int buf_len) { | |
| 539 net::CompletionCallback run_callback = | |
| 540 base::Bind(&ServiceWorkerCacheWriter::AsyncRunStateMachine, | |
| 541 weak_factory_.GetWeakPtr()); | |
| 542 scoped_refptr<AsyncOnlyCompletionCallbackAdaptor> adaptor( | |
| 543 new AsyncOnlyCompletionCallbackAdaptor(run_callback)); | |
| 544 reader->ReadData( | |
| 545 buf, buf_len, | |
| 546 base::Bind(&AsyncOnlyCompletionCallbackAdaptor::WrappedCallback, | |
| 547 adaptor)); | |
| 548 adaptor->set_async(true); | |
| 549 return adaptor->result(); | |
| 550 } | |
| 551 | |
| 552 int ServiceWorkerCacheWriter::WriteInfoHelper( | |
| 553 const scoped_ptr<ServiceWorkerResponseWriter>& writer, | |
| 554 HttpResponseInfoIOBuffer* buf) { | |
| 555 net::CompletionCallback run_callback = | |
| 556 base::Bind(&ServiceWorkerCacheWriter::AsyncRunStateMachine, | |
| 557 weak_factory_.GetWeakPtr()); | |
| 558 scoped_refptr<AsyncOnlyCompletionCallbackAdaptor> adaptor( | |
| 559 new AsyncOnlyCompletionCallbackAdaptor(run_callback)); | |
| 560 writer->WriteInfo( | |
| 561 buf, base::Bind(&AsyncOnlyCompletionCallbackAdaptor::WrappedCallback, | |
| 562 adaptor)); | |
| 563 adaptor->set_async(true); | |
| 564 return adaptor->result(); | |
| 565 } | |
| 566 | |
| 567 int ServiceWorkerCacheWriter::WriteDataHelper( | |
| 568 const scoped_ptr<ServiceWorkerResponseWriter>& writer, | |
| 569 net::IOBuffer* buf, | |
| 570 int buf_len) { | |
| 571 net::CompletionCallback run_callback = | |
| 572 base::Bind(&ServiceWorkerCacheWriter::AsyncRunStateMachine, | |
| 573 weak_factory_.GetWeakPtr()); | |
| 574 scoped_refptr<AsyncOnlyCompletionCallbackAdaptor> adaptor( | |
| 575 new AsyncOnlyCompletionCallbackAdaptor(run_callback)); | |
| 576 writer->WriteData( | |
| 577 buf, buf_len, | |
| 578 base::Bind(&AsyncOnlyCompletionCallbackAdaptor::WrappedCallback, | |
| 579 adaptor)); | |
| 580 adaptor->set_async(true); | |
| 581 return adaptor->result(); | |
| 582 } | |
| 583 | |
| 584 void ServiceWorkerCacheWriter::AsyncRunStateMachine(int result) { | |
| 585 result = state_machine_.Run(result); | |
| 586 // If the result is ERR_IO_PENDING, the pending callback will be run by a | |
| 587 // later invocation of AsyncRunStateMachine. | |
| 588 if (result != net::ERR_IO_PENDING) { | |
| 589 OnWriteCompleteCallback callback = pending_callback_; | |
| 590 pending_callback_.Reset(); | |
| 591 net::Error error = result >= 0 ? net::OK : static_cast<net::Error>(result); | |
| 592 callback.Run(error); | |
| 593 } | |
| 594 } | |
| 595 | |
| 596 } // namespace content | |
| OLD | NEW |