a xd8 @sddlZddlmZddlZddlmZddlmZm Z ddl m Z m Z dZ ejejhZedd ZGd d d e e d Zgd ZeZeD],ZzeeeeWqeyYq0qGddde ee d ZdS)N)contextmanager)socket)ConflictDetectorFinal)HalfCloseableStreamListeneric cs^z dVWnNtyX}z6|jtvr2tddntd||WYd}~n d}~00dS)Nzthis socket was already closedzsocket connection broken: {})OSErrorerrno_closed_stream_errnostrioClosedResourceErrorZBrokenResourceErrorformat)excrN/var/www/html/Ranjet/env/lib/python3.9/site-packages/trio/_highlevel_socket.py)_translate_socket_errors_to_stream_errorss  rc@sTeZdZdZddZddZddZdd Zdd d Zd dZ ddZ dddZ d S) SocketStreamufAn implementation of the :class:`trio.abc.HalfCloseableStream` interface based on a raw network socket. Args: socket: The Trio socket object to wrap. Must have type ``SOCK_STREAM``, and be connected. By default for TCP sockets, :class:`SocketStream` enables ``TCP_NODELAY``, and (on platforms where it's supported) enables ``TCP_NOTSENT_LOWAT`` with a reasonable buffer size (currently 16 KiB) – see `issue #72 `__ for discussion. You can of course override these defaults by calling :meth:`setsockopt`. Once a :class:`SocketStream` object is constructed, it implements the full :class:`trio.abc.HalfCloseableStream` interface. In addition, it provides a few extra features: .. attribute:: socket The Trio socket object that this stream wraps. cCst|tjstd|jtjkr(td||_td|_ z| tj tj dWnt y`Yn0ttdrz| tj tjdWnt yYn0dS)Nz*SocketStream requires a Trio socket objectz*SocketStream requires a SOCK_STREAM socketz;another task is currently sending data on this SocketStreamTTCP_NOTSENT_LOWATi@) isinstancetsocket SocketType TypeErrortype SOCK_STREAM ValueErrorrr_send_conflict_detector setsockopt IPPROTO_TCP TCP_NODELAYr hasattrr)selfrrrr__init__>s"      zSocketStream.__init__c s<|jjrtd|j tt|}|s|jdkrJtdtj IdHWdWdWddSd}|t |kr||d"}|j |IdH}Wdn1s0Y||7}qWdn1s0YWdn1s0YWdn1s.0YdS)Nz!can't send data after sending EOFzsocket was already closedr) rdid_shutdown_SHUT_WRr r rr memoryviewfilenolowlevel checkpointlensend)r!data total_sent remainingsentrrrsend_all`s    ( 0zSocketStream.send_allc st|jZ|jdkrtjt |jIdHWdn1sH0YWdn1sf0YdS)Nr#)rrr&r r rZ wait_writabler!rrrwait_send_all_might_not_blockqs z*SocketStream.wait_send_all_might_not_blockc s|jltjIdH|jjr0WddSt|jtj Wdn1sZ0YWdn1sx0YdSN) rr r'r(rr$rshutdownrSHUT_WRr0rrrsend_eofxs zSocketStream.send_eofNcsX|dur t}|dkrtdt"|j|IdHWdS1sJ0YdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZErrrrecv)r!Z max_bytesrrr receive_somes zSocketStream.receive_somecs|jtjIdHdSr2rcloser r'r(r0rrracloses zSocketStream.aclosecCs|j|||S)zlSet an option on the underlying socket. See :meth:`socket.socket.setsockopt` for details. )rr)r!leveloptionvaluerrrrszSocketStream.setsockoptrcCs*|dkr|j||S|j|||SdS)zCheck the current value of an option on the underlying socket. See :meth:`socket.socket.getsockopt` for details. rN)r getsockopt)r!r<r=Z buffersizerrrr?s zSocketStream.getsockopt)N)r) __name__ __module__ __qualname____doc__r"r/r1r5r8r;rr?rrrrr&s" r) metaclass)EPERM ECONNABORTEDZEPROTOZENETDOWNZ ENOPROTOOPTZ EHOSTDOWNZENONETZ EHOSTUNREACH EOPNOTSUPPZ ENETUNREACHZENOSRZESOCKTNOSUPPORTZEPROTONOSUPPORTZ ETIMEDOUT ECONNRESETc@s(eZdZdZddZddZddZdS) SocketListeneraA :class:`~trio.abc.Listener` that uses a listening socket to accept incoming connections as :class:`SocketStream` objects. Args: socket: The Trio socket object to wrap. Must have type ``SOCK_STREAM``, and be listening. Note that the :class:`SocketListener` "takes ownership" of the given socket; closing the :class:`SocketListener` will also close the socket. .. attribute:: socket The Trio socket object that this stream wraps. cCsft|tjstd|jtjkr(tdz|tjtj }Wnt yNYn0|s\td||_ dS)Nz,SocketListener requires a Trio socket objectz,SocketListener requires a SOCK_STREAM socketz*SocketListener requires a listening socket) rrrrrrrr? SOL_SOCKET SO_ACCEPTCONNr r)r!rZ listeningrrrr"Os   zSocketListener.__init__c shz|jIdH\}}Wn@tyX}z(|jtvr8tj|jtvrDWYd}~qd}~00t|SqdS)a#Accept an incoming connection. Returns: :class:`SocketStream` Raises: OSError: if the underlying call to ``accept`` raises an unexpected error. ClosedResourceError: if you already closed the socket. This method handles routine errors like ``ECONNABORTED``, but passes other errors on to its caller. In particular, it does *not* make any special effort to handle resource exhaustion errors like ``EMFILE``, ``ENFILE``, ``ENOBUFS``, ``ENOMEM``. N) racceptr r r r r _ignorable_accept_errnosr)r!sock_rrrrrL_s  zSocketListener.acceptcs|jtjIdHdS)z.Close this listener and its underlying socket.Nr9r0rrrr;{s zSocketListener.acloseN)r@rArBrCr"rLr;rrrrrI>srI)r contextlibrr rrZ_utilrrabcrrr6EBADFENOTSOCKr rrZ_ignorable_accept_errno_namessetrMnameaddgetattrAttributeErrorrIrrrrs*   z