def __add_sections()

in nightMARE/src/nightmare/malware/icedid/custom_pe.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

    def __add_sections(self, pe: lief.PE.Binary) -> lief.PE.Binary:
        for i, custom_section in enumerate(self.structure.sections):
            section = lief.PE.Section(f".mare{i}")
            section.virtual_address = custom_section.virtual_address
            section.virtual_size = custom_section.virtual_size
            section.characteristics = protection_to_lief_section_characteristics(
                custom_section.protection
            )

            section.content = list(
                self.__content[
                    custom_section.raw_offset : custom_section.raw_offset
                    + custom_section.raw_size
                ]
            )

            pe.add_section(section, lief.PE.SECTION_TYPES.DATA)
        return pe