3 \F @sddlZddlZddlZddlZddlTddlmZdddddd d d d d dg eZ[ejejej e e e fZ edZddZdJddddZdKddddZddddddddddd Zd d!Zd"d#Zd$dZd%d&Zddd'd Zejd(d)Zd*ej_d+ej_d,ej_d-ej_d.ej_d/ej _d0ej!_d1ej"_Gd2d d eZ#dd3d4d Z$d5d6Z%d7d8Z&dLd9d:Z'dNdddd?Z)ddd@dAZ*e(Z+dBdCZ,dDd Z-dEd Z.GdFddZ/dGdHZ0e1dIkre0dS)QN)*)__all__ code_infodis disassembledistbdiscofindlinestarts findlabels show_codeget_instructions InstructionBytecode FORMAT_VALUEc Cs6yt||d}Wn tk r0t||d}YnX|S)Nevalexec)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|drt ||dnJt|ttfrt||dn,t|tr(t||dnt d t|jdS) 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)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)Nr>r9z, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagrrr pretty_flags_srLcCsht|dr|j}t|dr |j}t|dr0|j}t|trDt|d}t|drR|Stdt|j dS)Nrrrz rz(don't know how to disassemble %s objects) r rrrr#r*rr&r,r-)r.rrr_get_code_objectms      rMcCs tt|S)N)_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)N)r)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@seZdZdddZdS)r 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) 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__rrrrrrr s) first_linecCsTt|}|j|j}tt|}|dk r4||j}nd}t|j|j|j |j |||S)Nr) rMr\r[dictr co_firstlineno_get_instructions_bytesrrZrYrW)r.rur] cell_names linestarts line_offsetrrrr s    cCs |}|dk r||}|t|fS)N)rh)Z const_indexZ const_listargvalrrr_get_const_infosr}cCs*|}|dk r||}|}nt|}||fS)N)rh)Z name_indexZ name_listr|rnrrr_get_name_infos r~c 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) Nr:zto r`r;r*rhasciir9z, z with format)rr*rhr)r _unpack_opargsrEZhasconstr}Zhasnamer~hasjrelrhZhaslocalZ hascompareZcmp_opZhasfreerr*rboolr rk)codevarnamesrI constantscellsrzr{labelsrfrioprmrgr|rnrrrrxsH         rxr9c Cs<|j|j}tt|}t|j||j|j|j|||ddS)N)r) r\r[rvr r)rrZrYrW)r]lastirryrzrrrrJs  )rr{c Cs||dk } | rdnd} xbt|||||||dD]H} | oF| jdk oF| jdk} | rVt|d| j|k} t| j| | |dq,WdS)Nr`r)r{)r)rxrfrir%rr)rrrrIrrrzrr{Z show_linenorqZinstrZnew_source_lineZis_current_instrrrrr)Qs       r)cCstt|d|ddS)Nz)r)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)Nr:)rrZhasjabsrC)rrrirrmZlabelrrrr tsccs|jddd}|jddd}d}|j}d}xPt||D]B\}}|rd||kr\||fV|}||7}|dkrt|d8}||7}q:W||kr||fVdS)Nrr:r9r@rA) co_lnotabrwzip)rZbyte_incrementsZline_incrementsZ lastlinenolinenoZaddrZ byte_incrZ line_incrrrrr s   c@sHeZdZdddddZddZddZed d Zd d Zd dZ dS)rN)rucurrent_offsetcCsdt||_}|dkr&|j|_d|_n||_||j|_|j|j|_tt ||_ ||_ ||_ dS)Nr) rMcodeobjrwru _line_offsetr\r[ _cell_namesrvr _linestarts_original_objectr)rpr.rurr]rrr__init__s zBytecode.__init__c Cs*|j}t|j|j|j|j|j|j|jdS)N)r{) rrxrrZrYrWrrr)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)N)r)r4r5r6r7)clsr8rrrfrom_tracebacks zBytecode.from_tracebackcCs t|jS)N)rNr)rprrrinfosz Bytecode.infocCs`|j}|jdk r|j}nd}tj2}t|j|j|j|j|j |j |j ||d |j SQRXdS)Nr9)rrIrrrzr{rr) rrioStringIOr)rrZrYrWrrrgetvalue)rpr]rioutputrrrrs   z Bytecode.dis) r-rsrtrrr classmethodrrrrrrrrs   c Csdddl}|j}|jd|jddd|j}|j}|j}WdQRXt||jjd}t |dS)Nrinfile?-)r,nargsdefaultr) argparseArgumentParser add_argumentZFileType parse_argsrreadrrr)rparserargsrrrrrr_testsr__main__)N)N)NNNNNrr)rr)rNNNNN)2r0types collectionsrZopcoderZ _opcodes_all MethodType FunctionTypeCodeTyper staticmethodr,r$ZopmaprrrrrDrLrMrrNr namedtupler_rk__doc__rmr|rnrirfrgr r r}r~rxrr)r+rrr r rrr-rrrrsv     #3  2 =