o
    í6WjD3  ã                   @  sú  U d dl mZ d dlZd dlZd dlmZ d dlmZmZm	Z	 d dl
mZ d dlmZ d dlmZmZ d dlmZ d d	lmZmZmZ d
dlmZ d
dlmZmZmZ ddlmZmZmZ ddl m!Z! e"eB eB Z#de$d< e%e"e&f Z'de$d< e'e"B Z(de$d< e%e)e'f Z*de$d< e%e)e"f Z+de$d< edƒZ,ej-fdddœd8d$d%„Z.G d&d'„ d'eƒZ/G d(d)„ d)eƒZ0G d*d+„ d+ee0ƒZ1G d,d-„ d-e1ƒZ2G d.d/„ d/ee1 e0ƒZ3G d0d1„ d1ee* e0ƒZ4G d2d3„ d3ee) e0ƒZ5G d4d5„ d5ee+ e0ƒZ6G d6d7„ d7ee) e0ƒZ7dS )9é    )ÚannotationsN)Úabstractmethod)ÚCallableÚ
CollectionÚMapping)ÚAsyncExitStack)ÚIOBase)ÚIPv4AddressÚIPv6Address)ÚAddressFamily)ÚAnyÚ	TypeAliasÚTypeVaré   )Úget_async_backend)ÚTypedAttributeProviderÚTypedAttributeSetÚtyped_attributeé   )Ú
ByteStreamÚListenerÚUnreliableObjectStream)Ú	TaskGroupr   ÚIPAddressTypeÚIPSockAddrTypeÚSockAddrTypeÚUDPPacketTypeÚUNIXDatagramPacketTypeÚT_RetvalF)Úrequire_connectedÚrequire_boundÚ
sock_or_fdúsocket.socket | intÚ	sock_typeúsocket.SocketKindÚaddr_familyúsocket.AddressFamilyr   Úboolr    Úreturnúsocket.socketc             
   C  s   t | tƒr4ztj| d�}W n: ty3 } z|jtjkr tdƒ|‚|r'tdƒ|‚|r.tdƒ|‚‚ d }~ww t | tjƒr=| }ntdt| ƒj	› d�ƒ‚zo|rcz| 
¡  W n tyb } ztdƒ|‚d }~ww |r�z|jtjtjfv rv| ¡ d }n| ¡ }W n ty†   d }Y nw |s�tdƒ‚|tjkr¤|j|kr¤td|j› d	|jj› �ƒ‚|j|kr¶td
|j› d	|jj› �ƒ‚W n tyÈ   t | tƒrÇ| ¡  ‚ w | d¡ |S )N)Úfilenoz.the file descriptor does not refer to a socketzthe socket must be connectedz+the socket must be bound to a local addresszexpected an int or socket, got z insteadr   z"address family mismatch: expected z, got zsocket type mismatch: expected F)Ú
isinstanceÚintÚsocketÚOSErrorÚerrnoÚENOTSOCKÚ
ValueErrorÚ	TypeErrorÚtypeÚ__qualname__ÚgetpeernameÚfamilyÚAF_INETÚAF_INET6ÚgetsocknameÚ	AF_UNSPECÚnameÚBaseExceptionÚdetachÚsetblocking)r!   r#   r%   r   r    ÚsockÚexcÚ
bound_addr© rB   úa/home/esfera/Documents/content_generation/venv/lib/python3.10/site-packages/anyio/abc/_sockets.pyÚ_validate_socket   sv   
ÿþ

€öÿ
€ÿ€ÿ
ÿÿ
ÿÿ
û
rD   c                   @  sf   e Zd ZU dZeƒ Zded< eƒ Zded< eƒ Zded< eƒ Z	ded	< eƒ Z
ded
< eƒ Zded< dS )ÚSocketAttributeaý  
    .. attribute:: family
        :type: socket.AddressFamily

        the address family of the underlying socket

    .. attribute:: local_address
        :type: tuple[str, int] | str

        the local address the underlying socket is connected to

    .. attribute:: local_port
        :type: int

        for IP based sockets, the local port the underlying socket is bound to

    .. attribute:: raw_socket
        :type: socket.socket

        the underlying stdlib socket object

    .. attribute:: remote_address
        :type: tuple[str, int] | str

        the remote address the underlying socket is connected to

    .. attribute:: remote_port
        :type: int

        for IP based sockets, the remote port the underlying socket is connected to
    r   r6   r   Úlocal_addressr,   Ú
local_portr)   Ú
raw_socketÚremote_addressÚremote_portN)Ú__name__Ú
__module__r4   Ú__doc__r   r6   Ú__annotations__rF   rG   rH   rI   rJ   rB   rB   rB   rC   rE   c   s   
  rE   c                   @  s,   e Zd Zed	dd„ƒZeed
dd„ƒƒZdS )Ú_SocketProviderr(   úMapping[Any, Callable[[], Any]]c                   sÌ   ddl m‰  tj‡fdd„tj‡ ‡fdd„tj‡fdd„i}z	ˆ ˆj ¡ ƒ‰W n ty2   d ‰Y nw ˆd ur@‡fdd„|tj	< ˆjjt
jt
jfv rd‡fdd„|tj< ˆd urdˆd	 ‰‡fd
d„|tj< |S )Nr   )Úconvert_ipv6_sockaddrc                     s   ˆ j jS ©N)Ú_raw_socketr6   rB   ©ÚselfrB   rC   Ú<lambda>’   s    z2_SocketProvider.extra_attributes.<locals>.<lambda>c                     s   ˆ ˆj  ¡ ƒS rR   ©rS   r9   rB   )ÚconvertrU   rB   rC   rV   “   s    ÿc                     s   ˆ j S rR   )rS   rB   rT   rB   rC   rV   –   ó    c                     ó   ˆ S rR   rB   rB   )ÚpeernamerB   rC   rV   Ÿ   ó    c                     s   ˆ j  ¡ d S )Nr   rW   rB   rT   rB   rC   rV   £   s   r   c                     rZ   rR   rB   rB   )rJ   rB   rC   rV   ¨   r\   )Ú_core._socketsrQ   rE   r6   rF   rH   rS   r5   r.   rI   r   r7   r8   rG   rJ   )rU   Ú
attributesrB   )rX   r[   rJ   rU   rC   Úextra_attributes�   s$   ûÿz _SocketProvider.extra_attributesr)   c                 C  s   d S rR   rB   rT   rB   rB   rC   rS   ¬   s   z_SocketProvider._raw_socketN)r(   rP   )r(   r)   )rK   rL   r4   Úpropertyr_   r   rS   rB   rB   rB   rC   rO   Œ   s    rO   c                   @  ó   e Zd ZdZeddd„ƒZdS )	ÚSocketStreamzu
    Transports bytes over a socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  ó$   �t |tjdd�}tƒ  |¡I dH S )aH  
        Wrap an existing socket object or file descriptor as a socket stream.

        The newly created socket wrapper takes ownership of the socket being passed in.
        The existing socket must already be connected.

        :param sock_or_fd: a socket object or file descriptor
        :return: a socket stream

        T©r   N)rD   r-   ÚSOCK_STREAMr   Úwrap_stream_socket©Úclsr!   r?   rB   rB   rC   Úfrom_socket¹   ó   €zSocketStream.from_socketN)r!   r"   r(   rb   ©rK   rL   r4   rM   Úclassmethodri   rB   rB   rB   rC   rb   ²   ó    rb   c                   @  s6   e Zd Zeddd„ƒZeddd„ƒZeddd„ƒZdS )ÚUNIXSocketStreamr!   r"   r(   c                 Ã  ó(   �t |tjtjdd�}tƒ  |¡I dH S )aR  
        Wrap an existing socket object or file descriptor as a UNIX socket stream.

        The newly created socket wrapper takes ownership of the socket being passed in.
        The existing socket must already be connected.

        :param sock_or_fd: a socket object or file descriptor
        :return: a UNIX socket stream

        Trd   N)rD   r-   re   ÚAF_UNIXr   Úwrap_unix_stream_socketrg   rB   rB   rC   ri   Ê   s
   €ÿzUNIXSocketStream.from_socketÚmessageÚbytesÚfdsúCollection[int | IOBase]ÚNonec                 Ã  ó   �dS )zð
        Send file descriptors along with a message to the peer.

        :param message: a non-empty bytestring
        :param fds: a collection of files (either numeric file descriptors or open file
            or socket objects)
        NrB   )rU   rr   rt   rB   rB   rC   Úsend_fdsÛ   rY   zUNIXSocketStream.send_fdsÚmsglenr,   Úmaxfdsútuple[bytes, list[int]]c                 Ã  rw   )a  
        Receive file descriptors along with a message from the peer.

        :param msglen: length of the message to expect from the peer
        :param maxfds: maximum number of file descriptors to expect from the peer
        :return: a tuple of (message, file descriptors)
        NrB   )rU   ry   rz   rB   rB   rC   Úreceive_fdså   rY   zUNIXSocketStream.receive_fdsN)r!   r"   r(   rn   )rr   rs   rt   ru   r(   rv   )ry   r,   rz   r,   r(   r{   )rK   rL   r4   rl   ri   r   rx   r|   rB   rB   rB   rC   rn   É   s    	rn   c                   @  s:   e Zd ZdZeddd„ƒZeddd	„ƒZ	
dddd„Zd
S )ÚSocketListenerz}
    Listens to incoming socket connections.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  rc   )a  
        Wrap an existing socket object or file descriptor as a socket listener.

        The newly created listener takes ownership of the socket being passed in.

        :param sock_or_fd: a socket object or file descriptor
        :return: a socket listener

        T©r    N)rD   r-   re   r   Úwrap_listener_socketrg   rB   rB   rC   ri   ÷   s   €zSocketListener.from_socketrb   c                 Ã  rw   )zAccept an incoming connection.NrB   rT   rB   rB   rC   Úaccept  rY   zSocketListener.acceptNÚhandlerúCallable[[SocketStream], Any]Ú
task_groupúTaskGroup | Nonerv   c              	   Ã  sp   �ddl m} tƒ 4 I d H š}|d u r| |ƒ ¡I d H }	 |  ¡ I d H }| ||¡ q1 I d H s1w   Y  d S )Nr   )Úcreate_task_group)Ú r…   r   Úenter_async_contextr€   Ú
start_soon)rU   r�   rƒ   r…   ÚstackÚstreamrB   rB   rC   Úserve  s   €þüzSocketListener.serve)r!   r"   r(   r}   )r(   rb   rR   )r�   r‚   rƒ   r„   r(   rv   )	rK   rL   r4   rM   rl   ri   r   r€   r‹   rB   rB   rB   rC   r}   ð   s    ýr}   c                   @  s(   e Zd ZdZeddd„ƒZddd„ZdS )Ú	UDPSocketz{
    Represents an unconnected UDP socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  rc   )aI  
        Wrap an existing socket object or file descriptor as a UDP socket.

        The newly created socket wrapper takes ownership of the socket being passed in.
        The existing socket must be bound to a local address.

        :param sock_or_fd: a socket object or file descriptor
        :return: a UDP socket

        Tr~   N)rD   r-   Ú
SOCK_DGRAMr   Úwrap_udp_socketrg   rB   rB   rC   ri   #  rj   zUDPSocket.from_socketÚdatars   ÚhostÚstrÚportr,   rv   c                 Ã  s   �|   |||ff¡I dH S )z^
        Alias for :meth:`~.UnreliableObjectSendStream.send` ((data, (host, port))).

        N©Úsend)rU   r�   r�   r’   rB   rB   rC   Úsendto2  s   €zUDPSocket.sendtoN)r!   r"   r(   rŒ   )r�   rs   r�   r‘   r’   r,   r(   rv   ©rK   rL   r4   rM   rl   ri   r•   rB   rB   rB   rC   rŒ     s
    rŒ   c                   @  ra   )	ÚConnectedUDPSocketzy
    Represents an connected UDP socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  rc   )aV  
        Wrap an existing socket object or file descriptor as a connected UDP socket.

        The newly created socket wrapper takes ownership of the socket being passed in.
        The existing socket must already be connected.

        :param sock_or_fd: a socket object or file descriptor
        :return: a connected UDP socket

        Trd   N)rD   r-   r�   r   Úwrap_connected_udp_socketrg   rB   rB   rC   ri   A  s   €ýzConnectedUDPSocket.from_socketN)r!   r"   r(   r—   rk   rB   rB   rB   rC   r—   :  rm   r—   c                   @  s(   e Zd ZdZeddd„ƒZddd„ZdS )ÚUNIXDatagramSocketz…
    Represents an unconnected Unix datagram socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  s$   �t |tjtjƒ}tƒ  |¡I dH S )a'  
        Wrap an existing socket object or file descriptor as a UNIX datagram
        socket.

        The newly created socket wrapper takes ownership of the socket being passed in.

        :param sock_or_fd: a socket object or file descriptor
        :return: a UNIX datagram socket

        N)rD   r-   r�   rp   r   Úwrap_unix_datagram_socketrg   rB   rB   rC   ri   ^  s   €zUNIXDatagramSocket.from_socketr�   rs   Úpathr‘   rv   c                 Ã  s   �|   ||f¡I dH S )zCAlias for :meth:`~.UnreliableObjectSendStream.send` ((data, path)).Nr“   )rU   r�   r›   rB   rB   rC   r•   p  s   €zUNIXDatagramSocket.sendtoN)r!   r"   r(   r™   )r�   rs   r›   r‘   r(   rv   r–   rB   rB   rB   rC   r™   U  s
    r™   c                   @  ra   )	ÚConnectedUNIXDatagramSocketz‚
    Represents a connected Unix datagram socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    r!   r"   r(   c                 Ã  ro   )ar  
        Wrap an existing socket object or file descriptor as a connected UNIX datagram
        socket.

        The newly created socket wrapper takes ownership of the socket being passed in.
        The existing socket must already be connected.

        :param sock_or_fd: a socket object or file descriptor
        :return: a connected UNIX datagram socket

        Trd   N)rD   r-   r�   rp   r   Ú#wrap_connected_unix_datagram_socketrg   rB   rB   rC   ri   |  s
   €ÿz'ConnectedUNIXDatagramSocket.from_socketN)r!   r"   r(   rœ   rk   rB   rB   rB   rC   rœ   u  rm   rœ   )r!   r"   r#   r$   r%   r&   r   r'   r    r'   r(   r)   )8Ú
__future__r   r/   r-   Úabcr   Úcollections.abcr   r   r   Ú
contextlibr   Úior   Ú	ipaddressr	   r
   r   Útypingr   r   r   Ú_core._eventloopr   Ú_core._typedattrr   r   r   Ú_streamsr   r   r   Ú_tasksr   r‘   r   rN   Útupler,   r   r   rs   r   r   r   r:   rD   rE   rO   rb   rn   r}   rŒ   r—   r™   rœ   rB   rB   rB   rC   Ú<module>   sH    ýúE)&',
ÿ 