Format: application/x-python-code
1 o
2
3 ;'h�3 � @ sP d dl mZ d dlZd dlZd dlZd dlZd dlmZmZ d dl m
4 Z
5 d dl m Z d dl
6 mZmZ ddlmZmZmZmZmZ dd lmZmZ dd
7 lmZmZm Z m Z ej d krbd d l
8 m Z m Z nd d l!m Z m Z ed
9 �Z"e d�Z#e$e$e%e%f df Z&e$e&df Z'G dd� de�Z(e
10 dd�G dd� de��Z)e
11 dd�G dd� de e) ��Z*dS )� )� annotationsN)�Callable�Mapping)� dataclass)�wraps)�Any�TypeVar� )�BrokenResourceError� EndOfStream�aclose_forcefully�get_cancelled_exc_class� to_thread)�TypedAttributeSet�typed_attribute)�
12 AnyByteStream�
13 ByteStream�Listener� TaskGroup)� � )� TypeVarTuple�Unpack�T_Retval�PosArgsT.c @ s� e Zd ZU dZe� Zded< e� Zded< e� Zded< e� Z ded < e� Z
14 d
15 ed < e� Z d ed
16 < e� Z ded< e� Z
17 ded< e� Zd ed< e� Zded< dS )� TLSAttributez5Contains Transport Layer Security related attributes.�
18 str | None�
19 alpn_protocol�bytes�channel_binding_tls_uniqueztuple[str, str, int]�cipherz*None | dict[str, str | _PCTRTTT | _PCTRTT]�peer_certificatez bytes | None�peer_certificate_binary�bool� server_sidez!list[tuple[str, str, int]] | None�shared_ciphers�
20 ssl.SSLObject�
21 ssl_object�standard_compatible�str� tls_versionN)�__name__�
22 __module__� __qualname__�__doc__r r �__annotations__r r r! r" r$ r% r' r( r* � r0 r0 �GC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/streams/tls.pyr ! s
23 r F)�eqc @ s� e Zd ZU dZded< ded< ded< ded < ded
24 < ed d d d d
25 �d2dd��Zd3d d �Zd4d d �Zd5d"d#�Z d6d7d(d)�Z
26 d8d+d,�Z d5d-d.�Z e
27 d9d0d1��Zd S ):� TLSStreama
28 A stream wrapper that encrypts all sent data and decrypts received data.
29
30 This class has no public initializer; use :meth:`wrap` instead.
31 All extra attributes from :class:`~TLSAttribute` are supported.
32
33 :var AnyByteStream transport_stream: the wrapped stream
34
35 r �transport_streamr# r( r& � _ssl_objectz
36 ssl.MemoryBIO� _read_bio�
37 _write_bioNT)r$ �hostname� ssl_contextr( r$ � bool | Noner8 r r9 �ssl.SSLContext | None�returnc � s� �|du r| }|s'|rt jjnt jj}t �|�}tt d�r'| jt j M _t �� }t �� }t |�t j
38 u r@|j ||||d�} nt �
39 |j ||||d�I dH } | ||| ||d�}
40 |
41 �| j�I dH |
42 S )a�
43 Wrap an existing stream with Transport Layer Security.
44
45 This performs a TLS handshake with the peer.
46
47 :param transport_stream: a bytes-transporting stream to wrap
48 :param server_side: ``True`` if this is the server side of the connection,
49 ``False`` if this is the client side (if omitted, will be set to ``False``
50 if ``hostname`` has been provided, ``False`` otherwise). Used only to create
51 a default context when an explicit context has not been provided.
52 :param hostname: host name of the peer (if host name checking is desired)
53 :param ssl_context: the SSLContext object to use (if not provided, a secure
54 default will be created)
55 :param standard_compatible: if ``False``, skip the closing handshake when
56 closing the connection, and don't raise an exception if the peer does the
57 same
58 :raises ~ssl.SSLError: if the TLS handshake fails
59
60 N�OP_IGNORE_UNEXPECTED_EOF)r$ �server_hostname)r4 r( r5 r6 r7 )�ssl�Purpose� CLIENT_AUTH� SERVER_AUTH�create_default_context�hasattr�optionsr= � MemoryBIO�type�
61 SSLContext�wrap_bior Zrun_sync�_call_sslobject_method� do_handshake) �clsr4 r$ r8 r9 r( �purposeZbio_inZbio_outr' �wrapperr0 r0 r1 �wrapO s@ � �
62
63 �
64 � �zTLSStream.wrap�func�&Callable[[Unpack[PosArgsT]], T_Retval]�args�Unpack[PosArgsT]r c
65 � s� � z||� }W n� t jyY z| jjr!| j�| j�� �I d H | j�� I d H }W n& ty8 | j �
66 � Y n t yP } z
67 | j �
68 � | j�
69 � t |�d }~ww | j �
70 |� Y no t jyn | j�| j�� �I d H Y nZ t jy� } z
71 | j �
72 � | j�
73 � t |�d }~w t jy� } z"| j �
74 � | j�
75 � t|t j�s�|jr�d|jv r�| jr�t |�td �� d }~ww | jjr�| j�| j�� �I d H |S q)NTZ UNEXPECTED_EOF_WHILE_READING)r? �SSLWantReadErrorr7 �pendingr4 �send�read�receiver r6 � write_eof�OSErrorr
76 �write�SSLWantWriteError�SSLSyscallError�SSLError�
77 isinstance� SSLEOFError�strerrorr( )�selfrP rR �result�data�excr0 r0 r1 rJ � sV �
78
79 �� �
80
81 �
82
83 �
84 ���z TLSStream._call_sslobject_method�tuple[AnyByteStream, bytes]c � s: �| � | jj�I dH | j�� | j�� | j| j�� fS )z�
85 Does the TLS closing handshake.
86
87 :return: a tuple of (wrapped byte stream, bytes left in the read buffer)
88
89 N)rJ r5 �unwrapr6 rY r7 r4 rW �rb r0 r0 r1 rg � s
90 �
91
92 zTLSStream.unwrap�Nonec � sP �| j r z | �� I d H W n ty t| j�I d H � w | j�� I d H d S �N)r( rg �
93 BaseExceptionr r4 �acloserh r0 r0 r1 rl � s � �zTLSStream.aclose� � max_bytes�intr c � s$ �| � | jj|�I d H }|st�|S rj )rJ r5 rW r )rb rn rd r0 r0 r1 rX � s
94 �zTLSStream.receive�itemc � s �| � | jj|�I d H d S rj )rJ r5 r[ )rb rp r0 r0 r1 rV � s �zTLSStream.sendc � sb �| � tj�}t�d|�}|r-t|�d��t|�d�p d�}}||fdk r-td|� ���td��)NzTLSv(\d+)(?:\.(\d+))?� r r )rq r z;send_eof() requires at least TLSv1.3; current session uses z7send_eof() has not yet been implemented for TLS streams)�extrar r* �re�matchro �group�NotImplementedError)rb r* rt �major�minorr0 r0 r1 �send_eof� s � " ���zTLSStream.send_eof� Mapping[Any, Callable[[], Any]]c s� i � j j�tj� jjtj� jjtj� jjtj � fdd�tj
95 � fdd�tj � fdd�tj � fdd�tj
96 � fdd�tj� fdd�tj� jji
97 �S )Nc � � j �d�S )NF�r5 � getpeercertr0 rh r0 r1 �<lambda>� s z,TLSStream.extra_attributes.<locals>.<lambda>c r{ )NTr| r0 rh r0 r1 r~ � s �c s � j jS rj )r5 r$ r0 rh r0 r1 r~ s c s � j jr � j �� S d S rj )r5 r$ r% r0 rh r0 r1 r~ s �c � � j S rj �r( r0 rh r0 r1 r~ � c r rj )r5 r0 rh r0 r1 r~ r� )r4 �extra_attributesr r r5 �selected_alpn_protocolr �get_channel_bindingr r! r" r$ r% r( r' r* �versionrh r0 rh r1 r� � s �
98
99
100 �zTLSStream.extra_attributes) r4 r r$ r: r8 r r9 r; r( r# r< r3 )rP rQ rR rS r< r )r< rf �r< ri )rm )rn ro r< r )rp r r< ri �r< rz )r+ r, r- r. r/ � classmethodrO rJ rg rl rX rV ry �propertyr� r0 r0 r0 r1 r3 = s*
101
102 �
103 F
104 .
105
106
107
108 r3 c @ sn e Zd ZU dZded< ded< dZded< d Zd
109 ed < ed dd��Z d!d"dd�Z d#dd �Z
110 e d$d d ��Z dS )%� TLSListenera�
111 A convenience listener that wraps another listener and auto-negotiates a TLS session
112 on every accepted connection.
113
114 If the TLS handshake times out or raises an exception,
115 :meth:`handle_handshake_error` is called to do whatever post-mortem processing is
116 deemed necessary.
117
118 Supports only the :attr:`~TLSAttribute.standard_compatible` extra attribute.
119
120 :param Listener listener: the listener to wrap
121 :param ssl_context: the SSL context object
122 :param standard_compatible: a flag passed through to :meth:`TLSStream.wrap`
123 :param handshake_timeout: time limit for the TLS handshake
124 (passed to :func:`~anyio.fail_after`)
125 z
126 Listener[Any]�listenerzssl.SSLContextr9 Tr# r( � �float�handshake_timeoutre rk �streamr r< ri c � sL �t |�I dH t| t� �st�t�jd| d� t| t�r#t| t� �r$� dS )a�
127 Handle an exception raised during the TLS handshake.
128
129 This method does 3 things:
130
131 #. Forcefully closes the original stream
132 #. Logs the exception (unless it was a cancellation exception) using the
133 ``anyio.streams.tls`` logger
134 #. Reraises the exception if it was a base exception or a cancellation exception
135
136 :param exc: the exception
137 :param stream: the original stream
138
139 NzError during TLS handshake)�exc_info)r r_ r
140 �logging� getLoggerr+ � exception� Exception)re r� r0 r0 r1 �handle_handshake_error" s �
141 ��z"TLSListener.handle_handshake_errorN�handler�Callable[[TLSStream], Any]�
142 task_group�TaskGroup | Nonec � s2 �t � �d� �fdd� �}�j�||�I d H d S )Nr� r r< ri c
143 � s� �ddl m} z$|�j�� tj| �j�jd�I d H }W d � n1 s%w Y W n tyG } z��|| �I d H W Y d }~d S d }~ww � |�I d H d S )Nr )�
144 fail_after)r9 r( ) � r� r� r3 rO r9 r( rk r� )r� r� Zwrapped_streamre �r� rb r0 r1 �handler_wrapperG s � � �� ��z*TLSListener.serve.<locals>.handler_wrapper)r� r r< ri )r r� �serve)rb r� r� r� r0 r� r1 r� B s �zTLSListener.servec � s �| j �� I d H d S rj )r� rl rh r0 r0 r1 rl Y s �zTLSListener.acloserz c s t j� fdd�iS )Nc r rj r� r0 rh r0 r1 r~ _ r� z.TLSListener.extra_attributes.<locals>.<lambda>)r r( rh r0 rh r1 r� \ s �z TLSListener.extra_attributes)re rk r� r r< ri rj )r� r� r� r� r< ri r� r� )
145 r+ r, r- r. r/ r( r� � staticmethodr� r� rl r� r� r0 r0 r0 r1 r�
146 s
147 " �
148 r� )+�
149 __future__r r� rs r? �sys�collections.abcr r � dataclassesr � functoolsr �typingr r r� r
150 r r r
151 r Z_core._typedattrr r �abcr r r r � version_infor r Ztyping_extensionsr r �tupler) Z_PCTRTTZ_PCTRTTTr r3 r� r0 r0 r0 r1 �<module> s2
152 M