lantz
f0c726a42f
This helps with virtualenv although it can open up another security hole if you end up using an unexpected python interpreter. Overall it seems to make sense to err on the side of usability but it's good to be aware of security. However, for the remaining utility scripts that require python 2, we explicitly note this with #!/usr/bin/python2.
62 lines
2.0 KiB
Python
Executable File
62 lines
2.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
limit.py: example of using link and CPU limits
|
|
"""
|
|
|
|
from mininet.net import Mininet
|
|
from mininet.link import TCIntf
|
|
from mininet.node import CPULimitedHost
|
|
from mininet.topolib import TreeTopo
|
|
from mininet.util import custom, quietRun
|
|
from mininet.log import setLogLevel, info
|
|
|
|
|
|
def testLinkLimit( net, bw ):
|
|
"Run bandwidth limit test"
|
|
info( '*** Testing network %.2f Mbps bandwidth limit\n' % bw )
|
|
net.iperf()
|
|
|
|
def limit( bw=10, cpu=.1 ):
|
|
"""Example/test of link and CPU bandwidth limits
|
|
bw: interface bandwidth limit in Mbps
|
|
cpu: cpu limit as fraction of overall CPU time"""
|
|
intf = custom( TCIntf, bw=bw )
|
|
myTopo = TreeTopo( depth=1, fanout=2 )
|
|
for sched in 'rt', 'cfs':
|
|
info( '*** Testing with', sched, 'bandwidth limiting\n' )
|
|
if sched == 'rt':
|
|
release = quietRun( 'uname -r' ).strip('\r\n')
|
|
output = quietRun( 'grep CONFIG_RT_GROUP_SCHED /boot/config-%s'
|
|
% release )
|
|
if output == '# CONFIG_RT_GROUP_SCHED is not set\n':
|
|
info( '*** RT Scheduler is not enabled in your kernel. '
|
|
'Skipping this test\n' )
|
|
continue
|
|
host = custom( CPULimitedHost, sched=sched, cpu=cpu )
|
|
net = Mininet( topo=myTopo, intf=intf, host=host, waitConnected=True )
|
|
net.start()
|
|
testLinkLimit( net, bw=bw )
|
|
net.runCpuLimitTest( cpu=cpu )
|
|
net.stop()
|
|
|
|
def verySimpleLimit( bw=150 ):
|
|
"Absurdly simple limiting test"
|
|
intf = custom( TCIntf, bw=bw )
|
|
net = Mininet( intf=intf, waitConnected=True )
|
|
h1, h2 = net.addHost( 'h1' ), net.addHost( 'h2' )
|
|
net.addLink( h1, h2 )
|
|
net.start()
|
|
net.pingAll()
|
|
net.iperf()
|
|
h1.cmdPrint( 'tc -s qdisc ls dev', h1.defaultIntf() )
|
|
h2.cmdPrint( 'tc -d class show dev', h2.defaultIntf() )
|
|
h1.cmdPrint( 'tc -s qdisc ls dev', h1.defaultIntf() )
|
|
h2.cmdPrint( 'tc -d class show dev', h2.defaultIntf() )
|
|
net.stop()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
setLogLevel( 'info' )
|
|
limit()
|