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 "content/browser/byte_stream.h" | 5 #include "content/browser/byte_stream.h" |
6 | 6 |
7 #include <deque> | 7 #include <deque> |
8 #include <set> | 8 #include <set> |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
(...skipping 30 matching lines...) Expand all Loading... |
41 }; | 41 }; |
42 | 42 |
43 // For both ByteStreamWriterImpl and ByteStreamReaderImpl, Construction and | 43 // For both ByteStreamWriterImpl and ByteStreamReaderImpl, Construction and |
44 // SetPeer may happen anywhere; all other operations on each class must | 44 // SetPeer may happen anywhere; all other operations on each class must |
45 // happen in the context of their SequencedTaskRunner. | 45 // happen in the context of their SequencedTaskRunner. |
46 class ByteStreamWriterImpl : public ByteStreamWriter { | 46 class ByteStreamWriterImpl : public ByteStreamWriter { |
47 public: | 47 public: |
48 ByteStreamWriterImpl(scoped_refptr<base::SequencedTaskRunner> task_runner, | 48 ByteStreamWriterImpl(scoped_refptr<base::SequencedTaskRunner> task_runner, |
49 scoped_refptr<LifetimeFlag> lifetime_flag, | 49 scoped_refptr<LifetimeFlag> lifetime_flag, |
50 size_t buffer_size); | 50 size_t buffer_size); |
51 virtual ~ByteStreamWriterImpl(); | 51 ~ByteStreamWriterImpl() override; |
52 | 52 |
53 // Must be called before any operations are performed. | 53 // Must be called before any operations are performed. |
54 void SetPeer(ByteStreamReaderImpl* peer, | 54 void SetPeer(ByteStreamReaderImpl* peer, |
55 scoped_refptr<base::SequencedTaskRunner> peer_task_runner, | 55 scoped_refptr<base::SequencedTaskRunner> peer_task_runner, |
56 scoped_refptr<LifetimeFlag> peer_lifetime_flag); | 56 scoped_refptr<LifetimeFlag> peer_lifetime_flag); |
57 | 57 |
58 // Overridden from ByteStreamWriter. | 58 // Overridden from ByteStreamWriter. |
59 virtual bool Write(scoped_refptr<net::IOBuffer> buffer, | 59 bool Write(scoped_refptr<net::IOBuffer> buffer, size_t byte_count) override; |
60 size_t byte_count) override; | 60 void Flush() override; |
61 virtual void Flush() override; | 61 void Close(int status) override; |
62 virtual void Close(int status) override; | 62 void RegisterCallback(const base::Closure& source_callback) override; |
63 virtual void RegisterCallback(const base::Closure& source_callback) override; | 63 size_t GetTotalBufferedBytes() const override; |
64 virtual size_t GetTotalBufferedBytes() const override; | |
65 | 64 |
66 // PostTask target from |ByteStreamReaderImpl::MaybeUpdateInput|. | 65 // PostTask target from |ByteStreamReaderImpl::MaybeUpdateInput|. |
67 static void UpdateWindow(scoped_refptr<LifetimeFlag> lifetime_flag, | 66 static void UpdateWindow(scoped_refptr<LifetimeFlag> lifetime_flag, |
68 ByteStreamWriterImpl* target, | 67 ByteStreamWriterImpl* target, |
69 size_t bytes_consumed); | 68 size_t bytes_consumed); |
70 | 69 |
71 private: | 70 private: |
72 // Called from UpdateWindow when object existence has been validated. | 71 // Called from UpdateWindow when object existence has been validated. |
73 void UpdateWindowInternal(size_t bytes_consumed); | 72 void UpdateWindowInternal(size_t bytes_consumed); |
74 | 73 |
(...skipping 26 matching lines...) Expand all Loading... |
101 // Only valid to access on peer_task_runner_ if | 100 // Only valid to access on peer_task_runner_ if |
102 // |*peer_lifetime_flag_ == true| | 101 // |*peer_lifetime_flag_ == true| |
103 ByteStreamReaderImpl* peer_; | 102 ByteStreamReaderImpl* peer_; |
104 }; | 103 }; |
105 | 104 |
106 class ByteStreamReaderImpl : public ByteStreamReader { | 105 class ByteStreamReaderImpl : public ByteStreamReader { |
107 public: | 106 public: |
108 ByteStreamReaderImpl(scoped_refptr<base::SequencedTaskRunner> task_runner, | 107 ByteStreamReaderImpl(scoped_refptr<base::SequencedTaskRunner> task_runner, |
109 scoped_refptr<LifetimeFlag> lifetime_flag, | 108 scoped_refptr<LifetimeFlag> lifetime_flag, |
110 size_t buffer_size); | 109 size_t buffer_size); |
111 virtual ~ByteStreamReaderImpl(); | 110 ~ByteStreamReaderImpl() override; |
112 | 111 |
113 // Must be called before any operations are performed. | 112 // Must be called before any operations are performed. |
114 void SetPeer(ByteStreamWriterImpl* peer, | 113 void SetPeer(ByteStreamWriterImpl* peer, |
115 scoped_refptr<base::SequencedTaskRunner> peer_task_runner, | 114 scoped_refptr<base::SequencedTaskRunner> peer_task_runner, |
116 scoped_refptr<LifetimeFlag> peer_lifetime_flag); | 115 scoped_refptr<LifetimeFlag> peer_lifetime_flag); |
117 | 116 |
118 // Overridden from ByteStreamReader. | 117 // Overridden from ByteStreamReader. |
119 virtual StreamState Read(scoped_refptr<net::IOBuffer>* data, | 118 StreamState Read(scoped_refptr<net::IOBuffer>* data, size_t* length) override; |
120 size_t* length) override; | 119 int GetStatus() const override; |
121 virtual int GetStatus() const override; | 120 void RegisterCallback(const base::Closure& sink_callback) override; |
122 virtual void RegisterCallback(const base::Closure& sink_callback) override; | |
123 | 121 |
124 // PostTask target from |ByteStreamWriterImpl::Write| and | 122 // PostTask target from |ByteStreamWriterImpl::Write| and |
125 // |ByteStreamWriterImpl::Close|. | 123 // |ByteStreamWriterImpl::Close|. |
126 // Receive data from our peer. | 124 // Receive data from our peer. |
127 // static because it may be called after the object it is targeting | 125 // static because it may be called after the object it is targeting |
128 // has been destroyed. It may not access |*target| | 126 // has been destroyed. It may not access |*target| |
129 // if |*object_lifetime_flag| is false. | 127 // if |*object_lifetime_flag| is false. |
130 static void TransferData( | 128 static void TransferData( |
131 scoped_refptr<LifetimeFlag> object_lifetime_flag, | 129 scoped_refptr<LifetimeFlag> object_lifetime_flag, |
132 ByteStreamReaderImpl* target, | 130 ByteStreamReaderImpl* target, |
(...skipping 326 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
459 ByteStreamReaderImpl* out = new ByteStreamReaderImpl( | 457 ByteStreamReaderImpl* out = new ByteStreamReaderImpl( |
460 output_task_runner, output_flag, buffer_size); | 458 output_task_runner, output_flag, buffer_size); |
461 | 459 |
462 in->SetPeer(out, output_task_runner, output_flag); | 460 in->SetPeer(out, output_task_runner, output_flag); |
463 out->SetPeer(in, input_task_runner, input_flag); | 461 out->SetPeer(in, input_task_runner, input_flag); |
464 input->reset(in); | 462 input->reset(in); |
465 output->reset(out); | 463 output->reset(out); |
466 } | 464 } |
467 | 465 |
468 } // namespace content | 466 } // namespace content |
OLD | NEW |