U gu=@sBUddlmZddlZddlmZmZddlmZmZddl Z ddl m Z ddl mZmZddlmZmZerdd lmZdd lmZdd lmZd ZejejhZed dddZeGdddeZdddddddddddddd d!gZe Z!d"e"d#<eD](Z#ee$e!%e&ee#W5QRXqeGd$d%d%eeZ'dS)&) annotationsN)contextmanagersuppress) TYPE_CHECKINGoverloadsocket)ConflictDetectorfinal)HalfCloseableStreamListener) Generator)Buffer) SocketTypeizGenerator[(None, None, None)]returnc cs\z dVWnLtk rV}z.|jtkr4tddntd||W5d}~XYnXdS)Nzthis socket was already closedzsocket connection broken: )OSErrorerrno_closed_stream_errnostrioClosedResourceErrorZBrokenResourceError)excr:/tmp/pip-unpacked-wheel-ks04xdmi/trio/_highlevel_socket.py)_translate_socket_errors_to_stream_errors#s   rc@seZdZdZddddZdddd 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. rrc Cst|tjstd|jtjkr(td||_td|_ t t | tj tjdW5QRXttdrt t | tj tjdW5QRXdS)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__Gs    zSocketStream.__init__zbytes | bytearray | memoryviewNone)datarc s|jjrtd|jtt|}|sz|jdkrHtdtj IdHW5QRW5QRW5QRdSd}|t |kr||d}|j |IdH}W5QRX||7}q~W5QRXW5QRXW5QRXdS)Nz!can't send data after sending EOFzsocket was already closedr) r did_shutdown_SHUT_WRrrr$r memoryviewfilenolowlevel checkpointlensend)r)r, total_sent remainingsentrrrsend_alles   " zSocketStream.send_allrc sL|j<|jdkrtjt|jIdHW5QRXW5QRXdS)Nr-)r$r r0rrrZ wait_writabler)rrrwait_send_all_might_not_blockvs z*SocketStream.wait_send_all_might_not_blockc s\|jLtjIdH|jjr.W5QRdSt|jtj W5QRXW5QRXdSN) r$rr1r2r r.rshutdownrSHUT_WRr9rrrsend_eof}s zSocketStream.send_eofNz int | Nonebytes) max_bytesrc sL|dkr t}|dkrtdt |j|IdHW5QRSQRXdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZEr#rr recv)r)r@rrr receive_somes zSocketStream.receive_somecs|jtjIdHdSr;r closerr1r2r9rrracloses zSocketStream.acloseintz int | Buffer)leveloptionvaluercCsdSr;r)r)rHrIrJrrrr%szSocketStream.setsockopt)rHrIrJlengthrcCsdSr;rr)rHrIrJrKrrrr%szint | Buffer | NonecCsR|dkr(|dkrtd|j|||S|dk r@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 r r%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)r rMrOrrrrMs )N)N)r) __name__ __module__ __qualname____doc__r*r8r:r>rCrFrr%rMrrrrr.s$ rEPERM ECONNABORTEDZEPROTOZENETDOWNZ ENOPROTOOPTZ EHOSTDOWNZENONETZ EHOSTUNREACH EOPNOTSUPPZ ENETUNREACHZENOSRZESOCKTNOSUPPORTZEPROTONOSUPPORTZ ETIMEDOUT ECONNRESETzset[int]_ignorable_accept_errnosc@s:eZdZdZddddZdddd Zd dd 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. rrcCsht|tjstd|jtjkr(tdz|tjtj }Wnt k rPYnX|s^td||_ dS)Nz,SocketListener requires a Trio socket objectz,SocketListener requires a SOCK_STREAM socketz*SocketListener requires a listening socket) rrrr r!r"r#rM SOL_SOCKET SO_ACCEPTCONNrr )r)r Z listeningrrrr*os  zSocketListener.__init__rrc shz|jIdH\}}Wn@tk rX}z"|jtkrsX