File: //lib/python3.9/site-packages/elftools/elf/__pycache__/dynamic.cpython-39.pyc
a
�#�_�5 � @ s� d dl Z d dlmZ ddlmZmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZ d d
lmZ d d lmZmZmZ G d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� dee�ZG dd� de e�ZdS )� N)� defaultdict� )� ELFHashTable� GNUHashTable)�Section�Symbol)�
ENUM_D_TAG)�Segment)�RelocationTable� )�ELFError)�
elf_assert� struct_parse�parse_cstring_from_streamc @ s e Zd ZdZdd� Zdd� ZdS )�_DynamicStringTablez� Bare string table based on values found via ELF dynamic tags and
loadable segments only. Good enough for get_string() only.
c C s || _ || _d S )N)�_stream�
_table_offset)�self�stream� table_offset� r �8/usr/lib/python3.9/site-packages/elftools/elf/dynamic.py�__init__ s z _DynamicStringTable.__init__c C s$ t | j| j| �}|r |�d�S dS )zI Get the string stored at the given offset in this string table.
zutf-8� )r r r �decode)r �offset�sr r r �
get_string s z _DynamicStringTable.get_stringN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ s< e Zd ZdZeg d��Zdd� Zdd� Zdd� Zd d
� Z d S ) �
DynamicTaga� Dynamic Tag object - representing a single dynamic tag entry from a
dynamic section.
Allows dictionary-like access to the dynamic structure. For special
tags (those listed in the _HANDLED_TAGS set below), creates additional
attributes for convenience. For example, .soname will contain the actual
value of DT_SONAME (fetched from the dynamic symbol table).
)Z DT_NEEDEDZDT_RPATHZ
DT_RUNPATHZ DT_SONAMEZDT_SUNW_FILTERc C sJ |d u rt d��|| _|j| jv rFt| |jdd � �� |�| jj�� d S )Nz(Creating DynamicTag without string table� )r �entry�d_tag�
_HANDLED_TAGS�setattr�lowerr �d_val)r r$ � stringtabler r r r 1 s �zDynamicTag.__init__c C s
| j | S )z/ Implement dict-like access to entries
)r$ )r �namer r r � __getitem__9 s zDynamicTag.__getitem__c C s d| j j| j f S )Nz<DynamicTag (%s): %r>)r$ r% �r r r r �__repr__> s zDynamicTag.__repr__c C sJ | j j| jv r.dt| | j jdd � �� � }n d| j j }d| j j|f S )Nz"%s"r# z%#xz<DynamicTag (%s) %s>)r$ r% r& �getattrr( �d_ptr)r r r r r �__str__A s zDynamicTag.__str__N)
r r r r! � frozensetr& r r, r. r1 r r r r r" $ s �r" c @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd d
�Zdd d �Zd
d� Z dd� Z
dd� Z dd� Z dS )�DynamiczA Shared functionality between dynamic sections and segments.
c C sF || _ |j| _|| _|s dnd| _|| _| jj�� | _|| _ || _
dS )a�
stream:
The file-like object from which to load data
elffile:
The parent elffile object
stringtable:
A stringtable reference to use for parsing string references in
entries
position:
The file offset of the dynamic segment/section
empty:
Whether this is a degenerate case with zero entries. Normally, every
dynamic table will have at least one entry, the DT_NULL terminator.
���r N) �elffileZstructs�
elfstructsr � _num_tags�_offset�Elf_Dyn�sizeof�_tagsize�_empty� _stringtable)r r r5 r* Zposition�emptyr r r r L s zDynamic.__init__c C sD d}| j |d�D ]}|d } q"qd}|r<t| j�|�d�}||fS )zH Return the virtual address and file offset of a dynamic table.
N��typer0 )�
_iter_tags�nextr5 Zaddress_offsets)r Ztag_nameZptr�tagr r r r �get_table_offsetj s zDynamic.get_table_offsetc C sJ | j r | j S | �d�\}}|dur6t| j|�| _ | j S | j�d�| _ | j S )z� Return a string table for looking up dynamic tag related strings.
This won't be a "full" string table object, but will at least
support the get_string() function.
Z DT_STRTABNz.dynstr)r= rD r r r5 Zget_section_by_name)r �_r r r r �_get_stringtablez s zDynamic._get_stringtableNc c sP | j r
dS t�� D ]8}| �|�}|du s4|d |kr:|V |d dkr qLqdS )z; Yield all raw tags (limit to |type| if specified)
Nr% �DT_NULL)r< � itertools�count�_get_tag)r r@ �nrC r r r rA � s
zDynamic._iter_tagsc c s& | j |d�D ]}t|| �� �V q dS )z7 Yield all tags (limit to |type| if specified)
r? N)rA r" rF )r r@ rC r r r � iter_tags� s zDynamic.iter_tagsc C s@ | j dkr || j kr t|��| j|| j }t| jj| j|d�S )z3 Get the raw tag at index #n from the file
r4 )Z
stream_pos)r7 �
IndexErrorr8 r; r r6 r9 r )r rK r r r r rJ � s �zDynamic._get_tagc C s t | �|�| �� �S )zC Get the tag at index #n from the file (DynamicTag object)
)r" rJ rF )r rK r r r �get_tag� s zDynamic.get_tagc C sL | j dkr| j S t�� D ].}| �|�}|jjdkr|d | _ | j S qdS )zG Number of dynamic tags in the file, including the DT_NULL tag
r4 rG r N)r7 rH rI rN r$ r% )r rK rC r r r �num_tags� s
zDynamic.num_tagsc C s" i }t | �d��rht| j| �d�d t| �d��d d�|d< t| �d��d }t|d j|kd| � t | �d ��r�t| j| �d �d t| �d
��d d �|d < t| �d
��d }t|d j|kd| � t | �d���r t| j| �d�d t| �d��d t| �d��d td k�|d< |S )z� Load all available relocation tables from DYNAMIC tags.
Returns a dictionary mapping found table types (REL, RELA,
JMPREL) to RelocationTable objects.
ZDT_RELr ZDT_RELSZr) FZRELZ DT_RELENTzExpected DT_RELENT to be %sZDT_RELAZ DT_RELASZTZRELAZ
DT_RELAENTz Expected DT_RELAENT to be %sZ DT_JMPRELZ DT_PLTRELSZZ DT_PLTRELZJMPREL) �listrL r
r5 rD rB r
Z
entry_sizer )r �resultZrelentszr r r �get_relocation_tables� s4 �� �� �z Dynamic.get_relocation_tables)N)N)
r r r r! r rD rF rA rL rJ rN rO rR r r r r r3 I s
r3 c @ s e Zd ZdZdd� ZdS )�DynamicSectionzH ELF dynamic table section. Knows how to process the list of tags.
c C sF t �| |||� |�|d �}t�| | j| j|| d | d dk� d S )N�sh_link� sh_offsetZsh_typeZ
SHT_NOBITS)r r � get_sectionr3 r r5 )r �headerr+ r5 r* r r r r � s
�zDynamicSection.__init__N)r r r r! r r r r r rS � s rS c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�DynamicSegmentzH ELF dynamic table segment. Knows how to process the list of tags.
c C s| d }|� � D ]0}t|t�r |d |d kr |�|d �} q>q t�| ||� t�| |||| d | d dk� d | _d | _d S )NrU Zp_offsetrT �p_fileszr ) Z
iter_sections�
isinstancerS rV r r r3 � _symbol_list�_symbol_name_map)r rW r r5 r* �sectionr r r r � s
�
�zDynamicSegment.__init__c C s" | j du rt| �� �| _ t| j �S )zA Number of symbols in the table recovered from DT_SYMTAB
N)r[ rP � iter_symbols�lenr- r r r � num_symbols s
zDynamicSegment.num_symbolsc C s" | j du rt| �� �| _ | j | S )zG Get the symbol at index #index from the table (Symbol object)
N)r[ rP r^ )r �indexr r r �
get_symbol s
zDynamicSegment.get_symbolc sb � j du r<tt�� _ t� �� �D ]\}}� j |j �|� q � j �|�}|r^� fdd�|D �S dS )za Get a symbol(s) by name. Return None if no symbol by the given name
exists.
Nc s g | ]}� � |��qS r )rb )�.0�ir- r r �
<listcomp> � z5DynamicSegment.get_symbol_by_name.<locals>.<listcomp>)r\ r rP � enumerater^ r+ �append�get)r r+ rd ZsymZsymnumsr r- r �get_symbol_by_name s
z!DynamicSegment.get_symbol_by_namec c s� | � d�\}}|du s |du r&td��| jj�� }d}| � d�\}}|durjt| j|| �}||�� | }|du r�| � d�\}}|dur�t| j|| �}||�� | }|du �rhd} | � � D ]V}
|
d } |
d dkr�||
d kr�td
|
d |f ��| |kr�| du �s| | kr�| } q�| du �rd| j�
� D ]:} | d |k�r(|| d | d k�r(| d | d } �q(| }|du �rztd
��| � � }
t || | �D ]:}t
| jj| j|| | �}|
�|d �}t||�V �q�dS )a0 Yield all symbols in this dynamic segment. The symbols are usually
the same as returned by SymbolTableSection.iter_symbols. However,
in stripped binaries, SymbolTableSection might have been removed.
This method reads from the mandatory dynamic tag DT_SYMTAB.
Z DT_SYMTABNz#Segment does not contain DT_SYMTAB.Z DT_GNU_HASHZDT_HASHr0 r% Z DT_SYMENTr) z DT_SYMENT (%d) != Elf_Sym (%d).Zp_vaddrrY z&Cannot determine the end of DT_SYMTAB.Zst_name)rD r r6 ZElf_Symr: r r5 Zget_number_of_symbolsr rL Z
iter_segmentsrF �ranger r r r )r Ztab_ptrZ
tab_offsetZ symbol_sizeZend_ptrrE Zgnu_hash_offsetZ hash_sectionZ hash_offsetZ nearest_ptrrC Ztag_ptrZsegmentZ string_tablerd �symbolZ symbol_namer r r r^ sb
�
�
���
�
�zDynamicSegment.iter_symbolsN) r r r r! r r` rb rj r^ r r r r rX � s rX )rH � collectionsr �hashr r Zsectionsr r Zenumsr �segmentsr Z
relocationr
Zcommon.exceptionsr Z common.utilsr
r r �objectr r" r3 rS rX r r r r �<module> s %