관리-도구
편집 파일: renderer.pyc
� w�Qc @ s� d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d Z d Z d Z d Z d e f d � � YZ d S( s* Help for building DNS wire format messagesi����Ni i i i t Rendererc B s� e Z d Z d d d d d � Z d � Z d � Z e j j d � Z d � Z d � Z d d � Z e j j 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 j � | _ | d k r3 t j d d � | _ n | | _ | | _ | | _ | | _ i | _ t | _ d d d d g | _ | j j 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�� s i t Ns ( t cStringIOt StringIOt outputt Nonet randomt randintt idt flagst max_sizet origint compresst QUESTIONt sectiont countst writet mac( t selfR R R R ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt __init__J s c C s~ | j j | � | j j � g } x9 | j j � D]( \ } } | | k r3 | j | � q3 q3 Wx | D] } | j | =qf 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 t seekt truncateR t iteritemst append( R t wheret keys_to_deletet kt v( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt _rollbackg s c C s= | j | k r9 | j | k r- t j j � 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 t dnst exceptiont FormError( R R ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt _set_sectiony s c C s� | j t � | j j � } | j | j | j | j � | j j t j d | | � � | j j � } | | j k r� | j | � t j j � n | j 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 t tellt to_wireR R R t structt packR R R R t TooBigR ( R t qnamet rdtypet rdclasst beforet after( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt add_question� s c K s� | j | � | j j � } | j | j | j | j | � } | j j � } | | j k rw | j | � t j j � n | j | 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 R R R R R% R ( R R t rrsett kwR) t nR* ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt add_rrset� s ! c K s� | j | � | j j � } | j | | j | j | j | � } | j j � } | | j k rz | j | � t j j � n | j | 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 R R R R R% R ( R R t namet rdatasetR- R) R. R* ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt add_rdataset� s c C s | d M} | | d >O} | j t � | j j � } | j j t j d d t j j | | d � � | d k r�| j j � } x� | D]� } t j d | j d � } | j j | � | j j � } | j | j � | j j � } | | d k s� t � | j j | d � t j d | | � } | j j | � | j j d d � q� W| j j � } | | d k slt � | j j | d � t j d | | � } | j j | � | j j d d � n | j j � } | | j k r�| j | � t j j � n | j 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( R t ADDITIONALR R! R R# R$ R t rdatatypet OPTR t otypeR" t AssertionErrorR R R R R% R ( R t ednst ednsflagst payloadt optionsR) t lstartt optt stufft startt endt lendR* ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt add_edns� s< ! c C s� | j t � | j j � } | j j � } t j j | | | t t j � � | | | | | d | � \ } | _ } | j | j | j | j � | j j t j d t j j t j j d d � � | j j � } | j j | � | j j � } | | d k st � | | j k r/| j | � t j j � n | j j | d � | j j t j d | | � � | j t c d 7<| j j d � | j j t j d | j t � � | j j 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 R3 R R! t getvalueR t tsigt signt intt timeR R" R R R R# R$ R4 t TSIGt rdataclasst ANYR7 R R R R% R R ( R t keynamet secrett fudgeR t tsig_errort other_datat request_macRC R) t st tsig_rdatat ctxt rdata_startR* ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt add_tsig� s: #c C st | j j d � | j j t j d | j | j | j d | j d | j d | j d � � | j j 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 ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt write_header1 s c C s | j j � S( s@ Return the wire format message. @rtype: string ( R RD ( R ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt get_wire? s N( t __name__t __module__t __doc__R R R R R RJ t INR+ R/ R2 RB RE t default_algorithmRV RW RX ( ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyR s ) 04 ( R[ R R# R RH t dns.exceptionR t dns.tsigR t ANSWERt AUTHORITYR3 t objectR ( ( ( s2 /usr/lib64/python2.7/site-packages/dns/renderer.pyt <module> s