o w[e4N @sdZddlZddlZddlZddlZddlTddlmZgdeZ[ejej ej e e e fZedZdedfed fed ffZed Zd Zd dZdVdddddZdVddddZddddddddddd Zd d!Zd"d#Zd$d%Zd&d'Zddd(d)Ze d*d+Z!d,e!j"_d-e!j_d.e!j#_d/e!j$_d0e!j%_d1e!j&_d2e!j'_d3e!j(_d4Z)d5Z*Gd6d7d7e!Z+dd8d9d:Z,d;d<Z-d=d>Z.  dWd?d@Z/dXdddBdCZ0ddddDdEZ1  dYdddFdGdHZ2dIdJZ3e0Z4dKdLZ5dMdNZ6dOdPZ7GdQdRdRZ8dSdTZ9e:dUkre9dSdS)Zz0Disassembler of Python byte code into mnemonics.N)*)__all__) code_infodis disassembledistbdiscofindlinestarts findlabels show_codeget_instructions InstructionBytecode FORMAT_VALUE)Nstrreprascii MAKE_FUNCTION)defaults kwdefaults annotationsclosurecCs4z t||d}W|Styt||d}Y|Sw)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/lib/python3.10/dis.py _try_compiles r"filedepthc Csh|dur t|ddSt|dr|j}t|dr|j}nt|dr%|j}nt|dr.|j}nt|dr6|j}t|drt|j }|D]9\}}t |t r}t d ||dz t |||d Wntyw}z t d ||dWYd}~nd}~wwt |dqDdSt|d rt|||d dSt |ttfrt||ddSt |trt|||d dStd t|j)a0Disassemble classes, methods, functions, and other compiled objects. With no argument, disassemble the last traceback. Compiled objects currently include generator objects, async generator objects, and coroutine objects, all of which store their code object in a special attribute. Nr$__func____code__gi_codeag_codecr_code__dict__zDisassembly of %s:r#zSorry:co_code(don't know how to disassemble %s objects)rhasattrr'r(r)r*r+sortedr,items isinstance _have_codeprintr TypeError_disassemble_recursivebytes bytearray_disassemble_bytesr_disassemble_strtype__name__)xr$r%r1rx1msgr r r!r+sH            rr&cCsV|durztj}Wn tytddw|jr|j}|jst|jj|j|ddS)z2Disassemble a traceback (default: last traceback).Nz no last traceback to disassembler&) syslast_tracebackAttributeError RuntimeErrortb_nextrtb_framef_codetb_lasti)tbr$r r r!rXs   r OPTIMIZED NEWLOCALSVARARGS VARKEYWORDSNESTED GENERATORNOFREE COROUTINEITERABLE_COROUTINEASYNC_GENERATOR)  @icCs`g}tdD]}d|>}||@r#|t|t|||N}|s#nq|t|d|S)z+Return pretty representation of code flags.rXrS, )rangeappendCOMPILER_FLAG_NAMESgethexjoin)flagsnamesiflagr r r! pretty_flagsrs  rgcCst|dr|j}t|dr|j}nt|dr|j}nt|dr#|j}nt|dr+|j}t|tr5t|d}t|dr<|St dt |j ) zDHelper to handle methods, compiled or raw code objects, and strings.r'r(r)r*r+z r-r.) r/r'r(r)r*r+r2rr"r5r;r<r=r r r!_get_code_objects"        ricCs tt|S)z1Formatted details of methods, functions, or code.)_format_code_inforirhr r r!rs rcCsxg}|d|j|d|j|d|j|d|j|d|j|d|j|d|j|dt|j |j r[|d t |j D] }|d |qQ|j rr|d t |j D] }|d |qh|j r|d t |j D] }|d |q|jr|dt |jD] }|d |q|jr|dt |jD] }|d |qd|S)NzName: %szFilename: %szArgument count: %szPositional-only arguments: %szKw-only arguments: %szNumber of locals: %szStack size: %szFlags: %sz Constants:z%4d: %rzNames:z%4d: %szVariable names:zFree variables:zCell variables: )r^co_name co_filename co_argcountco_posonlyargcountco_kwonlyargcount co_nlocals co_stacksizergco_flags co_consts enumerateco_names co_varnames co_freevars co_cellvarsrb)colinesi_ci_nr r r!rjs<      rjcCstt||ddS)z}Print details of methods, functions, or code to *file*. If *file* is not provided, the output is printed on stdout. r&N)r4r)rzr$r r r!r sr _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 FrUcCsg}|r|jdurd|}|||jn|d||r%|dn|d|jr3|dn|d|t|j|||jt|j duri|t|j t |j ri|d|j d d | S) a%Format 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 *offset_width* sets the width of the instruction offset field Nz%%%dd z-->z z>>z ()) starts_liner^is_jump_targetroffsetrjustopnameljust _OPNAME_WIDTHarg _OPARG_WIDTHargreprrbrstrip)self lineno_widthmark_as_current offset_widthfields lineno_fmtr r r! _disassembles&      zInstruction._disassembleN)rFrU)r< __module__ __qualname____doc__rr r r r!r s r ) first_linecCsTt|}|j|j}tt|}|dur||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) riryrxdictr co_firstlineno_get_instructions_bytesr-rwrvrt)r=rrz cell_names linestarts line_offsetr r r!r s   r cCs |}|dur ||}|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(). Nr) const_index const_listargvalr r r!_get_const_infos rcCs0|}|dur||}|}||fSt|}||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(). Nr) name_index name_listrrr r r!_get_name_info'src #st|}d}t|D]\} } |dur"|| d}|dur"||7}| |v} d} d} durˆ} | tvr@r|VqdS)rSNr ).0resrr r! js z*_get_instructions_bytes..)r _unpack_opargsr`hasconstrhasnamerhasjabsrhasjrelhaslocal hascomparecmp_ophasfreerFORMAT_VALUE_CONVERTERSboolrrbruMAKE_FUNCTION_FLAGSr r)codevarnamesrd constantscellsrrlabelsrroprrrr rr!r7sZ   rc Cs<|j|j}tt|}t|j||j|j|j|||ddS)zDisassemble a code object.r&N) ryrxrr r9r-rwrvrt)rzlastir$rrr r r!rps    rcCstt||d|dus|dkr6|dur|d}|jD]}t|dr5t|dtd|f|dt|||dqdSdS)Nr&rrSr-zDisassembly of %r:r#)rrtr/r4r6)rzr$r%r=r r r!r6ws    r6)r$rc Cst|} | rt||} | dkrtt| } nd} nd} t|d} | dkr/tt| } nd} t|||||||dD]%}| oI|jduoI|jdk}|rQt|d|j|k}t| | || |dqN)r6r")rkwargsr r r!r:sr:ccsjd}tdt|dD]'}||}|tkr(||d|B}|tkr%|d>nd}nd}d}|||fVq dS)NrrTrSrV)r]r HAVE_ARGUMENT EXTENDED_ARG)r extended_argrerrr r r!rsrcCsdg}t|D])\}}}|dur/|tvr|d|d}n |tvr%|d}nq||vr/||q|S)z`Detect all offsets in a byte code which are jump targets. Return the list of offsets. NrT)rrrr^)rrrrrlabelr r r!r s  r ccs<d}|D]\}}}|dur||kr|}||fVqdS)zqFind the offsets in a byte code which are start of lines in the source. Generate pairs (offset, lineno) N)co_lines)rlastlinestartendliner r r!r s r c@sLeZdZdZdddddZddZdd Zed d Zd d Z ddZ dS)raThe bytecode operations of a piece of code Instantiate this with a function, method, other compiled object, string of code, or a code object (as returned by compile()). Iterating over this yields the bytecode operations as Instruction instances. N)rcurrent_offsetcCsdt||_}|dur|j|_d|_n ||_||j|_|j|j|_tt ||_ ||_ ||_ dS)Nr) ricodeobjrr _line_offsetryrx _cell_namesrr _linestarts_original_objectr)rr=rrrzr r r!__init__s  zBytecode.__init__c Cs*|j}t|j|j|j|j|j|j|jdS)Nr) rrr-rwrvrtrrr)rrzr r r!__iter__s zBytecode.__iter__cCsd|jj|jS)Nz{}({!r}))format __class__r<rrr r r!__repr__s zBytecode.__repr__cCs$|jr |j}|js||jj|jdS)z/ Construct a Bytecode from the given traceback )r)rDrErFrG)clsrHr r r!from_tracebackszBytecode.from_tracebackcCs t|jS)z3Return formatted information about the code object.)rjrrr r r!infos z Bytecode.infoc Csx|j}|jdur |j}nd}t }t|j|j|j|j|j |j |j ||d | WdS1s5wYdS)z3Return a formatted view of the bytecode operations.Nr)rrdrrrrr$r) rrioStringIOr9r-rwrvrtrrrgetvalue)rrzroutputr r r!rs   $z Bytecode.dis) r<rrrrrr classmethodrrrr r r r!rs   rcCszddl}|}|jd|dddd|}|j }|}Wdn1s*wYt||jjd}t |dS) z*Simple test program to disassemble a file.rNinfilerb?-)r;nargsdefaultr) argparseArgumentParser add_argumentFileType parse_argsrreadrrr)rparserargsrrrr r r!_tests  r__main__)N)NNNNNr)r)rNNNNN);rr@types collectionsropcoder _opcodes_all MethodType FunctionTypeCodeTyper staticmethodr;r3opmaprrrrrrrr"rrr_rgrirrjr namedtupler~rrrrrrrrrr r rrrrr6r9r:rrr r rrr<r r r r!s   -  4  9    =