strings.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 strings.hpp
// @brief Class for managing the list of strings, these strings
// follow an specific format where a `string_data_off` is used
// to point to an `string_data_item`
 
#ifndef KUNAI_DEX_PARSER_STRINGS_HPP
#define KUNAI_DEX_PARSER_STRINGS_HPP
 
#include <iostream>
#include <vector>
#include <algorithm>
#include <unordered_map>
 
#include "Kunai/Utils/kunaistream.hpp"
 
namespace KUNAI
{
namespace DEX
{
    using ordered_strings_t = std::vector<std::string>;
 
    using offset_string_t = std::unordered_map<std::uint32_t, std::string*>;
 
    class Strings
    {
        std::uint32_t strings_offset;
        std::uint32_t number_of_strings;
        ordered_strings_t ordered_strings;
        offset_string_t offset_strings;
 
    public:
        Strings() = default;
 
        Strings(Strings &str);
 
        ~Strings() = default;
 
        void parse_strings(std::uint32_t strings_offset, 
            std::uint32_t number_of_strings, 
            stream::KunaiStream* stream);
 
        const ordered_strings_t& get_ordered_strings() const
        {
            return ordered_strings;
        }
 
        const offset_string_t& get_offset_strings() const
        {
            return offset_strings;
        }
 
        std::string* get_string_from_offset(std::uint32_t offset);
 
        std::string& get_string_by_id(std::uint32_t id);
        
        std::uint32_t get_number_of_strings() const
        {
            return number_of_strings;
        }
 
        friend std::ostream& operator<<(std::ostream& os, const Strings& entry);
 
        void to_xml(std::ofstream& fos);
    };
}
}
 
#endif // KUNAI_DEX_PARSER_DEXSTRINGS_HPP