File: //home/arjun/projects/env/lib64/python3.10/site-packages/dns/__pycache__/message.cpython-310.pyc
o
!we�� � @ sj d Z ddlZddlZddlZddlmZmZmZmZm Z m
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 Z2e e3e j4j5e j6j7e j8j9ee j8j9 ee j6j7 f Z:ee:e j;j<f Z=e
e3e>ee j;j< f Z?G d d!� d!�Z@G d"d#� d#�ZAG d$d%� d%e@�ZBd&d'� ZCd(d)� ZDG d*d+� d+�ZE , - - - - - - -d`d.eFd/ee d0eeF d1eGd2ee j4j5 d3ee
e jHjIe jHjJf d4eGd5eGd6eGd7eGd8eGd9eGd:e@fd;d<�ZKG d=d>� d>�ZL - ? dad@e>dAee j4jM d6eGd2ee j4j5 dBeGdCee j4j5 d:e@fdDdE�ZN -dbdFedAee j4jM d6eGd:e@fdGdH�ZOe j6jPdd-dddddde jQjRdf dIe
e j4j5e>f dJe
e j8j9e>f dKe
e j6j7e>f dLee
e3eGf dMeGdNee3 dOee3 dPee3 dQeee jSjT dAee j4jM dRee3 dSe3dTe3d:eBf dUdV�ZU - W X dcdYe@dZeGd[e3d\e3d]e3d:e@f d^d_�ZVe/jWZWe/jXZXe/jYZYe/jZZZdS )dz DNS Messages� N)�Any�Dict�List�Optional�Tuple�Unionc @ � e Zd ZdZdS )� ShortHeaderz2The DNS packet passed to from_wire() is too short.N��__name__�
__module__� __qualname__�__doc__� r r �D/home/arjun/projects/env/lib/python3.10/site-packages/dns/message.pyr - � r c @ r )� TrailingJunkzEThe DNS packet passed to from_wire() has extra junk at the end of it.Nr
r r r r r 1 r r c @ r )�UnknownHeaderFieldzVThe header field name was not recognized when converting from text
into a message.Nr
r r r r r 5 r r c @ r )�BadEDNSzLAn OPT record occurred somewhere other than
the additional data section.Nr
r r r r r : r r c @ r )�BadTSIGzWA TSIG record occurred somewhere other than the end of
the additional data section.Nr
r r r r r ? r r c @ r )�UnknownTSIGKeyz(A TSIG with an unknown key was received.Nr
r r r r r D r r c s. e Zd ZdZdhZ� fdd�Zdd� Z� ZS )� TruncatedzThe truncated flag is set.�messagec s t � j|i |�� d S �N)�super�__init__)�self�args�kwargs�� __class__r r r O s zTruncated.__init__c C �
| j d S )zbAs much of the message as could be processed.
Returns a ``dns.message.Message``.
r )r �r r r r r R s
zTruncated.message)r r r
r � supp_kwargsr r �
__classcell__r r r r r H s
r c @ r )�NotQueryResponsez%Message is not a response to a query.Nr
r r r r r% Z r r% c @ r )� ChainTooLongz The CNAME chain is too long.Nr
r r r r r& ^ r r& c @ r )�AnswerForNXDOMAINz.The rcode is NXDOMAIN but an answer was found.Nr
r r r r r' b r r' c @ r )�NoPreviousNamezNo previous name was known.Nr
r r r r r( f r r( c @ s, e Zd ZdZdZdZdZdZedd� �Z dS ) �MessageSectionzMessage sectionsr � � � c C s dS �Nr, r )�clsr r r �_maximumr � zMessageSection._maximumN)
r r r
r �QUESTION�ANSWER� AUTHORITY�
ADDITIONAL� classmethodr/ r r r r r) j s r) c @ s e Zd Zdedefdd�ZdS )� MessageError� exception�offsetc C s || _ || _d S r )r7 r8 )r r7 r8 r r r r x s
zMessageError.__init__N)r r r
� Exception�intr r r r r r6 w s r6 i� � c @ s� e Zd ZdZeZd~dee fdd�Ze de
e j j
fdd��Zejd d� �Ze de
e j j
fd
d ��Zejd d � �Ze de
e j j
fd
d��Zejdd� �Ze de
e j j
fdd��Zejdd� �Zdd� Zdd� Z ddee jj dedeeef defdd �Zd d � Z d d � Z d!d defd"d#�Z d$e
e j j
defd%d&�Z d'ede
e j j
fd(d)�Z e j!j"dd*d*dfd$e#d+e jjd,e j$j%d-e j!j&d.e j!j&d/ee j$j% d0ed1ed2ee jj' de j j
fd3d4�Z(e j!j"dd*d*dfd$e#d+e jjd,e j$j%d-e j!j&d.e j!j&d/ee j$j% d0ed1ed2ee jj' dee j j
fd5d6�Z)defd7d8�Z*defd9d:�Z+ ; * d�dee jj d<ed=ed>ee deeef de,f d?d@�Z-e.dAdB� �Z/ddCdd;dDe j0j1fdEedFee2e jjef dGedHee dIedJe,dKe2e jjef ddfdLdM�Z3e dee jj fdNdO��Z4e dee jj fdPdQ��Z5e dee, fdRdS��Z6e dee fdTdU��Z7e defdVdW��Z8e.d;e9dfdXdY��Z:d;d;e9ddd;fdZee2eef d[ed\ed]ee d^ee
e j;j< d_eddfd`da�Z=e defdbdc��Z;e defddde��Z>e>jdfde� �Z>e defdgdh��Z?e de@fdidj��ZAd�dkeddfdldm�ZBde jCjDfdndo�ZCdpe jCjDddfdqdr�ZEde jFjGfdsdt�ZFdue jFjGddfdvdw�ZHdxdy� ZIdzd{� ZJd|d}� ZKdS )��MessagezA DNS message.N�idc C s| |d u r t j�� | _n|| _d| _g g g g g| _d | _d| _d| _d | _ d | _
d| _ d| _ d | _
d | _i | _g | _d| _d S )Nr � Fg )�dns�entropy� random_16r= �flags�sections�opt�request_payload�pad�keyring�tsig� request_mac�xfr�origin�tsig_ctx�index�errors�time)r r= r r r r � s"
zMessage.__init__�returnc C r! )zThe question section.r �rC r" r r r �question� �
zMessage.questionc C � || j d< d S �Nr rQ �r �vr r r rR � � c C r! )zThe answer section.r* rQ r" r r r �answer� rS zMessage.answerc C rT )Nr* rQ rV r r r rY � rX c C r! )zThe authority section.r+ rQ r" r r r � authority� rS zMessage.authorityc C rT )Nr+ rQ rV r r r rZ � rX c C r! )z The additional data section.r, rQ r" r r r �
additional� rS zMessage.additionalc C rT r- rQ rV r r r r[ � rX c C s dt | j� d S )Nz<DNS message, ID �>)�reprr= r" r r r �__repr__� s zMessage.__repr__c C s | � � S r )�to_textr" r r r �__str__� s zMessage.__str__TrK �
relativize�kwc K s: t �� }|�d| j � |�dtj�| �� � � |�dtj�| �� � � |�dtj�| j� � | j dkrX|�d| j � | j
dkrP|�dtj� | j
� � |�d| j � | j
D ] }|�d |�� � q[| jj�� D ]'\}}|�d
|� d �� | �|�D ]}|�|j||fi |��� |�d � qqm|�� 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�writer= r? �opcoder_ �rcoderB �edns� ednsflags� edns_to_text�payload�options�
_section_enum� __members__�items�section_from_number�getvalue) r rK ra rb �srD �name�which�rrsetr r r r_ � s&
�zMessage.to_textc C s� t |t�sdS | j|jkrdS | j|jkrdS t| j�D ]#\}}|j| }|D ]
}||vr1 dS q'|D ]
}||vr> dS q4q dS )z�Two messages are equal if they have the same content in the
header, question, answer, and authority sections.
Returns a ``bool``.
FT)�
isinstancer< r= rB � enumeraterC )r �other�i�section�
other_section�nr r r �__eq__� s$
���zMessage.__eq__c C s | � |� S r )r� )r r{ r r r �__ne__ � zMessage.__ne__r{ c C s� |j tj j@ dks | j|jks tj�| j �tj�|j �kr dS |�� tjjtjjtjj tjj
hv r:t |j �dkr:dS tj�
| j �rCdS | j D ]
}||j vrP dS qF|j D ]
}|| j vr^ dS qTdS )zuIs *other*, also a ``dns.message.Message``, a response to this
message?
Returns a ``bool``.
r FT)rB r? �QRr= ri �
from_flagsrj �FORMERR�SERVFAIL�NOTIMP�REFUSED�lenrR � is_update)r r{ r r r r � is_response s. �
�
�zMessage.is_responser} c C s2 t | j�D ]\}}||u r| �|� S qtd��)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 section)rz rC rp �
ValueError)r r} r| � our_sectionr r r �section_number5 s
�zMessage.section_number�numberc 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``.
)rp �makerC )r r� r} r r r rs E s
zMessage.section_from_numberFrv �rdclass�rdtype�covers�deleting�create� force_unique�
idna_codecc
C s8 t |t�r
|}
| �|
�}nt |t�r t�|�}
| �|
�}n| �|�}
t |t�r/tjj|| d�}tj j
� |�}tj j
� |�}tj j
� |�}|durOtj j
� |�}|
|||||f} |s}| jdurk| j�| �} | durj| S n|D ]} | �|||||�r|| S qm|s�t�tj�|||||�} |�| � | jdur�| | j| < | S )a� Find the RRset with the given attributes in the specified section.
*section*, an ``int`` section number, a ``str`` section name, 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)
my_message.find_rrset("ANSWER", name, rdclass, rdtype)
*name*, a ``dns.name.Name`` or ``str``, the name of the RRset.
*rdclass*, an ``int`` or ``str``, the class of the RRset.
*rdtype*, an ``int`` or ``str``, the type of the RRset.
*covers*, an ``int`` or ``str``, the covers value of the RRset.
The default is ``dns.rdatatype.NONE``.
*deleting*, an ``int``, ``str``, 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.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
Raises ``KeyError`` if the RRset was not found and create was
``False``.
Returns a ``dns.rrset.RRset object``.
�r� N)ry r: rs �strr) � from_textr� r? rv � rdatatype� RdataTyper� �
rdataclass�
RdataClassrM �get�
full_match�KeyErrorrx �RRset�append)
r r} rv r� r� r� r� r� r� r� r� �keyrx r r r �
find_rrsetT s@
4
��
zMessage.find_rrsetc
C s: z| � ||||||||| � }
W |
S ty d}
Y |
S w )aj 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, a ``str`` section name, 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)
my_message.get_rrset("ANSWER", name, rdclass, rdtype)
*name*, a ``dns.name.Name`` or ``str``, the name of the RRset.
*rdclass*, an ``int`` or ``str``, the class of the RRset.
*rdtype*, an ``int`` or ``str``, the type of the RRset.
*covers*, an ``int`` or ``str``, the covers value of the RRset.
The default is ``dns.rdatatype.NONE``.
*deleting*, an ``int``, ``str``, 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.
*idna_codec*, a ``dns.name.IDNACodec``, specifies the IDNA
encoder/decoder. If ``None``, the default IDNA 2003 encoder/decoder
is used.
Returns a ``dns.rrset.RRset object`` or ``None``.
N)r� r� ) r r} rv r� r� r� r� r� r� r� rx r r r � get_rrset� s" 3�
��zMessage.get_rrsetc C sJ | j sdS d}| j d jD ]}|�� }|t|�d 7 }q
| jr#|d7 }|S )z:Compute the size required for the OPT RR, padding excludedr � � )rD ro �to_wirer� rF )r �size�option�wirer r r �_compute_opt_reserve� s z Message._compute_opt_reservec C s* | j sdS t�� }| j �|� t|�� �S )z)Compute the size required for the TSIG RRr )rH rf �BytesIOr� r� rt )r �fr r r �_compute_tsig_reserve� s
z Message._compute_tsig_reserver �max_size�multirL c K s� |du r | j dur | j }|dkr| jdkr| j}nd}|dk r"d}n|dkr(d}tj�| j| j||�}| �� }|�|� | � � }|�|� | j
D ] } |� | j | j
| j� qH| jD ]} |jtjj| fi |�� qX| jD ]} |jtjj| fi |�� qj| jD ]} |jtjj| fi |�� q||�� | jdur�|�| j| j||� |�� | jdur�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 )%rK rE r? �renderer�Rendererr= rB r� �reserver� rR � add_questionrv r� r� rY � add_rrsetr2 rZ r3 r[ r4 �release_reservedrD �add_optrF � write_headerrH �sign�get_wirerG r: rO rI �clear�addrL ) r rK r� r� rL rb �r� opt_reserve� tsig_reserverx �new_tsig�ctxr r r r� sX !
�
zMessage.to_wirec C s8 t jjj�t jjt jj|||||||� }t j�| d|�S rU )r? �rdtypes�ANY�TSIGr� r� rx �
from_rdata) �keyname� algorithm� time_signed�fudge�mac� original_id�errorr{ rH r r r �
_make_tsig[ s
� zMessage._make_tsig�, r> rG r� r� r� �
tsig_error�
other_datar� c
C s� t |tjj�r
|}|j}n0t|�r|| |�}n&t |t�r"tj�|�}|du r,tt |��}|| }t |t
�r=tj�|||�}|| _ |du rG| 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 � )ry r? rH �Keyrv �callabler� r� �next�iter�bytesrG r= r� r� � mac_sizes) r rG r� r� r� r� r� r� r� r r r �use_tsigl s2 /
�zMessage.use_tsigc C s | j r| j jS d S r )rH rv r" r r r r� � � zMessage.keynamec C � | j r | j d jS d S rU )rH r� r" r r r � keyalgorithm� � zMessage.keyalgorithmc C r� rU )rH r� r" r r r r� � r� z Message.macc C r� rU )rH r� r" r r r r� � r� zMessage.tsig_errorc C s
t | j�S r )�boolrH r" r r r �had_tsig� s
zMessage.had_tsigc C s4 t jjj�|t jj|p d�}t j�t jjt | �|�S )Nr )
r? r� r� �OPTr� rx r� rv �rootr: )rB rn ro rD r r r � _make_opt� s zMessage._make_optrk rl rn rE ro rF c C s� |du s|du r d}n|du rd}|dk r d| _ d| _dS |dM }||d> O }|du r-g }| �|||�| _ |du r;|}|| _|| _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.
NFre Tr l �~ r; )rD rE r� rF )r rk rl rn rE ro rF r r r �use_edns� s
zMessage.use_ednsc C s | j r
| jd@ d? S dS )Ni � r; re )rD rl r" r r r rk s z Message.ednsc C s | j r| j jS dS rU )rD �ttlr" r r r rl r� zMessage.ednsflagsc C s* | j r || j _d S |r| �|�| _ d S d S r )rD r� r� rV r r r rl s
�c C s | j r | j d jS dS rU )rD rn r" r r r rn # r� zMessage.payloadc C s | j r | j d jS dS )Nr r )rD ro r" r r r ro * r� zMessage.options�wantedc C s< |r
| j tjjO _ dS | jr | j tjj M _ dS 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)rl r? rB �DOrD )r r� r r r � want_dnssec1 s
�zMessage.want_dnssecc C s t j�t| j�t| j��S )zBReturn the rcode.
Returns a ``dns.rcode.Rcode``.
)r? rj r� r: rB rl r" r r r rj ? s z
Message.rcoderj 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)r? rj �to_flagsrB rl )r rj �value�evaluer r r � set_rcodeF s
zMessage.set_rcodec C s t j�t| j��S )zEReturn the opcode.
Returns a ``dns.opcode.Opcode``.
)r? ri r� r: rB r" r r r ri Q s zMessage.opcoderi c C s( | j dM _ | j tj�|�O _ dS )zZSet the opcode.
*opcode*, a ``dns.opcode.Opcode``, is the opcode to set.
i�� N)rB r? ri r� )r ri r r r �
set_opcodeX s zMessage.set_opcodec C s |S r r )r r� r r r �_get_one_rr_per_rrset` r0 z Message._get_one_rr_per_rrsetc C s ||d dfS �NFr )r r} rv r� r� r r r �_parse_rr_headerf r� zMessage._parse_rr_headerc C sl |t jjkr|tjks| js|t jjkrt�n|t jj kr0|tjks.|t jj
ks.||d kr0t �||d dfS )Nr* F) r? r� r� r) r4 rD rv r� r r� r� r )r r} �count�positionrv r� r� r r r �_parse_special_rr_headerk s
� �
z Message._parse_special_rr_headerr )NT)Nr FN)T)Lr r r
r r) rp r r: r �propertyr r? rx r� rR �setterrY rZ r[ r^ r` rv �Namer� r r� r r_ r� r� r� r� rs r� �NONE� SectionTyper� r� r� � IDNACodecr� r� r� r� r� r� � staticmethodr� rH �default_algorithmr r� r� r� r� r� r� �DEFAULT_EDNS_PAYLOADr� rk �Optionr� rl rn r ro r� rj �Rcoder� ri �Opcoder� r� r� r� r r r r r< � sr
�
��
�
�&&������
�� �
�
�[������
�� �
�
�C�
����
�
�O
��������
�J�������
�3
r< c @ s> e Zd ZdZdejjdeejj de
de ejj fdd�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_namerY � minimum_ttl�cnamesc C s || _ || _|| _|| _d S r )r rY r r )r r rY r r r r r r � s
zChainingResult.__init__N)
r r r
r r? rv r� r rx r� r: r r r r r r r } s �
��
�r c @ s, e Zd Zdefdd�Zdejjfdd�ZdS )� QueryMessagerP c C s� | j tj j@ dkr t�t| j�dkrtjj�| jd }|j}tj j
}d}d}g }|t k r�z| � | j
||j|j�}t||j �}W nA ty� |jtjjkr~z&| � | j
||jtjj�}|�|� t||j �}|D ]}|j} |d7 }W Y q( ty} Y Y nw Y nw |t kr�t�| �� tjjkr�|dur�t�|du r�|} z| � | j| |jtjj�}
t||
j |
d j �}W n ty� z| � � } W n tjj y� Y Y nw Y nw q�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) rB r? r� r% r� rR r7 � FormErrorrv r� �MAX_TTL� MAX_CHAINr� rY r� r� �minr� r� �CNAMEr� �targetr& rj �NXDOMAINr' rZ �SOA�minimum�parent�NoParentr ) r rR �qname�min_ttlrY r� r �crrset�rd�auname�srrsetr r r �resolve_chaining� st
� �
��� ����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 r? rv r� r r r r r r � s Kr c C s dd l } d S rU )�
dns.update)r? r r r �_maybe_import_update� s r c C s. | t jjkrtS | t jjkrt� t jjS tS r ) r? ri �QUERYr �UPDATEr �update�
UpdateMessager< )ri r r r � _message_factory_from_opcode s r c @ sF e Zd ZdZ ddd�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 r )
r? r� �Parser�parserr �initialize_message�
question_only�one_rr_per_rrset�ignore_trailingrG r� �continue_on_errorrN ) r r� r! r" r# r$ rG r� r% r r r r s
z_WireReader.__init__c C s~ | j dusJ �| j j| }t|�D ]+}| j�| j j�}| j�d�\}}| j �||||�\}}}}| j j||||ddd� qdS )zhRead the next *qcount* records from the wire data and add them to
the question section.
Nz!HHT�r� r� ) r rC �ranger �get_namerK �
get_structr� r� )r r� �qcountr} �_r r� r� r r r �
_get_question5 s � ��z_WireReader._get_questionc C s | j �t|| jj�� d S r )rN r� r6 r �current)r �er r r �
_add_errorE s z_WireReader._add_errorc C s� | j dusJ �| j j| }| j}t|�D �]Z}| jj}| j�� }| j jdur.|�| j j�}n|}| j� d�\} }
} } | t
j j t
j j
fv rT| j �|||||
| �\}
} }
}n
| j �|||
| �\}
} }
}| jj}z�|rw| dkrpt
jj�d}t
j j}n&| j�| �� t
j�|
| | j| j j�}W d � n1 s�w Y |�� }| j jr�| t
j jkr�d}| t
j j kr�t
j�|| |�| j _n�| t
j j
k�r)| j du r�t d��t | j t �r�| j � |�}t |t!�r�t
j"�#|||j$�}nt%| j �r�| � | j |�}n| j }|du r�t d| ��|| j _ t
j"�&| jj'|||t(t)�)� �| j j*|| j j+| j,� | j _+t
j�|d|�| j _"n | j �-|||
| ||
d|�}|du�rH| dk�rBd} |�.|| � W q t/�yo } z| j0�rd| �1|� | j�2|| � n� W Y d}~qd}~ww 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���)3r rC r# r' r r- r( rK ra r) r? r� r� r� r� r� r7 r r� � restrict_to�rdata�from_wire_parserr� rJ r rx r� rD rG r ry �dictr� r� rH r� r� r� �validater� r: rO rI rL r� r� r� r9 r% r/ �seek)r r� r� r} r� r| �rr_start�
absolute_namerv r� r� r� �rdlenr� �empty� rdata_startr r� r� rx r. r r r � _get_sectionH s�
�� �
� �
�
� �
�
����z_WireReader._get_sectionc
C sV | j �� dk r t�| j �d�\}}}}}}ttj�|��}||d�| _tj �
|�| j_ | � | j� | j� | j
�| _
zS| �tj|� | jrH| jW S | �tj|� | �tj|� | �tj|� | jsi| j �� dkrit�| jr�| jjr�| jjs�| jj�| j j � W | jS W | jS W | jS W | jS t y� } z| j r�| � |� n� W Y d}~| jS d}~ww )zNRead a wire format DNS message and build a dns.message.Message
object.� z!HHHHHH�r= r N) r � remainingr r) r r? ri r� r rB �Flagr! r� r# r, r) r1 r" r; r2 r3 r4 r$ r r� rL r� r r� r9 r% r/ ) r r= rB r* �ancount�aucount�adcount�factoryr. r r r �read� sL � �����
���z_WireReader.read)FFFNFF) r r r
r r r, r/ r; rD r r r r r
s
� `r r> Fr� rG rI rJ rK rL r� r" r# r$ �raise_on_truncationr% rP c s� �du rd�� ���fdd�} t | | ||| ||| �}
z|
�� }W n tjjy= |
jr<|
jjtjj@ r<|
r<t|
jd��� w |jtjj@ rL|
rLt|d��| rR|
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``.
Nr> c s �| _ �| _� | _�| _d S r )rI rJ rK rL �r �rK rI rL rJ r r r! s
z%from_wire.<locals>.initialize_messagerF )
r rD r? r7 r r rB �TCr rN )r� rG rI rJ rK rL r� r" r# r$ rE r% r! �reader�mr rG r � from_wire� s< F�
��� �
rK c @ sJ e Zd ZdZ ddd�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 )Nr� re r )r r? � tokenizer� Tokenizer�tok� last_namer# rK ra �
relativize_tor= rk rl r� rn rj ri r rB )r �textr� r# rK ra rQ r r r r G s
z_TextReader.__init__c C sx | j �� }|j}|dkr| j �� | _n�|dkr5 | j �� }|�� s(| j �|� n | jtj� |j�B | _qn�|dkrI| j �� | _
| j | j
d> B | _ nl|dkrs| j
dk rUd| _
| j �� }|�� sf| j �|� n | j tj� |j�B | _ qVnB|dkr�| j �� | _
| j
dk r�d| _
n/|d kr�| j �� }tj� |�| _| jtj�| j�B | _n|d
kr�| j �� }tj� |�| _nt�| j �� d S ) z5Process one line from the text format header section.r= rB Trk r; �eflagsr rn ri rj N)rO r� r� �get_intr= �
is_identifier�ungetrB r? r� rk rl �edns_from_textrn �
get_stringri r� rj r �get_eol)r r+ �token�whatrR r r r � _header_line_ sR
�
�
�
z_TextReader._header_linec C s | j j| }| jjdd�}|�� s | j�|| j j| j| j�| _ | j }|du r(t
�| j�� }|� � s5t j
j�zt j�|j�}| j�� }|� � sJt j
j�W n t j
jyX t j
j� tyd t jj}Y nw t j�|j�}| j �||||�\}}}}| j j||||ddd� | j�� dS )z7Process one line from the text format question section.T�� want_leadingNr& )r rC rO r� �
is_whitespace�as_namerK ra rQ rP r( rU r? r7 � SyntaxErrorr� r� r� r9 �INr� r� r� rY )r r� r} rZ rv r� r� r+ r r r �_question_line� s> �
� � � �z_TextReader._question_linec
C s | j j| }| jjdd�}|�� s | j�|| j j| j| j�| _ | j }|du r(t
�| j�� }|� � s5t j
j�zt|jd�}| j�� }|� � sIt j
j�W n t j
jyW t j
j� tya d}Y nw zt j�|j�}| j�� }|� � swt j
j�W n t j
jy� t j
j� ty� t jj}Y nw t j�|j�}| j �||||�\}}}} | j�� }| r�|�� s�t j
j�| s�|�� r�t j
j�|�� s�| j�|� t j�||| j| j j| j| j�}
|
�� } nd}
t jj } | j � ||||| |d| j �} |
dur�| � |
|� dS dS )zfProcess one line from the text format answer, authority, or
additional data sections.
Tr] Nr ) r rC rO r� r_ r` rK ra rQ rP r( rU r? r7 ra r: r� r9 r� r� rb r� r� �
is_eol_or_eof�
UnexpectedEndrV r1 r� r� r� r# r� )
r r� r} rZ rv r� r� r� r� r9 r r� rx r r r �_rr_line� s� �
� �
� � �
�
�
�z_TextReader._rr_linec C s` t | j�}|| jd�}| j|_| jdkr |�| j| j| j� | jr'|� | j� | j
r.| j
|_
|S )Nr= r ) r ri r= rB rk r� rl rn rj r� rK )r rC r r r r �
_make_message� s
z_TextReader._make_messagec C s� | j }d} | j�dd�}|�� rnW|�� r^|j�� }|dkr"| j }| jr)| j}n| �� }z!|j �
|�}| jsA|| _|� | j �| _ |t
jkrJ| j}n| j}W n tyW Y nw | j�� q| j�|� ||� q| jsq| �� | _| jS )zNRead a text format DNS message and build a dns.message.Message
object.Nr* T�HEADER)r\ rO r� rd �
is_commentr� �upperr rg rp r� r� r# r) r1 rc rf r9 rY rV )r � line_methodr� rZ �ur r r r rD sF
�
� �
�#
z_TextReader.read)FNTN)
r r r
r r r\ rc rf rg rD r r r r rL 9 s
�)#K rL TrR r� ra rQ 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``
)rL rD )rR r� r# rK ra rQ rI r r r r� 0 s * �r� r� c C sR t | t�r
t| �}nt�| �}|�} t| ||�W d � S 1 s"w Y J �)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``
N)ry r� �open�
contextlib� nullcontextr� )r� r� r# �cmr r r � from_file` s
�rq r r� r� r� r� rl rn rE ro r= rB rF c
C s� t | t�r
tjj| | d�} tjj�|�}tjj �|�}t
|
d�}
tj � | �|
_ |
j
|
j| ||ddd� i }|dur=||d< |durE||d< |durM||d< |durU||d < |r]|du r]d
}||d < | |d < |
jd
i |�� |
�|� |
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� r= Tr& Nrl rn rE ro r rk rF r )ry r� r? rv r� r� r� r� r� r� r rB r? r� rR r� r� )r r� r� r� r� rl rn rE ro r� r= rB rF rJ r r r r �
make_query� s.
F
rr � r� �query�recursion_available� our_payloadr� r� c C s� | j tj j@ r
tj�d��t| �� �}|| jd�}tj j| j tj j@ B |_ |r0| j tj j O _ |�
| �� � t | j �|_ | j
dkrK|�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 queryr= r Nr> )rB r? r� r7 r r ri r= �RD�RAr� �listrR rk r� rn r� r� rG r� r� r� rI )rt ru rv r� r� rC �responser r r �
make_response� s.
� r{ ) Nr> FNNFFFFFF)NFNTNr� )Frs r� r )[r rn rf rO �typingr r r r r r �dns.ednsr? � dns.entropy�dns.enum�
dns.exception� dns.flags�dns.name�
dns.opcode� dns.rcode� dns.rdata�dns.rdataclass�
dns.rdatatype�dns.rdtypes.ANY.OPT�dns.rdtypes.ANY.TSIG� dns.renderer� dns.rrset�dns.tsig�dns.ttl�dns.wirer7 r r r � DNSExceptionr r r r r r% r& r' ra r( �enum�IntEnumr) r6 r� r r: rv r� r� r� r� r� � IndexKeyTyperx r� � IndexTyper� r� r< r r r r r r� r� rH �HMACTSig�GSSTSigrK rL r� r� rq rb rB rw rk r� rr r{ r1 r2 r3 r4 r r r r �<module> s�
�� v ]
A�����
���� �
� � �
�n z��
��
��
�
�2��
��
�&��������� �
� � �
�
�d������
�;