| 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/edk/system/data_pipe_consumer_dispatcher.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "mojo/edk/system/data_pipe.h" | |
| 9 #include "mojo/edk/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); | |
| 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_); | |
| 29 } | |
| 30 | |
| 31 void DataPipeConsumerDispatcher::CancelAllAwakablesNoLock() { | |
| 32 lock().AssertAcquired(); | |
| 33 data_pipe_->ConsumerCancelAllAwakables(); | |
| 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 (flags & MOJO_READ_DATA_FLAG_PEEK)) | |
| 63 return MOJO_RESULT_INVALID_ARGUMENT; | |
| 64 DVLOG_IF(2, !elements.IsNull()) | |
| 65 << "Discard mode: ignoring non-null |elements|"; | |
| 66 return data_pipe_->ConsumerDiscardData( | |
| 67 num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | |
| 68 } | |
| 69 | |
| 70 if ((flags & MOJO_READ_DATA_FLAG_QUERY)) { | |
| 71 if ((flags & MOJO_READ_DATA_FLAG_PEEK)) | |
| 72 return MOJO_RESULT_INVALID_ARGUMENT; | |
| 73 DCHECK(!(flags & MOJO_READ_DATA_FLAG_DISCARD)); // Handled above. | |
| 74 DVLOG_IF(2, !elements.IsNull()) | |
| 75 << "Query mode: ignoring non-null |elements|"; | |
| 76 return data_pipe_->ConsumerQueryData(num_bytes); | |
| 77 } | |
| 78 | |
| 79 return data_pipe_->ConsumerReadData( | |
| 80 elements, num_bytes, !!(flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE), | |
| 81 !!(flags & MOJO_READ_DATA_FLAG_PEEK)); | |
| 82 } | |
| 83 | |
| 84 MojoResult DataPipeConsumerDispatcher::BeginReadDataImplNoLock( | |
| 85 UserPointer<const void*> buffer, | |
| 86 UserPointer<uint32_t> buffer_num_bytes, | |
| 87 MojoReadDataFlags flags) { | |
| 88 lock().AssertAcquired(); | |
| 89 | |
| 90 // These flags may not be used in two-phase mode. | |
| 91 if ((flags & MOJO_READ_DATA_FLAG_DISCARD) || | |
| 92 (flags & MOJO_READ_DATA_FLAG_QUERY) || (flags & MOJO_READ_DATA_FLAG_PEEK)) | |
| 93 return MOJO_RESULT_INVALID_ARGUMENT; | |
| 94 | |
| 95 return data_pipe_->ConsumerBeginReadData( | |
| 96 buffer, buffer_num_bytes, (flags & MOJO_READ_DATA_FLAG_ALL_OR_NONE)); | |
| 97 } | |
| 98 | |
| 99 MojoResult DataPipeConsumerDispatcher::EndReadDataImplNoLock( | |
| 100 uint32_t num_bytes_read) { | |
| 101 lock().AssertAcquired(); | |
| 102 | |
| 103 return data_pipe_->ConsumerEndReadData(num_bytes_read); | |
| 104 } | |
| 105 | |
| 106 HandleSignalsState DataPipeConsumerDispatcher::GetHandleSignalsStateImplNoLock() | |
| 107 const { | |
| 108 lock().AssertAcquired(); | |
| 109 return data_pipe_->ConsumerGetHandleSignalsState(); | |
| 110 } | |
| 111 | |
| 112 MojoResult DataPipeConsumerDispatcher::AddAwakableImplNoLock( | |
| 113 Awakable* awakable, | |
| 114 MojoHandleSignals signals, | |
| 115 uint32_t context, | |
| 116 HandleSignalsState* signals_state) { | |
| 117 lock().AssertAcquired(); | |
| 118 return data_pipe_->ConsumerAddAwakable(awakable, signals, context, | |
| 119 signals_state); | |
| 120 } | |
| 121 | |
| 122 void DataPipeConsumerDispatcher::RemoveAwakableImplNoLock( | |
| 123 Awakable* awakable, | |
| 124 HandleSignalsState* signals_state) { | |
| 125 lock().AssertAcquired(); | |
| 126 data_pipe_->ConsumerRemoveAwakable(awakable, signals_state); | |
| 127 } | |
| 128 | |
| 129 bool DataPipeConsumerDispatcher::IsBusyNoLock() const { | |
| 130 lock().AssertAcquired(); | |
| 131 return data_pipe_->ConsumerIsBusy(); | |
| 132 } | |
| 133 | |
| 134 } // namespace system | |
| 135 } // namespace mojo | |
| OLD | NEW |