2011-05-23 17:18:52 +10:00
|
|
|
/*
|
2011-06-21 20:26:32 +10:00
|
|
|
* This file is part of libgim.
|
2011-05-23 17:18:52 +10:00
|
|
|
*
|
2012-04-11 15:18:26 +10:00
|
|
|
* libgim is free software: you can redistribute it and/or modify it under the
|
2011-05-23 17:18:52 +10:00
|
|
|
* terms of the GNU General Public License as published by the Free Software
|
|
|
|
* Foundation, either version 3 of the License, or (at your option) any later
|
|
|
|
* version.
|
|
|
|
*
|
2012-04-11 15:18:26 +10:00
|
|
|
* libgim is distributed in the hope that it will be useful, but WITHOUT ANY
|
2011-05-23 17:18:52 +10:00
|
|
|
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
|
|
* FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
|
|
* details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
2011-06-21 20:26:32 +10:00
|
|
|
* along with libgim. If not, see <http://www.gnu.org/licenses/>.
|
2011-05-23 17:18:52 +10:00
|
|
|
*
|
2014-07-07 15:19:01 +10:00
|
|
|
* Copyright 2010-2014 Danny Robson <danny@nerdcruft.net>
|
2011-05-23 17:18:52 +10:00
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef __UTIL_IO_HPP
|
|
|
|
#define __UTIL_IO_HPP
|
|
|
|
|
2011-05-25 23:02:39 +10:00
|
|
|
#include "annotations.hpp"
|
2012-04-12 14:06:26 +10:00
|
|
|
#include "types.hpp"
|
2012-04-26 18:18:09 +10:00
|
|
|
#include "memory.hpp"
|
2011-05-25 23:02:39 +10:00
|
|
|
|
2011-05-23 17:18:52 +10:00
|
|
|
#include <cstdio>
|
|
|
|
#include <cstdint>
|
|
|
|
#include <memory>
|
2011-10-01 01:58:10 +10:00
|
|
|
#include <boost/filesystem/path.hpp>
|
2011-05-23 17:18:52 +10:00
|
|
|
|
|
|
|
|
2011-10-01 01:58:10 +10:00
|
|
|
namespace util {
|
|
|
|
/// Specifies bitwise combinations of IO access rights.
|
|
|
|
enum access_t {
|
|
|
|
ACCESS_READ = 1 << 0,
|
|
|
|
ACCESS_WRITE = 1 << 1,
|
|
|
|
ACCESS_READWRITE = ACCESS_READ | ACCESS_WRITE
|
|
|
|
};
|
2012-11-09 15:16:07 +11:00
|
|
|
|
2011-10-01 01:58:10 +10:00
|
|
|
/// Reads an entire file into memory. Caller frees the result. Guarantees a
|
|
|
|
/// null trailing byte.
|
2012-04-26 18:18:09 +10:00
|
|
|
std::unique_ptr<char []>
|
2011-10-01 01:58:10 +10:00
|
|
|
slurp (const boost::filesystem::path&) mustuse;
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2012-08-15 16:03:48 +10:00
|
|
|
void
|
|
|
|
write (const boost::filesystem::path &, const char *data, size_t len);
|
|
|
|
|
2014-07-15 19:47:15 +10:00
|
|
|
|
|
|
|
///------------------------------------------------------------------------
|
2011-10-01 01:58:10 +10:00
|
|
|
/// A simple RAII wrapper for file descriptors
|
|
|
|
struct fd_ref {
|
|
|
|
public:
|
|
|
|
int fd;
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2012-04-26 18:19:12 +10:00
|
|
|
explicit fd_ref (int _fd);
|
2014-07-29 02:20:02 +10:00
|
|
|
explicit fd_ref (const boost::filesystem::path&, access_t access);
|
2011-10-01 01:58:10 +10:00
|
|
|
~fd_ref ();
|
|
|
|
|
|
|
|
operator int (void) const;
|
|
|
|
};
|
2011-05-23 17:18:52 +10:00
|
|
|
|
|
|
|
|
2014-06-29 22:49:26 +10:00
|
|
|
//-------------------------------------------------------------------------
|
2012-11-09 15:16:07 +11:00
|
|
|
class indenter : public std::streambuf {
|
2012-04-12 14:06:26 +10:00
|
|
|
protected:
|
|
|
|
std::streambuf* m_dest;
|
|
|
|
bool m_line_start;
|
|
|
|
std::string m_indent;
|
|
|
|
std::ostream* m_owner;
|
|
|
|
|
|
|
|
protected:
|
|
|
|
virtual int overflow (int ch);
|
|
|
|
|
|
|
|
public:
|
|
|
|
explicit indenter (std::streambuf* _dest, size_t _indent = 4);
|
|
|
|
explicit indenter (std::ostream& _dest, size_t _indent = 4);
|
|
|
|
|
|
|
|
virtual ~indenter ();
|
|
|
|
};
|
|
|
|
|
2014-06-29 22:49:26 +10:00
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
template <typename T>
|
|
|
|
struct indented {
|
|
|
|
indented (const T &_data);
|
|
|
|
const T &data;
|
|
|
|
};
|
|
|
|
|
|
|
|
|
2014-08-01 20:43:51 +10:00
|
|
|
//-------------------------------------------------------------------------
|
2014-06-29 22:49:26 +10:00
|
|
|
template <typename T>
|
|
|
|
indented<T>
|
|
|
|
make_indented (const T &_data);
|
|
|
|
|
2014-08-01 20:43:51 +10:00
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
template <typename T>
|
|
|
|
std::ostream&
|
|
|
|
operator<< (std::ostream &os, const util::indented<T> &v);
|
|
|
|
|
|
|
|
|
2014-06-29 22:49:26 +10:00
|
|
|
//-------------------------------------------------------------------------
|
2012-04-26 18:19:37 +10:00
|
|
|
class scoped_cwd {
|
|
|
|
public:
|
|
|
|
scoped_cwd ();
|
|
|
|
~scoped_cwd ();
|
|
|
|
|
|
|
|
protected:
|
2012-05-08 15:04:55 +10:00
|
|
|
boost::filesystem::path m_original;
|
2012-04-26 18:19:37 +10:00
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
void set_cwd (const boost::filesystem::path &);
|
2012-04-12 14:06:26 +10:00
|
|
|
|
|
|
|
|
2011-07-04 16:26:08 +10:00
|
|
|
#if defined(HAVE_MMAP)
|
2011-10-01 01:58:10 +10:00
|
|
|
/// Wraps a mechanism to map a file into memory. Read only.
|
|
|
|
class mapped_file {
|
2014-07-29 02:20:02 +10:00
|
|
|
private:
|
2011-10-01 01:58:10 +10:00
|
|
|
fd_ref m_fd;
|
|
|
|
uint8_t *m_data;
|
|
|
|
size_t m_size;
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2014-07-29 02:20:02 +10:00
|
|
|
void load_fd (access_t);
|
|
|
|
|
|
|
|
protected:
|
|
|
|
int access_to_flags (access_t);
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2011-10-01 01:58:10 +10:00
|
|
|
public:
|
2014-07-29 02:20:02 +10:00
|
|
|
mapped_file (const boost::filesystem::path &path, access_t access = ACCESS_READWRITE);
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2014-07-07 15:19:01 +10:00
|
|
|
mapped_file (const mapped_file&) = delete;
|
|
|
|
mapped_file& operator= (const mapped_file&) = delete;
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2011-10-01 01:58:10 +10:00
|
|
|
~mapped_file ();
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2011-10-01 01:58:10 +10:00
|
|
|
const uint8_t* data (void) const;
|
2014-07-07 15:19:01 +10:00
|
|
|
uint8_t* data (void);
|
2011-10-01 01:58:10 +10:00
|
|
|
size_t size (void) const;
|
2014-07-15 19:48:59 +10:00
|
|
|
|
|
|
|
uint8_t* begin (void);
|
|
|
|
uint8_t* end (void);
|
|
|
|
|
|
|
|
const uint8_t* begin (void) const;
|
|
|
|
const uint8_t* end (void) const;
|
2011-10-01 01:58:10 +10:00
|
|
|
};
|
2011-07-04 16:26:08 +10:00
|
|
|
#endif
|
2012-04-26 18:19:12 +10:00
|
|
|
|
2012-11-09 15:16:07 +11:00
|
|
|
|
2012-04-26 18:19:12 +10:00
|
|
|
class path_error : public std::runtime_error {
|
|
|
|
public:
|
|
|
|
path_error (const boost::filesystem::path &path):
|
|
|
|
runtime_error ("Invalid path " + path.string ())
|
|
|
|
{ ; }
|
|
|
|
};
|
2011-10-01 01:58:10 +10:00
|
|
|
}
|
2011-05-23 17:18:52 +10:00
|
|
|
|
2014-06-29 22:49:26 +10:00
|
|
|
#include "io.ipp"
|
|
|
|
|
2011-05-23 17:18:52 +10:00
|
|
|
#endif
|