Gecko [ v3.cesa.co.za ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
rdtypes	[ DIR ]	drwxr-xr-x	2026-06-08 18:08
LICENSE	1.49 KB	-rw-r--r--	1980-01-01 08:00
__init__.py	1.37 KB	-rw-r--r--	1980-01-01 08:00
_compat.py	1.47 KB	-rw-r--r--	1980-01-01 08:00
dnssec.py	16.12 KB	-rw-r--r--	1980-01-01 08:00
e164.py	3.67 KB	-rw-r--r--	1980-01-01 08:00
edns.py	7.25 KB	-rw-r--r--	1980-01-01 08:00
entropy.py	4.66 KB	-rw-r--r--	1980-01-01 08:00
exception.py	4.76 KB	-rw-r--r--	1980-01-01 08:00
flags.py	2.86 KB	-rw-r--r--	1980-01-01 08:00
grange.py	2.01 KB	-rw-r--r--	1980-01-01 08:00
hash.py	1.29 KB	-rw-r--r--	1980-01-01 08:00
inet.py	3.3 KB	-rw-r--r--	1980-01-01 08:00
ipv4.py	2.05 KB	-rw-r--r--	1980-01-01 08:00
ipv6.py	5.4 KB	-rw-r--r--	1980-01-01 08:00
message.py	40.49 KB	-rw-r--r--	1980-01-01 08:00
name.py	30.53 KB	-rw-r--r--	1980-01-01 08:00
namedict.py	3.87 KB	-rw-r--r--	1980-01-01 08:00
node.py	6.2 KB	-rw-r--r--	1980-01-01 08:00
opcode.py	2.77 KB	-rw-r--r--	1980-01-01 08:00
query.py	23.14 KB	-rw-r--r--	1980-01-01 08:00
rcode.py	3.53 KB	-rw-r--r--	1980-01-01 08:00
rdata.py	14.11 KB	-rw-r--r--	1980-01-01 08:00
rdataclass.py	3.14 KB	-rw-r--r--	1980-01-01 08:00
rdataset.py	11.19 KB	-rw-r--r--	1980-01-01 08:00
rdatatype.py	5.78 KB	-rw-r--r--	1980-01-01 08:00
renderer.py	10.57 KB	-rw-r--r--	1980-01-01 08:00
resolver.py	49.03 KB	-rw-r--r--	1980-01-01 08:00
reversename.py	3.21 KB	-rw-r--r--	1980-01-01 08:00
rrset.py	6.01 KB	-rw-r--r--	1980-01-01 08:00
set.py	7.14 KB	-rw-r--r--	1980-01-01 08:00
tokenizer.py	17.62 KB	-rw-r--r--	1980-01-01 08:00
tsig.py	7.63 KB	-rw-r--r--	1980-01-01 08:00
tsigkeyring.py	1.77 KB	-rw-r--r--	1980-01-01 08:00
ttl.py	2.29 KB	-rw-r--r--	1980-01-01 08:00
update.py	10.1 KB	-rw-r--r--	1980-01-01 08:00
version.py	1.37 KB	-rw-r--r--	1980-01-01 08:00
wiredata.py	3.66 KB	-rw-r--r--	1980-01-01 08:00
zone.py	41.39 KB	-rw-r--r--	1980-01-01 08:00

Rename

# Copyright (C) Dnspython Contributors, see LICENSE for text of ISC license

# Copyright (C) 2001-2007, 2009-2011 Nominum, Inc.
#
# Permission to use, copy, modify, and distribute this software and its
# documentation for any purpose with or without fee is hereby granted,
# provided that the above copyright notice and this permission notice
# appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND NOMINUM DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NOMINUM BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
# OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

"""DNS TSIG support."""

import hashlib
import hmac
import struct

import dns.exception
import dns.rdataclass
import dns.name
from ._compat import long, string_types, text_type

class BadTime(dns.exception.DNSException):

"""The current time is not within the TSIG's validity time."""

class BadSignature(dns.exception.DNSException):

"""The TSIG signature fails to verify."""

class PeerError(dns.exception.DNSException):

"""Base class for all TSIG errors generated by the remote peer"""

class PeerBadKey(PeerError):

"""The peer didn't know the key we used"""

class PeerBadSignature(PeerError):

"""The peer didn't like the signature we sent"""

class PeerBadTime(PeerError):

"""The peer didn't like the time we sent"""

class PeerBadTruncation(PeerError):

"""The peer didn't like amount of truncation in the TSIG we sent"""

# TSIG Algorithms

HMAC_MD5 = dns.name.from_text("HMAC-MD5.SIG-ALG.REG.INT")
HMAC_SHA1 = dns.name.from_text("hmac-sha1")
HMAC_SHA224 = dns.name.from_text("hmac-sha224")
HMAC_SHA256 = dns.name.from_text("hmac-sha256")
HMAC_SHA384 = dns.name.from_text("hmac-sha384")
HMAC_SHA512 = dns.name.from_text("hmac-sha512")

_hashes = {
    HMAC_SHA224: hashlib.sha224,
    HMAC_SHA256: hashlib.sha256,
    HMAC_SHA384: hashlib.sha384,
    HMAC_SHA512: hashlib.sha512,
    HMAC_SHA1: hashlib.sha1,
    HMAC_MD5: hashlib.md5,
}

default_algorithm = HMAC_MD5

BADSIG = 16
BADKEY = 17
BADTIME = 18
BADTRUNC = 22

def sign(wire, keyname, secret, time, fudge, original_id, error,
         other_data, request_mac, ctx=None, multi=False, first=True,
         algorithm=default_algorithm):
    """Return a (tsig_rdata, mac, ctx) tuple containing the HMAC TSIG rdata
    for the input parameters, the HMAC MAC calculated by applying the
    TSIG signature algorithm, and the TSIG digest context.
    @rtype: (string, string, hmac.HMAC object)
    @raises ValueError: I{other_data} is too long
    @raises NotImplementedError: I{algorithm} is not supported
    """

if isinstance(other_data, text_type):
        other_data = other_data.encode()
    (algorithm_name, digestmod) = get_algorithm(algorithm)
    if first:
        ctx = hmac.new(secret, digestmod=digestmod)
        ml = len(request_mac)
        if ml > 0:
            ctx.update(struct.pack('!H', ml))
            ctx.update(request_mac)
    id = struct.pack('!H', original_id)
    ctx.update(id)
    ctx.update(wire[2:])
    if first:
        ctx.update(keyname.to_digestable())
        ctx.update(struct.pack('!H', dns.rdataclass.ANY))
        ctx.update(struct.pack('!I', 0))
    long_time = time + long(0)
    upper_time = (long_time >> 32) & long(0xffff)
    lower_time = long_time & long(0xffffffff)
    time_mac = struct.pack('!HIH', upper_time, lower_time, fudge)
    pre_mac = algorithm_name + time_mac
    ol = len(other_data)
    if ol > 65535:
        raise ValueError('TSIG Other Data is > 65535 bytes')
    post_mac = struct.pack('!HH', error, ol) + other_data
    if first:
        ctx.update(pre_mac)
        ctx.update(post_mac)
    else:
        ctx.update(time_mac)
    mac = ctx.digest()
    mpack = struct.pack('!H', len(mac))
    tsig_rdata = pre_mac + mpack + mac + id + post_mac
    if multi:
        ctx = hmac.new(secret, digestmod=digestmod)
        ml = len(mac)
        ctx.update(struct.pack('!H', ml))
        ctx.update(mac)
    else:
        ctx = None
    return (tsig_rdata, mac, ctx)

def hmac_md5(wire, keyname, secret, time, fudge, original_id, error,
             other_data, request_mac, ctx=None, multi=False, first=True,
             algorithm=default_algorithm):
    return sign(wire, keyname, secret, time, fudge, original_id, error,
                other_data, request_mac, ctx, multi, first, algorithm)

def validate(wire, keyname, secret, now, request_mac, tsig_start, tsig_rdata,
             tsig_rdlen, ctx=None, multi=False, first=True):
    """Validate the specified TSIG rdata against the other input parameters.

@raises FormError: The TSIG is badly formed.
    @raises BadTime: There is too much time skew between the client and the
    server.
    @raises BadSignature: The TSIG signature did not validate
    @rtype: hmac.HMAC object"""

(adcount,) = struct.unpack("!H", wire[10:12])
    if adcount == 0:
        raise dns.exception.FormError
    adcount -= 1
    new_wire = wire[0:10] + struct.pack("!H", adcount) + wire[12:tsig_start]
    current = tsig_rdata
    (aname, used) = dns.name.from_wire(wire, current)
    current = current + used
    (upper_time, lower_time, fudge, mac_size) = \
        struct.unpack("!HIHH", wire[current:current + 10])
    time = ((upper_time + long(0)) << 32) + (lower_time + long(0))
    current += 10
    mac = wire[current:current + mac_size]
    current += mac_size
    (original_id, error, other_size) = \
        struct.unpack("!HHH", wire[current:current + 6])
    current += 6
    other_data = wire[current:current + other_size]
    current += other_size
    if current != tsig_rdata + tsig_rdlen:
        raise dns.exception.FormError
    if error != 0:
        if error == BADSIG:
            raise PeerBadSignature
        elif error == BADKEY:
            raise PeerBadKey
        elif error == BADTIME:
            raise PeerBadTime
        elif error == BADTRUNC:
            raise PeerBadTruncation
        else:
            raise PeerError('unknown TSIG error code %d' % error)
    time_low = time - fudge
    time_high = time + fudge
    if now < time_low or now > time_high:
        raise BadTime
    (junk, our_mac, ctx) = sign(new_wire, keyname, secret, time, fudge,
                                original_id, error, other_data,
                                request_mac, ctx, multi, first, aname)
    if our_mac != mac:
        raise BadSignature
    return ctx

def get_algorithm(algorithm):
    """Returns the wire format string and the hash module to use for the
    specified TSIG algorithm

@rtype: (string, hash constructor)
    @raises NotImplementedError: I{algorithm} is not supported
    """

if isinstance(algorithm, string_types):
        algorithm = dns.name.from_text(algorithm)

try:
        return (algorithm.to_digestable(), _hashes[algorithm])
    except KeyError:
        raise NotImplementedError("TSIG algorithm " + str(algorithm) +
                                  " is not supported")

def get_algorithm_and_mac(wire, tsig_rdata, tsig_rdlen):
    """Return the tsig algorithm for the specified tsig_rdata
    @raises FormError: The TSIG is badly formed.
    """
    current = tsig_rdata
    (aname, used) = dns.name.from_wire(wire, current)
    current = current + used
    (upper_time, lower_time, fudge, mac_size) = \
        struct.unpack("!HIHH", wire[current:current + 10])
    current += 10
    mac = wire[current:current + mac_size]
    current += mac_size
    if current > tsig_rdata + tsig_rdlen:
        raise dns.exception.FormError
    return (aname, mac)