def __init__(self, dot_file, net):
self.dot_file = dot_file
self.dot_config = DotConfig(self.dot_file)
self.dot_config.export_inet(self.TOPOLOGY_FILE)
self.dot_config.export_net_settings(self.NET_SETTINGS_FILE)
self.net = net
class BackgroundTrafficConfig(object):
"""Add background traffic pattern to the NS3 Simulator
The background traffic is specifed by a DOT file
"""
TOPOLOGY_FILE = settings.ROOT + '/share/bg_topology.inet'
TOPOLOGY_TYPE = 'Inet'
NET_SETTINGS_FILE = settings.ROOT + '/share/bg_net_settings_tmp.py'
def __init__(self, dot_file, net):
self.dot_file = dot_file
self.dot_config = DotConfig(self.dot_file)
self.dot_config.export_inet(self.TOPOLOGY_FILE)
self.dot_config.export_net_settings(self.NET_SETTINGS_FILE)
self.net = net
def _add_onoff_app(self, start_time, end_time, local, remote, on_time,
off_time, data_rate, sport, dport):
"""add one ns3 onoff application to the network
"""
# ignore the network prefix length if there is
if '/' in remote: remote = remote.rsplit('/')[0]
if '/' in local: local = local.rsplit('/')[0]
### Install OnOff Application ###
socketType = "ns3::UdpSocketFactory"
# socketType = "ns3::TcpSocketFactory"
helper = ns3.OnOffHelper(socketType,
ns3.InetSocketAddress(remote, dport))
helper.SetAttribute("StartTime",
ns3.TimeValue(ns3.Time(str(start_time) + 's')))
helper.SetAttribute("StopTime",
ns3.TimeValue(ns3.Time(str(end_time) + 's')))
# helper.SetAttribute("Remote", ns3.AddressValue(ns3.Ipv4Address(remote)))
local_inet = ns3.InetSocketAddress(local, sport)
helper.SetAttribute("Local", ns3.AddressValue(local_inet))
helper.SetAttribute("DataRate",
ns3.StringValue(str(data_rate) + 'b/s'))
helper.SetAttribute("OnTime",
ns3.RandomVariableValue(on_time.to_ns3()))
helper.SetAttribute("OffTime",
ns3.RandomVariableValue(off_time.to_ns3()))
print('local, ', local)
local_node = self.net.search_node(local)
helper.Install(local_node)
### Install Sink Application ####
sinkLocalAddress = ns3.Address(
ns3.InetSocketAddress(ns3.Ipv4Address.GetAny(), dport))
sinkHelper = ns3.PacketSinkHelper(socketType, sinkLocalAddress)
helper.SetAttribute("StartTime",
ns3.TimeValue(ns3.Time(str(start_time) + 's')))
helper.SetAttribute("StopTime",
ns3.TimeValue(ns3.Time(str(end_time) + 's')))
remote_node = self.net.search_node(remote)
print('remote, ', remote)
sinkHelper.Install(remote_node)
print("""add an onoff application with
start_time: %f
end_time: %f
local: %s,
remote: %s,
on_time: %s,
off_time: %s""" %
(start_time, end_time, local, remote, on_time, off_time))
@staticmethod
def transform_para(flowsize, flowstart, ipsrc, ipdst, sport, dport,
**argv):
"""transform the parameter for generator desription to the NS3 onoff application parameters
"""
# data_rate = StringValue('1kbps') # constant data rate
data_rate = 1000
on_time = RV(flowsize) / data_rate
off_time = RV(flowstart)
return ipsrc, ipdst, on_time, off_time, data_rate, eval(
sport).next(), eval(dport).next()
def config_onoff_app(self):
""" config on off application according to the modulator profile"""
mod_prof = self.dot_config.load_mod_prof()
for node, mod_profs in mod_prof.iteritems(): # for each node
for start_time, end_time, num, generator in mod_profs: # for each modulator
gen = self.dot_config.get_gen(node, generator)
for i in xrange(num):
self._add_onoff_app(start_time, end_time,
*self.transform_para(**gen))
