
    QiD3                    ~   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z  ez  Z#de$d<   e%e"e&f   Z'de$d<   e'e"z  Z(de$d<   e%e)e'f   Z*de$d<   e%e)e"f   Z+de$d<    ed      Z,ejZ                  fddd	 	 	 	 	 	 	 	 	 	 	 d,dZ. G d de      Z/ G d de      Z0 G d dee0      Z1 G d  d!e1      Z2 G d" d#ee1   e0      Z3 G d$ d%ee*   e0      Z4 G d& d'ee)   e0      Z5 G d( d)ee+   e0      Z6 G d* d+ee)   e0      Z7y)-    )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_boundc               f   t        | t              r	 t        j                  |       }n?t        | t        j                        r| }n"t        dt        |       j                   d      	 |r	 |j                          |r^	 |j                  t        j                  t        j                  fv r|j                         d   }n|j                         }|st        d      |t        j                  k7  r>|j                  |k7  r/t        d|j                    d	|j                  j                          |j                  |k7  r/t        d
|j                    d	|j                  j                          	 |j'                  d       |S # t        $ rK}|j                  t        j
                  k(  rt        d      ||rt        d      ||rt        d      | d }~ww xY w# t        $ r}t        d      |d }~ww xY w# t        $ r d }Y 2w xY w# t"        $ r" t        | t              r|j%                           w xY w)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)
sock_or_fd	sock_typeaddr_familyr    r!   sockexc
bound_addrs           =/app/.venv/lib/python3.12/site-packages/anyio/abc/_sockets.py_validate_socketr?      s"    *c"	==
3D 
J	.-d:.>.K.K-LHU
 	
"J  " ";;6>>6??"CC!%!1!1!3A!6J!%!1!1!3J  !NOO&***t{{k/I4[5E5E4Ff;;##$& 
 99	!1)..1A		GWX  " 	UKo  
	yyENN* D # !?@cI !NOUXX
	,  J !?@cIJ  "!
"   j#&KKMsn   E? )H ,G <H ?AG3 BH ?	GAGG	G0G++G00H 3H>H HH +H0c                      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<   y)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   r/   r   local_addressr%   
local_portsocket.socket
raw_socketremote_addressremote_portN)__name__
__module__r-   __doc__r   r/   __annotations__rB   rC   rE   rF   rG        r>   rA   rA   c   sX    @ ,-FM-"1"3M<3%'J' / 1J1#2#4NL4&(K(rM   rA   c                  :    e Zd Zedd       Zeedd              Zy)_SocketProviderc                    ddl m t        j                   fdt        j                   fdt        j
                   fdi}	   j                  j                               fd|t        j                  <    j                  j                  t        j                  t        j                  fv r3 fd|t        j                  <   d   fd	|t        j                  <   |S # t        $ r d Y w xY w)
Nr   )convert_ipv6_sockaddrc                 0     j                   j                  S N)_raw_socketr/   selfs   r>   <lambda>z2_SocketProvider.extra_attributes.<locals>.<lambda>   s    D,<,<,C,CrM   c                 D      j                   j                               S rS   rT   r2   )convertrV   s   r>   rW   z2_SocketProvider.extra_attributes.<locals>.<lambda>   s    7  ,,.4rM   c                      j                   S rS   )rT   rU   s   r>   rW   z2_SocketProvider.extra_attributes.<locals>.<lambda>   s    0@0@rM   c                      S rS   rL   )peernames   r>   rW   z2_SocketProvider.extra_attributes.<locals>.<lambda>   s    rM   c                 >     j                   j                         d   S )Nr   rY   rU   s   r>   rW   z2_SocketProvider.extra_attributes.<locals>.<lambda>   s      ,,.q1rM   r   c                      S rS   rL   )rG   s   r>   rW   z2_SocketProvider.extra_attributes.<locals>.<lambda>   s    +rM   )_core._socketsrQ   rA   r/   rB   rE   rT   r.   r'   rF   r   r0   r1   rC   rG   )rV   
attributesrZ   r]   rG   s   ` @@@r>   extra_attributesz _SocketProvider.extra_attributes   s    E ""$C)) , &&(@4

	/6t7G7G7S7S7U/VH
 9IJ556 ""}'<'<m>T>T&UU6J112 #&qk:M
?667!  	H	s    C+ +C98C9c                     y rS   rL   rU   s    r>   rT   z_SocketProvider._raw_socket   s     	rM   N)returnzMapping[Any, Callable[[], Any]])rd   rD   )rH   rI   r-   propertyrb   r   rT   rL   rM   r>   rO   rO      s0     <   rM   rO   c                  "    e Zd ZdZedd       Zy)SocketStreamzu
    Transports bytes over a socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  d      }t               j	                  |       d{   S 7 w)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)r?   r&   SOCK_STREAMr   wrap_stream_socketclsr8   r;   s      r>   from_socketzSocketStream.from_socket   s7       
F,>,>RVW&(;;DAAAA   9AA AN)r8   socket.socket | intrd   rg   rH   rI   r-   rJ   classmethodrn   rL   rM   r>   rg   rg      s     B BrM   rg   c                  B    e Zd Zedd       Zedd       Zedd       Zy)UNIXSocketStreamc                   K   t        |t        j                  t        j                  d      }t	               j                  |       d{   S 7 w)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

        Tri   N)r?   r&   rj   AF_UNIXr   wrap_unix_stream_socketrl   s      r>   rn   zUNIXSocketStream.from_socket   sA       **FNNd
 '(@@FFFF   AA
AAc                   K   yw)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)
        NrL   )rV   messagefdss      r>   send_fdszUNIXSocketStream.send_fds           c                   K   yw)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)
        NrL   )rV   msglenmaxfdss      r>   receive_fdszUNIXSocketStream.receive_fds   r}   r~   N)r8   rp   rd   rt   )rz   bytesr{   zCollection[int | IOBase]rd   None)r   r%   r   r%   rd   ztuple[bytes, list[int]])rH   rI   r-   rr   rn   r   r|   r   rL   rM   r>   rt   rt      s>    G G     rM   rt   c                  R    e Zd ZdZe	 	 	 	 dd       Zedd       Z	 d	 	 	 	 	 d	dZy)
SocketListenerz}
    Listens to incoming socket connections.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  d      }t               j	                  |       d{   S 7 w)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)r?   r&   rj   r   wrap_listener_socketrl   s      r>   rn   zSocketListener.from_socket   s6       
F,>,>dS&(==dCCCCro   c                   K   yw)zAccept an incoming connection.NrL   rU   s    r>   acceptzSocketListener.accept  r}   r~   Nc                
  K   ddl m} t               4 d {   }||j                   |              d {   }	 | j	                          d {   }|j                  ||       +7 P7 27 # 1 d {  7  sw Y   y xY ww)Nr   )create_task_group) r   r   enter_async_contextr   
start_soon)rV   handler
task_groupr   stackstreams         r>   servezSocketListener.serve  st     
 	)!##u!#(#<#<=N=P#QQ
#{{},%%gv6 	 $Q - $##sT   BA(BA.A*A.A,A.(B*A.,A..B 4A75B <B)r8   rp   rd   r   )rd   rg   rS   )r   zCallable[[SocketStream], Any]r   zTaskGroup | Nonerd   r   )	rH   rI   r-   rJ   rr   rn   r   r   r   rL   rM   r>   r   r      sk     D'D 
D D  - - (,7.7 %7 
	7rM   r   c                  *    e Zd ZdZedd       ZddZy)	UDPSocketz{
    Represents an unconnected UDP socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  d      }t               j	                  |       d{   S 7 w)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)r?   r&   
SOCK_DGRAMr   wrap_udp_socketrl   s      r>   rn   zUDPSocket.from_socket#  s6       
F,=,=TR&(88>>>>ro   c                H   K   | j                  |||ff       d{   S 7 w)z^
        Alias for :meth:`~.UnreliableObjectSendStream.send` ((data, (host, port))).

        Nsend)rV   datahostports       r>   sendtozUDPSocket.sendto2  s&     
 YYtTl34444s   " "N)r8   rp   rd   r   )r   r   r   strr   r%   rd   r   rH   rI   r-   rJ   rr   rn   r   rL   rM   r>   r   r     s      ? ?5rM   r   c                  "    e Zd ZdZedd       Zy)ConnectedUDPSocketzy
    Represents an connected UDP socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  d      }t               j	                  |       d{   S 7 w)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

        Tri   N)r?   r&   r   r   wrap_connected_udp_socketrl   s      r>   rn   zConnectedUDPSocket.from_socketA  s=       "

 '(BB4HHHHro   N)r8   rp   rd   r   rq   rL   rM   r>   r   r   :  s     I IrM   r   c                  2    e Zd ZdZe	 	 	 	 dd       ZddZy)UNIXDatagramSocketz
    Represents an unconnected Unix datagram socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  t        j                        }t	               j                  |       d{   S 7 w)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)r?   r&   r   rv   r   wrap_unix_datagram_socketrl   s      r>   rn   zUNIXDatagramSocket.from_socket^  s:       
F,=,=v~~N&(BB4HHHHs   AAA	Ac                D   K   | j                  ||f       d{   S 7 w)zCAlias for :meth:`~.UnreliableObjectSendStream.send` ((data, path)).Nr   )rV   r   paths      r>   r   zUNIXDatagramSocket.sendtop  s     YYd|,,,,s     N)r8   rp   rd   r   )r   r   r   r   rd   r   r   rL   rM   r>   r   r   U  s6     I'I 
I I"-rM   r   c                  *    e Zd ZdZe	 	 	 	 dd       Zy)ConnectedUNIXDatagramSocketz
    Represents a connected Unix datagram socket.

    Supports all relevant extra attributes from :class:`~SocketAttribute`.
    c                   K   t        |t        j                  t        j                  d      }t	               j                  |       d{   S 7 w)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

        Tri   N)r?   r&   r   rv   r   #wrap_connected_unix_datagram_socketrl   s      r>   rn   z'ConnectedUNIXDatagramSocket.from_socket|  sA        ))6>>T
 '(LLTRRRRrx   N)r8   rp   rd   r   rq   rL   rM   r>   r   r   u  s1     S'S 
%S SrM   r   )r8   rp   r9   zsocket.SocketKindr:   zsocket.AddressFamilyr    boolr!   r   rd   rD   )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   rK   tupler%   r   r   r   r   r   r   r3   r?   rA   rO   rg   rt   r   r   r   r   r   rL   rM   r>   <module>r      s   "    9 9 %  .   * * 0 
 C B ,{:y :!#s(O	 +(3.i . !67y 7$)%*$5 	 5: )/(8(8B
 $B#B B &B
 B B BJ&)' &)R#, #LB: B.$| $N)7Xl+_ )7X5&}5 5<I/6 I6-12O-@S"8"? SrM   