75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
#!/usr/bin/python
|
|
|
|
from core.experience import Experience, ExperienceParameter, ExperienceParameter
|
|
from core.topo import Topo, TopoParameter
|
|
|
|
from mininet_builder import MininetBuilder
|
|
|
|
from experiences import EXPERIENCES
|
|
from topos import TOPO_CONFIGS, TOPOS
|
|
|
|
|
|
class Runner(object):
|
|
def __init__(self, builderType, topoParamFile, xpParamFile):
|
|
self.defParamXp(xpParamFile)
|
|
self.topoParam = TopoParameter(topoParamFile)
|
|
self.defBuilder(builderType)
|
|
self.defTopo()
|
|
self.defConfig()
|
|
self.startTopo()
|
|
self.runXp()
|
|
self.stopTopo()
|
|
|
|
def defParamXp(self, xpParamFile):
|
|
self.xpParam = ExperienceParameter(xpParamFile)
|
|
|
|
def defBuilder(self, builderType):
|
|
if builderType == Topo.mininetBuilder:
|
|
self.topoBuilder = MininetBuilder()
|
|
else:
|
|
raise Exception("I can not find the builder " +
|
|
builderType)
|
|
def defTopo(self):
|
|
t = self.topoParam.getParam(Topo.topoAttr)
|
|
if t in TOPOS:
|
|
self.Topo = TOPOS[t](self.topoBuilder, self.topoParam)
|
|
else:
|
|
raise Exception("Unknown topo: {}".format(t))
|
|
print(self.Topo)
|
|
|
|
def defConfig(self):
|
|
t = self.topoParam.getParam(Topo.topoAttr)
|
|
if t in TOPO_CONFIGS:
|
|
self.TopoConfig = TOPO_CONFIGS[t](self.Topo, self.topoParam)
|
|
else:
|
|
raise Exception("Unknown topo config: {}".format(t))
|
|
|
|
def startTopo(self):
|
|
self.Topo.startNetwork()
|
|
self.TopoConfig.configureNetwork()
|
|
|
|
def runXp(self):
|
|
xp = self.xpParam.getParam(ExperienceParameter.XPTYPE)
|
|
if xp in EXPERIENCES:
|
|
EXPERIENCES[xp](self.xpParam, self.Topo, self.TopoConfig)
|
|
else:
|
|
raise Exception("Unknown experience {}".format(xp))
|
|
|
|
def stopTopo(self):
|
|
self.Topo.stopNetwork()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="Minitopo, a wrapper of Mininet to run multipath experiments")
|
|
|
|
parser.add_argument("--topo_param_file", "-t", required=True,
|
|
help="path to the topo parameter file")
|
|
parser.add_argument("--experience_param_file", "-x", required=True,
|
|
help="path to the experience parameter file")
|
|
|
|
args = parser.parse_args()
|
|
# XXX Currently, there is no alternate topo builder...
|
|
Runner(Topo.mininetBuilder, args.topo_param_file, args.experience_param_file) |