o iuA @sUddlmZddlmZddlmZddlmZddlZddl Z ddl Z ddl m Z ddl m Z ddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddl#m$Z$ddl%m&Z&e 'dZ(e 'dZ)e rddl*m+Z,ddl-m.Z.ddl-m/Z/ddl0m1Z1edZ2eged fZ3e4e5Z6e&d!ee7d"ee7fd#d$Z8d%e7d"e7fd&d'Z9d(e7d)e jd"ee:e7ffd*d+Z; dYd%e7d)e jd(ee7d"ee7e:ffd,d-Zd%e7d3e d"e3fd4d5Z?d6d7Z@e&d8e=ee7e7fd"e7fd9d:ZAd;d<ZBGd=d>d>eCZDefd%e7d?eEd"e2fd@dAZFd%e7d"ejGfdBdCZHdaIee7eJdD<daKee7eJdE<dFZLdGe7dHe7d"e7fdIdJZMdKe7dHe7d"e7fdLdMZNdNe7d"eOe7ee7ePe7fffdOdPZQdYdNe7dQeeOd"eOe7e ffdRdSZReGdTdUdUZSdVePeSd"e=e:eOffdWdXZTdS)Z)contextmanager) dataclass) encode_noop)loadsN) TYPE_CHECKING)Any)Callable)ContextManager) Generator)Optional)Pattern)Union)parse) _USER_ID_KEY)unpatched_open)BLOCKED_RESPONSE_HTML)BLOCKED_RESPONSE_JSON)DEFAULT_TIMEOUT)SAMPLING_DECISION_TRACE_TAG_KEY)W3C_TRACESTATE_ORIGIN_KEY)W3C_TRACESTATE_PARENT_ID_KEY)$W3C_TRACESTATE_SAMPLING_PRIORITY_KEY)_get_metas_to_propagate)cachedz,|;|~|[^\x20-\x7E]+z,| |=|[^\x20-\x7E]+HTTPConnectionHTTPSConnectionUDSHTTPConnection)rrrhttplib.HTTPConnection header_namereturncCs|dur |SdS)z Normalizes an header name to lower case, stripping all its leading and trailing white spaces. :param header_name: the header name to normalize :type header_name: str :return: the normalized header name :rtype: str N)striplower)r!r%O/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/internal/utils/http.pynormalize_header_name1s r'urlcCs4|d\}}}|d\}}}|s|S|||S)z Strips the query string from a URL for use as tag in spans. :param url: The URL to be stripped :return: The given URL without query strings #?) partition)r(hqsfsfh_r%r%r&strip_query_string=s  r1 query_string query_string_obfuscation_patterncCs$t|tr|n|d}|d|S)Nutf-8s ) isinstancebytesencodesub)r2r3 bytes_queryr%r%r&redact_query_stringJs r:cCst|}d}|rt||}n |jrt|j|}|dur?t|dkr?|dd|f|dd}tdd|D}t||S|S)Ncss(|]}t|tr |n|dVqdS)r4N)r5r6r7).0xr%r%r& \s&zredact_url..)rurlparser:querylentuple urlunsplit)r(r3r2partsredacted_queryredacted_partsbytes_redacted_partsr%r%r& redact_urlOs    rI components. original_urlcCs|\}}}}}}|rd||f}|s|r2|dddkr2|r*|dddkr*d|}d|p.d|f}|r:d ||f}|sD|rJ|d d vrJd ||f}|sT|rZ|d d vrZd||f}|S)zK Adaptation from urlunsplit and urlunparse, using bytes components s%s;%sNs///s//%s%ss%s:%s)r*?s%s?%s)r)#s%s#%sr%)rJrKschemenetlocr(paramsrAfragmentr%r%r&rDcs    rDkwargsc s*tdttdddfffdd }|S)aCreate a connector context manager for the given URL. This function returns a context manager that wraps a connection object to perform HTTP requests against the given URL. Extra keyword arguments can be passed to the underlying connection object, if needed. Example:: >>> connect = connector("http://localhost:8080") >>> with connect() as conn: ... conn.request("GET", "/") ... ... r")r zhttplib.HTTPSConnectionNc3s$tfi}|V|dSN)get_connectionclose) connectionrWr(r%r&_connector_contexts z%connector.._connector_context)rr r )r(rWr]r%r\r& connectorws$r^c Cs8g}|jdur|dt|j|jr#|dtttd|jf|j t }|r8|dttd|f|j t }|rM|dttd|ft dd|D}t |D]<\}}|t t fvr|dd}dttd|fttd|f}|t|}|d ks|||t|7}qZtd |qZd |S) Nz{}:{}r0zt.dm:{}z t.usr.id:{}css|]}t|VqdSrX)rB)r=ir%r%r&r?sz)w3c_get_dd_list_member..z_dd.p.zt.zKtracestate would exceed 256 char limit with tag: %s. Tag will not be added.;)sampling_priorityappendformatr dd_originrw3c_encode_tag)_W3C_TRACESTATE_INVALID_CHARS_REGEX_VALUE_metagetrrsumrreplace'_W3C_TRACESTATE_INVALID_CHARS_REGEX_KEYrBlogdebugjoin) contexttagssampling_decisionusr_idcurrent_tags_lenkvnext_tagpotential_current_tags_lenr%r%r&w3c_get_dd_list_membersB           ryargscCs"|\}}}|||}|ddS)N=~)r8rk)rzpattern replacementtag_valr%r%r&rfs   rfcCsFdt|}d|vr|dd|dS|rd|d|Sd|S)Nz {}:{:016x}zdd=ra,)rdrrk) tracestatespan_idp_memberr%r%r&w3c_tracestate_add_ps  rc@s>eZdZdZgdZd ddZeddZdd Zd d Z dS) Responsea Custom API Response object to represent a response from calling the API. We do this to ensure we know expected properties will exist, and so we can call `resp.read()` and load the body once into an instance before we close the HTTPConnection used for the request. statusbodyreasonmsgNcCs||_||_||_||_dSrXr)selfrrrrr%r%r&__init__s zResponse.__init__cCs&||j|t|ddt|dddS)ad Build a ``Response`` from the provided ``HTTPResponse`` object. This function will call `.read()` to consume the body of the ``HTTPResponse`` object. :param resp: ``HTTPResponse`` object to build the ``Response`` from :type resp: ``HTTPResponse`` :rtype: ``Response`` :returns: A new ``Response`` rNrr)rreadgetattr)clsrespr%r%r&from_http_responses    zResponse.from_http_responsec Csz4|j}|std|WdSt|tst|dr|d}t|dr0|dr0tdWdSt|WSt t fyHtjd|d d YdSw) z0Helper to parse the body of this request as JSONz"Empty reply from Datadog Agent, %rNdecoder4 startswithOKzUCannot parse Datadog Agent response. This occurs because Datadog agent is out of datez/Unable to parse Datadog Agent JSON response: %rT)exc_info) rrmrnr5strhasattrrrr ValueError TypeError)rrr%r%r&get_jsons    zResponse.get_jsoncCsd|jj|j|j|j|jS)Nz6{0}(status={1!r}, body={2!r}, reason={3!r}, msg={4!r}))rd __class____name__rrrr)rr%r%r&__repr__szResponse.__repr__)NNNN) r __module__ __qualname____doc__ __slots__r classmethodrrrr%r%r%r&rs   rtimeoutcCst|}|jpd}|jp d}ddlm}ddlm}ddlm}|jdkr/|j ||j ||dS|jd kr>|j ||j ||dS|jd krL||||j |d St d |j) z+Return an HTTP connection to the given URL./rrrrhttps) base_pathrhttpunix)rzUnsupported protocol '%s') verify_urlhostnamepathddtrace.internal.httprrddtrace.internal.udsrrSwith_base_pathportr)r(rparsedrrrrrr%r%r&rYs        rYcCsnt|}d}|j|vrtd|j|d|f|jdvr'|js'td||jdkr5|js5td||S)zValidates that the given URL can be used as an intake Returns a parse.ParseResult. Raises a ``ValueError`` if the URL cannot be used as an intake )rrrz@Unsupported protocol '%s' in intake URL '%s'. Must be one of: %sz, )rrz#Invalid hostname in intake URL '%s'rz$Invalid file path in intake URL '%s')rr@rSrrorr)r(rschemesr%r%r&r$s    r_HTML_BLOCKED_TEMPLATE_CACHE_JSON_BLOCKED_TEMPLATE_CACHEz[security_response_id]templatesecurity_response_idcCs |t|SrX)rk_RESPONSE_ID_TEMPLATE)rrr%r%r&_format_template=s raccept_header_valuec Csd}|r d|vr d}|rtrtt|S|strtt|S|r&td}ntd}|rqz&t|d }|}Wdn1sBwY|rL|an|at||WStt fyp}zt d|t |WYd}~nd}~ww|rzt att|Statt|S)NFz text/htmlT$DD_APPSEC_HTTP_BLOCKED_TEMPLATE_HTML$DD_APPSEC_HTTP_BLOCKED_TEMPLATE_JSONrz(Could not load custom template at %s: %s)r$rrrosgetenvopenrOSErrorIOErrorrmwarningrrr)rrneed_html_template template_path template_filecontenter%r%r&_get_blocked_templateAs8         rrcCs|dd}t}|dD]6}|d\}}}|rDt|}t|}||d}|dur3|||<qt|tr>| |q||g||<q|S)2Return a dict of form data after HTTP form parsing+ &r{N) rkdictsplitr+runquoterir5listrc)r body_paramsreq_bodyitemkeyequalval prev_valuer%r%r&parse_form_paramsks        rheaderscsfddl}ddlddlmfdd|dur1|dp#|d}|d||f}|SiS) rrN)parse_qscs|rfdd|D}|S|d}|dvr#|}|S|dvr9ddlmm}||}|S|dvrE|}|S|dvrO|}|Sd }|S) Ncs&i|]}|jd|dd|qS)namezcontent-disposition)failobjheader) get_param get_filename)r=part) parse_messager%r& sz?parse_form_multipart..parse_message.. Content-Type)zapplication/jsonz text/json)zapplication/xmlztext/xmlr)zapplication/x-url-encodedz!application/x-www-form-urlencoded)z text/plainNr) is_multipart get_payloadrirddtrace.vendor.xmltodictvendor xmltodictr)rres content_typerjsonrrr%r&rs*    z+parse_form_multipart..parse_messagerz content-typez%MIME-Version: 1.0 Content-Type: %s %s)emailr urllib.parserrimessage_from_string)rrrrrr%rr&parse_form_multipart~s rc@s6eZdZUeed<eed<eeefed<eed<dS)FormDatarfilenamedatarN)rrrr__annotations__r r6r%r%r%r&rs  rrEc Csddlm}ddlm}ddlm}|d}|d=|D]}||j|jt}|j dd|j |j d|d=| |q|j |d d \}}} | td d |DfS) Nr)MIMEApplication) MIMEMultipart)HTTPz form-dataz MIME-VersionzContent-Disposition)rr)policys css|]}|dVqdS)z: N)r)r=r0r%r%r&r?szmultipart..)email.mime.applicationremail.mime.multipartr email.policyrrrr add_headerrrattachas_bytesr+rr splitlines) rErrrrrapprr0rr%r%r& multiparts    rrX)U contextlibr dataclassesremail.encodersrrrloggingrretypingrrrr r r r r urllibrddtrace.constantsrddtrace.internal._unpatchedrrddtrace.internal.constantsrrrrrrrddtrace.internal.utilsrddtrace.internal.utils.cachercompilergrl http.clientclienthttplibrrrrrConnectionType Connector getLoggerrrmrr'r1r6r:rIrCrDr^ryrfrobjectrfloatrY ParseResultrrrrrrrrrrrrrr%r%r%r&s                                 , D&*$$"