SDN-Routing/run.py at master · maxim-puchkov/SDN-Routing · 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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
#!/usr/bin/python
# -*- coding: utf-8 -*-

#
#  run.py
#  Test Network Demo
#
#  Created by mpuchkov on 2019-11-18.
#  Copyright © 2019 Maxim Puchkov. All rights reserved.
#

import sys

from TestNet.Topology import TestNetEnvironment
from TestNet.Utility import *


def __wait__( *commandList ):
	steps = len(commandList)
	for i in range( steps ):
		commandList[i]()
		display.prompt('\n\nPress <Return> to continue')
		try:
			x = input('')
		except:
			x = ''


# Run the demonstration
def __run__( argc, *argv ):

	#MARK: - Select Network Topology
	# 1. Create environment for creating, running, and testing
	#    network simulations from presets
	env = TestNetEnvironment()
	display.section("Cleaning up...")
	env.clean()
	# 2. Ask to input the index of a topology preset
	display.section("Select Network Topology")
	index = env.getNetworkTopologyIndex( argv )
	# 3. Select the preset with that index
	selectedTopoPreset = env.presetTopologies.select( index )
	display.message('Selected %s (index: %s)' % (selectedTopoPreset.displayName, index))


	#MARK: - Begin Network Simulation and do the First Test
	# 4. Initialize a new network with selected topology
	display.section("Starting network with topology preset")
	env.prepare( selectedTopoPreset )
	env.start()
	# 5. Launch the network
	display.section("Starting Configuration Interface")
	display.prompt('Input commands (optional):\n' + \
		' (E.g., track packets with \'sudo wireshark &\', \'tcpdump\', etc.)\n' + \
		' When you are ready to start the simulation tests, type \'exit\' or press CTRL-D.\n')
	env.startCLI()
	# 6. Test 1: Destination Host Unreachable
	display.section("Running tests...")
	(host1, host2) = env.getTestHosts()

	def do_test_1():
		test1 = env.nodeReachability( host1, host2 )
		display.section("Test 1: Hosts are unreachable...")
		display.highlight(test1)
	__wait__(do_test_1)

	#MARK: - Compute the Paths, Create Flow Tables, and do the Second Test
	# 7. Get the switches and link weights
	switches = env.net.topo.switches()
	linkWeights = env.net.topo._slinks
	# 8. Run LS Routing algorithm # 9. Add flow table entries
	env.updateRoutes( get_routing_decision, switches, linkWeights )

	def do_test_2():
	# 10. Test 2: Established connectivity
		display.section("Test 2: Hosts are now connected...")
		test2 = env.nodeReachability( host1, host2 )
		display.highlight(test2)
	__wait__(do_test_2)


	#MARK: - Disable one of the Links and do the Third Test
	# 11. Disable a link
	link = env.net.topo.problemLink
	display.section("Changing network conditions")
	display.message('Problem link in this simulation is {%s, %s}' % link )
	env.simulateLinkProblem(link)
	n1 = int( link[0][1:] )
	n2 = int( link[1][1:] )
	for (prev1, prev2, w) in linkWeights:
		if ((prev1 == n1 and prev2 == n2) or (prev1 == n2 and prev2 == n1)):
			linkWeights.remove( (prev1, prev2, w) )
			linkWeights.append( (str(n1), str(n2), 1000) )
			print('Increasing the cost of link %s to 1000: %s' % ([(prev1, prev2, w)], [(n1, n2, 1000)]) )
			break
	# 12. Test 3: Hosts
	def do_test_3():
		display.section("Test 3: Link failure affects optimal routes...")
		test3 = env.nodeReachability( host1, host2 )
		display.highlight(test3)
	__wait__(do_test_3)


	#MARK: - Recompute the Least-Cost Paths and do the Fourth Test
	# 13.
	env.updateRoutes( get_routing_decision, switches, linkWeights )
	# 14. Test 4:
	def do_test_4():
		display.section("Test 4: Hosts can communicate again...")
		test4 = env.nodeReachability(host1, host2)
		display.highlight(test4)
	__wait__(do_test_4)


	#MARK: - Command Line Interface
	# 15. Start the CLI to run other tests
	display.section("Starting Command Line Interface")
	print('Custom debugging commands:')
	customList = colorizeEach(['stats', 'flows', 'paths', 'routes', 'costs', 'weights', 'all'], UITextStyle.Color.magenta)
	for c in customList:
		print ('%s   ' % c),
	print('')
	env.startCLI()


	#MARK: - Stop Network Simulation
	# 15. Stop network
	display.section("Shutting down")
	env.stop()


if __name__ == '__main__':
	argc = len( sys.argv )
	__run__( argc, sys.argv )