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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
#!/usr/bin/env python3
#
# Copyright (C) 2019-2020 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import re
import os
import unittest
from vyos.configsession import ConfigSession, ConfigSessionError
from vyos.util import read_file
from vyos.util import process_named_running
CONFIG_FILE = '/run/powerdns/recursor.conf'
FORWARD_FILE = '/run/powerdns/recursor.forward-zones.conf'
PROCESS_NAME= 'pdns-r/worker'
base_path = ['service', 'dns', 'forwarding']
allow_from = ['192.0.2.0/24', '2001:db8::/32']
listen_adress = ['127.0.0.1', '::1']
def get_config_value(key, file=CONFIG_FILE):
tmp = read_file(file)
tmp = re.findall(r'\n{}=+(.*)'.format(key), tmp)
return tmp[0]
class TestServicePowerDNS(unittest.TestCase):
def setUp(self):
self.session = ConfigSession(os.getpid())
def tearDown(self):
# Delete DNS forwarding configuration
self.session.delete(base_path)
self.session.commit()
del self.session
def test_basic_forwarding(self):
""" Check basic DNS forwarding settings """
cache_size = '20'
negative_ttl = '120'
self.session.set(base_path + ['cache-size', cache_size])
self.session.set(base_path + ['negative-ttl', negative_ttl])
# check validate() - allow from must be defined
with self.assertRaises(ConfigSessionError):
self.session.commit()
for network in allow_from:
self.session.set(base_path + ['allow-from', network])
# check validate() - listen-address must be defined
with self.assertRaises(ConfigSessionError):
self.session.commit()
for address in listen_adress:
self.session.set(base_path + ['listen-address', address])
# configure DNSSEC
self.session.set(base_path + ['dnssec', 'validate'])
# commit changes
self.session.commit()
# Check configured cache-size
tmp = get_config_value('max-cache-entries')
self.assertEqual(tmp, cache_size)
# Networks allowed to query this server
tmp = get_config_value('allow-from')
self.assertEqual(tmp, ','.join(allow_from))
# Addresses to listen for DNS queries
tmp = get_config_value('local-address')
self.assertEqual(tmp, ','.join(listen_adress))
# Maximum amount of time negative entries are cached
tmp = get_config_value('max-negative-ttl')
self.assertEqual(tmp, negative_ttl)
# Check for running process
self.assertTrue(process_named_running(PROCESS_NAME))
def test_dnssec(self):
""" DNSSEC option testing """
for network in allow_from:
self.session.set(base_path + ['allow-from', network])
for address in listen_adress:
self.session.set(base_path + ['listen-address', address])
options = ['off', 'process-no-validate', 'process', 'log-fail', 'validate']
for option in options:
self.session.set(base_path + ['dnssec', option])
# commit changes
self.session.commit()
tmp = get_config_value('dnssec')
self.assertEqual(tmp, option)
# Check for running process
self.assertTrue(process_named_running(PROCESS_NAME))
def test_external_nameserver(self):
""" Externe Domain Name Servers (DNS) addresses """
for network in allow_from:
self.session.set(base_path + ['allow-from', network])
for address in listen_adress:
self.session.set(base_path + ['listen-address', address])
nameservers = ['192.0.2.1', '192.0.2.2']
for nameserver in nameservers:
self.session.set(base_path + ['name-server', nameserver])
# commit changes
self.session.commit()
tmp = get_config_value(r'\+.', file=FORWARD_FILE)
self.assertEqual(tmp, ', '.join(nameservers))
# Check for running process
self.assertTrue(process_named_running(PROCESS_NAME))
def test_domain_forwarding(self):
""" Externe Domain Name Servers (DNS) addresses """
for network in allow_from:
self.session.set(base_path + ['allow-from', network])
for address in listen_adress:
self.session.set(base_path + ['listen-address', address])
domains = ['vyos.io', 'vyos.net']
nameservers = ['192.0.2.1', '192.0.2.2']
for domain in domains:
for nameserver in nameservers:
self.session.set(base_path + ['domain', domain, 'server', nameserver])
# commit changes
self.session.commit()
for domain in domains:
tmp = get_config_value(domain, file=FORWARD_FILE)
self.assertEqual(tmp, ', '.join(nameservers))
# Check for running process
self.assertTrue(process_named_running(PROCESS_NAME))
if __name__ == '__main__':
unittest.main()
|
