| // Copyright 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "mojo/system/data_pipe_producer_dispatcher.h" |
| |
| #include "base/logging.h" |
| #include "mojo/system/data_pipe.h" |
| #include "mojo/system/memory.h" |
| |
| namespace mojo { |
| namespace system { |
| |
| DataPipeProducerDispatcher::DataPipeProducerDispatcher() { |
| } |
| |
| void DataPipeProducerDispatcher::Init(scoped_refptr<DataPipe> data_pipe) { |
| DCHECK(data_pipe.get()); |
| data_pipe_ = data_pipe; |
| } |
| |
| DataPipeProducerDispatcher::~DataPipeProducerDispatcher() { |
| // |Close()|/|CloseImplNoLock()| should have taken care of the pipe. |
| DCHECK(!data_pipe_.get()); |
| } |
| |
| void DataPipeProducerDispatcher::CancelAllWaitersNoLock() { |
| lock().AssertAcquired(); |
| data_pipe_->ProducerCancelAllWaiters(); |
| } |
| |
| MojoResult DataPipeProducerDispatcher::CloseImplNoLock() { |
| lock().AssertAcquired(); |
| data_pipe_->ProducerClose(); |
| data_pipe_ = NULL; |
| return MOJO_RESULT_OK; |
| } |
| |
| MojoResult DataPipeProducerDispatcher::WriteDataImplNoLock( |
| const void* elements, |
| uint32_t* num_elements, |
| MojoWriteDataFlags flags) { |
| lock().AssertAcquired(); |
| |
| if (!VerifyUserPointer<uint32_t>(num_elements, 1)) |
| return MOJO_RESULT_INVALID_ARGUMENT; |
| if (!VerifyUserPointerForSize(elements, data_pipe_->element_size(), |
| *num_elements)) |
| return MOJO_RESULT_INVALID_ARGUMENT; |
| |
| return data_pipe_->ProducerWriteData(elements, num_elements, flags); |
| } |
| |
| MojoResult DataPipeProducerDispatcher::BeginWriteDataImplNoLock( |
| void** buffer, |
| uint32_t* buffer_num_elements, |
| MojoWriteDataFlags flags) { |
| lock().AssertAcquired(); |
| |
| if (!VerifyUserPointer<void*>(buffer, 1)) |
| return MOJO_RESULT_INVALID_ARGUMENT; |
| if (!VerifyUserPointer<uint32_t>(buffer_num_elements, 1)) |
| return MOJO_RESULT_INVALID_ARGUMENT; |
| |
| return data_pipe_->ProducerBeginWriteData(buffer, buffer_num_elements, flags); |
| } |
| |
| MojoResult DataPipeProducerDispatcher::EndWriteDataImplNoLock( |
| uint32_t num_elements_written) { |
| lock().AssertAcquired(); |
| |
| return data_pipe_->ProducerEndWriteData(num_elements_written); |
| } |
| |
| MojoResult DataPipeProducerDispatcher::AddWaiterImplNoLock( |
| Waiter* waiter, |
| MojoWaitFlags flags, |
| MojoResult wake_result) { |
| lock().AssertAcquired(); |
| return data_pipe_->ProducerAddWaiter(waiter, flags, wake_result); |
| } |
| |
| void DataPipeProducerDispatcher::RemoveWaiterImplNoLock(Waiter* waiter) { |
| lock().AssertAcquired(); |
| data_pipe_->ProducerRemoveWaiter(waiter); |
| } |
| |
| scoped_refptr<Dispatcher> |
| DataPipeProducerDispatcher::CreateEquivalentDispatcherAndCloseImplNoLock() { |
| lock().AssertAcquired(); |
| |
| scoped_refptr<DataPipeProducerDispatcher> rv = |
| new DataPipeProducerDispatcher(); |
| rv->Init(data_pipe_); |
| data_pipe_ = NULL; |
| return scoped_refptr<Dispatcher>(rv.get()); |
| } |
| |
| } // namespace system |
| } // namespace mojo |