Format: application/x-python-code
1 o
2
3 ;'hv � @ s� d dl mZ d dlZd dlmZ d dlmZmZmZ d dl m
4 Z
5 d dl m Z d dl
6 mZmZ d dlmZ d d lmZ d d
7 lmZmZmZ d d lmZmZmZ d
8 dlm Z m Z m Z d
9 dl m Z ee!eef Z"e#e!e$f Z%ee%e!f Z&e#e'e%f Z(e#e'e!f Z)ed�Z*G dd� d�Z+G dd� de�Z,G dd� de�Z-G dd� de e-�Z.G dd� de.�Z/G dd � d e e. e-�Z0G d d � d e e( e-�Z1G d d � d e e' e-�Z2G d!d"� d"e e) e-�Z3G d#d$� d$e e' e-�Z4dS )%� )� annotationsN)�abstractmethod)�Callable�
10 Collection�Mapping)�AsyncExitStack)�IOBase)� IPv4Address� IPv6Address)�
11 AddressFamily)�
12 TracebackType)�Any�TypeVar�Union� )�TypedAttributeProvider�TypedAttributeSet�typed_attribute� )�
13 ByteStream�Listener�UnreliableObjectStream)� TaskGroup�T_Retvalc @ s e Zd Zddd�Zdd d
14 �ZdS )�_NullAsyncContextManager�return�Nonec � � �d S �N� ��selfr r �HC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/abc/_sockets.py�
15 __aenter__ s �z#_NullAsyncContextManager.__aenter__�exc_type�type[BaseException] | None�exc_val�BaseException | None�exc_tb�TracebackType | None� bool | Nonec � r r r )r! r$ r& r( r r r" � __aexit__! s �z"_NullAsyncContextManager.__aexit__N)r r )r$ r% r& r' r( r) r r* )�__name__�
16 __module__� __qualname__r# r+ r r r r" r s
17 r c @ sb e Zd ZU e� Zded< e� Zded< e� Zded< e� Zded< e� Z ded < e� Z
18 ded
19 < d S ) �SocketAttributer �family� SockAddrType�
20 local_address�int�
21 local_port�
22 socket.socket�
23 raw_socket�remote_address� remote_portN) r, r- r. r r0 �__annotations__r2 r4 r6 r7 r8 r r r r" r/ * s
24 r/ c @ s, e Zd Zed dd��Zeed
25 dd���ZdS ) �_SocketProviderr � Mapping[Any, Callable[[], Any]]c s� ddl m� tj�fdd�tj� �fdd�tj�fdd�i}z � �j�� ��W n ty2 d �Y nw �d ur@�fdd�|tj < �jjt
26 j t
27 j fv rd�fdd�|tj
28 < �d urd�d ��fd
29 d�|tj< |S ) Nr )�convert_ipv6_sockaddrc s � j jS r )� _raw_socketr0 r r r r" �<lambda>? s z2_SocketProvider.extra_attributes.<locals>.<lambda>c s � �j �� �S r �r= � getsocknamer )�convertr! r r" r> @ s �c s � j S r )r= r r r r" r> C � c � � S r r r )�peernamer r" r> L � c s � j �� d S )Nr r? r r r r" r> Q s r c rC r r r )r8 r r" r> U rE )Z_core._socketsr< r/ r0 r2 r6 r= � getpeername�OSErrorr7 r �AF_INET�AF_INET6r4 r8 )r! �
30 attributesr )rA rD r8 r! r" �extra_attributes: s& � �
31 �z _SocketProvider.extra_attributesr5 c C s d S r r r r r r" r= Y s z_SocketProvider._raw_socketN)r r; )r r5 )r, r- r. �propertyrK r r= r r r r" r: 9 s r: c @ � e Zd ZdZdS )� SocketStreamzu
32 Transports bytes over a socket.
33
34 Supports all relevant extra attributes from :class:`~SocketAttribute`.
35 N�r, r- r. �__doc__r r r r" rN _ � rN c @ s( e Zd Zeddd��Zedd
36 d��ZdS )�UNIXSocketStream�message�bytes�fds�Collection[int | IOBase]r r c � � �dS )z�
37 Send file descriptors along with a message to the peer.
38
39 :param message: a non-empty bytestring
40 :param fds: a collection of files (either numeric file descriptors or open file
41 or socket objects)
42 Nr )r! rS rU r r r" �send_fdsh rB zUNIXSocketStream.send_fds�msglenr3 �maxfds�tuple[bytes, list[int]]c � rW )a
43 Receive file descriptors along with a message from the peer.
44
45 :param msglen: length of the message to expect from the peer
46 :param maxfds: maximum number of file descriptors to expect from the peer
47 :return: a tuple of (message, file descriptors)
48 Nr )r! rY rZ r r r" � receive_fdsr rB z UNIXSocketStream.receive_fdsN)rS rT rU rV r r )rY r3 rZ r3 r r[ )r, r- r. r rX r\ r r r r" rR g s
49 rR c @ s, e Zd ZdZeddd��Z ddd d
50 �ZdS )�SocketListenerz}
51 Listens to incoming socket connections.
52
53 Supports all relevant extra attributes from :class:`~SocketAttribute`.
54 r rN c � rW )z Accept an incoming connection.Nr r r r r" �accept� rB zSocketListener.acceptN�handler� Callable[[SocketStream], Any]�
55 task_group�TaskGroup | Noner c � sp �ddl 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)� rc r �enter_async_contextr^ Z
56 start_soon)r! r_ ra rc �stack�streamr r r" �serve� s � ��zSocketListener.serve)r rN r )r_ r` ra rb r r )r, r- r. rP r r^ rh r r r r" r] } s �r] c @ s e Zd ZdZd
57 d
58 d �Zd S )� UDPSocketz{
59 Represents an unconnected UDP socket.
60
61 Supports all relevant extra attributes from :class:`~SocketAttribute`.
62 �datarT �host�str�portr3 r r c � s �| � |||ff�I dH S )z^
63 Alias for :meth:`~.UnreliableObjectSendStream.send` ((data, (host, port))).
64
65 N��send)r! rj rk rm r r r" �sendto� s �zUDPSocket.sendtoN)rj rT rk rl rm r3 r r �r, r- r. rP rp r r r r" ri � s ri c @ rM )�ConnectedUDPSocketzy
66 Represents an connected UDP socket.
67
68 Supports all relevant extra attributes from :class:`~SocketAttribute`.
69 NrO r r r r" rr � rQ rr c @ s e Zd ZdZd dd �Zd
70 S ) �UNIXDatagramSocketz�
71 Represents an unconnected Unix datagram socket.
72
73 Supports all relevant extra attributes from :class:`~SocketAttribute`.
74 rj rT �pathrl r r c � s �| � ||f�I dH S )zCAlias for :meth:`~.UnreliableObjectSendStream.send` ((data, path)).Nrn )r! rj rt r r r" rp � s �zUNIXDatagramSocket.sendtoN)rj rT rt rl r r rq r r r r" rs � s rs c @ rM )�ConnectedUNIXDatagramSocketz�
75 Represents a connected Unix datagram socket.
76
77 Supports all relevant extra attributes from :class:`~SocketAttribute`.
78 NrO r r r r" ru � rQ ru )5�
79 __future__r �socket�abcr �collections.abcr r r �
80 contextlibr �ior � ipaddressr r
81 r �typesr �typingr
82 r r Z_core._typedattrr r r Z_streamsr r r �_tasksr rl Z
83 IPAddressType�tupler3 ZIPSockAddrTyper1 rT Z
84 UDPPacketTypeZUNIXDatagramPacketTyper r r/ r: rN rR r] ri rr rs ru r r r r" �<module> s>
85 &
86 �