File: //usr/local/ssl/lib/python2.4/site-packages/dns/renderer.pyc
mò
w§Qc @ sn d Z d k Z d k Z d k Z d k Z d k Z d k Z d Z d Z d Z
d Z d e f d „ ƒ YZ
d S( s* Help for building DNS wire format messagesNi i i i t Rendererc B s‰ t Z d Z e d d e d „ Z d „ Z d „ Z e i i d „ Z
d „ Z d „ Z e d „ Z
e i i d
„ Z d „ Z d „ Z RS(
st Helper class for building DNS wire-format messages.
Most applications can use the higher-level L{dns.message.Message}
class and its to_wire() method to generate wire-format messages.
This class is for those applications which need finer control
over the generation of messages.
Typical use::
r = dns.renderer.Renderer(id=1, flags=0x80, max_size=512)
r.add_question(qname, qtype, qclass)
r.add_rrset(dns.renderer.ANSWER, rrset_1)
r.add_rrset(dns.renderer.ANSWER, rrset_2)
r.add_rrset(dns.renderer.AUTHORITY, ns_rrset)
r.add_edns(0, 0, 4096)
r.add_rrset(dns.renderer.ADDTIONAL, ad_rrset_1)
r.add_rrset(dns.renderer.ADDTIONAL, ad_rrset_2)
r.write_header()
r.add_tsig(keyname, secret, 300, 1, 0, '', request_mac)
wire = r.get_wire()
@ivar output: where rendering is written
@type output: cStringIO.StringIO object
@ivar id: the message id
@type id: int
@ivar flags: the message flags
@type flags: int
@ivar max_size: the maximum size of the message
@type max_size: int
@ivar origin: the origin to use when rendering relative names
@type origin: dns.name.Name object
@ivar compress: the compression table
@type compress: dict
@ivar section: the section currently being rendered
@type section: int (dns.renderer.QUESTION, dns.renderer.ANSWER,
dns.renderer.AUTHORITY, or dns.renderer.ADDITIONAL)
@ivar counts: list of the number of RRs in each section
@type counts: int list of length 4
@ivar mac: the MAC of the rendered message (if TSIG was used)
@type mac: string
i iÿÿ c C s¡ t i ƒ | _ | d j o t i d d ƒ | _ n
| | _ | | _ | | _ | | _
h | _ t | _
d d d d g | _ | i i d d ƒ d | _ d S( sô Initialize a new renderer.
@param id: the message id
@type id: int
@param flags: the DNS message flags
@type flags: int
@param max_size: the maximum message size; the default is 65535.
If rendering results in a message greater than I{max_size},
then L{dns.exception.TooBig} will be raised.
@type max_size: int
@param origin: the origin to use when rendering relative names
@type origin: dns.name.Namem or None.
i iÿÿ t i t N( t cStringIOt StringIOt selft outputt idt Nonet randomt randintt flagst max_sizet origint compresst QUESTIONt sectiont countst writet mac( R R R R R
( ( t0 /usr/lib/python2.4/site-packages/dns/renderer.pyt __init__J s
c C s€ | i i | ƒ | i i ƒ g } x; | i i ƒ D]* \ } } | | j o | i
| ƒ q3 q3 Wx | D] } | i | =qh Wd S( sÕ Truncate the output buffer at offset I{where}, and remove any
compression table entries that pointed beyond the truncation
point.
@param where: the offset
@type where: int
N( R R t seekt wheret truncatet keys_to_deleteR t iteritemst kt vt append( R R R R R ( ( R t _rollbackg s
c C sA | i | j o- | i | j o t i i ‚ n | | _ n d S( sZ Set the renderer's current section.
Sections must be rendered order: QUESTION, ANSWER, AUTHORITY,
ADDITIONAL. Sections may be empty.
@param section: the section
@type section: int
@raises dns.exception.FormError: an attempt was made to set
a section value less than the current section.
N( R R t dnst exceptiont FormError( R R ( ( R t _set_sectiony s
c C sª | i t ƒ | i i ƒ } | i | i | i | i ƒ | i i
t i d | | ƒ ƒ | i i ƒ } | | i j o | i | ƒ t i i ‚ n | i t c d 7<d S( s Add a question to the message.
@param qname: the question name
@type qname: dns.name.Name
@param rdtype: the question rdata type
@type rdtype: int
@param rdclass: the question rdata class
@type rdclass: int
s !HHi N( R R" R R t tellt beforet qnamet to_wireR R
R t structt packt rdtypet rdclasst afterR R R R t TooBigR ( R R% R) R* R+ R$ ( ( R t add_questionŠ s
c K s | i | ƒ | i i ƒ } | i | i | i | i | } | i i ƒ } | | i
j o | i | ƒ t i i ‚ n | i | c | 7<d S( s Add the rrset to the specified section.
Any keyword arguments are passed on to the rdataset's to_wire()
routine.
@param section: the section
@type section: int
@param rrset: the rrset
@type rrset: dns.rrset.RRset object
N( R R" R R R# R$ t rrsetR& R R
t kwt nR+ R R R R R, R ( R R R. R/ R+ R0 R$ ( ( R t add_rrsetŸ s
!
c K s“ | i | ƒ | i i ƒ } | i | | i | i | i
| } | i i ƒ } | | i j o | i | ƒ t i i ‚ n | i | c | 7<d S( s© Add the rdataset to the specified section, using the specified
name as the owner name.
Any keyword arguments are passed on to the rdataset's to_wire()
routine.
@param section: the section
@type section: int
@param name: the owner name
@type name: dns.name.Name object
@param rdataset: the rdataset
@type rdataset: dns.rdataset.Rdataset object
N( R R" R R R# R$ t rdatasetR& t nameR R
R/ R0 R+ R R R R R, R ( R R R3 R2 R/ R+ R0 R$ ( ( R t add_rdataset´ s
$
c
C s | d M} | | d >O} | i t ƒ | i i ƒ } | i i t i
d d t i i
| | d ƒ ƒ | d
j oS| i i ƒ } xÉ | D]Á } t i
d | i d ƒ }
| i i |
ƒ | i i ƒ } | i | i ƒ | i i ƒ } | | d j p t ‚ | i i | d ƒ t i
d | | ƒ }
| i i |
ƒ | i i d d ƒ q… W| i i ƒ } | | d j p t ‚ | i i | d ƒ t i
d | | ƒ }
| i i |
ƒ | i i d d ƒ n | i i ƒ } | | i j o | i | ƒ t i i ‚ n | i t c d 7<d
S( sØ Add an EDNS OPT record to the message.
@param edns: The EDNS level to use.
@type edns: int
@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 options: The EDNS options list
@type options: list of dns.edns.Option instances
@see: RFC 2671
l ÿ~ i s !BHHIHi s !HHi i s !Hi N(! t ednsflagst ednsR R" t
ADDITIONALR R# R$ R R' R( R t rdatatypet OPTt payloadt optionsR t lstartt optt otypet stufft startR& t endt AssertionErrorR t lendR+ R R R R, R (
R R6 R5 R: R; R= RA RC R+ R@ R? R< R$ ( ( R t add_ednsÍ s@
!
c C sÄ | i t ƒ | i i ƒ } | i i ƒ } t i i
| | | t
t i ƒ ƒ | | | | | d | ƒ \ } | _ } | i | i | i | i ƒ | i i t i d t i i t i i! d d ƒ ƒ | i i ƒ }
| i i | ƒ | i i ƒ }
|
|
d j p t$ ‚ |
| i% j o | i& | ƒ t i' i( ‚ n | i i) |
d ƒ | i i t i d |
|
ƒ ƒ | i* t c d 7<| i i) d ƒ | i i t i d | i* t ƒ ƒ | i i) d d ƒ d S(
sõ Add a TSIG signature to the message.
@param keyname: the TSIG key name
@type keyname: dns.name.Name object
@param secret: the secret to use
@type secret: string
@param fudge: TSIG time fudge
@type fudge: int
@param id: the message id to encode in the tsig signature
@type 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 request_mac: This message is a response to the request which
had the specified MAC.
@type request_mac: string
@param algorithm: the TSIG algorithm to use
@type algorithm: dns.name.Name object
t algorithms !HHIHi i i s !Hi i
N(+ R R" R7 R R# R$ t getvaluet sR t tsigt signt keynamet secrett intt timet fudgeR t
tsig_errort
other_datat request_macRE t
tsig_rdataR t ctxR& R R
R R' R( R8 t TSIGt
rdataclasst ANYt rdata_startR+ RB R R R R, R R ( R RJ RK RN R RO RP RQ RE RR R+ RS RG RW R$ ( ( R t add_tsigü s<
#c
C st | i i d ƒ | i i t i d | i | i | i d | i d | i d | i d ƒ ƒ | i i d d ƒ d S( s¾ Write the DNS message header.
Writing the DNS message header is done after all sections
have been rendered, but before the optional TSIG signature
is added.
i s !HHHHHHi i i N( R R R R R' R( R R R ( R ( ( R t write_header1 s c C s
| i i ƒ S( s@ Return the wire format message.
@rtype: string
N( R R RF ( R ( ( R t get_wire? s ( t __name__t
__module__t __doc__R R R R" R RU t INR- R1 R4 RD RH t default_algorithmRX RY RZ ( ( ( R R s ) /5 ( R] R R' R RM t
dns.exceptionR t dns.tsigR t ANSWERt AUTHORITYR7 t objectR (
R R' R R Rc R R RM Rb R7 ( ( R t ? s