byte_filer.cpp

Go to the documentation of this file.

#ifndef BYTE_FILER_IMPLEMENTATION_FILE
#define BYTE_FILER_IMPLEMENTATION_FILE

/*****************************************************************************\
*                                                                             *
*  Name   : byte_filer                                                        *
*  Author : Chris Koeritz                                                     *
*                                                                             *
*******************************************************************************
* Copyright (c) 2000-$now By Author.  This program is free software; you can  *
* redistribute it and/or modify it under the terms of the GNU General Public  *
* License as published by the Free Software Foundation; either version 2 of   *
* the License or (at your option) any later version.  This is online at:      *
*     http://www.fsf.org/copyleft/gpl.html                                    *
* Please send any updates to: fred@gruntose.com                               *
\*****************************************************************************/

#include "byte_filer.h"

#include <basis/byte_array.h>
#include <basis/convert_utf.h>
#include <basis/function.h>
#include <basis/istring.h>
#include <basis/portable.h>

#include <stdio.h>
#include <string.h>
#ifdef __UNIX__
  #include <unistd.h>
#endif
#ifdef __WIN32__
  #include <io.h>
#endif

//#define DEBUG_BYTE_FILER
  // uncomment for noisy version of class.

const size_t BTFL_FILE_TELL_LIMIT = size_t(2) * size_t(GIGABYTE);
  // the largest a long integer can represent in the tell system call.

class file_hider
{
public:
  FILE *fp;  // the real file pointer.

  file_hider() : fp(NIL) {}
};


byte_filer::byte_filer()
: _handle(new file_hider),
  _filename(new istring),
  _auto_close(true)
{}

byte_filer::byte_filer(const istring &filename, const istring &perms)
: _handle(new file_hider),
  _filename(new istring),
  _auto_close(true)
{ open(filename, perms); }

byte_filer::byte_filer(const char *filename, const char *perms)
: _handle(new file_hider),
  _filename(new istring),
  _auto_close(true)
{ open(filename, perms); }

byte_filer::byte_filer(bool auto_close, void *handle)
: _handle(new file_hider),
  _filename(new istring),
  _auto_close(auto_close)
{
  if (handle) {
    _handle->fp = (FILE *)handle;
  }
}

byte_filer::~byte_filer() { close(); WHACK(_handle); WHACK(_filename); }

istring byte_filer::filename() const { return *_filename; }

size_t byte_filer::file_size_limit() { return BTFL_FILE_TELL_LIMIT; }

bool byte_filer::open(const istring &filename, const istring &perms)
{
  close();
  _auto_close = true;  // reset since we know we're opening this.
  *_filename = filename;
#ifndef __WIN32__
  _handle->fp = filename.t()? fopen(filename.s(), perms.s()) : NIL;
#else
  _handle->fp = filename.t()? _wfopen((wchar_t *)(UTF16 *)transcode_to_utf16(filename),
      (wchar_t *)(UTF16 *)transcode_to_utf16(perms)) : NIL;

#ifdef DEBUG_BYTE_FILER
  if (!_handle->fp)
    wprintf((wchar_t *)(UTF16 *)transcode_to_utf16("could not open: %ls\n"),
        (wchar_t *)(UTF16 *)transcode_to_utf16(filename));
#endif

#endif
  return good();
}

void byte_filer::close()
{
  *_filename = "";
  if (_auto_close && _handle->fp) fclose(_handle->fp);
  _handle->fp = NIL;
}

bool byte_filer::good() { return !!_handle->fp; }

size_t byte_filer::tell()
{
  if (!_handle->fp) return 0;
  long to_return = ::ftell(_handle->fp);
  if (to_return == -1) {
    // if we couldn't get the size, either the file isn't there or the size
    // is too big for our OS to report.
    if (good()) return BTFL_FILE_TELL_LIMIT;
    else return 0;
  }
  return size_t(to_return);
}

void *byte_filer::file_handle() { return _handle->fp; }

bool byte_filer::eof() { return !_handle->fp ? true : !!feof(_handle->fp); }

int byte_filer::read(byte *buff, int size)
{ return !_handle->fp ? 0 : int(::fread((char *)buff, 1, size, _handle->fp)); }

int byte_filer::write(const byte *buff, int size)
{ return !_handle->fp ? 0 : int(::fwrite((char *)buff, 1, size, _handle->fp)); }

int byte_filer::read(byte_array &buff, int desired_size)
{
  buff.reset(desired_size);
  int to_return = read(buff.access(), desired_size); 
  buff.zap(to_return, buff.length() - 1);
  return to_return;
}

int byte_filer::write(const byte_array &buff)
{ return write(buff.observe(), buff.length()); }

size_t byte_filer::length()
{
  size_t current_posn = tell();
  seek(0, FROM_END);  // jump to end of file.
  size_t file_size = tell();  // get position.
  seek(int(current_posn), FROM_START);  // jump back to previous place.
  return file_size;
}

int byte_filer::read(istring &s, int desired_size)
{
  s.pad(desired_size + 2);
  int found = read((byte *)s.observe(), desired_size);
  if (non_negative(found)) s[found] = '\0';
  s.shrink();
  return found;
}

int byte_filer::write(const istring &s, bool add_null)
{
  int len = s.length();
  if (add_null) len++;
  return write((byte *)s.observe(), len);
}

void byte_filer::flush()
{
  if (!_handle->fp) return;
  ::fflush(_handle->fp);
}

bool byte_filer::truncate()
{
  flush();
  int fnum = fileno(_handle->fp);
#ifdef __WIN32__
  return SetEndOfFile((HANDLE)_get_osfhandle(fnum));
#else
  size_t posn = tell();
  // if we're at the highest point we can be, we no longer trust our
  // ability to truncate properly.
  if (posn >= file_size_limit())
    return false;
  return !ftruncate(fnum, posn);
#endif
}

bool byte_filer::seek(int where, origins origin)
{
  if (!_handle->fp) return false;
  int real_origin;
  switch (origin) {
    case FROM_START: real_origin = SEEK_SET; break;
    case FROM_END: real_origin = SEEK_END; break;
    case FROM_CURRENT: real_origin = SEEK_CUR; break;
    default: return false;  // not a valid choice.
  }
  int ret = ::fseek(_handle->fp, where, real_origin);
  return !ret;
}

int byte_filer::getline(byte *buff, int desired_size)
{
  if (!_handle->fp) return 0;
  char *ret = ::fgets((char *)buff, desired_size, _handle->fp);
  return !ret? 0 : int(strlen((char *)buff)) + 1;
}

int byte_filer::getline(byte_array &buff, int desired_size)
{
  buff.reset(desired_size + 1);
  return getline(buff.access(), desired_size);
}

int byte_filer::getline(istring &buff, int desired_size)
{
  buff.pad(desired_size + 1);
  int to_return = getline((byte *)buff.access(), desired_size);
  if (non_negative(to_return)) buff[to_return] = '\0';
  buff.shrink();
  return to_return;
}


#endif //BYTE_FILER_IMPLEMENTATION_FILE

---