o Oh> @s\UddlmZddlZddlmZmZddlmZmZddl Z ddl m Z ddl mZmZddlmZmZerCdd lmZdd lmZddlZejd krSdd lmZner[dd lmZd ZejejhZedddZeGdddeZ gdZ!e"Z#de$d<e!D]Z%ee&e#'e(ee%Wdn1swYqeGdddee Z)dS)) annotationsN)contextmanagersuppress) TYPE_CHECKINGoverload)socket)ConflictDetectorfinal)HalfCloseableStreamListener) Generator) SocketType) )BufferireturnGenerator[None, None, None]c csPzdVWdSty'}z|jtvrtddtd||d}~ww)Nzthis socket was already closedzsocket connection broken: )OSErrorerrno_closed_stream_errnostrioClosedResourceErrorBrokenResourceError)excrS/home/air/yokohama/back/venv/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. rrrNonecCst|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 ValueErrorrr _send_conflict_detectorrr setsockopt IPPROTO_TCP TCP_NODELAYhasattrr )selfrrrr__init__Os     "zSocketStream.__init__databytes | bytearray | memoryviewc 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) rdid_shutdown_SHUT_WRrrr'r memoryviewfilenolowlevel checkpointlensend)r,r. total_sent remainingsentrrrsend_allms8      "zSocketStream.send_allc s|j6|jdkrtjt|jIdHWdn1s%wYWddSWddS1s=wYdS)Nr0)r'rr3rrr wait_writabler,rrrwait_send_all_might_not_block~s"z*SocketStream.wait_send_all_might_not_blockc s|j@tjIdH|jjr WddSt|jtj Wdn1s/wYWddSWddS1sGwYdSN) r'rr4r5rr1rshutdownr"SHUT_WRr=rrrsend_eofs"zSocketStream.send_eofN max_bytes int | NonebytescsZ|durt}|dkrtdt|j|IdHWdS1s&wYdS)Nrzmax_bytes must be >= 1)DEFAULT_RECEIVE_SIZEr&rrrecv)r,rCrrr receive_somes$zSocketStream.receive_somecs |jtjIdHdSr?rcloserr4r5r=rrracloses zSocketStream.acloselevelintoptionvalue int | BuffercCdSr?r)r,rLrNrOrrrr(zSocketStream.setsockoptlengthcCrQr?rr,rLrNrOrSrrrr(rRint | 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)r#rr(rTrrrr(s  cCrQr?r)r,rLrNrrr getsockoptrRzSocketStream.getsockopt buffersizecCrQr?rr,rLrNrWrrrrVrRr 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)rrVrXrrrrVs rrrr)r.r/rrrrr?)rCrDrrE)rLrMrNrMrOrPrr) rLrMrNrMrOrrSrMrr) rLrMrNrMrOrUrSrDrr)rLrMrNrMrrM)rLrMrNrMrWrMrrE)r)rLrMrNrMrWrMrrY) __name__ __module__ __qualname____doc__r-r;r>rBrHrKrr(rVrrrrr6s&          r)EPERM ECONNABORTEDEPROTOENETDOWN ENOPROTOOPT EHOSTDOWNENONET EHOSTUNREACH EOPNOTSUPP ENETUNREACHENOSRESOCKTNOSUPPORTEPROTONOSUPPORT ETIMEDOUT ECONNRESETzset[int]_ignorable_accept_errnosc@s.eZdZdZdddZdd d 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. rrrrcCsft|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) r!r"rr#r$r%r&rV SOL_SOCKET SO_ACCEPTCONNrr)r,r listeningrrrr-xs    zSocketListener.__init__rc 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) racceptrrrrrror)r,sock_rrrrrts   zSocketListener.acceptcs |jtjIdHdS)z.Close this listener and its underlying socket.NrIr=rrrrKs zSocketListener.acloseNrZ)rrr[)r\r]r^r_r-rtrKrrrrrpfs   rp)rr)* __future__rr contextlibrrtypingrrrrr"_utilr r abcr r collections.abcr _socketrsys version_infortyping_extensionsrFEBADFENOTSOCKrrr_ignorable_accept_errno_namessetro__annotations__nameAttributeErroraddgetattrrprrrrsF