crashpad/util/file/file_reader.cc

// Copyright 2015 The Crashpad Authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#include "util/file/file_reader.h"

#include "base/logging.h"
#include "base/numerics/safe_conversions.h"
#include "build/build_config.h"

namespace crashpad {

bool FileReaderInterface::ReadExactly(void* data, size_t size) {
  FileOperationResult expect = base::checked_cast<FileOperationResult>(size);
  char* data_c = static_cast<char*>(data);

  FileOperationResult total_bytes = 0;
  while (size > 0) {
    FileOperationResult bytes = Read(data, size);
    if (bytes < 0) {
      // Read() will have logged its own error.
      return false;
    }

    DCHECK_LE(static_cast<size_t>(bytes), size);

    if (bytes == 0) {
      break;
    }

    data_c += bytes;
    size -= bytes;
    total_bytes += bytes;
  }

  if (total_bytes != expect) {
    LOG(ERROR) << "ReadExactly(): expected " << expect << ", observed "
               << total_bytes;
    return false;
  }

  return true;
}

WeakFileHandleFileReader::WeakFileHandleFileReader(FileHandle file_handle)
    : file_handle_(file_handle) {
}

WeakFileHandleFileReader::~WeakFileHandleFileReader() {
}

FileOperationResult WeakFileHandleFileReader::Read(void* data, size_t size) {
  DCHECK_NE(file_handle_, kInvalidFileHandle);

  base::checked_cast<FileOperationResult>(size);
  FileOperationResult rv = ReadFile(file_handle_, data, size);
  if (rv < 0) {
    PLOG(ERROR) << kNativeReadFunctionName;
    return -1;
  }

  return rv;
}

FileOffset WeakFileHandleFileReader::Seek(FileOffset offset, int whence) {
  DCHECK_NE(file_handle_, kInvalidFileHandle);
  return LoggingSeekFile(file_handle_, offset, whence);
}

FileReader::FileReader()
    : file_(),
      weak_file_handle_file_reader_(kInvalidFileHandle) {
}

FileReader::~FileReader() {
}

bool FileReader::Open(const base::FilePath& path) {
  CHECK(!file_.is_valid());
  file_.reset(LoggingOpenFileForRead(path));
  if (!file_.is_valid()) {
    return false;
  }

  weak_file_handle_file_reader_.set_file_handle(file_.get());
  return true;
}

void FileReader::Close() {
  CHECK(file_.is_valid());

  weak_file_handle_file_reader_.set_file_handle(kInvalidFileHandle);
  file_.reset();
}

FileOperationResult FileReader::Read(void* data, size_t size) {
  DCHECK(file_.is_valid());
  return weak_file_handle_file_reader_.Read(data, size);
}

FileOffset FileReader::Seek(FileOffset offset, int whence) {
  DCHECK(file_.is_valid());
  return weak_file_handle_file_reader_.Seek(offset, whence);
}

WeakStdioFileReader::WeakStdioFileReader(FILE* file)
    : file_(file) {
}

WeakStdioFileReader::~WeakStdioFileReader() {
}

FileOperationResult WeakStdioFileReader::Read(void* data, size_t size) {
  DCHECK(file_);

  size_t rv = fread(data, 1, size, file_);
  if (rv != size && ferror(file_)) {
    STDIO_PLOG(ERROR) << "fread";
    return -1;
  }
  if (rv > size) {
    LOG(ERROR) << "fread: expected " << size << ", observed " << rv;
    return -1;
  }

  return rv;
}

FileOffset WeakStdioFileReader::Seek(FileOffset offset, int whence) {
  DCHECK(file_);
  if (fseeko(file_, offset, whence) == -1) {
    STDIO_PLOG(ERROR) << "fseeko";
    return -1;
  }

  FileOffset new_offset = ftello(file_);
  if (new_offset == -1) {
    STDIO_PLOG(ERROR) << "ftello";
    return -1;
  }

  return new_offset;
}

}  // namespace crashpad