OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "mojo/edk/system/data_pipe_consumer_dispatcher.h" | 5 #include "mojo/edk/system/data_pipe_consumer_dispatcher.h" |
6 | 6 |
7 #include <utility> | 7 #include <utility> |
8 | 8 |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "mojo/edk/system/data_pipe.h" | 10 #include "mojo/edk/system/data_pipe.h" |
11 #include "mojo/edk/system/memory.h" | 11 #include "mojo/edk/system/memory.h" |
12 | 12 |
13 namespace mojo { | 13 namespace mojo { |
14 namespace system { | 14 namespace system { |
15 | 15 |
16 void DataPipeConsumerDispatcher::Init(RefPtr<DataPipe>&& data_pipe) { | 16 void DataPipeConsumerDispatcher::Init(RefPtr<DataPipe>&& data_pipe) { |
17 DCHECK(data_pipe); | 17 DCHECK(data_pipe); |
18 data_pipe_ = std::move(data_pipe); | 18 data_pipe_ = std::move(data_pipe); |
19 } | 19 } |
20 | 20 |
21 Dispatcher::Type DataPipeConsumerDispatcher::GetType() const { | 21 Dispatcher::Type DataPipeConsumerDispatcher::GetType() const { |
22 return Type::DATA_PIPE_CONSUMER; | 22 return Type::DATA_PIPE_CONSUMER; |
23 } | 23 } |
24 | 24 |
25 // static | 25 // static |
26 scoped_refptr<DataPipeConsumerDispatcher> | 26 RefPtr<DataPipeConsumerDispatcher> DataPipeConsumerDispatcher::Deserialize( |
27 DataPipeConsumerDispatcher::Deserialize(Channel* channel, | 27 Channel* channel, |
28 const void* source, | 28 const void* source, |
29 size_t size) { | 29 size_t size) { |
30 RefPtr<DataPipe> data_pipe; | 30 RefPtr<DataPipe> data_pipe; |
31 if (!DataPipe::ConsumerDeserialize(channel, source, size, &data_pipe)) | 31 if (!DataPipe::ConsumerDeserialize(channel, source, size, &data_pipe)) |
32 return nullptr; | 32 return nullptr; |
33 DCHECK(data_pipe); | 33 DCHECK(data_pipe); |
34 | 34 |
35 scoped_refptr<DataPipeConsumerDispatcher> dispatcher = Create(); | 35 auto dispatcher = DataPipeConsumerDispatcher::Create(); |
36 dispatcher->Init(std::move(data_pipe)); | 36 dispatcher->Init(std::move(data_pipe)); |
37 return dispatcher; | 37 return dispatcher; |
38 } | 38 } |
39 | 39 |
40 DataPipe* DataPipeConsumerDispatcher::GetDataPipeForTest() { | 40 DataPipe* DataPipeConsumerDispatcher::GetDataPipeForTest() { |
41 MutexLocker locker(&mutex()); | 41 MutexLocker locker(&mutex()); |
42 return data_pipe_.get(); | 42 return data_pipe_.get(); |
43 } | 43 } |
44 | 44 |
45 DataPipeConsumerDispatcher::DataPipeConsumerDispatcher() { | 45 DataPipeConsumerDispatcher::DataPipeConsumerDispatcher() { |
46 } | 46 } |
47 | 47 |
48 DataPipeConsumerDispatcher::~DataPipeConsumerDispatcher() { | 48 DataPipeConsumerDispatcher::~DataPipeConsumerDispatcher() { |
49 // |Close()|/|CloseImplNoLock()| should have taken care of the pipe. | 49 // |Close()|/|CloseImplNoLock()| should have taken care of the pipe. |
50 DCHECK(!data_pipe_); | 50 DCHECK(!data_pipe_); |
51 } | 51 } |
52 | 52 |
53 void DataPipeConsumerDispatcher::CancelAllAwakablesNoLock() { | 53 void DataPipeConsumerDispatcher::CancelAllAwakablesNoLock() { |
54 mutex().AssertHeld(); | 54 mutex().AssertHeld(); |
55 data_pipe_->ConsumerCancelAllAwakables(); | 55 data_pipe_->ConsumerCancelAllAwakables(); |
56 } | 56 } |
57 | 57 |
58 void DataPipeConsumerDispatcher::CloseImplNoLock() { | 58 void DataPipeConsumerDispatcher::CloseImplNoLock() { |
59 mutex().AssertHeld(); | 59 mutex().AssertHeld(); |
60 data_pipe_->ConsumerClose(); | 60 data_pipe_->ConsumerClose(); |
61 data_pipe_ = nullptr; | 61 data_pipe_ = nullptr; |
62 } | 62 } |
63 | 63 |
64 scoped_refptr<Dispatcher> | 64 RefPtr<Dispatcher> |
65 DataPipeConsumerDispatcher::CreateEquivalentDispatcherAndCloseImplNoLock() { | 65 DataPipeConsumerDispatcher::CreateEquivalentDispatcherAndCloseImplNoLock() { |
66 mutex().AssertHeld(); | 66 mutex().AssertHeld(); |
67 | 67 |
68 scoped_refptr<DataPipeConsumerDispatcher> rv = Create(); | 68 auto dispatcher = DataPipeConsumerDispatcher::Create(); |
69 rv->Init(std::move(data_pipe_)); | 69 dispatcher->Init(std::move(data_pipe_)); |
70 return scoped_refptr<Dispatcher>(rv.get()); | 70 return dispatcher; |
71 } | 71 } |
72 | 72 |
73 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock( | 73 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock( |
74 UserPointer<void> elements, | 74 UserPointer<void> elements, |
75 UserPointer<uint32_t> num_bytes, | 75 UserPointer<uint32_t> num_bytes, |
76 MojoReadDataFlags flags) { | 76 MojoReadDataFlags flags) { |
77 mutex().AssertHeld(); | 77 mutex().AssertHeld(); |
78 | 78 |
79 if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) { | 79 if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) { |
80 // These flags are mutally exclusive. | 80 // These flags are mutally exclusive. |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
143 Awakable* awakable, | 143 Awakable* awakable, |
144 HandleSignalsState* signals_state) { | 144 HandleSignalsState* signals_state) { |
145 mutex().AssertHeld(); | 145 mutex().AssertHeld(); |
146 data_pipe_->ConsumerRemoveAwakable(awakable, signals_state); | 146 data_pipe_->ConsumerRemoveAwakable(awakable, signals_state); |
147 } | 147 } |
148 | 148 |
149 void DataPipeConsumerDispatcher::StartSerializeImplNoLock( | 149 void DataPipeConsumerDispatcher::StartSerializeImplNoLock( |
150 Channel* channel, | 150 Channel* channel, |
151 size_t* max_size, | 151 size_t* max_size, |
152 size_t* max_platform_handles) { | 152 size_t* max_platform_handles) { |
153 DCHECK(HasOneRef()); // Only one ref => no need to take the lock. | 153 AssertHasOneRef(); // Only one ref => no need to take the lock. |
154 data_pipe_->ConsumerStartSerialize(channel, max_size, max_platform_handles); | 154 data_pipe_->ConsumerStartSerialize(channel, max_size, max_platform_handles); |
155 } | 155 } |
156 | 156 |
157 bool DataPipeConsumerDispatcher::EndSerializeAndCloseImplNoLock( | 157 bool DataPipeConsumerDispatcher::EndSerializeAndCloseImplNoLock( |
158 Channel* channel, | 158 Channel* channel, |
159 void* destination, | 159 void* destination, |
160 size_t* actual_size, | 160 size_t* actual_size, |
161 embedder::PlatformHandleVector* platform_handles) { | 161 embedder::PlatformHandleVector* platform_handles) { |
162 DCHECK(HasOneRef()); // Only one ref => no need to take the lock. | 162 AssertHasOneRef(); // Only one ref => no need to take the lock. |
163 | 163 |
164 bool rv = data_pipe_->ConsumerEndSerialize(channel, destination, actual_size, | 164 bool rv = data_pipe_->ConsumerEndSerialize(channel, destination, actual_size, |
165 platform_handles); | 165 platform_handles); |
166 data_pipe_ = nullptr; | 166 data_pipe_ = nullptr; |
167 return rv; | 167 return rv; |
168 } | 168 } |
169 | 169 |
170 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { | 170 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { |
171 mutex().AssertHeld(); | 171 mutex().AssertHeld(); |
172 return data_pipe_->ConsumerIsBusy(); | 172 return data_pipe_->ConsumerIsBusy(); |
173 } | 173 } |
174 | 174 |
175 } // namespace system | 175 } // namespace system |
176 } // namespace mojo | 176 } // namespace mojo |
OLD | NEW |