133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck# Copyright (C) 2003-2007, 2009, 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 dns.exception
1733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport dns.ipv4
1833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport dns.rdata
1933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckimport dns.tokenizer
2033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
2133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reckclass A(dns.rdata.Rdata):
2233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    """A record.
2333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
2433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    @ivar address: an IPv4 address
2533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    @type address: string (in the standard "dotted quad" format)"""
2633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
2733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    __slots__ = ['address']
2833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
2933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def __init__(self, rdclass, rdtype, address):
3033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        super(A, self).__init__(rdclass, rdtype)
3133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        # check that it's OK
3233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        junk = dns.ipv4.inet_aton(address)
3333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        self.address = address
3433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
3533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def to_text(self, origin=None, relativize=True, **kw):
3633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        return self.address
3733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
3833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def from_text(cls, rdclass, rdtype, tok, origin = None, relativize = True):
3933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        address = tok.get_identifier()
4033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        tok.get_eol()
4133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        return cls(rdclass, rdtype, address)
4233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
4333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    from_text = classmethod(from_text)
4433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
4533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def to_wire(self, file, compress = None, origin = None):
4633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        file.write(dns.ipv4.inet_aton(self.address))
4733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
4833259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def from_wire(cls, rdclass, rdtype, wire, current, rdlen, origin = None):
4933259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        address = dns.ipv4.inet_ntoa(wire[current : current + rdlen])
5033259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        return cls(rdclass, rdtype, address)
5133259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
5233259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    from_wire = classmethod(from_wire)
5333259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck
5433259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck    def _cmp(self, other):
5533259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        sa = dns.ipv4.inet_aton(self.address)
5633259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        oa = dns.ipv4.inet_aton(other.address)
5733259e44c8229f70ffe0cf3bb5ca9375c4feb2f9John Reck        return cmp(sa, oa)
58