3
@Wy
@ s d Z ddlmZ ddlmZ ddlZddlZddlZddl Zddl
Zddl Zddl Zddl
ZddlZddlZddlZddlZddlZddlZddlZddlZddlmZmZmZ G dd dejjZ G d d
d
ejjZ G d d d ejj Z G d
d dejjZ G dd dejjZ!G dd dejj Z"G dd de#Z$G dd de#Z%d(ddZ&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 )longxrange 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-ddZdd Zdd Zd.d
d Zd d
Zdd Z dd Z
dd Z e j
jdddfddZe j
jdddfddZd/ddZdd ddd e jjfd d Zd0d!d"Zd1d#d$Zd%d& Zd'd( Zd)d* Zd+d, ZdS )2Messagea,
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_16idflagsquestionanswer authority
additionaledns ednsflagspayloadoptionsrequest_payloadkeyringkeynametsigdefault_algorithm keyalgorithm request_mac
other_data
tsig_errorfudge original_idmacxfrorigintsig_ctxhad_tsigmultifirstindex)selfr r
r
r __init__ s:
zMessage.__init__c C s dt | j d S )Nz)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 qW |r,|jd n
|jd
x0| jD ]&}|j|j||f| |jd q>W |rz|jd n
|jd x0| jD ]&}|j|j||f| |jd qW |jd x0| jD ]&}|j|j||f| |jd qW |j dd S )zConvert 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/
relativizekwsZrcrA rrsetr
r
r r: sJ
zMessage.to_textc C s t |tsdS | j|jkr dS | j|jkr.dS x | jD ]}||jkr6dS q6W x |jD ]}|| jkrTdS qTW x | jD ]}||jkrrdS qrW x |jD ]}|| jkrdS qW x | jD ]}||jkrdS qW x |jD ]}|| jkrdS qW dS )zTwo 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 othernr
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 |jtjjkr\dS tjj | j rndS x | j
D ]}||j
krvdS qvW x |j
D ]}|| j
krdS qW 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 tdd 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 |sjttjj|||||}
|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 getmatchKeyErrorr rF ZRRsetappend) r5 rQ namerdclassrdtypecoversdeletingcreate force_uniquekeyrF 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| qW | jdkr|j| j| j| j| j x"| jD ]}|j tjj|f| qW |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_rrsetANSWERr 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|tr: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# listkeysr$ 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 dM }||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| jS )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 _| jtdM _| 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| jS )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 rdatatypeNONEr_ 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 ddZdd 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_wrapwiremessagecurrentupdating
rdataclassIN zone_rdclass
question_onlyone_rr_per_rrsetignore_trailing)r5 rv rw r} r~ r r
r
r r6 U s
z_WireReader.__init__c C s | j r|dkrtjjxtd|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 )zRead 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 structunpackr_ r r| )r5 qcountiqnameusedrY 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
jkrn|| jjk s|rt| | j_| | j_| d@ d? | j_g | j_| j}|
}xj|dkrdt j d | j||d
\}}|d
}tjj|| j||}| jjj| || }|d
| }qW d}n|
tj
jkrH|| jjko||d kst| jjdkrtd | jjj| }|dkrtd
| | | 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*ksz| tj)j+kr| }| j,} nd}|tj)j*ks|tj)j+kr|| jj-krtj
j+}d}n&tj.j| |
| j| j|
| jj }|j/ }| jj0r|
tj
j1krd}| 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- Zvalidateinttimer( 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 ZoptslenZotypeZolenoptZsecretr[ rZ rdrF r
r
r _get_sectionx s
z_WireReader._get_sectionc C s t | j}|dk rttjd| jdd \| j_| j_}}}}d| _t j
j | jjr\d| _ | j
| | jrpdS | j| jj| | j| jj| | j| jj| | j r| j|krt| 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 mreaderr
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 Tokenizertok 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|dkrx>| j j }|j sT| j j| P | jjt jj
|jB | j_q4W t j j | jjrd| _
n&|dkr| j j | j_| jj| jjd> B | j_n|dkr | jjdk rd| j_x| j j }|j s| j j| P | jjt jj|jB | j_qW n|dkrN| j j | j_| jjdk rd| j_nd|d kr| j j }| jjt j jt j j
|B | j_n.|d
kr| 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_identifierungetr 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 stj 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 )zConvert 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 | trtjj| } t |tr,tjj|}t |trBtjj|}t } | jtjj O _| j
| j | ||ddd i }
|dk r||
d< |dkrd}|dk r||
d< |dkrd}|dk r||
d< |dkrd}|dk r||
d< |dkrd}||
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@ rtjjdtjj| 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 sX v 7
5 :F