ObjectWriterWrapper.h

#pragma once
 
#include "nyxstone.h"
 
#include <llvm/Config/llvm-config.h>
 
#include <llvm/MC/MCAsmBackend.h>
#include <llvm/MC/MCAsmLayout.h>
#include <llvm/MC/MCAssembler.h>
#include <llvm/MC/MCContext.h>
#include <llvm/MC/MCExpr.h>
#include <llvm/MC/MCFixup.h>
#include <llvm/MC/MCFixupKindInfo.h>
#include <llvm/MC/MCFragment.h>
#include <llvm/MC/MCObjectWriter.h>
#include <llvm/MC/MCSection.h>
#include <llvm/MC/MCSymbol.h>
#include <llvm/MC/MCValue.h>
 
namespace nyxstone {
class ObjectWriterWrapper : public llvm::MCObjectWriter {
    // Wrapped MCObjectWriter, f. i., ELFSingleObjectWriter
    std::unique_ptr<llvm::MCObjectWriter> inner_object_writer;
 
    llvm::raw_pwrite_stream& stream;
 
    llvm::MCContext& context;
 
    bool write_text_section_only;
 
    u64 start_address;
 
    std::string& extended_error;
 
    std::vector<Nyxstone::Instruction>* instructions;
 
    void validate_fixups(const llvm::MCFragment& fragment, const llvm::MCAsmLayout& Layout);
 
public:
    ObjectWriterWrapper(std::unique_ptr<llvm::MCObjectWriter>&& object_writer, llvm::raw_pwrite_stream& stream,
        llvm::MCContext& context, bool write_text_section_only, u64 start_address, std::string& extended_error,
        std::vector<Nyxstone::Instruction>* instructions)
        : inner_object_writer(std::move(object_writer))
        , stream(stream)
        , context(context)
        , write_text_section_only(write_text_section_only)
        , start_address(start_address)
        , extended_error(extended_error)
        , instructions(instructions)
    {
    }
 
    static std::unique_ptr<llvm::MCObjectWriter> createObjectWriterWrapper(
        std::unique_ptr<llvm::MCObjectWriter>&& object_writer, llvm::raw_pwrite_stream& stream,
        llvm::MCContext& context, bool write_text_section_only, u64 start_address, std::string& extended_error,
        std::vector<Nyxstone::Instruction>* instructions);
 
    void executePostLayoutBinding(llvm::MCAssembler& Asm, const llvm::MCAsmLayout& Layout) override;
 
    bool resolve_relocation(llvm::MCAssembler& assembler, const llvm::MCAsmLayout& layout,
        const llvm::MCFragment* fragment, const llvm::MCFixup& fixup, llvm::MCValue target, uint64_t& fixed_value);
 
    void recordRelocation(llvm::MCAssembler& Asm, const llvm::MCAsmLayout& Layout, const llvm::MCFragment* Fragment,
        const llvm::MCFixup& Fixup, llvm::MCValue Target, uint64_t& FixedValue) override;
 
    uint64_t writeObject(llvm::MCAssembler& Asm, const llvm::MCAsmLayout& Layout) override;
};
} // namespace nyxstone