_socket.pyc
1 o 2 >��c�u � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlm Z d dl 3 Z d dl Z ddl mZ G dd� d�Zzd dlmZ W n eyI ejd krGd 4 ZY nw e�d �Ze�d �Zd d� Zdd� ZejejB Zd2dd�Zdd� Z dd� Z dd� Z eej ddd�d3dd ��Z ejd ks�e s�e ed �r�eej!ddd�d d � �Z!eej"ddd�d d � �Z"eejddd�ej#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 5 dd�Zdd� Zdd� Zdd � ZdS ) � _try_syncNc C s 6 || _ d S �N)�_blocking_exc_override)�selfZblocking_exc_override� r 7 ��C:\Users\Jacks.GUTTSPC\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.10_qbz5n2kfra8p0\LocalCache\local-packages\Python310\site-packages\trio\_socket.py�__init__ s 8 z_try_sync.__init__c C s | j d u r 9 t|t�S | � |�S r )r � 10 isinstance�BlockingIOError)r �excr 11 r 12 r �_is_blocking_io_error s 13 14 15 z _try_sync._is_blocking_io_errorc � s �t j�� I d H d S r )�trio�lowlevel�checkpoint_if_cancelled�r r 16 r 17 r � 18 __aenter__ s �z_try_sync.__aenter__c � s, �|d ur | � |�r dS tj�� I d H dS )NTF)r r r �cancel_shielded_checkpoint)r �etype�value�tbr 19 r 20 r � __aexit__# s 21 �z_try_sync.__aexit__r )�__name__� 22 __module__� __qualname__r r r r r 23 r 24 r 25 r r s 26 27 r )� IPPROTO_IPV6�win32�) �hostname_resolver�socket_factoryc C � t �d�}t �| � |S )a\ Set a custom hostname resolver. 28 29 By default, Trio's :func:`getaddrinfo` and :func:`getnameinfo` functions 30 use the standard system resolver functions. This function allows you to 31 customize that behavior. The main intended use case is for testing, but it 32 might also be useful for using third-party resolvers like `c-ares 33 <https://c-ares.haxx.se/>`__ (though be warned that these rarely make 34 perfect drop-in replacements for the system resolver). See 35 :class:`trio.abc.HostnameResolver` for more details. 36 37 Setting a custom hostname resolver affects all future calls to 38 :func:`getaddrinfo` and :func:`getnameinfo` within the enclosing call to 39 :func:`trio.run`. All other hostname resolution in Trio is implemented in 40 terms of these functions. 41 42 Generally you should call this function just once, right at the beginning 43 of your program. 44 45 Args: 46 hostname_resolver (trio.abc.HostnameResolver or None): The new custom 47 hostname resolver, or None to restore the default behavior. 48 49 Returns: 50 The previous hostname resolver (which may be None). 51 52 N)� _resolver�get�set)r! �oldr 53 r 54 r � set_custom_hostname_resolverB s 55 56 r( c C r# )a� Set a custom socket object factory. 57 58 This function allows you to replace Trio's normal socket class with a 59 custom class. This is very useful for testing, and probably a bad idea in 60 any other circumstance. See :class:`trio.abc.HostnameResolver` for more 61 details. 62 63 Setting a custom socket factory affects all future calls to :func:`socket` 64 within the enclosing call to :func:`trio.run`. 65 66 Generally you should call this function just once, right at the beginning 67 of your program. 68 69 Args: 70 socket_factory (trio.abc.SocketFactory or None): The new custom 71 socket factory, or None to restore the default behavior. 72 73 Returns: 74 The previous socket factory (which may be None). 75 76 N)�_socket_factoryr% r& )r" r' r 77 r 78 r �set_custom_socket_factoryb s 79 80 r* c 81 � s� �dd� }t |�4 I dH � t�| |||||tB �W d �I dH S 1 I dH s*w Y t| t�rLz| �d�} W n tyK tj| dd�} Y nw t � 82 d�}|durb|�| |||||�I dH S t j j tj| |||||dd�I dH S )aT Look up a numeric address given a name. 83 84 Arguments and return values are identical to :func:`socket.getaddrinfo`, 85 except that this version is async. 86 87 Also, :func:`trio.socket.getaddrinfo` correctly uses IDNA 2008 to process 88 non-ASCII domain names. (:func:`socket.getaddrinfo` uses IDNA 2003, which 89 can give the wrong result in some cases and cause you to connect to a 90 different host than the one you intended; see `bpo-17305 91 <https://bugs.python.org/issue17305>`__.) 92 93 This function's behavior can be customized using 94 :func:`set_custom_hostname_resolver`. 95 96 c S s t | tj�o | jtjkS r )r �_stdlib_socket�gaierror�errno� 97 EAI_NONAME)r r 98 r 99 r �numeric_only_failure� s 100 �z)getaddrinfo.<locals>.numeric_only_failureN�asciiT)�uts46�� cancellable)r r+ � getaddrinfo� _NUMERIC_ONLYr �str�encode�UnicodeEncodeError�_idnar$ r% r � to_thread�run_sync)�host�port�family�type�proto�flagsr/ �hrr 101 r 102 r r4 � s4 ��,� 103 � 104 �r4 c � sB �t �d�}|dur|�| |�I dH S tjjtj| |dd�I dH S )a Look up a name given a numeric address. 105 106 Arguments and return values are identical to :func:`socket.getnameinfo`, 107 except that this version is async. 108 109 This function's behavior can be customized using 110 :func:`set_custom_hostname_resolver`. 111 112 NTr2 )r$ r% � getnameinfor r: r; r+ )�sockaddrrA rB r 113 r 114 r rC � s � 115 116 117 �rC c � s �t jjtj| dd�I dH S )zkLook up a protocol number by name. (Rarely used.) 118 119 Like :func:`socket.getprotobyname`, but async. 120 121 Tr2 N)r r: r; r+ �getprotobyname)�namer 122 r 123 r rE � s � �rE c C s t | �S )z]Convert a standard library :class:`socket.socket` object into a Trio 124 socket object. 125 126 )� _SocketType)�sockr 127 r 128 r �from_stdlib_socket� s rI r 129 ��assigned�updatedc C s( t |||| �\}}}tt�| |||��S )z=Like :func:`socket.fromfd`, but returns a Trio socket object.)�_sniff_sockopts_for_filenorI r+ �fromfd)�fdr> r? r@ r 130 r 131 r rN � s rN � fromsharec O s t tj| i |���S r )rI r+ rP )�args�kwargsr 132 r 133 r rP � s c O s$ t j| i |��\}}t|�t|�fS )zTLike :func:`socket.socketpair`, but returns a pair of Trio socket 134 objects. 135 136 )r+ � 137 socketpairrI )rQ rR �left�rightr 138 r 139 r rS s rS c C sV |du rt �d�}|dur|�| ||�S n 140 t| |||�\} }}t�| |||�}t|�S )z�Create a new Trio socket, like :class:`socket.socket`. 141 142 This function's behavior can be customized using 143 :func:`set_custom_socket_factory`. 144 145 N)r) r% �socketrM r+ rI )r> r? r@ �fileno�sfZ stdlib_socketr 146 r 147 r rV s 148 �rV c C s� t jdkr 149 | ||fS ddlm}m}m}m} tj| |||d�}z|�||�} |�||�}|�||�}W |� � n|� � w | ||fS )zCCorrect SOCKOPTS for given fileno, falling back to provided values.�linuxr )� SO_DOMAIN� SO_PROTOCOL� 150 SOL_SOCKET�SO_TYPE)rW ) 151 �sys�platformrV rZ r[ r\ r] r+ � 152 getsockopt�detach) r> r? r@ rW rZ r[ r\ r] Zsockobjr 153 r 154 r rM " s 155 156 157 rM � SOCK_NONBLOCK� SOCK_CLOEXECc C s | t @ S r )�_SOCK_TYPE_MASK)�type_numr 158 r 159 r �real_socket_typeJ s rf Fc sT t tj| �� t� ddd�� �fdd��}d| � d�|_|r(| jd| � d �7 _|S ) 160 N)r r 161 rJ c � s �| � � ||��I d H S r )�_nonblocking_helper)r rQ rR ��fn�wait_fnr 162 r �wrapperQ s �z1_make_simple_sock_method_wrapper.<locals>.wrapperzLike :meth:`socket.socket.z`, but async. 163 164 z7Only available on platforms where :meth:`socket.socket.z` is available.)�getattrr+ rV �_wraps�__doc__)Zmethnamerj � maybe_availrk r 165 rh r � _make_simple_sock_method_wrapperN s 166 �rp c � s� �|t jkrt|t�rt|�dkstd��n+|t jkr5t|t�r0dt|� kr/dks4td�� td��n |t jkr?t� |�S |S |^}}}t|t 167 �rbz t � ||d � W |S t t fya Y nw |dkrhd }|dkrnd}d} |rw| t jO } |t jkr�|s�| t jO } t|||| || �I d H } 168 t| 169 �d ks�J �| 170 ^�^ }} }|t jkr�t| �} t| �dks�J �t|�d 171 kr�|d | d< t|�dkr�|d 172 | d 173 < 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� 174 ValueError�AF_INET6�AF_UNIX�os�fspath�int� inet_pton�OSError� TypeError� 175 AI_PASSIVE� AI_V4MAPPEDr4 �list) r? r> r@ � ipv6_v6only�address�localr< r= �_rA Zgai_resZnormedr 176 r 177 r �_resolve_address_nocpm s^ � 178 � 179 ���� 180 181 182 183 �� 184 185 186 r� c @ s e Zd Zdd� ZdS )� 187 SocketTypec C s t d��)Nz`SocketType is an abstract class; use trio.socket.socket if you want to construct a socket object)r� r r 188 r 189 r r � s �zSocketType.__init__N)r r r r r 190 r 191 r 192 r r� � s r� c s� e Zd Zdd� Zh d�Zdd� Z� fdd�Zdd � Zd 193 d � Ze d d � �Z 194 e dd� �Z e dd� �Z e dd� �Z dd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zed&ej�Zd'd(� Zd)d*� Z ed+ej�Z ed,ej�Z ed-ej�Z ed.ej�Z e!e"j#d/�r�ed/ejd0d1�Z$e!e"j#d2�r�ed2ejd0d1�Z%ed3ej�Z&e'e"j#j(d4d4d5�d6d7� �Z(e)j*d8ks�e+s�e!e"j#d9�r�e'e"j#j,d4d4d5�d:d;� �Z,� Z-S � Z-S � Z-S )<rG c C s> t |�tjurtd�t |�j���|| _| j�d� d| _d S )Nz0expected object of type 'socket.socket', not '{}F) r? r+ rV r� �formatr �_sock� setblocking�_did_shutdown_SHUT_WR)r rH r 195 r 196 r r � s �� 197 z_SocketType.__init__> 198 �listen� getpeernamer` � 199 setsockopt�get_inheritable�sharera � getsockname�set_inheritablerW c C s || j v r t| j|�S t|��r )�_forwardrl r� �AttributeError)r rF r 200 r 201 r � __getattr__� s 202 z_SocketType.__getattr__c s t � �� t| j� S r )�super�__dir__r� r� r �� __class__r 203 r r� � s z_SocketType.__dir__c C s | S r r 204 r r 205 r 206 r � __enter__� s z_SocketType.__enter__c G s | j j|� S r )r� �__exit__)r �exc_infor 207 r 208 r r� � s z_SocketType.__exit__c C � | j jS r )r� r> r r 209 r 210 r r> � � z_SocketType.familyc C s t | jj�S r )rf r� r? r r 211 r 212 r r? � s z_SocketType.typec C r� r )r� r@ r r 213 r 214 r r@ � r� z_SocketType.protoc C s | j S r )r� r r 215 r 216 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 217 r 218 r �__repr__� s z_SocketType.__repr__c C s t | j�� �S )z"Same as :meth:`socket.socket.dup`.)rG r� �dupr r 219 r 220 r r� s z_SocketType.dupc C s. | j �� dkrtj�| j � | j �� d S d S )N�����)r� rW r r �notify_closing�closer r 221 r 222 r r� s �z_SocketType.closec � sh �| j |dd�I d H }ttd�r&| jtjkr&|d r&tj�| jj |�I d H S tj 223 � � I d H | j� |�S )NT�r� rz r ) r� �hasattrr+ r> rz r r: r; r� �bindr � 224 checkpoint)r r� r 225 r 226 r r� s �� � z_SocketType.bindc C s* | j �|� |tjtjfv rd| _d S d S )NT)r� �shutdownr+ �SHUT_WR� SHUT_RDWRr� )r �flagr 227 r 228 r r� s 229 �z_SocketType.shutdownc C sR t jdkrt�| jgg g d�\}}}t|�S t�� }|�| jtj� t|�d��S )Nr r )r^ r_ �selectr� �bool�poll�register�POLLIN)r �rreadyr� �pr 230 r 231 r � is_readable$ s 232 z_SocketType.is_readablec � s �t �| j�I d H d S r )r � wait_writabler� r r 233 r 234 r r� - s �z_SocketType.wait_writablec � sD �| j tjkr| j�ttj�}nd}t| j| j | j |||d�I d H S )NF)r� r� r� ) 235 r> r+ ry r� r` r � IPV6_V6ONLYr� r? r@ )r r� r� r� r 236 r 237 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 s%w Y || j�I d H z || jg|�R i |��W S tyI Y nw q+r )r r� r )r ri rQ rR rj r 238 r 239 r rg @ s �,� ��z _SocketType._nonblocking_helper�acceptc � s �| � � I dH \}}t|�|fS )z-Like :meth:`socket.socket.accept`, but async.N)�_acceptrI )r rH �addrr 240 r 241 r r� f s � z_SocketType.acceptc � s� �z9| j |dd�I d H }t� 4 I d H � | j�|�W d �I d H W S 1 I d H s+w Y t�| j�I d H W n tjyH | j�� � w | j� t 242 j t 243 j �}|dkr`t |dt�|� ��d S )NFr� r zError in connect: )r� r r� �connectr r� r � Cancelledr� r` r+ r\ �SO_ERRORr r{ �strerror)r r� �errr 244 r 245 r r� o s � 246 0.�3 247 ��z_SocketType.connect�recv� recv_into�recvfrom� recvfrom_into�recvmsgT)ro � recvmsg_into�sendr 248 rJ c � sB �t |�}| j|d dd�I dH |d< | �tjj|i tj�I dH S )z3Similar to :meth:`socket.socket.sendto`, but async.r� Fr� N)r� r� rg r+ rV �sendtor r� �r rQ r 249 r 250 r r� � s � 251 �z_SocketType.sendtor �sendmsgc � sZ �t |�dkr |d dur t|�}| j|d dd�I dH |d< | �tjj|i tj�I dH S )z�Similar to :meth:`socket.socket.sendmsg`, but async. 252 253 Only available on platforms where :meth:`socket.socket.sendmsg` is 254 available. 255 256 rr r� NFr� ) rw r� r� rg r+ rV r� r r� r� r 257 r 258 r r� � s � 259 260 �z_SocketType.sendmsg).r r r r r� r� r� r� r� �propertyr> r? r@ r� r� r� r� r� r� r� r� r� rg rp r � wait_readabler� r� r� r� r� r� r� r� r+ rV r� r� r� rm r� r^ r_ r r� � __classcell__r 261 r 262 r� r rG � sh 263 264 265 266 $ J � � � 267 268 � 269 ��rG )r r r r )r )F)-r{ r^ r� rV r+ � functoolsr rm �typingr �idnar9 r rs r r r � ImportErrorr_ �RunVarr$ r) r( r* �AI_NUMERICHOST�AI_NUMERICSERVr5 r4 rC rE rI rN r� rP rS ru � SOCK_STREAMrM rl rd rf rp r� r� rG r 270 r 271 r 272 r �<module> sj 273 �� 274 275 276 277 ? 278 �� 279 280 � 281 282 �� 283 @