www.tinc-vpn.org Git - tinc/blob - test/integration/ns_ping.py

   1 #!/usr/bin/env python3
   2
   3 """Create two network namespaces and run ping between them."""
   4
   5 import typing as T
   6
   7 from testlib import external as ext, util, template, cmd
   8 from testlib.log import log
   9 from testlib.proc import Tinc, Script
  10 from testlib.test import Test
  11 from testlib.external import ping
  12
  13 util.require_root()
  14 util.require_command("ip", "netns", "list")
  15 util.require_path("/dev/net/tun")
  16
  17 IP_FOO = "192.168.1.1"
  18 IP_BAR = "192.168.1.2"
  19 MASK = 24
  20
  21
  22 def init(ctx: Test) -> T.Tuple[Tinc, Tinc]:
  23     """Initialize new test nodes."""
  24     foo, bar = ctx.node(), ctx.node()
  25
  26     log.info("create network namespaces")
  27     assert ext.netns_add(foo.name)
  28     assert ext.netns_add(bar.name)
  29
  30     log.info("initialize two nodes")
  31
  32     stdin = f"""
  33         init {foo}
  34         set Port 0
  35         set Subnet {IP_FOO}
  36         set Interface {foo}
  37         set Address localhost
  38         set AutoConnect no
  39     """
  40     foo.cmd(stdin=stdin)
  41     foo.add_script(Script.TINC_UP, template.make_netns_config(foo.name, IP_FOO, MASK))
  42     foo.start()
  43
  44     stdin = f"""
  45         init {bar}
  46         set Port 0
  47         set Subnet {IP_BAR}
  48         set Interface {bar}
  49         set Address localhost
  50         set AutoConnect no
  51     """
  52     bar.cmd(stdin=stdin)
  53     bar.add_script(Script.TINC_UP, template.make_netns_config(bar.name, IP_BAR, MASK))
  54
  55     cmd.exchange(foo, bar)
  56
  57     return foo, bar
  58
  59
  60 with Test("ns-ping") as context:
  61     foo_node, bar_node = init(context)
  62     bar_node.cmd("start")
  63
  64     log.info("waiting for nodes to come up")
  65     bar_node[Script.TINC_UP].wait()
  66
  67     log.info("ping must not work when there is no connection")
  68     assert not ping(IP_BAR, foo_node.name)
  69
  70     log.info("add script foo/host-up")
  71     bar_node.add_script(foo_node.script_up)
  72
  73     log.info("add ConnectTo clause")
  74     bar_node.cmd("add", "ConnectTo", foo_node.name)
  75
  76     log.info("bar waits for foo")
  77     bar_node[foo_node.script_up].wait()
  78
  79     log.info("ping must work after connection is up")
  80     assert ping(IP_BAR, foo_node.name)