
     ket1                        d Z ddlZddlZddlZddlmZ ddl ddl ddl	 ddl
 ddl g dZ	 ddl ddlmZ ddlmZ d	Z G d d      ZdedefdZdefdZdededefdZd Zdej                  de fdZ!dej                  de fdZ"dej                  dedej                  fd Z#dej                  defd!Z$y#  d
Z G d de      Z G d de      Z G d de      ZY xY w)"a_  
_http.py
websocket - WebSocket client library for Python

Copyright 2023 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   )*)
proxy_infoconnectread_headers)	ProxyType)ProxyTFc                       e Zd Zy)
ProxyErrorN__name__
__module____qualname__     1/usr/lib/python3/dist-packages/websocket/_http.pyr   r   )       r   r   c                       e Zd Zy)ProxyTimeoutErrorNr   r   r   r   r   r   ,   r   r   r   c                       e Zd Zy)ProxyConnectionErrorNr   r   r   r   r   r   /   r   r   r   c                       e Zd Zd Zy)r   c                    |j                  dd       | _        | j                  r|j                  dd      | _        |j                  dd       | _        |j                  dd       | _        |j                  dd      | _        |j                  dd       | _        | j
                  d	vrt        d
      y d| _        d | _        d | _        d| _        y )Nhttp_proxy_hosthttp_proxy_portr   http_proxy_authhttp_no_proxy
proxy_typehttp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optionss     r   __init__zproxy_info.__init__4   s    !++&7>??%kk*;Q?DO$5t<DI#KK>DM")++lF"CD!(-A4!HD"" +  !M   DODI DM"(Dr   N)r   r   r   r/   r   r   r   r   r   3   s    )r   r   urlreturnc                    t         st        d      t        |       \  }}}}|j                  dk(  rd}t        j
                  }ne|j                  dk(  rd}t        j
                  }nC|j                  dk(  rd}t        j                  }n!|j                  dk(  rd}t        j                  }t        j                  |j                  t        |j                        |j                  r|j                  d   nd |j                  r|j                  d	   nd 
      }	|	j                  |||j                        }
|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_SOCKSWebSocketException	parse_urlr+   r	   SOCKS4SOCKS5r
   creater'   intr(   r)   r   r,   HAVE_SSL_ssl_socketsslopt)r0   r.   proxyhostnamer4   resource	is_securer7   r   ws_proxysocks              r   _start_proxied_socketrJ   N   sG    L
 	
 +4C.'HdHix'%%
				*%%
				)%%
				*%%
||!!""'**A$"'**A$H HdE4G4GHDtW^^X>D$%9::(D(+++r   c                    |j                   r|s|j                  dk7  rt        | ||      S t        |       \  }}}}|r||||ffS t	        ||||      \  }}	}
|st        d| d|       d }	 t        ||j                  |j                        }|	rt        ||||
      }|r)t        rt        ||j                  |      }nt        d      ||||ffS #  |r|j                           xY w)Nr    zHost not found.: :r9   )r'   r+   rJ   r<   _get_addrinfo_listr;   _open_socketsockoptr8   _tunnelrA   rB   rC   close)r0   r.   rD   socketrE   port_from_urlrF   rG   addrinfo_listneed_tunnelr)   rI   s               r   r   r   y   s	    5+?+?6+I$S'5993<S>0HmXy-:::'9-E($M;  #4XJa!OPPDM7??GOOL4=$?D"4B()=>>hx888JJLs   ,A"C C$r4   rG   c                    t        | ||j                  |j                  |j                  |j                        \  }}}	 |s:t        j                  | |dt
        j                  t
        j                        }|dd fS |xr |xs d}t        j                  ||dt
        j                  t
        j                        }|d|fS # t
        j                  $ r}t        |      d }~ww xY w)Nr   FP   T)get_proxy_infor'   r(   r)   r*   rR   getaddrinfoSOCK_STREAMSOL_TCPgaierrorWebSocketAddressException)	rE   r4   rG   rD   phostpportpauthrT   es	            r   rM   rM      s    (

E5%+ "..$6#5#5v~~M !%--Oe)rE
 #..ua!3!3V^^M !$--?? +'**+s   ;B> :AB> >C!CC!c                    d }| D ]  }|d d \  }}}t        j                   |||      }|j                  |       t        D ]  }	 |j                  |	   |D ]  }	 |j                  |	   |d   }
d }|s	 |j	                  |
       	  |S  |r|S # t         j
                  $ r}|j                          t        |
d         |_        	 t        j                  t        j                  t        j                  f}n/# t        $ r# t        j                  t        j                  f}Y nw xY w|j                  |vr||}Y d }~d }~ww xY w)N      r   )rR   
settimeoutDEFAULT_SOCKET_OPTION
setsockoptr   errorrQ   str	remote_iperrnoECONNREFUSEDWSAECONNREFUSEDENETUNREACHAttributeError)rT   rO   r8   erraddrinfofamilysocktypeprotorI   optsaddressrh   eConnRefuseds                r   rN   rN      sc   
C! $"*2A,%}}VXu5 ) 	#DDOOT"	# 	#DDOOT"	# 1+W%"  	
 K A$F IK1 << 

"%gaj/K**--))$L
 & K$)$6$68I8I#JLK;;l2Ks<   3BE##D=/C76D=7)D# D="D##D==ErI   rC   c                    |j                  dd       }|s.t        j                  |j                  dt        j                              }t        j
                  j                  dd       |_        |j                  dt        j                        t        j                  k7  rq|j                  dd       }|j                  dd       }|s|r|j                  ||       n5t        |d      r)|j                  t        j                  j                         |j                  d	d       r6|j                  |d	   |j                  d
d       |j                  dd              |j                  dt        j                        t        j                  k(  r/|j                  dd      sd|_        t        j                  |_        n<|j                  dd      |_        |j                  dt        j                         |_        d|v r|j#                  |d          d|v r|d   \  }}}	|j                  |||	       d|v r|j%                  |d          |j'                  | |j                  dd      |j                  dd      |      S )Ncontextssl_versionSSLKEYLOGFILE	cert_reqsca_certsca_cert_path)cafilecapathload_default_certscertfilekeyfiler6   check_hostnameFTciphers
cert_chain
ecdh_curvedo_handshake_on_connectsuppress_ragged_eofs)r   r   server_hostname)r&   ssl
SSLContextPROTOCOL_TLS_CLIENTosenvironkeylog_filename	CERT_NONEload_verify_locationshasattrr   PurposeSERVER_AUTHload_cert_chainr   verify_modeCERT_REQUIREDset_ciphersset_ecdh_curvewrap_socket)
rI   rC   rE   r   ry   r   r   r   r   r6   s
             r   _wrap_sni_socketr      s   jjD)G..M3;R;R!ST
 #%**..$"G::k3==1S]]BZZ
D1FZZ5F--VF-K"67**3;;+B+BC::j$'##z"

9d+

:t, ::k3==1S]]B6::eL
 &+G""%--G%+ZZ0@$%GG""(**[#:K:K"LGy 126!*0*>'Hgx##Hgx@6!""6,#78 &

+Dd K#ZZ(>E 	   r   user_ssloptc                    t        t        j                        }|j                  |       t        j
                  j                  d      }|r7t        j                  j                  |      r|j                  dd       ||d<   n8|r6t        j                  j                  |      r|j                  dd       ||d<   |j                  dd       r|d   }|j                  dd      }t        | |||      } | S )N)r|   WEBSOCKET_CLIENT_CA_BUNDLEr}   r~   r   r   T)dictr   r   updater   r   r&   pathisfileisdirr   )rI   r   rE   rC   certPathr   s         r   rB   rB     s    #"3"34F
MM+zz~~:;HGGNN8$OOJ-5%zGGMM(#OOND19!)~zz#T*+,ZZ 0$7ND&(NCDKr   c                    t        d       d| d| d}|d| d| dz  }|rh|d   rc|d   }|d   r|d|d    z  }t        |j                               j                         j	                         j                  d	d
      }|d| dz  }|dz  }t        d|       t        | |       	 t        |       \  }}}	|dk7  rt        d|       | S # t        $ r}
t        t        |
            d }
~
ww xY w)NzConnecting proxy...zCONNECT rL   z HTTP/1.1
zHost: z
r   r   
 zProxy-Authorization: Basic zrequest header   z!failed CONNECT via proxy status: )debugbase64encodeencodestripdecodereplacedumpsendr   	ExceptionWebSocketProxyExceptionri   )rI   r3   r4   r)   connect_headerauth_strencoded_strstatusresp_headersstatus_messagera   s              r   rP   rP   8  s#   	
 vQtfM:NtfAdV400N Q77!DG9%H"8??#45;;=DDFNNtUWX7}DIIfN	>*~./;D/A,n }%(I&&RSSK  .%c!f--.s   )C 	C/C**C/c                 b   d }d }i }t        d       	 t        |       }|j                  d      j                         }|snt        |       |s4|j	                  dd      }t        |d         }t        |      dkD  r|d   }n|j	                  dd      }t        |      dk7  rt        d      |\  }}|j                         dk(  r:|j                  d      r)|j                  d      d	z   |j                         z   |d<   n!|j                         ||j                         <   t        d
       |||fS )Nz--- response header ---zutf-8    r   rL   zInvalid headerz
set-cookiez; z-----------------------)
trace	recv_liner   r   splitr@   lenr;   lowerr&   )	rI   r   r   headerslinestatus_infokvkeyvalues	            r   r   r   T  s    FNG	
#$
{{7#))+d**S!,KQ(F;!#!,QC#B2w!|()9::JCyy{l*w{{</H(/L(AD(H5;;=(X%',{{}		$' * 

#$7N**r   )%__doc__rk   r   rR   base64r   r   _exceptions_logging_socket_ssl_compat_url__all__python_socks._errorspython_socks._typesr	   python_socks.syncr
   r:   r   r   r   r   r   ri   tuplerJ   r   r@   boolrM   rN   r   r   rB   rP   r   r   r   r   <module>r      s&  $  	  .     
3&-') )6(,s (,u (,V" "J+s +t +u +@(V16== 1$ 1hfmm $ 6&-- S 6== 8+v}} + +]	
Y I y s   C   %C'