← Back
Editing: query.cpython-36.opt-1.pyc
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 ?
Save File
Cancel