Adding support for sdfabric topology deployment
Change-Id: If4f7a6016f8b3a91b47f68bc480f1cfa8ef5d5c3
diff --git a/mininet/toposcripts/mn_lib.py b/mininet/toposcripts/mn_lib.py
new file mode 100755
index 0000000..cbfad99
--- /dev/null
+++ b/mininet/toposcripts/mn_lib.py
@@ -0,0 +1,161 @@
+# SPDX-FileCopyrightText: 2022-present Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+
+from mininet.node import Host, Node
+import socket
+import struct
+
+DBUF_DROP_TIMEOUT_SEC = "30s"
+DBUF_NUM_QUEUES = 10
+DBUF_MAX_PKTS_PER_QUEUE = 16
+
+
+def ip2long(ip):
+ """
+ Convert an IP string to long
+ """
+ packedIP = socket.inet_aton(ip)
+ return struct.unpack("!L", packedIP)[0]
+
+
+class IPv4Host(Host):
+ """Host that can be configured with an IPv4 gateway (default route).
+ """
+
+ def config(self, mac=None, ip=None, defaultRoute=None, lo='up', gw=None, **_params):
+ super(IPv4Host, self).config(mac, ip, defaultRoute, lo, **_params)
+ self.cmd('ip -4 addr flush dev %s' % self.defaultIntf())
+ self.cmd('ip -6 addr flush dev %s' % self.defaultIntf())
+ self.cmd('sysctl -w net.ipv4.ip_forward=0')
+ self.cmd('ip -4 link set up %s' % self.defaultIntf())
+ self.cmd('ip -4 addr add %s dev %s' % (ip, self.defaultIntf()))
+ if gw:
+ self.cmd('ip -4 route add default via %s' % gw)
+ # Disable offload
+ for attr in ["rx", "tx", "sg"]:
+ cmd = "/sbin/ethtool --offload %s %s off" % (self.defaultIntf(), attr)
+ self.cmd(cmd)
+
+ def updateIP():
+ return ip.split('/')[0]
+
+ self.defaultIntf().updateIP = updateIP
+
+
+class TaggedIPv4Host(Host):
+ """VLAN-tagged host that can be configured with an IPv4 gateway
+ (default route).
+ """
+ vlanIntf = None
+
+ def config(self, mac=None, ip=None, defaultRoute=None, lo='up', gw=None,
+ vlan=None, **_params):
+ super(TaggedIPv4Host, self).config(mac, ip, defaultRoute, lo, **_params)
+ self.vlanIntf = "%s.%s" % (self.defaultIntf(), vlan)
+ # Replace default interface with a tagged one
+ self.cmd('ip -4 addr flush dev %s' % self.defaultIntf())
+ self.cmd('ip -6 addr flush dev %s' % self.defaultIntf())
+ self.cmd('ip -4 link add link %s name %s type vlan id %s' % (
+ self.defaultIntf(), self.vlanIntf, vlan))
+ self.cmd('ip -4 link set up %s' % self.vlanIntf)
+ self.cmd('ip -4 addr add %s dev %s' % (ip, self.vlanIntf))
+ if gw:
+ self.cmd('ip -4 route add default via %s' % gw)
+
+ self.defaultIntf().name = self.vlanIntf
+ self.nameToIntf[self.vlanIntf] = self.defaultIntf()
+
+ # Disable offload
+ for attr in ["rx", "tx", "sg"]:
+ cmd = "/sbin/ethtool --offload %s %s off" % (
+ self.defaultIntf(), attr)
+ self.cmd(cmd)
+
+ def updateIP():
+ return ip.split('/')[0]
+
+ self.defaultIntf().updateIP = updateIP
+
+ def terminate(self):
+ self.cmd('ip -4 link remove link %s' % self.vlanIntf)
+ super(TaggedIPv4Host, self).terminate()
+
+class DbufHost(IPv4Host):
+
+ def __init__(self, name, inNamespace=False, **params):
+ super(DbufHost, self).__init__(name, inNamespace, **params)
+
+ def config(self, drainIp=None, drainMac=None, **_params):
+ super(DbufHost, self).config(**_params)
+ self.setDrainIpAndMac(self.defaultIntf(), drainIp, drainMac)
+ self.startDbuf()
+
+ def startDbuf(self):
+ args = map(str, [
+ "-max_queues",
+ DBUF_NUM_QUEUES,
+ "-max_packet_slots_per_queue",
+ DBUF_MAX_PKTS_PER_QUEUE,
+ "-queue_drop_timeout",
+ DBUF_DROP_TIMEOUT_SEC,
+ ])
+ # Send to background
+ cmd = '/usr/local/bin/dbuf %s > /tmp/dbuf_%s.log 2>&1 &' \
+ % (" ".join(args), self.name)
+ print(cmd)
+ self.cmd(cmd)
+
+ def setDrainIpAndMac(self, intf, drainIp=None, drainMac=None):
+ if drainIp:
+ self.setHostRoute(drainIp, intf)
+ if drainMac:
+ self.setARP(drainIp, drainMac)
+
+
+class DualHomedIpv4Host(Host):
+ """A dual homed host that can be configured with an IPv4 gateway (default route).
+ """
+
+ def __init__(self, name, **kwargs):
+ super(DualHomedIpv4Host, self).__init__(name, **kwargs)
+ self.bond0 = None
+
+ def config(self, ip=None, gw=None, **kwargs):
+ super(DualHomedIpv4Host, self).config(**kwargs)
+ intf0 = self.intfs[0].name
+ intf1 = self.intfs[1].name
+ self.bond0 = "%s-bond0" % self.name
+ self.cmd('modprobe bonding')
+ self.cmd('ip link add %s type bond miimon 100 mode balance-xor xmit_hash_policy layer2+3' %
+ self.bond0)
+ self.cmd('ip link set %s down' % intf0)
+ self.cmd('ip link set %s down' % intf1)
+ self.cmd('ip link set %s master %s' % (intf0, self.bond0))
+ self.cmd('ip link set %s master %s' % (intf1, self.bond0))
+ self.cmd('ip addr flush dev %s' % intf0)
+ self.cmd('ip addr flush dev %s' % intf1)
+ self.cmd('ip link set %s up' % self.bond0)
+
+ self.cmd('sysctl -w net.ipv4.ip_forward=0')
+ self.cmd('ip -4 addr add %s dev %s' % (ip, self.bond0))
+ if gw:
+ self.cmd('ip -4 route add default via %s' % gw)
+ # Disable offload
+ for attr in ["rx", "tx", "sg"]:
+ cmd = "/sbin/ethtool --offload %s %s off" % (self.defaultIntf(), attr)
+ self.cmd(cmd)
+
+ def terminate(self, **kwargs):
+ self.cmd('ip link set %s down' % self.bond0)
+ self.cmd('ip link delete %s' % self.bond0)
+ super(DualHomedIpv4Host, self).terminate()
+
+
+class DualHomedDbufHost(DualHomedIpv4Host, DbufHost):
+
+ def __init__(self, name, inNamespace=False, **params):
+ super(DualHomedDbufHost, self).__init__(name, inNamespace=inNamespace, **params)
+
+ def config(self, drainIp=None, drainMac=None, **_params):
+ super(DualHomedDbufHost, self).config(**_params)
+ self.setDrainIpAndMac(self.bond0, drainIp, drainMac)