← Back
Editing: authority.cpython-38.pyc
U W[�@ � @ s� d Z ddlmZmZ ddlZddlZddlmZmZm Z ddl m Z ddl m Z ddlmZmZmZ ddlmZ dd d �ZG d d � d e j�ZG dd� de�ZG dd� de�ZdS )z Authoritative resolvers. � )�absolute_import�divisionN)�dns�error�common)�defer)�failure)�execfile� nativeString�_PY3)�FilePath�/tmp/twisted-names.serialc C s� t �d�}t�d�}z4tj�| �sFt| d��}|�|d � W 5 Q R X W 5 t�|� X t| d��}|�� � � \}}W 5 Q R X ||kr�t |�d p�d}t| d��}|�d||f � W 5 Q R X |d |f }|S ) ay Return a monotonically increasing (across program runs) integer. State is stored in the given file. If it does not exist, it is created with rw-/---/--- permissions. @param filename: Path to a file that is used to store the state across program runs. @type filename: L{str} @return: a monotonically increasing number @rtype: L{str} z%Y%m%d� �wz 0�r� r z%s %dz%02d) �time�strftime�os�umask�path�exists�open�write�readline�split�int)�filename�serial�o�fZ serialFileZ lastSerialZzoneID� r! �9/usr/lib/python3/dist-packages/twisted/names/authority.py� getSerial s r# c @ sh e Zd ZdZejejejfZej ej fZ dZ dZ dd� Zdd� Zdd� Zdd d �Zdd d �Zdd� ZdS )� FileAuthoritya� An Authority that is loaded from a file. @ivar _ADDITIONAL_PROCESSING_TYPES: Record types for which additional processing will be done. @ivar _ADDRESS_TYPES: Record types which are useful for inclusion in the additional section generated during additional processing. @ivar soa: A 2-tuple containing the SOA domain name as a L{bytes} and a L{dns.Record_SOA}. Nc C s t j�| � | �|� i | _d S �N)r � ResolverBase�__init__�loadFile�_cache)�selfr r! r! r" r' O s zFileAuthority.__init__c C s || _ d S r% )�__dict__)r* �stater! r! r" � __setstate__U s zFileAuthority.__setstate__c c sn || D ]`}|j | jkr|jjj}| j�|�� d�D ]2}|j| jkr4t j ||jt j |j pZ||dd�V q4qdS )aW Find locally known information that could be useful to the consumer of the response and construct appropriate records to include in the I{additional} section of that response. Essentially, implement RFC 1034 section 4.3.2 step 6. @param answer: A L{list} of the records which will be included in the I{answer} section of the response. @param authority: A L{list} of the records which will be included in the I{authority} section of the response. @param ttl: The default TTL for records for which this is not otherwise specified. @return: A generator of L{dns.RRHeader} instances for inclusion in the I{additional} section. These instances represent extra information about the records in C{answer} and C{authority}. r! T��authN) �type� _ADDITIONAL_PROCESSING_TYPESZpayload�name�records�get�lower�TYPE�_ADDRESS_TYPESr �RRHeader�IN�ttl)r* Zanswer� authorityr: �recordr2 �recr! r! r" �_additionalRecordsY s �z FileAuthority._additionalRecordsc C s� g }g }g }g }t | jd j| jd j�} | j�|�� �} | �r�| D ]�} | jdk rZ| j} n| } | jt j kr�|�� | jd �� kr�|� t j || jt j | | dd�� n4| j|ks�|t jkr�|� t j || jt j | | dd�� | jt jkrD|� t j || jt j | | dd�� qD|�s|}| �||| �} |�r.|�| � n |�| � |�sp|�sp|� t j | jd t jt j | | jd dd�� t�|||f�S t �|| jd ��r�t�t�t �|���S t�t�t�|���S dS )a� Determine a response to a particular DNS query. @param name: The name which is being queried and for which to lookup a response. @type name: L{bytes} @param cls: The class which is being queried. Only I{IN} is implemented here and this value is presently disregarded. @type cls: L{int} @param type: The type of records being queried. See the types defined in L{twisted.names.dns}. @type type: L{int} @param timeout: All processing is done locally and a result is available immediately, so the timeout value is ignored. @return: A L{Deferred} that fires with a L{tuple} of three sets of response records (to comprise the I{answer}, I{authority}, and I{additional} sections of a DNS response) or with a L{Failure} if there is a problem processing the query. r Nr Fr. T) �max�soa�minimum�expirer3 r4 r5 r: r6 r �NS�appendr8 r9 Z ALL_RECORDS�CNAMEr> �extend�SOAr �succeedZ_isSubdomainOf�failr �FailureZAuthoritativeDomainErrorr � DomainError)r* r2 �clsr0 �timeoutZcnames�resultsr; Z additional� default_ttlZdomain_recordsr< r: ZadditionalInformationr! r! r" �_lookupx s� � �� �� �� � ���zFileAuthority._lookup� c C s | j d �� |�� kr�t| j d j| j d j�}| j d jd k rN| j d j}n|}tj| j d tjtj || j d dd�g}| j � � D ]T\}}|D ]F}|jd k r�|j} n|} |j tjkr�|� tj||j tj | |dd�� q�q�|� |d � t�|ddf�S t�t�t�|���S )Nr r Tr. r! )r@ r5 r? rA rB r: r r8 rG r9 r3 �itemsr6 rD r rH rI r rJ rK ) r* r2 rM rO Zsoa_ttlrN �kr r= r: r! r! r" � lookupZone� sB �� ��zFileAuthority.lookupZonec C sb g g g }}}|D ]B}|d r|� |d d � |� |d d � |� |d d � q|||fS )Nr r � )rF )r* rN Zansr/ �add�resr! r! r" � _cbAllRecords� s zFileAuthority._cbAllRecords)N)rQ )�__name__� __module__� __qualname__�__doc__r rE ZMXrC r1 �AZAAAAr7 r@ r3 r' r- r> rP rT rX r! r! r! r" r$ ; s c r$ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �PySourceAuthorityzC A FileAuthority that is built up from Python source code. c C s| | � � i }}t|||� d|kr.td| ��i | _|d D ]:}t|d tj�rV|| _| j�|d � � g �� |d � q<d S )NZzonezNo zone defined in r r ) �setupConfigNamespacer � ValueErrorr3 � isinstancer Z Record_SOAr@ � setdefaultr5 rD )r* r �g�lZrrr! r! r" r( s zPySourceAuthority.loadFilec s � fdd�S )Nc s | � ||�fS r% r! )r2 �arg�kw�r0 r! r" �<lambda> � z.PySourceAuthority.wrapRecord.<locals>.<lambda>r! )r* r0 r! rg r" � wrapRecord s z PySourceAuthority.wrapRecordc C sN i }t j�� }dd� |D �D ],}tt |�}| �|�}|||td�d � < q |S )Nc S s g | ]}|� d �r|�qS )�Record_)� startswith)�.0�xr! r! r" � <listcomp> s z:PySourceAuthority.setupConfigNamespace.<locals>.<listcomp>rk )r r+ Ziterkeys�getattrrj �len)r* r rR r< r0 r r! r! r" r_ s z&PySourceAuthority.setupConfigNamespaceN)rY rZ r[ r\ r( rj r_ r! r! r! r" r^ s r^ c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )� BindAuthorityz� An Authority that loads U{BIND zone files <https://en.wikipedia.org/wiki/Zone_file>}. Supports only C{$ORIGIN} and C{$TTL} directives. c C sJ t |�}t|�� d �| _|�� �d�}| �|�}| �|�}| �|� dS )z} Load records from C{filename}. @param filename: file to read from @type filename: L{bytes} � .TN) r r �basename�originZ getContent� splitlines� stripComments�collapseContinuations� parseLines)r* r �fp�linesr! r! r" r( , s zBindAuthority.loadFilec C s dd� dd� |D �D �S )z� Strip comments from C{lines}. @param lines: lines to work on @type lines: iterable of L{bytes} @return: C{lines} sans comments. c s s2 | ]*}|� d �dkr|p(|d|� d �� V qdS )� ;���N)�find)rm �ar! r! r" � <genexpr>G s z.BindAuthority.stripComments.<locals>.<genexpr>c S s g | ] }|� � �qS r! )�strip)rm �br! r! r" ro H s z/BindAuthority.stripComments.<locals>.<listcomp>r! )r* r{ r! r! r" rw > s ��zBindAuthority.stripCommentsc C s� g }d}|D ]�}|dkrP|� d�dkr2|�|� q�|�|d|� d�� � d}q |� d�dkr�|d d|d|� d�� 7 < d}q |d d| 7 < q tddd � |D ��S ) z� Transform multiline statements into single lines. @param lines: lines to work on @type lines: iterable of L{bytes} @return: iterable of continuous lines r � (r} Nr � )� c s s | ]}|� � V qd S r% )r )rm �liner! r! r" r� f s z6BindAuthority.collapseContinuations.<locals>.<genexpr>)r~ rD �filter)r* r{ rd r, r� r! r! r" rx N s "z#BindAuthority.collapseContinuationsc C s� d}| j }i | _|D ]p}|d dkr4t�|d �}q|d dkrJ|d }q|d dkr`td��q|d dkrvtd ��q| �|||� q|| _ d S ) zs Parse C{lines}. @param lines: lines to work on @type lines: iterable of L{bytes} i0* r s $TTLr s $ORIGINs $INCLUDEz"$INCLUDE directive not implementeds $GENERATEz#$GENERATE directive not implementedN)ru r3 r Zstr2time�NotImplementedError�parseRecordLine)r* r{ r: ru r� r! r! r" ry i s �zBindAuthority.parseLinesc C sd |� d�s |d |dd� }n |dd� }t| d|f d�}|rR|||||� ntd|f ��dS )a Add a record to our authority. Expand domain with origin if necessary. @param owner: origin? @type owner: L{bytes} @param ttl: time to live for the record @type ttl: L{int} @param domain: the domain for which the record is to be added @type domain: L{bytes} @param type: record type @type type: L{str} @param cls: record class @type cls: L{str} @param rdata: record data @type rdata: L{list} of L{bytes} rs Nr} zclass_%sz Record class %r not supported)�endswithrp r� )r* �ownerr: r0 �domainrL �rdatar r! r! r" � addRecord� s �zBindAuthority.addRecordc C sj t tdt|�f d�}|rT||� }||_| j�|�� g ��|� |dkrf||f| _nt dt|�f ��dS )a> Simulate a class IN and recurse into the actual class. @param ttl: time to live for the record @type ttl: L{int} @param type: record type @type type: str @param domain: the domain @type domain: bytes @param rdata: @type rdate: bytes z Record_%sNrG z Record type %r not supported) rp r r r: r3 rb r5 rD r@ r� )r* r: r0 r� r� r< r r! r! r" �class_IN� s �zBindAuthority.class_INc C s� t r6tdd� tj�� D ��}tdd� tj�� D ��}n ttj�� �}ttj�� �}||B }d}|}|d dkr�|dd� }|}n,|d �� s�|d |kr�|d }|dd� }|d �� s�|d |kr�|} |}n|d } |dd� }|d |k�r,|d }|dd� }|d �� �rtt|d �}|dd� }nH|d �� �rtt|d �}|dd� }|d |k�rt|d }|dd� }|d } |dd� } | �||t | �| t |�| � dS ) a� Parse a C{line} from a zone file respecting C{origin} and C{ttl}. Add resulting records to authority. @param origin: starting point for the zone @type origin: L{bytes} @param ttl: time to live for the record @type ttl: L{int} @param line: zone file line to parse; split by word @type line: L{list} of L{bytes} c s s | ]}|� d �V qdS ��asciiN��encode)rm Zqcr! r! r" r� � s z0BindAuthority.parseRecordLine.<locals>.<genexpr>c s s | ]}|� d �V qdS r� r� )rm Zqtr! r! r" r� � s s INr � @r N) r �setr Z QUERY_CLASSES�valuesZ QUERY_TYPES�isdigitr r� r ) r* ru r: r� Z queryClassesZ queryTypesZmarkersrL r� r� r0 r� r! r! r" r� � sZ �� �z BindAuthority.parseRecordLineN) rY rZ r[ r\ r( rw rx ry r� r� r� r! r! r! r" rr % s # rr )r )r\ Z __future__r r r r Z twisted.namesr r r Ztwisted.internetr Ztwisted.pythonr Ztwisted.python.compatr r r Ztwisted.python.filepathr r# r& r$ r^ rr r! r! r! r" �<module> s % K
Save File
Cancel