dTddlmZddlmZddlmZmZddlmZddl m Z m Z m Z m Z mZmZmZmZmZddlmZGd d eZGd d eZGd deeZGddeZGddeZy)) namedtuple)ELFRelocationError) elf_assert struct_parse)Section) ENUM_RELOC_TYPE_i386ENUM_RELOC_TYPE_x64ENUM_RELOC_TYPE_MIPSENUM_RELOC_TYPE_ARMENUM_RELOC_TYPE_AARCH64ENUM_RELOC_TYPE_PPC64ENUM_RELOC_TYPE_BPFENUM_RELOC_TYPE_LOONGARCH ENUM_D_TAG) Containerc.eZdZdZdZdZdZdZdZy) Relocationz Relocation object - representing a single relocation entry. Allows dictionary-like access to the entry's fields. Can be either a REL or RELA relocation. c ||_||_yN)entryelffile)selfrrs 9/usr/lib/python3/dist-packages/elftools/elf/relocation.py__init__zRelocation.__init__s  cd|jvS)z6 Is this a RELA relocation? If not, it's REL. r_addendrrs ris_RELAzRelocation.is_RELAsTZZ''rc |j|S)z% Dict-like access to entries r )rnames r __getitem__zRelocation.__getitem__$szz$rcLd|jrdndd|jdS)Nz )r"rr!s r__repr__zRelocation.__repr__)s#,,.e3  rc"|jSr)r*r!s r__str__zRelocation.__str__.s}}rN) __name__ __module__ __qualname____doc__rr"r%r*r,rrrrs  (   rrc.eZdZdZdZdZdZdZdZy)RelocationTablezL Shared functionality between relocation sections and relocation tables c0|j|_||_|j|_||_||_||_|r|jj|_ n|jj|_ |jj|_ yr) stream_stream_elffilestructs _elfstructs_size_offset_is_relaElf_Rela entry_structElf_Relsizeof entry_size)rroffsetsizeis_relas rrzRelocationTable.__init__6sv~~  "??    $ 0 0 9 9D  $ 0 0 8 8D ++224rc|jS)z> Is this a RELA relocation section? If not, it's REL. )r<r!s rr"zRelocationTable.is_RELAEs}}rc4|j|jzSz. Number of relocations in the section )r:rAr!s rnum_relocationszRelocationTable.num_relocationsJszzT__,,rc|j||jzz}t|j|j|}t ||j S)M Get the relocation at index #n from the section (Relocation object)  stream_pos)r;rArr>r6rr7)rn entry_offsetrs rget_relocationzRelocationTable.get_relocationOsK||a$//&99     LL#%%//rc#nKt|jD]}|j|yw)2 Yield all the relocations in the section N)rangerHrO)ris riter_relocationsz RelocationTable.iter_relocationsYs6t++-. )A%%a( ( )s35N) r-r.r/r0rr"rHrOrTr1rrr3r32s  5 - 0)rr3ceZdZdZdZy)RelocationSectionzK ELF relocation section. Serves as a collection of Relocation entries. ctj||||tj||j|d|d|ddk(t |ddvdt |d|j k(d|dd |j y) N sh_offsetsh_sizesh_typeSHT_RELA)SHT_RELr[zUnknown relocation type section sh_entsizezExpected sh_entsize of z section to be )r rr3rrrArheaderr$rs rrzRelocationSection.__init__csvtW5  t||  tIy0AZ0O Q 6)$(?? - /6,'4??:y!4?? 4 5rN)r-r.r/r0rr1rrrVrV`s  5rrVc(eZdZdZdZdZdZdZy)RelrRelocationSectiona RELR compressed relocation section. This stores relative relocations in a compressed format. An entry with an even value serves as an 'anchor' that defines a base address. Following this entry are one or more bitmaps for consecutive addresses after the anchor which determine if the corresponding relocation exists (if the bit is 1) or if it is skipped. Addends are stored at the respective addresses (as in REL relocations). ctj|||||d|_|d|_|jj j |_|jj|_ d|_ y)NrXrY) r rr;r:rr8Elf_Relr _relr_structr@ _entrysize_cached_relocationsr^s rrzRelrRelocationSection.__init__wscvtW5K( )_  LL0099++224#' rc#K|j|jz}|j}d}||kr!t|j|jj |}|d}|dzdk(r*|}||j z }t||jnt|dudd} |dz }|dk(rnB|dzdk7r4|||j zz}tt||j|dz }M|d|j zdz |jjjd jzz }||j z }||kr yyw) rQNrKr_offsetrrz RELR bitmap without base address)rh) r;r:rrdrr5rerrrr8Elf_addrr@)rlimitrelrbaserrNrS calc_offsets rrTz&RelrRelocationSection.iter_relocationssb tzz)||Ul !2!2!%!4!4,02E!,Lq Q&$'  554t+-OP$0A$5L#q(%q(Q.&*Q-@&@ (k)J)-77FA T__,q0DLL4H4H4Q4QRT4U4\4\4^^^ DOO #DEUls E EEc|jt|j|_t|jSrG)rflistrTlenr!s rrHz%RelrRelocationSection.num_relocationss7  # # +'+D,A,A,C'DD $4++,,rct|jt|j|_|j|S)rJ)rfrqrT)rrMs rrOz$RelrRelocationSection.get_relocations7  # # +'+D,A,A,C'DD $''**rN)r-r.r/r0rrTrHrOr1rrrarans(*$X-+rraceZdZdZdZdZdZdZeddZ d>dZ d>d Z d>d Z d>d Z d>d Zd>d Zd>dZd>dZede dde ede ddeiZede dde ede dde ede ddeiZede dde ede dde iZede dde ede dde ede dde iZede dde ede ddeede dde iZed e dde ed!e dde ed"e dde iZed#e dde ed$e dde ed%e ddeed&e dde ed'e dde iZed(e dde ed)e dde ed*e ddeed+e dde ed,e dde ed-e dde iZ e!d.e dde e!d/e dde e!d0e dde e!d1e d2de e!d3e d2dee!d4e d5de e!d6e d5dee!d7e dde e!d8e ddee!d9e dde e!d:e ddee!d;e ddee!d<e ddei Z"y=)?RelocationHandlerz4 Handles the logic of relocations in ELF files. c||_yr)r)rrs rrzRelocationHandler.__init__s  rcd|jzd|jzf}|jjD]%}t|ts|j|vs#|cSy)z Given a section, find the relocation section for it in the ELF file. Return a RelocationSection object, or None if none was found. z.relz.relaN)r$r iter_sections isinstancerV)rsectionreloc_section_names relsections rfind_relocations_for_sectionz.RelocationHandler.find_relocations_for_sectionse %',,&( ,,446 "Jz+<=OO'::!! "rc|jj|d}|jD]}|j|||y)z Apply all relocations in reloc_section (a RelocationSection object) to the given stream, that contains the data of the section that is being relocated. The stream is modified as a result. sh_linkN)r get_sectionrT_do_apply_relocation)rr5 reloc_sectionsymtabrelocs rapply_section_relocationsz+RelocationHandler.apply_section_relocationssG ))- *BC"335 =E  % %feV < =rc|d|jk\rtd|dz|j|dd}|d}d}|jj dk(r<|j rtd|z|j j|d}nR|jj dk(r<|j std|z|jj|d}n|jj d k(r~|j rP|td k(r&|d d k7s|d d k7s|dd k7rtd|z|jj|d}n||jj|d}n^|jj dk(r<|j rtd|z|jj|d}n|jj dk(r|jj|d}n|jj dk(r|jj|d}n|jj dk(r|jj|d}nW|jj dk(r:|j std|z|j j|d}|td|z|j"dk(r&|jj$j'd}n|j"dk(r&|jj$j)d}n|j"dk(r&|jj$j+d}nM|j"dk(r&|jj$j-d}ntd|j"zt/|||d}|j1|||d|j2r|d nd !} |j5|d| d|j"dzzz} |j7| |y)"N r_info_symz0Invalid symbol reference in relocation: index %sst_value r_info_typex86z&Unexpected RELA relocation for x86: %sx64z%Unexpected REL relocation for x64: %sMIPS R_MIPS_64r_type2rr_type3r_ssymz9Multiple relocations in R_MIPS_64 are not implemented: %sARMz&Unexpected RELA relocation for ARM: %sAArch64z64-bit PowerPCz%Linux BPF - in-kernel virtual machine LoongArchz+Unexpected REL relocation for LoongArch: %szUnsupported relocation type: %srjrirrz"Invalid bytesize %s for relocationrhrKrvalue sym_valuerBaddend) num_symbolsr get_symbolrget_machine_archr"_RELOCATION_RECIPES_X86get_RELOCATION_RECIPES_X64r _RELOCATION_RECIPES_MIPS_RELA_RELOCATION_RECIPES_MIPS_REL_RELOCATION_RECIPES_ARM_RELOCATION_RECIPES_AARCH64_RELOCATION_RECIPES_PPC64_RELOCATION_RECIPES_EBPF_RELOCATION_RECIPES_LOONGARCHbytesizer8Elf_word Elf_word64Elf_byteElf_halfr calc_func has_addendseek build_stream) rr5rrr reloc_typerecipe value_structoriginal_valuerelocated_values rrz&RelocationHandler._do_apply_relocations  &"4"4"6 6$B,')* *%%eL&9::F =)  << ( ( *e 3}}(>z4P \\ * * , 5}}($5 BD D ??a <<//88L __ !<<//88rsn#23@@@":+)f+)\ 5 5I+GI+V^^r