3
ӕWZ @ sN d Z ddlmZ ddlZddlZddlZddlmZ ddlZ ddl
Z ddl Z ddl Z ddl
Z ddlZ ddlZ ddlZ ddlZ ddlZ ddlmZmZ ejd kZG dd d e jjZG d
d d eZG d d
d
eZ G dd deZ G dd de Z G dd de Z de j!j"de dddfddZ#de j!j"de dddfddZ$e ddfddZ%dS ) z
DNS Zones. )
generatorsN)BytesIO ) string_types text_type c @ s e Zd ZdZdS )BadZonezThe DNS zone is malformed.N)__name__
__module__ __qualname____doc__ r
r
/usr/lib/python3.6/zone.pyr ) s r c @ s e Zd ZdZdS )NoSOAz)The DNS zone has no SOA RR at its origin.N)r r
r r r
r
r
r r . s r c @ s e Zd ZdZdS )NoNSz+The DNS zone has no NS RRset at its origin.N)r r
r r r
r
r
r r 3 s r c @ s e Zd ZdZdS )
UnknownOriginz!The DNS zone's origin is unknown.N)r r
r r r
r
r
r r 8 s r c @ sj e Zd ZdZejjZddddgZej j
dfddZ d d
Z d d Z
d
d Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zd d Zd d ZeZd!d" Zd#d$ ZdCd&d'ZdDd(d)Z d*d+ Z ej j d%fd,d-Z ej j d%fd.d/Z!ej j fd0d1Z"d2d3 Z#ej j fd4d5Z$ej j fd6d7Z%ej j&ej j fd8d9Z'ej j&ej j fd:d;Z(dEd=d>Z)dFd?d@Z*dAdB Z+d rC r
r
r
find_rrsetk s
zZone.find_rrsetc C s0 y| j |||}W n tk
r* d}Y nX |S )aJ Look for rdata with the specified name and type in the zone,
and return an RRset encapsulating it.
The I{name}, I{rdtype}, and I{covers} parameters may be
strings, in which case they will be converted to their proper
type.
This method is less efficient than the similar L{get_rdataset}
because it creates an RRset instead of returning the matching
rdataset. It may be more convenient for some uses since it
returns an object which binds the owner name to the rdata.
This method may not be used to create new nodes or rdatasets;
use L{find_rdataset} instead.
None is returned if the name or type are not found.
Use L{find_rrset} if you want to have KeyError raised instead.
@param name: the owner name to look for
@type name: DNS.name.Name object or string
@param rdtype: the rdata type desired
@type rdtype: int or string
@param covers: the covered type (defaults to None)
@type covers: int or string
@rtype: dns.rrset.RRset object
N)rE r# )r r r< r= rC r
r
r get_rrset s
zZone.get_rrsetc c s~ t |trtjj|}t |tr,tjj|}xL| j D ]@\}}x6|D ].}|tjjksh|j|krD|j|krD||fV qDW q6W dS )a Return a generator which yields (name, rdataset) tuples for
all rdatasets in the zone which have the specified I{rdtype}
and I{covers}. If I{rdtype} is dns.rdatatype.ANY, the default,
then all rdatasets will be matched.
@param rdtype: int or string
@type rdtype: int or string
@param covers: the covered type (defaults to None)
@type covers: int or string
N) r r r r: r iteritemsANYr< r= )r r< r= r r6 rdsr
r
r iterate_rdatasets s
zZone.iterate_rdatasetsc c s t |trtjj|}t |tr,tjj|}x^| j D ]R\}}xH|D ]@}|tjjksh|j|krD|j|krDx|D ]}||j |fV qnW qDW q6W dS )a Return a generator which yields (name, ttl, rdata) tuples for
all rdatas in the zone which have the specified I{rdtype}
and I{covers}. If I{rdtype} is dns.rdatatype.ANY, the default,
then all rdatas will be matched.
@param rdtype: int or string
@type rdtype: int or string
@param covers: the covered type (defaults to None)
@type covers: int or string
N)
r r r r: r rG rH r< r= ttl)r r< r= r r6 rI rdatar
r
r iterate_rdatas s
zZone.iterate_rdatasNc C s8 t |trt|d}d}nd}t|dd}|dkr6d}|dkrPtjj|}d}n"t |trf|j|}n |}|j }z|rt| j }|j
n| j }x|D ]} | | j | | j
|d}
t |
tr|
j|} n |
} |
j }
y|j| |j| W q tk
r |j|
|j| Y qX qW W d|r2|j X dS ) a Write a zone to a file.
@param f: file or string. If I{f} is a string, it is treated
as the name of a file to open.
@param sorted: if True, the file will be written with the
names sorted in DNSSEC order from least to greatest. Otherwise
the names will be written in whatever order they happen to have
in the zone's dictionary.
@param relativize: if True, domain names in the output will be
relativized to the zone's origin (if possible).
@type relativize: bool
@param nl: The end of line string. If not specified, the
output will use the platform's native end-of-line marker (i.e.
LF on POSIX, CRLF on Windows, CR on Macintosh).
@type nl: string or None
wbTFencodingNzutf-8
)r r )r r opengetattroslinesepencodedecodelistr- sortr. to_textr r write TypeErrorclose) r fsortedr nl
want_closeZfile_encZnl_bnamesnlZl_br
r
r to_file sD
z Zone.to_filec C s* t }| j|||| |j }|j |S )a Return a zone's text as though it were written to a file.
@param sorted: if True, the file will be written with the
names sorted in DNSSEC order from least to greatest. Otherwise
the names will be written in whatever order they happen to have
in the zone's dictionary.
@param relativize: if True, domain names in the output will be
relativized to the zone's origin (if possible).
@type relativize: bool
@param nl: The end of line string. If not specified, the
output will use the platform's native end-of-line marker (i.e.
LF on POSIX, CRLF on Windows, CR on Macintosh).
@type nl: string or None
)r rd getvaluer\ )r r^ r r_ Z temp_bufferZ return_valuer
r
r rY $ s
z Zone.to_textc C sJ | j rtjj}n| j}| j|tjjdkr.t| j|tjj dkrFt
dS )zDo some simple checking of the zone's origin.
@raises dns.zone.NoSOA: there is no SOA RR
@raises dns.zone.NoNS: there is no NS RRset
@raises KeyError: there is no origin node
N) r r r emptyr r? r: ZSOAr ZNSr )r r r
r
r check_origin9 s
zZone.check_origin)F)F)TTN)TTN),r r
r r r r6 ZNoder4 __slots__
rdataclassINr r! r" r% r' r) r* r+ r. r- r0 r/ r1 rG r2 r3 r7 r8 r9 r: ZNONEr; r? r@ rB rE rF rH rJ rM rd rY rg r
r
r
r r = sJ
# !("
@
r c @ sH e Zd ZdZeddfddZdd Zdd Zd
d Zd d
Z dd Z
dS )
_MasterReadera Read a DNS master file
@ivar tok: The tokenizer
@type tok: dns.tokenizer.Tokenizer object
@ivar ttl: The default TTL
@type ttl: int
@ivar last_name: The last name read
@type last_name: dns.name.Name object
@ivar current_origin: The current origin
@type current_origin: dns.name.Name object
@ivar relativize: should names in the zone be relativized?
@type relativize: bool
@ivar zone: the zone
@type zone: dns.zone.Zone object
@ivar saved_state: saved reader state (used when processing $INCLUDE)
@type saved_state: list of (tokenizer, current_origin, last_name, file)
tuples.
@ivar current_file: the file object of the $INCLUDed file being parsed
(None if no $INCLUDE is active).
@ivar allow_include: is $INCLUDE allowed?
@type allow_include: bool
@ivar check_origin: should sanity checks of the origin node be done?
The default is True.
@type check_origin: bool
FTc C sb t |trtjj|}|| _|| _|| _d| _| j| _ ||||d| _
g | _ d | _ || _
|| _d S )Nr )r )r r r r r tokcurrent_originr rK last_namezone saved_state current_file
allow_includerg )r rl r r r zone_factoryrr rg r
r
r r f s
z_MasterReader.__init__c C s x| j j }|j rP qW d S )N)rl r2
is_eol_or_eof)r tokenr
r
r _eat_lineu s
z_MasterReader._eat_linec " C s | j dkrt| jjdd}|j s:tjj|j| j | _ n"| jj }|j
rPdS | jj | | j }|j | j
js|| j dS | jr|j| j
j}| jj }|j stjjy,tjj|j}| jj }|j stjjW n tjjk
r | j}Y nX y.tjj|j}| jj }|j s"tjjW n> tjjk
rF tjjY n tk
rb | j
j}Y nX || j
jkr~tjjdytjj|j}W n tjjd|j Y nX | j
jj|}|dkr| j
j }|| j
j|< y tj j||| j| j d}W nl tjjk
r, t j dd \}} | Y n< t j dd \}} tjjdt |t | f Y nX |j | j
j| j |j! }
|j"|||
d} | j#|| dS ) z(Process one line from a DNS master file.NT)Z want_leadingz RR class is not zone's classzunknown rdatatype '%s'F zcaught exception %s: %s)$rm r rl r2 Z
is_whitespacer r r r( rn rt ungetr$ ro r rv r
is_identifier exception SyntaxErrorrK BadTTLri Exceptionr r: r r4 rL sysexc_infostrchoose_relativityr= r; add) r ru r rK r r< rb rdtyvar= rI r
r
r _rr_line{ sx
z_MasterReader._rr_linec
C s t jd}t jd}t jd}|j|}|rJ|j \}}}} }
|dkrJd}|j|} | rz| j \}}}|dkrrd}d} d}
|j|} | r|j \}}}} |dkrd}|j d } d}
|p| p| sd}d}d}d} d}
|
dkrt |||| |
fS ) Nz"^.*\$({(\+|-?)(\d+),(\d+),(.)}).*$z^.*\$({(\+|-?)(\d+)}).*$z ^.*\$({(\+|-?)(\d+),(\d+)}).*$ +r drw )recompilematchgroupsNotImplementedError)
r ZsideZ is_generate1Z is_generate2Z is_generate3Zg1modZsignoffsetwidthbaseZg2Zg3r
r
r
_parse_modify s>
z_MasterReader._parse_modifyc ! 7 C s | j dkrt| jj }y2tjj|j\}}}| jj }|j sHtj j
W n tj j
Y nX y$|j}| jj }|j stj j
W n tj j
Y nX y,tj j|j}| jj }|j stj j
W n tj j k
r | j }Y nX y.tj
j|j}| jj }|j stj j
W n> tj j
k
r6 tj j
Y n tk
rR | jj}Y nX || jjkrntj j
dy.tjj|j}| jj }|j stj j
W n( tk
r tj j
d|j Y nX y
|j} W n tj j
Y nX | j|\}
} } }
}| j| \}}}}}xt||d |D ]}| dkr@|t| }n| dkrV|t| }|dkrn|t| }n|dkr|t| }t|jt|
}t|jt|}|jd|
|}| jd| |}tjj|| j | _| j}|j| jjs| j dS | j r|j | jj}| jj j|}|dkr@| jj }|| jj |< ytj j|||| j d} W nl tj j
k
r t!j" dd \} } | Y n< t!j" dd \} } tj j
d
t| t| f Y nX | j#| jj| j | j$ } |j%||| d } | j&| | q"W dS ) zRProcess one line containing the GENERATE statement from a DNS
master file.Nz RR class is not zone's classzunknown rdatatype '%s'r r -z$%sFrw zcaught exception %s: %sT)'rm r rl r2 r Zgranger r( ry rz r{ rK r| ri r} ro r r: r rangeintr zfillreplacer rn r$ r rv r r r4 rL r~ r r r= r; r )!r ru startstopstepZlhsrK r r< ZrhsZlmodZlsignZloffsetZlwidthZlbaseZrmodZrsignZroffsetZrwidthZrbaseiZlindexrindexZlzfindexZrzfindexr rL rb r r r r= rI r
r
r _generate_line s
z _MasterReader._generate_linec C s y x| j jdd}|j rh| jdk r2| jj t| jdkrb| jjd\| _ | _| _ | _| _
qP n|j rvqn|j r| j j
qnx|jd dkr|jj }|dkr| j j }|j stjjdtj
j|j| _
| j j
q|dkr$| j j | _| j j
| jjdkr| j| j_q|d kr| jr| j j }|j}| j j }|j rxtjj|j| j}| j j
n |j stjjd
n| j}| jj| j | j| j | j| j
f t |d | _tj j | j|| _ || _q|d kr| j qtjjd
| d q| j j | | j! qW W nZ tjjk
rz } z8| j j" \}}|dkrTd}tjjd|||f W Y dd}~X nX | j#r| jj# dS )zRead a DNS master file and build a zone object.
@raises dns.zone.NoSOA: No SOA RR was found at the zone origin
@raises dns.zone.NoNS: No NS RRset was found at the zone origin
TNr r $z$TTLzbad $TTLz$ORIGINz$INCLUDEzbad origin in $INCLUDErz $GENERATEz Unknown master file directive ''z syntax errorz %s:%d: %s)$rl r2 Zis_eofrq r\ rA rp poprm rn rK Zis_eolZ
is_commentZget_eolr( upperry r rz r{ r Zget_namero r rr r rt appendrQ tokenizer Tokenizerr rx r whererg )r ru cfilenameZ
new_originZdetailZ line_numberr
r
r reade s
"z_MasterReader.readN) r r
r r r r rv r r r r r
r
r
r rk J s L)urk TFc
C s> |dkr d}t jj| |}t|||||||d} | j | jS )ar Build a zone object from a master file format string.
@param text: the master file format input
@type text: string.
@param origin: The origin of the zone; if not specified, the first
$ORIGIN statement in the master file will determine the origin of the
zone.
@type origin: dns.name.Name object or string
@param rdclass: The zone's rdata class; the default is class IN.
@type rdclass: int
@param relativize: should names be relativized? The default is True
@type relativize: bool
@param zone_factory: The zone factory to use
@type zone_factory: function returning a Zone
@param filename: The filename to emit when describing where an error
occurred; the default is ''.
@type filename: string
@param allow_include: is $INCLUDE allowed?
@type allow_include: bool
@param check_origin: should sanity checks of the origin node be done?
The default is True.
@type check_origin: bool
@raises dns.zone.NoSOA: No SOA RR was found at the zone origin
@raises dns.zone.NoNS: No NS RRset was found at the zone origin
@rtype: dns.zone.Zone object
Nz)rr rg )r r r rk r ro )
textr r r rs r rr rg rl readerr
r
r r s " r c C sl t }d} t| |r.|dkr | }t| | } d}
n|dkr:d}d}
zt| |||||||} W d|
rf| j X | S )a Read a master file and build a zone object.
@param f: file or string. If I{f} is a string, it is treated
as the name of a file to open.
@param origin: The origin of the zone; if not specified, the first
$ORIGIN statement in the master file will determine the origin of the
zone.
@type origin: dns.name.Name object or string
@param rdclass: The zone's rdata class; the default is class IN.
@type rdclass: int
@param relativize: should names be relativized? The default is True
@type relativize: bool
@param zone_factory: The zone factory to use
@type zone_factory: function returning a Zone
@param filename: The filename to emit when describing where an error
occurred; the default is '', or the value of I{f} if I{f} is a
string.
@type filename: string
@param allow_include: is $INCLUDE allowed?
@type allow_include: bool
@param check_origin: should sanity checks of the origin node be done?
The default is True.
@type check_origin: bool
@raises dns.zone.NoSOA: No SOA RR was found at the zone origin
@raises dns.zone.NoNS: No NS RRset was found at the zone origin
@rtype: dns.zone.Zone object
ZrUNTzF)r r rQ r r\ ) r] r r r rs r rr rg Zstr_typeZoptsr` zr
r
r from_file s
r c C s d}x| D ]}|dkrH|r"|j }n |jd j}|jd j}||||d}x||jD ]r}|jj|j} | sz|j } | |j|j< | j|j|j|j d}
|
j
|j x$|D ] } | j |j | |
j
| qW qPW q
W |r|j |S )a Convert the output of a zone transfer generator into a zone object.
@param xfr: The xfr generator
@type xfr: generator of dns.message.Message objects
@param relativize: should names be relativized? The default is True.
It is essential that the relativize setting matches the one specified
to dns.query.xfr().
@type relativize: bool
@param check_origin: should sanity checks of the origin node be done?
The default is True.
@type check_origin: bool
@raises dns.zone.NoSOA: No SOA RR was found at the zone origin
@raises dns.zone.NoNS: No NS RRset was found at the zone origin
@rtype: dns.zone.Zone object
Nr )r T)r Zanswerr r r r2 r4 r; r< r= Z
update_ttlrK r r rg ) Zxfrrs r rg r r r r rC ZznodeZzrdsr r
r
r from_xfr s,
r )r )&r Z
__future__r r~ r rS ior Z
dns.exceptionr Zdns.nameZdns.nodeZdns.rdataclassZ
dns.rdatatypeZ dns.rdataZ dns.rrsetZ
dns.tokenizerZdns.ttlZ
dns.grangeZ_compatr r version_infor, rz Z DNSExceptionr r r r objectr rk ri rj r r r r
r
r
r sH
o*3