class WebsocketView(view_base.ViewBase):
def __init__(self, ws):
super(WebsocketView, self).__init__()
self.ws = ws
self.address = None
self.watcher = None
def run(self):
while True:
msg = self.ws.receive()
if msg is not None:
try:
msg = json.loads(msg)
except:
LOG.debug("json parse error: %s", msg)
continue
self._recv_message(msg)
else:
self.watcher.stop()
break
self.ws.close()
LOG.info('Websocket: closed.')
return self.null_response()
def _send_message(self, msg_name, address, body=None):
message = {}
message['message'] = msg_name
message['host'], message['port'] = address.split(':')
message['body'] = body
LOG.debug("Websocket: send msg.\n%s", json.dumps(message, indent=2))
self.ws.send(json.dumps(message))
def _recv_message(self, msg):
LOG.debug('Websocket: recv msg.\n%s', json.dumps(msg, indent=2))
message = msg.get('message')
body = msg.get('body')
if message == 'rest_update':
self._watcher_start(body)
elif message == 'watching_switch_update':
self._watching_switch_update(body)
else:
return
def _watcher_start(self, body):
address = '%s:%s' % (body['host'], body['port'])
self.address = address
if self.watcher:
self.watcher.stop()
self.watcher = TopologyWatcher(
update_handler=self.update_handler,
rest_error_handler=self.rest_error_handler)
self.watcher.start(address)
def _watching_switch_update(self, body):
pass
# called by watcher when topology update
def update_handler(self, address, delta):
if self.address != address:
# user be watching the another controller already
return
LOG.debug(delta)
self._send_message('rest_connected', address)
self._send_del_links(address, delta.deleted)
self._send_del_ports(address, delta.deleted)
self._send_del_switches(address, delta.deleted)
self._send_add_switches(address, delta.added)
self._send_add_ports(address, delta.added)
self._send_add_links(address, delta.added)
def _send_add_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('add_switches', address, body)
def _send_del_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('del_switches', address, body)
def _build_switches_message(self, topo):
body = []
for s in topo['switches']:
S = {'dpid': s.dpid, 'ports': {}}
for p in s.ports:
S['ports'][p.port_no] = p.to_dict()
body.append(S)
return body
def _send_add_ports(self, address, topo):
body = self._build_ports_message(topo)
if body:
self._send_message('add_ports', address, body)
def _send_del_ports(self, address, topo):
body = self._build_ports_message(topo)
if body:
self._send_message('del_ports', address, body)
def _build_ports_message(self, topo):
# send only except new added switches
ports = set(topo['ports'])
for s in topo['switches']:
ports -= set(s.ports)
body = []
for p in ports:
body.append(p.to_dict())
return body
def _send_add_links(self, address, topo):
body = self._build_links_message(topo)
if body:
self._send_message('add_links', address, body)
def _send_del_links(self, address, topo):
body = self._build_links_message(topo)
if body:
self._send_message('del_links', address, body)
def _build_links_message(self, topo):
body = []
for link in topo['links']:
# handle link as undirected
if link.src.dpid > link.dst.dpid:
continue
p1 = link.src.to_dict()
p2 = link.dst.to_dict()
L = {'p1': p1.copy(), 'p2': p2.copy()}
L['p1']['peer'] = p2.copy()
L['p2']['peer'] = p1.copy()
body.append(L)
return body
# called by watcher when rest api error
def rest_error_handler(self, address, e):
LOG.debug('REST API Error: %s', e)
self._send_message('rest_disconnected', address)
class WebsocketView(view_base.ViewBase):
def __init__(self, ws):
super(WebsocketView, self).__init__()
self.ws = ws
self.address = None
self.watcher = None
def run(self):
while True:
msg = self.ws.receive()
if msg is not None:
try:
msg = json.loads(msg)
except:
LOG.debug("json parse error: %s", msg)
continue
self._recv_message(msg)
else:
if self.watcher:
self.watcher.stop()
break
self.ws.close()
LOG.info('Websocket: closed.')
return self.null_response()
def _send_message(self, msg_name, address, body=None):
message = {}
message['message'] = msg_name
message['host'], message['port'] = address.split(':')
message['body'] = body
LOG.debug("Websocket: send msg.\n%s", json.dumps(message, indent=2))
self.ws.send(json.dumps(message))
def _recv_message(self, msg):
LOG.debug('Websocket: recv msg.\n%s', json.dumps(msg, indent=2))
message = msg.get('message')
body = msg.get('body')
if message == 'rest_update':
self._watcher_start(body)
elif message == 'watching_switch_update':
self._watching_switch_update(body)
elif message == 'open_wireshark':
self._open_wireshark(body)
else:
return
def _watcher_start(self, body):
address = '%s:%s' % (body['host'], body['port'])
self.address = address
if self.watcher:
self.watcher.stop()
self.watcher = TopologyWatcher(
update_handler=self.update_handler,
rest_error_handler=self.rest_error_handler)
self.watcher.start(address)
def _watching_switch_update(self, body):
pass
def _open_wireshark(self, body):
interface = '%s' % (body['interface'])
import os
os.system("sudo wireshark -i " + interface + " -k &")
# called by watcher when topology update
def update_handler(self, address, delta):
if self.address != address:
# user be watching the another controller already
return
LOG.debug(delta)
self._send_message('rest_connected', address)
self._send_del_links(address, delta.deleted)
self._send_del_ports(address, delta.deleted)
self._send_del_switches(address, delta.deleted)
self._send_add_switches(address, delta.added)
self._send_add_ports(address, delta.added)
self._send_add_links(address, delta.added)
def _send_add_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('add_switches', address, body)
def _send_del_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('del_switches', address, body)
def _build_switches_message(self, topo):
body = []
# Read switch positions from network.xml
from xml.dom import minidom
position = {}
xmldoc = None
try:
xmldoc = minidom.parse('network.xml')
itemlist = xmldoc.getElementsByTagName('node')
for s in itemlist:
n = s.attributes['id'].value
# Remove the N char at the beginning
n = int(n[1:])
x = s.getElementsByTagName('x')[0].firstChild.data
y = s.getElementsByTagName('y')[0].firstChild.data
position[n] = (int(float(x)), int(float(y)))
except Exception, detail:
print("Failure to parse network.xml: %s" % str(detail))
for s in topo['switches']:
position[int(float(s.dpid))] = (-1, -1)
# Nodes creation
for s in topo['switches']:
pos = {}
pos['x'] = position[int(float(s.dpid))][0]
pos['y'] = position[int(float(s.dpid))][1]
S = {'dpid': s.dpid, 'ports': {}, 'pos': pos}
for p in s.ports:
S['ports'][p.port_no] = p.to_dict()
body.append(S)
return body
class WebsocketView(view_base.ViewBase):
def __init__(self, ws):
super(WebsocketView, self).__init__()
self.ws = ws
self.address = None
self.watcher = None
def run(self):
while True:
msg = self.ws.receive()
if msg is not None:
try:
msg = json.loads(msg)
except:
LOG.debug("json parse error: %s", msg)
continue
self._recv_message(msg)
else:
if self.watcher:
self.watcher.stop()
break
self.ws.close()
LOG.info('Websocket: closed.')
return self.null_response()
def _send_message(self, msg_name, address, body=None):
message = {}
message['message'] = msg_name
message['host'], message['port'] = address.split(':')
message['body'] = body
LOG.debug("Websocket: send msg.\n%s", json.dumps(message, indent=2))
self.ws.send(json.dumps(message))
def _recv_message(self, msg):
LOG.debug('Websocket: recv msg.\n%s', json.dumps(msg, indent=2))
message = msg.get('message')
body = msg.get('body')
if message == 'rest_update':
self._watcher_start(body)
elif message == 'watching_switch_update':
self._watching_switch_update(body)
elif message == 'open_wireshark':
self._open_wireshark(body)
else:
return
def _watcher_start(self, body):
address = '%s:%s' % (body['host'], body['port'])
self.address = address
if self.watcher:
self.watcher.stop()
self.watcher = TopologyWatcher(
update_handler=self.update_handler,
rest_error_handler=self.rest_error_handler)
self.watcher.start(address)
def _watching_switch_update(self, body):
pass
def _open_wireshark(self, body):
interface = '%s' % (body['interface'])
import os
os.system("sudo wireshark -i "+interface+" -k &")
# called by watcher when topology update
def update_handler(self, address, delta):
if self.address != address:
# user be watching the another controller already
return
LOG.debug(delta)
self._send_message('rest_connected', address)
self._send_del_links(address, delta.deleted)
self._send_del_ports(address, delta.deleted)
self._send_del_switches(address, delta.deleted)
self._send_add_switches(address, delta.added)
self._send_add_ports(address, delta.added)
self._send_add_links(address, delta.added)
def _send_add_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('add_switches', address, body)
def _send_del_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('del_switches', address, body)
def _build_switches_message(self, topo):
body = []
# Read switch positions from network.xml
from xml.dom import minidom
position={}
xmldoc = None
try:
xmldoc = minidom.parse('network.xml')
itemlist = xmldoc.getElementsByTagName('node')
for s in itemlist:
n = s.attributes['id'].value
# Remove the N char at the beginning
n = int(n[1:])
x = s.getElementsByTagName('x')[0].firstChild.data
y = s.getElementsByTagName('y')[0].firstChild.data
position[n]=(int(float(x)), int(float(y)))
except Exception, detail:
print("Failure to parse network.xml: %s" % str(detail))
for s in topo['switches']:
position[int(float(s.dpid))]=(-1,-1)
# Nodes creation
for s in topo['switches']:
pos = {}
pos['x']=position[int(float(s.dpid))][0]
pos['y']=position[int(float(s.dpid))][1]
S = {'dpid': s.dpid, 'ports': {}, 'pos': pos}
for p in s.ports:
S['ports'][p.port_no] = p.to_dict()
body.append(S)
return body
class WebsocketView(view_base.ViewBase):
def __init__(self, ws):
super(WebsocketView, self).__init__()
self.ws = ws
self.address = None
self.watcher = None
def run(self):
while True:
msg = self.ws.receive()
if msg is not None:
try:
msg = json.loads(msg)
except:
LOG.debug("json parse error: %s", msg)
continue
self._recv_message(msg)
else:
if self.watcher:
self.watcher.stop()
break
self.ws.close()
LOG.info('Websocket: closed.')
return self.null_response()
def _send_message(self, msg_name, address, body=None):
message = {}
message['message'] = msg_name
message['host'], message['port'] = address.split(':')
message['body'] = body
LOG.debug("Websocket: send msg.\n%s", json.dumps(message, indent=2))
self.ws.send(json.dumps(message))
def _recv_message(self, msg):
LOG.debug('Websocket: recv msg.\n%s', json.dumps(msg, indent=2))
message = msg.get('message')
body = msg.get('body')
if message == 'rest_update':
self._watcher_start(body)
elif message == 'watching_switch_update':
self._watching_switch_update(body)
else:
return
def _watcher_start(self, body):
address = '%s:%s' % (body['host'], body['port'])
self.address = address
if self.watcher:
self.watcher.stop()
self.watcher = TopologyWatcher(
update_handler=self.update_handler,
rest_error_handler=self.rest_error_handler)
self.watcher.start(address)
def _watching_switch_update(self, body):
pass
# called by watcher when topology update
def update_handler(self, address, delta):
if self.address != address:
# user be watching the another controller already
return
LOG.debug(delta)
self._send_message('rest_connected', address)
self._send_del_links(address, delta.deleted)
self._send_del_ports(address, delta.deleted)
self._send_del_switches(address, delta.deleted)
self._send_add_switches(address, delta.added)
self._send_add_ports(address, delta.added)
self._send_add_links(address, delta.added)
def _send_add_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('add_switches', address, body)
def _send_del_switches(self, address, topo):
body = self._build_switches_message(topo)
if body:
self._send_message('del_switches', address, body)
def _build_switches_message(self, topo):
body = []
for s in topo['switches']:
S = {'dpid': s.dpid, 'ports': {}}
for p in s.ports:
S['ports'][p.port_no] = p.to_dict()
body.append(S)
return body
def _send_add_ports(self, address, topo):
body = self._build_ports_message(topo)
if body:
self._send_message('add_ports', address, body)
def _send_del_ports(self, address, topo):
body = self._build_ports_message(topo)
if body:
self._send_message('del_ports', address, body)
def _build_ports_message(self, topo):
# send only except new added switches
ports = set(topo['ports'])
for s in topo['switches']:
ports -= set(s.ports)
body = []
for p in ports:
body.append(p.to_dict())
return body
def _send_add_links(self, address, topo):
body = self._build_links_message(topo)
if body:
self._send_message('add_links', address, body)
def _send_del_links(self, address, topo):
body = self._build_links_message(topo)
if body:
self._send_message('del_links', address, body)
def _build_links_message(self, topo):
body = []
for link in topo['links']:
# handle link as undirected
if link.src.dpid > link.dst.dpid:
continue
p1 = link.src.to_dict()
p2 = link.dst.to_dict()
L = {'p1': p1.copy(), 'p2': p2.copy()}
L['p1']['peer'] = p2.copy()
L['p2']['peer'] = p1.copy()
body.append(L)
return body
# called by watcher when rest api error
def rest_error_handler(self, address, e):
LOG.debug('REST API Error: %s', e)
self._send_message('rest_disconnected', address)