recursive_traversal_disassembler.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 recursive_traversal_disassembler.hpp
// @brief Class for managing the Recursive Traversal Disassembly algorithm.
 
#ifndef KUNAI_DEX_DVM_RECURSIVE_TRAVERSAL_DISASSEMBLER_HPP
#define KUNAI_DEX_DVM_RECURSIVE_TRAVERSAL_DISASSEMBLER_HPP
 
#include "Kunai/DEX/DVM/disassembler.hpp"
#include <queue>
 
namespace KUNAI
{
namespace DEX
{
    class RecursiveTraversalDisassembler
    {
        Disassembler *disassembler;
 
        std::queue<std::uint64_t> Q;
 
        void analyze_switch(
            std::vector<std::unique_ptr<Instruction>> &instructions,
            std::unordered_map<std::uint64_t, Instruction *> &cache_instrs,
            std::vector<std::uint8_t> &buffer_bytes);
 
    public:
        RecursiveTraversalDisassembler() = default;
 
        void set_internal_disassembler(Disassembler *disassembler)
        {
            this->disassembler = disassembler;
        }
 
        void disassembly(std::vector<std::uint8_t> &buffer_bytes,
                            EncodedMethod *method,
                            std::vector<std::unique_ptr<Instruction>> &instructions);
    };
} // namespace DEX
} // namespace KUNAI
 
#endif