"eDdZddlZddlmZddlmZddlZddlZddlm Z Gdde Z e Z e Z e ZdZGd d e ZGd d e ZGd de ZiZdZGddeZGddeZeGddeZeGddeZeGddeZeGddeZeGddeZeGddeZeGd d!eZeGd"d#eZeGd$d%eZeGd&d'eZ eGd(d)eZ!e!Z"d/d*Z#d+Z$Gd,d-e Z%d.Z&y)0z Decompiler that can be used with the debugger (where statements correctly represent the line numbers). Note: this is a work in progress / proof of concept / not ready to be used. N)iter_instructions) pydev_log)StringIOceZdZdZdZdZy)_Stackcg|_yN) _contentsselfs F/usr/lib/python3/dist-packages/_pydevd_bundle/pydevd_code_to_source.py__init__z_Stack.__init__s c:|jj|yr )r append)r objs r pushz _Stack.pushs c"rc8|jjdS)N)r popr s r rz _Stack.pops~~!!"%%rN)__name__ __module__ __qualname__rrrrr rrs#&rrFc4eZdZdedddfdZdZdZdZeZy)_TokenNrFc^||_|tur||_nO|Btj|j rd|_n&t |j |_n td||_||_ ||_ t|_ t|_ |r|j|yy)z :param i_line: :param instruction: :param tok: :param priority: :param after: :param end_of_line: Marker to signal only after all the other tokens have been written. Nz,Either the tok or the instruction is needed.)i_line _SENTINELtokinspectiscodeargvalstrAssertionError instructionpriority end_of_lineset _after_tokens_after_handler_tokens mark_after)r rr'r!r(afterr)s r rz_Token.__init__'s i DH&>>+"4"45!DH";#5#56DH$%STT&  & U%(U"  OOE " rct|tr|jj|yt|tr|j j|yt d|)Nz Unhandled: ) isinstancerr+add _BaseHandlerr,r&)r vs r r-z_Token.mark_afterDsN a    " "1 % < (  & & * *1 -!A!78 8rc|jj}|jD]}|j|j|Sr )r+copyr,updatetokens)r rethandlers r get_after_tokensz_Token.get_after_tokensMs@  %%'11 'G JJw~~ & ' rcDd|jd|jdS)NzToken(z , after: ))r!r:r s r __repr__z_Token.__repr__Ss)-43H3H3JKKr) rrrr rr-r:r=__str__rrr rr%s++/YRVdi#:9 LGrrc*eZdZdZdZdZdZdZy)_Writerc0i|_t|_yr )line_to_contentsr* all_tokensr s r rz_Writer.__init__[s "%rcb|jj|}|gx}|j|<|Sr )rBget)r linelsts r get_linez_Writer.get_line_s7##''- ;02 2C$''- rcL|j|jtyr )rHr INDENT_MARKERr rFs r indentz_Writer.indente d""=1rcL|j|jtyr )rHr DEDENT_MARKERrKs r dedentz_Writer.dedenthrMrc||jvry|jj|t|tsJ|j |}|j |yr )rCr1r0rrHr)r rFtokenrGs r writez _Writer.writeksM DOO #  E"%(((mmD! 5rN)rrrrrHrLrPrSrrr r@r@Ys  22rr@c(eZdZdZdZdZdZeZy)r2cx||_||_||_||_||_g|_|j yr )rr'stackwriter disassemblerr7_handle)r rr'rVrWrXs r rz_BaseHandler.__init__vs7 &  (  rcr|jD](}|jj|j|*yr )r7rWrSr)r rRs r _write_tokensz_BaseHandler._write_tokenss-[[ 3E KK  ellE 2 3rct|r )NotImplementedErrorr s r rYz_BaseHandler._handles !$''rcr |jd|jS#tj|cYSxYw)Nz line:)r'robjectr=)r argskwargss r r=z_BaseHandler.__repr__s2 )#'#3#3T[[A A )??4( (s6N)rrrrr[rYr=r>rrr r2r2ts3() Grr2c,|t|j<|Sr )_op_name_to_handleropname)clss r _registerrfs&) # JrceZdZdZy)_BasePushHandlerc:|jj|yr )rVrr s r rYz_BasePushHandler._handles rNrrrrYrrr rhrhsrrhceZdZdZy)_BaseLoadHandlercztj|t|j|jg|_yr )rhrYrrr'r7r s r rYz_BaseLoadHandler._handles+  &dkk4+;+;<= rNrjrrr rlrls>rrlceZdZdZy)_LoadBuildClassLOAD_BUILD_CLASSNrrrrdrrr roros FrroceZdZdZy) _LoadConst LOAD_CONSTNrqrrr rsrss FrrsceZdZdZy) _LoadName LOAD_NAMENrqrrr rvrv FrrvceZdZdZy) _LoadGlobal LOAD_GLOBALNrqrrr rzrzs FrrzceZdZdZy) _LoadFast LOAD_FASTNrqrrr r}r}rxrr}ceZdZdZdZdZdZy)_GetIterz% Implements TOS = iter(TOS). GET_ITERNc|jj|_|jj |jj|jj |yr )rVr iter_targetr7extendrr s r rYz_GetIter._handlesC::>>+ 4++223 r)rrr__doc__rdrrYrrr rrsFKrrc$eZdZdZdZdZdZdZy)_ForIterz TOS is an iterator. Call its __next__() method. If this yields a new value, push it on the stack (leaving the iterator below it). If the iterator indicates it is exhausted TOS is popped, and the byte code counter is incremented by delta. FOR_ITERNcx|jj|_|jj|yr )rVriter_inrr s r rYz_ForIter._handles$zz~~'  rct|jdd}|jj||}t|j|j|}|jj||}t|jdd|}|jj||}|j}|j rs|j jD]+}|j |t||j}|}-|jj|j jt|jdd|}|jj||}|jy)Nzfor r.z in :) rrr7rr'rr-maxrr[) r store_name for_tokenprevt_namein_tokenmax_linet colon_tokens r store_in_namez_ForIter.store_in_names*4;;f5  9% )):+A+AN 6"*++T6F 8$$$ <<\\((  T"x2  KK  t||22 3T[[$4@  ;' r)rrrrrdrrYrrrr rrs FGrrceZdZdZdZdZy) _StoreNamez Implements name = TOS. namei is the index of name in the attribute co_names of the code object. The compiler tries to use STORE_FAST or STORE_GLOBAL if possible. STORE_NAMEc|jj}t|tr|j |yt|t r |j r|j}|jD]}t||j}t||j}t|dd|}|jj||jj||jD]}|j||jj|j|jyy)N=r)rVrr0rr _MakeFunction is_lambdarr7minrr'rr-rr[)r r3rFrrt_equals r rYz_StoreName._handles JJNN  a " OOD !a/1;;{{/AtQXX.D/ d&6&67 tS? ""6* ""7+*ALL)* ""188,""$4?rNrrrrrdrYrrr rrs F%rrceZdZdZdZdZy) _ReturnValuez9 Returns with TOS to the caller of the function. RETURN_VALUEcN|jj}t|jddd}|jj ||jD]}|j ||jj|j|jy)Nzreturn T)r)) rVrrrr7rr-rr[)r r3 return_tokenrRs r rYz_ReturnValue._handles{ JJNN dkk4M  <(XX +E   \ * + 188$ rNrrrr rrsFrrceZdZdZdZdZy) _CallFunctionaM CALL_FUNCTION(argc) Calls a callable object with positional arguments. argc indicates the number of positional arguments. The top of the stack contains positional arguments, with the right-most argument on top. Below the arguments is a callable object to call. CALL_FUNCTION pops all arguments and the callable object off the stack, calls the callable object with those arguments, and pushes the return value returned by the callable object. Changed in version 3.6: This opcode is used only for calls with positional arguments. CALL_FUNCTIONcg}t|jjdzD]-}|jj }|j |/t |}t|}|j}|jD]}|jj |t|jdd|}|jj ||} t|D]\} }|jD]U}|j||j| t||j}|jj |W|} | dkDsrt| jdd| } |jj | | } t|dd| } |jj | |j|jj|y)N(rr,r<)ranger'r$rVrrreversednextrr7r enumerater-rr[r) r r`_iargitnamerrtok_open_parensri comma_tokentok_close_parenss r rYz_CallFunction._handle<s((//!34 B**.."C KK  d^Bx;; "A KK  q ! "!dCtD ?+m #FAsZZ & T" T"x2 ""1%  & D1u$T[[$4H  "";/" #"(D#TB +,  rNrrrr rr*s F!rrceZdZdZdZdZdZy)_MakeFunctionPy3aA Pushes a new function object on the stack. From bottom to top, the consumed stack must consist of values if the argument carries a specified flag value 0x01 a tuple of default values for positional-only and positional-or-keyword parameters in positional order 0x02 a dictionary of keyword-only parameters' default values 0x04 an annotation dictionary 0x08 a tuple containing cells for free variables, making a closure the code associated with the function (at TOS1) the qualified name of the function (at TOS) MAKE_FUNCTIONFcv|j}|j|_|j|_d}|jj dzr|j}d|jj Dcgc]}|jc}vx}|_|s2t|jdd}|j j||jj D]1}|j j||r!|j3}t|jdd|}|j j||}|jjj } |r^tgt| jt|jj z zt!|jj z} ntgt| jz} t#| jD]\} } | dkDr6t|jdd|} |j j| | }t|jd| |}|j j|| | }|tust|jdd|}|j j||}t|jdt%||}|j j||}t|jdd |}|j j||j'|j)||j*j-|jdz|j*j/t1|j2j5| ycc}w) Nrzzdef rrrz, rz):)rVrqualified_namecoder'r$r7r!rrrrr-r len co_varnameslistrr%r[rrWrLrPrrX merge_code)r rV default_nodexr def_tokenrRropen_parens_tokenrdefaultsrrr arg_tokendefaulteq_token default_tokenrs r rYz_MakeFunctionPy3._handlevs  #iikIIK     " "T ) 99;L%/4CVCVC]C]3^aAEE3^%^^ DNt{{D&9I KK  y )((// ,E KK  u %  + ,"4;;cF ,- yy$$++ " s4+;+;'z$_print_after_info..s"KQ155"Ks z%s (NO REQUISITES))sysstdoutr:reprr!joinrS) line_contentsstreamrR after_tokensss r _print_after_infors ~F--/ UYYv{{"K%2H2H2J"KKKcQTA LLO LL2T%))_D EFrc0g}t}g}g}t|D]4\}}|js|j||j|6t |D]}||=~|rd}g}t|D]w\}}|j } | D] } | |vs| |vs 'd}|j ||j ||j|j|j|yt |D]}||=|s{|r|j||dd=|D]"}||vs|j|j$t} t|| tjd| jn|rdj|S)NFTz(Error. After markers are not correct: %sr)r*rr)rrr:r1r!rrrrcriticalgetvaluer) rprevious_line_tokensrGhandledadd_to_end_of_linedelete_indexesrrRaddedrr.rs r _compose_line_contentsrs CeGNm,%5     % %e ,  ! !! $%n % !  !-0 )HAu 113L% )'E9M,M )$((/ E" 599%%%a( ).) !Aa  !!$$%78&q)' *'JJuyy) *ZF mV 4   JFOOL] ^ C D 773<rc,eZdZddZdZdZdZdZy)_PyCodeToSourceNc|i}||_||_tt||_t |_t|_yr ) memocorr instructionsrrVr@rW)r rrs r rz_PyCodeToSource.__init__s? <D  !22!67X i rc|jjd}tj|j}|3||||j |j |}tr t|yytr td|yy)Nr UNHANDLED) rrrcrErdrVrWDEBUGprint)r rr' handler_classrs r _process_nextz_PyCodeToSource._process_nextsq''++A. +// 0B0BC  $fk4::t{{DQAak;/rc|j}ttj|}d}|j}|r:|d}|j |j }|||}|j||r:|jjS)Nr) rdictdisfindlinestartsrrEoffsetrrWrB)r rop_offset_to_linecurr_line_indexrr'new_line_indexs r build_line_to_contentsz&_PyCodeToSource.build_line_to_contentss WW !3!3B!78(( &q/K.22;3E3EFN)!-&4O    /{{+++rcNtr tdt||jj }g}t |j D]@\}}|j||jj|j|Btr td|S)Nzmerge code ----zend merge code ----) rrrrrsorteditemsrrWrHr)r rrBlinesrFcontentss r rz_PyCodeToSource.merge_codes  # $+4;RRT$%5%;%;%=> 8ND( LL  KK  & - -h 7 8  ' ( rcd}|j}t}d}d}t}t|j D]\}}||dz kr9|r|j |dzdn|j d|dz }||dz kr9g} d} |D]S} | t urtr td||dz }#| turtr td || dz } C| j| U|t| |z} |r|j |d | dn|j d | z| r |dd | z }|}|jS) NFrrrz. rz found indentz z found dedentz. z%s ) rrr*rrrSrJrrrOrrr) r show_linesrBr last_linerLrrrr dedents_foundpartrs r disassemblez_PyCodeToSource.disassemble.sj 668 "u &'7'='='? @  FHfqj(LLY]!<=LL'Q fqj(MM +=(nf5f$F=(nf5!Q&M$$T* +/ ?STTA FA67 Wq[) 51}#4!56I? B  rr )rrrrrrrrrrr rrs  0," (!rrcZt|j}tr t||S)z Converts a code object to source code to provide a suitable representation for the compiler when the actual source code is not found. This is a work in progress / proof of concept / not ready to be used. )rrrr)rr8s r code_obj_to_sourcerYs& "  ) ) +C  c Jrr )'rr+_pydevd_bundle.pydevd_collect_bytecode_infor _pydev_bundlerrr"iorr_rrJrOr rrr@r2rcrfrhrlrorsrvrzr}rrrrrrrrrrrrrr r s I#  &V &  H  1V1hf666 | >'>  &    !     "      |    '|' 'T %% %@ < $ 2L2 2j OD|OD ODd!  F0f_!f_!D r