Format: application/x-python-code
1 o
2
3 7'h.5 �
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� )� SyncBackend)�NetworkBackend�
9 NetworkStream)�ConnectionNotAvailable�
10 ProxyError)�URL�Origin�Request�Response�
11 enforce_bytes� enforce_url)�default_ssl_context)�Lock)�Trace� )�ConnectionPool)�HTTP11Connection)�ConnectionInterfacezhttpcore.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 C s� t j�� }|d u rt jjjnt jjj}|�t j�|g�� |�� }| � |� | j
12 dd�}|� |�}t |t jj
13 �s:J �|j|krWt�|d�} t�|jd�}
14 td| � d|
15 � d���|jt jjjkr�|d useJ �|\} } |�t j�| | �� |�� }| � |� | j
16 dd�}|� |�}t |t jj�s�J �|js�td��|�t jj�t jjj||f�� |�� }| � |� | j
17 dd�}|� |�}t |t jj�s�J �|jt jjj kr�t �|jd�}
18 td |
19 � d���d S )
20 Ni )Z max_bytes�UNKNOWNz
21 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�
22 isinstanceZSOCKS5AuthReply�method� AUTH_METHODS�getr Z SOCKS5UsernamePasswordRequestZSOCKS5UsernamePasswordReply�successZSOCKS5CommandRequest� from_addressZ
23 SOCKS5CommandZCONNECTZ SOCKS5Reply�
24 reply_codeZSOCKS5ReplyCodeZ SUCCEEDED� REPLY_CODES)r! r" r$ r �connZ auth_methodZoutgoing_bytesZincoming_bytes�response� requestedZ responded�username�passwordr5 � r< �PC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_sync/socks_proxy.py�_init_socks5_connection* sT
25
26 ��
27
28
29 �
30
31 ��
32
33 �r> c s@ e Zd ZdZ d"d#� fdd �
34 Zd$d d!�Z� ZS )%�
35 SOCKSProxyzB
36 A connection pool that sends requests via an HTTP proxy.
37 N�
38 TFr � proxy_url�URL | bytes | str�
39 proxy_auth�&tuple[bytes | str, bytes | str] | None� ssl_context�ssl.SSLContext | None�max_connections�
40 int | None�max_keepalive_connections�keepalive_expiry� float | None�http1�bool�http2�retriesr% �network_backend�NetworkBackend | Noner' r( c
41 sp t � j|||||||
42 | d� || _t|dd�| _|dur3|\} } t| dd�}
43 t| dd�}|
44 |f| _dS d| _dS )a&
45 A connection pool for making HTTP requests.
46
47 Parameters:
48 proxy_url: The URL to use when connecting to the proxy server.
49 For example `"http://127.0.0.1:8080/"`.
50 ssl_context: An SSL context to use for verifying connections.
51 If not specified, the default `httpcore.default_ssl_context()`
52 will be used.
53 max_connections: The maximum number of concurrent HTTP connections that
54 the pool should allow. Any attempt to send a request on a pool that
55 would exceed this amount will block until a connection is available.
56 max_keepalive_connections: The maximum number of idle HTTP connections
57 that will be maintained in the pool.
58 keepalive_expiry: The duration in seconds that an idle HTTP connection
59 may be maintained for before being expired from the pool.
60 http1: A boolean indicating if HTTP/1.1 requests should be supported
61 by the connection pool. Defaults to True.
62 http2: A boolean indicating if HTTP/2 requests should be supported by
63 the connection pool. Defaults to False.
64 retries: The maximum number of retries when trying to establish
65 a connection.
66 local_address: Local address to connect from. Can also be used to
67 connect using a particular address family. Using
68 `local_address="0.0.0.0"` will connect using an `AF_INET` address
69 (IPv4), while using `local_address="::"` will connect using an
70 `AF_INET6` address (IPv6).
71 uds: Path to a Unix Domain Socket to use instead of TCP sockets.
72 network_backend: A backend instance to use for handling network I/O.
73 )rE rG rI rJ rL rN rP rO rA )�nameNrC )�super�__init__� _ssl_contextr �
74 _proxy_urlr
75 � _proxy_auth)�selfrA rC rE rG rI rJ rL rN rO rP r: r; Zusername_bytesZpassword_bytes�� __class__r< r= rT n s( +�
76
77 �
78 zSOCKSProxy.__init__�originr
79 r c
80 C s( t | jj|| j| j| j| j| j| jd�S )N)� proxy_origin�
81 remote_originrC rE rJ rL rN rP ) �Socks5ConnectionrV r[ rW rU �_keepalive_expiry�_http1�_http2�_network_backend�rX r[ r< r< r= �create_connection� s �z SOCKSProxy.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
82 r' r )�__name__�
83 __module__� __qualname__�__doc__rT rd �
84 __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
85 d3d)d*�Z d3d+d,�Z dS )4r^ NTFr\ r
86 r] rC r& rE rF rJ rK rL rM rN rP rQ r' r( c C sV || _ || _|| _|| _|| _|| _|| _|d u r t� n|| _t � | _
87 d | _ d| _ d S )NF)
88 �
89 _proxy_origin�_remote_originrW rU r_ r` ra r rb r �
90 _connect_lock� _connection�_connect_failed) rX r\ r] rC rE rJ rL rN rP r< r< r= rT � s �
91 zSocks5Connection.__init__�requestr r c C sh |j �di �}|j �dd �}|�dd �}| j��
92 | jd u �rz�| jj�d�| jj|d�}tdt ||��}| j
93 j di |��}||_ W d � n1 sKw Y || j
94 j�d�| j
95 j| jd�}tdt ||��}tdi |�� ||_ W d � n1 s{w Y | j
96 jd kr�| jd u r�t� n| j}| jr�d
97 d gnd
98 g} |�| � ||p�| j
99 j�d�|d �}td
100 t ||��}|jdi |��}||_ W d � n1 s�w Y |�d�}
101 |
102 d uo�|
103 �� d k} | s�| jr�| js�ddlm} | | j
104 || jd�| _n
105 t | j
106 || jd�| _W n t �y }
107 zd| _ |
108 �d }
109 ~
110 ww | j� � �s t � �W d � n 1 �s)w Y | j�!|�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�
111 ssl_objectr )�HTTP2Connection)r[ r! rJ Tr< )"�
112 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_request)rX ro Ztimeoutsrq rp �kwargs�tracer! rE �alpn_protocolsry Zhttp2_negotiatedrz �excr< r< r= r� � s�
113 � � �� �
114 ���
115 � �
116
117 ���
118 ����� � � Nz Socks5Connection.handle_requestr[ c C s
119 || j kS �N)rk rc r< r< r= �can_handle_request- s
120 z#Socks5Connection.can_handle_requestc C s | j d ur | j �� d S d S r� )rm �close�rX r< r< r= r� 0 s
121 �zSocks5Connection.closec 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
122 ��
123 z Socks5Connection.is_availablec C � | j d u r| jS | j �� S r� )rm rn � has_expiredr� r< r< r= r� @ �
124
125 z Socks5Connection.has_expiredc C r� r� )rm rn �is_idler� r< r< r= r� E r� zSocks5Connection.is_idlec C r� r� )rm rn � is_closedr� r< r< r= r� J r� zSocks5Connection.is_closed�strc C s" | j d u r | jr
126 dS dS | j �� S )NzCONNECTION FAILEDZ
127 CONNECTING)rm rn �infor� r< r< r= r� O s
128
129 zSocks5Connection.infoc C s d| j j� d| �� � d�S )N�<z [z]>)rZ re r� r� r< r< r= �__repr__T s zSocks5Connection.__repr__)NNNTFN)r\ r
130 r] r
131 rC r& rE rF rJ rK rL rM rN rM rP rQ r' r( )ro r r' r )r[ r
132 r' rM )r' r( )r' rM )r' r� )
133 re rf rg rT r� r� r� r� r� r� r� r� r� r< r< r< r= r^ � s" �
134
135 U
136
137
138
139
140
141 r^ )
142 r! r r" r# r$ r% r r& r' r( )'�
143 __future__r �logging�sslr+ Z_backends.syncr Z_backends.baser r � _exceptionsr r Z_modelsr r
144 r r r
145 r �_sslr Z_synchronizationr �_tracer Zconnection_poolr Zhttp11r Z
146 interfacesr � getLoggerr} r1 r6 r> r? r^ r< r< r< r= �<module> sD
147 ���?T