a
    ñ»jT9  ã                   @   s˜  d Z ddlZddlZddlZddlmZ ddlmZm	Z	m
Z
 ddlmZmZmZ ddlmZmZmZ ddlmZmZ dd	lmZmZ g d
¢Zz4ddlmZmZmZ ddlmZ ddl m!Z! dZ"W n@   dZ"G dd„ de#ƒZG dd„ de#ƒZG dd„ de#ƒZY n0 G dd„ dƒZ$e%e&dœdd„Z'e%dœdd„Z(e)e*e&dœdd „Z+d!d"„ Z,eje-d#œd$d%„Z.eje-d&œd'd(„Z/eje)ejd)œd*d+„Z0eje&d,œd-d.„Z1dS )/a_  
_http.py
websocket - WebSocket client library for Python

Copyright 2025 engn33r

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
é    N)Úencodebytesé   )ÚWebSocketAddressExceptionÚWebSocketExceptionÚWebSocketProxyException)ÚdebugÚdumpÚtrace)ÚDEFAULT_SOCKET_OPTIONÚ	recv_lineÚsend)ÚHAVE_SSLÚssl)Úget_proxy_infoÚ	parse_url)Ú
proxy_infoÚconnectÚread_headers)ÚProxyConnectionErrorÚ
ProxyErrorÚProxyTimeoutError)Ú	ProxyType)ÚProxyTFc                   @   s   e Zd ZdS )r   N©Ú__name__Ú
__module__Ú__qualname__© r   r   ú5E:\soccer\ml_env\Lib\site-packages\websocket/_http.pyr   .   s   r   c                   @   s   e Zd ZdS )r   Nr   r   r   r   r   r   1   s   r   c                   @   s   e Zd ZdS )r   Nr   r   r   r   r   r   4   s   r   c                   @   s   e Zd Zdd„ ZdS )r   c                 K   sŠ   |  dd ¡| _| jrn|  dd¡| _|  dd ¡| _|  dd ¡| _|  dd¡| _|  dd ¡| _| jd	vr†td
ƒ‚nd| _d | _d | _d| _d S )NZhttp_proxy_hostZhttp_proxy_portr   Zhttp_proxy_authZhttp_no_proxyÚ
proxy_typeÚhttpZhttp_proxy_timeout)r    Úsocks4Úsocks4aÚsocks5Úsocks5hz7Only http, socks4, socks5 proxy protocols are supported)ÚgetÚ
proxy_hostÚ
proxy_portÚauthÚno_proxyÚproxy_protocolÚproxy_timeoutr   )ÚselfÚoptionsr   r   r   Ú__init__9   s    
ÿzproxy_info.__init__N)r   r   r   r.   r   r   r   r   r   8   s   r   )ÚurlÚreturnc                 C   sð   t stdƒ‚t| ƒ\}}}}|jdkr2d}tj}n@|jdkrHd}tj}n*|jdkr^d}tj}n|jdkrrd}tj}tj||j	t
|jƒ|jr”|jd nd |jr¦|jd	 nd |d
�}	|	j|||jd�}
|râtrÚt|
|j|ƒ}
ntdƒ‚|
|||ffS )Nz>Python Socks is needed for SOCKS proxying but is not availabler!   Fr"   Tr#   r$   r   r   )r   ÚhostÚportÚusernameÚpasswordÚrdns)ÚtimeoutúSSL not available.)ÚHAVE_PYTHON_SOCKSr   r   r*   r   ZSOCKS4ZSOCKS5r   Úcreater&   Úintr'   r(   r   r+   r   Ú_ssl_socketÚsslopt)r/   r-   ÚproxyÚhostnamer2   ÚresourceÚ	is_securer5   r   Zws_proxyÚsockr   r   r   Ú_start_proxied_socketS   s>    ÿ



ú	rB   )r/   c                 C   sä   |j r |s |jdkr t| ||ƒS t| ƒ\}}}}|rB||||ffS t||||ƒ\}}	}
|sntd|› d|› �ƒ‚d }zRt||j|jƒ}|	r–t	||||
ƒ}|r¶t
r®t||j|ƒ}ntdƒ‚||||ffW S    |rØ| ¡  ‚ Y n0 d S )Nr    zHost not found.: ú:r7   )r&   r*   rB   r   Ú_get_addrinfo_listr   Ú_open_socketÚsockoptr6   Ú_tunnelr   r;   r<   Úclose)r/   r-   r=   Úsocketr>   Zport_from_urlr?   r@   Úaddrinfo_listZneed_tunnelr(   rA   r   r   r   r   ~   s0    ÿ
r   )r2   r@   r0   c           	   
   C   s¬   t | ||j|j|j|jƒ\}}}zX|sHt | |dtjtj¡}|dd fW S |rP|pRd}t ||dtjtj¡}|d|fW S W n. tj	y¦ } zt
|ƒ‚W Y d }~n
d }~0 0 d S )Nr   FéP   T)r   r&   r'   r(   r)   rI   ÚgetaddrinfoÚSOCK_STREAMÚSOL_TCPÚgaierrorr   )	r>   r2   r@   r=   ZphostZpportZpauthrJ   Úer   r   r   rD   £   s*    ú
ÿÿrD   c                 C   s.  d }| D �]}|d d… \}}}t   |||¡}| |¡ tD ]}	|j|	Ž  q<|D ]}	|j|	Ž  qP|d }
d }|sz| |
¡ W n� t j�y } zt| ¡  t|
d ƒ|_zt	j
t	jt	jf}W n tyÚ   t	j
t	jf}Y n0 |j	|vrê|‚|}W Y d }~qlW Y d }~qld }~0 0 �qqlq �q*q|�r*|‚|S )Né   é   r   )rI   Ú
settimeoutr
   Ú
setsockoptr   ÚerrorrH   ÚstrZ	remote_ipÚerrnoZECONNREFUSEDZWSAECONNREFUSEDZENETUNREACHÚAttributeError)rJ   rF   r6   ÚerrZaddrinfoÚfamilyÚsocktypeÚprotorA   ÚoptsÚaddressrU   ZeConnRefusedr   r   r   rE   Ã   sD    

ý
"rE   )rA   r<   c              
   C   s|  |  dd ¡}|�sZt |  dtj¡¡}tj  d¡}|d ur@||_|  dtj¡tjk�r|  dd ¡}|  dd ¡}|sv|rÆz|j||d� W n: t	tj
tfyÂ } ztd|› �ƒ‚W Y d }~n
d }~0 0 nVt|d	ƒ�rz| tjj¡ W n6 tj
�y } ztd
|› �ƒ‚W Y d }~n
d }~0 0 |  dd ¡�r¾z&| |d |  dd ¡|  dd ¡¡ W nl t	tf�yˆ } ztd|› �ƒ‚W Y d }~n>d }~0  tj
�y¼ } ztd|› �ƒ‚W Y d }~n
d }~0 0 |  dtj¡tjk�rò|  dd¡�sòd|_tj|_n|  dd¡|_|  dtj¡|_d|v �rdz| |d ¡ W n6 tj
�yb } ztd|› �ƒ‚W Y d }~n
d }~0 0 d|v �rzJ|d }	t|	ttfƒ�r–t|	ƒdk�ržtdƒ‚|	\}
}}| |
||¡ W nN t�yÎ   ‚ Y n: t	tj
f�y } ztd|› �ƒ‚W Y d }~n
d }~0 0 d|v �rZz| |d ¡ W n4 t�yX } ztd|› �ƒ‚W Y d }~n
d }~0 0 |j| |  dd¡|  dd¡|d�S )NÚcontextÚssl_versionÚSSLKEYLOGFILEÚ	cert_reqsÚca_certsÚca_cert_path)ÚcafileÚcapathz#SSL CA certificate loading failed: Úload_default_certsz(SSL default certificate loading failed: ÚcertfileÚkeyfiler4   z'SSL client certificate loading failed: Úcheck_hostnameFTÚciphersz!SSL cipher configuration failed: Ú
cert_chainrQ   z@cert_chain must be a tuple/list of (certfile, keyfile, password)z-SSL client certificate configuration failed: Z
ecdh_curvez%SSL ECDH curve configuration failed: Údo_handshake_on_connectÚsuppress_ragged_eofs)rm   rn   Úserver_hostname)r%   r   Ú
SSLContextÚPROTOCOL_TLS_CLIENTÚosÚenvironÚkeylog_filenameÚ	CERT_NONEÚload_verify_locationsÚFileNotFoundErrorÚSSLErrorÚ
ValueErrorr   Úhasattrrg   ÚPurposeÚSERVER_AUTHÚload_cert_chainrj   Úverify_modeÚCERT_REQUIREDÚset_ciphersÚ
isinstanceÚtupleÚlistÚlenÚset_ecdh_curveÚwrap_socket)rA   r<   r>   rj   r_   Zkeylog_filere   rf   rP   rl   rh   ri   r4   r   r   r   Ú_wrap_sni_socketî   sŒ    &ÿ

ý"$ÿ

$
ÿ
ÿ
$

ür‡   )rA   Úuser_ssloptc                 C   s¤   dt ji}| |¡ tj d¡}|rJtj |¡rJ| dd ¡d u rJ||d< n(|rrtj |¡rr| dd ¡d u rr||d< | dd ¡r†|d }| dd¡}t	| |||ƒ} | S )Nrb   ZWEBSOCKET_CLIENT_CA_BUNDLErc   rd   ro   rj   T)
r   r   Úupdaterr   rs   r%   ÚpathÚisfileÚisdirr‡   )rA   rˆ   r>   r<   Z	cert_pathrj   r   r   r   r;   C  s,    

ÿ
þý
ÿ
þýr;   )rA   r2   r0   c           
   
   C   s  t dƒ d|› d|› d�}|d|› d|› d�7 }|rŠ|d rŠ|d }|d r^|d|d › �7 }t| ¡ ƒ ¡  ¡  d	d
¡}|d|› d�7 }|d7 }td|ƒ t| |ƒ zt| ƒ\}}}W n6 t	j
tfyî }	 ztt|	ƒƒ‚W Y d }	~	n
d }	~	0 0 |dk�rtd|› �ƒ‚| S )NzConnecting proxy...zCONNECT rC   z HTTP/1.1
zHost: z
r   r   Ú
Ú zProxy-Authorization: Basic zrequest headeréÈ   z!failed CONNECT via proxy status: )r   Úbase64encodeÚencodeÚstripÚdecodeÚreplacer   r   r   rI   rU   r   r   rV   )
rA   r1   r2   r(   Zconnect_headerÚauth_strZencoded_strÚstatusÚ_rP   r   r   r   rG   ^  s&    

"
rG   )rA   r0   c           
      C   s  d }d }i }t dƒ t| ƒ}| d¡ ¡ }|s0qðt |ƒ |sj| dd¡}t|d ƒ}t|ƒdkrî|d }q| dd¡}t|ƒdkrŠtdƒ‚|\}}| ¡ dkrÞ| 	d¡rÞ| 	d¡}	|	d urÐ|	d	 | ¡  |d< qî| ¡ |d< q| ¡ || ¡ < qt d
ƒ |||fS )Nz--- response header ---zutf-8ú é   r   rC   zInvalid headerz
set-cookiez; z-----------------------)
r	   r   r“   r’   Úsplitr:   r„   r   Úlowerr%   )
rA   r–   Zstatus_messageÚheadersÚlineZstatus_infoÚkvÚkeyÚvalueZexisting_cookier   r   r   r   z  s4    

r   )2Ú__doc__rW   rr   rI   Úbase64r   r�   Ú_exceptionsr   r   r   Z_loggingr   r   r	   Ú_socketr
   r   r   Z_ssl_compatr   r   Ú_urlr   r   Ú__all__Zpython_socks._errorsr   r   r   Zpython_socks._typesr   Zpython_socks.syncr   r8   Ú	Exceptionr   rV   r‚   rB   r   r:   ÚboolrD   rE   Údictr‡   r;   rG   r   r   r   r   r   Ú<module>   s:   +% +U