Your IP : 3.145.83.149
3
�Of�L � @ s� d Z ddlmZ ddlZddlZddlZddlZddlZddlZddl Z
ddl Z
ddl Z
ddl
Z
ddlZ
ddlZ
ddlmZmZ ejd/kr�eZnejZejZG dd� de
jj�ZG d d
� d
e
jj�Z d d � Z d
d� Z dd� Z dd� Z dd� Z!e"ed�r�e a#ne a#dd� Z$dd� Z%dd� Z&d d � Z'd d � Z(d0d"d#�Z)d$d%� Z*d&d'� Z+d(d)� Z,d1d*d+�Z-e
j.j/e
j0j1dd ddd,dddddd!e
j2j3fd-d.�Z4dS )2zTalk to a DNS server.� )�
generatorsN� )�long� string_types� c @ s e Zd ZdZdS )�UnexpectedSourcez=A DNS query response came from an unexpected address or port.N)�__name__�
__module__� __qualname__�__doc__� r r �/usr/lib/python3.6/query.pyr , s r c @ s e Zd ZdZdS )� BadResponsez<A DNS query response does not respond to the question asked.N)r r r
r r r r r
r 1 s r c C s | d kr d S t j � | S d S )N)�time)�timeoutr r r
�_compute_expiration6 s r c C sj d}|r|t jO }|r |t jO }|r.|t jO }t j� }|j| |� |rZ|jt|d ��}n|j� }t|�S )aq Poll polling backend.
@param fd: File descriptor
@type fd: int
@param readable: Whether to wait for readability
@type readable: bool
@param writable: Whether to wait for writability
@type writable: bool
@param timeout: Deadline timeout (expiration time, in seconds)
@type timeout: float
@return True on success, False on timeout
r i� )�selectZPOLLINZPOLLOUTZPOLLERR�poll�registerr �bool)�fd�readable�writable�errorr Z
event_maskZpollableZ
event_listr r r
� _poll_for= s
r c C sr g g g }}}|r| g}|r$| g}|r.| g}|dkrLt j |||�\}} }
nt j ||||�\}} }
t|pn| pn|
�S )as Select polling backend.
@param fd: File descriptor
@type fd: int
@param readable: Whether to wait for readability
@type readable: bool
@param writable: Whether to wait for writability
@type writable: bool
@param timeout: Deadline timeout (expiration time, in seconds)
@type timeout: float
@return True on success, False on timeout
N)r r ) r r r r r ZrsetZwsetZxsetZrcountZwcountZxcountr r r
� _select_for\ s r c C s� d}x�|s�|d krd }n |t j � }|dkr4tjj�y t| ||||�sNtjj�W n6 tk
r� } z|jd tjkrv|�W Y d d }~X nX d}qW d S )NFg r T) r �dns� exceptionZTimeout�_polling_backend� select_error�args�errnoZEINTR)r r r r �
expiration�doner �er r r
� _wait_fory s r% c C s | a dS )z#
Internal API. Do not use.
N)r )�fnr r r
�_set_polling_backend� s r' r c C s t | ddd|� d S )NTF)r% )�sr" r r r
�_wait_for_readable� s r) c C s t | ddd|� d S )NFT)r% )r( r" r r r
�_wait_for_writable� s r* c C sD t jj| |d �}t jj| |d �}||koB|dd � |dd � kS )Nr r )r �inetZ inet_pton)�afZa1Za2Zn1Zn2r r r
�_addresses_equal� s r- c C s` |sdS t | ||�s<tjj|d �r@|dd � |dd � kr@dS |rHdS td|� d|� ���d S )NTr r Fzgot a response from z instead of )r- r r+ Z is_multicastr )r, � from_address� destination�ignore_unexpectedr r r
�_matches_destination� s (r1 c C s� | d kr6yt jj|�} W n tk
r4 t jj} Y nX | t jjkrp||f}|d k sZ|dkr�|d krfd}||f}n@| t jjkr�||ddf}|d k s�|dkr�|d kr�d}||ddf}| ||fS )Nr z0.0.0.0z::)r r+ Zaf_for_address� ExceptionZAF_INETZAF_INET6)r, �where�port�source� source_portr/ r r r
�_destination_and_source� s$
r7 �5 Fc
C s" | j � }
t|||||�\}} }t|tjd�} d}
z�t|�}| jd� |dk rV| j|� t| |� t j � }
| j
|
| � x�t | |� | j d�\}
}t
| j|| |�s�qvt j � |
}y"tjj|
| j| j|d�}||_ W n tk
r� | r�wvn� Y nX | j|�� r�|S | �rqvqvt�qvW W d| j� X dS )a� Return the response obtained after sending a query via UDP.
@param q: the query
@type q: dns.message.Message
@param where: where to send the message
@type where: string containing an IPv4 or IPv6 address
@param timeout: The number of seconds to wait before the query times out.
If None, the default, wait forever.
@type timeout: float
@param port: The port to which to send the message. The default is 53.
@type port: int
@param af: the address family to use. The default is None, which
causes the address family to use to be inferred from the form of where.
If the inference attempt fails, AF_INET is used.
@type af: int
@rtype: dns.message.Message object
@param source: source address. The default is the wildcard address.
@type source: string
@param source_port: The port from which to send the message.
The default is 0.
@type source_port: int
@param ignore_unexpected: If True, ignore responses from unexpected
sources. The default is False.
@type ignore_unexpected: bool
@param one_rr_per_rrset: Put each RR into its own RRset
@type one_rr_per_rrset: bool
r Ni�� )�keyring� request_mac�one_rr_per_rrset)�to_wirer7 �socket_factory�socket�
SOCK_DGRAMr � setblocking�bindr* r Zsendtor) �recvfromr1 Zfamilyr �message� from_wirer9 �macr2 � is_responser �close)�qr3 r r4 r, r5 r6 r0 r; �
ignore_errors�wirer/ r( �
begin_timer" r. �
response_time�rr r r
�udp� sD
rN c C sJ d}x@|dkrDt | |� | j|�}|dkr.t�|t|� }|| }qW |S )z�Read the specified number of bytes from sock. Keep trying until we
either get the desired amount, or we hit EOF.
A Timeout exception will be raised if the operation is not completed
by the expiration time.
� r )r) Zrecv�EOFError�len)�sock�countr" r( �nr r r
� _net_read s
rU c C s> d}t |�}x,||k r8t| |� || j||d� �7 }qW dS )z�Write the specified data to the socket.
A Timeout exception will be raised if the operation is not completed
by the expiration time.
r N)rQ r* �send)rR �datar" Zcurrent�lr r r
�
_net_write/ s
rY c C sp y| j |� W n\ tjk
rj tj� d d� \}}t|d�rF|j}n|d }|tjtjtj gkrf|�Y nX d S )N� r! r )
Zconnectr> r �sys�exc_info�hasattrr! Z EINPROGRESSZ EWOULDBLOCKZEALREADY)r( ZaddressZty�vZv_errr r r
�_connect; s
r_ c C s | j � }t|||||�\}} }t|tjd�}
d} z�t|�} |
jd� tj� } |dk r^|
j|� t |
| � t
|�}
t j d|
�| }t
|
|| � t|
d| �}t jd|�\}
t|
|
| �}W d| dkr�d}n tj� | }|
j� X tjj|| j| j|d�}||_| j|��st�|S )a Return the response obtained after sending a query via TCP.
@param q: the query
@type q: dns.message.Message object
@param where: where to send the message
@type where: string containing an IPv4 or IPv6 address
@param timeout: The number of seconds to wait before the query times out.
If None, the default, wait forever.
@type timeout: float
@param port: The port to which to send the message. The default is 53.
@type port: int
@param af: the address family to use. The default is None, which
causes the address family to use to be inferred from the form of where.
If the inference attempt fails, AF_INET is used.
@type af: int
@rtype: dns.message.Message object
@param source: source address. The default is the wildcard address.
@type source: string
@param source_port: The port from which to send the message.
The default is 0.
@type source_port: int
@param one_rr_per_rrset: Put each RR into its own RRset
@type one_rr_per_rrset: bool
r Nz!HrZ )r9 r: r; )r< r7 r= r> � SOCK_STREAMr r@ r rA r_ rQ �struct�packrY rU �unpackrG r rC rD r9 rE rF r )rH r3 r r4 r, r5 r6 r; rJ r/ r( rK r" rX �tcpmsg�ldatarL rM r r r
�tcpI s8
rf Tc & c s� t |t�rtjj|�}t |t�r,tjj|�}tjj|||�}|tjjkrltj j|dddd|
�}|j
j |� |dk r�|j |||d� |j
� }t| | || | �\} }} |r�|tjjkr�td��t| tjd�}nt| tjd�}|jd� | dk r�|j| � t|
�}t||� t|�}|�r*t||� |j|� n tj d|�| }t |||� d }d
}d }d}|�rj|} tjj } nd} |} d} d
} �x6|�s�t|�} | dk�s�|dk �r�| |k�r�|} |�r�t ||� |j d �\}}!n&t!|d | �}"tj"d|"�\}t!||| �}|tjjk}#tjj#||j$|j%d
| | d
| |#d
� }$|$j&} d } d}%|dk�r�|$j' �sV|$j'd j| k�rbtj(j)d��|$j'd }|j*tjj+k�r�tj(j)d��d}%|j,� }|tjjk�r�|d j-|
k�r�d
}nd
}x�|$j'|%d� D ]�}|�r�tj(j)d��|j*tjj+k�rl|j| k�rl|�r&|d j-|
k�r tj(j)d��d }n|tjjk�r:| }||k�r~|tjj.k�sf|tjjk�r~|�r~d
}n|�r�tjj.}d }�q�W |�r�|j$�r�|$j/ �r�tj(j)d��|$V �q~W |j0� dS )a� Return a generator for the responses to a zone transfer.
@param where: where to send the message
@type where: string containing an IPv4 or IPv6 address
@param zone: The name of the zone to transfer
@type zone: dns.name.Name object or string
@param rdtype: The type of zone transfer. The default is
dns.rdatatype.AXFR.
@type rdtype: int or string
@param rdclass: The class of the zone transfer. The default is
dns.rdataclass.IN.
@type rdclass: int or string
@param timeout: The number of seconds to wait for each response message.
If None, the default, wait forever.
@type timeout: float
@param port: The port to which to send the message. The default is 53.
@type port: int
@param keyring: The TSIG keyring to use
@type keyring: dict
@param keyname: The name of the TSIG key to use
@type keyname: dns.name.Name object or string
@param relativize: If True, all names in the zone will be relativized to
the zone origin. It is essential that the relativize setting matches
the one specified to dns.zone.from_xfr().
@type relativize: bool
@param af: the address family to use. The default is None, which
causes the address family to use to be inferred from the form of where.
If the inference attempt fails, AF_INET is used.
@type af: int
@param lifetime: The total number of seconds to spend doing the transfer.
If None, the default, then there is no limit on the time the transfer may
take.
@type lifetime: float
@rtype: generator of dns.message.Message objects.
@param source: source address. The default is the wildcard address.
@type source: string
@param source_port: The port from which to send the message.
The default is 0.
@type source_port: int
@param serial: The SOA serial number to use as the base for an IXFR diff
sequence (only meaningful if rdtype == dns.rdatatype.IXFR).
@type serial: int
@param use_udp: Use UDP (only meaningful for IXFR)
@type use_udp: bool
@param keyalgorithm: The TSIG algorithm to use; defaults to
dns.tsig.default_algorithm
@type keyalgorithm: string
r �IN�SOAz. . %u 0 0 0 0N)� algorithmzcannot do a UDP AXFRz!HFTi�� rZ )r9 r: �xfr�origin�tsig_ctxZmulti�firstr; z No answer or RRset not for qnamezfirst RRset is not an SOAr zanswers after final SOAzIXFR base serial mismatchz missing TSIG)1�
isinstancer r �nameZ from_text� rdatatyperC Z
make_queryZIXFR�rrsetZ authority�appendZuse_tsigr< r7 �
ValueErrorr= r> r? r` r@ rA r r_ rQ r* rV ra rb rY �emptyr) rB rU rc rD r9 rE rl Zanswerr � FormError�rdtyperh �copy�serial�AXFRZhad_tsigrG )&r3 Zzonerv Zrdclassr r4 r9 ZkeynameZ
relativizer, Zlifetimer5 r6 rx Zuse_udpZ keyalgorithmrH rq rJ r/ r( r" rX rd r# Z delete_modeZ
expecting_SOAZ soa_rrsetrk Zonamerl rm Z mexpirationr. re Zis_ixfrrM Z answer_indexr r r
rj � s� 5
rj )r )Nr8 NNr FFF)Nr8 NNr F)5r Z
__future__r r! r r> ra r[ r Z
dns.exceptionr Zdns.inetZdns.nameZ dns.messageZdns.rdataclassZ
dns.rdatatypeZ_compatr r � version_info�OSErrorr r r= r Z DNSExceptionr ru r r r r r% r' r] r r) r* r- r1 r7 rN rU rY r_ rf rp ry Z
rdataclassrg ZtsigZdefault_algorithmrj r r r r
�<module> sX
I
?