U Rh^ @sddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddlm Z mZddlmZmZddlmZmZmZmZdd lmZdd lmZmZmZmZmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%d dl&m'Z'd dl(m)Z)d dl*m+Z+d dl,m-Z-m.Z.ej/dkr,ddl0m1Z1e2eddZ3eej4ej5ej6fZ7eej5ej6fZ8edddddddddddddd d!d"Z9eddddd#dddd$dd%ddd d&d"Z9edddddd'dddd(ddd%ddd) d*d"Z9edddddd'dddd+ddd%dd,d) d-d"Z9eddd.ddddd,d/d0d"Z9dd1dd2dd3d4ddddddd%dd5d) d6d"Z9d7d8d9d:d;Z:ddejj4dddd?d@dAdBZ;ddddddJdKdLdMZ=ej4ddd1dNddd>ddddOdPdQdRZ>dddSdTdDdUdVdWdXZ?dddSd7dTdDdYdZd[d\Z@ddddd]d^d_d`dadddbdcdddeZAdzdfddgdhdidjZBdkdldmdndoZCdkdldmdpdqZDdrdsdtdudvZEdTdDddkdwdxdyZFdS){) annotationsN) Awaitable) IPv6Address ip_address)PathLikechmod) AddressFamily SocketKind)AnyLiteralcastoverload) to_thread) ConnectedUDPSocketConnectedUNIXDatagramSocket IPAddressTypeIPSockAddrTypeSocketListener SocketStream UDPSocketUNIXDatagramSocketUNIXSocketStream) MultiListener) TLSStream)get_async_backend)aclose_forcefully)Event)create_task_group move_on_after) )ExceptionGroup IPPROTO_IPV6).) local_host ssl_contexttls_standard_compatiblehappy_eyeballs_delayrintzIPAddressType | Nonezssl.SSLContext | Noneboolstrfloatr) remote_host remote_portr&r'r( tls_hostnamer)returncsdSNr.r/r&r'r(r0r)r3r3:./venv/lib/python3.8/site-packages/anyio/_core/_sockets.py connect_tcp.s r6)r&r(r0r)zssl.SSLContextz str | NonecsdSr2r3r4r3r3r5r6<s )r&r'r(r0r)z Literal[True]) r.r/r&tlsr'r(r0r)r1csdSr2r3r.r/r&r7r'r(r0r)r3r3r5r6Js zLiteral[False]rcsdSr2r3r8r3r3r5r6Ys )r&r))r.r/r&r)r1csdSr2r3)r.r/r&r)r3r3r5r6hsFTg?)r&r7r'r(r0r)zSocketStream | TLSStreamc s^dddddfdd }tdtj} |r\tt|dIdH} | d^} } t|} z t|} Wntk r(t| | tjd IdH} d }}g}| D]x^}}}|tjkr|sd }| d||dfq|tj kr|s|rd }| d ||dfq| ||dfqYn,Xt | t rFtj| jfg}ntj | jfg}gt4IdHTt|D]D\}\}}t}|||t||IdHW5QRXqpW5QIdHRXdkrtd krdntd }td||s |s |rZz&tjd |p"t|||dIdHWStk rXtIdHYnXS)a~ Connect to a host using the TCP protocol. This function implements the stateless version of the Happy Eyeballs algorithm (RFC 6555). If ``remote_host`` is a host name that resolves to multiple IP addresses, each one is tried until one connection attempt succeeds. If the first attempt does not connected within 250 milliseconds, a second attempt is started using the next address in the list, and so on. On IPv6 enabled systems, an IPv6 address (if available) is tried first. When the connection has been established, a TLS handshake will be done if either ``ssl_context`` or ``tls_hostname`` is not ``None``, or if ``tls`` is ``True``. :param remote_host: the IP address or host name to connect to :param remote_port: port on the target host to connect to :param local_host: the interface address or name to bind the socket to before connecting :param tls: ``True`` to do a TLS handshake with the connected stream and return a :class:`~anyio.streams.tls.TLSStream` instead :param ssl_context: the SSL context object to use (if omitted, a default context is created) :param tls_standard_compatible: If ``True``, performs the TLS shutdown handshake before closing the stream and requires that the server does this as well. Otherwise, :exc:`~ssl.SSLEOFError` may be raised during reads from the stream. Some protocols, such as HTTP, require this option to be ``False``. See :meth:`~ssl.SSLContext.wrap_socket` for details. :param tls_hostname: host name to check the server certificate against (defaults to the value of ``remote_host``) :param happy_eyeballs_delay: delay (in seconds) before starting the next connection attempt :return: a socket stream object if no TLS handshake was done, otherwise a TLS stream :raises OSError: if the connection attempt fails Nr,rNone)r.eventr1c szzz|IdH}Wn6tk rP}z|WYW.try_connectrfamilytypeFTrz#multiple connection attempts failedzAll connection attempts failed)Z server_sideZhostnamer'Zstandard_compatible)rsocket AF_UNSPEC getaddrinfor,r ValueError SOCK_STREAMAF_INET6insertAF_INETr= isinstancer compressedr enumeraterZ start_soonr waitlenr#r<rwrap BaseExceptionr)r.r/r&r7r'r(r0r)rFrHgai_res_Z target_hostZaddr_objZv6_foundZv4_foundZ target_addrsafrestZsaiaddrr:causer3rBr5r6rsr.     ,    zstr | bytes | PathLike[Any]r)pathr1cst|}t|IdHS)z Connect to the given UNIX socket. Not available on Windows. :param path: path to the socket :return: a socket stream object N)osfspathr connect_unix)r`r3r3r5rcs rc)r& local_portrHbacklog reuse_portAnyIPAddressFamilyzMultiListener[SocketStream])r&rerHrfrgr1cst}t|d}|dk r t|nd}t|||tjdkr>tjjndtj tj BdIdH}g}z t t |D]^}} } } tjdkr| tjk rqjt|} | dtjdkr| tjtjdn| tjtjd|r| tjtjd|tjkr:| ttjdd| dkr:| ddd\} }| | ddt|f} | | | ||| }||qjWn2tk r|D]}|IdHq|YnXt|S) a Create a TCP socket listener. :param local_port: port number to listen on :param local_host: IP address of the interface to listen on. If omitted, listen on all IPv4 and IPv6 interfaces. To listen on all interfaces on a specific address family, use ``0.0.0.0`` for IPv4 or ``::`` for IPv6. :param family: address family (used if ``local_host`` was omitted) :param backlog: maximum number of queued incoming connections (up to a maximum of 2**16, or 65536) :param reuse_port: ``True`` to allow multiple sockets to bind to the same address/port (not supported on Windows) :return: a list of listener objects rdNwin32rrHrIflagsFr%)rminr,rLsysplatformrJr rN AI_PASSIVE AI_ADDRCONFIGsortedr; setblockingZ setsockoptZ SOL_SOCKETZSO_EXCLUSIVEADDRUSEZ SO_REUSEADDRZ SO_REUSEPORTrOr$Z IPV6_V6ONLYsplitr*bindlistencreate_tcp_listenerr=rXr?r)r&rerHrfrgrCrYZ listenersZfamkindrZsockaddr raw_socketr^scope_idZlistenerr3r3r5rwsH         rw)moderfz int | Noner)r`r|rfr1csZt|d}t||tjIdH}z||t|WStk rT|YnXdS)a Create a UNIX socket listener. Not available on Windows. :param path: path of the socket :param mode: permissions to set on the socket :param backlog: maximum number of queued incoming connections (up to a maximum of 2**16, or 65536) :return: a listener object .. versionchanged:: 3.0 If a socket already exists on the file system in the given path, it will be removed first. rdN) rmsetup_unix_local_socketrJrNrvrcreate_unix_listenerrXclose)r`r|rfrzr3r3r5r~Ss  r~)r&rergr)rHr&rergr1cs|tjkr|std|r`tt|||tjtjtjBdIdH}t t |dd}|dd}n|tj krpd}nd}t ||d|IdH}t t|S)uY Create a UDP socket. If ``port`` has been given, the socket will be bound to this port on the local machine, making this socket suitable for providing UDP based services. :param family: address family (``AF_INET`` or ``AF_INET6``) – automatically determined from ``local_host`` if omitted :param local_host: IP address or host name of the local interface to bind to :param local_port: local port to bind to :param reuse_port: ``True`` to allow multiple sockets to bind to the same address/port (not supported on Windows) :return: a UDP socket z-Either "family" or "local_host" must be givenrjNr)z::r)z0.0.0.0r)rrKrMrLr,rJ SOCK_DGRAMrprqr rhrOrcreate_udp_socketr)rHr&rergrYrDsockr3r3r5rss,   r)rHr&rergr)r.r/rHr&rergr1c sd}|rLtt|||tjtjtjBdIdH}tt|dd}|dd}tt|||tjdIdH}tt|dd}|dd}t ||||IdH} tt | S)u Create a connected UDP socket. Connected UDP sockets can only communicate with the specified remote host/port, an any packets sent from other sources are dropped. :param remote_host: remote host to set as the default target :param remote_port: port on the remote host to set as the default target :param family: address family (``AF_INET`` or ``AF_INET6``) – automatically determined from ``local_host`` or ``remote_host`` if omitted :param local_host: IP address or host name of the local interface to bind to :param local_port: local port to bind to :param reuse_port: ``True`` to allow multiple sockets to bind to the same address/port (not supported on Windows) :return: a connected UDP socket NrjrrrG) rLr,rJrrprqr rhrrr) r.r/rHr&rergrDrYZremote_addressrr3r3r5create_connected_udp_sockets4     r) local_path local_modez"None | str | bytes | PathLike[Any]r)rrr1cs(t||tjIdH}t|dIdHS)a8 Create a UNIX datagram socket. Not available on Windows. If ``local_path`` has been given, the socket will be bound to this path, making this socket suitable for receiving datagrams from other processes. Other processes can send datagrams to this socket only if ``local_path`` is set. If a socket already exists on the file system in the ``local_path``, it will be removed first. :param local_path: the path on which to bind to :param local_mode: permissions to set on the local socket :return: a UNIX datagram socket N)r}rJrrcreate_unix_datagram_socket)rrrzr3r3r5rs  rr) remote_pathrrr1cs2t|}t||tjIdH}t||IdHS)a Create a connected UNIX datagram socket. Connected datagram sockets can only communicate with the specified remote path. If ``local_path`` has been given, the socket will be bound to this path, making this socket suitable for receiving datagrams from other processes. Other processes can send datagrams to this socket only if ``local_path`` is set. If a socket already exists on the file system in the ``local_path``, it will be removed first. :param remote_path: the path to set as the default target :param local_path: the path on which to bind to :param local_mode: permissions to set on the local socket :return: a connected UNIX datagram socket N)rarbr}rJrrr)rrrrzr3r3r5%create_connected_unix_datagram_sockets  rrHrIprotorkzbytes | str | Nonezstr | int | Nonezint | AddressFamilyzint | SocketKindzAlist[tuple[AddressFamily, SocketKind, int, str, tuple[str, int]]])hostportrHrIrrkr1c svt|trFz|d}WqJtk rBddl}|j|dd}YqJXn|}tj||||||dIdH}dd|DS) a Look up a numeric IP address given a host name. Internationalized domain names are translated according to the (non-transitional) IDNA 2008 standard. .. note:: 4-tuple IPv6 socket addresses are automatically converted to 2-tuples of (host, port), unlike what :func:`socket.getaddrinfo` does. :param host: host name :param port: port number :param family: socket family (`'AF_INET``, ...) :param type: socket type (``SOCK_STREAM``, ...) :param proto: protocol number :param flags: flags to pass to upstream ``getaddrinfo()`` :return: list of tuples containing (family, type, proto, canonname, sockaddr) .. seealso:: :func:`socket.getaddrinfo` asciirNT)Zuts46rcSs(g|] \}}}}}||||t|fqSr3)convert_ipv6_sockaddr).0rHrIrZ canonnameryr3r3r5 ;s zgetaddrinfo..)rRr,encodeUnicodeEncodeErroridnarrL) rrrHrIrrkZ encoded_hostrrYr3r3r5rLs$  rLrzAwaitable[tuple[str, str]])ryrkr1cCst||S)a Look up the host name of an IP address. :param sockaddr: socket address (e.g. (ipaddress, port) for IPv4) :param flags: flags to pass to upstream ``getnameinfo()`` :return: a tuple of (host name, service name) .. seealso:: :func:`socket.getnameinfo` )r getnameinfo)ryrkr3r3r5rAs rz socket.socketzAwaitable[None])rr1cCs t|S)aI Wait until the given socket has data to be read. This does **NOT** work on Windows when using the asyncio backend with a proactor event loop (default on py3.8+). .. warning:: Only use this on raw sockets that have not been wrapped by any higher level constructs like socket streams! :param sock: a socket object :raises ~anyio.ClosedResourceError: if the socket was closed while waiting for the socket to become readable :raises ~anyio.BusyResourceError: if another task is already waiting for the socket to become readable )rwait_socket_readablerr3r3r5rOsrcCs t|S)aG Wait until the given socket can be written to. This does **NOT** work on Windows when using the asyncio backend with a proactor event loop (default on py3.8+). .. warning:: Only use this on raw sockets that have not been wrapped by any higher level constructs like socket streams! :param sock: a socket object :raises ~anyio.ClosedResourceError: if the socket was closed while waiting for the socket to become writable :raises ~anyio.BusyResourceError: if another task is already waiting for the socket to become writable )rwait_socket_writablerr3r3r5rcsrz+tuple[str, int, int, int] | tuple[str, int]ztuple[str, int])ryr1cCsXt|trPt|dkrP|\}}}}|rF|dd}|d||fS||fSn|SdS)a Convert a 4-tuple IPv6 socket address to a 2-tuple (address, port) format. If the scope ID is nonzero, it is added to the address, separated with ``%``. Otherwise the flow id and scope id are simply cut off from the tuple. Any other kinds of socket addresses are returned as-is. :param sockaddr: the result of :meth:`~socket.socket.getsockname` :return: the converted socket address rlrN)rRtuplerVrt)ryrrZflowinfor{r3r3r5r|s  r)r`r|socktyper1c s |dk rt|}|dszt|}Wn>tk rh}z |jtjtjtjtj fkrXW5d}~XYqXt |j rt |nd}t t j|}|d|dk rzs   ,        " " $$ "S!24%1