Your IP : 3.140.195.8
3
�@�Wy� �
@ s� d Z ddlmZ ddlmZ ddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlZddlZddlZddlZddlZddlmZmZmZ G dd� dejj�Z G d d
� d
ejj�Z G d d � d ejj �Z G d
d� dejj�Z G dd� dejj�Z!G dd� dejj �Z"G dd� de#�Z$G dd� de#�Z%d(dd�Z&G d d � d e#�Z'd d � Z(d d!� Z)ej*j+ddddddfd"d#�Z,d)d&d'�Z-dS )*z DNS Messages� )�absolute_import)�StringION� )�long�xrange� string_typesc @ s e Zd ZdZdS )� ShortHeaderz2The DNS packet passed to from_wire() is too short.N)�__name__�
__module__� __qualname__�__doc__� r
r
� /usr/lib/python3.6/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 / 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 4 s r c @ s e Zd ZdZdS )�BadEDNSzVOPT record occurred somewhere other than the start of
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 F s r c @ s� e Zd ZdZd-dd�Zdd� Zdd� Zd.d
d �Zd d
� Zdd� Z dd� Z
dd� Z e j
jdddfdd�Ze j
jdddfdd�Zd/dd�Zdd ddd e jjfd d �Zd0d!d"�Zd1d#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� ZdS )2�Messagea,
A DNS message.
@ivar id: The query id; the default is a randomly chosen id.
@type id: int
@ivar flags: The DNS flags of the message. @see: RFC 1035 for an
explanation of these flags.
@type flags: int
@ivar question: The question section.
@type question: list of dns.rrset.RRset objects
@ivar answer: The answer section.
@type answer: list of dns.rrset.RRset objects
@ivar authority: The authority section.
@type authority: list of dns.rrset.RRset objects
@ivar additional: The additional data section.
@type additional: list of dns.rrset.RRset objects
@ivar edns: The EDNS level to use. The default is -1, no Edns.
@type edns: int
@ivar ednsflags: The EDNS flags
@type ednsflags: long
@ivar payload: The EDNS payload size. The default is 0.
@type payload: int
@ivar options: The EDNS options
@type options: list of dns.edns.Option objects
@ivar request_payload: The associated request's EDNS payload size.
@type request_payload: int
@ivar keyring: The TSIG keyring to use. The default is None.
@type keyring: dict
@ivar keyname: The TSIG keyname to use. The default is None.
@type keyname: dns.name.Name object
@ivar keyalgorithm: The TSIG algorithm to use; defaults to
dns.tsig.default_algorithm. Constants for TSIG algorithms are defined
in dns.tsig, and the currently implemented algorithms are
HMAC_MD5, HMAC_SHA1, HMAC_SHA224, HMAC_SHA256, HMAC_SHA384, and
HMAC_SHA512.
@type keyalgorithm: string
@ivar request_mac: The TSIG MAC of the request message associated with
this message; used when validating TSIG signatures. @see: RFC 2845 for
more information on TSIG fields.
@type request_mac: string
@ivar fudge: TSIG time fudge; default is 300 seconds.
@type fudge: int
@ivar original_id: TSIG original id; defaults to the message's id
@type original_id: int
@ivar tsig_error: TSIG error code; default is 0.
@type tsig_error: int
@ivar other_data: TSIG other data.
@type other_data: string
@ivar mac: The TSIG MAC for this message.
@type mac: string
@ivar xfr: Is the message being used to contain the results of a DNS
zone transfer? The default is False.
@type xfr: bool
@ivar origin: The origin of the zone in messages which are used for
zone transfers or for DNS dynamic updates. The default is None.
@type origin: dns.name.Name object
@ivar tsig_ctx: The TSIG signature context associated with this
message. The default is None.
@type tsig_ctx: hmac.HMAC object
@ivar had_tsig: Did the message decoded from wire format have a TSIG
signature?
@type had_tsig: bool
@ivar multi: Is this message part of a multi-message sequence? The
default is false. This variable is used when validating TSIG signatures
on messages which are part of a zone transfer.
@type multi: bool
@ivar first: Is this message standalone, or the first of a multi
message sequence? This variable is used when validating TSIG signatures
on messages which are part of a zone transfer.
@type first: bool
@ivar index: An index of rrsets in the message. The index key is
(section, name, rdclass, rdtype, covers, deleting). Indexing can be
disabled by setting the index to None.
@type index: dict
Nc C s� |d krt jj� | _n|| _d| _g | _g | _g | _g | _d| _ d| _
d| _ g | _ d| _
d | _d | _t jj| _d| _d| _d| _d| _| j| _d| _d| _d | _d | _d| _ d| _ d| _ i | _ d S )Nr r � i, FT���) �dnsZentropyZ random_16�id�flags�question�answer� authority�
additional�edns� ednsflags�payload�options�request_payload�keyring�keyname�tsig�default_algorithm� keyalgorithm� request_mac�
other_data�
tsig_error�fudge� original_id�mac�xfr�origin�tsig_ctx�had_tsig�multi�first�index)�selfr r
r
r �__init__� s:
zMessage.__init__c C s dt | j� d S )Nz<DNS message, ID �>)�reprr )r5 r
r
r �__repr__� s zMessage.__repr__c C s | j � S )N)�to_text)r5 r
r
r �__str__� s zMessage.__str__Tc K s t � }|jd| j � |jdtjjtjj| j�� � tjj| j| j �}|jdtjj|� � |jdtjj| j� � | j
dkr�|jd| j
� | j dkr�|jdtjj | j � � |jd| j � tjj
| j�}|r�|jd � n
|jd
� x.| jD ]$}|j|j||f|�� |jd � q�W |�r,|jd � n
|jd
� x0| jD ]&}|j|j||f|�� |jd � �q>W |�rz|jd� n
|jd� x0| jD ]&}|j|j||f|�� |jd � �q�W |jd� x0| jD ]&}|j|j||f|�� |jd � �q�W |j� dd� S )z�Convert the message to text.
The I{origin}, I{relativize}, and any other keyword
arguments are passed to the rrset to_wire() method.
@rtype: string
zid %d
z
opcode %s
z rcode %s
z flags %s
r zedns %s
z
eflags %s
z payload %d
z;ZONE
z
;QUESTION
�
z;PREREQ
z;ANSWER
z;UPDATE
z ;AUTHORITY
z ;ADDITIONAL
Nr r )r �writer r �opcoder: �
from_flagsr �rcoder r Z edns_to_textr � is_updater r r r �getvalue)r5 r/ �
relativize�kw�sZrcrA �rrsetr
r
r r: � sJ
zMessage.to_textc C s� t |t�sdS | j|jkr dS | j|jkr.dS x | jD ]}||jkr6dS q6W x |jD ]}|| jkrTdS qTW x | jD ]}||jkrrdS qrW x |jD ]}|| jkr�dS q�W x | jD ]}||jkr�dS q�W x |jD ]}|| jkr�dS q�W dS )z�Two messages are equal if they have the same content in the
header, question, answer, and authority sections.
@rtype: boolFT)�
isinstancer r r r r r )r5 �other�nr
r
r �__eq__� s2
zMessage.__eq__c C s | j |� S )z0Are two messages not equal?
@rtype: bool)rJ )r5 rH r
r
r �__ne__ s zMessage.__ne__c C s� |j tj j@ dks:| j|jks:tjj| j �tjj|j �kr>dS tjj|j |j�tjjkr\dS tjj | j �rndS x | j
D ]}||j
krvdS qvW x |j
D ]}|| j
kr�dS q�W dS )z1Is other a response to self?
@rtype: boolr FT) r r �QRr r> r? r@ r ZNOERRORrA r )r5 rH rI r
r
r � is_response s"
zMessage.is_responsec C sD || j krdS || jkr dS || jkr*dS || jkr8dS td��d S )Nr r � � zunknown section)r r r r �
ValueError)r5 �sectionr
r
r �section_number- s
zMessage.section_numberFc C s� | j |�|||||f} |sb| jdk r>| jj| �}
|
dk rb|
S n$x"|D ]}
|
j|||||�rD|
S qDW |sjt�tjj|||||�}
|j|
� | jdk r�|
| j| < |
S )a Find the RRset with the given attributes in the specified section.
@param section: the section of the message to look in, e.g.
self.answer.
@type section: list of dns.rrset.RRset objects
@param name: the name of the RRset
@type name: dns.name.Name object
@param rdclass: the class of the RRset
@type rdclass: int
@param rdtype: the type of the RRset
@type rdtype: int
@param covers: the covers value of the RRset
@type covers: int
@param deleting: the deleting value of the RRset
@type deleting: int
@param create: If True, create the RRset if it is not found.
The created RRset is appended to I{section}.
@type create: bool
@param force_unique: If True and create is also True, create a
new RRset regardless of whether a matching RRset exists already.
@type force_unique: bool
@raises KeyError: the RRset was not found and create was False
@rtype: dns.rrset.RRset objectN) rR r4 �get�match�KeyErrorr rF ZRRset�append) r5 rQ �name�rdclass�rdtype�covers�deleting�create� force_unique�keyrF r
r
r �
find_rrset9 s"
zMessage.find_rrsetc
C s: y | j ||||||||�} W n tk
r4 d} Y nX | S )a Get the RRset with the given attributes in the specified section.
If the RRset is not found, None is returned.
@param section: the section of the message to look in, e.g.
self.answer.
@type section: list of dns.rrset.RRset objects
@param name: the name of the RRset
@type name: dns.name.Name object
@param rdclass: the class of the RRset
@type rdclass: int
@param rdtype: the type of the RRset
@type rdtype: int
@param covers: the covers value of the RRset
@type covers: int
@param deleting: the deleting value of the RRset
@type deleting: int
@param create: If True, create the RRset if it is not found.
The created RRset is appended to I{section}.
@type create: bool
@param force_unique: If True and create is also True, create a
new RRset regardless of whether a matching RRset exists already.
@type force_unique: bool
@rtype: dns.rrset.RRset object or NoneN)r_ rU )
r5 rQ rW rX rY rZ r[ r\ r] rF r
r
r � get_rrsetg s
zMessage.get_rrsetr c K sR |dkr | j dkr| j }nd}|dk r,d}n |dkr8d}tjj| j| j||�}x"| jD ]}|j|j|j |j
� qVW x"| j D ]}|j tjj
|f|� qzW x"| jD ]}|j tjj|f|� q�W | jdkr�|j| j| j| j| j� x"| jD ]}|j tjj|f|� q�W |j� | jdk �rJ|j| j| j| j | j| j | j | j | j | j � |j!| _!|j"� S )a7 Return a string containing the message in DNS compressed wire
format.
Additional keyword arguments are passed to the rrset to_wire()
method.
@param origin: The origin to be appended to any relative names.
@type origin: dns.name.Name object
@param max_size: The maximum size of the wire format output; default
is 0, which means 'the message's request payload, if nonzero, or
65536'.
@type max_size: int
@raises dns.exception.TooBig: max_size was exceeded
@rtype: string
r i�� i N)#r" r ZrendererZRendererr r r Z add_questionrW rY rX r Z add_rrset�ANSWERr � AUTHORITYr Zadd_ednsr r r! r �
ADDITIONALZ write_headerr$ Zadd_tsigr# r+ r, r* r) r( r' r- Zget_wire)r5 r/ Zmax_sizerD �rrF r
r
r �to_wire� s6
zMessage.to_wirei, r c C st || _ |dkr$t| j j� �d | _n t|t�r:tjj|�}|| _|| _ || _
|dkr^| j | _ n|| _ || _
|| _dS )aw When sending, a TSIG signature using the specified keyring
and keyname should be added.
@param keyring: The TSIG keyring to use; defaults to None.
@type keyring: dict
@param keyname: The name of the TSIG key to use; defaults to None.
The key must be defined in the keyring. If a keyring is specified
but a keyname is not, then 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 keyring is used, unless
they know the keyring contains only one key.
@type keyname: dns.name.Name or string
@param fudge: TSIG time fudge; default is 300 seconds.
@type fudge: int
@param original_id: TSIG original id; defaults to the message's id
@type original_id: int
@param tsig_error: TSIG error code; default is 0.
@type tsig_error: int
@param other_data: TSIG other data.
@type other_data: string
@param algorithm: The TSIG algorithm to use; defaults to
dns.tsig.default_algorithm
Nr )r# �list�keysr$ rG r r rW � from_textr' r+ r r, r* r) )r5 r# r$ r+ r, r* r) � algorithmr
r
r �use_tsig� s
zMessage.use_tsig� c C s� |dks|dkrd}|dkr d}|dkr,|}|dk rFd}d}d}g }n$|t d�M }||d> O }|dkrjg }|| _|| _|| _|| _|| _dS ) a_ Configure EDNS behavior.
@param edns: 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'.
@type edns: int or bool or None
@param ednsflags: EDNS flag values.
@type ednsflags: int
@param payload: The EDNS sender's payload field, which is the maximum
size of UDP datagram the sender can handle.
@type payload: int
@param request_payload: The EDNS payload size to use when sending
this message. If not specified, defaults to the value of payload.
@type request_payload: int or None
@param options: The EDNS options
@type options: None or list of dns.edns.Option objects
@see: RFC 2671
NFr Tr l �~ � r )r r r r r! r" )r5 r r r r" r! r
r
r �use_edns� s( zMessage.use_ednsc C sL |r*| j dk r| j� | jtjjO _n | j dkrH| jtjj M _dS )a
Enable or disable 'DNSSEC desired' flag in requests.
@param wanted: Is DNSSEC desired? If True, EDNS is enabled if
required, and then the DO bit is set. If False, the DO bit is
cleared if EDNS is enabled.
@type wanted: bool
r N)r rm r r r ZDO)r5 Zwantedr
r
r � want_dnssec s
zMessage.want_dnssecc C s t jj| j| j�S )z.Return the rcode.
@rtype: int
)r r@ r? r r )r5 r
r
r r@ s z
Message.rcodec C sj t jj|�\}}| jdM _| j|O _| jtd�M _| j|O _| jdkrf| jdk rfd| _dS )zPSet the rcode.
@param rcode: the rcode
@type rcode: int
i� i��� r N)r r@ �to_flagsr r r r )r5 r@ �valueZevaluer
r
r � set_rcode" s zMessage.set_rcodec C s t jj| j�S )z/Return the opcode.
@rtype: int
)r r> r? r )r5 r
r
r r> / s zMessage.opcodec C s( | j dM _ | j tjj|�O _ dS )zTSet the opcode.
@param opcode: the opcode
@type opcode: int
i�� N)r r r> ro )r5 r> r
r
r �
set_opcode5 s zMessage.set_opcode)N)NT)Nr )r r rk NN)T)r r
r r r6 r9 r; r: rJ rK rM rR r � rdatatype�NONEr_ r` re r% r& rj rm rn r@ rq r> rr r
r
r
r r K s0 K
7
-
!
.)
*
r c @ s2 e Zd ZdZd dd�Zdd� Zdd� Zd d
� Zd S )
� _WireReadera� Wire format reader.
@ivar wire: the wire-format message.
@type wire: string
@ivar message: The message object being built
@type message: dns.message.Message object
@ivar current: When building a message object from wire format, this
variable contains the offset from the beginning of wire of the next octet
to be read.
@type current: int
@ivar updating: Is the message a dynamic update?
@type updating: bool
@ivar one_rr_per_rrset: Put each RR into its own RRset?
@type one_rr_per_rrset: bool
@ivar ignore_trailing: Ignore trailing junk at end of request?
@type ignore_trailing: bool
@ivar zone_rdclass: The class of the zone in messages which are
DNS dynamic updates.
@type zone_rdclass: int
Fc C s@ t jj|�| _|| _d| _d| _t jj| _ || _
|| _ || _ d S )Nr F)
r ZwiredataZ
maybe_wrap�wire�message�current�updating�
rdataclass�IN� zone_rdclass�
question_only�one_rr_per_rrset�ignore_trailing)r5 rv rw r} r~ r r
r
r r6 U s
z_WireReader.__init__c C s� | j r|dkrtjj�x�td|�D ]�}tjj| j| j�\}}| j j
dk rV|j | j j
�}| j| | _t j
d| j| j| jd � �\}}| jd | _| j j| j j|||ddd� | j r"|| _q"W dS )z�Read the next I{qcount} records from the wire data and add them to
the question section.
@param qcount: the number of questions in the message
@type qcount: intr r Nz!HH� T)r\ r] )ry r � exception� FormErrorr rW � from_wirerv rx rw r/ rC �struct�unpackr_ r r| )r5 �qcount�i�qname�usedrY rX r
r
r �
_get_question` s z_WireReader._get_questionc
C sH | j s | jrd}nd}d}�x&td|�D �]}| j}tjj| j| j�\}}|} | jj dk rh|j
| jj �}| j| | _t j d| j| j| jd � �\}
} } }
| jd | _|
tj
jk�rn|| jjk s�|r�t�| | j_| | j_| d@ d? | j_g | j_| j}|
}xj|dk�rdt j d | j||d
� �\}}|d
}tjj|| j||�}| jjj|� || }|d
| }q�W d}�n�|
tj
jk�rH|| jjk�o�||d k�s�t�| jjdk�r�td ��| jjj| �}|dk�r�td
| ��| | j_ tj j | j| j|
�\| j_ | j_ tj j!| j| |t"t#j#� �| jj$|| j|
| jj%| jj&| jj'� | j_%d| j_(n�| dk �rVd} | j �r�| tj)j*k�sz| tj)j+k�r�| }| j,} nd}|tj)j*k�s�|tj)j+k�r�|| jj-k�r�tj
j+}d}n&tj.j| |
| j| j|
| jj �}|j/� }| jj0�r|
tj
j1k�rd}| jj2||| |
||d|�}|dk �r4|j3|| � | j|
| _q(W dS )a/ Read the next I{count} records from the wire data and add them to
the specified section.
@param section: the section of the message to which to add records
@type section: list of dns.rrset.RRset objects
@param count: the number of records to read
@type count: intTFr Nz!HHIH�
i � rl z!HHr� r z"got signed message without keyringzkey '%s' unknown)4ry r~ r rx r rW r� rv rw r/ rC r� r� rs �OPTr r r r r r! Zoption_from_wirerV ZTSIGr r# r rS r$ r% Zget_algorithm_and_macr' r- Zvalidate�int�timer( r0 r2 r3 r1 rz �ANYrt r| r �rdatarZ r. ZSOAr_ �add)r5 rQ �countr] Zseen_optr� Zrr_startrW r� Z
absolute_namerY rX �ttlZrdlenrx ZoptslenZotypeZolen�optZsecretr[ rZ �rdrF r
r
r � _get_sectionx s�
z_WireReader._get_sectionc C s� t | j�}|dk rt�tjd| jdd� �\| j_| j_}}}}d| _t j
j | jj�r\d| _ | j
|� | jrpdS | j| jj|� | j| jj|� | j| jj|� | j r�| j|kr�t�| jjr�| jjr�| jj r�| jjj| j� dS )zNRead a wire format DNS message and build a dns.message.Message
object.� z!HHHHHHNT)�lenrv r r� r� rw r r rx r r> rA ry r� r} r� r r r r r r2 r0 r1 �update)r5 �lr� ZancountZaucountZadcountr
r
r �read� s$
*
z_WireReader.readN)FFF)r r
r r r6 r� r� r� r
r
r
r ru > s
cru r FTc
C sP t dd�} || _|| _|| _|| _|| _|| _|| _t| | || |
�} | j � | S )a) Convert a DNS wire format message into a message
object.
@param keyring: The keyring to use if the message is signed.
@type keyring: dict
@param request_mac: If the message is a response to a TSIG-signed request,
I{request_mac} should be set to the MAC of that request.
@type request_mac: string
@param xfr: Is this message part of a zone transfer?
@type xfr: bool
@param origin: If the message is part of a zone transfer, I{origin}
should be the origin name of the zone.
@type origin: dns.name.Name object
@param tsig_ctx: The ongoing TSIG context, used when validating zone
transfers.
@type tsig_ctx: hmac.HMAC object
@param multi: Is this message part of a multiple message sequence?
@type multi: bool
@param first: Is this message standalone, or the first of a multi
message sequence?
@type first: bool
@param question_only: Read only up to the end of the question section?
@type question_only: bool
@param one_rr_per_rrset: Put each RR into its own RRset
@type one_rr_per_rrset: bool
@param ignore_trailing: Ignore trailing junk at end of request?
@type ignore_trailing: bool
@raises ShortHeader: The message is less than 12 octets long.
@raises TrailingJunk: There were octets in the message past the end
of the proper DNS message.
@raises BadEDNS: An OPT record was in the wrong section, or occurred more
than once.
@raises BadTSIG: A TSIG record was not the last record of the additional
data section.
@rtype: dns.message.Message objectr )r )
r r# r( r. r/ r0 r2 r3 ru r� )
rv r# r( r. r/ r0 r2 r3 r} r~ r �m�readerr
r
r r� � s (
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� _TextReadera Text format reader.
@ivar tok: the tokenizer
@type tok: dns.tokenizer.Tokenizer object
@ivar message: The message object being built
@type message: dns.message.Message object
@ivar updating: Is the message a dynamic update?
@type updating: bool
@ivar zone_rdclass: The class of the zone in messages which are
DNS dynamic updates.
@type zone_rdclass: int
@ivar last_name: The most recently read name when building a message object
from text format.
@type last_name: dns.name.Name object
c C s. || _ tjj|�| _d | _tjj| _d| _ d S )NF)
rw r Z tokenizer� Tokenizer�tok� last_namerz r{ r| ry )r5 �textrw r
r
r r6 > s
z_TextReader.__init__c C s� | j j� }|j}|dkr*| j j� | j_�n�|dkr�x>| j j� }|j� sT| j j|� P | jjt jj
|j�B | j_q4W t j j | jj�r�d| _
�n&|dkr�| j j� | j_| jj| jjd> B | j_n�|dk�r | jjdk r�d| j_x�| j j� }|j� � s�| j j|� P | jjt jj|j�B | j_q�W n�|dk�rN| j j� | j_| jjdk �r�d| j_nd|d k�r�| j j� }| jjt j jt j j
|��B | j_n.|d
k�r�| j j� }| jjt jj
|�� nt�| j j� d S ) z5Process one line from the text format header section.r r Tr rl Zeflagsr r r> r@ N)r� rS rp Zget_intrw r �
is_identifier�ungetr r rh r> rA ry r r Zedns_from_textr Z
get_stringro rq r@ r �get_eol)r5 rQ �tokenZwhatr� r
r
r � _header_lineE sR
z_TextReader._header_linec C s� | j jdd�}|j� s(tjj|jd�| _| j}| j j� }|j� sHtj j
�y,tj j|j�}| j j� }|j� srtj j
�W n: tj j
k
r� tj j
�Y n t k
r� tj j
}Y nX tjj|j�}| jj| jj|||ddd� | jr�|| _| j j� dS )z7Process one line from the text format question section.T)� want_leadingN)r\ r] )r� rS �
is_whitespacer rW rh rp r� r� r� � SyntaxErrorrz � Exceptionr{ rs rw r_ r ry r| r� )r5 rQ r� rW rX rY r
r
r �_question_lines s.
z_TextReader._question_linec C s� d}| j jdd�}|j� s,tjj|jd�| _| j}| j j� }|j� sLtj j
�y*t |jd�}| j j� }|j� sttj j
�W n6 tj j
k
r� tj j
�Y n t k
r� d}Y nX yPtj
j|j�}| j j� }|j� s�tj j
�|tj
jks�|tj
jk� r�|}| j}W n> tj j
k
�r tj j
�Y n t k
�r< tj
j}Y nX tjj|j�}| j j� }|j� �s�| j j|� tjj||| j d�}|j� } n d}tjj} | jj||||| |d| j�}
|dk �r�|
j||� dS )zfProcess one line from the text format answer, authority, or
additional data sections.
NT)r� r )r� rS r� r rW rh rp r� r� r� r� r� r� rz r� rt r| r{ rs �
is_eol_or_eofr� r� rZ rw r_ ry r� ) r5 rQ r[ r� rW r� rX rY r� rZ rF r
r
r �_rr_line� sT
z_TextReader._rr_linec C s� | j }d}x�| jjdd�}|j� r$P |j� r�|jj� }|dkrF| j }nv|dksV|dkrf| j}| jj }nV|dksv|dkr�| j
}| jj }n6|dks�|d kr�| j
}| jj }n|d
kr�| j
}| jj
}| jj� q | jj|� ||� q W dS ) zNRead a text format DNS message and build a dns.message.Message
object.NTZHEADERZQUESTIONZZONEra ZPREREQrb ZUPDATErc )r� r� rS r� Z
is_commentrp �upperr� rw r r� r r r r� r� )r5 Z line_methodrQ r� �ur
r
r r� � s4
z_TextReader.readN) r r
r r r6 r� r� r� r� r
r
r
r r� , s . 5r� c C s t � }t| |�}|j� |S )z�Convert the text format message into a message object.
@param text: The text format message.
@type text: string
@raises UnknownHeaderField:
@raises dns.exception.SyntaxError:
@rtype: dns.message.Message object)r r� r� )r� r� r� r
r
r rh � s
rh c
C sF t }d}t| |�r"t| |�} d}nd}z t| �}W d|r@| j� X |S )a Read the next text format message from the specified file.
@param f: file or string. If I{f} is a string, it is treated
as the name of a file to open.
@raises UnknownHeaderField:
@raises dns.exception.SyntaxError:
@rtype: dns.message.Message objectZrUTFN)r rG �openrh �close)�fZstr_typeZoptsZ
want_closer� r
r
r � from_file� s
r� c C s t | t�rtjj| �} t |t�r,tjj|�}t |t�rBtjj|�}t� } | jtjj O _| j
| j | ||ddd� i }
|dk r�||
d< |dkr�d}|dk r�||
d< |dkr�d}|dk r�||
d< |dkr�d}|dk r�||
d< |dkr�d}||
d < | j f |
� | j
|� | S )
aV 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.
@param qname: The query name.
@type qname: dns.name.Name object or string
@param rdtype: The desired rdata type.
@type rdtype: int
@param rdclass: The desired rdata class; the default is class IN.
@type rdclass: int
@param use_edns: The EDNS level to use; the default is None (no EDNS).
See the description of dns.message.Message.use_edns() for the possible
values for use_edns and their meanings.
@type use_edns: int or bool or None
@param want_dnssec: Should the query indicate that DNSSEC is desired?
@type want_dnssec: bool
@param ednsflags: EDNS flag values.
@type ednsflags: int
@param payload: The EDNS sender's payload field, which is the maximum
size of UDP datagram the sender can handle.
@type payload: int
@param request_payload: The EDNS payload size to use when sending
this message. If not specified, defaults to the value of payload.
@type request_payload: int or None
@param options: The EDNS options
@type options: None or list of dns.edns.Option objects
@see: RFC 2671
@rtype: dns.message.Message objectT)r\ r] Nr r r r" r! r )rG r r rW rh rs rz r r �RDr_ r rm rn ) r� rY rX rm rn r r r" r! r� �kwargsr
r
r �
make_query s> $
r� � �, c C s� | j tj j@ rtjjd��tjj| j�}tj j| j tj j@ B |_ |rV| j tj j O _ |j
| j � � t | j
�|_
| jdkr�|jdd|| j� | jr�|j| j| j|ddd| 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.
@param query: the query to respond to
@type query: dns.message.Message object
@param recursion_available: should RA be set in the response?
@type recursion_available: bool
@param our_payload: payload size to advertise in EDNS responses; default
is 8192.
@type our_payload: int
@param fudge: TSIG time fudge; default is 300 seconds.
@type fudge: int
@rtype: dns.message.Message objectz&specified query message is not a queryr Nr )r r rL r� r� rw r r r� ZRArr r> rf r r rm r r1 rj r# r$ r' r- r( )ZqueryZrecursion_availableZ our_payloadr+ Zresponser
r
r �
make_response\ s
r� )
Nr FNNFTFFF)Fr� r� ).r Z
__future__r �ior r� r� Zdns.ednsr Z
dns.exceptionZ dns.flagsZdns.nameZ
dns.opcodeZ dns.entropyZ dns.rcodeZ dns.rdataZdns.rdataclassZ
dns.rdatatypeZ dns.rrsetZ dns.rendererZdns.tsigZ dns.wiredataZ_compatr r r r� r� r r Z DNSExceptionr r r r �objectr ru r� r� rh r� rz r{ r� r� r
r
r
r �<module> sX v 7
5 :F