o iaB @sjddlmZddlmZddlZddlmZddlZddlm Z ddl Z ddl m Z ddlmZejdkr:ejd ksd?Z5e6ej)d@gZ7d%e dAe dBe8dCe8de$e eff dDdEZ9dS)G)ABC) dataclassN)Enum)CodeType)HookType) CoverageLines) )r c@s*eZdZdZdZededdfddZdS) JumpDirectionopcodereturncCsdtj|vr |jS|jS)NBACKWARD)disopnamerFORWARD)clsrrd/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/coverage/instrumentation_py3_11.py from_opcodeszJumpDirection.from_opcodeN)__name__ __module__ __qualname__rr classmethodintrrrrrr s r c@s"eZdZdededdfddZdS)JumpstartargrNcCs||_d|_||_dSN)rendr)selfrrrrr__init__s z Jump.__init__)rrrrr#rrrrrsrcs8eZdZeejZdedededdffdd Z Z S)RJumprr directionrNcs2t||||_||jd>|jd|_dS)Nr )superr#r%rr!)r"rrr% __class__rrr#%szRJump.__init__) rrrsetrhasjrel __opcodes__rr r# __classcell__rrr(rr$"s &r$c@s*eZdZdZdedededdfddZdS) Instructionoffsetrrtargetsr0rrrNcCs||_||_||_g|_dSr r/)r"r0rrrrrr#/s zInstruction.__init__)rrr __slots__rr#rrrrr.,sr.c@s4eZdZdededdfddZedefddZdS) Branchrr!rNcCs||_||_dSr rr!)r"rr!rrrr#7s zBranch.__init__cCst|jj|jjdd?S)Nr&r )absr!r0r)r"rrrr;sz Branch.arg)rrrr.r#propertyrrrrrrr36sr3r rrrcCsFtt||d@g}|dL}|r!|dttt|d@|dL}|s|S)Nr)r. NO_OFFSETinsert EXTENDED_ARG)rr instructionsrrrinstr_with_argDsr=iteratorcCs@t|}|d@}|d@r|dK}t|}||d@O}|d@s |SN?@)next)r>bvalrrr from_varintMs rFFvalueset_begin_markercCslt}|dkr td|r!|d|d@|rdndB|dL}|s |p'tdg}|r2|ddO<t|S)NrzInvalid value for varintr@rArB) bytearray ValueErrorr:bytes)rGrHtemprrr to_varintWsrNstreamcCsZt}t|}|||d@}|d@r)t|}|||d>|d@B}|d@st|Sr?)rJrCappendrL)rOarDrGrrrconsume_varintfs  rRcodetrap_mapext_arg_offsetsc Cs"t}|j}t|}tt|}t|d\}}d} } zt} t|} | | | d@dd>} | d?d@}|dkrQ| t|tdD] }| t |qFn:|d kr]| t|n.d |krgd krxnntd D] }| t|qmnd|krd krnn| t|| |vr|| }t |d\}}t|D]}|dq|r|d|dB| |d>7} | | 7} | | 7} |dur| |krd| }| dt |t t|7<|t t|kr| dt t||B| |d>7} t|d\}}|| Wn tyYt|Swq)N)NNrTr r r r& r8r7)rJ co_linetableitersortedrCrPextendconsume_signed_varintrangerRdivmodmintcastr StopIterationrL)rSrTrUnew_datadata data_iterext_arg_offset_iterext_arg_offset ext_arg_sizeoriginal_offsetr0chunkrD offset_deltaloc_code_ trap_sizenrroomrrrupdate_location_dataysb         rwc@sLeZdZUejeefed<ejeefed<ejeefed<eed<dS)ExceptionTableEntryrr!target depth_lastiN)rrrreUnionrr.__annotations__rrrrrxs  rxccsnt|j}z& t|d>}t|d>}||d}t|d>}t|}t||||Vqty6YdSw)NTr r&)r^co_exceptiontablerFrxrg)rSr>rlengthr!rydlrrrparse_exception_tables      r exc_tablecCs|t}|D]4}|jj|jjd}|t|jjd?d|t|d?|t|jjd?|t|jqt|S)Nr&r T) rJr!r0rr`rNryrzrL)rtableentrysizerrrcompile_exception_tablesr PUSH_NULL LOAD_CONSTPRECALLCACHECALLPOP_TOPRESUME RETURN_VALUE IMPORT_NAME IMPORT_FROM trap_index arg_index.cCstttdgtt|tt|tttdtttdtttdtttdtttdtttdtttdtttdRS)Nrr ) r.r9rr=rrrrr)rrrrr trap_calls,         r END_ASYNC_FORhookpathpackagec: Cs||}}g}t|j}t|}||t} tt|} dd| D} i} i} i}i}tt|}t }t dt|j dD] }|j |t krL|}nq?|j dkra|ddikra|j tkraddi}d}d}d}d}d}ztt|j }g} t|\}}|| vrt|d>| |<||vr||kr||}|j |tvrt|t|}t|||<||d}|j dkrt|t|jdkr|df}||||f|d||<| |t|\}} t|d>}!|t||| |tur|| qv|}|}tjg|| d d d }||tkrA|j|}"|j|}|"dkr-d |d d|" dn|}|d d|d d||fff|d <|tkrt|d |j|}#|d d|d d|d ddt t|d dd|#gff|d <|t!j"vrt!||t#$|| |!<qwt%yYnwdd| &D}$t|D]\}%}&|%d>}'|&j'|$vs|&j'| vr|'| |&j'<|'|&_'qg}(| &D],})| |)j(}*| |)j)}+|*|)j)||+d?},t+||*d?|,}-|,j,|-|(|-q| D]!}.|| |.j(d?|._(|| |.j)d?|._)|| |.j-d?|._-qd}/g}0|/rd }/|(D]}-|-j(}1|-j.}2|2d@|1_.|2dL}2d}3|1j'd?}%|2r|%rQ||%dj/turQ|%d8}%|2d@||%_.n8t|%d>t|2d@}4|0|%|4|3d7}3|1j,r|1j,D]}5|5j)|1urxt1d|4|5_)ql|4j,|1j,|1j,|2dL}2|2s8|3r|0|4|3ft||%dd|%dD] \}6}&|6d>|&_'qd}/q|/st2}7|D]}&|7|&j/|7|&j.qt|jD]\}}8t3|8t4rt5|8|||\||<}9| 6|9q|j7t8|7t ||j9dt:||dd|0Dt;| d| fS)NcSs&h|]}|j|j|jfD]}|q qSr)rr!ry).0errrrr s&z'instrument_all_lines..rr&zTr )bigF)signed.r cSs"h|] }|j|jfD]}|q qSrr4)rjrrrrrrts"r7r8zInvalid targetcSsg|] \}}|j|fqSr)r0)rinstrsrrr sz(instrument_all_lines..)co_code co_consts co_stacksizer]r})<listrlenrPrrdictrfindlinestartsr9rbrrco_nameEMPTY_BYTECODEr^ enumeraterC SKIP_LINESrr`addr.r;r from_bytesclearrco_namesjoinsplitrtupler$r,r rrgvaluesr0rr!getr3r1ryrrr:rKrJ isinstancerinstrument_all_linesupdatereplacerLrrwr):rSrrr trap_functrap_argr< new_constsr seen_linesrexc_table_offsets offset_mapjumpstrapsline_map line_starts resume_offseti current_arg previous_arg_previous_previous_argcurrent_import_namecurrent_import_package code_iterextrnrlinetrap_instructions package_deprrrr0 import_depthimport_from_name jump_targetsindexr new_offsetbranchesjump new_startnew_end target_instrbranchrprocess_branchesexts jump_instrnew_argc ext_instrry instr_indexnew_code nested_code nested_linesrrrrs$                 *     , J             ".   r)F):abcr dataclassesrrenumrsystypesrtypingre#ddtrace.internal.bytecode_injectionr/ddtrace.internal.test_visibility.coverage_linesr version_inforr rr$r.r3r;r9rr=IteratorrFboolrLrNIterablerRrarrrwrxrropmaprrrrrrrrrrrr frozensetrstrrrrrrsR           .>         *