ELFStreamerWrapper.h

#pragma once
 
#include "nyxstone.h"
 
#include <llvm/MC/MCAsmBackend.h>
#include <llvm/MC/MCCodeEmitter.h>
#include <llvm/MC/MCELFStreamer.h>
#include <llvm/MC/MCObjectWriter.h>
 
namespace nyxstone {
class ELFStreamerWrapper : public llvm::MCELFStreamer {
    // Sink to record instruction details
    std::vector<Nyxstone::Instruction>* instructions;
 
    // Reference to the nyxstone error string
    std::string& extended_error;
 
    // Instruction printer
    llvm::MCInstPrinter& instruction_printer;
 
public:
    ELFStreamerWrapper(llvm::MCContext& context, std::unique_ptr<llvm::MCAsmBackend>&& assembler_backend,
        std::unique_ptr<llvm::MCObjectWriter>&& object_writer, std::unique_ptr<llvm::MCCodeEmitter>&& code_emitter,
        std::vector<Nyxstone::Instruction>* instructions, std::string& extended_error,
        llvm::MCInstPrinter& instruction_printer)
        : llvm::MCELFStreamer(context, std::move(assembler_backend), std::move(object_writer), std::move(code_emitter))
        , instructions(instructions)
        , extended_error(extended_error)
        , instruction_printer(instruction_printer)
    {
    }
 
    static std::unique_ptr<llvm::MCStreamer> createELFStreamerWrapper(llvm::MCContext& context,
        std::unique_ptr<llvm::MCAsmBackend>&& assembler_backend, std::unique_ptr<llvm::MCObjectWriter>&& object_writer,
        std::unique_ptr<llvm::MCCodeEmitter>&& code_emitter, bool RelaxAll,
        std::vector<Nyxstone::Instruction>* instructions, std::string& extended_error,
        llvm::MCInstPrinter& instruction_printer);
 
    void emitInstruction(const llvm::MCInst& Inst, const llvm::MCSubtargetInfo& STI) override;
};
} // namespace nyxstone