V-Gears/V-Gears-Installer/include/decompiler/Disassembler.h

/*
 * Copyright (C) 2022 The V-Gears Team
 *
 * This file is part of V-Gears
 *
 * V-Gears is free software: you can redistribute it and/or modify it under
 * terms of the GNU General Public License as published by the Free Software
 * Foundation, version 3.0 (GPLv3) of the License.
 *
 * V-Gears is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 */

#pragma once

#include <iostream>
#include <vector>
#include "common/BinaryReader.h"
#include "instruction.h"
#include "unknown_opcode_exception.h"
#include "objectFactory.h"

/**
 * Base class for disassemblers.
 */
class Disassembler{

    public:

        /**
         * Constructor for Disassembler.
         *
         * @param insts[in] Where disassembled instructions will be placed.
         */
        Disassembler(InstVec &insts);

        /**
         * Destructor.
         */
        virtual ~Disassembler() = default;

        /**
         * Open a file for disassembly.
         *
         * @param filename[in] The file to Disassemble.
         */
        virtual void Open(const char *filename);

        /**
         * Request disassembled instructions.
         */
        void Disassemble();

        /**
         * Outputs the disassembled code.
         *
         * Disassembles code if this has not already been done.
         *
         * @param output[out] The output stream.
         */
        void DumpDisassembly(std::ostream &output);

    protected:

        /**
         * Performs disassembly.
         *
         * @throws UnknownOpcodeException on unknown opcode.
         * @throws std::exception on other failures.
         */
        virtual void DoDisassemble() = 0;

        /**
         * Outputs the disassembled code.
         *
         * @param output[out] The output stream.
         */
        virtual void DoDumpDisassembly(std::ostream &output);

        /**
         * Used to perform file I/O.
         */
        std::unique_ptr<BinaryReader> stream_;

        /**
         * List of disassembled instructions.
         */
        InstVec &insts_;

        /**
         * Base address where the script starts.
         */
        uint32 address_base_;
};