libcruft-util/io.cpp

#include "io.hpp"

#include "debug.hpp"
#include "except.hpp"

#include <cstdio>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>


using namespace std;
using namespace util;

//----------------------------------------------------------------------------
uint8_t *
util::slurp (const boost::filesystem::path& path)  {
    fd_ref fd(open (path.string ().c_str (), O_RDONLY)); // | O_CLOEXEC));
    
    // Calculate the total file size
    off_t size = lseek (fd, 0, SEEK_END);
    if (size == (off_t)-1)
        throw errno_error();

    if (lseek (fd, 0, SEEK_SET) == (off_t)-1)
        throw errno_error ();

    // Allocate a buffer, and keep reading until it's full. We provide a null
    // padding at the tail as a 'just in case' measure for string manipulation.
    unique_ptr <uint8_t[]> buffer (new uint8_t[size + 1]);
    buffer.get()[size] = '\0';

    check_hard (size >= 0);
    size_t remaining = (size_t)size;
    uint8_t *cursor = buffer.get();

    while (remaining) {
        ssize_t consumed = read (fd, cursor, remaining);
        if (consumed == -1)
            throw errno_error();
        check_hard (        consumed >= 0);
        check_hard ((size_t)consumed <= remaining);

        remaining -= (size_t)consumed;
        cursor    += consumed;
    }

    return buffer.release();
}

//----------------------------------------------------------------------------
fd_ref::fd_ref (int _fd):
    fd (_fd)
{
    if (fd < 0) 
        throw invalid_argument ("invalid file descriptor");
}


fd_ref::~fd_ref () {
    check (fd >= 0);
    close (fd);
}


fd_ref::operator int (void) const
    { return fd; }



//----------------------------------------------------------------------------
int
indenter::overflow (int ch) {
    if (m_line_start && ch != '\n')
        m_dest->sputn (m_indent.data (), sign_cast<std::streamsize> (m_indent.size ()));

    m_line_start = ch == '\n';
    return m_dest->sputc (ch);
}


indenter::indenter (std::streambuf* _dest, size_t _indent)
    : m_dest       (_dest)
    , m_line_start (true)
    , m_indent     (_indent, ' ')
    , m_owner      (NULL)
{ ; }


indenter::indenter (std::ostream& _dest, size_t _indent)
    : m_dest       (_dest.rdbuf())
    , m_line_start (true)
    , m_indent     (_indent, ' ')
    , m_owner      (&_dest)
{
    m_owner->rdbuf (this);
}


indenter::~indenter ()
{
    if (m_owner != NULL)
        m_owner->rdbuf (m_dest);
}


//----------------------------------------------------------------------------
#if defined(HAVE_MMAP)
#include <sys/mman.h>


mapped_file::mapped_file (const char *_path):
    m_fd (open (_path, O_RDONLY))
{ load_fd (); } 


mapped_file::mapped_file (const std::string &_path):
    m_fd (open (_path.c_str (), O_RDONLY))
{ load_fd (); } 


mapped_file::mapped_file (const boost::filesystem::path &_path):
    m_fd (open (_path.native ().c_str (), O_RDONLY))
{ load_fd (); } 


mapped_file::~mapped_file () {
    check (m_data != NULL);
    munmap (m_data, m_size);
}


void
mapped_file::load_fd (void) {
    struct stat meta;
    if (fstat (m_fd, &meta) < 0)
        throw errno_error ();

    m_size = (size_t)meta.st_size;
    m_data = (uint8_t *)mmap (NULL, m_size, PROT_READ, MAP_PRIVATE, m_fd, 0);
    if (m_data == MAP_FAILED)
        throw errno_error ();
}


size_t
mapped_file::size (void) const {
    check (m_size > 0);
    check (m_data != NULL);

    return m_size;
}


const uint8_t*
mapped_file::data (void) const {
    check (m_size > 0);
    check (m_data != NULL);

    return m_data;
}
#endif