example-topos/simple_aqm.py at master · L4STeam/example-topos · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import subprocess

from ipmininet.clean import cleanup
from ipmininet.cli import IPCLI
from ipmininet.ipnet import IPNet
from ipmininet.iptopo import IPTopo
from ipmininet.router.config import OSPF
from ipmininet.router.config.base import RouterConfig
from ipmininet.router.config.ospf import OSPFRedistributedRoute


class MinimalAQMNet(IPTopo):
    """

    client1 ---- r1 ---- r2 ---- r3 ---- server1
                 |                |
    client2 ---- +                + ---- server2

    """
    def __init__(self, *args, **kwargs):
        super(MinimalAQMNet, self).__init__(*args, **kwargs)

    def build(self, *args, **kwargs):
        r1 = self.addRouter("r1", config=RouterConfig)
        r1.addDaemon(OSPF)
        r2 = self.addRouter("r2", config=RouterConfig)
        r2.addDaemon(OSPF)
        r3 = self.addRouter("r3", config=RouterConfig)
        r3.addDaemon(OSPF)
        self.addLink(r1, r2, params1={"ip": "10.0.0.1/24"}, params2={"ip": "10.0.0.2/24"})
        self.addLink(r2, r3, params1={"ip": "10.0.1.1/24"}, params2={"ip": "10.0.1.2/24"})

        client1 = self.addHost("client1")
        self.addLink(r1, client1, params1={"ip": "10.1.0.1/24"}, params2={"ip": "10.1.0.2/24"})
        client2 = self.addHost("client2")
        self.addLink(r1, client2, params1={"ip": "10.1.1.1/24"}, params2={"ip": "10.1.1.2/24"})

        server1 = self.addHost("server1")
        self.addLink(r3, server1, params1={"ip": "10.2.0.1/24"}, params2={"ip": "10.2.0.2/24"})
        server2 = self.addHost("server2")
        self.addLink(r3, server2, params1={"ip": "10.2.1.1/24"}, params2={"ip": "10.2.1.2/24"})

        super(MinimalAQMNet, self).build(*args, **kwargs)

if __name__ == "__main__":
    try:
        subprocess.check_call(["modprobe", "prague"])
        net = IPNet(topo=MinimalAQMNet(), allocate_IPs=False)
        net.start()

        # Set TCP Prague for client1 and server1 and cubic for client2 and server2 Congestion Control on client1 and server1
        net["client1"].cmd("sysctl -w net.ipv4.tcp_congestion_control=prague")
        net["server1"].cmd("sysctl -w net.ipv4.tcp_congestion_control=prague")
        net["client2"].cmd("sysctl -w net.ipv4.tcp_congestion_control=cubic")
        net["server2"].cmd("sysctl -w net.ipv4.tcp_congestion_control=cubic")

        # Setup AQM l4s_dualq
        net["r2"].cmd("tc qdisc del dev r2-eth0")
        net["r2"].cmd("tc qdisc add dev r2-eth0 root handle 1:0 htb default 1 direct_qlen 1000")
        net["r2"].cmd("tc class add dev r2-eth0 parent 1:0 classid 1:1 htb rate 40Mbit ceil 40Mbit")
        net["r2"].cmd("tc qdisc add dev r2-eth0 parent 1:1 handle 2:0 dualpi2 target 20ms l4s_ecn l_thresh 1ms l4s_dualq limit 1000")

        IPCLI(net)
        net.stop()
    finally:
        cleanup()