def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = 'domain_controller' self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 10 self.local_mac = ['10.108.93.100', '10.108.90.201'] self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.num = 0 self.taskId_match = {} self.border_table = [] self.arp_table = {} self.port_to_localmac = {} self.timesOfQos = 0 self.qosSwitch = [] self.completeTopo = None ############################################ self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} # self.shortestPath = {(513,518):[513,515,518],(518,513):[518,515,513]} ##################################################### self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag # if self.monitorThreadFlag: # self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain ' + str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.deviceInfo = {} #gqz #shortestPath is a dict ,the key is a tuple of (in_dpid,in_port_no,out_dpid,out_port_no), # and the value is the shortestpath between them #edgePort is a list consist by tuples of (dpid,port_no),which line to a switch controlled by a neighboured-domainController #borderPeer is a list consist by tuples of (dpid,port_no),which Packet-in a arp packet. #outDomainSwitch is a dict ,the key is a tuple of (dpid,port_no),which in the edgePort, # and the value is a tuple of (dpid,port_no) which line to the key and is not the switch controlled by this DC self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} #gqz self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = 'domain_controller' self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain ' + str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} #self.shortestPath = {(773,769):[773,770,769],(769,773):[769,770,773]} self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.wsapi_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.deviceInfo = {} self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.deviceInfo = {} #gqz #shortestPath is a dict ,the key is a tuple of (in_dpid,in_port_no,out_dpid,out_port_no), # and the value is the shortestpath between them #edgePort is a list consist by tuples of (dpid,port_no),which line to a switch controlled by a neighboured-domainController #borderPeer is a list consist by tuples of (dpid,port_no),which Packet-in a arp packet. #outDomainSwitch is a dict ,the key is a tuple of (dpid,port_no),which in the edgePort, # and the value is a tuple of (dpid,port_no) which line to the key and is not the switch controlled by this DC self.edgePort=[] self.borderPeer=[] self.outDomainSwitch={} self.shortestPath={} #gqz self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = "domain_controller" self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name="Domain " + str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 10 self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.num = 0 ############################################ self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} # self.shortestPath = {(513,518):[513,515,518],(518,513):[518,515,513]} ##################################################### self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag # if self.monitorThreadFlag: # self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs["wsgi"] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data)
class DomainController(app_manager.RyuApp): _CONTEXTS = {"wsgi": WSGIApplication, "switches": switches.Switches, "arp": IcmpResponder} OFP_VERSIONS = [ofproto_v1_0.OFP_VERSION, ofproto_v1_3.OFP_VERSION] def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = "domain_controller" self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name="Domain " + str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 10 self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.num = 0 ############################################ self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} # self.shortestPath = {(513,518):[513,515,518],(518,513):[518,515,513]} ##################################################### self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag # if self.monitorThreadFlag: # self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs["wsgi"] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data) def _get_QueueQos(self, dpid): assert dpid in self.QoS_dict return self.QoS_dict[dpid] def _get_datapath(self, dpid): assert dpid in self.dps return self.dps[dpid] @set_ev_cls(ofp_event.EventOFPSwitchFeatures, CONFIG_DISPATCHER) def switch_features_handler(self, ev): msg = ev.msg datapath = msg.datapath self.logger.debug("switch features ev %s", msg) dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwitchFeatures(dpid)) switch.initFieds(msg.version, msg.capabilities, msg.n_buffers, msg.n_tables, msg.auxiliary_id) if dpid not in self.QoS_dict: queueQoSInstance = QueueQos(datapath, self.CONF) self.QoS_dict[dpid] = queueQoSInstance ip = datapath.address[0] ovs_addr = "tcp:" + ip + ":" + DEFAULTOVSDBADDR queueQoSInstance.set_ovsdb_addr(dpid, ovs_addr) self.logger.info("QueueQoS Entering. dpid: %016x, addr: %s" % (dpid, ovs_addr)) # msg: {'body': [OFPPort(port_no=9,hw_addr='48:6e:73:02:03:08', # name='ge-1/1/9',config=0,state=1,curr=10240 @set_ev_cls(ofp_event.EventOFPPortDescStatsReply, CONFIG_DISPATCHER) def multipart_reply_handler(self, ev): msg = ev.msg datapath = msg.datapath dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwitchFeatures(dpid)) # self.num = self.num+1 # print self.num # ports = switch.getPorts() # print 'msg: ', msg.__dict__ for portdesc in msg.body: # print 'portdesc: ', portdesc portNo = portdesc.port_no if portNo > 10000: # print 11111111111111111111111111 name = portdesc.name switch.setName(name) else: port = PortFeatures(portNo) port.initFields(portdesc) # print 'port: ',port.__dict__ switch.ports[portNo] = port self.QoS_dict[dpid].queueInfo[portNo] = QueuePort(portNo, dpid) if self.superExist: self._send_switch_features_message(dpid) def _send_switch_features_message(self, dpid): send_message = self._make_switch_features_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_switch_features_message(self, dpid): assert dpid in self.deviceInfo switchFeaturs = self.deviceInfo[dpid] info = switchFeaturs.makeFeaturesMessage() to_send = {} to_send[TYPE] = "switchFeature" to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[SWITCHFEATURE] = info send_message = json.dumps(to_send) return send_message def remove_flow(self, datapath, match): ofproto = datapath.ofproto parser = datapath.ofproto_parser if ofproto.OFP_VERSION == ofproto_v1_0.OFP_VERSION: mod = parser.OFPFlowMod( datapath=datapath, command=ofproto.OFPFC_DELETE, out_port=ofproto.OFPP_ANY, match=match ) elif ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: mod = parser.OFPFlowMod( datapath=datapath, command=ofproto.OFPFC_DELETE, out_group=ofproto.OFPG_ANY, out_port=ofproto.OFPP_ANY, match=match, ) datapath.send_msg(mod) def _get_ofproto_info(self, dpid): assert dpid in self.dps dp = self.dps[dpid] parser = dp.ofproto_parser ofproto = dp.ofproto return dp, parser, ofproto def _get_flow_mod(self, dpid, match, actions, priority=None, buffer_id=None): dp, parser, ofproto = self._get_ofproto_info(dpid) if not priority: priority = ofproto.OFP_DEFAULT_PRIORITY if ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: inst = [parser.OFPInstructionActions(ofproto.OFPIT_APPLY_ACTIONS, actions)] mod = parser.OFPFlowMod(datapath=dp, priority=priority, match=match, instructions=inst) else: raise ValueError("We only support OpenFlow 1.3 now") return mod def _send_flow(self, datapath, mod): datapath.send_msg(mod) def pushMplsFlow(self, dpid, pushLabel, srcIp, dstIp, outPortNo, queueId, pathType): assert dpid in self.dps dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS # print 'srcIp,dstIp: ',srcIp,dstIp match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) # match = parser.OFPMatch(eth_type=eth_IP,ipv4_src=('10.108.92.0','255.255.255.0'), # ipv4_dst=('10.108.93.0','255.255.255.0')) actions = [] actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions) if pathType == "main": self._send_flow(dp, mod) elif pathType == "backup": self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsFlow(self, dpid, popLabel, outPort, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod def swapMplsFlow(self, dpid, pushLabel, popLabel, outPortNo, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod def noMplsFlow(self, dpid, srcIp, dstIp, outPortNo, queueId, pathType): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP # eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) actions = [] actions.append(parser.OFPActionOutput(outPortNo)) actions.append(parser.OFPActionSetQueue(queueId)) mod = self._get_flow_mod(dpid, match, actions) if pathType == "main": self._send_flow(dp, mod) elif pathType == "backup": self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsMacFlow(self, dpid, popLabel, outPortNo, local_mac, next_mac, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) # print 'local_mac,next_mac: ',local_mac,next_mac actions.append(parser.OFPActionSetField(eth_src=local_mac)) ## actions.append(parser.OFPActionSetField(eth_dst=next_mac)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod #### def noMplsMacFlow(self, dpid, srcIp, dstIp, outPortNo, local_mac, next_mac, queueId, pathType): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP # eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) actions = [] actions.append(parser.OFPActionOutput(outPortNo)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionSetField(eth_src=local_mac)) ## actions.append(parser.OFPActionSetField(eth_dst=next_mac)) mod = self._get_flow_mod(dpid, match, actions) if pathType == "main": self._send_flow(dp, mod) elif pathType == "backup": self.logger.info("Backup path's First switch should not install flow now!") return match, mod @set_ev_cls(event.EventSwitchEnter, [MAIN_DISPATCHER, CONFIG_DISPATCHER]) def switchEnterHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid not in self.switches: self.switches.append(dpid) if dpid not in self.dps: self.dps[dpid] = dp assert len(self.switches) == len(self.dps) self.topo.addNode(dpid) self.logger.info("Switch %016x enter in local topo", dpid) if self.superExist: self._send_switch_enter_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_enter_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[TYPE] = "switchEnter" send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventSwitchLeave, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def switchLeaveHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid in self.switches: self.switches.remove(dpid) if dpid in self.dps: del self.dps[dpid] if dpid not in self.topo.nodes(): self.logger.warning("Switch %016x not in local topo", dpid) else: self.topo.removeNode(dpid) self.logger.info("Switch %016x leave the local topo", dpid) if self.superExist: self._send_switch_leave_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_leave_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "switchLeave" to_send[DPID] = dpid send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkAdd, [CONFIG_DISPATCHER, MAIN_DISPATCHER]) def linkAddHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no # print 'src_switch: ',src_switch,type(src_switch) edge = (src_switch, dst_switch) ######################################################################## if src_switch in self.dps: if edge not in self.topo.edges(): self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info( "Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port) ) else: assert src_switch not in self.dps # self.outDomainSwitch[(dst_switch, dst_port)] = (src_switch, src_port) if src_switch not in self.outDomainSwitch.values(): self.outDomainSwitch[dst_switch] = src_switch self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.reportPath() dst_switch = str(dst_switch) ##################################################################################### if self.superExist: self._send_link_add_msg(src_switch, src_port, dst_switch, dst_port) dst_switch = int(dst_switch) self.logger.info( "TO Super controller.Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port) ) def _send_link_add_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "linkAdd" to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkDelete, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def linkDelHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge in self.topo.edges(): self.topo.removeEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info( "Link delete: src %016x port_no %8d-> dst %016x port_no %8d" % (src_switch, src_port, dst_switch, dst_port) ) if self.superExist: self._send_link_delete_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info( "TO Super controller.Link Delete: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port) ) def _send_link_delete_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "linkDelete" to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def _monitor(self): hub.sleep(5) while True: for k in self.dps.keys(): # self._request_stats(dp) self._request_stats(self.dps[k]) hub.sleep(self.sleep) def _request_stats(self, datapath): self.logger.info("send stats request: %016x", datapath.id) ofproto = datapath.ofproto parser = datapath.ofproto_parser req = parser.OFPFlowStatsRequest(datapath) datapath.send_msg(req) req = parser.OFPPortStatsRequest(datapath, 0, ofproto.OFPP_ANY) datapath.send_msg(req) def _keepAlive(self): while True: if self.superExist: self._send_keep_alive_message() self.logger.info("send keep alive") hub.sleep(5) def _send_keep_alive_message(self): to_send = {} to_send[TYPE] = "keepAlive" to_send[DOMAINID] = self.domainId to_send[DOMAINWSGIIP] = self.domainWsgiIp to_send[DOMAINWSGIPORT] = self.domainWsgiPort send_message = json.dumps(to_send) # print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(ofp_event.EventOFPFlowStatsReply, MAIN_DISPATCHER) def _flow_stats_reply_handler(self, ev): body = ev.msg.body self.logger.info("datapath in_port eth_dst out_port packets bytes ") self.logger.info("---------------- -------- -------- -------- -------- -------- ") for stat in sorted( [flow for flow in body if flow.priority == 1], key=lambda flow: (flow.match["in_port"], flow.match["eth_dst"]), ): self.logger.info( "%016x %8x %17s %8x %8d %8d", ev.msg.datapath.id, stat.match["in_port"], stat.match["eth_dst"], stat.instructions[0].actions[0].port, stat.packet_count, stat.byte_count, ) @set_ev_cls(ofp_event.EventOFPPortStatsReply, MAIN_DISPATCHER) def _port_stats_reply_handler(self, ev): msg = ev.msg body = msg.body dpid = msg.datapath.id switchCollect = self.lastCollect.setdefault(dpid, SwitchStats(dpid)) ports = switchCollect.getPotrs() self.logger.info( "datapath port " "rx-pkts rx-bytes rx-error" "tx-pkts tx-bytes tx-error" ) self.logger.info( "----------- --------------" "------- -------- --------" "------- -------- --------" ) for stat in sorted(body, key=attrgetter("port_no")): self.logger.info( "%016x %8x %8d %8d %8d %8d %8d %8d ", ev.msg.datapath.id, stat.port_no, stat.rx_packets, stat.rx_bytes, stat.rx_errors, stat.tx_packets, stat.tx_bytes, stat.tx_errors, ) for stat in sorted(body, key=attrgetter("port_no")): portNo = stat.port_no if portNo < 10000: port = ports.setdefault(portNo, Portstats(portNo)) port.setFieds(stat, time.time()) if self.superExist: pass # self._send_port_stats(dpid) def _send_port_stats(self, dpid): send_message = self._make_port_stats_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_port_stats_message(self, dpid): assert dpid in self.lastCollect switchCollect = self.lastCollect[dpid] portsMsg = switchCollect.getPortMessageByRx_bytes() to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "portStats" to_send[DPID] = dpid to_send[PORTSTATS] = portsMsg send_message = json.dumps(to_send) # print send_message return send_message def sendTaskAssignReply(self, taskId, pathType): self.logger.info("send Task Assign Reply") send_message = self._make_task_assign_reply(taskId, pathType) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_assign_reply(self, taskId, pathType): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "taskAssignReply" to_send[TASK_ID] = taskId to_send[PATHTYPE] = pathType send_message = json.dumps(to_send) return send_message def sendTaskDeleteReply(self, taskId): self.logger.info("send Task Delete Reply") send_message = self._make_task_delete_reply(taskId) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_delete_reply(self, taskId): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "taskDeleteReply" to_send[TASK_ID] = taskId send_message = json.dumps(to_send) return send_message def _to_commad(self, send_message, returnType=False): command = "curl -X " if returnType: command += "GET -d '" else: command += "PUT -d '" command += send_message command += "' http://" command += self.superWsgiIp command += ":" command += str(self.superWsgiPort) command += super_url_no_reutrn command += " 2> /dev/null" return command # @staticmethod def send_no_return_command(self, command): print "command: ", command os.popen2(command) # print "to_print:", to_print @set_ev_cls(arpevent, dispatchers=None) def arp_handler(self, ev): ev_dict = self.object2dict(ev) self.borderPeer.extend(ev.dpid_table) self.borderPeer = list(set(self.borderPeer)) self.reportPath() ev_dict[TYPE] = "ArpMessage" # print ev_dict send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) # print 12333333333333333333 @set_ev_cls(mactoportEvent, dispatchers=None) def mac_to_port_handler(self, ev): ev_dict = self.object2dict(ev) ev_dict[TYPE] = "mactoportmessage" send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) def object2dict(self, obj): d = {} d["__class__"] = obj.__class__.__name__ d["__module__"] = obj.__module__ d.update(obj.__dict__) return d def reportPath(self): self.edgePort = list(set(self.borderPeer) | set(self.outDomainSwitch.keys())) tmpPathList = self.pathAnalysisByShortest(self.edgePort) self._send_path_report_msg(tmpPathList) print "self.edgePort:", self.edgePort print "tmpPathList:", tmpPathList # print 'tmpPathList: ',tmpPathList ############################################### def pathAnalysisByShortest(self, edgePort): pathList = [] if edgePort is None: LOG.info("This Domain have no edgePort!") return None # if len(edgePort) == 0 or len(edgePort) == 1: # return for i in edgePort: for m in edgePort: if i == m: continue tmppath = self.topo.getShortestPath(i, m) pathList.append([i, m, len(tmppath) - 1]) self.shortestPath[(i, m)] = tmppath self.logger.info("in_switch out_switch length ") self.logger.info("---------------- ---------------- -------- ") for stat in pathList: if len(stat) > 2: print ("%016x %016x %8d" % (stat[0], stat[1], stat[2])) self.logger.info("pathList end") print "self.edgePort", self.edgePort print "self.borderPeer", self.borderPeer print "self.outDomainSwitch", self.outDomainSwitch print "self.topoe.dges:", self.topo.edges() print "self.topo.nodes:", self.topo.nodes() # print 'self.shortestPath: ', self.shortestPath return pathList def _send_path_report_msg(self, pathList): self.logger.info("send Path Report Msg") send_message = self._make_path_report_msg(pathList) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_path_report_msg(self, pathList): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = "pathReport" to_send[PATHLIST] = pathList send_message = json.dumps(to_send) return send_message
class DomainController(app_manager.RyuApp): _CONTEXTS = {'wsgi': WSGIApplication, 'switches': switches.Switches, 'arp': IcmpResponder} OFP_VERSIONS = [ofproto_v1_0.OFP_VERSION, ofproto_v1_3.OFP_VERSION] def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = 'domain_controller' self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 10 self.local_mac = ['10.108.93.100', '10.108.90.201'] self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.num = 0 self.taskId_match = {} self.border_table = [] self.arp_table = {} self.port_to_localmac = {} self.timesOfQos = 0 self.qosSwitch = [] self.completeTopo = None ############################################ self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} # self.shortestPath = {(513,518):[513,515,518],(518,513):[518,515,513]} ##################################################### self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag # if self.monitorThreadFlag: # self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data) def _get_QueueQos(self, dpid): assert dpid in self.QoS_dict return self.QoS_dict[dpid] def _get_datapath(self, dpid): assert dpid in self.dps return self.dps[dpid] @set_ev_cls(ofp_event.EventOFPSwitchFeatures, CONFIG_DISPATCHER) def switch_features_handler(self, ev): msg = ev.msg datapath = msg.datapath self.logger.debug('switch features ev %s', msg) dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwitchFeatures(dpid)) switch.initFieds(msg.version, msg.capabilities, msg.n_buffers, msg.n_tables, msg.auxiliary_id) if dpid not in self.QoS_dict: queueQoSInstance = QueueQos(datapath, self.CONF) self.QoS_dict[dpid] = queueQoSInstance ip = datapath.address[0] ovs_addr = 'tcp:' + ip + ':' + DEFAULTOVSDBADDR queueQoSInstance.set_ovsdb_addr(dpid, ovs_addr) self.logger.info("QueueQoS Entering. dpid: %016x, addr: %s" % (dpid, ovs_addr)) #msg: {'body': [OFPPort(port_no=9,hw_addr='48:6e:73:02:03:08', # name='ge-1/1/9',config=0,state=1,curr=10240 @set_ev_cls(ofp_event.EventOFPPortDescStatsReply, CONFIG_DISPATCHER) def multipart_reply_handler(self, ev): msg = ev.msg datapath = msg.datapath dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwitchFeatures(dpid)) # self.num = self.num+1 # print self.num #ports = switch.getPorts() #print 'msg: ', msg.__dict__ for portdesc in msg.body: #print 'portdesc: ', portdesc portNo = portdesc.port_no if portNo > 10000: # print 11111111111111111111111111 name = portdesc.name switch.setName(name) else: port = PortFeatures(portNo) port.initFields(portdesc) #print 'port: ',port.__dict__ switch.ports[portNo] = port self.QoS_dict[dpid].queueInfo[portNo] = QueuePort(portNo, dpid) if self.superExist: self._send_switch_features_message(dpid) def _send_switch_features_message(self, dpid): send_message = self._make_switch_features_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_switch_features_message(self, dpid): assert dpid in self.deviceInfo switchFeaturs = self.deviceInfo[dpid] info = switchFeaturs.makeFeaturesMessage() to_send = {} to_send[TYPE] = 'switchFeature' to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[SWITCHFEATURE] = info send_message = json.dumps(to_send) return send_message def remove_flow(self, datapath, match): ofproto = datapath.ofproto parser = datapath.ofproto_parser if ofproto.OFP_VERSION == ofproto_v1_0.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_port=ofproto.OFPP_ANY, match=match) elif ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_group=ofproto.OFPG_ANY,out_port=ofproto.OFPP_ANY, match=match) datapath.send_msg(mod) def _get_ofproto_info(self, dpid): if dpid in self.dps: dp = self.dps[dpid] parser = dp.ofproto_parser ofproto = dp.ofproto return dp, parser, ofproto def _get_flow_mod(self, dpid, match, actions, priority=None, instruction=None): dp, parser, ofproto = self._get_ofproto_info(dpid) if not priority: priority = ofproto.OFP_DEFAULT_PRIORITY if ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: inst = [parser.OFPInstructionActions(ofproto.OFPIT_APPLY_ACTIONS, actions)] mod = parser.OFPFlowMod(datapath=dp, priority=priority, match=match, instructions=inst) else: raise ValueError('We only support OpenFlow 1.3 now') return mod def _send_flow(self, datapath, mod): datapath.send_msg(mod) def pushMplsFlow(self, dpid, pushLabel, srcIp, dstIp, outPortNo, queueId, pathType, src_mac,dst_mac,priority): assert dpid in self.dps dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS #print 'srcIp,dstIp: ',srcIp,dstIp match = parser.OFPMatch(eth_type=eth_IP,ipv4_src=srcIp,ipv4_dst=dstIp) #match = parser.OFPMatch(eth_type=eth_IP,ipv4_src=('10.108.92.0','255.255.255.0'), # ipv4_dst=('10.108.93.0','255.255.255.0')) actions = [] actions.append(parser.OFPActionSetField(eth_src=src_mac)) actions.append(parser.OFPActionSetField(eth_dst=dst_mac)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsFlow(self, dpid, popLabel, outPort, queueId, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions =[] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions, priority) self._send_flow(dp, mod) return match, mod def swapMplsFlow(self, dpid, pushLabel, popLabel, outPortNo, queueId, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) self._send_flow(dp, mod) return match, mod def noMplsFlow(self, dpid, srcIp, dstIp, outPortNo, queueId, pathType, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP #eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) actions = [] actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsMacFlow(self, dpid, popLabel, outPortNo, local_mac, next_mac, queueId, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions =[] #actions_inst = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) #local_mac = self.port_to_localmac[outPortNo] #print 'local_mac,next_mac: ',local_mac,next_mac #actions.append(parser.OFPActionSetField(eth_src=local_mac)) ## #actions.append(parser.OFPActionSetField(eth_dst=next_mac)) actions.append(parser.OFPActionOutput(outPortNo)) #inst = parser.OFPInstructionGotoTable(2) #actions_inst.append(inst) mod = self._get_flow_mod(dpid, match, actions, priority) #print 'mod: ', mod.__dict__ self._send_flow(dp, mod) return match, mod #### def noMplsMacFlow(self, dpid, srcIp, dstIp, outPortNo, local_mac, next_mac, queueId, pathType, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP #eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) actions = [] #actions_inst = [] #local_mac = self.port_to_localmac[outPortNo] actions.append(parser.OFPActionSetField(eth_src=local_mac)) ## actions.append(parser.OFPActionSetField(eth_dst=next_mac)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) #inst = (parser.OFPInstructionGotoTable(2)) mod = self._get_flow_mod(dpid, match, actions, priority) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod @set_ev_cls(event.EventSwitchEnter, [MAIN_DISPATCHER, CONFIG_DISPATCHER]) def switchEnterHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid not in self.switches: self.switches.append(dpid) if dpid not in self.dps: self.dps[dpid] = dp assert len(self.switches) == len(self.dps) self.topo.addNode(dpid) self.logger.info("Switch %d enter in local topo", dpid) if self.superExist: self._send_switch_enter_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %d", dpid) def _send_switch_enter_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[TYPE] = 'switchEnter' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventSwitchLeave, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def switchLeaveHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid in self.switches: self.switches.remove(dpid) if dpid in self.dps: del self.dps[dpid] if dpid not in self.topo.nodes(): self.logger.warning("Switch %d not in local topo", dpid) else: self.topo.removeNode(dpid) self.logger.info("Switch %d leave the local topo", dpid) if self.superExist: self._send_switch_leave_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %d", dpid) def _send_switch_leave_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'switchLeave' to_send[DPID] = dpid send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def changeToBackPath(self, dpid, nextdpid): # topo1 = copy.deepcopy(self.topo) # topo1.remove_edge(dpid, nextdpid) self.completeTopo = copy.deepcopy(self.topo) self.topo.topo.remove_edge(dpid, nextdpid) self.pathAnalysisByShortest(self.edgePort) @set_ev_cls(event.EventLinkAdd, [CONFIG_DISPATCHER, MAIN_DISPATCHER]) def linkAddHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no #print 'src_switch: ',src_switch,type(src_switch) edge = (src_switch, dst_switch) ######################################################################## if src_switch in self.dps: if edge not in self.topo.edges(): self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link add: src %d port_no %8d-> dst %d %8d" % (src_switch, src_port, dst_switch, dst_port)) else: assert src_switch not in self.dps print "src_switch:",src_switch print "outdomian:",self.outDomainSwitch #self.outDomainSwitch[(dst_switch, dst_port)] = (src_switch, src_port) if src_switch not in self.outDomainSwitch.values(): self.outDomainSwitch[dst_switch] = src_switch self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.reportPath() dst_switch = str(dst_switch) ##################################################################################### if self.superExist: self._send_link_add_msg(src_switch, src_port, dst_switch, dst_port) dst_switch = int(dst_switch) self.logger.info("TO Super controller.Link add: src %d port_no %8d-> dst %d %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_add_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkAdd' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkDelete, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def linkDelHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge in self.topo.edges(): self.topo.removeEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link delete: src %d port_no %8d-> dst %d port_no %8d" % (src_switch, src_port, dst_switch, dst_port)) if self.superExist: self._send_link_delete_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info("TO Super controller.Link Delete: src %d port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_delete_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkDelete' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def _monitor(self): hub.sleep(5) while True: for k in self.dps.keys(): # self._request_stats(dp) self._request_stats(self.dps[k]) hub.sleep(self.sleep) def _request_stats(self, datapath): self.logger.info('send stats request: %d', datapath.id) ofproto = datapath.ofproto parser = datapath.ofproto_parser #req = parser.OFPFlowStatsRequest(datapath) #datapath.send_msg(req) req = parser.OFPPortStatsRequest(datapath, 0, ofproto.OFPP_ANY) datapath.send_msg(req) def _keepAlive(self): while True: if self.superExist: self._send_keep_alive_message() self.logger.info("send keep alive") hub.sleep(5) def _send_keep_alive_message(self): to_send = {} to_send[TYPE] = 'keepAlive' to_send[DOMAINID] = self.domainId to_send[DOMAINWSGIIP] =self.domainWsgiIp to_send[DOMAINWSGIPORT] = self.domainWsgiPort send_message = json.dumps(to_send) #print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(ofp_event.EventOFPFlowStatsReply, MAIN_DISPATCHER) def _flow_stats_reply_handler(self, ev): body = ev.msg.body self.logger.info('datapath ' 'in_port eth_dst ' 'out_port packets bytes') self.logger.info('---------------- ' '-------- -------- ' '-------- -------- -------- ') for stat in sorted([flow for flow in body if flow.priority == 1], key=lambda flow: (flow.match['in_port'], flow.match['ipv4_dst'])): self.logger.info('%016x %8x %17s %8x %8d %8d', ev.msg.datapath.id, stat.match['in_port'], stat.match['ipv4_dst'], stat.instructions[0].actions[0].port, stat.packet_count, stat.byte_count) @set_ev_cls(ofp_event.EventOFPPortStatsReply, MAIN_DISPATCHER) def _port_stats_reply_handler(self, ev): msg = ev.msg body = msg.body dpid = msg.datapath.id switchCollect = self.lastCollect.setdefault(dpid, SwitchStats(dpid)) ports = switchCollect.getPotrs() self.logger.info('datapath port ' 'rx-pkts rx-bytes rx-error ' 'tx-pkts tx-bytes tx-error') self.logger.info('----------- ------ ' '------- -------- ----- ' '------ ------- --------') for stat in sorted(body, key=attrgetter('port_no')): if stat.port_no > 10000: continue self.logger.info('%016x %8d %8d %8d %8d %8d %8d %8d ', ev.msg.datapath.id, stat.port_no, stat.rx_packets, stat.rx_bytes, stat.rx_errors, stat.tx_packets, stat.tx_bytes, stat.tx_errors) for stat in sorted(body, key=attrgetter('port_no')): portNo = stat.port_no if portNo < 10000: port = ports.setdefault(portNo, Portstats(portNo)) port.setFieds(stat, time.time()) if self.superExist: self._send_port_stats(dpid) def _send_port_stats(self, dpid): send_message = self._make_port_stats_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_port_stats_message(self, dpid): assert dpid in self.lastCollect switchCollect = self.lastCollect[dpid] portsMsg = switchCollect.getPortMessageByRx_bytes() to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'portStats' to_send[DPID] = dpid to_send[PORTSTATS] = portsMsg send_message = json.dumps(to_send) return send_message def sendPathChangeReply(self, src, dst): to_send = {} to_send['src'] = src to_send['dst'] = dst to_send['domainId'] = self.domainId to_send['type'] = 'pathChangeReply' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def sendTaskAssignReply(self, taskId, pathType, src, dst): self.logger.info("send Task Assign Reply") send_message = self._make_task_assign_reply(taskId, pathType, src, dst) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_assign_reply(self, taskId, pathType, src, dst): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskAssignReply' to_send[TASK_ID] = taskId to_send[PATHTYPE] = pathType to_send['src'] = src to_send['dst'] = dst send_message = json.dumps(to_send) return send_message def sendTaskDeleteReply(self, taskId, src, dst): self.logger.info("send Task Delete Reply") send_message = self._make_task_delete_reply(taskId, src, dst) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_delete_reply(self, taskId, src, dst): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskDeleteReply' to_send[TASK_ID] = taskId to_send['src'] = src to_send['dst'] = dst send_message = json.dumps(to_send) return send_message def _to_commad(self, send_message, returnType=False): command = 'curl -X ' if returnType: command += 'GET -d \'' else: command += 'PUT -d \'' command += send_message command += '\' http://' command += self.superWsgiIp command += ':' command += str(self.superWsgiPort) command += super_url_no_return command += ' 2> /dev/null' return command # @staticmethod def send_no_return_command(self, command): print 'command: ',command os.popen2(command) # print "to_print:", to_print @set_ev_cls(arpevent, dispatchers=None) def arp_handler(self, ev): ev_dict = self.object2dict(ev) self.borderPeer.extend(ev.dpid_table) self.borderPeer = list(set(self.borderPeer)) self.reportPath() self.port_to_localmac = ev_dict['port_to_localmac'] self.arp_table.update(ev_dict['ip_to_mac']) ev_dict[TYPE] = 'ArpMessage' #print ev_dict send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) #print 12333333333333333333 @set_ev_cls(mactoportEvent, dispatchers=None) def mac_to_port_handler(self,ev): ev_dict = self.object2dict(ev) ev_dict[TYPE] = 'mactoportmessage' send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) def object2dict(self,obj): d={} d['__class__'] = obj.__class__.__name__ d['__module__'] = obj.__module__ d.update(obj.__dict__) return d def reportPath(self): self.edgePort = list(set(self.borderPeer) | set(self.outDomainSwitch.keys())) tmpPathList = self.pathAnalysisByShortest(self.edgePort) self._send_path_report_msg(tmpPathList) #print "self.edgePort:" , self.edgePort #print "tmpPathList:" , tmpPathList #print 'tmpPathList: ',tmpPathList ############################################### def pathAnalysisByShortest(self, edgePort): pathList=[] if edgePort is None: LOG.info("This Domain have no edgePort!") return None # if len(edgePort) == 0 or len(edgePort) == 1: # return print edgePort for i in edgePort: for m in edgePort: if i == m: continue tmppath = self.topo.getShortestPath(i, m) pathList.append([i, m, len(tmppath)-1]) self.shortestPath[(i, m)] = tmppath self.logger.info('in_switch out_switch length ') self.logger.info('---------------- ---------------- -------- ') for stat in pathList: if len(stat) > 2: pass #print ('%016x %016x %8d'%(stat[0], stat[1], stat[2])) self.logger.info("pathList end") # print "self.edgePort", self.edgePort # print "self.borderPeer", self.borderPeer # print "self.outDomainSwitch",self.outDomainSwitch # print "self.topoe.dges:",self.topo.edges() # print "self.topo.nodes:",self.topo.nodes() #print 'self.shortestPath: ', self.shortestPath return pathList def _send_path_report_msg(self, pathList): #self.logger.info("send Path Report Msg") send_message = self._make_path_report_msg(pathList) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_path_report_msg(self, pathList): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'pathReport' to_send[PATHLIST] = pathList send_message = json.dumps(to_send) return send_message
class DomainController(app_manager.RyuApp): _CONTEXTS = {'wsgi': WSGIApplication, 'switches': switches.Switches} OFP_VERSIONS = [ofproto_v1_0.OFP_VERSION, ofproto_v1_3.OFP_VERSION] def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.wsapi_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.deviceInfo = {} self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data) def _get_QueueQos(self, dpid): assert dpid in self.QoS_dict return self.QoS_dict[dpid] def _get_datapath(self, dpid): assert dpid in self.dps return self.dps[dpid] @set_ev_cls(ofp_event.EventOFPSwitchFeatures, CONFIG_DISPATCHER) def switch_features_handler(self, ev): msg = ev.msg datapath = msg.datapath self.logger.debug('switch features ev %s', msg) dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) switch.initFieds(msg.version, msg.capabilities, msg.n_buffers, msg.n_tables, msg.auxiliary_id) if dpid not in self.QoS_dict: queueQoSInstance = QueueQos(datapath, self.CONF) self.QoS_dict[dpid] = queueQoSInstance ip = datapath.address[0] ovs_addr = 'tcp:' + ip + ':' + DEFAULTOVSDBADDR queueQoSInstance.set_ovsdb_addr(dpid, ovs_addr) self.logger.info("QueueQoS Entering. dpid: %0x16, addr: %s" % (dpid, ovs_addr)) @set_ev_cls(ofp_event.EventOFPPortDescStatsReply, CONFIG_DISPATCHER) def multipart_reply_handler(self, ev): msg = ev.msg datapath = msg.datapath dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) ports = switch.getPorts() for portdesc in msg.body: portNo = portdesc.port_no if portNo > 10000: name = portdesc.name switch.setName(name) port = PortFeatures(portNo) port.initFields(portdesc) ports[portNo] = port if self.superExist: self._send_switch_features_message(dpid) def _send_switch_features_message(self, dpid): send_message = self._make_switch_features_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_switch_features_message(self, dpid): assert dpid in self.deviceInfo switchFeaturs = self.deviceInfo[dpid] info = switchFeaturs.makeFeaturesMessage() to_send = {} to_send[TYPE] = 'switchFeature' to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[SWITCHFEATURE] = info send_message = json.dumps(to_send) return send_message def remove_flow(self, datapath, match): ofproto = datapath.ofproto parser = datapath.ofproto_parser if ofproto.OFP_VERSION == ofproto_v1_0.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_port=ofproto.OFPP_ANY, match=match) elif ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_group=ofproto.OFPG_ANY,out_port=ofproto.OFPP_ANY, match=match) datapath.send_msg(mod) def _get_ofproto_info(self, dpid): assert dpid in self.dps dp = self.dps[dpid] parser = dp.ofproto_parser ofproto = dp.ofproto return dp, parser, ofproto def _get_flow_mod(self, dpid, match, actions, priority=None, buffer_id=None): dp, parser, ofproto = self._get_ofproto_info(dpid) if not priority: priority = ofproto.OFP_DEFAULT_PRIORITY if ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: inst = [parser.OFPInstructionActions(ofproto.OFPIT_APPLY_ACTIONS, actions)] mod = parser.OFPFlowMod(datapath=dp, priority=priority, match=match, instructions=inst) else: raise ValueError('We only support OpenFlow 1.3 now') return mod def _send_flow(self, datapath, mod): datapath.send_msg(mod) def pushMplsFlow(self, dpid, pushLabel, srcIp, dstIp, outPort, queueId, pathType): assert dpid in self.dps dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS #match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp,nw_src_mask=24, nw_dst_mask=24) actions = [] actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsFlow(self, dpid, popLabel, outPort, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS,mpls_label=popLabel) actions =[] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod def swapMplsFlow(self, dpid, pushLabel, popLabel, outPort, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod def noMplsFlow(self, dpid, srcIp, dstIp, outPort, queueId, pathType): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS #match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp,nw_src_mask=24, nw_dst_mask=24) actions = [] actions.append(parser.OFPActionOutput(outPort)) actions.append(parser.OFPActionSetQueue(queueId)) mod = self._get_flow_mod(dpid, match, actions) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsMacFlow(self, dpid, popLabel, outPort, next_mac, local_mac, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions =[] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) actions.append(parser.OFPActionSetField(eth_src=local_mac, eth_dst=next_mac)) ## mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod #### def noMplsMacFlow(self, dpid, srcIp, dstIp, outPort, next_mac, local_mac, queueId, pathType): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS #match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp) match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcIp, ipv4_dst=dstIp,nw_src_mask=24, nw_dst_mask=24) actions = [] actions.append(parser.OFPActionOutput(outPort)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionSetField(eth_src=local_mac, eth_dst=next_mac)) ## mod = self._get_flow_mod(dpid, match, actions) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod @set_ev_cls(event.EventSwitchEnter, [MAIN_DISPATCHER, CONFIG_DISPATCHER]) def switchEnterHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid not in self.switches: self.switches.append(dpid) if dpid not in self.dps: self.dps[dpid] = dp assert len(self.switches) == len(self.dps) self.topo.addNode(dpid) self.logger.info("Switch %d enter in local topo", dpid) if self.superExist: self._send_switch_enter_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %d", dpid) def _send_switch_enter_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[TYPE] = 'switchEnter' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventSwitchLeave, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def switchLeaveHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid in self.switches: self.switches.remove(dpid) if dpid in self.dps: del self.dps[dpid] if dpid not in self.topo.nodes(): self.logger.warning("Swtich %d not in local topo", dpid) else: self.topo.removeNode(dpid) self.logger.info("Swtich %d leave the local topo", dpid) if self.superExist: self._send_switch_leave_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %d", dpid) def _send_switch_leave_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'switchLeave' to_send[DPID] = dpid send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkAdd, [CONFIG_DISPATCHER, MAIN_DISPATCHER]) def linkAddHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge not in self.topo.edges(): self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link add: src %d port_no %8x-> dst %d %8x" % (src_switch, src_port, dst_switch, dst_port)) if self.superExist: self._send_link_add_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info("TO Super controller.Link add: src %d port_no %8d-> dst %d %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_add_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkAdd' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkDelete, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def linkDelHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge in self.topo.edges(): self.topo.removeEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link delete: src %d port_no %8d-> dst %d port_no %8d" % (src_switch, src_port, dst_switch, dst_port)) if self.superExist: self._send_link_delete_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info("TO Super controller.Link Delete: src %d port_no %8d-> dst %d %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_delete_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkDelete' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def _monitor(self): hub.sleep(5) while True: for k in self.dps.keys(): # self._request_stats(dp) self._request_stats(self.dps[k]) hub.sleep(self.sleep) def _request_stats(self, datapath): # self.logger.info('send stats request: %016x', datapath.id) ofproto = datapath.ofproto parser = datapath.ofproto_parser # req = parser.OFPFlowStatsRequest(datapath) # datapath.send_msg(req) req = parser.OFPPortStatsRequest(datapath, 0, ofproto.OFPP_ANY) datapath.send_msg(req) def _keepAlive(self): while True: if self.superExist: self._send_keep_alive_message() self.logger.info("send keep alive") hub.sleep(5) def _send_keep_alive_message(self): to_send = {} to_send[TYPE] = 'keepAlive' to_send[DOMAINID] = self.domainId to_send[DOMAINWSGIIP] =self.domainWsgiIp to_send[DOMAINWSGIPORT] = self.domainWsgiPort send_message = json.dumps(to_send) # print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(ofp_event.EventOFPFlowStatsReply, MAIN_DISPATCHER) def _flow_stats_reply_handler(self, ev): body = ev.msg.body self.logger.info('datapath in_port eth_dst out_port packets bytes ') self.logger.info('---------------- -------- -------- -------- -------- -------- ') for stat in sorted([flow for flow in body if flow.priority == 1], key=lambda flow: (flow.match['in_port'], flow.match['eth_dst'])): self.logger.info('%016x %8x %17s %8x %8d %8d', ev.msg.datapath.id, stat.match['in_port'], stat.match['eth_dst'], stat.instructions[0].actions[0].port, stat.packet_count, stat.byte_count) @set_ev_cls(ofp_event.EventOFPPortStatsReply, MAIN_DISPATCHER) def _port_stats_reply_handler(self, ev): msg = ev.msg body = msg.body dpid = msg.datapath.id switchCollect = self.lastCollect.setdefault(dpid, SwitchStats(dpid)) ports = switchCollect.getPotrs() for stat in sorted(body, key=attrgetter('port_no')): portNo = stat.port_no if portNo < 10000: port = ports.setdefault(portNo, Portstats(portNo)) port.setFieds(stat, time.time()) if self.superExist: pass # self._send_port_stats(dpid) def _send_port_stats(self, dpid): send_message = self._make_port_stats_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_port_stats_message(self, dpid): assert dpid in self.lastCollect switchCollect = self.lastCollect[dpid] portsMsg = switchCollect.getPortMessageByRx_bytes() to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'portStats' to_send[DPID] = dpid to_send[PORTSTATS] = portsMsg send_message = json.dumps(to_send) # print send_message return send_message def sendTaskAssignReply(self, taskId, pathType): self.logger.info("send Task Assign Reply") send_message = self._make_task_assign_reply(taskId, pathType) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_assign_reply(self, taskId, pathType): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskAssignReply' to_send[TASK_ID] = taskId to_send[PATHTYPE] = pathType send_message = json.dumps(to_send) return send_message def sendTaskDeleteReply(self, taskId): self.logger.info("send Task Delete Reply") send_message = self._make_task_delete_reply(taskId) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_delete_reply(self, taskId): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskDeleteReply' to_send[TASK_ID] = taskId send_message = json.dumps(to_send) return send_message def _to_commad(self, send_message, returnType=False): command = 'curl -X ' if returnType: command += 'GET -d \'' else: command += 'PUT -d \'' command += send_message command += '\' http://' command += self.superWsgiIp command += ':' command += str(self.superWsgiPort) command += super_url_no_reutrn command += ' 2> /dev/null' return command # @staticmethod def send_no_return_command(self, command): # print command os.popen2(command) # print "to_print:", to_print from ryu.app.arp import arpevent @set_ev_cls(arpevent, MAIN_DISPATCHER) def arp_handler(self, ev): ev_dict = self.object2dict(ev) send_msg = json.dumps(ev_dict) send_msg['TYPE'] = 'ArpMessage' command = self._to_commad(send_msg) self.send_no_return_command(command) def object2dict(self,obj): d={} d['__class__'] = obj.__class__.__name__ d['__module__'] = obj.__module__ d.update(obj.__dict__) return d
class DomainController(app_manager.RyuApp): _CONTEXTS = { 'wsgi': WSGIApplication, 'switches': switches.Switches, 'simple_switch_13': SimpleSwitch13 } OFP_VERSIONS = [ofproto_v1_0.OFP_VERSION, ofproto_v1_3.OFP_VERSION] def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = 'domain_controller' self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain ' + str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.edgePort = [] self.borderPeer = [] self.outDomainSwitch = {} self.shortestPath = {} #self.shortestPath = {(773,769):[773,770,769],(769,773):[769,770,773]} self.QoS_dict = {} self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data) def _get_QueueQos(self, dpid): assert dpid in self.QoS_dict return self.QoS_dict[dpid] def _get_datapath(self, dpid): assert dpid in self.dps return self.dps[dpid] @set_ev_cls(ofp_event.EventOFPSwitchFeatures, CONFIG_DISPATCHER) def switch_features_handler(self, ev): msg = ev.msg datapath = msg.datapath self.logger.debug('switch features ev %s', msg) dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) switch.initFieds(msg.version, msg.capabilities, msg.n_buffers, msg.n_tables, msg.auxiliary_id) if dpid not in self.QoS_dict: queueQoSInstance = QueueQos(datapath, self.CONF) self.QoS_dict[dpid] = queueQoSInstance ip = datapath.address[0] ovs_addr = 'tcp:' + ip + ':' + DEFAULTOVSDBADDR queueQoSInstance.set_ovsdb_addr(dpid, ovs_addr) self.logger.info("QueueQoS Entering. dpid: %016x, addr: %s" % (dpid, ovs_addr)) @set_ev_cls(ofp_event.EventOFPPortDescStatsReply, CONFIG_DISPATCHER) def multipart_reply_handler(self, ev): msg = ev.msg datapath = msg.datapath dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) #ports = switch.getPorts() for portdesc in msg.body: #print 'portdesc: ',portdesc portNo = portdesc.port_no if portNo > 10000: name = portdesc.name switch.setName(name) else: port = PortFeatures(portNo) port.initFields(portdesc) switch.ports[portNo] = port self.QoS_dict[dpid].queueInfo[portNo] = QueuePort(portNo, dpid) if self.superExist: self._send_switch_features_message(dpid) def _send_switch_features_message(self, dpid): send_message = self._make_switch_features_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_switch_features_message(self, dpid): assert dpid in self.deviceInfo switchFeaturs = self.deviceInfo[dpid] info = switchFeaturs.makeFeaturesMessage() to_send = {} to_send[TYPE] = 'switchFeature' to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[SWITCHFEATURE] = info send_message = json.dumps(to_send) return send_message def remove_flow(self, datapath, match): ofproto = datapath.ofproto parser = datapath.ofproto_parser if ofproto.OFP_VERSION == ofproto_v1_0.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_port=ofproto.OFPP_ANY, match=match) elif ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_group=ofproto.OFPG_ANY, out_port=ofproto.OFPP_ANY, match=match) datapath.send_msg(mod) def _get_ofproto_info(self, dpid): assert dpid in self.dps dp = self.dps[dpid] parser = dp.ofproto_parser ofproto = dp.ofproto return dp, parser, ofproto def _get_flow_mod(self, dpid, match, actions, priority=None, buffer_id=None): dp, parser, ofproto = self._get_ofproto_info(dpid) if not priority: priority = ofproto.OFP_DEFAULT_PRIORITY if ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: inst = [ parser.OFPInstructionActions(ofproto.OFPIT_APPLY_ACTIONS, actions) ] mod = parser.OFPFlowMod(datapath=dp, priority=priority, match=match, instructions=inst) else: raise ValueError('We only support OpenFlow 1.3 now') return mod def _send_flow(self, datapath, mod): datapath.send_msg(mod) def pushMplsFlow(self, dpid, pushLabel, srcMac, dstMac, outPortNo, queueId, pathType): assert dpid in self.dps dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, eth_src=srcMac, eth_dst=dstMac) actions = [] actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info( "Backup path's First switch should not install flow now!") return match, mod def popMplsFlow(self, dpid, popLabel, outPortNo, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod def swapMplsFlow(self, dpid, pushLabel, popLabel, outPort, queueId): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions) self._send_flow(dp, mod) return match, mod # def popMplsMacFlow(self, dpid, popLabel, outPort, queueId,local_mac,next_mac): # dp, parser, ofproto = self._get_ofproto_info(dpid) # eth_IP = ether.ETH_TYPE_IP # eth_MPLS = ether.ETH_TYPE_MPLS # match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) # # actions =[] # actions.append(parser.OFPActionPopMpls(eth_IP)) # actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) # actions.append(parser.OFPActionOutput(outPort)) # actions.append(parser.OFPActionSetField(eth_src=local_mac, eth_dst=next_mac)) ## # # mod = self._get_flow_mod(dpid, match, actions) # # self._send_flow(dp, mod) # return match, mod #### def noMplsFlow(self, dpid, srcMac, dstMac, outPortNo, queueId, pathType): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP #eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, eth_src=srcMac, eth_dst=dstMac) actions = [] actions.append(parser.OFPActionOutput(outPortNo)) actions.append(parser.OFPActionSetQueue(queueId)) mod = self._get_flow_mod(dpid, match, actions) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info( "Backup path's First switch should not install flow now!") return match, mod # def noMplsMacFlow(self, dpid, srcMac, dstMac, outPortNo,queueId, pathType): # dp, parser, ofproto = self._get_ofproto_info(dpid) # eth_IP = ether.ETH_TYPE_IP # #eth_MPLS = ether.ETH_TYPE_MPLS # match = parser.OFPMatch(eth_type=eth_IP, ipv4_src=srcMac, ipv4_dst=dstMac) # # actions = [] # actions.append(parser.OFPActionOutput(outPortNo)) # actions.append(parser.OFPActionSetQueue(queueId)) # # mod = self._get_flow_mod(dpid, match, actions) # # if pathType == 'main': # self._send_flow(dp, mod) # elif pathType == 'backup': # self.logger.info("Backup path's First switch should not install flow now!") # # return match, mod def reportPath(self): self.edgePort = list( set(self.borderPeer) | set(self.outDomainSwitch.keys())) tempPathList = self.BoardShortestPath(self.edgePort) print 'self.edgePort: ', self.edgePort self._send_path_report_msg(tempPathList) def BoardShortestPath(self, edgeSwitch): pathList = [] if edgeSwitch is None: LOG.info("This Domain have no edgePort!") return None # if len(edgeSwitch) == 1 or len(edgeSwitch) == 0: # return for i in edgeSwitch: for m in edgeSwitch: if i == m: continue temppath = self.topo.getShortestPath(i, m) #print 'temppath: ',temppath pathList.append([i, m, len(temppath) - 1]) self.shortestPath[(i, m)] = temppath self.logger.info('in_switch out_switch length ') self.logger.info('---------------- ---------------- -------- ') # for stat in pathList: # self.logger.info('%016x %016x %8d' % stat[0], stat[1], stat[2]) return pathList #Send pathList which created by the function named athAnalysisByShortest to SuperController def _send_path_report_msg(self, pathList): self.logger.info("send Path Report Msg") send_message = self._make_path_report_msg(pathList) command = self._to_commad(send_message) self.send_no_return_command(command) # Implement the message for SuperController path # At last, if SuperController exists,this function transmit the pathlist to SuperController. def _make_path_report_msg(self, pathList): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'pathReport' to_send[PATHLIST] = pathList send_message = json.dumps(to_send) return send_message @set_ev_cls(event.EventSwitchEnter, [MAIN_DISPATCHER, CONFIG_DISPATCHER]) def switchEnterHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid not in self.switches: self.switches.append(dpid) if dpid not in self.dps: self.dps[dpid] = dp assert len(self.switches) == len(self.dps) self.topo.addNode(dpid) self.logger.info("Switch %016x enter in local topo", dpid) if self.superExist: self._send_switch_enter_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_enter_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[TYPE] = 'switchEnter' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventSwitchLeave, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def switchLeaveHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid in self.switches: self.switches.remove(dpid) if dpid in self.dps: del self.dps[dpid] if dpid not in self.topo.nodes(): self.logger.warning("Switch %016x not in local topo", dpid) else: self.topo.removeNode(dpid) self.logger.info("Switch %016x leave the local topo", dpid) if self.superExist: self._send_switch_leave_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_leave_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'switchLeave' to_send[DPID] = dpid send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkAdd, [CONFIG_DISPATCHER, MAIN_DISPATCHER]) def linkAddHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if src_switch in self.dps: if edge not in self.topo.edges(): self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info( "Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) else: assert src_switch not in self.dps #self.outDomainSwitch[(dst_switch,dst_port)] = (src_switch, src_port) self.outDomainSwitch[dst_switch] = src_switch self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info( "Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) self.reportPath() dst_switch = str(dst_switch) if self.superExist: self._send_link_add_msg(src_switch, src_port, dst_switch, dst_port) dst_switch = int(dst_switch) self.logger.info( "TO Super controller.Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_add_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkAdd' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkDelete, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def linkDelHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge in self.topo.edges(): self.topo.removeEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info( "Link delete: src %016x port_no %8d-> dst %016x port_no %8d" % (src_switch, src_port, dst_switch, dst_port)) if self.superExist: self._send_link_delete_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info( "TO Super controller.Link Delete: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_delete_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkDelete' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def _monitor(self): hub.sleep(5) while True: for k in self.dps.keys(): # self._request_stats(dp) self._request_stats(self.dps[k]) hub.sleep(self.sleep) def _request_stats(self, datapath): # self.logger.info('send stats request: %016x', datapath.id) ofproto = datapath.ofproto parser = datapath.ofproto_parser # req = parser.OFPFlowStatsRequest(datapath) # datapath.send_msg(req) req = parser.OFPPortStatsRequest(datapath, 0, ofproto.OFPP_ANY) datapath.send_msg(req) def _keepAlive(self): while True: if self.superExist: self._send_keep_alive_message() self.logger.info("send keep alive") hub.sleep(5) def _send_keep_alive_message(self): to_send = {} to_send[TYPE] = 'keepAlive' to_send[DOMAINID] = self.domainId to_send[DOMAINWSGIIP] = self.domainWsgiIp to_send[DOMAINWSGIPORT] = self.domainWsgiPort send_message = json.dumps(to_send) print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(ofp_event.EventOFPFlowStatsReply, MAIN_DISPATCHER) def _flow_stats_reply_handler(self, ev): body = ev.msg.body self.logger.info( 'datapath in_port eth_dst out_port packets bytes ') self.logger.info( '---------------- -------- -------- -------- -------- -------- ') for stat in sorted([flow for flow in body if flow.priority == 1], key=lambda flow: (flow.match['in_port'], flow.match['eth_dst'])): self.logger.info('%016x %8x %17s %8x %8d %8d', ev.msg.datapath.id, stat.match['in_port'], stat.match['eth_dst'], stat.instructions[0].actions[0].port, stat.packet_count, stat.byte_count) @set_ev_cls(ofp_event.EventOFPPortStatsReply, MAIN_DISPATCHER) def _port_stats_reply_handler(self, ev): msg = ev.msg body = msg.body dpid = msg.datapath.id switchCollect = self.lastCollect.setdefault(dpid, SwitchStats(dpid)) ports = switchCollect.getPorts() for stat in sorted(body, key=attrgetter('port_no')): portNo = stat.port_no if portNo < 10000: port = ports.setdefault(portNo, Portstats(portNo)) port.setFieds(stat, time.time()) if self.superExist: pass # self._send_port_stats(dpid) def _send_port_stats(self, dpid): send_message = self._make_port_stats_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_port_stats_message(self, dpid): assert dpid in self.lastCollect switchCollect = self.lastCollect[dpid] portsMsg = switchCollect.getPortMessageByRx_bytes() to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'portStats' to_send[DPID] = dpid to_send[PORTSTATS] = portsMsg send_message = json.dumps(to_send) # print send_message return send_message def sendTaskAssignReply(self, taskId, pathType): self.logger.info("send Task Assign Reply") send_message = self._make_task_assign_reply(taskId, pathType) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_assign_reply(self, taskId, pathType): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskAssignReply' to_send[TASK_ID] = taskId to_send[PATHTYPE] = pathType send_message = json.dumps(to_send) return send_message def sendTaskDeleteReply(self, taskId): self.logger.info("send Task Delete Reply") send_message = self._make_task_delete_reply(taskId) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_delete_reply(self, taskId): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskDeleteReply' to_send[TASK_ID] = taskId send_message = json.dumps(to_send) return send_message def _to_commad(self, send_message, returnType=False): command = 'curl -X ' if returnType: command += 'GET -d \'' else: command += 'PUT -d \'' command += send_message command += '\' http://' command += self.superWsgiIp command += ':' command += str(self.superWsgiPort) command += super_url_no_reutrn command += ' 2> /dev/null' return command # @staticmethod def send_no_return_command(self, command): print 'command: ', command os.popen2(command) # try: # os.popen2(command) # except: # self.logger.debug('command exceute fail.Fail Command: %s' % command) # return # print "to_print:", to_print ## # @set_ev_cls(arpevent, MAIN_DISPATCHER) # def arp_handler(self, ev): # ev_dict = self.object2dict(ev) # send_msg = json.dumps(ev_dict) # send_msg['TYPE'] = 'ArpMessage' # command = self._to_commad(send_msg) # self.send_no_return_command(command) @set_ev_cls(mactoportEvent, dispatchers=None) def mac_to_port_handler(self, ev): ev_dict = self.object2dict(ev) for dpid in ev_dict['mac_to_port'].keys(): self.borderPeer.append(dpid) self.borderPeer = list(set(self.borderPeer)) self.reportPath() ev_dict[TYPE] = 'mactoportmessage' send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) def object2dict(self, obj): d = {} d['__class__'] = obj.__class__.__name__ d['__module__'] = obj.__module__ d.update(obj.__dict__) return d
class DomainController(app_manager.RyuApp): _CONTEXTS = {'wsgi': WSGIApplication, 'switches': switches.Switches} OFP_VERSIONS = [ofproto_v1_0.OFP_VERSION, ofproto_v1_3.OFP_VERSION] def __init__(self, *args, **kwargs): super(DomainController, self).__init__(*args, **kwargs) self.domainId = self.CONF.domain_id self.name = 'domain_controller' self.domainWsgiIp = self.CONF.domain_wsgi_ip self.domainWsgiPort = self.CONF.domain_port self.topo = DomainTopo(name='Domain '+str(self.domainId)) self.dps = {} self.switches = [] self.sleep = 2 self.TASK_LIST = {} self.LabelsPool = MplsLabelsPool() self.LabelsPool.initPool() self.deviceInfo = {} self.edgePort = [] self.borderPeer = [] self.timeOfQos = 0 self.qosSwitch = [] self.outDomainSwitch = {} self.shortestPath = {} self.taskId_match = {} self.QoS_dict = {} self.completeTopo = None self.superExist = self.CONF.super_exist if self.superExist: self.superWsgiIp = self.CONF.super_wsgi_ip self.superWsgiPort = self.CONF.super_wsgi_port self.superLastEcho = time.time() self.monitorThreadFlag = self.CONF.monitor_thread_flag if self.monitorThreadFlag: #self.monitorThread = hub.spawn(self._monitor) self.lastCollect = {} self.keepAliveThread = hub.spawn(self._keepAlive) wsgi = kwargs['wsgi'] data = {} data[DOMAINCONTROLLER] = self data[DomainReplyController] = DomainReplyController() wsgi.register(DomainWsgiController, data) def _keepAlive(self): while True: if self.superExist: self._send_keep_alive_message() self.logger.info("send keep alive") hub.sleep(5) def _send_keep_alive_message(self): to_send = {} to_send[TYPE] = 'keepAlive' to_send[DOMAINID] = self.domainId to_send[DOMAINWSGIIP] =self.domainWsgiIp to_send[DOMAINWSGIPORT] = self.domainWsgiPort send_message = json.dumps(to_send) print send_message command = self._to_commad(send_message) self.send_no_return_command(command) def _monitor(self): hub.sleep(5) while True: for k in self.dps.keys(): self._request_stats(self.dps[k]) hub.sleep(self.sleep) def _request_stats(self, datapath): self.logger.info('send stats request: %016x', datapath.id) ofproto = datapath.ofproto parser = datapath.ofproto_parser req = parser.OFPPortStatsRequest(datapath, 0, ofproto.OFPP_ANY) datapath.send_msg(req) @set_ev_cls(ofp_event.EventOFPFlowStatsReply, MAIN_DISPATCHER) def _flow_stats_reply_handler(self, ev): body = ev.msg.body self.logger.info('datapath in_port eth_dst out_port packets bytes ') self.logger.info('---------------- -------- -------- -------- -------- -------- ') for stat in sorted([flow for flow in body if flow.priority == 1], key=lambda flow: (flow.match['in_port'], flow.match['eth_dst'])): self.logger.info('%016x %8x %17s %8x %8d %8d', ev.msg.datapath.id, stat.match['in_port'], stat.match['eth_dst'], stat.instructions[0].actions[0].port, stat.packet_count, stat.byte_count) @set_ev_cls(ofp_event.EventOFPPortStatsReply, MAIN_DISPATCHER) def _port_stats_reply_handler(self, ev): msg = ev.msg body = msg.body dpid = msg.datapath.id switchCollect = self.lastCollect.setdefault(dpid, SwitchStats(dpid)) ports = switchCollect.getPorts() self.logger.info('datapath port ' 'rx-pkts rx-bytes rx-error ' 'tx-pkts tx-bytes tx-error') self.logger.info('----------- ------ ' '------- ------- ----- ' '------ ------- --------') for stat in sorted(body, key=attrgetter('port_no')): if stat.port_no > 10000: continue self.logger.info('%016x %8d %8d %8d %8d %8d %8d %8d ', ev.msg.datapath.id, stat.port_no, stat.rx_packets, stat.rx_bytes, stat.rx_errors, stat.tx_packets, stat.tx_bytes, stat.tx_errors) for stat in sorted(body, key=attrgetter('port_no')): portNo = stat.port_no if portNo < 10000: port = ports.setdefault(portNo, Portstats(portNo)) port.setFieds(stat, time.time()) if self.superExist: self._send_port_stats(dpid) def _send_port_stats(self, dpid): send_message = self._make_port_stats_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_port_stats_message(self, dpid): assert dpid in self.lastCollect switchCollect = self.lastCollect[dpid] portsMsg = switchCollect.getPortMessageByRx_bytes() to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'portStats' to_send[DPID] = dpid to_send[PORTSTATS] = portsMsg send_message = json.dumps(to_send) return send_message def _get_QueueQos(self, dpid): assert dpid in self.QoS_dict return self.QoS_dict[dpid] def _get_datapath(self, dpid): assert dpid in self.dps return self.dps[dpid] @set_ev_cls(ofp_event.EventOFPSwitchFeatures, CONFIG_DISPATCHER) def switch_features_handler(self, ev): msg = ev.msg datapath = msg.datapath self.logger.debug('switch features ev %s', msg) dpid = datapath.id #print 'dpid in domain: ', dpid switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) switch.initFieds(msg.version, msg.capabilities, msg.n_buffers, msg.n_tables, msg.auxiliary_id) if dpid not in self.QoS_dict: queueQoSInstance = QueueQos(datapath, self.CONF) self.QoS_dict[dpid] = queueQoSInstance ip = datapath.address[0] ovs_addr = 'tcp:' + ip + ':' + DEFAULTOVSDBADDR queueQoSInstance.set_ovsdb_addr(dpid, ovs_addr) self.logger.info("QueueQoS Entering. dpid: %016x, addr: %s" % (dpid, ovs_addr)) @set_ev_cls(ofp_event.EventOFPPortDescStatsReply, CONFIG_DISPATCHER) def multipart_reply_handler(self, ev): msg = ev.msg datapath = msg.datapath dpid = datapath.id switch = self.deviceInfo.setdefault(dpid, SwtichFeatures(dpid)) for portdesc in msg.body: #print 'portdesc: ',portdesc portNo = portdesc.port_no if portNo > 10000: name = portdesc.name switch.setName(name) else: port = PortFeatures(portNo) port.initFields(portdesc) switch.ports[portNo] = port self.QoS_dict[dpid].queueInfo[portNo] = QueuePort(portNo,dpid) if self.superExist: self._send_switch_features_message(dpid) def _send_switch_features_message(self, dpid): send_message = self._make_switch_features_message(dpid) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_switch_features_message(self, dpid): assert dpid in self.deviceInfo switchFeaturs = self.deviceInfo[dpid] info = switchFeaturs.makeFeaturesMessage() to_send = {} to_send[TYPE] = 'switchFeature' to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[SWITCHFEATURE] = info send_message = json.dumps(to_send) return send_message @set_ev_cls(event.EventSwitchEnter, [MAIN_DISPATCHER, CONFIG_DISPATCHER]) def switchEnterHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid not in self.switches: self.switches.append(dpid) if dpid not in self.dps: self.dps[dpid] = dp assert len(self.switches) == len(self.dps) self.topo.addNode(dpid) self.logger.info("Switch %016x enter in local topo", dpid) if self.superExist: self._send_switch_enter_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_enter_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[DPID] = dpid to_send[TYPE] = 'switchEnter' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkAdd, [CONFIG_DISPATCHER, MAIN_DISPATCHER]) def linkAddHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if src_switch in self.dps: if edge not in self.topo.edges(): self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) #print 'self.linkWithPort: ', self.topo.linkWithPort self.logger.info("Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) else: assert src_switch not in self.dps #self.outDomainSwitch[(dst_switch,dst_port)] = (src_switch, src_port) self.outDomainSwitch[dst_switch] = src_switch self.topo.addEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) self.reportPath() dst_switch = str(dst_switch) if self.superExist: self._send_link_add_msg(src_switch, src_port, dst_switch, dst_port) dst_switch = int(dst_switch) self.logger.info("TO Super controller.Link add: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_add_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkAdd' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) print send_message command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventSwitchLeave, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def switchLeaveHandler(self, ev): switch = ev.switch dp = switch.dp dpid = dp.id if dpid in self.switches: self.switches.remove(dpid) if dpid in self.dps: del self.dps[dpid] if dpid not in self.topo.nodes(): self.logger.warning("Switch %016x not in local topo", dpid) else: self.topo.removeNode(dpid) self.logger.info("Switch %016x leave the local topo", dpid) if self.superExist: self._send_switch_leave_msg(dpid) self.logger.info("To super controller: Switch Enter->dpid %016x", dpid) def _send_switch_leave_msg(self, dpid): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'switchLeave' to_send[DPID] = dpid send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) @set_ev_cls(event.EventLinkDelete, [MAIN_DISPATCHER, DEAD_DISPATCHER]) def linkDelHandler(self, ev): link = ev.link src = link.src dst = link.dst src_switch = src.dpid dst_switch = dst.dpid src_port = src.port_no dst_port = dst.port_no edge = (src_switch, dst_switch) if edge in self.topo.edges(): self.topo.removeEdge(src_switch, src_port, dst_switch, dst_port) self.logger.info("Link delete: src %016x port_no %8d-> dst %016x port_no %8d" % (src_switch, src_port, dst_switch, dst_port)) if self.superExist: self._send_link_delete_msg(src_switch, src_port, dst_switch, dst_port) self.logger.info("TO Super controller.Link Delete: src %016x port_no %8d-> dst %016x %8d" % (src_switch, src_port, dst_switch, dst_port)) def _send_link_delete_msg(self, ss, sp, ds, dp): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'linkDelete' to_send[SRC_SWITCH] = ss to_send[SRC_PORT] = sp to_send[DST_SWITCH] = ds to_send[DST_PORT] = dp send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def _get_ofproto_info(self, dpid): assert dpid in self.dps dp = self.dps[dpid] parser = dp.ofproto_parser ofproto = dp.ofproto return dp, parser, ofproto def _get_flow_mod(self, dpid, match, actions, priority=None, buffer_id=None): dp, parser, ofproto = self._get_ofproto_info(dpid) if not priority: priority = ofproto.OFP_DEFAULT_PRIORITY if ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: inst = [parser.OFPInstructionActions(ofproto.OFPIT_APPLY_ACTIONS, actions)] mod = parser.OFPFlowMod(datapath=dp, priority=priority, match=match, instructions=inst) else: raise ValueError('We only support OpenFlow 1.3 now') return mod def _send_flow(self, datapath, mod): datapath.send_msg(mod) def pushMplsFlow(self, dpid, pushLabel, srcMac, dstMac, outPortNo, queueId, pathType, priority): assert dpid in self.dps dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_IP, eth_src=srcMac, eth_dst=dstMac) actions = [] actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def popMplsFlow(self, dpid, popLabel, outPortNo,queueId,local_mac, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions =[] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(dp.ofproto_parser.OFPActionSetQueue(queueId)) #actions.append(parser.OFPActionSetField(eth_src=local_mac)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) self._send_flow(dp, mod) return match, mod def swapMplsFlow(self, dpid, pushLabel, popLabel, outPort, queueId, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP eth_MPLS = ether.ETH_TYPE_MPLS match = parser.OFPMatch(eth_type=eth_MPLS, mpls_label=popLabel) actions = [] actions.append(parser.OFPActionPopMpls(eth_IP)) actions.append(parser.OFPActionPushMpls(eth_MPLS)) f = parser.OFPMatchField.make(ofproto.OXM_OF_MPLS_LABEL, pushLabel) actions.append(parser.OFPActionSetField(f)) actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPort)) mod = self._get_flow_mod(dpid, match, actions, priority) self._send_flow(dp, mod) return match, mod def noMplsFlow(self, dpid, srcMac, dstMac, outPortNo, queueId, pathType, priority): dp, parser, ofproto = self._get_ofproto_info(dpid) eth_IP = ether.ETH_TYPE_IP match = parser.OFPMatch(eth_type=eth_IP, eth_src=srcMac, eth_dst=dstMac) actions = [] actions.append(parser.OFPActionSetQueue(queueId)) actions.append(parser.OFPActionOutput(outPortNo)) mod = self._get_flow_mod(dpid, match, actions, priority) if pathType == 'main': self._send_flow(dp, mod) elif pathType == 'backup': self.logger.info("Backup path's First switch should not install flow now!") return match, mod def changeToBackPath(self, dpid, nextdpid): #topo1 = copy.deepcopy(self.topo) self.completeTopo = copy.deepcopy(self.topo) self.topo.topo.remove_edge(dpid, nextdpid) self.BoardShortestPath(self.edgePort) @set_ev_cls(mactoportEvent, dispatchers=None) def mac_to_port_handler(self, ev): ev_dict = self.object2dict(ev) for dpid in ev_dict['mac_to_port'].keys(): self.borderPeer.append(dpid) self.borderPeer = list(set(self.borderPeer)) self.reportPath() ev_dict[TYPE] = 'mactoportmessage' send_msg = json.dumps(ev_dict) command = self._to_commad(send_msg) self.send_no_return_command(command) print "mac to port:", ev_dict def object2dict(self,obj): d={} d['__class__'] = obj.__class__.__name__ d['__module__'] = obj.__module__ d.update(obj.__dict__) return d def reportPath(self): self.edgePort = list(set(self.borderPeer) | set(self.outDomainSwitch.keys())) tempPathList = self.BoardShortestPath(self.edgePort) print 'self.edgePort: ', self.edgePort self._send_path_report_msg(tempPathList) def BoardShortestPath(self, edgeSwitch): pathList=[] if edgeSwitch is None: LOG.info("This Domain have no edgePort!") return None for i in edgeSwitch: for m in edgeSwitch: if i == m: continue temppath = self.topo.getShortestPath(i, m) #print 'temppath: ',temppath pathList.append([i, m, len(temppath)-1]) self.shortestPath[(i, m)] = temppath self.logger.info('in_switch out_switch length ') self.logger.info('---------------- ---------------- -------- ') # for stat in pathList: # self.logger.info('%016x %016x %8d' % stat[0], stat[1], stat[2]) print self.shortestPath return pathList #Send pathList which created by the function named athAnalysisByShortest to SuperController def _send_path_report_msg(self, pathList): self.logger.info("send Path Report Msg") send_message = self._make_path_report_msg(pathList) command = self._to_commad(send_message) self.send_no_return_command(command) # Implement the message for SuperController path # At last, if SuperController exists,this function transmit the pathlist to SuperController. def _make_path_report_msg(self, pathList): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'pathReport' to_send[PATHLIST] = pathList send_message = json.dumps(to_send) print "local_topo:", self.topo.edges() return send_message def sendPathChangeReply(self, src, dst): to_send = {} to_send['src'] = src to_send['dst'] = dst to_send['domainId'] = self.domainId to_send['type'] = 'pathChangeReply' send_message = json.dumps(to_send) command = self._to_commad(send_message) self.send_no_return_command(command) def sendTaskAssignReply(self, taskId, pathType, src, dst): self.logger.info("send Task Assign Reply") send_message = self._make_task_assign_reply(taskId, pathType, src, dst) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_assign_reply(self, taskId, pathType, src, dst): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskAssignReply' to_send[TASK_ID] = taskId to_send[PATHTYPE] = pathType to_send['src'] = src to_send['dst'] = dst send_message = json.dumps(to_send) return send_message def sendTaskDeleteReply(self, taskId, src, dst): self.logger.info("send Task Delete Reply") send_message = self._make_task_delete_reply(taskId, src, dst) command = self._to_commad(send_message) self.send_no_return_command(command) def _make_task_delete_reply(self, taskId, src, dst): to_send = {} to_send[DOMAINID] = self.domainId to_send[TYPE] = 'taskDeleteReply' to_send[TASK_ID] = taskId to_send['src'] = src to_send['dst'] = dst send_message = json.dumps(to_send) return send_message def remove_flow(self, datapath, match): ofproto = datapath.ofproto parser = datapath.ofproto_parser if ofproto.OFP_VERSION == ofproto_v1_0.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_port=ofproto.OFPP_ANY, match=match) elif ofproto.OFP_VERSION == ofproto_v1_3.OFP_VERSION: mod = parser.OFPFlowMod(datapath=datapath, command=ofproto.OFPFC_DELETE, out_group=ofproto.OFPG_ANY, out_port=ofproto.OFPP_ANY, match=match) datapath.send_msg(mod) def _to_commad(self, send_message, returnType=False): command = 'curl -X ' if returnType: command += 'GET -d \'' else: command += 'PUT -d \'' command += send_message command += '\' http://' command += self.superWsgiIp command += ':' command += str(self.superWsgiPort) command += super_url_no_reutrn command += ' 2> /dev/null' return command def send_no_return_command(self, command): print 'command: ', command os.popen2(command)