File: //lib/python3.6/site-packages/DNS/__pycache__/Base.cpython-36.opt-1.pyc
3
�W&_�E � @ s\ d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z m
Z
m Z ddl Z yddl
mZ e� Z
W n ddl
Z
Y nX G dd� de�ZG dd � d e�ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� de�ZddlmZ dde je jddddg d� ad&dd�Zdd� Zd d � Z d d � Z G d d!� d!�Z G d"d#� d#e e j �Z d$d%� Z!dS )'u<
$Id$
This file is part of the py3dns project.
Homepage: https://launchpad.net/py3dns
This code is covered by the standard Python License. See LICENSE for details.
Changes for Python3 port © 2011-14 Scott Kitterman <scott@kitterman.com>
Base functionality. Request and Response classes, that sort of thing.
� N� )�Type�Class�Opcode)� SystemRandomc @ s e Zd ZdS )�DNSErrorN)�__name__�
__module__� __qualname__� r r �/usr/lib/python3.6/Base.pyr s r c @ s e Zd ZdS )�
ArgumentErrorN)r r r
r r r r r
s r
c @ s e Zd ZdS )� SocketErrorN)r r r
r r r r r ! s r c @ s e Zd ZdS )� TimeoutErrorN)r r r
r r r r r " s r c @ s e Zd Zdd� ZdS )� ServerErrorc C s t j| ||� || _|| _d S )N)r �__init__�message�rcode)�selfr r r r r r % s zServerError.__init__N)r r r
r r r r r r $ s r c @ s e Zd ZdS )�IncompleteReplyErrorN)r r r
r r r r r * s r )�Lib�udp�5 � ) �protocol�port�opcode�qtype�rdZtiming�timeout�
server_rotate�server�/etc/resolv.confc C sR y t | d��
}t|�S Q R X W n. tk
rL } ztjt|�� dS d}~X nX dS )zCparses the /etc/resolv.conf file and sets defaults for name servers�rN)�open�ParseResolvConfFromIterable�FileNotFoundError�warnings�warn�str)Z resolv_path�stream�er r r �ParseResolvConf3 s r, c C s� x�| D ]�}|j � }| s|d dks|d dkr2q|j� }t|�dk rHq|d dkrlt|�dkrl|d td< |d dkrx|d dkr�|d d kr�|d d
krtd j|d � qW d S )
zHparses a resolv.conf formatted stream and sets defaults for name serversr �;�#� Zdomainr �searchZoptionsZsortlistZ
nameserverr! N)�strip�split�len�defaults�append)�lines�lineZfieldsr r r r% < s
r% c C sH dd l } | jdkr,ddlm} |j� td< n| jdkr>t� nt� S d S ) Nr �win32�ntr )�win32dnsr! �darwin)r8 r9 )�sys�platform� r: ZRegistryResolver4 �ParseOSXSysConfigr, )r< r: r r r �_DiscoverNameServersQ s
r@ c C s dS )zbDon't call, only here for backward compatability. We do discovery for
you automatically.
Nr r r r r �DiscoverNameServers[ s rA c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �
DnsRequestz high level Request object c O sP d | _ d | _i | _| j||� | j| _d| _d| _t| jd �dkrLtd��d S )Nr r> r! z"No working name servers discovered) �donefunc�py3asyncr4 �argparse�args�tid�
resulttyper3 r )r �namerF r r r r c s zDnsRequest.__init__c C s | r d| j kr | j d |d< t|�tks6t|�tkr@||d< n t|�dkr`|d r`|d |d< t d r�tt d �tjkr�t d dd � t d d d� t d< xBtt j� �D ]2}||kr�|| j kr�| j | ||< q�t | ||< q�W t|d �tk�st|d �tk�r|d g|d< || _ d S )NrI r r r r! )
r4 �type�bytesr) r3 �typesZListType�list�keysrF )r rI rF �ir r r rE n s$
$
$zDnsRequest.argparsec C s t j ||�| _d S )N)�socket�s)r �a�br r r �
socketInit� s zDnsRequest.socketInitc C sh | j dkr6tj| jgg g | j �\}}}t|�s6td��| jjd�\| _| _tj� | _ | j
| j d< | j � S )Nr �Timeouti�� r! )
r �selectrQ r3 r Zrecvfrom�reply� from_address�time� time_finish�nsrF � processReply)r r# �wr+ r r r �processUDPReply� s
zDnsRequest.processUDPReplyc C s� |j |�}x|t|�|k r�| jdkrR| j| j tj� }|dkrFtd��| jj|� |j |t|� �}|s|tdt|�|f ��||7 }q W |S )Nr rU z incomplete reply - %d of %d read)�readr3 r �
time_startrY r rQ �
settimeout)r �f�count�resZrem�bufr r r �_readall� s
zDnsRequest._readallc C s� | j dkr| jj| j � n | jjd � | jjd�}z(| j|d�}tj|�}| j||�| _W d |j� X t j � | _
| j | j d< | j
� S )Nr �rbr/ r! )r rQ ra �makefilerf r Z unpack16bitrW �closerY rZ r[ rF r\ )r rb �headerrc r r r �processTCPReply� s
zDnsRequest.processTCPReplyc C s� | j | j d | jd< | js*tj| j�}nn| jdkrBtj| j�}nV| jdkrZtj| j�}n>| jdkrrtj | j�}n&| jdkr�tj
| j�}nt d| j ��tj || j�}| j|_|S )Ni� �elapsed�defaultZbinary�textZintegerzUnknown resulttype: )
rZ r` rF rH r Z MunpackerrW ZMunpackerDefaultZMunpackerBinaryZ
MunpackerTextZMunpackerInteger� SyntaxErrorZ DnsResult)r �ur# r r r r\ � s
zDnsRequest.processReplyc C s` xZy"t jdd�}| jjd|f� P W q tjk
rV } z|jtjkrF� W Y dd}~X qX qW dS )z<Pick random source port to avoid DNS cache poisoning attack.i i�� r> N)�random�randintrQ ZbindrP �error�errnoZ
EADDRINUSE)r Z source_port�msgr r r � getSource� s zDnsRequest.getSourcec C s | j � | jj| j| jf� d S )N)rv rQ �connectr[ r )r r r r �conn� s zDnsRequest.connc O s� | j ||� | jd }| jd | _tjdd�| _| jd | _| jd }| jd }| jd }d | jkrr| jd | _nd
| _t| jd �t ks�t| jd �t
kr�y t t t
| jd j
� ��}W q� tk
r� td ��Y q�X n
| jd }d
| jkr�t| j� td��| jd
}|t jk�r*|dk�r*td� d}tj� } | j| jd|dd|ddddddd�
| j||tj� | j� | _y$|dk�r�| j|� n
| j|� W n. t j k
�r� }
zt |
��W Y dd}
~
X nX | j �r�dS | j S dS )aW
Request function for the DnsRequest class. In addition to standard
DNS args, the special pydns arg 'resulttype' can optionally be passed.
Valid resulttypes are 'default', 'text', 'decimal', and 'binary'.
Defaults are configured to be compatible with pydns:
AAAA: decimal
Others: text
r r r i�� r r r r! rH rm r zunknown query typerI znothing to lookup�tcpz'Query type AXFR, protocol forced to TCPr r N)!rE rF r rq rr rG r rH rJ rK r) �getattrr �upper�AttributeErrorr
�print�AXFRr �Mpacker� addHeader� addQuestionr �IN�getbuf�request�sendUDPRequest�sendTCPRequestrP rs r rD �response) r rI rF r r r r! r �qname�m�reasonr r r �qry� sN
$
zDnsRequest.qryc O s� | j ||� y"| jd r,tdj| jd ���W n Y nX | jd }| jd | _tjdd�| _| jd | _| jd }| jd }| jd
}t | jd �t
ks�t | jd �t kr�y t t
t | jd j� ��}W q� tk
r� td ��Y q�X n
| jd }d
| jk�rt| j� td��| jd
}|t
jk�r>|dk�r>td� d}tj� } | j| jd|dd|ddddddd�
| j||tj� | j� | _y$|dk�r�| j|� n
| j|� W n. t j k
�r� }
zt |
��W Y dd}
~
X nX | j �r�dS | j S dS )z needs a refactoring rH zERestulttype {0} set with DNS.req, use DNS.qry to specify result type.r r r i�� r r r r! r zunknown query typerI znothing to lookupry z'Query type AXFR, protocol forced to TCPr r N)!rE rF r
�formatr rq rr rG r rJ rK r) rz r r{ r| r} r~ r r r� r� r r� r� r� r� r� rP rs r rD r� ) r rI rF r r r r! r r� r� r� r r r �req sR
$
zDnsRequest.reqc C s^ d}d| _ �x8|D �].| _y�| jjd�rNttd�rJtjrJ| jtjtj� q^wn| jtj tj� zft
j
� | _ | j � | j
s�| jj| j� | j� }x,|jd | jks�| jd | jkr�| j� }q�W || _ W d| j
s�| jj� X W n^ tjk
�r
} z|p�|}wW Y dd}~X n0 tk
�r8 } z|�p$|}wW Y dd}~X nX | j rP qW | j �rZ|�rZ|�dS )z refactor meN�:�has_ipv6�idr )r� r[ rc �hasattrrP r� rT �AF_INET6Z
SOCK_DGRAM�AF_INETrY r` rx rD rQ �sendr� r^ rj rG rX r ri rs r )r r! �first_socket_errorr# r+ �tr r r r� R s>
zDnsRequest.sendUDPRequestc C sV d}d| _ �x0|D �]&| _y�| jjd�rNttd�rJtjrJ| jtjtj� q^wn| jtj tj� zdt
j
� | _ | j � t
jt| j��| j }| jjd� | jj|� | j� }|jd | jkr�|| _ P W d| jj� X W n^ tjk
�r } z|p�|}wW Y dd}~X n0 tk
�r0 } z|�p |}wW Y dd}~X nX | j rP qW | j �rR|�rR|�dS )z& do the work of sending a TCP request Nr� r� r r� )r� r[ rc r� rP r� rT r� Z SOCK_STREAMr� rY r` rx r Z pack16bitr3 r� rQ Z setblockingZsendallrk rj rG ri rs r )r r! r� re r# r+ r� r r r r� s<
zDnsRequest.sendTCPRequestN)r r r
�__doc__r rE rT r^ rf rk r\ rv rx r� r� r� r� r r r r rB a s
% ;4-rB c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�DnsAsyncRequestz> an asynchronous request object. out of date, probably broken c O s@ t j| f|�|� d|kr.|d r.|d | _n| j| _d| _d S )N�doner )rB r rC �
showResultrD )r rI rF r r r r � s
zDnsAsyncRequest.__init__c C sH | j � | j| j| jf� tj� | _d| jkrD| jd rDtjj | � d S )N�start)
rv rw r[ r rY r` rF �asyncore�
dispatcherZgo)r r r r rx � s
zDnsAsyncRequest.connc C s" | j ||� tjj| � | | _d S )N)Z
create_socketr� r� r rQ )r rR rS r r r rT � s zDnsAsyncRequest.socketInitc C s. | j d dkr*| j� | _| jr*| j| f� d S )Nr r )rF r^ r� rC )r r r r � handle_read� s
zDnsAsyncRequest.handle_readc C s | j | j� d S )N)r� r� )r r r r �handle_connect� s z DnsAsyncRequest.handle_connectc C s d S )Nr )r r r r � handle_write� s z DnsAsyncRequest.handle_writec G s | j j� d S )N)r� Zshow)r rQ r r r r� � s zDnsAsyncRequest.showResultN) r r r
r� r rx rT r� r� r� r� r r r r r� � s r� c C s� ddl } ddl}| jdd�}|jd�}g }d}xn|j� }|s>P |j� }t|�dk s`|d tjkrfd}q0|j |�}|r0|dkr�g }|j
|� |j
|j d�� q0W |j � x|D ]}t
d j|� q�W dS )zMRetrieves the current Mac OS X resolver settings using the scutil(8) command.r Nz/usr/sbin/scutil --dnsr# z$^\s+nameserver[]0-9[]*\s*\:\s*(\S+)$r r! )�os�re�popen�compile�readline�rstripr3 �stringZ
whitespace�matchr5 �groupri r4 �extend)r� r� ZscutilZres_reZsetsZ
currentset�lr� r r r r? � s,
r? )r" )"r� rP r� rL rY rV r' rt r> r r r r� rq r � Exceptionr r
r r r r r ZQUERY�Ar4 r, r% r@ rA rB Zdispatcher_with_sendr� r? r r r r �<module> s8 0
K!