コード例 #1
0
ファイル: writer.py プロジェクト: doublereedkurt/PeachPy
    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:
            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.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)
コード例 #2
0
ファイル: writer.py プロジェクト: moneytech/PeachPy
    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)
コード例 #3
0
 def __init__(self, abi, source=None):
     from peachpy.formats.elf.section import null_section, StringSection, SymbolSection, SectionIndex
     from peachpy.formats.elf.symbol import Symbol, SymbolBinding, SymbolType
     self.abi = abi
     self.shstrtab = StringSection(".shstrtab")
     self.strtab = StringSection(".strtab")
     self.symtab = SymbolSection(string_table=self.strtab)
     self.sections = [null_section, self.shstrtab, self.strtab, self.symtab]
     self._section_names = set(
         [self.shstrtab.name, self.strtab.name, self.symtab.name])
     if source:
         source_symbol = Symbol()
         source_symbol.value = 0
         source_symbol.binding = SymbolBinding.local
         source_symbol.type = SymbolType.file
         source_symbol.name = source
         source_symbol.size = 0
         source_symbol.section = SectionIndex.absolute
         self.symtab.add(source_symbol)
コード例 #4
0
ファイル: image.py プロジェクト: Rinoahu/PeachPy
    def __init__(self, abi, source=None):
        from peachpy.formats.elf.section import null_section, StringSection, SymbolSection, SectionIndex
        from peachpy.formats.elf.symbol import Symbol, SymbolBinding, SymbolType

        self.abi = abi
        self.shstrtab = StringSection(".shstrtab")
        self.strtab = StringSection(".strtab")
        self.symtab = SymbolSection(string_table=self.strtab)
        self.sections = [null_section, self.shstrtab, self.strtab, self.symtab]
        self._section_names = set([self.shstrtab.name, self.strtab.name, self.symtab.name])
        if source:
            source_symbol = Symbol()
            source_symbol.value = 0
            source_symbol.binding = SymbolBinding.local
            source_symbol.type = SymbolType.file
            source_symbol.name = source
            source_symbol.size = 0
            source_symbol.section = SectionIndex.absolute
            self.symtab.add(source_symbol)
コード例 #5
0
    def add_function(self, function):
        import peachpy.x86_64.function
        from peachpy.util import roundup
        assert isinstance(function, peachpy.x86_64.function.ABIFunction), \
            "Function must be finalized with an ABI before its assembly can be used"

        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)

        const_offset = 0
        if encoded_function.const_section.content:
            if self.rodata_section is None:
                from peachpy.formats.elf.section import ReadOnlyDataSection
                self.rodata_section = ReadOnlyDataSection()
                self.image.add_section(self.rodata_section)
            const_offset = self.rodata_section.get_content_size(self.abi)
            const_padding = bytearray([encoded_function.const_section.alignment_byte] *
                                      (roundup(const_offset, encoded_function.const_section.alignment) - const_offset))
            self.rodata_section.content += const_padding
            const_offset += len(const_padding)
            self.rodata_section.content += encoded_function.const_section.content
            self.rodata_section.alignment = max(self.rodata_section.alignment, encoded_function.const_section.alignment)

        # Map from symbol name to symbol index
        from peachpy.formats.elf.symbol import Symbol, SymbolBinding, SymbolType
        symbol_map = dict()
        for symbol in encoded_function.const_section.symbols:
            const_symbol = Symbol()
            const_symbol.name = function.mangled_name + "." + symbol.name
            const_symbol.value = const_offset + symbol.offset
            const_symbol.size = symbol.size
            const_symbol.section = self.rodata_section
            const_symbol.binding = SymbolBinding.local
            const_symbol.type = SymbolType.data_object
            self.image.symtab.add(const_symbol)
            symbol_map[symbol] = const_symbol

        if encoded_function.code_section.relocations:
            if self.text_rela_section is None:
                from peachpy.formats.elf.section import RelocationsWithAddendSection
                self.text_rela_section = RelocationsWithAddendSection(self.text_section, self.image.symtab)
                self.image.add_section(self.text_rela_section)

            from peachpy.formats.elf.symbol import RelocationWithAddend, RelocationType
            for relocation in encoded_function.code_section.relocations:
                elf_relocation = RelocationWithAddend(RelocationType.x86_64_pc32,
                                                      code_offset + relocation.offset,
                                                      symbol_map[relocation.symbol],
                                                      relocation.offset - relocation.program_counter)
                self.text_rela_section.add(elf_relocation)

        function_symbol = Symbol()
        function_symbol.name = function.mangled_name
        function_symbol.value = code_offset
        function_symbol.content_size = len(encoded_function.code_section)
        function_symbol.section = self.text_section
        function_symbol.binding = SymbolBinding.global_
        function_symbol.type = SymbolType.function
        self.image.symtab.add(function_symbol)
コード例 #6
0
ファイル: writer.py プロジェクト: hotelzululima/PeachPy
    def add_function(self, function):
        import peachpy.x86_64.function
        from peachpy.util import roundup
        assert isinstance(function, peachpy.x86_64.function.ABIFunction), \
            "Function must be bindinded to an ABI before its assembly can be used"

        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)

        const_offset = 0
        if encoded_function.const_section.content:
            if self.rodata_section is None:
                from peachpy.formats.elf.section import ReadOnlyDataSection
                self.rodata_section = ReadOnlyDataSection()
                self.image.add_section(self.rodata_section)
            const_offset = self.rodata_section.get_content_size(self.abi)
            const_padding = bytearray([encoded_function.const_section.alignment_byte] *
                                      (roundup(const_offset, encoded_function.const_section.alignment) - const_offset))
            self.rodata_section.content += const_padding
            const_offset += len(const_padding)
            self.rodata_section.content += encoded_function.const_section.content
            self.rodata_section.alignment = max(self.rodata_section.alignment, encoded_function.const_section.alignment)

        # Map from symbol name to symbol index
        from peachpy.formats.elf.symbol import Symbol, SymbolBinding, SymbolType
        symbol_map = dict()
        for symbol in encoded_function.const_section.symbols:
            const_symbol = Symbol()
            const_symbol.name = symbol.name
            const_symbol.value = const_offset + symbol.offset
            const_symbol.size = symbol.size
            const_symbol.section = self.rodata_section
            const_symbol.binding = SymbolBinding.local
            const_symbol.type = SymbolType.data_object
            self.image.symtab.add(const_symbol)
            symbol_map[symbol] = const_symbol

        if encoded_function.code_section.relocations:
            if self.text_rela_section is None:
                from peachpy.formats.elf.section import RelocationsWithAddendSection
                self.text_rela_section = RelocationsWithAddendSection(self.text_section, self.image.symtab)
                self.image.add_section(self.text_rela_section)

            from peachpy.formats.elf.symbol import RelocationWithAddend, RelocationType
            for relocation in encoded_function.code_section.relocations:
                elf_relocation = RelocationWithAddend(RelocationType.x86_64_pc32,
                                                      relocation.offset,
                                                      symbol_map[relocation.symbol],
                                                      -4)
                self.text_rela_section.add(elf_relocation)

        function_symbol = Symbol()
        function_symbol.name = function.name
        function_symbol.value = code_offset
        function_symbol.content_size = len(encoded_function.code_section)
        function_symbol.section = self.text_section
        function_symbol.binding = SymbolBinding.global_
        function_symbol.type = SymbolType.function
        self.image.symtab.add(function_symbol)