Implement MojoGetBufferInformation(), part 1.

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 101 matching lines @@
         ~0u                    // |flags|.
     };
     MojoCreateSharedBufferOptions unused;
     EXPECT_EQ(MOJO_RESULT_UNIMPLEMENTED,
               SharedBufferDispatcher::ValidateCreateOptions(
                   MakeUserPointer(&options), &unused));
   }
 }
 
 TEST_F(SharedBufferDispatcherTest, CreateAndMapBuffer) {
+  const uint64_t kSize = 100u;
+
   MojoResult result = MOJO_RESULT_INTERNAL;
   auto dispatcher = SharedBufferDispatcher::Create(
-      platform_support(), SharedBufferDispatcher::kDefaultCreateOptions, 100,
+      platform_support(), SharedBufferDispatcher::kDefaultCreateOptions, kSize,
       &result);
   EXPECT_EQ(MOJO_RESULT_OK, result);
 
   ASSERT_TRUE(dispatcher);
   EXPECT_EQ(Dispatcher::Type::SHARED_BUFFER, dispatcher->GetType());
 
+  // Get information about the buffer (in particular the size) and check it.
+  MojoBufferInformation info = {};
+  EXPECT_EQ(MOJO_RESULT_OK,
+            dispatcher->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);
+
+  // Also check that some invalid calls to |GetBufferInformation()| fail in the
+  // expected way.
+  EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
+            dispatcher->GetBufferInformation(MakeUserPointer(&info), 0u));
+  EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
+            dispatcher->GetBufferInformation(MakeUserPointer(&info), 15u));
+
+  // It's also valid to call it with a larger-than-required |info_num_bytes|.
+  // (Note: The pointer must be aligned, so we use an array of two
+  // |MojoBufferInformation|s.)
+  MojoBufferInformation infos[2] = {};
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->GetBufferInformation(
+                                MakeUserPointer(&infos[0]),
+                                static_cast<uint32_t>(sizeof(infos[0]) + 1)));
+  EXPECT_EQ(kSize, infos[0].num_bytes);
+
   // Make a couple of mappings.
   std::unique_ptr<PlatformSharedBufferMapping> mapping1;
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer(
-                                0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping1));
+  EXPECT_EQ(
+      MOJO_RESULT_OK,
+      dispatcher->MapBuffer(0u, kSize, MOJO_MAP_BUFFER_FLAG_NONE, &mapping1));
   ASSERT_TRUE(mapping1);
   ASSERT_TRUE(mapping1->GetBase());
-  EXPECT_EQ(100u, mapping1->GetLength());
+  EXPECT_EQ(kSize, mapping1->GetLength());
   // Write something.
   static_cast<char*>(mapping1->GetBase())[50] = 'x';
 
   std::unique_ptr<PlatformSharedBufferMapping> mapping2;
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer(
-                                50, 50, MOJO_MAP_BUFFER_FLAG_NONE, &mapping2));
+  EXPECT_EQ(MOJO_RESULT_OK,
+            dispatcher->MapBuffer(kSize / 2, kSize / 2,
+                                  MOJO_MAP_BUFFER_FLAG_NONE, &mapping2));
   ASSERT_TRUE(mapping2);
   ASSERT_TRUE(mapping2->GetBase());
-  EXPECT_EQ(50u, mapping2->GetLength());
+  EXPECT_EQ(kSize / 2, mapping2->GetLength());
   EXPECT_EQ('x', static_cast<char*>(mapping2->GetBase())[0]);
 
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close());
 
   // Check that we can still read/write to mappings after the dispatcher has
   // gone away.
   static_cast<char*>(mapping2->GetBase())[1] = 'y';
-  EXPECT_EQ('y', static_cast<char*>(mapping1->GetBase())[51]);
+  EXPECT_EQ('y', static_cast<char*>(mapping1->GetBase())[kSize / 2 + 1]);
 }
 
 TEST_F(SharedBufferDispatcherTest, DuplicateBufferHandle) {
   MojoResult result = MOJO_RESULT_INTERNAL;
   auto dispatcher1 = SharedBufferDispatcher::Create(
       platform_support(), SharedBufferDispatcher::kDefaultCreateOptions, 100,
       &result);
   EXPECT_EQ(MOJO_RESULT_OK, result);
 
   // Map and write something.
@@ skipping 112 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