File: //lib/python3.9/site-packages/elftools/elf/__pycache__/relocation.cpython-39.pyc
a
�#�_�/ � @ s� d dl mZ ddlmZ ddlmZmZ ddlmZ ddl m
Z
m Z m Z m
Z
mZmZ G dd � d e�ZG d
d � d e�ZG d d
� d
ee�ZG dd� de�ZdS )� )�
namedtuple� )�ELFRelocationError)�
elf_assert� struct_parse� )�Section)�ENUM_RELOC_TYPE_i386�ENUM_RELOC_TYPE_x64�ENUM_RELOC_TYPE_MIPS�ENUM_RELOC_TYPE_ARM�ENUM_RELOC_TYPE_AARCH64�
ENUM_D_TAGc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�
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 C s || _ || _d S �N)�entry�elffile)�selfr r � r �;/usr/lib/python3.9/site-packages/elftools/elf/relocation.py�__init__ s zRelocation.__init__c C s
d| j v S )z6 Is this a RELA relocation? If not, it's REL.
�r_addend�r �r r r r �is_RELA s zRelocation.is_RELAc C s
| j | S )z% Dict-like access to entries
r )r �namer r r � __getitem__" s zRelocation.__getitem__c C s d| � � rdnd| jf S )Nz<Relocation (%s): %s>ZRELAZREL)r r r r r r �__repr__' s �zRelocation.__repr__c C s | � � S r )r r r r r �__str__, s zRelocation.__str__N) �__name__�
__module__� __qualname__�__doc__r r r r r r r r r r s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�RelocationTablezL Shared functionality between relocation sections and relocation tables
c C sR |j | _|| _|j| _|| _|| _|| _|r8| jj| _ n
| jj
| _ | j � � | _ d S r )
�stream�_stream�_elffile�structsZ _elfstructs�_size�_offset�_is_relaZElf_Rela� entry_structZElf_RelZsizeof�
entry_size)r r �offset�sizeZis_relar r r r 4 s
zRelocationTable.__init__c C s | j S )z> Is this a RELA relocation section? If not, it's REL.
)r* r r r r r C s zRelocationTable.is_RELAc C s | j | j S )z. Number of relocations in the section
)r( r, r r r r �num_relocationsH s z RelocationTable.num_relocationsc C s. | j || j }t| j| j|d�}t|| j�S )zM Get the relocation at index #n from the section (Relocation object)
�Z
stream_pos)r) r, r r+ r% r r& )r �nZ entry_offsetr r r r �get_relocationM s �z RelocationTable.get_relocationc c s" t | �� �D ]}| �|�V q dS )z2 Yield all the relocations in the section
N)�ranger/ r2 )r �ir r r �iter_relocationsW s z RelocationTable.iter_relocationsN) r r r! r" r r r/ r2 r5 r r r r r# 0 s
r# c @ s e Zd ZdZdd� ZdS )�RelocationSectionzK ELF relocation section. Serves as a collection of Relocation entries.
c C sl t �| |||� t�| | j| d | d |d dk� t|d dv d� t|d | jkd|d | jf � d S ) NZ sh_offsetZsh_sizeZsh_type�SHT_RELA)ZSHT_RELr7 z Unknown relocation type sectionZ
sh_entsizez*Expected sh_entsize of %s section to be %s)r r r# r r r, )r �headerr r r r r r a s
� �
��zRelocationSection.__init__N)r r r! r" r r r r r r6 ^ s r6 c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zed
d �Z d.d
d�Z
d/dd�Z d0dd�Z d1dd�Z
d2dd�Zd3dd�Zed e dde d �ed e dded �iZed e d d e
d �ed! e dd e
d �ed" e dd ed �iZed# e dde
d �ed$ e dde d �iZed% e dde
d �ed& e dde d �ed' e dde d �iZed( e d d e
d �ed) e d d e
d �ed* e dd ed �ed+ e dd e
d �ed, e dd e
d �iZd-S )4�RelocationHandlerz4 Handles the logic of relocations in ELF files.
c C s
|| _ d S r )r )r r r r r r p s zRelocationHandler.__init__c C sD d|j d|j f}| j�� D ] }t|t�r |j |v r | S q dS )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 Z
iter_sections�
isinstancer6 )r �sectionZreloc_section_namesZ
relsectionr r r � find_relocations_for_sections s �
�
z.RelocationHandler.find_relocations_for_sectionc C s0 | j �|d �}|�� D ]}| �|||� qdS )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.
Zsh_linkN)r Z get_sectionr5 �_do_apply_relocation)r r$ Z
reloc_section�symtab�relocr r r �apply_section_relocations� s z+RelocationHandler.apply_section_relocationsc
C s� |d |� � kr td|d ��|�|d �d }|d }d }| j�� dkrp|�� r`td| ��| j�|d �}n�| j�� dkr�|�� s�td| ��| j�|d �}n�| j�� d kr�|�� r�td
| ��| j �|d �}nR| j�� d k�r|�� r�td | ��| j
�|d �}n | j�� d
k�r&| j �|d �}|d u �r<td| ��|j dk�rX| jj
�d�}n*|j dk�rt| jj
�d�}ntd|j ��t|||d d�}|j|||d |j�r�|d ndd�} |�|d � | d|j d } |�| |� d S )NZ
r_info_symz0Invalid symbol reference in relocation: index %sZst_valueZ r_info_typeZx86z&Unexpected RELA relocation for x86: %sZx64z%Unexpected REL relocation for x64: %sZMIPSz'Unexpected RELA relocation for MIPS: %sZARMz&Unexpected RELA relocation for ARM: %sZAArch64z Unsupported relocation type: %s� � � z"Invalid bytesize %s for relocationZr_offsetr0 r r ��value� sym_valuer- Zaddendr )Z num_symbolsr Z
get_symbolr Zget_machine_archr �_RELOCATION_RECIPES_X86�get�_RELOCATION_RECIPES_X64�_RELOCATION_RECIPES_MIPS�_RELOCATION_RECIPES_ARM�_RELOCATION_RECIPES_AARCH64�bytesizer' ZElf_wordZ
Elf_word64r � calc_func�
has_addend�seekZ build_stream)
r r$ r? r> rF Z
reloc_typeZrecipeZ value_struct�original_valueZrelocated_valuer r r r= � sx ������
� ���z&RelocationHandler._do_apply_relocation�_RELOCATION_RECIPE_TYPEz bytesize has_addend calc_funcr c C s | S r r rD r r r �_reloc_calc_identity� s z&RelocationHandler._reloc_calc_identityc C s || S r r rD r r r �_reloc_calc_sym_plus_value� s z,RelocationHandler._reloc_calc_sym_plus_valuec C s || | S r r rD r r r � _reloc_calc_sym_plus_value_pcrel� s z2RelocationHandler._reloc_calc_sym_plus_value_pcrelc C s || S r r rD r r r �_reloc_calc_sym_plus_addend� s z-RelocationHandler._reloc_calc_sym_plus_addendc C s || | S r r rD r r r �!_reloc_calc_sym_plus_addend_pcrel� s z3RelocationHandler._reloc_calc_sym_plus_addend_pcrelc C s |d | |d S )NrA r rD r r r �$_arm_reloc_calc_sym_plus_value_pcrel� s z6RelocationHandler._arm_reloc_calc_sym_plus_value_pcrelZ R_ARM_ABS32rA F)rM rO rN Z
R_ARM_CALLZR_AARCH64_ABS64rC TZR_AARCH64_ABS32ZR_AARCH64_PREL32Z R_MIPS_NONEZ R_MIPS_32Z
R_386_NONEZR_386_32Z
R_386_PC32Z
R_X86_64_NONEZ R_X86_64_64Z
R_X86_64_PC32Z R_X86_64_32Z R_X86_64_32SN)r )r )r )r )r )r )r r r! r" r r< r@ r= r rR rS rT rU rV rW rX r rK r
rL r rJ r rG r
rI r r r r r9 m s�
Q�
���
���� ��� ���� ������r9 N)� collectionsr Zcommon.exceptionsr Z common.utilsr r Zsectionsr Zenumsr r
r r r
r �objectr r# r6 r9 r r r r �<module> s .