def arpRequestReplicas(self):
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['mac'] == '':
Globals.log.info("Issuing ARP Request for " + Globals.REPLICAS[i]['ip'])
arpRequest = Arps.create_arp_request(octstr_to_array(Globals.VMAC), ipstr_to_int(Globals.VIP), ipstr_to_int(Globals.REPLICAS[i]['ip']))
Multipath.flood(None, arpRequest, openflow.OFPP_FLOOD, openflow.OFPP_CONTROLLER)
self.post_callback(5, self.arpRequestReplicas)
def __init__(self, ctxt):
Component.__init__(self, ctxt)
self.Init = True
self.myEvalRules = EvalRules()
self.Multipath = Multipath(self)
self.myIPTransition = IPTransition(self, self.Multipath)
self.myLastAlphaMod = 0
self.AllMacsFound = False
def actionDeleteRule(rule):
Globals.INSTALLLOG.write("***Removing Rule: " + str(rule) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == rule['replica']:
(flow, defaultActions,
rewriteActions) = IPs.get_forwarding_dstrule(
ipstr_to_int(rule['ip']), rule['wild'], Globals.VIP,
Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'],
Globals.REPLICAS[i]['port'])
Multipath.delete_flow(flow, rule['replica'])
def actionTransitionRule(rule):
Globals.INSTALLLOG.write("***Transition Rule: " + str(rule) + '\n')
(flow, defaultActions,
rewriteActions) = IPs.get_controller_dstrule(ipstr_to_int(rule['ip']),
rule['wild'], Globals.VIP)
Multipath.install_controller_flow(flow, openflow.OFP_FLOW_PERMANENT,
openflow.OFP_FLOW_PERMANENT,
defaultActions, None,
openflow.OFP_DEFAULT_PRIORITY, 0, None,
rule['replica'], rewriteActions)
Globals.COMPONENT.post_callback(Globals.HARD_TIMEOUT,
lambda: hardTimeoutHandler(rule))
def actionInstallMicroflow(packet, replica):
Globals.INSTALLLOG.write("***Microflow " + ip_to_str(packet.next.srcip) +
" to replica " + str(replica) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == replica:
(flow, defaultActions, rewriteActions) = IPs.get_microflow_dstrule(
packet, Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'],
replica)
Multipath.install_replica_flow(flow, Globals.SOFT_TIMEOUT,
openflow.OFP_FLOW_PERMANENT,
defaultActions, None,
openflow.OFP_DEFAULT_PRIORITY + 10,
0, None, replica, rewriteActions)
def actionInstallRule(rule):
Globals.INSTALLLOG.write("***Installing Rule: " + str(rule) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == rule['replica']:
(flow, defaultActions,
rewriteActions) = IPs.get_forwarding_dstrule(
ipstr_to_int(rule['ip']), rule['wild'], Globals.VIP,
Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'],
Globals.REPLICAS[i]['port'])
Multipath.install_replica_flow(flow, openflow.OFP_FLOW_PERMANENT,
openflow.OFP_FLOW_PERMANENT,
defaultActions, None,
openflow.OFP_DEFAULT_PRIORITY, 0,
None, rule['replica'],
rewriteActions)
def packet_in_callback(self, dpid, inport, reason, len, bufid, packet):
if packet.type == Globals.ARP_TYPE:
# ARP Response from replicas for our ARP Requests
if packet.dst == octstr_to_array(Globals.VMAC):
self.foundMac(dpid, mac_to_str(packet.src), ip_to_str(packet.next.protosrc), inport)
# Request for VIP, respond with ARP Response
elif packet.next.protodst == ipstr_to_int(Globals.VIP):
Globals.STATSLOG.write(mac_to_str(dpid) + ' received ' + ip_to_str(packet.next.protosrc) + '\n')
srcIP = ip_to_str(packet.next.protosrc)
# Install Rewrite Rules
(flow, defaultActions, rewriteActions) = IPs.get_forwarding_srcrule(srcIP, mac_to_str(packet.src), Globals.VMAC, Globals.VIP, inport)
Multipath.install_microflow_flow(flow, openflow.OFP_FLOW_PERMANENT, openflow.OFP_FLOW_PERMANENT, defaultActions, None, openflow.OFP_DEFAULT_PRIORITY + 20, 0, None, dpid, inport, rewriteActions)
# Response
arpResponse = Arps.create_virtual_arp_response(packet, octstr_to_array(Globals.VMAC), ipstr_to_int(Globals.VIP))
Multipath.send(dpid, None, arpResponse, [[openflow.OFPAT_OUTPUT, [0, inport]]], openflow.OFPP_CONTROLLER)
ip = ip_to_str(packet.next.protosrc)
for i, switchJ in enumerate(Globals.SWITCHES):
if switchJ['mac'] == dpid:
found = False
for i, client in enumerate(switchJ['clients']):
if client['ip'] == ip:
found = True
if not found:
Globals.STATSLOG.write(mac_to_str(dpid) + ' found ' + ip + '\n')
switchJ['clients'].append({'port': inport, 'ip': ip, 'oldCount': 0L, 'newCount': 0L, 'avg': 0L})
elif packet.src != octstr_to_array(Globals.VMAC):
Globals.STATSLOG.write(mac_to_str(dpid) + ' received REQ ' + ip_to_str(packet.next.protosrc) + '\n')
arpResponse = Arps.create_arp_response(packet)
Multipath.send(dpid, None, arpResponse, [[openflow.OFPAT_OUTPUT, [0, inport]]], openflow.OFPP_CONTROLLER)
elif packet.type == Globals.IP_TYPE:
if reason == openflow.OFPR_ACTION:
IPTransition.handleControllerAction(packet)
def alphaFileUpdate():
Globals.printNewPeriod()
Globals.PERIOD += 1
rulePairList = EvalRules.updateAlphas()
IPTransition.handleRules(rulePairList)
IPTransition.printTargetInstallPairs(Globals.TARGETRULES, Globals.INSTALLEDRULES, Globals.TRANSITRULES)
def hardTimeoutHandler(timeoutRule):
Globals.printNewPeriod()
Globals.PERIOD += 1
IPTransition.handleTimeoutRule(timeoutRule)
IPTransition.printTargetInstallPairs(Globals.TARGETRULES, Globals.INSTALLEDRULES, Globals.TRANSITRULES)
# Init Forwarding Table
Multipath.calcForwardingTable()
# NEW ALPHAS
os.system('cp ' + Globals.HOME + 'alpha1.txt ' + Globals.HOME + 'alpha.txt')
time.sleep(1)
alphaFileUpdate()
Globals.TARGETRULES[0]['traffic'] = 99
alphaFileUpdate()
Globals.TARGETRULES[0]['traffic'] = 1
# NEW ALPHAS
os.system('cp ' + Globals.HOME + 'alpha2.txt ' + Globals.HOME + 'alpha.txt')
time.sleep(1)
class lbtest(Component):
# ==========================================
# INIT
# ==========================================
def __init__(self, ctxt):
Component.__init__(self, ctxt)
self.Init = True
self.myEvalRules = EvalRules()
self.Multipath = Multipath(self)
self.myIPTransition = IPTransition(self, self.Multipath)
self.myLastAlphaMod = 0
self.AllMacsFound = False
# ==========================================
# Check Alpha Callback
# ==========================================
def checkAlphaFile(self):
lastAlphaMod = os.path.getmtime(Globals.ALPHAFILE)
if lastAlphaMod > self.myLastAlphaMod:
Globals.RULESLOG.write('Alpha File Modification')
self.myLastAlphaMod = lastAlphaMod
rulePairList = self.myEvalRules.updateAlphas()
self.myIPTransition.installRules(rulePairList)
self.post_callback(2, self.checkAlphaFile)
# ==========================================
# Packet In Callback
# ==========================================
def packet_in_callback(self, dpid, inport, reason, len, bufid, packet):
# Install initial setup forwarding rules
if self.Init == True:
self.post_callback(Globals.PORT_STATS_PERIOD, lambda : self.port_timer())
self.Init = False
# Controller Attention
if reason == openflow.OFPR_ACTION:
self.myIPTransition.controllerAction(packet)
# Incoming ARPs
elif packet.type == Globals.ARP_TYPE:
# Response to our ARP requests
# 1. Record MAC
# 2. Install appropriate IP Forwarding Rule: Change dest VIP to dest Replica i
if packet.dst == octstr_to_array(Globals.VMAC):
foundMacs = True
for i in range(0, Globals.NUMREPLICAS):
if ((packet.next.protosrc == ipstr_to_int(Globals.REPLICAS[i]['ip'])) and (Globals.REPLICAS[i]['mac'] != mac_to_str(packet.src))):
Globals.REPLICAS[i]['mac'] = mac_to_str(packet.src)
Globals.REPLICAS[i]['port'] = inport
for j, switchJ in enumerate(Globals.SWITCHES):
if switchJ['mac'] == dpid:
switchJ['replicas'].append(Globals.REPLICAS[i]['no'])
if (Globals.REPLICAS[i]['mac'] == ''):
foundMacs = False
if foundMacs == True and self.AllMacsFound == False:
Globals.log.info('REPLICAS ' + str(Globals.REPLICAS))
Globals.log.info('SWITCHES ' + str(Globals.SWITCHES))
Globals.log.info('\n')
Globals.log.info('>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
rulePairList = self.myEvalRules.updateAlphas()
self.myIPTransition.installRules(rulePairList)
Globals.log.info('<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<')
Globals.log.info('\n')
self.AllMacsFound = True
# Requests for VIP respond with ARP Response
elif packet.next.protodst == ipstr_to_int(Globals.VIP):
srcIP = ip_to_str(packet.next.protosrc)
# IP Rules
(flow, defaultActions, rewriteActions) = IPs.get_forwarding_srcrule(srcIP, mac_to_str(packet.src), Globals.VMAC, Globals.VIP, inport)
self.Multipath.install_microflow_flow(flow, openflow.OFP_FLOW_PERMANENT, openflow.OFP_FLOW_PERMANENT, defaultActions, None, openflow.OFP_DEFAULT_PRIORITY, 0, None, dpid, inport, rewriteActions)
arpResponse = Arps.create_virtual_arp_response(packet, octstr_to_array(Globals.VMAC), ipstr_to_int(Globals.VIP))
self.Multipath.send(dpid, None, arpResponse, [[openflow.OFPAT_OUTPUT, [0, inport]]], openflow.OFPP_CONTROLLER)
else:
arpResponse = Arps.create_arp_response(packet)
self.Multipath.send(dpid, None, arpResponse, [[openflow.OFPAT_OUTPUT, [0, inport]]], openflow.OFPP_CONTROLLER)
elif packet.type == Globals.IP_TYPE:
Globals.log.info("Warning! S1 SHOULD NOT BE RECEIVING IP Traffic!!!")
# self.myIPTransition.controllerAction(packet)
return CONTINUE
# =======================================
# ARP Request Replicas to retrieve MACs
# =======================================
def arpRequestReplicas(self):
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['mac'] == '':
arpRequest = Arps.create_arp_request(octstr_to_array(Globals.VMAC), ipstr_to_int(Globals.VIP), ipstr_to_int(Globals.REPLICAS[i]['ip']))
self.Multipath.flood(None, arpRequest, openflow.OFPP_FLOOD, openflow.OFPP_CONTROLLER)
self.post_callback(5, self.arpRequestReplicas)
# =======================================
# Install Events
# =======================================
def install(self):
self.register_for_packet_in(self.packet_in_callback)
self.post_callback(3, self.arpRequestReplicas)
self.post_callback(5, self.checkAlphaFile)
self.register_for_port_stats_in(self.port_stats_in_handler)
def getInterface(self):
return str(lbtest)
# =======================================
# Port Stats
# =======================================
def port_stats_in_handler(self, dpid, ports):
for port, info in enumerate(ports):
timer = str(time.time())
portno = str(ports[port]['port_no'])
rxpackets = str(ports[port]['rx_packets'])
rxbytes = str(ports[port]['rx_bytes'])
txpackets = str(ports[port]['tx_packets'])
txbytes = str(ports[port]['tx_bytes'])
Globals.STATSFILE.write(portno + ':' + rxpackets + ':' + rxbytes + ':' + txpackets + ':' + txbytes + ':' + timer + '\n')
def port_timer(self):
self.ctxt.send_port_stats_request(Globals.SWITCHES[0]['mac'])
self.post_callback(Globals.PORT_STATS_PERIOD, lambda : self.port_timer())
def actionDeleteRule(rule):
Globals.INSTALLLOG.write("***Removing Rule: " + str(rule) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == rule['replica']:
(flow, defaultActions, rewriteActions) = IPs.get_forwarding_dstrule(ipstr_to_int(rule['ip']), rule['wild'], Globals.VIP, Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'], Globals.REPLICAS[i]['port'])
Multipath.delete_flow(flow, rule['replica'])
def actionTransitionRule(rule):
Globals.INSTALLLOG.write("***Transition Rule: " + str(rule) + '\n')
(flow, defaultActions, rewriteActions) = IPs.get_controller_dstrule(ipstr_to_int(rule['ip']), rule['wild'], Globals.VIP)
Multipath.install_controller_flow(flow, openflow.OFP_FLOW_PERMANENT, openflow.OFP_FLOW_PERMANENT, defaultActions, None, openflow.OFP_DEFAULT_PRIORITY, 0, None, rule['replica'], rewriteActions)
Globals.COMPONENT.post_callback(Globals.HARD_TIMEOUT, lambda : hardTimeoutHandler(rule))
def actionInstallRule(rule):
Globals.INSTALLLOG.write("***Installing Rule: " + str(rule) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == rule['replica']:
(flow, defaultActions, rewriteActions) = IPs.get_forwarding_dstrule(ipstr_to_int(rule['ip']), rule['wild'], Globals.VIP, Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'], Globals.REPLICAS[i]['port'])
Multipath.install_replica_flow(flow, openflow.OFP_FLOW_PERMANENT, openflow.OFP_FLOW_PERMANENT, defaultActions, None, openflow.OFP_DEFAULT_PRIORITY, 0, None, rule['replica'], rewriteActions)
def actionInstallMicroflow(packet, replica):
Globals.INSTALLLOG.write("***Microflow " + ip_to_str(packet.next.srcip) + " to replica " + str(replica) + '\n')
for i in range(0, Globals.NUMREPLICAS):
if Globals.REPLICAS[i]['no'] == replica:
(flow, defaultActions, rewriteActions) = IPs.get_microflow_dstrule(packet, Globals.REPLICAS[i]['mac'], Globals.REPLICAS[i]['ip'], replica)
Multipath.install_replica_flow(flow, Globals.SOFT_TIMEOUT, openflow.OFP_FLOW_PERMANENT, defaultActions, None, openflow.OFP_DEFAULT_PRIORITY + 10, 0, None, replica, rewriteActions)
def __init__(self, ctxt):
Component.__init__(self, ctxt)
Globals.COMPONENT = self
Multipath.calcForwardingTable()
