Drag on linux

ui/base/x/selection_utils.cc

 // Copyright (c) 2013 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/x/selection_utils.h"
 
 #include <set>
 
 #include "base/i18n/icu_string_conversions.h"
 #include "base/logging.h"
 #include "base/strings/utf_string_conversions.h"
 #include "ui/base/clipboard/clipboard.h"
 #include "ui/base/x/x11_atom_cache.h"
+#include "ui/base/x/x11_util.h"
 
 namespace ui {
 
 const char kMimeTypeMozillaURL[] = "text/x-moz-url";
 const char kString[] = "STRING";
 const char kText[] = "TEXT";
 const char kUtf8String[] = "UTF8_STRING";
 
 const char* kSelectionDataAtoms[] = {
   Clipboard::kMimeTypeHTML,
@@ skipping 44 matching lines @@
   for (InternalMap::iterator it = data_.begin(); it != data_.end(); ++it)
     to_delete.insert(it->second.first);
 
   for (std::set<char*>::iterator it = to_delete.begin(); it != to_delete.end();
        ++it) {
     delete [] *it;
   }
 }
 
 void SelectionFormatMap::Insert(::Atom atom, char* data, size_t size) {
-  DCHECK(data_.find(atom) == data_.end());
+  // Views code often inserts the same content multiple times, so we have to
+  // free old data. Only call delete when it's the last pointer we have to that
+  // data.
+  InternalMap::iterator exists_it = data_.find(atom);
+  if (exists_it != data_.end()) {
+    int count = 0;
+    for (InternalMap::iterator it = data_.begin(); it != data_.end(); ++it) {
+      if (it->second.first == exists_it->second.first)
+        count++;
+    }
+
+    if (count == 1)
+      delete [] exists_it->second.first;
+  }
+
   data_.insert(std::make_pair(atom, std::make_pair(data, size)));
 }
 
+ui::SelectionData* SelectionFormatMap::GetFirstOf(
+    const std::vector< ::Atom>& requested_types) const {
+  for (std::vector< ::Atom>::const_iterator it = requested_types.begin();
+       it != requested_types.end(); ++it) {
+    const_iterator data_it = data_.find(*it);
+    if (data_it != data_.end()) {
+      ui::SelectionData* data = new SelectionData;
+      data->Set(data_it->first, data_it->second.first, data_it->second.second,
+                false);
+      return data;
+    }
+  }
+
+  return NULL;
+}
+
+std::vector< ::Atom> SelectionFormatMap::GetTypes() const {
+  std::vector< ::Atom> atoms;
+  for (const_iterator it = data_.begin(); it != data_.end(); ++it)
+    atoms.push_back(it->first);
+
+  return atoms;
+}
+
+SelectionFormatMap* SelectionFormatMap::Clone() const {

[Daniel Erat, 2013/06/17 22:09:57]

return a scoped_ptr instead?

+  SelectionFormatMap* ret = new SelectionFormatMap;
+
+  for (const_iterator it = data_.begin(); it != data_.end(); ++it) {
+    char* data_copy = new char[it->second.second];
+    memcpy(data_copy, it->second.first, it->second.second);
+    ret->Insert(it->first, data_copy, it->second.second);
+  }
+
+  return ret;
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 
-SelectionData::SelectionData(Display* x_display)
+SelectionData::SelectionData()
     : type_(None),
       data_(NULL),
       size_(0),
       owned_(false),
-      atom_cache_(x_display, kSelectionDataAtoms) {
+      atom_cache_(ui::GetXDisplay(), kSelectionDataAtoms) {
 }
 
 SelectionData::~SelectionData() {
   if (owned_)
     XFree(data_);
 }
 
 void SelectionData::Set(::Atom type, char* data, size_t size, bool owned) {
   if (owned_)
     XFree(data_);
@@ skipping 49 matching lines @@
 
 void SelectionData::AssignTo(std::string* result) const {
   result->assign(data_, size_);
 }
 
 void SelectionData::AssignTo(string16* result) const {
   result->assign(reinterpret_cast<base::char16*>(data_), size_ / 2);
 }
 
 }  // namespace ui