a
    ,2a.                    @   s  d Z dZdZdZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlmZ ddlmZ ddlmZ dd	lmZ dd
lmZ ddlZddlZddlZeded eZdddZedddd Zedddd Zdd ZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)d Z*d!Z+d"Z,d#Z-d$Z.d%Z/d&Z0d'Z1d(Z2d)Z3d*d+ Z4g d,Z5e4e5Z6g d-Z7e4e7Z8g d.Z9e4e9Z:g d/Z;e4e;Z<g d0Z=e4e=Z>g d1Z?e4e?Z@g d2ZAe4eAZBg d3ZCe4eCZDdZEg d4ZFe4eFZGg d5ZHe4eHZIdZJd6ZKd7ZLd8ZMd9ZNd:ZOd;ZPd<ZQd=ZRd>ZSg d?ZTe4eTZUg d@ZVe4eVZWg dAZXe4eXZYeZeXZYeXD ].\Z[Z\e\eYv rzeYe\ ]e[ n
e[geYe\< qXdBdC Z^dDdE Z_dFdG Z`dHdI ZadJdK ZbdLdM ZcG dNdO dOZdG dPdQ dQeeZfG dRdS dSZgd6d6d6d6d7d7d9d9d9d9d9d<d<d<d6dTZhedddUdV ZieddWdXdYdZ ZjG d[d\ d\ZkG d]d^ d^ekZledddXd_d` ZmG dadb dbekZnG dcdd ddZoG dedf dfeoZpG dgdh dheoZqG didj djeoZrG dkdl dleoZsG dmdn dneoZtG dodp dpeoZuG dqdr dreoZvG dsdt dteoZwG dudv dveoZxG dwdx dxeoZyG dydz dzeoZzG d{d| d|eoZ{G d}d~ d~eoZ|G dd deoZ}G dd deoZ~G dd denZG dd dZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd dZece
je
j e
j d Zdd Zece
je
j e
j d Zedddd ZG dd dZdd Zedkre  dS )ab  pefile, Portable Executable reader module

All the PE file basic structures are available with their default names as
attributes of the instance returned.

Processed elements such as the import table are made available with lowercase
names, to differentiate them from the upper case basic structure names.

pefile has been tested against many edge cases such as corrupted and malformed
PEs as well as malware, which often attempts to abuse the format way beyond its
standard use. To the best of my knowledge most of the abuse is handled
gracefully.

Copyright (c) 2005-2021 Ero Carrera <ero.carrera@gmail.com>
zEro Carreraz2021.9.3zero.carrera@gmail.com    N)Counter)sha1)sha256)sha512)md5backslashreplace_backslashreplace   Fc                    s"   |st  S  fdd}|S )Nc                    s*   t |  t |  fdd}|S )Nc                     s   t  | i |S N)copymodcopy)argskwargsZcached_func W/home/tom/ab/renpy-build/tmp/install.linux-x86_64/lib/python3.9/site-packages/pefile.pywrapper9   s    z-lru_cache.<locals>.decorator.<locals>.wrapper)	functools	lru_cachewraps)fr   maxsizetypedr   r   	decorator6   s    zlru_cache.<locals>.decorator)r   r   )r   r   r   r   r   r   r   r   2   s    
r      )r   c                 C   s   |t k r| S t| d d S )N   )FILE_ALIGNMENT_HARDCODED_VALUEint)valfile_alignmentr   r   r   cache_adjust_FileAlignmentC   s    r!   c                 C   s,   |dk r|}|r(| | r(|t | |  S | S )N   )r   )r   section_alignmentr    r   r   r   cache_adjust_SectionAlignmentJ   s
    r$   c                 C   s0   z|  d}W n ty*   |  d}Y n0 |S )N r   )count	TypeError)datar&   r   r   r   count_zeroesZ   s
    r)          r          iMZ  iZM  iNE  iLE  iLX  iVZ  iPE             l            i  i  c                 C   s   t dd | D |  S )Nc                 S   s   g | ]}|d  |d fqS )   r   r   ).0er   r   r   
<listcomp>       z two_way_dict.<locals>.<listcomp>)dict)pairsr   r   r   two_way_dict   s    r7   ))IMAGE_DIRECTORY_ENTRY_EXPORTr   )IMAGE_DIRECTORY_ENTRY_IMPORTr0   )IMAGE_DIRECTORY_ENTRY_RESOURCE   )IMAGE_DIRECTORY_ENTRY_EXCEPTION   )IMAGE_DIRECTORY_ENTRY_SECURITY   )IMAGE_DIRECTORY_ENTRY_BASERELOC   )IMAGE_DIRECTORY_ENTRY_DEBUG   )ZIMAGE_DIRECTORY_ENTRY_COPYRIGHT   )ZIMAGE_DIRECTORY_ENTRY_GLOBALPTR   )IMAGE_DIRECTORY_ENTRY_TLS	   )!IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG
   )"IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT   )ZIMAGE_DIRECTORY_ENTRY_IAT   )"IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT   )Z$IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR   )ZIMAGE_DIRECTORY_ENTRY_RESERVED   ))ZIMAGE_FILE_RELOCS_STRIPPEDr0   )IMAGE_FILE_EXECUTABLE_IMAGEr;   )ZIMAGE_FILE_LINE_NUMS_STRIPPEDr?   )ZIMAGE_FILE_LOCAL_SYMS_STRIPPEDrE   )ZIMAGE_FILE_AGGRESIVE_WS_TRIMr.   )ZIMAGE_FILE_LARGE_ADDRESS_AWAREr-   )ZIMAGE_FILE_16BIT_MACHINE@   )ZIMAGE_FILE_BYTES_REVERSED_LOr	   )ZIMAGE_FILE_32BIT_MACHINE   )ZIMAGE_FILE_DEBUG_STRIPPEDr   )Z"IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP   )ZIMAGE_FILE_NET_RUN_FROM_SWAPr   )ZIMAGE_FILE_SYSTEMr"   )IMAGE_FILE_DLLr+   )ZIMAGE_FILE_UP_SYSTEM_ONLY @  )ZIMAGE_FILE_BYTES_REVERSED_HIr,   ).)ZIMAGE_SCN_TYPE_REGr   )ZIMAGE_SCN_TYPE_DSECTr0   )ZIMAGE_SCN_TYPE_NOLOADr;   )ZIMAGE_SCN_TYPE_GROUPr?   )ZIMAGE_SCN_TYPE_NO_PADrE   )ZIMAGE_SCN_TYPE_COPYr.   )ZIMAGE_SCN_CNT_CODEr-   )ZIMAGE_SCN_CNT_INITIALIZED_DATArR   )Z IMAGE_SCN_CNT_UNINITIALIZED_DATAr	   )ZIMAGE_SCN_LNK_OTHERrS   )ZIMAGE_SCN_LNK_INFOr   )ZIMAGE_SCN_LNK_OVERrT   )ZIMAGE_SCN_LNK_REMOVEr   )ZIMAGE_SCN_LNK_COMDATr"   )ZIMAGE_SCN_MEM_PROTECTEDrV   )ZIMAGE_SCN_NO_DEFER_SPEC_EXCrV   )ZIMAGE_SCN_GPRELr,   )ZIMAGE_SCN_MEM_FARDATAr,   )ZIMAGE_SCN_MEM_SYSHEAP   )ZIMAGE_SCN_MEM_PURGEABLE   )ZIMAGE_SCN_MEM_16BITrX   )ZIMAGE_SCN_MEM_LOCKEDi   )ZIMAGE_SCN_MEM_PRELOADi   )ZIMAGE_SCN_ALIGN_1BYTESr*   )ZIMAGE_SCN_ALIGN_2BYTESi    )ZIMAGE_SCN_ALIGN_4BYTESi  0 )ZIMAGE_SCN_ALIGN_8BYTESi  @ )ZIMAGE_SCN_ALIGN_16BYTESi  P )ZIMAGE_SCN_ALIGN_32BYTESi  ` )ZIMAGE_SCN_ALIGN_64BYTESi  p )ZIMAGE_SCN_ALIGN_128BYTESi   )ZIMAGE_SCN_ALIGN_256BYTESi   )ZIMAGE_SCN_ALIGN_512BYTESi   )ZIMAGE_SCN_ALIGN_1024BYTESi   )ZIMAGE_SCN_ALIGN_2048BYTESi   )ZIMAGE_SCN_ALIGN_4096BYTESi   )ZIMAGE_SCN_ALIGN_8192BYTESi   )ZIMAGE_SCN_ALIGN_MASKi   )ZIMAGE_SCN_LNK_NRELOC_OVFLi   )ZIMAGE_SCN_MEM_DISCARDABLEi   )ZIMAGE_SCN_MEM_NOT_CACHEDi   )ZIMAGE_SCN_MEM_NOT_PAGED   )ZIMAGE_SCN_MEM_SHARED   )IMAGE_SCN_MEM_EXECUTEi    )ZIMAGE_SCN_MEM_READi   @)IMAGE_SCN_MEM_WRITEr/   ))ZIMAGE_DEBUG_TYPE_UNKNOWNr   )ZIMAGE_DEBUG_TYPE_COFFr0   )ZIMAGE_DEBUG_TYPE_CODEVIEWr;   )ZIMAGE_DEBUG_TYPE_FPOr=   )ZIMAGE_DEBUG_TYPE_MISCr?   )ZIMAGE_DEBUG_TYPE_EXCEPTIONrA   )ZIMAGE_DEBUG_TYPE_FIXUPrC   )ZIMAGE_DEBUG_TYPE_OMAP_TO_SRCrD   )ZIMAGE_DEBUG_TYPE_OMAP_FROM_SRCrE   )ZIMAGE_DEBUG_TYPE_BORLANDrG   )ZIMAGE_DEBUG_TYPE_RESERVED10rI   )ZIMAGE_DEBUG_TYPE_CLSIDrK   )ZIMAGE_DEBUG_TYPE_VC_FEATURErL   )ZIMAGE_DEBUG_TYPE_POGOrN   )ZIMAGE_DEBUG_TYPE_ILTCGrO   )ZIMAGE_DEBUG_TYPE_MPXrP   )ZIMAGE_DEBUG_TYPE_REPROr.   )Z&IMAGE_DEBUG_TYPE_EX_DLLCHARACTERISTICS   ))ZIMAGE_SUBSYSTEM_UNKNOWNr   )IMAGE_SUBSYSTEM_NATIVEr0   )ZIMAGE_SUBSYSTEM_WINDOWS_GUIr;   )ZIMAGE_SUBSYSTEM_WINDOWS_CUIr=   )ZIMAGE_SUBSYSTEM_OS2_CUIrA   )ZIMAGE_SUBSYSTEM_POSIX_CUIrD   )IMAGE_SUBSYSTEM_NATIVE_WINDOWSrE   )ZIMAGE_SUBSYSTEM_WINDOWS_CE_GUIrG   )ZIMAGE_SUBSYSTEM_EFI_APPLICATIONrI   )Z'IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVERrK   )Z"IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVERrL   )ZIMAGE_SUBSYSTEM_EFI_ROMrN   )ZIMAGE_SUBSYSTEM_XBOXrO   )Z(IMAGE_SUBSYSTEM_WINDOWS_BOOT_APPLICATIONr.   ))ZIMAGE_FILE_MACHINE_UNKNOWNr   )IMAGE_FILE_MACHINE_I386iL  )ZIMAGE_FILE_MACHINE_R3000ib  )ZIMAGE_FILE_MACHINE_R4000if  )ZIMAGE_FILE_MACHINE_R10000ih  )ZIMAGE_FILE_MACHINE_WCEMIPSV2ii  )ZIMAGE_FILE_MACHINE_ALPHAi  )ZIMAGE_FILE_MACHINE_SH3i  )ZIMAGE_FILE_MACHINE_SH3DSPi  )ZIMAGE_FILE_MACHINE_SH3Ei  )ZIMAGE_FILE_MACHINE_SH4i  )ZIMAGE_FILE_MACHINE_SH5i  )ZIMAGE_FILE_MACHINE_ARMi  )ZIMAGE_FILE_MACHINE_THUMBi  )ZIMAGE_FILE_MACHINE_ARMNTi  )ZIMAGE_FILE_MACHINE_AM33i  )ZIMAGE_FILE_MACHINE_POWERPCi  )ZIMAGE_FILE_MACHINE_POWERPCFPi  )IMAGE_FILE_MACHINE_IA64r   )ZIMAGE_FILE_MACHINE_MIPS16if  )ZIMAGE_FILE_MACHINE_ALPHA64  )ZIMAGE_FILE_MACHINE_AXP64rb   )ZIMAGE_FILE_MACHINE_MIPSFPUif  )ZIMAGE_FILE_MACHINE_MIPSFPU16if  )ZIMAGE_FILE_MACHINE_TRICOREi   )ZIMAGE_FILE_MACHINE_CEFi  )ZIMAGE_FILE_MACHINE_EBCi  )IMAGE_FILE_MACHINE_AMD64id  )ZIMAGE_FILE_MACHINE_M32RiA  )ZIMAGE_FILE_MACHINE_ARM64id  )ZIMAGE_FILE_MACHINE_CEEi  ))IMAGE_REL_BASED_ABSOLUTEr   )IMAGE_REL_BASED_HIGHr0   )IMAGE_REL_BASED_LOWr;   )IMAGE_REL_BASED_HIGHLOWr=   )IMAGE_REL_BASED_HIGHADJr?   )ZIMAGE_REL_BASED_MIPS_JMPADDRrA   )ZIMAGE_REL_BASED_SECTIONrC   )ZIMAGE_REL_BASED_RELrD   )ZIMAGE_REL_BASED_MIPS_JMPADDR16rG   )ZIMAGE_REL_BASED_IA64_IMM64rG   )IMAGE_REL_BASED_DIR64rI   )ZIMAGE_REL_BASED_HIGH3ADJrK   ))ZIMAGE_LIBRARY_PROCESS_INITr0   )ZIMAGE_LIBRARY_PROCESS_TERMr;   )ZIMAGE_LIBRARY_THREAD_INITr?   )ZIMAGE_LIBRARY_THREAD_TERMrE   )Z(IMAGE_DLLCHARACTERISTICS_HIGH_ENTROPY_VAr-   )Z%IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASErR   )Z(IMAGE_DLLCHARACTERISTICS_FORCE_INTEGRITYr	   )Z"IMAGE_DLLCHARACTERISTICS_NX_COMPATrS   )Z%IMAGE_DLLCHARACTERISTICS_NO_ISOLATIONr   )ZIMAGE_DLLCHARACTERISTICS_NO_SEHrT   )Z IMAGE_DLLCHARACTERISTICS_NO_BINDr   )Z%IMAGE_DLLCHARACTERISTICS_APPCONTAINERr"   )Z#IMAGE_DLLCHARACTERISTICS_WDM_DRIVERr+   )Z!IMAGE_DLLCHARACTERISTICS_GUARD_CFrV   )Z.IMAGE_DLLCHARACTERISTICS_TERMINAL_SERVER_AWAREr,   ))UNW_FLAG_EHANDLERr0   )UNW_FLAG_UHANDLERr;   )UNW_FLAG_CHAININFOr?   ))ZRAXr   )ZRCXr0   )ZRDXr;   )ZRBXr=   )ZRSPr?   )ZRBPrA   )ZRSIrC   )ZRDIrD   )ZR8rE   )ZR9rG   )ZR10rI   )ZR11rK   )ZR12rL   )ZR13rN   )ZR14rO   )ZR15rP   r0   r;   r=   r?   rA   rC   rE   rG   rI   ))Z	RT_CURSORr0   )Z	RT_BITMAPr;   )ZRT_ICONr=   )ZRT_MENUr?   )Z	RT_DIALOGrA   )	RT_STRINGrC   )Z
RT_FONTDIRrD   )ZRT_FONTrE   )ZRT_ACCELERATORrG   )Z	RT_RCDATArI   )ZRT_MESSAGETABLErK   )ZRT_GROUP_CURSORrL   )ZRT_GROUP_ICONrO   )
RT_VERSIONr.   )ZRT_DLGINCLUDE   )ZRT_PLUGPLAY   )ZRT_VXDr]   )ZRT_ANICURSOR   )Z
RT_ANIICON   )ZRT_HTML   )ZRT_MANIFEST   )^)ZLANG_NEUTRALr   )ZLANG_INVARIANT   )ZLANG_AFRIKAANS6   )ZLANG_ALBANIAN   )ZLANG_ARABICr0   )ZLANG_ARMENIAN+   )ZLANG_ASSAMESEM   )Z
LANG_AZERI,   )ZLANG_BASQUE-   )ZLANG_BELARUSIAN#   )ZLANG_BENGALIE   )ZLANG_BULGARIANr;   )ZLANG_CATALANr=   )ZLANG_CHINESEr?   )ZLANG_CROATIAN   )Z
LANG_CZECHrA   )ZLANG_DANISHrC   )ZLANG_DIVEHIe   )Z
LANG_DUTCHrp   )ZLANG_ENGLISHrG   )ZLANG_ESTONIAN%   )ZLANG_FAEROESE8   )Z
LANG_FARSI)   )ZLANG_FINNISHrK   )ZLANG_FRENCHrL   )ZLANG_GALICIANV   )ZLANG_GEORGIAN7   )ZLANG_GERMANrD   )Z
LANG_GREEKrE   )ZLANG_GUJARATIG   )ZLANG_HEBREWrN   )Z
LANG_HINDI9   )ZLANG_HUNGARIANrO   )ZLANG_ICELANDICrP   )ZLANG_INDONESIAN!   )ZLANG_ITALIANr.   )ZLANG_JAPANESEro   )ZLANG_KANNADAK   )ZLANG_KASHMIRI`   )Z
LANG_KAZAK?   )ZLANG_KONKANIW   )ZLANG_KOREAN   )ZLANG_KYRGYZrR   )ZLANG_LATVIAN&   )ZLANG_LITHUANIAN'   )ZLANG_MACEDONIAN/   )Z
LANG_MALAY>   )ZLANG_MALAYALAML   )ZLANG_MANIPURIX   )ZLANG_MARATHIN   )ZLANG_MONGOLIANP   )ZLANG_NEPALIa   )ZLANG_NORWEGIANr]   )Z
LANG_ORIYAH   )ZLANG_POLISHrq   )ZLANG_PORTUGUESErr   )ZLANG_PUNJABIF   )ZLANG_ROMANIANrt   )ZLANG_RUSSIAN   )ZLANG_SANSKRITO   )ZLANG_SERBIANr~   )ZLANG_SINDHIY   )ZLANG_SLOVAK   )ZLANG_SLOVENIAN$   )ZLANG_SPANISHrI   )ZLANG_SWAHILIA   )ZLANG_SWEDISH   )ZLANG_SYRIACZ   )Z
LANG_TAMILI   )Z
LANG_TATARD   )ZLANG_TELUGUJ   )Z	LANG_THAI   )ZLANG_TURKISH   )ZLANG_UKRAINIAN"   )Z	LANG_URDUr-   )Z
LANG_UZBEKC   )ZLANG_VIETNAMESE*   )ZLANG_GAELIC<   )ZLANG_MALTESE:   )Z
LANG_MAORI(   )ZLANG_RHAETO_ROMANCErs   )Z
LANG_SAAMI;   )ZLANG_SORBIAN.   )Z	LANG_SUTU0   )ZLANG_TSONGA1   )ZLANG_TSWANA2   )Z
LANG_VENDA3   )Z
LANG_XHOSA4   )Z	LANG_ZULU5   )ZLANG_ESPERANTO   )Z
LANG_WALON   )ZLANG_CORNISH   )Z
LANG_WELSH   )ZLANG_BRETON   )g)ZSUBLANG_NEUTRALr   )ZSUBLANG_DEFAULTr0   )ZSUBLANG_SYS_DEFAULTr;   )ZSUBLANG_ARABIC_SAUDI_ARABIAr0   )ZSUBLANG_ARABIC_IRAQr;   )ZSUBLANG_ARABIC_EGYPTr=   )ZSUBLANG_ARABIC_LIBYAr?   )ZSUBLANG_ARABIC_ALGERIArA   )ZSUBLANG_ARABIC_MOROCCOrC   )ZSUBLANG_ARABIC_TUNISIArD   )ZSUBLANG_ARABIC_OMANrE   )ZSUBLANG_ARABIC_YEMENrG   )ZSUBLANG_ARABIC_SYRIArI   )ZSUBLANG_ARABIC_JORDANrK   )ZSUBLANG_ARABIC_LEBANONrL   )ZSUBLANG_ARABIC_KUWAITrN   )ZSUBLANG_ARABIC_UAErO   )ZSUBLANG_ARABIC_BAHRAINrP   )ZSUBLANG_ARABIC_QATARr.   )ZSUBLANG_AZERI_LATINr0   )ZSUBLANG_AZERI_CYRILLICr;   )ZSUBLANG_CHINESE_TRADITIONALr0   )ZSUBLANG_CHINESE_SIMPLIFIEDr;   )ZSUBLANG_CHINESE_HONGKONGr=   )ZSUBLANG_CHINESE_SINGAPOREr?   )ZSUBLANG_CHINESE_MACAUrA   )ZSUBLANG_DUTCHr0   )ZSUBLANG_DUTCH_BELGIANr;   )ZSUBLANG_ENGLISH_USr0   )ZSUBLANG_ENGLISH_UKr;   )ZSUBLANG_ENGLISH_AUSr=   )ZSUBLANG_ENGLISH_CANr?   )ZSUBLANG_ENGLISH_NZrA   )ZSUBLANG_ENGLISH_EIRErC   )ZSUBLANG_ENGLISH_SOUTH_AFRICArD   )ZSUBLANG_ENGLISH_JAMAICArE   )ZSUBLANG_ENGLISH_CARIBBEANrG   )ZSUBLANG_ENGLISH_BELIZErI   )ZSUBLANG_ENGLISH_TRINIDADrK   )ZSUBLANG_ENGLISH_ZIMBABWErL   )ZSUBLANG_ENGLISH_PHILIPPINESrN   )ZSUBLANG_FRENCHr0   )ZSUBLANG_FRENCH_BELGIANr;   )ZSUBLANG_FRENCH_CANADIANr=   )ZSUBLANG_FRENCH_SWISSr?   )ZSUBLANG_FRENCH_LUXEMBOURGrA   )ZSUBLANG_FRENCH_MONACOrC   )ZSUBLANG_GERMANr0   )ZSUBLANG_GERMAN_SWISSr;   )ZSUBLANG_GERMAN_AUSTRIANr=   )ZSUBLANG_GERMAN_LUXEMBOURGr?   )ZSUBLANG_GERMAN_LIECHTENSTEINrA   )ZSUBLANG_ITALIANr0   )ZSUBLANG_ITALIAN_SWISSr;   )ZSUBLANG_KASHMIRI_SASIAr;   )ZSUBLANG_KASHMIRI_INDIAr;   )ZSUBLANG_KOREANr0   )ZSUBLANG_LITHUANIANr0   )ZSUBLANG_MALAY_MALAYSIAr0   )ZSUBLANG_MALAY_BRUNEI_DARUSSALAMr;   )ZSUBLANG_NEPALI_INDIAr;   )ZSUBLANG_NORWEGIAN_BOKMALr0   )ZSUBLANG_NORWEGIAN_NYNORSKr;   )ZSUBLANG_PORTUGUESEr;   )ZSUBLANG_PORTUGUESE_BRAZILIANr0   )ZSUBLANG_SERBIAN_LATINr;   )ZSUBLANG_SERBIAN_CYRILLICr=   )ZSUBLANG_SPANISHr0   )ZSUBLANG_SPANISH_MEXICANr;   )ZSUBLANG_SPANISH_MODERNr=   )ZSUBLANG_SPANISH_GUATEMALAr?   )ZSUBLANG_SPANISH_COSTA_RICArA   )ZSUBLANG_SPANISH_PANAMArC   )Z"SUBLANG_SPANISH_DOMINICAN_REPUBLICrD   )ZSUBLANG_SPANISH_VENEZUELArE   )ZSUBLANG_SPANISH_COLOMBIArG   )ZSUBLANG_SPANISH_PERUrI   )ZSUBLANG_SPANISH_ARGENTINArK   )ZSUBLANG_SPANISH_ECUADORrL   )ZSUBLANG_SPANISH_CHILErN   )ZSUBLANG_SPANISH_URUGUAYrO   )ZSUBLANG_SPANISH_PARAGUAYrP   )ZSUBLANG_SPANISH_BOLIVIAr.   )ZSUBLANG_SPANISH_EL_SALVADORro   )ZSUBLANG_SPANISH_HONDURASr   )ZSUBLANG_SPANISH_NICARAGUArp   )ZSUBLANG_SPANISH_PUERTO_RICOr]   )ZSUBLANG_SWEDISHr0   )ZSUBLANG_SWEDISH_FINLANDr;   )ZSUBLANG_URDU_PAKISTANr0   )ZSUBLANG_URDU_INDIAr;   )ZSUBLANG_UZBEK_LATINr0   )ZSUBLANG_UZBEK_CYRILLICr;   )ZSUBLANG_DUTCH_SURINAMr=   )ZSUBLANG_ROMANIANr0   )ZSUBLANG_ROMANIAN_MOLDAVIAr;   )ZSUBLANG_RUSSIANr0   )ZSUBLANG_RUSSIAN_MOLDAVIAr;   )ZSUBLANG_CROATIANr0   )ZSUBLANG_LITHUANIAN_CLASSICr;   )ZSUBLANG_GAELICr0   )ZSUBLANG_GAELIC_SCOTTISHr;   )ZSUBLANG_GAELIC_MANXr=   c                 C   s@   t | d}t|g D ]}||v r|  S qt|dgd S )N	*unknown*r   )LANGgetSUBLANG)Z
lang_valuesublang_valueZ	lang_namesublang_namer   r   r   get_sublang_name_for_lang  s
    
r   c                 C   s   d}d}|t | k r| ||d  }t |dk r2qtd|d }|d7 }|dkrd|d   krnt | krn nXz&t| |||d   d||< W n ty   |d7 }Y n0 |dkrq||d 7 }|d7 }qd S )Nr   r;   z<hutf-16ler0   r=   )lenstructunpackbdecodeUnicodeDecodeError)r(   counterlierror_countZ
data_sliceZlen_r   r   r   parse_strings  s$    (&r   c                    s    fdd   D S )zRead the flags from a dictionary and return them in a usable form.

    Will return a list of (flag, value) for all flags in "flag_dict"
    matching the filter "flag_filter".
    c                    s0   g | ](}t |ttfr|r| | fqS r   )
isinstancestrbytes
startswith)r1   flagZ	flag_dictZflag_filterr   r   r3     s   z"retrieve_flags.<locals>.<listcomp>)keysr   r   r   r   retrieve_flags  s    r   c                 C   s0   |D ]&\}}||@ r d| j |< qd| j |< qdS )a
  Will process the flags and set attributes in the object accordingly.

    The object "obj" will gain attributes named after the flags provided in
    "flags" and valued True/False, matching the results of applying each
    flag value from "flags" to flag_field.
    TFN)__dict__)objZ
flag_fieldflagsr   valuer   r   r   	set_flags  s    r   c                 C   s   | dko| | d @ dkS )Nr   r0   r   )r   r   r   r   power_of_two  s    r   c                 C   s"   t | ttfrt| S t| dS )Ncp1252)r   r   	bytearraycodecsencode)xr   r   r   r     s    r   c                   @   s`   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dd Z
dd Zdd Zdd ZdS )!UnicodeStringWrapperPostProcessorzThis class attempts to help the process of identifying strings
    that might be plain Unicode or Pascal. A list of strings will be
    wrapped on it with the hope the overlappings will help make the
    decision about their type.c                 C   s   || _ || _d | _d S r
   )perva_ptrstring)selfr   r   r   r   r   __init__  s    z*UnicodeStringWrapperPostProcessor.__init__c                 C   s   | j S )zGet the RVA of the string.)r   r   r   r   r   get_rva  s    z)UnicodeStringWrapperPostProcessor.get_rvac                 C   s   |  ddS )z6Return the escaped UTF-8 representation of the string.utf-8r   )r   r   r   r   r   __str__  s    z)UnicodeStringWrapperPostProcessor.__str__c                 G   s   | j s
dS | j j| S )N )r   r   )r   r   r   r   r   r     s    z(UnicodeStringWrapperPostProcessor.decodec                 C   s   d} dS )z>Make this instance None, to express it's no known string type.Nr   r   r   r   r   
invalidate  s    z,UnicodeStringWrapperPostProcessor.invalidatec              	   C   sT   z | j j| jd |  d| _W n. tyN   | j  d| jd  Y n0 d S )Nr;   
max_lengthzCFailed rendering pascal string, attempting to read from RVA 0x{0:x})	r   get_string_u_at_rvar   get_pascal_16_lengthr   PEFormatErrorget_warningsappendformatr   r   r   r   render_pascal_16  s    
z2UnicodeStringWrapperPostProcessor.render_pascal_16c                 C   s   |  | jS r
   )9_UnicodeStringWrapperPostProcessor__get_word_value_at_rvar   r   r   r   r   r     s    z6UnicodeStringWrapperPostProcessor.get_pascal_16_lengthc                 C   sH   z| j |d}W n ty&   Y dS 0 t|dk r8dS td|d S )Nr;   F<Hr   )r   get_datar   r   r   r   )r   rvar(   r   r   r   Z__get_word_value_at_rva  s    z9UnicodeStringWrapperPostProcessor.__get_word_value_at_rvac                 C   s&   |  |d dkr"|| j | _dS dS )zThe next RVA is taken to be the one immediately following this one.

        Such RVA could indicate the natural end of the string and will be checked
        to see if there's a Unicode NULL character there.
        r;   r   TF)r   r   length)r   Znext_rva_ptrr   r   r   ask_unicode_16  s    z0UnicodeStringWrapperPostProcessor.ask_unicode_16c                 C   sD   z| j | j| _W n* ty>   | j  d| j Y n0 d S )NzDFailed rendering unicode string, attempting to read from RVA 0x{0:x})r   r   r   r   r   r   r   r   r   r   r   r   render_unicode_16  s    
z3UnicodeStringWrapperPostProcessor.render_unicode_16N)__name__
__module____qualname____doc__r   r   r   r   r   r   r   r   r   r   r   r   r   r   r     s   r   c                   @   s    e Zd ZdZdd Zdd ZdS )r   z"Generic PE format error exception.c                 C   s
   || _ d S r
   )r   )r   r   r   r   r   r   *  s    zPEFormatError.__init__c                 C   s
   t | jS r
   )reprr   r   r   r   r   r   -  s    zPEFormatError.__str__N)r   r   r   r   r   r   r   r   r   r   r   '  s   r   c                   @   sN   e Zd ZdZdd ZdddZdddZdd	d
Zdd Zdd Z	dd Z
dS )Dumpz1Convenience class for dumping the PE information.c                 C   s
   g | _ d S r
   )textr   r   r   r   r   4  s    zDump.__init__r   c                 C   s   |D ]}|  || qdS )zeAdds a list of lines.

        The list can be indented with the optional argument 'indent'.
        Nadd_line)r   txtindentliner   r   r   	add_lines7  s    zDump.add_linesc                 C   s   |  |d | dS )z\Adds a line.

        The line can be indented with the optional argument 'indent'.
        
N)addr   r  r  r   r   r   r  ?  s    zDump.add_linec                 C   s   | j dd| | dS )z|Adds some text, no newline will be appended.

        The text can be indented with the optional argument 'indent'.
        z{0}{1} N)r   r   r   r	  r   r   r   r  F  s    zDump.addc                 C   s   |  dd| dS )zAdds a header element.z
{0}{1}{0}
z
----------N)r  r   )r   r  r   r   r   
add_headerM  s    zDump.add_headerc                 C   s   | j d dS )zAdds a newline.r  N)r   r   r   r   r   r   add_newlineQ  s    zDump.add_newlinec                 C   s   d dd | jD S )z"Get the text in its current state.r   c                 s   s   | ]}d  |V  qdS ){0}Nr   )r1   r   r   r   r   	<genexpr>W  r4   z Dump.get_text.<locals>.<genexpr>)joinr   r   r   r   r   get_textU  s    zDump.get_textN)r   )r   )r   )r   r   r   r   r   r  r  r  r  r  r  r   r   r   r   r   1  s   


r   )r   cr   BhHr   Ir   Lr   qQdsc                 C   sN   d}| }| d t jv rBtddd | D }ddd | D }t| | S )Nr0   r   r   c                 S   s   g | ]}|t jv r|qS r   r   digitsr1   r  r   r   r   r3   s  r4   zsizeof_type.<locals>.<listcomp>c                 S   s   g | ]}|t jvr|qS r   r  r  r   r   r   r3   t  r4   )r   r  r   r  STRUCT_SIZEOF_TYPES)tr&   Z_tr   r   r   sizeof_typem  s    r!  T)r   r   c                    s   d}g }i }g }d}d}| D ]}d|v r| dd\} ||7 }|d    d}	g }
|	D ]F  |v r fdd|D }| }d | |
  || < q\|t|7 }||
 qt|}|||||fS )N<r   ,r0   c                    s   g | ]}|d t   qS r
   )r   )r1   r   elm_namer   r   r3     r4   zset_format.<locals>.<listcomp>z	{0}_{1:d})splitr   r&   r   r!  r   calcsize)r   
__format____unpacked_data_elms____field_offsets____keys____format_length__offsetelmelm_typeZ	elm_namesnamesZsearch_listZ	occ_countr   r$  r   
set_formatx  s:    





r1  c                   @   s   e Zd ZdZd ddZdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zd!ddZdd ZdS )"	StructurezPrepare structure object to extract members from data.

    Format is a list containing definitions for the elements
    of the structure.
    Nc                 C   s   d| _ g | _d| _i | _g | _|d }t|d ts@t|d }t|\| _ | _| _| _| _d| _|| _	|rt|| _
n
|d | _
d S )Nr"  r   r0   F)r(  r+  r,  r*  r)  r   tupler1  __all_zeroes____file_offset__name)r   r   r6  file_offsetr  r   r   r   r     s(    zStructure.__init__c                 C   s   | j S r
   )r(  r   r   r   r   __get_format__  s    zStructure.__get_format__c                 C   s   | j | j|  S )zLReturn the offset within the field for the requested field in the structure.)r5  r*  r   
field_namer   r   r   get_field_absolute_offset  s    z#Structure.get_field_absolute_offsetc                 C   s
   | j | S )z?Return the offset within the structure for the requested field.)r*  r9  r   r   r   get_field_relative_offset  s    z#Structure.get_field_relative_offsetc                 C   s   | j S r
   r5  r   r   r   r   get_file_offset  s    zStructure.get_file_offsetc                 C   s
   || _ d S r
   r=  r   r-  r   r   r   set_file_offset  s    zStructure.set_file_offsetc                 C   s   | j S )z/Returns true is the unpacked data is all zeros.)r4  r   r   r   r   
all_zeroes  s    zStructure.all_zeroesc                 C   s   | j S )zReturn size of the structure.)r,  r   r   r   r   sizeof  s    zStructure.sizeofc                 C   s   t |}t|| jkr&|d | j }nt|| jk r<tdt|t|krRd| _t| j|| _	t
| j	D ]$\}}| j| D ]}t| || q~qld S )Nz-Data length less than expected header length.T)r   r   r,  r   r)   r4  r   r   r(  r)  	enumerater+  setattr)r   r(   idxr   keyr   r   r   
__unpack__  s    zStructure.__unpack__c                 C   s\   g }t | jD ]8\}}| j| D ]}t| |}||kr  q<q || qtj| jg|R  S r
   )rC  r)  r+  getattrr   r   packr(  )r   
new_valuesrE  r   rF  new_valr   r   r   __pack__  s    
zStructure.__pack__c                 C   s   d |  S )Nr  r  dumpr   r   r   r   r     s    zStructure.__str__c                 C   s   dd dd |  D  S )Nz<Structure: %s>r
  c                 S   s   g | ]}d  | qS )r
  )r  r&  r1   r  r   r   r   r3     r4   z&Structure.__repr__.<locals>.<listcomp>rM  r   r   r   r   __repr__  s    zStructure.__repr__r   c              
      s<  g }| d| j dd tjD  | jD ]}|D ]}t| |}t|tt	fr|
drhd|}n
d|}|dks|dkrz|d	tt| 7 }W n ty   |d
7 }Y n0 nLt|}|
drddd |dD }nd fdd|dD }| d| j| | j | j| |d |f  q6q,|S )z1Returns a string representation of the structure.z[{0}]c                 S   s   g | ]}|t jvrt|qS r   )r   
whitespaceordr1   r   r   r   r   r3     s   z"Structure.dump.<locals>.<listcomp>Z
Signature_z{:<8X}z0x{:<8X}TimeDateStampdwTimeStampz	 [%s UTC]z [INVALID TIME]	Signaturer   c                 S   s   g | ]}d  |qS )z{:02X}r  rS  r   r   r   r3   2  r4       c                    s&   g | ]}| v rt |nd |qS )z	\x{0:02x})chrr   rS  Zprintable_bytesr   r   r3   6  s   
z0x%-8X 0x%-3X %-30s %s:)r   r   r6  r   	printabler+  rH  r   r   longr   timeasctimegmtime
ValueErrorr   r  rstripr*  r5  )r   indentationrN  r   rF  r   Zval_strr   rY  r   rN    sJ    




	
zStructure.dumpc              
   C   s   i }| j |d< | jD ]}|D ]}t| |}t|ttfr|dksH|dkrzd|tt|f }W q t	y~   d| }Y q0 nd
dd d	d
 |D D }| j| | j | j| |d||< qq|S )z5Returns a dictionary representation of the structure.r2  rT  rU  z0x%-8X [%s UTC]z0x%-8X [INVALID TIME]r   c                 s   s,   | ]$}t |tjv rt |nd | V  qdS )z\x%02xNrX  r   r[  r  r   r   r   r  a  s   z&Structure.dump_dict.<locals>.<genexpr>c                 S   s"   g | ]}t |tst|n|qS r   r   r   rR  r1   r  r   r   r   r3   c  r4   z'Structure.dump_dict.<locals>.<listcomp>)Z
FileOffsetOffsetValue)r6  r+  rH  r   r   r\  r]  r^  r_  r`  r  r*  r5  )r   	dump_dictr   rF  r   r   r   r   rh  J  s,    




zStructure.dump_dict)NN)r   )r   r   r   r   r   r8  r;  r<  r>  r@  rA  rB  rG  rL  r   rP  rN  rh  r   r   r   r   r2    s   

7r2  c                   @   s   e Zd ZdZdd Zdd Zdd Zd#d	d
Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" ZdS )$SectionStructurez#Convenience section handling class.c                 O   s@   d|v r|d | _ |d= tj| g|R i | d | _d | _d S )Nr   )r   r2  r   PointerToRawData_adjVirtualAddress_adj)r   ZarglZargdr   r   r   r   r  s    
zSectionStructure.__init__c                 C   s2   | j d u r,| jd ur,| j| j| jjj| _ | j S r
   )rj  PointerToRawDatar   adjust_FileAlignmentOPTIONAL_HEADERFileAlignmentr   r   r   r   get_PointerToRawData_adj{  s    

z)SectionStructure.get_PointerToRawData_adjc                 C   s:   | j d u r4| jd ur4| j| j| jjj| jjj| _ | j S r
   )rk  VirtualAddressr   adjust_SectionAlignmentrn  SectionAlignmentro  r   r   r   r   get_VirtualAddress_adj  s    

z'SectionStructure.get_VirtualAddress_adjNc                 C   sn   |du r|   }n||   |    }|dur8|| }n
|| j }|| j| j kr^| j| j }| jj|| S )a\  Get data chunk from a section.

        Allows to query data from the section by passing the
        addresses where the PE file would be loaded by default.
        It is then possible to retrieve code and data by their real
        addresses as they would be if loaded.

        Returns bytes() under Python 3.x and set() under Python 2.7
        N)rp  rt  SizeOfRawDatarl  r   __data__)r   startr   r-  endr   r   r   r     s    



zSectionStructure.get_datac                 C   sr   |dkr t td}t| || nDd|v rdt| |rd|rN| jd  t| O  < n| jd  t| N  < || j|< d S )NCharacteristics
IMAGE_SCN_)r   SECTION_CHARACTERISTICSr   hasattrr   )r   r6  r   section_flagsr   r   r   __setattr__  s    
zSectionStructure.__setattr__c                 C   s   ||    |   S r
   )rp  rt  r?  r   r   r   get_rva_from_offset  s    z$SectionStructure.get_rva_from_offsetc                 C   s   ||    |   S r
   )rt  rp  r   r   r   r   r   get_offset_from_rva  s    z$SectionStructure.get_offset_from_rvac                 C   s4   | j du rdS |  }||  ko.|| j k S   S )z<Check whether the section contains the file offset provided.NF)rl  rp  ru  )r   r-  rj  r   r   r   contains_offset  s
    
z SectionStructure.contains_offsetc                 C   s   |   }t| jj|   | jk r*| j}nt| j| j}| jdurf| j| j	krf|| | jkrf| j| }||  ko||| k S   S )z8Check whether the section contains the address provided.N)
rt  r   r   rv  rp  ru  Misc_VirtualSizemaxnext_section_virtual_addressrq  )r   r   rk  sizer   r   r   contains_rva  s    

zSectionStructure.contains_rvac                 C   s
   |  |S r
   )r  r  r   r   r   contains  s    zSectionStructure.containsc                 C   s   |  |  S )z1Calculate and return the entropy for the section.)	entropy_Hr   r   r   r   r   get_entropy  s    zSectionStructure.get_entropyc                 C   s   t durt |   S dS )z/Get the SHA-1 hex-digest of the section's data.N)r   r   	hexdigestr   r   r   r   get_hash_sha1  s    zSectionStructure.get_hash_sha1c                 C   s   t durt |   S dS )z1Get the SHA-256 hex-digest of the section's data.N)r   r   r  r   r   r   r   get_hash_sha256  s    z SectionStructure.get_hash_sha256c                 C   s   t durt |   S dS )z1Get the SHA-512 hex-digest of the section's data.N)r   r   r  r   r   r   r   get_hash_sha512   s    z SectionStructure.get_hash_sha512c                 C   s   t durt |   S dS )z-Get the MD5 hex-digest of the section's data.N)r   r   r  r   r   r   r   get_hash_md5  s    zSectionStructure.get_hash_md5c                 C   sN   |sdS t t|}d}| D ](}t|t| }||t|d 8 }q |S )z)Calculate the entropy of a chunk of data.g        r   r;   )r   r   valuesfloatr   mathlog)r   r(   Z
occurencesZentropyr   Zp_xr   r   r   r    s    zSectionStructure.entropy_H)NN)r   r   r   r   r   rp  rt  r   r~  r  r  r  r  r  r  r  r  r  r  r  r   r   r   r   ri  o  s"   	

 ri  c                 C   sT  G dd d}g }i }|||}| d D ]}d|vrH|   || q(|dd\}}d|v rhtd|dd\}}t|}|| ks|| kr|   || ||| q(|   t	t
|\}	}
}}}g }t|D ]`\}}||vr|| q|| \}
}dd |D }|| |D ]}||d	  ||d	 < q&q|	|||||fS )
Nc                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )z)set_bitfields_format.<locals>.Accumulatorc                 S   s(   g | _ d| _d | _d| _|| _|| _d S )N~r   )
_subfields_name_type
_bits_left_comp_fields_format)r   fmtcomp_fieldsr   r   r   r     s    z2set_bitfields_format.<locals>.Accumulator.__init__c                 S   sX   | j d krd S | j| j d | j  | j | jf| jt| jd < d| _d | _ g | _d S )Nr#  r0   r  )r  r  r   r  r  r  r   r   r   r   r   wrap_up)  s    
z1set_bitfields_format.<locals>.Accumulator.wrap_upc                 S   s   t | d | _|| _d S NrE   )r  r  r  )r   tpr   r   r   new_type2  s    z2set_bitfields_format.<locals>.Accumulator.new_typec                 S   s0   |  j |7  _ |  j|8  _| j||f d S r
   )r  r  r  r   )r   r6  Zbitcntr   r   r   add_subfield6  s    z6set_bitfields_format.<locals>.Accumulator.add_subfieldc                 S   s   | j S r
   )r  r   r   r   r   get_type;  s    z2set_bitfields_format.<locals>.Accumulator.get_typec                 S   s   | j S r
   )r  r   r   r   r   get_name>  s    z2set_bitfields_format.<locals>.Accumulator.get_namec                 S   s   | j S r
   )r  r   r   r   r   get_bits_leftA  s    z7set_bitfields_format.<locals>.Accumulator.get_bits_leftN)
r   r   r   r   r  r  r  r  r  r  r   r   r   r   Accumulator  s   
	r  r0   rZ  r#  z3Structures with bitfields do not support unions yetc                 S   s   g | ]}|t j gqS r   )StructureWithBitfieldsBTF_NAME_IDX)r1   r   r   r   r   r3   f  r4   z(set_bitfields_format.<locals>.<listcomp>r   )r  r   r&  NotImplementedErrorr   r  r  r  r  r1  r3  rC  extend)r   r  Zold_fmtr  Zacr.  r/  r%  Zelm_bitsZ
format_str_Zfield_offsetsr   Zformat_lengthZextended_keysrE  r   ZsbfZbf_namesnr   r   r   set_bitfields_format  sB    &




r  c                       sp   e Zd ZdZdZdZdZdZdddZ fddZ	 fd	d
Z
d fdd	Z fddZdd Zdd Z  ZS )r  aV  
    Extends Structure's functionality with support for bitfields such as:
        ('B:4,LowerHalf', 'B:4,UpperHalf')
    To this end, two lists are maintained:
        * self.__keys__ that contains compound fields, for example
          ('B,~LowerHalfUpperHalf'), and is used during packing/unpaking
        * self.__keys_ext__ containing a separate key for each field (ex., LowerHalf,
          UpperHalf) to simplify implementation of dump()
    This way the implementation of unpacking/packing and dump() from Structure can be
    reused.

    In addition, we create a dictionary:
        <comound_field_index_in_keys> -->
            (data type, [ (subfield name, length in bits)+ ] )
    that facilitates bitfield paking and unpacking.

    With lru_cache() creating only once instance per format string, the memory
    overhead is negligible.
    r   r0   Nc                 C   s\   t |\| _| _| _| _| _| _dd t| jD | _d| _	|| _
|d krN|n|d | _d S )Nc                 S   s   g | ]}d qS r
   r   rS  r   r   r   r3     r4   z3StructureWithBitfields.__init__.<locals>.<listcomp>Fr   )r  r(  r,  r*  r+  __keys_ext____compound_fields__ranger)  r4  r5  r6  )r   r   r6  r7  r   r   r   r     s    zStructureWithBitfields.__init__c                    s   t t| | |   d S r
   )superr  rG  _unpack_bitfield_attributesr   r(   	__class__r   r   rG    s    z!StructureWithBitfields.__unpack__c                    s2   |    ztt|  }W |   n
|   0 |S r
   )_pack_bitfield_attributesr  r  rL  r  r  r  r   r   rL    s
    zStructureWithBitfields.__pack__c                    s6   | j }| j| _ ztt| |}W || _ n|| _ 0 |S r
   )r+  r  r  r  rN  )r   rb  tkretr  r   r   rN    s    zStructureWithBitfields.dumpc                    s4   | j }| j| _ ztt|  }W || _ n|| _ 0 |S r
   )r+  r  r  r  rh  )r   r  r  r  r   r   rh    s    z StructureWithBitfields.dump_dictc                 C   s   | j  D ]}| j| d }t| |}t| | d}| j | tj D ]F}d|tj > d }||K }t| |tj	 ||@ |?  ||tj 7 }qDq
dS )zaReplace compound attributes corresponding to bitfields with separate
        sub-fields.
        r   r0   N)
r  r   r+  rH  delattrr  CF_SUBFLD_IDXBTF_BITCNT_IDXrD  r  )r   r   cf_nameZcvaloffstsfmaskr   r   r   r    s    


z2StructureWithBitfields._unpack_bitfield_attributesc                 C   s   | j  D ]|}| j| d }d\}}| j | tj D ]D}d|tj > d }t| |tj |@ }|||> O }||tj 7 }q4t| || q
dS )z(Pack attributes into a compound bitfieldr   r   r   r0   N)	r  r   r+  r  r  r  rH  r  rD  )r   r   r  r  Zacc_valr  r  Z	field_valr   r   r   r    s    z0StructureWithBitfields._pack_bitfield_attributes)NN)r   )r   r   r   r   r  r  ZCF_TYPE_IDXr  r   rG  rL  rN  rh  r  r  __classcell__r   r   r  r   r  n  s   
		r  c                       s    e Zd ZdZ fddZ  ZS )DataContainerzGeneric data container.c                    s0   t t| j}t| D ]\}}||| qd S r
   )r  r  r~  listitems)r   r   Zbare_setattrrF  r   r  r   r   r     s    zDataContainer.__init__)r   r   r   r   r   r  r   r   r  r   r    s   r  c                   @   s   e Zd ZdZdS )ImportDescDatazHolds import descriptor information.

    dll:        name of the imported DLL
    imports:    list of imported symbols (ImportData instances)
    struct:     IMAGE_IMPORT_DESCRIPTOR structure
    Nr   r   r   r   r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdd ZdS )
ImportDatazHolds imported symbol's information.

    ordinal:    Ordinal of the symbol
    name:       Name of the symbol
    bound:      If the symbol is bound, this contains
                the address.
    c                 C   sh  t | drZt | drZt | drZ|dkr| jjtkr>t}n| jjtkrNt}||d@ B | j_| jj| j_	| jj| j_
| jj| j_n|dkr| jd ur|| j_	| jj	| j_	| jj	| j_
| jj	| j_n|dkr|| j_	| jj	| j_| jj	| j_
| jj	| j_n`|dkrZ| jrZ| j| j}| j| jd|B  t|t| jkrJtd| j| j| || j|< d S )Nordinalboundr6    addressr   9The export name provided is longer than the existing one.)r|  r   PE_TYPEOPTIONAL_HEADER_MAGIC_PEIMAGE_ORDINAL_FLAGOPTIONAL_HEADER_MAGIC_PE_PLUSIMAGE_ORDINAL_FLAG64struct_tableOrdinalAddressOfDataFunctionForwarderString
struct_iatname_offsetr  set_dword_at_offsetordinal_offsetr   r6  r   set_bytes_at_offsetr   )r   r6  r   ordinal_flagZname_rvar   r   r   r~    sL    


zImportData.__setattr__Nr   r   r   r   r~  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )ExportDirDatazHolds export directory information.

    struct:     IMAGE_EXPORT_DIRECTORY structure
    symbols:    list of exported symbols (ExportData instances)Nr  r   r   r   r   r  .  s   r  c                   @   s   e Zd ZdZdd ZdS )
ExportDataad  Holds exported symbols' information.

    ordinal:    ordinal of the symbol
    address:    address of the symbol
    name:       name of the symbol (None if the symbol is
                exported by ordinal only)
    forwarder:  if the symbol is forwarded it will
                contain the name of the target symbol,
                None otherwise.
    c                 C   s   t | drt | drt | drt | dr|dkrB| j| j| n|dkr\| j| j| nf|dkrt|t| jkr~td| j	| j
| n2|dkrt|t| jkrtd| j	| j| || j|< d S )Nr  r  	forwarderr6  r  z<The forwarder name provided is longer than the existing one.)r|  r   set_word_at_offsetr  r  address_offsetr   r6  r   r  r  r  forwarder_offsetr   r   r6  r   r   r   r   r~  A  s2    zExportData.__setattr__Nr  r   r   r   r   r  5  s   r  c                   @   s   e Zd ZdZdS )ResourceDirDatazHolds resource directory information.

    struct:     IMAGE_RESOURCE_DIRECTORY structure
    entries:    list of entries (ResourceDirEntryData instances)
    Nr  r   r   r   r   r  f  s   r  c                   @   s   e Zd ZdZdS )ResourceDirEntryDataaF  Holds resource directory entry data.

    struct:     IMAGE_RESOURCE_DIRECTORY_ENTRY structure
    name:       If the resource is identified by name this
                attribute will contain the name string. None
                otherwise. If identified by id, the id is
                available at 'struct.Id'
    id:         the id, also in struct.Id
    directory:  If this entry has a lower level directory
                this attribute will point to the
                ResourceDirData instance representing it.
    data:       If this entry has no further lower directories
                and points to the actual resource data, this
                attribute will reference the corresponding
                ResourceDataEntryData instance.
    (Either of the 'directory' or 'data' attribute will exist,
    but not both.)
    Nr  r   r   r   r   r  n  s   r  c                   @   s   e Zd ZdZdS )ResourceDataEntryDatazHolds resource data entry information.

    struct:     IMAGE_RESOURCE_DATA_ENTRY structure
    lang:       Primary language ID
    sublang:    Sublanguage ID
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )	DebugDatazHolds debug information.

    struct:     IMAGE_DEBUG_DIRECTORY structure
    entries:    list of entries (IMAGE_DEBUG_TYPE instances)
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )BaseRelocationDatazHolds base relocation information.

    struct:     IMAGE_BASE_RELOCATION structure
    entries:    list of relocation data (RelocationData instances)
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdd ZdS )RelocationDatazHolds relocation information.

    type:       Type of relocation
                The type string can be obtained by
                RELOCATION_TYPE[type]
    rva:        RVA of the relocation
    c                 C   sj   t | dr\| jj}|dkr,|d> |d@ B }n(|dkrTt|| j d}|d@ |d@ B }|| j_|| j|< d S )Nr   typerL     r   r   i   )r|  r   Datar  base_rvar   )r   r6  r   wordr-  r   r   r   r~    s    
zRelocationData.__setattr__Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )TlsDatazJHolds TLS information.

    struct:     IMAGE_TLS_DIRECTORY structure
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )BoundImportDescDataa  Holds bound import descriptor data.

    This directory entry will provide information on the
    DLLs this PE file has been bound to (if bound at all).
    The structure will contain the name and timestamp of the
    DLL at the time of binding so that the loader can know
    whether it differs from the one currently present in the
    system and must, therefore, re-bind the PE's imports.

    struct:     IMAGE_BOUND_IMPORT_DESCRIPTOR structure
    name:       DLL name
    entries:    list of entries (BoundImportRefData instances)
                the entries will exist if this DLL has forwarded
                symbols. If so, the destination DLL will have an
                entry in this list.
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )LoadConfigDatazlHolds Load Config data.

    struct:     IMAGE_LOAD_CONFIG_DIRECTORY structure
    name:       dll name
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )BoundImportRefDatazHolds bound import forwarder reference data.

    Contains the same information as the bound descriptor but
    for forwarded DLLs, if any.

    struct:     IMAGE_BOUND_FORWARDER_REF structure
    name:       dll name
    Nr  r   r   r   r   r    s   r  c                   @   s   e Zd ZdZdS )ExceptionsDirEntryDatazHolds the data related to SEH (and stack unwinding, in particular)

    struct      an instance of RUNTIME_FUNTION
    unwindinfo  an instance of UNWIND_INFO
    Nr  r   r   r   r   r    s   r  c                       st   e Zd ZdZd fdd	Z fddZd fdd	Z fd	d
Zdd Zdd Z	 fddZ
dd Zdd Z  ZS )
UnwindInfozHandles the complexities of UNWIND_INFO structure:
    * variable number of UWIND_CODEs
    * optional ExceptionHandler and FunctionEntry fields
    r   c                    sH   t t| jd|d t t|  | _d | _tddd| _d | _d| _	d S )N)ZUNWIND_INFO)zB:3,Versionz	B:5,FlagszB,SizeOfPrologzB,CountOfCodeszB:4,FrameRegisterzB:4,FrameOffsetr7  ZUNWIND_CODE)B,CodeOffsetB:4,UnwindOp
B:4,OpInfor   F)
r  r  r   rB  
_full_size_opt_field_namer  
_code_info_chained_entry_finished_unpacking)r   r7  r  r   r   r     s    
zUnwindInfo.__init__c           	         s  | j r
dS tt| | | jd d@ }tt|  || j   }|| jdkrTdntd  | _	t
|| j	k rrdS | jdkr| jdkrdt| j S g | _tt|  }| j}|dkrZ| j|||| j    t| j}|du rdt| j|  S || j| }| j | }|| j||||  | | j|  ||7 }||8 }| j| q| jsj| jrpd	| _| jr~d
| _| jdkrt| | jtd|||td   d  d| _ dS )zUnpacks the UNWIND_INFO "in two calls", with the first call establishing
        a full size of the structure and the second, performing the actual unpacking.
        Nr0   r   r  r;   z&Unsupported version of UNWIND_INFO at zUnknown UNWIND_CODE at ZExceptionHandlerFunctionEntry<IT)r  r  r  rG  ZCountOfCodesrB  r  Flagsr  r  r   Versionhexr5  UnwindCodesPrologEpilogOpsFactorycreatelength_in_code_structures
initializer   rj   rk   r  rl   rD  r   r   )	r   r(   Zcodes_cnt_maxZhdlr_offsetroZ
codes_leftZucodeZlen_in_codesZopc_sizer  r   r   unpack_in_stages  s`    
zUnwindInfo.unpack_in_stagesc                    s    j d kr0 jtd   j j <  j j g z(tt |}W  j d krp j	  n j d krn j	  0 |dd
 fddtD   |dd
dd  jD   |S )	Nr  Flags: , c                    s"   g | ]}t  |d  r|d  qS r   )rH  rO  r   r   r   r3   a  r4   z#UnwindInfo.dump.<locals>.<listcomp>zUnwind codes: z; c                 S   s   g | ]}|  rt|qS r   )is_validr   re  r   r   r   r3   e  r4   )r  r  r  r*  r  r   r  r  rN  popr  unwind_info_flagsr  )r   rb  rN  r  r   r   rN  N  s*    



zUnwindInfo.dumpc                    sr   | j d kr0| jtd  | j| j < | j| j g z&tt|  }W | j d krn| j	  n| j d krl| j	  0 |S )Nr  )
r  r  r  r*  r  r   r  r  rh  r  )r   r  r  r   r   rh  i  s    



zUnwindInfo.dump_dictc                 C   sh   |dkrt | |t nDd|v rZt| |rZ|rD| jd  t| O  < n| jd  t| N  < || j|< d S )Nr  Z	UNW_FLAG_)r   r  r|  r   UNWIND_INFO_FLAGSr  r   r   r   r~  v  s    zUnwindInfo.__setattr__c                 C   s   | j S r
   )r  r   r   r   r   rB    s    zUnwindInfo.sizeofc                    s   t | j}tt|  |dtt|  < tt|  }| jD ]F}||j  | jkrZ q|j ||||j  < ||j 7 }q>| jd krt	dt
| | j|| jtd  | j< |S )Nr   r  r  )r   r  r  r  rL  rB  r  r   r  rI  rH  r  )r   r(   Z
cur_offsetZucr  r   r   rL    s    
 

zUnwindInfo.__pack__c                 C   s   | j S r
   )r  r   r   r   r   get_chained_function_entry  s    z%UnwindInfo.get_chained_function_entryc                 C   s   | j d krtd|| _ d S )Nz(Chained function entry cannot be changed)r  r   )r   entryr   r   r   set_chained_function_entry  s    
z%UnwindInfo.set_chained_function_entry)r   )r   )r   r   r   r   r   r  rN  rh  r~  rB  rL  r  r  r  r   r   r  r   r    s   >
r  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpzMeant as an abstract class representing a generic unwind code.
    There is a subclass of PrologEpilogOp for each member of UNWIND_OP_CODES enum.
    c                 C   s$   t | ||d| _| j| d S )Nr  )r  _get_formatr   rG  r   unw_coder(   unw_infor7  r   r   r   r	    s    
zPrologEpilogOp.initializec                 C   s   dS )zComputes how many UNWIND_CODE structures UNWIND_CODE occupies.
        May be called before initialize() and, for that reason, should not rely on
        the values of intance attributes.
        r0   r   r   r  r  r   r   r   r    s    z(PrologEpilogOp.length_in_code_structuresc                 C   s   dS )NTr   r   r   r   r   r    s    zPrologEpilogOp.is_validc                 C   s   dS )Nr  r   r   r  r   r   r   r    s    zPrologEpilogOp._get_formatN)r   r   r   r   r	  r  r  r  r   r   r   r   r    s
   r  c                   @   s    e Zd ZdZdd Zdd ZdS )PrologEpilogOpPushRegUWOP_PUSH_NONVOLc                 C   s   dS )N)ZUNWIND_CODE_PUSH_NONVOL)r  r  B:4,Regr   r  r   r   r   r    s    z!PrologEpilogOpPushReg._get_formatc                 C   s   dt | jj  S )Nz	.PUSHREG )	REGISTERSr   Regr   r   r   r   r     s    zPrologEpilogOpPushReg.__str__N)r   r   r   r   r  r   r   r   r   r   r    s   r  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpAllocLargeUWOP_ALLOC_LARGEc                 C   s   dddd|j dkrdndffS )NZUNWIND_CODE_ALLOC_LARGEr  r  r  r   zH,AllocSizeInQwordszI,AllocSizeOpInfor  r   r   r   r    s    z$PrologEpilogOpAllocLarge._get_formatc                 C   s   |j dkrdS dS )Nr   r;   r=   r$  r  r   r   r   r    s    z2PrologEpilogOpAllocLarge.length_in_code_structuresc                 C   s    | j jdkr| j jd S | j jS )Nr   rE   )r   r%  ZAllocSizeInQwordsZ	AllocSizer   r   r   r   get_alloc_size  s    
z'PrologEpilogOpAllocLarge.get_alloc_sizec                 C   s   dt |   S Nz.ALLOCSTACK r  r&  r   r   r   r   r     s    z PrologEpilogOpAllocLarge.__str__N)r   r   r   r   r  r  r&  r   r   r   r   r   r"    s
   r"  c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	PrologEpilogOpAllocSmallUWOP_ALLOC_SMALLc                 C   s   dS )N)ZUNWIND_CODE_ALLOC_SMALL)r  r  zB:4,AllocSizeInQwordsMinus8r   r  r   r   r   r    s    z$PrologEpilogOpAllocSmall._get_formatc                 C   s   | j jd d S r  )r   ZAllocSizeInQwordsMinus8r   r   r   r   r&    s    z'PrologEpilogOpAllocSmall.get_alloc_sizec                 C   s   dt |   S r'  r(  r   r   r   r   r     s    z PrologEpilogOpAllocSmall.__str__N)r   r   r   r   r  r&  r   r   r   r   r   r)    s   r)  c                       s(   e Zd ZdZ fddZdd Z  ZS )PrologEpilogOpSetFPUWOP_SET_FPREGc                    s.   t t| |||| |j| _|jd | _d S Nr.   )r  r+  r	  ZFrameRegister_frame_registerZFrameOffset_frame_offsetr  r  r   r   r	    s
    
zPrologEpilogOpSetFP.initializec                 C   s   dt | j  d t| j S )Nz
.SETFRAME r  )r   r.  r  r/  r   r   r   r   r     s    zPrologEpilogOpSetFP.__str__)r   r   r   r   r	  r   r  r   r   r  r   r+    s   r+  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpSaveRegUWOP_SAVE_NONVOLc                 C   s   dS Nr;   r   )r   unwcoder  r   r   r   r    s    z/PrologEpilogOpSaveReg.length_in_code_structuresc                 C   s   | j jd S r  )r   ZOffsetInQwordsr   r   r   r   
get_offset  s    z PrologEpilogOpSaveReg.get_offsetc                 C   s   dS )N)ZUNWIND_CODE_SAVE_NONVOL)r  r  r  zH,OffsetInQwordsr   r  r   r   r   r  	  s    z!PrologEpilogOpSaveReg._get_formatc                 C   s    dt | jj  d t|   S Nz	.SAVEREG r  )r   r   r!  r  r4  r   r   r   r   r     s    zPrologEpilogOpSaveReg.__str__Nr   r   r   r   r  r4  r  r   r   r   r   r   r0     s
   r0  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpSaveRegFarUWOP_SAVE_NONVOL_FARc                 C   s   dS Nr=   r   r  r   r   r   r    s    z2PrologEpilogOpSaveRegFar.length_in_code_structuresc                 C   s   | j jS r
   r   rf  r   r   r   r   r4    s    z#PrologEpilogOpSaveRegFar.get_offsetc                 C   s   dS )N)ZUNWIND_CODE_SAVE_NONVOL_FARr  r  r  zI,Offsetr   r  r   r   r   r    s    z$PrologEpilogOpSaveRegFar._get_formatc                 C   s    dt | jj  d t| jj S r5  )r   r   r!  r  rf  r   r   r   r   r   "  s    z PrologEpilogOpSaveRegFar.__str__Nr6  r   r   r   r   r7    s
   r7  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpSaveXMMUWOP_SAVE_XMM128c                 C   s   dS )N)ZUNWIND_CODE_SAVE_XMM128)r  r  r  zH,OffsetIn2Qwordsr   r  r   r   r   r  )  s    z!PrologEpilogOpSaveXMM._get_formatc                 C   s   dS r2  r   r  r   r   r   r  /  s    z/PrologEpilogOpSaveXMM.length_in_code_structuresc                 C   s   | j jd S r-  )r   ZOffsetIn2Qwordsr   r   r   r   r4  2  s    z PrologEpilogOpSaveXMM.get_offsetc                 C   s    dt | jj d t|   S Nz.SAVEXMM128 XMMr  )r   r   r!  r  r4  r   r   r   r   r   5  s    zPrologEpilogOpSaveXMM.__str__Nr   r   r   r   r  r  r4  r   r   r   r   r   r<  &  s
   r<  c                   @   s0   e Zd ZdZdd Zdd Zdd Zdd	 Zd
S )PrologEpilogOpSaveXMMFarUWOP_SAVE_XMM128_FARc                 C   s   dS )N)ZUNWIND_CODE_SAVE_XMM128_FARr;  r   r  r   r   r   r  <  s    z$PrologEpilogOpSaveXMMFar._get_formatc                 C   s   dS r9  r   r  r   r   r   r  B  s    z2PrologEpilogOpSaveXMMFar.length_in_code_structuresc                 C   s   | j jS r
   r:  r   r   r   r   r4  E  s    z#PrologEpilogOpSaveXMMFar.get_offsetc                 C   s    dt | jj d t| jj S r>  )r   r   r!  r  rf  r   r   r   r   r   H  s    z PrologEpilogOpSaveXMMFar.__str__Nr?  r   r   r   r   r@  9  s
   r@  c                   @   s   e Zd ZdZdd ZdS )PrologEpilogOpPushFrameUWOP_PUSH_MACHFRAMEc                 C   s   d| j jrdnd S )Nz
.PUSHFRAMEz <code>r   )r   r%  r   r   r   r   r   O  s    zPrologEpilogOpPushFrame.__str__N)r   r   r   r   r   r   r   r   r   rB  L  s   rB  c                       sH   e Zd ZdZ fddZdd Zdd Zdd	 Zd
d Zdd Z	  Z
S )PrologEpilogOpEpilogMarkerUWOP_EPILOGc                    s\   d| _ t|d | _tt| |||| | jrPt|d| jj |j	d@ dk| _ |j
| _d S )NTSizeOfEpilogr0   r   )_long_offstr|  _firstr  rD  r	  rD  r   Sizer%  rF  _epilog_sizer  r  r   r   r	  V  s    
z%PrologEpilogOpEpilogMarker.initializec                 C   s(   | j r d|jd@ dkrdndfS dS d S )NUNWIND_CODE_EPILOGr0   )zB,OffsetLow,Sizer  	B:4,Flags)zB,Sizer  rL  B,OffsetLowz
B:4,UnusedB:4,OffsetHigh)rK  )rM  r  rN  )rH  r%  r  r   r   r   r  a  s    z&PrologEpilogOpEpilogMarker._get_formatc                 C   s    t |ds|jd@ dkrdS dS )NrF  r0   r   r;   )r|  r%  r  r   r   r   r  x  s    z4PrologEpilogOpEpilogMarker.length_in_code_structuresc                 C   s   | j j| jr| j jd> ndB S )NrE   r   )r   Z	OffsetLowrG  Z
OffsetHighr   r   r   r   r4    s    z%PrologEpilogOpEpilogMarker.get_offsetc                 C   s   |   dkS )Nr   )r4  r   r   r   r   r    s    z#PrologEpilogOpEpilogMarker.is_validc                 C   s.   |   dkr*dt| j d t|    S dS )Nr   zEPILOG: size=z, offset from the end=-r   )r4  r  rJ  r   r   r   r   r     s    

z"PrologEpilogOpEpilogMarker.__str__)r   r   r   r   r	  r  r  r4  r  r   r  r   r   r  r   rD  S  s   rD  c                   @   sH   e Zd ZdZeeeeee	e
eeeeeeeeeeeeei
Zedd ZdS )r  zBA factory for creating unwind codes based on the value of UnwindOpc                 C   s    | j }|tjv rtj|  S d S r
   )ZUnwindOpr  _class_dict)r3  coder   r   r   r    s
    zPrologEpilogOpsFactory.createN)r   r   r   r   r  r  r#  r"  r*  r)  r,  r+  r1  r0  r8  r7  r=  r<  rA  r@  rC  rB  rE  rD  rO  staticmethodr  r   r   r   r   r    s   r  z!#$%&'()-@^_`{}~+,.;=[]c                    s>   | d u st | tttfsdS td  t fddt| D S )NFs   \/c                 3   s   | ]}| v V  qd S r
   r   re  allowedr   r   r    r4   z(is_valid_dos_filename.<locals>.<genexpr>)r   r   r   r   allowed_filenameallsetr  r   rR  r   is_valid_dos_filename  s    rX  z_?@$()<>c                 C   s.   | d uo,t | tttfo,tdd t| D S )Nc                 s   s   | ]}|t v V  qd S r
   )allowed_function_namere  r   r   r   r    r4   z)is_valid_function_name.<locals>.<genexpr>)r   r   r   r   rU  rV  rW  r   r   r   is_valid_function_name  s
    rZ  c                   @   s6  e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!d!d!e#d"fd#d$Z$d%d& Z%d'd( Z&d)d* Z'd+d, Z(d-d. Z)d/d0 Z*d1d2 Z+dd3d4Z,d5d6 Z-dd8d9Z.d:d; Z/d<d= Z0d>d? Z1d@dA Z2dBdC Z3dDdE Z4dFdG Z5ddIdJZ6dKdL Z7dMdN Z8dOdP Z9ddQdRZ:dSdT Z;dUdV Z<dWdX Z=ddZd[Z>d\d] Z?dd^d_Z@dd`daZAddbdcZBddedfZCdgdh ZDddidjZEdkdl ZFdmdn ZGeHfdodpZIdqdr ZJdsdt ZKddvdwZLdxdy ZMdzd{ ZNd|d} ZOd~d ZPdddZQdddZRdd ZSdd ZTdd ZUdd ZVdd ZWdd ZXdd ZYdd ZZdd Z[dd Z\dd Z]dd Z^dd Z_dd Z`dd Zadd Zbdd Zcdd Zddd Zedd Zfdd Zgdd Zhdd Zidd Zjdd Zkdd Zldd Zmdd Zndd Zodd ZpddÄ Zqddń ZrddǄ ZsddɄ Ztd!S )PEa	  A Portable Executable representation.

    This class provides access to most of the information in a PE file.

    It expects to be supplied the name of the file to load or PE data
    to process and an optional argument 'fast_load' (False by default)
    which controls whether to load all the directories information,
    which can be quite time consuming.

    pe = pefile.PE('module.dll')
    pe = pefile.PE(name='module.dll')

    would load 'module.dll' and process it. If the data is already
    available in a buffer the same can be achieved with:

    pe = pefile.PE(data=module_dll_data)

    The "fast_load" can be set to a default by setting its value in the
    module itself by means, for instance, of a "pefile.fast_load = True".
    That will make all the subsequent instances not to load the
    whole PE structure. The "full_load" method can be used to parse
    the missing data at a later stage.

    Basic headers information will be available in the attributes:

    DOS_HEADER
    NT_HEADERS
    FILE_HEADER
    OPTIONAL_HEADER

    All of them will contain among their attributes the members of the
    corresponding structures as defined in WINNT.H

    The raw data corresponding to the header (from the beginning of the
    file up to the start of the first section) will be available in the
    instance's attribute 'header' as a string.

    The sections will be available as a list in the 'sections' attribute.
    Each entry will contain as attributes all the structure's members.

    Directory entries will be available as attributes (if they exist):
    (no other entries are processed at this point)

    DIRECTORY_ENTRY_IMPORT (list of ImportDescData instances)
    DIRECTORY_ENTRY_EXPORT (ExportDirData instance)
    DIRECTORY_ENTRY_RESOURCE (ResourceDirData instance)
    DIRECTORY_ENTRY_DEBUG (list of DebugData instances)
    DIRECTORY_ENTRY_BASERELOC (list of BaseRelocationData instances)
    DIRECTORY_ENTRY_TLS
    DIRECTORY_ENTRY_BOUND_IMPORT (list of BoundImportData instances)

    The following dictionary attributes provide ways of mapping different
    constants. They will accept the numeric value and return the string
    representation and the opposite, feed in the string and get the
    numeric constant:

    DIRECTORY_ENTRY
    IMAGE_CHARACTERISTICS
    SECTION_CHARACTERISTICS
    DEBUG_TYPE
    SUBSYSTEM_TYPE
    MACHINE_TYPE
    RELOCATION_TYPE
    RESOURCE_TYPE
    LANG
    SUBLANG
    )ZIMAGE_DOS_HEADER)z	H,e_magiczH,e_cblpzH,e_cpzH,e_crlczH,e_cparhdrzH,e_minalloczH,e_maxalloczH,e_sszH,e_spzH,e_csumzH,e_ipzH,e_csz
H,e_lfarlczH,e_ovnoz8s,e_resz	H,e_oemidzH,e_oeminfoz
20s,e_res2z
I,e_lfanew)ZIMAGE_FILE_HEADER)z	H,MachinezH,NumberOfSectionsI,TimeDateStampzI,PointerToSymbolTablezI,NumberOfSymbolszH,SizeOfOptionalHeaderzH,Characteristics)ZIMAGE_DATA_DIRECTORY)I,VirtualAddressI,Size)ZIMAGE_OPTIONAL_HEADER)H,MagicB,MajorLinkerVersionB,MinorLinkerVersionI,SizeOfCodeI,SizeOfInitializedDataI,SizeOfUninitializedDataI,AddressOfEntryPointI,BaseOfCodezI,BaseOfDatazI,ImageBaseI,SectionAlignmentI,FileAlignmentH,MajorOperatingSystemVersionH,MinorOperatingSystemVersionH,MajorImageVersionH,MinorImageVersionH,MajorSubsystemVersionH,MinorSubsystemVersionI,Reserved1I,SizeOfImageI,SizeOfHeaders
I,CheckSumH,SubsystemH,DllCharacteristicszI,SizeOfStackReservezI,SizeOfStackCommitzI,SizeOfHeapReservezI,SizeOfHeapCommitI,LoaderFlagsI,NumberOfRvaAndSizes)ZIMAGE_OPTIONAL_HEADER64)r_  r`  ra  rb  rc  rd  re  rf  zQ,ImageBaserg  rh  ri  rj  rk  rl  rm  rn  ro  rp  rq  rr  rs  rt  zQ,SizeOfStackReservezQ,SizeOfStackCommitzQ,SizeOfHeapReservezQ,SizeOfHeapCommitru  rv  )ZIMAGE_NT_HEADERS)I,Signature)ZIMAGE_SECTION_HEADER)
z8s,Namez,I,Misc,Misc_PhysicalAddress,Misc_VirtualSizer]  zI,SizeOfRawDataI,PointerToRawDatazI,PointerToRelocationszI,PointerToLinenumberszH,NumberOfRelocationszH,NumberOfLinenumbersI,Characteristics)ZIMAGE_DELAY_IMPORT_DESCRIPTOR)z	I,grAttrszI,szNamezI,phmodzI,pIATzI,pINTzI,pBoundIATzI,pUnloadIATzI,dwTimeStamp)ZIMAGE_IMPORT_DESCRIPTOR)z$I,OriginalFirstThunk,Characteristicsr\  zI,ForwarderChainI,NamezI,FirstThunk)ZIMAGE_EXPORT_DIRECTORY)ry  r\  H,MajorVersionH,MinorVersionrz  zI,BasezI,NumberOfFunctionszI,NumberOfNameszI,AddressOfFunctionszI,AddressOfNameszI,AddressOfNameOrdinals)ZIMAGE_RESOURCE_DIRECTORY)ry  r\  r{  r|  zH,NumberOfNamedEntrieszH,NumberOfIdEntries)ZIMAGE_RESOURCE_DIRECTORY_ENTRY)rz  I,OffsetToData)ZIMAGE_RESOURCE_DATA_ENTRY)r}  r^  z
I,CodePagez
I,Reserved)VS_VERSIONINFOzH,LengthzH,ValueLengthzH,Type)VS_FIXEDFILEINFO)rw  zI,StrucVersionzI,FileVersionMSzI,FileVersionLSzI,ProductVersionMSzI,ProductVersionLSzI,FileFlagsMaskzI,FileFlagszI,FileOSz
I,FileTypezI,FileSubtypezI,FileDateMSzI,FileDateLS)ZStringFileInfor  )StringTabler  )Stringr  )Varr  )IMAGE_THUNK_DATA)z0I,ForwarderString,Function,Ordinal,AddressOfData)r  )z0Q,ForwarderString,Function,Ordinal,AddressOfData)ZIMAGE_DEBUG_DIRECTORY)ry  r\  r{  r|  zI,TypezI,SizeOfDatazI,AddressOfRawDatarx  )ZIMAGE_BASE_RELOCATION)r]  zI,SizeOfBlock)ZIMAGE_BASE_RELOCATION_ENTRY)zH,Data)IMAGE_TLS_DIRECTORY)zI,StartAddressOfRawDatazI,EndAddressOfRawDatazI,AddressOfIndexzI,AddressOfCallBacksI,SizeOfZeroFillry  )r  )zQ,StartAddressOfRawDatazQ,EndAddressOfRawDatazQ,AddressOfIndexzQ,AddressOfCallBacksr  ry  )IMAGE_LOAD_CONFIG_DIRECTORY)r^  r\  r{  r|  I,GlobalFlagsClearI,GlobalFlagsSetI,CriticalSectionDefaultTimeoutzI,DeCommitFreeBlockThresholdzI,DeCommitTotalFreeThresholdzI,LockPrefixTablezI,MaximumAllocationSizezI,VirtualMemoryThresholdI,ProcessHeapFlagszI,ProcessAffinityMaskH,CSDVersionH,Reserved1z
I,EditListzI,SecurityCookiezI,SEHandlerTablezI,SEHandlerCountzI,GuardCFCheckFunctionPointerzI,Reserved2zI,GuardCFFunctionTablezI,GuardCFFunctionCountI,GuardFlags)r  )r^  r\  r{  r|  r  r  r  zQ,DeCommitFreeBlockThresholdzQ,DeCommitTotalFreeThresholdzQ,LockPrefixTablezQ,MaximumAllocationSizezQ,VirtualMemoryThresholdzQ,ProcessAffinityMaskr  r  r  z
Q,EditListzQ,SecurityCookiezQ,SEHandlerTablezQ,SEHandlerCountzQ,GuardCFCheckFunctionPointerzQ,Reserved2zQ,GuardCFFunctionTablezQ,GuardCFFunctionCountr  )ZIMAGE_BOUND_IMPORT_DESCRIPTOR)r\  H,OffsetModuleNamezH,NumberOfModuleForwarderRefs)ZIMAGE_BOUND_FORWARDER_REF)r\  r  z
H,Reserved)ZRUNTIME_FUNCTION)zI,BeginAddresszI,EndAddresszI,UnwindDataNx   c                 C   s   || _ || _g | _g | _d | _|d u r6|d u r6tdg | _d | _d| _d| _	d| _
d| _d| _|plt d }z| ||| W n   |    Y n0 d S )NzMust supply either name or dataFr   	fast_load)max_symbol_exportsmax_repeated_symbolsections_PE__warningsr  r`  __structures___PE__from_fileFileAlignment_WarningSectionAlignment_Warning!_PE__total_resource_entries_count_PE__total_resource_bytes_PE__total_import_symbolsglobals	__parse__close)r   r6  r(   r  r  r  r   r   r   r   
  s(    	zPE.__init__c                 C   sR   | j du rNt| drNttjtr.t| jtjs@dtt| jv rN| j  | `d S )NTrv  z	mmap.mmap)r  r|  r   mmapr  rv  r   r  r   r   r   r   r  
  s    

zPE.closec              
   C   sl   t ||d}z|| W n@ tyZ } z(| jd|d || W Y d}~dS d}~0 0 | j| |S )zyApply structure format to raw data.

        Returns an unpacked structure object if successful, None otherwise.
        r  z7Corrupt header "{0}" at file offset {1}. Exception: {2}r   N)r2  rG  r   r  r   r   r  )r   r   r(   r7  	structureerrr   r   r   __unpack_data__
  s    
zPE.__unpack_data__c              
      s@  |durt |}|jdkr$tdd}zzTt|d}|  _ttdr`t jdtj _	ntj jdtj
d _	d _W nJ ty } z2d|}|od	| }td
||W Y d}~n
d}~0 0 W |dur|  n|dur|  0 n|dur| _	d _t j	 _d _|stt j	 D ]p\}}	|dkr\d|	 t j	 dks~|dkr2d|	 t j	 dkr2 jd|d|	 t j	  q2 j	dd }
t|
dkrtd j j|
dd _ jjtkrtd jr jjtkrtd jjt j	kr,td jj} j j j	||d  |d _  j rh j j!sptdd j j!@ t"krtdd j j!@ t#krtdd j j!@ t$krtdd j j!@ t%krtd j j!t&krtd j j' j	|d |d d   |d d _(t)t*d!} j(s6td"t+ j( j(j,| |d  j(-  }| j(j. } j j/ j	||d#  |d _0d$} j0du rt j	||d%  |krd&} j	||d%  d'|  } j j/||d _0 j0dur j0j1t2krt2 _3n j0j1t4krt4 _3 j j5 j	||d%  |d _0d(} j0du rt j	||d%  |krd&} j	||d%  d'|  } j j5||d _0 j(std" j0du rtd) j3du r jd* j0j1 t)t6d+}t+ j0 j0j7| g  j0_8| j0-  } j( j _( j0 j _0 j0j9 j0j:k rF jd,  j0j;d-krh jd. j0j;  d#}t<t=d/ j0j;@ D ]}t j	| dkr qft j	| dk rƈ j	|d d0 }n j	|||  } j j>||d}|du r qfzt?| |_@W n  tAtBfy(   Y  qfY n0 ||- 7 } j0j8| || j0-  d& kr qfq C|} fd1d2 jDD }t|dkrtE|}nd}|r||k r j	d|  _Fn j	d|  _F G j0j9dur H j0j9}|t j	kr. jd3 j0j9  n jd4 j0j9  |s< I  dS )5zParse a Portable Executable file.

        Loads a PE file, parsing all its structures and making them available
        through the instance's attributes.
        Nr   zThe file is emptyrbMAP_PRIVATE)accessTr  z: %szUnable to access file '{0}'{1}Fg      ?g      ?g333333?zeByte 0x{0:02x} makes up {1:.4f}% of the file's contents. This may indicate truncation / malformation.g      Y@rR   z9Unable to read the DOS Header, possibly a truncated file.r  z)Probably a ZM Executable (not a PE file).zDOS Header magic not found.z.Invalid e_lfanew value, probably not a PE filerE   zNT Headers not found.r  z0Invalid NT Headers signature. Probably a NE filez0Invalid NT Headers signature. Probably a LE filez0Invalid NT Headers signature. Probably a LX filez0Invalid NT Headers signature. Probably a TE filezInvalid NT Headers signature.r?   r-   IMAGE_FILE_zFile Header missingrS   r}   r   r	   rW  r   z5No Optional Header found, invalid PE32 or PE32+ file.z*Invalid type 0x{0:04x} in Optional Header.IMAGE_DLLCHARACTERISTICS_zXSizeOfHeaders is smaller than AddressOfEntryPoint: this file cannot run under Windows 8.r.   zsSuspicious NumberOfRvaAndSizes in the Optional Header. Normal values are never larger than 0x10, the value is: 0x%xs           c                    s(   g | ] }|j d kr |j  jjqS r  )rl  rm  rn  ro  rO  r   r   r   r3     s
   

z PE.__parse__.<locals>.<listcomp>z[Possibly corrupt file. AddressOfEntryPoint lies outside the file. AddressOfEntryPoint: 0x%xzTAddressOfEntryPoint lies outside the sections' boundaries. AddressOfEntryPoint: 0x%x)Josstatst_sizer   openfilenor|  r  r  rv  ZACCESS_READr  IOErrorr   	Exceptionr  r   $_PE__resource_size_limit_upperbounds _PE__resource_size_limit_reachedr   r   r  r  r   r  __IMAGE_DOS_HEADER_format__
DOS_HEADERZe_magicIMAGE_DOSZM_SIGNATUREIMAGE_DOS_SIGNATUREZe_lfanew__IMAGE_NT_HEADERS_format__
NT_HEADERSrV  IMAGE_NE_SIGNATUREIMAGE_LE_SIGNATUREIMAGE_LX_SIGNATUREIMAGE_TE_SIGNATUREIMAGE_NT_SIGNATURE__IMAGE_FILE_HEADER_format__FILE_HEADERr   IMAGE_CHARACTERISTICSr   ry  rB  SizeOfOptionalHeader __IMAGE_OPTIONAL_HEADER_format__rn  ZMagicr  r  r  "__IMAGE_OPTIONAL_HEADER64_format__DLL_CHARACTERISTICSDllCharacteristicsDATA_DIRECTORYZAddressOfEntryPointZSizeOfHeadersZNumberOfRvaAndSizesr  r   __IMAGE_DATA_DIRECTORY_format__DIRECTORY_ENTRYr6  KeyErrorAttributeErrorparse_sectionsr  minheaderget_section_by_rvar  	full_load)r   fnamer(   r  r  fdexcpZexception_msgZbyteZ
byte_countZdos_header_dataZnt_headers_offsetimage_flagsZoptional_header_offsetZsections_offsetZ&MINIMUM_VALID_OPTIONAL_HEADER_RAW_SIZEpadding_lengthZpadded_datadll_characteristics_flagsr-  Z)MAX_ASSUMED_VALID_NUMBER_OF_RVA_AND_SIZESr   	dir_entryZrawDataPointersZlowest_section_offsetZ	ep_offsetr   r   r   r  
  s   









"



	









	

zPE.__parse__c              	   C   s  d}d}| j dd| j }|dkr*dS zb| j d|d  }|ddtt|d   }ttd		tt|d |}||vrW dS W n t
y   Y dS 0 td
|||d  }d|i}|d|d }||d< dd }	t }
t|D ]*\}}|
|	||	||t|  A  qt|
|d< |d }|d |A |ksb|d |ksb|d |krfdS ||d< g }||d< |dd }ttt|d D ]h}|d|  |kr|d| d  |kr| jd  q||d|  |A |d| d  |A g7 }q|S )a"  Parses the rich header
        see http://www.ntcore.com/files/richsign.htm for more information

        Structure:
        00 DanS ^ checksum, checksum, checksum, checksum
        10 Symbol RVA ^ checksum, Symbol size ^ checksum...
        ...
        XX Rich, checksum, 0, 0,...
        iDanSiRichs   Richr	   NrE   r?   z<{0}I<Lr0   rF  raw_datac                 S   s   t | tst| S | S r
   rd  )r  r   r   r   <lambda>|  r4   z&PE.parse_rich_header.<locals>.<lambda>
clear_datar   r;   r=   checksumr  zRich Header is malformed)rv  findrn  r>  r   r   r  r   r   r   r   rI  indexr   rC  r   r   r  r  )r   ZDANSZRICHZ
rich_indexZ	rich_datar(   rF  resultr  Zord_r  rE  r   r  Zheadervaluesr   r   r   r   parse_rich_headerN  sP    
$.,zPE.parse_rich_headerc                 C   s   | j S )zReturn the list of warnings.

        Non-critical problems found when parsing the PE file are
        appended to a list of warnings. This method returns the
        full list.
        )r  r   r   r   r   r     s    zPE.get_warningsc                 C   s   | j D ]}td| qdS )zPrint the list of warnings.

        Non-critical problems found when parsing the PE file are
        appended to a list of warnings. This method prints the
        full list to standard output.
        >N)r  print)r   warningr   r   r   show_warnings  s    
zPE.show_warningsc                 C   s   |    G dd d}|  }|r|| | _|dd| j_|dd| j_|dd| j_|dd| j_|dd| j_nd| _dS )	zProcess the data directories.

        This method will load the data directories which might not have
        been loaded if the "fast_load" option was used.
        c                   @   s   e Zd ZdS )z PE.full_load.<locals>.RichHeaderN)r   r   r   r   r   r   r   
RichHeader  s   r  r  Nr  rF  r  r  )	parse_data_directoriesr  RICH_HEADERr   r  r  rF  r  r  )r   r  Zrich_headerr   r   r   r    s    zPE.full_loadc              
   C   sZ  t | j}| jD ],}t | }| }||||t| < qt| dr,t| dr,| jD ]}|D ]}t|drd|jD ]}t	|j
 D ]\}	}|j|	 }
|j|	 }t||d kr|dd}|d|d d  ||
d |
d |d d  < q|dd}|||
d |
d t| < qqxqdq\|}|s:|S t|d	}|| |  dS )
a  Write the PE file.

        This function will process all headers and components
        of the PE file and include all changes made (by just
        assigning to attributes in the PE objects) and write
        the changes back to a file whose name is provided as
        an argument. The filename is optional, if not
        provided the data will be returned as a 'str' object.
        r~  FileInfor  r0   r   r   Nr;   zwb+)r   rv  r  rL  r>  r   r|  r  r  r  entriesr  entries_offsetsentries_lengthsr   r   r  writer  )r   filenameZ	file_datar  Zstruct_datar-  finfor  st_entryrF  offsetsZlengthsr   Zencoded_dataZnew_file_datar   r   r   r   r    sD    










zPE.writec           
      C   s  g | _ d}t| jjD ]t}|tkrB| jd| jjt  qd}t| j	| d}|s^ q||
 |  }|| | j|||
   }t||
 kr| jd| d  q|s| jd| d  q|| | j| |j|j t| jkr |d7 }| jd	| d
 | |j| jjt| jkrZ|d7 }| jd	| d |jdkr|d7 }| jd| d | |j| jj| jjdkr|d7 }| jd| d | jjdkr|j| jj dkr|d7 }| jd	| d ||kr| jd  qttd}t||j| |jddr|jddr|j !ddkrl| " rln| jd| d | j | q| j j#dd d t$| j D ]8\}	}|	t| j d krd|_%n| j |	d  j|_%q| jjdkr| j r|| j d 
 | jj  S |S dS )a  Fetch the PE file sections.

        The sections will be readily available in the "sections" attribute.
        Its attributes will contain all the section information plus "data"
        a buffer containing the section's data.

        The "Characteristics" member will be processed and attributes
        representing the section characteristics (with the 'IMAGE_SCN_'
        string trimmed from the constant's names) will be added to the
        section instance.

        Refer to the SectionStructure class for additional info.
        r=   zToo many sections {0} (>={1})r   )r   zInvalid section z. Contents are null-bytes.z8. No data in the file (is this corkami's virtsectblXP?).r0   zError parsing section z$. SizeOfRawData is larger than file.z5. PointerToRawData points beyond the end of the file.rZ   z'Suspicious value found parsing section z*. VirtualSize is extremely large > 256MiB.z&. VirtualAddress is beyond 0x10000000.z. PointerToRawData should normally be a multiple of FileAlignment, this might imply the file is trying to confuse tools which parse this incorrectly.z,Too many warnings parsing section. Aborting.rz  r\   Fr[   rW  s   PAGEz!Suspicious flags set for section zf. Both IMAGE_SCN_MEM_WRITE and IMAGE_SCN_MEM_EXECUTE are set. This might indicate a packed executable.c                 S   s   | j S r
   )rq  )ar   r   r   r    r4   z#PE.parse_sections.<locals>.<lambda>)rF  N)&r  r  r  ZNumberOfSectionsMAX_SECTIONSr  r   r   ri  __IMAGE_SECTION_HEADER_format__rB  r@  rv  r)   rG  r  ru  rl  r   rm  rn  ro  r  rr  rq  rs  r   r{  r   ry  r   r   Namera  	is_driversortrC  r  )
r   r-  ZMAX_SIMULTANEOUS_ERRORSr   Zsimultaneous_errorssectionZsection_offsetsection_datar}  rE  r   r   r   r    s    









	


zPE.parse_sectionsFc           
      C   s  d| j fd| jfd| jfd| jfd| jfd| jfd| jfd| jfd	| jfd
| j	ff
}|durpt
|ttfsp|g}|D ]J}zt|d  }| jj| }W n ty   Y  qY n0 |du s||v r|jr|r|d dkr|d |j|jdd}n|r |d dkr |d |j|jdd}nZz|d |j|j}W nB tyx }	 z(| jd|d  d|	  W Y d}	~	n
d}	~	0 0 |rt| |d dd | |durtt
|trt|d |v rt|| qtdS )aS  Parse and process the PE file's data directories.

        If the optional argument 'directories' is given, only
        the directories at the specified indexes will be parsed.
        Such functionality allows parsing of areas of interest
        without the burden of having to parse all others.
        The directories can then be specified as:

        For export / import only:

          directories = [ 0, 1 ]

        or (more verbosely):

          directories = [ DIRECTORY_ENTRY['IMAGE_DIRECTORY_ENTRY_IMPORT'],
            DIRECTORY_ENTRY['IMAGE_DIRECTORY_ENTRY_EXPORT'] ]

        If 'directories' is a list, the ones that are processed will be removed,
        leaving only the ones that are not present in the image.

        If `forwarded_exports_only` is True, the IMAGE_DIRECTORY_ENTRY_EXPORT
        attribute will only contain exports that are forwarded to another DLL.

        If `import_dllnames_only` is True, symbols will not be parsed from
        the import table and the entries in the IMAGE_DIRECTORY_ENTRY_IMPORT
        attribute will not have a `symbols` attribute.
        r9   r8   r:   rB   r@   rF   rH   rM   rJ   r<   Nr   r0   T)forwarded_only)dllnames_onlyzFailed to process directoty "z": rC   )parse_import_directoryparse_export_directoryparse_resources_directoryparse_debug_directoryparse_relocations_directoryparse_directory_tlsparse_directory_load_configparse_delay_import_directoryparse_directory_bound_importsparse_exceptions_directoryr   r3  r  r  rn  r  
IndexErrorrq  rI  r   r  r   rD  remove)
r   directoriesZforwarded_exports_onlyZimport_dllnames_onlyZdirectory_parsingr  Zdirectory_indexr  r   r  r   r   r   r    sj     




zPE.parse_data_directoriesc                 C   s2  | j jtd kr$| j jtd kr$dS t| j}| }i }g }i }t|| D ]}| j| j| ||| 	|d}|du r qdd}	|j
d@ dkr:|j
|v r||j
 }	nt| 	|j
d}	|	||j
< |	| |j
|	 }
|
dkr| j|
  qd|	| |j
|	 }
|
dkr.| j|
  qd| j|	 t||	d}|| |||j< ||7 }qN|D ]}|jdkr|qht|jdsqh|jj|vr| jd	|j d
d qhz|j||jj  W nT ty( } z:| jd|j d
d|  W Y d}~qhW Y d}~n
d}~0 0 qh|S )zParses exception directory

        All the code related to handling exception directories is documented in
        https://auscitte.github.io/systems%20blog/Exception-Directory-pefile#implementation-details
        rc   ra   Nr  r0   r   )r   
unwindinfor   z FunctionEntry of UNWIND_INFO at r   z' points to an entry that does not existz/Failed parsing FunctionEntry of UNWIND_INFO at z: )r  MachineMACHINE_TYPEr2  __RUNTIME_FUNCTION_format__rB  r  r  r   r  Z
UnwindDatar  r  r  r   r  r  ZBeginAddressr  r|  r   r   r>  r  r   )r   r   r  rfZrf_sizeZrva2rtZrt_funcsZ	rva2infosr  Zuiwsr  r  r   r   r   r    s|    	








(zPE.parse_exceptions_directoryc                    sB  t | j}| }|}g }| j| j| j|||  |d}|du rR| jd dS | r^q>|| 7 }| |}| 	| |du rt
| j  } fdd| jD }	|	rt|	}
| |
}|dur|j  }n|jt
|    }|s| jd| dS g }tt|jt|d D ]}| j| j| j|||  |d}|sRtd|| 7 }||j }| d	| j||t  }|rd
d t|D }t
|dks|r q|t||d q ||j }| d	| j||t  }|rdd t|D }t
|dks>|rq>|s(q>|t|||d q|S )r   r  Nz7The Bound Imports directory exists but can't be parsed.c                    s   g | ]}|j  kr|j qS r   )rl  rO  r  r   r   r3   m  s   
z4PE.parse_directory_bound_imports.<locals>.<listcomp>zHRVA of IMAGE_BOUND_IMPORT_DESCRIPTOR points to an invalid address: {0:x}rE   z(IMAGE_BOUND_FORWARDER_REF cannot be readr   c                 S   s   g | ]}t |tjvr|qS r   rc  re  r   r   r   r3     s   rS   )r   r6  c                 S   s   g | ]}t |tjvr|qS r   rc  re  r   r   r   r3     s   )r   r6  r  )r2  (__IMAGE_BOUND_IMPORT_DESCRIPTOR_format__rB  r  rv  r  r   rA  get_section_by_offsetr  r   r  r  rl  r   r   r  ZNumberOfModuleForwarderRefsr   $__IMAGE_BOUND_FORWARDER_REF_format__r   ZOffsetModuleNameget_string_from_dataMAX_STRING_LENGTHr   r  r  )r   r   r  Z	bnd_descrZbnd_descr_sizerw  Zbound_importsr  safety_boundaryZsections_after_offsetZfirst_section_after_offsetZforwarder_refsr  Zbnd_frwd_refr-  Zname_strZinvalid_charsr   r  r   r  K  s    





	

z PE.parse_directory_bound_importsc                 C   sz   | j }| jtkr| j}z*| j|| |t| | |d}W n& t	yf   | j
d|  d}Y n0 |spdS t|dS )r   r  z5Invalid TLS information. Can't read data at RVA: 0x%xNr   )__IMAGE_TLS_DIRECTORY_format__r  r   __IMAGE_TLS_DIRECTORY64_format__r  r   r2  rB  r  r   r  r   r  )r   r   r  r   Z
tls_structr   r   r   r     s"    


zPE.parse_directory_tlsc                 C   s   | j tkr| j}n"| j tkr$| j}n| jd dS d}z*| j|| |t	|
 | |d}W n" ty   | jd|  Y n0 |sdS t|dS )r   zGDon't know how to parse LOAD_CONFIG information for non-PE32/PE32+ fileNr  z=Invalid LOAD_CONFIG information. Can't read data at RVA: 0x%xr  )r  r  &__IMAGE_LOAD_CONFIG_DIRECTORY_format__r  (__IMAGE_LOAD_CONFIG_DIRECTORY64_format__r  r   r  r   r2  rB  r  r   r  )r   r   r  r   Zload_config_structr   r   r   r    s,    



zPE.parse_directory_load_configc                 C   s   t | j }|| }g }||k rz$| j| j| ||| |d}W n& tyl   | jd|  d}Y n0 |stq|j	| j
jkr| jd|j	  q|j| j
jkr| jd|j  q| || |j	|j| }|t||d |jsq||j7 }q|S )r   r  z<Invalid relocation information. Can't read data at RVA: 0x%xNzEInvalid relocation information. VirtualAddress outside of Image: 0x%xz9Invalid relocation information. SizeOfBlock too large: %dr   r  )r2   __IMAGE_BASE_RELOCATION_format__rB  r  r   r  r   r  r   rq  rn  SizeOfImageZSizeOfBlockparse_relocationsr  )r   r   r  Zrlc_sizerx  ZrelocationsZrlcZreloc_entriesr   r   r   r    sR    


zPE.parse_relocations_directoryc              	   C   s"  z|  ||}| |}W n* tyD   | jd|d g  Y S 0 g }g }ttt|d D ]}| j| j	||d |d d  |d}	|	s q|	j
}
|
d? }|
d@ }||f|v r| jd||    qt|d	kr|  |d
||f |t|	|||| d ||	 7 }qb|S )r   zBad RVA in relocation data: 0xr   r;   r0   r  rL   r  z3Overlapping offsets in relocation data at RVA: 0x%x  r   )r   r  r  r   )r   r  r   r  r   r  r   r   r  &__IMAGE_BASE_RELOCATION_ENTRY_format__r  r  insertr  rB  )r   Zdata_rvar   r  r(   r7  r  Zoffsets_and_typerE  r  r  Z
reloc_typeZreloc_offsetr   r   r   r  7  sH    
zPE.parse_relocationsc              	   C   s0  t | j }g }tt|| D ]}z| |||  |}W n& tyf   | jd|  Y  dS 0 | j	| j|| 
|||  d}|s dS d}|jdkrnt|jdkr|j}	|j}
| j|	|	|
  }|dd dkr*dg d	g}|
t |  }|d
kr|d d| | 	|||	}nZ|dd dkrdg dg}|
t |  }|d
krv|d d| | 	|||	}n|jdkr|j}	|j}
| j|	|	|
  }dg dg}| 	|||	}|r|jdv r|
t |  }|d
kr
|d d| | 	|||	}|t||d q"|S )r   z7Invalid debug information. Can't read data at RVA: 0x%xNr  r0   r;   r?   s   RSDSZCV_INFO_PDB70)zI,CvSignaturezI,Signature_Data1zH,Signature_Data2zH,Signature_Data3z8s,Signature_Data4I,Ager   z{0}s,PdbFileNames   NB10ZCV_INFO_PDB20)zI,CvHeaderSignaturezI,CvHeaderOffsetrw  r   ZIMAGE_DEBUG_MISC)z
I,DataTypezI,Lengthz	B,UnicodezB,Reserved1zH,Reserved2r   r0   z	{0}s,Data)r   r  )r2   __IMAGE_DEBUG_DIRECTORY_format__rB  r  r   r   r   r  r   r  r  Typerl  Z
SizeOfDatarv  r   ZUnicoder  )r   r   r  Zdbg_sizedebugrE  r(   dbgZdbg_typeZdbg_type_offsetZdbg_type_sizeZdbg_type_dataZ__CV_INFO_PDB70_format__ZpdbFileName_sizeZ__CV_INFO_PDB20_format__Z___IMAGE_DEBUG_MISC_format__Zdbg_type_partial	data_sizer   r   r   r  d  s    










zPE.parse_debug_directoryr   c           %      C   s  |du r|g}|du r|}|t kr:| jd|t f  dS z| |t| j }W n$ tyx   | jd|  Y dS 0 | j| j|| 	|d}|du r| jd|  dS g }|| 7 }|j
|j }	d}
|	|
kr| jd|	|
f  dS |  j|	7  _| jtkr"| jd| jtf  dS g }d}t|	D ]}| jsl| j| jkrld	| _| jd
| j| jf  | |}|du r| jd||f   qd}d}|jd@ d? }|s|j}n||j }z|t| |}|  j| 7  _|r(|d |k r(|d |kr(|  | jd|  W  q|||  f}|| W n$ tyh   | jd|  Y n0 |jr||j |v r q| j||j |||  ||d |||j g d}|s qd}|td kri }|jD ]}t|dri }|jjD ]}|du st|dr|jj j!du s|j"du r8q|jj j#}|jj j!}|j"}z| ||}W n,   | jd|dd|  Y qY n0 t$|t%|d d | |&| q||j_'q|t(||||d n^| )||j }|r2|  j|j!7  _t*||jd@ |jd? d}|t(||||d n q|dkr|j+td kr|r`|d  }z|jjd jj}W n   Y n@0 |D ]8} d}!z| jj }!W n   Y n0 |!dur| ,|! q|| 7 }q2d!d" |D }"|"-  t.|D ]\}}#|#/  qt0||d#}$|$S )$a  Parse the resources directory.

        Given the RVA of the resources directory, it will process all
        its entries.

        The root will have the corresponding member of its structure,
        IMAGE_RESOURCE_DIRECTORY plus 'entries', a list of all the
        entries in the directory.

        Those entries will have, correspondingly, all the structure's
        members (IMAGE_RESOURCE_DIRECTORY_ENTRY) and an additional one,
        "directory", pointing to the IMAGE_RESOURCE_DIRECTORY structure
        representing upper layers of the tree. This one will also have
        an 'entries' attribute, pointing to the 3rd, and last, level.
        Another directory with more entries. Those last entries will
        have a new attribute (both 'leaf' or 'data_entry' can be used to
        access it). This structure finally points to the resource data.
        All the members of this structure, IMAGE_RESOURCE_DATA_ENTRY,
        are available as its attributes.
        NzNError parsing the resources directory. Excessively nested table depth %d (>%s)zCInvalid resources directory. Can't read directory data at RVA: 0x%xr  zDInvalid resources directory. Can't parse directory data at RVA: 0x%xr"   zNError parsing the resources directory. The directory contains %d entries (>%s)zRError parsing the resources directory. The file contains at least %d entries (>%d)TzGResource size 0x%x exceeds file size 0x%x, overlapping resources found.zHError parsing the resources directory, Entry %d is invalid, RVA = 0x%x. r/   r   r   r0   z^Error parsing the resources directory, attempting to read entry name. Entry names overlap 0x%xznError parsing the resources directory, attempting to read entry name. Can't read unicode string at offset 0x%x)r  leveldirsrm   	directoryr(   z2Error parsing resource of type RT_STRING at RVA 0xr   z with size r.   )r   r6  idr)  i  rI   )r   langsublang)r   r6  r*  r(   rn   r  c                 S   s   g | ]}|  qS r   )r   rO  r   r   r   r3   %  r4   z0PE.parse_resources_directory.<locals>.<listcomp>r  )1MAX_RESOURCE_DEPTHr  r   r   r2  #__IMAGE_RESOURCE_DIRECTORY_format__rB  r   r  r  ZNumberOfNamedEntriesZNumberOfIdEntriesr  MAX_RESOURCE_ENTRIESr  r  r  r  parse_resource_entryr  
NameOffsetr   r   r  DataIsDirectoryOffsetToDirectoryr  RESOURCE_TYPEr  r|  r)  r(   r   rI  r*  OffsetToDatar   r   updatestringsr  parse_resource_data_entryr  Idparse_version_informationr  rC  r   r  )%r   r   r  r  r'  r(  r(   resource_dirZdir_entriesZnumber_of_entriesZMAX_ALLOWED_ENTRIESZstrings_to_postprocessZlast_name_begin_endrE  resZ
entry_nameZentry_idZname_is_stringustr_offsetZentry_directoryr7  resource_idZresource_stringsresource_langZstring_entry_rvaZstring_entry_sizeZstring_entry_idZstring_entry_datar   Z
entry_dataZ
last_entryZversion_entriesZversion_entryZrt_version_structZstring_rvasr  Zresource_directory_datar   r   r   r    s   

	












zPE.parse_resources_directoryc                 C   s\   z|  |t| j }W n$ ty>   | jd|  Y dS 0 | j| j|| |d}|S )z0Parse a data entry from the resources directory.zGError parsing a resource directory data entry, the RVA is invalid: 0x%xNr  )	r   r2  $__IMAGE_RESOURCE_DATA_ENTRY_format__rB  r   r  r   r  r  )r   r   r(   Z
data_entryr   r   r   r8  1  s"    zPE.parse_resource_data_entryc                 C   s   z|  |t| j }W n ty.   Y dS 0 | j| j|| |d}|du rTdS |jd@ |_|jd@ |_	|jd@ |_
|jd@ d? |_|jd@ |_|S )z5Parse a directory entry from the resources directory.Nr  r        r  r/   r   )r   r2  )__IMAGE_RESOURCE_DIRECTORY_ENTRY_format__rB  r   r  r  r  r1  Z_PE__padr9  r5  r2  r3  )r   r   r(   resourcer   r   r   r0  I  s&    zPE.parse_resource_entryc           #      C   s  z|  |j}W n( ty8   | jd|j Y dS 0 | j|||j  }| j| j	||d}|du rldS |j|
  }| |}d}|r|jt|j|j }d}z4|du r| j|dd}n| j||| d? dd}W n" ty   | jd|  Y n0 |dkr| jd| dS |dur|d	krt|d
krv|dd
 d}	|	d|	d }	td|	t|}| jd|ddd dS t| dsg | _|}
||
_| j|
 |du rd}| |
 dt|d   |j}| j| j||d || d}|sdS t| ds,g | _| j| | ||
  |j}t| ds`g | _g }| j| j||d || d}|du r| jd dS |j| |
  }z| |}W n, ty   | jd| Y qY n0 ||_|| |r|dr|jdv rz|j dkrz| ||
  dt|d   |j}g |_!| j| j"||d || d}|sxqz|j| |
  }z| |}W n, ty   | jd| Y qzY n0 ||_#i |_$i |_%i |_&|j!| | ||
  dt|d   |j}|||j' k r|| j| j(||d || d}|sHq||j| |
  }z| |}|  |}W n, ty   | jd| Y q|Y n0 | dt|d  | |
  |j}|j| }z| j||j d}|  |}W n. ty   | jd|d Y q|Y n0 |j'dkr6||j' }n| |j'| |j}||j$|< ||f|j%|< t|t|f|j&|< q| |j'| |j}||krqz|}||j'krPqzqPn|rz|drz|}d|_)|jdv rz|j dkrz| ||
  dt|d   |j}g |_*| j| j+||d || d}|sBqz|j| |
  }z| |}W n, ty   | jd | Y qzY n0 |du rqz|j*| | dt|d  | |
  |j}|} || |j  k rN| ,|||d  d}!| ,||d |d!  d}"|d!7 }t-|!t.rt-|"t.r|d"|!|"f i|_/q| ||j' |j}|||j' krqzq| |j'| |j}|j'dks||j'krdqqd| j| dS )#a  Parse version information structure.

        The date will be made available in three attributes of the PE object.

        VS_VERSIONINFO   will contain the first three fields of the main structure:
            'Length', 'ValueLength', and 'Type'

        VS_FIXEDFILEINFO will hold the rest of the fields, accessible as sub-attributes:
            'Signature', 'StrucVersion', 'FileVersionMS', 'FileVersionLS',
            'ProductVersionMS', 'ProductVersionLS', 'FileFlagsMask', 'FileFlags',
            'FileOS', 'FileType', 'FileSubtype', 'FileDateMS', 'FileDateLS'

        FileInfo    is a list of all StringFileInfo and VarFileInfo structures.

        StringFileInfo structures will have a list as an attribute named 'StringTable'
        containing all the StringTable structures. Each of those structures contains a
        dictionary 'entries' with all the key / value version information string pairs.

        VarFileInfo structures will have a list as an attribute named 'Var' containing
        all Var structures. Each Var structure will have a dictionary as an attribute
        named 'entry' which will contain the name and value of the Var.
        zWError parsing the version information, attempting to read OffsetToData with RVA: 0x{:x}Nr  asciiencodingr0   zzError parsing the version information, attempting to read VS_VERSION_INFO string. Can't read unicode string at offset 0x%xz"Invalid VS_VERSION_INFO block: {0}s   VS_VERSION_INFOr	   z\uz({0} ... ({1} bytes, too long to display)r%   z\00r~  r   r;   r  r  z/Error parsing StringFileInfo/VarFileInfo structz|Error parsing the version information, attempting to read StringFileInfo string. Can't read unicode string at offset 0x{0:x}s   StringFileInfor!  r   zyError parsing the version information, attempting to read StringTable string. Can't read unicode string at offset 0x{0:x}z}Error parsing the version information, attempting to read StringTable Key string. Can't read unicode string at offset 0x{0:x}r   zzError parsing the version information, attempting to read StringTable Value string. Can't read unicode string at offset 0xr   s   VarFileInfoZVarFileInfoz}Error parsing the version information, attempting to read VarFileInfo Var string. Can't read unicode string at offset 0x{0:x}r?   z0x%04x 0x%04x)0r  r5  r   r  r   r   rv  rI  r  __VS_VERSIONINFO_format__rB  r  rq  r  ru  r  r   r   r   rfindr   replacer|  r~  ZKeydword_align__VS_FIXEDFILEINFO_format__r  r  __StringFileInfo_format__r   r#  ZValueLengthr  __StringTable_format__LangIDr  r  r  ZLength__String_format__r6  r  __Var_format__get_word_from_datar   r   r  )#r   Zversion_structZstart_offsetr  Zversioninfo_structr=  r  Zsection_endZversioninfo_stringZexcerptZvinfoZfixedfileinfo_offsetZfixedfileinfo_structZstringfileinfo_offsetr  Zstringfileinfo_structZstringfileinfo_stringZstringtable_offsetZstringtable_structZstringtable_stringZentry_offsetZstring_structrF  
key_offsetZvalue_offsetr   Znew_stringtable_offsetZvarfileinfo_structZ
var_offsetZ
var_structZ
var_stringZvarword_offsetZorig_varword_offsetZword1Zword2r   r   r   r:  h  sv   






















zPE.parse_version_informationc                    s  z. j  j |t j  |d}W n$ tyR    jd|  Y dS 0 |s\dS  fdd}zd |j	t
||j	|jd } |jt
||j|jd } |jt
||j|jd }W n$ ty    jd|  Y dS 0 g }	d}
 |j	}t j}|r.|jt|  |j	 }tt}d}tt
|jt|d D ]T} ||}|dur|d t|k r ||}n dS |du sT|d	krqT||kr||| k r |}z |}W n ty   Y qTY n0 n|rqTd}d} ||}|du r@|
d
8 }
|
d	kr@d} q |t}t|s`d} qz |}W n~ ty   |
d
8 }
|
d	krd}Y  qz |}W n< ty   |
d
8 }
|
d	krd}Y Y  qY Y qTY n0 Y n0 |||f  d
7  < |||f dkr8 jd| d|dd  qn*t| jkrb jd j  q|	t |j|  |jd|  | |jd|  ||||d	 qT|sȈ jd|j	d dd |	D }d}
 |j}t j}|r|jt|  |j }tt}d}tt
|jt|d D ],}||j |vr4z ||}W n tyr   d}Y n0 |du r|
d
8 }
|
d	krd} qd|d	krq4|dur||kr||| k rֈ |}nd}||  d
7  < ||  j kr jd j |  qdn,t| jkrD jd j d  qd|	t|j| |d|d q4|s jd|jd dS |	s|! rdS t"||	 |j#dS )a  Parse the export directory.

        Given the RVA of the export directory, it will process all
        its entries.

        The exports will be made available as a list of ExportData
        instances in the 'IMAGE_DIRECTORY_ENTRY_EXPORT' PE attribute.
        r  z+Error parsing export directory at RVA: 0x%xNc                    s   t  j |  S r
   )r   rv  r  )r   r   r   r   length_until_eof;  s    z3PE.parse_export_directory.<locals>.length_until_eofr?   rI   Tr   r0   Fz9Export directory contains more than 10 repeated entries (r  z#02xz). Assuming corrupt.zHExport directory contains more than {} symbol entries. Assuming corrupt.r;   )	r   r  r  r  r  r6  r  r  r  zIRVA AddressOfNames in the export directory points to an invalid address: r   c                 S   s   h | ]
}|j qS r   )r  )r1   expr   r   r   	<setcomp>  r4   z,PE.parse_export_directory.<locals>.<setcomp>z[Export directory contains more than {} repeated ordinal entries (0x{:x}). Assuming corrupt.z$Export directory contains more than z# ordinal entries. Assuming corrupt.)r  r  r6  r  zMRVA AddressOfFunctions in the export directory points to an invalid address: )r   symbolsr6  )$r  !__IMAGE_EXPORT_DIRECTORY_format__r   r2  rB  r  r   r  r   ZAddressOfNamesr  ZNumberOfNamesZAddressOfNameOrdinalsZAddressOfFunctionsZNumberOfFunctionsr  r   rv  rq  collectionsdefaultdictr   r  rQ  get_dword_from_dataget_string_at_rvaMAX_SYMBOL_NAME_LENGTHrZ  r  r   r  ZBaser  rA  r  r  )r   r   r  r  Z
export_dirrS  Zaddress_of_namesZaddress_of_name_ordinalsZaddress_of_functionsexportsZ#max_failed_entries_before_giving_upr  r  Zsymbol_countsZ&export_parsing_loop_completed_normallyr   Zsymbol_ordinalZsymbol_addressZforwarder_strr  Zsymbol_name_addressZsymbol_nameZsymbol_name_offsetZordinalsrE  r   r   r   r    s   



















	
zPE.parse_export_directoryc                 C   s   || d d@ |d@  S )Nr=   l    r   )r   r-  baser   r   r   rJ  &  s    zPE.dword_alignc                 C   s4   | j j}| j j| j j }||kr0||k r0||8 }|S r
   )rn  	ImageBaser  )r   vaZbegin_of_imageZend_of_imager   r   r   normalize_import_va)  s
    zPE.normalize_import_vac              
   C   s\  g }d}z|  |t| j }W n( tyJ   | jd|  Y qXY n0 | |}| j| j||d}|rt|	 rxqXd}|j
dkr| jjtd kr| |j|_| |j|_| |j|_| |j|_| |j|_| |j|_d}|| 7 }t| j| }	||jks ||jkr6t||j ||j }	g }
z| |j|jd|	|}
W n< ty } z"| jd||j W Y d}~n
d}~0 0 |d	kr| jd
| qX|
s|d7 }q| jtkr| jd| jtf  qX| |jt}t|st d}|r|
D ]0}|j!du rt"#|$ |j%}|r||_!q|t&||
|d q|S )z*Walk and parse the delay import directory.r   z5Error parsing the Delay import directory at RVA: 0x%xr  Fr`   TNzSError parsing the Delay import directory. Invalid import data at RVA: 0x{0:x} ({1})rA   zWToo many errors parsing the Delay import directory. Invalid import data at RVA: 0x{0:x}r0   z)Error, too many imported symbols %d (>%s)	*invalid*r   importsdll)'r   r2  (__IMAGE_DELAY_IMPORT_DESCRIPTOR_format__rB  r   r  r   r  r  rA  ZgrAttrsr  r	  r
  ra  Z	pBoundIATZpIATZpINTZ
pUnloadIATZphmodZszNamer   rv  r  parse_importsr   r   r  MAX_IMPORT_SYMBOLSr[  MAX_DLL_LENGTHrX  r   r6  	ordlookup	ordLookuplowerr  r  )r   r   r  import_descsr   r(   r7  import_desccontains_addressesmax_lenimport_datar  re  symbolfuncnamer   r   r   r  5  s    
	


zPE.parse_delay_import_directoryr   c                 C   s   t | dr| jd u rdS |dkr0t| jj S |dkrHt| jj S |dkr`t| jj S |dkrxt| jj S tdd S )Nr  r   r   r   r   r   z#Invalid hashing algorithm specified)	r|  r  r   r  r  r   r   r   r  )r   	algorithmr   r   r   get_rich_header_hash  s    zPE.get_rich_header_hashc                 C   s  g }g d}t | dsdS | jD ]}t|jtr@|j  }n
|j }|dd}t|dkrv|d |v rv|d }|j	D ]~}d }|j
stj|j |jdd}|std	|j d
|jdn|j
}|sq|t|tr| }|d| | f  q|q td|  S )N)Zocxsysre  DIRECTORY_ENTRY_IMPORTr   .r0   r   T)Z	make_namezUnable to look up ordinal rZ  Z04xz%s.%sr#  )r|  rw  r   re  r   r   rl  rsplitr   rd  r6  rj  rk  r  r   r   r   r  r   r  )r   ZimpstrsZextsr  libnamepartsimprs  r   r   r   get_imphash  s8    




zPE.get_imphashc              
   C   s  g }d}z|  |t| j }W n, tyN   | jd|d Y qY n0 | |}| j| j||d}|rx|	 r|q|| 7 }t
| j| }	||jks||jkrt||j ||j }	g }
|s^z| j|j|j|j|	d}
W nD ty* } z*| jd|dd|j d W Y d	}~n
d	}~0 0 |d
krN| jd|d q|
s^|d7 }q| |jt}t|s~td}|r|
D ]0}|jd	u rt| |j}|r||_q|t||
|d q|stddg}d}d}|D ]v}|jD ]h}|D ]T}|r|jsq|j}t|jt kr6|j!d}|"|r|d7 } qTq|d7 }qq|t
|kr|dk r| jd |S )z$Walk and parse the import directory.r   z-Error parsing the import directory at RVA: 0xr   r  r   zBError parsing the import directory. Invalid Import data at RVA: 0xz ()NrA   zLToo many errors parsing the import directory. Invalid import data at RVA: 0xr0   rb  rc  LoadLibraryZGetProcAddressr   r]   z?Imported symbols contain entries typical of packed executables.)#r   r2  "__IMAGE_IMPORT_DESCRIPTOR_format__rB  r   r  r   r  r  rA  r   rv  ZOriginalFirstThunkZ
FirstThunkr  rg  ZForwarderChainr   r[  r  ri  rX  r   r6  rj  rk  rl  r  r  rV  rd  r  r   r   r   )r   r   r  r  rm  r   r(   r7  rn  rp  rq  r2   re  rr  rs  Zsuspicious_importsZsuspicious_imports_countZtotal_symbolsZimp_dllZsuspicious_symbolr6  r   r   r   r    s    








zPE.parse_import_directoryc                 C   s  g }|  |||}|  |||}|r0t|dkr`|r@t|dkr`| jd|dd|d g S d}	|rn|}	n|rx|}	ndS d}
d}| jtkrt}n| jtkrt}d}
d	}nt}d}t	|	D ]\}}d}d}d}d}d}|j
r|j
|@ rd
}|j
d@ }d}d}ntd}zZ|j
|@ }| |d}| |d}| |j
d t}t|sRtd}| |j
d }W n tyx   Y n0 | }| |}|| jj ||
  }d}z>|r|r|| j
|| j
kr|| j
}|| }nd}W n ty   d}Y n0 |dkr|dkrtd|tdkrL|dkrB||krBtd|d7 }q|sV|r|t| ||||| ||||||||d q|S )zParse the imported symbols.

        It will fill a list, which will be available as the dictionary
        attribute "imports". Its keys will be the DLL names and the values
        of all the symbols imported from that object.
        r   z\Damaged Import Table information. ILT and/or IAT appear to be broken. OriginalFirstThunk: 0xr   z FirstThunk: 0xNr?   r  rE   l    Tr  Fr;   rb  z"Invalid entries, aborting parsing.r  z)Too many invalid names, aborting parsing.r0   )r   r  r  import_by_ordinalr  r  hintr6  r  r  r  hint_name_table_rvathunk_offset	thunk_rva)get_import_tabler   r  r   r  r  r  r  r  rC  r  r   rQ  r[  MAX_IMPORT_NAME_LENGTHrZ  r   r  r   r>  r  rn  r_  r  r  )r   Zoriginal_first_thunkZfirst_thunkZforwarder_chainr   ro  Zimported_symbolsZiltZiattableZ
imp_offsetZaddress_maskr  Znum_invalidrE  Z	tbl_entryZimp_ordZimp_hintZimp_namer  r  r  r(   r  r  Zimp_addressr  Z	imp_boundr   r   r   rg  ?  s     






"



zPE.parse_importsc                 C   s  g }| j tkrt}| j}n | j tkr0t}| j}n
t}| j}d}d}d}	t }
t }|}|r|d ur||| kr| j	d q| j
tkr| j	d| j
tf  q|  j
d7  _
|	|krg S |rt|t| |krg S |
rt|
t|
 |krg S d}z| |t| }W n ty*   d}Y n0 |sHt|t| kr\| j	d	|  d S | j||| |d
}|r| |j|_| |j|_| |j|_| |j|_|r|j|kr|j|kr| j	d|  q|rX|jrX|j|@ r|jd@ dkrXg S nF|j|v s*|j|
v r2|	d7 }	|jdkrL|
|j n||j |r| rlq|| 7 }|	| qV|S )NrY   rP   r   z9Error parsing the import table. Entries go beyond bounds.z$Excessive number of imports %d (>%s)r0   FTz9Error parsing the import table. Invalid data at RVA: 0x%xr  z\Error parsing the import table. AddressOfData overlaps with THUNK_DATA for THUNK at RVA 0x%xr  r          )r  r  r  __IMAGE_THUNK_DATA_format__r  r  __IMAGE_THUNK_DATA64_format__rV  r  r   r  rh  r  r  r   r2  rB  r   r   r  r  ra  r  r  r  r  r  rA  )r   r   r   ro  r  r  r   ZMAX_ADDRESS_SPREADZMAX_REPEATED_ADDRESSESZrepeated_addressZaddresses_of_data_set_64Zaddresses_of_data_set_32Z	start_rvafailedr(   Z
thunk_datar   r   r   r    s    



zPE.get_import_tablerZ   c           
      C   s  |dur| j }| | | j dd }| jD ]}|jdkrF|jdkrFq,|j}| |j| jj}| 	|j
| jj| jj}|t| j ks,|t| j ks,|| t| j ks,||krq,|t| }	|	dkr|d|	 7 }n|	dk r|d|	 }|| 7 }q,|dur|| _ |S )a  Returns the data corresponding to the memory layout of the PE file.

        The data includes the PE header and the sections loaded at offsets
        corresponding to their relative virtual addresses. (the VirtualAddress
        section header member).
        Any offset in this data corresponds to the absolute memory address
        ImageBase+offset.

        The optional argument 'max_virtual_address' provides with means of limiting
        which sections are processed.
        Any section with their VirtualAddress beyond this value will be skipped.
        Normally, sections with values beyond this range are just there to confuse
        tools. It's a common trick to see in packed executables.

        If the 'ImageBase' optional argument is supplied, the file's relocations
        will be applied to the image by calling the 'relocate_image()' method. Beware
        that the relocation information is applied permanently.
        Nr   rW  )rv  relocate_imager  r  ru  rm  rl  rn  ro  rr  rq  rs  r   r   )
r   Zmax_virtual_addressr_  Zoriginal_dataZmapped_datar  ZsrdZprdrk  r  r   r   r   get_memory_mapped_imageg  sD    



zPE.get_memory_mapped_imagec                 C   sv   g }t | drr| jjD ]Z}t |dr|jjD ]B}t |dr,t |jdr,|jjr,t|jj D ]}|| q^q,q|S )a  Returns a list of all the strings found withing the resources (if any).

        This method will scan all entries in the resources directory of the PE, if
        there is one, and will return a [] with the strings.

        An empty list will be returned otherwise.
        DIRECTORY_ENTRY_RESOURCEr)  r7  )r|  r  r  r)  r7  r  r  r   )r   Zresources_stringsres_typer>  
res_stringr   r   r   get_resources_strings  s    	




zPE.get_resources_stringsc                 C   sl   |  |}|r|| }nd}|s`|t| jk r<| j|| S |t| jk rX| j|| S td|||S )zGet data regardless of the section where it lies on.

        Given a RVA and the size of the chunk to retrieve, this method
        will find the section where the data lies and return the data.
        Nz-data at RVA can't be fetched. Corrupt header?)r  r   r  rv  r   r   )r   r   r   r  rx  r   r   r   r     s    

zPE.get_datac                    sJ     |}|s@ jr<t fdd jD }||k r8|S dS |S ||S )z.Get the RVA corresponding to this file offset.c                    s$   g | ]}  |j jj jjqS r   )rr  rq  rn  rs  ro  rO  r   r   r   r3     s   z*PE.get_rva_from_offset.<locals>.<listcomp>N)r  r  r  r  )r   r-  r  Z
lowest_rvar   r   r   r    s    


	zPE.get_rva_from_offsetc                 C   s<   |  |}|s2|t| jk r |S td|dd||S )zGet the file offset corresponding to this RVA.

        Given a RVA , this method will find the section where the
        data lies and return the offset within the file.
        zdata at RVA 0xr   z can't be fetched)r  r   rv  r   r  )r   r   r  r   r   r   r    s    
zPE.get_offset_from_rvac                 C   sJ   |du rdS |  |}|s4| d| j|||  S | d|j||dS )z1Get an ASCII string located at the given address.Nr   )r   )r  r  rv  r   )r   r   r   r  r   r   r   r[  "  s    
zPE.get_string_at_rvac                 C   s2   |t |krdS ||d }t|tr.t|S |S )rx  r4   N)r   r   r   r   )r   r-  r(   r  r   r   r   get_bytes_from_data-  s    
zPE.get_bytes_from_datac                 C   s.   |  ||}|d}|dkr*|d| }|S )zGet an ASCII string from data.rW  r   N)r  r  )r   r-  r(   r  rx  r   r   r   r  6  s
    
zPE.get_string_from_datarW   c           
      C   s  |dkrdS |  |d}|dK }t|d}|  ||}d}|d|d }|dkrt|}||k sj||krxt|d? }q||  || || 7 }|d }|}q:|d dkr:|dL }qq:td||d	|d  }d
tt	|}	|rt
|	|dS t
|	ddS )z3Get an Unicode string located at the given address.r   r4   r;   r0   rS   r  s     z<{:d}HNr   r   r   )r   r  r  r   r   r   r   r  maprX  r   r   )
r   r   r   rF  r(   Z	requestedZ
null_indexZdata_lengthZuchrsr  r   r   r   r   >  s0    
zPE.get_string_u_at_rvac                 C   s"   | j D ]}||r|  S qdS )z1Get the section containing the given file offset.N)r  r  )r   r-  r  r   r   r   r  i  s    


zPE.get_section_by_offsetc                 C   s"   | j D ]}||r|  S qdS )z-Get the section containing the given address.N)r  r  )r   r   r  r   r   r   r  r  s    


zPE.get_section_by_rvac                 C   s   |   S r
   )	dump_infor   r   r   r   r   {  s    z
PE.__str__c                 C   s
   t | dS )z.Checks if the PE file has relocation directoryDIRECTORY_ENTRY_BASERELOC)r|  r   r   r   r   
has_relocs~  s    zPE.has_relocsr   c                 C   s   t | j|d dS )z=Print all the PE header information in a human readable from.rE  N)r  r  )r   rF  r   r   r   
print_info  s    zPE.print_inforD  c           !         s   du rt   |  }|r@ d |D ]} |    q( d  | j      d  | j      d  | j	  t
td} d g }t|D ]"}t| j	|d r||d  q d	|    t| d
r,| jdur, d
  | j  t
td} d g }t|D ]&}t| j|d rL||d  qL d	|     d t
td}	| jD ]}
 |
   d g }t|	D ]$}t|
|d r||d  qΈ d	|  d|
  tdur6 d|
  tdurR d|
   tdurn d|
   tdur d|
      qt| d
rt| jdr d | jj D ]}|dur |  q   t| drt!| j"D ]\}}t#| j"dkr0 d|d   n
 d |durR |     t| dr | j$|      t| dr t#| j%|kr | j%| D ]6} |     t|drd|j&D ]} fdd| D   d|j'(|d     tt)|j*+ D ]0} d!|d (|d |d (|d  q$qֈ   nzt|d"r|j,D ]^}t|d#rv fd$d| D   d!t)|j-. d (d%d t)|j-/ d  qv   qq t| d&r d'  | j0j1      d(d)  | j0j2D ]r}|j3dur,t4d*}|j5rR|j5} d+|j6|j3|(|f  |j7r d,|j7(|d  n   q,   t| d-rڈ d. | j8D ]} |j1  |j9s d/| :|j1j;(|d        |j9D ]}|j<d0u r||j5dur^ d1|j=(d%|j5(d%|j6 n d2|j=(d%|j6 n* d3|j=(|d |j5(|d |j> |j?r d4|j? n   q   qt| d5rt d6 | j@D ]|} |j1   d7|j5(|d     |j*D ]<} |j1 d8  d7|j5(|d d8    q2qt| d9r@ d: | jAD ]} |j1     |j9D ]}|j<d0u r d;|j=(|d |j6 n* d<|j=(|d |j5(|d |j> |j?r( d4|j? n   q   qt| d=
r d>  | jBj1  | jBj*D ]}|j5dur|j5(|d } d?| d@dA n0tCD|j1jEdB} dC|j1jEdDdE| dFdA  |j1 dA t|dG
rx |jFj1 d8 |jFj*D ]d}|j5du	rF|j5(d%d } d?| d@dH n dC|j1jEdDd@dH  |j1 dH t|dG	r |jFj1 dI |jFj*D ]r}t|dJ	r dK|jGjH|jGjItJD|jGjHdLtK|jGjH|jGjIf dI  |j1 dM  |jGj1 dN 	qt|jFdO	r|jFjL	r dPdM t)t|jFjL+ D ],\}} dQ||MdRdS(dTdN 
qF	q   qp   t| dU
r| jN
r| jNj1
rΈ dV  | jNj1     t| dWr| jOr| jOj1r dX  | jOj1     t| dYr dZ | jPD ]} |j1  z d[tQ|j1jR   W n( tSy    d\|j1jR Y n0    |j-r, |j- d8    q,| T rP d] | jUD ]} |j1  |j*D ]\}z( d^|jVtW|jX d_d f d8 W n, tSy>    d`|jV|jXf d8 Y n0 q   qt| dart#| jYdkr db | jYD ]@}  | j1  t| dcr|| jZdur| | jZ d8 q| [ S )dz>Dump all the PE header information into human readable string.NParsing Warningsr  r  r  r  r  r   r  rn  r  zDllCharacteristics: PE Sectionsrz  z!Entropy: {0:f} (Min=0.0, Max=8.0)zMD5     hash: {0}zSHA-1   hash: %szSHA-256 hash: %szSHA-512 hash: %sr  Directoriesr~  r0   zVersion Information Version Informationr  r  r  c                    s   g | ]}  d | qS z  r  r1   r  rN  r   r   r3     r4   z PE.dump_info.<locals>.<listcomp>z  LangID: {0}r   z    {0}: {1}r  r  c                    s   g | ]}  d | qS r  r  r  r  r   r   r3     s   r   DIRECTORY_ENTRY_EXPORTExported symbolsz%-10s   %-10s  %sr  RVAr  Nonez%-10d 0x%08X    %sz forwarder: {0}rw  Imported symbolsz  Name -> {0}Tz*{0}.{1} Ordinal[{2}] (Imported by Ordinal)z&{0} Ordinal[{1}] (Imported by Ordinal)z{0}.{1} Hint[{2:d}]z Bound: 0x{0:08X}DIRECTORY_ENTRY_BOUND_IMPORTBound importszDLL: {0}r?   DIRECTORY_ENTRY_DELAY_IMPORTDelay Imported symbolsz({0} Ordinal[{1:d}] (Imported by Ordinal)z{0}.{1} Hint[{2}]r  Resource directoryzName: []r;   -zId: [0xXz] (r~  r)  rC   rE   r(   z\--- LANG [%d,%d][%s,%s]r   rI   rL   r7  z	[STRINGS]z{0:6d}: {1}unicode-escaper   rD  DIRECTORY_ENTRY_TLSTLSDIRECTORY_ENTRY_LOAD_CONFIGLOAD_CONFIGDIRECTORY_ENTRY_DEBUGDebug informationzType: zType: 0x{0:x}(Unknown)Base relocationsz%08Xh %sr.   z0x%08X 0x%x(Unknown)DIRECTORY_ENTRY_EXCEPTIONz"Unwind data for exception handlingr  )\r   r   r  r  r  r  r  rN  r  r  r   r  r  sortedrH  r   r  r|  rn  r  r{  r  r   r  r   r  r   r  r   r  r   r  r  rC  r~  r   r  r  r  rN  r   r  r  r  r  r  r   r  r  r   rV  r  r   r6  r  r  rw  rd  r[  r  r  re  r  r  r  r  r  r4  r   r9  r)  r(   r+  r,  r   r   r7  r   r  r  r  
DEBUG_TYPEr#  r  r  r  r   RELOCATION_TYPEr  r  r  r  )!r   rN  rF  warningsr  r  r   r   r  r}  r  r)  rE  Zvinfo_entryr  r  	str_entry	var_entryexportr6  modulerr  bound_imp_descbound_imp_refr  Zres_type_idr>  r?  r  r%  
base_relocrelocr  r   r  r   r    s   

























	

























zPE.dump_infoc           .   
   C   s`	  i }|   }|r||d< | j |d< | j |d< | j |d< ttd}g |d< |D ]&}t| j|d rX|d |d  qXt	| dr| j
d	ur| j
 |d< ttd
}g |d< |D ]&}t| j
|d r|d |d  qg |d< ttd}| jD ]}| }|d | g |d< |D ](}t||d r|d |d  q| |d< td	urj| |d< td	ur| |d< td	ur| |d< td	ur| |d< qt	| drt	| j
drg |d< t| j
jD ]&\}	}
|
d	ur|d |
  qt	| drg |d< t| jD ]b\}	}g }||  t	| dr\|| j|	   t	| drtt| j|	krtg }|| | j|	 D ]}||  t	|dr
i }|jD ]D}||  |j|d< t |j!" D ]}|d ||d < qq|| nft	|dr|j#D ]R}i }t	|dr||  t |j$% d |t |j$& d < || qq|d | q t	| drg |d< |d | j'j(  | j'j)D ]N}i }|j*d	ur|+|j,|j*|j-d  |j.r|j.|d!< |d | qt	| d"rg |d#< | j/D ]}g }|d# | ||j(  |j0D ]f}i }|j1d$u rx|j2|d%< |j,|d&< n|j2|d%< |j-|d'< |j3|d(< |j4r|j4|d)< || qNq"t	| d*r4g |d+< | j5D ]^}i }|d+ | |+|j(  |j-|d%< |j!D ]$}i }|+|j(  |j-|d%< q
qt	| d,rg |d-< | j6D ]}g }|d- | ||j(  |j0D ]f}i }|j1d$u r|j2|d%< |j,|d&< n|j2|d%< |j-|d'< |j3|d(< |j4r|j4|d)< || qzqNt	| d.rg |d/< |d/ | j7j(  | j7j!D ]}i } |j-d	ur:|j-| d'< n|j(j8t9:|j(j8d0f| d1< | +|j(  |d/ |  t	|d2rg }!|!|j;j(  |d/ |! |j;j!D ]N}"i }#|"j-d	ur|"j-|#d'< n|"j(j8|#d1< |#+|"j(  |!|# t	|"d2rg }$|$|"j;j(  |!|$ |"j;j!D ]}%t	|%d3r(i }&|%j<j=|&d4< |%j<j>|&d5< t?:|%j<j=d6|&d7< t@|%j<j=|%j<j>|&d8< |&+|%j(  |&+|%j<j(  |$|& q(t	|"j;d9r|"j;jArt |"j;jA" D ]"\}	}'|$|'Bd:d;Cd< q֐qqt	| d=r0| jDr0| jDj(r0| jDj( |d>< t	| d?r^| jEr^| jEj(r^| jEj( |d@< t	| dArg |dB< | jFD ]@}(i })|dB |) |)+|(j(  tG:|(j(jH|(j(jH|)dC< qx| I 	r\g |dD< | jJD ]}*g }+|dD |+ |+|*j(  |*j!D ]X},i }-|+|- |,jK|-dE< ztL|,jM dFd	 |-dC< W n tN	yR   |,jM|-dC< Y n0 qq|S )Gz5Dump all the PE header information into a dictionary.r  r  r  r  r  r  r   rn  Nr  r  r  rz  ZEntropyZMD5ZSHA1ZSHA256ZSHA512r  r  r~  r  r  r  r  rN  r0   r  r  r  r  r  r  rw  r  TZDLLr  r  ZHintZBoundr  r  r  r  r  r  r  r9  r)  r(   r   r   r   Z	LANG_NAMEZSUBLANG_NAMEr7  r  r   rD  r  r  r  r  r  r  r#  r  r  r.   )Or   r  rh  r  r  r   r  rH  r   r|  rn  r  r{  r  r  r   r  r   r  r   r  r   r  rC  r  r~  r  r   r  r  r  rN  r  r  r  r  r  r  r   r  r   rV  r  r6  r  r6  r  rw  rd  r  re  r  r  r  r  r  r9  r4  r   r)  r(   r+  r,  r   r   r7  r   r   r  r  r  r  r#  r  r  r   r  r  r  ).r   rh  r  r  r   r  r}  r  Zsection_dictrE  r)  Zvs_vinfoZversion_info_listZfileinfo_listr  Zstringtable_dictr  r  r  Zvar_dictr  Zexport_dictr  Zimport_listrr  Zsymbol_dictr  Zbound_imp_desc_dictr  Zbound_imp_ref_dictZmodule_listr  Zresource_type_dictZdirectory_listr>  Zresource_id_dictZresource_id_listr?  Zresource_lang_dictr  r%  Zdbg_dictr  Zbase_reloc_listr  Z
reloc_dictr   r   r   rh  %  s   











































zPE.dump_dictc                 C   s&   z|  |W S  ty    Y dS 0 dS )z;Gets the physical address in the PE file from an RVA value.N)r  r  r  r   r   r   get_physical_by_rvaK  s    zPE.get_physical_by_rvac                 C   s   t d|d@ S )zMReturn a four byte string representing the double word value (little endian).r      r   rI  )r   dwordr   r   r   get_data_from_dwordV  s    zPE.get_data_from_dwordc                 C   s<   |d d t |krdS td||d |d d  d S )a  Convert four bytes of data to a double word (little endian)

        'offset' is assumed to index into a dword array. So setting it to
        N will return a dword out of the data starting at offset N*4.

        Returns None if the data can't be turned into a double word.
        r0   r?   Nr  r   r   r   r   r   r(   r-  r   r   r   rZ  Z  s    	zPE.get_dword_from_datac                 C   s0   z|  | |ddW S  ty*   Y dS 0 dS )zReturn the double word value at the given RVA.

        Returns None if the value can't be read, i.e. the RVA can't be mapped
        to a file offset.
        r?   r   N)rZ  r   r   r  r   r   r   get_dword_at_rvah  s    zPE.get_dword_at_rvac                 C   s0   |d t | jkrdS | | j||d  dS )zFReturn the double word value at the given file offset. (little endian)r?   Nr   )r   rv  rZ  r?  r   r   r   get_dword_from_offsett  s    zPE.get_dword_from_offsetc                 C   s   |  || |S )zLSet the double word value at the file offset corresponding to the given RVA.)set_bytes_at_rvar  )r   r   r  r   r   r   set_dword_at_rva|  s    zPE.set_dword_at_rvac                 C   s   |  || |S )z3Set the double word value at the given file offset.)r  r  )r   r-  r  r   r   r   r    s    zPE.set_dword_at_offsetc                 C   s   t d|S )zFReturn a two byte string representing the word value. (little endian).r   r  r   r  r   r   r   get_data_from_word  s    zPE.get_data_from_wordc                 C   s<   |d d t |krdS td||d |d d  d S )a  Convert two bytes of data to a word (little endian)

        'offset' is assumed to index into a word array. So setting it to
        N will return a dword out of the data starting at offset N*2.

        Returns None if the data can't be turned into a word.
        r0   r;   Nr   r   r  r  r   r   r   rQ    s    	zPE.get_word_from_datac                 C   s6   z|  | |dd dW S  ty0   Y dS 0 dS )zReturn the word value at the given RVA.

        Returns None if the value can't be read, i.e. the RVA can't be mapped
        to a file offset.
        Nr;   r   )rQ  r   r   r  r   r   r   get_word_at_rva  s    zPE.get_word_at_rvac                 C   s0   |d t | jkrdS | | j||d  dS )z?Return the word value at the given file offset. (little endian)r;   Nr   )r   rv  rQ  r?  r   r   r   get_word_from_offset  s    zPE.get_word_from_offsetc                 C   s   |  || |S )zESet the word value at the file offset corresponding to the given RVA.)r  r  )r   r   r  r   r   r   set_word_at_rva  s    zPE.set_word_at_rvac                 C   s   |  || |S )z,Set the word value at the given file offset.)r  r  )r   r-  r  r   r   r   r    s    zPE.set_word_at_offsetc                 C   s   t d|S )zMReturn an eight byte string representing the quad-word value (little endian).<Qr  r  r   r   r   get_data_from_qword  s    zPE.get_data_from_qwordc                 C   s<   |d d t |krdS td||d |d d  d S )a  Convert eight bytes of data to a word (little endian)

        'offset' is assumed to index into a word array. So setting it to
        N will return a dword out of the data starting at offset N*8.

        Returns None if the data can't be turned into a quad word.
        r0   rE   Nr  r   r  r  r   r   r   get_qword_from_data  s    	zPE.get_qword_from_datac                 C   s6   z|  | |dd dW S  ty0   Y dS 0 dS )zReturn the quad-word value at the given RVA.

        Returns None if the value can't be read, i.e. the RVA can't be mapped
        to a file offset.
        NrE   r   )r  r   r   r  r   r   r   get_qword_at_rva  s    zPE.get_qword_at_rvac                 C   s0   |d t | jkrdS | | j||d  dS )zDReturn the quad-word value at the given file offset. (little endian)rE   Nr   )r   rv  r  r?  r   r   r   get_qword_from_offset  s    zPE.get_qword_from_offsetc                 C   s   |  || |S )zJSet the quad-word value at the file offset corresponding to the given RVA.)r  r  )r   r   qwordr   r   r   set_qword_at_rva  s    zPE.set_qword_at_rvac                 C   s   |  || |S )z1Set the quad-word value at the given file offset.)r  r  )r   r-  r  r   r   r   set_qword_at_offset  s    zPE.set_qword_at_offsetc                 C   s0   t |tstd| |}|s$dS | ||S )zOverwrite, with the given string, the bytes at the file offset corresponding
        to the given RVA.

        Return True if successful, False otherwise. It can fail if the
        offset is outside the file's boundaries.
        data should be of type: bytesF)r   r   r'   r  r  )r   r   r(   r-  r   r   r   r    s    

zPE.set_bytes_at_rvac                 C   sd   t |tstdd|  kr,t| jk r\n n,| jd| | | j|t| d  | _ndS dS )zOverwrite the bytes at the given file offset with the given string.

        Return True if successful, False otherwise. It can fail if the
        offset is outside the file's boundaries.
        r  r   NFT)r   r   r'   r   rv  )r   r-  r(   r   r   r   r    s    
&zPE.set_bytes_at_offsetc                 C   sn   | j D ]b}| |j| jj}||j }|t| jk r|t| jk r| jd| |  | j|d  | _qdS )zeUpdate the PE image content with any individual section data that has been
        modified.
        N)	r  rm  rl  rn  ro  ru  r   rv  r   )r   r  Zsection_data_startZsection_data_endr   r   r   merge_modified_section_data  s    


zPE.merge_modified_section_datac           	      C   s  || j j }t| j jdkr| j jd jrt| dsJ| jtd gd t| dsd| j	d n\| j
D ]R}d}|t|jk rj|j| }|d7 }|jtd	 krqt|jtd
 kr| |j| |j| d? d@  qt|jtd kr| |j| |j| d@  qt|jtd kr2| |j| |j|  qt|jtd kr|t|jkrTqj|j| }|d7 }| |j| |jd> |j | d@ d?  qt|jtd krt| |j| |j|  qtqj|| j _t| dr| jD ]"}|jD ]}| j|7  _qqt| drR| jj j|7  _| jj j|7  _| jj j|7  _| jj j|7  _t| dr| jjjr|| jj j|7  _| jjj r| jj j |7  _ | jjj!r| jj j!|7  _!| jjj"r| jj j"|7  _"| jjj#r| jj j#|7  _#| jjj$r| jj j$|7  _$dS )a2  Apply the relocation information to the image using the provided image base.

        This method will apply the relocation information to the image. Given the new
        base, all the relocations will be processed and both the raw data and the
        section's data will be fixed accordingly.
        The resulting image can be retrieved as well through the method:

            get_memory_mapped_image()

        In order to get something that would more closely match what could be found in
        memory once the Windows loader finished its work.
        rC   rA   r  r@   r  zZRelocating image but PE does not have (or pefile cannot parse) a DIRECTORY_ENTRY_BASERELOCr   r0   rd   re   r.   r  rf   rg   rh   rA  ri   rw  r  r  N)%rn  r_  r   r  rI  r|  r  r  r  r   r  r  r  r  r  r   r  r  r  r  r  rw  rd  r  r  r   ZStartAddressOfRawDataZEndAddressOfRawDataZAddressOfIndexZAddressOfCallBacksr  ZLockPrefixTableZEditListZSecurityCookieZSEHandlerTableZGuardCFCheckFunctionPointerZGuardCFFunctionTable)	r   Znew_ImageBaseZrelocation_differencer  Z	entry_idxr  Z
next_entryre  funcr   r   r   r  #  s    





	














zPE.relocate_imagec                 C   s   | j j|  kS r
   )rn  ZCheckSumgenerate_checksumr   r   r   r   verify_checksum  s    zPE.verify_checksumc                 C   s(  |   | _| j d }d}t| jd }t| jd| |dk  }tt|d D ]}|t|d krjqT|d t|d kr|rtd| j|d d  dd|   d }n&td| j|d |d d  d }||7 }|dkrT|d@ |d	?  }qT|d
@ |d?  }||d?  }|d
@ }|t| j S )NrR   r   r?   r0   r  rW  r  r  r-   r  r.   )	r  rv  rn  r>  r   r  r   r   r   )r   Zchecksum_offsetr  	remainderZdata_lenr   r  r   r   r   r    s,    
&zPE.generate_checksumc                 C   s0   t d }|  s,|  s,|| jj@ |kr,dS dS )zCheck whether the file is a standard executable.

        This will return true only if the file has the IMAGE_FILE_EXECUTABLE_IMAGE flag
        set and the IMAGE_FILE_DLL not set and the file does not appear to be a driver
        either.
        rQ   TF)r  is_dllr  r  ry  )r   ZEXE_flagr   r   r   is_exe  s    z	PE.is_exec                 C   s    t d }|| jj@ |krdS dS )zCheck whether the file is a standard DLL.

        This will return true only if the image has the IMAGE_FILE_DLL flag set.
        rU   TF)r  r  ry  )r   ZDLL_flagr   r   r   r    s    z	PE.is_dllc                 C   s   t | ds| jtd gd t | ds*dS td}|dd | jD rLdS td	}|d
d | jD r| jjt	d t	d fv rdS dS )zCheck whether the file is a Windows driver.

        This will return true only if there are reliable indicators of the image
        being a driver.
        rw  r9   r  F)s   ntoskrnl.exes   hal.dlls   ndis.syss   bootvid.dlls	   kdcom.dllc                 S   s   g | ]}|j  qS r   )re  rl  )r1   r|  r   r   r   r3   G  r4   z PE.is_driver.<locals>.<listcomp>T)s   pages   pagedc                 S   s   g | ]}|j  d qS )rW  )r  rl  ra  )r1   r  r   r   r   r3   M  r4   r^   r_   )
r|  r  r  rV  intersectionrw  r  rn  Z	SubsystemSUBSYSTEM_TYPE)r   Zsystem_DLLsZdriver_like_section_namesr   r   r   r    s2    

	zPE.is_driverc              	      s   d t | jf fdd	}t| dr:|| j | jjf | jD ]}||j|j	f q@t
d g}t| jjD ]D\}}||v r~qlz|| |j|jf W ql ty   Y qlY ql0 qlt | jt krt S dS )zoGet the offset of data appended to the file and not contained within
        the area described in the headers.r  c                    s$   t | |kr t | t  kr | S  S r
   )sum)Zoffset_and_size	file_sizeZlargest_offset_and_sizer   r   'update_if_sum_is_larger_and_within_file_  s
    zQPE.get_overlay_data_start_offset.<locals>.update_if_sum_is_larger_and_within_filern  r>   N)r   rv  r|  rn  r>  r  r  r  rl  ru  r  rC  r  r  rq  rI  r   r  )r   r  r  Zskip_directoriesrE  r)  r   r  r   get_overlay_data_start_offsetY  s4    	



z PE.get_overlay_data_start_offsetc                 C   s"   |   }|dur| j|d S dS )zeGet the data appended to the file and not contained within the area described
        in the headers.Nr  rv  r   Zoverlay_data_offsetr   r   r   get_overlay  s    zPE.get_overlayc                 C   s,   |   }|dur| jd| S | jdd S )zKReturn the just data defined by the PE headers, removing any overlaid data.Nr  r  r   r   r   trim  s    zPE.trimc                 C   s:   |t kr0| jdu r0t|s0| jd|  d| _t||S )NFz=If FileAlignment > 0x200 it should be a power of 2. Value: %xT)r   r  r   r  r   r!   )r   r   r    r   r   r   rm    s    zPE.adjust_FileAlignmentc                 C   s@   |t k r4||kr4| jdu r4| jd||f  d| _t|||S )NFzAIf FileAlignment(%x) < 0x200 it should equal SectionAlignment(%x)T)r   r  r  r   r$   )r   r   r#   r    r   r   r   rr    s    zPE.adjust_SectionAlignment)N)NFF)r   Nr   N)F)r   )F)NF)NF)rZ   N)r   N)rW   N)r   )NrD  )ur   r   r   r   r  r  r  r  r  r  r  rf  r  rW  r.  rB  r@  rG  rK  rL  rM  rO  rP  r  r  r"  r  r  r  r  r  r  r  r  r  MAX_SYMBOL_EXPORT_COUNTr   r  r  r  r  r   r  r  r  r  r  r  r  r   r  r  r  r  r  r8  r0  r:  r  rJ  ra  r  ru  r}  r  rg  r  r  r  r   r  r  r  r[  r  r  r   r  r  r   r  r  r  rh  r  r  rZ  r  r  r  r  r  rQ  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  rm  rr  r   r   r   r   r[    s  H$#
.  rQ

7  
aZw:- 
  9   8
  m
%
n  
 
 
I
"	
+		

   "  ( )0:.r[  c                  C   s   dd l } d}| jdd  s$t| nx| jd dkr| jdd  sJ| d t| jd }|jjD ]"}tt|jj	|j
 |j|j q`ntt| jd   d S )Nr   z1pefile.py <filename>
pefile.py exports <filename>r0   r]  r;   zerror: <filename> required)rv  argvr  exitr[  r  rV  r  rn  r_  r  r6  r  r  )rv  usager   rT  r   r   r   main  s    

r  __main__)r	   FF)r   
__author____version____contact__rX  r  r   r   r]  r  r   r  r   Zhashlibr   r   r   r   r   r   r   rj  register_errorlookup_errorr   r\  r   r!   r$   r)   r  r  rh  r  ri  r\  r  r/  r-  r  r  r  r  r  r  r  r  Z IMAGE_NUMBEROF_DIRECTORY_ENTRIESr  r  r  r  r7   Zdirectory_entry_typesr  Zimage_characteristicsr  Zsection_characteristicsr{  Zdebug_typesr  Zsubsystem_typesr  Zmachine_typesr
  Zrelocation_typesr  Zdll_characteristicsr  r   r  r  Z	registersr   r  r#  r*  r,  r1  r8  rE  r=  rA  rC  Zresource_typer4  r+  r   r,  r   r5   r   r   r   r   r   r   r   r   r   r   r  r   r   r  r!  r1  r2  ri  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r"  r)  r+  r0  r7  r<  r@  rB  rD  r  ascii_lowercaseascii_uppercaser  rT  rX  rY  rZ  r[  r  r   r   r   r   r   <module>   s  



1"aj
K
*



+ L .

Ql		B1	! -
@"
                                        
