chrome/common/safe_browsing/protobuf_message_param_traits.h

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_COMMON_SAFE_BROWSING_PROTOBUF_MESSAGE_PARAM_TRAITS_H_
   6 #define CHROME_COMMON_SAFE_BROWSING_PROTOBUF_MESSAGE_PARAM_TRAITS_H_
   7
   8 #include <string>
   9
  10 #include "base/pickle.h"
  11 #include "ipc/ipc_message.h"
  12 #include "ipc/ipc_param_traits.h"
  13 #include "third_party/protobuf/src/google/protobuf/repeated_field.h"
  14
  15 namespace IPC {
  16
  17 template <class Element>
  18 struct ParamTraits<google::protobuf::RepeatedPtrField<Element>> {
  19   typedef google::protobuf::RepeatedPtrField<Element> param_type;
  20
  21   static void Write(Message* m, const param_type& p) {
  22     WriteParam(m, p.size());
  23     for (const auto& element : p)
  24       WriteParam(m, element);
  25   }
  26
  27   static bool Read(const Message* m, PickleIterator* iter, param_type* p) {
  28     int size;
  29     if (!iter->ReadLength(&size) || size < 0)
  30       return false;
  31     if (INT_MAX / static_cast<int>(sizeof(void*)) <= size)
  32       return false;
  33     p->Clear();
  34     p->Reserve(size);
  35     while (size--) {
  36       if (!ReadParam(m, iter, p->Add()))
  37         return false;
  38     }
  39     return true;
  40   }
  41
  42   static void Log(const param_type& p, std::string* l) {
  43     bool logged_first = false;
  44     for (const auto& element : p) {
  45       if (logged_first)
  46         l->push_back(' ');
  47       else
  48         logged_first = true;
  49       LogParam(element, l);
  50     }
  51   }
  52 };
  53
  54 }  // namespace IPC
  55
  56 #endif  // CHROME_COMMON_SAFE_BROWSING_PROTOBUF_MESSAGE_PARAM_TRAITS_H_