3 \F @sdZddlZddlZddlZddlZddlTddlmZddddd d d d d ddg eZ[ejej ej e e e fZedZddZdKddddZdLddddZddddddddddd Zd!d"Zd#d$Zd%dZd&d'Zddd(d Zejd)d*Zd+ej_d,ej_d-ej_d.ej_d/ej_d0ej _d1ej!_d2ej"_Gd3ddeZ#dd4d5d Z$d6d7Z%d8d9Z&dMd:d;Z'dOddd=dZ(dQddd>d?d@Z)dddAdBZ*e(Z+dCdDZ,dEd Z-dFd Z.GdGddZ/dHdIZ0e1dJkre0dS)Rz0Disassembler of Python byte code into mnemonics.N)*)__all__ code_infodis disassembledistbdiscofindlinestarts findlabels show_codeget_instructions InstructionBytecode FORMAT_VALUEc Cs6yt||d}Wn tk r0t||d}YnX|S)zAttempts to compile the given source, first as an expression and then as a statement if the first approach fails. Utility function to accept strings in functions that otherwise expect code objects evalexec)compile SyntaxError)sourcenamecr/usr/lib64/python3.6/dis.py _try_compiles r)filecCs>|dkrt|ddSt|dr&|j}t|dr6|j}t|drF|j}t|drt|jj}x|D]n\}}t|t rdt d||dyt ||dWn0t k r}zt d||dWYdd}~XnXt |dqdWnbt|d rt ||dnJt|ttfrt||dn,t|tr(t||dnt d t|jdS) zzDisassemble classes, methods, functions, generators, or code. With no argument, disassemble the last traceback. N)r__func____code__gi_code__dict__zDisassembly of %s:zSorry:co_codez(don't know how to disassemble %s objects)rhasattrrrrsortedritems isinstance _have_codeprintr TypeErrorrbytes bytearray_disassemble_bytesstr_disassemble_strtype__name__)xrr"rZx1msgrrrr"s6         c CsZ|dkrBy tj}Wntk r.tdYnXx|jr@|j}q2Wt|jj|j|ddS)z2Disassemble a traceback (default: last traceback).Nz no last traceback to disassemble)r) syslast_tracebackAttributeError RuntimeErrortb_nextrtb_framef_codetb_lasti)tbrrrrrEs  Z OPTIMIZEDZ NEWLOCALSZVARARGSZ VARKEYWORDSZNESTEDZ GENERATORZNOFREEZ COROUTINEZITERABLE_COROUTINEZASYNC_GENERATOR)  @icCsbg}xRtdD]8}d|>}||@r|jtj|t|||N}|sPqW|jt|dj|S)z+Return pretty representation of code flags.r>r9z, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagrrr pretty_flags_srLcCsht|dr|j}t|dr |j}t|dr0|j}t|trDt|d}t|drR|Stdt|j dS)zMHelper to handle methods, functions, generators, strings and raw code objectsrrrz rz(don't know how to disassemble %s objectsN) r rrrr#r*rr&r,r-)r.rrr_get_code_objectms      rMcCs tt|S)z1Formatted details of methods, functions, or code.)_format_code_inforM)r.rrrr|scCsg}|jd|j|jd|j|jd|j|jd|j|jd|j|jd|j|jdt|j|j r|jdx t |j D]}|jd |qW|j r|jd x t |j D]}|jd |qW|j r|jd x t |j D]}|jd |qW|j rF|jd x"t |j D]}|jd |q.W|jr||jdx"t |jD]}|jd |qdWdj|S)NzName: %szFilename: %szArgument count: %szKw-only arguments: %szNumber of locals: %szStack size: %szFlags: %sz Constants:z%4d: %rzNames:z%4d: %szVariable names:zFree variables:zCell variables: )rCco_name co_filename co_argcountco_kwonlyargcount co_nlocals co_stacksizerLco_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrG)colinesZi_cZi_nrrrrNs:     rNcCstt||ddS)z}Print details of methods, functions, or code to *file*. If *file* is not provided, the output is printed on stdout. )rN)r%r)r]rrrrr s _InstructionzBopname opcode arg argval argrepr offset starts_line is_jump_targetz!Human readable name for operationzNumeric code for operationz6Numeric argument to operation (if any), otherwise Nonez4Resolved arg value (if known), otherwise same as argz0Human readable description of operation argumentz1Start index of operation within bytecode sequencez4Line started by this opcode (if any), otherwise Nonez1True if other code jumps to here, otherwise Falsec@seZdZdZdddZdS)r aKDetails for a bytecode operation Defined fields: opname - human readable name for operation opcode - numeric code for operation arg - numeric argument to operation (if any), otherwise None argval - resolved arg value (if known), otherwise same as arg argrepr - human readable description of operation argument offset - start index of operation within bytecode sequence starts_line - line started by this opcode (if any), otherwise None is_jump_target - True if other code jumps to here, otherwise False FcCsg}|r:|jdk r,d|}|j||jn|jd||rJ|jdn |jd|jrf|jdn |jd|jt|jjd|j|jjd |jdk r|jt|jjd |j r|jd |j d dj |j S) zFormat instruction details for inclusion in disassembly output *lineno_width* sets the width of the line number field (0 omits it) *mark_as_current* inserts a '-->' marker arrow as part of the line Nz%%%dd z-->z z>>z r;()) starts_linerCis_jump_targetreproffsetrjustopnameljustargargreprrGrstrip)self lineno_widthZmark_as_currentZfieldsZ lineno_fmtrrr _disassembles&      zInstruction._disassembleN)r`F)r- __module__ __qualname____doc__rrrrrrr s ) first_linecCsTt|}|j|j}tt|}|dk r4||j}nd}t|j|j|j |j |||S)aIterator for the opcodes in methods, functions or code Generates a series of Instruction named tuples giving the details of each operations in the supplied code. If *first_line* is not None, it indicates the line number that should be reported for the first source line in the disassembled code. Otherwise, the source line information (if any) is taken directly from the disassembled code object. Nr) rMr\r[dictr co_firstlineno_get_instructions_bytesrrZrYrW)r.rvr] cell_names linestarts line_offsetrrrr s    cCs |}|dk r||}|t|fS)zHelper to get optional details about const references Returns the dereferenced constant and its repr if the constant list is defined. Otherwise returns the constant index and its repr(). N)rh)Z const_indexZ const_listargvalrrr_get_const_infosr~cCs*|}|dk r||}|}nt|}||fS)zHelper to get optional details about named references Returns the dereferenced name as both value and repr if the name list is defined. Otherwise returns the name index and its repr(). N)rh)Z name_indexZ name_listr}rnrrr_get_name_infos rc csxt|}d}xdt|D]V\} } } |dk rH|j| d}|dk rH||7}| |k} d} d}| dk rT| } | tkr~t| |\} }n| tkrt| |\} }n| tkr| d| } dt| }n| t krt| |\} }n| t krt | } | }nn| t krt| |\} }nV| t krTdtttf| d@t| d@f} d | d@}| d rT|rL|d 7}|d 7}tt| | | | || || VqWdS)a&Iterate over the instructions in a bytecode string. Generates a sequence of Instruction namedtuples giving the details of each opcode. Additional information about the code's runtime environment (e.g. variable names, constants) can be specified using optional arguments. Nr:zto r`r;r*rhasciir9z, z with format)rr*rhr)r _unpack_opargsrEZhasconstr~ZhasnamerhasjrelrhZhaslocalZ hascompareZcmp_opZhasfreerr*rboolr rk)codevarnamesrI constantscellsr{r|labelsrfrioprmrgr}rnrrrrysH         ryr9c Cs<|j|j}tt|}t|j||j|j|j|||ddS)zDisassemble a code object.)rN) r\r[rwr r)rrZrYrW)r]lastirrzr{rrrrJs  )rr|c Cs||dk } | rdnd} xbt|||||||dD]H} | oF| jdk oF| jdk} | rVt|d| j|k} t| j| | |dq,WdS)Nr`r)r|)r)ryrfrir%rr)rrrrIrrr{rr|Z show_linenorqZinstrZnew_source_lineZis_current_instrrrrr)Qs       r)cCstt|d|ddS)z)rN)rr)rrrrrr+csr+ccshd}x^tdt|dD]J}||}|tkrP||d|B}|tkrJ|d>nd}nd}|||fVqWdS)Nrr:r9r<)rBlenZ HAVE_ARGUMENTZ EXTENDED_ARG)rZ extended_argrJrrmrrrrisrcCs`g}xVt|D]J\}}}|dk r|tkr6|d|}n|tkr|}nq||kr|j|qW|S)z`Detect all offsets in a byte code which are jump targets. Return the list of offsets. Nr:)rrZhasjabsrC)rrrirrmZlabelrrrr tsccs|jddd}|jddd}d}|j}d}xPt||D]B\}}|rd||kr\||fV|}||7}|dkrt|d8}||7}q:W||kr||fVdS)zFind the offsets in a byte code which are start of lines in the source. Generate pairs (offset, lineno) as described in Python/compile.c. rNr:r9r@rA) co_lnotabrxzip)rZbyte_incrementsZline_incrementsZ lastlinenolinenoZaddrZ byte_incrZ line_incrrrrr s   c@sLeZdZdZdddddZddZdd Zed d Zd d Z ddZ dS)rzThe bytecode operations of a piece of code Instantiate this with a function, method, string of code, or a code object (as returned by compile()). Iterating over this yields the bytecode operations as Instruction instances. N)rvcurrent_offsetcCsdt||_}|dkr&|j|_d|_n||_||j|_|j|j|_tt ||_ ||_ ||_ dS)Nr) rMcodeobjrxrv _line_offsetr\r[ _cell_namesrwr _linestarts_original_objectr)rpr.rvrr]rrr__init__s zBytecode.__init__c Cs*|j}t|j|j|j|j|j|j|jdS)N)r|) rryrrZrYrWrrr)rpr]rrr__iter__s zBytecode.__iter__cCsdj|jj|jS)Nz{}({!r}))format __class__r-r)rprrr__repr__s zBytecode.__repr__cCs$x|jr|j}qW||jj|jdS)z/ Construct a Bytecode from the given traceback )r)r4r5r6r7)clsr8rrrfrom_tracebacks zBytecode.from_tracebackcCs t|jS)z3Return formatted information about the code object.)rNr)rprrrinfosz Bytecode.infocCs`|j}|jdk r|j}nd}tj2}t|j|j|j|j|j |j |j ||d |j SQRXdS)z3Return a formatted view of the bytecode operations.Nr9)rrIrrr{r|rr) rrioStringIOr)rrZrYrWrrrgetvalue)rpr]rioutputrrrrs   z Bytecode.dis) r-rsrtrurrr classmethodrrrrrrrrs  c Csdddl}|j}|jd|jddd|j}|j}|j}WdQRXt||jjd}t |dS)z*Simple test program to disassemble a file.rNinfile?-)r,nargsdefaultr) argparseArgumentParser add_argumentZFileType parse_argsrreadrrr)rparserargsrrrrrr_testsr__main__)N)N)NNNNNrr)rr)rNNNNN)2rur0types collectionsrZopcoderZ _opcodes_all MethodType FunctionTypeCodeTyper staticmethodr,r$ZopmaprrrrrDrLrMrrNr namedtupler_rkrmr}rnrirfrgr r r~rryrr)r+rrr r rrr-rrrrsx     #3  2 =