U
|
Z ±du ã @ sÔ d dl Z d dlZd dlZd dlZd dlmZ d dlm Z d dl
|
Z d dl Z ddl mZ G dd dZzd dlmZ W n" ek
|
r ejd krd
|
ZY nX e d ¡Ze d ¡Zd d Zdd ZejejB Zd2ddZdd Z dd Z dd Z eej dddd3dd Z ejd ks e s8e ed r8eej!dddd d Z!eej"dddd d Z"eejdddej#ej$d dfd!d"Zd#d$ Z%e&ed%d e&ed&d B Z'd'd( Z(d4d*d+Z)d,d- Z*G d.d/ d/Z+G d0d1 d1e+Z,dS )5é N)Úwraps)Ú TYPE_CHECKINGé )Ú_corec @ s. e Zd Zd
|
ddZdd Zdd Zdd ZdS ) Ú _try_syncNc C s
|
|| _ d S ©N)Ú_blocking_exc_override)ÚselfZblocking_exc_override© r
|
úCd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_socket.pyÚ__init__ s z_try_sync.__init__c C s" | j d krt|tS | |¡S d S r )r Ú
|
isinstanceÚBlockingIOError)r Úexcr
|
r
|
r Ú_is_blocking_io_error s
|
|
z _try_sync._is_blocking_io_errorc à s t j ¡ I d H d S r )ÚtrioÚlowlevelZcheckpoint_if_cancelled©r r
|
r
|
r Ú
|
__aenter__ s z_try_sync.__aenter__c à s. |d k r| |¡rdS tj ¡ I d H dS d S )NTF)r r r Zcancel_shielded_checkpoint)r ÚetypeÚvalueÚtbr
|
r
|
r Ú __aexit__# s z_try_sync.__aexit__)N)Ú__name__Ú
|
__module__Ú __qualname__r r r r r
|
r
|
r
|
r r s
|
r )Ú IPPROTO_IPV6Úwin32é) Úhostname_resolverÚsocket_factoryc C s t d¡}t | ¡ |S )a\ Set a custom hostname resolver.
|
|
By default, Trio's :func:`getaddrinfo` and :func:`getnameinfo` functions
|
use the standard system resolver functions. This function allows you to
|
customize that behavior. The main intended use case is for testing, but it
|
might also be useful for using third-party resolvers like `c-ares
|
<https://c-ares.haxx.se/>`__ (though be warned that these rarely make
|
perfect drop-in replacements for the system resolver). See
|
:class:`trio.abc.HostnameResolver` for more details.
|
|
Setting a custom hostname resolver affects all future calls to
|
:func:`getaddrinfo` and :func:`getnameinfo` within the enclosing call to
|
:func:`trio.run`. All other hostname resolution in Trio is implemented in
|
terms of these functions.
|
|
Generally you should call this function just once, right at the beginning
|
of your program.
|
|
Args:
|
hostname_resolver (trio.abc.HostnameResolver or None): The new custom
|
hostname resolver, or None to restore the default behavior.
|
|
Returns:
|
The previous hostname resolver (which may be None).
|
|
N)Ú _resolverÚgetÚset)r Úoldr
|
r
|
r Ú set_custom_hostname_resolverB s
|
|
r% c C s t d¡}t | ¡ |S )aÛ Set a custom socket object factory.
|
|
This function allows you to replace Trio's normal socket class with a
|
custom class. This is very useful for testing, and probably a bad idea in
|
any other circumstance. See :class:`trio.abc.HostnameResolver` for more
|
details.
|
|
Setting a custom socket factory affects all future calls to :func:`socket`
|
within the enclosing call to :func:`trio.run`.
|
|
Generally you should call this function just once, right at the beginning
|
of your program.
|
|
Args:
|
socket_factory (trio.abc.SocketFactory or None): The new custom
|
socket factory, or None to restore the default behavior.
|
|
Returns:
|
The previous socket factory (which may be None).
|
|
N)Ú_socket_factoryr" r# )r r$ r
|
r
|
r Úset_custom_socket_factoryb s
|
|
r' c à sà dd }t |4 I dH , t | |||||tB ¡W 5 Q I dH R £ S Q I dH R X t| trz| d¡} W n" tk
|
r tj| dd} Y nX t
|
d¡}|dk r¸| | |||||¡I dH S t j j tj| |||||ddI dH S dS )aT Look up a numeric address given a name.
|
|
Arguments and return values are identical to :func:`socket.getaddrinfo`,
|
except that this version is async.
|
|
Also, :func:`trio.socket.getaddrinfo` correctly uses IDNA 2008 to process
|
non-ASCII domain names. (:func:`socket.getaddrinfo` uses IDNA 2003, which
|
can give the wrong result in some cases and cause you to connect to a
|
different host than the one you intended; see `bpo-17305
|
<https://bugs.python.org/issue17305>`__.)
|
|
This function's behavior can be customized using
|
:func:`set_custom_hostname_resolver`.
|
|
c S s t | tjo| jtjkS r )r Ú_stdlib_socketÚgaierrorÚerrnoÚ
|
EAI_NONAME)r r
|
r
|
r Únumeric_only_failure s
|
þz)getaddrinfo.<locals>.numeric_only_failureNÚasciiT)Úuts46©Z cancellable)r r( Ú getaddrinfoÚ _NUMERIC_ONLYr ÚstrÚencodeÚUnicodeEncodeErrorÚ_idnar! r" r Ú to_threadÚrun_sync)ÚhostÚportÚfamilyÚtypeÚprotoÚflagsr, Úhrr
|
r
|
r r0 s8 ÿ"
|
|
ør0 c à sD t d¡}|dk r$| | |¡I dH S tjjtj| |ddI dH S dS )a Look up a name given a numeric address.
|
|
Arguments and return values are identical to :func:`socket.getnameinfo`,
|
except that this version is async.
|
|
This function's behavior can be customized using
|
:func:`set_custom_hostname_resolver`.
|
|
NTr/ )r! r" Ú getnameinfor r6 r7 r( )Zsockaddrr= r> r
|
r
|
r r? Ã s
|
|
ÿr? c à s t jjtj| ddI dH S )zkLook up a protocol number by name. (Rarely used.)
|
|
Like :func:`socket.getprotobyname`, but async.
|
|
Tr/ N)r r6 r7 r( Úgetprotobyname)Únamer
|
r
|
r r@ Ö s
|
ÿr@ c C s t | S )z]Convert a standard library :class:`socket.socket` object into a Trio
|
socket object.
|
|
)Ú _SocketType)Úsockr
|
r
|
r Úfrom_stdlib_socketé s rD r
|
©ÚassignedÚupdatedc C s( t |||| \}}}tt | |||¡S )z=Like :func:`socket.fromfd`, but returns a Trio socket object.)Ú_sniff_sockopts_for_filenorD r( Úfromfd)Úfdr: r; r< r
|
r
|
r rI ñ s rI Ú fromsharec O s t tj| |S r )rD r( rK )ÚargsÚkwargsr
|
r
|
r rK ü s c O s t j| |\}}t|t|fS )zTLike :func:`socket.socketpair`, but returns a pair of Trio socket
|
objects.
|
|
)r( Ú
|
socketpairrD )rL rM ÚleftÚrightr
|
r
|
r rN s rN c C sV |dkr*t d¡}|dk r>| | ||¡S nt| |||\} }}t | |||¡}t|S )zCreate a new Trio socket, like :class:`socket.socket`.
|
|
This function's behavior can be customized using
|
:func:`set_custom_socket_factory`.
|
|
N)r& r" ÚsocketrH r( rD )r: r; r< ÚfilenoZsfZ stdlib_socketr
|
r
|
r rQ s
|
rQ c C s| t jdkr| ||fS ddlm}m}m}m} tj| |||d}z(| ||¡} | ||¡}| ||¡}W 5 | ¡ X | ||fS )zCCorrect SOCKOPTS for given fileno, falling back to provided values.Úlinuxr )Ú SO_DOMAINÚ SO_PROTOCOLÚ
|
SOL_SOCKETÚSO_TYPE)rR )
|
ÚsysÚplatformrQ rT rU rV rW r( ÚdetachÚ
|
getsockopt) r: r; r< rR rT rU rV rW Zsockobjr
|
r
|
r rH " s
|
|
|
rH Z SOCK_NONBLOCKZ SOCK_CLOEXECc C s | t @ S r )Ú_SOCK_TYPE_MASK)Ztype_numr
|
r
|
r Úreal_socket_typeJ s r] Fc sT t tj| t ddd fdd}d| d|_|rP| jd| d 7 _|S )
|
N)r r
|
rE c s | ||¡I d H S r )Ú_nonblocking_helper)r rL rM ©ÚfnÚwait_fnr
|
r ÚwrapperQ s z1_make_simple_sock_method_wrapper.<locals>.wrapperzLike :meth:`socket.socket.z`, but async.
|
|
z7Only available on platforms where :meth:`socket.socket.z` is available.)Úgetattrr( rQ Ú_wrapsÚ__doc__)Zmethnamera Ú maybe_availrb r
|
r_ r Ú _make_simple_sock_method_wrapperN s
|
ÿrg c à s¤ |t jkr*t|tr t|dksztdnP|t jkrbt|trXdt| krVdkszn tdn|t jkrvt |¡S |S |^}}}t|t
|
rÀzt ||d ¡ W n t t fk
|
rº Y nX |S |dkrÌd }|dkrØd}d} |rê| t jO } |t jkr|s| t jO } t|||| || I d H }
|
t|
|
d ks0t|
|
^^ }} }|t jkr t| } t| dksdtt|d
|
kr~|d | d< t|dkr|d
|
| d
|
< t| } | S ) Né z&address should be a (host, port) tupleé z=address should be a (host, port, [flowinfo, [scopeid]]) tupler Ú z <broadcast>z255.255.255.255r é )r( ÚAF_INETr ÚtupleÚlenÚ
|
ValueErrorÚAF_INET6ÚAF_UNIXÚosÚfspathÚintÚ inet_ptonÚOSErrorÚ TypeErrorÚ
|
AI_PASSIVEÚ AI_V4MAPPEDr0 ÚAssertionErrorÚlist) r; r: r< Ú ipv6_v6onlyÚaddressÚlocalr8 r9 Ú_r= Zgai_resÚnormedr
|
r
|
r Ú_resolve_address_nocpm sN
|
|
|
$ÿ
|
|
|
|
|
|
r c @ s e Zd Zdd ZdS )Ú
|
SocketTypec C s t dd S )Nz`SocketType is an abstract class; use trio.socket.socket if you want to construct a socket object)rw r r
|
r
|
r r ® s ÿzSocketType.__init__N)r r r r r
|
r
|
r
|
r r s r c
|
s´ e Zd Zdd Zddddddd d
|
d d h
|
Zd d Z fddZdd Zdd Ze dd Z
|
e dd Z e dd Z e dd Z d d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zed/ejZd0d1 Zd2d3 Z ed4ejZ ed5ejZ ed6ejZ ed7ejZ e!e"j#d8r.ed8ejd9d:Z$e!e"j#d;rLed;ejd9d:Z%ed<ejZ&e'e"j#j(d=d=d>d?d@ Z(e)j*dAkse+s¬e!e"j#dBr¬e'e"j#j,d=d=d>dCdD Z, Z-S )ErB c C s> t |tjk r"td t |j¡|| _| j d¡ d| _d S )Nz0expected object of type 'socket.socket', not '{}F) r; r( rQ rw Úformatr Ú_sockÚ setblockingÚ_did_shutdown_SHUT_WR)r rC r
|
r
|
r r ¶ s ÿÿ z_SocketType.__init__rZ Úget_inheritableÚset_inheritablerR Ú getpeernameÚ getsocknamer[ Ú
|
setsockoptÚlistenÚsharec C s" || j krt| j|S t|d S r )Ú_forwardrc r ÚAttributeError)r rA r
|
r
|
r Ú __getattr__Þ s
|
z_SocketType.__getattr__c s t ¡ t| j S r )ÚsuperÚ__dir__r{ r r ©Ú __class__r
|
r r ã s z_SocketType.__dir__c C s | S r r
|
r r
|
r
|
r Ú __enter__æ s z_SocketType.__enter__c G s | j j| S r )r Ú__exit__)r Úexc_infor
|
r
|
r r é s z_SocketType.__exit__c C s | j jS r )r r: r r
|
r
|
r r: ì s z_SocketType.familyc C s t | jjS r )r] r r; r r
|
r
|
r r; ð s z_SocketType.typec C s | j jS r )r r< r r
|
r
|
r r< ÷ s z_SocketType.protoc C s | j S r )r r r
|
r
|
r Údid_shutdown_SHUT_WRû s z _SocketType.did_shutdown_SHUT_WRc C s t | j dd¡S )Nz socket.socketztrio.socket.socket)Úreprr Úreplacer r
|
r
|
r Ú__repr__ÿ s z_SocketType.__repr__c C s t | j ¡ S )z"Same as :meth:`socket.socket.dup`.)rB r Údupr r
|
r
|
r r s z_SocketType.dupc C s* | j ¡ dkr&tj | j ¡ | j ¡ d S )Néÿÿÿÿ)r rR r r Znotify_closingÚcloser r
|
r
|
r r s z_SocketType.closec à sj | j |ddI d H }ttdrJ| jtjkrJ|d rJtj | jj |¡I d H S tj
|
¡ I d H | j |¡S d S )NT©r~ rq r ) r Úhasattrr( r: rq r r6 r7 r Úbindr Ú
|
checkpoint)r r} r
|
r
|
r r¡ s ÿ
|
þýz_SocketType.bindc C s& | j |¡ |tjtjfkr"d| _d S )NT)r Úshutdownr( ÚSHUT_WRÚ SHUT_RDWRr )r Úflagr
|
r
|
r r£ s z_SocketType.shutdownc C sR t jdkr,t | jgg g d¡\}}}t|S t ¡ }| | jtj¡ t| d¡S )Nr r )rX rY Úselectr ÚboolÚpollÚregisterÚPOLLIN)r Úrreadyr Úpr
|
r
|
r Ú is_readable$ s
|
z_SocketType.is_readablec à s t | j¡I d H d S r )r Ú wait_writabler r r
|
r
|
r r¯ - s z_SocketType.wait_writablec à sB | j tjkr | j ttj¡}nd}t| j| j | j |||dI d H S )NF)r| r} r~ )
|
r: r( rp r r[ r Ú IPV6_V6ONLYr r; r< )r r} r~ r| r
|
r
|
r r 0 s ÿúz!_SocketType._resolve_address_nocpc à s t 4 I d H & || jf||W 5 Q I d H R £ S Q I d H R X || jI d H z|| jf||W S tk
|
rx Y q@X q@d S r )r r r )r r` rL rM ra r
|
r
|
r r^ @ s 0z _SocketType._nonblocking_helperÚacceptc à s | ¡ I dH \}}t||fS )z-Like :meth:`socket.socket.accept`, but async.N)Ú_acceptrD )r rC Úaddrr
|
r
|
r r± f s z_SocketType.acceptc à s¼ zf| j |ddI d H }t 4 I d H " | j |¡W 5 Q I d H R £ W S Q I d H R X t | j¡I d H W n" tjk
|
r | j ¡ Y nX | j t
|
j t
|
j ¡}|dkr¸t |dt |¡ d S )NFr r zError in connect: )r r r Úconnectr r¯ r Z Cancelledr r[ r( rV ÚSO_ERRORrv rr Ústrerror)r r} Úerrr
|
r
|
r r´ o s 0,
|
z_SocketType.connectÚrecvÚ recv_intoÚrecvfromÚ recvfrom_intoÚrecvmsgT)rf Ú recvmsg_intoÚsendr
|
rE c Ç s@ t |}| j|d ddI dH |d< | tjj|i tj¡I dH S )z3Similar to :meth:`socket.socket.sendto`, but async.r Fr N)r{ r r^ r( rQ Úsendtor r¯ ©r rL r
|
r
|
r r¿ ë s ÿz_SocketType.sendtor Úsendmsgc Ç sX t |dkr<|d dk r<t|}| j|d ddI dH |d< | tjj|i tj¡I dH S )z©Similar to :meth:`socket.socket.sendmsg`, but async.
|
|
Only available on platforms where :meth:`socket.socket.sendmsg` is
|
available.
|
|
ri r NFr ) rn r{ r r^ r( rQ rÁ r r¯ rÀ r
|
r
|
r rÁ þ s
|
ÿz_SocketType.sendmsg).r r r r r r r r r Úpropertyr: r; r< r r r r r¡ r£ r® r¯ r r^ rg r Z wait_readabler² r± r´ r¸ r¹ rº r» r r( rQ r¼ r½ r¾ rd r¿ rX rY r rÁ Ú __classcell__r
|
r
|
r r rB µ s ö
|
|
|
|
$ J ÿ ÿ ÿ
|
ÿ
|
ÿrB )r r r r )r )F)-rr rX r§ rQ r( Ú functoolsr rd Útypingr Úidnar5 r rj r r r Ú ImportErrorrY ZRunVarr! r& r% r' ÚAI_NUMERICHOSTÚAI_NUMERICSERVr1 r0 r? r@ rD rI r rK rN rl Ú SOCK_STREAMrH rc r\ r] rg r r rB r
|
r
|
r
|
r Ú<module> sd
|
|
|
|
|
? ÿÿ
|
|
ü
|
|
ÿÿ
|
@
|
