linear_sweep_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 linear_sweep_disassembler.hpp
// @brief Class for managing the Linear Sweep Disassembly algorithm.
 
#ifndef KUNAI_DEX_DVM_LINEAR_SWEEP_DISASSEMBLER_HPP
#define KUNAI_DEX_DVM_LINEAR_SWEEP_DISASSEMBLER_HPP
 
#include "Kunai/DEX/DVM/disassembler.hpp"
 
namespace KUNAI
{
namespace DEX
{
    class LinearSweepDisassembler
    {
        Disassembler *disassembler;
 
        void assign_switch_if_any(
            std::vector<std::unique_ptr<Instruction>> &instructions,
            std::unordered_map<std::uint64_t, Instruction *> &cache_instructions);
 
    public:
        LinearSweepDisassembler() = default;
 
        void set_internal_disassembler(Disassembler *disassembler)
        {
            this->disassembler = disassembler;
        }
 
        void disassembly(std::vector<std::uint8_t> &buffer_bytes,
                            std::vector<std::unique_ptr<Instruction>> &instructions);
    };
} // DEX
} // KUNAI
 
#endif