3 ^2>@sdZddlZddlZddlmZddlmZmZmZm Z ddl m Z ddl m Z mZerrddlmZddlmZn ddlmZejejejefZd d Zd d Zd dZddZdddZddZGdddeZddZ dS)zyPython bytecode and instruction classes Extracted from Python 3 dis module but generalized to allow running on Python 2. N)PYTHON3)get_code_objectformat_code_infoinstruction_sizeop_has_argument) Instruction)code2numnum2code)StringIO)reducecCs ||j>S)N)ZEXTENDED_ARG_SHIFT)opcvalrJ/home/psgendb/BIRCHDEV/python/lib/python3.6/site-packages/xdis/bytecode.pyextended_arg_val&srcCst|dks||ddkr dSd}t|d}||dd}x\||kr||d|krd|d}n&||d|kr~|d}n ||dS||dd}qBW|t|kr|t|ddS||dS)alinestarts is expected to be a *list) of (offset, line number) where both offset and line number are in increasing order. Return the closes line number at or below the offset. If offset is less than the first line number given in linestarts, return line number 0. r)len)offset linestartslowhighZmidrrr offset2line)s       rcCsN|}|dk r||}t|trBt|tddddgkrB|d|fS|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(). Nnanz-naninfz-infz float('%s')) isinstancefloatstr frozensetrepr) const_index const_listargvalrrr_get_const_infoCs " r#cCs6|}|dk r&|t|kr&||}|}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)rr) name_index name_listr"argreprrrr_get_name_infoUs r'ccs&|j||}d} |jdkrdnd} d} t|} d} d}d} t|j|}x| | kr t|| }| }|dk r|j| d} | dk r| |7} | |krd}nd}| d7} d}d}d}t||}d}|r| rt|| | B}||jkr|d>nd} | d7} n@t|| t|| dd | }| d 7} ||jkr.|d nd} |}||jkrXt ||\}}d }n"||j krzt ||\}}d }n||j kr| |}dt |}d}n||jkr|}dt |}d}n||jkrt ||\}}d}n||jkr|j|}|}d}nx||jkr"t ||\}}d}nX||jkrjd}| pD|j|dkszdt|| d t|| df}n||jkrzd}t|dr|j||jkr|j|j||}n| r| d7} |j|}t||||}||jk}t|||||||||| ||dk V||jkr|dnd}qJWdS)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. rg @TFNrriconstnamezto ZjrelZjabslocalZcomparefreenargs RAISE_VARARGSDUP_TOPX MAKE_FUNCTIONz%d positional, %d namedZvargsopcode_arg_fmt)r0r1r2) findlabelspython_versionrr EXTENDED_ARGrgetr CONST_OPSr#ZNAME_OPSr'ZJREL_OPSrZJABS_OPSZ LOCAL_OPSZ COMPARE_OPScmp_opZFREE_OPSZ NARGS_OPSopnameZ VARGS_OPShasattrr3Znofollowr)bytecoder varnamesnames constantscellsr line_offsetlabels extended_argZ python_36 starts_lineniZextended_arg_countZextended_arg_sizeopris_jump_targetargr"r&has_argZoptyper:Z inst_sizeZ fallthroughrrrget_instructions_bytesgs                       rKcCs|t||S)N)r)rGr rrrr next_offsetsrLc @s|eZdZdZdddZddZdd Zed d Zd d Z dddZ dddZ dddddde j dddf ddZdddZdS)BytecodezBytecode operations involving a Python code object. 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. NTcCst||_}d|_f|_|jdkr`|dkr6|j|_n||_||j|_|jdkr`|j|j|_t |j ||d|_ ||_ ||_ |j|_||_dS)Nrg?g@) dup_lines)rcodeobj _line_offset _cell_namesversionco_firstlineno first_line co_cellvars co_freevarsdictfindlinestarts _linestarts_original_objectr r:Zopnamescurrent_offset)selfxr rTr[rNcorrr__init__s    zBytecode.__init__c Cs.|j}t|j|j|j|j|j|j|j|j dS)N)rA) rOrKco_coder co_varnamesco_names co_constsrQrYrP)r\r^rrr__iter__s zBytecode.__iter__cCsdj|jj|jS)Nz{}({!r}))format __class____name__rZ)r\rrr__repr__s zBytecode.__repr__cCs$x|jr|j}qW||jj|jdS)z/ Construct a Bytecode from the given traceback )r[)tb_nexttb_framef_codetb_lasti)clstbrrrfrom_tracebacks zBytecode.from_tracebackcCst|j|jjS)z3Return formatted information about the code object.)rrOr rR)r\rrrinfo sz Bytecode.infoFc Csv|j}|jdk r|j}nd}t}|jjdkr<|j}|j}nd}d}|j|j|j |j |j |||j ||||d |j S)z3Return a formatted view of the bytecode operations.Nrg@) r=r>r?r@rrAfilelasti asm_format show_bytes)rOr[r r rRrQrYdisassemble_bytesr`rarbrcrPgetvalue)r\rsrtr^routputr@rrrrdiss(   z Bytecode.disc CsX|dkrBy tj}Wntk r.tdYnXx|jr@|j}q2W|j|jj|jdS)z2Disassemble a traceback (default: last traceback).Nz no last traceback to disassemble) syslast_tracebackAttributeError RuntimeErrorri disassemblerjrkrl)r\rnrrrdistb*s  zBytecode.distbrrc  Cs|dk } | rdnd} xlt||j|||||| dD]N}| oJ|jdk oJ|jdk}|rZ|jd|j|k}|j|j| || | dq0WdS)Nr)rA )rKr rDrwriter~)r\coderrr=r>r?r@rrqrArsrt show_lineno lineno_widthinstrnew_source_lineis_current_instrrrrrv4s        zBytecode.disassemble_bytesc Cs\t|}|j|j}t|jj|}|dk r8||j}nd}t|j|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) rrUrVrWr rXrSrKr`rarbrc)r\r]rTr^ cell_namesrrArrrget_instructionsKs    zBytecode.get_instructions)NNT)FF)Nru)N)rg __module__ __qualname____doc__r_rdrh classmethodrorpryrrzstdoutrvrrrrrrMs    rMc sg}xt|D]\}}|d}|dd||jkrFtd||f|j|}|j|t|fddD} xL| D]D} ||jkr|n|j| } | d krtd||fq||t| 7}q|WqW|j dkrt d d |d St rt |St t |SdS) z^Convert list/tuple of list/tuples to bytecode _names_ contains a list of name objects rrNz+error at item %d [%s, %s], opcode not validc3s|]}r|VqdS)Nr).0j)operandsrr psz list2bytecode..z'operand %s [%s, %s], not found in namesg@cSs |t|S)N)chr)abrrr}szlist2bytecode..r(ru) enumerater: TypeErroropmapappendprintr8indexr r5r rbytes bytearray) lr r=ZconstsZbcrFZopcodesr:opcodegenrkr)rr list2bytecodeas4        r)NNNNNr)!rrztypesZxdis.version_inforZxdis.cross_disrrrrZxdis.instructionrZ xdis.utilrr ior functoolsr MethodType FunctionTypeCodeTypetype _have_coderrr#r'rKrLobjectrMrrrrrs(     q