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

require 'msf/core'
require 'msf/base'
require 'msf/ui'

require 'msf/ui/gtk2/controls'
require 'msf/ui/gtk2/app'
require 'msf/ui/gtk2/about'
require 'msf/ui/gtk2/frame'
require 'msf/ui/gtk2/dialogs'
require 'msf/ui/gtk2/console'
require 'msf/ui/gtk2/view'
require 'msf/ui/gtk2/search'

require 'msf/ui/gtk2/framework_event_manager'

module Msf
module Ui
module Gtk2

###
#
# This class implements a user interface driver on a gtk2 interface.
#
###
class Driver < Msf::Ui::Driver
	
	attr_accessor :session_tree, :module_tree, :job_tree, :log_text, :module_model
	attr_accessor :module_completion
	
	include Msf::Ui::Gtk2::FrameworkEventManager
	
	ConfigCore  = "framework/core"
	ConfigGroup = "framework/ui/gtk2"
	
	#
	# The default resource directory for msfgui
	#
	DefaultResourceDirectory = Msf::Config.data_directory + File::SEPARATOR + "msfgui"
	
	#
	# Initializes a gtk2 driver instance
	#
	def initialize(opts = {})
		# Call the parent
		super()

		# Set the passed options hash for referencing later on.
		self.opts = opts

		# Initialize configuration
		Msf::Config.init
		
		# Initialize logging
		initialize_logging

		# Initialize attributes
		self.framework = Msf::Simple::Framework.create
		
		# Create the gtk2driver global
		$gtk2driver = self
		
		# Init GTK2
		Gtk.init
		
		# Initialize the Gtk2 application object
		@gtk2app = Msf::Ui::Gtk2::MyApp.new()
		
		# Register event handlers
		register_event_handlers
	end

	#
	# Starts the main gtk2 loop
	#
	def run
		ilog("msfgui has been started", LogSource)
		Gtk.main
		true
	end

	#
	# Returns the local directory that will hold the files to be serviced.
	#
	def resource_directory
		opts['ResourceDirectory'] || DefaultResourceDirectory
	end
	
	#
	# Returns a new Gdk::Pixbuf object
	#
	def get_icon(name)
		Gdk::Pixbuf.new(File.join(resource_directory, 'pix', name))
	end
	
	#
	# Adds text to the main logging screen
	#
	def append_log_view(data)
	
		data = Time.now.strftime("%H:%m:%S") + " " + data
		
		return if not self.log_text
	
		view = self.log_text
		buff = view.buffer
		
		if (not buff.get_mark('end_mark'))
			buff.create_mark('end_mark', buff.end_iter, false)
		end
		
		buff.insert(buff.end_iter, Rex::Text.to_utf8(data))
		buff.move_mark('end_mark', buff.end_iter)
		view.scroll_mark_onscreen(buff.get_mark('end_mark'))			
	end
	

	#
	# The framework instance associated with this driver.
	#
	attr_reader   :framework

protected

	attr_writer   :framework # :nodoc:
	attr_accessor :opts      # :nodoc:

	#
	# Initializes logging for the Gtk2 driver.
	#
	def initialize_logging
		level = (opts['LogLevel'] || 0).to_i

		Msf::Logging.enable_log_source(LogSource, level)
	end

end

end
end
end