LqjVJddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z mZmZddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZm Z dd l!m"Z"m#Z#ddl$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*ddl m+Z+m,Z,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3ddl4m5Z5gdZ6Gdde3Z7ddddddddedddddddddddd d0dZ8 d1 d2d Z9 dd!l:m;Z;dd"l   .26(2 $ ...&. .  .8 ) % %r;r'deflateTr*r)ir+)r7sslserver_hostnameorigin extensions subprotocols compressionrGrHproxy proxy_sslproxy_server_hostname open_timeoutr-r.r/max_sizer0loggercreate_connectionc |/d|vr+|jd}tjdtt |}|j s | t d| t||dk(r t|}n|t d||tjd}|t}|jdd }|jd d}|r| | t d | | t d |rd} |d} | d ur t|} t| } |Q|rgtjtjtj }|j#|j%|J|j'|n| t)| }|j*dddk(r t-||||jdd}n|j*dddk(r/|j*dk7r | t dt/|||f| | | d|}nWt1d|j3d|j%tj4|j6|j8ffi|}|j#d|s/|j;tj<tj>d |j r|tAjB}| |j6}|j#|j%| |jE||}n(tG|||}tItj|}|j#dtK||||||}|||||||} |jQ|| |j%|jY|S#tL$r||jOwxYw#tL$r,|jS|jTjWwxYw)a Connect to the WebSocket server at ``uri``. This function returns a :class:`ClientConnection` instance, which you can use to send and receive messages. :func:`connect` may be used as a context manager:: from websockets.sync.client import connect with connect(...) as websocket: ... The connection is closed automatically when exiting the context. Args: uri: URI of the WebSocket server. sock: Preexisting TCP socket. ``sock`` overrides the host and port from ``uri``. You may call :func:`socket.create_connection` to create a suitable TCP socket. ssl: Configuration for enabling TLS on the connection. server_hostname: Host name for the TLS handshake. ``server_hostname`` overrides the host name from ``uri``. origin: Value of the ``Origin`` header, for servers that require it. extensions: List of supported extensions, in order in which they should be negotiated and run. subprotocols: List of supported subprotocols, in order of decreasing preference. compression: The "permessage-deflate" extension is enabled by default. Set ``compression`` to :obj:`None` to disable it. See the :doc:`compression guide <../../topics/compression>` for details. additional_headers: Arbitrary HTTP headers to add to the handshake request. user_agent_header: Value of the ``User-Agent`` request header. It defaults to ``"Python/x.y.z websockets/X.Y"``. Setting it to :obj:`None` removes the header. proxy: If a proxy is configured, it is used by default. Set ``proxy`` to :obj:`None` to disable the proxy or to the address of a proxy to override the system configuration. See the :doc:`proxy docs <../../topics/proxies>` for details. proxy_ssl: Configuration for enabling TLS on the proxy connection. proxy_server_hostname: Host name for the TLS handshake with the proxy. ``proxy_server_hostname`` overrides the host name from ``proxy``. open_timeout: Timeout for opening the connection in seconds. :obj:`None` disables the timeout. ping_interval: Interval between keepalive pings in seconds. :obj:`None` disables keepalive. ping_timeout: Timeout for keepalive pings in seconds. :obj:`None` disables timeouts. close_timeout: Timeout for closing the connection in seconds. :obj:`None` disables the timeout. max_size: Maximum size of incoming messages in bytes. :obj:`None` disables the limit. You may pass a ``(max_message_size, max_fragment_size)`` tuple to set different limits for messages and fragments when you expect long messages sent in short fragments. max_queue: High-water mark of the buffer where frames are received. It defaults to 16 frames. The low-water mark defaults to ``max_queue // 4``. You may pass a ``(high, low)`` tuple to set the high-water and low-water marks. If you want to disable flow control entirely, you may set it to ``None``, although that's a bad idea. logger: Logger for this client. It defaults to ``logging.getLogger("websockets.client")``. See the :doc:`logging guide <../../topics/logging>` for details. create_connection: Factory for the :class:`ClientConnection` managing the connection. Set it to a wrapper or a subclass to customize connection handling. Any other keyword arguments are passed to :func:`~socket.create_connection`. Raises: InvalidURI: If ``uri`` isn't a valid WebSocket URI. InvalidProxy: If ``proxy`` isn't a valid proxy. OSError: If the TCP connection fails. InvalidHandshake: If the opening handshake fails. TimeoutError: If the opening handshake times out. N ssl_contextzssl_context was renamed to sslz-ssl argument is incompatible with a ws:// URIr_zunsupported compression: zwebsockets.clientunixFpathzmissing path argumentzpath is incompatible with sockTsockssource_address) local_addrhttphttpsz8proxy_ssl argument is incompatible with an http:// proxyrHr`raz&parse_proxy returned unsupported proxyrIra)rbrcrdrjrkr,)-popwarningswarnDeprecationWarningr!secure ValueErrorrrlogging getLoggerr'rr$socketAF_UNIX SOCK_STREAM settimeoutrIr%rschemeconnect_socks_proxyconnect_http_proxyAssertionErrorrArlhostport setsockopt IPPROTO_TCP TCP_NODELAY ssl_modulecreate_default_context wrap_socket SSLSSLSocketr r ExceptioncloserJ close_socketrecv_events_threadjoinstart_keepalive)urir7r`rarbrcrdrerGrHrfrgrhrir-r.r/rjr0rkrlkwargsws_urirorpdeadline proxy_parsedsock_2r8 connections r:r%r%sd {}.jj' ,  s^F ==S_HIIl+i5jA  4[MBCC ~""#67 ,FE*Dzz&$/D  >   }&!  %H` <}}V^^V5G5GH 0 0 23''' T""*51 &&r*g5.$ $*::.>#E D"((!,6#**g5):O(V.$ +<%(= !D))QRR!!)X-=-=-?@//[[&++. OOD ! OOF..0B0BD I =={ 779&"(++ OOH,,. / t_M%dCQFMM62 OOD !" !%  '  '%'           #    JJL  !%%**, s IN !N N 5Oc r|'|jd|jdd}nd}td|d|d|S)a Connect to a WebSocket server listening on a Unix socket. This function accepts the same keyword arguments as :func:`connect`. It's only available on Unix. It's mainly useful for debugging servers listening on Unix sockets. Args: path: File system path to the Unix socket. uri: URI of the WebSocket server. ``uri`` defaults to ``ws://localhost/`` or, when a ``ssl`` is provided, to ``wss://localhost/``. r`rnzws://localhost/zwss://localhost/T)rrorp)getr%)rprrs r:r&r&sG* { ::e  $M)B)J#C$C  ;sD ;F ;;r;) ProxyType)r)socks5hsocks5socks4asocks4Fc tt|j|j|j|j |j t|j}|jd|j |j|j|jfi|S#tttjf$rt$r}td|d}~wwxYw)z0Connect via a SOCKS proxy and return the socket.rIz failed to connect to SOCKS proxyN) SocksProxySOCKS_PROXY_TYPESrrrusernamepasswordSOCKS_PROXY_RDNSrArIr%OSErrorrErrr)rfrrr socks_proxyexcs r:rrs! ell + JJ JJ NN NN U\\ *   )X%5%5%78 J&;&&v{{FKKJ6J Jv~~6   J?@c I Js<'B$$'C CCc td)Nz6connecting through a SOCKS proxy requires python-socks) ImportError)rfrrrs r:rrs RSSr;ct}tj|j|j|j d} |j |j|jd}|r|j|n|jt|`#t$rr}t|jtsJ|j}d|jcxkrdkr%n t!||cYd}~|j dSt!|d}~wt"j$r t%dt&$r}t)d|d}~wwxYw#|j dwxYw)NT)rfii,z(timed out while connecting to HTTP proxyz0did not receive a valid HTTP response from proxy)rrparse read_line read_exact read_to_eofrrIrecv feed_datafeed_eofnext StopIterationrMvalue status_coderrrErr )r7rreaderparserdatarrLs r:read_connect_responsers; ^F ^^ F  OOH,,. /99T?D  &! L /#))X...99 (&& , ,%X. .O %X. . >>GEFF ! >   sDA!B$$ E-A D8E9E D)E EEEE*rxc |jd|jtj|j|j ffi|}|j dk(rg|tj}| |j}|j|j|j||}|jdt|||}|j| t|||S#t$r|jwxYw)NrIrwry)rArIrrlrrrrrrrrsendallrrr) rfrrrHr`rarr7r>s r:rr s i!1!1!34  # #UZZ$< G GD ||w ;335C  "#jjO ((*+t_E &eV5FGGLLdH- K   s " C00D TF.)boundcVeZdZdZdZ d d dZd dZd dZddZddZ dd Z y)rzr Socket-like object providing TLS-in-TLS. Only methods that are used by websockets are implemented. iNctj|_tj|_||_|j |j|j||_|j|j jy)Nry) r MemoryBIOincomingoutgoing ssl_socketwrap_bio ssl_objectrun_io do_handshake)r6r7rnras r:r5zSSLSSLSocket.__init__Ask #,,. ",,. %.. MM MM+/  DOO001r;c d}d} ||}|jj }|r|j j ||r^|j j|j}|r|jj|n|jj|rS#tj$rd}Ytj$rd}YwxYw)NTF) rSSLWantReadErrorSSLWantWriteErrorrreadrrr recv_bufsizerwrite write_eof)r6funcargs want_read want_writeresultrs r:rzSSLSSLSocket.run_ioQsIJ "t==%%'D''-++D,=,=>MM''-MM++-M-.. ! // "!  "sB((C?CCc |j|jj|S#tj$rYywxYw)Nr;)rrrr SSLEOFError)r6buflens r:rzSSLSSLSocket.recvos: ;;t33V< <%%  s %(>>cN|j|jj|SN)rrr)r6rs r:sendzSSLSSLSocket.sendus{{4??00$77r;cd}t|5}|jd5}t|}||kr||j||dz }||krddddddy#1swYxYw#1swYyxYw)NrB) memoryviewr lenr)r6rcountview byte_viewamounts r:rzSSLSSLSocket.sendallxsz   6tyy~ 6^F&.9UV#455&. 6 6 6 6 6 6s"A+-AA+A( $A++A4c.t|j|Sr)getattrr)r6names r: __getattr__zSSLSSLSocket.__getattr__st--r;r)r7rSrnzssl_module.SSLContextrarYrVrW)rzCallable[..., T]rrrVr)rintrVbytes)rrrVr)rrrVrW)rstrrVr) rZr[r\r]rr5rrrrrrr;r:rr7sWL '+ 22+2$ 2  2 < 86.r;r).rrr7zsocket.socket | Noner`ssl_module.SSLContext | NonerarYrbz Origin | Nonercz'Sequence[ClientExtensionFactory] | NonerdzSequence[Subprotocol] | NonererYrGrXrHrYrfzstr | Literal[True] | NonergrrhrYrirTr-rTr.rTr/rTrjrUr0rUrkzLoggerLike | Nonerlztype[ClientConnection] | NonerrrVr')NN)rprYrrYrrrVr') rfrrr rr$rrrVrS)r7rSrr$rVr)rfrrr rr$rHrYr`rrarYrrrVrS)I __future__rrrr`rr2r{collections.abcrtypingrrrrr clientr datastructuresr exceptionsr rrextensions.baserextensions.permessage_deflaterr?rhttp11rrr8rrrfrrrrstreamsrrrrrrr r!rr#utilsr$__all__r'r%r& python_socksrpython_socks.syncrSOCKS5SOCKS4rrrrrrrrrrr;r:rsA" $88#(LL4L+)(JJ"??)" :b%zb%R"&(,"& :>15'-1$.(,.2(,!#"$!#"$;@<> $7;;O O  O & O  O O8O/OO+O"O &O ,!O"&#O&'O( )O*+O, -O091O2:3O6 7O:5;O>?O@AOf< < << <<4J&5##""##""  JJJJ J  J4N%)(,"&$ $ $$ " $ & $ $$$N CL CxQ'(M.M.ATTTTT T  TTs> F F"!F"