csit/libraries/MininetTopo/vlan_vtn_test.py

   1 #!/usr/bin/env python
   2
   3 from mininet.node import Host
   4 from mininet.topo import Topo
   5
   6
   7 class VLANHost(Host):
   8     """Host connected to VLAN interface"""
   9
  10     def config(self, vlan=100, **params):
  11         """Configure VLANHost according to (optional) parameters
  12         vlan: VLAN ID for default interface"""
  13
  14         hostid = super(Host, self).config(**params)
  15
  16         intf = self.defaultIntf()
  17         # remove IP from default, "physical" interface
  18         self.cmd("ifconfig %s inet 0" % intf)
  19         # create VLAN interface
  20         self.cmd("vconfig add %s %d" % (intf, vlan))
  21         # assign the host's IP to the VLAN interface
  22         self.cmd("ifconfig %s.%d inet %s" % (intf, vlan, params["ip"]))
  23         # update the intf name and host's intf map
  24         new_name = "%s.%d" % (intf, vlan)
  25         # update the (Mininet) interface to refer to VLAN interface name
  26         intf.name = new_name
  27         # add VLAN interface to host's name to intf map
  28         self.nameToIntf[new_name] = intf
  29         return hostid
  30
  31
  32 class VlanTopo(Topo):
  33
  34     """Simple topology example."""
  35
  36     def __init__(self):
  37         """Create custom topo."""
  38
  39         # Initialize topology
  40         Topo.__init__(self)
  41
  42         # Add hosts and switches
  43         host1 = self.addHost("h1", cls=VLANHost, vlan=200)
  44         host2 = self.addHost("h2", cls=VLANHost, vlan=300)
  45         host3 = self.addHost("h3", cls=VLANHost, vlan=200)
  46         host4 = self.addHost("h4", cls=VLANHost, vlan=300)
  47         host5 = self.addHost("h5", cls=VLANHost, vlan=200)
  48         host6 = self.addHost("h6", cls=VLANHost, vlan=300)
  49
  50         s1 = self.addSwitch("s1")
  51         s2 = self.addSwitch("s2")
  52         s3 = self.addSwitch("s3")
  53
  54         self.addLink(s1, s2)
  55         self.addLink(s2, host1)
  56         self.addLink(s2, host2)
  57         self.addLink(s2, host3)
  58         self.addLink(s1, s3)
  59         self.addLink(s3, host4)
  60         self.addLink(s3, host5)
  61         self.addLink(s3, host6)
  62
  63
  64 topos = {"vlantopo": (lambda: VlanTopo())}