Signed-iff-by: David Gibson <david(a)gibson.dropbear.id.au> --- test/tasst/__main__.py | 1 + test/tasst/ndp.py | 106 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 107 insertions(+) create mode 100644 test/tasst/ndp.py diff --git a/test/tasst/__main__.py b/test/tasst/__main__.py index 251edae5..92319d46 100644 --- a/test/tasst/__main__.py +++ b/test/tasst/__main__.py @@ -18,6 +18,7 @@ import exeter MODULES = [ 'cmdsite', 'ip', + 'ndp', 'transfer', 'unshare', 'veth', diff --git a/test/tasst/ndp.py b/test/tasst/ndp.py new file mode 100644 index 00000000..0ea2f75e --- /dev/null +++ b/test/tasst/ndp.py @@ -0,0 +1,106 @@ +#! /usr/bin/env avocado-runner-avocado-classless + +# SPDX-License-Identifier: GPL-2.0-or-later +# +# Copyright Red Hat +# Author: David Gibson <david(a)gibson.dropbear.id.au> + +""" +Test A Simple Socket Transport + +ndp.py - Helpers for testing NDP +""" + +import dataclasses +import ipaddress +import os +import tempfile +from typing import Iterator + +import exeter + +from . import cmdsite, ip, unshare, veth + + +(a)dataclasses.dataclass +class NdpScenario(exeter.Scenario): + client: cmdsite.CmdSite + ifname: str + network: ip.Net + gateway: ip.Addr + + @exeter.scenariotest + def ndp_addr(self) -> None: + # Wait for NDP to do its thing + (addr,) = ip.addr_wait(self.client, self.ifname, + family='inet6', scope='global') + + # The SLAAC address is derived from the guest ns MAC, so + # probably won't exactly match the host address (we need + # DHCPv6 for that). It should be in the right network though. + exeter.assert_eq(addr.network, self.network) + + @exeter.scenariotest + def ndp_route(self) -> None: + defroutes = ip.routes6(self.client, dst='default') + while not defroutes: + defroutes = ip.routes6(self.client, dst='default') + + exeter.assert_eq(len(defroutes), 1) + gw = ipaddress.ip_address(defroutes[0]['gateway']) + exeter.assert_eq(gw, self.gateway) + + +IFNAME = 'clientif' +NETWORK = ip.TEST_NET6_TASST_A +ipa = ip.IpiAllocator(NETWORK) +(ROUTER_IP6,) = ipa.next_ipis() + + +def setup_radvd() -> Iterator[NdpScenario]: + router_ifname = 'routerif' + + with unshare.unshare('client', '-Un') as client, \ + unshare.unshare('router', '-n', + parent=client, privilege=True) as router, \ + tempfile.TemporaryDirectory() as tmpdir, \ + veth.veth(client, IFNAME, router_ifname, router): + + # Configure the simulated router + confpath = os.path.join(tmpdir, 'radvd.conf') + pidfile = os.path.join(tmpdir, 'radvd.pid') + open(confpath, 'w', encoding='UTF-8').write( + f''' + interface {router_ifname} {{ + AdvSendAdvert on; + prefix {NETWORK} {{ + }}; + }}; + ''' + ) + + ip.ifup(router, 'lo') + ip.ifup(router, 'routerif', ROUTER_IP6) + + # Configure the client + ip.ifup(client, 'lo') + ip.ifup(client, IFNAME) + + # Get the router's link-local-address + (router_ll,) = ip.addr_wait(router, router_ifname, + family='inet6', scope='link') + + # Run radvd + router.fg('radvd', '-c', '-C', f'{confpath}') + radvd_cmd = ['radvd', '-C', f'{confpath}', '-n', + '-p', f'{pidfile}', '-d', '5'] + with router.bg(*radvd_cmd, privilege=True) as radvd: + yield NdpScenario(client=client, + ifname=IFNAME, + network=NETWORK, + gateway=router_ll.ip) + radvd.terminate() + + +def selftests() -> None: + NdpScenario.test(setup_radvd) -- 2.46.0