kunaistream.hpp

//--------------------------------------------------------------------*- C++ -*-
// Kunai-static-analyzer: library for doing analysis of dalvik files
// @author Farenain <kunai.static.analysis@gmail.com>
//
// @file kunaistream.hpp
// @brief Manage the possible streams for reading and writing, as well as
// possibly reading data types that need special handling.
#ifndef KUNAI_UTILS_KUNAISTREAM_HPP
#define KUNAI_UTILS_KUNAISTREAM_HPP
 
#include <fstream>
#include "Kunai/Exceptions/stream_exception.hpp"
 
namespace KUNAI
{
    namespace stream
    {
        class KunaiStream
        {
            std::ifstream& input_file;
            
            std::size_t file_size;
 
            void initialize();
 
        public:
            const std::int32_t MAX_ANSII_STR_SIZE = 256;
 
            KunaiStream(std::ifstream& input_file) : input_file(input_file)
            {
                if (!input_file.is_open())
                    throw exceptions::StreamException("KunaiStream: error input_file not open");
                initialize();
            }
 
            ~KunaiStream() = default;
 
            std::size_t get_size() const
            {
                return file_size;
            }
 
            template <typename T>
            void read_data(T& buffer, std::int32_t read_size)
            {
                if (read_size < 0)
                    throw exceptions::StreamException("read_data(): read_size given incorrect");
                // read the data
                input_file.read(reinterpret_cast<char*>(&buffer), read_size);
 
                if(!input_file)
                    throw exceptions::StreamException("read_data(): error reading input file");
            }
 
            std::streampos tellg() const
            {
                return input_file.tellg();
            }
 
            void seekg(std::streamoff off, std::ios_base::seekdir dir)
            {
                if (off >= file_size)
                    throw exceptions::StreamException("seekg(): offset provided is out of bound");
                input_file.seekg(off, dir);
            }
 
            std::string read_ansii_string(std::int64_t offset);
 
            std::string read_dex_string(std::int64_t offset);
 
            std::uint64_t read_uleb128();
 
            std::int64_t read_sleb128();
        };
    }   
} // namespace KUNAI
 
 
#endif // KUNAI_UTILS_KUNAISTREAM_HPP