← Back
Editing: resolver.cpython-36.pyc
3 >D�W�� � @ s� d Z ddlZddlZddlZddlZy ddlZW n ek rL ddlZY nX ddl Z ddl Z ddl Z ddl Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlmZmZ ejdkr�y ddlZW n ek r� ddlZY nX G dd� de j j �Z G dd � d e j j �Z e j j Z G d d � d e j j �Z!G d d � d e j j �Z"G dd� de j j �Z#G dd� de j j �Z$G dd� de j j �Z%G dd� de&�Z'G dd� de&�Z(G dd� de&�Z)G dd� de&�Z*G d d � d e&�Z+da,d d � Z-d d!� Z.e j/j0e j1j2d"dd#dfd$d%�Z3e j1j2d"dfd&d'�Z4ej5ej6gej7ej8giZ9da:ej;Z<ej=Z>ej?Z@ejAZBejCZDejEZFddejGdddfd(d)�ZHd8d*d+�ZId9d,d-�ZJd.d/� ZKd0d1� ZLd2d3� ZMd:d4d5�ZNd6d7� ZOdS );z{DNS stub resolver. @var default_resolver: The default resolver object @type default_resolver: dns.resolver.Resolver object� N� )�xrange� string_types�win32c sV e Zd ZdZeddg�ZdZddd�Z� fdd�Zd d � Z e e d d �Z d d� Z � Z S )�NXDOMAINz"The DNS query name does not exist.�qnames� responsesNc 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)�selfr r �kwargs� r � /usr/lib/python3.6/resolver.py� _check_kwargs9 s zNXDOMAIN._check_kwargsc s^ d| j krtt| �j� S | j d }t|�dkr4d}n| jd d� }djtt|��}d||f S )Nr r z None of DNS query names existz, z%s: %s���) r �superr �__str__r �__doc__�join�map�str)r r �msg)� __class__r r r E s zNXDOMAIN.__str__c C s� d| j krtd��tjj}tjj}d }xj| j d D ]\}| j d | }x4|jD ]*}|j|ksL|j |krfqL|j d j j � }qLW |d k r2tj j|�S q2W | j d d S )Nr z parametrized exception requiredr r )r � TypeError�dns� rdataclass�IN� rdatatype�CNAME�answer�rdtype�rdclass�items�target�to_text�name� from_text)r r! r# �cname�qname�responser$ r r r �canonical_nameP s zNXDOMAIN.canonical_namez%Return the unresolved canonical name.)�docc C s| t | jjdg ��}t| jjdi ��}|jjdi �}x<|jjdg �D ]*}||krX|j|� ||krB|| ||< qBW t||d�S )z3Augment by results from another NXDOMAIN exception.r r )r r )r r �getr �appendr )r Ze_nxZqnames0Z responses0Z responses1Zqname1r r r �__add__b s zNXDOMAIN.__add__)N) �__name__� __module__� __qualname__r r � supp_kwargs�fmtr r r/ �propertyr3 � __classcell__r r )r r r 3 s r c @ s e Zd ZdZdS )�YXDOMAINz8The DNS query name is too long after DNAME substitution.N)r4 r5 r6 r r r r r r; o s r; c s. e Zd ZdZdZedg�Z� fdd�Z� ZS )�NoAnswerz<The DNS response does not contain an answer to the question.z,The DNS response does not contain an answer zto the question: {query}r. c s t t| �j|d jd�S )Nr. )�query)r r< � _fmt_kwargs�question)r r )r r r r> � s zNoAnswer._fmt_kwargszDThe DNS response does not contain an answer to the question: {query}) r4 r5 r6 r r8 r r7 r> r: r r )r r r< z s r< c s@ e Zd ZdZdZdedd � Zeddg�Z� fdd �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}Nr �request�errorsc sd g }x>|d D ]2}|j d|d |d r*dnd|d |d f � qW tt| �j|d jd j|�d �S ) NrB z Server %s %s port %s answered %sr r ZTCPZUDP� � rA z; )r= rB )r2 r r@ r> r? r )r r Zsrv_msgs�err)r r r r> � s & zNoNameservers._fmt_kwargsr ) r4 r5 r6 r r r8 r r7 r> r: r r )r r r@ � s r@ c @ s e Zd ZdZdS )� NotAbsolutezEAn absolute domain name is required but a relative name was provided.N)r4 r5 r6 r r r r r rF � s rF c @ s e Zd ZdZdS )� NoRootSOAzBThere is no SOA RR at the DNS root name. This should never happen!N)r4 r5 r6 r r r r r rG � s rG c @ s e Zd ZdZdS )� NoMetaqueriesz DNS metaqueries are not allowed.N)r4 r5 r6 r r r r r rH � s rH c @ sB e Zd ZdZddd�Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�Answera9 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. 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 node's name might not be the query name. @ivar qname: The query name @type qname: dns.name.Name object @ivar rdtype: The query type @type rdtype: int @ivar rdclass: The query class @type rdclass: int @ivar response: The response message @type response: dns.message.Message object @ivar rrset: The answer @type rrset: dns.rrset.RRset object @ivar expiration: The time when the answer expires @type expiration: float (seconds since the epoch) @ivar canonical_name: The canonical name of the query name @type canonical_name: dns.name.Name object Tc C s� || _ || _|| _|| _d}d }x�tdd�D ]�}y0|j|j|||�}|dksV|j|k r\|j}P W q, tk �r |t j j kr�yJ|j|j||t j j �} |dks�| j|k r�| j}x| D ] } | j }P q�W w,W n" tk r� |r�t |d��Y nX |� r�t |d��Y q,X q,W |d k�r |�r t |d��|| _|| _|d k�r�x�yR|j|j||t j j�} |dk�sf| j|k �rl| j}| d j|k �r�| d j}P W n@ tk �r� y |j� }W n t jjk �r� P Y nX Y nX �q8W tj� | | _d S ) Nr r � )r. r r r r )r- r% r& r. r Z find_rrsetr$ �ttl�KeyErrorr r" r# r( r< r/ �rrsetZ authority�SOAZminimum�parentr* �NoParent�time� expiration) r r- r% r& r. �raise_on_no_answerZmin_ttlrM �countZcrrsetZrdZsrrsetr r r �__init__� sf zAnswer.__init__c C s\ |dkr| j jS |dkr | j jS |dkr0| j jS |dkr@| j jS |dkrP| j jS t|��d S )Nr* rK �coversr& r% )rM r* rK rV r& r% r )r �attrr r r � __getattr__ s zAnswer.__getattr__c C s | j rt| j �pdS )Nr )rM r )r r r r �__len__ s zAnswer.__len__c C s | j rt| j �ptt� �S )N)rM �iterr )r r r r �__iter__ s zAnswer.__iter__c C s | j | S )N)rM )r �ir r r � __getitem__ s zAnswer.__getitem__c C s | j |= d S )N)rM )r r\ r r r � __delitem__ s zAnswer.__delitem__N)T) r4 r5 r6 r rU rX rY r[ r] r^ r r r r rI � s 7rI c @ s< e Zd ZdZddd�Zdd� Zdd� Zd d � Zdd d �Zd S )�Cacheaq Simple DNS answer cache. @ivar data: A dictionary of cached data @type data: dict @ivar cleaning_interval: The number of seconds between cleanings. The default is 300 (5 minutes). @type cleaning_interval: float @ivar next_cleaning: The time the cache should next be cleaned (in seconds since the epoch.) @type next_cleaning: float � �r@c C s* i | _ || _tj� | j | _tj� | _dS )z�Initialize a DNS cache. @param cleaning_interval: the number of seconds between periodic cleanings. The default is 300.0 @type cleaning_interval: float. N)�data�cleaning_intervalrQ � next_cleaning� _threading�Lock�lock)r rb r r r rU - s zCache.__init__c C sp t j � }| j|krlg }x*| jj� D ] \}}|j|kr"|j|� q"W x|D ] }| j|= qHW t j � }|| j | _dS )z&Clean the cache if it's time to do so.N)rQ rc ra r' rR r2 rb )r �nowZkeys_to_delete�k�vr r r � _maybe_clean: s zCache._maybe_cleanc C sN z<| j j� | j� | jj|�}|dks6|jtj� kr:dS |S | j j� X dS )a) Get the answer associated with I{key}. Returns None if no answer is cached for the key. @param key: the key @type key: (dns.name.Name, int, int) tuple whose values are the query name, rdtype, and rdclass. @rtype: dns.resolver.Answer object or None N)rf �acquirerj ra r1 rR rQ �release)r �keyri r r r r1 H s z Cache.getc C s2 z | j j� | j� || j|< W d| j j� X dS )a Associate key and value in the cache. @param key: the key @type key: (dns.name.Name, int, int) tuple whose values are the query name, rdtype, and rdclass. @param value: The answer being cached @type value: dns.resolver.Answer object N)rf rk rj ra rl )r rm �valuer r r �put[ s z Cache.putNc C sR z@| j j� |dk r(|| jkr>| j|= ni | _tj� | j | _W d| j j� X dS )z�Flush the cache. If I{key} is specified, only that item is flushed. Otherwise the entire cache is flushed. @param key: the key to flush @type key: (dns.name.Name, int, int) tuple or None N)rf rk ra rQ rb rc rl )r rm r r r �flushk s z Cache.flush)r` )N) r4 r5 r6 r rU rj r1 ro rp r r r r r_ s r_ c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � LRUCacheNodezLRUCache node. c C s || _ || _| | _| | _d S )N)rm rn �prev�next)r rm rn r r r rU � s zLRUCacheNode.__init__c C s |j | _ || _| |j _| |_ d S )N)rr rs )r �noder r r � link_before� s zLRUCacheNode.link_beforec C s || _ |j| _| |j_ | |_d S )N)rr rs )r rt r r r � link_after� s zLRUCacheNode.link_afterc C s | j | j_ | j| j _d S )N)rr rs )r r r r �unlink� s zLRUCacheNode.unlinkN)r4 r5 r6 r rU ru rv rw r r r r rq � s rq c @ s< e Zd ZdZddd�Zdd� Zdd� Zd d � Zdd d �Zd S )�LRUCacheaO 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. @ivar data: A dictionary of cached data @type data: dict @ivar sentinel: sentinel node for circular doubly linked list of nodes @type sentinel: LRUCacheNode object @ivar max_size: The maximum number of nodes @type max_size: int 順 c C s* i | _ | j|� tdd�| _tj� | _dS )z�Initialize a DNS cache. @param max_size: The maximum number of nodes to cache; the default is 100,000. Must be greater than 1. @type max_size: int N)ra � set_max_sizerq �sentinelrd re rf )r �max_sizer r r rU � s zLRUCache.__init__c C s |dk r d}|| _ d S )Nr )r| )r r| r r r rz � s zLRUCache.set_max_sizec C sl zZ| j j� | jj|�}|dkr$dS |j� |jjtj� krJ| j|j= dS |j | j � |jS | j j � X dS )a) Get the answer associated with I{key}. Returns None if no answer is cached for the key. @param key: the key @type key: (dns.name.Name, int, int) tuple whose values are the query name, rdtype, and rdclass. @rtype: dns.resolver.Answer object or None N) rf rk ra r1 rw rn rR rQ rm rv r{ rl )r rm rt r r r r1 � s z LRUCache.getc C s� z�| j j� | jj|�}|dk r2|j� | j|j= x.t| j�| jkr`| jj }|j� | j|j= q4W t ||�}|j | j� || j|< W d| j j � X dS )a Associate key and value in the cache. @param key: the key @type key: (dns.name.Name, int, int) tuple whose values are the query name, rdtype, and rdclass. @param value: The answer being cached @type value: dns.resolver.Answer object N) rf rk ra r1 rw rm r r| r{ rr rq rv rl )r rm rn rt r r r ro � s z LRUCache.putNc C s� zr| j j� |dk r<| jj|�}|dk rp|j� | j|j= n4| jj}x$|| jkrh|j}d|_d|_|}qFW i | _W d| j j � X dS )z�Flush the cache. If I{key} is specified, only that item is flushed. Otherwise the entire cache is flushed. @param key: the key to flush @type key: (dns.name.Name, int, int) tuple or None N) rf rk ra r1 rw rm r{ rs rr rl )r rm rt rs r r r rp � s zLRUCache.flush)ry )N) r4 r5 r6 r rU rz r1 ro rp r r r r rx � s rx c @ s� e Zd ZdZd%dd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zejjejjdddd fd d �Zdejjfd d �Zd!d"� Zd#d$� ZdS )&�Resolvera- DNS stub resolver @ivar domain: The domain of this host @type domain: dns.name.Name object @ivar nameservers: A list of nameservers to query. Each nameserver is a string which contains the IP address of a nameserver. @type nameservers: list of strings @ivar search: The search list. If the query name is a relative name, the resolver will construct an absolute query name by appending the search names one by one to the query name. @type search: list of dns.name.Name objects @ivar port: The port to which to send queries. The default is 53. @type port: int @ivar timeout: The number of seconds to wait for a response from a server, before timing out. @type timeout: float @ivar lifetime: The total number of seconds to spend trying to get an answer to the question. If the lifetime expires, a Timeout exception will occur. @type lifetime: float @ivar keyring: The TSIG keyring to use. The default is None. @type keyring: dict @ivar keyname: The TSIG keyname to use. The default is None. @type keyname: dns.name.Name object @ivar keyalgorithm: The TSIG key algorithm to use. The default is dns.tsig.default_algorithm. @type keyalgorithm: string @ivar edns: The EDNS level to use. The default is -1, no Edns. @type edns: int @ivar ednsflags: The EDNS flags @type ednsflags: int @ivar payload: The EDNS payload size. The default is 0. @type payload: int @ivar flags: The message flags to use. The default is None (i.e. not overwritten) @type flags: int @ivar cache: The cache to use. The default is None. @type cache: dns.resolver.Cache object @ivar retry_servfail: should we retry a nameserver if it says SERVFAIL? The default is 'false'. @type retry_servfail: bool �/etc/resolv.confTc C s� d| _ d| _d| _d| _d| _d| _d| _d| _d| _d| _ d| _ d| _ d| _ d| _ d| _d| _d| _| j� |r�tjdkr�| j� n|r�| j|� dS )aJ Initialize a resolver instance. @param filename: The filename of a configuration file in standard /etc/resolv.conf format. This parameter is meaningful only when I{configure} is true and the platform is POSIX. @type filename: string or file object @param configure: If True (the default), the resolver instance is configured in the normal fashion for the operating system the resolver is running on. (I.e. a /etc/resolv.conf file on POSIX systems and from the registry on Windows systems.) @type configure: boolNFr )�domain� nameservers�nameserver_ports�port�search�timeout�lifetime�keyring�keyname� keyalgorithm�edns� ednsflags�payload�cache�flags�retry_servfail�rotate�reset�sys�platform� read_registry�read_resolv_conf)r �filenameZ configurer r r rU 9 s. zResolver.__init__c C s� t jjt jjtj� �dd� �| _t| j�dkr:t jj| _g | _ i | _ d| _ g | _ d| _ d| _d| _d| _t jj| _d| _d| _d| _d| _d| _d| _d| _dS ) z1Reset all resolver configuration to the defaults.r Nr �5 g @g >@Fr )r r* �Namer+ �socket� gethostnamer r �rootr� r� r� r� r� r� r� r� �tsig�default_algorithmr� r� r� r� r� r� r� r� )r r r r r� _ s& " zResolver.resetc C sZ t |t�r<yt|d�}W n tk r4 dg| _dS X d}nd}z�x�|D ]�}t|�dksH|d dksH|d dkrrqH|j� }t|�d k r�qH|d d kr�| jj|d � qH|d d kr�tj j |d �| _ qH|d d k�r xL|d d� D ]}| j jtj j |�� q�W qH|d dkrHd|d d� krHd| _ qHW W d|�r8|j� X t| j�dk�rV| jjd� dS )z�Process f as a file in the /etc/resolv.conf format. If f is a string, it is used as the name of the file to open; otherwise it is treated as the file itself.�rz 127.0.0.1NTFr �#�;rC � nameserverr r r� Zoptionsr� )r r �open�IOErrorr� r �splitr2 r r* r+ r r� r� �close)r �fZ want_close�l�tokens�suffixr r r r� v s: $ zResolver.read_resolv_confc C s0 |j d�dkrd}n|j d�dkr(d}nd}|S )N� r �,)�find)r �entry� split_charr r r �_determine_split_char� s z Resolver._determine_split_charc C sD t |�}| j|�}|j|�}x"|D ]}|| jkr"| jj|� q"W dS )z&Configure a NameServer registry entry.N)r r� r� r� r2 )r r� r� Zns_list�nsr r r �_config_win32_nameservers� s z"Resolver._config_win32_nameserversc C s t jjt|��| _dS )z"Configure a Domain registry entry.N)r r* r+ r r )r r r r r �_config_win32_domain� s z Resolver._config_win32_domainc C sL t |�}| j|�}|j|�}x*|D ]"}|| jkr"| jjtjj|�� q"W dS )z"Configure a Search registry entry.N)r r� r� r� r2 r r* r+ )r r� r� Z search_list�sr r r �_config_win32_search� s z Resolver._config_win32_searchc # C s. yt j|d�\}}W n tk r, d}Y nX |rv| j|� y"t j|d�\}}|r\| j|� W q� tk rr Y q�X ntyt j|d�\}}W n tk r� d}Y nX |r�| j|� y"t j|d�\}}|r�| j|� W n tk r� Y nX yt j|d�\}}W n tk �r d}Y nX |�r*| j|� dS )z%Extract DNS info from a registry key.Z NameServerNZDomainZDhcpNameServerZ DhcpDomainZ SearchList)�_winreg� QueryValueEx� WindowsErrorr� r� r� )r rm ZserversZrtypeZdomr� r r r �_config_win32_fromkey� s< z Resolver._config_win32_fromkeyc , C s t jdt j�}d}z�yt j|d�}d}W n tk rH t j|d�}Y nX z| j|� W d|j� X |r�t j|d�}ztd}xjyNt j||�}|d7 }t j||�}| j|||�s�w|z| j|� W d|j� X W q| tk r� P Y q|X q|W W d|j� X W d|j� X dS ) z9Extract resolver configuration from the Windows registry.NFz2SYSTEM\CurrentControlSet\Services\Tcpip\ParametersTz+SYSTEM\CurrentControlSet\Services\VxD\MSTCPz=SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\Interfacesr r ) r� ZConnectRegistry�HKEY_LOCAL_MACHINE�OpenKey�EnvironmentErrorr� �CloseZEnumKey�_win32_is_nic_enabled)r �lmZ want_scanZ tcp_paramsZ interfacesr\ �guidrm r r r r� � s@ zResolver.read_registryc C s� y�t j|d| �}zft j|d�\}}|t jkr2t�t j|d| �}z(t j|d�\}}|t jkrbt�|d@ S |j� X W d |j� X W nF ttfk r� yt j|d�\} }| d k S tk r� dS X Y nX d S )Nz]SYSTEM\CurrentControlSet\Control\Network\{4D36E972-E325-11CE-BFC1-08002BE10318}\%s\ConnectionZ PnpInstanceIDz SYSTEM\CurrentControlSet\Enum\%sZ ConfigFlagsr ZNTEContextListF) r� r� r� ZREG_SZ� ValueErrorZ REG_DWORDr� r� r� ) r r� r� Z interface_keyZconnection_keyZpnp_idZttypeZ device_keyr� Znter r r r� s6 z Resolver._win32_is_nic_enabledc C sV t j � }|| }|dk r0|dk r,t|d��n|}|| jkrDt|d��t| j| | j�S )Nr r )r� r )rQ �Timeoutr� �minr� )r �startrg Zdurationr r r �_compute_timeoutM s zResolver._compute_timeoutFNr c C s t |t�rtjj|d�}t |t�r.tjj|�}tjj|�r>t�t |t�rTtjj|�}tjj |�rdt�g }|j � r||j |� nZt |�dkr�|j |j tjj�� | jr�x2| jD ]} |j |j | �� q�W n|j |j | j�� d} i } tj� } d} �x�|D �]�} | j�r@| jj| ||f�}|dk �r@|jdk�r<|�r<t|jd��n|S tjj| ||�}| jdk �rr|j| j| j| j d� |j | j | j | j � | j!dk �r�| j!|_!d}| j"dd� }g }| j#�r�t$j%|� d}�x�|dk�r�t |�dk�r�t&||d���x`|dd� D �]N}| j'| �}| j(j|| j)�}yx|}|�rDtj*j+||||||d �}nNtj*j,||||||d �}|j!tj!j-@ �r�d}| j'| �}tj*j+||||||d �}W �n t.j/tj0j1fk �r� } z"|j |||||f� d}�w�W Y dd}~X n� tj*j2k �r" } z"|j |||||f� d}�w�W Y dd}~X n� tj0j3k �rn } z,|j4|� |j |||||f� d}�w�W Y dd}~X nJ t5k �r� } z,|j4|� |j |||||f� d}�w�W Y dd}~X nX |j6� }|tj6j7k�r�t7� }|j |||||f� |�|tj6j8k�s|tj6j9k�r P |tj6j:k�s"| j; �r,|j4|� |j |||tj6j<|�|f� d}�q�W |dk �r^P t |�dk�r�| j'| �}t=||�}|d 9 }tj>|� �q�W |j6� tj6j9k�r�|| | < q�d } P q�W | �r�t9|| d ��t?| ||||�}| j�r�| jj@| ||f|� |S ) a Query nameservers to find the answer to the question. The I{qname}, I{rdtype}, and I{rdclass} parameters may be objects of the appropriate type, or strings that can be converted into objects of the appropriate type. E.g. For I{rdtype} the integer 2 and the the string 'NS' both mean to query for records with DNS rdata type NS. @param qname: the query name @type qname: dns.name.Name object or string @param rdtype: the query type @type rdtype: int or string @param rdclass: the query class @type rdclass: int or string @param tcp: use TCP to make the query (default is False). @type tcp: bool @param source: bind to this IP address (defaults to machine default IP). @type source: IP address in dotted quad notation @param raise_on_no_answer: raise NoAnswer if there's no answer (defaults is True). @type raise_on_no_answer: bool @param source_port: The port from which to send the message. The default is 0. @type source_port: int @rtype: dns.resolver.Answer instance @raises Timeout: no answers could be found in the specified lifetime @raises NXDOMAIN: the query name does not exist @raises YXDOMAIN: the query name is too long after DNAME substitution @raises NoAnswer: the response did not contain an answer and raise_on_no_answer is True. @raises NoNameservers: no non-broken nameservers are available to answer the question.Nr T)r. )� algorithmg�������?r )rA rB )�source� source_portrC F)r r )Ar r r r* r+ r" Z is_metatyperH r Z is_metaclass� is_absoluter2 r Z concatenater� r� r rQ r� r1 rM r< r. �messageZ make_queryr� �use_tsigr� r� �use_ednsr� r� r� r� r� r� �randomZshuffler@ r� r� r� r= �tcp�udpZTCr� �error� exceptionr� ZUnexpectedSourceZ FormError�remove�EOFError�rcoder; ZNOERRORr ZSERVFAILr� r) r� ZsleeprI ro )r r- r% r& r� r� rS r� Z qnames_to_tryr� Z all_nxdomainZnxdomain_responsesr� Z_qnamer$ rA r. r� rB Zbackoffr� r� r� Z tcp_attemptZexr� Z sleep_timer r r r= ] s� # zResolver.queryc C s4 || _ |dkr$t| j j� �d | _n|| _|| _dS )a� Add a TSIG signature to the query. @param keyring: The TSIG keyring to use; defaults to None. @type keyring: dict @param keyname: The name of the TSIG key to use; defaults to None. The key must be defined in the keyring. If a keyring is specified but a keyname is not, then the key used will be the first key in the keyring. Note that the order of keys in a dictionary is not defined, so applications should supply a keyname when a keyring is used, unless they know the keyring contains only one key. @param algorithm: The TSIG key algorithm to use. The default is dns.tsig.default_algorithm. @type algorithm: stringNr )r� r �keysr� r� )r r� r� r� r r r r� " s zResolver.use_tsigc C s" |dkr d}|| _ || _|| _dS )a Configure Edns. @param edns: The EDNS level to use. The default is -1, no Edns. @type edns: int @param ednsflags: The EDNS flags @type ednsflags: int @param payload: The EDNS payload size. The default is 0. @type payload: intNr r )r� r� r� )r r� r� r� r r r r� 8 s zResolver.use_ednsc C s || _ dS )z�Overrides the default flags with your own @param flags: The flags to overwrite the default with @type flags: intN)r� )r r� r r r � set_flagsH s zResolver.set_flags)r~ T)r4 r5 r6 r rU r� r� r� r� r� r� r� r� r� r� r r" �Ar r! r= r� r� r� r� r� r r r r r} s& + &) "-7 Er} c C s t dkrt� t S )z7Get the default resolver, initializing it if necessary.N)�default_resolver�reset_default_resolverr r r r �get_default_resolverR s r� c C s t � adS )zSRe-initialize default resolver. resolv.conf will be re-read immediatelly. N)r} r� r r r r r� Y s r� FTc C s t � j| ||||||�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: L{dns.resolver.Resolver.query} for more information on the parameters.)r� r= )r- r% r& r� r� rS r� r r r r= b s r= c C s� t | t�r tjj| tjj�} |dkr*t� }| j� s:t| ��xxy(|j | tj j ||�}|j j| krb| S W n tj jtj jfk r� Y nX y | j� } W q< tjjk r� t�Y q<X q<W dS )a� Find the name of the zone which contains the specified name. @param name: the query name @type name: absolute dns.name.Name object or string @param rdclass: The query class @type rdclass: int @param tcp: use TCP to make the query (default is False). @type tcp: bool @param resolver: the resolver to use @type resolver: dns.resolver.Resolver object or None @rtype: dns.name.NameN)r r r r* r+ r� r� r� rF r= r" rN rM �resolverr r< rO rP rG )r* r& r� r� r$ r r r � zone_for_nameo s" r� c , C s� |t jt jB @ dkrt�| d kr4|d kr4t jt j��g }g }d }y�| d kr�d}|t j@ dkrr|jd� |jd� q�|jd� |jd� n>| jd�} t | �dkr�| d } n| } t j j | �} |j| � | }W �np t k �r: y t jj | �} |j| � | }W �n8 t k �r4 |t j@ dk�r0y�|t jk�s8|t jk�r�tj| t jjd d �} | j} | jjd �}| jd k �r�x | jD ]} |j| j� �qrW |t jk�s�|t jk�r�tj| t jj d d �}|j} |jjd �}|jd k �r�x |jD ]} |j| j� �q�W W n< t j j k �r t jt j��Y n t jt j ��Y nX Y nX Y nX d }y |d k�rRd}nt |�}W nL t k �r� |t j!@ dk�r�yt j"|�}W n t k �r� Y nX Y nX |d k�r�t jt j��g }|dk�r�t j#t j$g}n|g}|t j%@ dk�r�|}nd }|t jk�s|t jk�rfxP|D ]H} x@|D ]8}x0t&| D ]$}|jt j|||| |ddff� �q2W �q$W �qW |t jk�s~|t jk�r�xL|D ]D} x<|D ]4}x,t&| D ] }|jt j|||| |ff� �q�W �q�W �q�W t |�dk�r�t jt j��|S ) Nr Z localhostz::z0.0.0.0z::1z 127.0.0.1�%rC F)rS T� )'r� Z AI_ADDRCONFIGZ AI_V4MAPPED�NotImplementedError�gaierror� EAI_NONAMEZ AI_PASSIVEr2 r� r r �ipv6� inet_aton� ExceptionZipv4ZAI_NUMERICHOST�AF_INET6� AF_UNSPEC� _resolverr= r" ZAAAAr- r/ r) rM Zaddress�AF_INETr� r� r Z EAI_SYSTEM�intZAI_NUMERICSERVZ getservbyname� SOCK_DGRAM� SOCK_STREAM� AI_CANONNAME�_protocols_for_socktype)�host�service�familyZsocktype�protor� Zv6addrsZv4addrsr/ �partsZahost�addrZv6ZrdataZv4r� �tuplesZ socktypesr, r r r � _getaddrinfo� s� $ r� c C sf | d }| d }t | �dkr,| d }tj}n d }tj}t|||tjtjd�}t |�dkrbtjd��|d d d }|tj@ r�d}nd}t j j |�} |tj @ dk�r y"t j| d�} | jd jjd �} W nV t jjt jjfk �r |tj@ r�tjtj��|} |d k �r| d t|� 7 } Y nX n |} |d k �r<| d t|� 7 } |tj@ �rRt|�} n tj||�} | | fS ) Nr r � rD z'sockaddr resolved to multiple addressesr� r� ZPTRTr� )r r� r� r� r� r� �SOL_TCPr� ZNI_DGRAMr Z reversenameZ from_addressZNI_NUMERICHOSTr� r= rM r( r) r� r r< � NI_NAMEREQDr� r� r ZNI_NUMERICSERVZ getservbyport) �sockaddrr� r� r� Zscoper� r� r� Zpnamer- r$ Zhostnamer� r r r � _getnameinfo sB r� c C sD | d krt j� } ytt| d�d d �d S tk r> | S X d S )N�P r r� )r� r� r� r� r� )r* r r r �_getfqdn. s r� c C s t | �d d S )NrC r )�_gethostbyname_ex)r* r r r �_gethostbyname7 s r� c C sX g }g }t | dtjtjtjtj�}|d d }x |D ]}|j|d d � q4W |||fS )Nr rD r� )r� r� r� r� r� r� r2 )r* �aliases� addressesr� � canonical�itemr r r r� ; s r� c C s� y"t jj| � | dddf}tj}W n" tk rD | df}tj}Y nX t|tj�\}}g }g }t |d|tj tj tj �}|d d }x |D ]} |j | d d � q�W |||fS )Nr� r rD r� )r r� r� r� r� r� r� r� r� r� r� r� r� r2 ) Zipr� r� r* r� r� r� r� r� r r r r �_gethostbyaddrG s 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. @param resolver: the resolver to use @type resolver: dns.resolver.Resolver object or None N)r� r� r� 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 )z4Undo the effects of override_system_resolver(). N)r� �_original_getaddrinfor� r �_original_getnameinfor �_original_getfqdnr �_original_gethostbynamer �_original_gethostbyname_exr �_original_gethostbyaddrr r r r r �restore_system_resolveru s r )r )N)N)Pr r� r� rQ r� Z threadingrd � ImportErrorZdummy_threadingZ dns.exceptionr Z dns.flagsZdns.ipv4Zdns.ipv6Z dns.messageZdns.nameZ dns.queryZ dns.rcodeZdns.rdataclassZ dns.rdatatypeZdns.reversenameZdns.tsigZ_compatr r r� �winregr� r� Z DNSExceptionr r; r� r< r@ rF rG rH �objectrI r_ rq rx r} r� r� r� r" r� r r! r= r� r� ZSOL_UDPr� 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 r r �<module> s� < sb o G & c (
Save File
Cancel