metasploit-framework/lib/msf/ui/gtk2/framework_event_manager.rb

64 lines
1.3 KiB
Ruby

module Msf
module Ui
module Gtk2
###
#
# Handles events of various types that are sent from the framework.
#
###
module FrameworkEventManager
include Msf::SessionEvent
#
# Subscribes to the framework as a subscriber of various events.
#
def register_event_handlers
framework.events.add_session_subscriber(self)
end
#
# Unsubscribes from the framework.
#
def deregister_event_handlers
framework.events.remove_session_subscriber(self)
end
#
# Called when a session is registered with the framework.
#
def on_session_open(session)
$gtk2driver.append_log_view("[*] Session #{session.sid} created for #{session.tunnel_peer}\n")
$gtk2driver.session_tree.add_session(session)
# remove job if not a passive exploit
rhost = session.tunnel_peer.split(':')[0]
$gtk2driver.job_tree.remove_job(rhost, session.via_exploit)
if (Msf::Logging.session_logging_enabled? == true)
Msf::Logging.start_session_log(session)
end
end
#
# Called when a session is closed and removed from the framework.
#
def on_session_close(session)
$gtk2driver.append_log_view("[*] Session #{session.sid} closed for #{session.tunnel_peer}\n")
$gtk2driver.session_tree.remove_session(session)
# If logging had been enabled for this session, stop it now.
Msf::Logging::stop_session_log(session)
end
end
end
end
end