a h= @s>UddlmZddlZddlmZmZddlmZmZddl Z ddl m Z ddl mZmZddlmZmZerdd lmZdd lmZdd lmZd ZejejhZed dddZeGdddeZgdZe Z!de"d<eD]>Z#ee$ e!%e&ee#Wdq1s0YqeGdddeeZ'dS)) annotationsN)contextmanagersuppress) TYPE_CHECKINGoverload)socket)ConflictDetectorfinal)HalfCloseableStreamListener) Generator)Buffer) SocketTypeizGenerator[None, None, None]returnc cs^z dVWnNtyX}z6|jtvr2tddntd||WYd}~n d}~00dS)Nzthis socket was already closedzsocket connection broken: )OSErrorerrno_closed_stream_errnostrioClosedResourceErrorZBrokenResourceError)excr[/var/www/html/swiplay.fr/scambot/venv/lib/python3.9/site-packages/trio/_highlevel_socket.py)_translate_socket_errors_to_stream_errors#s   rc@seZdZdZdddddZdddd d Zdd d d Zdd ddZd)dddddZdd ddZ e dddddddZ e ddddddddZ d*ddddddd dZ e dddd!d"d#Z e ddddd$d%d#Z d+dddd'd$d(d#Z dS), 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. rNonerrcCst|tjstd|jtjkr(td||_td|_ t t "| tj tjdWdn1sh0Yttdrt t "| tj tjdWdn1s0YdS)Nz*SocketStream requires a Trio socket objectz*SocketStream requires a SOCK_STREAM socketz;another task is currently sending data on this SocketStreamTTCP_NOTSENT_LOWATi@) isinstancetsocketr TypeErrortype SOCK_STREAM ValueErrorrr _send_conflict_detectorrr setsockopt IPPROTO_TCP TCP_NODELAYhasattrr)selfrrrr__init__Gs   0 zSocketStream.__init__zbytes | bytearray | memoryview)datarc 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_WRrrr%r memoryviewfilenolowlevel checkpointlensend)r*r, total_sent remainingsentrrrsend_alles    ( 0zSocketStream.send_allrc st|jZ|jdkrtjt |jIdHWdn1sH0YWdn1sf0YdS)Nr-)r%rr0rrrZ wait_writabler*rrrwait_send_all_might_not_blockvs z*SocketStream.wait_send_all_might_not_blockc s|jltjIdH|jjr0WddSt|jtj Wdn1sZ0YWdn1sx0YdSN) r%rr1r2rr.rshutdownr SHUT_WRr9rrrsend_eof}s zSocketStream.send_eofNz int | Nonebytes) max_bytesrcsX|dur t}|dkrtdt"|j|IdHWdS1sJ0YdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZEr$rrrecv)r*r@rrr receive_somes zSocketStream.receive_somecs|jtjIdHdSr;rcloserr1r2r9rrracloses zSocketStream.acloseintz int | Buffer)leveloptionvaluercCsdSr;r)r*rHrIrJrrrr&szSocketStream.setsockopt)rHrIrJlengthrcCsdSr;rr*rHrIrJrKrrrr&szint | Buffer | NonecCsR|dur(|durtd|j|||S|dur@td|d|j||||S)zlSet an option on the underlying socket. See :meth:`socket.socket.setsockopt` for details. NzKinvalid value for argument 'value', must not be None when specifying lengthz$invalid value for argument 'value': z%, must be None when specifying optlen)r!rr&rLrrrr&s  )rHrIrcCsdSr;r)r*rHrIrrr getsockoptszSocketStream.getsockopt)rHrI buffersizercCsdSr;rr*rHrIrNrrrrMsrz int | bytescCs*|dkr|j||S|j|||SdS)zCheck the current value of an option on the underlying socket. See :meth:`socket.socket.getsockopt` for details. rN)rrMrOrrrrMs )N)N)r) __name__ __module__ __qualname____doc__r+r8r:r>rCrFrr&rMrrrrr.s$ r)EPERM ECONNABORTEDZEPROTOZENETDOWNZ ENOPROTOOPTZ EHOSTDOWNZENONETZ EHOSTUNREACH EOPNOTSUPPZ ENETUNREACHZENOSRZESOCKTNOSUPPORTZEPROTONOSUPPORTZ ETIMEDOUT ECONNRESETzset[int]_ignorable_accept_errnosc@s<eZdZdZdddddZddd d Zddd d Zd S)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. rrrcCsft|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) rr rr!r"r#r$rM SOL_SOCKET SO_ACCEPTCONNrr)r*rZ listeningrrrr+os   zSocketListener.__init__rrc sjz|jIdH\}}WnBtyZ}z*|jtvr:tjd|jtvrFWYd}~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) racceptrrrrrrXr)r*sock_rrrrr\s  zSocketListener.acceptcs|jtjIdHdS)z.Close this listener and its underlying socket.NrDr9rrrrFs zSocketListener.acloseN)rPrQrRrSr+r\rFrrrrrY]srY)( __future__rr contextlibrrtypingrrrrr Z_utilr r abcr r collections.abcr Ztyping_extensionsr_socketrrAEBADFENOTSOCKrrrZ_ignorable_accept_errno_namessetrX__annotations__nameAttributeErroraddgetattrrYrrrrs8      2