def __init__(self,
srcnode,
dstnode=None,
action=None,
ipdstfilt=None,
ipsrcfilt=None,
ipprotofilt=None):
TrafficGenerator.__init__(self, srcnode)
self.dstnode = dstnode
self.logger.debug(
'subtractive: %s %s %s %s %s %s' %
(srcnode, dstnode, action, ipdstfilt, ipsrcfilt, ipprotofilt))
self.ipdstfilt = self.ipsrcfilt = ''
self.ipprotofilt = 0
assert (action)
self.action = eval(action)
if ipdstfilt:
self.ipdstfilt = ipaddr.IPNetwork(ipdstfilt)
if ipsrcfilt:
self.ipsrcfilt = ipaddr.IPNetwork(ipsrcfilt)
if ipprotofilt:
self.ipprotofilt = int(ipprotofilt)
def __init__(self, srcnode, dstnode=None, action=None, ipdstfilt=None,
ipsrcfilt=None, ipprotofilt=None):
TrafficGenerator.__init__(self, srcnode)
self.dstnode = dstnode
self.logger.debug('subtractive: %s %s %s %s %s %s' % (srcnode,dstnode,action,ipdstfilt, ipsrcfilt, ipprotofilt))
self.ipdstfilt = self.ipsrcfilt = ''
self.ipprotofilt = 0
assert(action)
self.action = eval(action)
if ipdstfilt:
self.ipdstfilt = ipaddr.IPNetwork(ipdstfilt)
if ipsrcfilt:
self.ipsrcfilt = ipaddr.IPNetwork(ipsrcfilt)
if ipprotofilt:
self.ipprotofilt = int(ipprotofilt)
def __init__(self, srcnode, ipsrc='0.0.0.0', ipdst='0.0.0.0', sport=0, dport=0, flowsize=1500, pktsize=1500, flowstart=0, ipproto=socket.IPPROTO_TCP, lossrate=0.001, mss=1460, iptos=0x0, xopen=True, tcpmodel='csa00'):
TrafficGenerator.__init__(self, srcnode)
self.logger = get_logger('fs.harpoon')
self.srcnet = ipaddr.IPNetwork(ipsrc)
self.dstnet = ipaddr.IPNetwork(ipdst)
if haveIPAddrGen:
self.ipsrcgen = ipaddrgen.initialize_trie(int(self.srcnet), self.srcnet.prefixlen, 0.61)
self.ipdstgen = ipaddrgen.initialize_trie(int(self.dstnet), self.dstnet.prefixlen, 0.61)
if isinstance(ipproto, (str,unicode)):
self.ipproto = eval(ipproto)
else:
self.ipproto = randomchoice(ipproto)
if isinstance(sport, (str,unicode)):
self.srcports = eval(sport)
else:
self.srcports = randomchoice(sport)
if isinstance(dport, (str,unicode)):
self.dstports = eval(dport)
else:
self.dstports = randomchoice(dport)
if isinstance(flowsize, (str,unicode)):
self.flowsizerv = eval(flowsize)
else:
self.flowsizerv = randomchoice(flowsize)
if isinstance(pktsize, (str,unicode)):
self.pktsizerv = eval(pktsize)
else:
self.pktsizerv = randomchoice(pktsize)
if isinstance(flowstart, (str,unicode)):
self.flowstartrv = eval(flowstart)
else:
self.flowstartrv = randomchoice(flowstart)
if isinstance(lossrate, (str,unicode)):
self.lossraterv = eval(lossrate)
else:
self.lossraterv = randomchoice(lossrate)
if isinstance(mss, (str,unicode)):
self.mssrv = eval(mss)
else:
self.mssrv = randomchoice(mss)
if isinstance(iptos, (str,unicode)):
self.iptosrv = eval(iptos)
else:
self.iptosrv = randomchoice(iptos)
self.xopen = xopen
self.activeflows = {}
try:
self.tcpmodel = import_module("tcpmodels.{}".format(tcpmodel))
except ImportError,e:
raise InvalidFlowConfiguration('Unrecognized tcp model for harpoon: {} (Error on import: {})'.format(tcpmodel, str(e)))
def __init__(self,
srcnode,
ipsrc=None,
ipdst=None,
ipproto=None,
dport=None,
sport=None,
continuous=True,
flowlets=None,
tcpflags=None,
iptos=None,
fps=None,
pps=None,
bps=None,
pkts=None,
bytes=None,
pktsize=None,
icmptype=None,
icmpcode=None,
interval=None,
autoack=False):
TrafficGenerator.__init__(self, srcnode)
# assume that all keyword params arrive as strings
# print ipsrc,ipdst
self.ipsrc = IPNetwork(ipsrc)
self.ipdst = IPNetwork(ipdst)
if haveIPAddrGen:
self.ipsrcgen = ipaddrgen.initialize_trie(int(self.ipsrc),
self.ipsrc.prefixlen,
0.61)
self.ipdstgen = ipaddrgen.initialize_trie(int(self.ipdst),
self.ipdst.prefixlen,
0.61)
self.sport = self.dport = None
self.icmptype = self.icmpcode = None
self.autoack = False
if autoack and isinstance(autoack, (str, unicode)):
self.autoack = eval(autoack)
else:
self.autoack = autoack
try:
self.ipproto = int(ipproto)
except:
if ipproto == 'tcp':
self.ipproto = IPPROTO_TCP
elif ipproto == 'udp':
self.ipproto = IPPROTO_UDP
elif ipproto == 'icmp':
self.ipproto = IPPROTO_ICMP
else:
raise InvalidFlowConfiguration('Unrecognized protocol:' +
str(ipproto))
if not iptos:
self.iptos = randomchoice(0x0)
else:
if isinstance(iptos, int):
self.iptos = randomchoice(self.iptos)
elif isinstance(iptos, (str, unicode)):
self.iptos = eval(iptos)
if self.ipproto == IPPROTO_ICMP:
xicmptype = xicmpcode = 0
if icmptype:
xicmptype = eval(icmptype)
if icmpcode:
xicmpcode = eval(icmpcode)
if isinstance(xicmptype, int):
xicmptype = randomchoice(xicmptype)
if isinstance(xicmpcode, int):
xicmpcode = randomchoice(xicmpcode)
self.icmptype = xicmptype
self.icmpcode = xicmpcode
elif self.ipproto == IPPROTO_UDP or self.ipproto == IPPROTO_TCP:
self.dport = eval(dport)
if isinstance(self.dport, int):
self.dport = randomchoice(self.dport)
self.sport = eval(sport)
if isinstance(self.sport, int):
self.sport = randomchoice(self.sport)
# print 'sport,dport',self.sport, self.dport
if self.ipproto == IPPROTO_TCP:
self.tcpflags = randomchoice('')
if tcpflags:
if re.search('\(\S+\)', tcpflags):
self.tcpflags = eval(tcpflags)
else:
self.tcpflags = randomchoice(tcpflags)
else:
self.dport = self.sport = 0
self.continuous = None
self.nflowlets = None
if continuous:
if isinstance(continuous, (str, unicode)):
self.continuous = eval(continuous)
else:
self.continuous = continuous
if flowlets:
self.nflowlets = eval(flowlets)
if isinstance(self.nflowlets, (int, float)):
self.nflowlets = randomchoice(self.nflowlets)
if not self.nflowlets:
self.nflowlets = randomchoice(1)
if not fps and not interval:
raise InvalidFlowConfiguration(
'Need one of fps or interval in rawflow configuration.')
self.fps = self.interval = None
if fps:
fps = eval(fps)
if isinstance(fps, int):
fps = randomchoice(fps)
self.fps = fps
elif interval:
self.interval = eval(interval)
if isinstance(self.interval, (int, float)):
self.interval = randomchoice(self.interval)
assert (bytes)
self.bytes = eval(bytes)
if isinstance(self.bytes, int):
self.bytes = randomchoice(self.bytes)
self.pkts = self.pktsize = None
if pkts:
self.pkts = eval(pkts)
if isinstance(self.pkts, int):
self.pkts = randomchoice(self.pkts)
if pktsize:
self.pktsize = eval(pktsize)
if isinstance(self.pktsize, int):
self.pktsize = randomchoice(self.pktsize)
assert (self.fps or self.interval)
assert (self.pkts or self.pktsize)
def __init__(self, srcnode, ipsrc=None, ipdst=None, ipproto=None,
dport=None, sport=None, continuous=True, flowlets=None, tcpflags=None, iptos=None,
fps=None, pps=None, bps=None, pkts=None, bytes=None, pktsize=None,
icmptype=None, icmpcode=None, interval=None, autoack=False):
TrafficGenerator.__init__(self, srcnode)
# assume that all keyword params arrive as strings
# print ipsrc,ipdst
self.ipsrc = IPNetwork(ipsrc)
self.ipdst = IPNetwork(ipdst)
if haveIPAddrGen:
self.ipsrcgen = ipaddrgen.initialize_trie(int(self.ipsrc), self.ipsrc.prefixlen, 0.61)
self.ipdstgen = ipaddrgen.initialize_trie(int(self.ipdst), self.ipdst.prefixlen, 0.61)
self.sport = self.dport = None
self.icmptype = self.icmpcode = None
self.autoack = False
if autoack and isinstance(autoack, (str,unicode)):
self.autoack = eval(autoack)
else:
self.autoack = autoack
try:
self.ipproto = int(ipproto)
except:
if ipproto == 'tcp':
self.ipproto = IPPROTO_TCP
elif ipproto == 'udp':
self.ipproto = IPPROTO_UDP
elif ipproto == 'icmp':
self.ipproto = IPPROTO_ICMP
else:
raise InvalidFlowConfiguration('Unrecognized protocol:'+str(ipproto))
if not iptos:
self.iptos = randomchoice(0x0)
else:
if isinstance(iptos, int):
self.iptos = randomchoice(self.iptos)
elif isinstance(iptos, (str,unicode)):
self.iptos = eval(iptos)
if self.ipproto == IPPROTO_ICMP:
xicmptype = xicmpcode = 0
if icmptype:
xicmptype = eval(icmptype)
if icmpcode:
xicmpcode = eval(icmpcode)
if isinstance(xicmptype, int):
xicmptype = randomchoice(xicmptype)
if isinstance(xicmpcode, int):
xicmpcode = randomchoice(xicmpcode)
self.icmptype = xicmptype
self.icmpcode = xicmpcode
elif self.ipproto == IPPROTO_UDP or self.ipproto == IPPROTO_TCP:
self.dport = eval(dport)
if isinstance(self.dport, int):
self.dport = randomchoice(self.dport)
self.sport = eval(sport)
if isinstance(self.sport, int):
self.sport = randomchoice(self.sport)
# print 'sport,dport',self.sport, self.dport
if self.ipproto == IPPROTO_TCP:
self.tcpflags = randomchoice('')
if tcpflags:
if re.search('\(\S+\)', tcpflags):
self.tcpflags = eval(tcpflags)
else:
self.tcpflags = randomchoice(tcpflags)
else:
self.dport = self.sport = 0
self.continuous = None
self.nflowlets = None
if continuous:
if isinstance(continuous, (str,unicode)):
self.continuous = eval(continuous)
else:
self.continuous = continuous
if flowlets:
self.nflowlets = eval(flowlets)
if isinstance(self.nflowlets, (int, float)):
self.nflowlets = randomchoice(self.nflowlets)
if not self.nflowlets:
self.nflowlets = randomchoice(1)
if not fps and not interval:
raise InvalidFlowConfiguration('Need one of fps or interval in rawflow configuration.')
self.fps = self.interval = None
if fps:
fps = eval(fps)
if isinstance(fps, int):
fps = randomchoice(fps)
self.fps = fps
elif interval:
self.interval = eval(interval)
if isinstance(self.interval, (int, float)):
self.interval = randomchoice(self.interval)
assert(bytes)
self.bytes = eval(bytes)
if isinstance(self.bytes, int):
self.bytes = randomchoice(self.bytes)
self.pkts = self.pktsize = None
if pkts:
self.pkts = eval(pkts)
if isinstance(self.pkts, int):
self.pkts = randomchoice(self.pkts)
if pktsize:
self.pktsize = eval(pktsize)
if isinstance(self.pktsize, int):
self.pktsize = randomchoice(self.pktsize)
assert(self.fps or self.interval)
assert(self.pkts or self.pktsize)
def __init__(self, srcnode, ipsrc='0.0.0.0', ipdst='0.0.0.0', sport=0, dport=0, flowstarttime=0, flowsize=1500, pktsize=1500, flowstart=0, ipproto=socket.IPPROTO_TCP, lossrate=0.0045, mss=1460, iptos=0x0, xopen=True, tcpmodel='csa00',matrix='../conf/input.matrix'):
TrafficGenerator.__init__(self, srcnode)
self.logger = get_logger('fs.matrix')
trafficmatrix = open(matrix, 'r')
flowList = []
flowStartList = []
srcipList = []
dstipList = []
flowSizeList = []
for line in trafficmatrix:
if line.strip()!= "":
flowList.append(line.strip().split(" "))
for item in flowList:
try:
if nodename_to_ipaddr(srcnode)==item[1]:
flowStartList.append(item[0])
srcipList.append(item[1])
dstipList.append(item[2])
#srcipList.append(nodename_to_ipaddr(item[1]))
#dstipList.append(nodename_to_ipaddr(item[2]))
flowSizeList.append(item[3])
except:
print "Invalid Flow"
print item
pass
self.srcipList=srcipList
self.srcnet = eval('sequentialchoice(srcipList)')
self.dstnet = eval('sequentialchoice(dstipList)')
self.flowstarttimerv = eval('sequentialchoice(flowStartList)')
self.flowsizerv = eval('sequentialchoice(flowSizeList)')
if haveIPAddrGen:
self.ipsrcgen = ipaddrgen.initialize_trie(int(self.srcnet), self.srcnet.prefixlen, 0.61)
self.ipdstgen = ipaddrgen.initialize_trie(int(self.dstnet), self.dstnet.prefixlen, 0.61)
if isinstance(ipproto, (str,unicode)):
#print eval(ipproto)
self.ipproto = eval(ipproto)
else:
self.ipproto = randomchoice(ipproto)
if isinstance(sport, (str,unicode)):
self.srcports = eval(sport)
else:
self.srcports = randomchoice(sport)
if isinstance(dport, (str,unicode)):
self.dstports = eval(dport)
else:
self.dstports = randomchoice(dport)
if isinstance(pktsize, (str,unicode)):
self.pktsizerv = eval(pktsize)
else:
self.pktsizerv = randomchoice(pktsize)
if isinstance(flowstart, (str,unicode)):
self.flowstartrv = eval(flowstart)
else:
self.flowstartrv = randomchoice(flowstart)
if isinstance(lossrate, (str,unicode)):
self.lossraterv = eval(lossrate)
else:
self.lossraterv = randomchoice(lossrate)
if isinstance(mss, (str,unicode)):
self.mssrv = eval(mss)
else:
self.mssrv = randomchoice(mss)
if isinstance(iptos, (str,unicode)):
self.iptosrv = eval(iptos)
else:
self.iptosrv = randomchoice(iptos)
self.xopen = xopen
self.activeflows = {}
try:
self.tcpmodel = import_module("tcpmodels.{}".format(tcpmodel))
except ImportError,e:
raise InvalidFlowConfiguration('Unrecognized tcp model for matrix: {} (Error on import: {})'.format(tcpmodel, str(e)))
def __init__(self, srcnode, ipsrc='0.0.0.0', ipdst='0.0.0.0', sport=0, dport=0, flowsize=1500, pktsize=1500, flowstart=0, ipproto=socket.IPPROTO_TCP, lossrate=0.001, mss=1460, iptos=0x0, xopen=True, tcpmodel='csa00'):
TrafficGenerator.__init__(self, srcnode)
self.logger = get_logger('fs.harpoon')
self.srcnet = ipaddr.IPNetwork(ipsrc)
self.dstnet = ipaddr.IPNetwork(ipdst)
if haveIPAddrGen:
self.ipsrcgen = ipaddrgen.initialize_trie(int(self.srcnet), self.srcnet.prefixlen, 0.61)
self.ipdstgen = ipaddrgen.initialize_trie(int(self.dstnet), self.dstnet.prefixlen, 0.61)
if isinstance(ipproto, (str,unicode)):
self.ipproto = eval(ipproto)
else:
self.ipproto = randomchoice(ipproto)
if isinstance(sport, (str,unicode)):
self.srcports = eval(sport)
else:
self.srcports = randomchoice(sport)
if isinstance(dport, (str,unicode)):
self.dstports = eval(dport)
else:
self.dstports = randomchoice(dport)
if isinstance(flowsize, (str,unicode)):
self.flowsizerv = eval(flowsize)
else:
self.flowsizerv = randomchoice(flowsize)
if isinstance(pktsize, (str,unicode)):
self.pktsizerv = eval(pktsize)
else:
self.pktsizerv = randomchoice(pktsize)
if isinstance(flowstart, (str,unicode)):
self.flowstartrv = eval(flowstart)
else:
self.flowstartrv = randomchoice(flowstart)
if isinstance(lossrate, (str,unicode)):
self.lossraterv = eval(lossrate)
else:
self.lossraterv = randomchoice(lossrate)
if isinstance(mss, (str,unicode)):
self.mssrv = eval(mss)
else:
self.mssrv = randomchoice(mss)
if isinstance(iptos, (str,unicode)):
self.iptosrv = eval(iptos)
else:
self.iptosrv = randomchoice(iptos)
self.xopen = xopen
self.activeflows = {}
try:
self.tcpmodel = import_module("tcpmodels.{}".format(tcpmodel))
except ImportError,e:
raise InvalidFlowConfiguration('Unrecognized tcp model for harpoon: {} (Error on import: {})'.format(tcpmodel, str(e)))
