Rewrite clipboard write IPC handling to be easier to understand.

ui/base/clipboard/clipboard_unittest.cc

 // Copyright (c) 2012 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 "build/build_config.h"
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 313 matching lines @@
   clipboard().ReadAsciiText(CLIPBOARD_TYPE_COPY_PASTE, &ascii_text);
   EXPECT_EQ(UTF16ToUTF8(url), ascii_text);
 
 #if defined(OS_POSIX) && !defined(OS_MACOSX) && !defined(OS_ANDROID)
   ascii_text.clear();
   clipboard().ReadAsciiText(CLIPBOARD_TYPE_SELECTION, &ascii_text);
   EXPECT_EQ(UTF16ToUTF8(url), ascii_text);
 #endif
 }
 
-// TODO(dcheng): The tests for copying to the clipboard also test IPC
-// interaction... consider moving them to a different layer so we can
-// consolidate the validation logic.
-// Note that |bitmap_data| is not premultiplied!
 static void TestBitmapWrite(Clipboard* clipboard,
-                            const uint32* bitmap_data,
-                            size_t bitmap_data_size,
-                            const gfx::Size& size) {
-  // Create shared memory region.
-  base::SharedMemory shared_buf;
-  ASSERT_TRUE(shared_buf.CreateAndMapAnonymous(bitmap_data_size));
-  memcpy(shared_buf.memory(), bitmap_data, bitmap_data_size);
-  // CBF_SMBITMAP expects premultiplied bitmap data so do that now.
-  uint32* pixel_buffer = static_cast<uint32*>(shared_buf.memory());
-  for (int j = 0; j < size.height(); ++j) {
-    for (int i = 0; i < size.width(); ++i) {
-      uint32& pixel = pixel_buffer[i + j * size.width()];
-      pixel = SkPreMultiplyColor(pixel);
-    }
+                            const gfx::Size& size,
+                            const uint32* bitmap_data) {
+  {
+    ScopedClipboardWriter scw(CLIPBOARD_TYPE_COPY_PASTE);
+    SkBitmap bitmap;
+    ASSERT_TRUE(bitmap.setInfo(
+        SkImageInfo::MakeN32Premul(size.width(), size.height())));
+    bitmap.setPixels(
+        const_cast<void*>(reinterpret_cast<const void*>(bitmap_data)));
+    scw.WriteImage(bitmap);
   }
-  base::SharedMemoryHandle handle_to_share;
-  base::ProcessHandle current_process = base::kNullProcessHandle;
-#if defined(OS_WIN)
-  current_process = GetCurrentProcess();
-#endif
-  shared_buf.ShareToProcess(current_process, &handle_to_share);
-  ASSERT_TRUE(shared_buf.Unmap());
-
-  // Setup data for clipboard().
-  Clipboard::ObjectMapParam placeholder_param;
-  Clipboard::ObjectMapParam size_param;
-  const char* size_data = reinterpret_cast<const char*>(&size);
-  for (size_t i = 0; i < sizeof(size); ++i)
-    size_param.push_back(size_data[i]);
-
-  Clipboard::ObjectMapParams params;
-  params.push_back(placeholder_param);
-  params.push_back(size_param);
-
-  Clipboard::ObjectMap objects;
-  objects[Clipboard::CBF_SMBITMAP] = params;
-  ASSERT_TRUE(Clipboard::ReplaceSharedMemHandle(
-      &objects, handle_to_share, current_process));
-
-  ClipboardTest::WriteObjectsToClipboard(clipboard, objects);
 
   EXPECT_TRUE(clipboard->IsFormatAvailable(Clipboard::GetBitmapFormatType(),
                                            CLIPBOARD_TYPE_COPY_PASTE));
   const SkBitmap& image = clipboard->ReadImage(CLIPBOARD_TYPE_COPY_PASTE);
   EXPECT_EQ(size, gfx::Size(image.width(), image.height()));
   SkAutoLockPixels image_lock(image);
   for (int j = 0; j < image.height(); ++j) {
     const uint32* row_address = image.getAddr32(0, j);
     for (int i = 0; i < image.width(); ++i) {
       int offset = i + j * image.width();
-      uint32 pixel = SkPreMultiplyColor(bitmap_data[offset]);
-      EXPECT_EQ(pixel, row_address[i])
-          << "i = " << i << ", j = " << j;
+      EXPECT_EQ(bitmap_data[offset], row_address[i]) << "i = " << i
+                                                     << ", j = " << j;
     }
   }
 }
 
 TEST_F(ClipboardTest, SharedBitmapTest) {
   const uint32 fake_bitmap_1[] = {
-    0x46155189, 0xF6A55C8D, 0x79845674, 0xFA57BD89,
-    0x78FD46AE, 0x87C64F5A, 0x36EDC5AF, 0x4378F568,
-    0x91E9F63A, 0xC31EA14F, 0x69AB32DF, 0x643A3FD1,
+      0x46061626, 0xf69f5988, 0x793f2937, 0xfa55b986,
+      0x78772152, 0x87692a30, 0x36322a25, 0x4320401b,
+      0x91848c21, 0xc3177b3c, 0x6946155c, 0x64171952,
   };
   {
     SCOPED_TRACE("first bitmap");
-    TestBitmapWrite(
-        &clipboard(), fake_bitmap_1, sizeof(fake_bitmap_1), gfx::Size(4, 3));
+    TestBitmapWrite(&clipboard(), gfx::Size(4, 3), fake_bitmap_1);
   }
 
   const uint32 fake_bitmap_2[] = {
-    0x46155189, 0xF6A55C8D,
-    0x79845674, 0xFA57BD89,
-    0x78FD46AE, 0x87C64F5A,
-    0x36EDC5AF, 0x4378F568,
-    0x91E9F63A, 0xC31EA14F,
-    0x69AB32DF, 0x643A3FD1,
-    0xA6DF041D, 0x83046278,
+      0x46061626, 0xf69f5988,
+      0x793f2937, 0xfa55b986,
+      0x78772152, 0x87692a30,
+      0x36322a25, 0x4320401b,
+      0x91848c21, 0xc3177b3c,
+      0x6946155c, 0x64171952,
+      0xa6910313, 0x8302323e,
   };
   {
     SCOPED_TRACE("second bitmap");
-    TestBitmapWrite(
-        &clipboard(), fake_bitmap_2, sizeof(fake_bitmap_2), gfx::Size(2, 7));
+    TestBitmapWrite(&clipboard(), gfx::Size(2, 7), fake_bitmap_2);
   }
 }
 
-namespace {
-// A size class that just happens to have the same layout as gfx::Size!
-struct UnsafeSize {
-  int width;
-  int height;
-};
-COMPILE_ASSERT(sizeof(UnsafeSize) == sizeof(gfx::Size),
-               UnsafeSize_must_be_same_size_as_gfx_Size);
-}  // namespace
-
-TEST_F(ClipboardTest, SharedBitmapWithTwoNegativeSizes) {
-  Clipboard::ObjectMapParam placeholder_param;
-  void* crash_me = reinterpret_cast<void*>(57);
-  placeholder_param.resize(sizeof(crash_me));
-  memcpy(&placeholder_param.front(), &crash_me, sizeof(crash_me));
-
-  Clipboard::ObjectMapParam size_param;
-  UnsafeSize size = {-100, -100};
-  size_param.resize(sizeof(size));
-  memcpy(&size_param.front(), &size, sizeof(size));
-
-  Clipboard::ObjectMapParams params;
-  params.push_back(placeholder_param);
-  params.push_back(size_param);
-
-  Clipboard::ObjectMap objects;
-  objects[Clipboard::CBF_SMBITMAP] = params;
-
-  WriteObjectsToClipboard(objects);
-  EXPECT_FALSE(clipboard().IsFormatAvailable(Clipboard::GetBitmapFormatType(),
-                                             CLIPBOARD_TYPE_COPY_PASTE));
-}
-
-TEST_F(ClipboardTest, SharedBitmapWithOneNegativeSize) {
-  Clipboard::ObjectMapParam placeholder_param;
-  void* crash_me = reinterpret_cast<void*>(57);
-  placeholder_param.resize(sizeof(crash_me));
-  memcpy(&placeholder_param.front(), &crash_me, sizeof(crash_me));
-
-  Clipboard::ObjectMapParam size_param;
-  UnsafeSize size = {-100, 100};
-  size_param.resize(sizeof(size));
-  memcpy(&size_param.front(), &size, sizeof(size));
-
-  Clipboard::ObjectMapParams params;
-  params.push_back(placeholder_param);
-  params.push_back(size_param);
-
-  Clipboard::ObjectMap objects;
-  objects[Clipboard::CBF_SMBITMAP] = params;
-
-  WriteObjectsToClipboard(objects);
-  EXPECT_FALSE(clipboard().IsFormatAvailable(Clipboard::GetBitmapFormatType(),
-                                             CLIPBOARD_TYPE_COPY_PASTE));
-}
-
-TEST_F(ClipboardTest, BitmapWithSuperSize) {
-  Clipboard::ObjectMapParam placeholder_param;
-  void* crash_me = reinterpret_cast<void*>(57);
-  placeholder_param.resize(sizeof(crash_me));
-  memcpy(&placeholder_param.front(), &crash_me, sizeof(crash_me));
-
-  Clipboard::ObjectMapParam size_param;
-  // Width just big enough that bytes per row won't fit in a 32-bit
-  // representation.
-  gfx::Size size(0x20000000, 1);
-  size_param.resize(sizeof(size));
-  memcpy(&size_param.front(), &size, sizeof(size));
-
-  Clipboard::ObjectMapParams params;
-  params.push_back(placeholder_param);
-  params.push_back(size_param);
-
-  Clipboard::ObjectMap objects;
-  objects[Clipboard::CBF_SMBITMAP] = params;
-
-  WriteObjectsToClipboard(objects);
-  EXPECT_FALSE(clipboard().IsFormatAvailable(Clipboard::GetBitmapFormatType(),
-                                             CLIPBOARD_TYPE_COPY_PASTE));
-}
-
-TEST_F(ClipboardTest, BitmapWithSuperSize2) {
-  Clipboard::ObjectMapParam placeholder_param;
-  void* crash_me = reinterpret_cast<void*>(57);
-  placeholder_param.resize(sizeof(crash_me));
-  memcpy(&placeholder_param.front(), &crash_me, sizeof(crash_me));
-
-  Clipboard::ObjectMapParam size_param;
-  // Width and height large enough that SkBitmap::getSize() will be truncated.
-  gfx::Size size(0x0fffffff, 0x0fffffff);
-  size_param.resize(sizeof(size));
-  memcpy(&size_param.front(), &size, sizeof(size));
-
-  Clipboard::ObjectMapParams params;
-  params.push_back(placeholder_param);
-  params.push_back(size_param);
-
-  Clipboard::ObjectMap objects;
-  objects[Clipboard::CBF_SMBITMAP] = params;
-
-  WriteObjectsToClipboard(objects);
-  EXPECT_FALSE(clipboard().IsFormatAvailable(Clipboard::GetBitmapFormatType(),
-                                             CLIPBOARD_TYPE_COPY_PASTE));
-}
-
 TEST_F(ClipboardTest, DataTest) {
   const ui::Clipboard::FormatType kFormat =
       ui::Clipboard::GetFormatType("chromium/x-test-format");
   std::string payload("test string");
   Pickle write_pickle;
   write_pickle.WriteString(payload);
 
   {
     ScopedClipboardWriter clipboard_writer(CLIPBOARD_TYPE_COPY_PASTE);
     clipboard_writer.WritePickledData(write_pickle, kFormat);
@@ skipping 252 matching lines @@
   EXPECT_TRUE(clipboard().IsFormatAvailable(
       Clipboard::GetPlainTextWFormatType(), CLIPBOARD_TYPE_COPY_PASTE));
 
   // Make sure the text is what we inserted while simulating the other app
   std::string contents;
   clipboard().ReadAsciiText(CLIPBOARD_TYPE_COPY_PASTE, &contents);
   EXPECT_EQ(contents, new_value);
 }
 #endif
 }  // namespace ui