Revert of Rewrite clipboard write IPC handling to be easier to understand.

ui/base/clipboard/clipboard.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 "ui/base/clipboard/clipboard.h"
 
 #include <iterator>
 #include <limits>
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/size.h"
 
 namespace ui {
 
+namespace {
+
+// Valides a shared bitmap on the clipboard.
+// Returns true if the clipboard data makes sense and it's safe to access the
+// bitmap.
+bool ValidateAndMapSharedBitmap(size_t bitmap_bytes,
+                                base::SharedMemory* bitmap_data) {
+  using base::SharedMemory;
+
+  if (!bitmap_data || !SharedMemory::IsHandleValid(bitmap_data->handle()))
+    return false;
+
+  if (!bitmap_data->Map(bitmap_bytes)) {
+    PLOG(ERROR) << "Failed to map bitmap memory";
+    return false;
+  }
+  return true;
+}
+
+}  // namespace
+
 base::LazyInstance<Clipboard::AllowedThreadsVector>
     Clipboard::allowed_threads_ = LAZY_INSTANCE_INITIALIZER;
 base::LazyInstance<Clipboard::ClipboardMap> Clipboard::clipboard_map_ =
     LAZY_INSTANCE_INITIALIZER;
 base::LazyInstance<base::Lock>::Leaky Clipboard::clipboard_map_lock_ =
     LAZY_INSTANCE_INITIALIZER;
 
 // static
 void Clipboard::SetAllowedThreads(
     const std::vector<base::PlatformThreadId>& allowed_threads) {
@@ skipping 40 matching lines @@
   ClipboardMap* clipboard_map = clipboard_map_.Pointer();
   base::PlatformThreadId id = base::PlatformThread::CurrentId();
   ClipboardMap::iterator it = clipboard_map->find(id);
   if (it != clipboard_map->end()) {
     delete it->second;
     clipboard_map->erase(it);
   }
 }
 
 void Clipboard::DispatchObject(ObjectType type, const ObjectMapParams& params) {
+  // All types apart from CBF_WEBKIT need at least 1 non-empty param.
+  if (type != CBF_WEBKIT && (params.empty() || params[0].empty()))
+    return;
+  // Some other types need a non-empty 2nd param.
+  if ((type == CBF_BOOKMARK || type == CBF_SMBITMAP || type == CBF_DATA) &&
+      (params.size() != 2 || params[1].empty()))
+    return;
   switch (type) {
     case CBF_TEXT:
       WriteText(&(params[0].front()), params[0].size());
       break;
 
     case CBF_HTML:
       if (params.size() == 2) {
         if (params[1].empty())
           return;
         WriteHTML(&(params[0].front()), params[0].size(),
@@ skipping 10 matching lines @@
     case CBF_BOOKMARK:
       WriteBookmark(&(params[0].front()), params[0].size(),
                     &(params[1].front()), params[1].size());
       break;
 
     case CBF_WEBKIT:
       WriteWebSmartPaste();
       break;
 
     case CBF_SMBITMAP: {
-      // Usually, the params are just UTF-8 strings. However, for images,
-      // ScopedClipboardWriter actually sizes the buffer to sizeof(SkBitmap*),
-      // aliases the contents of the vector to a SkBitmap**, and writes the
-      // pointer to the actual SkBitmap in the clipboard object param.
-      const char* packed_pointer_buffer = &params[0].front();
-      WriteBitmap(
-          **reinterpret_cast<SkBitmap* const*>(packed_pointer_buffer));
+      using base::SharedMemory;
+      using base::SharedMemoryHandle;
+
+      if (params[0].size() != sizeof(SharedMemory*) ||
+          params[1].size() != sizeof(gfx::Size)) {
+        return;
+      }
+
+      SkBitmap bitmap;
+      const gfx::Size* unvalidated_size =
+          reinterpret_cast<const gfx::Size*>(&params[1].front());
+      // Let Skia do some sanity checking for us (no negative widths/heights, no
+      // overflows while calculating bytes per row, etc).
+      if (!bitmap.setInfo(SkImageInfo::MakeN32Premul(
+          unvalidated_size->width(), unvalidated_size->height()))) {
+        return;
+      }
+      // Make sure the size is representable as a signed 32-bit int, so
+      // SkBitmap::getSize() won't be truncated.
+      if (!sk_64_isS32(bitmap.computeSize64()))
+        return;
+
+      // It's OK to cast away constness here since we map the handle as
+      // read-only.
+      const char* raw_bitmap_data_const =
+          reinterpret_cast<const char*>(&params[0].front());
+      char* raw_bitmap_data = const_cast<char*>(raw_bitmap_data_const);
+      scoped_ptr<SharedMemory> bitmap_data(
+          *reinterpret_cast<SharedMemory**>(raw_bitmap_data));
+
+      if (!ValidateAndMapSharedBitmap(bitmap.getSize(), bitmap_data.get()))
+        return;
+      bitmap.setPixels(bitmap_data->memory());
+
+      WriteBitmap(bitmap);
       break;
     }
 
     case CBF_DATA:
       WriteData(
           FormatType::Deserialize(
               std::string(&(params[0].front()), params[0].size())),
           &(params[1].front()),
           params[1].size());
       break;
 
     default:
       NOTREACHED();
   }
 }
 
+// static
+bool Clipboard::ReplaceSharedMemHandle(ObjectMap* objects,
+                                       base::SharedMemoryHandle bitmap_handle,
+                                       base::ProcessHandle process) {
+  using base::SharedMemory;
+  bool has_shared_bitmap = false;
+
+  for (ObjectMap::iterator iter = objects->begin(); iter != objects->end();
+       ++iter) {
+    if (iter->first == CBF_SMBITMAP) {
+      // The code currently only accepts sending a single bitmap over this way.
+      // Fail if we ever encounter more than one shmem bitmap structure to fill.
+      if (has_shared_bitmap)
+        return false;
+
+#if defined(OS_WIN)
+      SharedMemory* bitmap = new SharedMemory(bitmap_handle, true, process);
+#else
+      SharedMemory* bitmap = new SharedMemory(bitmap_handle, true);
+#endif
+
+      // There must always be two parameters associated with each shmem bitmap.
+      if (iter->second.size() != 2)
+        return false;
+
+      // We store the shared memory object pointer so it can be retrieved by the
+      // UI thread (see DispatchObject()).
+      iter->second[0].clear();
+      for (size_t i = 0; i < sizeof(SharedMemory*); ++i)
+        iter->second[0].push_back(reinterpret_cast<char*>(&bitmap)[i]);
+      has_shared_bitmap = true;
+    }
+  }
+  return true;
+}
+
 }  // namespace ui