3 /*^L@s2dZddlmZddlmZddlmZddlZddlmZm Z ddl m Z ddl Z ddl m Z mZmZmZmZmZed3Zed d!eDZded"<erejZd#Zd$d%Zned#ZGd&d'd'eZGd(d)d)eZd*d+Zd4d-d.Zed/kr.ddl Z ddlZe j!j"Z#eej$e d0Z%e%j&e#id1d2\Z'Z(dS)5zl scanner/ingestion module. From here we call various version-specific scanners, e.g. for Python 2.7 or 3.4. )print_function)array) namedtupleN)PYTHON3IS_PYPY)Token)Bytecodecanonic_python_versioncode2numinstruction_sizeextended_arg_val next_offset?皙??ffffff??皙?@皙@ffffff@333333@@@皙@@@皙 @ffffff @333333 @ @ @皙 @ffffff@333333@ccs|]}tt||fVqdS)N)r str).0vr(O/home/psgendb/BIRCHDEV/python/lib/python3.6/site-packages/uncompyle6/scanner.py Asr*z3.5.2icCs|S)Nr()lr(r(r)longMsr,c@seZdZdZdddZdS)Codez Class for representing code-objects. This is similar to the original code object, but additionally the diassembled code is stored in the attribute '_tokens'. NcCsFx,t|D] }|jdr t||t||q W|j||\|_|_dS)NZco_)dir startswithsetattrgetattringestZ_tokensZ _customize)selfcoscanner classnameir(r(r)__init__]s z Code.__init__)N)__name__ __module__ __qualname____doc__r8r(r(r(r)r-Usr-c@seZdZd3ddZddZddZd d Zd d Zd dZddZ d4ddZ ddZ ddZ ddZ d5ddZd6ddZd7dd Zd8d!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2ZdS)9ScannerNFcCs|||_||_||_|tkrZ|r0dt|d}ndt|d}td|td|n td||jj|_|j dS)Nz opcode_%spypy z opcode_%szfrom xdis.opcodes import %sz self.opc = %sz'%s is not a Python version I know about) versionshow_asmis_pypyPYTHON_VERSIONSintexec TypeErroropcopnameresetTokenClass)r3r?r@rAv_strr(r(r)r8es   zScanner.__init__cCsltd|j|_t||j}|j|jt||_|j ||_ i|_ x"t |jD]\}}||j |j <qPW|S)z Create a list of instructions (a structured object rather than an array of bytes) and store that in self.insts B)rco_codecoderrF build_prev_opremove_extended_argslistinstsbuild_lines_datalinesoffset2inst_index enumerateoffset)r3r4bytecoder7instr(r(r)build_instructionsys  zScanner.build_instructionsc Cs|jdkrt|jj|}n ddgg}t||_g}tdddg}|d\}}d}xB|ddD]2\}} x$||kr|j||||d7}qjW| }q`Wt|j } x$|| kr|j||| |d7}qW|S)z< Generate various line-related helper data. gffffff?r LineTuplel_nonextN) r?rOrFfindlinestartsdict linestartsrappendlenrL) r3Zcode_objr_rRrZ_Z prev_line_norUZ start_offsetZline_nocodelenr(r(r)rQs$         zScanner.build_lines_datacCsd|j}t|}dg|_|_xB|jd|D]2}||}x$tt||jD]}|jj|qHWq*WdS)zv Compose 'list-map' which allows to jump to previous op, given offset of current op as index. rN) rLraprevprev_opop_rangeranger rFr`)r3rLrcrUoprbr(r(r)rMszScanner.build_prev_opcCs2|j|j}|dkrdS|dkr$dS||j|kS)z Return True if the code at offset is some sort of jump forward. That is, it is ether "JUMP_FORWARD" or an absolute jump that goes forward. JUMP_FORWARDT JUMP_ABSOLUTEF)get_instrG get_target)r3rUrGr(r(r)is_jump_forwards  zScanner.is_jump_forwardcCs|j|j|djS)NrY)rPrSrU)r3rUr(r(r) prev_offsetszScanner.prev_offsetcCsD||jkr4|t|jj|j8}|j||jjks4t|j|j|S)N)rSr rF EXTENDED_ARGrLAssertionErrorrP)r3rUr(r(r)rks zScanner.get_instrcCs>|j|}|j|jj|jjBkr(|j}nt|j|j|j}|S)z| Get next instruction offset for op located at given . NOTE: extended_arg is no longer used )rkopcoderFZJREL_OPSZJABS_OPSargvalr rU)r3rU extended_argrWtargetr(r(r)rls  zScanner.get_targetcCs$|j|d|j|dd}|S)NrY)rL)r3posargr(r(r) get_arguments zScanner.get_argumentcCstj||j|S)N)xdisr rF)r3rhrUr(r(r)r szScanner.next_offsetcCspxj|jdt|jD]T}|j|}||jkrR|j||}td||j||fqtd||j|fqWdS)Nrz%i %s %iz%i %s )rfrarLZJUMP_OPSrlprintrG)r3r7rhdestr(r(r)print_bytecodes    zScanner.print_bytecodeTc Cs|j}|dkr|t|kstt|ts.|g}d}t|}xf|j||D]V} || } | |krH|dkrh| S|j| } | |kr~| S|sHt|| } | |krH| }| }qHW|S)a Find the first in the block from start to end. is any python bytecode instruction or a list of opcodes If is an opcode with a target (like a jump), a target destination can be specified which must match precisely if exact is True, or if exact is False, the instruction which has a target closest to will be returned. Return index to it or None if not found. rN)rLrarp isinstancerOrfrlabs) r3startendinstrrtexactrL result_offsetcurrent_distancerUrhr| new_distancer(r(r) first_instrs(    zScanner.first_instrcCs|j}|dko|t|ksdSt|ts.|g}d}|jdj|jdj}d} x|j||D]} || } | |jjkrt || d| B} t |j| } q\| |kr|dkr| }n>|j | | } | |krd}| }n |st || }||kr|}| }d} q\W|S)a Find the last in the block from start to end. is any python bytecode instruction or a list of opcodes If is an opcode with a target (like a jump), a target destination can be specified which must match precisely if exact is True, or if exact is False, the instruction which has a target closest to will be returned. Return index to it or None if not found. rNrY) rLrar~rOrPrUrfrFror r rlr)r3rrrrtrrLrrrsrUrhrxr|rr(r(r) last_instrs:      zScanner.last_instrc Csy d|kWn|g}YnX|j|}g}x||j|dD]j}|j|kr|dkrb|j|jn:|j|j} |r| |kr|j|jn| |kr|j|j|j|kr>Pq>W|S)aB Find all `instr` in the block from start to end. `instr` is a Python opcode or a list of opcodes If `instr` is an opcode with a target (like a jump), a target destination can be specified which must match precisely. Return a list with indexes to them or [] if none found. N)rSrPrqr`rUrl) r3rrrrtinclude_beyond_targetfirstresultrWtr(r(r) inst_matchesPs&         zScanner.inst_matchesc Cs|j}|dkr|t|ksty d|kWn|g}YnXg}d}x|j||D]} || } | |jjkrt|| d|B} t|j| }qT| |kr|dkr|j| n6|j | |} |r| |kr|j| n| |kr|j| d}qTW|S)aD Find all `instr` in the block from start to end. `instr` is any Python opcode or a list of opcodes If `instr` is an opcode with a target (like a jump), a target destination can be specified which must match precisely. Return a list with indexes to them or [] if none found. rNrY) rLrarprfrFror r r`rl) r3rrrrtrrLrrsrUrhrxrr(r(r) all_instrzs2          zScanner.all_instrcCs|jj|j|S)N)rFrGrL)r3rUr(r(r)opname_for_offsetszScanner.opname_for_offsetcCs |jj|S)N)rFrG)r3rhr(r(r)op_nameszScanner.op_nameccs.x(||kr(|V|t|j||j7}qWdS)zS Iterate through positions of opcodes, skipping arguments. N)r rLrF)r3rrr(r(r)rfs zScanner.op_rangecCsg}d}t|}xt|D]\}}|jdkrb|d|krb||djdkrbd}|j}|j}|j} q|r|j||| d} | }||kr|j||j} ||dj} |j| } x,|j| | kr| |kr| |j| <| d7} qWd}|j|qW|S)zGo through instructions removing extended ARG. get_instruction_bytes previously adjusted the operand values to account for theseFrorY MAKE_FUNCTIONT) starts_lineis_jump_targetrU) rarTrGrrrU_replacerer`)r3Z instructionsZnew_instructionsZlast_was_extargnr7rWrrrUZnew_instZnew_prevjZold_prevr(r(r)rNs8     zScanner.remove_extended_argscCsfg}x\|D]T}|j|j|j|djkrT|j|j|j|j|jj|jjfkrTq |j|q W|S)z^ Go through passed offsets, filtering ifs located somewhere mid-line. ) rRr[rLrdr\rFZPJITZPJIFr`)r3Zifsfilteredr7r(r(r)remove_mid_line_ifss  zScanner.remove_mid_line_ifscCs |jtS)N) setTokenClassr)r3r(r(r)rHszScanner.resetTokenClasscCs,|d|ko|dkns(|d}|S)z/Restrict target to parent structure boundaries.rrr()r3rtparentr(r(r)restrict_to_parents zScanner.restrict_to_parentcCs ||_|jS)N)r)r3Z tokenClassr(r(r)rszScanner.setTokenClass)NF)r)NT)NT)NF)NF)r9r:r;r8rXrQrMrmrnrkrlryr r}rrrrrrrfrNrrHrrr(r(r(r)r=ds. %   " 6 * . )r=cCs|d@|d?d@|d?d@fS)Nir()Zargcr(r(r)parse_fn_countssrFc Cst|trF|tkrtd|t|}|tkr>td||ft|}|tkrdt|d}y.ddl}|r~|jd|}n|jd|}WnBt k r|rt d|t t nt d |t t YnX|rt d |t t }nt d |t t }n td ||S) Nz!Unknown Python version in xdis %sz*Unsupported Python version %s (canonic %s)z%sr>rzuncompyle6.scanners.pypy%szuncompyle6.scanners.scanner%sz)import uncompyle6.scanners.pypy%s as scanz,import uncompyle6.scanners.scanner%s as scanz%scan.ScannerPyPy%s(show_asm=show_asm)z!scan.Scanner%s(show_asm=show_asm)zUnsupported Python version %s)r~r%r RuntimeErrorCANONIC2VERSIONrBrC importlib import_module ImportErrorrDlocalsglobalseval)r?rAr@Zcanonic_versionrIrscanr5r(r(r) get_scanner sD      r__main__Tafter)r@)rrrrrrrrrrrrrrrrrrr r!r"r#r$)FN))r< __future__rr collectionsrsys uncompyle6rrZuncompyle6.scanners.tokrrzrr r r r r frozensetrBr^rinternZL65536r,objectr-r=rrr9inspect currentframef_coder4ZPYTHON_VERSIONr5r2tokensZ customizer(r(r(r)sf      $ 4