a
xdu @ 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 ey ejd krd
ZY n0 ed Zed Zd
d Zdd ZejejB Zd2ddZdd Z dd Z dd Z eej dddd3dd Z ejd ks e s6e ed r6eej!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
D/var/www/html/Ranjet/env/lib/python3.9/site-packages/trio/_socket.py__init__ s z_try_sync.__init__c C s" | j d u rt|tS | |S d S r )r
isinstanceBlockingIOError)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 )triolowlevelZcheckpoint_if_cancelledr r
r
r
__aenter__ s z_try_sync.__aenter__c s. |d ur| |rdS tj I d H dS d S )NTF)r r r Zcancel_shielded_checkpoint)r etypevaluetbr
r
r __aexit__# s z_try_sync.__aexit__)N)__name__
__module__ __qualname__r r r r r
r
r
r r s
r ) IPPROTO_IPV6win32) hostname_resolversocket_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
`__ (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) _resolvergetset)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 d I dH S 1 I dH sR0 Y t| trz| d} W n ty tj| dd} Y n0 t
d}|dur|| |||||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
`__.)
This function's behavior can be customized using
:func:`set_custom_hostname_resolver`.
c S s t | tjo| jtjkS r )r
_stdlib_socketgaierrorerrno
EAI_NONAME)r r
r
r numeric_only_failure s
z)getaddrinfo..numeric_only_failureNasciiT)uts46Z cancellable)r r( getaddrinfo
_NUMERIC_ONLYr
strencodeUnicodeEncodeError_idnar! r" r to_threadrun_sync)hostportfamilytypeprotoflagsr, hrr
r
r r0 s. .
r0 c sD t d}|dur$|| |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
assignedupdatedc 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| i |S r )rD r( rK )argskwargsr
r
r rK s c O s$ t j| i |\}}t|t|fS )zTLike :func:`socket.socketpair`, but returns a pair of Trio socket
objects.
)r(
socketpairrD )rL rM leftrightr
r
r rN s rN c C sV |du r*t d}|dur>|| ||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}z0|||} |||}|||}W | n
| 0 | ||fS )zCCorrect SOCKOPTS for given fileno, falling back to provided values.linuxr ) SO_DOMAIN SO_PROTOCOL
SOL_SOCKETSO_TYPE)rR )
sysplatformrQ rT rU rV rW r(
getsockoptdetach) r: r; r< rR rT rU rV rW Zsockobjr
r
r rH " s
rH
SOCK_NONBLOCK 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 fnwait_fnr
r wrapperQ s z1_make_simple_sock_method_wrapper..wrapperzLike :meth:`socket.socket.z`, but async.
z7Only available on platforms where :meth:`socket.socket.z` is available.)getattrr( rQ _wraps__doc__)Zmethnamerc maybe_availrd r
ra r _make_simple_sock_method_wrapperN s
ri c s |t jkr*t|tr t|dksztdnP|t jkrbt|trXdt| krVdkszn tdn|t jkrvt |S |S |^}}}t|t
rzt ||d W n t t
fy Y n0 |S |dkrd }|dkrd}d} |r| t jO } |t jkr|s| t jO } t|||| || I d H }
t|
d ks.J |
^^ }} }|t jkrt| } t| dksbJ t|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 z255.255.255.255r )r( AF_INETr
tuplelen
ValueErrorAF_INET6AF_UNIXosfspathint inet_ptonOSError TypeError
AI_PASSIVE AI_V4MAPPEDr0 list) r; r: r< ipv6_v6onlyaddresslocalr8 r9 _r= Zgai_resnormedr
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)ry 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 Zh dZdd Z fddZdd Zd
d Ze d d
Z
e dd Z e dd Z e dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd d! Zd"d# Zd$d% Zed&ejZd'd( Zd)d* Z ed+ejZ ed,ejZ ed-ejZ ed.ejZ e!e"j#d/r ed/ejd0d1Z$e!e"j#d2r t |tjur"tdt |j|| _| jd d| _d S )Nz0expected object of type 'socket.socket', not '{}F) r; r( rQ ry formatr _sock setblocking_did_shutdown_SHUT_WR)r rC r
r
r r s z_SocketType.__init__>
rZ listen getpeernameZsharerR set_inheritable
setsockoptget_inheritabler[ getsocknamec C s" || j v rt| j|S t|d S r )_forwardre 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 | jddS )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_closingcloser 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 )NTr rs r ) r hasattrr( r: rs r r6 r7 r bindr
checkpoint)r r~ r
r
r r s
z_SocketType.bindc C s& | j | |tjtjfv r"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|dS )Nr r )rX rY selectr boolpollregisterPOLLIN)r rreadyr pr
r
r is_readable$ s
z_SocketType.is_readablec s t | jI d H d S r )r
wait_writabler r r
r
r r - s z_SocketType.wait_writablec sB | j tjkr | jttj}nd}t| j| j | j |||dI d H S )NF)r} r~ r )
r: r( rr r rZ 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 . || jg|R i |W d I d H S 1 I d H sH0 Y || jI d H z|| jg|R i |W S ty Y qR0 qRd S r )r r r )r rb rL rM rc r
r
r r` @ s B z _SocketType._nonblocking_helperacceptc 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 zr| j |ddI d H }t 4 I d H $ | j|W d I d H W S 1 I d H sT0 Y t| jI d H W n tjy | j Y n0 | j t
j t
j }|dkrt
|dt| d S )NFr r zError in connect: )r r r connectr r r Z Cancelledr rZ r( rV SO_ERRORrx rt strerror)r r~ errr
r
r r o s 08
z_SocketType.connectrecv recv_intorecvfrom
recvfrom_intorecvmsgT)rh recvmsg_intosendr
rE c s@ t |}| j|d ddI dH |d< | tjj|i tjI 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 dur sd
?
@