| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "mojo/system/data_pipe_consumer_dispatcher.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "mojo/system/data_pipe.h" | |
| 9 #include "mojo/system/memory.h" | |
| 10 | |
| 11 namespace mojo { | |
| 12 namespace system { | |
| 13 | |
| 14 DataPipeConsumerDispatcher::DataPipeConsumerDispatcher() { | |
| 15 } | |
| 16 | |
| 17 void DataPipeConsumerDispatcher::Init(scoped_refptr<DataPipe> data_pipe) { | |
| 18 DCHECK(data_pipe.get()); | |
| 19 data_pipe_ = data_pipe; | |
| 20 } | |
| 21 | |
| 22 Dispatcher::Type DataPipeConsumerDispatcher::GetType() const { | |
| 23 return kTypeDataPipeConsumer; | |
| 24 } | |
| 25 | |
| 26 DataPipeConsumerDispatcher::~DataPipeConsumerDispatcher() { | |
| 27 // |Close()|/|CloseImplNoLock()| should have taken care of the pipe. | |
| 28 DCHECK(!data_pipe_.get()); | |
| 29 } | |
| 30 | |
| 31 void DataPipeConsumerDispatcher::CancelAllWaitersNoLock() { | |
| 32 lock().AssertAcquired(); | |
| 33 data_pipe_->ConsumerCancelAllWaiters(); | |
| 34 } | |
| 35 | |
| 36 void DataPipeConsumerDispatcher::CloseImplNoLock() { | |
| 37 lock().AssertAcquired(); | |
| 38 data_pipe_->ConsumerClose(); | |
| 39 data_pipe_ = nullptr; | |
| 40 } | |
| 41 | |
| 42 scoped_refptr<Dispatcher> | |
| 43 DataPipeConsumerDispatcher::CreateEquivalentDispatcherAndCloseImplNoLock() { | |
| 44 lock().AssertAcquired(); | |
| 45 | |
| 46 scoped_refptr<DataPipeConsumerDispatcher> rv = | |
| 47 new DataPipeConsumerDispatcher(); | |
| 48 rv->Init(data_pipe_); | |
| 49 data_pipe_ = nullptr; | |
| 50 return scoped_refptr<Dispatcher>(rv.get()); | |
| 51 } | |
| 52 | |
| 53 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock( | |
| 54 UserPointer<void> elements, | |
| 55 UserPointer<uint32_t> num_bytes, | |
| 56 MojoReadDataFlags flags) { | |
| 57 lock().AssertAcquired(); | |
| 58 | |
| 59 if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) { | |
| 60 // These flags are mutally exclusive. | |
| 61 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) | |
| 62 return MOJO_RESULT_INVALID_ARGUMENT; | |
| 63 DVLOG_IF(2, !elements.IsNull()) | |
| 64 << "Discard mode: ignoring non-null |elements|"; | |
| 65 return data_pipe_->ConsumerDiscardData( | |
| 66 num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | |
| 67 } | |
| 68 | |
| 69 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) { | |
| 70 DCHECK(!(flags & MOJO_READ_DATA_FLAG_DISCARD)); // Handled above. | |
| 71 DVLOG_IF(2, !elements.IsNull()) | |
| 72 << "Query mode: ignoring non-null |elements|"; | |
| 73 return data_pipe_->ConsumerQueryData(num_bytes); | |
| 74 } | |
| 75 | |
| 76 return data_pipe_->ConsumerReadData( | |
| 77 elements, num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | |
| 78 } | |
| 79 | |
| 80 MojoResult DataPipeConsumerDispatcher::BeginReadDataImplNoLock( | |
| 81 UserPointer<const void*> buffer, | |
| 82 UserPointer<uint32_t> buffer_num_bytes, | |
| 83 MojoReadDataFlags flags) { | |
| 84 lock().AssertAcquired(); | |
| 85 | |
| 86 // These flags may not be used in two-phase mode. | |
| 87 if ((flags & MOJO_READ_DATA_FLAG_DISCARD) || | |
| 88 (flags & MOJO_READ_DATA_FLAG_QUERY)) | |
| 89 return MOJO_RESULT_INVALID_ARGUMENT; | |
| 90 | |
| 91 return data_pipe_->ConsumerBeginReadData( | |
| 92 buffer, buffer_num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | |
| 93 } | |
| 94 | |
| 95 MojoResult DataPipeConsumerDispatcher::EndReadDataImplNoLock( | |
| 96 uint32_t num_bytes_read) { | |
| 97 lock().AssertAcquired(); | |
| 98 | |
| 99 return data_pipe_->ConsumerEndReadData(num_bytes_read); | |
| 100 } | |
| 101 | |
| 102 HandleSignalsState DataPipeConsumerDispatcher::GetHandleSignalsStateImplNoLock() | |
| 103 const { | |
| 104 lock().AssertAcquired(); | |
| 105 return data_pipe_->ConsumerGetHandleSignalsState(); | |
| 106 } | |
| 107 | |
| 108 MojoResult DataPipeConsumerDispatcher::AddWaiterImplNoLock( | |
| 109 Waiter* waiter, | |
| 110 MojoHandleSignals signals, | |
| 111 uint32_t context, | |
| 112 HandleSignalsState* signals_state) { | |
| 113 lock().AssertAcquired(); | |
| 114 return data_pipe_->ConsumerAddWaiter(waiter, signals, context, signals_state); | |
| 115 } | |
| 116 | |
| 117 void DataPipeConsumerDispatcher::RemoveWaiterImplNoLock( | |
| 118 Waiter* waiter, | |
| 119 HandleSignalsState* signals_state) { | |
| 120 lock().AssertAcquired(); | |
| 121 data_pipe_->ConsumerRemoveWaiter(waiter, signals_state); | |
| 122 } | |
| 123 | |
| 124 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { | |
| 125 lock().AssertAcquired(); | |
| 126 return data_pipe_->ConsumerIsBusy(); | |
| 127 } | |
| 128 | |
| 129 } // namespace system | |
| 130 } // namespace mojo | |
| OLD | NEW |