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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
from common import process, netns, vlan, iface
import time
import math
# creates veth pair. if ok returns 0
def create_pair(name_a, name_b):
veth, out, err = process.run(
["ip", "link", "add", name_a, "type", "veth", "peer", "name", name_b]
)
print("veth.create: exit=%d out=%s err=%s" % (veth, out, err))
return veth
# put veth to netns. if ok returns 0
def assign_netns(veth, netns):
veth, out, err = process.run(["ip", "link", "set", veth, "netns", netns])
print("veth.assign_netns: exit=%d out=%s err=%s" % (veth, out, err))
return veth
# creates netns, creates veth pair and place second link to netns
# creates vlans over veth interfaces according to veth_pair_vlans_config
# return dict with 'netns', 'veth_a', 'veth_b'
def create_veth_pair_netns(veth_pair_vlans_config):
name = str(math.floor(time.time() * 1000) % 1000000)
netns_name = "N" + name
netns_status = netns.create(netns_name)
print("create_veth_pair_netns: netns_status=%d" % netns_status)
veth_a = "A" + name
veth_b = "B" + name
pair_status = create_pair(veth_a, veth_b)
print("create_veth_pair_netns: pair_status=%d" % pair_status)
iface.up(veth_a, None)
assign_status = assign_netns(veth_b, netns_name)
print("create_veth_pair_netns: assign_status=%d" % assign_status)
iface.up(veth_b, netns_name)
vlans_a = veth_pair_vlans_config["vlans_a"]
for vlan_num in vlans_a:
vlan.create(veth_a, vlan_num, None)
iface.up(veth_a + "." + str(vlan_num), None)
vlans_b = veth_pair_vlans_config["vlans_b"]
for vlan_num in vlans_b:
vlan.create(veth_b, vlan_num, netns_name)
iface.up(veth_b + "." + str(vlan_num), netns_name)
return {
"netns": netns_name,
"veth_a": veth_a,
"veth_b": veth_b,
"vlans_a": vlans_a,
"vlans_b": vlans_b,
}
# deletes veth pair and netns created by create_veth_pair_netns
def delete_veth_pair_netns(veth_pair_netns):
vlans_a = veth_pair_netns["vlans_a"]
veth_a = veth_pair_netns["veth_a"]
vlans_b = veth_pair_netns["vlans_b"]
veth_b = veth_pair_netns["veth_b"]
netns_name = veth_pair_netns["netns"]
# remove vlans on top of veth interface
for vlan_num in vlans_a:
iface.delete(veth_a + "." + str(vlan_num), None)
for vlan_num in vlans_b:
iface.delete(veth_b + "." + str(vlan_num), netns_name)
veth_status = iface.delete(veth_a, None)
print("delete_veth_pair_netns: veth_status=%d" % veth_status)
netns_status = netns.delete(netns_name)
print("delete_veth_pair_netns: netns_status=%d" % netns_status)
|
