CasperSecurity
o
�b� � @ sn d Z ddlZddlZddlmZmZ ddlmZmZm Z m
Z
ddl m Z G dd� de j
�Z
dd � Zd
d � ZdS ) z
Domain Name Server
� N)�internet�service)� authority�dns� secondary�server)�usagec @ s� e Zd Zg d�g d�g d�g d�gZg d�g d�g d�gZejde�� id �Zd
Z d
Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zd
S )�Options)� interface�i� z The interface to which to bind)�port�p�53zThe port on which to listen)� resolv-confNNz6Override location of resolv.conf (implies --recursive))�
hosts-fileNNz!Perform lookups with a hosts file)�cache�czEnable record caching)� recursive�rzPerform recursive lookups)�verbose�vz
Log verboselyr
)�
optActionsNc C s* t j�| � d| d< g | _g | _g | _d S )Nr r )r r �__init__� bindfiles� zonefiles� secondaries��self� r �3/usr/lib/python3/dist-packages/twisted/names/tap.pyr + s
zOptions.__init__c C �* t j�|�s
t�|d ��| j�|� dS )z7Specify the filename of a Python syntax zone definition�: No such fileN)�os�path�existsr �
UsageErrorr �append�r �filenamer r r �
opt_pyzone2 � zOptions.opt_pyzonec C r! )z6Specify the filename of a BIND9 syntax zone definitionr" N)r# r$ r% r r& r r' r( r r r � opt_bindzone8 r+ zOptions.opt_bindzonec C s� |� dd�}t|�dkrt�d��|d � d�}t|�dkr&|d tjf}n zt|d �}W n ty? t�d|d ����w |d |f}| j� ||d gf� dS ) zwAct as secondary for the specified domain, performing
zone transfers from the specified IP (IP/domain)
�/� � z-Argument must be of the form IP[:port]/domainr �:z$Specify an integer port number, not N)
�split�lenr r& r �PORT�int�
ValueErrorr r' )r � ip_domain�args�addressr
r r r �
opt_secondary> s
�� zOptions.opt_secondaryc C s | d d7 < dS )zIncrement verbosity levelr r. Nr r r r r � opt_verboseR s zOptions.opt_verbosec C s | d rd| d< g | _ g | _| jD ] }z | j�t�|�� W q ty0 t�� t �
d| ��w | j D ] }z | j�t� |�� W q4 tyS t�� t �
d| ��w | j
D ]}tjj|� }| j �|� | j�| j d �� � qWz t| d �| d< W d S ty� t �
d�| d ���w )Nr Tr zInvalid syntax in ���r
zInvalid port: {!r})�svcs�zonesr r' r �PySourceAuthority� Exception� traceback� print_excr r& r �
BindAuthorityr r �SecondaryAuthorityService�fromServerAddressAndDomains� getAuthorityr4 r5 �format)r �f�svcr r r � postOptionsV s6
�
�
�zOptions.postOptions)�__name__�
__module__� __qualname__�
optParameters�optFlagsr � Completions�CompleteNetInterfaces�compDatar= r r r* r, r9 r: rI r r r r r s* �
�
� r c C s� ddl m}m}m} g g }}| d r |�|j| d d�� | d r-|�|j| d d�� | d r<|�|j| d d
�� ||fS ) a%
Build DNS resolver instances in an order which leaves recursive
resolving as a last resort.
@type config: L{Options} instance
@param config: Parsed command-line configuration
@return: Two-item tuple of a list of cache resovers and a list of client
resolvers
r )r �client�hostsr r )r r )�filer r )�
resolvconf)�
twisted.namesr rR rS r' �
CacheResolver�Resolver�createResolver)�configr rR rS �ca�clr r r �_buildResolversr s
r] c
C s� t | �\}}t�| j||| d �}t�|�}d|_t�� }t j
|ft j |ffD ]\}}|| d || d d�}|� |� q'| j
D ]} | � |� q?|S )Nr r r
r
)r
)r] r �DNSServerFactoryr= r �DNSDatagramProtocol�noisyr � MultiServicer � TCPServer� UDPServer�setServiceParentr< )
rZ r[ r\ rG r �ret�klass�arg�srH r r r � makeService� s
ri )�__doc__r# r@ �twisted.applicationr r rV r r r r �twisted.pythonr r r] ri r r r r �<module> s a