def do_monitor(self, line):
"""Start/stop or collect monitoring network usage of the nodes"""
print("Not implemented")
parser = argparse.ArgumentParser()
# parser.add_argument('command',
# metavar = 'COMMAND',
# choices = ['start', 'stop', 'collect', 'reset'],
# dest = 'command',
# help = 'Monitor command to run')
#
# parser.add_argument('host',
# # action = 'append',
# # metavar = 'host',
# nargs = '*',
# help = 'Host to monitor')
#
subp = parser.add_subparsers(dest = 'subparser_name')
# parser for the add command
# # subp1 = subp.add_parser('start')
#
# # parser for the do command
# # subp2 = subp.add_parser('stop')
# # subp2.add_argument('test', metavar = 'test',
# # help = 'The message you want to send to the probe')
# # subp2.add_argument('options', nargs = argparse.REMAINDER)
# # subp2.set_defaults(func = self.stop)
#
# # parse for the remove command
subp3 = subp.add_parser('collect')
subp3.add_argument('file',
help = "File to collect the data to")
# # subp3.set_defaults(func = self.collect)
# # subp4 = subp.add_parser('reset')
# # subp4.set_defaults(func = self.reset)
#
args = parser.parse_args(shlex.split(line))
for host in self.mn.hosts():
monitor.collect(host, args.file)
def runTopo(topoFile, simParams, hostOptions, checkLevel, controller, switch):
topo = CustomTopo(topoFilePath = topoFile, simParams = simParams, hostOptions = hostOptions)
if checkLevel > 1:
topo.setNetOption('link', TCLink)
# net = CustomMininet(topo = topo, controller = Beacon, autoSetMacs = True, **topo.getNetOptions())
# net = CustomMininet(topo = topo, controller = Beacon, **topo.getNetOptions())
net = CustomMininet(topo = topo, controller = controller, switch = switch, **topo.getNetOptions())
global netprobes
netprobes = collections.OrderedDict()
try:
lg.output('Constructing virtual network..\n')
start(net)
check(net, checkLevel)
lg.output("Starting hosts")
lg.info(": ")
for host in net.hosts:
lg.info("%s " % host.name)
if host.monitor_rules is not None:
monitor.start(host, host.monitor_rules)
if host.command is not None:
lg.info("cmd ")
host.command = host.command.format(commandOpts = host.commandOpts, name = host.name).format(name = host.name)
if host.isXHost:
t = makeTerm(host, cmd = host.command)
if len(t) < 1:
lg.error("Error while starting terminal for host %s\n" % host.name)
continue
if len(t) == 2:
tunnel, term = t
else:
term = t
try:
if term.poll() is not None:
lg.error(
"Terminal with command %s ended early for host %s : %s\n" % (host.command, host.name, repr(term.communicate())))
except:
pass
netprobes[host.name] = term
else:
netprobes[host.name] = runCommand(host)
# print(netprobes[host.name].communicate())
else:
if host.isXHost:
makeTerm(host)
lg.info("term ")
lg.info("done ")
lg.output("\n")
EventsManager.startClock(net)
interract(net)
mon = False
counter = monitor.Counter()
for host in net.hosts:
if host.monitor_rules is not None:
monitor.collect(host, monitor_file, counter)
monitor.stop(host, host.monitor_rules)
mon = True
for name, probe in netprobes.iteritems():
lg.info("Send sigint to probe %s\n" % name)
import signal
try:
probe.send_signal(signal.SIGINT)
time.sleep(0.05)
except OSError as e:
lg.error("Failed to send SIGINT to %s : %s\n" % ( name, e))
if mon:
monitor.writeSummary(monitor_file, counter)
finally:
stop(net)
# cleanup !
lg.info("Stopping remaining processes...\n")
kill = 0
for name, probe in netprobes.iteritems():
if probe.poll() is None:
kill += 1
if kill > 0:
lg.info("Found %s process(es) to kill\n" % kill)
time.sleep(3)
for name, probe in netprobes.iteritems():
if probe.poll() is None:
try:
lg.info("Send terminate signal to %s\n" % name)
probe.terminate()
time.sleep(0.001)
except OSError as e:
lg.error("Failed to terminate %s : %s\n" % (name, e))
time.sleep(3)
for name, probe in netprobes.iteritems():
if probe.poll() is None:
try:
lg.info("Send kill signal to %s\n" % name)
probe.kill()
except OSError as e:
lg.error("Failed to kill %s : %s\n" % (name, e))
lg.output("\nAll done\n")
def runTopo(topoFile, simParams, hostOptions, checkLevel, controller, switch):
topo = CustomTopo(topoFilePath=topoFile,
simParams=simParams,
hostOptions=hostOptions)
if checkLevel > 1:
topo.setNetOption('link', TCLink)
# net = CustomMininet(topo = topo, controller = Beacon, autoSetMacs = True, **topo.getNetOptions())
# net = CustomMininet(topo = topo, controller = Beacon, **topo.getNetOptions())
net = CustomMininet(topo=topo,
controller=controller,
switch=switch,
**topo.getNetOptions())
global netprobes
netprobes = collections.OrderedDict()
try:
lg.output('Constructing virtual network..\n')
start(net)
check(net, checkLevel)
lg.output("Starting hosts")
lg.info(": ")
for host in net.hosts:
lg.info("%s " % host.name)
if host.monitor_rules is not None:
monitor.start(host, host.monitor_rules)
if host.command is not None:
lg.info("cmd ")
host.command = host.command.format(
commandOpts=host.commandOpts,
name=host.name).format(name=host.name)
if host.isXHost:
t = makeTerm(host, cmd=host.command)
if len(t) < 1:
lg.error(
"Error while starting terminal for host %s\n" %
host.name)
continue
if len(t) == 2:
tunnel, term = t
else:
term = t
try:
if term.poll() is not None:
lg.error(
"Terminal with command %s ended early for host %s : %s\n"
% (host.command, host.name,
repr(term.communicate())))
except:
pass
netprobes[host.name] = term
else:
netprobes[host.name] = runCommand(host)
# print(netprobes[host.name].communicate())
else:
if host.isXHost:
makeTerm(host)
lg.info("term ")
lg.info("done ")
lg.output("\n")
EventsManager.startClock(net)
interract(net)
mon = False
counter = monitor.Counter()
for host in net.hosts:
if host.monitor_rules is not None:
monitor.collect(host, monitor_file, counter)
monitor.stop(host, host.monitor_rules)
mon = True
for name, probe in netprobes.iteritems():
lg.info("Send sigint to probe %s\n" % name)
import signal
try:
probe.send_signal(signal.SIGINT)
time.sleep(0.05)
except OSError as e:
lg.error("Failed to send SIGINT to %s : %s\n" % (name, e))
if mon:
monitor.writeSummary(monitor_file, counter)
finally:
stop(net)
# cleanup !
lg.info("Stopping remaining processes...\n")
kill = 0
for name, probe in netprobes.iteritems():
if probe.poll() is None:
kill += 1
if kill > 0:
lg.info("Found %s process(es) to kill\n" % kill)
time.sleep(3)
for name, probe in netprobes.iteritems():
if probe.poll() is None:
try:
lg.info("Send terminate signal to %s\n" % name)
probe.terminate()
time.sleep(0.001)
except OSError as e:
lg.error("Failed to terminate %s : %s\n" % (name, e))
time.sleep(3)
for name, probe in netprobes.iteritems():
if probe.poll() is None:
try:
lg.info("Send kill signal to %s\n" % name)
probe.kill()
except OSError as e:
lg.error("Failed to kill %s : %s\n" % (name, e))
lg.output("\nAll done\n")
def SvcDoRun(self):
while self.isAlive:
monitor.collect()
