Format: application/x-python-code
1 o
2
3 7'h) � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZddlmZm Z m
4 Z
5 m Z m Z m
6 Z
7 mZmZ ddlmZ ddlmZmZmZ G dd � d e�ZG d
8 d � d e�ZG d d
9 � d
10 e�ZdS )� )� annotationsN� )� ConnectError�ConnectTimeout�ExceptionMapping� ReadError� ReadTimeout�
11 WriteError� WriteTimeout�map_exceptions)�is_socket_readable� )�
12 SOCKET_OPTION�NetworkBackend�
13 NetworkStreamc @ sj e Zd ZdZdZ d&d'd d
14 �Zd(dd�Zd)d*dd�Zd)d+dd �Zd,d d �Z d&d-d d!�Z
15 d.d$d%�Z dS )/�TLSinTLSStreamz�
16 Because the standard `SSLContext.wrap_socket` method does
17 not work for `SSLSocket` objects, we need this class
18 to implement TLS stream using an underlying `SSLObject`
19 instance in order to support TLS on top of TLS.
20 i @ N�sock�
21 socket.socket� ssl_context�ssl.SSLContext�server_hostname�
22 str | None�timeout� float | Nonec C sN || _ t�� | _t�� | _|j| j| j|d�| _| j �|� | �| jj � d S )N)�incoming�outgoingr )
23 �_sock�ssl� MemoryBIO� _incoming� _outgoing�wrap_bio�ssl_obj�
24 settimeout� _perform_io� do_handshake)�selfr r r r � r' �MC:\Users\shubh\Videos\agent\nova\Lib\site-packages\httpcore/_backends/sync.py�__init__" s
25
26 � zTLSinTLSStream.__init__�func� typing.Callable[..., typing.Any]�return�
27 typing.Anyc
28 C s� d } d }z|� }W n t jt jfy# } z |j}W Y d }~nd }~ww | j�| j�� � |t jkrG| j� | j
29 �}|rB| j � |� n| j �
30 � |d u rM|S q�N)r �SSLWantReadError�SSLWantWriteError�errnor �sendallr �read�SSL_ERROR_WANT_READ�recv�TLS_RECORD_SIZEr �write� write_eof)r& r* �retr1 �e�bufr' r' r( r$ 6 s$
31 ��
32
33 �zTLSinTLSStream._perform_io� max_bytes�int�bytesc
34 C sd t jttti}t|�� | j�|� t� t
35 | � t �
36 | jj|���W d � S 1 s+w Y d S r. )�socketr r �OSErrorr r r r# �typing�castr> r$ � functools�partialr" r3 �r& r< r �exc_mapr' r' r( r3 O s
37 �$�zTLSinTLSStream.read�buffer�Nonec C s� t jttti}t|��, | j�|� |r/| �t �
38 | j j |��}||d � }|sW d � d S W d � d S 1 s:w Y d S r. )
39 r? r r
40 r@ r r r r# r$ rC rD r" r7 )r& rG r rF Znsentr' r' r( r7 W s
41 ��"�zTLSinTLSStream.writec C � | j �� d S r. �r �close�r& r' r' r( rK _ � zTLSinTLSStream.closer c C s t � �r. )�NotImplementedError)r& r r r r' r' r( � start_tlsb s zTLSinTLSStream.start_tls�info�strc C sV |dkr| j S |dkr| j�� S |dkr| j�� S |dkr | jS |dkr)t| j�S d S �N�
42 ssl_object� client_addrZ server_addrr? Z is_readable)r" r � getsockname� getpeernamer �r& rP r' r' r( �get_extra_infoj s
43
44
45 z TLSinTLSStream.get_extra_info�NN)r r r r r r r r )r* r+ r, r- r. �r< r= r r r, r> �rG r> r r r, rH �r, rH �r r r r r r r, r �rP rQ r, r- ) �__name__�
46 __module__� __qualname__�__doc__r6 r) r$ r3 r7 rK rO rX r' r' r' r( r s �
47
48 �r c @ sR e Zd Zd dd�Zd!d"d
49 d�Zd!d#dd�Zd$dd�Z d%d&dd�Zd'd d �ZdS )(�
50 SyncStreamr r r, rH c C s
51 || _ d S r. )r )r& r r' r' r( r) y s
52 zSyncStream.__init__Nr< r= r r r> c C sR t jttti}t|�� | j�|� | j�|�W d � S 1 s"w Y d S r. ) r? r r r@ r r r r# r5 rE r' r' r( r3 | s
53
54
55 $�zSyncStream.readrG c C s� |sd S t jttti}t|��' |r.| j�|� | j�|�}||d � }|sW d � d S W d � d S 1 s9w Y d S r. ) r? r r
56 r@ r r r r# �send)r& rG r rF �nr' r' r( r7 � s
57 ��"�zSyncStream.writec C rI r. rJ rL r' r' r( rK � rM zSyncStream.closer r r r r c C s� t jttti}t|��E z't| jtj �r$t
58 | j|||�W W d � S | j� |� |j | j|d�}W n t
59 yE } z| �� |�d }~ww W d � t|�S 1 sSw Y t|�S )N)r )r? r r r@ r r �
60 isinstancer r � SSLSocketr r# � wrap_socket� ExceptionrK rc )r& r r r rF r �excr' r' r( rO � s0 �
61
62 ��
63
64 ����
65 ��zSyncStream.start_tlsrP rQ r- c C sf |dkrt | jtj�r| jjS |dkr| j�� S |dkr!| j�� S |dkr(| jS |dkr1t| j�S d S rR )rf r r rg �_sslobjrU rV r rW r' r' r( rX � s
66
67
68 zSyncStream.get_extra_info)r r r, rH r. rZ r[ r\ rY r] r^ ) r_ r` ra r) r3 r7 rK rO rX r' r' r' r( rc x s
69
70 � rc c @ s. e Zd Z dddd�Z dddd�ZdS )� SyncBackendN�hostrQ �portr= r r �
71 local_addressr �socket_options�%typing.Iterable[SOCKET_OPTION] | Noner, r c C s� |d u rg }||f}|d u rd n|df}t jttti}t|��& t j|||d�} |D ]}
72 | j|
73 � q*| �t jt j d� W d � t
74 | �S 1 sHw Y t
75 | �S )Nr )�source_addressr
76 ) r? r r r@ r r �create_connection�
77 setsockopt� IPPROTO_TCP� TCP_NODELAYrc ) r& rm rn r ro rp �addressrr rF r �optionr' r' r( � connect_tcp� s(
78 �
79 �
80 � � zSyncBackend.connect_tcp�pathc C s� t jdkr td��|d u rg }tjttti}t|��' t�tj tj
81 �}|D ]}|j |� q%|� |� |�
82 |� W d � t|�S 1 sDw Y t|�S )N�win32z:Attempted to connect to a UNIX socket on a Windows system.)�sys�platform� RuntimeErrorr? r r r@ r r �AF_UNIX� SOCK_STREAMrt r# �connectrc )r& rz r rp rF r rx r' r' r( �connect_unix_socket� s&
83 ��
84
85
86 ��z SyncBackend.connect_unix_socket)NNN) rm rQ rn r= r r ro r rp rq r, r rY )rz rQ r r rp rq r, r )r_ r` ra ry r� r' r' r' r( rl � s �!�rl )�
87 __future__r rC r? r r| rA � _exceptionsr r r r r r r
88 r �_utilsr �baser r r r rc rl r' r' r' r( �<module> s (
89 aC