File: //opt/alt/python37/lib/python3.7/site-packages/dns/__pycache__/message.cpython-37.pyc
B
��f�� � @ s� d Z ddlmZmZmZmZmZmZ ddlZddl Z ddl
Z
ddl Z ddl
Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddlZ ddl Z ddl Z G dd� de j j �Z G dd� de j j �Z!G dd � d e j j"�Z#G d
d � d e j j �Z$G d d
� d
e j j �Z%G dd� de j j"�Z&G dd� de j j"�Z'G dd� de j j"�Z(G dd� de j j"�Z)G dd� de j j"�Z*G dd� de j j+�Z,G dd� de j-j.�Z/G d d � d �Z0d Z1d Z2ee3e j4j5e j6j7e j8j9ee j8j9 ee j6j7 f Z:ee:e j;j<f Z=ee3ee j;j< f Z>G d d!� d!�Z?G d"d#� d#�Z@G d$d%� d%e?�ZAd&d'� ZBd(d)� ZCG d*d+� d+�ZDdBeEee eeE eFee j4j5 eee jGjHe jGjIf eFeFeFeFeFeFe?d.�
d/d0�ZJG d1d2� d2�ZKdCeLee j4jM eFee j4j5 eFee j4j5 e?d4�d5d6�ZNdDeee j4jM eFe?d7�d8d9�ZOe j6jPdd-dddddde jQjRdf ee j4j5eLf ee j8j9eLf ee j6j7eLf eee3eFf eFee3 ee3 ee3 eee jSjT ee j4jM ee3 e3e3eAd:�d;d<�ZUdEe?eFe3e3e3e?d?�d@dA�ZVe/jWZWe/jXZXe/jYZYe/jZZZdS )Fz DNS Messages� )�Any�Dict�List�Optional�Tuple�UnionNc @ s e Zd ZdZdS )� ShortHeaderz2The DNS packet passed to from_wire() is too short.N)�__name__�
__module__� __qualname__�__doc__� r
r
�</opt/alt/python37/lib/python3.7/site-packages/dns/message.pyr . s r c @ s e Zd ZdZdS )� TrailingJunkzEThe DNS packet passed to from_wire() has extra junk at the end of it.N)r r
r r r
r
r
r r 2 s r c @ s e Zd ZdZdS )�UnknownHeaderFieldzVThe header field name was not recognized when converting from text
into a message.N)r r
r r r
r
r
r r 6 s r c @ s e Zd ZdZdS )�BadEDNSzLAn OPT record occurred somewhere other than
the additional data section.N)r r
r r r
r
r
r r ; s r c @ s e Zd ZdZdS )�BadTSIGzWA TSIG record occurred somewhere other than the end of
the additional data section.N)r r
r r r
r
r
r r @ s r c @ s e Zd ZdZdS )�UnknownTSIGKeyz(A TSIG with an unknown key was received.N)r r
r r r
r
r
r r E s r c s. e Zd ZdZdhZ� fdd�Zdd� Z� ZS )� TruncatedzThe truncated flag is set.�messagec s t � j||� d S )N)�super�__init__)�self�args�kwargs)� __class__r
r r P s zTruncated.__init__c C s
| j d S )zbAs much of the message as could be processed.
Returns a ``dns.message.Message``.
r )r )r r
r
r r S s zTruncated.message)r r
r r Z supp_kwargsr r �
__classcell__r
r
)r r r I s r c @ s e Zd ZdZdS )�NotQueryResponsez%Message is not a response to a query.N)r r
r r r
r
r
r r [ s r c @ s e Zd ZdZdS )� ChainTooLongz The CNAME chain is too long.N)r r
r r r
r
r
r r _ s r c @ s e Zd ZdZdS )�AnswerForNXDOMAINz.The rcode is NXDOMAIN but an answer was found.N)r r
r r r
r
r
r r c s r c @ s e Zd ZdZdS )�NoPreviousNamezNo previous name was known.N)r r
r r r
r
r
r r g s r c @ s, e Zd ZdZdZdZdZdZedd� �Z dS ) �MessageSectionzMessage sectionsr � � � c C s dS )Nr$ r
)�clsr
r
r �_maximums s zMessageSection._maximumN)
r r
r r �QUESTION�ANSWER� AUTHORITY�
ADDITIONAL� classmethodr& r
r
r
r r! k s r! c @ s e Zd Zeed�dd�ZdS )� MessageError)� exception�offsetc C s || _ || _d S )N)r- r. )r r- r. r
r
r r y s zMessageError.__init__N)r r
r � Exception�intr r
r
r
r r, x s r, i� � c @ s e Zd ZdZeZdhee d�dd�Ze e
e j j
d�dd��Zejd d� �Ze e
e j j
d�d
d ��Zejd d � �Ze e
e j j
d�d
d��Zejdd� �Ze e
e j j
d�dd��Zejdd� �Zdd� Zdd� Zdiee jj eeeef ed�dd�Zdd � Z d d � Z d ed �d d!�Z e
e j j
ed"�d#d$�Z ee
e j j
d%�d&d'�Z e j!j"dd(d(fe#e jje j$j%e j!j&e j!j&ee j$j% eee j j
d)� d*d+�Z'e j!j"dd(d(fe#e jje j$j%e j!j&e j!j&ee j$j% eeee j j
d)� d,d-�Z(ed�d.d/�Z)ed�d0d1�Z*djee jj eeee eeef e+d3�d4d5�Z,e-d6d7� �Z.dd8dd2d9e j/j0feee1e jjef eee ee+e1e jjef dd:�d;d<�Z2e ee jj d�d=d>��Z3e ee jj d�d?d@��Z4e ee+ d�dAdB��Z5e ee d�dCdD��Z6e ed�dEdF��Z7e-d2e8dfdGdH��Z9d2d2e8ddd2fee1eef eeee ee
e j:j; eddI�dJdK�Z<e ed�dLdM��Z:e ed�dNdO��Z=e=jdPdO� �Z=e ed�dQdR��Z>e e?d�dSdT��Z@dkeddU�dVdW�ZAe jBjCd�dXdY�ZBe jBjCddZ�d[d\�ZDe jEjFd�d]d^�ZEe jEjFdd_�d`da�ZGdbdc� ZHddde� ZIdfdg� ZJdS )l�MessagezA DNS message.N)�idc C s| |d krt j�� | _n|| _d| _g g g g g| _d | _d| _d| _d | _ d | _
d| _ d| _ d | _
d | _i | _g | _d| _d S )Nr � Fg )�dnsZentropyZ random_16r3 �flags�sections�opt�request_payload�pad�keyring�tsig� request_mac�xfr�origin�tsig_ctx�index�errors�time)r r3 r
r
r r � s" zMessage.__init__)�returnc C s
| j d S )zThe question section.r )r7 )r r
r
r �question� s zMessage.questionc C s || j d< d S )Nr )r7 )r �vr
r
r rE � s c C s
| j d S )zThe answer section.r" )r7 )r r
r
r �answer� s zMessage.answerc C s || j d< d S )Nr" )r7 )r rF r
r
r rG � s c C s
| j d S )zThe authority section.r# )r7 )r r
r
r � authority� s zMessage.authorityc C s || j d< d S )Nr# )r7 )r rF r
r
r rH � s c C s
| j d S )z The additional data section.r$ )r7 )r r
r
r �
additional� s zMessage.additionalc C s || j d< d S )Nr$ )r7 )r rF r
r
r rI � s c C s dt | j� d S )Nz<DNS message, ID �>)�reprr3 )r r
r
r �__repr__� s zMessage.__repr__c C s | � � S )N)�to_text)r r
r
r �__str__� s zMessage.__str__T)r? �
relativize�kwrD c K sD t �� }|�d| j � |�dtj�| �� � � |�dtj�| �� � � |�dtj�| j� � | j dkr�|�d| j � | j
dkr�|�dtj� | j
� � |�d| j � x | j
D ]}|�d |�� � q�W x`| jj�� D ]P\}}|�d
|� d �� x4| �|�D ]&}|�|j||f|�� |�d � �qW q�W |�� d d
� S )z�Convert the message to text.
The *origin*, *relativize*, and any other keyword
arguments are passed to the RRset ``to_wire()`` method.
Returns a ``str``.
zid %d
z
opcode %s
z rcode %s
z flags %s
r zedns %s
z
eflags %s
z payload %d
z
option %s
�;�
N���)�io�StringIO�writer3 r5 �opcoderM �rcoder6 �edns� ednsflagsZ edns_to_text�payload�options�
_section_enum� __members__�items�section_from_number�getvalue) r r? rO rP �sr8 �name�which�rrsetr
r
r rM � s$
zMessage.to_textc C s� t |t�sdS | j|jkr dS | j|jkr.dS xTt| j�D ]F\}}|j| }x|D ]}||krRdS qRW x|D ]}||krldS qlW q:W dS )z�Two messages are equal if they have the same content in the
header, question, answer, and authority sections.
Returns a ``bool``.
FT)�
isinstancer2 r3 r6 � enumerater7 )r �other�i�sectionZ
other_section�nr
r
r �__eq__� s
zMessage.__eq__c C s | � |� S )N)rl )r rh r
r
r �__ne__
s zMessage.__ne__)rh rD c C s� |j tj j@ dks:| j|jks:tj�| j �tj�|j �kr>dS |�� tjjtjjtjj tjj
hkrtt |j �dkrtdS tj�
| j �r�dS x | j D ]}||j kr�dS q�W x |j D ]}|| j kr�dS q�W dS )zuIs *other*, also a ``dns.message.Message``, a response to this
message?
Returns a ``bool``.
r FT)r6 r5 �QRr3 rW �
from_flagsrX ZFORMERRZSERVFAILZNOTIMPZREFUSED�lenrE Z is_update)r rh rk r
r
r � is_response s(
zMessage.is_response)rj rD c C s6 x(t | j�D ]\}}||kr | �|�S q W td��dS )z�Return the "section number" of the specified section for use
in indexing.
*section* is one of the section attributes of this message.
Raises ``ValueError`` if the section isn't known.
Returns an ``int``.
zunknown sectionN)rg r7 r] �
ValueError)r rj ri Z our_sectionr
r
r �section_number6 s zMessage.section_number)�numberrD c C s | j �|�}| j| S )a Return the section list associated with the specified section
number.
*number* is a section number `int` or the text form of a section
name.
Raises ``ValueError`` if the section isn't known.
Returns a ``list``.
)r] �maker7 )r rt rj r
r
r r` F s zMessage.section_from_numberF) rj rc �rdclass�rdtype�covers�deleting�create� force_uniquerD c
C s� t |t�r|} | �| �}
n| �|�} |}
| |||||f} |s�| jdk r`| j�| �} | dk r�| S n$x"|
D ]} | �|||||�rf| S qfW |s�t�tj �
|||||�} |
� | � | jdk r�| | j| < | S )aP Find the RRset with the given attributes in the specified section.
*section*, an ``int`` section number, or one of the section
attributes of this message. This specifies the
the section of the message to search. For example::
my_message.find_rrset(my_message.answer, name, rdclass, rdtype)
my_message.find_rrset(dns.message.ANSWER, name, rdclass, rdtype)
*name*, a ``dns.name.Name``, the name of the RRset.
*rdclass*, an ``int``, the class of the RRset.
*rdtype*, an ``int``, the type of the RRset.
*covers*, an ``int`` or ``None``, the covers value of the RRset.
The default is ``None``.
*deleting*, an ``int`` or ``None``, the deleting value of the RRset.
The default is ``None``.
*create*, a ``bool``. If ``True``, create the RRset if it is not found.
The created RRset is appended to *section*.
*force_unique*, a ``bool``. If ``True`` and *create* is also ``True``,
create a new RRset regardless of whether a matching RRset exists
already. The default is ``False``. This is useful when creating
DDNS Update messages, as order matters for them.
Raises ``KeyError`` if the RRset was not found and create was
``False``.
Returns a ``dns.rrset.RRset object``.
N) rf r0 r` rs rA �getZ
full_match�KeyErrorr5 re �RRset�append)
r rj rc rv rw rx ry rz r{ rs Z the_section�keyre r
r
r �
find_rrsetU s* .
zMessage.find_rrsetc
C s: y | � ||||||||�} W n tk
r4 d} Y nX | S )a5 Get the RRset with the given attributes in the specified section.
If the RRset is not found, None is returned.
*section*, an ``int`` section number, or one of the section
attributes of this message. This specifies the
the section of the message to search. For example::
my_message.get_rrset(my_message.answer, name, rdclass, rdtype)
my_message.get_rrset(dns.message.ANSWER, name, rdclass, rdtype)
*name*, a ``dns.name.Name``, the name of the RRset.
*rdclass*, an ``int``, the class of the RRset.
*rdtype*, an ``int``, the type of the RRset.
*covers*, an ``int`` or ``None``, the covers value of the RRset.
The default is ``None``.
*deleting*, an ``int`` or ``None``, the deleting value of the RRset.
The default is ``None``.
*create*, a ``bool``. If ``True``, create the RRset if it is not found.
The created RRset is appended to *section*.
*force_unique*, a ``bool``. If ``True`` and *create* is also ``True``,
create a new RRset regardless of whether a matching RRset exists
already. The default is ``False``. This is useful when creating
DDNS Update messages, as order matters for them.
Returns a ``dns.rrset.RRset object`` or ``None``.
N)r� r} )
r rj rc rv rw rx ry rz r{ re r
r
r � get_rrset� s -
zMessage.get_rrsetc C sN | j s
dS d}x,| j d jD ] }|�� }|t|�d 7 }q W | jrJ|d7 }|S )z:Compute the size required for the OPT RR, padding excludedr � � )r8 r\ �to_wirerp r: )r �size�option�wirer
r
r �_compute_opt_reserve� s z Message._compute_opt_reservec C s* | j s
dS t�� }| j �|� t|�� �S )z)Compute the size required for the TSIG RRr )r< rT �BytesIOr� rp ra )r �fr
r
r �_compute_tsig_reserve� s
z Message._compute_tsig_reserver )r? �max_size�multir@ rP rD c K s� |dkr| j dk r| j }|dkr6| jdkr2| j}nd}|dk rDd}n |dkrPd}tj�| j| j||�}| �� }|�|� | � � }|�|� x"| j
D ]} |� | j | j
| j� q�W x"| jD ]} |jtjj| f|� q�W x"| jD ]} |jtjj| f|� q�W x"| jD ]} |jtjj| f|� q�W |�� | jdk �rB|�| j| j||� |�� | jdk �r�tj� |� � | j | jd t t � � �| j!||�\}
} | j�"� | j�#|
� |�tjj| j� |�� |�r�| | _$|� � S )a� Return a string containing the message in DNS compressed wire
format.
Additional keyword arguments are passed to the RRset ``to_wire()``
method.
*origin*, a ``dns.name.Name`` or ``None``, the origin to be appended
to any relative names. If ``None``, and the message has an origin
attribute that is not ``None``, then it will be used.
*max_size*, an ``int``, the maximum size of the wire format
output; default is 0, which means "the message's request
payload, if nonzero, or 65535".
*multi*, a ``bool``, should be set to ``True`` if this message is
part of a multiple message sequence.
*tsig_ctx*, a ``dns.tsig.HMACTSig`` or ``dns.tsig.GSSTSig`` object, the
ongoing TSIG context, used when signing zone transfers.
Raises ``dns.exception.TooBig`` if *max_size* was exceeded.
Returns a ``bytes``.
Nr i�� i )%r? r9 r5 ZrendererZRendererr3 r6 r� Zreserver� rE Z add_questionrc rw rv rG Z add_rrsetr( rH r) rI r* Zrelease_reservedr8 Zadd_optr: Z write_headerr< �signZget_wirer; r0 rC r= �clear�addr@ ) r r? r� r� r@ rP �rZ opt_reserveZ tsig_reservere Znew_tsig�ctxr
r
r r� � sV !
zMessage.to_wirec C s8 t jjj�t jjt jj|||||||� }t j�| d|�S )Nr )r5 �rdtypes�ANY�TSIG�
rdataclass� rdatatypere �
from_rdata) �keyname� algorithmZ time_signed�fudge�mac� original_id�errorrh r< r
r
r �
_make_tsig? s
zMessage._make_tsigi, r4 )r; r� r� r� �
tsig_error�
other_datar� rD c
C s� t |tjj�r|}|j}n`t|�r.|| |�}nLt |t�rDtj�|�}|dkrXtt |��}|| }t |t
�rztj�|||�}|| _ |dkr�| j }| �
|| j jd|dtjj| j j |||�| _dS )a� When sending, a TSIG signature using the specified key
should be added.
*key*, a ``dns.tsig.Key`` is the key to use. If a key is specified,
the *keyring* and *algorithm* fields are not used.
*keyring*, a ``dict``, ``callable`` or ``dns.tsig.Key``, is either
the TSIG keyring or key to use.
The format of a keyring dict is a mapping from TSIG key name, as
``dns.name.Name`` to ``dns.tsig.Key`` or a TSIG secret, a ``bytes``.
If a ``dict`` *keyring* is specified but a *keyname* is not, the key
used will be the first key in the *keyring*. Note that the order of
keys in a dictionary is not defined, so applications should supply a
keyname when a ``dict`` keyring is used, unless they know the keyring
contains only one key. If a ``callable`` keyring is specified, the
callable will be called with the message and the keyname, and is
expected to return a key.
*keyname*, a ``dns.name.Name``, ``str`` or ``None``, the name of
this TSIG key to use; defaults to ``None``. If *keyring* is a
``dict``, the key must be defined in it. If *keyring* is a
``dns.tsig.Key``, this is ignored.
*fudge*, an ``int``, the TSIG time fudge.
*original_id*, an ``int``, the TSIG original id. If ``None``,
the message's id is used.
*tsig_error*, an ``int``, the TSIG error code.
*other_data*, a ``bytes``, the TSIG other data.
*algorithm*, a ``dns.name.Name`` or ``str``, the TSIG algorithm to use. This is
only used if *keyring* is a ``dict``, and the key entry is a ``bytes``.
Nr � )rf r5 r< �Keyrc �callable�str� from_text�next�iter�bytesr; r3 r� r� Z mac_sizes) r r; r� r� r� r� r� r� r� r
r
r �use_tsigP s0 /
zMessage.use_tsigc C s | j r| j jS d S d S )N)r< rc )r r
r
r r� � s zMessage.keynamec C s | j r| j d jS d S d S )Nr )r< r� )r r
r
r � keyalgorithm� s zMessage.keyalgorithmc C s | j r| j d jS d S d S )Nr )r< r� )r r
r
r r� � s z Message.macc C s | j r| j d jS d S d S )Nr )r< r� )r r
r
r r� � s zMessage.tsig_errorc C s
t | j�S )N)�boolr< )r r
r
r �had_tsig� s zMessage.had_tsigc C s4 t jjj�|t jj|pd�}t j�t jjt | �|�S )Nr
)
r5 r� r� �OPTr� re r� rc �rootr0 )r6 r[ r\ r8 r
r
r � _make_opt� s zMessage._make_opt)rY rZ r[ r9 r\ r: rD c C s� |dks|dkrd}n |dkr"d}|dk r8d| _ d| _nH|dM }||d> O }|dkrXg }| �|||�| _ |dkrt|}|| _|| _dS )a' Configure EDNS behavior.
*edns*, an ``int``, is the EDNS level to use. Specifying ``None``, ``False``,
or ``-1`` means "do not use EDNS", and in this case the other parameters are
ignored. Specifying ``True`` is equivalent to specifying 0, i.e. "use EDNS0".
*ednsflags*, an ``int``, the EDNS flag values.
*payload*, an ``int``, is the EDNS sender's payload field, which is the maximum
size of UDP datagram the sender can handle. I.e. how big a response to this
message can be.
*request_payload*, an ``int``, is the EDNS payload size to use when sending this
message. If not specified, defaults to the value of *payload*.
*options*, a list of ``dns.edns.Option`` objects or ``None``, the EDNS options.
*pad*, a non-negative ``int``. If 0, the default, do not pad; otherwise add
padding bytes to make the message size a multiple of *pad*. Note that if
padding is non-zero, an EDNS PADDING option will always be added to the
message.
NFrS Tr l �~ r1 )r8 r9 r� r: )r rY rZ r[ r9 r\ r: r
r
r �use_edns� s zMessage.use_ednsc C s | j r| jd@ d? S dS d S )Ni � r1 rS )r8 rZ )r r
r
r rY � s z Message.ednsc C s | j r| j jS dS d S )Nr )r8 �ttl)r r
r
r rZ � s zMessage.ednsflagsc C s$ | j r|| j _n|r | �|�| _ d S )N)r8 r� r� )r rF r
r
r rZ s
c C s | j r| j d jS dS d S )Nr )r8 r[ )r r
r
r r[ s zMessage.payloadc C s | j r| j d jS dS d S )Nr r
)r8 r\ )r r
r
r r\ s zMessage.options)�wantedrD c C s6 |r| j tjjO _ n| jr2| j tjj M _ dS )a! Enable or disable 'DNSSEC desired' flag in requests.
*wanted*, a ``bool``. If ``True``, then DNSSEC data is
desired in the response, EDNS is enabled if required, and then
the DO bit is set. If ``False``, the DO bit is cleared if
EDNS is enabled.
N)rZ r5 r6 ZDOr8 )r r� r
r
r � want_dnssec s zMessage.want_dnssecc C s t j�t| j�t| j��S )zBReturn the rcode.
Returns a ``dns.rcode.Rcode``.
)r5 rX ro r0 r6 rZ )r r
r
r rX # s z
Message.rcode)rX rD c C sL t j�|�\}}| jdM _| j|O _| jdM _| j|O _dS )zUSet the rcode.
*rcode*, a ``dns.rcode.Rcode``, is the rcode to set.
i� i��� N)r5 rX �to_flagsr6 rZ )r rX �valueZevaluer
r
r � set_rcode* s
zMessage.set_rcodec C s t j�t| j��S )zEReturn the opcode.
Returns a ``dns.opcode.Opcode``.
)r5 rW ro r0 r6 )r r
r
r rW 5 s zMessage.opcode)rW rD c C s( | j dM _ | j tj�|�O _ dS )zZSet the opcode.
*opcode*, a ``dns.opcode.Opcode``, is the opcode to set.
i�� N)r6 r5 rW r� )r rW r
r
r �
set_opcode< s zMessage.set_opcodec C s |S )Nr
)r r� r
r
r �_get_one_rr_per_rrsetD s z Message._get_one_rr_per_rrsetc C s ||d dfS )NFr
)r rj rc rv rw r
r
r �_parse_rr_headerJ s zMessage._parse_rr_headerc C sl |t jjkr.|tjks(| js(|t jjkr`t�n2|t jj kr`|tjks\|t jj
ks\||d kr`t �||d dfS )Nr" F) r5 r� r� r! r* r8 rc r� r r� r� r )r rj �count�positionrc rv rw r
r
r �_parse_special_rr_headerO s
z Message._parse_special_rr_header)N)NT)Nr FN)T)Kr r
r r r! r] r r0 r �propertyr r5 re r~ rE �setterrG rH rI rL rN rc �Namer� r r� r rM rl rm rq rs r` r� �NONE� SectionTyper� �
RdataClass� RdataTyper� r� r� r� r� r� � staticmethodr� r< Zdefault_algorithmr r� r� r� r� r� r� �DEFAULT_EDNS_PAYLOADr� rY �Optionr� rZ r[ r r\ r� rX ZRcoder� rW ZOpcoder� r� r� r� r
r
r
r r2 � s�
!&,B,+
H 6A,+ r2 c @ s8 e Zd ZdZejjeejj e
e ejj d�dd�Z dS )�ChainingResulta� The result of a call to dns.message.QueryMessage.resolve_chaining().
The ``answer`` attribute is the answer RRSet, or ``None`` if it doesn't
exist.
The ``canonical_name`` attribute is the canonical name after all
chaining has been applied (this is the same name as ``rrset.name`` in cases
where rrset is not ``None``).
The ``minimum_ttl`` attribute is the minimum TTL, i.e. the TTL to
use if caching the data. It is the smallest of all the CNAME TTLs
and either the answer TTL if it exists or the SOA TTL and SOA
minimum values for negative answers.
The ``cnames`` attribute is a list of all the CNAME RRSets followed to
get to the canonical name.
)�canonical_namerG � minimum_ttl�cnamesc C s || _ || _|| _|| _d S )N)r� rG r� r� )r r� rG r� r� r
r
r r t s zChainingResult.__init__N)
r r
r r r5 rc r� r re r~ r0 r r r
r
r
r r� a s
r� c @ s, e Zd Zed�dd�Zejjd�dd�ZdS )� QueryMessage)rD c C s� | j tj j@ dkrt�t| j�dkr,tjj�| jd }|j}tj j
}d}d}g }x�|t k �r y(| � | j
||j|j�}t||j �}P W qR tk
�r |jtjjk�ryR| � | j
||jtjj�}|�|� t||j �}x|D ] }|j}P q�W |d7 }wRW n tk
�r P Y nX nP Y qRX qRW |t k�r.t�| �� tjjk�rN|dk �rNt�|dk�r�|} xzy2| � | j| |jtjj�}
t||
j |
d j �}P W n@ tk
�r� y | � � } W n tjj k
�r� P Y nX Y nX �q^W t ||||�S )a� Follow the CNAME chain in the response to determine the answer
RRset.
Raises ``dns.message.NotQueryResponse`` if the message is not
a response.
Raises ``dns.message.ChainTooLong`` if the CNAME chain is too long.
Raises ``dns.message.AnswerForNXDOMAIN`` if the rcode is NXDOMAIN
but an answer was found.
Raises ``dns.exception.FormError`` if the question count is not 1.
Returns a ChainingResult object.
r r" N) r6 r5 rn r rp rE r- � FormErrorrc r� ZMAX_TTL� MAX_CHAINr� rG rv rw �minr} r� ZCNAMEr �targetr rX ZNXDOMAINr rH �SOAZminimum�parentZNoParentr� ) r rE �qnameZmin_ttlrG r� r� Zcrrset�rdZaunameZsrrsetr
r
r �resolve_chaining� sb
z QueryMessage.resolve_chainingc C s
| � � jS )a� Return the canonical name of the first name in the question
section.
Raises ``dns.message.NotQueryResponse`` if the message is not
a response.
Raises ``dns.message.ChainTooLong`` if the CNAME chain is too long.
Raises ``dns.message.AnswerForNXDOMAIN`` if the rcode is NXDOMAIN
but an answer was found.
Raises ``dns.exception.FormError`` if the question count is not 1.
)r� r� )r r
r
r r� � s zQueryMessage.canonical_nameN) r r
r r� r� r5 rc r� r� r
r
r
r r� � s Kr� c C s dd l } d S )Nr )Z
dns.update)r5 r
r
r �_maybe_import_update� s r� c C s2 | t jjkrtS | t jjkr*t� t jjS tS d S )N) r5 rW �QUERYr� ZUPDATEr� �updateZ
UpdateMessager2 )rW r
r
r � _message_factory_from_opcode� s r� c @ s: e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� ZdS )� _WireReaderai Wire format reader.
parser: the binary parser
message: The message object being built
initialize_message: Callback to set message parsing options
question_only: Are we only reading the question?
one_rr_per_rrset: Put each RR into its own RRset?
keyring: TSIG keyring
ignore_trailing: Ignore trailing junk at end of request?
multi: Is this message part of a multi-message sequence?
DNS dynamic updates.
continue_on_error: try to extract as much information as possible from
the message, accumulating MessageErrors in the *errors* attribute instead of
raising them.
FNc C sH t j�|�| _d | _|| _|| _|| _|| _|| _ || _
|| _ g | _ d S )N)
r5 r� �Parser�parserr �initialize_message�
question_only�one_rr_per_rrset�ignore_trailingr; r� �continue_on_errorrB ) r r� r� r� r� r� r; r� r� r
r
r r s z_WireReader.__init__c C s� | j dk st�| j j| }xbt|�D ]V}| j�| j j�}| j�d�\}}| j �||||�\}}}}| j j ||||ddd� q$W dS )zhRead the next *qcount* records from the wire data and add them to
the question section.
Nz!HHT)rz r{ )
r �AssertionErrorr7 �ranger� �get_namer? �
get_structr� r� )r rs �qcountrj �_r� rw rv r
r
r �
_get_question s z_WireReader._get_questionc C s | j �t|| jj�� d S )N)rB r r, r� �current)r �er
r
r �
_add_error) s z_WireReader._add_errorc C s� | j dk st�| j j| }| j}�x�t|�D �]�}| jj}| j�� }| j jdk r`|� | j j�}n|}| j�
d�\} }
} } | t j j
t j jfkr�| j �|||||
| �\}
} }
}n| j �|||
| �\}
} }
}| jj}�y�|r�| dkr�t jj�d}t j j}n8| j�| �� t j�|
| | j| j j�}W dQ R X |�� }| j j�rH| t j jk�rHd}| t j j
k�rnt j�|| |�| j _ �n&| t j jk�rV| j dk�r�t d��t | j t ��r�| j �!|�}t |t"��r�t j#�$|||j%�}n"t&| j ��r�| � | j |�}n| j }|dk�rt d| ��|| j _ t j#�'| jj(|||t)t*�*� �| j j+|| j j,| j-� | j _,t j�|d|�| j _#n>| j �.|||
| ||
d|�}|dk �r�| dk�r�d} |�/|| � W q, t0k
�r� } z*| j1�r�| �2|� | j�3|| � n� W dd}~X Y q,X q,W dS )z�Read the next I{count} records from the wire data and add them to
the specified section.
section_number: the section of the message to which to add records
count: the number of records to read
Nz!HHIHr Tz"got signed message without keyringzkey '%s' unknowni���)4r r� r7 r� r� r� r� r� r? rO r� r5 r� r� r� r� r� r- r� r� Z restrict_to�rdataZfrom_wire_parserrx r> r� re r� r8 r; r rf �dictr| r� r< r� r� r� �validater� r0 rC r= r@ r� r� r� r/ r� r� �seek)r rs r� rj r{ ri Zrr_startZ
absolute_namerc rw rv r� Zrdlenry �emptyZ rdata_startr� rx r� re r� r
r
r � _get_section, s�
z_WireReader._get_sectionc
C s< | j �� dk rt�| j �d�\}}}}}}ttj�|��}||d�| _tj �
|�| j_ | � | j� | j� | j
�| _
y�| �tj|� | jr�| jS | �tj|� | �tj|� | �tj|� | js�| j �� dkr�t�| jr�| jjr�| jjs�| jj�| j j � W n: t k
�r4 } z| j �r"| � |� n� W dd}~X Y nX | jS )zNRead a wire format DNS message and build a dns.message.Message
object.� z!HHHHHH)r3 r N) r� � remainingr r� r� r5 rW ro r r6 �Flagr� r� r� r� r! r' r� r� r( r) r* r� r r� r@ r� r� r� r/ r� r� ) r r3 r6 r� ZancountZaucountZadcount�factoryr� r
r
r �read� s4
z_WireReader.read)FFFNFF) r r
r r r r� r� r� r� r
r
r
r r� � s
`r� r4 F)
r� r; r= r> r? r@ r� r� r� r� �raise_on_truncationr� rD c s� �dkr d�� ���fdd�} t | | ||| ||| �}
y |
�� }W nB tjjk
r� |
jr||
jjtjj@ r||
r|t|
jd��n� Y nX |jtjj@ r�|
r�t|d��| r�|
j |_ |S )a� Convert a DNS wire format message into a message object.
*keyring*, a ``dns.tsig.Key`` or ``dict``, the key or keyring to use if the message
is signed.
*request_mac*, a ``bytes`` or ``None``. If the message is a response to a
TSIG-signed request, *request_mac* should be set to the MAC of that request.
*xfr*, a ``bool``, should be set to ``True`` if this message is part of a zone
transfer.
*origin*, a ``dns.name.Name`` or ``None``. If the message is part of a zone
transfer, *origin* should be the origin name of the zone. If not ``None``, names
will be relativized to the origin.
*tsig_ctx*, a ``dns.tsig.HMACTSig`` or ``dns.tsig.GSSTSig`` object, the ongoing TSIG
context, used when validating zone transfers.
*multi*, a ``bool``, should be set to ``True`` if this message is part of a multiple
message sequence.
*question_only*, a ``bool``. If ``True``, read only up to the end of the question
section.
*one_rr_per_rrset*, a ``bool``. If ``True``, put each RR into its own RRset.
*ignore_trailing*, a ``bool``. If ``True``, ignore trailing junk at end of the
message.
*raise_on_truncation*, a ``bool``. If ``True``, raise an exception if the TC bit is
set.
*continue_on_error*, a ``bool``. If ``True``, try to continue parsing even if
errors occur. Erroneous rdata will be ignored. Errors will be accumulated as a
list of MessageError objects in the message's ``errors`` attribute. This option is
recommended only for DNS analysis tools, or for use in a server as part of an error
handling path. The default is ``False``.
Raises ``dns.message.ShortHeader`` if the message is less than 12 octets long.
Raises ``dns.message.TrailingJunk`` if there were octets in the message past the end
of the proper DNS message, and *ignore_trailing* is ``False``.
Raises ``dns.message.BadEDNS`` if an OPT record was in the wrong section, or
occurred more than once.
Raises ``dns.message.BadTSIG`` if a TSIG record was not the last record of the
additional data section.
Raises ``dns.message.Truncated`` if the TC flag is set and *raise_on_truncation* is
``True``.
Returns a ``dns.message.Message``.
Nr4 c s �| _ �| _� | _�| _d S )N)r= r> r? r@ )r )r? r= r@ r> r
r r� � s z%from_wire.<locals>.initialize_message)r )
r� r� r5 r- r� r r6 ZTCr rB )r� r; r= r> r? r@ r� r� r� r� r� r� r� �reader�mr
)r? r= r@ r> r � from_wire� s2 F
r� c @ sB e Zd ZdZddd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dS )� _TextReaderat Text format reader.
tok: the tokenizer.
message: The message object being built.
DNS dynamic updates.
last_name: The most recently read name when building a message object.
one_rr_per_rrset: Put each RR into its own RRset?
origin: The origin for relative names
relativize: relativize names?
relativize_to: the origin to relativize to.
FNTc C sh d | _ tjj||d�| _d | _|| _|| _|| _|| _ d | _
d| _ d| _ t
| _d | _tjj| _d| _d S )N)�
idna_codecrS r )r r5 � tokenizer� Tokenizer�tok� last_namer� r? rO �
relativize_tor3 rY rZ r� r[ rX rW r� r6 )r �textr� r� r? rO r r
r
r r + s
z_TextReader.__init__c C s� | j �� }|j}|dkr(| j �� | _�nR|dkrpx:| j �� }|�� sR| j �|� P | jtj� |j�B | _q2W �n
|dkr�| j �� | _
| j | j
d> B | _ n�|dkr�| j
dk r�d| _
x�| j �� }|�� s�| j �|� P | j tj� |j�B | _ q�W n�|dk�r| j �� | _
| j
dk �rzd| _
nb|dk�rR| j �� }tj� |�| _| jtj�| j�B | _n(|d k�rv| j �� }tj� |�| _nt�| j �� d
S ) z5Process one line from the text format header section.r3 r6 rY r1 Zeflagsr r[ rW rX N)r r| r� Zget_intr3 �
is_identifier�ungetr6 r5 r� rY rZ Zedns_from_textr[ Z
get_stringrW r� rX r �get_eol)r r� �token�whatr r
r
r � _header_lineC sH
z_TextReader._header_linec C s | j j| }| jjdd�}|�� s>| j�|| j j| j| j�| _ | j }|dkrPt
�| j�� }|� � sjt j
j�y,t j�|j�}| j�� }|� � s�t j
j�W n: t j
jk
r� t j
j�Y n tk
r� t jj}Y nX t j�|j�}| j �||||�\}}}}| j j||||ddd� | j�� dS )z7Process one line from the text format question section.T)� want_leadingN)rz r{ )r r7 r r| �
is_whitespace�as_namer? rO r r r r r5 r- � SyntaxErrorr� r� r� r/ �INr� r� r� r )r rs rj r rc rv rw r� r
r
r �_question_linel s4
z_TextReader._question_linec
C s | j j| }| jjdd�}|�� s>| j�|| j j| j| j�| _ | j }|dkrPt
�| j�� }|� � sjt j
j�y*t|jd�}| j�� }|� � s�t j
j�W n6 t j
jk
r� t j
j�Y n tk
r� d}Y nX y,t j�|j�}| j�� }|� � s�t j
j�W n> t j
jk
�r t j
j�Y n tk
�r6 t jj}Y nX t j�|j�}| j �||||�\}}}} | j�� }| �r�|�� �s�t j
j�| �s�|�� �r�t j
j�|�� �s�| j�|� t j�||| j| j j| j| j�}
|
�� } n d}
t jj } | j � ||||| |d| j �} |
dk �r| � |
|� dS )zfProcess one line from the text format answer, authority, or
additional data sections.
T)r
Nr ) r r7 r r| r r r? rO r r r r r5 r- r
r0 r� r/ r� r� r r� r� �
is_eol_or_eofZ
UnexpectedEndr r� rx r� r� r� r� )
r rs rj r rc r� rv rw ry r� r� rx re r
r
r �_rr_line� sx
z_TextReader._rr_linec C s` t | j�}|| jd�}| j|_| jdkr<|�| j| j| j� | jrN|� | j� | j
r\| j
|_
|S )N)r3 r ) r� rW r3 r6 rY r� rZ r[ rX r� r? )r r� r r
r
r �
_make_message� s
z_TextReader._make_messagec C s� | j }d}x�| j�dd�}|�� r$P |�� r�|j�� }|dkrD| j }| jrR| j}n| �� }yB|j �
|�}| js�|| _|� | j �| _ |t
jkr�| j}n| j}W n tk
r� Y nX | j�� q | j�|� ||� q W | js�| �� | _| jS )zNRead a text format DNS message and build a dns.message.Message
object.NTZHEADER)r r r| r Z
is_commentr� �upperr r r] r� r� r� r! r' r r r/ r r )r Z line_methodrs r �ur r
r
r r� � s>
z_TextReader.read)FNTN)
r r
r r r r r r r r� r
r
r
r r� s
)#K r� T)r r� r� r? rO r rD c C s t | |||||�}|�� S )a Convert the text format message into a message object.
The reader stops after reading the first blank line in the input to
facilitate reading multiple messages from a single file with
``dns.message.from_file()``.
*text*, a ``str``, the text format message.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
*one_rr_per_rrset*, a ``bool``. If ``True``, then each RR is put
into its own rrset. The default is ``False``.
*origin*, a ``dns.name.Name`` (or ``None``), the
origin to use for relative names.
*relativize*, a ``bool``. If true, name will be relativized.
*relativize_to*, a ``dns.name.Name`` (or ``None``), the origin to use
when relativizing names. If not set, the *origin* value will be used.
Raises ``dns.message.UnknownHeaderField`` if a header is unknown.
Raises ``dns.exception.SyntaxError`` if the text is badly formed.
Returns a ``dns.message.Message object``
)r� r� )r r� r� r? rO r r� r
r
r r� s *r� )r� r� r� rD c C sB t | t�rt| �}n
t�| �}|�} t| ||�S Q R X ds>t�dS )a� Read the next text format message from the specified file.
Message blocks are separated by a single blank line.
*f*, a ``file`` or ``str``. If *f* is text, it is treated as the
pathname of a file to open.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
*one_rr_per_rrset*, a ``bool``. If ``True``, then each RR is put
into its own rrset. The default is ``False``.
Raises ``dns.message.UnknownHeaderField`` if a header is unknown.
Raises ``dns.exception.SyntaxError`` if the text is badly formed.
Returns a ``dns.message.Message object``
NF)rf r� �open�
contextlib� nullcontextr� r� )r� r� r� �cmr
r
r � from_fileD s
r )r� rw rv r� r� rZ r[ r9 r\ r� r3 r6 r: rD c
C s� t | t�rtjj| | d�} tjj�|�}
tjj �|�}t
|
d�}tj � | �|_ |j
|j| ||
ddd� i }|dk rz||d< |dk r�||d< |dk r�||d< |dk r�||d < |r�|dkr�d
}||d < | |d < |jf |� |�|� |S )
a� Make a query message.
The query name, type, and class may all be specified either
as objects of the appropriate type, or as strings.
The query will have a randomly chosen query id, and its DNS flags
will be set to dns.flags.RD.
qname, a ``dns.name.Name`` or ``str``, the query name.
*rdtype*, an ``int`` or ``str``, the desired rdata type.
*rdclass*, an ``int`` or ``str``, the desired rdata class; the default
is class IN.
*use_edns*, an ``int``, ``bool`` or ``None``. The EDNS level to use; the
default is ``None``. If ``None``, EDNS will be enabled only if other
parameters (*ednsflags*, *payload*, *request_payload*, or *options*) are
set.
See the description of dns.message.Message.use_edns() for the possible
values for use_edns and their meanings.
*want_dnssec*, a ``bool``. If ``True``, DNSSEC data is desired.
*ednsflags*, an ``int``, the EDNS flag values.
*payload*, an ``int``, is the EDNS sender's payload field, which is the
maximum size of UDP datagram the sender can handle. I.e. how big
a response to this message can be.
*request_payload*, an ``int``, is the EDNS payload size to use when
sending this message. If not specified, defaults to the value of
*payload*.
*options*, a list of ``dns.edns.Option`` objects or ``None``, the EDNS
options.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
*id*, an ``int`` or ``None``, the desired query id. The default is
``None``, which generates a random query id.
*flags*, an ``int``, the desired query flags. The default is
``dns.flags.RD``.
*pad*, a non-negative ``int``. If 0, the default, do not pad; otherwise add
padding bytes to make the message size a multiple of *pad*. Note that if
padding is non-zero, an EDNS PADDING option will always be added to the
message.
Returns a ``dns.message.QueryMessage``
)r� )r3 T)rz r{ NrZ r[ r9 r\ r rY r: )rf r� r5 rc r� r� r� ru r� r� r� r6 r� r� rE r� r� )r� rw rv r� r� rZ r[ r9 r\ r� r3 r6 r: Z
the_rdtypeZ the_rdclassr� r r
r
r �
make_queryg s0 F
r � �, )�query�recursion_available� our_payloadr� r� rD c C s� | j tj j@ rtj�d��t| �� �}|| jd�}tj j| j tj j@ B |_ |r`| j tj j O _ |�
| �� � t | j �|_ | j
dkr�|�dd|| j� | jr�|�| j| j|d|d| j� | j|_|S )a` Make a message which is a response for the specified query.
The message returned is really a response skeleton; it has all
of the infrastructure required of a response, but none of the
content.
The response's question section is a shallow copy of the query's
question section, so the query's question RRsets should not be
changed.
*query*, a ``dns.message.Message``, the query to respond to.
*recursion_available*, a ``bool``, should RA be set in the response?
*our_payload*, an ``int``, the payload size to advertise in EDNS
responses.
*fudge*, an ``int``, the TSIG time fudge.
*tsig_error*, an ``int``, the TSIG error.
Returns a ``dns.message.Message`` object whose specific class is
appropriate for the query. For example, if query is a
``dns.update.UpdateMessage``, response will be too.
z&specified query message is not a query)r3 r Nr4 )r6 r5 rn r- r� r� rW r3 �RDZRAr� �listrE rY r� r[ r� r� r; r� r� r� r= )r r r r� r� r� �responser
r
r �
make_response� s,
r# ) Nr4 FNNFFFFFF)NFNTN)NF)Fr r r )[r �typingr r r r r r r rT rC Zdns.wirer5 Zdns.ednsZdns.enumZ
dns.exceptionZ dns.flagsZdns.nameZ
dns.opcodeZ dns.entropyZ dns.rcodeZ dns.rdataZdns.rdataclassZ
dns.rdatatypeZ dns.rrsetZ dns.rendererZdns.ttlZdns.tsigZdns.rdtypes.ANY.OPTZdns.rdtypes.ANY.TSIGr- r� r r Z DNSExceptionr r r r r r r r r
r �enum�IntEnumr! r, r� r� r0 rc r� r� r� r� r� Z IndexKeyTypere r~ Z IndexTyper� r2 r� r� r� r� r� r� r� r< ZHMACTSigZGSSTSigr� r� r� Z IDNACodecr� r r r6 r rY r� r r# r' r( r) r* r
r
r
r �<module> s�
Y ]
A >a z &+ "rX 5