o ۷i @s.ddlZddlZddlmZddlZddlZddlmZmZddl m Z m Z m Z m Z mZmZmZddlZddZeZdedefd d Zd ed eedefd dZddefddZed edeefddZd edefddZGdddejZ  dded eedBdejedBdBdej fddZ!dS) N)Path) lru_cachewraps)Any AwaitableCallable CoroutineLiteralTTuplecCsHttjd}t|d}|WdS1swYdS)Nz version.txtr)r__file__parentopenreadstrip) version_filefrH/home/ubuntu/vllm_env/lib/python3.10/site-packages/safehttpx/__init__.py get_version s  $ripreturncCsBzt|}|jp|jp|jp|jp|j WSty YdSw)NF) ipaddress ip_address is_private is_loopback is_link_local is_multicast is_reserved ValueError)rip_objrrr is_public_ips  r"hostnamedomain_whitelistcCsr|r|sdS|}|D]*}|sq |}|dr/|dd}||ks+|d|r.dSq ||kr6dSq dS)NFz*..T)lower startswithendswith)r#r$domainsuffixrrrmatches_domain_whitelist s   r,maxsizecs8dtdttttffdtdttfffdd }|S)N async_func.rc s4tdtdtdtdttffdd }|S)Nr.argskwargsrcst|i|SN)asyncio create_task)r1r2r/rrwrapper5sz3lru_cache_async..decorator..wrapper)rrrrr )r/r7r0r6r decorator2s"z"lru_cache_async..decorator)rrrr r)r.r8rr0rlru_cache_async1s  r9c st4IdHW}z@|d|dIdH}|d|dIdH}g}||fD]}|dd|dgDq-|WWdIdHSty_gYWdIdHSw1IdHsfwYdS)Nz https://dns.google/resolve?name=z&type=Az &type=AAAAcSsg|]}|dqS)datar).0answerrrr Lsz1async_resolve_hostname_google..Answer)httpx AsyncClientgetjsonextend Exception)r#client response_v4 response_v6ipsresponserrrasync_resolve_hostname_google?s&      rJc szt}||dIdH}Wntjy(}z td|d||d}~ww|D]\}}}}}|d}|tjtjfvrFt|rF|Sq+t |IdHD] }t|rX|SqNtd|d)NzUnable to resolve hostname z: rz Hostname z failed validation) r4get_event_loop getaddrinfosocketgaierrorr AF_INETAF_INET6r"rJ)r#loopaddrinfoefamily_sockaddrrrrrasync_validate_urlRs$rWcsHeZdZdeffdd Zdejdeee e ej fffdd Z Z S)AsyncSecureTransport verified_ipcs||_tdSr3)rYsuper__init__)selfrY __class__rrr[fszAsyncSecureTransport.__init__requestrcsH|j}|j}|j|jd}||_||jd<d|i|_t|IdHS)N)hostHost sni_hostname)urlr` copy_withrYheaders extensionsrZhandle_async_request)r\r_ original_url original_hostnew_urlr]rrrgjs  z)AsyncSecureTransport.handle_async_request) __name__ __module__ __qualname__strr[r?Requestr intbytesHeadersrg __classcell__rrr]rrXesrXrc _transportFc st|}|j}|std|d|durg}|r|}n|dus't||r*d}n t|IdH}t|}tj|d4IdH}|j|fddi|IdHWdIdHS1IdHs^wYdS)a< This is the main function that should be used to make async HTTP GET requests. It will automatically use a secure transport for non-whitelisted domains, unless a proxy is set in the environment variables (HTTP_PROXY, HTTPS_PROXY, http_proxy, https_proxy). Parameters: - url (str): The URL to make a GET request to. - domain_whitelist (list[str] | None): A list of domains to whitelist, which will not use a secure transport. Supports wildcard subdomains with "*.domain.com" format (asterisk must be at the beginning). - _transport (httpx.AsyncBaseTransport | Literal[False] | None): A custom transport to use for the request. Takes precedence over domain_whitelist. Set to False to use no transport. - **kwargs: Additional keyword arguments to pass to the httpx.AsyncClient.get() function. zURL z does not have a valid hostnameNF) transportfollow_redirects) r?URLr`r r,rWrXr@rA) rcr$rtr2 parsed_urlr#rurYrErrrrAvs  0rA)r-)NN)"r4rpathlibrrMos functoolsrrtypingrrrrr r r r?r __version__rnboolr"listr,rpr9rJrWAsyncHTTPTransportrXAsyncBaseTransportResponserArrrrs8 $