[PATCH v3 12/15] tasst: Helpers for testing NDP behaviour

[David Gibson <david@gibson.dropbear.id.au>]

Signed-iff-by: David Gibson <david@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@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
+
+
+@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)
-- 
@@ -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@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
+
+
+@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