header.hpp

//--------------------------------------------------------------------*- C++ -*-
// Kunai-static-analyzer: library for doing analysis of dalvik files
// @author Farenain <kunai.static.analysis@gmail.com>
// @author Ernesto Java <javaernesto@gmail.com>
//
// @file header.hpp
// @brief File to manage the structure of the DEX header
// this will be useful for the other classes that will use
// the values from this header
#ifndef KUNAI_DEX_PARSER_HEADER_HPP
#define KUNAI_DEX_PARSER_HEADER_HPP
 
 
#include "Kunai/Utils/logger.hpp"
#include "Kunai/Utils/kunaistream.hpp"
#include "Kunai/Exceptions/parser_exception.hpp"
#include <iostream>
#include <iomanip>
 
namespace KUNAI
{
    namespace DEX
    {
        class Header
        {
        public:
#pragma pack(1)
            struct dexheader_t
            {
                std::uint8_t magic[8];          
                std::int32_t checksum;          
                std::uint8_t signature[20];     
                std::uint32_t file_size;        
                std::uint32_t header_size;      
                std::uint32_t endian_tag;       
                std::uint32_t link_size;        
                std::uint32_t link_off;         
                std::uint32_t map_off;          
                std::uint32_t string_ids_size;  
                std::uint32_t string_ids_off;   
                std::uint32_t type_ids_size;    
                std::uint32_t type_ids_off;     
                std::uint32_t proto_ids_size;   
                std::uint32_t proto_ids_off;    
                std::uint32_t field_ids_size;   
                std::uint32_t field_ids_off;    
                std::uint32_t method_ids_size;  
                std::uint32_t method_ids_off;   
                std::uint32_t class_defs_size;  
                std::uint32_t class_defs_off;   
                std::uint32_t data_size;        
                std::uint32_t data_off;         
            };
#pragma pack()
 
        private:
            struct dexheader_t dexheader;
 
        public:
            Header() = default;
 
            void parse_headers(stream::KunaiStream* stream);
 
            Header(Header& header)
            {
                memcpy(&dexheader, &header.dexheader, sizeof(dexheader_t));
            }
 
            ~Header() = default;
 
            const dexheader_t &get_dex_header_const() const
            {
                return dexheader;
            }
 
            dexheader_t &get_dex_header()
            {
                return dexheader;
            }
 
            std::uint64_t get_dex_header_size() const
            {
                return sizeof(dexheader_t);
            }
 
            friend std::ostream &operator<<(std::ostream &os, const Header &entry);
 
            void to_xml(std::ofstream &fos);
        };
    } // namespace DEX
} // namespace KUNAI
 
#endif