o ACe= @s8UddlmZddlZddlmZddlmZmZddlm Z m Z ddl Z ddl m ZddlmZmZdd lmZmZe rIdd lmZdd lmZd ZejejhZedddZeGdddeZgdZe Z!de"d<eD]Z#ee$e!%e&ee#Wdn1swYqoeGdddeeZ'dS)) annotationsN) Generator)contextmanagersuppress) TYPE_CHECKINGoverload)socket)ConflictDetectorfinal)HalfCloseableStreamListener)Buffer) SocketTypeireturnGenerator[None, None, None]c csPzdVWdSty'}z|jtvrtddtd||d}~ww)Nzthis socket was already closedzsocket connection broken: )OSErrorerrno_closed_stream_errnostrioClosedResourceErrorZBrokenResourceError)excr:usr/lib/python3.10/site-packages/trio/_highlevel_socket.py)_translate_socket_errors_to_stream_errors"s   rc@seZdZdZd*ddZd+d d Zd,d d Zd,ddZd-d.ddZd,ddZ e d/ddZ e d0d dZ d-d1d"dZ e d2d#d$Z e d3d&d$Z d4d5d)d$Z dS)6 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. r rcCst|tjs td|jtjkrtd||_td|_ t t | tj tjdWdn1s4wYttdr^t t | tj tjdWddS1sWwYdSdS)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 ValueErrorr r _send_conflict_detectorrr setsockopt IPPROTO_TCP TCP_NODELAYhasattrr)selfr rrr__init__Fs     "zSocketStream.__init__databytes | bytearray | memoryviewrNonec sV|jjr td|jtxt|d}|sB|jdkr%tdtj IdH WdWdWddSd}|t |krs||d}|j |IdH}Wdn1sdwY||7}|t |ksJWdn1s}wYWdn1swYWddSWddS1swYdS)Nz!can't send data after sending EOFzsocket was already closedr) r did_shutdown_SHUT_WRrrr#r memoryviewfilenolowlevel checkpointlensend)r(r* total_sent remainingsentrrrsend_allds8      "zSocketStream.send_allc s|j6|jdkrtjt|jIdHWdn1s%wYWddSWddS1s=wYdS)Nr-)r#r r0rrrZ wait_writabler(rrrwait_send_all_might_not_blockus"z*SocketStream.wait_send_all_might_not_blockc s|j@tjIdH|jjr WddSt|jtj Wdn1s/wYWddSWddS1sGwYdSN) r#rr1r2r r.rshutdownrSHUT_WRr9rrrsend_eof|s"zSocketStream.send_eofN max_bytes int | NonebytescsZ|durt}|dkrtdt|j|IdHWdS1s&wYdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZEr"rr recv)r(r?rrr receive_somes$zSocketStream.receive_somecs |jtjIdHdSr;r closerr1r2r9rrracloses zSocketStream.acloselevelintoptionvalue int | BuffercCdSr;r)r(rHrJrKrrrr$zSocketStream.setsockoptlengthcCrMr;rr(rHrJrKrOrrrr$rNint | Buffer | NonecCsR|dur|dur td|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)rr r$rPrrrr$s  cCrMr;r)r(rHrJrrr getsockoptrNzSocketStream.getsockopt buffersizecCrMr;rr(rHrJrSrrrrRrNr int | bytescCs&|dkr |j||S|j|||S)zCheck the current value of an option on the underlying socket. See :meth:`socket.socket.getsockopt` for details. r)r rRrTrrrrRs r r)r*r+rr,rr,r;)r?r@rrA)rHrIrJrIrKrLrr,) rHrIrJrIrKr,rOrIrr,) rHrIrJrIrKrQrOr@rr,)rHrIrJrIrrI)rHrIrJrIrSrIrrA)r)rHrIrJrIrSrIrrU) __name__ __module__ __qualname____doc__r)r8r:r>rDrGrr$rRrrrrr-s&          r)EPERMZ ECONNABORTEDZEPROTOZENETDOWNZ ENOPROTOOPTZ EHOSTDOWNZENONETZ EHOSTUNREACH EOPNOTSUPPZ ENETUNREACHZENOSRZESOCKTNOSUPPORTZEPROTONOSUPPORTZ ETIMEDOUTZ ECONNRESETzset[int]_ignorable_accept_errnosc@s.eZdZdZdddZddd Zdd 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. r rcCsft|tjs td|jtjkrtdz |tjtj }Wn t y'Ynw|s.td||_ dS)Nz,SocketListener requires a Trio socket objectz,SocketListener requires a SOCK_STREAM socketz*SocketListener requires a listening socket) rrrrr r!r"rR SOL_SOCKET SO_ACCEPTCONNrr )r(r Z listeningrrrr)rs    zSocketListener.__init__rrc sh z|jIdH\}}Wt|Sty2}z|jtvr"tjd|jtvr(WYd}~nd}~wwq)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``. TN) r acceptrrrrrr^r)r(sock_rrrrrbs   zSocketListener.acceptr,cs |jtjIdHdS)z.Close this listener and its underlying socket.NrEr9rrrrGs zSocketListener.acloseNrV)rrrW)rXrYrZr[r)rbrGrrrrr_`s   r_)rr)( __future__rrcollections.abcr contextlibrrtypingrrrr rZ_utilr r abcr r typing_extensionsr_socketrrBEBADFENOTSOCKrrrZ_ignorable_accept_errno_namessetr^__annotations__nameAttributeErroraddgetattrr_rrrrs>