map_item.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 map_item.hpp
// The map contains all the different data from the DEX file contained in
// the data section. Androguard retrieves all the information from here, but
// in some cases, contains less data than the header.
#ifndef KUNAI_DEX_PARSER_MAP_ITEM_HPP
#define KUNAI_DEX_PARSER_MAP_ITEM_HPP
 
#include "Kunai/Utils/kunaistream.hpp"
#include <unordered_map>
 
namespace KUNAI
{
namespace DEX
{
    class MapList
    {
    public:
        enum type_codes : std::uint16_t
        {
            TYPE_HEADER_ITEM = 0x0000,
            TYPE_STRING_ID_ITEM = 0x0001,
            TYPE_TYPE_ID_ITEM = 0x0002,
            TYPE_PROTO_ID_ITEM = 0x0003,
            TYPE_FIELD_ID_ITEM = 0x0004,
            TYPE_METHOD_ID_ITEM = 0x0005,
            TYPE_CLASS_DEF_ITEM = 0x0006,
            TYPE_CALL_SITE_ID_ITEM = 0x0007,
            TYPE_METHOD_HANDLE_ITEM = 0x0008,
            TYPE_MAP_LIST = 0x1000,
            TYPE_TYPE_LIST = 0x1001,
            TYPE_ANNOTATION_SET_REF_LIST = 0x1002,
            TYPE_ANNOTATION_SET_ITEM = 0x1003,
            TYPE_CLASS_DATA_ITEM = 0x2000,
            TYPE_CODE_ITEM = 0x2001,
            TYPE_STRING_DATA_ITEM = 0x2002,
            TYPE_DEBUG_INFO_ITEM = 0x2003,
            TYPE_ANNOTATION_ITEM = 0x2004,
            TYPE_ENCODED_ARRAY_ITEM = 0x2005,
            TYPE_ANNOTATIONS_DIRECTORY_ITEM = 0x2006,
            TYPE_HIDDENAPI_CLASS_DATA_ITEM = 0xF000
        };
 
        struct map_item
        {
            type_codes type;            
            std::uint16_t unused;       
            std::uint32_t size;         
            std::uint32_t offset;       
        };
        
    private:
        std::unordered_map<type_codes, map_item> items;
    public:
        MapList() = default;
        ~MapList() = default;
 
        void parse_map_list(stream::KunaiStream* stream, std::uint32_t map_off);
 
        const std::unordered_map<type_codes, map_item>& get_map_items() const
        {
            return items;
        }
 
        std::unordered_map<type_codes, map_item>& get_map_items()
        {
            return items;
        }
    };
} // namespace DEX
} // namespace KUNAI
 
 
#endif