blob: 5d955e5187e7dd9c30b5a48caff6f5af0abadbd6 [file] [log] [blame]
Brian O'Connoreb27c452014-12-07 02:43:58 -08001#!/usr/bin/python
2
3'''
4Notes:
5
6This file contains classes and methods useful for integrating LincOE with Mininet,
Marc De Leenheer6ff97642015-07-08 19:21:16 +00007such as startOE, stopOE, LINCLink, and OpticalSwitch
Brian O'Connoreb27c452014-12-07 02:43:58 -08008
9- $ONOS_ROOT ust be set
10- Need to run with sudo -E to preserve ONOS_ROOT env var
11- We assume LINC-Config-Generator is named LINC-Config-Generator
12- We also assume linc-oe is named linc-oe
13- LINC-config-generator and linc-oe must be subdirectories of the user's
14 home directory
15
16 TODO
17 -----------
18 - clean up files after runtime
19 - maybe save the old files in a separate directory?
20 - modify script to allow startOE to run before net.start()
21 - add ONOS as a controller in script
22
23 Usage:
24 ------------
Marc De Leenheer6ff97642015-07-08 19:21:16 +000025 - import LINCLink and OpticalSwitch from this module
Brian O'Connoreb27c452014-12-07 02:43:58 -080026 - import startOE and stopOE from this module
27 - create topology as you would a normal topology. when
Marc De Leenheer6ff97642015-07-08 19:21:16 +000028 to an optical switch with topo.addLink, always specify cls=LINCLink
Brian O'Connoreb27c452014-12-07 02:43:58 -080029 - when creating an optical switch, use cls=OpticalSwitch in topo.addSwitch
30 - for annotations on links and switches, a dictionary must be passed in as
31 the annotations argument
32 - startOE must be run AFTER net.start() with net as an argument.
33 - stopOE can be run at any time
34
35I created a separate function to start lincOE to avoid subclassing Mininet.
36In case anyone wants to write something that DOES subclass Mininet, I
37thought I would outline how:
38
39If we want an object that starts lincOE within the mininet class itself,
40we need to add another object to Mininet that contains all of the json object
41information for each switch. We would still subclass switch and link, but these
42classes would basically be dummy classes that store their own json information
43in the Mininet class object. We may also change the default switch class to add
44it's tap interfaces from lincOE during startup. The start() method for mininet would
45grab all of the information from these switches and links, write configuration files
46for lincOE using the json module, start lincOE, then run the start methodfor each
47switch. The new start() method for each switch would parse through the sys.config
48file that was created and find the tap interface it needs to connect to, similar
49to the findTap function that I currently use. After all of the controllers and
50switches have been started, the new Mininet start() method should also push the
51Topology configuration file to ONOS.
52
53'''
Marc De Leenheer6ff97642015-07-08 19:21:16 +000054import sys
Brian O'Connoreb27c452014-12-07 02:43:58 -080055import re
56import json
57import os
58from time import sleep
Marc De Leenheer6ff97642015-07-08 19:21:16 +000059import urllib2
Brian O'Connoreb27c452014-12-07 02:43:58 -080060
61from mininet.node import Switch, RemoteController
62from mininet.topo import Topo
63from mininet.util import quietRun
64from mininet.net import Mininet
65from mininet.log import setLogLevel, info, error, warn
66from mininet.link import Link, Intf
67from mininet.cli import CLI
68
Marc De Leenheer6ff97642015-07-08 19:21:16 +000069# Sleep time and timeout values in seconds
70SLEEP_TIME = 2
71TIMEOUT = 60
Marc De Leenheer16f857b2015-05-05 20:50:24 -070072
Marc De Leenheer6ff97642015-07-08 19:21:16 +000073class OpticalSwitch(Switch):
74 """
75 For now, same as Switch class.
76 """
77 pass
78
79class OpticalIntf(Intf):
80 """
81 For now,same as Intf class.
82 """
83 pass
84
85class OpticalLink(Link):
86 """
87 For now, same as Link.
88 """
89 pass
90
91class LINCSwitch(OpticalSwitch):
92 """
93 LINCSwitch class
94 """
95 # FIXME:Sometimes LINC doesn't remove pipes and on restart increase the pipe
96 # number from erlang.pipe.1.* to erlang.pipe.2.*, so should read and write
97 # from latest pipe files. For now we are removing all the pipes before
98 # starting LINC.
99 ### User Name ###
100 user = os.getlogin()
101 ### pipes ###
102 readPipe = "/tmp/home/{}/linc-oe/rel/linc/erlang.pipe.1.r".format(user)
103 writePipe = "/tmp/home/{}/linc-oe/rel/linc/erlang.pipe.1.w".format(user)
104 ### sys.config path ###
105 sysConfig = "/home/{}/linc-oe/rel/linc/releases/1.0/sys.config".format(user)
106 ### method, mapping dpid to LINC switchId ###
107 @staticmethod
108 def dpids_to_ids(sysConfig):
109 '''
110 return the dict containing switch dpids as key and LINC switch id as values
111 '''
112 dpids_to_ids = {}
113 fd = None
114 try:
115 with open(sysConfig, 'r', 0) as fd:
116 switch_id = 1
117 for line in fd:
118 dpid = re.search(r'([0-9A-Fa-f]{2}[:-]){7}([0-9A-Fa-f]{2})+', line, re.I)
119 if dpid:
120 dpids_to_ids[dpid.group().replace(':', '')] = switch_id
121 switch_id += 1
122 return dpids_to_ids
123 except:
124 print "Error working with {}\nError: {}\n".format(sysConfig, sys.exc_info())
125 fd.close()
126 return None
127 ### dict of containing dpids as key and corresponding LINC switchId as values ###
128 dpidsToLINCSwitchId = dpids_to_ids.__func__(sysConfig)
129 @staticmethod
130 def findDir(directory, userName):
131 "finds and returns the path of any directory in the user's home directory"
132 homeDir = '/home/' + userName
133 Dir = quietRun('find %s -maxdepth 1 -name %s -type d' % (homeDir, directory)).strip('\n')
134 DirList = Dir.split('\n')
135 if not Dir:
136 return None
137 elif len(DirList) > 1 :
138 warn('***WARNING: Found multiple instances of %s; using %s\n'
139 % (directory, DirList[ 0 ]))
140 return DirList[ 0 ]
141 else:
142 return Dir
143 ### ONOS Directory ###
144 try:
145 onosDir = os.environ[ 'ONOS_ROOT' ]
146 except:
147 onosDir = findDir('onos', user)
148 if not onosDir:
149 error('Please set ONOS_ROOT environment variable!\n')
150 else:
151 os.environ[ 'ONOS_ROOT' ] = onosDir
Ayaka Koshibeb0d70582015-09-11 11:29:36 -0700152 ### REST USER/PASS ###
153 try:
154 restUser = os.environ[ 'ONOS_WEB_USER' ]
155 restPass = os.environ[ 'ONOS_WEB_PASS' ]
156 except:
157 error('***WARNING: $ONOS_WEB_USER and $ONOS_WEB_PASS aren\'t set!\n')
158 error('***WARNING: Setting (probably) sane WEB user/pass values\n')
159 restUser = 'onos'
160 restPass = 'rocks'
161 os.environ[ 'ONOS_WEB_USER' ] = restUser
162 os.environ[ 'ONOS_WEB_PASS' ] = restPass
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000163 ### LINC-directory
164 lincDir = findDir.__func__('linc-oe', user)
165 if not lincDir:
166 error("***ERROR: Could not find linc-oe in user's home directory\n")
167 ### LINC config generator directory###
168 configGen = findDir.__func__('LINC-config-generator', user)
169 if not configGen:
170 error("***ERROR: Could not find LINC-config-generator in user's home directory\n")
171 # list of all the controllers
172 controllers = None
173 def __init__(self, name, dpid=None, allowed=True,
174 switchType='ROADM', topo=None, annotations={}, controller=None, **params):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800175 params[ 'inNamespace' ] = False
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000176 Switch.__init__(self, name, dpid=dpid, **params)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800177 self.name = name
178 self.annotations = annotations
179 self.allowed = allowed
180 self.switchType = switchType
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000181 self.configDict = {} # dictionary that holds all of the JSON configuration data
182 self.crossConnects = []
183 self.deletedCrossConnects = []
184 self.controller = controller
185 self.lincId = self._get_linc_id() # use to communicate with LINC
186 self.lincStarted = False
Brian O'Connoreb27c452014-12-07 02:43:58 -0800187
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000188 def start(self, *opts, **params):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800189 '''Instead of starting a virtual switch, we build the JSON
190 dictionary for the emulated optical switch'''
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000191 # TODO:Once LINC has the ability to spawn network element dynamically
192 # we need to use this method to spawn new logical LINC switch rather then
193 # bulding JSON.
194 # if LINC is started then we can start and stop logical switches else create JSON
195 if self.lincStarted:
196 return self.start_oe()
Brian O'Connoreb27c452014-12-07 02:43:58 -0800197 self.configDict[ 'uri' ] = 'of:' + self.dpid
198 self.configDict[ 'annotations' ] = self.annotations
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000199 self.configDict[ 'annotations' ].setdefault('name', self.name)
200 self.configDict[ 'hw' ] = 'LINC-OE'
Brian O'Connoreb27c452014-12-07 02:43:58 -0800201 self.configDict[ 'mfr' ] = 'Linc'
202 self.configDict[ 'mac' ] = 'ffffffffffff' + self.dpid[-2] + self.dpid[-1]
203 self.configDict[ 'type' ] = self.switchType
204 self.configDict[ 'ports' ] = []
205 for port, intf in self.intfs.items():
206 if intf.name == 'lo':
207 continue
208 else:
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000209 self.configDict[ 'ports' ].append(intf.json())
210 self.lincStarted = True
fahad44e62c72015-03-04 14:55:35 -0800211
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000212 def stop(self, deleteIntfs=False):
213 '''
214 stop the existing switch
215 '''
216 # TODO:Add support for deleteIntf
217 self.stop_oe()
fahad44e62c72015-03-04 14:55:35 -0800218
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000219 def dpctl( self, *args ):
220 "Run dpctl command: ignore for now"
fahad44e62c72015-03-04 14:55:35 -0800221 pass
222
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000223 def write_to_cli(self, command):
224 '''
225 send command to LINC
226 '''
227 fd = None
228 try:
229 fd = open(self.writePipe, 'w', 0)
230 fd.write(command)
231 fd.close()
232 except:
233 print "Error working with {}\nError: {}\n".format(self.writePipe, sys.exc_info())
234 if fd:
235 fd.close()
Nikhil Cheerla5f8f8f02015-07-07 16:01:17 -0700236
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000237 def read_from_cli(self):
238 '''
239 read the output from the LINC CLI
240 '''
241 response = None
242 fd = None
243 try:
244 fd = open(self.readPipe, 'r', 0)
245 fcntl.fcntl(fd, fcntl.F_SETFL, os.O_NONBLOCK) # for non-blocking read
246 # FIXME:Due to non-blocking read most for the time we read nothing
247 response = fd.read()
248 fd.close()
249 except :
250 # print "Error working with {}\nError: {}\n".format(self.readPipe, sys.exc_info())
251 if fd:
252 fd.close()
253 return response
254
255 def _get_linc_id(self):
256 '''
257 return the corresponding LINC switchId.
258 '''
259 return LINCSwitch.dpidsToLINCSwitchId.get(self.dpid)
260 #--------------------------------------------------------------------------
261 # LINC CLI commands
262 #--------------------------------------------------------------------------
263 def start_oe(self):
264 '''
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700265 existing LINC switch
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000266 '''
267 #starting Switch
268 cmd = "linc:start_switch({}).\r\n".format(self.lincId)
269 self.write_to_cli(cmd)
270 #hanlding taps interfaces related to the switch
271 crossConnectJSON = {}
272 linkConfig = []
273 for i in range(0,len(self.deletedCrossConnects)):
274 crossConnect = self.deletedCrossConnects.pop()
275 tap = None
276 if isinstance(crossConnect.intf1.node, LINCSwitch):
277 intf = crossConnect.intf2
278 tapPort = crossConnect.intf1.port
279 else:
280 intf = crossConnect.intf1
281 tapPort = crossConnect.intf2.port
282 tap = LINCSwitch.findTap(self, tapPort)
283 if tap:
284 LINCSwitch.setupInts([tap])
285 intf.node.attach(tap)
286 self.crossConnects.append(crossConnect)
287 linkConfig.append(crossConnect.json())
288 #Sending crossConnect info to the ONOS.
289 crossConnectJSON['links'] = linkConfig
290 with open("crossConnect.json", 'w') as fd:
291 json.dump(crossConnectJSON, fd, indent=4, separators=(',', ': '))
292 info('*** Pushing crossConnect.json to ONOS\n')
293 output = quietRun('%s/tools/test/bin/onos-topo-cfg %s\
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700294 Topology.json network/configuration/' % (self.onosDir, self.controllers[ 0 ].ip), shell=True)
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000295
296 def stop_oe(self):
297 '''
298 stop the existing LINC switch
299 '''
300 cmd = "linc:stop_switch({}).\r\n".format(self.lincId)
301 self.write_to_cli(cmd)
302 #handling taps if any
303 for i in range(0, len(self.crossConnects)):
304 crossConnect = self.crossConnects.pop()
305 if isinstance(crossConnect.intf1.node, LINCSwitch):
306 intf = crossConnect.intf2
307 tapPort = crossConnect.intf1.port
308 else:
309 intf = crossConnect.intf1
310 tapPort = crossConnect.intf2.port
311 intf.node.detach(LINCSwitch.findTap(self, tapPort))
312 self.deletedCrossConnects.append(crossConnect)
313
314 def w_port_up(self, port):
315 '''
316 port_up
317 '''
318 cmd = "linc:port_up({},{}).\r\n".format(self.lincId, port)
319 self.write_to_cli(cmd)
320
321 def w_port_down(self, port):
322 '''
323 port_down
324 '''
325 cmd = "linc:port_down({},{}).\r\n".format(self.lincId, port)
326 self.write_to_cli(cmd)
327
328 # helper functions
329 @staticmethod
330 def switchJSON(switch):
331 "Returns the json configuration for a packet switch"
332 configDict = {}
333 configDict[ 'uri' ] = 'of:' + switch.dpid
334 configDict[ 'mac' ] = quietRun('cat /sys/class/net/%s/address' % switch.name).strip('\n').translate(None, ':')
335 configDict[ 'hw' ] = 'PK' # FIXME what about OVS?
336 configDict[ 'mfr' ] = 'Linc' # FIXME what about OVS?
337 configDict[ 'type' ] = 'SWITCH' # FIXME what about OVS?
338 annotations = switch.params.get('annotations', {})
339 annotations.setdefault('name', switch.name)
340 configDict[ 'annotations' ] = annotations
341 ports = []
342 for port, intf in switch.intfs.items():
343 if intf.name == 'lo':
344 continue
345 portDict = {}
346 portDict[ 'port' ] = port
Ayaka Koshibe144bab02015-10-22 12:56:59 -0700347 portType = 'COPPER'
348 if isinstance(intf.link, LINCLink):
349 portType = 'OCH' if intf.link.isCrossConnect else 'OMS'
350 portDict[ 'type' ] = portType
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000351 intfList = [ intf.link.intf1, intf.link.intf2 ]
352 intfList.remove(intf)
353 portDict[ 'speed' ] = intfList[ 0 ].speed if isinstance(intf.link, LINCLink) else 0
354 ports.append(portDict)
355 configDict[ 'ports' ] = ports
356 return configDict
357
358 @staticmethod
359 def bootOE(net):
360 "Start the LINC optical emulator within a mininet instance"
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700361 LINCSwitch.opticalJSON = {}
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000362 linkConfig = []
363 devices = []
364 #setting up the controllers for LINCSwitch class
365 LINCSwitch.controllers = net.controllers
366
367 for switch in net.switches:
368 if isinstance(switch, OpticalSwitch):
369 devices.append(switch.json())
370 else:
371 devices.append(LINCSwitch.switchJSON(switch))
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700372 LINCSwitch.opticalJSON[ 'devices' ] = devices
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000373
374 for link in net.links:
375 if isinstance(link, LINCLink) :
376 linkConfig.append(link.json())
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700377 LINCSwitch.opticalJSON[ 'links' ] = linkConfig
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000378
379 info('*** Writing Topology.json file\n')
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700380 topoJSON = LINCSwitch.makeTopoJSON()
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000381 with open('Topology.json', 'w') as outfile:
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700382 json.dump(topoJSON, outfile, indent=4, separators=(',', ': '))
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000383
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700384 info('*** Converting Topology.json to linc-oe format (TopoConfig.json) file (no oecfg) \n')
385
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700386 topoConfigJson = {}
387 dpIdToName = {}
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700388
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700389 topoConfigJson["switchConfig"] = LINCSwitch.getSwitchConfig(dpIdToName)
390 topoConfigJson["linkConfig"] = LINCSwitch.getLinkConfig(dpIdToName)
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700391
392 #Writing to TopoConfig.json
393 with open( 'TopoConfig.json', 'w' ) as outfile:
394 json.dump( topoConfigJson, outfile, indent=4, separators=(',', ': ') )
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000395
396 info('*** Creating sys.config...\n')
397 output = quietRun('%s/config_generator TopoConfig.json %s/sys.config.template %s %s'
398 % (LINCSwitch.configGen, LINCSwitch.configGen, LINCSwitch.controllers[ 0 ].ip, LINCSwitch.controllers[ 0 ].port), shell=True)
399 if output:
400 error('***ERROR: Error creating sys.config file: %s\n' % output)
401 return False
402
403 info ('*** Setting multiple controllers in sys.config...\n')
404 searchStr = '\[{"Switch.*$'
405 ctrlStr = ''
406 for index in range(len(LINCSwitch.controllers)):
407 ctrlStr += '{"Switch%d-Controller","%s",%d,tcp},' % (index, net.controllers[index].ip, net.controllers[index].port)
408 replaceStr = '[%s]},' % ctrlStr[:-1] # Cut off last comma
409 sedCmd = 'sed -i \'s/%s/%s/\' sys.config' % (searchStr, replaceStr)
410 output = quietRun(sedCmd, shell=True)
411
412 info('*** Copying sys.config to linc-oe directory: ', output + '\n')
413 output = quietRun('cp -v sys.config %s/rel/linc/releases/1.0/' % LINCSwitch.lincDir, shell=True).strip('\n')
414 info(output + '\n')
415
416 info('*** Adding taps and bringing them up...\n')
417 LINCSwitch.setupInts(LINCSwitch.getTaps())
418
419 info('*** removing pipes if any \n')
420 quietRun('rm /tmp/home/%s/linc-oe/rel/linc/*' % LINCSwitch.user, shell=True)
421
422 info('*** Starting linc OE...\n')
423 output = quietRun('%s/rel/linc/bin/linc start' % LINCSwitch.lincDir, shell=True)
424 if output:
425 error('***ERROR: LINC-OE: %s' % output + '\n')
426 quietRun('%s/rel/linc/bin/linc stop' % LINCSwitch.lincDir, shell=True)
427 return False
428
429 info('*** Waiting for linc-oe to start...\n')
430 LINCSwitch.waitStarted(net)
431
432 info('*** Adding cross-connect (tap) interfaces to packet switches...\n')
433 for link in net.links:
434 if isinstance(link, LINCLink):
435 if link.annotations[ 'optical.type' ] == 'cross-connect':
436 for intf in [ link.intf1, link.intf2 ]:
437 if not isinstance(intf, LINCIntf):
438 intfList = [ intf.link.intf1, intf.link.intf2 ]
439 intfList.remove(intf)
440 intf2 = intfList[ 0 ]
441 intf.node.attach(LINCSwitch.findTap(intf2.node, intf2.node.ports[ intf2 ]))
442
443 info('*** Waiting for all devices to be available in ONOS...\n')
444 url = 'http://%s:8181/onos/v1/devices' % LINCSwitch.controllers[0].ip
445 time = 0
Ayaka Koshibec9eed382015-09-03 14:38:55 -0700446 # Set up password authentication
447 pw_mgr = urllib2.HTTPPasswordMgrWithDefaultRealm()
Ayaka Koshibeb0d70582015-09-11 11:29:36 -0700448 pw_mgr.add_password(None, url, LINCSwitch.restUser, LINCSwitch.restPass)
Ayaka Koshibec9eed382015-09-03 14:38:55 -0700449 handler = urllib2.HTTPBasicAuthHandler(pw_mgr)
450 opener = urllib2.build_opener(handler)
451 opener.open(url)
452 urllib2.install_opener(opener)
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000453 while True:
454 response = json.load(urllib2.urlopen(url))
455 devs = response.get('devices')
456
457 # Wait for all devices to be registered
458 if (len(devices) != len(devs)):
459 continue
460
461 # Wait for all devices to available
462 available = True
463 for d in devs:
464 available &= d['available']
465 if available:
466 break
467
468 if (time >= TIMEOUT):
469 error('***ERROR: ONOS did not register devices within %s seconds\n' % TIMEOUT)
470 break
471
472 time += SLEEP_TIME
473 sleep(SLEEP_TIME)
474
475 info('*** Pushing Topology.json to ONOS\n')
476 for index in range(len(LINCSwitch.controllers)):
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700477 output = quietRun('%s/tools/test/bin/onos-topo-cfg %s Topology.json network/configuration/ &'\
478 % (LINCSwitch.onosDir, LINCSwitch.controllers[ index ].ip), shell=True)
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000479 # successful output contains the two characters '{}'
480 # if there is more output than this, there is an issue
481 if output.strip('{}'):
482 warn('***WARNING: Could not push topology file to ONOS: %s\n' % output)
483
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700484 #converts node ids to linc-oe format, with colons every two chars
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700485 @staticmethod
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700486 def dpId(id):
487 nodeDpid = ""
488 id = id.split("/", 1)[0]
489 for i in range(3, len(id) - 1, 2):
490 nodeDpid += (id[i:(i + 2):]) + ":"
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700491 return nodeDpid[0:-1];
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700492
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700493 @staticmethod
Ayaka Koshibed88b81d2015-09-17 17:52:27 -0700494 def makeTopoJSON():
495 """
496 Builds ONOS network conifg system compatible dicts to be written as Topology.json file.
497 """
498 topology = {}
499 links = {}
500 devices = {}
501 ports = {}
502
503 for switch in LINCSwitch.opticalJSON[ 'devices' ]:
504 # build device entries - keyed on uri (DPID) and config key 'basic'
505 devDict = {}
506 devDict[ 'driver' ] = switch[ 'hw' ]
507 devDict[ 'mfr' ] = switch[ 'mfr' ]
508 devDict[ 'mac' ] = switch[ 'mac' ]
509 devDict[ 'type' ] = switch[ 'type' ]
510 devDict.update(switch[ 'annotations' ])
511
512 devSubj = switch[ 'uri' ]
513 devices[ devSubj ] = { 'basic': devDict }
514
515 # build port entries - keyed on "uri/port" and config key 'optical'
516 for port in switch[ 'ports' ]:
517 portSubj = devSubj + '/' + str(port[ 'port' ])
518 ports[ portSubj ] = { 'optical': port }
519
520 # build link entries - keyed on "uri/port-uri/port" and config key 'basic'
521 for link in LINCSwitch.opticalJSON[ 'links' ]:
522 linkDict = {}
523 linkDict[ 'type' ] = link[ 'type' ]
524 linkDict.update(link[ 'annotations' ])
525
526 linkSubj = link[ 'src' ] + '-' + link[ 'dst' ]
527 links[ linkSubj ] = { 'basic': linkDict }
528
529 topology[ 'links' ] = links
530 topology[ 'devices' ] = devices
531 topology[ 'ports' ] = ports
532
533 return topology
534
535 @staticmethod
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700536 def getSwitchConfig (dpIdToName):
537 switchConfig = [];
538 #Iterate through all switches and convert the ROADM switches to linc-oe format
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700539 for switch in LINCSwitch.opticalJSON["devices"]:
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700540 if switch.get("type", "none") == "ROADM":
541 builtSwitch = {}
542
543 #set basic switch params based on annotations
544 builtSwitch["allowed"] = True;
545 builtSwitch["latitude"] = switch["annotations"].get("latitude", 0.0);
546 builtSwitch["longitude"] = switch["annotations"].get("longitude", 0.0);
547
548 #assumed that all switches have this entry
549 nodeId = switch["uri"]
550
551 #convert the nodeId to linc-oe format
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700552 nodeDpid = LINCSwitch.dpId(nodeId);
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700553
554 builtSwitch["name"] = switch.get("name", "none");
555
556 #keep track of the name corresponding to each switch dpid
557 dpIdToName[nodeDpid] = builtSwitch["name"];
558
559 builtSwitch["nodeDpid"] = nodeDpid
560
561 #set switch params and type
562 builtSwitch["params"] = {};
563 builtSwitch["params"]["numregens"] = switch["annotations"].get("optical.regens", 0);
564 builtSwitch["type"] = "Roadm"
565
566 #append to list of switches
567 switchConfig.append(builtSwitch);
568 return switchConfig
569
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700570 @staticmethod
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700571 def getLinkConfig (dpIdToName):
572 newLinkConfig = [];
573 #Iterate through all optical links and convert them to linc-oe format
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700574 for link in LINCSwitch.opticalJSON["links"]:
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700575 if link.get("type", "none") == "OPTICAL":
576 builtLink = {}
577
578 #set basic link params for src and dst
579 builtLink["allowed"] = True;
Nikhil Cheerlaf9391e12015-07-21 10:49:12 -0700580 builtLink["nodeDpid1"] = LINCSwitch.dpId(link["src"])
581 builtLink["nodeDpid2"] = LINCSwitch.dpId(link["dst"])
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700582
583 #set more params such as name/bandwidth/port/waves if they exist
584 params = {}
585 params["nodeName1"] = dpIdToName.get(builtLink["nodeDpid1"], "none")
586 params["nodeName2"] = dpIdToName.get(builtLink["nodeDpid2"], "none")
587
588 params["port1"] = int(link["src"].split("/")[1])
589 params["port2"] = int(link["dst"].split("/")[1])
590
591 if "bandwidth" in link["annotations"]:
592 params["bandwidth"] = link["annotations"]["bandwidth"]
593
594 if "optical.waves" in link["annotations"]:
595 params["numWaves"] = link["annotations"]["optical.waves"]
596
597 builtLink["params"] = params
598
599 #set type of link (WDM or pktOpt)
600 if link["annotations"].get("optical.type", "cross-connect") == "WDM":
601 builtLink["type"] = "wdmLink"
602 else:
603 builtLink["type"] = "pktOptLink"
604
605 newLinkConfig.append(builtLink);
606 return newLinkConfig
607
608
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000609 @staticmethod
610 def waitStarted(net, timeout=TIMEOUT):
611 "wait until all tap interfaces are available"
612 tapCount = 0
613 time = 0
614 for link in net.links:
615 if isinstance(link, LINCLink):
616 if link.annotations[ 'optical.type' ] == 'cross-connect':
617 tapCount += 1
618
619 while True:
620 if str(tapCount) == quietRun('ip addr | grep tap | wc -l', shell=True).strip('\n'):
621 return True
622 if timeout:
623 if time >= TIMEOUT:
624 error('***ERROR: LINC OE did not start within %s seconds\n' % TIMEOUT)
625 return False
626 time += SLEEP_TIME
627 sleep(SLEEP_TIME)
628
629 @staticmethod
630 def shutdownOE():
631 "stop the optical emulator"
632 info('*** Stopping linc OE...\n')
633 quietRun('%s/rel/linc/bin/linc stop' % LINCSwitch.lincDir, shell=True)
634
635 @staticmethod
636 def setupInts(intfs):
637 '''
638 add taps and bring them up.
639 '''
640 for i in intfs:
641 quietRun('ip tuntap add dev %s mode tap' % i)
642 quietRun('ip link set dev %s up' % i)
643 info('*** Intf %s set\n' % i)
644
645 @staticmethod
646 def getTaps(path=None):
647 '''
648 return list of all the tops in sys.config
649 '''
650 if path is None:
651 path = '%s/rel/linc/releases/1.0/sys.config' % LINCSwitch.lincDir
652 fd = open(path, 'r', 0)
653 sys_data = fd.read()
654 taps = re.findall('tap\d+', sys_data)
655 fd.close()
656 return taps
657
658 @staticmethod
659 def findUser():
660 "Try to return logged-in (usually non-root) user"
661 try:
662 # If we're running sudo
663 return os.environ[ 'SUDO_USER' ]
664 except:
665 try:
666 # Logged-in user (if we have a tty)
667 return quietRun('who am i').split()[ 0 ]
668 except:
669 # Give up and return effective user
670 return quietRun('whoami')
671
672
673 @staticmethod
674 def findTap(node, port, path=None):
675 '''utility function to parse through a sys.config
676 file to find tap interfaces for a switch'''
677 switch = False
678 portLine = ''
679 intfLines = []
680
681 if path is None:
682 path = '%s/rel/linc/releases/1.0/sys.config' % LINCSwitch.lincDir
683
684 with open(path) as f:
685 for line in f:
686 if 'tap' in line:
687 intfLines.append(line)
688 if node.dpid in line.translate(None, ':'):
689 switch = True
690 continue
691 if switch:
692 if 'switch' in line:
693 switch = False
694 if 'port_no,%s}' % port in line:
695 portLine = line
696 break
697
698 if portLine:
699 m = re.search('port,\d+', portLine)
700 port = m.group(0).split(',')[ 1 ]
701 else:
702 error('***ERROR: Could not find any ports in sys.config\n')
703 return
704
705 for intfLine in intfLines:
706 if 'port,%s' % port in intfLine:
707 return re.findall('tap\d+', intfLine)[ 0 ]
708
709 def json(self):
710 "return json configuration dictionary for switch"
711 return self.configDict
712
713 def terminate(self):
714 pass
715
Nikhil Cheerla7d7f3be2015-07-09 13:13:40 -0700716
717
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000718class LINCLink(Link):
719 """
720 LINC link class
721 """
722 def __init__(self, node1, node2, port1=None, port2=None, allowed=True,
Brian O'Connoreb27c452014-12-07 02:43:58 -0800723 intfName1=None, intfName2=None, linkType='OPTICAL',
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000724 annotations={}, speed1=0, speed2=0, **params):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800725 "Creates a dummy link without a virtual ethernet pair."
726 self.allowed = allowed
727 self.annotations = annotations
728 self.linkType = linkType
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000729 self.port1 = port1
730 self.port2 = port2
Brian O'Connoreb27c452014-12-07 02:43:58 -0800731 params1 = { 'speed': speed1 }
732 params2 = { 'speed': speed2 }
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000733 # self.isCrossConnect = True if self.annotations.get('optical.type') == 'cross-connect' else False
734 if isinstance(node1, LINCSwitch) and isinstance(node2, LINCSwitch):
735 self.isCrossConnect = False
736 else:
737 self.isCrossConnect = True
738 if isinstance(node1, LINCSwitch):
739 cls1 = LINCIntf
740 if self.isCrossConnect:
741 node1.crossConnects.append(self)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800742 else:
743 cls1 = Intf
744 # bad hack to stop error message from appearing when we try to set up intf in a packet switch,
745 # and there is no interface there( because we do not run makeIntfPair ). This way, we just set lo up
746 intfName1 = 'lo'
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000747 if isinstance(node2, LINCSwitch):
748 cls2 = LINCIntf
749 if self.isCrossConnect:
750 node2.crossConnects.append(self)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800751 else:
752 cls2 = Intf
753 intfName2 = 'lo'
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000754 Link.__init__(self, node1, node2, port1=port1, port2=port2,
Brian O'Connoreb27c452014-12-07 02:43:58 -0800755 intfName1=intfName1, intfName2=intfName2, cls1=cls1,
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000756 cls2=cls2, params1=params1, params2=params2)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800757
758 @classmethod
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000759 def makeIntfPair(_cls, intfName1, intfName2, *args, **kwargs):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800760 pass
761
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000762 def json(self):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800763 "build and return the json configuration dictionary for this link"
764 configData = {}
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000765 configData[ 'src' ] = ('of:' + self.intf1.node.dpid +
766 '/%s' % self.intf1.node.ports[ self.intf1 ])
767 configData[ 'dst' ] = ('of:' + self.intf2.node.dpid +
768 '/%s' % self.intf2.node.ports[ self.intf2 ])
Brian O'Connoreb27c452014-12-07 02:43:58 -0800769 configData[ 'type' ] = self.linkType
770 configData[ 'annotations' ] = self.annotations
771 return configData
772
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000773class LINCIntf(OpticalIntf):
774 """
775 LINC interface class
776 """
777 def __init__(self, name=None, node=None, speed=0,
778 port=None, link=None, **params):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800779 self.node = node
780 self.speed = speed
781 self.port = port
782 self.link = link
783 self.name = name
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000784 node.addIntf(self, port=port)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800785 self.params = params
786 self.ip = None
787
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000788 def json(self):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800789 "build and return the JSON information for this interface( not used right now )"
790 configDict = {}
791 configDict[ 'port' ] = self.port
792 configDict[ 'speed' ] = self.speed
Ayaka Koshibe144bab02015-10-22 12:56:59 -0700793 portType = 'COPPER'
794 if isinstance(self.link, LINCLink):
795 portType = 'OCH' if self.link.isCrossConnect else 'OMS'
796 configDict[ 'type' ] = portType
Brian O'Connoreb27c452014-12-07 02:43:58 -0800797 return configDict
798
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000799 def config(self, *args, **kwargs):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800800 "dont configure a dummy interface"
801 pass
802
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000803 def ifconfig(self, status):
804 "configure the status"
805 if status == "up":
806 return self.node.w_port_up(self.port)
807 elif status == "down":
808 return self.node.w_port_down(self.port)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800809
810
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000811class MininetOE(Mininet):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800812 "Mininet with Linc-OE support (starts and stops linc-oe)"
813
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000814 def start(self):
815 Mininet.start(self)
816 LINCSwitch.bootOE(self)
Brian O'Connoreb27c452014-12-07 02:43:58 -0800817
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000818 def stop(self):
819 Mininet.stop(self)
820 LINCSwitch.shutdownOE()
Brian O'Connoreb27c452014-12-07 02:43:58 -0800821
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000822 def addControllers(self, controllers):
Brian O'Connoreb27c452014-12-07 02:43:58 -0800823 i = 0
824 for ctrl in controllers:
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000825 self.addController(RemoteController('c%d' % i, ip=ctrl))
826 i += 1
Brian O'Connoreb27c452014-12-07 02:43:58 -0800827
828if __name__ == '__main__':
Marc De Leenheer6ff97642015-07-08 19:21:16 +0000829 pass