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

module Msf
module Ui
module Gtk2

##
# Skeleton class for all parameters stuff
# title = Options title (menu)
# items = Array
##
class SkeletonOption < Gtk::Dialog
	
	attr_accessor :frame
	
	def initialize(title, items)
		super("", $gtk2driver.main, Gtk::Dialog::DESTROY_WITH_PARENT,
			[ Gtk::Stock::OK, Gtk::Dialog::RESPONSE_NONE ],
			[ Gtk::Stock::CLOSE, Gtk::Dialog::RESPONSE_NONE ])
		
		self.border_width = 6
		self.resizable = true
		self.has_separator = true
		self.vbox.spacing = 12
		self.vbox.set_homogeneous(false)
		self.title = title
		self.set_default_size(500, 400)
		
		model = create_model(items)
		treeview = Gtk::TreeView.new(model)
		treeview.set_size_request(5, 200)
		
		@hbox = Gtk::HBox.new(false, 10)
		
		# ScrolledWindow
		sw = Gtk::ScrolledWindow.new
		sw.set_policy(Gtk::POLICY_NEVER, Gtk::POLICY_AUTOMATIC)
		@hbox.pack_start(sw)
		sw.add(treeview)
		
		renderer = Gtk::CellRendererText.new
		column = Gtk::TreeViewColumn.new('Select an item', renderer, 'text' => 0)
		#column.fixed_width = 20
		column.pack_start(renderer, false)
		treeview.append_column(column)
		
		self.vbox.pack_start(@hbox, false, false, 0)
		@frame = Gtk::Frame.new
		@frame.set_size_request(300, 400)
		@hbox.pack_end(@frame, true, true, 0)
		
		# Signal
		selection = treeview.selection
		selection.mode = Gtk::SELECTION_SINGLE
		selection.signal_connect('changed') do |s|
			selection_changed(s)
		end
	end
	
	#
	# Create and return the model
	#
	def create_model(items)
		store = Gtk::ListStore.new(String)
		
		items.each do |item|
			iter = store.append
			iter[0] = item
		end
		
		return store
	end
	
	#
	# Destroy all children widgets in the frame
	#
	def clean_frame
		@frame.each do |w|
			w.destroy
		end
	end
	
	#
	# Dummy 
	#
	def selection_changed(selection)
		#
		# must be present in the real class
		#
	end
	
	#
	# Return a dialog file chooser
	#
	def file_chooser(title=nil)
		dialog = Gtk::FileChooserDialog.new(title, 
							nil,
							Gtk::FileChooser::ACTION_OPEN,
							nil,
							[Gtk::Stock::OPEN, Gtk::Dialog::RESPONSE_ACCEPT],
							[Gtk::Stock::CANCEL, Gtk::Dialog::RESPONSE_CANCEL]
							)
	end
end

class MsfParameters
	##
	# Display the preference parameters
	##
	class Preferences < Msf::Ui::Gtk2::SkeletonOption
		def initialize
			menu = ["Exploits", "Payloads"]
			super("Preferences", menu)
			
			w_exploits
			
			show_all and run
			destroy
		end
		
		#
		# Describe the exploits widget
		#
		def w_exploits
			frame.set_label("Exploits")
			frame.add(Gtk::Label.new("Add specifics widgets here"))
			frame.show_all
		end
		
		#
		# Describe the payloads widget
		#
		def w_payloads
			frame.set_label("Payloads")
			frame.shadow_type = Gtk::SHADOW_IN
			#frame.add(Gtk::Label.new("Add specifics widgets here"))
			
			#
			# add specific widget here
			#
			vbox_frame = Gtk::VBox.new(false, 0)
			
			# VNC			
			frame.add(vbox_frame)
			
			table_vnc = Gtk::Table.new(2, 2, false)
			table_vnc.set_row_spacings(4)
			table_vnc.set_column_spacings(4)
			vbox_frame.pack_start(table_vnc, false, false, 5)
			
			label_vnc = Gtk::Label.new("Viewer VNC path: ")
			table_vnc.attach_defaults(label_vnc, 0, 1, 0, 1)
			
			hbox_vnc = Gtk::HBox.new(false, 5)			
			entry_vnc = Gtk::Entry.new
			table_vnc.attach_defaults(entry_vnc, 0, 2, 1, 2)
			
			button_vnc = Gtk::HButtonBox.new
			button_vnc.layout_style = Gtk::ButtonBox::END
			button = Gtk::Button.new(Gtk::Stock::OPEN)
			button_vnc.add(button)
			#button_vnc = Gtk::FileChooserButton.new("Select a file", Gtk::FileChooser::ACTION_OPEN)
			table_vnc.attach_defaults(button_vnc, 1, 2, 0, 1)
			button.signal_connect('clicked') do
				dialog = file_chooser("Select your VNC viewer")
				if dialog.run == Gtk::Dialog::RESPONSE_ACCEPT
					entry_vnc.set_text(dialog.uri)
				end
				dialog.destroy
			end
			
			frame.show_all
		end
		
		#
		# Signals
		#
		def selection_changed(selection)
			iter = selection.selected
			case iter[0]
				when "Exploits"
					clean_frame and w_exploits
				when "Payloads"
					clean_frame and w_payloads
			end
		end
	end
	
	
	##
	# Display the databases parameters
	##
	class Databases < Msf::Ui::Gtk2::SkeletonOption
		def initialize
			menu = ["AutoPOWN", "OPCODES"]
			super("Databases", menu)
			
			show_all and run
			destroy
		end
		
		#
		# Describe the autopown widget
		#
		def w_autopown
			frame.set_label("AutoPWN")
		end
		
		#
		# Describe the opcode widget opcodes
		#
		def w_opcode
			frame.set_label("OPCODES")
		end
		
		#
		# Signals
		#
		def selection_changed(selection)
			iter = selection.selected
			case iter[0]
				when "AutoPOWN"
					clean_frame and w_autopown
				when "OPCODES"
					clean_frame and w_opcode
			end
		end
	end
	
	###
	# Display the options parameters
	###
	class Options < Msf::Ui::Gtk2::SkeletonOption
		def initialize
			menu = ["Debug", "Tips", "Proxy", "Update"]
			super("Options", menu)
			
			show_all and run
			destroy
		end
		
		def w_debug
			frame.set_label("Debug")
		end
		
		#
		# Want to show tips ... or not !
		#
		def w_tips
			frame.set_label("Tips")
		end
		
		#
		# Need a proxy ?
		#
		def w_proxy
			frame.set_label("Proxy")
		end
		
		#
		# Update stuff
		#
		def w_update
			frame.set_label("Update")
		end
		
		#
		# Signals
		#
		def selection_changed(selection)
			iter = selection.selected
			case iter[0]
				when "Debug"
					clean_frame and w_debug
				when "Tips"
					clean_frame and w_tips
				when "Proxy"
					clean_frame and w_proxy
				when "Update"
					clean_frame and w_update
			end
		end
	end
end

end
end
end