Format: application/x-python-code
1 o
2
3 7'h6 �
4 @ s d dl mZ d dlZd dlZd dlZddlmZ ddlmZm Z ddl
5 m Z m Z ddl
6 mZmZmZmZmZmZ ddlmZ dd lmZ dd
7 lmZ d d lmZ d d
8 l m Z d dl m Z e� d�Z!ddddd�Z"dddddddd d d � Z#dd �d0d*d+�Z$G d,d-� d-e�Z%G d.d/� d/e �Z&dS )1� )� annotationsN� )� AutoBackend)�AsyncNetworkBackend�AsyncNetworkStream)�ConnectionNotAvailable�
9 ProxyError)�URL�Origin�Request�Response�
10 enforce_bytes� enforce_url)�default_ssl_context)� AsyncLock)�Trace� )�AsyncConnectionPool)�AsyncHTTP11Connection)�AsyncConnectionInterfacezhttpcore.sockszNO AUTHENTICATION REQUIREDZGSSAPIzUSERNAME/PASSWORDzNO ACCEPTABLE METHODS)� � � � �Z Succeededz General SOCKS server failurez!Connection not allowed by rulesetzNetwork unreachablezHost unreachablezConnection refusedz TTL expiredzCommand not supportedzAddress type not supported) r r r � � � � � � )�auth�streamr �host�bytes�port�intr �tuple[bytes, bytes] | None�return�Nonec � s� �t j�� }|d u rt jjjnt jjj}|�t j�|g�� |�� }| � |�I d H | j
11 dd�I d H }|� |�}t |t jj
12 �sAJ �|j|kr^t�|d�} t�|jd�}
13 td| � d|
14 � d���|jt jjjkr�|d uslJ �|\} } |�t j�| | �� |�� }| � |�I d H | j
15 dd�I d H }|� |�}t |t jj�s�J �|js�td��|�t jj�t jjj||f�� |�� }| � |�I d H | j
16 dd�I d H }|� |�}t |t jj�s�J �|jt jjj kr�t �|jd�}
17 td |
18 � d���d S )
19 Ni )Z max_bytes�UNKNOWNz
20 Requested z from proxy server, but got �.zInvalid username/passwordZUNKOWNz Proxy Server could not connect: ) �socksioZsocks5ZSOCKS5ConnectionZSOCKS5AuthMethodZNO_AUTH_REQUIREDZUSERNAME_PASSWORD�sendZSOCKS5AuthMethodsRequestZ data_to_send�write�readZ receive_data�
21 isinstanceZSOCKS5AuthReply�method� AUTH_METHODS�getr Z SOCKS5UsernamePasswordRequestZSOCKS5UsernamePasswordReply�successZSOCKS5CommandRequest� from_addressZ
22 SOCKS5CommandZCONNECTZ SOCKS5Reply�
23 reply_codeZSOCKS5ReplyCodeZ SUCCEEDED� REPLY_CODES)r! r" r$ r �connZ auth_methodZoutgoing_bytesZincoming_bytes�response� requestedZ responded�username�passwordr5 � r< �QC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_async/socks_proxy.py�_init_socks5_connection* sV �
24
25 ��
26
27 �
28 ��
29 �r> c s@ e Zd ZdZ d"d#� fdd �
30 Zd$d d!�Z� ZS )%�AsyncSOCKSProxyzB
31 A connection pool that sends requests via an HTTP proxy.
32 N�
33 TFr � proxy_url�URL | bytes | str�
34 proxy_auth�&tuple[bytes | str, bytes | str] | None� ssl_context�ssl.SSLContext | None�max_connections�
35 int | None�max_keepalive_connections�keepalive_expiry� float | None�http1�bool�http2�retriesr% �network_backend�AsyncNetworkBackend | Noner' r( c
36 sp t � j|||||||
37 | d� || _t|dd�| _|dur3|\} } t| dd�}
38 t| dd�}|
39 |f| _dS d| _dS )a&
40 A connection pool for making HTTP requests.
41
42 Parameters:
43 proxy_url: The URL to use when connecting to the proxy server.
44 For example `"http://127.0.0.1:8080/"`.
45 ssl_context: An SSL context to use for verifying connections.
46 If not specified, the default `httpcore.default_ssl_context()`
47 will be used.
48 max_connections: The maximum number of concurrent HTTP connections that
49 the pool should allow. Any attempt to send a request on a pool that
50 would exceed this amount will block until a connection is available.
51 max_keepalive_connections: The maximum number of idle HTTP connections
52 that will be maintained in the pool.
53 keepalive_expiry: The duration in seconds that an idle HTTP connection
54 may be maintained for before being expired from the pool.
55 http1: A boolean indicating if HTTP/1.1 requests should be supported
56 by the connection pool. Defaults to True.
57 http2: A boolean indicating if HTTP/2 requests should be supported by
58 the connection pool. Defaults to False.
59 retries: The maximum number of retries when trying to establish
60 a connection.
61 local_address: Local address to connect from. Can also be used to
62 connect using a particular address family. Using
63 `local_address="0.0.0.0"` will connect using an `AF_INET` address
64 (IPv4), while using `local_address="::"` will connect using an
65 `AF_INET6` address (IPv6).
66 uds: Path to a Unix Domain Socket to use instead of TCP sockets.
67 network_backend: A backend instance to use for handling network I/O.
68 )rE rG rI rJ rL rN rP rO rA )�nameNrC )�super�__init__� _ssl_contextr �
69 _proxy_urlr
70 � _proxy_auth)�selfrA rC rE rG rI rJ rL rN rO rP r: r; Zusername_bytesZpassword_bytes�� __class__r< r= rT n s( +�
71
72 �
73 zAsyncSOCKSProxy.__init__�originr
74 r c
75 C s( t | jj|| j| j| j| j| j| jd�S )N)� proxy_origin�
76 remote_originrC rE rJ rL rN rP ) �AsyncSocks5ConnectionrV r[ rW rU �_keepalive_expiry�_http1�_http2�_network_backend�rX r[ r< r< r= �create_connection� s �z!AsyncSOCKSProxy.create_connection) NNr@ NNTFr N)rA rB rC rD rE rF rG rH rI rH rJ rK rL rM rN rM rO r% rP rQ r' r( )r[ r
77 r' r )�__name__�
78 __module__� __qualname__�__doc__rT rd �
79 __classcell__r< r< rY r= r? i s �Br? c @ s~ e Zd Z d-d.dd�Zd/dd�Zd0d d �Zd1d d �Zd2d d!�Zd2d"d#�Zd2d$d%�Z d2d&d'�Z
80 d3d)d*�Z d3d+d,�Z dS )4r^ NTFr\ r
81 r] rC r& rE rF rJ rK rL rM rN rP rQ r' r( c C sV || _ || _|| _|| _|| _|| _|| _|d u r t� n|| _t � | _
82 d | _ d| _ d S )NF)
83 �
84 _proxy_origin�_remote_originrW rU r_ r` ra r rb r �
85 _connect_lock� _connection�_connect_failed) rX r\ r] rC rE rJ rL rN rP r< r< r= rT � s �
86 z AsyncSocks5Connection.__init__�requestr r c � s� �|j �di �}|j �dd �}|�dd �}| j4 I d H ��< | jd u �rF�z| jj�d�| jj|d�}tdt ||�4 I d H �}| j
87 j di |��I d H }||_ W d �I d H n 1 I d H s^w Y || j
88 j�d�| j
89 j| jd�}tdt ||�4 I d H �}tdi |��I d H ||_ W d �I d H n 1 I d H s�w Y | j
90 jd kr�| jd u r�t� n| j}| jr�d
91 d gnd
92 g} |�| � ||p�| j
93 j�d�|d �}td
94 t ||�4 I d H �}|jdi |��I d H }||_ W d �I d H n 1 I d H s�w Y |�d�}
95 |
96 d u�o |
97 �� d k} | �s| j�r(| j�s(ddlm} | | j
98 || jd�| _n
99 t | j
100 || jd�| _W n t �yE }
101 zd| _ |
102 �d }
103 ~
104 ww | j� � �sOt � �W d �I d H n 1 I d H �s`w Y | j�!|�I d H S )N�timeout� sni_hostname�connect�ascii)r" r$ rp � connect_tcp)r! r" r$ r Zsetup_socks5_connection� httpszhttp/1.1�h2)rE �server_hostnamerp � start_tls�
105 ssl_objectr )�AsyncHTTP2Connection)r[ r! rJ Tr< )"�
106 extensionsr2 rl rm rj r" �decoder$ r �loggerrb rt � return_valuerk rW r> �schemerU r ra �set_alpn_protocolsrx �get_extra_info�selected_alpn_protocolr` rN rz r_ r � Exceptionrn � is_availabler �handle_async_request)rX ro Ztimeoutsrq rp �kwargs�tracer! rE �alpn_protocolsry Zhttp2_negotiatedrz �excr< r< r= r� � s� � �(� � �(�
107 ���
108 �(�
109
110
111 ���
112 ����� *�Nz*AsyncSocks5Connection.handle_async_requestr[ c C s
113 || j kS �N)rk rc r< r< r= �can_handle_request- s
114 z(AsyncSocks5Connection.can_handle_requestc � s$ �| j d ur| j �� I d H d S d S r� )rm �aclose�rX r< r< r= r� 0 s �
115 �z AsyncSocks5Connection.aclosec C s6 | j d u r| jo| jjdkp| j o| j S | j �� S )Nru )rm ra rk r r` rn r� r� r< r< r= r� 4 s
116 ��
117 z"AsyncSocks5Connection.is_availablec C � | j d u r| jS | j �� S r� )rm rn � has_expiredr� r< r< r= r� @ �
118
119 z!AsyncSocks5Connection.has_expiredc C r� r� )rm rn �is_idler� r< r< r= r� E r� z AsyncSocks5Connection.is_idlec C r� r� )rm rn � is_closedr� r< r< r= r� J r� z AsyncSocks5Connection.is_closed�strc C s" | j d u r | jr
120 dS dS | j �� S )NzCONNECTION FAILEDZ
121 CONNECTING)rm rn �infor� r< r< r= r� O s
122
123 zAsyncSocks5Connection.infoc C s d| j j� d| �� � d�S )N�<z [z]>)rZ re r� r� r< r< r= �__repr__T s z AsyncSocks5Connection.__repr__)NNNTFN)r\ r
124 r] r
125 rC r& rE rF rJ rK rL rM rN rM rP rQ r' r( )ro r r' r )r[ r
126 r' rM )r' r( )r' rM )r' r� )
127 re rf rg rT r� r� r� r� r� r� r� r� r� r< r< r< r= r^ � s" �
128
129 U
130
131
132
133
134
135 r^ )
136 r! r r" r# r$ r% r r& r' r( )'�
137 __future__r �logging�sslr+ Z_backends.autor Z_backends.baser r � _exceptionsr r Z_modelsr r
138 r r r
139 r �_sslr Z_synchronizationr �_tracer Zconnection_poolr Zhttp11r Z
140 interfacesr � getLoggerr} r1 r6 r> r? r^ r< r< r< r= �<module> sD
141 ���?T