123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960 |
- // Copyright 2017 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.
- #ifndef IPC_IPC_MESSAGE_PROTOBUF_UTILS_H_
- #define IPC_IPC_MESSAGE_PROTOBUF_UTILS_H_
- #include "build/build_config.h"
- #include "base/pickle.h"
- #include "ipc/ipc_param_traits.h"
- #include "ipc/ipc_message_utils.h"
- #include "third_party/protobuf/src/google/protobuf/repeated_field.h"
- namespace IPC {
- template <class RepeatedFieldLike, class StorageType>
- struct RepeatedFieldParamTraits {
- typedef RepeatedFieldLike param_type;
- static void Write(base::Pickle* m, const param_type& p) {
- WriteParam(m, p.size());
- for (int i = 0; i < p.size(); i++)
- WriteParam(m, p.Get(i));
- }
- static bool Read(const base::Pickle* m,
- base::PickleIterator* iter,
- param_type* r) {
- size_t size;
- if (!iter->ReadLength(&size))
- return false;
- // Avoid integer overflow / assertion failure in Reserve() function.
- if (size > INT_MAX / sizeof(StorageType))
- return false;
- r->Reserve(size);
- for (size_t i = 0; i < size; i++) {
- if (!ReadParam(m, iter, r->Add()))
- return false;
- }
- return true;
- }
- static void Log(const param_type& p, std::string* l) {
- for (int i = 0; i < p.size(); ++i) {
- if (i != 0)
- l->append(" ");
- LogParam(p.Get(i), l);
- }
- }
- };
- template <class P>
- struct ParamTraits<google::protobuf::RepeatedField<P>> :
- RepeatedFieldParamTraits<google::protobuf::RepeatedField<P>, P> {};
- template <class P>
- struct ParamTraits<google::protobuf::RepeatedPtrField<P>> :
- RepeatedFieldParamTraits<google::protobuf::RepeatedPtrField<P>, void*> {};
- } // namespace IPC
- #endif // IPC_IPC_MESSAGE_PROTOBUF_UTILS_H_
|