133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# Copyright (C) 2010 Nominum, Inc. 233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# 333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# Permission to use, copy, modify, and distribute this software and its 433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# documentation for any purpose with or without fee is hereby granted, 533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# provided that the above copyright notice and this permission notice 633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# appear in all copies. 733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# 833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES 933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF 1033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR 1133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES 1233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN 1333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT 1433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. 1533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 1633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport struct 1733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 1833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport dns.rdata 1933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport dns.rdatatype 2033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 2133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckclass DSBase(dns.rdata.Rdata): 2233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck """Base class for rdata that is like a DS record 2333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 2433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @ivar key_tag: the key tag 2533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @type key_tag: int 2633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @ivar algorithm: the algorithm 2733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @type algorithm: int 2833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @ivar digest_type: the digest type 2933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @type digest_type: int 3033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @ivar digest: the digest 3133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @type digest: int 3233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck @see: draft-ietf-dnsext-delegation-signer-14.txt""" 3333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 3433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck __slots__ = ['key_tag', 'algorithm', 'digest_type', 'digest'] 3533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 3633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def __init__(self, rdclass, rdtype, key_tag, algorithm, digest_type, 3733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest): 3833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck super(DSBase, self).__init__(rdclass, rdtype) 3933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.key_tag = key_tag 4033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.algorithm = algorithm 4133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.digest_type = digest_type 4233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.digest = digest 4333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 4433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def to_text(self, origin=None, relativize=True, **kw): 4533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck return '%d %d %d %s' % (self.key_tag, self.algorithm, 4633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.digest_type, 4733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck dns.rdata._hexify(self.digest, 4833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck chunksize=128)) 4933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 5033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def from_text(cls, rdclass, rdtype, tok, origin = None, relativize = True): 5133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck key_tag = tok.get_uint16() 5233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck algorithm = tok.get_uint8() 5333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest_type = tok.get_uint8() 5433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck chunks = [] 5533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck while 1: 5633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck t = tok.get().unescape() 5733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck if t.is_eol_or_eof(): 5833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck break 5933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck if not t.is_identifier(): 6033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck raise dns.exception.SyntaxError 6133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck chunks.append(t.value) 6233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest = ''.join(chunks) 6333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest = digest.decode('hex_codec') 6433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck return cls(rdclass, rdtype, key_tag, algorithm, digest_type, 6533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest) 6633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 6733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck from_text = classmethod(from_text) 6833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 6933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def to_wire(self, file, compress = None, origin = None): 7033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck header = struct.pack("!HBB", self.key_tag, self.algorithm, 7133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.digest_type) 7233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck file.write(header) 7333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck file.write(self.digest) 7433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 7533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def from_wire(cls, rdclass, rdtype, wire, current, rdlen, origin = None): 7633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck header = struct.unpack("!HBB", wire[current : current + 4]) 7733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck current += 4 7833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck rdlen -= 4 7933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck digest = wire[current : current + rdlen] 8033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck return cls(rdclass, rdtype, header[0], header[1], header[2], digest) 8133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 8233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck from_wire = classmethod(from_wire) 8333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck 8433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck def _cmp(self, other): 8533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck hs = struct.pack("!HBB", self.key_tag, self.algorithm, 8633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck self.digest_type) 8733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck ho = struct.pack("!HBB", other.key_tag, other.algorithm, 8833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck other.digest_type) 8933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck v = cmp(hs, ho) 9033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck if v == 0: 9133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck v = cmp(self.digest, other.digest) 9233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck return v 93