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