EDK: Add Core::DuplicateHandleWithReducedRights().

mojo/edk/system/shared_buffer_dispatcher_unittest.cc

 // Copyright 2014 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/shared_buffer_dispatcher.h"
 
 #include <limits>
 #include <memory>
 
 #include "mojo/edk/embedder/simple_platform_support.h"
@@ skipping 185 matching lines @@
   EXPECT_FALSE(d->SupportsEntrypointClass(EntrypointClass::DATA_PIPE_PRODUCER));
   EXPECT_FALSE(d->SupportsEntrypointClass(EntrypointClass::DATA_PIPE_CONSUMER));
   EXPECT_TRUE(d->SupportsEntrypointClass(EntrypointClass::BUFFER));
 
   // TODO(vtl): Check that it actually returns |MOJO_RESULT_INVALID_ARGUMENT|
   // for methods in unsupported entrypoint classes.
 
   EXPECT_EQ(MOJO_RESULT_OK, d->Close());
 }
 
+TEST_F(SharedBufferDispatcherTest, DuplicateDispatcher) {
+  const uint64_t kSize = 100u;
+
+  MojoResult result = MOJO_RESULT_INTERNAL;
+  auto dispatcher1 = SharedBufferDispatcher::Create(
+      platform_support(), SharedBufferDispatcher::kDefaultCreateOptions, kSize,
+      &result);
+  EXPECT_EQ(MOJO_RESULT_OK, result);
+
+  // Map and write something.
+  std::unique_ptr<PlatformSharedBufferMapping> mapping;
+  EXPECT_EQ(
+      MOJO_RESULT_OK,
+      dispatcher1->MapBuffer(0u, kSize, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
+  static_cast<char*>(mapping->GetBase())[0] = 'x';
+  mapping.reset();
+
+  // Duplicate |dispatcher1| and then close it.
+  RefPtr<Dispatcher> dispatcher2;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->DuplicateDispatcher(&dispatcher2));
+  ASSERT_TRUE(dispatcher2);
+  EXPECT_EQ(Dispatcher::Type::SHARED_BUFFER, dispatcher2->GetType());
+
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close());
+
+  // Make sure that |dispatcher2| still reports the right information.
+  MojoBufferInformation info = {};
+  EXPECT_EQ(MOJO_RESULT_OK,
+            dispatcher2->GetBufferInformation(
+                MakeUserPointer(&info), static_cast<uint32_t>(sizeof(info))));
+  EXPECT_EQ(sizeof(MojoBufferInformation), info.struct_size);
+  EXPECT_EQ(MOJO_BUFFER_INFORMATION_FLAG_NONE, info.flags);
+  EXPECT_EQ(kSize, info.num_bytes);
+
+  // Map |dispatcher2| and read something.
+  EXPECT_EQ(
+      MOJO_RESULT_OK,
+      dispatcher2->MapBuffer(0u, kSize, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
+  EXPECT_EQ('x', static_cast<char*>(mapping->GetBase())[0]);
+
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->Close());
+}
+
 TEST_F(SharedBufferDispatcherTest, DuplicateBufferHandle) {
   const uint64_t kSize = 100u;
 
   MojoResult result = MOJO_RESULT_INTERNAL;
   auto dispatcher1 = SharedBufferDispatcher::Create(
       platform_support(), SharedBufferDispatcher::kDefaultCreateOptions, kSize,
       &result);
   EXPECT_EQ(MOJO_RESULT_OK, result);
 
   // Map and write something.
@@ skipping 123 matching lines @@
   EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
             dispatcher->MapBuffer(0, 0, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
   EXPECT_FALSE(mapping);
 
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close());
 }
 
 }  // namespace
 }  // namespace system
 }  // namespace mojo