Verify FormatType in ScopedClipboardWriter::WritePickledData.

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 <set>
 
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/synchronization/lock.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/size.h"
 
 namespace ui {
 
@@ skipping 25 matching lines @@
 
 // Mapping from threads to clipboard objects.
 typedef std::map<base::PlatformThreadId, Clipboard*> ClipboardMap;
 static base::LazyInstance<ClipboardMap> g_clipboard_map =
     LAZY_INSTANCE_INITIALIZER;
 
 // Mutex that controls access to |g_clipboard_map|.
 static base::LazyInstance<base::Lock>::Leaky
     g_clipboard_map_lock = LAZY_INSTANCE_INITIALIZER;
 
+// Set of registered custom formats.
+static base::LazyInstance<std::set<Clipboard::FormatType> >
+    g_registered_formats = LAZY_INSTANCE_INITIALIZER;
+
+// Mutex that controls access to |g_registered_formats|.
+static base::LazyInstance<base::Lock>::Leaky g_registered_formats_lock =
+    LAZY_INSTANCE_INITIALIZER;
+
 }  // namespace
 
 // static
 void Clipboard::SetAllowedThreads(
     const std::vector<base::PlatformThreadId>& allowed_threads) {
   base::AutoLock lock(g_clipboard_map_lock.Get());
 
   g_allowed_threads.Get().clear();
   std::copy(allowed_threads.begin(), allowed_threads.end(),
             std::back_inserter(g_allowed_threads.Get()));
@@ skipping 34 matching lines @@
 
   ClipboardMap* clipboard_map = g_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);
   }
 }
 
+// static
+Clipboard::FormatType Clipboard::GetFormatType(
+    const std::string& format_string) {
+  FormatType format = GetFormatTypeInternal(format_string);
+  {
+    base::AutoLock lock(g_registered_formats_lock.Get());
+    g_registered_formats.Get().insert(format);
+  }
+  return format;
+}
+
+bool Clipboard::IsRegisteredFormatType(const FormatType& format) {
+  // Terrible hack to force web/pepper custom types to be registered.
+  // TODO(dcheng): There's got to a better way...
+  GetWebCustomDataFormatType();
+  GetPepperCustomDataFormatType();
+
+  base::AutoLock lock(g_registered_formats_lock.Get());
+  const std::set<FormatType>& registered_formats = g_registered_formats.Get();
+  return registered_formats.find(format) != registered_formats.end();
+}
+
 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:
@@ skipping 107 matching lines @@
       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