
    hiǚ                       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ZddlZddlZddlZddlmZ ddlmZmZmZmZ ddlmZ ddlZ ej4                  d	 ej6                  d
             eZddZ e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Z0dZ1dZ2dZ3dZ4d Z5d!Z6d" Z7g d#Z8 e7e8      Z9g d$Z: e7e:      Z;g d%Z< e7e<      Z=g d&Z> e7e>      Z?g d'Z@ e7e@      ZAg d(ZB e7eB      ZCg d)ZD e7eD      ZEg d*ZF e7eF      ZGg d+ZH e7eH      ZIdZJdZKg d,ZL e7eL      ZMg d-ZN e7eN      ZOdZPd.ZQd/ZRd0ZSd1ZTd2ZUd3ZVd4ZWd5ZXd6ZYg d7ZZ e7eZ      Z[g d8Z\ e7e\      Z]g d9Z^ e7e^      Z_ e`e^      Z_e^D ]$  \  ZaZbebe_v re_eb   j                  ea       eage_eb<   & d: Zdd; Zed< Zfd= Zgd> Zh G d? d@ei      Zj G dA dB      Zk G dC dDel      Zm G dE dF      Znd.d.d.d.d/d/d1d1d1d1d1d4d4d4d.dGZo ed      dH        Zp eddIJ      dK        Zq G dL dM      Zr G dN dOer      Zs eddJ      dP        Zt G dQ dRer      Zu G dS dT      Zv G dU dVev      Zw G dW dXev      Zx G dY dZev      Zy G d[ d\ev      Zz G d] d^ev      Z{ G d_ d`ev      Z| G da dbev      Z} G dc ddev      Z~ G de dfev      Z G dg dhev      Z G di djev      Z G dk dlev      Z G dm dnev      Z G do dpev      Z G dq drev      Z G ds dtev      Z G du dvev      Z G dw dxev      Z G dy dzev      Z G d{ d|eu      Z G d} d~      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d de      Z G d d      Zej.                  ej0                  z   ej2                  z   dz   j5                         Zd Zej.                  ej0                  z   ej2                  z   j5                         Z ed      	 ddeeeef   dedefd       Z G d d      Zd Zedk(  r e        yy)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-2024 Ero Carrera <ero.carrera@gmail.com>
zEro Carreraz	2024.8.26zero.carrera@gmail.com    N)Counter)md5sha1sha256sha512)Unionbackslashreplace_backslashreplaceFc                 F     |st        j                         S  fd}|S )Nc                 |     t        j                        |       t        j                  |       fd       }|S )Nc                  :    t        j                   | i |      S N)copymodcopy)argskwargscached_funcs     2C:\des-py\Monitor\venv\Lib\site-packages\pefile.pywrapperz-lru_cache.<locals>.decorator.<locals>.wrapper9   s     <<T <V <==    )	functools	lru_cachewraps)fr   r   maxsizetypeds     @r   	decoratorzlru_cache.<locals>.decorator6   s>    9i))'59!<			> 
	> r   )r   r   )r   r   r   r   s   ``  r   r   r   2   s&    ""7E22 r      )r   c                 D    |dk  r|}|r| |z  r|t        | |z        z  S | S N   )int)valsection_alignmentfile_alignments      r   cache_adjust_SectionAlignmentr&   C   s8    6!* S#44 C.?(?$@AAJr   c                 $    | j                  d      S Nr   )count)datas    r   count_zeroesr+   S   s    ::a=r                    iMZ  iZM  iNE  iLE  iLX  iVZ  iPE             l            i  i  c                 T    t        | D cg c]  }|d   |d   f c}| z         S c c}w )N   r   )dict)pairses     r   two_way_dictr8      s,    u-!!A$!-566-s   %))IMAGE_DIRECTORY_ENTRY_EXPORTr   )IMAGE_DIRECTORY_ENTRY_IMPORTr4   )IMAGE_DIRECTORY_ENTRY_RESOURCE   )IMAGE_DIRECTORY_ENTRY_EXCEPTION   )IMAGE_DIRECTORY_ENTRY_SECURITY   )IMAGE_DIRECTORY_ENTRY_BASERELOC   )IMAGE_DIRECTORY_ENTRY_DEBUG   )IMAGE_DIRECTORY_ENTRY_COPYRIGHT   )IMAGE_DIRECTORY_ENTRY_GLOBALPTR   )IMAGE_DIRECTORY_ENTRY_TLS	   )!IMAGE_DIRECTORY_ENTRY_LOAD_CONFIG
   )"IMAGE_DIRECTORY_ENTRY_BOUND_IMPORT   )IMAGE_DIRECTORY_ENTRY_IAT   )"IMAGE_DIRECTORY_ENTRY_DELAY_IMPORT   )$IMAGE_DIRECTORY_ENTRY_COM_DESCRIPTOR   )IMAGE_DIRECTORY_ENTRY_RESERVED   ))IMAGE_FILE_RELOCS_STRIPPEDr4   )IMAGE_FILE_EXECUTABLE_IMAGEr<   )IMAGE_FILE_LINE_NUMS_STRIPPEDr@   )IMAGE_FILE_LOCAL_SYMS_STRIPPEDrH   )IMAGE_FILE_AGGRESIVE_WS_TRIMr1   )IMAGE_FILE_LARGE_ADDRESS_AWAREr0   )IMAGE_FILE_16BIT_MACHINE@   )IMAGE_FILE_BYTES_REVERSED_LO   )IMAGE_FILE_32BIT_MACHINE   )IMAGE_FILE_DEBUG_STRIPPEDr.   )"IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP   )IMAGE_FILE_NET_RUN_FROM_SWAPr   )IMAGE_FILE_SYSTEMr!   )IMAGE_FILE_DLLr-   )IMAGE_FILE_UP_SYSTEM_ONLY @  )IMAGE_FILE_BYTES_REVERSED_HIr/   ).)IMAGE_SCN_TYPE_REGr   )IMAGE_SCN_TYPE_DSECTr4   )IMAGE_SCN_TYPE_NOLOADr<   )IMAGE_SCN_TYPE_GROUPr@   )IMAGE_SCN_TYPE_NO_PADrH   )IMAGE_SCN_TYPE_COPYr1   )IMAGE_SCN_CNT_CODEr0   )IMAGE_SCN_CNT_INITIALIZED_DATAr^   ) IMAGE_SCN_CNT_UNINITIALIZED_DATAr`   )IMAGE_SCN_LNK_OTHERrb   )IMAGE_SCN_LNK_INFOr.   )IMAGE_SCN_LNK_OVERre   )IMAGE_SCN_LNK_REMOVEr   )IMAGE_SCN_LNK_COMDATr!   )IMAGE_SCN_MEM_PROTECTEDrj   )IMAGE_SCN_NO_DEFER_SPEC_EXCrj   )IMAGE_SCN_GPRELr/   )IMAGE_SCN_MEM_FARDATAr/   )IMAGE_SCN_MEM_SYSHEAP   )IMAGE_SCN_MEM_PURGEABLE   )IMAGE_SCN_MEM_16BITr   )IMAGE_SCN_MEM_LOCKEDi   )IMAGE_SCN_MEM_PRELOADi   )IMAGE_SCN_ALIGN_1BYTESr,   )IMAGE_SCN_ALIGN_2BYTESi    )IMAGE_SCN_ALIGN_4BYTESi  0 )IMAGE_SCN_ALIGN_8BYTESi  @ )IMAGE_SCN_ALIGN_16BYTESi  P )IMAGE_SCN_ALIGN_32BYTESi  ` )IMAGE_SCN_ALIGN_64BYTESi  p )IMAGE_SCN_ALIGN_128BYTESi   )IMAGE_SCN_ALIGN_256BYTESi   )IMAGE_SCN_ALIGN_512BYTESi   )IMAGE_SCN_ALIGN_1024BYTESi   )IMAGE_SCN_ALIGN_2048BYTESi   )IMAGE_SCN_ALIGN_4096BYTESi   )IMAGE_SCN_ALIGN_8192BYTESi   )IMAGE_SCN_ALIGN_MASKi   )IMAGE_SCN_LNK_NRELOC_OVFLi   )IMAGE_SCN_MEM_DISCARDABLEi   )IMAGE_SCN_MEM_NOT_CACHEDi   )IMAGE_SCN_MEM_NOT_PAGED   )IMAGE_SCN_MEM_SHARED   )IMAGE_SCN_MEM_EXECUTEi    )IMAGE_SCN_MEM_READi   @)IMAGE_SCN_MEM_WRITEr2   ))IMAGE_DEBUG_TYPE_UNKNOWNr   )IMAGE_DEBUG_TYPE_COFFr4   )IMAGE_DEBUG_TYPE_CODEVIEWr<   )IMAGE_DEBUG_TYPE_FPOr>   )IMAGE_DEBUG_TYPE_MISCr@   )IMAGE_DEBUG_TYPE_EXCEPTIONrB   )IMAGE_DEBUG_TYPE_FIXUPrD   )IMAGE_DEBUG_TYPE_OMAP_TO_SRCrF   )IMAGE_DEBUG_TYPE_OMAP_FROM_SRCrH   )IMAGE_DEBUG_TYPE_BORLANDrJ   )IMAGE_DEBUG_TYPE_RESERVED10rL   )IMAGE_DEBUG_TYPE_CLSIDrN   )IMAGE_DEBUG_TYPE_VC_FEATURErP   )IMAGE_DEBUG_TYPE_POGOrR   )IMAGE_DEBUG_TYPE_ILTCGrT   )IMAGE_DEBUG_TYPE_MPXrV   )IMAGE_DEBUG_TYPE_REPROr1   )&IMAGE_DEBUG_TYPE_EX_DLLCHARACTERISTICS   ))IMAGE_SUBSYSTEM_UNKNOWNr   )IMAGE_SUBSYSTEM_NATIVEr4   )IMAGE_SUBSYSTEM_WINDOWS_GUIr<   )IMAGE_SUBSYSTEM_WINDOWS_CUIr>   )IMAGE_SUBSYSTEM_OS2_CUIrB   )IMAGE_SUBSYSTEM_POSIX_CUIrF   )IMAGE_SUBSYSTEM_NATIVE_WINDOWSrH   )IMAGE_SUBSYSTEM_WINDOWS_CE_GUIrJ   )IMAGE_SUBSYSTEM_EFI_APPLICATIONrL   )'IMAGE_SUBSYSTEM_EFI_BOOT_SERVICE_DRIVERrN   )"IMAGE_SUBSYSTEM_EFI_RUNTIME_DRIVERrP   )IMAGE_SUBSYSTEM_EFI_ROMrR   )IMAGE_SUBSYSTEM_XBOXrT   )(IMAGE_SUBSYSTEM_WINDOWS_BOOT_APPLICATIONr1   )$)IMAGE_FILE_MACHINE_UNKNOWNr   )IMAGE_FILE_MACHINE_I386iL  )IMAGE_FILE_MACHINE_R3000ib  )IMAGE_FILE_MACHINE_R4000if  )IMAGE_FILE_MACHINE_R10000ih  )IMAGE_FILE_MACHINE_WCEMIPSV2ii  )IMAGE_FILE_MACHINE_ALPHAi  )IMAGE_FILE_MACHINE_SH3i  )IMAGE_FILE_MACHINE_SH3DSPi  )IMAGE_FILE_MACHINE_SH3Ei  )IMAGE_FILE_MACHINE_SH4i  )IMAGE_FILE_MACHINE_SH5i  )IMAGE_FILE_MACHINE_ARMi  )IMAGE_FILE_MACHINE_THUMBi  )IMAGE_FILE_MACHINE_ARMNTi  )IMAGE_FILE_MACHINE_AM33i  )IMAGE_FILE_MACHINE_POWERPCi  )IMAGE_FILE_MACHINE_POWERPCFPi  )IMAGE_FILE_MACHINE_IA64r.   )IMAGE_FILE_MACHINE_MIPS16if  )IMAGE_FILE_MACHINE_ALPHA64  )IMAGE_FILE_MACHINE_AXP64r   )IMAGE_FILE_MACHINE_MIPSFPUif  )IMAGE_FILE_MACHINE_MIPSFPU16if  )IMAGE_FILE_MACHINE_TRICOREi   )IMAGE_FILE_MACHINE_CEFi  )IMAGE_FILE_MACHINE_EBCi  )IMAGE_FILE_MACHINE_RISCV32i2P  )IMAGE_FILE_MACHINE_RISCV64idP  )IMAGE_FILE_MACHINE_RISCV128i(Q  )IMAGE_FILE_MACHINE_LOONGARCH32i2b  )IMAGE_FILE_MACHINE_LOONGARCH64idb  )IMAGE_FILE_MACHINE_AMD64id  )IMAGE_FILE_MACHINE_M32RiA  )IMAGE_FILE_MACHINE_ARM64id  )IMAGE_FILE_MACHINE_CEEi  ))IMAGE_REL_BASED_ABSOLUTEr   )IMAGE_REL_BASED_HIGHr4   )IMAGE_REL_BASED_LOWr<   )IMAGE_REL_BASED_HIGHLOWr>   )IMAGE_REL_BASED_HIGHADJr@   )IMAGE_REL_BASED_MIPS_JMPADDRrB   )IMAGE_REL_BASED_SECTIONrD   )IMAGE_REL_BASED_RELrF   )IMAGE_REL_BASED_MIPS_JMPADDR16rJ   )IMAGE_REL_BASED_IA64_IMM64rJ   )IMAGE_REL_BASED_DIR64rL   )IMAGE_REL_BASED_HIGH3ADJrN   ))IMAGE_LIBRARY_PROCESS_INITr4   )IMAGE_LIBRARY_PROCESS_TERMr<   )IMAGE_LIBRARY_THREAD_INITr@   )IMAGE_LIBRARY_THREAD_TERMrH   )(IMAGE_DLLCHARACTERISTICS_HIGH_ENTROPY_VAr0   )%IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASEr^   )(IMAGE_DLLCHARACTERISTICS_FORCE_INTEGRITYr`   )"IMAGE_DLLCHARACTERISTICS_NX_COMPATrb   )%IMAGE_DLLCHARACTERISTICS_NO_ISOLATIONr.   )IMAGE_DLLCHARACTERISTICS_NO_SEHre   ) IMAGE_DLLCHARACTERISTICS_NO_BINDr   )%IMAGE_DLLCHARACTERISTICS_APPCONTAINERr!   )#IMAGE_DLLCHARACTERISTICS_WDM_DRIVERr-   )!IMAGE_DLLCHARACTERISTICS_GUARD_CFrj   ).IMAGE_DLLCHARACTERISTICS_TERMINAL_SERVER_AWAREr/   ))&IMAGE_DLLCHARACTERISTICS_EX_CET_COMPATr4   )2IMAGE_DLLCHARACTERISTICS_EX_CET_COMPAT_STRICT_MODEr<   )FIMAGE_DLLCHARACTERISTICS_EX_CET_SET_CONTEXT_IP_VALIDATION_RELAXED_MODEr@   ):IMAGE_DLLCHARACTERISTICS_EX_CET_DYNAMIC_APIS_ALLOW_IN_PROCrH   )*IMAGE_DLLCHARACTERISTICS_EX_CET_RESERVED_1r1   )*IMAGE_DLLCHARACTERISTICS_EX_CET_RESERVED_2r0   ))UNW_FLAG_EHANDLERr4   )UNW_FLAG_UHANDLERr<   )UNW_FLAG_CHAININFOr@   ))RAXr   )RCXr4   )RDXr<   )RBXr>   )RSPr@   )RBPrB   )RSIrD   )RDIrF   )R8rH   )R9rJ   )R10rL   )R11rN   )R12rP   )R13rR   )R14rT   )R15rV   r4   r<   r>   r@   rB   rD   rH   rJ   rL   ))	RT_CURSORr4   )	RT_BITMAPr<   )RT_ICONr>   )RT_MENUr@   )	RT_DIALOGrB   )	RT_STRINGrD   )
RT_FONTDIRrF   )RT_FONTrH   )RT_ACCELERATORrJ   )	RT_RCDATArL   )RT_MESSAGETABLErN   )RT_GROUP_CURSORrP   )RT_GROUP_ICONrT   )
RT_VERSIONr1   )RT_DLGINCLUDE   )RT_PLUGPLAY   )RT_VXDr   )RT_ANICURSOR   )
RT_ANIICON   )RT_HTML   )RT_MANIFEST   )^)LANG_NEUTRALr   )LANG_INVARIANT   )LANG_AFRIKAANS6   )LANG_ALBANIAN   )LANG_ARABICr4   )LANG_ARMENIAN+   )LANG_ASSAMESEM   )
LANG_AZERI,   )LANG_BASQUE-   )LANG_BELARUSIAN#   )LANG_BENGALIE   )LANG_BULGARIANr<   )LANG_CATALANr>   )LANG_CHINESEr@   )LANG_CROATIAN   )
LANG_CZECHrB   )LANG_DANISHrD   )LANG_DIVEHIe   )
LANG_DUTCHr)  )LANG_ENGLISHrJ   )LANG_ESTONIAN%   )LANG_FAEROESE8   )
LANG_FARSI)   )LANG_FINNISHrN   )LANG_FRENCHrP   )LANG_GALICIANV   )LANG_GEORGIAN7   )LANG_GERMANrF   )
LANG_GREEKrH   )LANG_GUJARATIG   )LANG_HEBREWrR   )
LANG_HINDI9   )LANG_HUNGARIANrT   )LANG_ICELANDICrV   )LANG_INDONESIAN!   )LANG_ITALIANr1   )LANG_JAPANESEr'  )LANG_KANNADAK   )LANG_KASHMIRI`   )
LANG_KAZAK?   )LANG_KONKANIW   )LANG_KOREAN   )LANG_KYRGYZr^   )LANG_LATVIAN&   )LANG_LITHUANIAN'   )LANG_MACEDONIAN/   )
LANG_MALAY>   )LANG_MALAYALAML   )LANG_MANIPURIX   )LANG_MARATHIN   )LANG_MONGOLIANP   )LANG_NEPALIa   )LANG_NORWEGIANr   )
LANG_ORIYAH   )LANG_POLISHr,  )LANG_PORTUGUESEr.  )LANG_PUNJABIF   )LANG_ROMANIANr2  )LANG_RUSSIAN   )LANG_SANSKRITO   )LANG_SERBIANrK  )LANG_SINDHIY   )LANG_SLOVAK   )LANG_SLOVENIAN$   )LANG_SPANISHrL   )LANG_SWAHILIA   )LANG_SWEDISH   )LANG_SYRIACZ   )
LANG_TAMILI   )
LANG_TATARD   )LANG_TELUGUJ   )	LANG_THAI   )LANG_TURKISH   )LANG_UKRAINIAN"   )	LANG_URDUr0   )
LANG_UZBEKC   )LANG_VIETNAMESE*   )LANG_GAELIC<   )LANG_MALTESE:   )
LANG_MAORI(   )LANG_RHAETO_ROMANCEr0  )
LANG_SAAMI;   )LANG_SORBIAN.   )	LANG_SUTU0   )LANG_TSONGA1   )LANG_TSWANA2   )
LANG_VENDA3   )
LANG_XHOSA4   )	LANG_ZULU5   )LANG_ESPERANTO   )
LANG_WALON   )LANG_CORNISH   )
LANG_WELSH   )LANG_BRETON   )g)SUBLANG_NEUTRALr   )SUBLANG_DEFAULTr4   )SUBLANG_SYS_DEFAULTr<   )SUBLANG_ARABIC_SAUDI_ARABIAr4   )SUBLANG_ARABIC_IRAQr<   )SUBLANG_ARABIC_EGYPTr>   )SUBLANG_ARABIC_LIBYAr@   )SUBLANG_ARABIC_ALGERIArB   )SUBLANG_ARABIC_MOROCCOrD   )SUBLANG_ARABIC_TUNISIArF   )SUBLANG_ARABIC_OMANrH   )SUBLANG_ARABIC_YEMENrJ   )SUBLANG_ARABIC_SYRIArL   )SUBLANG_ARABIC_JORDANrN   )SUBLANG_ARABIC_LEBANONrP   )SUBLANG_ARABIC_KUWAITrR   )SUBLANG_ARABIC_UAErT   )SUBLANG_ARABIC_BAHRAINrV   )SUBLANG_ARABIC_QATARr1   )SUBLANG_AZERI_LATINr4   )SUBLANG_AZERI_CYRILLICr<   )SUBLANG_CHINESE_TRADITIONALr4   )SUBLANG_CHINESE_SIMPLIFIEDr<   )SUBLANG_CHINESE_HONGKONGr>   )SUBLANG_CHINESE_SINGAPOREr@   )SUBLANG_CHINESE_MACAUrB   )SUBLANG_DUTCHr4   )SUBLANG_DUTCH_BELGIANr<   )SUBLANG_ENGLISH_USr4   )SUBLANG_ENGLISH_UKr<   )SUBLANG_ENGLISH_AUSr>   )SUBLANG_ENGLISH_CANr@   )SUBLANG_ENGLISH_NZrB   )SUBLANG_ENGLISH_EIRErD   )SUBLANG_ENGLISH_SOUTH_AFRICArF   )SUBLANG_ENGLISH_JAMAICArH   )SUBLANG_ENGLISH_CARIBBEANrJ   )SUBLANG_ENGLISH_BELIZErL   )SUBLANG_ENGLISH_TRINIDADrN   )SUBLANG_ENGLISH_ZIMBABWErP   )SUBLANG_ENGLISH_PHILIPPINESrR   )SUBLANG_FRENCHr4   )SUBLANG_FRENCH_BELGIANr<   )SUBLANG_FRENCH_CANADIANr>   )SUBLANG_FRENCH_SWISSr@   )SUBLANG_FRENCH_LUXEMBOURGrB   )SUBLANG_FRENCH_MONACOrD   )SUBLANG_GERMANr4   )SUBLANG_GERMAN_SWISSr<   )SUBLANG_GERMAN_AUSTRIANr>   )SUBLANG_GERMAN_LUXEMBOURGr@   )SUBLANG_GERMAN_LIECHTENSTEINrB   )SUBLANG_ITALIANr4   )SUBLANG_ITALIAN_SWISSr<   )SUBLANG_KASHMIRI_SASIAr<   )SUBLANG_KASHMIRI_INDIAr<   )SUBLANG_KOREANr4   )SUBLANG_LITHUANIANr4   )SUBLANG_MALAY_MALAYSIAr4   )SUBLANG_MALAY_BRUNEI_DARUSSALAMr<   )SUBLANG_NEPALI_INDIAr<   )SUBLANG_NORWEGIAN_BOKMALr4   )SUBLANG_NORWEGIAN_NYNORSKr<   )SUBLANG_PORTUGUESEr<   )SUBLANG_PORTUGUESE_BRAZILIANr4   )SUBLANG_SERBIAN_LATINr<   )SUBLANG_SERBIAN_CYRILLICr>   )SUBLANG_SPANISHr4   )SUBLANG_SPANISH_MEXICANr<   )SUBLANG_SPANISH_MODERNr>   )SUBLANG_SPANISH_GUATEMALAr@   )SUBLANG_SPANISH_COSTA_RICArB   )SUBLANG_SPANISH_PANAMArD   )"SUBLANG_SPANISH_DOMINICAN_REPUBLICrF   )SUBLANG_SPANISH_VENEZUELArH   )SUBLANG_SPANISH_COLOMBIArJ   )SUBLANG_SPANISH_PERUrL   )SUBLANG_SPANISH_ARGENTINArN   )SUBLANG_SPANISH_ECUADORrP   )SUBLANG_SPANISH_CHILErR   )SUBLANG_SPANISH_URUGUAYrT   )SUBLANG_SPANISH_PARAGUAYrV   )SUBLANG_SPANISH_BOLIVIAr1   )SUBLANG_SPANISH_EL_SALVADORr'  )SUBLANG_SPANISH_HONDURASrt  )SUBLANG_SPANISH_NICARAGUAr)  )SUBLANG_SPANISH_PUERTO_RICOr   )SUBLANG_SWEDISHr4   )SUBLANG_SWEDISH_FINLANDr<   )SUBLANG_URDU_PAKISTANr4   )SUBLANG_URDU_INDIAr<   )SUBLANG_UZBEK_LATINr4   )SUBLANG_UZBEK_CYRILLICr<   )SUBLANG_DUTCH_SURINAMr>   )SUBLANG_ROMANIANr4   )SUBLANG_ROMANIAN_MOLDAVIAr<   )SUBLANG_RUSSIANr4   )SUBLANG_RUSSIAN_MOLDAVIAr<   )SUBLANG_CROATIANr4   )SUBLANG_LITHUANIAN_CLASSICr<   )SUBLANG_GAELICr4   )SUBLANG_GAELIC_SCOTTISHr<   )SUBLANG_GAELIC_MANXr>   c                     t         j                  | d      }t        j                  |g       D ]
  }||v s|c S  t        j                  |dg      d   S )N	*unknown*r   )LANGgetSUBLANG)
lang_valuesublang_value	lang_namesublang_names       r   get_sublang_name_for_langrD    sV    [1IM26   $	  ;;}{m4Q77r   c                 z   d}d}|t        |       k  r| ||dz    }t        |      dk  ry t        j                  d|      d   }|dz  }|dk7  rFd|dz  cxk  rt        |       k  r/n n,	 | |||dz  z    j                  d      ||<   |dk\  ry ||dz  z  }|dz  }|t        |       k  ry y # t        $ r |dz  }Y 4w xY w)Nr   r<   z<hutf-16ler4   r>   )lenstructunpackdecodeUnicodeDecodeError)r*   counterlierror_count
data_slicelen_s          r   parse_stringsrR    s    	AK
c$i-!a!e_
z?Q}}T:.q1	Q19dQh3#d)3!!!a$(l3:::F'
 aMA1 c$i- & !q !s   (B) )B:9B:c                     | j                         D cg c]0  }t        |t        t        f      r|j	                  |      r|| |   f2 c}S c c}w )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".
    )keys
isinstancestrbytes
startswith)	flag_dictflag_filterflags      r   retrieve_flagsr\    sM     NN$dS%L)dook.J 
y  s   5Ac                 `    |D ])  \  }}||z  rd| j                   |<   d| j                   |<   + y)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__)obj
flag_fieldflagsr[  values        r   	set_flagsrc    s;      'e:!%CLL!&CLL	'r   c                 &    | dk7  xr | | dz
  z  dk(  S )Nr   r4    )r#   s    r   power_of_tworf    s    !8.aQ..r   c                   .     e Zd Z fdZ fdZd Z xZS )
AddressSetc                 >    t         |           d | _        d | _        y r   )super__init__minmax)self	__class__s    r   rk  zAddressSet.__init__  s    r   c                     t         |   |       | j                  |nt        | j                  |      | _        | j                  || _        y t        | j                  |      | _        y r   )rj  addrl  rm  )rn  rb  ro  s     r   rq  zAddressSet.add  sM    E HH,5#dhh2F HH,5#dhh2Fr   c                 h    | j                   | j                  dS | j                  | j                   z
  S r(   )rl  rm  rn  s    r   diffzAddressSet.diff  s,    HH$(8qQdhh>QQr   )__name__
__module____qualname__rk  rq  rt  __classcell__ro  s   @r   rh  rh    s    
G
Rr   rh  c                   L    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y)!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                 .    || _         || _        d | _        y r   )perva_ptrstring)rn  r}  r~  s      r   rk  z*UnicodeStringWrapperPostProcessor.__init__  s    r   c                     | j                   S )zGet the RVA of the string.)r~  rs  s    r   get_rvaz)UnicodeStringWrapperPostProcessor.get_rva  s    ||r   c                 &    | j                  dd      S )z6Return the escaped UTF-8 representation of the string.utf-8r	   )rJ  rs  s    r   __str__z)UnicodeStringWrapperPostProcessor.__str__  s    {{7$788r   c                 N    | j                   sy | j                   j                  | S )N )r  rJ  )rn  r   s     r   rJ  z(UnicodeStringWrapperPostProcessor.decode  s#    {{!t{{!!4((r   c                     d} y)z>Make this instance None, to express it's no known string type.Nre  rs  s    r   
invalidatez,UnicodeStringWrapperPostProcessor.invalidate   s    r   c                 (   	 | j                   j                  | j                  dz   | j                               | _        y # t
        $ rH | j                   j                         j                  dj                  | j                  dz                Y y w xY w)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formatrs  s    r   render_pascal_16z2UnicodeStringWrapperPostProcessor.render_pascal_16  s}    	''55q T-F-F-H 6 DK  	GG  "))66<fT\\A=M6N	s   =A   ABBc                 8    | j                  | j                        S r   )9_UnicodeStringWrapperPostProcessor__get_word_value_at_rvar~  rs  s    r   r  z6UnicodeStringWrapperPostProcessor.get_pascal_16_length  s    ++DLL99r   c                     	 | j                   j                  |d      }t        |      dk  ryt	        j
                  d|      d   S # t        $ r Y yw xY w)Nr<   F<Hr   )r}  get_datar  rG  rH  rI  )rn  rvar*   s      r   __get_word_value_at_rvaz9UnicodeStringWrapperPostProcessor.__get_word_value_at_rva  sW    	77##C+D t9q=}}T4(++  		s   A 	AAc                 \    | j                  |dz
        dk(  r|| j                  z
  | _        yy)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)rn  next_rva_ptrs     r   ask_unicode_16z0UnicodeStringWrapperPostProcessor.ask_unicode_16  s2     ''q(89Q>&5DKr   c                     	 | j                   j                  | j                        | _        y # t        $ rE | j                   j                         j                  dj                  | j                               Y y w xY w)NzDFailed rendering unicode string, attempting to read from RVA 0x{0:x})r}  r  r~  r  r  r  r  r  rs  s    r   render_unicode_16z3UnicodeStringWrapperPostProcessor.render_unicode_16)  s^    	''55dllCDK 	GG  "))66<fT\\6J	s   *- AA;:A;N)ru  rv  rw  __doc__rk  r  r  rJ  r  r  r  r  r  r  re  r   r   r{  r{    s9    "

9)
	:	,
r   r{  c                       e Zd ZdZd Zd Zy)r  z"Generic PE format error exception.c                     || _         y r   )rb  )rn  rb  s     r   rk  zPEFormatError.__init__6  s	    
r   c                 ,    t        | j                        S r   )reprrb  rs  s    r   r  zPEFormatError.__str__9  s    DJJr   N)ru  rv  rw  r  rk  r  re  r   r   r  r  3  s    , r   r  c                   @    e Zd ZdZd Zd
dZd
dZd
dZd Zd Z	d Z
y	)Dumpz1Convenience class for dumping the PE information.c                     g | _         y r   )textrs  s    r   rk  zDump.__init__@  s	    	r   c                 6    |D ]  }| j                  ||        y)zeAdds a list of lines.

        The list can be indented with the optional argument 'indent'.
        N)add_line)rn  txtindentlines       r   	add_lineszDump.add_linesC  s!    
  	(DMM$'	(r   c                 .    | j                  |dz   |       y)z\Adds a line.

        The line can be indented with the optional argument 'indent'.
        
N)rq  rn  r  r  s      r   r  zDump.add_lineK  s    
 	tV$r   c                 `    | j                   j                  dj                  d|z  |             y)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  s      r   rq  zDump.addR  s%    
 			vs;<r   c                 F    | j                  dj                  d|             y)zAdds a header element.z
{0}{1}{0}
z
----------N)r  r  )rn  r  s     r   
add_headerzDump.add_headerY  s    m**8S9:r   c                 :    | j                   j                  d       y)zAdds a newline.r  N)r  r  rs  s    r   add_newlinezDump.add_newline]  s    		r   c                 F    dj                  d | j                  D              S )z"Get the text in its current state.r  c              3   >   K   | ]  }d j                  |        yw){0}N)r  ).0bs     r   	<genexpr>z Dump.get_text.<locals>.<genexpr>c  s     :1u||A:s   )joinr  rs  s    r   get_textzDump.get_texta  s    ww:		:::r   Nr   )ru  rv  rw  r  rk  r  r  rq  r  r  r  re  r   r   r  r  =  s(    ;(%=;;r   r  )xcr  BhHrN  IrM  Lr   qQdsc           	      :   d}| }| d   t         j                  v rmt        dj                  | D cg c]  }|t         j                  v s| c}            }dj                  | D cg c]  }|t         j                  vs| c}      }t        |   |z  S c c}w c c}w )Nr4   r   r  )r  digitsr"   r  STRUCT_SIZEOF_TYPES)tr)   _tr  s       r   sizeof_typer  y  s    E	
Btv}}BGG@1Q&---?Q@ABWW=Aav}}&<a=>r"U** A=s   BB$B<BT)r   r   c           
         d}g }i }g }d}d}| D ]  }d|v s|j                  dd      \  }}	||z  }|j                  d        |	j                  d      }
g }|
D ]Z  }	|	|v r>|D cg c]  }|d t        |	        }}|j                  |	      }dj	                  |	|      }	|j                  |	       |||	<   \ |t        |      z  }|j                  |        t        j                  |      }|||||fS c c}w )N<r   ,r4   z	{0}_{1:d})splitr  rG  r)   r  r  rH  calcsize)r  __format_str____unpacked_data_elms____field_offsets____keys____format_length__offsetelmelm_typeelm_name	elm_namesnamesr  search_list	occ_counts                  r   
set_formatr    s8   NHF ##:!$3!2Hhh&N"))$/ s+IE% 5x'?G"H!1_s8}#5"HK"H + 1 1( ;I*11(IFHX&.4!(+5 k(++F
 OOE"+#. 7 	  #Is   "C1c                   n    e Zd ZdZddZdefdZd Zd Zd Z	d	 Z
d
 Zd Zd Zd Zd Zd ZddZd Zy)	StructurezPrepare structure object to extract members from data.

    Format is a list containing definitions for the elements
    of the structure.
    Nc                 "   d| _         g | _        d| _        i | _        g | _        |d   }t        |t              st        |      }t        |      \  | _         | _        | _        | _        | _        d| _        || _	        |r|| _
        y |d   | _
        y )Nr  r   r4   F)r  r  r  r  r  rU  tupler  __all_zeroes____file_offset__name)rn  r  r  file_offsetr  s        r   rk  zStructure.__init__  s    !!"!#&(#1I!U#aA qM	
'"M" $*DIq	DIr   returnc                     | j                   S r   )r  rs  s    r   __get_format__zStructure.__get_format__      """r   c                 :    | j                   | j                  |   z   S )zLReturn the offset within the field for the requested field in the structure.)r  r  rn  
field_names     r   get_field_absolute_offsetz#Structure.get_field_absolute_offset  s    ##d&<&<Z&HHHr   c                      | j                   |   S )z?Return the offset within the structure for the requested field.)r  r  s     r   get_field_relative_offsetz#Structure.get_field_relative_offset  s    %%j11r   c                     | j                   S r   r  rs  s    r   get_file_offsetzStructure.get_file_offset  s    ###r   c                     || _         y r   r  rn  r  s     r   set_file_offsetzStructure.set_file_offset  s
    %r   c                     | j                   S )z/Returns true is the unpacked data is all zeros.)r  rs  s    r   
all_zeroeszStructure.all_zeroes  s     """r   c                     | j                   S )zReturn size of the structure.)r  rs  s    r   sizeofzStructure.sizeof  s     %%%r   c                    |}t        |      | j                  kD  r|d | j                   }n#t        |      | j                  k  rt        d      t        |      t        |      k(  rd| _        t        j                  | j                  |      | _        t        | j                        D ]&  \  }}| j                  |   D ]  }t        | ||        ( y )Nz-Data length less than expected header length.T)rG  r  r  r+   r  rH  rI  r  r  	enumerater  setattr)rn  r*   idxr#   keys        r   
__unpack__zStructure.__unpack__  s    t9t---0$001D Y/// OPPT*"&D&,mmD4G4G&N#!$"="=> 	(HC}}S) (c3'(	(r   c                     g }t        | j                        D ]?  \  }}d }| j                  |   D ]  }t        | |      }||k7  s n |j	                  |       A t        j                  | j                  g| S r   )r  r  r  getattrr  rH  packr  )rn  
new_valuesr  r#   new_valr  s         r   __pack__zStructure.__pack__  s    
!$"="=> 	'HCG}}S) !$, c> g&	' {{4..<<<r   c                 @    dj                  | j                               S )Nr  )r  dumprs  s    r   r  zStructure.__str__  s    yy%%r   c           
          ddj                  | j                         D cg c]!  }dj                  |j                               # c}      z  S c c}w )Nz<Structure: %s>r  )r  r  r  )rn  r  s     r   __repr__zStructure.__repr__  s>     HH499;?achhqwwy)?@
 	
?s   &A
c           
      >   g }|j                  dj                  | j                               t        j                  D cg c]   }|t        j
                  vst        |      " }}| j                  D ]  }|D ]{  }t        | |      }t        |t        t        f      rn|j                  d      rdj                  |      }ndj                  |      }|dk(  s|dk(  r	 |dt        j                  t        j                  |            z  z  }nt#        |      }|j                  d	      r>d
j%                  |j'                  d      D cg c]  }dj                  |       c}      }nLd
j%                  |j'                  d      D cg c]"  }||v rt)        |      ndj                  |      $ c}      }|j                  d| j*                  |   | j,                  z   | j*                  |   |dz   |fz         ~  |S c c}w # t         $ r |dz  }Y _w xY wc c}w c c}w )z1Returns a string representation of the structure.z[{0}]
Signature_z{:<8X}z0x{:<8X}TimeDateStampdwTimeStampz	 [%s UTC]z [INVALID TIME]	Signaturer      z{:02X}z	\x{0:02x}z0x%-8X 0x%-3X %-30s %s:)r  r  r  r  	printable
whitespaceordr  r  rU  r"   longrX  timeasctimegmtime
ValueError	bytearrayr  rstripchrr  r  )	rn  indentationr  rN  printable_bytesrT  r  r#   val_strs	            r   r  zStructure.dump  s    GNN499-. #,,
9J9J0JCF
 
 MM %	D $dC(cC;/~~l3"*//#"6","3"3C"8o-1E9#{T\\$++cBR5S'SSG (nG~~k2"$''9@9PQAX__Q/Q# #%''
 *1)@	 %& %&$8 !$A%1%8%8%;!<# ,..s3d6J6JJ..s3c		9$%	N [
"  * 9#'88G9 Rs)   G<G<.HH'HHHc                 "   i }| j                   |d<   | j                  D ]  }|D ]  }t        | |      }t        |t        t
        f      r9|dk(  s|dk(  rp	 d|t        j                  t        j                  |            fz  }nAdj                  d |D cg c]  }t        |t              st        |      n|! c}D              }| j                  |   | j                  z   | j                  |   |d||<     |S # t        $ r d|z  }Y Fw xY wc c}w )	z5Returns a dictionary representation of the structure.r  r  r  z0x%-8X [%s UTC]z0x%-8X [INVALID TIME]r  c              3   r   K   | ]/  }t        |      t        j                  v rt        |      nd |z   1 yw)z\x%02xN)r"  r  r  )r  r  s     r   r  z&Structure.dump_dict.<locals>.<genexpr>e  s6      " #&a&F,<,<"<A)a-O"s   57)
FileOffsetOffsetValue)r  r  r  rU  r"   r  r  r  r  r  r  r  r  r  )rn  	dump_dictrT  r  r#   r  s         r   r+  zStructure.dump_dictO  s,    	!%	+ MM 	D dC(cC;/o-1E@"3 # $T[[-= >7 #C '' "SV!Wa
1c0B#a&"I!W" C #'"8"8"=@T@T"T"44S9 "	##	0   * @"9C"?C@
 "Xs   -C8$D8D	D	NNr  )ru  rv  rw  r  rk  rV  r  r  r  r  r  r  r   r  r  r  r  r  r+  re  r   r   r  r    sV    "8# #I2$&#
&
(*=&

4l!r   r  c                   ~     e Zd ZdZ fdZd Z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 xZS )SectionStructurez#Convenience section handling class.c                     d|v r|d   | _         |d= d | _        d | _        d | _        d | _        t        |   |i | d | _        d | _        d | _	        d | _
        y )Nr}  )r}  PointerToRawDataVirtualAddressSizeOfRawDataMisc_VirtualSizerj  rk  PointerToRawData_adjVirtualAddress_adjsection_min_addrsection_max_addr)rn  r   r   ro  s      r   rk  zSectionStructure.__init__v  sn    6>TlDGt $"! $$)&)$(!"& $ $r   c                 2   | j                   | j                  t| j                  j                  | j                        }| j                  j                  j
                  dk  r%| j                  | j                  k(  r| j                  }|| _         | j                   S r    )r4  r0  r}  adjust_PointerToRawDataOPTIONAL_HEADERSectionAlignmentr1  )rn  ptrds     r   get_PointerToRawData_adjz)SectionStructure.get_PointerToRawData_adj  s    $$,1F1F1R772243H3HID
 ww&&77&@((D,?,??..D(,D%(((r   c                    | j                   t| j                  h| j                  j                  | j                  | j                  j                  j
                  | j                  j                  j                        | _         | j                   S r   )r5  r1  r}  adjust_SectionAlignmentr:  r;  FileAlignmentrs  s    r   get_VirtualAddress_adjz'SectionStructure.get_VirtualAddress_adj  sn    ""*"".*.''*I*I''GG++<<GG++99+'
 &&&r   c                    || j                         }n$|| j                         z
  | j                         z   }|||z   }n| j                  || j                  z   }n|}|r||t        ||| j                  z         }| j
                  A| j                  5|| j
                  | j                  z   kD  r| j
                  | j                  z   }| 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.

        Note that sections on disk can include padding that would
        not be loaded to memory. That is the case if `section.SizeOfRawData`
        is greater than `section.Misc_VirtualSize`, and that means
        that data past `section.Misc_VirtualSize` is padding.
        In case you are not interested in this padding, passing
        `ignore_padding=True` will truncate the result in order
        not to return the padding (if any).

        Returns bytes().
        )r=  rA  r2  rl  r3  r0  r}  __data__)rn  startr  ignore_paddingr  ends         r   r  zSectionStructure.get_data  s    & =224F 3355--/0F 6/C+4---CCco&2Dc6D$9$99:C
   ,1C1C1OT**T-?-???++d.@.@@wws++r   c                    |dk(  rt        t        d      }t        | ||       nOd|v rKt        | |      r?|r| j                  dxx   t        |   z  cc<   n| j                  dxx   t        |   z  cc<   || j                  |<   y )NCharacteristics
IMAGE_SCN_)r\  SECTION_CHARACTERISTICSrc  hasattrr^  )rn  r  r#   section_flagss       r   __setattr__zSectionStructure.__setattr__  s~    $$*+BLQM dC/T!gdD&9/04KD4QQ0/04KD4QQ0!dr   c                 J    || j                         z
  | j                         z   S r   )r=  rA  r  s     r   get_rva_from_offsetz$SectionStructure.get_rva_from_offset  s$    5577$:U:U:WWWr   c                 J    || j                         z
  | j                         z   S r   )rA  r=  rn  r  s     r   get_offset_from_rvaz$SectionStructure.get_offset_from_rva  s$    T0022T5R5R5TTTr   c                 v    | j                   y| j                         }||cxk  xr || j                  z   k  S c S )z<Check whether the section contains the file offset provided.F)r0  r=  r2  )rn  r  r4  s      r   contains_offsetz SectionStructure.contains_offset  sJ       ( #<<> FV-ADDVDV-VV	
V	
r   c                 B   | j                   0| j                  $| j                   |cxk  xr | j                  k  S c S | j                         }t        | j                  j
                        | j                         z
  | j                  k  r| j                  }n t        | j                  | j                        }| j                  :| j                  | j                  kD  r!||z   | j                  kD  r| j                  |z
  }|| _         ||z   | _        ||cxk  xr ||z   k  S c S )z8Check whether the section contains the address provided.)r6  r7  rA  rG  r}  rC  r=  r2  r3  rm  next_section_virtual_addressr1  )rn  r  r5  sizes       r   contains_rvazSectionStructure.contains_rva  s      ,1F1F1R((CG$2G2GGGGG!88: tww 4#@#@#BBTEWEWW ((Dt))4+@+@AD --911D4G4GG"T)D,M,MM447IID 2 2T 9!SD+=+DDDDDr   c                 $    | j                  |      S r   )rX  rQ  s     r   containszSectionStructure.contains  s      %%r   c                 @    | j                  | j                               S )z1Calculate and return the entropy for the section.)	entropy_Hr  rs  s    r   get_entropyzSectionStructure.get_entropy  s     ~~dmmo..r   c                 ^    t         't        | j                               j                         S y)z/Get the SHA-1 hex-digest of the section's data.N)r   r  	hexdigestrs  s    r   get_hash_sha1zSectionStructure.get_hash_sha1  s)     (2244 r   c                 ^    t         't        | j                               j                         S y)z1Get the SHA-256 hex-digest of the section's data.N)r   r  r_  rs  s    r   get_hash_sha256z SectionStructure.get_hash_sha256  )     $--/*4466 r   c                 ^    t         't        | j                               j                         S y)z1Get the SHA-512 hex-digest of the section's data.N)r   r  r_  rs  s    r   get_hash_sha512z SectionStructure.get_hash_sha512#  rc  r   c                 ^    t         't        | j                               j                         S y)z-Get the MD5 hex-digest of the section's data.N)r   r  r_  rs  s    r   get_hash_md5zSectionStructure.get_hash_md5)  s(     ?t}}'1133 r   c                     |syt        t        |            }d}|j                         D ]5  }t        |      t	        |      z  }||t        j                  |d      z  z  }7 |S )z)Calculate the entropy of a chunk of data.g        r   r<   )r   r   valuesfloatrG  mathlog)rn  r*   
occurencesentropyr  p_xs         r   r\  zSectionStructure.entropy_H/  sh     Yt_-
""$ 	.A(SY&CsTXXc1---G	. r   )NNF)ru  rv  rw  r  rk  r=  rA  r  rM  rO  rR  rT  rX  rZ  r]  r`  rb  re  rg  r\  rx  ry  s   @r   r.  r.  s  s[    -%)'*,X"XU

$EL&/
5774r   r.  c                 ,    G d d      }g }i } |||      }| d   D ]  }d|vr"|j                          |j                  |       )|j                  dd      \  }}d|v rt        d      |j                  dd      \  }}t	        |      }||j                         k7  s||j                         kD  r!|j                          |j                  |       |j                  ||        |j                          t        t        |            \  }	}
}}}g }t        |      D ]j  \  }}||vr|j                  |       ||   \  }
}|D cg c]  }|t        j                     g }}|j                  |       |D ]  }||d      ||d   <    l |	|||||fS c c}w )Nc                   6    e Zd Zd Zd Zd Zd Zd Zd Zd Z	y)	)set_bitfields_format.<locals>.Accumulatorc                 X    g | _         d| _        d | _        d| _        || _        || _        y )N~r   )
_subfields_name_type
_bits_left_comp_fields_format)rn  fmtcomp_fieldss      r   rk  z2set_bitfields_format.<locals>.Accumulator.__init__B  s/     DO DJDJDO +DDLr   c                 (   | j                   y | j                  j                  | j                   dz   | j                  z          | j                   | j                  f| j
                  t        | j                        dz
  <   d| _        d | _         g | _        y )Nr  r4   rt  )rw  rz  r  rv  ru  ry  rG  rs  s    r   wrap_upz1set_bitfields_format.<locals>.Accumulator.wrap_upL  ss    zz!LL

S 04:: =>8<

DOO7TDc$,,/!34DJDJ DOr   c                 4    t         |   dz  | _        || _        y NrH   )r  rx  rw  )rn  tps     r   new_typez2set_bitfields_format.<locals>.Accumulator.new_typeU  s    1"59DODJr   c                     | xj                   |z  c_         | xj                  |z  c_        | j                  j                  ||f       y r   )rv  rx  ru  r  )rn  r  bitcnts      r   add_subfieldz6set_bitfields_format.<locals>.Accumulator.add_subfieldY  s4    JJ$JOOv%OOO""D&>2r   c                     | j                   S r   )rw  rs  s    r   get_typez2set_bitfields_format.<locals>.Accumulator.get_type^      ::r   c                     | j                   S r   )rv  rs  s    r   get_namez2set_bitfields_format.<locals>.Accumulator.get_namea  r  r   c                     | j                   S r   )rx  rs  s    r   get_bits_leftz7set_bitfields_format.<locals>.Accumulator.get_bits_leftd  s    ??"r   N)
ru  rv  rw  rk  r~  r  r  r  r  r  re  r   r   Accumulatorrr  A  s%    		!		3
			#r   r  r4   r  r  z3Structures with bitfields do not support unions yetr   )r~  r  r  NotImplementedErrorr"   r  r  r  r  r  r  r  StructureWithBitfieldsBTF_NAME_IDXextend)r  r  old_fmtr|  acr  r  r  elm_bits
format_str_field_offsetsrT  format_lengthextended_keysr  r#   sbfr   bf_namesns                        r   set_bitfields_formatr  ?  s   $# $#L GK	Wk	*Bay ,c>JJLNN3 YYsA.((?%E  &^^C3(x=r{{}$23C3C3E(EJJLKK!
(+',( JJL8B5>8R5J=$MdO 8Sk!  %S!3FIJQ-::;<JJX& 	8A"/A"7M!A$	88 }dM;WW Ks   Fc                   f     e Zd ZdZdZdZdZdZddZ fdZ	 fdZ
d fd	Z fdZd	 Zd
 Z x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   r4   c                    t        |      \  | _        | _        | _        | _        | _        | _        t        | j                        D cg c]  }d  c}| _        d| _	        || _
        |d k7  r|| _        y |d   | _        y c c}w )NFr   )r  r  r  r  r  __keys_ext____compound_fields__ranger  r  r  r  )rn  r  r  r  rN  s        r   rk  zStructureWithBitfields.__init__  s     !(	
""M$ 6;4;Q;Q5R&St&S##* DLD	fQi	 'Ts   	A?c                 D    t         |   |       | j                          y r   )rj  r  _unpack_bitfield_attributesrn  r*   ro  s     r   r  z!StructureWithBitfields.__unpack__  s     	4 ((*r   c                     | j                          	 t        | 	         }| j                          |S # | j                          w xY wr   )_pack_bitfield_attributesrj  r  r  r  s     r   r  zStructureWithBitfields.__pack__  sB    &&(	/7#%D,,. ,,.s	   3 Ac                     | j                   }| j                  | _         	 t        |   |      }|| _         |S # || _         w xY wr   )r  r  rj  r  )rn  r#  tkretro  s       r   r  zStructureWithBitfields.dump  sC    ]]))	',{+CDM
 DMs	   8 	Ac                     | j                   }| j                  | _         	 t        |          }|| _         |S # || _         w xY wr   )r  r  rj  r+  )rn  r  r  ro  s      r   r+  z StructureWithBitfields.dump_dict  sB    ]]))	'#%CDM
 DMs	   7 	A c                    | j                   j                         D ]  }| j                  |   d   }t        | |      }t	        | |       d}| j                   |   t
        j                     D ]Z  }d|t
        j                     z  dz
  }||z  }t        | |t
        j                     ||z  |z	         ||t
        j                     z  }\  y)zaReplace compound attributes corresponding to bitfields with separate
        sub-fields.
        r   r4   N)
r  rT  r  r  delattrr  CF_SUBFLD_IDXBTF_BITCNT_IDXr  r  )rn  rN  cf_namecvaloffstsfmasks          r   r  z2StructureWithBitfields._unpack_bitfield_attributes  s     ))..0 	CAmmA&q)G4)DD'"E..q12H2V2VW CR 6 E EFF!K-::;D[U*
 2AABBC	Cr   c                    | j                   j                         D ]  }| j                  |   d   }d\  }}| j                   |   t        j                     D ]Y  }d|t        j
                     z  dz
  }t        | |t        j                           |z  }|||z  z  }||t        j
                     z  }[ t        | ||        y)z(Pack attributes into a compound bitfieldr   r   r   r4   N)	r  rT  r  r  r  r  r  r  r  )rn  rN  r  r  acc_valr  r  	field_vals           r   r  z0StructureWithBitfields._pack_bitfield_attributes  s    ))..0 
	,AmmA&q)G!NE7..q12H2V2VW CR 6 E EFF!KD"%;%H%H"IJTQ  9--2AABBC D'7+
	,r   r,  r  )ru  rv  rw  r  r  r  CF_TYPE_IDXr  rk  r  r  r  r+  r  r  rx  ry  s   @r   r  r    sB    ( LNKM8 +C&,r   r  c                   "     e Zd ZdZ fdZ xZS )DataContainerzGeneric data container.c                 \    t         |   }|j                         D ]  \  }} |||        y r   )rj  rM  items)rn  r   bare_setattrr  rb  ro  s        r   rk  zDataContainer.__init__   s0    w* ,,. 	%JCe$	%r   )ru  rv  rw  r  rk  rx  ry  s   @r   r  r    s    !% %r   r  c                       e Zd ZdZy)ImportDescDatazHolds import descriptor information.

    dll:        name of the imported DLL
    imports:    list of imported symbols (ImportData instances)
    struct:     IMAGE_IMPORT_DESCRIPTOR structure
    Nru  rv  rw  r  re  r   r   r  r        r   r  c                       e Zd ZdZd Zy)
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                    t        | d      rt        | d      rt        | d      r|dk(  r| j                  j                  t        k(  rt        }n#| j                  j                  t
        k(  rt        }|dz  z  | j                  _        | j                  j                  | j                  _	        | j                  j                  | j                  _
        | j                  j                  | j                  _        n|dk(  r| j                  || j                  _	        | j                  j                  | j                  _	        | j                  j                  | j                  _
        | j                  j                  | j                  _        n7|dk(  r|| j                  _	        | j                  j                  | j                  _        | j                  j                  | j                  _
        | j                  j                  | j                  _        n|dk(  r| j                  r| j                  j                  | j                        }| j                  j                  | j                   d|z         t#        |      t#        | j$                        kD  rt'        d      | j                  j)                  | j                  |       || j*                  |<   y )Nordinalboundr    addressr   9The export name provided is longer than the existing one.)rK  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_offsetrO  set_dword_at_offsetordinal_offsetrG  r  r  set_bytes_at_offsetr^  )rn  r  r#   ordinal_flagname_rvas        r   rM  zImportData.__setattr__  s    D)$g&f%y 77??&>>#5LWW__(EE#7L -9C&L,I!!)262C2C2K2K!!/-1->->-F-F!!*484E4E4M4M!!1??.47DOO148OO4Q4QDOO1/3/L/LDOO,6:oo6S6SDOO3"25!!/,0,=,=,K,K!!)-1->->-L-L!!*484E4E4S4S!!1 ###ww::4;K;KLHGG//++g-A 3x#dii.0+W  GG//0@0@#F!dr   Nru  rv  rw  r  rM  re  r   r   r  r    s    2"r   r  c                       e Zd ZdZy)ExportDirDatazHolds export directory information.

    struct:     IMAGE_EXPORT_DIRECTORY structure
    symbols:    list of exported symbols (ExportData instances)Nr  re  r   r   r  r  M  s    Cr   r  c                       e Zd ZdZd Zy)
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                    t        | d      r.t        | d      r!t        | d      rt        | d      r|dk(  r'| j                  j                  | j                  |       n|dk(  r'| j                  j	                  | j
                  |       n|dk(  rSt        |      t        | j                        kD  rt        d      | j                  j                  | j                  |       nW|dk(  rRt        |      t        | j                        kD  rt        d      | j                  j                  | j                  |       || j                  |<   y )Nr  r  	forwarderr  r  z<The forwarder name provided is longer than the existing one.)rK  r}  set_word_at_offsetr  r  address_offsetrG  r  r  r  r  r  forwarder_offsetr^  rn  r  r#   s      r   rM  zExportData.__setattr__`  s    D)$i(k*f%y **4+>+>D"++D,?,?E s8c$))n,'S  ++D,<,<cB$ s8c$..11'V  ++D,A,A3G!dr   Nr  re  r   r   r  r  T  s    	 "r   r  c                       e Zd ZdZy)ResourceDirDatazHolds resource directory information.

    struct:     IMAGE_RESOURCE_DIRECTORY structure
    entries:    list of entries (ResourceDirEntryData instances)
    Nr  re  r   r   r  r        r   r  c                       e Zd ZdZy)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  re  r   r   r  r    s    r   r  c                       e Zd ZdZy)ResourceDataEntryDatazHolds resource data entry information.

    struct:     IMAGE_RESOURCE_DATA_ENTRY structure
    lang:       Primary language ID
    sublang:    Sublanguage ID
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZy)	DebugDatazHolds debug information.

    struct:     IMAGE_DEBUG_DIRECTORY structure
    entries:    list of entries (IMAGE_DEBUG_TYPE instances)
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZy)DynamicRelocationDataa  Holds dynamic relocation information.

    struct:        IMAGE_DYNAMIC_RELOCATION structure
    symbol:        Symbol to which dynamic relocations must be applied
    relocations:   List of dynamic relocations for this symbol (BaseRelocationData instances)
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZy)FunctionOverrideDataa  Holds Function and bdd dynamic relocation information.

    struct:        IMAGE_DYNAMIC_RELOCATION structure
    symbol:        Symbol to which dynamic relocations must be applied
    bdd_relocs:    List of bdd dynamic relocations (BddDynamicRelocationData instances)
    func_relocs:   List of function override dynamic relocations (FunctionOverrideDynamicRelocationData instances)
    Nr  re  r   r   r  r        r   r  c                       e Zd ZdZy)%FunctionOverrideDynamicRelocationDataa0  Holds Function override dynamic relocation information.

    struct:        IMAGE_FUNCTION_OVERRIDE_DYNAMIC_RELOCATION structure
    func_rva:      Original function rva
    override_rvas: List of overriding function rvas
    relocations:   List of dynamic relocations (BaseRelocationData instances)
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZy)BddDynamicRelocationDataziHolds Bdd dynamic relocation information.

    struct:        IMAGE_BDD_DYNAMIC_RELOCATION structure
    Nr  re  r   r   r  r        r   r  c                       e Zd ZdZy)BaseRelocationDatazHolds base relocation information.

    struct:     IMAGE_BASE_RELOCATION structure
    entries:    list of relocation data (RelocationData instances)
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZd Zy)RelocationDatazHolds relocation information.

    type:       Type of relocation
                The type string can be obtained by
                RELOCATION_TYPE[type]
    rva:        RVA of the relocation
    c                     t        | d      ra| j                  j                  }|dk(  r|dz  |dz  z  }n)|dk(  r$t        || j                  z
  d      }|dz  |dz  z  }|| j                  _        || j
                  |<   y )NrH  typerP     r  r   i   )rK  rH  Datarm  base_rvar^  )rn  r  r#   wordr  s        r   rM  zRelocationData.__setattr__  s    
 4" ;;##Dv~r	dUl3S4==0!4v&5.9  $DKK!dr   Nr  re  r   r   r  r    s    "r   r  c                       e Zd ZdZy)TlsDatazJHolds TLS information.

    struct:     IMAGE_TLS_DIRECTORY structure
    Nr  re  r   r   r  r    r  r   r  c                       e Zd ZdZy)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  re  r   r   r  r    s    r   r  c                       e Zd ZdZy)LoadConfigDatazHolds Load Config data.

    struct:     IMAGE_LOAD_CONFIG_DIRECTORY structure
    name:       dll name
    dynamic_relocations: dynamic relocation information, if present
    Nr  re  r   r   r	  r	    r  r   r	  c                       e Zd ZdZy)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  re  r   r   r  r     s    r   r  c                       e Zd ZdZy)ExceptionsDirEntryDatazHolds the data related to SEH (and stack unwinding, in particular)

    struct      an instance of RUNTIME_FUNTION
    unwindinfo  an instance of UNWIND_INFO
    Nr  re  r   r   r  r  +  r  r   r  c                   f     e Zd ZdZd fd	Z fdZd fd	Z fdZd Zd Z	 fdZ
d	 Zd
 Z xZS )
UnwindInfozHandles the complexities of UNWIND_INFO structure:
    * variable number of UWIND_CODEs
    * optional ExceptionHandler and FunctionEntry fields
    c                     t         |   d|       t         | 	         | _        d | _        t        dd      | _        d | _        d| _        y )N)UNWIND_INFO)zB:3,Versionz	B:5,FlagszB,SizeOfPrologzB,CountOfCodeszB:4,FrameRegisterzB:4,FrameOffsetr  UNWIND_CODE)B,CodeOffsetB:4,UnwindOp
B:4,OpInfor   F)	rj  rk  r   
_full_size_opt_field_namer  
_code_info_chained_entry_finished_unpacking)rn  r  ro  s     r   rk  zUnwindInfo.__init__9  s[    
 $ 	 	
  '.*#0K
 ##( r   c                    | j                   ryt        	| 	  |       | j                  dz   dz  }t        	|          || j
                  j	                         z  z   }|| j                  dk(  rdnt        d   z   | _        t        |      | j                  k  ry| j                  dk7  r'| j                  dk7  rdt        | j                        z   S g | _        t        	|          }| j                  }|dkD  r| j
                  j                  |||| j
                  j	                         z           t        j                  | j
                        }|dt        | j                  |z         z   S |j!                  | j
                  |       }| j
                  j	                         |z  }|j#                  | j
                  ||||z    | | j                  |z          ||z  }||z  }| j                  j%                  |       |dkD  r| j&                  s| j(                  rd	| _        | j,                  rd
| _        | j*                  dk7  r;t/        | | j*                  t1        j2                  d|||t        d   z          d          d| _         y)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.
        Nr4   r   r  r<   z&Unsupported version of UNWIND_INFO at zUnknown UNWIND_CODE at ExceptionHandlerFunctionEntry<IT)r  rj  r  CountOfCodesr   r  Flagsr  r  rG  Versionhexr  UnwindCodesPrologEpilogOpsFactorycreatelength_in_code_structures
initializer  r  r  r  r  r  rH  rI  )
rn  r*   codes_cnt_maxhdlr_offsetro
codes_leftucodelen_in_codesopc_sizero  s
            r   unpack_in_stageszUnwindInfo.unpack_in_stagesQ  sW    ##4 **Q."4gn&9O9O9Q)QQ%qA&9#&>
 t9t&<<1!2;c$BVBV>WWWW^&&
1nOO&&tBdoo6L6L6N1N'OP*11$//BE}03t7K7Kb7P3QQQ ::4??DQL--/,>HR"x-($$r)	 (NB,&J##E*! 1n$ !!T%;%;#5D ""#2D 4'$$${[;Ns;S-ST $( r   c                    | j                   d k7  rS| j                  t        d   z
  | j                  | j                   <   | j                  j                  | j                   g       	 t        |   |      }| j                   d k7  r| j                  j                          	 |j                  ddj                  t        D cg c]  }t        | |d         s|d    c}      z          |j                  ddj                  | j                  D cg c]  }|j                         st        |        c}      z          |S # | j                   d k7  r| j                  j                          w w xY wc c}w c c}w )Nr  Flags: , r   zUnwind codes: z; )r  r  r  r  r  r  rj  r  popr  unwind_info_flagsr  r&  is_validrV  )rn  r#  r  r  r  ro  s        r   r  zUnwindInfo.dump  sF    4'"5c":: ""4#7#78 $$d&:&:%;<	(7<,D##t+!!%%'ii'8P!GD!A$<O1PQR	
 	ii)9)9JAQZZ\QJKL	
  ##t+!!%%' ,
 Q Ks$   %D= =E,E,E1!E1=,E)c                    | j                   d k7  rS| j                  t        d   z
  | j                  | j                   <   | j                  j                  | j                   g       	 t        |          }| j                   d k7  r| j                  j                          |S # | j                   d k7  r| j                  j                          w w xY w)Nr  )	r  r  r  r  r  r  rj  r+  r6  )rn  r  ro  s     r   r+  zUnwindInfo.dump_dict  s    4'"5c":: ""4#7#78 $$d&:&:%;<	('#%C##t+!!%%'
 ##t+!!%%' ,s   %B ,C
c                     |dk(  rt        | |t               nOd|v rKt        | |      r?|r| j                  dxx   t        |   z  cc<   n| j                  dxx   t        |   z  cc<   || j                  |<   y )Nr#  	UNW_FLAG_)rc  r7  rK  r^  UNWIND_INFO_FLAGSr  s      r   rM  zUnwindInfo.__setattr__  sl    7?dC!23D WT4%8g&*;D*AA&g&*;D*AA&!dr   c                     | j                   S r   )r  rs  s    r   r   zUnwindInfo.sizeof  s    r   c                 `   t        | j                        }t        |          |dt        |           t        |          }| j
                  D ]  }||j                  j	                         z   | j                  kD  r nW|j                  j                         ||||j                  j	                         z    ||j                  j	                         z  } | j                  d k7  rKt        j                  dt        | | j                              || j                  t        d   z
  | j                   |S )Nr   r!  r  )r   r  rj  r  r   r&  rH  r  r	  r  r  )rn  r*   
cur_offsetucro  s       r   r  zUnwindInfo.__pack__  s   )%*W%5%7Q!"W^%
"" 	-BBII,,..@ACASASAUDj299+;+;+==>"))**,,J		- 4' D'$0D0D"EF "5c"::T__ r   c                     | j                   S r   )r  rs  s    r   get_chained_function_entryz%UnwindInfo.get_chained_function_entry  r  r   c                 F    | j                   d k7  rt        d      || _         y )Nz(Chained function entry cannot be changed)r  r  )rn  entrys     r   set_chained_function_entryz%UnwindInfo.set_chained_function_entry  s$    $& JKK#r   r  )ru  rv  rw  r  rk  r2  r  r+  rM  r   r  rB  rE  rx  ry  s   @r   r  r  3  s6    
)0:x6"$#$r   r  c                   (    e Zd ZdZd Zd Zd Zd Zy)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                 |    t        | j                  |      |      | _        | j                  j                  |       y )Nr  )r  _get_formatrH  r  )rn  unw_coder*   unw_infor  s        r   r*  zPrologEpilogOp.initialize  s2    ,X&K
 	t$r   c                      y)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.
        r4   re  rn  rJ  rK  s      r   r)  z(PrologEpilogOp.length_in_code_structures  s    
 r   c                      y)NTre  rs  s    r   r8  zPrologEpilogOp.is_valid  s    r   c                      y)Nr  re  rn  rJ  s     r   rI  zPrologEpilogOp._get_format  s    Nr   N)ru  rv  rw  r  r*  r)  r8  rI  re  r   r   rG  rG    s    %Or   rG  c                       e Zd ZdZd Zd Zy)PrologEpilogOpPushRegUWOP_PUSH_NONVOLc                      y)N)UNWIND_CODE_PUSH_NONVOL)r  r  B:4,Regre  rP  s     r   rI  z!PrologEpilogOpPushReg._get_format  s    Wr   c                 B    dt         | j                  j                     z   S )Nz	.PUSHREG )	REGISTERSrH  Regrs  s    r   r  zPrologEpilogOpPushReg.__str__  s    Yt{{777r   N)ru  rv  rw  r  rI  r  re  r   r   rR  rR    s    X8r   rR  c                   (    e Zd ZdZd Zd Zd Zd Zy)PrologEpilogOpAllocLargeUWOP_ALLOC_LARGEc                 8    dddd|j                   dk(  rdffS dffS )NUNWIND_CODE_ALLOC_LARGEr  r  r  r   zH,AllocSizeInQwordszI,AllocSizeOpInforP  s     r   rI  z$PrologEpilogOpAllocLarge._get_format  sB    %)1A)=%	
 	
 DQ	
 	
r   c                 (    |j                   dk(  rdS dS )Nr   r<   r>   r_  rM  s      r   r)  z2PrologEpilogOpAllocLarge.length_in_code_structures  s    OOq(q/a/r   c                     | j                   j                  dk(  r| j                   j                  dz  S | j                   j                  S )Nr   rH   )rH  r`  AllocSizeInQwords	AllocSizers  s    r   get_alloc_sizez'PrologEpilogOpAllocLarge.get_alloc_size  sC     {{!!Q& KK))A-	
 &&	
r   c                 :    dt        | j                               z   S Nz.ALLOCSTACK r%  re  rs  s    r   r  z PrologEpilogOpAllocLarge.__str__      D$7$7$9 :::r   N)ru  rv  rw  r  rI  r)  re  r  re  r   r   r[  r[     s    	
0
;r   r[  c                   "    e Zd ZdZd Zd Zd Zy)PrologEpilogOpAllocSmallUWOP_ALLOC_SMALLc                      y)N)UNWIND_CODE_ALLOC_SMALL)r  r  zB:4,AllocSizeInQwordsMinus8re  rP  s     r   rI  z$PrologEpilogOpAllocSmall._get_format      
r   c                 :    | j                   j                  dz  dz   S r  )rH  AllocSizeInQwordsMinus8rs  s    r   re  z'PrologEpilogOpAllocSmall.get_alloc_size%  s    {{22Q6::r   c                 :    dt        | j                               z   S rg  rh  rs  s    r   r  z PrologEpilogOpAllocSmall.__str__(  ri  r   N)ru  rv  rw  r  rI  re  r  re  r   r   rk  rk    s    
;;r   rk  c                   (     e Zd ZdZ fdZd Z xZS )PrologEpilogOpSetFPUWOP_SET_FPREGc                 t    t         |   ||||       |j                  | _        |j                  dz  | _        y Nr1   )rj  r*  FrameRegister_frame_registerFrameOffset_frame_offsetrn  rJ  r*   rK  r  ro  s        r   r*  zPrologEpilogOpSetFP.initialize/  s7    8T8[A'55%11B6r   c                 `    dt         | j                     z   dz   t        | j                        z   S )Nz
.SETFRAME r5  )rX  ry  r%  r{  rs  s    r   r  zPrologEpilogOpSetFP.__str__4  s:    ,,-. $$$%&	
r   )ru  rv  rw  r  r*  r  rx  ry  s   @r   rt  rt  ,  s    7

r   rt  c                   (    e Zd ZdZd Zd Zd Zd Zy)PrologEpilogOpSaveRegUWOP_SAVE_NONVOLc                      yNr<   re  )rn  unwcoderK  s      r   r)  z/PrologEpilogOpSaveReg.length_in_code_structures@      r   c                 4    | j                   j                  dz  S r  )rH  OffsetInQwordsrs  s    r   
get_offsetz PrologEpilogOpSaveReg.get_offsetC  s    {{))A--r   c                      y)N)UNWIND_CODE_SAVE_NONVOL)r  r  rV  zH,OffsetInQwordsre  rP  s     r   rI  z!PrologEpilogOpSaveReg._get_formatF  ro  r   c                 |    dt         | j                  j                     z   dz   t        | j	                               z   S Nz	.SAVEREG r5  )rX  rH  rY  r%  r  rs  s    r   r  zPrologEpilogOpSaveReg.__str__L  s0    Yt{{77$>T__EVAWWWr   Nru  rv  rw  r  r)  r  rI  r  re  r   r   r  r  =  s    .
Xr   r  c                   (    e Zd ZdZd Zd Zd Zd Zy)PrologEpilogOpSaveRegFarUWOP_SAVE_NONVOL_FARc                      yNr>   re  rM  s      r   r)  z2PrologEpilogOpSaveRegFar.length_in_code_structuresS  r  r   c                 .    | j                   j                  S r   rH  r)  rs  s    r   r  z#PrologEpilogOpSaveRegFar.get_offsetV      {{!!!r   c                      y)N)UNWIND_CODE_SAVE_NONVOL_FARr  r  rV  zI,Offsetre  rP  s     r   rI  z$PrologEpilogOpSaveRegFar._get_formatY  ro  r   c                     dt         | j                  j                     z   dz   t        | j                  j                        z   S r  )rX  rH  rY  r%  r)  rs  s    r   r  z PrologEpilogOpSaveRegFar.__str___  s3    Yt{{77$>T[[EWEWAXXXr   Nr  re  r   r   r  r  P  s    "
Yr   r  c                   (    e Zd ZdZd Zd Zd Zd Zy)PrologEpilogOpSaveXMMUWOP_SAVE_XMM128c                      y)N)UNWIND_CODE_SAVE_XMM128)r  r  rV  zH,OffsetIn2Qwordsre  rP  s     r   rI  z!PrologEpilogOpSaveXMM._get_formatf  ro  r   c                      yr  re  rM  s      r   r)  z/PrologEpilogOpSaveXMM.length_in_code_structuresl  r  r   c                 4    | j                   j                  dz  S rw  )rH  OffsetIn2Qwordsrs  s    r   r  z PrologEpilogOpSaveXMM.get_offseto  s    {{**R//r   c                     dt        | j                  j                        z   dz   t        | j	                               z   S Nz.SAVEXMM128 XMMr5  )rV  rH  rY  r%  r  rs  s    r   r  zPrologEpilogOpSaveXMM.__str__r  s0     3t{{#77$>T__EVAWWWr   Nru  rv  rw  r  rI  r)  r  r  re  r   r   r  r  c  s    
0Xr   r  c                   (    e Zd ZdZd Zd Zd Zd Zy)PrologEpilogOpSaveXMMFarUWOP_SAVE_XMM128_FARc                      y)N)UNWIND_CODE_SAVE_XMM128_FARr  re  rP  s     r   rI  z$PrologEpilogOpSaveXMMFar._get_formaty  ro  r   c                      yr  re  rM  s      r   r)  z2PrologEpilogOpSaveXMMFar.length_in_code_structures  r  r   c                 .    | j                   j                  S r   r  rs  s    r   r  z#PrologEpilogOpSaveXMMFar.get_offset  r  r   c                     dt        | j                  j                        z   dz   t        | j                  j                        z   S r  )rV  rH  rY  r%  r)  rs  s    r   r  z PrologEpilogOpSaveXMMFar.__str__  s3     3t{{#77$>T[[EWEWAXXXr   Nr  re  r   r   r  r  v  s    
"Yr   r  c                       e Zd ZdZd Zy)PrologEpilogOpPushFrameUWOP_PUSH_MACHFRAMEc                 @    d| j                   j                  rdz   S dz   S )Nz
.PUSHFRAMEz <code>r  )rH  r`  rs  s    r   r  zPrologEpilogOpPushFrame.__str__  s!    DKK,>,>yGGBGGr   N)ru  rv  rw  r  r  re  r   r   r  r    s    Hr   r  c                   @     e Zd ZdZ fdZd Zd Zd Zd Zd Z	 xZ
S )PrologEpilogOpEpilogMarkerUWOP_EPILOGc                    d| _         t        |d       | _        t        |   ||||       | j                  r8t        |d| j                  j                         |j                  dz  dk(  | _         |j                  | _
        y )NTSizeOfEpilogr4   r   )_long_offstrK  _firstrj  r*  r  rH  Sizer`  r  _epilog_sizer|  s        r   r*  z%PrologEpilogOpEpilogMarker.initialize  sp    !(N;;8T8[A;;Hndkk.>.>?'2a7D$11r   c                 N    | j                   rd|j                  dz  dk(  rdfS dfS y)NUNWIND_CODE_EPILOGr4   )zB,OffsetLow,Sizer  	B:4,Flags)zB,Sizer  r  B,OffsetLowz
B:4,UnusedB:4,OffsetHigh)r  )r  r  r  )r  r`  rP  s     r   rI  z&PrologEpilogOpEpilogMarker._get_format  sC     ;;$??Q&!+ B 	 r   c                 F    t        |d      s|j                  dz  dk(  rdS dS )Nr  r4   r   r<   )rK  r`  rM  s      r   r)  z4PrologEpilogOpEpilogMarker.length_in_code_structures  s3     8^4(//A:MRS9S 	
 	
r   c                     | j                   j                  | j                  r| j                   j                  dz  z  S dz  S )NrH   r   )rH  	OffsetLowr  
OffsetHighrs  s    r   r  z%PrologEpilogOpEpilogMarker.get_offset  s@    {{$$+/+;+;DKK""a'
 	
AB
 	
r   c                 (    | j                         dkD  S r(   )r  rs  s    r   r8  z#PrologEpilogOpEpilogMarker.is_valid  s     1$$r   c                     | j                         dkD  r5dt        | j                        z   dz   t        | j                               z   S dS )Nr   zEPILOG: size=z, offset from the end=-r  )r  r%  r  rs  s    r   r  z"PrologEpilogOpEpilogMarker.__str__  sX      1$	 $##$%'( $//#$%	
 	
r   )ru  rv  rw  r  r*  rI  r)  r  r8  r  rx  ry  s   @r   r  r    s$    2.


%	
r   r  c                   L    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        Zy)r'  zBA factory for creating unwind codes based on the value of UnwindOpc                 p    | j                   }|t        j                  v rt        j                  |          S d S r   )UnwindOpr'  _class_dict)r  codes     r   r(  zPrologEpilogOpsFactory.create  s@     -999 #..t46	
 	
r   N)ru  rv  rw  r  rS  rR  r\  r[  rl  rk  ru  rt  r  r  r  r  r  r  r  r  r  r  r  r  r  staticmethodr(  re  r   r   r'  r'    s^    L 	/22+/6/64/K 
 
r   r'  z!#$%&'()-@^_`{}~+,.;=[]:c                     | t        | t        t        t        f      syt        dz   t        fdt        |       D              S )NFs   \/c              3   &   K   | ]  }|v  
 y wr   re  )r  r  alloweds     r   r  z(is_valid_dos_filename.<locals>.<genexpr>  s     ,qG|,s   )rU  rV  rW  r   allowed_filenameallset)r  r  s    @r   is_valid_dos_filenamer    s:    y
1sE9&=>'G,SV,,,r   r  relax_allowed_charactersr  c                     d|rd| d uxr: t        | t        t        t        f      xr t	        fdt        |       D              S )Ns	   ._?@$()<>s   !"#$%&'()*+,-./:<>?[\]^_`{|}~@c              3   :   K   | ]  }|t         v xs |v   y wr   )allowed_function_name)r  r  allowed_extras     r   r  z)is_valid_function_name.<locals>.<genexpr>	  s$     Sq++AqM/AASs   )rU  rV  rW  r   r  r  )r  r  r  s     @r   is_valid_function_namer  	  sN     !M;	 	Tq3y12	TSCPQFSSr   c                      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!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/Z0d0 Z1d1 Z2d2 Z3d3 Z4d4 Z5d5 Z6d6 Z7d7 Z8d8 Z9d9 Z:d: Z;dd;Z<d< Z=	 dd=Z>d> Z?d? Z@d@ ZAdA ZBdB ZCdC ZDdD ZEddEZFdF ZGdG ZHdH ZIddIZJdJ ZKdK ZLdL ZMddMZNdN ZOdO ZPdP ZQddQZRdR ZSdS ZTddTZU	 	 ddUZVddVZWddWZXdX ZYddYZZdZ Z[d[ Z\e]fd\Z^d] Z_d^ Z`dd_Zad` Zbda Zcdb Zddc Zedd ZfddeZgddfZhdg Zidh Zjdi Zkdj Zldk Zmdl Zndm Zodn Zpdo Zqdp Zrdq Zsdr Ztds Zudt Zvdu Zwdv Zxdw Zydx Zzdy Z{dz Z|d{ Z}d| Z~d}ed~efdZd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zy-)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
    )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)IMAGE_FILE_HEADER)z	H,MachinezH,NumberOfSectionsI,TimeDateStampzI,PointerToSymbolTablezI,NumberOfSymbolszH,SizeOfOptionalHeaderzH,Characteristics)IMAGE_DATA_DIRECTORY)I,VirtualAddressI,Size)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)IMAGE_OPTIONAL_HEADER64)r  r  r  r  r  r  r  r  zQ,ImageBaser  r  r  r  r  r  r  r  r  r  r  r  r  r  zQ,SizeOfStackReservezQ,SizeOfStackCommitzQ,SizeOfHeapReservezQ,SizeOfHeapCommitr  r  )IMAGE_NT_HEADERS)I,Signature)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)IMAGE_DELAY_IMPORT_DESCRIPTOR)z	I,grAttrszI,szNamezI,phmodzI,pIATzI,pINTzI,pBoundIATzI,pUnloadIATzI,dwTimeStamp)IMAGE_IMPORT_DESCRIPTOR)z$I,OriginalFirstThunk,Characteristicsr  zI,ForwarderChainI,NamezI,FirstThunk)IMAGE_EXPORT_DIRECTORY)r  r  H,MajorVersionH,MinorVersionr   zI,BasezI,NumberOfFunctionszI,NumberOfNameszI,AddressOfFunctionszI,AddressOfNameszI,AddressOfNameOrdinals)IMAGE_RESOURCE_DIRECTORY)r  r  r  r  zH,NumberOfNamedEntrieszH,NumberOfIdEntries)IMAGE_RESOURCE_DIRECTORY_ENTRY)r   I,OffsetToData)IMAGE_RESOURCE_DATA_ENTRY)r  r  z
I,CodePagez
I,Reserved)VS_VERSIONINFOzH,LengthzH,ValueLengthzH,Type)VS_FIXEDFILEINFO)r  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)StringFileInfor	  )StringTabler	  )Stringr	  )Varr	  )IMAGE_THUNK_DATA)z0I,ForwarderString,Function,Ordinal,AddressOfData)r  )z0Q,ForwarderString,Function,Ordinal,AddressOfData)IMAGE_DEBUG_DIRECTORY)r  r  r  r  zI,TypezI,SizeOfDatazI,AddressOfRawDatar  )IMAGE_BASE_RELOCATION)r  zI,SizeOfBlock)IMAGE_BASE_RELOCATION_ENTRY)zH,Data)0IMAGE_IMPORT_CONTROL_TRANSFER_DYNAMIC_RELOCATION)zI:12,PageRelativeOffsetzI:1,IndirectCallzI:19,IATIndex)/IMAGE_INDIR_CONTROL_TRANSFER_DYNAMIC_RELOCATION)H:12,PageRelativeOffsetzH:1,IndirectCallzH:1,RexWPrefixzH:1,CfgCheckzH:1,Reserved)+IMAGE_SWITCHTABLE_BRANCH_DYNAMIC_RELOCATION)r  zH:4,RegisterNumber)IMAGE_FUNCTION_OVERRIDE_HEADER)zI,FuncOverrideSize)*IMAGE_FUNCTION_OVERRIDE_DYNAMIC_RELOCATION)zI,OriginalRvazI,BDDOffsetz	I,RvaSizeI,BaseRelocSize)IMAGE_BDD_INFO)	I,Versionz	I,BDDSize)IMAGE_BDD_DYNAMIC_RELOCATION)zH,LeftzH,RightzI,Value)IMAGE_TLS_DIRECTORY)zI,StartAddressOfRawDatazI,EndAddressOfRawDatazI,AddressOfIndexzI,AddressOfCallBacksI,SizeOfZeroFillr  )r  )zQ,StartAddressOfRawDatazQ,EndAddressOfRawDatazQ,AddressOfIndexzQ,AddressOfCallBacksr  r  )IMAGE_LOAD_CONFIG_DIRECTORY)4r  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,DependentLoadFlagsz
I,EditListzI,SecurityCookiezI,SEHandlerTablezI,SEHandlerCountzI,GuardCFCheckFunctionPointerz I,GuardCFDispatchFunctionPointerzI,GuardCFFunctionTablezI,GuardCFFunctionCountI,GuardFlagsH,CodeIntegrityFlagsH,CodeIntegrityCatalogI,CodeIntegrityCatalogOffsetI,CodeIntegrityReservedz I,GuardAddressTakenIatEntryTablez I,GuardAddressTakenIatEntryCountzI,GuardLongJumpTargetTablezI,GuardLongJumpTargetCountzI,DynamicValueRelocTablezI,CHPEMetadataPointerzI,GuardRFFailureRoutinez&I,GuardRFFailureRoutineFunctionPointerI,DynamicValueRelocTableOffsetH,DynamicValueRelocTableSectionH,Reserved2z*I,GuardRFVerifyStackPointerFunctionPointerI,HotPatchTableOffsetI,Reserved3zI,EnclaveConfigurationPointerzI,VolatileMetadataPointerzI,GuardEHContinuationTablezI,GuardEHContinuationCountzI,GuardXFGCheckFunctionPointerz!I,GuardXFGDispatchFunctionPointerz&I,GuardXFGTableDispatchFunctionPointerz"I,CastGuardOsDeterminedFailureModezI,GuardMemcpyFunctionPointer)r  )4r  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,GuardCFDispatchFunctionPointerzQ,GuardCFFunctionTablezQ,GuardCFFunctionCountr&  r'  r(  r)  r*  z Q,GuardAddressTakenIatEntryTablez Q,GuardAddressTakenIatEntryCountzQ,GuardLongJumpTargetTablezQ,GuardLongJumpTargetCountzQ,DynamicValueRelocTablezQ,CHPEMetadataPointerzQ,GuardRFFailureRoutinez&Q,GuardRFFailureRoutineFunctionPointerr+  r,  r-  z*Q,GuardRFVerifyStackPointerFunctionPointerr.  r/  zQ,EnclaveConfigurationPointerzQ,VolatileMetadataPointerzQ,GuardEHContinuationTablezQ,GuardEHContinuationCountzQ,GuardXFGCheckFunctionPointerz!Q,GuardXFGDispatchFunctionPointerz&Q,GuardXFGTableDispatchFunctionPointerz"Q,CastGuardOsDeterminedFailureModezQ,GuardMemcpyFunctionPointer)IMAGE_DYNAMIC_RELOCATION_TABLE)r  r  )IMAGE_DYNAMIC_RELOCATION)I,Symbolr  )IMAGE_DYNAMIC_RELOCATION64)Q,Symbolr  )IMAGE_DYNAMIC_RELOCATION_V2)I,HeaderSizeI,FixupInfoSizer2  I,SymbolGroupI,Flags)IMAGE_DYNAMIC_RELOCATION64_V2)r6  r7  r4  r8  r9  )IMAGE_BOUND_IMPORT_DESCRIPTOR)r  H,OffsetModuleNamezH,NumberOfModuleForwarderRefs)IMAGE_BOUND_FORWARDER_REF)r  r<  z
H,Reserved)RUNTIME_FUNCTION)zI,BeginAddresszI,EndAddresszI,UnwindDataNx   c                    || _         || _        d | _        g | _        g | _        d | _        ||t        d      g | _        d | _        d| _	        d| _
        d| _        d| _        d| _        t        j                  t        j                   t        j"                  d| _        ||nt'               d   }	 | j)                  |||       y #  | j+                           xY w)NzMust supply either name or dataFr   )r>   r@   rB   	fast_load)max_symbol_exportsmax_repeated_symbol_get_section_by_rva_last_usedsections_PE__warningsr  r  __structures___PE__from_fileFileAlignment_WarningSectionAlignment_Warning!_PE__total_resource_entries_count_PE__total_resource_bytes_PE__total_import_symbolsr  ;__IMAGE_IMPORT_CONTROL_TRANSFER_DYNAMIC_RELOCATION_format__:__IMAGE_INDIR_CONTROL_TRANSFER_DYNAMIC_RELOCATION_format__6__IMAGE_SWITCHTABLE_BRANCH_DYNAMIC_RELOCATION_format__#dynamic_relocation_format_by_symbolglobals	__parse__close)rn  r  r*   rA  rB  rC  s         r   rk  zPE.__init__M  s     #5#6 -1*<DL>??
 ! &+"(-% /0+ '(#&'# MMLLHH4
0 "+!6IGIk<R		NN4y1	JJLs   2C Cc                     | S r   re  rs  s    r   	__enter__zPE.__enter__  s    r   c                 $    | j                          y r   )rT  )rn  r  rb  	tracebacks       r   __exit__zPE.__exit__  s    

r   c                 \   | j                   du rt        | d      r~t        t        j                  t              r$t        | j
                  t        j                        s dt        t	        | j
                              v r| j
                  j                          | `t        j                          y )NTrC  z	mmap.mmap)
rH  rK  rU  mmapr  rC  r  rT  gccollectrs  s    r   _close_datazPE._close_data  so    $j)DIIt,DMM4991U$tDMM':";; MM!


r   c                 $    | j                          y r   )r^  rs  s    r   rT  zPE.close  s    r   c                    t        ||      }	 |j                  |       | j                  j	                  |       |S # t        $ r9}| j                  j	                  dj                  |d   ||             Y d}~yd}~ww xY w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)r  r  r  rF  r  r  rG  rn  r  r*   r  	structureerrs         r   __unpack_data__zPE.__unpack_data__  s     f+>		  & 	""9-  	OO""IPP1I{C
 	   = 	A?/A::A?c                    t        ||      }	 |j                  |       | j                  j	                  |       |S # t        $ r9}| j                  j	                  dj                  |d   ||             Y d}~yd}~ww xY wra  )r  r  r  rF  r  r  rG  rb  s         r   __unpack_data_with_bitfields__z!PE.__unpack_data_with_bitfields__  s     +6{K		  & 	""9-  	OO""IPP1I{C
 	rf  c           
      .   |t        j                  |      }|j                  dk(  rt        d      d}	 t	        |d      }|j                         | _        t        t        d      r5t        j                  | j
                  dt        j                        | _	        n5t        j                  | j
                  dt        j                        | _	        d| _        	 |!|j                          n||| _	        d| _        t!        | j                        | _        d| _        |st'        t)        | j                              j+                         D ]  \  }}	|dk(  r|	t!        | j                        z  dkD  s"|dk7  s,|	t!        | j                        z  dkD  sH| j,                  j/                  dj                  |d|	z  t!        | j                        z                | j                  dd }
t!        |
      dk7  rt        d      | j1                  | j2                  |
d      | _        | j4                  j6                  t8        k(  rt        d      | j4                  r| j4                  j6                  t:        k7  rt        d      | j4                  j<                  t!        | j                        kD  rt        d      | j4                  j<                  }| j1                  | j>                  | j                  ||dz    |      | _         | j@                  r| j@                  jB                  st        d      d| j@                  jB                  z  tD        k(  rt        d      d| j@                  jB                  z  tF        k(  rt        d      d| j@                  jB                  z  tH        k(  rt        d      d| j@                  jB                  z  tJ        k(  rt        d      | j@                  jB                  tL        k7  rt        d      | j1                  | jN                  | j                  |dz   |dz   dz    |dz         | _(        tS        tT        d       }| jP                  st        d!      tW        | jP                  | jP                  jX                  |       |dz   | jP                  j[                         z   }|| jP                  j\                  z   }| j1                  | j^                  | j                  ||d"z    |      | _0        d#}| j`                  [t!        | j                  ||d$z          |k\  r=d%}| j                  ||d$z    d&|z  z   }| j1                  | j^                  ||      | _0        | j`                  | j`                  jb                  td        k(  rtd        | _3        n| j`                  jb                  th        k(  rth        | _3        | j1                  | jj                  | j                  ||d$z    |      | _0        d'}| j`                  [t!        | j                  ||d$z          |k\  r=d%}| j                  ||d$z    d&|z  z   }| j1                  | jj                  ||      | _0        | jP                  st        d!      | j`                  t        d(      | jf                  >| j,                  j/                  d)j                  | j`                  jb                               tS        tl        d*      }tW        | j`                  | j`                  jn                  |       g | j`                  _8        || j`                  j[                         z   }| jP                  | j@                  _(        | j`                  | j@                  _0        | j`                  jr                  | j`                  jt                  k  r| j,                  j/                  d+       | j`                  jv                  d,kD  r2| j,                  j/                  d-| j`                  jv                  z         d"}ty        t{        d.| j`                  jv                  z              D ]  }t!        | j                        |z
  dk(  r nt!        | j                        |z
  dk  r| j                  |d d/z   }n| j                  |||z    }| j1                  | j|                  ||      }| nn	 t~        |   |_@        ||j[                         z  }| j`                  jp                  j/                  |       ||| j`                  j[                         z   d%z   k\  s n | j                  |      }| j                  D cg c],  }|j                  dkD  r| j                  |j                        . }}t!        |      dkD  rt        |      }nd}|r||k  r| j                  d| | _H        n| j                  d| | _H        | j                  | j`                  jr                        	 p| j                  | j`                  jr                        }|t!        | j                        kD  re| j,                  j/                  d0| j`                  jr                  z         n2| j,                  j/                  d1| j`                  jr                  z         |s| j                          yy# t        $ r:}dj                  |      }|xr d	|z  }t        d
j                  ||            d}~ww xY w# ||j                          w w xY w# t        t        f$ r Y  w xY wc c}w )2zParse 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      ?g333333?zeByte 0x{0:02x} makes up {1:.4f}% of the file's contents. This may indicate truncation / malformation.g      Y@r^   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 filerH   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@   r0   IMAGE_FILE_zFile Header missingrb   rF  r.   r`   r  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.r1   zsSuspicious NumberOfRvaAndSizes in the Optional Header. Normal values are never larger than 0x10, the value is: 0x%xs           z[Possibly corrupt file. AddressOfEntryPoint lies outside the file. AddressOfEntryPoint: 0x%xzTAddressOfEntryPoint lies outside the sections' boundaries. AddressOfEntryPoint: 0x%x)Losstatst_sizer  openfilenorK  r[  rk  rC  ACCESS_READrH  IOErrorr  	ExceptionrT  rG  $_PE__resource_size_limit_upperbounds _PE__resource_size_limit_reachedr   r   r  rF  r  re  __IMAGE_DOS_HEADER_format__
DOS_HEADERe_magicIMAGE_DOSZM_SIGNATUREIMAGE_DOS_SIGNATUREe_lfanew__IMAGE_NT_HEADERS_format__
NT_HEADERSr  IMAGE_NE_SIGNATUREIMAGE_LE_SIGNATUREIMAGE_LX_SIGNATUREIMAGE_TE_SIGNATUREIMAGE_NT_SIGNATURE__IMAGE_FILE_HEADER_format__FILE_HEADERr\  IMAGE_CHARACTERISTICSrc  rH  r   SizeOfOptionalHeader __IMAGE_OPTIONAL_HEADER_format__r:  Magicr  r  r  "__IMAGE_OPTIONAL_HEADER64_format__DLL_CHARACTERISTICSDllCharacteristicsDATA_DIRECTORYAddressOfEntryPointSizeOfHeadersNumberOfRvaAndSizesr  r"   __IMAGE_DATA_DIRECTORY_format__DIRECTORY_ENTRYr  KeyErrorAttributeErrorparse_sectionsrE  r0  r9  rl  headerget_section_by_rvarR  	full_load)rn  fnamer*   rA  rq  fdexcpexception_msgbyte
byte_countdos_header_datant_headers_offsetimage_flagsoptional_header_offsetsections_offset&MINIMUM_VALID_OPTIONAL_HEADER_RAW_SIZEpadding_lengthpadded_datadll_characteristics_flagsr  )MAX_ASSUMED_VALID_NUMBER_OF_RVA_AND_SIZESrN  	dir_entryr  rawDataPointerslowest_section_offset	ep_offsets                              r   rS  zPE.__parse__  s
    775>D||q #$788B%& iik4/$(IIdkk1d>N>N$ODM %)IIdkk1TEUEU$VDM#'  >HHJ DM$D 25T]]1C.-2*$+Idmm,D$E$K$K$M  j
 AI*s4==/A"AC"GAI*s4==/A"AD"HOO**L &uz'9C<N'NO	 --,2%K  ..,,o1 / 
 ??""&;; KLL$//"9"9=P"P =>> ??##c$--&88 PQQ OO44..,,MM+.?!.CD) / 
 doo&?&? 788T__...3EE RSST__...3EE RSST__...3EE RSST__...3EE RSS??$$(:: ?@@//--MM+a/2Ca2G"2LM)A- 0 

 %%:MJ 566 	$""D$4$4$D$DkR!2Q!69I9I9P9P9R!R 143C3C3X3XX#3311MM03IC3OP.	  4  
  24.   (47MPU7UV 66 !N --&)?%)G')K $(#7#7552 $8 $D  +##))-EE7%%++/LL<'+';';;;MM.1G%1O !7 (< ($ :@6 ((025Ke5S
 >> &)N"&--.1G%1O#/#1K ,0+?+???#$: ,@ ,D(  566 ' WXX<<OO""<CC((.. %3!<%
!
 	    33%	
 /1+'$*>*>*E*E*GG&*&6&6#*.*>*>'
   44""001 OO"". 33d:OO""O&&::; 5:1s:(<(<(P(PPQR )	A4==!F*a/4==!F*Q.}}VW-	9}}V&OO ,,44d - I  !0!3	 i&&((F  //66yA &)=)=)D)D)FFO S)	V $$_5 ]]
!!A% ((););<
 
 !#$'$8!$(!$(=(F--0DK--(>)>?DK
 ##D$8$8$L$LM
 00$$88I 3t}}--&&7**>>? OO"",.2.B.B.V.VW
 NN M
   %T 2 - J6M3I4;;E=Q  >HHJ "t n- :
s<   B"j 6k;1l	k &5kk  k# #k8;llc                 b   d}d}| j                   j                  dd| j                  j                               }|dk(  ry	 | j                   d|dz    }|ddt	        |      dz  z   }t        t        j                  d	j                  t	        |      dz        |            }||vry	 t        j                  d
||j                  |      dz            }d|i}|d|j                  d       }||d<   d }	t               }
t        |      D ]4  \  }}|
j                   |	|       |	||t	        |      z           z         6 t        |
      |d<   t         j#                  |d      }|d   |z  |k7  s|d   |k7  s|d   |k7  s|d   |k7  r| j$                  j                  d       ||d<   g }||d<   |dd }t'        t	        |      dz        D ]U  }|d|z     |k(  r,|d|z  dz      |k7  r| j$                  j                  d        |S ||d|z     |z  |d|z  dz      |z  gz  }W |S # t        $ r Y yw xY w)a#  Parses the rich header
        see https://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`   NrH   r@   z<{0}I<Lr4   r  raw_datac                 <    t        | t              st        |       S | S r   )rU  r"   r  )r  s    r   <lambda>z&PE.parse_rich_header.<locals>.<lambda>P  s    z!S'9Q q r   
clear_datalittler   r<   r>   z:Rich Header is not in Microsoft format, possibly malformedchecksumri  zRich Header is malformed)rC  findr:  r  rG  listrH  rI  r  r  r	  indexr   r  r  rW  r"   
from_bytesrF  r  )rn  DANSRICH
rich_index	rich_datar*   r  resultr  ord_r  r  r#   r  headervaluesrN  s                   r   parse_rich_headerzPE.parse_rich_header$  s    ]]''T4//??A

 	 dZ!^<I ""=AY1)<$=>IgnnS^q5H&I9UVD4   kk$TZZ%5%9 :;6y~~g67%z@[
!(+ 	EHCd3i$s3S>/B*CCD	E$Z0| >>#x0 Gh$&Aw("Aw("Aw("OO""L &z'xABxs4yA~& 	QA AE{d" A	?h.OO**+EF
  T!a%[83T!a%!)_x5OPPL	Q c  		s   A$H" "	H.-H.c                     | 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.
        )rF  rs  s    r   r  zPE.get_warningsy  s     r   c                 >    | j                   D ]  }t        d|        y)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)rF  print)rn  warnings     r   show_warningszPE.show_warnings  s!      	 G#w	 r   c                    | j                           G d d      }| j                         }|r |       | _        |j                  dd      | j                  _        |j                  dd      | j                  _        |j                  dd      | j                  _        |j                  dd      | j                  _        |j                  dd      | j                  _        yd| _        y)	zProcess the data directories.

        This method will load the data directories which might not have
        been loaded if the "fast_load" option was used.

        It also parses the rich header, which may or may not present.
        c                       e Zd Zy) PE.full_load.<locals>.RichHeaderN)ru  rv  rw  re  r   r   
RichHeaderr    s    r   r  r  Nri  r  r  r  )	parse_data_directoriesr  RICH_HEADERr>  r  ri  r  r  r  )rn  r  rich_headers      r   r  zPE.full_load  s     	##%	 	 ,,.)|D(3
D(ID%&1ooh&ED##.??5$#?D (3
D(ID%*5//,*MD'#Dr   c                 r   t        | j                        }| j                  D ]<  }t        |j                               }|j	                         }||||t        |      z    > t        | d      rt        | d      r| j                  D ]  }|D ]  }t        |d      s|j                  D ]  }t        |j                  j                               D ]  \  }	}|j                  |	   }
|j                  |	   }t        |      |d   kD  r>|j                  d      j                  d      }|d|d   dz   ||
d   |
d   |d   dz  z    s|j                  d      j                  d      }|||
d   |
d   t        |      z        |}|s|S t!        |d	      5 }|j#                  |       ddd       y# 1 sw Y   yxY w)
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  r4   r  rF  Nr<   zwb+)r   rC  rG  r  r  rG  rK  r  r  r  entriesr  entries_offsetsentries_lengthsrJ  encoders  write)rn  filename	file_datarc  struct_datar  finforD  st_entryr  offsetslengthsrM  encoded_datanew_file_datar   s                   r   r  zPE.write  s    dmm,	,, 	HI#I$6$6$89K..0F<GIfvK(889	H
 4)*tZ(!]] 9E!& 9"5-8,1,=,= 926x7G7G7M7M7O2P !9JC
 /7.F.Fs.KG.6.F.Fs.KG'*5zGAJ'>,1LL,A,H,H,T -..>
Q,? )2,3AJgajSTn9T)* 8=||G7L7S7S,68*
 -9 )2,3AJc,FW9W)*#!99992 "  (E" 	#aGGM"	# 	# 	#s   F--F6c                 	   g | _         d}t        | j                  j                        D ]  }|t        k\  rF| j
                  j                  dj                  | j                  j                  t                      nd}t        | j                  |       }|s nl||j                         |z  z   }|j                  |       | j                  |||j                         z    }t        |      |j                         k(  r"| j
                  j                  d| d        n|s"| j
                  j                  d| d        n|j                  |       | j                  j                  |       |j                   |j"                  z   t%        | j                        kD  r$|dz  }| j
                  j                  d	| d
       | j'                  |j"                        t%        | j                        kD  r$|dz  }| j
                  j                  d	| d       |j(                  dkD  r$|dz  }| j
                  j                  d| d       | j+                  |j,                  | j.                  j0                  | j.                  j2                        dkD  r$|dz  }| j
                  j                  d| d       | j.                  j2                  dk7  rJ|j"                  | j.                  j2                  z  dk7  r$|dz  }| j
                  j                  d	| d       ||k\  r| j
                  j                  d        nt5        t6        d      }t9        ||j:                  |       |j<                  j?                  dd      rj|j<                  j?                  dd      rN|j@                  jC                  d      dk(  r| jE                         rn| j
                  j                  d| d       | j                   j                  |        | j                   jG                  d        tI        | j                         D ]I  \  }	}|	t%        | j                         dz
  k(  rd|_%        )| j                   |	dz      j,                  |_%        K | j                  j                  dkD  rC| j                   r7|| j                   d   j                         | j                  j                  z  z   S |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?).r4   zError parsing section z$. SizeOfRawData is larger than file.z5. PointerToRawData points beyond the end of the file.r   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.rI  r   Fr   r  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                     | j                   S r   )r1  )as    r   r  z#PE.parse_sections.<locals>.<lambda>X  s    )9)9 r   )r  N)&rE  r  r  NumberOfSectionsMAX_SECTIONSrF  r  r  r.  __IMAGE_SECTION_HEADER_format__r   r  rC  r+   r  rG  r2  r0  rG  r9  r3  r?  r1  r:  r;  r@  r\  rJ  rc  rH  r^  r>  Namer!  	is_driversortr  rV  )
rn  r  MAX_SIMULTANEOUS_ERRORSrN  simultaneous_errorssectionsection_offsetsection_datarL  r  s
             r   r  zPE.parse_sections  s    "#t''889 g	*AL &&3::((99<
 "#&t'K'KPTUG#gnn&6&::N##N3=='..2B!BL L)W^^-==&&)9!<V'WX&&&qc *& & |,&&w/$$w'?'??#dmmBTT#q(#&&,QC/ST ++G,D,DEI  $q(#&&,QC 0+ +
 ''*4#q(#&&=aS A0 0 ,,**((99((66
  $q(#&&=aS A) ) $$22a7--0D0D0R0RRWXX#q(#&&0 4S S #&==&&'UV*+BLQM gw66F##%u""&&'>F<<&&w/7:t~~?O OO**;A3 ?C C MM  )Og	*X 	9:%dmm4 	!LCc$--(1,,7;47;}}!G8 . 4		! ,,q0T]]q)002T5E5E5V5VVV Mr   c                    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
}|t        |t        t        f      s|g}|D ]  }	 t        |d      }| j                  j                  |   }|||v rd}|j"                  r|r+|d   dk(  r# |d   |j"                  |j$                  d      }nN|r+|d   dk(  r# |d   |j"                  |j$                  d      }n!	  |d   |j"                  |j$                        }|rt-        | |d   dd |       |t        |t              s|d   |v s|j/                  |        y# t         $ r Y  yw xY w# t&        $ r.}	| j(                  j+                  d|d    d|	        Y d}	~	d}	~	ww xY w)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.
        r:   r9   r;   rC   rA   rI   rK   rQ   rM   r=   Nr   r4   T)forwarded_only)dllnames_onlyzFailed to process directory "z": rD   )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_directoryrU  r  r  r  r:  r  
IndexErrorr1  r  r  rF  r  r  remove)
rn  directoriesforwarded_exports_onlyimport_dllnames_onlydirectory_parsingrD  directory_indexr  rb  r  s
             r   r  zPE.parse_data_directoriesh  s+   @ ,T-H-HI+T-H-HI-t/M/MN*D,F,FG.0P0PQ($*B*BC0$2R2RS143T3TU143U3UV.0O0OP
 "kE4=9*m& -	4E"1%("; 00??P	 "o&D++.!!H(FF (a%44%NN+/! -!!H(FF (a%44innTX!
$,E!HY-E-Ey~~$VE
 eAhqrlE: ({D11X,""?3[-	4  :  -  OO22"?azTF S s*   &%F: F/	F,+F,/	G&8$G!!G&c                 ^   | j                   j                  t        d   k7  r!| j                   j                  t        d   k7  ryt        | j                        }|j                         }i }g }i }t        ||z        D ]  }| j                  | j                  | j                  ||      | j                  |            }| nmd}	|j                  dz  dk(  r#|j                  |v r||j                     }	n4t        | j                  |j                              }	|	||j                  <   |	j                  | j                  |j                  |	j                                     }
|
dk7  r| j                  j                  |
        n|	j                  | j                  |j                  |	j                                     }
|
dk7  r| j                  j                  |
        nP| j                  j                  |	       t!        ||	      }|j                  |       |||j"                  <   ||z  } |D ]  }|j$                  t'        |j$                  d      s'|j$                  j(                  |vr9| j                  j                  d	|j*                  j-                         d
d       x	 |j$                  j/                  ||j$                  j(                             |S # t0        $ rD}| j                  j                  d|j*                  j-                         d
d|        Y d}~d}~ww xY w)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
        r   r   Nr  r4   r   )rH  
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_TYPEr  __RUNTIME_FUNCTION_format__r   r  re  r  rR  
UnwindDatar  r2  rF  r  rG  r  BeginAddressr  rK  r   rH  r  rE  r  )rn  r  rW  rfrf_sizerva2rtrt_funcs	rva2infosr  uiwsrD  r  s                r   r  zPE.parse_exceptions_directory  s    $$5O(PP  ((L9R,SSt778))+	tw' (	A%%00c7+ 44S9 & B zB#) MMY."2==1B#0H0H0WXB/1Ibmm,((r}}biik)RS:OO**2.((r}}biik)RS:OO**2.##**2.*"DEOOE"&+F2??#7NCQ(	V  	B}}$
 2==/:}}**&8&&6ryy7P7P7RST6U=> 	882==667!	4  ! &&Eyy00215Rv? s   )2K	L,(:L''L,c           	         t        | j                        }|j                         }|}g }	 | j                  | j                  | j                  |||z    |      }|| j
                  j                  d       y|j                         r	 |S ||j                         z  }| j                  |      }| j                  |      }|zt        | j                        |z
  }	| j                  D 
cg c]  }
|
j                  |kD  r|
j                   }}
|rWt        |      }| j                  |      }|9|j                  |z
  }	n)|j                  t        |j                               z   |z
  }	|s+| j
                  j                  dj                  |             yg }t!        t        |j"                  t%        |	dz                    D ]  }| j                  | j&                  | j                  |||z    |      }|st)        d      ||j                         z  }||j*                  z   }| j-                  d| j                  ||t.        z          }|rFt1        |      D cg c]   }t3        |      t4        j6                  vs|" }}t        |      dkD  s|r n|j                  t9        ||	              ||j*                  z   }| j-                  d| j                  ||t.        z          }|rGt1        |      D cg c]   }t3        |      t4        j6                  vs|" }}t        |      dkD  s|r	 |S |s	 |S |j                  t;        |||
             !c c}
w c c}w c c}w )r  r  Nz7The Bound Imports directory exists but can't be parsed.zHRVA of IMAGE_BOUND_IMPORT_DESCRIPTOR points to an invalid address: {0:x}rH   z(IMAGE_BOUND_FORWARDER_REF cannot be readr   rb   )rH  r  )rH  r  r  )r  (__IMAGE_BOUND_IMPORT_DESCRIPTOR_format__r   re  rC  rF  r  r  get_section_by_offsetrR  rG  rE  r0  rl  r  r  r  NumberOfModuleForwarderRefsr"   $__IMAGE_BOUND_FORWARDER_REF_format__r  OffsetModuleNameget_string_from_dataMAX_STRING_LENGTHr   r"  r  r  r  r  )rn  r  rW  	bnd_descrbnd_descr_sizerD  bound_importsr  r  safety_boundaryr  sections_after_offsetfirst_section_after_offsetforwarder_refsr  bnd_frwd_refr  name_strr  invalid_charss                       r   r  z PE.parse_directory_bound_imports!  s    dKKL	"))+,,==cC.$89 - I
  
 &&M ##%v s 9##%%C005G2237K"%dmm"4{"B "]])))K7 &&)% )
 ) 255J1K."889STG**1*B*B[*P ,,s73C3C3E/FFT   &&7fSk	 NI993QR?R;ST "
  $33==MM#n(<= #  4   $'(RSS|**,,!>!>>44t}}Vf7H.HI #,X#6%#a&HXHX:X%M % 8}s*m%%&lJA"H Y777F004==&3D*DEH (2!c!fFDTDT6TA! ! x=3&-      #$8^O 4)j% !s   '"M M4M. MMc           	      j   | j                   }| j                  t        k(  r| j                  }	 | j	                  || j                  |t        |      j                               | j                  |            }|syt        |      S # t        $ r# | j                  j                  d|z         d}Y :w xY w)r  r  z5Invalid TLS information. Can't read data at RVA: 0x%xNrH  )__IMAGE_TLS_DIRECTORY_format__r  r   __IMAGE_TLS_DIRECTORY64_format__re  r  r  r   rR  r  rF  r  r  )rn  r  rW  r  
tls_structs        r   r  zPE.parse_directory_tls  s     44<<88::F
	--c9V#4#;#;#=> 44S9 . J j))  	OO""JSP J		s   A
B )B21B2c           	         | j                   t        k(  r| j                  |      }| j                  }nM| j                   t        k(  r| j                  |      }| j
                  }n| j                  j                  d       yd}d}|d   D ],  }|dz  }|t        |j                  d      d      z  }||k(  s, n |d   |d   d| f}d}	 | j                  || j                  |t        |      j                               | j                  |            }|syd}	|dkD  r&| j!                  |j"                  |j$                        }	t'        ||		      S # t        $ r! | j                  j                  d|z         Y fw xY w)
r  zGDon't know how to parse LOAD_CONFIG information for non-PE32/PE32+ fileNr   r4   r  r  z=Invalid LOAD_CONFIG information. Can't read data at RVA: 0x%xrD  )rH  dynamic_relocations)r  r  get_dword_at_rva&__IMAGE_LOAD_CONFIG_DIRECTORY_format__r  (__IMAGE_LOAD_CONFIG_DIRECTORY64_format__rF  r  r  r  re  r  r  r   rR  r  parse_dynamic_relocationsDynamicValueRelocTableOffsetDynamicValueRelocTableSectionr	  )
rn  r  rW  load_config_dir_szr  fields_countercumulative_szfieldload_config_structr*  s
             r   r  zPE.parse_directory_load_config  s    <<33!%!6!6s!;@@F\\::!%!6!6s!;BBFOO""  AY 	EaN0S1A!1DEEM 22		
 )VAY78!		!%!5!5c9V#4#;#;#=> 44S9 "6 " ""B"&"@"@"??"@@#
 %;N
 	
  	OO""RUXX	s   A
E 'E87E8c                 j   |sy |sy |t        | j                        kD  ry | j                  |dz
     }|j                  |z   }d }t        | j                        j                         }	 | j                  | j                  | j                  ||      | j                  |            }|j                  dk7  r)| j                  j                  d|j                  z         y ||z  }||j                  z   }g }||k  r^| j                  }	| j                  t         k(  r| j"                  }	t        |	      j                         }
	 | j                  |	| j                  ||
      | j                  |            }|s	 |S ||
z  }|j$                  }|j&                  }d|cxk  rdk  rAn n>| j)                  ||| j*                  |         }|j                  t-        |||             nl|d	k(  r3| j/                  |      \  }}|j                  t1        ||||
             n4|dkD  r/| j)                  ||      }|j                  t-        |||             ||z  }||k  r^|S # t        $ r! | j                  j                  d|z         Y y w xY w# t        $ r$ | j                  j                  d|z         d }Y Aw xY w)Nr4   r  zPInvalid IMAGE_DYNAMIC_RELOCATION_TABLE information. Can't read data at RVA: 0x%xzDNo parsing available for IMAGE_DYNAMIC_RELOCATION_TABLE.Version = %d<Invalid relocation information. Can't read data at RVA: 0x%xr>   rB   )rH  symbolrelocationsrF   )rH  r8  
bdd_relocsfunc_relocs)rG  rE  r1  r  )__IMAGE_DYNAMIC_RELOCATION_TABLE_format__r   re  r  rR  r  rF  r  r$  r  #__IMAGE_DYNAMIC_RELOCATION_format__r  r  %__IMAGE_DYNAMIC_RELOCATION64_format__SymbolBaseRelocSize parse_image_base_relocation_listrQ  r  parse_function_override_datar  )rn   dynamic_value_reloc_table_offset!dynamic_value_reloc_table_sectionr  r   image_dynamic_reloc_table_structreloc_table_sizerF  r*  r  rlc_sizedynamic_rlcr8  rW  r9  r;  r:  s                    r   r.  zPE.parse_dynamic_relocations  s    00,s4==/AA-- AA EF$$'GG+/($::

&( 		/3/C/C>>c#34 44S9 0D 0, ,33q8OO""V2::; 4999 Ci==F||<<CC (//1H#"22MM#x0 $ 8 8 = 3  J #"G 8OC ''F,,DFa"CCtGGO $**)*6{ 1*.*K*KC*P'Z#**(*%#-$/	 !"CCCN#**)*6{ 4KCs Civ #"[  	OO""$&)* 	@ ! #&&(*-. ##s$   '=I =3J 'JJ)J21J2c           
         g }g }| j                   }| j                  || j                  |t        |      j	                               | j                  |            }|s"| j                  j                  d|z         ||fS |t        |      j	                         z  }||j                  z   }||k  rK| j                  }| j                  || j                  |t        |      j	                               | j                  |            }|s"| j                  j                  d|z         ||fS |t        |      j	                         z  }g }t        |j                  dz        D ]?  }	|j                  t        j                  d| j                  |d            d          |dz  }A | j                  ||j                        }
||j                  z  }|j                  t!        ||j"                  ||
             ||k  rK| j$                  }| j                  || j                  |t        |      j	                               | j                  |            }|s"| j                  j                  d|z         ||fS |t        |      j	                         z  }t        |j&                  dz        D ]  }	| j(                  }| j                  || j                  |t        |      j	                               | j                  |            }|s$| j                  j                  d	|z         ||fc S |t        |      j	                         z  }|j                  t+        |
              ||fS )r  z>Invalid function override header. Can't read data at RVA: 0x%xz<Invalid function override info. Can't read data at RVA: 0x%xr@   r!  r   )rH  func_rvaoverride_rvasr9  z.Invalid bdd info. Can't read data at RVA: 0x%xrH   z<Invalid bdd dynamic relocation. Can't read data at RVA: 0x%xr%  ))__IMAGE_FUNCTION_OVERRIDE_HEADER_format__re  r  r  r   rR  rF  r  FuncOverrideSize5__IMAGE_FUNCTION_OVERRIDE_DYNAMIC_RELOCATION_format__r  RvaSizerH  rI  rA  r@  r  OriginalRva__IMAGE_BDD_INFO_format__BDDSize'__IMAGE_BDD_DYNAMIC_RELOCATION_format__r  )rn  r  r;  r:  r  func_headerfunc_end	func_inforK  rN  r9  bdd_info	bdd_relocs                r   rB  zPE.parse_function_override_dataO  s^   
??**MM#y0779:$$S)

 OO""$&)* 
**y ''))555HnOOF,,c9V#4#;#;#=>((-I
 &&(*-. #J..9V$++--CM9,,12 $$V]]4sA9N%OPQ%RSq ??Y,,K 9***C5$&22"/ +	3 HnD //''MM#y0779:$$S)

 OO""CcI 
**y ''))x''1,- 	JAAAF,,c9V#4#;#;#=>((-I
 &&(*-. #J..9V$++--C6iHI	J" J&&r   c                 &    | j                  ||      S )r  )rA  )rn  r  rW  s      r   r  zPE.parse_relocations_directory  s     44S$??r   c                    t        | j                        j                         }||z   }g }||k  r~	 | j                  | j                  | j	                  ||      | j                  |            }|s	 |S |j                  | j                  j                  kD  r+| j                  j                  d|j                  z         	 |S |j                  | j                  j                  kD  r+| j                  j                  d|j                  z         	 |S |.| j                  ||z   |j                  |j                  |z
        }n.| j                  ||z   |j                  |j                  |z
  |      }|j                  t        ||             |j                  s	 |S ||j                  z  }||k  r~|S # t        $ r$ | j                  j                  d|z         d }Y nw xY w)Nr  r7  zEInvalid relocation information. VirtualAddress outside of Image: 0x%xz9Invalid relocation information. SizeOfBlock too large: %drH  r  )r   __IMAGE_BASE_RELOCATION_format__r   re  r  rR  r  rF  r  r1  r:  SizeOfImageSizeOfBlockparse_relocationsparse_relocations_with_formatr  )	rn  r  rW  r{  rG  rF  r9  rlcreloc_entriess	            r   rA  z#PE.parse_image_base_relocation_list  s   TBBCJJLDjCi**99MM#x0 $ 8 8 = +  D ? !!D$8$8$D$DD&&&(+(:(:; 4 - !5!5!A!AA&& __- "  { $ 6 6(NC$6$6(8R! !% B B(NC$6$6(8RTW! 1mTU??  3??"Ce Cih U ! &&(*-. s   =F1 1)GGc           
      j   	 | j                  ||      }| j                  |      }g }t               }t        t        |      dz        D ]  }| j                  | j                  ||dz  |dz   dz   |      }	|	s |S |	j                  }
|
dz	  }|
dz  }||f|v r$| j                  j	                  d||z   z          |S |j                  ||f       |j	                  t        |	||||z   	             ||	j                         z  } |S # t        $ r$ | j                  j	                  d|d       g cY S w xY w)
r  Bad RVA in relocation data: 0xr  r<   r4   r  rP   r   3Overlapping offsets in relocation data at RVA: 0x%x)rH  r  r  r  )r  rR  r  rF  r  r  r  rG  re  &__IMAGE_BASE_RELOCATION_ENTRY_format__r  rq  r  r   )rn  data_rvar  rW  r*   r  r  offsets_and_typer  rD  r  
reloc_typereloc_offsets                r   r_  zPE.parse_relocations  sw   	==40D228<K
 5Ta( 	*C((;;S1Wa1}-' ) E * ) ::DJ&=Lj)-==&&#&2S&8:     ,
!;<NN zC\TWEW
 5<<>)K7	*: G  	OO""%CHQ<#PQI	s   #D *D21D2c           	      N   	 | j                  ||      }| j                  |      }t        |      j                         }g }t               }	t        t        |      |z        D ]  }
| j                  |||
|z  |
dz   |z   |      }|s |S |j                  }||	v r$| j                  j	                  d||z   z          |S |	j                  |       |j	                  t        ||||z                ||z  } |S # t        $ r$ | j                  j	                  d|d       g cY S w xY w)r  rd  r  r4   r  re  )rH  r  r  )r  rR  r  rF  r  r  r   r  r  rG  rh  PageRelativeOffsetrq  r  )rn  rg  r  rW  r  r*   r  
entry_sizer  r  r  rD  rj  s                r   r`  z PE.parse_relocations_with_format  s\   	==40D228<K
 ,F3::<
%Tj01 	&C77S:%qJ(>?' 8 E    !33Lw&&&#&2S&8:   KK%NNec|c?QR :%K-	&0 ?  	OO""%CHQ<#PQI	s   #C7 7*D$#D$c                    t        | j                        j                         }g }t        t	        ||z              D ]  }	 | j                  |||z  z   |      }| j                  | j                  || j                  |||z  z               }|s yd}|j                  dk(  rn|j                  dk(  r|j                  }	|j                  }
| j                  |	|	|
z    }|dd dk(  r dg d	g}|
t        |      j                         z
  }|d
kD  r#|d   j                  dj                  |             | j                  |||	      }|Zt!        j"                  dd|j$                  z         d
   |_        t)        t+        j,                  |j.                  |j0                  |j2                  |j4                  |j6                  |j&                  f            j9                  dd       j;                         |j<                  dz   |_        n|dd dk(  rdg dg}|
t        |      j                         z
  }|d
kD  r#|d   j                  dj                  |             | j                  |||	      }n2|j                  dk(  r|j                  }	|j                  }
| j                  |	|	|
z    }dg dg}| j                  |||	      }|r|j@                  dv r|
t        |      j                         z
  }|d
kD  r#|d   j                  dj                  |             | j                  |||	      }nx|j                  dk(  ri|j                  }	|j                  }
| j                  |	|	|
z    }ddgg}| j                  |||	      }tC        tD        d      }tG        ||jH                  |       |j                  tK        ||              |S # t        $ r" | j                  j                  d|z         Y  yw xY w)r  z7Invalid debug information. Can't read data at RVA: 0x%xNr  r4   r<   r@   s   RSDSCV_INFO_PDB70)z4s,CvSignaturezI,Signature_Data1zH,Signature_Data2zH,Signature_Data3zB,Signature_Data4zB,Signature_Data5z6s,Signature_Data6I,Ager   z{0}s,PdbFileNamez>Q     )fields-Xs   NB10CV_INFO_PDB20)zI,CvHeaderSignaturezI,CvHeaderOffsetr  rp  IMAGE_DEBUG_MISC)z
I,DataTypezI,Lengthz	B,UnicodezB,Reserved1r-  r   r4   z	{0}s,Datar   !IMAGE_DEBUG_EX_DLLCHARACTERISTICSzI,ExDllCharacteristicsIMAGE_DLLCHARACTERISTICS_EX_)rH  rD  )&r   __IMAGE_DEBUG_DIRECTORY_format__r   r  r"   r  r  rF  r  re  rR  Typer0  
SizeOfDatarC  r  rH  rI  Signature_Data6Signature_Data6_valuerV  uuidUUIDSignature_Data1Signature_Data2Signature_Data3Signature_Data4Signature_Data5replaceupperAgeSignature_StringUnicoder\  EX_DLL_CHARACTERISTICSrc  ExDllCharacteristicsr  )rn  r  rW  dbg_sizedebugr  r*   dbgdbg_typedbg_type_offsetdbg_type_sizedbg_type_data__CV_INFO_PDB70_format__pdbFileName_size__CV_INFO_PDB20_format_____IMAGE_DEBUG_MISC_format__dbg_type_partial	data_size-___IMAGE_DEBUG_EX_DLLCHARACTERISTICS_format__ex_dll_characteristics_flagss                       r   r  zPE.parse_debug_directory8  sl    TBBCJJLTH_-. 	@C}}S8c>%98D &&55 44S8c>5IJ ' C 
 Hxx1}Q"%"6"6 # $#o&E! !!$/ (	0, &	2J(K(R(R(TT % (!+03::.556FG  $330- H  +9? 'H,D,D"D::6   $		(0(@(@(0(@(@(0(@(@(0(@(@(0(@(@(0(F(F,&	!" %WS"-"UW!)a 02 !1$ #2A&'1 (0, &	2J(K(R(R(TT % (!+03::.556FG  $330- H Q"%"6"6 # $#o&E! '	0, $(#7#70-$  $'//69)'(DELLNO " %q=8;BB + 2 29 = $(#7#78-$ R"%"6"6 # $#o&E! 80A=  //A!# 0>*,J0, 110 LL#X>?	@B } ! &&PSVV 	s   O'O-,O-c           
      R   ||g}||}|t         kD  r%| j                  j                  d|t         fz         y	 | j                  |t	        | j
                        j                               }| j                  | j
                  || j                  |            }|| j                  j                  d|z         yg }||j                         z  }|j                  |j                  z   }	d}
|	|
kD  r!| j                  j                  d|	|
fz         y| xj                  |	z  c_        | j                  t        kD  r/| j                  j                  d| j                  t        fz         yg }d}t        |	      D ].  }| j                  sT| j                   | j"                  kD  r;d	| _        | j                  j                  d
| j                   | j"                  fz         | j%                  |      }|#| j                  j                  d||fz          nd}d}|j&                  dz  dz	  }|s|j&                  }n||j(                  z   }	 t+        | |      }| xj                   |j-                         z  c_        |rA|d   |k  r9|d   |k\  r1|j/                          | j                  j                  d|z          n|||j-                         z   f}|j                  |       |j0                  r||j2                  z   |v r n| j5                  ||j2                  z   |||z
  z
  ||dz   |||j2                  z   gz         }|s nhd}|t6        d   k(  ri }|j8                  D ]  }t;        |d      si }|j<                  j8                  D ]  }|8t;        |d      r,|j>                  j@                  jB                  |jD                  >|j>                  j@                  jF                  }|j>                  j@                  jB                  }|jD                  }	 | j                  ||      }tI        |tK        |      dz
  dz  |       |jM                  |        ||j<                  _'         |j                  tQ        ||||             n| jS                  ||j2                  z         }|rf| xj                   |jB                  z  c_        tU        ||j&                  dz  |j&                  dz	        }|j                  tQ        ||||             n n|dk(  r|jV                  t6        d   k(  rj|r|d    }	 j<                  j8                  d   j<                  j8                  }|D ]/  } d}!	 | j>                  j@                  }!|!| jY                  |!       1 	 ||j                         z  }1 |D "cg c]  }"|"j[                          }#}"|#j]                          t_        |      D ]  \  }}"|"ja                           tc        ||!      }$|$S # t        $ r! | j                  j                  d|z         Y yw xY w# t        $ r" | j                  j                  d|z         Y w xY w# t        $ r& | j                  j                  d|dd|        Y 
w xY w#  Y xY w#  Y xY wc c}"w )"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. r2   r  r   r4   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dirsr  	directoryr*   z2Error parsing resource of type RT_STRING at RVA 0xr  z with size r1   )rH  r  idr  i  rL   )rH  langsublang)rH  r  r  r*   r%  r  r[  )2MAX_RESOURCE_DEPTHrF  r  r  r  #__IMAGE_RESOURCE_DIRECTORY_format__r   r  re  rR  NumberOfNamedEntriesNumberOfIdEntriesrK  MAX_RESOURCE_ENTRIESr  ry  rL  rx  parse_resource_entryr  
NameOffsetr{  r  r6  DataIsDirectoryOffsetToDirectoryr  RESOURCE_TYPEr  rK  r  r*   rH  r  r  OffsetToDatarR  r"   updatestringsr  parse_resource_data_entryr  Idparse_version_informationr  r  r  r  r  )%rn  r  rW  r  r  r  r*   resource_dirdir_entriesnumber_of_entriesMAX_ALLOWED_ENTRIESstrings_to_postprocesslast_name_begin_endr  res
entry_nameentry_idname_is_stringustr_offsetentry_directoryr  resource_idresource_stringsresource_langstring_entry_rvastring_entry_sizestring_entry_idstring_entry_datarH  
entry_data
last_entryversion_entriesversion_entryrt_version_structr  string_rvasresource_directory_datas%                                        r   r  zPE.parse_resources_directory  sm   , <5DH%%OO"":=BDV<WX 	 ==YtGGHOOQD ++44005 , 

  OO"".034 
 	|""$$ --0N0NN 	
 #22OO"":$&9:;
 ++/@@+..1EEOO"">668LMN
 !#
 #*+ }	 C66//$2X2XX592&&' 33>> ++C0C{&&8;>*E JH!hh3:N!88&7 !B4!UJ//:3R3R3TT/ ++A.</2kA /224..7:EG
  $#j&E&E&GG+'
 +11*= "" c333t;"&"@"@s444C(N+%!)C,A,A!A BB #A # ' }[99 G'6'>'> 'M";</1,1<1F1F1N1N !A$1$9+2=&+I'4'9'9'@'@'E'E'M'2~~'=$, %2$6$6$=$=$J$J !1 5B4F4F4M4M4R4R 12=..
!-8<(8:K9&$5 !.$5%(%9A%=$C$4!"
 !(/? @C!AF =MK119O'MR ""("'#"1	 77s444 //6;;>/!6%CHHu,<chhRTn"J  &&,#&ZH:  zcffl(CC!,RJN&0&:&:&B&B1&E&O&O&W&WO *9 
N,0)!0=0B0B0I0I- -8 ::;LM
N 3::<C{}	 ~ -CCqqyy{CC 67 	!FC 	! #2#
 '&M  	OO"".034 	n % OO**CFQSx (5 !-$(OO$:$:+11A!0D E00A/B)D%&
 %-!-~! !   DsZ   3X A/X47&X4Y"0-Z%Z,Z$'X10X14'YY"+Z	Z	ZZ!c                 "   	 | j                  |t        | j                        j                               }| j                  | j                  || j                  |            }|S # t        $ r! | j
                  j                  d|z         Y yw xY w)z0Parse a data entry from the resources directory.zGError parsing a resource directory data entry, the RVA is invalid: 0x%xNr  )	r  r  $__IMAGE_RESOURCE_DATA_ENTRY_format__r   r  rF  r  re  rR  )rn  r  r*   
data_entrys       r   r  zPE.parse_resource_data_entry2  s    	 ==YtHHIPPRD ))55005 * 

   	OO""+.13 	s   3A$ $'BBc                    	 | j                  |t        | j                        j                               }| j                  | j                  || j                  |            }|y|j                  dz  |_        |j                  dz  |_	        |j                  dz  |_
        |j                  dz  dz	  |_        |j                  dz  |_        |S # t        $ r Y yw xY w)z5Parse a directory entry from the resources directory.Nr  ro        r  r2   r  )r  r  )__IMAGE_RESOURCE_DIRECTORY_ENTRY_format__r   r  re  rR  r  r  _PE__padr  r  r  r  )rn  r  r*   resources       r   r  zPE.parse_resource_entryJ  s    	==YtMMNUUWD ''::005 ( 
  'mmj8!3mmj0$,$9$9J$F2#M %-%:%:Z%G"-  		s   3C 	CCc                    	 | j                  |j                        }| j                  |||j                  z    }| j                  | j                  ||      }|y|j                  |j                         z   }| j                  |      }d}|r-|j                  t        |j                  |j                        z   }d}	 || j!                  |d      }n| j!                  |||z
  dz	  d      }|+| j                  j	                  dj                  |             y||d	k7  rt#        |      d
kD  rQ|dd
 j%                  d      }	|	d|	j'                  d       }	dj                  |	t#        |            j)                         }| j                  j	                  dj                  |j%                  d      j+                  dd                   yt-        | d      sg | _        |}
||
_        | j.                  j	                  |
       |d}| j3                  |j                         dt#        |      dz   z  z   |j                        }| j                  | j4                  ||d ||z         }|syt-        | d      sg | _        | j6                  j	                  |       | j3                  ||j                         z   |j                        }t-        | d      sg | _        g }	 | j                  | j:                  ||d ||z         }|| j                  j	                  d       y|j                  |z   |j                         z   }	 | j!                  |      }||_        |j	                  |       |r'|j=                  d      r|j>                  dv r|j@                  dk(  r| j3                  ||j                         z   dt#        |      dz   z  z   |j                        }g |_!        	 | j                  | jD                  ||d ||z         }|sn|j                  |z   |j                         z   }	 | j!                  |      }||_#        i |_$        i |_%        i |_&        |jB                  j	                  |       | j3                  ||j                         z   dt#        |      dz   z  z   |j                        }|||jN                  z   k  r| j                  | jP                  ||d ||z         }|sn_|j                  |z   |j                         z   }	 | j!                  |      }| j                  |      }| j3                  dt#        |      dz   z  |z   |j                         z   |j                        }|j                  |z   }	 | j!                  ||j@                        }| j                  |      }|jN                  dk(  r||jN                  z   }n)| j3                  |jN                  |z   |j                        }||jH                  |<   ||f|jJ                  |<   t#        |      t#        |      f|jL                  |<   |||jN                  z   k  r| j3                  |jN                  |z   |j                        }||k(  rn|}||jN                  k\  rn	|r|j=                  d      r|}d|_)        |j>                  dv r|j@                  dk(  r| j3                  ||j                         z   dt#        |      dz   z  z   |j                        }g |_*        	 | j                  | jV                  ||d ||z         }|sn[|j                  |z   |j                         z   }	 | j!                  |      }|n%|jT                  j	                  |       | j3                  dt#        |      dz   z  |z   |j                         z   |j                        }|} || |j@                  z   k  ry| jY                  |||dz    d      }!| jY                  ||dz   |d!z    d      }"|d!z  }t[        |!t\              rt[        |"t\              r|d"|!|"fz  i|_/        || |j@                  z   k  ry| j3                  ||jN                  z   |j                        }|||jN                  z   k  rn| j3                  |jN                  |z   |j                        }|jN                  dk(  s||jN                  k\  rn| j8                  j	                  |       y# t        $ r7 | j                  j	                  dj                  |j                               Y yw xY w# t        $ r" | j                  j	                  d|z         Y |w xY w# t        $ r- | j                  j	                  dj                  |             Y w xY w# t        $ r. | j                  j	                  dj                  |             Y Fw xY w# t        $ r. | j                  j	                  dj                  |             Y w xY w# t        $ r# | j                  j	                  d|d       Y w xY w# t        $ r. | j                  j	                  d j                  |             Y w xY w)#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encodingr4   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) 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   StringFileInforw  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   VarFileInfo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)0rR  r  r  rF  r  r  rC  r  re  __VS_VERSIONINFO_format__r   r  r1  rm  r2  r3  r  rG  rJ  rfindr  r  rK  r  Keydword_align__VS_FIXEDFILEINFO_format__r
  r  __StringFileInfo_format__rX  r{  ValueLengthr  __StringTable_format__LangIDr  r  r  Length__String_format__r  r  __Var_format__get_word_from_datarU  r"   rD  )#rn  version_structstart_offsetr  versioninfo_structr  r  section_endversioninfo_stringexcerptvinfofixedfileinfo_offsetfixedfileinfo_structstringfileinfo_offsetr  stringfileinfo_structstringfileinfo_stringstringtable_offsetstringtable_structstringtable_stringentry_offsetstring_structr  
key_offsetvalue_offsetrb  new_stringtable_offsetvarfileinfo_struct
var_offset
var_struct
var_stringvarword_offsetorig_varword_offsetword1word2s#                                      r   r  zPE.parse_version_informationi  s'   4		33N4O4OPL ==~?R?R0RS "11**H, 2 
 %$114F4M4M4OO))+6!003%%w'?'?4 K "
	"%)%=%=' &> &" &*%=%=+";!AG &> &" %OO""4;;<NO  ).@DV.V%&,,Tc299'B!"8GMM%$89%O%V%VS!34&&( # OO""4;;&--g6>>ufM
 t-."$D # '	""5)%!##//%%'!s3E/F/J*KK'' 
  $33,,)*+$';;  4  
 $t/0$&D! 	$$%9:
 !% 0 0 #7#>#>#@@''!
 tZ(DM$($8$8.../0(+@@ %9 %! %,&&E  ++'($++-. 
(,(@(@(M% )>!% LL./ %)>)I)I!* *..&8-99Q>)-)9)9-/6689s#89A=>? '33	*& 9;)5 -1-A-A 77$%7%89(47I(I .B .*  2! +77010779: $

"151I1I+1V. 5G*157*2=?*:=?*:-99@@AST'+'7'7.0779:3'9#:Q#>?@ +77	( )03E3L3LLM -1,@,@ $ 6 6 ( 7,8<,G -A -M $1 % !/ ; ;".!/"/"6"6"8!9 (
&&*&>&>{&K-1-E-Ek-R
 ,0+;+; !SX\ 2".!/"/"6"6"8!9 !/ ; ;	,L +9*E*E*TK&(,(@(@$/M<U<U )A )" 04/G/G/T  -33q8$69K9R9R$R !- 04/?/?$1$8$8<$G$2$?$?0"
 ?D.66s; * ,G.>>sC
 !$C #E
G.>>sCE )03E3L3LLMN 261A1A.558JJ*772. 25GG!-C*-1F1M1MM!} B '+@+K+K, &;"*7"' '++v5*66!;!%!1!1-,3356s#89A=>? '33	"J .0&* %)%9%9 //$Z[1(4z(A &: &
  *! +77()(//12 $

")-)A)A+)NJ &-!*..55jA)-)9)9Z1!45()(//12 +77	* /=+ +1J4J4JJK %)$;$; (.1:L Mq%E %)$;$; (!);nq>P QST%E +a/N)%5*UC:P$.5%.0P4"
 0 +1J4J4JJK  &*%5%5&):)::N<W<W&
 &j6G6G)GG!} B %)$4$4%,,/DD++%! &,,1(,>,E,EEo r 	U#]  	OO""CCI6"//D 	R  	OO""58CE	D ! &&<<BF;<O
 b  - " OO22!HHN$/I" ""\ $1 & $ 6 6%LLRF(3M&!" !&&. $1 & $ 6 6%BBMa%R!"
 !&&b  - " OO22!NNTf$/O" ""s}   c; <0d> ;e, (f% "g <.h i ;=d;:d;>'e)(e),3f"!f"%3gg3hh(ii3i?>i?c                      	  j                   j                   j                  |t         j                        j	                                j                  |            }|sy fd}	  j                  |j                  t         ||j                        |j                  dz              } j                  |j                  t         ||j                        |j                  dz              } j                  |j                  t         ||j                        |j                  dz              }g }	d}
 j                  |j                        }t!         j"                        }|r3|j$                  t!        |j                               z   |j                  z
  }t'        j(                  t*              }d}t-        t        |j                  t+        |dz                    D ]  } j/                  ||      }|$|dz  t!        |      k  r j1                  ||      }n y||dk(  rF||k\  r,|||z   k  r$ j3                  |      }	  j                  |      }n|rzd}d} j1                  ||      }||
d	z  }
|
dk  rd
} n7 j3                  |t4              }t7        |d      sd
} n	  j                  |      }|||fxx   d	z  cc<   |||f   dkD  r% j                  j                  d| d|dd        nt!        |       j8                  kD  r6 j                  j                  dj;                   j8                                nq|	j                  t=         |j>                  |z    j                  |j                  d|z  z         | j                  |j                  d|z  z         ||||	              |s) j                  j                  d|j                  d       |	D ch c]  }|j@                   }}d}
 j                  |j                        }t!         j"                        }|r3|j$                  t!        |j                               z   |j                  z
  }t'        j(                  t*              }d}t-        t        |j                  t+        |dz                    D ]%  }||j>                  z   |vs	  j1                  ||      }||
d	z  }
|
dk  rd
} n|dk(  r?|||k\  r|||z   k  r j3                  |      }nd}||xx   d	z  cc<   ||    jB                  kD  r7 j                  j                  dj;                   jB                  |              nqt!        |       j8                  kD  r+ j                  j                  d j8                   d        n.|	j                  t=        |j>                  |z   |d|             ( |s* j                  j                  d|j                  d       y|	s|jE                         rytG        ||	 j3                  |jH                              S # t        $ r!  j                  j                  d|z         Y yw xY w# t        $ r!  j                  j                  d|z         Y yw xY w# t        $ r Y w xY w# t        $ rI |
d	z  }
|
dk  rd
}Y  	  j                  |      }n"# t        $ r |
d	z  }
|
dk  rd
}Y Y  0Y Y w xY wY 8w xY wc c}w # t        $ r d}Y 8w xY w)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                 R    t        j                        j                  |       z
  S r   )rG  rC  rR  )r  rn  s    r   length_until_eofz3PE.parse_export_directory.<locals>.length_until_eof3  s"    t}}%(@(@(EEEr   r@   rL   Tr   r4   F)r  z9Export directory contains more than 10 repeated entries (r5  z#02xz). Assuming corrupt.zHExport directory contains more than {} symbol entries. Assuming corrupt.r<   )	r}  r  r  r  r  r  r  r  r  zIRVA AddressOfNames in the export directory points to an invalid address: r  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  r  r  zMRVA AddressOfFunctions in the export directory points to an invalid address: )rH  symbolsr  )%re  !__IMAGE_EXPORT_DIRECTORY_format__r  r  r   rR  r  rF  r  AddressOfNamesrl  NumberOfNamesAddressOfNameOrdinalsAddressOfFunctionsNumberOfFunctionsr  rG  rC  r1  collectionsdefaultdictr"   r  r  get_dword_from_dataget_string_at_rvaMAX_SYMBOL_NAME_LENGTHr  rB  r  r  Baser  rC  r  r  r  )rn  r  rW  r  
export_dirr  address_of_namesaddress_of_name_ordinalsaddress_of_functionsexports#max_failed_entries_before_giving_upr  r  symbol_counts&export_parsing_loop_completed_normallyrN  symbol_ordinalsymbol_addressforwarder_strr  symbol_name_addresssymbol_namesymbol_name_offsetexpordinalsr  s   `                         r   r  zPE.parse_export_directory  s   	--664#I#IJQQS !44S9 . J 
	F	#}}))$Z%>%>?,,q0  (,}}00$Z%E%EF,,q0($ $(==--$Z%B%BC0014$  .0+))**C*CDdmm,&&g&&()*++,  $//415.s:33S19L5MNO ]	A!445MqQN)nq.@3$D / "&!9!9(." %1)<
 $#*)D $ 6 6~ F'+'?'?'O$ " $#' "&":":;KQ"O"*3q836!;=B:00#%;K *+PTU9>6%)%=%=>Q%R"$ ;78A=8k>:;b@&&#}B~d&;;OQ ]#d&=&==&&((.t/F/F(G NN&OOn<#'#;#;"881q5@$ +#'#;#;"55N8JJ$ % 2+%5[]	~ 6OO""&55a8:
 ,33CCKK33.0+))**G*GHdmm,&&g&&()*//0  $//415.Z993QR?R;STU 5	CZ__$H4*%)%=%=>RTW%XN ")71<7:a?AF>!Q& #.&#-&t3$($:$:>$JM$(M
 n-2- 043K3KKOO**FFLf 44nG '$*A*AAOO**>2233VX  *# 5 .!"/	]5	n 6OO""&99!<> :002''
8
 	
a  	OO""=E 		J  	OO""=E 		Z % . ! 3q836!;=B:)-)A)ABU)V&$ 71<7:a?AF> 'n 4( % *%)N*s   AX: *C	Y' Z,Z$)[9[>:'Y$#Y$''ZZ	Z! Z!$[6>[[6[/%[6*[6.[//[65[6>\\c                 $    ||z   dz   dz  |dz  z
  S )Nr>   l    re  )rn  r  bases      r   r  zPE.dword_align  s     $"j0TJ5FGGr   c                     | j                   j                  }| j                   j                  | j                   j                  z   }||cxk  r|k  r	n |S ||z  }|S r   )r:  	ImageBaser]  )rn  vabegin_of_imageend_of_images       r   normalize_import_vazPE.normalize_import_va   sZ    --77++558L8L8X8XX R.,.	 . B	r   c                 >   g }d}	 	 | j                  |t        | j                        j                               }| j                  |      }| j                  | j                  ||      }|r|j                         r	 |S d}|j                  dk(  r| j                  j                  t        d   k(  r| j                  |j                        |_        | j                  |j                         |_        | j                  |j"                        |_        | j                  |j$                        |_        | j                  |j$                        |_        | j                  |j(                        |_        d}||j                         z  }t+        | j,                        |z
  }	||j"                  kD  s||j                   kD  r&t/        ||j"                  z
  ||j                   z
        }	g }
	 | j1                  |j"                  |j                   d|	|      }
|d	kD  r-| j
                  j                  d
j3                  |             	 |S |
s|dz  }B| j6                  t8        kD  r1| j
                  j                  d| j6                  t8        fz         	 |S | j;                  |j(                  t<              }t?        |      sd}|ri|
D ]G  }|j@                  tC        jD                  |jG                         |jH                        }|sA||_         I |j                  tK        ||
|             # t        $ r" | j
                  j                  d|z         Y |S w xY w# t        $ r@}| j
                  j                  dj3                  ||j4                               Y d}~d}~ww xY w)z*Walk and parse the delay import directory.r   Tz5Error parsing the Delay import directory at RVA: 0x%xr  Fr   NzSError parsing the Delay import directory. Invalid import data at RVA: 0x{0:x} ({1})rB   zWToo many errors parsing the Delay import directory. Invalid import data at RVA: 0x{0:x}r4   z)Error, too many imported symbols %d (>%s)	   *invalid*rH  importsdll)&r  r  (__IMAGE_DELAY_IMPORT_DESCRIPTOR_format__r   r  rF  r  rR  re  r  grAttrsr  r  r  r+  	pBoundIATpIATpINT
pUnloadIATphmodszNamerG  rC  rm  parse_importsr  rb  rM  MAX_IMPORT_SYMBOLSr  MAX_DLL_LENGTHr  r  	ordlookup	ordLookuplowerr  r  )rn  r  rW  import_descsrO  r*   r  import_desccontains_addressesmax_lenimport_datar  r0  r8  funcnames                  r   r  zPE.parse_delay_import_directory+  s     }}dKKLSSU 2237K..==' / K +"8"8":^ ] "' ##q($$,,=V0WW(,(@(@AVAV(W%#'#;#;K<L<L#M #'#;#;K<L<L#M )-)A)A***& %)$<$<[=S=S$T!%)%=%=k>P>P%Q"%)";%%''C
 $--(;6G[%%%{/?/?)?cK$4$44cK<L<L6LMK"00$$$$& Q&&::@&+ 6 3 q **-??&&?224FGH    ((););^LC(-") 3F{{*#,#6#6syy{FNN#S#*2FK	3
 ##"+{PSTE  ! &&KsS t }p ! &&@@FsDJJ@W s)   3L% #)M %'MM	N5NNc                    t        | d      r| j                  y|dk(  r-t        | j                  j                        j	                         S |dk(  r-t        | j                  j                        j	                         S |dk(  r-t        | j                  j                        j	                         S |dk(  r-t        | j                  j                        j	                         S t        d      )Nr  r  r   r   r   r   z#Invalid hashing algorithm specified)	rK  r  r   r  r_  r   r   r   rw  )rn  	algorithms     r   get_rich_header_hashzPE.get_rich_header_hash  s    t]+t/?/?/Gt''223==??& ((334>>@@("$**556@@BB("$**556@@BB=>>r   c           	         g }g d}t        | d      sy| j                  D ]p  }t        |j                  t              r)|j                  j                         j                         }n|j                  j                         }|j                  dd      }t        |      dkD  r|d   |v r|d   }|j                  j                         }|j                  D ]  }d}|j                  sJt        j                  ||j                  d	      }|s2t        d
|j                   d|j                  d      |j                  }|sjt        |t              r|j                         }|j                  |j                         d|j                                 s t!        dj#                  |      j%                               j'                         S )a?  Return the imphash of the PE file.

        Creates a hash based on imported symbol names and their specific order within
        the executable:
        https://www.mandiant.com/resources/blog/tracking-malware-import-hashing

        Returns:
            the hexdigest of the MD5 hash of the exported symbols.
        )ocxsysr0  DIRECTORY_ENTRY_IMPORTr  .r4   r   NT)	make_namezUnable to look up ordinal r  04xr  )rK  rK  rU  r0  rW  rJ  r>  rsplitrG  r/  r  r<  r=  r  r  r  r   r  r  r_  )	rn  impstrsextsrD  libnamepartsentry_dll_lowerimprD  s	            r   get_imphashzPE.get_imphash  s    $t5600 	NE%))U+))**,224))//+NN3*E5zA~%(d"2(#iioo/O}} Nxx(22' H $+81S[[QTDUV   #xxHh.'0H'--/8>>;KLM%N	N> 388G$++-.88::r   c                    t        | d      syt        | j                  d      sy| j                  j                  D cg c]8  }|r4|j                  (|j                  j	                         j                         : }}t        |      dk(  ryt        dj                  |      j                               j                         S c c}w )zReturn the exphash of the PE file.

        Similar to imphash, but based on exported symbol names and their specific order.

        Returns:
            the hexdigest of the SHA256 hash of the exported symbols.
        DIRECTORY_ENTRY_EXPORTr  r  r   r  )rK  rX  r  r  rJ  r>  rG  r   r  r  r_  )rn  r7   export_lists      r   get_exphashzPE.get_exphash  s     t56t22I> 0088
QVV' FFMMO!!#
 

 {q 388K(//12<<>>
s   =Cc                 P   g }d}t        | j                        j                         }	 	 | j                  ||      }| j                  |      }| j                  | j                  ||      }	|	r|	j                         rno||	j                         z  }t        | j                        |z
  }
||	j                  kD  s||	j                  kD  r&t        ||	j                  z
  ||	j                  z
        }
g }|sb	 | j                  |	j                  |	j                  |	j                   |
      }|d
kD  r | j
                  j                  d|d       n|s|dz  },| j%                  |	j&                  t(              }t+        |      sd}|ri|D ]G  }|j,                  t/        j0                  |j3                         |j4                        }|sA||_        I |j                  t7        |	||             |sddh}d}d}|D ]  }|j8                  D ]x  }|D ]l  }|r|j,                  s|j,                  }t;        |j,                        t<        k(  r|j,                  j?                  d      }|jA                  |      sg|dz  } n |dz  }z  |t        |      k(  r |dk  r| j
                  j                  d       |S # t        $ r" | j
                  j                  d|d       Y w xY w# t        $ r8}| j
                  j                  d|dd|j"                   d       Y d	}~d	}~ww xY w)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 ()NrB   zLToo many errors parsing the import directory. Invalid import data at RVA: 0xr4   r-  r.  LoadLibraryGetProcAddressr  r   z?Imported symbols contain entries typical of packed executables.)!r  "__IMAGE_IMPORT_DESCRIPTOR_format__r   r  r  rF  r  rR  re  r  rG  rC  OriginalFirstThunk
FirstThunkrm  r9  ForwarderChainrb  r  r  r;  r  r  r<  r=  r>  r  r  r/  r  rW  rJ  rX  )rn  r  rW  r  r?  rO  image_import_descriptor_sizer*   r  r@  rB  rC  r7   r0  r8  rD  suspicious_importssuspicious_imports_counttotal_symbolsimp_dllsuspicious_symbolr  s                         r   r  zPE.parse_import_directory  sf    '033(

&( 	%  }}S*FG 2237K..77; / K
 +"8"8":;%%''C
 $--(;6G[333s[=S=S7S+888#@V@V:V K "&"4"4#66#..#22#*	 #5 #K ?OO**99<QA "1$K(()9)9>JC(-") 3F{{*#,#6#6syy{FNN#S#*2FK	3
 ##"+{PSTI P "/1A!B'($M' '%oo 
'F-? ")%V[[$%{{,5#);;#5#5g#>D??+<=494!" "Q&M
'' )C0B,CC!B&&&U w ! &&CC7K 	F % OO**99<Qr!''!M s)   J6 33K$ 6(K! K!$	L%--L  L%c                 @   g }| j                  |||      }| j                  |||      }|rt        |      dk(  r5|rt        |      dk(  r%| j                  j                  d|dd|d       g S d}	|r|}	n|r|}	nyd}
d}| j                  t
        k(  rt        }n$| j                  t        k(  rt        }d}
d	}nt        }d}t        |	      D ]  \  }}d}d}d}d}d}d
}|j                  r|j                  |z  rd}|j                  dz  }d}d}nd
}	 |j                  |z  }| j                  |d      }| j                  |d      }| j                  |j                  dz   t              }t        |      sd}| j!                  |j                  dz         }|j%                         }| j'                  |      }|| j(                  j*                  z   ||
z  z   }d}	 |r6|r4||   j                  ||   j                  k7  r||   j                  }||   }nd}||t#        d      |dk(  r|dkD  r||k(  rt#        d      |dz  }u|s|s{|j                  t/        | |||||j%                         ||||||              |S # t"        $ r Y w xY w# t,        $ r d}Y w xY w)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@   ro  rH   l    FTr  r<   r-  z"Invalid entries, aborting parsing.i  z)Too many invalid names, aborting parsing.r4   )r}  r  r  import_by_ordinalr  r  hintr  r  r  r  hint_name_table_rvathunk_offset	thunk_rva)get_import_tablerG  rF  r  r  r  r  r  r  r  r  r  r  r  MAX_IMPORT_NAME_LENGTHr  rR  r  r  rO  r:  r'  r  r  )rn  original_first_thunkfirst_thunkforwarder_chainr  rA  imported_symbolsiltiattable
imp_offsetaddress_maskr  num_invalidr  	tbl_entryimp_ordimp_hintimp_namer  rl  rj  r*   rm  rn  imp_addressr  	imp_bounds                               r   r9  zPE.parse_importsZ  sD     ## *.@
 ##K=OP s3x1}sc#h!mOO""))=a(@ A!!,Q1 IEE
!<<33-L\\::/LJ-L
 .L'. ^	NCGHHK"& %&& **\9(,%'55>G#H"&K(-%.7.E.E.T+#}}-@!D#'#:#:4#C#'#9#9%33a79O$  6h?'3H&*&>&>%33a7'  )88: 44\B	 d22<<<sZ?OO  J!33s8#9#9SX=S=S#S #C 6 6I!$SJ $I 8#3#$HII
 <'%+*<'(STTq ( ''%.#-*; ''0'@'@'B%%$/' +,?%1"+[^	@  A ) $  ! 	!s%   BJ :J 	JJJJc                 *   g }| j                   t        k(  rt        }| j                  }n8| j                   t        k(  rt
        }| j                  }nt        }| j                  }t        |      j                         }d}d}	d}
d}t               }t               }|}|rf|&|||z   k\  r| j                  j                  d       	 |S | j                  t        kD  r1| j                  j                  d| j                  t        fz         	 |S | xj                  dz  c_        ||
k\  rg S |j                         |kD  rg S |j                         |kD  rg S d}	 | j                  ||      }|st#              |k7  r| j                  j                  d
|z         y | j%                  ||| j'                  |            }|r| j)                  |j*                        |_        | j)                  |j,                        |_        | j)                  |j.                        |_        | j)                  |j0                        |_        |r?|j*                  |k\  r0|j*                  |k  r!| j                  j                  d|z         	 |S |rK|j*                  r?|j*                  }||z  r
|dz  dkD  r&g S ||	k\  r|}n|}||v r|dz  }|j3                  |       |r|j5                         r	 |S ||j                         z  }|j                  |       |rf|S # t         $ r d	}Y w xY w)Nr           rV   r   z9Error parsing the import table. Entries go beyond bounds.z$Excessive number of imports %d (>%s)r4   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%xro  r  )r  r  r  __IMAGE_THUNK_DATA_format__r  r  __IMAGE_THUNK_DATA64_format__r  r   rh  rF  r  rM  r:  rt  r  r  rG  re  rR  r+  r  r  r  r  rq  r  )rn  r  r  rA  rw  r  r  expected_sizeMAX_ADDRESS_SPREADADDR_4GBMAX_REPEATED_ADDRESSESrepeated_addressaddresses_of_data_set_64addresses_of_data_set_32	start_rvafailedr*   
thunk_dataaddr_of_datathe_sets                       r   ro  zPE.get_import_table  s]   
 <<33-L55F\\::/L77F
 .L55F!&)002(!##-< #-< 	%#Z1G*G&&O B  **-??&&:224FGH t q ''1,'  #99	
 (,,.1CC	',,.1CC	F}}S-8 Tm3&&RUXX --$*B*B3*G . J
 "+/+C+C,,,
( .2-E-E...
* '+&>&>z?R?R&S
#%)%=%=j>P>P%Q
" ,,	9,,3&&(+.0
 > ; j66)77,. $j069!	 $x/":":#w.(A-(KK-!6!6!8 	 :$$&&CLL$I L M ! s   L LLc                    || j                   }| j                  |       | j                  }| j                  D ]  }|j                  dk(  r|j
                  dk(  r#|j
                  }| j                  |j                        }| j                  |j                  | j                  j                  | j                  j                        }|t        | j                         kD  s8|t        | j                         kD  s ||z   t        | j                         kD  s||k\  r|t        |      z
  }	|	dkD  r	|d|	z  z  }n
|	dk  r|d|	 }||j                         z  } || _         |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   r  )rC  relocate_imager  rE  r3  r2  r9  r0  r?  r1  r:  r;  r@  rG  r  )
rn  max_virtual_addressr'  original_datamapped_datar  srdprdr5  r  s
             r   get_memory_mapped_imagezPE.get_memory_mapped_image{  sb   ,   !MMM	* kk}} 	.G ''1,1F1F!1K''C..w/G/GHC!%!=!=&&$$55$$22" c$--((T]]++9s4==11%)<</#k2BBN!u~55!#)/>:7++--K;	.B  )DMr   c                    g }t        | d      r| j                  j                  D ]  }t        |d      s|j                  j                  D ]  }t        |d      st        |j                  d      s'|j                  j                  s>t        |j                  j                  j                               D ]  }|j                  |          |S )a  Returns a list of all the strings found within 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  r  )rK  r  r  r  r  r  ri  r  )rn  resources_stringsres_typer  
res_strings        r   get_resources_stringszPE.get_resources_strings  s     434 99AA I8[1'/'9'9'A'A 	I";< '(=(=y I$/$9$9$A$A26$/$9$9$A$A$H$H$J3" !IJ %6$<$<Z$H!I	II ! r   c                    | j                  |      }|d}n||z   }|sY|t        | j                        k  r| j                  || S |t        | j                        k  r| j                  || S t	        d      |j                  ||      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  rG  r  rC  r  r  )rn  r  r  r  rF  s        r   r  zPE.get_data  s     ##C(>C,CS%%{{3s++ S''}}S-- OPPzz#v&&r   c           
      L   | j                  |      }|s|| j                  rnt        | j                  D cg c]G  }| j                  |j                  | j
                  j                  | j
                  j                        I c}      }||k  r|S y|S |j                  |      S c c}w )z.Get the RVA corresponding to this file offset.N)	r  rE  rl  r?  r1  r:  r;  r@  rO  )rn  r  r  
lowest_rvas       r   rO  zPE.get_rva_from_offset  s     &&v.}}  "&  44,, 00AA 00>>	
 J& "M$$V,,-s   AB!c                     | j                  |      }|s*|t        | j                        k  r|S t        d|dd      |j	                  |      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  rG  rC  r  rR  )rn  r  r  s      r   rR  zPE.get_offset_from_rva  sW     ##C(
 S''
.Q7H IJJ$$S))r   c                     |y| j                  |      }|s"| j                  d| j                  |||z          S | j                  d|j                  ||            S )z1Get an ASCII string located at the given address.Nr   )r  )r  r  rC  r  )rn  r  r  r  s       r   r  zPE.get_string_at_rva0  sb     ;##C(,,QcC*DT0UVV((AJJs:J,NOOr   c                 d    |t        |      kD  ry||d }t        |t              rt        |      S |S )rL  r   N)rG  rU  r   rW  )rn  r  r*   r  s       r   get_bytes_from_datazPE.get_bytes_from_data;  s4    CIMa#8Or   c                 `    | j                  ||      }|j                  d      }|dk\  r|d| }|S )zGet an ASCII string from data.r  r   N)r  r  )rn  r  r*   r  rF  s        r   r  zPE.get_string_from_dataD  s8    $$VT2ffUm!8$3Ar   c                 >   |dk(  ry| j                  |d      }|dz  }t        |d      }| j                  ||      }d}	 |j                  d|dz         }|dk(  rGt        |      }||k  s||k(  rt        |      dz	  }n2|| j                  ||z   ||z
        z  }|dz
  }|}n|dz  dk(  r|dz  }npt	        j
                  dj                  |      |d	|dz         }d
j                  t        t        |            }	|r|	j                  |d      S |	j                  dd      S )z3Get an Unicode string located at the given address.r   r   r<   r4   rb   r  rq  z<{:d}HNr  r	   r  )r  rl  r  rG  rH  rI  r  r  mapr"  r  )
rn  r  r  r  r*   	requested
null_indexdata_lengthuchrsr  s
             r   r  zPE.get_string_u_at_rvaL  sC    ? }}S!$ 	q

C(	}}S),
;
Q?JR!$i*kZ.G!$TaJ cK&7k9QRR&]
&	a1$q 
 $ hooj94@P*q.;QRGGCUO$88H&9::xx!455r   c                 P    | j                   D ]  }|j                  |      s|c S  y)z1Get the section containing the given file offset.N)rE  rT  )rn  r  r  s      r   r  zPE.get_section_by_offsetw  s/     }} 	G&&v.	 r   c                     | j                   '| j                   j                  |      r| j                   S | j                  D ]  }|j                  |      s|| _         |c S  y)z-Get the section containing the given address.N)rD  rX  rE  )rn  r  r  s      r   r  zPE.get_section_by_rva  sc    
 --911>>sC999}} 	G##C(5<2	
 r   c                 "    | j                         S r   )	dump_infors  s    r   r  z
PE.__str__  s    ~~r   c                     t        | d      S )z.Checks if the PE file has relocation directoryDIRECTORY_ENTRY_BASERELOC)rK  rs  s    r   
has_relocszPE.has_relocs  s    t899r   c                 J    t        | d      r| j                  j                  ryy)NDIRECTORY_ENTRY_LOAD_CONFIGTF)rK  r  r*  rs  s    r   has_dynamic_relocszPE.has_dynamic_relocs  s"    467//CCr   c                 :    t        | j                  |             y)z=Print all the PE header information in a human readable from.r  N)r  r  )rn  r  s     r   
print_infozPE.print_info  s    dnnhn/0r   c                 &/   |
t               }| j                         }|r9|j                  d       |D ]#  }|j                  |       |j	                          % |j                  d       |j                  | j                  j                                |j	                          |j                  d       |j                  | j                  j                                |j	                          |j                  d       |j                  | j                  j                                t        t        d      }|j                  d       g }t        |      D ]0  }t        | j                  |d         s|j                  |d          2 |j                  d	j!                  |             |j	                          t#        | d
      rF| j$                  :|j                  d
       |j                  | j$                  j                                t        t&        d      }|j                  d       g }t        |      D ]0  }t        | j$                  |d         s|j                  |d          2 |j                  d	j!                  |             t        t(        d      }	|	rg }t#        | d      r|| j*                  p| j*                  D ]a  }
|
j,                  j.                  t0        d   k(  s$t        |	      D ]0  }t        |
j2                  |d         s|j                  |d          2 c |r1|j                  d       |j                  d	j!                  |             |j	                          |j                  d       t        t4        d      }| j6                  D ]s  }|j                  |j                                |j                  d       g }t        |      D ]&  }t        ||d         s|j                  |d          ( |j                  d	j!                  |             |j                  dj9                  |j;                                      t<        .|j                  dj9                  |j?                                      t@        "|j                  d|jC                         z         tD        "|j                  d|jG                         z         tH        "|j                  d|jK                         z         |j	                          v t#        | d
      rtt#        | j$                  d      r^|j                  d       | j$                  jL                  D ]$  }||j                  |j                                & |j	                          t#        | d      rKtO        | jP                        D ]2  \  }}tS        | jP                        dkD  r|j                  d|dz           n|j                  d       ||j                  |j                                |j	                          t#        | d      r<|j                  | jT                  |   j                                |j	                          t#        | d      stS        | jV                        |kD  s| jV                  |   D ]9  }|j                  |j                                |j	                          t#        |d       r|jX                  D ]  }|j                         D cg c]  }|j                  d!|z          c} |j                  d"j9                  |jZ                  j]                  |d#                   |j	                          t        t_        |j`                  jc                                     D ]I  }|j                  d$j9                  |d   j]                  |d#      |d   j]                  |d#                   K  |j	                          Ut#        |d%      sc|jd                  D ]  }t#        |d&      s|j                         D cg c]  }|j                  d!|z          c} |j                  d$j9                  t_        |j2                  jg                               d   j]                  d'd#      t_        |j2                  ji                               d                 |j	                          < 5 t#        | d(      rK|j                  d)       |j                  | jj                  j,                  j                                |j	                          |j                  d*d+z         | jj                  jl                  D ]  }|jn                  d,}|jp                  r|jp                  }|j                  d-|jr                  |jn                  |j]                  |      fz         |jt                  r;|j                  d.j9                  |jt                  j]                  |d#                   |j	                           |j	                          t#        | d/      rU|j                  d0       | jv                  D ]4  }|j                  |j,                  j                                |jx                  sc|j                  d1j9                  | j{                  |j,                  j|                        j]                  |d#                   |j	                          |j	                          |jx                  D ]i  }|j~                  d2u r|jp                  _|j                  d3j9                  |j                  j]                  d'      |jp                  j]                  d'      |jr                               n|j                  d4j9                  |j                  j]                  d'      |jr                               n`|j                  d5j9                  |j                  j]                  |d#      |jp                  j]                  |d#      |j                               |j                  r,|j                  d6j9                  |j                               Z|j	                          l |j	                          7 t#        | d7      r|j                  d8       | j                  D ]  }|j                  |j,                  j                                |j                  d9j9                  |jp                  j]                  |d#                   |j	                          |j`                  D ]w  }|j                  |j,                  j                         d:       |j                  d9j9                  |jp                  j]                  |d#            d:       |j	                          y  t#        | d;      ry|j                  d<       | j                  D ]X  }|j                  |j,                  j                                |j	                          |jx                  D ]  }|j~                  d2u rF|j                  d=j9                  |j                  j]                  |d#      |jr                               n`|j                  d>j9                  |j                  j]                  |d#      |jp                  j]                  |d#      |j                               |j                  r+|j                  d6j9                  |j                               |j	                           |j	                          [ t#        | d?      rY|j                  d@       |j                  | j                  j,                  j                                | j                  j`                  D ]  }|jp                  3|jp                  j]                  |d#      }|j                  dA| dBdC       nXt        j                  |j,                  j                  dD      }|j                  dE|j,                  j                  dFdG| dHdC       |j                  |j,                  j                         dC       t#        |dI      r
|j                  |j                  j,                  j                         d:       |j                  j`                  D ]  }|jp                  3|jp                  j]                  d'd#      }|j                  dA| dBdJ       n+|j                  dE|j,                  j                  dFdBdJ       |j                  |j,                  j                         dJ       t#        |dI      s|j                  |j                  j,                  j                         dK       |j                  j`                  D ]	  }t#        |dL      s|j                  dM|j                  j                  |j                  j                  t        j                  |j                  j                  dN      t        |j                  j                  |j                  j                        fz  dK       |j                  |j,                  j                         dO       |j                  |j                  j,                  j                         dP        t#        |j                  dQ      s|j                  j                  s,|j                  dRdO       t_        t        |j                  j                  jc                                     D ]F  \  }}|j                  dSj9                  ||j                  dTdU      j]                  dV            dP       H  |j	                           |j	                          t#        | dW      rv| j                  rj| j                  j,                  rT|j                  dX       |j                  | j                  j,                  j                                |j	                          t#        | dY      rv| j                  rj| j                  j,                  rT|j                  dZ       |j                  | j                  j,                  j                                |j	                          t#        | d      r|j                  d[       | j*                  D ]  } |j                  | j,                  j                                	 |j                  d\t0        | j,                  j.                     z          |j	                          | j2                  sy|j                  | j2                  j                         d:       |j	                           | j                         r|j                  d^       | j                  D ]  }!|j                  |!j,                  j                                |!j`                  D ]8  }"	 |j                  d_|"j                  t        |"j                     d`d fz  d:       : |j	                           t#        | db      rtS        | j                        dkD  r|j                  dc       | j                  D ]o  }#|j                  |#j,                  j                                t#        |#dd      s9|#j                  F|j                  |#j                  j                         d:       q |j                         S c c}w c c}w # t        $ r8 |j                  d]j9                  | j,                  j.                               Y w xY w# t        $ r/ |j                  da|"j                  |"j                  fz  d:       Y w xY w)ez>Dump all the PE header information into human readable string.NParsing Warningsr{  r  r  rm  r4  r   r5  r:  rn  zDllCharacteristics: ry  DIRECTORY_ENTRY_DEBUGr   zExDllCharacteristics: PE SectionsrI  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  r4   zVersion Information Version Informationr
  r  r  z  z  LangID: {0}r	   z    {0}: {1}r  rD  r  rX  Exported symbolsz%-10s   %-10s  %sr  RVAr  s   Nonez%-10d 0x%08X    %sz forwarder: {0}rK  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<   rs  zId: [0xrt  z] (r\  r  rD   rH   r*   z\--- LANG [%d,%d][%s,%s]r<  rL   rP   r  z	[STRINGS]z{0:6d}: {1}unicode-escaper
   r  DIRECTORY_ENTRY_TLSTLSr  LOAD_CONFIGDebug informationzType: zType: 0x{0:x}(Unknown)Base relocationsz%08Xh %sr1   z0x%08X 0x%x(Unknown)DIRECTORY_ENTRY_EXCEPTIONz"Unwind data for exception handlingr  )\r  r  r  r  r  r  r{  r  r  r  r\  r  rq  sortedr  r  r  rK  r:  r  r  r  rH  r{  
DEBUG_TYPErD  rJ  rE  r  r]  r   rg  r   r`  r   rb  r   re  r  r  r  rG  r
  r  r  r  rJ  r  r  r  r  rT  ri  rX  r  r  r  r  r  rK  r/  r  r  rj  r0  rk  r  r  r  r  r  r>  r  r  r*   r  r  r=  rD  r  r  r  r  r  r  r  r  RELOCATION_TYPEr  r  r  r  )$rn  r  r  warningsr  r  ra  r[  r  r  debug_entryrL  r  r  r  vinfo_entryrD  r  r  	str_entry	var_entryexportr  moduler8  bound_imp_descbound_imp_refr  res_type_idr  r  r  r  
base_relocrelocr  s$                                       r   r  zPE.dump_info  sZ    <6D$$&OO./# #g&  "# 	%t++-.%t++-.&t'',,./$%:MJ;' 	&Dt''a1T!W%	& 	dii&'4*+0D0D0POO-.NN4//4467$2!<%
! 	'(45 	&Dt++T!W5T!W%	& 	dii&''5"$B(
$ (E56..:#'#=#= 6K#**//%&NOP %++G$H 6D&{'8'8$q'B %T!W 566 12dii./&&'>M}} 	GNN7<<>*HHYE}- *7DG,LLa)* MM$))E*+MM3::7;N;N;PQ 1889M9M9OPQ073H3H3JJK!073J3J3LLM!073J3J3LLM'	* 4*+  "21
 OOM*!11@@ 5	(NN9>>#345 4)*$-d.A.A$B </ [t**+a/OO&:37)$DEOO$9:*NN;#3#3#56  "4!34NN4#8#8#=#B#B#DE$$&4,T]]1Cc1I!%s!3 ./uzz|4((*"5-8,1,=,= &HP Xtd{!; X $$3$:$:(0(>(>,46I)*%&!" !% 0 0 217X=M=M=S=S=U8V1W !&I$(MM(6(=(=,5aL,?,?08:M-. -6aL,?,?08:M-.	)*	%&!&&. !,,.$UE2-2YY &	#*9g#> 5>NN4D%&,0 )-dTk(B%& %)MM(6(=(=,01E1E1G,H,K,R,R079L-. -11G1G1I,J1,M	)*%&& !,,.]./</| 412OO./NN466==BBDEMM-0JJK55== +>>-"D{{%{{HH,!>>6>>4;;x;PQR ''-44 & 0 0 7 7BU V ((*!+$ 412OO./55 *#v}}1134~~HH'.. 226==3E3EFMM (*= $$&  "$nn +F//47!;;2 HH L S S$*JJ$5$5g$>$*KK$6$6w$?$*NN!" !HH H O O$*JJ$5$5g$>!" 188 &

 1 1(<O P & 2 28=P Q & ||&9&@&@&NO((*9+:   "U*#X 478OOO,"&"C"C '~4499;<%%&++228=PQ
   "%3%;%; 'MNN=#7#7#<#<#>BMM")))..55h@ST 	 $$&''& 478OO45;; #v}}1134  "$nn +F//47FMM &

 1 1(<O P & /66 &

 1 1(<O P & 2 28=P Q & ||&9&@&@&NO((*)+*   "3#6 434OO01NN488??DDFG 99AA G#==,#==//:MNDMM!$q)
 #0"3"3HOO4F4F"LKMM!(//"4"4Q!7s;-qI
 x335q98[1NN8#5#5#<#<#A#A#CQG'/'9'9'A'A 2&&++7#.#3#3#:#:7DW#XD MM")$q 1 !
 !MMGK4F4F4I4I!3LA*NPQR{'9'9'>'>'@!D";< NN;+@+@+G+G+L+L+NPQR1<1F1F1N1N Y#*=&#A$(MM(C,9,>,>,C,C,9,>,>,F,F,0HH0=0B0B0G0G-. -F0=0B0B0G0G0=0B0B0J0J-.
+*)* )*%& %)NN=3G3G3L3L3NPR$S$(NN=3E3E3L3L3Q3Q3SUW$X%Y( !((=(=y I$/$9$9$A$A $k2 >7;$*;+@+@+H+H+N+N+P$Q8" !&OC %)MM(5(<(<,/,6,=,=0@BT-..4fWo	)* )+%&!&O2&h   "OG#R  D/0((((//OOE"NN433::??AB D78000077OOM*NN4;;BBGGIJ401OO/011 	'szz01TMM(Z

-H"HI   "99NN399>>#3Q7$$&	' ??OO./"<< #
z005578'// 	E&%))_UZZ5PQSQT5U)VV	   "# D56D223a7OO@A44 <ryy~~/02|,1JNN2==#5#5#7;<
 }}Y	 !Y6%&`   TMM":"A"A#**//"RST" $ 2eii5LLas>   A\

6A\
T/A\W>5A]\=A]]A]]4A^^A^c           	         i }| j                         }|r||d<   | j                  j                         |d<   | j                  j                         |d<   | j                  j                         |d<   t        t        d      }g |d<   |D ]3  }t        | j                  |d         s|d   j                  |d          5 t        | d      r)| j                  | j                  j                         |d<   t        t        d
      }g |d<   |D ]3  }t        | j                  |d         s|d   j                  |d          5 g |d<   t        t        d      }| j                  D ]  }|j                         }|d   j                  |       g |d<   |D ])  }t        ||d         s|d   j                  |d          + |j                         |d<   t        |j!                         |d<   t"        |j%                         |d<   t&        |j)                         |d<   t*        |j-                         |d<    t        | d      rgt        | j                  d      rQg |d<   t/        | j                  j0                        D ]*  \  }	}
|
	|d   j                  |
j                                , t        | d      r;g |d<   t/        | j2                        D ]  \  }	}g }|j                  |j                                t        | d      r,|j                  | j4                  |	   j                                t        | d      rt7        | j8                        |	kD  rg }|j                  |       | j8                  |	   D ]^  }|j                  |j                                t        |d      ri }|j:                  D ]c  }|j=                  |j                                |j>                  |d<   tA        |jB                  jE                               D ]  }|d   ||d   <    e |j                  |       t        |d      s|jF                  D ]  }i }t        |d      s|j=                  |j                                tA        |jH                  jK                               d   |tA        |jH                  jM                               d   <   |j                  |        a |d   j                  |         t        | d      rg |d<   |d   j                  | jN                  jP                  j                                | jN                  jR                  D ]r  }i }|jT                  N|jW                  |jX                  |jT                  |jZ                  d        |j\                  r|j\                  |d!<   |d   j                  |       t t        | d"      rg |d#<   | j^                  D ]  }g }|d#   j                  |       |j                  |jP                  j                                |j`                  D ]  }i }|jb                  d$u r|jd                  |d%<   |jX                  |d&<   n-|jd                  |d%<   |jZ                  |d'<   |jf                  |d(<   |jh                  r|jh                  |d)<   |j                  |         t        | d*      rg |d+<   | jj                  D ]  }i }|d+   j                  |       |jW                  |jP                  j                                |jZ                  |d%<   |jB                  D ]<  }i }|jW                  |jP                  j                                |jZ                  |d%<   >  t        | d,      rg |d-<   | jl                  D ]  }g }|d-   j                  |       |j                  |jP                  j                                |j`                  D ]  }i }|jb                  d$u r|jd                  |d%<   |jX                  |d&<   n-|jd                  |d%<   |jZ                  |d'<   |jf                  |d(<   |jh                  r|jh                  |d)<   |j                  |         t        | d.      rg |d/<   |d/   j                  | jn                  jP                  j                                | jn                  jB                  D ]  }i } |jZ                  |jZ                  | d'<   nC|jP                  jp                  tr        ju                  |jP                  jp                  d0      f| d1<   | jW                  |jP                  j                                |d/   j                  |        t        |d2      sg }!|!j                  |jv                  jP                  j                                |d/   j                  |!       |jv                  jB                  D ]  }"i }#|"jZ                  |"jZ                  |#d'<   n|"jP                  jp                  |#d1<   |#jW                  |"jP                  j                                |!j                  |#       t        |"d2      sg }$|$j                  |"jv                  jP                  j                                |!j                  |$       |"jv                  jB                  D ]  }%t        |%d3      si }&|%jx                  jz                  |&d4<   |%jx                  j|                  |&d5<   t~        ju                  |%jx                  jz                  d6      |&d7<   t        |%jx                  jz                  |%jx                  j|                        |&d8<   |&jW                  |%jP                  j                                |&jW                  |%jx                  jP                  j                                |$j                  |&        t        |"jv                  d9      s|"jv                  j                  s'tA        |"jv                  j                  jE                               D ]5  \  }	}'|$j                  |'j                  d:d;      j                  d<             7   t        | d=      rI| j                  r=| j                  jP                  r'| j                  jP                  j                         |d><   t        | d?      rI| j                  r=| j                  jP                  r'| j                  jP                  j                         |d@<   t        | dA      rg |dB<   | j                  D ]  }(i })|dB   j                  |)       |)jW                  |(jP                  j                                t        ju                  |(jP                  j                  |(jP                  j                        |)dC<    | j                         rg |dD<   | j                  D ]  }*g }+|dD   j                  |+       |+j                  |*jP                  j                                |*jB                  D ]>  },i }-|+j                  |-       |,j                  |-dE<   	 t        |,j                     dFd	 |-dC<   @  |S # t        $ r |,j                  |-dC<   Y _w xY w)Gz5Dump all the PE header information into a dictionary.r  r{  r  r  rm  r#  r   r:  Nrn  r  r  rI  EntropyMD5SHA1SHA256SHA512r  r  r  r  r
  r  r  r  r4   r  rD  rX  r  r  r  rK  r  TDLLr  r  HintBoundr  r  r  r  r  r  rs  r  r  r*   r=  r?  r<  	LANG_NAMESUBLANG_NAMEr  r  r
   r  r  r  r  r  r  r  r{  r  r  r1   )Or  r{  r+  r  r  r\  r  r  r  rK  r:  r  rJ  rE  r]  r   rg  r   r`  r   rb  r   re  r  r  r  r
  rG  r  r  r  r  r  r  r  r  rD  ri  rT  rX  rH  r  r  r  r  r  r  rK  r/  rj  r0  rk  r  r  r  r  r  r  r>  r  r*   r  r  r=  rD  r  r  rJ  r  r  r  r  r{  r  r  r  r  r  r  ).rn  r+  r  r  r[  r  rL  r  section_dictr  r  vs_vinfoversion_info_listfileinfo_listrD  stringtable_dictr  r  r  var_dictr  export_dictr  import_listr8  symbol_dictr  bound_imp_desc_dictr  bound_imp_ref_dictmodule_listr  resource_type_dictdirectory_listr  resource_id_dictresource_id_listr  resource_lang_dictr  r  dbg_dictr  base_reloc_listr  
reloc_dicts.                                                 r   r+  zPE.dump_dictN  s    	$$&,4I()"&//";";"=	,"&//";";"=	,#'#3#3#=#=#?	- $%:MJ	' 	3Dt''a1'"))$q'2	3 4*+0D0D0P+/+?+?+I+I+KI'($2!<%
! +-	&'- 	@Dt++T!W5./66tAw?	@ $&	- &'>M}} 	CG",,.Lm$++L9$&L!% :7DG, )00a9: '.&9&9&;L#&-&:&:&<U#'.'<'<'>V$!)0)@)@)BX&!)0)@)@)BX&!	C$ 4*+  "21
 (*Im$"+D,@,@,O,O"P KY(m,33I4G4G4IJK 4)*/1I+,!*4+>+>!?  KX$&!!((););)=>4!34%,,T-B-B3-G-Q-Q-ST4,T]]1Cc1I$&M%,,];!%s!3 C%,,U__->?"5-8/1,,1,=,= R - 4 4X5G5G5I J=E__ 0 :15h6F6F6L6L6N1O !RIENq\$4Yq\$B!RR
 *001AB$UE2-2YY C	+-#*9g#>$1$8$89L9L9N$OPT(1(>(>(@Q&&'Q)HT)//2F2F2H-I!-L$M %2$8$8$BCC, /0778IJA KD 412,.I()()00++22<<> 55== B >>-&&'-~~#)>>$*KK ''393C3CK0,-44[AB 412,.I()55 4 ,-44[A""6==#:#:#<=$nn 4F"$K//47-3ZZE*17I.-3ZZE*.4kkF+.4kkF+||/5||G,&&{34	4$ 478)+Io&"&"C"C 
C&(#/*112EF#**>+@+@+J+J+LM-;-@-@#E*%3%;%; CM)+&&--m.B.B.L.L.NO0=0B0B&u-C
C 47824I./;; 4 23::;G""6==#:#:#<=$nn 4F"$K//47-3ZZE*17I.-3ZZE*.4kkF+.4kkF+||/5||G,&&{344& 434.0I*+*+22--44>>@ !99AA G&%'"==,19&v. !**%))(//*<*<cB0&t,
 #))(//*C*C*EF./667IJ8[1%'N"))(*<*<*C*C*M*M*OP23::>J'/'9'9'A'A 4&+-(&++77B7G7G,V45@5G5G5J5J,T2(//0B0B0L0L0NO&--.>?";</1,,33 + 5 5 < < F F H +112BC1<1F1F1N1N P#*=&#A9;$6ANASASAXAX$6v$> )6(:(:(B(B %7(1%& GKhh(5(:(:(?(?G&$6{$C
 )B(5(:(:(?(?(5(:(:(B(B)& %7(6%& %7$=$=(5(<(<(F(F(H%& %7$=$=(5(:(:(A(A(K(K(M%& %5$;$;<N$O-P0 !((=(=y I$/$9$9$A$A7;$/$9$9$A$A$G$G$I8" !&OC %5$;$;(2(9(9,<>P)**0&/%&!&[4&'G&T D/0((((//#77>>HHJIe D78000077 0077AAC  401-/I)*11 T-.55h?

 4 4 67#->>#**//3::??#S 	T ??,.I()"<< 8
"$,-44_E&&z'8'8'B'B'DE'// 8E!#J#**:6(-		Ju%8-<UZZ-H-M
6*8	8  $ 8-2ZZ
6*8s   y//z
	z
c                 D    	 | j                  |      S # t        $ r Y yw xY w)z;Gets the physical address in the PE file from an RVA value.N)rR  rw  rQ  s     r   get_physical_by_rvazPE.get_physical_by_rvar  s*    	++C00 		s    	c                 4    t        j                  d|dz        S )zMReturn a four byte string representing the double word value (little endian).r      rH  r	  )rn  dwords     r   get_data_from_dwordzPE.get_data_from_dword}  s    {{4!344r   c                 v    |dz   dz  t        |      kD  ryt        j                  d||dz  |dz   dz         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.
        r4   r@   Nr!  r   rG  rH  rI  rn  r*   r  s      r   r  zPE.get_dword_from_data  F     QJ!c$i'}}T4
fqjA5E#FGJJr   c                 f    	 | j                  | j                  |d      d      S # t        $ r Y yw xY w)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)r  r  r  rQ  s     r   r+  zPE.get_dword_at_rva  s7    	++DMM#q,A1EE 		s   !$ 	00c                 ~    |dz   t        | j                        kD  ry| j                  | j                  ||dz    d      S )zFReturn the double word value at the given file offset. (little endian)r@   Nr   )rG  rC  r  r  s     r   get_dword_from_offsetzPE.get_dword_from_offset  >     A:DMM**''fvz(JANNr   c                 D    | j                  || j                  |            S )zLSet the double word value at the file offset corresponding to the given RVA.)set_bytes_at_rvar  )rn  r  r  s      r   set_dword_at_rvazPE.set_dword_at_rva       $$S$*B*B5*IJJr   c                 D    | j                  || j                  |            S )z3Set the double word value at the given file offset.)r  r  )rn  r  r  s      r   r  zPE.set_dword_at_offset       ''0H0H0OPPr   c                 .    t        j                  d|      S )zFReturn a two byte string representing the word value. (little endian).r  r  rn  r  s     r   get_data_from_wordzPE.get_data_from_word      {{4&&r   c                 v    |dz   dz  t        |      kD  ryt        j                  d||dz  |dz   dz         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.
        r4   r<   Nr  r   r  r  s      r   r  zPE.get_word_from_data  r  r   c                 j    	 | j                  | j                  |      dd d      S # t        $ r Y yw xY w)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   )r  r  r  rQ  s     r   get_word_at_rvazPE.get_word_at_rva  s<    	**4==+=bq+A1EE 		   #& 	22c                 ~    |dz   t        | j                        kD  ry| j                  | j                  ||dz    d      S )z?Return the word value at the given file offset. (little endian)r<   Nr   )rG  rC  r  r  s     r   get_word_from_offsetzPE.get_word_from_offset  s>     A:DMM**&&t}}Vfqj'I1MMr   c                 D    | j                  || j                  |            S )zESet the word value at the file offset corresponding to the given RVA.)r  r  )rn  r  r  s      r   set_word_at_rvazPE.set_word_at_rva  s     $$S$*A*A$*GHHr   c                 D    | j                  || j                  |            S )z,Set the word value at the given file offset.)r  r  )rn  r  r  s      r   r  zPE.set_word_at_offset  s     ''0G0G0MNNr   c                 .    t        j                  d|      S )zMReturn an eight byte string representing the quad-word value (little endian).<Qr  r  s     r   get_data_from_qwordzPE.get_data_from_qword  r  r   c                 v    |dz   dz  t        |      kD  ryt        j                  d||dz  |dz   dz         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.
        r4   rH   Nr!  r   r  r  s      r   get_qword_from_datazPE.get_qword_from_data  r  r   c                 j    	 | j                  | j                  |      dd d      S # t        $ r Y yw xY w)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.
        NrH   r   )r$  r  r  rQ  s     r   get_qword_at_rvazPE.get_qword_at_rva  s<    	++DMM#,>r,BAFF 		r  c                 ~    |dz   t        | j                        kD  ry| j                  | j                  ||dz    d      S )zDReturn the quad-word value at the given file offset. (little endian)rH   Nr   )rG  rC  r$  r  s     r   get_qword_from_offsetzPE.get_qword_from_offset  r  r   c                 D    | j                  || j                  |            S )zJSet the quad-word value at the file offset corresponding to the given RVA.)r  r"  )rn  r  qwords      r   set_qword_at_rvazPE.set_qword_at_rva  r  r   c                 D    | j                  || j                  |            S )z1Set the quad-word value at the given file offset.)r  r"  )rn  r  r*  s      r   set_qword_at_offsetzPE.set_qword_at_offset  r  r   c                     t        |t              st        d      | j                  |      }|sy| j	                  ||      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)rU  rW  	TypeErrorr  r  )rn  r  r*   r  s       r   r  zPE.set_bytes_at_rva  sC     $&;<<))#.''55r   c                     t        |t              st        d      d|cxk  rt        | j                        k  rn y| j                  ||       yy)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   FT)rU  rW  r0  rG  rC  set_data_bytes)rn  r  r*   s      r   r  zPE.set_bytes_at_offset$  sQ     $&;<<+T]]++  -  r   r  r*   c                     t        | j                  t              s,t        | j                        }| j                          || _        || j                  ||t	        |      z    y r   )rU  rC  r   r^  rG  )rn  r  r*   new_datas       r   r2  zPE.set_data_bytes5  sJ    $--3 /H$DM59fvD	12r   c                    | j                   D ]~  }| j                  |j                        }||j                  z   }|t	        | j
                        k  sF|t	        | j
                        k  s_| j                  ||j                                 y)zeUpdate the PE image content with any individual section data that has been
        modified.
        N)rE  r9  r0  r2  rG  rC  r2  r  )rn  r  section_data_startsection_data_ends       r   merge_modified_section_datazPE.merge_modified_section_data=  s    
 }} 	LG!%!=!=g>V>V!W1G4I4II!C$66;KcO < ##$68H8H8JK	Lr   c                 6   || j                   j                  z
  }t        | j                   j                        dk\  r| j                   j                  d   j                  rt        | d      s| j                  t        d   g       t        | d      s| j                  j                  d       nM| j                  D ]=  }d}|t        |j                        k  s|j                  |   }|dz  }|j                  t        d	   k(  rn|j                  t        d
   k(  r@| j                  |j                  | j!                  |j                        |z   dz	  dz         n|j                  t        d   k(  r=| j                  |j                  | j!                  |j                        |z   dz         n0|j                  t        d   k(  r9| j#                  |j                  | j%                  |j                        |z          n|j                  t        d   k(  r}|t        |j                        k(  rs|j                  |   }|dz  }| j                  |j                  | j!                  |j                        dz  |j                  z   |z   dz  dz	         nN|j                  t        d   k(  r8| j'                  |j                  | j)                  |j                        |z          |t        |j                        k  r@ || j                   _        t        | d      r7| j*                  D ](  }|j,                  D ]  }|xj.                  |z  c_         * t        | d      r| j0                  j2                  xj4                  |z  c_        | j0                  j2                  xj6                  |z  c_        | j0                  j2                  xj8                  |z  c_        | j0                  j2                  xj:                  |z  c_        t        | d      r
| j<                  j2                  }	t        |	d      r!|	j>                  r|	xj>                  |z  c_        t        |	d      r!|	j@                  r|	xj@                  |z  c_         t        |	d      r!|	jB                  r|	xjB                  |z  c_!        t        |	d      r!|	jD                  r|	xjD                  |z  c_"        t        |	d      r!|	jF                  r|	xjF                  |z  c_#        t        |	d      r!|	jH                  r|	xjH                  |z  c_$        t        |	d      r!|	jJ                  r|	xjJ                  |z  c_%        t        |	d      r!|	jL                  r|	xjL                  |z  c_&        t        |	d      r!|	jN                  r|	xjN                  |z  c_'        t        |	d      r!|	jP                  r|	xjP                  |z  c_(        | jR                  tT        k(  r-t        |	d      r!|	jV                  r|	xjV                  |z  c_+        t        |	d       r!|	jX                  r|	xjX                  |z  c_,        t        |	d!      r!|	jZ                  r|	xj\                  |z  c_.        t        |	d"      r!|	j\                  r|	xj\                  |z  c_.        t        |	d#      r!|	j^                  r|	xj^                  |z  c_/        t        |	d$      r!|	j`                  r|	xj`                  |z  c_0        t        |	d%      r!|	jb                  r|	xjb                  |z  c_1        t        |	d&      r!|	jd                  r|	xjd                  |z  c_2        t        |	d'      r!|	jf                  r|	xjf                  |z  c_3        t        |	d(      r!|	jh                  r|	xjh                  |z  c_4        t        |	d)      r!|	jj                  r|	xjj                  |z  c_5        t        |	d*      r#|	jl                  r|	xjl                  |z  c_6        y+y+y+y+y+y+),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.
        rD   rB   r  rA   r  zZRelocating image but PE does not have (or pefile cannot parse) a DIRECTORY_ENTRY_BASERELOCr   r4   r   r   r1   r  r   r   r   r  r   rK  r  r  LockPrefixTableEditListSecurityCookieSEHandlerTableGuardCFCheckFunctionPointerGuardCFDispatchFunctionPointerGuardCFFunctionTableGuardAddressTakenIatEntryTableGuardLongJumpTargetTableDynamicValueRelocTableCHPEMetadataPointerGuardRFFailureRoutine$GuardRFFailureRoutineFunctionPointer(GuardRFVerifyStackPointerFunctionPointerEnclaveConfigurationPointerVolatileMetadataPointerGuardEHContinuationTableGuardXFGCheckFunctionPointerGuardXFGDispatchFunctionPointer$GuardXFGTableDispatchFunctionPointer CastGuardOsDeterminedFailureModeGuardMemcpyFunctionPointerN)7r:  r'  rG  r  r  rK  r  r  rF  r  r  r  r  r  r  r  r  r  r+  r+  r&  rK  r/  r  r  rH  StartAddressOfRawDataEndAddressOfRawDataAddressOfIndexAddressOfCallBacksr  r;  r<  r=  r>  r?  r@  rA  rB  rC  rD  r  r  rE  rF  rG  rH  rI  rJ  rK  rL  rM  rN  rO  rP  )
rn  new_ImageBaserelocation_differencer  	entry_idxrD  
next_entryr0  funcload_configs
             r   r  zPE.relocate_imageJ  sO    !.0D0D0N0N N $$3349$$33A6;;4!<=++!01R!S T ,  4!<=&&9
 ";; ZE
 !"I#c%--&88 %i 8!Q	 ::9S)TT "ZZ?;Q+RR !00 %		$($8$8$C&;%<')%* #)!) #ZZ?;P+QQ !00 %		$($8$8$C&;%< #)	!) #ZZ?;T+UU
 !11 %		 $ 5 5eii @"7!8 #ZZ?;T+UU  )C,>> %).y)AJ%NI 00 %		%)%9%9%))%D%J&0nn%5&;%< '1%1
 $&!&	 #ZZ?;R+SS !11 %		 $ 5 5eii @"7!8a $c%--&88Zx .;D  * t5666 >C # >(==>> t23((//EE)E ((//CC)C ((//>>BWW>((//BB)B t:;">>EEK):;#33//3HH/;
38L8L((,AA(K)9:#22..2GG.K)9:#22..2GG.K)FG#??;;?TT;K)IJ#BB>>BWW>K)?@#88448MM4K)IJ#BB>>BWW>K)CD#<<88<QQ8K)AB#::66:OO6LL$AA-BC#77337LL3K)@A#99559NN5K)OP#HHHH-H K)ST#LLHH-H K)FG#??;;?TT;K)BC#;;77;PP7K)CD#<<88<QQ8K)GH#@@<<@UU<K)JK#CC??CXX?K)OP#HHDD-D K)KL#DD@@-@ K)EF#>>::>SS: ? Ge <u < :r   c                 P    | j                   j                  | j                         k(  S r   )r:  CheckSumgenerate_checksumrs  s    r   verify_checksumzPE.verify_checksumM  s"    ##,,0F0F0HHHr   c                    | j                         }| j                          || _        | j                  j	                         dz   }d}t        | j                        dz  }t        | j                        d|z
  |dk7  z  z   }t        |dz        D ]  }||dz  k(  r|dz   |dz  k(  r5|r3t        j                  d| j                  |dz  d  dd|z
  z  z         d   }n/t        j                  d| j                  |dz  |dz  dz          d   }||z  }|dk\  s|dz  |d	z	  z   } |d
z  |dz	  z   }||dz	  z   }|d
z  }|t        | j                        z   S )Nr^   r   r@   r4   r  r  r  r  r0   r  r1   )	r  r^  rC  r:  r  rG  r  rH  rI  )rn  r4  checksum_offsetr  	remainderdata_lenrN  r  s           r   r]  zPE.generate_checksumP  s    ::<  ..>>@4G &*	t}}%!i-IN)KLx1}% 	FAOq((1uQ'Iq1uw/5A	M3JK c4==QQ+KLQOH5 $z1h"nE	F v%(b.9R0f$ #dmm,,,r   c                     t         d   }| j                         s-| j                         s|| j                  j                  z  |k(  ryy)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.
        rX   TF)r  is_dllr  r  rH  )rn  EXE_flags     r   is_exez	PE.is_exe  sC     ))FG ^^%D,,<<<Ir   c                 P    t         d   }|| j                  j                  z  |k(  ryy)zCheck whether the file is a standard DLL.

        This will return true only if the image has the IMAGE_FILE_DLL flag set.
        rh   TF)r  r  rH  )rn  DLL_flags     r   rd  z	PE.is_dll  s.     ))9:t''777HDr   c                    t        | d      s| j                  t        d   g       t        | d      syh d}|j                  | j                  D ch c]  }|j
                  j                          c}      ryddh}|j                  | j                  D ch c]+  }|j                  j                         j                  d	      - c}      r)| j                  j                  t        d
   t        d   fv ryyc c}w c c}w )zCheck whether the file is a Windows driver.

        This will return true only if there are reliable indicators of the image
        being a driver.
        rK  r:   r:  F>      hal.dll   ndis.sys	   kdcom.dll   bootvid.dll   ntoskrnl.exeTs   pages   pagedr  r   r   )rK  r  r  intersectionrK  r0  r>  rE  r  r!  r:  	SubsystemSUBSYSTEM_TYPE)rn  system_DLLsrU  driver_like_section_namesr  s        r   r  zPE.is_driver  s    , t56'',-KLM (  t56

 ##(,(C(CDSWW]]_D
 %,h$7!$11AEOgW\\!((1O
   **78?@ ! E Ps   !C80C=c                 ^   dt        | j                        ffd	}t        | d      r6 || j                  j	                         | j
                  j                  f      | j                  D ]   } ||j                  |j                  f      " t        d   g}t        | j                  j                        D ]8  \  }}||v r	  || j                  |j                        |j                  f      : t        | j                        t#              kD  rt#              S y# t         $ r Y sw xY w)zoGet the offset of data appended to the file and not contained within
        the area described in the headers.r  c                 L    t              t        |       cxk  r|k  r| S  S S r   )sum)offset_and_size	file_sizelargest_offset_and_sizes     r   'update_if_sum_is_larger_and_within_filezQPE.get_overlay_data_start_offset.<locals>.update_if_sum_is_larger_and_within_file  s5     *+c/.BOiO&& P****r   r:  r?   N)rG  rC  rK  r:  r  r  r  rE  r0  r2  r  r  r  rR  r1  r  r  rv  )rn  rz  r  skip_directoriesr  r  ry  s         @r   get_overlay_data_start_offsetz PE.get_overlay_data_start_offset  s:    #) (+4=='9	+ 4*+&M((88:$$99'# }} 	G&M))7+@+@A'#	
 ,,LMN'(<(<(K(KL 		NC&&*Q--i.F.FGX+'			 t}}$; <<.// ! s   -D  	D,+D,c                 F    | j                         }|| j                  |d S y)zeGet the data appended to the file and not contained within the area described
        in the headers.Nr|  rC  rn  overlay_data_offsets     r   get_overlayzPE.get_overlay  s/     #@@B*==!4!566r   c                 b    | j                         }|| j                  d| S | j                  dd S )zKReturn the just data defined by the PE headers, removing any overlaid data.Nr~  r  s     r   trimzPE.trim  s;     #@@B*==!5"566}}Qr   c                    | j                   j                  t        k\  rf| j                  du rXt	        | j                   j                        s9| j
                  j                  d| j                   j                  z         d| _        |dz  S )NFz=If FileAlignment > 0x200 it should be a power of 2. Value: %xTi )r:  r@  MIN_VALID_FILE_ALIGNMENTrI  rf  rF  r  )rn  r#   s     r   r9  zPE.adjust_PointerToRawData&  s|    --1II))U2<$$22< &&S++99; .2* V|r   c                     |dk  r>||k7  r9| j                   du r+| j                  j                  d|dd|dd       d| _         t        |||      S )Nr!   FzIf SectionAlignment(0xr  z+) < 0x1000 it should equal FileAlignment(0xr\  T)rJ  rF  r  r&   )rn  r#   r$   r%   s       r   r?  zPE.adjust_SectionAlignment;  sp     v%"3311U:&&,->q,A B--;A,>aA 15-,S2C^TTr   r   )NFF)r   Nr   NF)r   )NF)r   N)r   N)r   N)r  )Nr  )ru  rv  rw  r  rz  r  r  r  r  r  r  r1  r_  r  r  r  r  r  r  r  r  r  r  r  r  rz  r\  rf  rN  rO  rP  rL  rN  rQ  rS  r&  r'  r,  r-  r<  r=  r>  &__IMAGE_DYNAMIC_RELOCATION_V2_format__(__IMAGE_DYNAMIC_RELOCATION64_V2_format__r  r  r  MAX_SYMBOL_EXPORT_COUNTrk  rV  rY  r^  rT  re  rh  rS  r  r  r  r  r  r  r  r  r  r  r  r.  rB  r  rA  r_  r`  r  r  r  r  r  r  r  r+  r  rG  rV  rZ  r  r9  ro  r  r  r  rO  rR  r  r  r  r  r  r  r  r  r  r  r  r  r+  r  r  r  r+  r  r  r  r  r  r  r  r  r  r"  r$  r&  r(  r+  r-  r  r  r"   rW  r2  r8  r  r^  r]  rf  rd  r  r|  r  r  r9  r?  re  r   r   r  r  	  s   BP#2$ '#
"($H!*&F #I'# 0,	*&)%"
+'1-
,(
!
#&!
 VKEN#
%!
($($
.*
C?
	B>>:
1-
=9!
/+

&"
($8.*t80,t1-
+'
-)
.*
0,
0,
,(
# 23j,,]~
Sj	 $22#hK\ TY]4~Xtun*83
jd#LU'n@
9v)V%NGRo'b	0>j$XF
PH	kZ?.;`?2h^  \ |AFCJ!6 'D->*( 1B 	P)6V  :1jXbH	5K
OKQ'K
NIO'K
OKQ6"":S : :LATFI0-d&;z*X	  *Ur   r  c                     dd l } d}| j                  dd  st        |       y | j                  d   dk(  r| j                  dd  s| j                  d       t	        | j                  d         }|j
                  j                  D ]M  }t        t        |j                  j                  |j                  z         |j                  |j                         O y t        t	        | j                  d         j                                y )Nr   z1pefile.py <filename>
pefile.py exports <filename>r4   r  r<   zerror: <filename> required)rJ  argvr  exitr  rX  r  r%  r:  r'  r  r  r  r  )rJ  usager}  r"  s       r   mainr  L  s     E 88AB<e	!		!xx|HH12_,,44 	CB&&003;;>?3;;	
 	b!o'')*r   __main__)r`   FFr  )r  
__author____version____contact__codecsr  r   r   r   r\  rk  r[  rp  r  rH  r  r  r   hashlibr   r   r   r   typingr   r<  register_errorlookup_errorr"   r  r   r&   r+   rA  r  r:  rp  r;  r  r  r  r  r  r~  r}  r  r  r  r  r   IMAGE_NUMBEROF_DIRECTORY_ENTRIESr  r  r  r  r8   directory_entry_typesr  image_characteristicsr  section_characteristicsrJ  debug_typesr  subsystem_typesrq  machine_typesr  relocation_typesr  dll_characteristicsr  ex_dll_characteristicsr  r  SECTOR_SIZEr7  r<  	registersrX  rS  r\  rl  ru  r  r  r  r  r  r  resource_typer  r  r=  r  r?  r5   rC  rA  r  rD  rR  r\  rc  rf  r  rh  r{  rw  r  r  r  r  r  r  r.  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r  r	  r  r  r  rG  rR  r[  rk  rt  r  r  r  r  r  r  r'  ascii_lowercaseascii_uppercaser  r  r  r  r  rV  rW  r   boolr  r  r  ru  re  r   r   <module>r     sU    
%     	   	      - -     )+>6+>+>?Q+R S
" 4  	
             !        #%   )    % 7 ( 45 & %%:; / b ''>? * +&
" o.%N M*  /0 $ ##67   &&<=   
  !!23 	& #	         0 ]+_B DhT w
 w-#* 0L-%%l3".	08*'/R RH HV I  &; &;T 
	
	
	
	
	
	
	
	
	
	
	
	
	
	
 & 4+ + 4d#' $'TA AHIy IX 4e$NX %NXbi,Y i,X% %] ;" ;"|CM C," ,"^m = *M  M = M }  "] "@m - &]  ] g$' g$TO O28N 8;~ ;8;~ ; 
. 
"XN X&Y~ Y&XN X&Y~ Y&Hn H;
 ;
|
 
F mm !! 	&( - V333fmmC&( 
 4FK
S%"#
?C
	
 
pXU pXUfq+* zF r   