관리-도구
편집 파일: resolver.cpython-38.pyc
U /?�fP � @ s� U d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZ ddlmZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Zddl Zddl Zddl Zddl Zddl!Zddl"Zddl#Zddl$Zej%dk�rddl&ZG dd� dej'j(�Z)G dd � d ej'j(�Z*ee e+ e,e-ee.e+f e ej/j0 f Z1e e1 e e+ d �d d �Z2G d d� dej'j3�Z4e4Z3G dd� dej'j(�Z5G dd� dej'j(�Z6G dd� dej'j(�Z7G dd� dej'j(�Z8G dd� dej'j(�Z9G dd� dej'j(�Z:G dd � d �Z;G d d � d e<�Z=G d d � d e=�Z>G d!d"� d"�Z?G d#d$� d$�Z@eejAjBejCjDejEjFf ZGG d%d&� d&e@�ZHG d'd(� d(�ZIG d)d*� d*e@�ZJG d+d,� d,�ZKG d-d.� d.�ZLG d/d0� d0eL�ZMdaNe eM eOd1< eMd2�d3d4�ZPdd2�d5d6�ZQejCjRejEjSd7dd8dddfeejAjBe+f eejCjDe+f eejEjFe+f e,e e+ e,e-e eT e e, e;d9� d:d;�ZUejCjRejEjSd7dd8ddfeejAjBe+f eejCjDe+f eejEjFe+f e,e e+ e,e-e eT e;d<� d=d>�ZVe+e e e;d?�d@dA�ZWejXfeejAjBe+f e-e e>dB�dCdD�ZYeejAjBe+f ejAjBdE�dFdG�ZZdgeTddI�dJdK�Z[ejEjSd7ddfeejAjBe+f ejEjFe,e eM e eT ejAjBdL�dMdN�Z\dOejXdfeejAjBe+f e-e-e eM eMdP�dQdR�Z]ejCjRejEjSd7dd8ddddOejXdf eejAjBe+f eejAjBe+f eejCjDe+f eejEjFe+f e,e e+ e,e-e eT e e, e-e-e eM e;dS�dTdU�Z^ej_ej`gejaejbgiZcdadejeZfejgZhejiZjejkZlejmZnejoZpddejXdddfdVdW�ZqdhdXdY�ZrdidZd[�Zsd\d]� Ztd^d_� Zud`da� Zvdje eM ddb�dcdd�Zwdd2�dedf�ZxdS )k�DNS stub resolver.� N)�Any�Dict�Iterator�List�Optional�Sequence�Tuple�Union)�urlparse�win32c st e Zd ZdZddhZdZ� fdd�Zddd�Zed �� fd d � Z e d d � �Z dd� Z dd� Z dd� Zdd� Z� ZS )�NXDOMAINz"The DNS query name does not exist.�qnames� responsesNc s t � j||� d S �N��super�__init__��self�args�kwargs�� __class__� �:/opt/hc_python/lib/python3.8/site-packages/dns/resolver.pyr > s zNXDOMAIN.__init__c C s\ t |tttf�std��t|�dkr,td��|d kr:i }nt |t�sLtd��t||d�}|S )Nz#qnames must be a list, tuple or setr z(qnames must contain at least one elementz(responses must be a dict(qname=response)�r r )� isinstance�list�tuple�set�AttributeError�len�dict)r r r r r r r � _check_kwargsA s zNXDOMAIN._check_kwargs��returnc sP d| j krt� �� S | j d }t|�dkr0d}nd}d�tt|��}d�||�S )Nr � z None of DNS query names existz!The DNS query name does not existz, z{}: {})r r �__str__r" �join�map�str�format)r r �msgr r r r( M s zNXDOMAIN.__str__c C sr d| j krtd��| j d D ]F}| j d | }z |�� }||krJ|W S W q tk r` Y q X q | j d d S )z%Return the unresolved canonical name.r z parametrized exception requiredr r )r � TypeError�canonical_name� Exception)r �qname�response�cnamer r r r/ X s zNXDOMAIN.canonical_namec C sx t | j�dg ��}t| j�di ��}|j�di �}|j�dg �D ]*}||krV|�|� ||kr@|| ||< q@t||d�S )z3Augment by results from another NXDOMAIN exception.r r r )r r �getr# �appendr )r Ze_nxZqnames0Z responses0Z responses1Zqname1r r r �__add__i s zNXDOMAIN.__add__c C s | j d S )zXAll of the names that were tried. Returns a list of ``dns.name.Name``. r �r �r r r r r u s zNXDOMAIN.qnamesc C s | j d S )z�A map from queried names to their NXDOMAIN responses. Returns a dict mapping a ``dns.name.Name`` to a ``dns.message.Message``. r r7 r8 r r r r | s zNXDOMAIN.responsesc C s | j d | S )zTThe response for query *qname*. Returns a ``dns.message.Message``. r r7 )r r1 r r r r2 � s zNXDOMAIN.response)N)�__name__� __module__� __qualname__�__doc__� supp_kwargs�fmtr r$ r+ r( �propertyr/ r6 r r r2 � __classcell__r r r r r 4 s r c @ s e Zd ZdZdS )�YXDOMAINz8The DNS query name is too long after DNAME substitution.N�r9 r: r; r< r r r r rA � s rA )�errorsr& c C s, g }| D ] }|� d�|d |d �� q|S )z3Turn a resolution errors trace into a list of text.zServer {} answered {}r � )r5 r, )rC Ztexts�errr r r �_errors_to_text� s rF c sH e Zd ZdZdZdedd� ZddhZ� fdd�Z� fd d �Z� Z S ) �LifetimeTimeoutz The resolution lifetime expired.z(%s after {timeout:.3f} seconds: {errors}N����timeoutrC c s t � j||� d S r r r r r r r � s zLifetimeTimeout.__init__c s& t |d �}t� j|d d�|�d�S )NrC rI �; �rI rC )rF r � _fmt_kwargsr) �r r Zsrv_msgsr r r rL � s �zLifetimeTimeout._fmt_kwargs� r9 r: r; r< r- r> r= r rL r@ r r r r rG � s rG c s> e Zd ZdZdZdhZ� fdd�Z� fdd�Zdd � Z� Z S ) �NoAnswerz<The DNS response does not contain an answer to the question.zDThe DNS response does not contain an answer to the question: {query}r2 c s t � j||� d S r r r r r r r � s zNoAnswer.__init__c s t � j|d jd�S )Nr2 )�query)r rL �question)r r r r r rL � s zNoAnswer._fmt_kwargsc C s | j d S )Nr2 r7 r8 r r r r2 � s zNoAnswer.response) r9 r: r; r< r> r= r rL r2 r@ r r r r rO � s rO c sH e Zd ZdZdZdedd� ZddhZ� fdd �Z� fd d �Z� Z S ) � NoNameserversz�All nameservers failed to answer the query. errors: list of servers and respective errors The type of errors is [(server IP address, any object convertible to string)]. Non-empty errors list will add explanatory message () z+All nameservers failed to answer the query.z%s {query}: {errors}NrH �requestrC c s t � j||� d S r r r r r r r � s zNoNameservers.__init__c s( t |d �}t� j|d jd�|�d�S )NrC rS rJ )rP rC )rF r rL rQ r) rM r r r rL � s �zNoNameservers._fmt_kwargsrN r r r r rR � s rR c @ s e Zd ZdZdS )� NotAbsolutezEAn absolute domain name is required but a relative name was provided.NrB r r r r rT � s rT c @ s e Zd ZdZdS )� NoRootSOAzBThere is no SOA RR at the DNS root name. This should never happen!NrB r r r r rU � s rU c @ s e Zd ZdZdS )� NoMetaqueriesz DNS metaqueries are not allowed.NrB r r r r rV � s rV c @ s e Zd ZdZdS )�NoResolverConfigurationzEResolver configuration could not be read or specified no nameservers.NrB r r r r rW � s rW c @ sr e Zd ZdZdejjejjej j ej j e e e e dd�dd�Zdd� Zed�d d �Zd d � Zd d� Zdd� ZdS )�Answera& DNS stub resolver answer. Instances of this class bundle up the result of a successful DNS resolution. For convenience, the answer object implements much of the sequence protocol, forwarding to its ``rrset`` attribute. E.g. ``for a in answer`` is equivalent to ``for a in answer.rrset``. ``answer[i]`` is equivalent to ``answer.rrset[i]``, and ``answer[i:j]`` is equivalent to ``answer.rrset[i:j]``. Note that CNAMEs or DNAMEs in the response may mean that answer RRset's name might not be the query name. N)r1 �rdtype�rdclassr2 � nameserver�portr& c C sX || _ || _|| _|| _|| _|| _|�� | _| jj| _| jj | _ t � � | jj | _ d S r )r1 rY rZ r2 r[ r\ Zresolve_chainingZchaining_resultr/ �answer�rrset�timeZ minimum_ttl� expiration)r r1 rY rZ r2 r[ r\ r r r r s zAnswer.__init__c C s\ |dkr| j jS |dkr | j jS |dkr0| j jS |dkr@| j jS |dkrP| j jS t|��d S )N�name�ttl�coversrZ rY )r^ ra rb rc rZ rY r! )r �attrr r r � __getattr__ s zAnswer.__getattr__r% c C s | j rt| j �pdS �Nr )r^ r" r8 r r r �__len__) s zAnswer.__len__c C s | j rt| j �ptt� �S r )r^ �iterr r8 r r r �__iter__, s zAnswer.__iter__c C s | j d krt�| j | S r �r^ � IndexError�r �ir r r � __getitem__/ s zAnswer.__getitem__c C s | j d krt�| j |= d S r rj rl r r r � __delitem__4 s zAnswer.__delitem__)NN)r9 r: r; r< �dnsra �Name� rdatatype� RdataType� rdataclass� RdataClass�message� QueryMessager r+ �intr re rg ri rn ro r r r r rX � s" �� rX c @ s e Zd ZdZdS )�Answersz5A dict of DNS stub resolver answers, indexed by type.NrB r r r r ry : s ry c @ s� e Zd ZdZedee ee ed d�dd��Ze j fe e e ee f d�dd �Ze j fe e e d�d d �Zejjd �d d�ZdS )� HostAnswerszTA dict of DNS stub resolver answers to a host name lookup, indexed by type. NT)�v6�v4� add_emptyr& c C sF t � }|d k r$|s|jr$||tjj< |d k rB|s6|jrB||tjj< |S r )rz r^ rp rr �AAAA�A)�clsr{ r| r} �answersr r r �makeC s zHostAnswers.make)�familyr& c c s� |t jkr2| �t j�E d H | �t j�E d H d S |t jkrL| �tjj�}n(|t jkrf| �tjj �}nt d|� ���|r�|D ]}|j |fV q|d S )N�unknown address family ) �socket� AF_UNSPEC�addresses_and_families�AF_INET6�AF_INETr4 rp rr r~ r �NotImplementedError�address)r r� r] Zrdatar r r r� S s z"HostAnswers.addresses_and_familiesc C s dd� | � |�D �S )Nc s s | ]}|d V qdS )r Nr )�.0�pairr r r � <genexpr>g s z(HostAnswers.addresses.<locals>.<genexpr>)r� )r r� r r r � addressesf s zHostAnswers.addressesr% c C s | � tjj| � tjj��}|jS r )r4 rp rr r~ r r/ )r r] r r r r/ j s zHostAnswers.canonical_name)NNT)r9 r: r; r< � classmethodr rX �boolr� r� r� rx r r r+ r� r� rp ra rq r/ r r r r rz > s$ ���� rz c @ s@ e Zd ZdZd eedd�dd�Zdd�dd �Zd d�d d �ZdS ) �CacheStatisticszCache Statisticsr N)�hits�missesr& c C s || _ || _d S r �r� r� )r r� r� r r r r r s zCacheStatistics.__init__r% c C s d| _ d| _d S rf r� r8 r r r �resetv s zCacheStatistics.resetc C s t | j| j�S r )r� r� r� r8 r r r �clonez s zCacheStatistics.clone)r r )r9 r: r; r< rx r r� r� r r r r r� o s r� c @ sR e Zd Zdd�dd�Zdd�dd�Zed�dd�Zed�d d �Zed�d d �Z dS ) � CacheBaseNr% c C s t �� | _t� | _d S r )� threading�Lock�lockr� � statisticsr8 r r r r s zCacheBase.__init__c C s | j � | j�� W 5 Q R X dS )z Reset all statistics to zero.N)r� r� r� r8 r r r �reset_statistics� s zCacheBase.reset_statisticsc C s& | j � | jjW 5 Q R � S Q R X dS )z How many hits has the cache had?N)r� r� r� r8 r r r r� � s zCacheBase.hitsc C s& | j � | jjW 5 Q R � S Q R X dS )z"How many misses has the cache had?N)r� r� r� r8 r r r r� � s zCacheBase.missesc C s( | j � | j�� W 5 Q R � S Q R X dS )z�Return a consistent snapshot of all the statistics. If running with multiple threads, it's better to take a snapshot than to call statistics methods such as hits() and misses() individually. N)r� r� r� r8 r r r �get_statistics_snapshot� s z!CacheBase.get_statistics_snapshot) r9 r: r; r r� rx r� r� r� r� r r r r r� ~ s r� c st e Zd ZdZdedd�� fdd� Zdd�dd �Zeee d �d d �Z ee dd �dd�Z dee dd �dd�Z � Z S )�Cachez$Simple thread-safe DNS answer cache.� �r@N)�cleaning_intervalr& c s* t � �� i | _|| _t�� | j | _dS )zf*cleaning_interval*, a ``float`` is the number of seconds between periodic cleanings. N)r r �datar� r_ � next_cleaning)r r� r r r r � s zCache.__init__r% c C sh t � � }| j|krdg }| j�� D ] \}}|j|kr |�|� q |D ] }| j|= qBt � � }|| j | _dS )z&Clean the cache if it's time to do so.N)r_ r� r� �itemsr` r5 r� )r �nowZkeys_to_delete�k�vr r r � _maybe_clean� s zCache._maybe_clean��keyr& c C sz | j �j | �� | j�|�}|dks2|jt�� krP| j jd7 _W 5 Q R � dS | j jd7 _|W 5 Q R � S Q R X dS �aK Get the answer associated with *key*. Returns None if no answer is cached for the key. *key*, a ``(dns.name.Name, dns.rdatatype.RdataType, dns.rdataclass.RdataClass)`` tuple whose values are the query name, rdtype, and rdclass respectively. Returns a ``dns.resolver.Answer`` or ``None``. Nr' ) r� r� r� r4 r` r_ r� r� r� )r r� r� r r r r4 � s z Cache.get�r� �valuer& c C s( | j � | �� || j|< W 5 Q R X dS �a Associate key and value in the cache. *key*, a ``(dns.name.Name, dns.rdatatype.RdataType, dns.rdataclass.RdataClass)`` tuple whose values are the query name, rdtype, and rdclass respectively. *value*, a ``dns.resolver.Answer``, the answer. N)r� r� r� �r r� r� r r r �put� s z Cache.putc C sH | j �8 |dk r$|| jkr:| j|= ni | _t�� | j | _W 5 Q R X dS �a2 Flush the cache. If *key* is not ``None``, only that item is flushed. Otherwise the entire cache is flushed. *key*, a ``(dns.name.Name, dns.rdatatype.RdataType, dns.rdataclass.RdataClass)`` tuple whose values are the query name, rdtype, and rdclass respectively. N)r� r� r_ r� r� )r r� r r r �flush� s z Cache.flush)r� )N)r9 r: r; r< �floatr r� �CacheKeyr rX r4 r� r� r@ r r r r r� � s r� c @ s6 e Zd ZdZdd� Zd dd�dd�Zdd�d d �ZdS ) � LRUCacheNodezLRUCache node.c C s" || _ || _d| _| | _| | _d S rf )r� r� r� �prev�nextr� r r r r � s zLRUCacheNode.__init__N)�noder& c C s || _ |j| _| |j_ | |_d S r �r� r� )r r� r r r � link_after� s zLRUCacheNode.link_afterr% c C s | j | j_ | j| j _d S r r� r8 r r r �unlink� s zLRUCacheNode.unlink)r9 r: r; r< r r� r� r r r r r� � s r� c s� e Zd ZdZdedd�� fdd� Zedd�dd�Zeee d �d d �Z eed �d d �Z ee dd�dd�Z dee dd �dd�Z � ZS )�LRUCachea` Thread-safe, bounded, least-recently-used DNS answer cache. This cache is better than the simple cache (above) if you're running a web crawler or other process that does a lot of resolutions. The LRUCache has a maximum number of nodes, and when it is full, the least-recently used node is removed to make space for a new one. 順 N)�max_sizer& c s> t � �� i | _| �|� tdd�| _| j| j_| j| j_dS )zl*max_size*, an ``int``, is the maximum number of nodes to cache; it must be greater than 0. N)r r r� � set_max_sizer� �sentinelr� r� �r r� r r r r s zLRUCache.__init__c C s |dk r d}|| _ d S )Nr' )r� r� r r r r� s zLRUCache.set_max_sizer� c C s� | j �� | j�|�}|dkr:| j jd7 _W 5 Q R � dS |�� |jjt�� krz| j|j = | j jd7 _W 5 Q R � dS |� | j � | j j d7 _ | j d7 _ |jW 5 Q R � S Q R X dS r� ) r� r� r4 r� r� r� r� r` r_ r� r� r� r� �r r� r� r r r r4 s z LRUCache.getc C sZ | j �J | j�|�}|dks,|jjt�� kr:W 5 Q R � dS |jW 5 Q R � S W 5 Q R X dS )zBReturn the number of cache hits associated with the specified key.Nr )r� r� r4 r� r` r_ r� r� r r r �get_hits_for_key; s zLRUCache.get_hits_for_keyr� c C s� | j �x | j�|�}|dk r.|�� | j|j= t| j�| jkrZ| jj}|�� | j|j= q.t ||�}|� | j� || j|< W 5 Q R X dS r� ) r� r� r4 r� r� r" r� r� r� r� r� )r r� r� r� �gnoder r r r� D s z LRUCache.putc C sr | j �b |dk r8| j�|�}|dk rd|�� | j|j= n,| jj}|| jkr^|j}|�� |}q@i | _W 5 Q R X dS r� )r� r� r4 r� r� r� r� )r r� r� r� r� r r r r� Z s zLRUCache.flush)r� )N)r9 r: r; r< rx r r� r� r rX r4 r� r� r� r@ r r r r r� s r� c @ s� e Zd ZdZdeejjef eej j ef eej j ef e e ee dd�dd�Zeeejj ee f d�dd �Zeejje ef d�d d �Zeejj ee eee e f d �d d�ZdS )� _Resolutiona Helper class for dns.resolver.Resolver.resolve(). All of the "business logic" of resolution is encapsulated in this class, allowing us to have multiple resolve() implementations using different I/O schemes without copying all of the complicated logic. This class is a "friend" to dns.resolver.Resolver and manipulates resolver data structures directly. � BaseResolverN)�resolverr1 rY rZ �tcp�raise_on_no_answer�searchr& c C s� t |t�rtj�|d �}tjj�|�}tj�|�r6t �tj j �|�}tj � |�rTt �|| _ |�||�| _| jd d � | _|| _|| _|| _|| _i | _tjj| _g | _g | _g | _d | _d| _ d| _ d | _ d| _ d S )NF� ) r r+ rp ra � from_textrr rs r� Z is_metatyperV rt ru Z is_metaclassr� �_get_qnames_to_try� qnames_to_tryr rY rZ r� r� �nxdomain_responses�emptyr1 � nameservers�current_nameserversrC r[ � tcp_attempt�retry_with_tcprS �backoff)r r� r1 rY rZ r� r� r� r r r r s2 z_Resolution.__init__r% c C s� t | j�dk�r�| j�d�| _| jjr�| jj�| j| j| jf�}|dk rn|j dkrf| j rft |j d��nd|fS | jj�| jt jj| jf�}|dk r�|j �� t jjkr�|j | j| j< q t j�| j| j| j�}| jjdk r�|j| jj| jj| jjd� |j| jj| jj| jj | jj d� | jj dk �r,| jj |_ | j� | jj | jj!| jj"�| _#| jj$�r`t%�&| j#� | j#dd� | _'g | _(d| _)d| _*d| _+|| _,d| _-|dfS t| j.| jd��dS ) z�Get the next request to send, and check the cache. Returns a (request, answer) tuple. At most one of request or answer will not be None. r N�r2 )� algorithm)�optionsFg�������?r )/r" r �popr1 r� �cacher4 rY rZ r^ r� rO r2 rp rr �ANY�rcoder r� rv Z make_query�keyname�use_tsig�keyring� keyalgorithm�use_edns�edns� ednsflags�payload� ednsoptions�flags�_enrich_nameservers� _nameservers�nameserver_portsr\ r� �rotate�random�shuffler� rC r[ r� r� rS r� r� )r r] rS r r r � next_request� s^ �� �� � z_Resolution.next_requestc C s� | j r:| jd k st�| j�� r"t�d| _d| _ | jddfS d}| js�t| j�dkrbt| j | j d��| jd d � | _| j }t | j d d�| _ | j� d�| _| jp�| j�� | _| j| j|fS )NTFr r� )rS rC � )r� r[ �AssertionErrorZis_always_max_sizer� r� r" r� rR rS rC r� �minr� r� )r r� r r r �next_nameserver� s z_Resolution.next_nameserver)r2 �exr& c C s | j d k st�|r�|d ks t�| j�t| j �| j| j �� ||f� t|tj j �snt|t �snt|t �snt|t �r~| j�| j � n*t|tjj�r�| jr�| j�| j � nd| _dS |d k s�t�t|tjj�s�t�|�� }|tjjk�r�z(t| j| j| j|| j �� | j �� �}W n\ tk �rd } z<| j�t| j �| j| j �� ||f� | j�| j � W Y �dS d }~X Y nX | j j �r�| j j � | j| j| jf|� |j d k�r�| j �r�t!|j"d��|dfS |tjj#k�rtz t| jtj$j%tj&j'|�}W n\ tk �r: } z<| j�t| j �| j| j �� ||f� | j�| j � W Y �dS d }~X Y nX || j(| j< | j j �rp| j j � | jtj$j%| jf|� dS |tjj)k�r�t)� }| j�t| j �| j| j �� ||f� |�nV|tjj*k�s�| j j+�s�| j�| j � | j�t| j �| j| j �� tj�,|�|f� dS d S )NT)NFr� )NT)-r[ r� rC r5 r+ r� Z answer_portr rp � exceptionZ FormError�EOFError�OSErrorr� r� �removerv Z Truncatedr� rw r� ZNOERRORrX r1 rY rZ Zanswer_nameserverr0 r� r� r� r^ r� rO r2 r rr r� rt �INr� rA ZSERVFAIL�retry_servfail�to_text)r r2 r� r� r] �eZyexr r r � query_result s� �� ���� ��� ��� ��� �� z_Resolution.query_result) r9 r: r; r< r rp ra rq r+ rr rs rt ru r� r r r rv rw rX r� r[ � Nameserverr� r� �Messager0 r� r r r r r� s s" � '� K �r� c @ s� e Zd ZU dZejjed< ee e f ed< e ed< e ejj ed< e ed< e ed< e ed< ee ed < eeejje f ed < eejje f ed < e ed < e ed < ee ejj ed< e ed< eed< ee ed< e ed< e ed< ee ed< eee ejjf ed< d;e e dd�dd�Zdd �d d �Zedd �d d!�Zdd �d"d#�Zd<e ee ee e e d$�d%d&�Zejjee e ejj d'�d(d)�Z dej j feeeejje f eejje f dd*�d+d,�Z d-d-ej j!dfeee e f e e ee ejj dd.�d/d0�Z"e dd1�d2d3�Z#e$eee ejjf ee e f e e ejj d4�d5d6��Z%e&eee ejjf d �d7d8��Z'e'j(eee ejjf dd9�d:d8��Z'dS )=r� r �domainr� r\ r� �use_search_by_defaultrI �lifetimer� r� r� r� r� r� r� r� r� r� r� �ndotsr� �/etc/resolv.confTN)�filename� configurer& c C s2 | � � |r.tjdkr | �� n|r.| �|� dS )a� *filename*, a ``str`` or file object, specifying a file in standard /etc/resolv.conf format. This parameter is meaningful only when *configure* is true and the platform is POSIX. *configure*, a ``bool``. If True (the default), the resolver instance is configured in the normal fashion for the operating system the resolver is running on. (I.e. by reading a /etc/resolv.conf file on POSIX systems and from the registry on Windows systems.) r N)r� �sys�platform� read_registry�read_resolv_conf)r r r r r r r � s zBaseResolver.__init__r% c C s� t j�t j�t�� �dd� �| _t| j�dkr:t jj| _g | _ i | _ d| _ g | _ d| _ d| _d| _d| _d| _t jj| _d| _d| _d| _d| _d| _d| _d| _d| _ d| _ dS ) z1Reset all resolver configuration to the defaults.r' Nr �5 Fg @� @rH ) rp ra rq r� r� � gethostnamer� r" �rootr� r� r\ r� r� rI r r� r� �tsig�default_algorithmr� r� r� r� r� r� r� r� r� r r8 r r r r� � s, " zBaseResolver.reset)�fr& c C s g }t |t�r@z t|�}W qJ tk r< td|� ���Y qJX n t�|�}|���}|D �]�}t|�dksV|d dksV|d dkr�qV|�� }t|�dk r�qV|d dkr�|� |d � qV|d dkr�t j � |d �| _ g | _qV|d d k�rg | _|dd � D ]}| j� t j � |�� q�qV|d d krV|dd � D ]�}|d k�rDd | _n�|dk�rX| �� n~d|k�r�zt|�d�d �| _W n ttfk �r� Y nX n>d|k�r.zt|�d�d �| _W n ttfk �r� Y nX �q.qVW 5 Q R X t|�dk�r�td��|| _d S )a� Process *f* as a file in the /etc/resolv.conf format. If f is a ``str``, it is used as the name of the file to open; otherwise it is treated as the file itself. Interprets the following items: - nameserver - name server IP address - domain - local domain name - search - search list for host-name lookup - options - supported options are rotate, timeout, edns0, and ndots z cannot open r �#�;r� r[ r' r� r� Nr� r� TZedns0rI �:r zno nameservers)r r+ �openr� rW � contextlib� nullcontextr" �splitr5 rp ra r� r� r� r� r� rx rI � ValueErrorrk r r� )r r r� �cm�l�tokens�suffix�optr r r r � sT $ z BaseResolver.read_resolv_confc C sN z0t j�� }|jdk r |j| _|j| _|j| _W n tk rH t�Y nX dS )z9Extract resolver configuration from the Windows registry.N)rp Z win32utilZ get_dns_infor� r� r� r! r� )r �infor r r r s zBaseResolver.read_registry)�startr rC r& c C st |d kr| j n|}t�� }|| }|d kr.g }|dk rP|dk rLt||d��nd}||krdt||d��t|| | j�S )Nr rH rK )r r_ rG r� rI )r r r rC r� �durationr r r �_compute_timeout s z BaseResolver._compute_timeout)r1 r� r& c C s� |d kr| j }g }|�� r&|�|� n�|�tjj�}|r�t| j�dkrV| jd d � }n&| j tjjkrx| j d k rx| j g}ng }| j d kr�d}n| j }|D ]}|�|| � q�t|�|kr�|� d|� q�|�|� n |�|� |S )Nr r' ) r� � is_absoluter5 Z concatenaterp ra r r" r� r� r �insert)r r1 r� r� Z abs_qnameZ search_listr r r r r r� 6 s, z BaseResolver._get_qnames_to_try)r� r� r� r& c C s || _ || _|| _dS )z�Add a TSIG signature to each query. The parameters are passed to ``dns.message.Message.use_tsig()``; see its documentation for details. N)r� r� r� )r r� r� r� r r r r� _ s zBaseResolver.use_tsigr )r� r� r� r� r& c C s> |dks|dkrd}n |dkr"d}|| _ || _|| _|| _dS )a� Configure EDNS behavior. *edns*, an ``int``, is the EDNS level to use. Specifying ``None``, ``False``, or ``-1`` means "do not use EDNS", and in this case the other parameters are ignored. Specifying ``True`` is equivalent to specifying 0, i.e. "use EDNS0". *ednsflags*, an ``int``, the EDNS flag values. *payload*, an ``int``, is the EDNS sender's payload field, which is the maximum size of UDP datagram the sender can handle. I.e. how big a response to this message can be. *options*, a list of ``dns.edns.Option`` objects or ``None``, the EDNS options. NFrH Tr )r� r� r� r� )r r� r� r� r� r r r r� o s zBaseResolver.use_edns)r� r& c C s || _ dS )zkOverrides the default flags with your own. *flags*, an ``int``, the message flags to use. N)r� )r r� r r r � set_flags� s zBaseResolver.set_flags)r� r� � default_portr& c C s� g }t |t�r�|D ]�}t |tjj�r*|}nptj�|�rR|�||�}tj�||�}nHzt |�j dkrft �W n$ t k r� t d|� d���Y nX tj�|�}|�|� qnt d�t|����|S )N�httpsz nameserver z\ is not a dns.nameserver.Nameserver instance or text form, IP address, nor a valid https URLz.nameservers must be a list or tuple (not a {}))r r rp r[ r� �inet� is_addressr4 �Do53Nameserverr �schemer� r0 r Z DoHNameserverr5 r, �type)r� r� r� r$ Zenriched_nameserversr[ Zenriched_nameserverr\ r r r r� � s4 � � ��z BaseResolver._enrich_nameserversc C s | j S r )r� r8 r r r r� � s zBaseResolver.nameservers)r� r& c C s | � || j| j� || _dS )a *nameservers*, a ``list`` of nameservers, where a nameserver is either a string interpretable as a nameserver, or a ``dns.nameserver.Nameserver`` instance. Raises ``ValueError`` if *nameservers* is not a list of nameservers. N)r� r� r\ r� )r r� r r r r� � s )r T)NN))r9 r: r; r< rp ra rq �__annotations__r r+ rx r r� r� r r r r� �Optionr r[ r� r r� r r � ErrorTupler r� r r r� rv ZDEFAULT_EDNS_PAYLOADr� r# r� r� r? r� �setterr r r r r� � s� � � G � � � ,�� �� ! �%��r� c @ sH e Zd ZdZejjejjddddddfe ej j e f e ejj e f e ejje f eee eeee ee ed� dd�Zejjejjdddddfe ej j e f e ejj e f e ejje f eee eeee ed � d d �Ze eeed �d d�Zejfe ej j e f eeed�dd�Ze ej j e f ej j d�dd�Z dedd�dd�Z dS )�Resolverr FNTr � r1 rY rZ r� �sourcer� � source_portr r� r& c C s� t | |||||| �} t�� } | �� \} } | dk r4| S | dk s@t�d}|s | �� \}}}|rdt�|� | �| || j�}z|j| ||||d�}W n: t k r� } z | � d|�\}}W Y �qDW 5 d}~X Y nX | � |d�\} }| dk rD| S qDq dS )a� Query nameservers to find the answer to the question. The *qname*, *rdtype*, and *rdclass* parameters may be objects of the appropriate type, or strings that can be converted into objects of the appropriate type. *qname*, a ``dns.name.Name`` or ``str``, the query name. *rdtype*, an ``int`` or ``str``, the query type. *rdclass*, an ``int`` or ``str``, the query class. *tcp*, a ``bool``. If ``True``, use TCP to make the query. *source*, a ``str`` or ``None``. If not ``None``, bind to this IP address when making queries. *raise_on_no_answer*, a ``bool``. If ``True``, raise ``dns.resolver.NoAnswer`` if there's no answer to the question. *source_port*, an ``int``, the port from which to send the message. *lifetime*, a ``float``, how many seconds a query should run before timing out. *search*, a ``bool`` or ``None``, determines whether the search list configured in the system's resolver configuration are used for relative names, and whether the resolver's domain may be added to relative names. The default is ``None``, which causes the value of the resolver's ``use_search_by_default`` attribute to be used. Raises ``dns.resolver.LifetimeTimeout`` if no answers could be found in the specified lifetime. Raises ``dns.resolver.NXDOMAIN`` if the query name does not exist. Raises ``dns.resolver.YXDOMAIN`` if the query name is too long after DNAME substitution. Raises ``dns.resolver.NoAnswer`` if *raise_on_no_answer* is ``True`` and the query name exists but has no RRset of the desired type and class. Raises ``dns.resolver.NoNameservers`` if no non-broken nameservers are available to answer the question. Returns a ``dns.resolver.Answer`` instance. NF)rI r1 r2 r� ) r� r_ r� r� r� �sleepr rC rP r0 r� )r r1 rY rZ r� r1 r� r2 r r� � resolutionr rS r] �doner[ r� rI r2 r� �_r r r �resolve� sD ? � � zResolver.resolve� r1 rY rZ r� r1 r� r2 r r& c C s* t jdtdd� | �||||||||d� S )a$ Query nameservers to find the answer to the question. This method calls resolve() with ``search=True``, and is provided for backwards compatibility with prior versions of dnspython. See the documentation for the resolve() method for further details. z2please use dns.resolver.Resolver.resolve() insteadr� �� stacklevelT��warnings�warn�DeprecationWarningr7 ) r r1 rY rZ r� r1 r� r2 r r r r rP = s ��zResolver.query��ipaddrr r r& c O s@ i }|� |� tjj|d< tjj|d< | jtj�|�f|�|�S )a� Use a resolver to run a reverse query for PTR records. This utilizes the resolve() method to perform a PTR lookup on the specified IP address. *ipaddr*, a ``str``, the IPv4 or IPv6 address you want to get the PTR record for. All other arguments that can be passed to the resolve() function except for rdtype and rdclass are also supported by this function. rY rZ ) �updaterp rr �PTRrt r� r7 � reversename� from_address)r r@ r r �modified_kwargsr r r �resolve_address` s ���zResolver.resolve_address�ra r� r r& c K s, i }|� |� |�dd� tjj|d< |tjkrP| j|tjj f|�}t j |d�S |tj krz| j|tjj f|�}t j |d�S |tjkr�td|� ���|�dd�}|�d d�}t�� } | j|tjj fd | �| |�d �|��}|j}| j|tjj fd | �| |�d �|��}t j ||| d �} | �s(t|jd ��| S )a Use a resolver to query for address records. This utilizes the resolve() method to perform A and/or AAAA lookups on the specified name. *qname*, a ``dns.name.Name`` or ``str``, the name to resolve. *family*, an ``int``, the address family. If socket.AF_UNSPEC (the default), both A and AAAA records will be retrieved. All other arguments that can be passed to the resolve() function except for rdtype and rdclass are also supported by this function. rY NrZ )r| )r{ r� r� Tr F)r� r )r{ r| r} r� )rA r� rp rt r� r� r� r7 rr r rz r� r� r~ r� r� r_ r r1 rO r2 ) r ra r� r rE r| r{ r� r r r� r r r � resolve_namex sL � �� � �� zResolver.resolve_name�ra r& c C sJ z| j |dd�}|j}W n, tjjk rD } z |j}W 5 d}~X Y nX |S )a� Determine the canonical name of *name*. The canonical name is the name the resolver uses for queries after all CNAME and DNAME renamings have been applied. *name*, a ``dns.name.Name`` or ``str``, the query name. This method can raise any exception that ``resolve()`` can raise, other than ``dns.resolver.NoAnswer`` and ``dns.resolver.NXDOMAIN``. Returns a ``dns.name.Name``. F)r� N)r7 r/ rp r� r )r ra r] r/ r� r r r r/ � s zResolver.canonical_namer �r r& c C sj zPt � � | }| jtjjd|d�}tj�|�}tj�||�}t|�dkrN|| _ W n t k rd Y nX dS )a� Try to update the resolver's nameservers using Discovery of Designated Resolvers (DDR). If successful, the resolver will subsequently use DNS-over-HTTPS or DNS-over-TLS for future queries. *lifetime*, a float, is the maximum time to spend attempting DDR. The default is 5 seconds. If the SVCB query is successful and results in a non-empty list of nameservers, then the resolver's nameservers are set to the returned servers in priority order. The current implementation does not use any address hints from the SVCB record, nor does it resolve addresses for the SCVB target name, rather it assumes that the bootstrap nameserver will always be one of the addresses and uses it. A future revision to the code may offer fuller support. The code verifies that the bootstrap nameserver is in the Subject Alternative Name field of the TLS certficate. ZSVCB�r r N) r_ r7 rp Z_ddrZ_local_resolver_namerP Z _remainingZ_get_nameservers_syncr" r� r0 )r r r` r] rI r� r r r �try_ddr� s � zResolver.try_ddr)r ) r9 r: r; r< rp rr r rt r� r ra rq r+ rs ru r� r rx r� rX r7 rP r rF r� r� rz rH r/ rL r r r r r/ � sb �� i�� #�� C r/ �default_resolverr% c C s t dkrt� t dk st�t S )z7Get the default resolver, initializing it if necessary.N)rM �reset_default_resolverr� r r r r �get_default_resolver� s rO c C s t � adS )z�Re-initialize default resolver. Note that the resolver configuration (i.e. /etc/resolv.conf on UNIX systems) will be re-read immediately. N)r/ rM r r r r rN � s rN FTr0 c C s t � �| ||||||||� S )z�Query nameservers to find the answer to the question. This is a convenience function that uses the default resolver object to make the query. See ``dns.resolver.Resolver.resolve`` for more information on the parameters. )rO r7 ) r1 rY rZ r� r1 r� r2 r r� r r r r7 s �r7 r8 c C s( t jdtdd� t| |||||||d� S )a Query nameservers to find the answer to the question. This method calls resolve() with ``search=True``, and is provided for backwards compatibility with prior versions of dnspython. See the documentation for the resolve() method for further details. z)please use dns.resolver.resolve() insteadr� r9 Tr; )r1 rY rZ r� r1 r� r2 r r r r rP * s ��rP r? c O s t � j| f|�|�S )z�Use a resolver to run a reverse query for PTR records. See ``dns.resolver.Resolver.resolve_address`` for more information on the parameters. )rO rF )r@ r r r r r rF K s rF rG c K s t � j| |f|�S )z�Use a resolver to query for address records. See ``dns.resolver.Resolver.resolve_name`` for more information on the parameters. )rO rH )ra r� r r r r rH U s rH rI c C s t � �| �S )z�Determine the canonical name of *name*. See ``dns.resolver.Resolver.canonical_name`` for more information on the parameters and possible exceptions. )rO r/ �ra r r r r/ a s r/ r rJ c C s t � �| �S )a Try to update the default resolver's nameservers using Discovery of Designated Resolvers (DDR). If successful, the resolver will subsequently use DNS-over-HTTPS or DNS-over-TLS for future queries. See :py:func:`dns.resolver.Resolver.try_ddr` for more information. )rO rL rK r r r rL k s rL )ra rZ r� r� r r& c C s� t | t�r tj�| tjj�} |dkr*t� }| �� s:t| ��t � � }|dk rT|| }nd}zb|dk r||t � � }|dkr�d}nd}|j | tj j |||d�}|j dk s�t�|j j| kr�| W S W n� tjjtjjfk �rr } z�t | tjj�r�| �� �| �} n| �� } | �rb| jD ]T} | jtj j k�r | j|k�r | j�| �\} } } | tjjk�r | j W Y � S �q W 5 d} ~ X Y nX z | �� } W qX tjjk �r� t �Y qXX qXdS )a� Find the name of the zone which contains the specified name. *name*, an absolute ``dns.name.Name`` or ``str``, the query name. *rdclass*, an ``int``, the query class. *tcp*, a ``bool``. If ``True``, use TCP to make the query. *resolver*, a ``dns.resolver.Resolver`` or ``None``, the resolver to use. If ``None``, the default, then the default resolver is used. *lifetime*, a ``float``, the total time to allow for the queries needed to determine the zone. If ``None``, the default, then only the individual query limits of the resolver apply. Raises ``dns.resolver.NoRootSOA`` if there is no SOA RR at the DNS root. (This is only likely to happen if you're using non-default root servers in your network and they are misconfigured.) Raises ``dns.resolver.LifetimeTimeout`` if the answer could not be found in the allotted lifetime. Returns a ``dns.name.Name``. Nr rK ) r r+ rp ra r� r rO r! rT r_ r7 rr ZSOAr^ r� r� r rO r r4 r2 � authorityrY rZ Z fullcompareZNAMERELN_SUPERDOMAIN�parentZNoParentrU )ra rZ r� r� r r r` Z rlifetimer] r� r2 Zrrs�nrr6 r r r � zone_for_nameu sP � ( rT r )�wherer\ r� r� r& c C s� |dkrt � }g }t| t�r>tj�| �r>|�tj�| |�� n*|� | |�� � D ]}|�tj�||�� qNtj j dd�}||_ |S )a Make a stub resolver using the specified destination as the full resolver. *where*, a ``dns.name.Name`` or ``str`` the domain name or IP address of the full resolver. *port*, an ``int``, the port to use. If not specified, the default is 53. *family*, an ``int``, the address family to use. This parameter is used if *where* is not an address. The default is ``socket.AF_UNSPEC`` in which case the first address returned by ``resolve_name()`` will be used, otherwise the first address of the specified family will be used. *resolver*, a ``dns.resolver.Resolver`` or ``None``, the resolver to use for resolution of hostnames. If not specified, the default resolver will be used. Returns a ``dns.resolver.Resolver`` or raises an exception. NF)r )rO r r+ rp r& r' r5 r[ r( rH r� r� r/ r� )rU r\ r� r� r� r� �resr r r �make_resolver_at� s rW )rU r1 rY rZ r� r1 r� r2 r r� r\ r� r� r& c C s$ t | | | | ��||||||||| � S )a} Query nameservers to find the answer to the question. This is a convenience function that calls ``dns.resolver.make_resolver_at()`` to make a resolver, and then uses it to resolve the query. See ``dns.resolver.Resolver.resolve`` for more information on the resolution parameters, and ``dns.resolver.make_resolver_at`` for information about the resolver parameters *where*, *port*, *family*, and *resolver*. If making more than one query, it is more efficient to call ``dns.resolver.make_resolver_at()`` and then use that resolver for the queries instead of calling ``resolve_at()`` multiple times. )rW r7 ) rU r1 rY rZ r� r1 r� r2 r r� r\ r� r� r r r � resolve_at� s �rX c C s^ |t j@ dkr t| |||||�S |t jt jB @ dkrBt �t jd��| d kr`|d kr`t �t jd��g }d }| d kr�t| |||||�S z tj � | � t| |||||�W S t k r� Y nX z&t � | |�}|�� }|�� �d�}W nJ tjjk �r t �t jd��Y n$ t k �r( t �t jd��Y nX d } z |d k�r@d} nt|�} W nL t k �r� |t j@ dk�r�zt �|�} W n t k �r� Y nX Y nX | d k�r�t �t jd��g } |dk�r�t jt jg} n|g} |t j@ dk�r�|} nd} |D ]L\} }| D ]<}t| D ],}tj �| | f|�}| � |||| |f� �q�q��q�t | �dk�rZt �t jd��| S )Nr z*Non-recoverable failure in name resolution�Name or service not knownTz$Temporary failure in name resolution� ) r� �AI_NUMERICHOST�_original_getaddrinfo� AI_ADDRCONFIG� AI_V4MAPPED�gaierror�EAI_FAIL� EAI_NONAMErp r& Zaf_for_addressr0 � _resolverrH r� r/ r� r� r � EAI_AGAINrx �AI_NUMERICSERV� getservbyname� SOCK_DGRAM� SOCK_STREAM� AI_CANONNAME�_protocols_for_socktypeZlow_level_address_tupler5 r" )�host�servicer� �socktype�protor� �addrsr/ r� r\ �tuplesZ socktypesr3 �addr�afZ addr_tupler r r � _getaddrinfo* sl � rr c C sh | d }| d }t | �dkr,| d }tj}n d }tj}t|||tjtjd�}t |�dkrbt�d��|d d d }|tj@ r�d}nd}t j � |�} |tj @ dk�r z"t �| d�} | jd j�d �} W nX t jjt jjfk �r |tj@ r�t�tjd ��|} |d k �r| d t|� 7 } Y nX n |} |d k �r>| d t|� 7 } |tj@ �rTt|�} n t�||�} | | fS ) Nr r' � rD z'sockaddr resolved to multiple addressesZudpr� rB TrY �%)r" r� r� r� rr rg �SOL_TCP�error�NI_DGRAMrp rC rD �NI_NUMERICHOSTrb r7 r^ �targetr� r� r rO � NI_NAMEREQDr_ ra r+ �NI_NUMERICSERV� getservbyport) �sockaddrr� rj r\ �scoper� ro rp Zpnamer1 r] �hostnamerk r r r � _getnameinfo~ s@ r� c C s< | d krt �� } zt| �\} }}W n tk r6 Y nX | S r )r� r �_gethostbyaddrr0 )ra r6 r r r �_getfqdn� s r� c C s t | �d d S )Nr� r )�_gethostbyname_exrP r r r �_gethostbyname� s r� c C sT g }g }t | dtjtjtjtj�}|d d }|D ]}|�|d d � q2|||fS )Nr rD rs )rr r� r� rg ru rh r5 )ra �aliasesr� ro � canonical�itemr r r r� � s � r� c C s z"t j�| � | dddf}tj}W nV tk rx zt j�| � W n" tk rd t�tjd��Y nX | df}tj }Y nX t |tj �\}}g }g }t |d|tj tjtj�}|d d }t j�|| �} |D ]0} | d d } t j�|| �} | | kr�|�| � q�|||fS )N�P r rY rD rs )rp Zipv6� inet_atonr� r� r0 Zipv4r_ ra r� r� rz rr rg ru rh r& � inet_ptonr5 ) �ipr} r� ra r6 r� r� ro r� Zbin_ipr� rp Zbin_addrr r r r� � s< � r� )r� r& c C s: | dkrt � } | att_tt_tt_t t_ t t_ t t_dS )a� Override the system resolver routines in the socket module with versions which use dnspython's resolver. This can be useful in testing situations where you want to control the resolution behavior of python code without having to change the system's resolver settings (e.g. /etc/resolv.conf). The resolver to use may be specified; if it's not, the default resolver will be used. resolver, a ``dns.resolver.Resolver`` or ``None``, the resolver to use. N)rO rb rr r� � getaddrinfor� � getnameinfor� �getfqdnr� � gethostbynamer� �gethostbyname_exr� � gethostbyaddr)r� r r r �override_system_resolver� s r� c C s, da tt_tt_tt_tt_ t t_ t t_ dS )z5Undo the effects of prior override_system_resolver().N)rb r\ r� r� �_original_getnameinfor� �_original_getfqdnr� �_original_gethostbynamer� �_original_gethostbyname_exr� �_original_gethostbyaddrr� r r r r �restore_system_resolver� s r� )r )r )N)N)yr< r r� r� r r� r_ r< �typingr r r r r r r r � urllib.parser Zdns._ddrrp Zdns.ednsZ dns.exceptionZ dns.flagsZdns.inetZdns.ipv4Zdns.ipv6Z dns.messageZdns.nameZdns.nameserverZ dns.queryZ dns.rcodeZdns.rdataclassZ dns.rdatatypeZdns.rdtypes.svcbbaseZdns.reversenameZdns.tsigr Z dns.win32utilr� Z DNSExceptionr rA r+ r� rx r0 rv r� r- rF �TimeoutrG rO rR rT rU rV rW rX r# ry rz r� r� ra rq rr rs rt ru r� r� r� r� r� r� r/ rM r+ rO rN r r� r� r7 rP rF r� rH r/ rL rT rW rX rf �SOL_UDPrg ru ri rb r� r\ r� r� r� r� r� r� r� r� r� r� rr r� r� r� r� r� r� r� r r r r �<module> s~ ( X �� E1 Oo U ! �� #�� ! � � �� W�� '�� / � � T '