You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 1 | """ |
Jeremy Ronquillo | 23fb216 | 2017-09-15 14:59:57 -0700 | [diff] [blame] | 2 | Copyright 2016 Open Networking Foundation ( ONF ) |
Jeremy Ronquillo | b27ce4c | 2017-07-17 12:41:28 -0700 | [diff] [blame] | 3 | |
| 4 | Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>, |
| 5 | the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>, |
| 6 | or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg> |
| 7 | |
| 8 | TestON is free software: you can redistribute it and/or modify |
| 9 | it under the terms of the GNU General Public License as published by |
| 10 | the Free Software Foundation, either version 2 of the License, or |
Jeremy Ronquillo | 23fb216 | 2017-09-15 14:59:57 -0700 | [diff] [blame] | 11 | ( at your option ) any later version. |
Jeremy Ronquillo | b27ce4c | 2017-07-17 12:41:28 -0700 | [diff] [blame] | 12 | |
| 13 | TestON is distributed in the hope that it will be useful, |
| 14 | but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 15 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 16 | GNU General Public License for more details. |
| 17 | |
| 18 | You should have received a copy of the GNU General Public License |
| 19 | along with TestON. If not, see <http://www.gnu.org/licenses/>. |
| 20 | """ |
Jeremy Ronquillo | b27ce4c | 2017-07-17 12:41:28 -0700 | [diff] [blame] | 21 | """ |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 22 | This file contains classes for CHOTestMonkey that are related to network event |
| 23 | Author: you@onlab.us |
| 24 | """ |
| 25 | from tests.CHOTestMonkey.dependencies.events.Event import EventType, EventStates, Event |
| 26 | from tests.CHOTestMonkey.dependencies.elements.NetworkElement import NetworkElement, Device, Host, Link |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 27 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 28 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 29 | class LinkEvent( Event ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 30 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 31 | def __init__( self ): |
| 32 | Event.__init__( self ) |
| 33 | self.linkA = None |
| 34 | self.linkB = None |
| 35 | |
| 36 | def startLinkEvent( self ): |
| 37 | return EventStates().PASS |
| 38 | |
| 39 | def startEvent( self, args ): |
| 40 | """ |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 41 | args are the names of the two link ends, e.g. [ 's1', 's2' ] |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 42 | """ |
| 43 | with self.eventLock: |
Jeremy Ronquillo | 23fb216 | 2017-09-15 14:59:57 -0700 | [diff] [blame] | 44 | # main.log.info( "%s - starting event" % ( self.typeString ) ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 45 | if len( args ) < 2: |
| 46 | main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) ) |
| 47 | return EventStates().ABORT |
| 48 | elif len( args ) > 2: |
| 49 | main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) ) |
| 50 | return EventStates().ABORT |
| 51 | if args[ 0 ] == 'random' or args[ 1 ] == 'random': |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 52 | if self.typeIndex == EventType().NETWORK_LINK_DOWN: |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 53 | with main.mininetLock: |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 54 | linkRandom = main.Mininet1.getLinkRandom( switchClasses=r"(OVSSwitch)", |
| 55 | excludeNodes=[ 'bgp', 'cs', 'nat', 'dhcp', 'r' ] ) |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 56 | if linkRandom is None: |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 57 | main.log.warn( "No link available, aborting event" ) |
| 58 | return EventStates().ABORT |
| 59 | args[ 0 ] = linkRandom[ 0 ] |
| 60 | args[ 1 ] = linkRandom[ 1 ] |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 61 | elif self.typeIndex == EventType().NETWORK_LINK_UP: |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 62 | import random |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 63 | with main.variableLock: |
| 64 | downLinks = [] |
| 65 | for link in main.links: |
| 66 | if link.isDown(): |
| 67 | downLinks.append( link ) |
| 68 | if len( downLinks ) == 0: |
| 69 | main.log.warn( "None of the links are in 'down' state, aborting event" ) |
| 70 | return EventStates().ABORT |
| 71 | linkList = random.sample( downLinks, 1 ) |
| 72 | self.linkA = linkList[ 0 ] |
| 73 | self.linkB = linkList[ 0 ].backwardLink |
| 74 | elif args[ 0 ] == args[ 1 ]: |
| 75 | main.log.warn( "%s - invalid arguments: %s" % ( self.typeString, args ) ) |
| 76 | return EventStates().ABORT |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 77 | if self.linkA is None or self.linkB is None: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 78 | for link in main.links: |
| 79 | if link.deviceA.name == args[ 0 ] and link.deviceB.name == args[ 1 ]: |
| 80 | self.linkA = link |
| 81 | elif link.deviceA.name == args[ 1 ] and link.deviceB.name == args[ 0 ]: |
| 82 | self.linkB = link |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 83 | if self.linkA is not None and self.linkB is not None: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 84 | break |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 85 | if self.linkA is None or self.linkB is None: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 86 | main.log.warn( "Bidirectional link %s - %s does not exist: " % ( args[ 0 ], args[ 1 ] ) ) |
| 87 | return EventStates().ABORT |
| 88 | main.log.debug( "%s - %s" % ( self.typeString, self.linkA ) ) |
| 89 | return self.startLinkEvent() |
| 90 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 91 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 92 | class LinkDown( LinkEvent ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 93 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 94 | """ |
| 95 | Generate a link down event giving the two ends of the link |
| 96 | """ |
| 97 | def __init__( self ): |
| 98 | LinkEvent.__init__( self ) |
| 99 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 100 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 101 | |
| 102 | def startLinkEvent( self ): |
| 103 | # TODO: do we need to handle a unidirectional link? |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 104 | assert self.linkA is not None and self.linkB is not None |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 105 | with main.variableLock: |
| 106 | if self.linkA.isDown() or self.linkB.isDown(): |
| 107 | main.log.warn( "Link Down - link already down" ) |
| 108 | return EventStates().ABORT |
| 109 | elif self.linkA.isRemoved() or self.linkB.isRemoved(): |
| 110 | main.log.warn( "Link Down - link has been removed" ) |
| 111 | return EventStates().ABORT |
You Wang | 5216320 | 2016-07-14 16:37:15 -0700 | [diff] [blame] | 112 | main.log.info( "Event recorded: {} {} {} {}".format( self.typeIndex, self.typeString, self.linkA.deviceA.name, self.linkA.deviceB.name ) ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 113 | with main.mininetLock: |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 114 | """ |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 115 | result = main.Mininet1.link( END1=self.linkA.deviceA.name, |
| 116 | END2=self.linkA.deviceB.name, |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 117 | OPTION="down" ) |
| 118 | """ |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 119 | result = main.Mininet1.delLink( self.linkA.deviceA.name, |
| 120 | self.linkA.deviceB.name ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 121 | if not result: |
| 122 | main.log.warn( "%s - failed to bring down link" % ( self.typeString ) ) |
| 123 | return EventStates().FAIL |
| 124 | with main.variableLock: |
| 125 | self.linkA.bringDown() |
| 126 | self.linkB.bringDown() |
| 127 | return EventStates().PASS |
| 128 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 129 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 130 | class LinkUp( LinkEvent ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 131 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 132 | """ |
| 133 | Generate a link up event giving the two ends of the link |
| 134 | """ |
| 135 | def __init__( self ): |
| 136 | LinkEvent.__init__( self ) |
| 137 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 138 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 139 | |
| 140 | def startLinkEvent( self ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 141 | assert self.linkA is not None and self.linkB is not None |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 142 | with main.variableLock: |
| 143 | if self.linkA.isUp() or self.linkB.isUp(): |
| 144 | main.log.warn( "Link Up - link already up" ) |
| 145 | return EventStates().ABORT |
| 146 | if self.linkA.isRemoved() or self.linkB.isRemoved(): |
| 147 | main.log.warn( "Link Up - link has been removed" ) |
| 148 | return EventStates().ABORT |
You Wang | 5216320 | 2016-07-14 16:37:15 -0700 | [diff] [blame] | 149 | main.log.info( "Event recorded: {} {} {} {}".format( self.typeIndex, self.typeString, self.linkA.deviceA.name, self.linkA.deviceB.name ) ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 150 | with main.mininetLock: |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 151 | """ |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 152 | result = main.Mininet1.link( END1=self.linkA.deviceA.name, |
| 153 | END2=self.linkA.deviceB.name, |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 154 | OPTION="up" ) |
| 155 | """ |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 156 | result = main.Mininet1.addLink( self.linkA.deviceA.name, |
| 157 | self.linkA.deviceB.name ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 158 | if not result: |
| 159 | main.log.warn( "%s - failed to bring up link" % ( self.typeString ) ) |
| 160 | return EventStates().FAIL |
| 161 | with main.variableLock: |
| 162 | self.linkA.bringUp() |
| 163 | self.linkB.bringUp() |
| 164 | return EventStates().PASS |
| 165 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 166 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 167 | class DeviceEvent( Event ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 168 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 169 | def __init__( self ): |
| 170 | Event.__init__( self ) |
| 171 | self.device = None |
| 172 | |
| 173 | def startDeviceEvent( self ): |
| 174 | return EventStates().PASS |
| 175 | |
| 176 | def startEvent( self, args ): |
| 177 | """ |
| 178 | args are the names of the device, e.g. 's1' |
| 179 | """ |
| 180 | with self.eventLock: |
Jeremy Ronquillo | 23fb216 | 2017-09-15 14:59:57 -0700 | [diff] [blame] | 181 | # main.log.info( "%s - starting event" % ( self.typeString ) ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 182 | if len( args ) < 1: |
| 183 | main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) ) |
| 184 | return EventStates().ABORT |
| 185 | elif len( args ) > 1: |
| 186 | main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) ) |
| 187 | return EventStates().ABORT |
| 188 | if args[ 0 ] == 'random': |
| 189 | import random |
| 190 | if self.typeIndex == EventType().NETWORK_DEVICE_DOWN: |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 191 | with main.mininetLock: |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 192 | switchRandom = main.Mininet1.getSwitchRandom( switchClasses=r"(OVSSwitch)", |
| 193 | excludeNodes=[ 'bgp', 'cs', 'nat', 'dhcp', 'r' ] ) |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 194 | if switchRandom is None: |
You Wang | 221db32 | 2016-06-03 15:45:52 -0700 | [diff] [blame] | 195 | main.log.warn( "No switch available, aborting event" ) |
| 196 | return EventStates().ABORT |
| 197 | args[ 0 ] = switchRandom |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 198 | elif self.typeIndex == EventType().NETWORK_DEVICE_UP: |
| 199 | with main.variableLock: |
| 200 | removedDevices = [] |
| 201 | for device in main.devices: |
| 202 | if device.isRemoved(): |
| 203 | removedDevices.append( device ) |
| 204 | if len( removedDevices ) == 0: |
| 205 | main.log.warn( "None of the devices are removed, aborting event" ) |
| 206 | return EventStates().ABORT |
| 207 | deviceList = random.sample( removedDevices, 1 ) |
| 208 | self.device = deviceList[ 0 ] |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 209 | if self.device is None: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 210 | for device in main.devices: |
| 211 | if device.name == args[ 0 ]: |
| 212 | self.device = device |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 213 | if self.device is None: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 214 | main.log.warn( "Device %s does not exist: " % ( args[ 0 ] ) ) |
| 215 | return EventStates().ABORT |
| 216 | main.log.debug( "%s - %s" % ( self.typeString, self.device ) ) |
| 217 | return self.startDeviceEvent() |
| 218 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 219 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 220 | class DeviceDown( DeviceEvent ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 221 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 222 | """ |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 223 | Generate a device down event ( which actually removes this device for now ) giving its name |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 224 | """ |
| 225 | def __init__( self ): |
| 226 | DeviceEvent.__init__( self ) |
| 227 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 228 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 229 | |
| 230 | def startDeviceEvent( self ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 231 | assert self.device is not None |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 232 | with main.variableLock: |
| 233 | if self.device.isRemoved(): |
| 234 | main.log.warn( "Device Down - device has been removed" ) |
| 235 | return EventStates().ABORT |
You Wang | 5216320 | 2016-07-14 16:37:15 -0700 | [diff] [blame] | 236 | main.log.info( "Event recorded: {} {} {}".format( self.typeIndex, self.typeString, self.device.name ) ) |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 237 | result = main.TRUE |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 238 | with main.mininetLock: |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 239 | # Disable ports toward dual-homed hosts |
| 240 | for host, port in self.device.hosts.items(): |
| 241 | if host.isDualHomed: |
| 242 | main.log.info( "Disable port {}/{} which connects to a dual-homed host before bringing down this device".format( self.device.dpid, port ) ) |
| 243 | result = result and main.Cluster.active( 0 ).CLI.portstate( dpid=self.device.dpid, port=port, state="disable" ) |
| 244 | # result = main.Mininet1.delSwitch( self.device.name ) |
| 245 | result = result and main.Mininet1.switch( SW=self.device.name, OPTION="stop" ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 246 | if not result: |
| 247 | main.log.warn( "%s - failed to bring down device" % ( self.typeString ) ) |
| 248 | return EventStates().FAIL |
| 249 | with main.variableLock: |
| 250 | self.device.setRemoved() |
| 251 | for link in self.device.outgoingLinks: |
| 252 | link.setRemoved() |
| 253 | link.backwardLink.setRemoved() |
| 254 | for host in self.device.hosts: |
| 255 | host.setRemoved() |
You Wang | 2b687c0 | 2016-05-13 17:01:31 -0700 | [diff] [blame] | 256 | for intent in main.intents: |
| 257 | if intent.deviceA == self.device or intent.deviceB == self.device: |
| 258 | intent.setFailed() |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 259 | return EventStates().PASS |
| 260 | |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 261 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 262 | class DeviceUp( DeviceEvent ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 263 | |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 264 | """ |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 265 | Generate a device up event ( which re-adds this device in case the device is removed ) giving its name |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 266 | """ |
| 267 | def __init__( self ): |
| 268 | DeviceEvent.__init__( self ) |
| 269 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 270 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 271 | |
| 272 | def startDeviceEvent( self ): |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 273 | assert self.device is not None |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 274 | with main.variableLock: |
| 275 | if self.device.isUp(): |
| 276 | main.log.warn( "Device Up - device already up" ) |
| 277 | return EventStates().ABORT |
| 278 | # Re-add the device |
You Wang | 5216320 | 2016-07-14 16:37:15 -0700 | [diff] [blame] | 279 | main.log.info( "Event recorded: {} {} {}".format( self.typeIndex, self.typeString, self.device.name ) ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 280 | with main.mininetLock: |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 281 | # result = main.Mininet1.addSwitch( self.device.name, dpid=self.device.dpid[ 3: ] ) |
| 282 | result = main.Mininet1.switch( SW=self.device.name, OPTION='start' ) |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 283 | if not result: |
| 284 | main.log.warn( "%s - failed to re-add device" % ( self.typeString ) ) |
| 285 | return EventStates().FAIL |
| 286 | with main.variableLock: |
| 287 | self.device.bringUp() |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 288 | ''' |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 289 | # Re-add links |
| 290 | # We add host-device links first since we did the same in mininet topology file |
| 291 | # TODO: a more rubust way is to add links according to the port info of the device |
| 292 | for host in self.device.hosts: |
| 293 | # Add host-device link |
| 294 | with main.mininetLock: |
| 295 | result = main.Mininet1.addLink( self.device.name, host.name ) |
| 296 | if not result: |
| 297 | main.log.warn( "%s - failed to re-connect host %s to device" % ( self.typeString, host.name ) ) |
| 298 | return EventStates().FAIL |
| 299 | for link in self.device.outgoingLinks: |
| 300 | neighbor = link.deviceB |
| 301 | # Skip bringing up any link that connecting this device to a removed neighbor |
| 302 | if neighbor.isRemoved(): |
| 303 | continue |
| 304 | with main.mininetLock: |
| 305 | result = main.Mininet1.addLink( self.device.name, neighbor.name ) |
| 306 | if not result: |
| 307 | main.log.warn( "%s - failed to re-add link to %s" % ( self.typeString, neighbor.name ) ) |
| 308 | return EventStates().FAIL |
| 309 | with main.variableLock: |
| 310 | link.bringUp() |
| 311 | link.backwardLink.bringUp() |
You Wang | 2b687c0 | 2016-05-13 17:01:31 -0700 | [diff] [blame] | 312 | for intent in main.intents: |
| 313 | if intent.isFailed(): |
| 314 | if intent.deviceA == self.device and intent.deviceB.isUp() or\ |
Jon Hall | 2bb3e21 | 2017-05-24 17:07:25 -0700 | [diff] [blame] | 315 | intent.deviceB == self.device and intent.deviceA.isUp(): |
You Wang | 2b687c0 | 2016-05-13 17:01:31 -0700 | [diff] [blame] | 316 | intent.setInstalled() |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 317 | ''' |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 318 | # Re-assign mastership for the device |
| 319 | with main.mininetLock: |
Devin Lim | 142b534 | 2017-07-20 15:22:39 -0700 | [diff] [blame] | 320 | ips = main.Cluster.getIps() |
| 321 | main.Mininet1.assignSwController( sw=self.device.name, ip=ips ) |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 322 | for link in self.device.outgoingLinks: |
| 323 | neighbor = link.deviceB |
| 324 | # Skip bringing up any link that connecting this device to a removed neighbor |
| 325 | if neighbor.isRemoved(): |
| 326 | continue |
| 327 | # Bring down again any link that was brought down before the device was down |
| 328 | if int( link.portB ) in link.deviceB.downPorts: |
| 329 | with main.variableLock: |
| 330 | link.bringDown() |
| 331 | link.backwardLink.bringDown() |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 332 | else: |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 333 | with main.variableLock: |
| 334 | link.bringUp() |
| 335 | link.backwardLink.bringUp() |
| 336 | # Re-discover hosts |
| 337 | if self.device.hosts: |
| 338 | main.Mininet1.discoverHosts( hostList=[ host.name for host in self.device.hosts ] ) |
| 339 | for host in self.device.hosts: |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 340 | with main.variableLock: |
| 341 | host.bringUp() |
You Wang | 7d14d64 | 2019-01-23 15:10:08 -0800 | [diff] [blame] | 342 | self.device.downPorts = [] |
| 343 | return EventStates().PASS |
| 344 | |
| 345 | |
| 346 | class PortEvent( Event ): |
| 347 | |
| 348 | def __init__( self ): |
| 349 | Event.__init__( self ) |
| 350 | self.device = None |
| 351 | self.port = None |
| 352 | self.link = None |
| 353 | |
| 354 | def startPortEvent( self ): |
| 355 | return EventStates().PASS |
| 356 | |
| 357 | def startEvent( self, args ): |
| 358 | """ |
| 359 | args are the device name and port number, e.g. [ 's1', '5' ] |
| 360 | """ |
| 361 | with self.eventLock: |
| 362 | # main.log.info( "%s - starting event" % ( self.typeString ) ) |
| 363 | if len( args ) < 2: |
| 364 | main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) ) |
| 365 | return EventStates().ABORT |
| 366 | elif len( args ) > 2: |
| 367 | main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) ) |
| 368 | return EventStates().ABORT |
| 369 | if args[ 0 ] == 'random' or args[ 1 ] == 'random': |
| 370 | if self.typeIndex == EventType().NETWORK_PORT_DOWN: |
| 371 | with main.mininetLock: |
| 372 | linkRandom = main.Mininet1.getLinkRandom( switchClasses=r"(OVSSwitch)", |
| 373 | excludeNodes=[ 'bgp', 'cs', 'nat', 'dhcp', 'r' ]) |
| 374 | if linkRandom is None: |
| 375 | main.log.warn( "No link available, aborting event" ) |
| 376 | return EventStates().ABORT |
| 377 | for link in main.links: |
| 378 | if link.deviceA.name == linkRandom[ 0 ] and link.deviceB.name == linkRandom[ 1 ]: |
| 379 | self.device = link.deviceA |
| 380 | self.port = int( link.portA ) |
| 381 | if not self.device: |
| 382 | main.log.warn( "Failed to get a radnom device port, aborting event" ) |
| 383 | return EventStates().ABORT |
| 384 | elif self.typeIndex == EventType().NETWORK_PORT_UP: |
| 385 | import random |
| 386 | with main.variableLock: |
| 387 | downPorts = {} |
| 388 | for link in main.links: |
| 389 | if link.isDown(): |
| 390 | if int( link.portA ) in link.deviceA.downPorts: |
| 391 | downPorts[ link.deviceA ] = link.portA |
| 392 | if len( downPorts ) == 0: |
| 393 | main.log.warn( "None of the links are in 'down' state, aborting event" ) |
| 394 | return EventStates().ABORT |
| 395 | deviceList = random.sample( downPorts, 1 ) |
| 396 | self.device = deviceList[ 0 ] |
| 397 | self.port = int( downPorts[ self.device ] ) |
| 398 | if self.device is None: |
| 399 | for device in main.devices: |
| 400 | if device.name == args[ 0 ]: |
| 401 | self.device = device |
| 402 | if self.device is None: |
| 403 | main.log.warn( "Device %s does not exist: " % ( args[ 0 ] ) ) |
| 404 | return EventStates().ABORT |
| 405 | if self.port is None: |
| 406 | try: |
| 407 | self.port = int( args[ 1 ] ) |
| 408 | except Exception: |
| 409 | main.log.warn( "Device port is not a number: {}".format( args[ 1 ] ) ) |
| 410 | return EventStates().ABORT |
| 411 | if self.link is None: |
| 412 | for link in main.links: |
| 413 | if link.deviceA.name == self.device.name and int( link.portA ) == self.port: |
| 414 | self.link = link |
| 415 | if self.link is None: |
| 416 | main.log.warn( "There's no link on device {} port {}".format( self.device.name, self.port ) ) |
| 417 | return EventStates().ABORT |
| 418 | main.log.debug( "%s - %s:%s" % ( self.typeString, self.device, self.port ) ) |
| 419 | return self.startPortEvent() |
| 420 | |
| 421 | |
| 422 | class PortDown( PortEvent ): |
| 423 | |
| 424 | """ |
| 425 | Generate a port down event giving the device name and port number |
| 426 | """ |
| 427 | def __init__( self ): |
| 428 | PortEvent.__init__( self ) |
| 429 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 430 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 431 | |
| 432 | def startPortEvent( self ): |
| 433 | assert self.device is not None and self.port is not None and self.link is not None |
| 434 | if self.link.isDown(): |
| 435 | main.log.warn( "port Down - link already down" ) |
| 436 | return EventStates().ABORT |
| 437 | elif self.link.isRemoved(): |
| 438 | main.log.warn( "port Down - link has been removed" ) |
| 439 | return EventStates().ABORT |
| 440 | main.log.info( "Event recorded: {} {} {} {}".format( self.typeIndex, self.typeString, self.device.name, self.port ) ) |
| 441 | with main.mininetLock: |
| 442 | result = main.Cluster.active( 0 ).CLI.portstate( dpid=self.device.dpid, port=self.port, state="disable" ) |
| 443 | if not result: |
| 444 | main.log.warn( "%s - failed to bring down port" % ( self.typeString ) ) |
| 445 | return EventStates().FAIL |
| 446 | with main.variableLock: |
| 447 | self.device.downPorts.append( self.port ) |
| 448 | self.link.bringDown() |
| 449 | self.link.backwardLink.bringDown() |
| 450 | return EventStates().PASS |
| 451 | |
| 452 | |
| 453 | class PortUp( PortEvent ): |
| 454 | |
| 455 | """ |
| 456 | Generate a port up event giving the device name and port number |
| 457 | """ |
| 458 | def __init__( self ): |
| 459 | PortEvent.__init__( self ) |
| 460 | self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| 461 | self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| 462 | |
| 463 | def startPortEvent( self ): |
| 464 | assert self.device is not None and self.port is not None and self.link is not None |
| 465 | if self.link.isUp(): |
| 466 | main.log.warn( "port up - link already up" ) |
| 467 | return EventStates().ABORT |
| 468 | elif self.link.isRemoved(): |
| 469 | main.log.warn( "port up - link has been removed" ) |
| 470 | return EventStates().ABORT |
| 471 | main.log.info( "Event recorded: {} {} {} {}".format( self.typeIndex, self.typeString, self.device.name, self.port ) ) |
| 472 | with main.mininetLock: |
| 473 | result = main.Cluster.active( 0 ).CLI.portstate( dpid=self.device.dpid, port=self.port, state="enable" ) |
| 474 | if not result: |
| 475 | main.log.warn( "%s - failed to bring up port " % ( self.typeString ) ) |
| 476 | return EventStates().FAIL |
| 477 | with main.variableLock: |
| 478 | self.device.downPorts.remove( self.port ) |
| 479 | self.link.bringUp() |
| 480 | self.link.backwardLink.bringUp() |
You Wang | db927a5 | 2016-02-26 11:03:28 -0800 | [diff] [blame] | 481 | return EventStates().PASS |