3 ^) @s6dZddlZddlZddlZddlZddlZddlmZddlZddlm Z m Z ddl m Z ddl mZmZddlmZmZddlmZdd lmZdd lmZdd lmZd d ZejddddddfddZejddddddddf ddZdddZddZ ddZ!ejdddfddZ"ddZ#e$dkr2e#dS) a CPython independent disassembly routines There are two reasons we can't use Python's built-in routines from dis. First, the bytecode we are extracting may be from a different version of Python (different magic number) than the version of Python that is doing the extraction. Second, we need structured instruction information for the (de)-parsing step. Python 3.4 and up provides this, but we still do want to run on Python 2.7. N)deque)IS_PYPYPYTHON_VERSION)Bytecode)iscodecodeType2Portable)check_object_path load_module)PYTHON_MAGIC_INT)format_code_info)VERSION) op_importscCsnt|ttfkr$djdd|D}t|}|r8|d7}|tjkrLt|S|rVd}nd}td||fdS)N.cSsg|] }t|qS)str).0xrrF/home/psgendb/BIRCHDEV/python/lib/python3.6/site-packages/xdis/main.py 3szget_opcode..Zpypyz for pypyz)%s is not a Python version%s I know about)typelisttuplejoinrr keys TypeError)versionis_pypylookupZpypy_strrrr get_opcode0s rFTc  Cs|ptj} |rd} nd} tr"d} nd} |rfd} |r:t|} | jdt| |d| | djtjjdft dkr|dkr| jd| jd |dk rt j j |}| jd || j|j d |dk r| jd ||dk r| jd || r| jd|j dS)NzPyPy rzO# pydisasm version %s # %sPython bytecode %s%s # Disassembled from %sPython %s z (%s)z #  g@z? ## **Warning** bytecode strings will be converted to strings. z7## To avoid loss, run this from Python 3.0 or greater z# Timestamp in code: %dz (%Y-%m-%d %H:%M:%S) z'# Source code size mod 2**32: %d bytes z# SipHash: 0x%x z# Embedded file name: %s )sysstdoutrrwriter rrsplitrdatetime fromtimestampstrftime co_filename)bytecode_versionco timestampoutr magic_int source_sizesip_hashheader show_filenamereal_outZ co_pypy_strZ run_pypy_strZ magic_strvaluerrrshow_module_header@s@    r4c  Cst|s tt|||||||||dd |p0tj} |jrR| rR| jt||dt||} | rxt | ||| it gnt |g}t | ||| | dddS)z: diassembles and deparses a given code block 'co' F)r1r T) show_bytes dup_linesN) rAssertionErrorr4r!r"r(r#r rdisco_loop_asm_formatsetr disco_loop)r)r*r+r,rr-r.r/r0 asm_formatr5r6r2opcqueuerrrdiscozs*      r>c Csxt|dkr|j}|jdkr8|jdt||dt|||d}|j|j|ddx |jD]}t|rd|j |qdWqWdS) aDisassembles a queue of code objects. If we discover another code object which will be found in co_consts, we add the new code to the list. Note that the order of code discovery is in the order of first encountered which is not amenable for the format used by a disassembler where code objects should be defined before using them in other functions. However this is not recursive and will overall lead to less memory consumption at run time. r?r )r6)r5N)r?r@) lenpopleftco_namer#r rdis co_constsrappend) r<rr=r2r6r5r*bytecodecrrrr:s    r:cCsd|t|fS)Nz%s_0x%x)id)basenameco_coderrr code_uniquifysrLcCst|}|j}|j||}g}x|jD]} t| rt| tjrHt| } n| } t||| |||t j dt | } | r| j d} | dkrt | | j}|| _| j|j| q&|j| q&W||_t j d|j} | s||kr*||kr|} n | j d} | dkrt | |j}|}||kst| ||<||_n$||krNt ||j}|||<||_|j}|j||jdkst|jr|jdt|||dt||dd} |j| jdd dd S) zProduces disassembly in a format more conducive to automatic assembly by producing inner modules before they are used by outer ones. Since this is recusive, we'll use more stack space at runtime. z.* object <(.+)> atmodulez^<(.+)>$zr T)r6)r;N)rrCgetrEr isinstancetypesCodeTyper8rematchrgrouprLrKZfreezerFr7addr(r#r rrD)r<rr*r2Z fn_name_mapZall_fnsrCZ mapped_nameZ new_constsrHZc_compatmrJrGrrrr8sX              r8cCsd}y"t|}t|\}}}} } } } WnV|dkr8tj|} t|dj}t||d} t} t}d} | j } | j }t }YnX|}|rt || ||| || | dd nt || ||| || | ||d || |||| | | fS)z disassemble Python byte-code file (.pyc) If given a Python source file (".py") file, we'll try to find the corresponding compiled object. If that fails we'll compile internally for the Python version currently running NrexecrT)r1)r;r5)rr osstatopenreadcompilerr st_sizest_mtimerr4r>)filenameZ outstreamr;r0r5Z pyc_filenamerr+r-r*rr.r/r[sourcerrrdisassemble_file sP     rccCsXttj}|dkrB|dkr&tjr&t}qLtjjdttjdn tjd}t |dS)z*Simple test program to disassemble a file.rMz$usage: %s [-|CPython compiled file] N) rAr!argvxdisZPYTHON3__file__stderrr#exitrc)Zargcfnrrr_testLs   rk__main__)FF)%__doc__r%rZrSr!rQ collectionsrrfrrZ xdis.bytecoderZ xdis.codetyperrZ xdis.loadrr Z xdis.magicsr Zxdis.cross_disr Z xdis.versionr Zxdis.op_importsr rr"r4r>r:rLr8rcrk__name__rrrrsF(       4 & B?