#!/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 . import re import os import unittest from psutil import process_iter from vyos.configsession import ConfigSession from vyos.configsession import ConfigSessionError from vyos.util import read_file from vyos.util import process_named_running from vyos.validate import is_ipv6 PROCESS_NAME = 'in.tftpd' base_path = ['service', 'tftp-server'] dummy_if_path = ['interfaces', 'dummy', 'dum69'] address_ipv4 = '192.0.2.1' address_ipv6 = '2001:db8::1' class TestServiceTFTPD(unittest.TestCase): def setUp(self): self.session = ConfigSession(os.getpid()) self.session.set(dummy_if_path + ['address', address_ipv4 + '/32']) self.session.set(dummy_if_path + ['address', address_ipv6 + '/128']) def tearDown(self): self.session.delete(base_path) self.session.delete(dummy_if_path) self.session.commit() del self.session def test_01_tftpd_single(self): directory = '/tmp' port = '69' # default port self.session.set(base_path + ['allow-upload']) self.session.set(base_path + ['directory', directory]) self.session.set(base_path + ['listen-address', address_ipv4]) # commit changes self.session.commit() config = read_file('/etc/default/tftpd0') # verify listen IP address self.assertIn(f'{address_ipv4}:{port} -4', config) # verify directory self.assertIn(directory, config) # verify upload self.assertIn('--create --umask 000', config) # Check for running process self.assertTrue(process_named_running(PROCESS_NAME)) def test_02_tftpd_multi(self): directory = '/tmp' address = [address_ipv4, address_ipv6] port = '70' self.session.set(base_path + ['directory', directory]) for addr in address: self.session.set(base_path + ['listen-address', addr]) self.session.set(base_path + ['port', port]) # commit changes self.session.commit() for idx in range(0, len(address)): config = read_file(f'/etc/default/tftpd{idx}') addr = address[idx] # verify listen IP address if is_ipv6(addr): addr = f'[{addr}]' self.assertIn(f'{addr}:{port} -6', config) else: self.assertIn(f'{addr}:{port} -4', config) # verify directory self.assertIn(directory, config) # Check for running processes - one process is spawned per listen # IP address, wheter it's IPv4 or IPv6 count = 0 for p in process_iter(): if PROCESS_NAME in p.name(): count += 1 self.assertEqual(count, len(address)) if __name__ == '__main__': unittest.main()