Пример #1
0
    def as_bytearray(self):
        from peachpy.formats.elf.file import FileHeader

        file_header = FileHeader(self.abi)
        file_header.section_header_table_offset = file_header.size
        file_header.section_header_entries_count = len(self.sections)
        file_header.section_name_string_table_index = 1
        data = file_header.as_bytearray

        # Update section offsets
        data_offset = file_header.size + self.sections[0].header.size * len(self.sections)
        for section in self.sections:
            if section.header.address_alignment != 0:
                if data_offset % section.header.address_alignment != 0:
                    padding_length = section.header.address_alignment - data_offset % section.header.address_alignment
                    data_offset += padding_length
            section.header.offset = data_offset
            data_offset += section.header.content_size

        # Write section headers
        for section in self.sections:
            data += section.header.as_bytearray

        # Write section content
        for section in self.sections:
            if section.header.address_alignment != 0:
                if len(data) % section.header.address_alignment != 0:
                    padding_length = section.header.address_alignment - len(data) % section.header.address_alignment
                    padding_data = bytearray([0] * padding_length)
                    data += padding_data
            data += section.content

        return data
Пример #2
0
    def as_bytearray(self):
        import six
        from peachpy.formats.elf.file import FileHeader
        from peachpy.formats.elf.section import Section, StringSection, SymbolSection
        from peachpy.util import roundup

        file_header = FileHeader(self.abi)
        file_header.section_header_table_offset = file_header.size
        file_header.section_header_entries_count = len(self.sections)
        file_header.section_name_string_table_index = 1
        data = file_header.as_bytearray

        # Collect strings from sections
        for section in self.sections:
            self.shstrtab.add(section.name)
            if isinstance(section, StringSection):
                pass
            elif isinstance(section, SymbolSection):
                for symbol in six.iterkeys(section.symbol_index_map):
                    self.strtab.add(symbol.name)

        # Layout sections
        data_offset = file_header.size + Section.get_header_size(
            self.abi) * len(self.sections)
        section_offsets = []
        for section in self.sections:
            if section.alignment != 0:
                data_offset = roundup(data_offset, section.alignment)
            section_offsets.append(data_offset)
            data_offset += section.get_content_size(self.abi)

        from peachpy.encoder import Encoder
        encoder = Encoder(self.abi.endianness, self.abi.elf_bitness)

        section_index_map = {
            section: index
            for index, section in enumerate(self.sections)
        }
        # Write section headers
        for section, offset in zip(self.sections, section_offsets):
            data += section.encode_header(encoder,
                                          self.shstrtab._string_index_map,
                                          section_index_map, offset)

        # Write section content
        for section in self.sections:
            padding = bytearray(
                roundup(len(data), section.alignment) - len(data))
            data += padding
            data += section.encode_content(encoder,
                                           self.strtab._string_index_map,
                                           section_index_map,
                                           self.symtab.symbol_index_map)

        return data
Пример #3
0
    def as_bytearray(self):
        import six
        from peachpy.formats.elf.file import FileHeader
        from peachpy.formats.elf.section import Section, StringSection, SymbolSection
        from peachpy.util import roundup

        file_header = FileHeader(self.abi)
        file_header.section_header_table_offset = file_header.size
        file_header.section_header_entries_count = len(self.sections)
        file_header.section_name_string_table_index = 1
        data = file_header.as_bytearray

        # Collect strings from sections
        for section in self.sections:
            self.shstrtab.add(section.name)
            if isinstance(section, StringSection):
                pass
            elif isinstance(section, SymbolSection):
                for symbol in six.iterkeys(section.symbol_index_map):
                    self.strtab.add(symbol.name)

        # Layout sections
        data_offset = file_header.size + Section.get_header_size(self.abi) * len(self.sections)
        section_offsets = []
        for section in self.sections:
            if section.alignment != 0:
                data_offset = roundup(data_offset, section.alignment)
            section_offsets.append(data_offset)
            data_offset += section.get_content_size(self.abi)

        from peachpy.encoder import Encoder

        encoder = Encoder(self.abi.endianness, self.abi.elf_bitness)

        section_index_map = {section: index for index, section in enumerate(self.sections)}
        # Write section headers
        for section, offset in zip(self.sections, section_offsets):
            data += section.encode_header(encoder, self.shstrtab._string_index_map, section_index_map, offset)

        # Write section content
        for section in self.sections:
            padding = bytearray(roundup(len(data), section.alignment) - len(data))
            data += padding
            data += section.encode_content(
                encoder, self.strtab._string_index_map, section_index_map, self.symtab.symbol_index_map
            )

        return data