garnet/bin/ktrace_provider/device_reader.cc

// Copyright 2019 The Fuchsia 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 "garnet/bin/ktrace_provider/device_reader.h"

#include <fcntl.h>
#include <limits.h>
#include <sys/stat.h>
#include <unistd.h>

#include <src/lib/files/eintr_wrapper.h>

namespace ktrace_provider {

namespace {
constexpr char kTraceDev[] = "/dev/misc/ktrace";
}  // namespace

DeviceReader::DeviceReader() : Reader(buffer_, kChunkSize), fd_(open(kTraceDev, O_RDONLY)) {}

void DeviceReader::ReadMoreData() {
  memcpy(buffer_, current_, AvailableBytes());
  char* new_marker = buffer_ + AvailableBytes();

  while (new_marker < end_) {
    int bytes_read = HANDLE_EINTR(
        read(fd_.get(), new_marker, std::distance(const_cast<const char*>(new_marker), end_)));

    if (bytes_read <= 0)
      break;

    new_marker += bytes_read;
  }

  marker_ = new_marker;
  current_ = buffer_;
}

}  // namespace ktrace_provider