1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
# Copyright (C) Dnspython Contributors, see LICENSE for text of ISC license
import hashlib
import unittest
import time
import dns.rcode
import dns.tsig
import dns.tsigkeyring
import dns.message
keyring = dns.tsigkeyring.from_text(
{
'keyname.' : 'NjHwPsMKjdN++dOfE5iAiQ=='
}
)
keyname = dns.name.from_text('keyname')
class TSIGTestCase(unittest.TestCase):
def test_get_context(self):
key = dns.tsig.Key('foo.com', 'abcd', 'hmac-sha256')
ctx = dns.tsig.get_context(key)
self.assertEqual(ctx.name, 'hmac-sha256')
key = dns.tsig.Key('foo.com', 'abcd', 'hmac-sha512')
ctx = dns.tsig.get_context(key)
self.assertEqual(ctx.name, 'hmac-sha512')
bogus = dns.tsig.Key('foo.com', 'abcd', 'bogus')
with self.assertRaises(NotImplementedError):
dns.tsig.get_context(bogus)
def test_sign_and_validate(self):
m = dns.message.make_query('example', 'a')
m.use_tsig(keyring, keyname)
w = m.to_wire()
# not raising is passing
dns.message.from_wire(w, keyring)
def test_sign_and_validate_with_other_data(self):
m = dns.message.make_query('example', 'a')
other = b'other data'
m.use_tsig(keyring, keyname, other_data=b'other')
w = m.to_wire()
# not raising is passing
dns.message.from_wire(w, keyring)
def make_message_pair(self, qname='example', rdtype='A', tsig_error=0):
q = dns.message.make_query(qname, rdtype)
q.use_tsig(keyring=keyring, keyname=keyname)
q.to_wire() # to set q.mac
r = dns.message.make_response(q, tsig_error=tsig_error)
return(q, r)
def test_peer_errors(self):
items = [(dns.rcode.BADSIG, dns.tsig.PeerBadSignature),
(dns.rcode.BADKEY, dns.tsig.PeerBadKey),
(dns.rcode.BADTIME, dns.tsig.PeerBadTime),
(dns.rcode.BADTRUNC, dns.tsig.PeerBadTruncation),
(99, dns.tsig.PeerError),
]
for err, ex in items:
q, r = self.make_message_pair(tsig_error=err)
w = r.to_wire()
def bad():
dns.message.from_wire(w, keyring=keyring, request_mac=q.mac)
self.assertRaises(ex, bad)
|
