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
|
#!/usr/bin/env python
__metaclass__ = type
import sys
import subprocess
from script_commands import (
helps,
run_subcommand,
UserError,
)
def tc(command):
"""Run a tc command under sudo.
:param tc: The remainder of the command (leaving out tc).
"""
subprocess.call('sudo tc ' + command, shell=True)
@helps(delay='Length of delay in miliseconds (each way).',
port='Port to induce delay on.')
def start(delay=500, port=443):
"""Add artificial latency to the lo interface on the specified port."""
tc('qdisc add dev lo root handle 1: prio')
tc('qdisc add dev lo parent 1:3 handle 30: netem delay %dms' % delay)
tc('filter add dev lo protocol ip parent 1:0 prio 3 u32 match ip'
' dport %d 0xffff flowid 1:3' % port)
tc('filter add dev lo protocol ip parent 1:0 prio 3 u32 match ip'
' sport %d 0xffff flowid 1:3' % port)
def stop():
"""Remove latency from the lo."""
tc('qdisc del dev lo root')
subcommands = {
'start': start,
'stop': stop,
}
if __name__ == "__main__":
try:
run_subcommand(subcommands, sys.argv[1:])
except UserError as e:
sys.stderr.write(str(e)+'\n')
|