o ib@sddlmZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl)m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ddl0m2Z2ddl3m4Z4e"e5Z6e7d dure68d!e 9d"e:e j;d"d#d$d%e(d&e(d d%e/d'e/d'e/e(d(d%d#oe/e(d)d%d#e/e(d)d%d#e/e(d*d+d#d,d-Zfd0d1Z?d2e e>e@fd/e>fd3d4ZAd5ee>efd6ee>efd/ee>e>ffd7d8ZBd5ee>efd/ee>effd9d:ZCd;d<ZDd5ee>efd/ee1fd=d>ZEd5ee>efd?ed@efdAdBZFdCee>e>fd/ee>effdDdEZGd5ee>effdFdGZHGdHdIdIZIdS)J)wrapsN)Any)Callable)Mapping)Optional)Union)parse)config) SPAN_KIND) trace_utils)guarantee_single_callable)SpanKind) SpanTypes)http)core)BlockingException)find_exception) is_valid_ip) COMPONENT) get_logger)schematize_url_operation) SpanDirection) _get_config) get_blocked) set_blocked)DDTraceDeprecationWarning)asbool)Span)tracer) deprecateDD_ASGI_TRACE_WEBSOCKETzDD_ASGI_TRACE_WEBSOCKET is deprecated and will be removed in a future version. Use DD_TRACE_WEBSOCKET_MESSAGES_ENABLED instead.asgi)default asgi.requestT#DD_TRACE_WEBSOCKET_MESSAGES_ENABLED)r"modifier,DD_TRACE_WEBSOCKET_MESSAGES_INHERIT_SAMPLING+DD_TRACE_WEBSOCKET_MESSAGES_SEPARATE_TRACESDD_ASGI_OBFUSCATE_404_RESOURCEF) service_namerequest_span_namedistributed_tracingtrace_asgi_websocket_messages(asgi_websocket_messages_inherit_sampling"websocket_messages_separate_tracesobfuscate_404_resourcez asgi.versionzasgi.spec_versionreturncCsdS)Nr2r2r2\/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/asgi/middleware.py get_versionFsr4 str_or_bytescCst|tr |jddS|S)Nignoreerrors) isinstancebytesdecode)r5r2r2r3 bytes_to_strJr<scopeintegration_configcCsZi}|d}|r||tj<|d}|rd|vr|d|t<|r+d|vr+|d|t<|S)z? Extract HTTP and ASGI version information from scope. http_versionr!version spec_version)getrVERSION ASGI_VERSIONASGI_SPEC_VERSION)r>r?tagsr@ scope_asgir2r2r3_extract_versions_from_scopeNs       rIcCs$|d}|rtdd|DSiS)z Extract and decode headers from ASGI scope. ASGI headers are stored as byte strings; this method decodes them to UTF-8 strings for easier processing. headerscss$|] \}}t|t|fVqdS)N)r<).0kvr2r2r3 ms"z#_extract_headers..)rCdict)r>rJr2r2r3_extract_headerscs rPcCs|tjddS)z&Default handler for exception for span500N) _set_tag_strr STATUS_CODE)excspanr2r2r3_default_handle_exception_spanqsrVcCs|diddgdS)Ndatadog request_spansr)rC)r>r2r2r3span_from_scopevr=rYreceivesendcs0|ddgdIdH|dddIdHdS)Nhttp.response.start)typestatusrJhttp.response.body)r^bodyr2)r>rZr[r2r2r3_blocked_asgi_appzsrcresponse_headerscCsfi}z|ddjddd}t|dkr|\}}|||<W|SW|Sty2tjddd Y|Sw) Nz set-cookier1=)maxsplitz"failed to extract response cookiesTexc_info)rCsplitlen Exceptionlogdebug)rdcookiesresult cookie_key cookie_valuer2r2r3_parse_response_cookiess   rtcCs6|did}|r||ddddSdS)NrWcurrent_receive_span)rCfinishpop)r>rur2r2r3_cleanup_previous_receives rxc @s,eZdZdZdddddZdejedfddZde e e fd e d e fd d Z de e e fd e e e fdefddZde e e fd e e e fdefddZde e e fd e e e fdede dee e e ff ddZdejde e e fd e e e fdedef ddZde e e fd e e e fdefddZdS)TraceMiddlewarea ASGI application middleware that traces HTTP and websocket requests. Provides distributed tracing for ASGI applications, including support for websocket connections with configurable message-level tracing. When websocket instrumentation is enabled, the middleware creates spans for: - websocket handshake (HTTP upgrade) - Individual message receive/send operations - Connection close events Pi)rhttpswswssNcCs8t||_|durtddtdd||_||_||_dS)Nz"The tracer parameter is deprecatedz'The global tracer will be used instead.z5.0.0)messagecategoryremoval_version)r apprrr?handle_exception_span span_modifier)selfrrr?rrr2r2r3__init__s  zTraceMiddleware.__init__r>rZr[c#sddkr dnddkrjjrdn IdHSzt}Wnty;tjdddi}Yn wtjt j|d d d g}j d d }ddkrct |t jdd}tjd  d|d||tjtdj|djdljjjrd}tD]!\}} |dkrz| }WntytjdddYnwnq d d}  dd} tt dd}  d d} |rd| || n5| rt| dkr| d}j | d}| d|dur ||kr dt|nd}d| || nd d}|r0t|}r0d|jj s7d}d}t!d|fj"}|rM|j#IdH\} d}t$|t%rit|rit&|dri|d}nd}tj'j||| ||dd t(j}|D] \}} )|| qt*fd!d"}t*d#t+tt,fffd$d% }d#t+tt,fffd&d' }ddkr|n}z5zMt-d(d)||IdHWWt-d*f d+i d,}|r|vr|.ddkrd+vrt/WdWdSt0y~}zRt1|j2dt3|IdHWYd}~Wt-d*f d+i d,}|r[|vr[|.ddkrkd+vrkt/WdWdSd}~wty}zt45\}} }!6|| |!7|d}~wt8y}"z[t9|"t0}rt1|j2dt3|IdHWYd}"~"Wt-d*f d+i d,}|r|vr|.ddkrd+vrt/WdWdSd}"~"wwt-d*f d+i d,}|r$|vr$|.ddkr5d+vr6t/www1s;wYWddS1sLwYdS)-aJ Handle ASGI application calls with tracing. Processes ASGI requests and responses, creating spans for: - HTTP requests and responses - websocket handshakes and message exchanges - Error handling and exception tracking Raises: BlockingException: When request is blocked Exception: Re-raises any exceptions from the wrapped application Span Types Created: - HTTP: asgi.request spans with HTTP metadata - websocket: websocket.receive, websocket.send, websocket.close spans r^rmethod websocketNz0failed to decode headers for distributed tracingTri) int_configrequest_headers pathr*r#) directionprotocol REMOTE_ADDR) remote_addrrJheaders_case_sensitiveenviron middleware span_nameresource span_typeservicedistributed_headersactivate_distributed_headersr>r?shostzKfailed to decode host header, host from http headers will not be consideredserverscheme query_stringrar1z {}://{}{}rhrfr:?zasgi.request.parse.bodyclient) rurlqueryrraw_uri parsed_query request_bodypeer_ipheaders_are_case_sensitivec sjjjs IdHSdid}tjdjdjdddtjjjs,ndd d d y}|rA| d dd|j }zVzBIdH}d d krb|d dkrb |||n |d dkro ||WWdvs|ddvr| WdSty|jt| wdvsddvr| ww1swYdS)a Wrapped receive function that instruments websocket message reception. Intercepts websocket receive operations to create ExecutionContexts for: - Message receive timing and metadata - Message type (text/binary) - Message length measurement - Connection close events When websocket instrumentation is enabled, creates ExecutionContexts for: - websocket.receive: For incoming messages - websocket.close: For peer disconnect events Note: - Spans are linked to the handshake span - Receive spans are finished exactly when the next receive operation starts NrWruasgi.websocket.receive.messagezwebsocket.receive websocket rr1FT)r?rrrrchild_of call_traceactivater^rzwebsocket.disconnect)r?r,rCrcontext_with_datarr WEBSOCKETr.rvrwrU!_handle_websocket_receive_message$_handle_websocket_disconnect_messagerm set_exc_infosysrj)ructx recv_spanr~)rZr>rrUr2r3wrapped_receive/sN      z1TraceMiddleware.__call__..wrapped_receiver~c szcddkr&jjr&|ddkr&rjdkr|IdHWSddkr?jjr?|ddkr?|n ddkr_jjr_|ddkr_||IdHWSt|}Wntyvt j dd d d}Ynw ||t d |d |ft}|rt|z!|IdHW|dd kr|ddsjdkrSSSS|dd kr|ddsΈjdkrψwwww)a Wrapped ASGI send function that traces websocket message transmission. Intercepts websocket send operations to create spans for: - Message transmission timing and metadata - Message type (text/binary) - Message length measurement - Connection close operations When websocket instrumentation is enabled, creates ExecutionContexts for: - websocket.send: For outgoing messages - websocket.close: For application-initiated close events Note: - Spans are linked to the handshake span - Context (baggage, sampling) is propagated from handshake span - Each sent message is attached to the current context - Close spans include additional metadata like close codes and reasons r^rzwebsocket.acceptrNwebsocket.sendwebsocket.closez"failed to extract response headersTriasgi.finalize_responserbr` more_bodyF)r?r,rCerrorrv_handle_websocket_send_message_handle_websocket_close_messagerPrmrnwarning_handle_http_responserdispatchrr)r~rdblocked)rr>rr[rUr2r3 wrapped_sendls\           z.TraceMiddleware.__call__..wrapped_sendc s8tdfj}|r|j\}}}ndgd}}}r7|ddkr7||d<t||d<tdd|fn#|dd krZt|trE|n|j d d d |d <d|d<td|dfz$|IdHWt j j ||d|dd kr}j dkr~SSSt j j ||d|dd krj dkrwww)Nzasgi.block.startedr]rar^r\rJr_rr`zutf-8r6r7rbFr) status_coderdr)rdispatch_with_resultsstatus_headers_contentvaluerCintrr9r:encoder set_http_metar?rrv)r~rqr_rJcontent)rrr[rUrr2r3wrapped_blocked_sends@     z6TraceMiddleware.__call__..wrapped_blocked_sendzasgi.start_requestr!zweb.request.final_tagsrWrX):r?r,rrPrmrnrr rrjoinrCrrINBOUNDrrrWEB int_servicerUritemsrUnicodeDecodeErrorrparse_qsr<formatrl default_portsstrtrace_query_stringrawait_receive_and_bodyrr9listrrrIrRrrrrremoverxrrargsrcrrjrr BaseExceptionr)#rr>rZr[rJroperation_name host_headerkeyrrrr full_pathport default_port server_hostrrbrqrrrGnamerrr wrapped_recvrXerTexc_typeexc_valexc_tb exceptionr2)rrrZr>rr[rUrr3__call__sR              ,   $  <$"F  x  {    8zTraceMiddleware.__call__r~ request_spancCs|did}|jjr|r|}n|}tjd|jd|jd|ddtj|dd t|jj t t j id }t d|||fWddS1sIwYdS) NrWruzasgi.websocket.send.messagerrrr1FT r?rrrrrrrrG)rCr?r.rrrrrrintegration_namer r PRODUCERrrr>r~rru parent_spanrr2r2r3rs&  "z.TraceMiddleware._handle_websocket_send_messagec Cs|did}|jjr|r|}n|}tjd|jdd|ddtj|dd t|jjt t j id }t d|||fWdn1sFwYt |dS) NrWruzasgi.websocket.close.messagerrrr1FT)r?rrrrrrrG)rCr?r.rrrrrrr r rrrxrr2r2r3rs&   z/TraceMiddleware._handle_websocket_close_messagerUrrdcCs||r8|ddkr:d|vrr~rUrrdrprr2r2r3rs z%TraceMiddleware._handle_http_responserrcCs*t|td|||f||dd<dS)NrrWru)rxrr)rrr>r~rrr2r2r3r1sz1TraceMiddleware._handle_websocket_receive_messagecCst|tjd|jd|jd|ddtj|jjs|ndddt |jj t t j id }td|||fWddS1s?wYdS) Nz!asgi.websocket.disconnect.messagerrrr1FTr)rxrrr?rrCrrr.rrr r CONSUMERr)rr>r~rrr2r2r3r>s  "z4TraceMiddleware._handle_websocket_disconnect_message)__name__ __module__ __qualname____doc__rr r!rVrrrrrrrrrrrrExecutionContextrrr2r2r2r3rysT  &9&        ry)J functoolsrosrtypingrrrrrurllibrddtracer ddtrace.constantsr ddtrace.contribr #ddtrace.contrib.internal.asgi.utilsr ddtrace.extr rrddtrace.internalrddtrace.internal._exceptionsrrddtrace.internal.compatrddtrace.internal.constantsrddtrace.internal.loggerrddtrace.internal.schemar-ddtrace.internal.schema.span_attribute_schemar!ddtrace.internal.settings._configrddtrace.internal.utilsrr#ddtrace.internal.utils.deprecationsrddtrace.internal.utils.formatsr ddtrace.tracerrddtrace.vendor.debtcollectorrrrngetenvr_addrO _get_servicerErFrr4r:r<rIrPrVrYrcrtrxryr2r2r2r3s                                  .""