File: //home/arjun/projects/env/lib/python3.10/site-packages/dns/__pycache__/_ddr.cpython-310.pyc
o
!we � @ s� d dl Z d dlZd dlmZ d dlZd dlZd dlZd dlZd dl Zd dl
Zej � d�Z
G dd� d�Zdd� Zdd � Zd
d � ZdS ) � N)�urlparsez_dns.resolver.arpac @ s6 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd S )
� _SVCBInfoc C s || _ || _|| _|| _d S �N)�bootstrap_address�port�hostname� nameservers)�selfr r r r � r
�A/home/arjun/projects/env/lib/python3.10/site-packages/dns/_ddr.py�__init__ s
z_SVCBInfo.__init__c C s. |d D ]\}}|dkr|| j kr dS qdS )zIVerify that the _SVCBInfo's address is in the cert's subjectAltName (SAN)�subjectAltNamez
IP AddressTF)r )r �cert�name�valuer
r
r �ddr_check_certificate! s
�z _SVCBInfo.ddr_check_certificatec C s t jj}|�� }|jj|_|S r )�dns�query�ssl�create_default_context�
TLSVersion�TLSv1_2�minimum_version)r r �ctxr
r
r �make_tls_context( s
z_SVCBInfo.make_tls_contextc C s� | � � }t�� | }t�| j| jf|��>}|j|| jd��%}|�t j
� |�� |� � |�
� }| �|�W d � W d � S 1 sEw Y W d � d S 1 sUw Y d S )N)�server_hostname)r �time�socket�create_connectionr r � wrap_socketr �
settimeoutr r �
_remaining� do_handshake� getpeercertr )r �lifetimer �
expiration�s�tsr r
r
r �ddr_tls_check_sync. s ���"�z _SVCBInfo.ddr_tls_check_syncNc
� s� �|d u r
t j�� }| �� }t�� | }|�t j�| j�t j
dd | j| j f||| j �I d H 4 I d H � }|�
t j�|��I d H }| �|�W d �I d H S 1 I d H sTw Y d S )Nr )r � asyncbackend�get_default_backendr r � make_socket�inet�af_for_addressr r � SOCK_STREAMr r r# r r! r )r r$ �backendr r% r' r r
r
r �ddr_tls_check_async: s& �
� 0�z _SVCBInfo.ddr_tls_check_asyncr )�__name__�
__module__� __qualname__r r r r( r0 r
r
r
r r s r c C s� | j }tj�|�s g S g }| j�� D ]�}g }|j�tjj j
j �}|d u r%qt |j
�}|jjdd�}d }|j�tjj j
j�}|d urD|j}d|v r�|j�tjj j
j�}|d u s\|j�d�s]q|jd d� �� } | �d�sod| } |d u rud}d|� d |� | � �}
zt|
� |�tj �|
|�� W n ty� Y nw d
|v r�|d u r�d }|�tj �|||�� d |v r�|d u r�d }|�tj � ||d|�� t |�d
kr�|�t ||||�� q|S )NT)�omit_final_dots h2s {?dns}i�����/i� zhttps://�:s dotiU s doqr ) �
nameserverr r, �
is_address�rrset�processing_order�params�get�rdtypes�svcbbase�ParamKey�ALPN�set�ids�target�to_text�PORTr �DOHPATHr �endswith�decode�
startswithr �append�
DoHNameserver� Exception�
DoTNameserver�
DoQNameserver�lenr ) �answerr �infos�rrr �param�alpns�hostr �path�urlr
r
r � _extract_nameservers_from_svcbM s^
��� �rX c C sF g }t | �}|D ]}z
|�|�r|�|j� W q ty Y qw |S )�VReturn a list of TLS-validated resolver nameservers extracted from an SVCB
answer.)rX r( �extendr rL �rP r$ r rQ �infor
r
r �_get_nameservers_sync� s
� �r] c � sN �g }t | �}|D ]}z|�|�I dH r|�|j� W q ty$ Y q w |S )rY N)rX r0 rZ r rL r[ r
r
r �_get_nameservers_async� s � � �r^ )r r � urllib.parser �dns.asyncbackendr �dns.inet�dns.name�dns.nameserver� dns.query�dns.rdtypes.svcbbaser � from_text�_local_resolver_namer rX r] r^ r
r
r
r �<module> s 34