a
Mer @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dlZ d dl
Z d dl Z dZ
dZed ZG dd deZG dd d ZG d
d d ZG d d
d
ZG dd deZG dd dZG dd deZdS ) N)AnyOptionali c @ s e Zd ZdS )
UnexpectedEOFN)__name__
__module__ __qualname__ r
r
4/usr/lib/python3.9/site-packages/dns/quic/_common.pyr s r c @ s4 e Zd Zdd Zdd Zdd Zdd Zd d
Zd S ) Bufferc C s d| _ d| _d S )N F)_buffer _seen_endselfr
r
r __init__ s zBuffer.__init__c C s& | j r
d S | j|7 _|r"d| _ d S )NT)r r r dataZis_endr
r
r put s
z
Buffer.putc C s t | j|krdS | jr tdS )NTF)lenr r r )r amountr
r
r have& s
z Buffer.havec C s | j S N)r r r
r
r seen_end- s zBuffer.seen_endc C s" | j d | }| j |d | _ |S r )r )r r r r
r
r get0 s z
Buffer.getN)r r r r r r r r r
r
r
r r s
r c @ sD e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dS )BaseQuicStreamc C s || _ || _t | _d| _d S Nr ) _connection
_stream_idr r
_expecting)r
connection stream_idr
r
r r 8 s zBaseQuicStream.__init__c C s | j S r )r r r
r
r id> s zBaseQuicStream.idc C s |d urt | }nd }|S r )time)r timeout
expirationr
r
r _expiration_from_timeoutA s z'BaseQuicStream._expiration_from_timeoutc C s$ |d ur t |t d}nd }|S )Ng )maxr$ )r r& r% r
r
r _timeout_from_expirationH s z'BaseQuicStream._timeout_from_expirationc C s t |}td|| S )Nz!H)r structpack)r Zdatagramlr
r
r _encapsulateR s zBaseQuicStream._encapsulatec C sB | j || z| jdko&| j | jW S ty< Y dS 0 d S )Nr T)r r r r r r r
r
r _common_add_inputV s
z BaseQuicStream._common_add_inputc C s | j | j | jdd d S )Nr
T)r close_streamr r r r r
r
r _close] s zBaseQuicStream._closeN)
r r r r r# r' r) r- r. r0 r
r
r
r r 7 s
r c @ s0 e Zd Zd ddZdd Zd
dd Zd
d ZdS )BaseQuicConnectionNr c C s d| _ || _|| _|| _d| _|| _i | _tj || _
tj ||f| _ |d u r|dkr| j
t
jkrld}n| j
t
jkr~d}nt|r||f| _nd | _d S )NFr z0.0.0.0z::)Z_doner Z_addressZ_port_closedZ_manager_streamsdnsinetZaf_for_addressZ_aflow_level_address_tupleZ_peersocketZAF_INETZAF_INET6NotImplementedError_source)r r! addressportsource source_portZmanagerr
r
r r c s$ zBaseQuicConnection.__init__c C s | j |= d S r )r3 )r r" r
r
r r/ { s z BaseQuicConnection.close_streamTc C sL t }| j }|d u r"|d }t|| d}| jrD|rDt|d}||fS )Ni r g?)r$ r Z get_timerr( r2 min)r Zclosed_is_specialnowr& intervalr
r
r _get_timer_values~ s
z$BaseQuicConnection._get_timer_valuesc C s t }||kr | j| d S r )r$ r Z handle_timer)r r& r? r
r
r
_handle_timer s z BaseQuicConnection._handle_timer)Nr N)T)r r r r r/ rA rB r
r
r
r r1 b s
r1 c @ s" e Zd Zdee edddZdS )AsyncQuicConnectionN)r% returnc s d S r r
)r r% r
r
r make_stream s z AsyncQuicConnection.make_stream)N)r r r r floatr rE r
r
r
r rC s rC c @ s0 e Zd Zd
ddZdddZd d
Zd d ZdS )BaseQuicManagerNc C sd i | _ || _i | _|d u rZd }t|tr0|}d}tjjjddg||d}|d urZ| | || _
d S )NTZdoqzdoq-i03)Zalpn_protocols verify_mode server_name) _connections_connection_factory_session_tickets
isinstancestraioquicquic
configurationZQuicConfigurationZload_verify_locations_conf)r confrH Zconnection_factoryrI Z verify_pathr
r
r r s
zBaseQuicManager.__init__U r Tc C s | j ||f}|d ur |dfS | j}|rtz$| j||f}t|}||_W n ty` Y n0 t | j
||} nd } t j j
j|| d}
tj||f} |
| t | |
||||| }|| j ||f< |dfS )NF)rQ session_ticket_handlerT)rJ r rR rL popcopysession_ticketKeyError functoolspartialsave_session_ticketrO rP r! ZQuicConnectionr4 r5 r6 connectr$ rK ) r r: r; r< r= Zwant_session_ticketr! rS rX rU ZqconnZ lladdressr
r
r _connect s4
zBaseQuicManager._connectc C s( z| j ||f= W n ty" Y n0 d S r )rJ rY )r r: r; r
r
r closed s zBaseQuicManager.closedc C sL t | j}|tkr:t| j dt }|D ] }| j|= q,|| j||f< d S r )r rL MAX_SESSION_TICKETSlistkeysSESSIONS_TO_DELETE)r r: r; Zticketr, Zkeys_to_deletekeyr
r
r r\ s
z#BaseQuicManager.save_session_ticket)N)rT Nr T)r r r r r^ r_ r\ r
r
r
r rG s
#rG c @ s e Zd ZdddZdS )AsyncQuicManagerrT Nr c C s t d S r )r8 )r r: r; r< r= r
r
r r] s zAsyncQuicManager.connect)rT Nr )r r r r] r
r
r
r re s re )rW rZ r7 r* r$ typingr r Zaioquic.quic.configurationrO Zaioquic.quic.connectionZdns.inetr4 ZQUIC_MAX_DATAGRAMr` rc Exceptionr r r r1 rC rG re r
r
r
r s$ +/H