Add Mininet entries to /etc/hosts in 8r-1c-srv6-pm topology

Author: cscarpitta

nets/8r-1c-srv6-pm/isis8d.py

@@ -25,6 +25,7 @@
             exec(code, {'__file__': venv_path})
 
 from argparse import ArgumentParser
+import python_hosts
 import shutil
 from dotenv import load_dotenv
 from mininet.topo import Topo
@@ -41,6 +42,11 @@
 
 PRIVDIR = '/var/priv'
 
+# Path of the file containing the entries (ip-hostname)
+# to be added to /etc/hosts
+ETC_HOSTS_FILE = './etc-hosts'
+
+# Define whether to start the node managers on the routers or not
 START_NODE_MANAGERS = False
 
 # Load environment variables from .env file
@@ -275,6 +281,31 @@ def create_topo(my_net):
     add_link(my_net, r8, sw)
 
 
+def add_nodes_to_etc_hosts():
+    # Get /etc/hosts
+    etc_hosts = python_hosts.hosts.Hosts()
+    # Import host-ip mapping defined in etc-hosts file
+    count = etc_hosts.import_file(ETC_HOSTS_FILE)
+    # Print results
+    print('*** Added %s entries to /etc/hosts\n' % count['write_result']['total_written'])
+
+
+def remove_nodes_from_etc_hosts(net):
+    print('*** Removing entries from /etc/hosts\n')
+    # Get /etc/hosts
+    etc_hosts = python_hosts.hosts.Hosts()
+    for host in net.hosts:
+        # Remove all the nodes from /etc/hosts
+        etc_hosts.remove_all_matching(name=str(host))
+    # Remove entries related to the management network
+    # These entries are in the form *.m (e.g. r1.m, controller.m)
+    # therefore they are not removed during the previous loop
+    for host in net.hosts:
+        etc_hosts.remove_all_matching(name='%s.m' % host)
+    # Write changes to /etc/hosts
+    etc_hosts.write()
+
+
 def stopAll():
     # Clean Mininet emulation environment
     os.system('sudo mn -c')
@@ -308,7 +339,14 @@ def simpleTest():
         for host in net.hosts:
             file.write("%s %d\n" % (host, extractHostPid( repr(host) )) )
 
+    # Add Mininet nodes to /etc/hosts
+    add_nodes_to_etc_hosts()
+
     CLI( net ) 
+
+    # Remove Mininet nodes from /etc/hosts
+    remove_nodes_from_etc_hosts(net)
+ 
     net.stop() 
     stopAll()