a (j> @sdUddlmZddlZddlmZmZddlmZmZddl Z ddl m Z ddl mZmZddlmZmZerdd lmZdd lmZddlZejd krdd lmZnerdd lmZd ZejejhZeddddZeGdddeZ gdZ!e"Z#de$d<e!D]@Z%ee& e#'e(ee%Wdn1s:0YqeGdddee Z)dS)) annotationsN)contextmanagersuppress) TYPE_CHECKINGoverload)socket)ConflictDetectorfinal)HalfCloseableStreamListener) Generator) SocketType) )BufferizGenerator[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`__ 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__Os   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_allms    ( 0zSocketStream.send_allrc st|jZ|jdkrtjt |jIdHWdn1sH0YWdn1sf0YdS)Nr/)r'rr2rrrZ wait_writabler,rrrwait_send_all_might_not_block~s z*SocketStream.wait_send_all_might_not_blockc s|jltjIdH|jjr0WddSt|jtj Wdn1sZ0YWdn1sx0YdSN) r'rr3r4rr0rshutdownr"SHUT_WRr;rrrsend_eofs zSocketStream.send_eofNz int | Nonebytes) max_bytesrcsX|dur t}|dkrtdt"|j|IdHWdS1sJ0YdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZEr&rrrecv)r,rBrrr receive_somes zSocketStream.receive_somecs|jtjIdHdSr=rcloserr3r4r;rrracloses zSocketStream.acloseintz int | Buffer)leveloptionvaluercCsdSr=r)r,rJrKrLrrrr(szSocketStream.setsockopt)rJrKrLlengthrcCsdSr=rr,rJrKrLrMrrrr(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(rNrrrr(s  )rJrKrcCsdSr=r)r,rJrKrrr getsockoptszSocketStream.getsockopt)rJrK buffersizercCsdSr=rr,rJrKrPrrrrOsrz 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)rrOrQrrrrOs )N)N)r) __name__ __module__ __qualname____doc__r-r:r<r@rErHrr(rOrrrrr6s$ r)ZEPERM ECONNABORTEDZEPROTOZENETDOWN 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) r!r"rr#r$r%r&rO SOL_SOCKET SO_ACCEPTCONNrr)r,rZ listeningrrrr-xs   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) racceptrrrrrrZr)r,sock_rrrrr^s  zSocketListener.acceptcs|jtjIdHdS)z.Close this listener and its underlying socket.NrFr;rrrrHs zSocketListener.acloseN)rRrSrTrUr-r^rHrrrrr[fsr[)* __future__rr contextlibrrtypingrrrrr"Z_utilr r abcr r collections.abcr _socketrsys version_inforZtyping_extensionsrCEBADFENOTSOCKrrrZ_ignorable_accept_errno_namessetrZ__annotations__nameAttributeErroraddgetattrr[rrrrs@       4