| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright (c) 2013 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/streams/stream.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/message_loop_proxy.h" | |
| 9 #include "content/browser/streams/stream_read_observer.h" | |
| 10 #include "content/browser/streams/stream_registry.h" | |
| 11 #include "content/browser/streams/stream_write_observer.h" | |
| 12 #include "net/base/io_buffer.h" | |
| 13 | |
| 14 namespace { | |
| 15 // Start throttling the connection at about 1MB. | |
| 16 const size_t kDeferSizeThreshold = 40 * 32768; | |
| 17 } | |
| 18 | |
| 19 namespace content { | |
| 20 | |
| 21 Stream::Stream(StreamRegistry* registry, | |
| 22 StreamWriteObserver* write_observer, | |
| 23 const GURL& security_origin, | |
| 24 const GURL& url) | |
| 25 : bytes_read_(0), | |
| 26 can_add_data_(true), | |
| 27 security_origin_(security_origin), | |
| 28 url_(url), | |
| 29 data_length_(0), | |
| 30 registry_(registry), | |
| 31 read_observer_(NULL), | |
| 32 write_observer_(write_observer) { | |
| 33 CreateByteStream(base::MessageLoopProxy::current(), | |
| 34 base::MessageLoopProxy::current(), | |
| 35 kDeferSizeThreshold, | |
| 36 &writer_, | |
| 37 &reader_); | |
| 38 | |
| 39 // Setup callback for writing. | |
| 40 writer_->RegisterCallback(base::Bind(&Stream::OnSpaceAvailable, this)); | |
| 41 reader_->RegisterCallback(base::Bind(&Stream::OnDataAvailable, this)); | |
| 42 | |
| 43 registry_->RegisterStream(this); | |
| 44 } | |
| 45 | |
| 46 Stream::~Stream() { | |
| 47 } | |
| 48 | |
| 49 bool Stream::SetReadObserver(StreamReadObserver* observer) { | |
| 50 if (read_observer_) | |
| 51 return false; | |
| 52 read_observer_ = observer; | |
| 53 return true; | |
| 54 } | |
| 55 | |
| 56 void Stream::RemoveReadObserver(StreamReadObserver* observer) { | |
| 57 DCHECK(observer == read_observer_); | |
| 58 read_observer_ = NULL; | |
| 59 } | |
| 60 | |
| 61 void Stream::AddData(scoped_refptr<net::IOBuffer> buffer, size_t size) { | |
| 62 can_add_data_ = writer_->Write(buffer, size); | |
| 63 } | |
| 64 | |
| 65 void Stream::Finalize() { | |
| 66 writer_->Close(DOWNLOAD_INTERRUPT_REASON_NONE); | |
| 67 writer_.reset(NULL); | |
| 68 | |
| 69 OnDataAvailable(); | |
| 70 } | |
| 71 | |
| 72 Stream::StreamState Stream::ReadRawData(net::IOBuffer* buf, | |
| 73 int buf_size, | |
| 74 int* bytes_read) { | |
| 75 if (!data_) { | |
| 76 data_length_ = 0; | |
| 77 bytes_read_ = 0; | |
| 78 ByteStreamReader::StreamState state = reader_->Read(&data_, &data_length_); | |
| 79 switch (state) { | |
| 80 case ByteStreamReader::STREAM_HAS_DATA: | |
| 81 break; | |
| 82 case ByteStreamReader::STREAM_COMPLETE: | |
| 83 registry_->UnregisterStream(url()); | |
| 84 return STREAM_COMPLETE; | |
| 85 case ByteStreamReader::STREAM_EMPTY: | |
| 86 return STREAM_EMPTY; | |
| 87 } | |
| 88 } | |
| 89 | |
| 90 size_t remaining_bytes = data_length_ - bytes_read_; | |
| 91 size_t to_read = | |
| 92 static_cast<size_t>(buf_size) < remaining_bytes ? | |
| 93 buf_size : remaining_bytes; | |
| 94 memcpy(buf->data(), data_->data() + bytes_read_, to_read); | |
| 95 bytes_read_ += to_read; | |
| 96 if (bytes_read_ >= data_length_) | |
| 97 data_ = NULL; | |
| 98 | |
| 99 *bytes_read = to_read; | |
| 100 return STREAM_HAS_DATA; | |
| 101 } | |
| 102 | |
| 103 void Stream::OnSpaceAvailable() { | |
| 104 can_add_data_ = true; | |
| 105 write_observer_->OnSpaceAvailable(this); | |
| 106 } | |
| 107 | |
| 108 void Stream::OnDataAvailable() { | |
| 109 read_observer_->OnDataAvailable(this); | |
| 110 } | |
| 111 | |
| 112 } // namespace content | |
| 113 | |
| OLD | NEW |
|---|
Chromium Code Reviews
