def graph(self):
"""
Graph the stats of the running router
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
if options["graphing"]:
try:
from UI.GraphWindow import GraphWindow
except:
mainWidgets["log"].append(
"Error: matplotlib required for graphing capabilities")
else:
self.rstatsWindow = GraphWindow(self.getName(),
mainWidgets["canvas"])
self.rstatsWindow.show()
def graph(self):
"""
Graph the stats of the running router
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
if options["graphing"]:
try:
from UI.GraphWindow import GraphWindow
except:
mainWidgets["log"].append(
"Error: matplotlib required for graphing capabilities")
else:
self.rstatsWindow = GraphWindow(self.getName(),
mainWidgets["canvas"])
self.rstatsWindow.show()
class Router(Interfaceable):
device_type = "Router"
def __init__(self):
Interfaceable.__init__(self)
self.menu.addAction("Wireshark", self.wireshark)
self.menu.addAction("Graph", self.graph)
self.tail = None
self.wshark = None
self.rstatsWindow = None
self.lightPoint = QPoint(-19, -6)
def graph(self):
"""
Graph the stats of the running router
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
if options["graphing"]:
try:
from UI.GraphWindow import GraphWindow
except:
mainWidgets["log"].append(
"Error: matplotlib required for graphing capabilities")
else:
self.rstatsWindow = GraphWindow(self.getName(),
mainWidgets["canvas"])
self.rstatsWindow.show()
def wireshark(self):
"""
Open wireshark with the running device.
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
progName = "wireshark"
if environ["os"] == "Windows":
progName += ".exe"
if Core.util.progExists(progName):
outfile = environ["tmp"].replace("\\", "/") + \
str(self.getName()) + ".out"
if not os.access(outfile, os.F_OK):
open(outfile, "w").close()
client = mainWidgets["client"]
if client:
client.send("wshark " + self.getName())
command = ["tail", "-n", "+1", "-f", "%s" % outfile]
command2 = [progName, "-k", "-i", "-"]
try:
self.tail = subprocess.Popen(command, stdout=subprocess.PIPE)
except:
mainWidgets["log"].append(
"Error: tail not found in path!\nPlease add GINI_HOME/bin to PATH."
)
self.wshark = subprocess.Popen(command2, stdin=self.tail.stdout)
else:
mainWidgets["log"].append("Error: wireshark not found in path")
class Router(Interfaceable):
type = "Router"
def __init__(self):
Interfaceable.__init__(self)
self.menu.addAction("Wireshark", self.wireshark)
self.menu.addAction("Graph", self.graph)
self.tail = None
self.wshark = None
self.rstatsWindow = None
self.lightPoint = QPoint(-19,-3)
def graph(self):
"""
Graph the stats of the running router
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
if options["graphing"]:
try:
from UI.GraphWindow import GraphWindow
except:
mainWidgets["log"].append(
"Error: matplotlib required for graphing capabilities")
else:
self.rstatsWindow = GraphWindow(self.getName(),
mainWidgets["canvas"])
self.rstatsWindow.show()
def wireshark(self):
"""
Open wireshark with the running device.
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
progName = "wireshark"
if environ["os"] == "Windows":
progName += ".exe"
if Core.util.progExists(progName):
outfile = environ["tmp"].replace("\\", "/") + \
str(self.getName()) + ".out"
if not os.access(outfile, os.F_OK):
open(outfile, "w").close()
client = mainWidgets["client"]
if client:
client.send("wshark " + self.getName())
command = ["tail", "-n", "+1", "-f", "%s" % outfile]
command2 = [progName, "-k", "-i", "-"]
try:
self.tail = subprocess.Popen(command, stdout=subprocess.PIPE)
except:
mainWidgets["log"].append("Error: tail not found in path!\nPlease add GINI_HOME/bin to PATH.")
self.wshark = subprocess.Popen(command2, stdin=self.tail.stdout)
else:
mainWidgets["log"].append(
"Error: wireshark not found in path")
class Router(Interfaceable):
device_type = "Router"
def __init__(self):
super(Router, self).__init__()
self.menu.addAction("Graph", self.graph)
self.wireshark_sessions = []
self.router_stats_window = None
self.wireshark_menu = self.menu.addMenu("Wireshark")
self.wireshark_menu.aboutToShow.connect(self.load_wireshark_menu)
self.lightPoint = QPoint(-19, -6)
def stop(self):
"""
Override parent class' stop method
"""
super(Router, self).stop()
try:
if self.router_stats_window:
self.router_stats_window.close()
for session in self.wireshark_sessions:
session.terminate()
except:
print "Error occurred when stopping %s" % self.getName()
def graph(self):
"""
Graph the stats of the running router
"""
if not mainWidgets["main"].isRunning():
mainWidgets["log"].append("You must start the topology first!")
return
if options["graphing"]:
try:
from UI.GraphWindow import GraphWindow
except:
mainWidgets["log"].append(
"Error: matplotlib required for graphing capabilities")
else:
self.router_stats_window = GraphWindow(self.getName(),
mainWidgets["canvas"])
self.router_stats_window.show()
def load_wireshark_menu(self):
"""Get connected interfaces and display as options on Wireshark menu"""
# TODO: see if I can send this information through TCP sockets instead of writing to files.
self.wireshark_menu.clear()
router_tmp_file = environ["tmp"] + self.getName() + ".json"
if not os.access(router_tmp_file, os.R_OK):
return
with open(router_tmp_file, "r") as f:
interfaces_map = json.load(f, encoding="utf-8")
for key, value in interfaces_map.items():
wireshark_action = partial(self.wireshark, value)
self.wireshark_menu.addAction(key, wireshark_action)
def wireshark(self, interface):
"""
Open wireshark with the running device.
"""
program_name = "wireshark"
command_to_execute = [program_name, "-k", "-i", interface]
wireshark_process = subprocess.Popen(command_to_execute)
if Core.util.progExists(program_name):
self.wireshark_sessions.append(wireshark_process)
else:
mainWidgets["log"].append("Error: wireshark not found in path!")
