system-proxy/protobuf_util.cc

// Copyright 2020 The Chromium OS 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 "system-proxy/protobuf_util.h"

#include <vector>

#include <base/files/file_util.h>

namespace system_proxy {

bool ReadProtobuf(int in_fd, google::protobuf::MessageLite* message) {
  size_t proto_size = 0;
  // The first part of the message will be the size of the actual message.
  // Because the message is a serialized protobuf, we need to read the whole
  // message before deserializing it.
  if (!base::ReadFromFD(in_fd, reinterpret_cast<char*>(&proto_size),
                        sizeof(proto_size)))
    return false;
  std::vector<char> buf(proto_size);
  // Tries to read exactly buf.size() bytes from in_fd and returns true if
  // succeeded, false otherwise. If the read() get interrupted by EINTR it will
  // resume reading by itself with a limited number of attempts.
  if (!base::ReadFromFD(in_fd, buf.data(), buf.size()))
    return false;

  return message->ParseFromArray(buf.data(), buf.size());
}

bool WriteProtobuf(int out_fd, const google::protobuf::MessageLite& message) {
  size_t size = message.ByteSizeLong();
  char* size_data = reinterpret_cast<char*>(&size);
  std::vector<char> buf(size_data, size_data + sizeof(size));
  buf.resize(size + sizeof(size));

  if (!message.SerializeToArray(buf.data() + sizeof(size), size))
    return false;

  return base::WriteFileDescriptor(out_fd, buf.data(), buf.size());
}
}  // namespace system_proxy