EDK: Add implementation of data pipe consumer read threshold stuff.

mojo/edk/system/data_pipe_consumer_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/edk/system/data_pipe_consumer_dispatcher.h"
 
 #include <utility>
 
 #include "base/logging.h"
 #include "mojo/edk/system/data_pipe.h"
 #include "mojo/edk/system/memory.h"
+#include "mojo/edk/system/options_validation.h"
 
 using mojo::platform::ScopedPlatformHandle;
 using mojo::util::MutexLocker;
 using mojo::util::RefPtr;
 
 namespace mojo {
 namespace system {
 
 void DataPipeConsumerDispatcher::Init(RefPtr<DataPipe>&& data_pipe) {
   DCHECK(data_pipe);
@@ skipping 45 matching lines @@
 
 RefPtr<Dispatcher>
 DataPipeConsumerDispatcher::CreateEquivalentDispatcherAndCloseImplNoLock() {
   mutex().AssertHeld();
 
   auto dispatcher = DataPipeConsumerDispatcher::Create();
   dispatcher->Init(std::move(data_pipe_));
   return dispatcher;
 }
 
+MojoResult DataPipeConsumerDispatcher::SetDataPipeConsumerOptionsImplNoLock(
+    UserPointer<const MojoDataPipeConsumerOptions> options) {
+  mutex().AssertHeld();
+
+  UserOptionsReader<MojoDataPipeConsumerOptions> reader(options);
+  if (!reader.is_valid())
+    return MOJO_RESULT_INVALID_ARGUMENT;
+
+  if (!OPTIONS_STRUCT_HAS_MEMBER(MojoDataPipeConsumerOptions,
+                                 read_threshold_num_bytes, reader))
+    return MOJO_RESULT_OK;
+
+  return data_pipe_->ConsumerSetOptions(
+      reader.options().read_threshold_num_bytes);
+}
+
+MojoResult DataPipeConsumerDispatcher::GetDataPipeConsumerOptionsImplNoLock(
+    UserPointer<MojoDataPipeConsumerOptions> options,
+    uint32_t options_num_bytes) {
+  mutex().AssertHeld();
+
+  // Note: If/when |MojoDataPipeConsumerOptions| is extended beyond its initial
+  // definition, more work will be necessary. (See the definition of
+  // |MojoGetDataPipeConsumerOptions()| in mojo/public/c/system/data_pipe.h.)
+  static_assert(sizeof(MojoDataPipeConsumerOptions) == 8u,
+                "MojoDataPipeConsumerOptions has been extended!");
+
+  if (options_num_bytes < sizeof(MojoDataPipeConsumerOptions))
+    return MOJO_RESULT_INVALID_ARGUMENT;
+
+  uint32_t read_threshold_num_bytes = 0;
+  data_pipe_->ConsumerGetOptions(&read_threshold_num_bytes);
+  MojoDataPipeConsumerOptions model_options = {
+      sizeof(MojoDataPipeConsumerOptions),  // |struct_size|.
+      read_threshold_num_bytes,             // |read_threshold_num_bytes|.
+  };
+  options.Put(model_options);
+  return MOJO_RESULT_OK;
+}
+
 MojoResult DataPipeConsumerDispatcher::ReadDataImplNoLock(
     UserPointer<void> elements,
     UserPointer<uint32_t> num_bytes,
     MojoReadDataFlags flags) {
   mutex().AssertHeld();
 
   if ((flags & MOJO_READ_DATA_FLAG_DISCARD)) {
     // These flags are mutally exclusive.
     if ((flags & MOJO_READ_DATA_FLAG_QUERY) ||
         (flags & MOJO_READ_DATA_FLAG_PEEK))
@@ skipping 84 matching lines @@
   return rv;
 }
 
 bool DataPipeConsumerDispatcher::IsBusyNoLock() const {
   mutex().AssertHeld();
   return data_pipe_->ConsumerIsBusy();
 }
 
 }  // namespace system
 }  // namespace mojo