a

�#�_�M�@s�dd
lm
Z
mZmZmZmZmZmZmZm	Z	m
Z
mZmZm
Z
mZmZmZmZmZmZmZmZ
ddlmZ
ddlTGdd�de�ZdS)�)�UBInt8�UBInt16�UBInt32�UBInt64�ULInt8�ULInt16�ULInt32�ULInt64�SBInt32�SLInt32�SBInt64�SLInt64�Struct�Array�Enum�Padding�	BitStruct�BitField�Value�String�CString)
�ULEB128�
)
�
*c@s�eZ
dZd
Zd5dd�
Zdd�Zdd	�Zd
d�Zd6d
d�
Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Zd'd(�Zd)d*�Zd7d+d,�
Zd-d.�Zd/d0�Zd1d2�Zd3d4�ZdS)8�
ELFStructsa Accessible attributes:

            Elf_{byte|half|word|word64|addr|offset|sword|xword|xsword}:
                Data chunks, as specified by the ELF standard, adjusted for
                correct endianness and word-size.

            Elf_Ehdr:
                ELF file header

            Elf_Phdr:
                Program header

            Elf_Shdr:
                Section header

            Elf_Sym:
                Symbol table entry

            Elf_Rel, Elf_Rela:
                Entries in relocation sections
    T� cCs"|
|_||_
d|_d|_d|_dS�
N��
little_endian�elfclass�e_type�	e_machine�
e_ident_osabi)�selfrr�r$�8/usr/lib/python3.9/site-packages/elftools/elf/structs.py�__init__*s




zELFStructs.__init__c

Cs|j|j
|j|j|jfSrr�
r#r$r$r%�__getstate__2s
zELFStructs.__getstate__cCs,|
\|_|_
}}}|��
|�|||�
dSr)rr�create_basic_structs�create_advanced_structs)r#�stater r!Ze_osabir$r$r%�__setstate__5s


zELFStructs.__setstate__c

Cs�|jrjt
|_t|_t|_t|_|j	d
kr,tnt|_
|j
|_t|_
|j	d
krNtnt|_|j	d
krbtnt|_nbt|_t|_t|_t|_|j	d
kr�tnt|_
|j
|_t|_
|j	d
kr�tnt|_|j	d
kr�tnt|_|��
|��
|��
dS)zr Create word-size related structs and ehdr struct needed for
            initial determining of ELF type.
        rN)rr�Elf_byter�Elf_halfr�Elf_wordr	Z
Elf_word64r�Elf_addr�
Elf_offsetrZ	Elf_sword�	Elf_xwordr
�
Elf_sxwordrrrrr
r�_create_ehdr�_create_leb128�_create_ntbsr'r$r$r%r):s,



















zELFStructs.create_basic_structsNcCs�|
|_||_
||_|��
|��
|��
|��
|��
|��
|�	�
|�
�
|��
|��
|�
�
|�|
�

|��
|��
|��
|��
d
S)z� Create all ELF structs except the ehdr. They may possibly depend
            on provided e_type and/or e_machine parsed from ehdr.
        N)r r!r"�_create_phdr�_create_shdr�_create_chdr�_create_sym�_create_rel�_create_dyn�_create_sunw_syminfo�_create_gnu_verneed�_create_gnu_verdef�_create_gnu_versym�_create_gnu_abi�_create_note�
_create_stabs�_create_arm_attributes�_create_elf_hash�_create_gnu_hash)r#r r!r"r$r$r%r*Vs&

















z"ELFStructs.create_advanced_structsc

Cs
td
tdt
d|�d�
�t|�d�
f
it�
�
t|�d�
f
it�
�
t|�d�
f
it�
�
t|�d�
f
it�
�
|�d	�
td
�
�t|�	d�
f
it
�
�
t|�	d�
f
it�
�
t|�d
�
f
it�
�
|�
d�
|�d�
|�d�
|�d�
|�	d�
|�	d�
|�	d�
|�	d�
|�	d�
|�	d�
�|_dS)N�Elf_EhdrZe_ident�ZEI_MAGZEI_CLASSZEI_DATAZ
EI_VERSIONZEI_OSABIZ
EI_ABIVERSION�r r!Z	e_versionZe_entryZe_phoffZe_shoffZe_flagsZe_ehsizeZe_phentsizeZe_phnumZe_shentsizeZe_shnumZ
e_shstrndx)rrr-rZ
ENUM_EI_CLASSZENUM_EI_DATAZENUM_E_VERSIONZ
ENUM_EI_OSABIrr.ZENUM_E_TYPEZENUM_E_MACHINEr/r0r1rGr'r$r$r%r4qs0








�	











�zELFStructs._create_ehdrc

Cs
t|_
dSr)r�Elf_uleb128r'r$r$r%r5�s
zELFStructs._create_leb128c

Cs
t|_
dSr)r�Elf_ntbsr'r$r$r%r6�s
zELFStructs._create_ntbsc
Cs�t}
|j
d
krt}
n|j
dkr$t}
n|j
dkr2t}
|jdkr�tdt|�d�
f
i|
�
�
|�	d�
|�
d�
|�
d	�
|�d
�
|�d�
|�d�
|�d
�
�	|_nVtdt|�d�
f
i|
�
�
|�d�
|�	d�
|�
d�
|�
d	�
|�d
�
|�d�
|�d
�
�	|_dS)N�EM_ARMZ
EM_AARCH64�EM_MIPSr�Elf_PhdrZp_typeZp_offsetZp_vaddrZp_paddrZp_fileszZp_memszZp_flagsZp_align)
ZENUM_P_TYPE_BASEr!ZENUM_P_TYPE_ARMZENUM_P_TYPE_AARCH64ZENUM_P_TYPE_MIPSrrrr/r1r0rNr2)r#Zp_type_dictr$r$r%r7�s8



















�







�zELFStructs._create_phdrc
Cs�t}
|j
d
krt}
n|j
dkr$t}
n|j
dkr2t}
td|�d�
t|�d�
f
i|
�
�
|�d�
|�	d�
|�
d	�
|�d
�
|�d�
|�d�
|�d
�
|�d�
�|_dS)zmSection header parsing.

        Depends on e_machine because of machine-specific values in sh_type.
        rLZ	EM_X86_64rM�Elf_ShdrZsh_nameZsh_typeZsh_flagsZsh_addrZ	sh_offsetZsh_sizeZsh_linkZsh_infoZsh_addralignZ
sh_entsizeN)ZENUM_SH_TYPE_BASEr!ZENUM_SH_TYPE_ARMZENUM_SH_TYPE_AMD64ZENUM_SH_TYPE_MIPSrr/rr2r0r1rO)r#Zsh_type_dictr$r$r%r8�s&


















�zELFStructs._create_shdrc
CsZt|�
d
�
f
it�
�
|�d�
|�d�
g}
|jdkrD|
�d|�
d�
�
tdg
|
�
R�|_dS)NZch_typeZch_sizeZch_addralign�@rZch_reserved�Elf_Chdr)rr/ZENUM_ELFCOMPRESS_TYPEr2r�insertrrQ)r#�fieldsr$r$r%r9�s

�


zELFStructs._create_chdrc
Cs
|jd
kr0|�
d�
tddd��tddd��g}
n�|jdkr�|�d	�
|�d
�
|�d�
|�d�
|�d
�
tddd��tddd��tddd��tddd��tddd��tddd��g}
n$|�
d�
tddd��tddd��g}
td|�d�
g|
�
R�|_|
|�	d�
g
}td|�d�
g|�
R�|_
dS)Nr�r_infoZ
r_info_symc

Ss|d
d?d@S)NrT�i���r$�
�ctxr$r$r%�<lambda>��z(ELFStructs._create_rel.<locals>.<lambda>Zr_info_typec

Ss|d
d@S)NrT�r$rVr$r$r%rX�rYrM�r_sym�r_ssym�r_type3�r_type2�r_typec

Ss|d
S)Nr[r$rVr$r$r%rX�rYZr_info_ssymc

Ss|d
S)Nr\r$rVr$r$r%rX�rYc

Ss|d
S)Nr_r$rVr$r$r%rX�rYZr_info_type2c

Ss|d
S)Nr^r$rVr$r$r%rX�rYZr_info_type3c

Ss|d
S)Nr]r$rVr$r$r%rX�rYc

Ss8|d
d>|dd>B|dd>B|dd>B|d	BS)
Nr[rr\�r]�r^rUr_r$rVr$r$r%rX�s


�
�
��c

Ss|d
d?d@S)NrTr���r$rVr$r$r%rX
rYc

Ss|d
d@S)NrTrbr$rVr$r$r%rX
rY�Elf_RelZr_offsetZr_addend�Elf_Rela)rr2rr!r/r-rr0rcr3rd)r#rSZfields_and_addendr$r$r%r;�sV



�
��










��

�
��
��


��zELFStructs._create_relc
Csntt
�
}
|jtvr$|
�t|j�

n|jd
kr8|
�t�

tdt|�	d�
f
i|
�
�
|�
d�
tddd���|_dS)NZELFOSABI_SOLARIS�Elf_DynZd_tag�d_valZd_ptrc

Ss|d
S)Nrfr$rVr$r$r%rX
rYz(ELFStructs._create_dyn.<locals>.<lambda>)
�dictZENUM_D_TAG_COMMONr!ZENUMMAP_EXTRA_D_TAG_MACHINE�updater"ZENUM_D_TAG_SOLARISrrr3r2rre)r#Z
d_tag_dictr$r$r%r<
s










�zELFStructs._create_dync
Cs�td
t
tdd�f
it�
�
t
tdd�f
it�
�
�}
tdtd�
t
tdd�f
it�
�
�}|jd	kr�td
|�	d�
|�
d�
|�	d
�
|
|t
|�d�
f
it�
�
�|_
n:td
|�	d�
|
|t
|�d�
f
it�
�
|�
d�
|�d
�
�|_
dS)NZst_infoZbindrH�typeZst_other�Z
visibility�r�Elf_SymZst_nameZst_value�st_sizeZst_shndx)rrrZENUM_ST_INFO_BINDZENUM_ST_INFO_TYPErZENUM_ST_VISIBILITYrrr/r0r.Z
ENUM_ST_SHNDXrlr2)r#Zst_info_structZst_other_structr$r$r%r:
s2

�

�







�	





�zELFStructs._create_symc

Cs*td
t
|�d�
f
it�
�
|�d�
�|_dS)N�Elf_Sunw_SyminfoZ
si_boundtoZsi_flags)rrr.ZENUM_SUNW_SYMINFO_BOUNDTOrnr'r$r$r%r=9
s


�zELFStructs._create_sunw_syminfoc

	Cshtd
|�
d�
|�
d�
|�d�
|�d�
|�d�
�|_td|�d�
|�
d	�
|�
d
�
|�d�
|�d�
�|_dS)
N�Elf_VerneedZ
vn_versionZvn_cntZvn_fileZvn_auxZvn_next�Elf_VernauxZvna_hashZ	vna_flagsZ	vna_otherZvna_nameZvna_next)rr.r/rorpr'r$r$r%r>?
s




�




�zELFStructs._create_gnu_verneedc

Cs`td
|�
d�
|�
d�
|�
d�
|�
d�
|�d�
|�d�
|�d�
�|_td	|�d
�
|�d�
�|_dS)N�
Elf_VerdefZ
vd_versionZvd_flagsZvd_ndxZvd_cntZvd_hashZvd_auxZvd_next�Elf_VerdauxZvda_nameZvda_next)rr.r/rqrrr'r$r$r%r?Q
s






�	

�zELFStructs._create_gnu_verdefc

Cs"td
t
|�d�
f
it�
�
�|_dS)N�
Elf_VersymZndx)rrr.ZENUM_VERSYMrsr'r$r$r%r@b
s
�zELFStructs._create_gnu_versymc

Cs:td
t
|�d�
f
it�
�
|�d�
|�d�
|�d�
�|_dS)N�Elf_abiZabi_osZ	abi_majorZ	abi_minorZabi_tiny)rrr/ZENUM_NOTE_ABI_TAG_OSrtr'r$r$r%rAi
s



�zELFStructs._create_gnu_abicCs�
td
|�
d�
|�
d�
t|�
d�
f
i|
dkr.tnt�
�
�|_|jdkr�td|�d�
td	d
�|�d�
|�d�
|�	d
�
|�
d�
|�
d�
|�
d�
|�
d�
|�
d�
|�
d�
tdd�tdd��|_nxtd|�d�
td	d
�|�d�
|�d�
td�
|�	d
�
|�
d�
|�
d�
|�
d�
|�
d�
|�
d�
|�
d�
tdd�tdd��|_td|�	d�
|�	d�
t
dd�td|�d�
|�d �
|�d!�
��t
d"d�td#�
��|_dS)$N�Elf_NhdrZn_nameszZn_descsz�n_typeZET_COREr�Elf_PrpsinfoZpr_stateZpr_snamerZpr_zombZpr_niceZpr_flagZpr_uidZpr_gidZpr_pidZpr_ppidZpr_pgrpZpr_sidZpr_fnameraZ	pr_psargs�PrH�Elf_Nt_File�num_map_entriesZ	page_sizec


Ss|jSr�
rzrVr$r$r%rX�
rYz)ELFStructs._create_note.<locals>.<lambda>ZElf_Nt_File_EntryZvm_startZvm_endZpage_offsetc


Ss|jSrr{rVr$r$r%rX�
rY�filename)rr/rZENUM_NOTE_N_TYPEZENUM_CORE_NOTE_N_TYPErurr-rr2r.rwrrr0r1rry)r#r r$r$r%rBs
sl




��















�













�






��
��zELFStructs._create_notec

	Cs6td
|�
d�
|�d�
|�d�
|�d�
|�
d�
�|_dS)N�	Elf_StabsZn_strxrvZn_otherZn_descZn_value)rr/r-r.r}r'r$r$r%rC�
s




�zELFStructs._create_stabsc

Cs@td
|�
d�
|jddd��|_tdt|�d�
f
it�
�
�|_dS)NZElf_Attr_Subsection�lengthZvendor_namezutf-8)
�encoding�Elf_Attribute_Tag�tag)rr/rKZElf_Attr_Subsection_HeaderrrJZENUM_ATTR_TAG_ARMr�r'r$r$r%rD�
s


��

��z!ELFStructs._create_arm_attributesc


CsBtd
|�
d�
|�
d�
tdd�|�
d�
�tdd�|�
d�
��|_dS)	N�Elf_Hash�nbuckets�nchainsc

Ss|d
S�Nr�r$rVr$r$r%rX�
rYz-ELFStructs._create_elf_hash.<locals>.<lambda>�bucketsc

Ss|d
S)Nr�r$rVr$r$r%rX�
rYZchains)rr/rr�r'r$r$r%rE�
s



�zELFStructs._create_elf_hashc

CsRtd
|�
d�
|�
d�
|�
d�
|�
d�
tdd�|�d�
�td	d�|�
d
�
��|_dS)N�Gnu_Hashr�Z	symoffset�
bloom_sizeZbloom_shiftc

Ss|d
S)Nr�r$rVr$r$r%rX�
rYz-ELFStructs._create_gnu_hash.<locals>.<lambda>Zbloomc

Ss|d
Sr�r$rVr$r$r%rX�
rYr�)rr/rr2r�r'r$r$r%rF�
s





�zELFStructs._create_gnu_hash)Tr)NNN)
N)�__name__�
__module__�__qualname__�__doc__r&r(r,r)r*r4r5r6r7r8r9r;r<r:r=r>r?r@rArBrCrDrErFr$r$r$r%rs2


 5


=
rN)Z	constructrrrrrrrr	r
rrr
rrrrrrrrrZcommon.construct_utilsrZenums�objectrr$r$r$r%�<module>
s\