168 lines
5.7 KiB
Python
168 lines
5.7 KiB
Python
from mpConfig import MpConfig
|
|
from mpECMPSingleInterfaceTopo import MpECMPSingleInterfaceTopo
|
|
from mpParamTopo import MpParamTopo
|
|
from mpTopo import MpTopo
|
|
from struct import *
|
|
|
|
class MpECMPSingleInterfaceConfig(MpConfig):
|
|
def __init__(self, topo, param):
|
|
MpConfig.__init__(self, topo, param)
|
|
|
|
def configureRoute(self):
|
|
i = 0
|
|
mask = len(self.topo.routers) - 1
|
|
for l in self.topo.routers:
|
|
cmd = self.getIptableRuleICMP(mask, i)
|
|
self.topo.commandTo(self.client, cmd)
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
cmd = self.getIptableRuleTCPPortClient(mask, i)
|
|
self.topo.commandTo(self.client, cmd)
|
|
cmd = self.getIptableRuleTCPPortServer(mask, i)
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
cmd = self.getIpRuleCmd(i)
|
|
self.topo.commandTo(self.client, cmd)
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
cmd = self.getDefaultRouteCmd(self.getRouterIPClient(i),
|
|
i)
|
|
self.topo.commandTo(self.client, cmd)
|
|
cmd = self.getDefaultRouteCmd(self.getRouterIPServer(i),
|
|
i)
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
i = i + 1
|
|
|
|
###
|
|
cmd = self.addRouteDefaultSimple(self.getRouterIPServer(0))
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
cmd = self.addRouteDefaultSimple(self.getRouterIPClient(0))
|
|
self.topo.commandTo(self.client, cmd)
|
|
|
|
self.topo.commandTo(self.client, "ip route flush cache")
|
|
self.topo.commandTo(self.server, "ip route flush cache")
|
|
|
|
def getIptableRuleICMP(self, mask, id):
|
|
s = 'iptables -t mangle -A OUTPUT -m u32 --u32 ' + \
|
|
'"6&0xFF=0x1 && ' + \
|
|
'24&0x' + \
|
|
pack('>I',(mask)).encode('hex') + \
|
|
'=0x' + pack('>I',id).encode('hex') + \
|
|
'" -j MARK --set-mark ' + str(id + 1)
|
|
print (s)
|
|
return s
|
|
|
|
def getIptableRuleTCPPortClient(self, mask, id):
|
|
s = 'iptables -t mangle -A OUTPUT -m u32 --u32 ' + \
|
|
'"6&0xFF=0x6 && ' + \
|
|
'18&0x' + \
|
|
pack('>I',(mask)).encode('hex') + \
|
|
'=0x' + pack('>I',id).encode('hex') + \
|
|
'" -j MARK --set-mark ' + str(id + 1)
|
|
print (s)
|
|
return s
|
|
|
|
def getIptableRuleTCPPortServer(self, mask, id):
|
|
s = 'iptables -t mangle -A OUTPUT -m u32 --u32 ' + \
|
|
'"6&0xFF=0x6 && ' + \
|
|
'20&0x' + \
|
|
pack('>I',(mask)).encode('hex') + \
|
|
'=0x' + pack('>I',id).encode('hex') + \
|
|
'" -j MARK --set-mark ' + str(id + 1)
|
|
print (s)
|
|
return s
|
|
|
|
def getIpRuleCmd(self, id):
|
|
s = 'ip rule add fwmark ' + str(id + 1) + ' table ' + \
|
|
str(id + 1)
|
|
print(s)
|
|
return s
|
|
|
|
def getDefaultRouteCmd(self, via, id):
|
|
s = 'ip route add default via ' + via + ' table ' + str(id + 1)
|
|
print(s)
|
|
return s
|
|
|
|
def configureInterfaces(self):
|
|
self.client = self.topo.getHost(MpTopo.clientName)
|
|
self.server = self.topo.getHost(MpTopo.serverName)
|
|
self.routers = []
|
|
i = 0
|
|
netmask = "255.255.255.0"
|
|
for l in self.topo.routers:
|
|
self.routers.append(self.topo.getHost(
|
|
MpTopo.routerNamePrefix + str(i)))
|
|
cmd = self.interfaceUpCommand(
|
|
self.getRouterInterfaceLSwitch(i),
|
|
self.getRouterIPClient(i), netmask)
|
|
self.topo.commandTo(self.routers[-1] , cmd)
|
|
|
|
cmd = self.interfaceUpCommand(
|
|
self.getRouterInterfaceRSwitch(i),
|
|
self.getRouterIPServer(i), netmask)
|
|
self.topo.commandTo(self.routers[-1] , cmd)
|
|
|
|
i = i + 1
|
|
|
|
cmd = self.interfaceUpCommand(self.getClientInterface(0),
|
|
self.getClientIP(0), netmask)
|
|
self.topo.commandTo(self.client, cmd)
|
|
|
|
cmd = self.interfaceUpCommand(self.getServerInterface(),
|
|
self.getServerIP(), netmask)
|
|
self.topo.commandTo(self.server, cmd)
|
|
|
|
def getClientIP(self, interfaceID):
|
|
lSubnet = self.param.getParam(MpParamTopo.LSUBNET)
|
|
clientIP = lSubnet + str(interfaceID) + ".1"
|
|
return clientIP
|
|
|
|
def getClientSubnet(self, interfaceID):
|
|
lSubnet = self.param.getParam(MpParamTopo.LSUBNET)
|
|
clientSubnet = lSubnet + str(interfaceID) + ".0/24"
|
|
return clientSubnet
|
|
|
|
def getRouterIPClient(self, id):
|
|
lSubnet = self.param.getParam(MpParamTopo.LSUBNET)
|
|
routerIP = lSubnet + "0." + str(id + 2)
|
|
return routerIP
|
|
|
|
def getRouterIPServer(self, id):
|
|
rSubnet = self.param.getParam(MpParamTopo.RSUBNET)
|
|
routerIP = rSubnet + "0." + str(id + 2)
|
|
return routerIP
|
|
|
|
def getServerIP(self):
|
|
rSubnet = self.param.getParam(MpParamTopo.RSUBNET)
|
|
serverIP = rSubnet + "0.1"
|
|
return serverIP
|
|
|
|
def getClientInterfaceCount(self):
|
|
return 1
|
|
|
|
def getRouterInterfaceLSwitch(self, id):
|
|
return MpTopo.routerNamePrefix + str(id) + "-eth0"
|
|
|
|
def getRouterInterfaceRSwitch(self, id):
|
|
return MpTopo.routerNamePrefix + str(id) + "-eth1"
|
|
|
|
def getClientInterface(self, interfaceID):
|
|
return MpTopo.clientName + "-eth" + str(interfaceID)
|
|
|
|
def getServerInterface(self):
|
|
return MpTopo.serverName + "-eth0"
|
|
|
|
def getMidLeftName(self, id):
|
|
return MpTopo.routerNamePrefix + str(id)
|
|
|
|
def getMidRightName(self, id):
|
|
return MpTopo.switchNamePrefix + "1"
|
|
|
|
def getMidL2RInterface(self, id):
|
|
return self.getMidLeftName(id) + "-eth1"
|
|
|
|
def getMidR2LInterface(self, id):
|
|
return self.getMidRightName(id) + "-eth" + str(id+2)
|