class MSCOFFWriter(ImageWriter): def __init__(self, output_path, abi, input_path=None): super(MSCOFFWriter, self).__init__(output_path) from peachpy.formats.mscoff import Image, TextSection, ReadOnlyDataSection self.output_path = output_path self.abi = abi self.image = Image(abi, input_path) self.text_section = TextSection() self.image.add_section(self.text_section) self.rdata_section = ReadOnlyDataSection() self.image.add_section(self.rdata_section) def encode(self): return self.image.encode() def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be finalized with an ABI before its assembly can be used" from peachpy.util import roundup from peachpy.formats.mscoff import Symbol, SymbolType, StorageClass, Relocation, RelocationType encoded_function = function.encode() code_offset = len(self.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) self.text_section.content += code_padding code_offset += len(code_padding) self.text_section.content += encoded_function.code_section.content self.text_section.alignment = \ max(self.text_section.alignment, encoded_function.code_section.alignment) rdata_offset = self.rdata_section.content_size rdata_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(rdata_offset, encoded_function.const_section.alignment) - rdata_offset)) self.rdata_section.content += rdata_padding rdata_offset += len(rdata_padding) self.rdata_section.content += encoded_function.const_section.content self.rdata_section.alignment = \ max(self.rdata_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: mscoff_symbol = Symbol() mscoff_symbol.name = symbol.name mscoff_symbol.value = rdata_offset + symbol.offset mscoff_symbol.section = self.rdata_section mscoff_symbol.symbol_type = SymbolType.non_function mscoff_symbol.storage_class = StorageClass.static self.image.add_symbol(mscoff_symbol) symbol_map[symbol] = mscoff_symbol for relocation in encoded_function.code_section.relocations: relocation_type_map = { 4: RelocationType.x86_64_relocation_offset32, 5: RelocationType.x86_64_relocation_plus_1_offset32, 6: RelocationType.x86_64_relocation_plus_2_offset32, 7: RelocationType.x86_64_relocation_plus_3_offset32, 8: RelocationType.x86_64_relocation_plus_4_offset32, 9: RelocationType.x86_64_relocation_plus_5_offset32 } relocation_type = relocation_type_map[relocation.program_counter - relocation.offset] mscoff_relocation = Relocation(relocation_type, code_offset + relocation.offset, symbol_map[relocation.symbol]) self.text_section.relocations.append(mscoff_relocation) function_symbol = Symbol() function_symbol.name = function.mangled_name function_symbol.value = code_offset function_symbol.section = self.text_section function_symbol.symbol_type = SymbolType.function function_symbol.storage_class = StorageClass.external self.image.add_symbol(function_symbol)
class MachOWriter(ImageWriter): def __init__(self, output_path, abi): super(MachOWriter, self).__init__(output_path) from peachpy.formats.macho.image import Image self.abi = abi self.image = Image(abi) def encode(self): return self.image.encode() def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be finalized with an ABI before its assembly can be used" from peachpy.formats.macho.symbol import Symbol, SymbolType, SymbolDescription, SymbolVisibility, \ Relocation, RelocationType from peachpy.util import roundup encoded_function = function.encode() code_offset = len(self.image.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) self.image.text_section.content += code_padding code_offset += len(code_padding) self.image.text_section.content += encoded_function.code_section.content self.image.text_section.alignment = \ max(self.image.text_section.alignment, encoded_function.code_section.alignment) const_offset = self.image.const_section.content_size const_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(const_offset, encoded_function.const_section.alignment) - const_offset)) self.image.const_section.content += const_padding const_offset += len(const_padding) self.image.const_section.content += encoded_function.const_section.content self.image.const_section.alignment = \ max(self.image.const_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: macho_symbol = Symbol("_" + function.mangled_name + "." + symbol.name, SymbolType.section_relative, self.image.const_section, const_offset + symbol.offset) macho_symbol.description = SymbolDescription.defined self.image.symbol_table.add_symbol(macho_symbol) symbol_map[symbol] = macho_symbol for relocation in encoded_function.code_section.relocations: macho_relocation = Relocation(RelocationType.x86_64_signed, code_offset + relocation.offset, 4, symbol_map[relocation.symbol], is_pc_relative=True) relocation_addend = relocation.offset + 4 - relocation.program_counter if relocation_addend != 0: self.image.text_section.content[code_offset + relocation.offset] = relocation_addend & 0xFF self.image.text_section.content[code_offset + relocation.offset + 1] = (relocation_addend >> 8) & 0xFF self.image.text_section.content[code_offset + relocation.offset + 2] = (relocation_addend >> 16) & 0xFF self.image.text_section.content[code_offset + relocation.offset + 3] = (relocation_addend >> 24) & 0xFF self.image.text_section.relocations.append(macho_relocation) function_symbol = Symbol("_" + function.mangled_name, SymbolType.section_relative, self.image.text_section, value=code_offset) function_symbol.description = SymbolDescription.defined function_symbol.visibility = SymbolVisibility.external self.image.symbol_table.add_symbol(function_symbol)
class MSCOFFWriter: def __init__(self, output_path, abi, input_path=None): from peachpy.formats.mscoff import Image, TextSection, ReadOnlyDataSection self.output_path = output_path self.previous_writer = None self.abi = abi self.image = Image(abi, input_path) self.text_section = TextSection() self.image.add_section(self.text_section) self.rdata_section = ReadOnlyDataSection() self.image.add_section(self.rdata_section) def __enter__(self): global active_writer self.previous_writer = active_writer active_writer = self self.output_file = open(self.output_path, "w", buffering=0) return self def __exit__(self, exc_type, exc_value, traceback): global active_writer active_writer = self.previous_writer self.previous_writer = None if exc_type is None: self.output_file.write(self.image.encode()) self.output_file.close() self.output_file = None else: import os os.unlink(self.output_file.name) self.output_file = None raise def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be bindinded to an ABI before its assembly can be used" from peachpy.util import roundup from peachpy.formats.mscoff import Symbol, SymbolType, StorageClass, Relocation, RelocationType encoded_function = function.encode() code_offset = len(self.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) code_offset += len(code_padding) self.text_section.content += encoded_function.code_section.content self.text_section.alignment = \ max(self.text_section.alignment, encoded_function.code_section.alignment) rdata_offset = self.rdata_section.content_size rdata_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(rdata_offset, encoded_function.const_section.alignment) - rdata_offset)) self.rdata_section.content += rdata_padding rdata_offset += len(rdata_padding) self.rdata_section.content += encoded_function.const_section.content self.rdata_section.alignment = \ max(self.rdata_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: mscoff_symbol = Symbol() mscoff_symbol.name = symbol.name mscoff_symbol.value = rdata_offset + symbol.offset mscoff_symbol.section = self.rdata_section mscoff_symbol.symbol_type = SymbolType.non_function mscoff_symbol.storage_class = StorageClass.static self.image.add_symbol(mscoff_symbol) symbol_map[symbol] = mscoff_symbol for relocation in encoded_function.code_section.relocations: mscoff_relocation = Relocation(RelocationType.x86_64_relocation_offset32, code_offset + relocation.offset, symbol_map[relocation.symbol]) self.text_section.relocations.append(mscoff_relocation) function_symbol = Symbol() function_symbol.name = function.name function_symbol.value = code_offset function_symbol.section = self.text_section function_symbol.symbol_type = SymbolType.function function_symbol.storage_class = StorageClass.external self.image.add_symbol(function_symbol)
class MachOWriter: def __init__(self, output_path, abi): from peachpy.formats.macho.image import Image self.output_path = output_path self.previous_writer = None self.abi = abi self.image = Image(abi) def __enter__(self): global active_writer self.previous_writer = active_writer active_writer = self self.output_file = open(self.output_path, "w", buffering=0) return self def __exit__(self, exc_type, exc_value, traceback): global active_writer active_writer = self.previous_writer self.previous_writer = None if exc_type is None: self.output_file.write(self.image.encode()) self.output_file.close() self.output_file = None else: import os os.unlink(self.output_file.name) self.output_file = None raise def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be bindinded to an ABI before its assembly can be used" from peachpy.formats.macho.symbol import Symbol, SymbolType, SymbolDescription, SymbolVisibility, \ Relocation, RelocationType from peachpy.util import roundup encoded_function = function.encode() code_offset = len(self.image.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) self.image.text_section.content += code_padding code_offset += len(code_padding) self.image.text_section.content += encoded_function.code_section.content self.image.text_section.alignment = \ max(self.image.text_section.alignment, encoded_function.code_section.alignment) const_offset = self.image.const_section.content_size const_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(const_offset, encoded_function.const_section.alignment) - const_offset)) self.image.text_section.content += const_padding const_offset += len(const_padding) self.image.const_section.content += encoded_function.const_section.content self.image.const_section.alignment = \ max(self.image.const_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: macho_symbol = Symbol(symbol.name, SymbolType.section_relative, self.image.const_section, const_offset + symbol.offset) macho_symbol.description = SymbolDescription.defined self.image.symbol_table.add_symbol(macho_symbol) symbol_map[symbol] = macho_symbol for relocation in encoded_function.code_section.relocations: macho_relocation = Relocation(RelocationType.x86_64_signed, code_offset + relocation.offset, 4, symbol_map[relocation.symbol], is_pc_relative=True) self.image.text_section.relocations.append(macho_relocation) function_symbol = Symbol("_" + function.name, SymbolType.section_relative, self.image.text_section, value=code_offset) function_symbol.description = SymbolDescription.defined function_symbol.visibility = SymbolVisibility.external self.image.symbol_table.add_symbol(function_symbol)
class MSCOFFWriter(ImageWriter): def __init__(self, output_path, abi, input_path=None): super(MSCOFFWriter, self).__init__(output_path) from peachpy.formats.mscoff import Image, TextSection, ReadOnlyDataSection self.output_path = output_path self.abi = abi self.image = Image(abi, input_path) self.text_section = TextSection() self.image.add_section(self.text_section) self.rdata_section = ReadOnlyDataSection() self.image.add_section(self.rdata_section) def encode(self): return self.image.encode() def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be finalized with an ABI before its assembly can be used" from peachpy.util import roundup from peachpy.formats.mscoff import Symbol, SymbolType, StorageClass, Relocation, RelocationType encoded_function = function.encode() code_offset = len(self.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) code_offset += len(code_padding) self.text_section.content += encoded_function.code_section.content self.text_section.alignment = \ max(self.text_section.alignment, encoded_function.code_section.alignment) rdata_offset = self.rdata_section.content_size rdata_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(rdata_offset, encoded_function.const_section.alignment) - rdata_offset)) self.rdata_section.content += rdata_padding rdata_offset += len(rdata_padding) self.rdata_section.content += encoded_function.const_section.content self.rdata_section.alignment = \ max(self.rdata_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: mscoff_symbol = Symbol() mscoff_symbol.name = symbol.name mscoff_symbol.value = rdata_offset + symbol.offset mscoff_symbol.section = self.rdata_section mscoff_symbol.symbol_type = SymbolType.non_function mscoff_symbol.storage_class = StorageClass.static self.image.add_symbol(mscoff_symbol) symbol_map[symbol] = mscoff_symbol for relocation in encoded_function.code_section.relocations: relocation_type_map = { 4: RelocationType.x86_64_relocation_offset32, 5: RelocationType.x86_64_relocation_plus_1_offset32, 6: RelocationType.x86_64_relocation_plus_2_offset32, 7: RelocationType.x86_64_relocation_plus_3_offset32, 8: RelocationType.x86_64_relocation_plus_4_offset32, 9: RelocationType.x86_64_relocation_plus_5_offset32 } relocation_type = relocation_type_map[relocation.program_counter - relocation.offset] mscoff_relocation = Relocation(relocation_type, code_offset + relocation.offset, symbol_map[relocation.symbol]) self.text_section.relocations.append(mscoff_relocation) function_symbol = Symbol() function_symbol.name = function.mangled_name function_symbol.value = code_offset function_symbol.section = self.text_section function_symbol.symbol_type = SymbolType.function function_symbol.storage_class = StorageClass.external self.image.add_symbol(function_symbol)
class MachOWriter(ImageWriter): def __init__(self, output_path, abi): super(MachOWriter, self).__init__(output_path) from peachpy.formats.macho.image import Image self.abi = abi self.image = Image(abi) def encode(self): return self.image.encode() def add_function(self, function): import peachpy.x86_64.function assert isinstance(function, peachpy.x86_64.function.ABIFunction), \ "Function must be finalized with an ABI before its assembly can be used" from peachpy.formats.macho.symbol import Symbol, SymbolType, SymbolDescription, SymbolVisibility, \ Relocation, RelocationType from peachpy.util import roundup encoded_function = function.encode() code_offset = len(self.image.text_section.content) code_padding = bytearray([encoded_function.code_section.alignment_byte] * (roundup(code_offset, encoded_function.code_section.alignment) - code_offset)) self.image.text_section.content += code_padding code_offset += len(code_padding) self.image.text_section.content += encoded_function.code_section.content self.image.text_section.alignment = \ max(self.image.text_section.alignment, encoded_function.code_section.alignment) const_offset = self.image.const_section.content_size const_padding = bytearray([encoded_function.const_section.alignment_byte] * (roundup(const_offset, encoded_function.const_section.alignment) - const_offset)) self.image.const_section.content += const_padding const_offset += len(const_padding) self.image.const_section.content += encoded_function.const_section.content self.image.const_section.alignment = \ max(self.image.const_section.alignment, encoded_function.const_section.alignment) # Map from PeachPy symbol to Mach-O symbol symbol_map = dict() for symbol in encoded_function.const_section.symbols: macho_symbol = Symbol("_" + function.mangled_name + "." + symbol.name, SymbolType.section_relative, self.image.const_section, const_offset + symbol.offset) macho_symbol.description = SymbolDescription.defined self.image.symbol_table.add_symbol(macho_symbol) symbol_map[symbol] = macho_symbol for relocation in encoded_function.code_section.relocations: macho_relocation = Relocation(RelocationType.x86_64_signed, code_offset + relocation.offset, 4, symbol_map[relocation.symbol], is_pc_relative=True) relocation_addend = relocation.offset + 4 - relocation.program_counter if relocation_addend != 0: self.image.text_section.content[code_offset + relocation.offset] = relocation_addend & 0xFF self.image.text_section.content[code_offset + relocation.offset + 1] = (relocation_addend >> 8) & 0xFF self.image.text_section.content[code_offset + relocation.offset + 2] = (relocation_addend >> 16) & 0xFF self.image.text_section.content[code_offset + relocation.offset + 3] = (relocation_addend >> 24) & 0xFF self.image.text_section.relocations.append(macho_relocation) function_symbol = Symbol("_" + function.mangled_name, SymbolType.section_relative, self.image.text_section, value=code_offset) function_symbol.description = SymbolDescription.defined function_symbol.visibility = SymbolVisibility.external self.image.symbol_table.add_symbol(function_symbol)