o i< @sddlmZddlZddlmZddlZddlmZddlZ ddl m Z ddl m Z ejdkrGdd d eeZGd d d eZGd d d eZGdddeZGdddZGdddeZGdddeZGdddeZejZdZdededeefddZdedeeefd eeeefdefd!d"Zej d#Z!ej d$Z"ej d%Z#ej d&Z$ej d'Z%d(ed)edeed*ffd+d,Z&ded-e d.e'd/e'deee ff d0d1Z(dSdS)2)ABCN)Enum)CodeType)HookType) CoverageLines) c@s*eZdZdZdZededdfddZdS) JumpDirectionopcodereturncCsdtj|vr |jS|jS)NBACKWARD)disopnamerFORWARD)clsr rc/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/coverage/instrumentation_py3_9.py from_opcodeszJumpDirection.from_opcodeN)__name__ __module__ __qualname__rr classmethodintrrrrrr s r c@s"eZdZdededdfddZdS)Jumpstartargr NcCs||_|||_dSN)rrselfrrrrr__init__s z Jump.__init__)rrrrr!rrrrrsrcs4eZdZeejZdededdffdd ZZ S)AJumprrr Ncst|||j|_dSr)superr!rendr __class__rrr!#s zAJump.__init__) rrrsetrhasjabs __opcodes__rr! __classcell__rrr%rr" s "r"cs8eZdZeejZdedededdffdd Z Z S)RJumprr directionr Ncs.t||||_||j|jd|_dS)N)r#r!r,rr$)r rrr,r%rrr!*szRJump.__init__) rrrr'rhasjrelr)rr r!r*rrr%rr+'s &r+c@s*eZdZdZdedededdfddZdS) Instructionoffsetr rtargetsr1r rr NcCs||_||_||_g|_dSrr0)r r1r rrrrr!2s zInstruction.__init__)rrr __slots__rr!rrrrr//sr/c@s4eZdZdededdfddZedefddZdS) Branchrr$r NcCs||_||_dSrrr$)r rr$rrrr!9s zBranch.__init__cCstr)NotImplementedErrorr rrrr=sz Branch.arg)rrrr/r!propertyrrrrrrr48sr4c@eZdZedefddZdS)RBranchr cCst|jj|jjdd?S)Nr-r )absr$r1rr7rrrrBsz RBranch.argNrrrr8rrrrrrr:Ar:c@r9)ABranchr cCs |jjd?S)Nr )r$r1r7rrrrGs z ABranch.argNr<rrrrr>Fr=r>r r rr cCsFtd||d@g}|dL}|r!|dttt|d@|dL}|s|S)Nr r)r/insert NO_OFFSET EXTENDED_ARG)r r instructionsrrrinstr_with_argNsrEcodetrap_mapext_arg_offsetscCsr|jdkr|jSt}|j}tt|}t|d\}}d}|dd>} } tdt|dD]} || } || d} | | 7} || 7}| | 7} | dkrJq-| dkrY|d|| q-|dur|dur| |kr| |d>7} | |d>7} t|d\}}|dur|dur| |kse| dkr| dkr|d|d| d8} | dks|| || ||dd>} | | 7} q-t |S)N)NNrr r-r?) co_lnotab bytearrayitersortednextrangelenappendgetbytes)rFrGrHnew_datadataext_arg_offset_iterext_arg_offset ext_arg_sizecurrent_orig_offsetcurrent_new_offsetaccumulated_new_offsetiorig_offset_delta line_deltarrrupdate_location_dataVsJ           r_ LOAD_CONST CALL_FUNCTIONPOP_TOP IMPORT_NAME IMPORT_FROM trap_index arg_index.cCs2gtt|tt|tttdtttdRS)Nr r)rEr`r/rBCALLrb)rerfrrr trap_calls  rhhookpathpackagec6Csz||}}g}t|j}t|}||t} i} i} i} i} tt|}d}d}d}d}d}ztt |j }g} t |\}}||vr||}t |t|}t|| |<| |d}|jdkrst|t|jdkrs|df}||||f|d| |<| |t |\}}t|d>}|t||||tur||q<|}|}tjg||ddd}||tkr|j|}|j|}|dkrd |d d| dn|}|d d|d d||fff|d <|tkr!|d |j|}|d d|d d|d d dtt|d d d|gff|d <|tjvr/t||| |<n|tjvr@t||t|| |<|turK||n|q=t yYYnwd d | !D} t |D]\}!}"|!d>}#|"j"| vrz|#| |"j"<|#|"_"qgg}$| !D];}%| |%j#}&| |%j$}'| %|%j$||'d?}(t&|%trt'||&d?|(nt(||&d?|(})|(j)|)|$|)qd}*g}+|*rjd}*|$D]})|)j#},|)j*d>}-|-d@|,_*|-dL}-d}.|,j"d?}!|-rA|!r||!dj+tur|!d8}!|-d@||!_*n8t|!d>t|-d@}/|,|!|/|.d7}.|,j)r:|,j)D]}0|0j$|,ur)t-d|/|0_$q|/j) |,j)|,j)|-dL}-|-s|.re|+|/|.ft ||!dd|!dD] \}1}"|1d>|"_"qXd}*q|*st.}2|D]}"|2|"j+|2|"j*qot |jD]\}}3t&|3t/rt0|3|||\||<}4| 1|4qdd|+D}5|j2t3|2t||j4dt5|| |5d| fS)NrTzr )bigF)signed.r r-cSs"h|] }|j|jfD]}|q qSrr5).0j_rrr s"z'instrument_all_lines..r?r@zInvalid targetcSsg|] \}}|j|fqSr)r1)rpinstrsrrr nsz(instrument_all_lines..)co_code co_consts co_stacksizerJ)6listryrPrQrdictrfindlinestartsrL enumeraterxrNrhextendco_nameaddr/rCr from_bytesclearrcco_namesjoinsplitrdtupler"r)r+r r StopIterationvaluesr1rr$rR isinstancer:r>r2rr rA ValueErrorrKrinstrument_all_linesupdatereplacerSrzr_)6rFrirjrk trap_functrap_argrD new_constsre seen_lines offset_mapjumpstrapsline_map line_starts current_arg previous_argprevious_previous_argcurrent_import_namecurrent_import_package code_iterextoriginal_offsetr linetrap_instructions package_deprrrr1 import_depthimport_from_name jump_targetsindexrt new_offsetbranchesjump new_startnew_end target_instrbranchprocess_branchesexts jump_instrnew_argc ext_instrtarget instr_indexnew_code nested_code nested_linesrHrrrrs              (     ,    K               ".    r))abcrrenumrsystypesrtypingt#ddtrace.internal.bytecode_injectionr/ddtrace.internal.test_visibility.coverage_linesr version_inforr rr"r+r/r4r:r>rCrBr{rEr|rrSr_opmapr`rgrbrcrdrhstrrrrrrs<        . A    *