arc/vm/vsock_proxy/pipe_stream.cc

// Copyright 2018 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 "arc/vm/vsock_proxy/pipe_stream.h"

#include <unistd.h>

#include <utility>

#include <base/files/file_util.h>
#include <base/logging.h>
#include <base/posix/eintr_wrapper.h>

#include "arc/vm/vsock_proxy/message.pb.h"

namespace arc {

PipeStream::PipeStream(base::ScopedFD pipe_fd) : pipe_fd_(std::move(pipe_fd)) {}

PipeStream::~PipeStream() = default;

bool PipeStream::Read(arc_proxy::VSockMessage* message) {
  char buf[4096];
  ssize_t size = HANDLE_EINTR(read(pipe_fd_.get(), buf, sizeof(buf)));
  if (size < 0) {
    PLOG(ERROR) << "Failed to read";
    return false;
  }

  if (size == 0)
    message->mutable_close();
  else
    message->mutable_data()->set_blob(buf, size);
  return true;
}

bool PipeStream::Write(arc_proxy::Data* data) {
  // WriteFileDescriptor takes care of the short write.
  const auto& blob = data->blob();
  if (!base::WriteFileDescriptor(pipe_fd_.get(), blob.data(), blob.size())) {
    PLOG(ERROR) << "Failed to write";
    return false;
  }
  return true;
}

bool PipeStream::Pread(uint64_t count,
                       uint64_t offset,
                       arc_proxy::PreadResponse* response) {
  LOG(ERROR) << "Pread for pipe file descriptor is unsupported.";
  return false;
}

}  // namespace arc