168 lines
4.9 KiB
Python
168 lines
4.9 KiB
Python
from core.experience import Experience, ExperienceParameter
|
|
|
|
|
|
class MpPvAt(object):
|
|
def __init__(self, at, cmd):
|
|
self.at = at
|
|
self.cmd = cmd
|
|
self.delta = 0
|
|
|
|
def __str__(self):
|
|
return "Pv... at " + str(self.at) + "(" + str(self.delta) + \
|
|
") will be " + self.cmd
|
|
|
|
|
|
class ExperienceNCPV(Experience):
|
|
"""
|
|
NC PV : NetCat and Pipe Viewer
|
|
"""
|
|
SERVER_NC_LOG = "netcat_server"
|
|
CLIENT_NC_LOG = "netcat_client"
|
|
NC_BIN = "/usr/local/bin/nc"
|
|
PV_BIN = "/usr/local/bin/pv"
|
|
PING_OUTPUT = "ping.log"
|
|
|
|
def __init__(self, xpParamFile, mpTopo, mpConfig):
|
|
Experience.__init__(self, xpParamFile, mpTopo, mpConfig)
|
|
self.loadParam()
|
|
self.ping()
|
|
Experience.classicRun(self)
|
|
|
|
def ping(self):
|
|
self.mpTopo.commandTo(self.mpConfig.client, "rm " + \
|
|
ExperienceNCPV.PING_OUTPUT )
|
|
count = self.xpParam.getParam(ExperienceParameter.PINGCOUNT)
|
|
for i in range(0, self.mpConfig.getClientInterfaceCount()):
|
|
cmd = self.pingCommand(self.mpConfig.getClientIP(i),
|
|
self.mpConfig.getServerIP(), n = count)
|
|
self.mpTopo.commandTo(self.mpConfig.client, cmd)
|
|
|
|
def pingCommand(self, fromIP, toIP, n=5):
|
|
s = "ping -c " + str(n) + " -I " + fromIP + " " + toIP + \
|
|
" >> " + ExperienceNCPV.PING_OUTPUT
|
|
print(s)
|
|
return s
|
|
|
|
def loadParam(self):
|
|
self.pvg = self.xpParam.getParam(ExperienceParameter.PVG)
|
|
self.pvz = self.xpParam.getParam(ExperienceParameter.PVZ)
|
|
self.pvRateLimit = self.xpParam.getParam(ExperienceParameter.PVRATELIMIT)
|
|
self.ddibs = self.xpParam.getParam(ExperienceParameter.DDIBS)
|
|
self.ddobs = self.xpParam.getParam(ExperienceParameter.DDOBS)
|
|
self.ddcount = self.xpParam.getParam(ExperienceParameter.DDCOUNT)
|
|
self.ncServerPort = self.xpParam.getParam(ExperienceParameter.NCSERVERPORT)
|
|
self.pvRateLimit = self.xpParam.getParam(ExperienceParameter.PVRATELIMIT)
|
|
self.ncClientPort = []
|
|
for k in sorted(self.xpParam.paramDic):
|
|
if k.startswith(ExperienceParameter.NCCLIENTPORT):
|
|
port = self.xpParam.paramDic[k]
|
|
self.ncClientPort.append(port)
|
|
if len(self.ncClientPort) == 0:
|
|
d = self.xpParam.getParam(ExperienceParameter.NCCLIENTPORT)
|
|
self.ncClientPort.append(d)
|
|
self.loadPvAt()
|
|
|
|
def loadPvAt(self):
|
|
self.changePvAt = []
|
|
self.changePv = self.xpParam.getParam(ExperienceParameter.CHANGEPV)
|
|
if self.changePv != "yes":
|
|
print("Don't change pv rate...")
|
|
return
|
|
changePvAt = self.xpParam.getParam(ExperienceParameter.CHANGEPVAT)
|
|
if not isinstance(changePvAt, list):
|
|
changePvAt = [changePvAt]
|
|
for p in changePvAt:
|
|
tab = p.split(",")
|
|
if len(tab)==2:
|
|
o = MpPvAt(float(tab[0]), tab[1])
|
|
self.addPvAt(o)
|
|
else:
|
|
print("pv wrong line : " + p)
|
|
|
|
def addPvAt(self, p):
|
|
if len(self.changePvAt) == 0 :
|
|
p.delta = p.at
|
|
else:
|
|
if p.at > self.changePvAt[-1].at:
|
|
p.delta = p.at - self.changePvAt[-1].at
|
|
else:
|
|
print("Do not take into account " + p.__str__() + \
|
|
"because ooo !")
|
|
return
|
|
|
|
self.changePvAt.append(p)
|
|
|
|
def getPvChangeCmd(self):
|
|
cmd = ""
|
|
for p in self.changePvAt:
|
|
cmd = cmd + "sleep " + str(p.delta)
|
|
cmd = cmd + " && "
|
|
cmd = cmd + ExperienceNCPV.PV_BIN + " -R " + self.pvPid
|
|
cmd = cmd + " " + p.cmd + " && "
|
|
cmd = cmd + " true &"
|
|
return cmd
|
|
|
|
def prepare(self):
|
|
Experience.prepare(self)
|
|
self.mpTopo.commandTo(self.mpConfig.client, "rm " + \
|
|
ExperienceNCPV.CLIENT_NC_LOG )
|
|
self.mpTopo.commandTo(self.mpConfig.server, "rm " + \
|
|
ExperienceNCPV.SERVER_NC_LOG )
|
|
|
|
def getNCServerCmd(self, id):
|
|
s = ExperienceNCPV.NC_BIN + " -d " + \
|
|
" -l " + self.ncServerPort + \
|
|
" 1>/dev/null 2>" + ExperienceNCPV.SERVER_NC_LOG + \
|
|
"_" + str(id) + ".log &"
|
|
print(s)
|
|
return s
|
|
|
|
def getNCClientCmd(self, id):
|
|
s = "dd if=/dev/urandom ibs=" + self.ddibs + \
|
|
" obs=" + self.ddobs + \
|
|
" count=" + self.ddcount + \
|
|
" | " + ExperienceNCPV.PV_BIN + \
|
|
" -g " + self.pvg + " -z " + self.pvz + \
|
|
" -q --rate-limit " + self.pvRateLimit + \
|
|
" | " + ExperienceNCPV.NC_BIN + " " + \
|
|
" -p " + self.ncClientPort[id] + " " + \
|
|
self.mpConfig.getServerIP() + " " + \
|
|
self.ncServerPort + " " + \
|
|
"&>" + ExperienceNCPV.CLIENT_NC_LOG + \
|
|
"_" + str(id) + ".log"
|
|
print(s)
|
|
return s
|
|
def getPvPidCmd(self):
|
|
s = "pgrep -n pv"
|
|
return s
|
|
|
|
def clean(self):
|
|
Experience.clean(self)
|
|
#todo use cst
|
|
self.mpTopo.commandTo(self.mpConfig.server, "killall netcat")
|
|
|
|
|
|
def run(self):
|
|
for i in range(0, len(self.ncClientPort)):
|
|
cmd = self.getNCServerCmd(i)
|
|
self.mpTopo.commandTo(self.mpConfig.server, cmd)
|
|
|
|
cmd = self.getNCClientCmd(i)
|
|
self.mpConfig.client.sendCmd(cmd)
|
|
|
|
cmd = self.getPvPidCmd()
|
|
self.pvPid = None
|
|
while self.pvPid == None or self.pvPid == "":
|
|
self.pvPid = self.mpTopo.commandTo(self.mpConfig.server, cmd)[:-1]
|
|
print("guessing pv pid ... :" + str(self.pvPid))
|
|
|
|
cmd = self.getPvChangeCmd()
|
|
print(cmd)
|
|
self.mpTopo.commandTo(self.mpConfig.server, cmd)
|
|
|
|
|
|
self.mpConfig.client.waitOutput()
|
|
|
|
self.mpTopo.commandTo(self.mpConfig.client, "sleep 1")
|
|
|