48 lines
1.1 KiB
Ruby
48 lines
1.1 KiB
Ruby
|
require_relative 'elfobject'
|
||
|
|
|
||
|
|
module Asm
|
||
|
|
|
||
|
|
class ObjectWriter
|
||
|
|
def initialize(target)
|
||
|
|
@object = ELF::ObjectFile.new(target)
|
||
|
|
|
||
|
|
sym_strtab = ELF::StringTableSection.new(".strtab")
|
||
|
|
@object.add_section sym_strtab
|
||
|
|
@symbol_table = ELF::SymbolTableSection.new(".symtab", sym_strtab)
|
||
|
|
@object.add_section @symbol_table
|
||
|
|
|
||
|
|
@text = ELF::TextSection.new(".text")
|
||
|
|
@object.add_section @text
|
||
|
|
|
||
|
|
@reloc_table = ELF::RelocationTableSection.new(".text.rel", @symbol_table, @text)
|
||
|
|
@object.add_section @reloc_table
|
||
|
|
end
|
||
|
|
|
||
|
|
def set_text(text)
|
||
|
|
@text.text = text
|
||
|
|
add_symbol "_start", 0
|
||
|
|
end
|
||
|
|
|
||
|
|
def add_symbol(name, offset, linkage = Elf::Constants::STB_GLOBAL)
|
||
|
|
@symbol_table.add_func_symbol name, offset, @text, linkage
|
||
|
|
end
|
||
|
|
|
||
|
|
def add_reloc_symbol(name)
|
||
|
|
@symbol_table.add_func_symbol name, 0, nil, Elf::Constants::STB_GLOBAL
|
||
|
|
end
|
||
|
|
|
||
|
|
def add_reloc(offset, label, type)
|
||
|
|
@reloc_table.add_reloc offset, label, type
|
||
|
|
end
|
||
|
|
|
||
|
|
def save(filename)
|
||
|
|
File.open(filename, 'wb') { |fp|
|
||
|
|
write fp
|
||
|
|
}
|
||
|
|
end
|
||
|
|
|
||
|
|
def write(io)
|
||
|
|
@object.write io
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|