Mojo: More data pipe implementation.

mojo/system/data_pipe_producer_dispatcher.cc

+// 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