Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 1 | import socket |
| 2 | |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 3 | from mininet.log import error, info |
| 4 | from mininet.node import Switch |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 5 | |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 6 | BMV2_TARGET = 'simple_switch_grpc' |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 7 | |
| 8 | class ONOSBmv2Switch(Switch): |
| 9 | """BMv2 software switch """ |
| 10 | |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 11 | deviceId = 0 |
Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 12 | instanceCount = 0 |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 13 | |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 14 | def __init__(self, name, debugger=False, loglevel="warn", elogger=False, persistent=False, |
| 15 | logflush=False, **kwargs): |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 16 | Switch.__init__(self, name, **kwargs) |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 17 | self.thriftPort = ONOSBmv2Switch.pickUnusedPort() |
| 18 | self.grpcPort = ONOSBmv2Switch.pickUnusedPort() |
| 19 | if self.dpid: |
| 20 | self.deviceId = int(self.dpid, 0 if 'x' in self.dpid else 16) |
Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 21 | else: |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 22 | self.deviceId = ONOSBmv2Switch.deviceId |
| 23 | ONOSBmv2Switch.deviceId += 1 |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 24 | self.debugger = debugger |
| 25 | self.loglevel = loglevel |
| 26 | self.logfile = '/tmp/bmv2-%d.log' % self.deviceId |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 27 | self.elogger = elogger |
| 28 | self.persistent = persistent |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 29 | self.logflush = logflush |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 30 | if persistent: |
| 31 | self.exectoken = "/tmp/bmv2-%d-exec-token" % self.deviceId |
| 32 | self.cmd("touch %s" % self.exectoken) |
Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 33 | # Store thrift port for future uses. |
| 34 | self.cmd("echo %d > /tmp/bmv2-%d-thrift-port" % (self.thriftPort, self.deviceId)) |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 35 | self.cmd("echo %d > /tmp/bmv2-%d-grpc-port" % (self.grpcPort, self.deviceId)) |
Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 36 | |
| 37 | @classmethod |
| 38 | def pickUnusedPort(cls): |
| 39 | s = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 40 | s.bind(('localhost', 0)) |
| 41 | addr, port = s.getsockname() |
| 42 | s.close() |
| 43 | return port |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 44 | |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 45 | def start(self, controllers): |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 46 | args = [BMV2_TARGET, '--device-id %s' % str(self.deviceId)] |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 47 | for port, intf in self.intfs.items(): |
| 48 | if not intf.IP(): |
| 49 | args.append('-i %d@%s' % (port, intf.name)) |
| 50 | if self.thriftPort: |
| 51 | args.append('--thrift-port %d' % self.thriftPort) |
| 52 | if self.elogger: |
| 53 | nanomsg = 'ipc:///tmp/bmv2-%d-log.ipc' % self.deviceId |
| 54 | args.append('--nanolog %s' % nanomsg) |
| 55 | if self.debugger: |
| 56 | args.append('--debugger') |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 57 | args.append('--log-file %s -L%s' % (self.logfile, self.loglevel)) |
| 58 | if self.logflush: |
| 59 | args.append('--log-flush') |
| 60 | args.append('--no-p4') |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 61 | args.append('--') |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 62 | args.append('--enable-swap') |
| 63 | if self.grpcPort: |
| 64 | args.append('--grpc-server-addr 0.0.0.0:%d' % self.grpcPort) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 65 | |
| 66 | bmv2cmd = " ".join(args) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 67 | info("\nStarting BMv2 target: %s\n" % bmv2cmd) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 68 | if self.persistent: |
| 69 | # Re-exec the switch if it crashes. |
| 70 | cmdStr = "(while [ -e {} ]; " \ |
| 71 | "do {} ; " \ |
| 72 | "sleep 1; " \ |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 73 | "done;) &".format(self.exectoken, bmv2cmd) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 74 | else: |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 75 | cmdStr = "{} &".format(bmv2cmd) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 76 | self.cmd(cmdStr) |
| 77 | |
| 78 | def stop(self): |
| 79 | "Terminate switch." |
Carmelo Cascone | 977ae3f | 2016-06-23 19:28:28 -0700 | [diff] [blame] | 80 | self.cmd("rm -f /tmp/bmv2-%d-*" % self.deviceId) |
Carmelo Cascone | 75e9799 | 2017-06-05 02:32:47 -0400 | [diff] [blame^] | 81 | # wildcard end as BMv2 might create log files with .txt extension |
| 82 | self.cmd("rm -f /tmp/bmv2-%d.log*" % self.deviceId) |
| 83 | self.cmd('kill %' + BMV2_TARGET) |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 84 | self.deleteIntfs() |
| 85 | |
| 86 | |
| 87 | ### Exports for bin/mn |
Carmelo Cascone | 785fada | 2016-06-16 18:34:16 -0700 | [diff] [blame] | 88 | switches = {'onosbmv2': ONOSBmv2Switch} |