shaka-packager/packager/media/file/local_file.cc

// Copyright 2014 Google Inc. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd

#include "packager/media/file/local_file.h"

#include <stdio.h>

#include "packager/base/files/file_util.h"
#include "packager/base/logging.h"

namespace edash_packager {
namespace media {

LocalFile::LocalFile(const char* file_name, const char* mode)
    : File(file_name), file_mode_(mode), internal_file_(NULL) {}

bool LocalFile::Close() {
  bool result = true;
  if (internal_file_) {
    result = base::CloseFile(internal_file_);
    internal_file_ = NULL;
  }
  delete this;
  return result;
}

int64_t LocalFile::Read(void* buffer, uint64_t length) {
  DCHECK(buffer != NULL);
  DCHECK(internal_file_ != NULL);
  return fread(buffer, sizeof(char), length, internal_file_);
}

int64_t LocalFile::Write(const void* buffer, uint64_t length) {
  DCHECK(buffer != NULL);
  DCHECK(internal_file_ != NULL);
  return fwrite(buffer, sizeof(char), length, internal_file_);
}

int64_t LocalFile::Size() {
  DCHECK(internal_file_ != NULL);

  // Flush any buffered data, so we get the true file size.
  if (!Flush()) {
    LOG(ERROR) << "Cannot flush file.";
    return -1;
  }

  int64_t file_size;
  if (!base::GetFileSize(base::FilePath(file_name()), &file_size)) {
    LOG(ERROR) << "Cannot get file size.";
    return -1;
  }
  return file_size;
}

bool LocalFile::Flush() {
  DCHECK(internal_file_ != NULL);
  return ((fflush(internal_file_) == 0) && !ferror(internal_file_));
}

bool LocalFile::Seek(uint64_t position) {
  return fseeko(internal_file_, position, SEEK_SET) >= 0;
}

bool LocalFile::Tell(uint64_t* position) {
  off_t offset = ftello(internal_file_);
  if (offset < 0)
    return false;
  *position = offset;
  return true;
}

LocalFile::~LocalFile() {}

bool LocalFile::Open() {
  internal_file_ =
      base::OpenFile(base::FilePath(file_name()), file_mode_.c_str());
  return (internal_file_ != NULL);
}

bool LocalFile::Delete(const char* file_name) {
  return base::DeleteFile(base::FilePath(file_name), false);
}

}  // namespace media
}  // namespace edash_packager