a
�M�e�& � @ s� d dl mZmZ d dlmZ d dlZd dlZd dlZd dl Zd dl
ZG dd� d�Z G dd� de �Z G dd � d e �Z
G d
d � d e �ZG d d
� d
e �ZG dd� de �ZdS )� )�Optional�Union)�urlparseNc @ s� e Zd Zdd� Zdd� Zed�dd�Zed�dd �Zed�d
d �Z e
d�d d
�Z de j
jeee e
eeee j
jd�dd�Zde j
jeee e
ee jjeee j
jd� dd�ZdS )�
Nameserverc C s d S �N� ��selfr r �2/usr/lib/python3.9/site-packages/dns/nameserver.py�__init__ s zNameserver.__init__c C s t �d S r ��NotImplementedErrorr r r r
�__str__ s zNameserver.__str__��returnc C s t �d S r r r r r r
�kind s zNameserver.kindc C s t �d S r r r r r r
�is_always_max_size s z Nameserver.is_always_max_sizec C s t �d S r r r r r r
�answer_nameserver s z Nameserver.answer_nameserverc C s t �d S r r r r r r
� answer_port s zNameserver.answer_portF��request�timeout�source� source_port�max_size�one_rr_per_rrset�ignore_trailingr c C s t �d S r r �r r r r r r r r r r r
�query s
zNameserver.query� r r r r r �backendr r r c � s t �d S r r � r r r r r r r r r r r r
� async_query* s zNameserver.async_queryN)FF)FF)�__name__�
__module__� __qualname__r r �strr �boolr r �intr �dns�message� QueryMessage�floatr �Messager � asyncbackend�Backendr" r r r r
r s>
�� ��r c sd e Zd Zeed�� fdd� Zed�dd�Zed�dd�Zd d
� Z ed�d d �Z
ed�d
d�Z � Z S )�AddressAndPortNameserver��address�portc s t � �� || _|| _d S r )�superr r2 r3 �r r2 r3 �� __class__r r
r 9 s
z!AddressAndPortNameserver.__init__r c C s t �d S r r r r r r
r > s z AddressAndPortNameserver.kindc C s dS )NFr r r r r
r A s z+AddressAndPortNameserver.is_always_max_sizec C s | � � }|� d| j� d| j� �S )N�:�@)r r2 r3 )r Zns_kindr r r
r D s z AddressAndPortNameserver.__str__c C s | j S r )r2 r r r r
r H s z*AddressAndPortNameserver.answer_nameserverc C s | j S r )r3 r r r r
r K s z$AddressAndPortNameserver.answer_port)
r# r$ r% r& r( r r r' r r r r �
__classcell__r r r6 r
r0 8 s r0 c s� e Zd Zdeed�� fdd�
Zdd� Zdejj e
e e ee e e ejj
d�d d
�Zdejj e
e e ee ejje e ejj
d � d d
�Z� ZS )�Do53Nameserver�5 r1 c s t � �||� d S r )r4 r r5 r6 r r
r P s zDo53Nameserver.__init__c C s dS )NZDo53r r r r r
r S s zDo53Nameserver.kindFr c
C sP |r&t jj|| j|| j||||d�}n&t jj|| j|| j||d||ddd� }|S )N)r r3 r r r r T) r r3 r r �raise_on_truncationr r �
ignore_errors�ignore_unexpected)r) r �tcpr2 r3 �udp) r r r r r r r r �responser r r
r V s2
� �
zDo53Nameserver.queryr c
� s` |r.t jj|| j|| j|||||d� I d H } n.t jj|| j|| j||d|||ddd� I d H } | S )N)r r3 r r r r r T)
r r3 r r r= r r r r> r? )r) �
asyncqueryr@ r2 r3 rA )
r r r r r r r r r rB r r r
r" { s6 � � zDo53Nameserver.async_query)r< )FF)FF)r# r$ r% r& r( r r r) r* r+ r, r r' r- r r. r/ r" r: r r r6 r
r; O s6
�� - ��r; c s� e Zd Zdeee eeef ed�� fdd�
Zdd� Zed �d
d �Z d d
� Z
ed �dd�Z e d �dd�Z
dejjeee e eeeejjd�dd�Zdejjeee e eejjeeejjd� dd�Z� ZS )�
DoHNameserverNTF)�url�bootstrap_address�verify�want_getc s& t � �� || _|| _|| _|| _d S r )r4 r rE rF rG rH )r rE rF rG rH r6 r r
r � s
zDoHNameserver.__init__c C s dS )NZDoHr r r r r
r � s zDoHNameserver.kindr c C s dS )NTr r r r r
r � s z DoHNameserver.is_always_max_sizec C s | j S r �rE r r r r
r � s zDoHNameserver.__str__c C s | j S r rI r r r r
r � s z DoHNameserver.answer_nameserverc C s t | j�j}|d u rd}|S )Ni� )r rE r3 )r r3 r r r
r � s zDoHNameserver.answer_portr c C s* t jj|| j|||| j||| j| j d�
S �N)r r r rF r r rG Zpost)r) r �httpsrE rF rG rH r r r r
r � s
�zDoHNameserver.queryr c � s0 t jj|| j|||| j||| j| j d�
I d H S rJ )r) rC rK rE rF rG rH r! r r r
r" � s �zDoHNameserver.async_query)NTF)FFF)FF)r# r$ r% r& r r r' r r r r r r( r r) r* r+ r, r- r r. r/ r" r: r r r6 r
rD � sP �
�
�� ��rD c s� e Zd Zdeeee eeef d�� fdd�
Zdd� Z de
j j e
ee eeeee
j jd
�d d �Zde
j j e
ee eee
jjeee
j jd
� dd�Z� ZS )�
DoTNameserver�U NT)r2 r3 �hostnamerG c s t � �||� || _|| _d S r )r4 r rN rG )r r2 r3 rN rG r6 r r
r � s zDoTNameserver.__init__c C s dS )NZDoTr r r r r
r s zDoTNameserver.kindFr c
C s$ t jj|| j| j|||| j| jd�S �N)r3 r r r �server_hostnamerG )r) r �tlsr2 r3 rN rG r r r r
r s
�zDoTNameserver.queryr c
� s* t jj|| j| j|||| j| jd�I d H S rO )r) rC rQ r2 r3 rN rG r! r r r
r" s �zDoTNameserver.async_query)rM NT)FFF)FF)r# r$ r% r& r( r r r' r r r) r* r+ r, r- r r. r/ r" r: r r r6 r
rL � sH �
� �� ��rL c s� e Zd Zdeeeeef ee d�� fdd�
Zdd� Z de
j j e
ee eeeee
j jd
�d d �Zde
j j e
ee eee
jjeee
j jd
� dd�Z� ZS )�
DoQNameserverrM TN)r2 r3 rG rP c s t � �||� || _|| _d S r )r4 r rG rP )r r2 r3 rG rP r6 r r
r 0 s zDoQNameserver.__init__c C s dS )NZDoQr r r r r
r ; s zDoQNameserver.kindFr c
C s$ t jj|| j| j|||| j| jd�S �N)r3 r r r rG rP )r) r �quicr2 r3 rG rP r r r r
r > s
�zDoQNameserver.queryr c
� s* t jj|| j| j|||| j| jd�I d H S rS )r) rC rT r2 r3 rG rP r! r r r
r" S s �zDoQNameserver.async_query)rM TN)FFF)FF)r# r$ r% r& r( r r' r r r r) r* r+ r, r- r r. r/ r" r: r r r6 r
rR / sH �
� �� ��rR )�typingr r � urllib.parser Zdns.asyncbackendr) Zdns.asyncqueryZdns.inetZ dns.messageZ dns.queryr r0 r; rD rL rR r r r r
�<module> s -UP;