def __init__(self, netzob):
self.netzob = netzob
self.currentActor = None
self._view = SimulatorView(self)
self.log = logging.getLogger(__name__)
class SimulatorController(object):
def __init__(self, netzob):
self.netzob = netzob
self.currentActor = None
self._view = SimulatorView(self)
self.log = logging.getLogger(__name__)
@property
def view(self):
return self._view
def restart(self):
"""Restart the view"""
logging.debug("Restarting the simulator perspective")
self._view.restart()
def activate(self):
"""Activate the perspective"""
self.restart()
def deactivate(self):
pass
def getCurrentProject(self):
"""Return the current project (can be None)"""
return self.netzob.getCurrentProject()
def getCurrentWorkspace(self):
"""Return the current workspace"""
return self.netzob.getCurrentWorkspace()
def getCurrentActor(self):
"""Return the current actor to display (mostly on the right panel)"""
return self.currentActor
def createNetworkActor_activate_cb(self, event):
"""Callback executed when the user wants to create a network actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
controller = CreateNetworkActorController(self)
controller.run()
def editActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to edit a selected actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
def deleteActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to delete a selected actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
model, iter = self.view.listOfActorsTreeView.get_selection().get_selected()
if iter is not None:
actorID = model[iter][2]
actor = self.getCurrentProject().getSimulator().getActorByID(actorID)
if actor is not None:
questionMsg = _("Click yes to confirm the deletion of the selected actor")
result = NetzobQuestionMessage(questionMsg)
if result == Gtk.ResponseType.YES:
self.getCurrentProject().getSimulator().removeActor(actor)
self._view.refreshListOfActors()
self._view.deleteActorButton.set_sensitive(False)
def stopActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to stop a selected actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
def startActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to start a selected actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
def stopCurrentActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to start the current actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
self.currentActor.stop()
def startCurrentActorButton_clicked_cb(self, event):
"""Callback executed when the user wants to stop the current Actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
# reset the thread (to allows it reuses)
self.currentActor = self.currentActor.clone()
if self.currentActor is not None and not self.currentActor.isActive():
# self.currentActor.setStatusModification_cb(self._view.updateStatusOfCurrentActor)
self.currentActor.getAbstractionLayer().setInputSymbolReception_cb(self._view.registerInputSymbolOfCurrentActor)
self.currentActor.getAbstractionLayer().setOutputSymbolSending_cb(self._view.registerOutputSymbolOfCurrentActor)
self.currentActor.getAbstractionLayer().getMemory().setMemoryAccess_cb(self._view.registerMemoryAccess)
# Restart the view
self._view.updateCurrentActor()
self.currentActor.start()
def listOfActorsSelection_changed_cb(self, selection):
"""Callback executed when the user selects a different actor"""
if self.getCurrentProject() is None:
logging.info("No project loaded.")
return
model, iter = selection.get_selected()
if iter is not None:
actorID = model[iter][2]
self.currentActor = self.getCurrentProject().getSimulator().getActorByID(actorID)
self._view.deleteActorButton.set_sensitive(True)
self._view.updateCurrentActor()