U
    g2                     @   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 d
ddgZz(ddlT 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X 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 2024 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)*)	ProxyType)ProxyTFc                   @   s   e Zd ZdS )
ProxyErrorN__name__
__module____qualname__ r   r   3/tmp/pip-unpacked-wheel-0xk_xrpx/websocket/_http.pyr   -   s   r   c                   @   s   e Zd ZdS )ProxyTimeoutErrorNr   r   r   r   r   r   0   s   r   c                   @   s   e Zd ZdS )ProxyConnectionErrorNr   r   r   r   r   r   3   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	krt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__8   s    
zproxy_info.__init__N)r   r   r   r/   r   r   r   r   r   7   s   )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}
|rtrt|
|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)r0   r.   proxyhostnamer3   resource	is_securer6   r    Zws_proxysockr   r   r   _start_proxied_socketR   s>    



	rC   )r0   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}|	rt	||||
}|rt
rt||j|}ntd||||ffW S    |r|   Y nX d S )Nr!   zHost not found.: :r8   )r'   r+   rC   r   _get_addrinfo_listr   _open_socketsockoptr7   _tunnelr   r<   r=   close)r0   r.   r>   socketr?   Zport_from_urlr@   rA   addrinfo_listZneed_tunnelr)   rB   r   r   r   r   }   s6       
)r3   rA   r1   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	k
r } zt
|W 5 d }~X Y nX d S )Nr   FP   T)r   r'   r(   r)   r*   rJ   getaddrinfoSOCK_STREAMSOL_TCPgaierrorr   )	r?   r3   rA   r>   ZphostZpportZpauthrK   er   r   r   rE      s:    
        rE   c                 C   s*  d }| D ]}|d d \}}}t  |||}|| tD ]}	|j|	  q<|D ]}	|j|	  qP|d }
d }|sz||
 W n t jk
r
 } zj|  t|
d |_zt	j
t	jt	jf}W n  tk
r   t	j
t	jf}Y nX |j	|kr||}W Y qlW 5 d }~X Y qlX qqlq q&q|r&||S )N      r   )rJ   
settimeoutr
   
setsockoptr   errorrI   strZ	remote_iperrnoZECONNREFUSEDZWSAECONNREFUSEDZENETUNREACHAttributeError)rK   rG   r7   errZaddrinfofamilysocktypeprotorB   optsaddressrV   ZeConnRefusedr   r   r   rF      sD    


rF   )rB   r=   c           
      C   s  | dd }|sdt| dtj}tj dd |_| dtjtjkr| dd }| dd }|sj|rz|j||d nt	|dr|
tjj | d	d r||d	 | d
d | dd  | dtjtjkr| ddsd|_tj|_n| dd|_| dtj|_d|kr&||d  d|krL|d \}}}	||||	 d|krd||d  |j| | dd| dd|dS )Ncontextssl_versionSSLKEYLOGFILE	cert_reqsca_certsca_cert_path)cafilecapathload_default_certscertfilekeyfiler5   check_hostnameFTciphersZ
cert_chainZ
ecdh_curve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hasattrrh   PurposeSERVER_AUTHload_cert_chainrk   verify_modeCERT_REQUIREDset_ciphersset_ecdh_curvewrap_socket)
rB   r=   r?   rk   r`   rf   rg   ri   rj   r5   r   r   r   _wrap_sni_socket   sL    


	 





r   )rB   user_ssloptc                 C   s   dt ji}|| tjd}|rJtj|rJ|dd d krJ||d< n(|rrtj|rr|dd d krr||d< |dd r|d }|dd}t	| |||} | S )Nrc   ZWEBSOCKET_CLIENT_CA_BUNDLErd   re   ro   rk   T)
r   r|   updaterr   rs   r&   pathisfileisdirr   )rB   r   r?   r=   Z	cert_pathrk   r   r   r   r<   !  s,    




r<   )rB   r3   r1   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 n. t	k
r }	 zt
t|	W 5 d }	~	X Y nX |dkrt
d| | S )NzConnecting proxy...zCONNECT rD   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   	Exceptionr   rW   )
rB   r2   r3   r)   Zconnect_headerauth_strZencoded_strstatus_rQ   r   r   r   rH   <  s&    

rH   )rB   r1   c           	      C   s   d }d }i }t d t| }|d }|s0qt | |sj|dd}t|d }t|dkr|d }q|dd}t|dkrtd|\}}| dkr|	dr|	dd	 |  |d< q| || < qt d
 |||fS )Nz--- response header ---zutf-8    r   rD   zInvalid headerz
set-cookiez; z-----------------------)
r	   r   r   r   splitr;   lenr   lowerr&   )	rB   r   Zstatus_messageheaderslineZstatus_infokvkeyvaluer   r   r   r   X  s.    
)2__doc__rX   rr   rJ   base64r   r   _exceptionsr   r   r   Z_loggingr   r   r	   _socketr
   r   r   Z_ssl_compatr   r   _urlr   r   __all__Zpython_socks._errorsZpython_socks._typesr   Zpython_socks.syncr   r9   r   r   r   r   r   rW   tuplerC   r   r;   boolrE   rF   dictr   r<   rH   r   r   r   r   r   <module>   s:   
+% +4