U Zh.5 @s(ddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z ddl mZmZmZmZmZmZddlmZdd lmZdd lmZd d lmZd d lmZd dlmZe dZ!dddddZ"dddddddddd Z#ddd d!d"d#d$d%d&d'Z$Gd(d)d)eZ%Gd*d+d+eZ&dS),) annotationsN) SyncBackend)NetworkBackend NetworkStream)ConnectionNotAvailable ProxyError)URLOriginRequestResponse enforce_bytes enforce_url)default_ssl_context)Lock)Trace)ConnectionPool)HTTP11Connection)ConnectionInterfacezhttpcore.sockszNO AUTHENTICATION REQUIREDZGSSAPIzUSERNAME/PASSWORDzNO ACCEPTABLE METHODS)Z SucceededzGeneral SOCKS server failurez!Connection not allowed by rulesetzNetwork unreachablezHost unreachablezConnection refusedz TTL expiredzCommand not supportedzAddress type not supported) rrr)authrbytesinttuple[bytes, bytes] | NoneNone)streamhostportr returncCstj}|dkrtjjjntjjj}|tj|g|}| ||j dd}| |}t |tjj stt|j|krt|d} t|jd} td| d| d|jtjjjkr4|dk st|\} } |tj| | |}| ||j dd}| |}t |tjjs$t|js4td|tjjtjjj||f|}| ||j dd}| |}t |tjjst|jtjjjkrt|jd} td | ddS) Ni)Z max_bytesUNKNOWNz Requested z from proxy server, but got .zInvalid username/passwordZUNKOWNz Proxy Server could not connect: )socksioZsocks5ZSOCKS5ConnectionZSOCKS5AuthMethodZNO_AUTH_REQUIREDZUSERNAME_PASSWORDsendZSOCKS5AuthMethodsRequestZ data_to_sendwritereadZ receive_data isinstanceZSOCKS5AuthReplyAssertionErrormethod AUTH_METHODSgetrZSOCKS5UsernamePasswordRequestZSOCKS5UsernamePasswordReplysuccessZSOCKS5CommandRequest from_addressZ SOCKS5CommandZCONNECTZ SOCKS5Reply reply_codeZSOCKS5ReplyCodeZ SUCCEEDED REPLY_CODES)r%r&r'r connZ auth_methodZoutgoing_bytesZincoming_bytesresponse requestedZ respondedusernamepasswordr6r=>/tmp/pip-unpacked-wheel-s37ty00c/httpcore/_sync/socks_proxy.py_init_socks5_connection*sT              r?c sLeZdZdZdddd d d d d d d ddd fdd ZdddddZZS) SOCKSProxyzB A connection pool that sends requests via an HTTP proxy. N TFrzURL | bytes | strz&tuple[bytes | str, bytes | str] | Nonessl.SSLContext | Nonez int | None float | Noneboolr"NetworkBackend | Noner$) proxy_url proxy_auth ssl_contextmax_connectionsmax_keepalive_connectionskeepalive_expiryhttp1http2retriesnetwork_backendr(c  sntj||||||| | d||_t|dd|_|dk rd|\} } t| dd} t| dd}| |f|_nd|_dS)a& A connection pool for making HTTP requests. Parameters: proxy_url: The URL to use when connecting to the proxy server. For example `"http://127.0.0.1:8080/"`. ssl_context: An SSL context to use for verifying connections. If not specified, the default `httpcore.default_ssl_context()` will be used. max_connections: The maximum number of concurrent HTTP connections that the pool should allow. Any attempt to send a request on a pool that would exceed this amount will block until a connection is available. max_keepalive_connections: The maximum number of idle HTTP connections that will be maintained in the pool. keepalive_expiry: The duration in seconds that an idle HTTP connection may be maintained for before being expired from the pool. http1: A boolean indicating if HTTP/1.1 requests should be supported by the connection pool. Defaults to True. http2: A boolean indicating if HTTP/2 requests should be supported by the connection pool. Defaults to False. retries: The maximum number of retries when trying to establish a connection. local_address: Local address to connect from. Can also be used to connect using a particular address family. Using `local_address="0.0.0.0"` will connect using an `AF_INET` address (IPv4), while using `local_address="::"` will connect using an `AF_INET6` address (IPv6). uds: Path to a Unix Domain Socket to use instead of TCP sockets. network_backend: A backend instance to use for handling network I/O. )rHrIrJrKrLrMrOrNrF)nameNrG)super__init__ _ssl_contextr _proxy_urlr _proxy_auth)selfrFrGrHrIrJrKrLrMrNrOr;r<Zusername_bytesZpassword_bytes __class__r=r>rRns(+   zSOCKSProxy.__init__r roriginr(c Cs(t|jj||j|j|j|j|j|jdS)N) proxy_origin remote_originrGrHrKrLrMrO) Socks5ConnectionrTrZrUrS_keepalive_expiry_http1_http2_network_backendrVrZr=r=r>create_connectionszSOCKSProxy.create_connection) NNrANNTFrN)__name__ __module__ __qualname____doc__rRrc __classcell__r=r=rWr>r@is(Br@c @seZdZd&dddddddd d d d d ZdddddZdddddZd dddZddddZddddZddddZ dddd Z d!dd"d#Z d!dd$d%Z dS)'r]NTFr r#rBrCrDrEr$) r[r\rGrHrKrLrMrOr(c CsV||_||_||_||_||_||_||_|dkr8tn||_t |_ d|_ d|_ dS)NF) _proxy_origin_remote_originrUrSr^r_r`rrar _connect_lock _connection_connect_failed) rVr[r\rGrHrKrLrMrOr=r=r>rRs zSocks5Connection.__init__r r )requestr(c Cs&|jdi}|jdd}|dd}|j|jdkrz|jjd|jj|d}tdt ||}|j j f|}||_ W5QRX||j jd|j j|jd}tdt ||}tf|||_ W5QRX|j jd kr^|jdkrtn|j}|jrd d gnd g} || ||p*|j jd|d }td t ||}|jf|}||_ W5QRX|d} | dk o|| d k} | s|jr|jsddlm} | |j ||jd|_nt|j ||jd|_Wn.tk r} zd|_| W5d} ~ XYnXn|jst W5QRX|j!|S)Ntimeout sni_hostnameconnectascii)r&r'ro connect_tcp)r%r&r'r Zsetup_socks5_connectionhttpszhttp/1.1h2)rHserver_hostnamero start_tls ssl_objectr)HTTP2Connection)rZr%rKT)" extensionsr3rkrlrir&decoder'rloggerrarsZ return_valuerjrUr?schemerSrr`set_alpn_protocolsrwZget_extra_infoselected_alpn_protocolr_rMryr^r Exceptionrm is_availablerhandle_request)rVrnZtimeoutsrprokwargstracer%rHalpn_protocolsrxZhttp2_negotiatedryexcr=r=r>rs                zSocks5Connection.handle_requestrYcCs ||jkSN)rjrbr=r=r>can_handle_request-sz#Socks5Connection.can_handle_request)r(cCs|jdk r|jdSr)rlcloserVr=r=r>r0s zSocks5Connection.closecCs6|jdkr,|jo*|jjdks$|j o*|j S|jS)Nrt)rlr`rjr}r_rmrrr=r=r>r4s zSocks5Connection.is_availablecCs|jdkr|jS|jSr)rlrm has_expiredrr=r=r>r@s zSocks5Connection.has_expiredcCs|jdkr|jS|jSr)rlrmis_idlerr=r=r>rEs zSocks5Connection.is_idlecCs|jdkr|jS|jSr)rlrm is_closedrr=r=r>rJs zSocks5Connection.is_closedstrcCs"|jdkr|jrdSdS|jS)NzCONNECTION FAILEDZ CONNECTING)rlrminforr=r=r>rOs zSocks5Connection.infocCsd|jjd|dS)N)rXrdrrr=r=r>__repr__TszSocks5Connection.__repr__)NNNTFN) rdrerfrRrrrrrrrrrr=r=r=r>r]s  U r])' __future__rloggingsslr+Z_backends.syncrZ_backends.baserr _exceptionsrrZ_modelsr r r r r r_sslrZ_synchronizationrZ_tracerZconnection_poolrZhttp11r interfacesr getLoggerr|r2r7r?r@r]r=r=r=r>sB          ?T