U
|
Z ±d¥8 ã @ sð d dl Z d dlmZ d dlZddlmZ ddlmZm Z ddl
|
m Z m Z dZ e je jhZedd ZG d
|
d d e e d Zd ddddddddddddddgZe ZeD ].Zze ee e¡ W q¤ ek
|
rÐ Y q¤X q¤G d d d e e e d ZdS ) é N)Úcontextmanageré )Úsocket)ÚConflictDetectorÚFinal)ÚHalfCloseableStreamÚListeneri c
|
c s\ z
|
d V W nL t k
|
rV } z.| jtkr4t d¡d nt d | ¡¡| W 5 d } ~ X Y nX d S )Nz this socket was already closedz socket connection broken: {})ÚOSErrorÚerrnoÚ_closed_stream_errnosÚtrioÚClosedResourceErrorZBrokenResourceErrorÚformat)Úexc© r úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_highlevel_socket.pyÚ)_translate_socket_errors_to_stream_errors s
|
|
ÿþr c @ sT e Zd ZdZdd Zdd Zdd Zdd Zdd d Zd d Z dd Z
|
dddZ d
|
S )Ú SocketStreamuf An 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
|
<https://github.com/python-trio/trio/issues/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.
|
|
c C s t |tjstd|jtjkr(td|| _td| _ z|
|
tj tj d¡ W n t k
|
rb Y nX ttdrz|
|
tj tjd¡ W n t k
|
r Y nX d S )Nz*SocketStream requires a Trio socket objectz*SocketStream requires a SOCK_STREAM socketz;another task is currently sending data on this SocketStreamTÚTCP_NOTSENT_LOWATi @ )Ú
|
isinstanceÚtsocketÚ
|
SocketTypeÚ TypeErrorÚtypeÚ SOCK_STREAMÚ
|
ValueErrorr r Ú_send_conflict_detectorÚ
|
setsockoptÚ IPPROTO_TCPÚ TCP_NODELAYr Úhasattrr )Úselfr r r r Ú__init__> s" ÿ
|
|
zSocketStream.__init__c à sà | j jrt d¡| j¾ t ¬ t|}|sz| j ¡ dkrHt d¡tj ¡ I d H W 5 Q R £ W 5 Q R £ W 5 Q R £ d S d}|t
|
|k r¾||d
}| j |¡I d H }W 5 Q R X ||7 }q~W 5 Q R X W 5 Q R X W 5 Q R X d S )Nz!can't send data after sending EOFéÿÿÿÿzsocket was already closedr ) r Údid_shutdown_SHUT_WRr r r r Ú
|
memoryviewÚfilenoÚlowlevelÚ
|
checkpointÚlenÚsend)r! ÚdataÚ
|
total_sentÚ remainingÚsentr r r Úsend_all` s
|
|
|
" zSocketStream.send_allc
|
à sL | j < | j ¡ dkr tjt | j ¡ I d H W 5 Q R X W 5 Q R X d S )Nr# )r r r& r r r Z wait_writable©r! r r r Ú wait_send_all_might_not_blockq s
|
z*SocketStream.wait_send_all_might_not_blockc
|
à s\ | j L tj ¡ I d H | jjr.W 5 Q R £ d S t | j tj ¡ W 5 Q R X W 5 Q R X d S ©N)
|
r r r' r( r r$ r Úshutdownr ÚSHUT_WRr0 r r r Úsend_eofx s zSocketStream.send_eofNc
|
à sL |d kr t }|dk r tdt | j |¡I d H W 5 Q R £ S Q R X d S )Nr zmax_bytes must be >= 1)ÚDEFAULT_RECEIVE_SIZEr r r Úrecv)r! Z max_bytesr r r Ú receive_some s zSocketStream.receive_somec à s | j ¡ tj ¡ I d H d S r2 ©r Úcloser r' r( r0 r r r Úaclose s
|
zSocketStream.aclosec C s | j |||¡S )zlSet an option on the underlying socket.
|
|
See :meth:`socket.socket.setsockopt` for details.
|
|
)r r )r! ÚlevelÚoptionÚvaluer r r r s zSocketStream.setsockoptr c C s* |dkr| j ||¡S | j |||¡S dS )zCheck the current value of an option on the underlying socket.
|
|
See :meth:`socket.socket.getsockopt` for details.
|
|
r N)r Ú
|
getsockopt)r! r< r= Z
|
buffersizer r r r? s zSocketStream.getsockopt)N)r ) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r" r/ r1 r5 r8 r; r r? r r r r r & s "
|
|
r )Ú metaclassÚEPERMÚ ECONNABORTEDZEPROTOZENETDOWNZ ENOPROTOOPTZ EHOSTDOWNZENONETZ EHOSTUNREACHÚ
|
EOPNOTSUPPZ ENETUNREACHZENOSRZESOCKTNOSUPPORTZEPROTONOSUPPORTZ ETIMEDOUTÚ
|
ECONNRESETc @ s( e Zd ZdZdd Zdd Zdd ZdS ) ÚSocketListeneraâ A :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.
|
|
c C sh t |tjstd|jtjkr(tdz| tjtj ¡}W n t
|
k
|
rP Y nX |s^td|| _ d S )Nz,SocketListener requires a Trio socket objectz,SocketListener requires a SOCK_STREAM socketz*SocketListener requires a listening socket) r r r r r r r r? Ú
|
SOL_SOCKETÚ SO_ACCEPTCONNr r )r! r Z listeningr r r r" O s zSocketListener.__init__c
|
à sf z| j ¡ I dH \}}W n> tk
|
rV } z |jtkr:tj|jtkrF W 5 d}~X Y q X t|S q dS )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) r Úacceptr r
|
r r r Ú_ignorable_accept_errnosr )r! ÚsockÚ_r r r r rL _ s
|
|
zSocketListener.acceptc à s | j ¡ tj ¡ I dH dS )z.Close this listener and its underlying socket.Nr9 r0 r r r r; { s
|
zSocketListener.acloseN)r@ rA rB rC r" rL r; r r r r rI > s rI )r
|
Ú
|
contextlibr r Ú r r Z_utilr r Úabcr r r6 ÚEBADFÚENOTSOCKr r r Z _ignorable_accept_errno_namesÚsetrM ÚnameÚaddÚgetattrÚAttributeErrorrI r r r r Ú<module> sH ü
|
|í
|
