15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(dcheng): For efficiency reasons, consider passing custom data around 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// as a vector instead. It allows us to append a 75d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles)// std::pair<base::string16, base::string16> and swap the deserialized values. 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/clipboard/custom_data_helper.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <utility> 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/pickle.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace ui { 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SkippablePickle : public Pickle { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkippablePickle(const void* data, size_t data_len); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool SkipString16(PickleIterator* iter); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)SkippablePickle::SkippablePickle(const void* data, size_t data_len) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : Pickle(reinterpret_cast<const char*>(data), data_len) { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool SkippablePickle::SkipString16(PickleIterator* iter) { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(iter); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int len; 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!ReadLength(iter, &len)) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 35a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) return iter->SkipBytes(len * sizeof(base::char16)); 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReadCustomDataTypes(const void* data, 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t data_length, 42a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) std::vector<base::string16>* types) { 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkippablePickle pickle(data, data_length); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PickleIterator iter(pickle); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint64 size = 0; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadUInt64(&iter, &size)) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Keep track of the original elements in the types vector. On failure, we 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // truncate the vector to the original size since we want to ignore corrupt 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // custom data pickles. 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint64 original_size = types->size(); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (uint64 i = 0; i < size; ++i) { 56a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) types->push_back(base::string16()); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadString16(&iter, &types->back()) || 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) !pickle.SkipString16(&iter)) { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) types->resize(original_size); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReadCustomDataForType(const void* data, 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t data_length, 67a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) const base::string16& type, 68a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16* result) { 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkippablePickle pickle(data, data_length); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PickleIterator iter(pickle); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint64 size = 0; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadUInt64(&iter, &size)) 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (uint64 i = 0; i < size; ++i) { 77a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 deserialized_type; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadString16(&iter, &deserialized_type)) 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (deserialized_type == type) { 814e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) ignore_result(pickle.ReadString16(&iter, result)); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.SkipString16(&iter)) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ReadCustomDataIntoMap(const void* data, 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t data_length, 91a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) std::map<base::string16, base::string16>* result) { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Pickle pickle(reinterpret_cast<const char*>(data), data_length); 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PickleIterator iter(pickle); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) uint64 size = 0; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadUInt64(&iter, &size)) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (uint64 i = 0; i < size; ++i) { 100a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 type; 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadString16(&iter, &type)) { 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Data is corrupt, return an empty map. 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->clear(); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 106a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) std::pair<std::map<base::string16, base::string16>::iterator, bool> 107a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) insert_result = result->insert(std::make_pair(type, base::string16())); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!pickle.ReadString16(&iter, &insert_result.first->second)) { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Data is corrupt, return an empty map. 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result->clear(); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 116a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)void WriteCustomDataToPickle( 117a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) const std::map<base::string16, base::string16>& data, 118a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) Pickle* pickle) { 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle->WriteUInt64(data.size()); 120a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) for (std::map<base::string16, base::string16>::const_iterator it = 121a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) data.begin(); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) it != data.end(); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ++it) { 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle->WriteString16(it->first); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) pickle->WriteString16(it->second); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace ui 130