from core.experience import Experience, ExperienceParameter
from mpPvAt import MpPvAt

"""
Should be the mother of ExperienceNCPV, shame on me, should rewrite
ExperienceNCPV as daughter class of this one.
"""

class  ExperienceNC(Experience): 
	SERVER_NC_LOG = "netcat_server"
	CLIENT_NC_LOG = "netcat_client"
	NC_BIN = "netcat"

	def __init__(self, xpParamFile, mpTopo, mpConfig):
		Experience.__init__(self, xpParamFile, mpTopo, mpConfig)
		self.loadParam()
		Experience.classicRun(self)
	
	def loadParam(self):
		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.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)

	def prepare(self):
		Experience.prepare(self)
		self.mpTopo.commandTo(self.mpConfig.client, "rm " + \
				ExperienceNC.CLIENT_NC_LOG )
		self.mpTopo.commandTo(self.mpConfig.server, "rm " + \
				ExperienceNC.SERVER_NC_LOG )

	def getNCServerCmd(self, id):
		s = "dd if=/dev/urandom ibs=" + self.ddibs + \
				" obs=" + self.ddobs + \
				" count=" + self.ddcount + \
				" | " + \
				ExperienceNC.NC_BIN + \
				" -l " + self.ncServerPort  + \
				" &>" + ExperienceNC.SERVER_NC_LOG + \
				"_" + str(id) + ".log"
		print(s)
		return s

	def getNCClientCmd(self, id):
		s = ExperienceNC.NC_BIN + " " + \
				" -p " + self.ncClientPort[id] + " " + \
				self.mpConfig.getServerIP() + " " + \
				self.ncServerPort + " " + \
				"&>" + ExperienceNC.CLIENT_NC_LOG + \
				"_" + str(id) + ".log"
		print(s)
		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.mpConfig.server.sendCmd(cmd)
			
			cmd = self.getNCClientCmd(i)
			self.mpTopo.commandTo(self.mpConfig.client, cmd)

			self.mpConfig.server.waitOutput()
			
			self.mpTopo.commandTo(self.mpConfig.client, "sleep 1")