metasploit-framework/msfgui

#!/usr/bin/env ruby
#
# This is a basic user interface using the Gtk2 GUI library
#

msfbase = File.symlink?(__FILE__) ? File.readlink(__FILE__) : __FILE__
$:.unshift(File.join(File.dirname(msfbase), 'lib'))


require 'rex'
require 'msf/base'
require 'msf/ui'

# Check for ruby packages
begin
	require 'gtk2'
	require 'libglade2'
rescue ::Exception => e
	$stderr.puts "[*] The msfgui interface requires the ruby-gtk2 and ruby-libglade2 package"
	exit(0)
end

# Check for Gtk+ version
# Returns: nil if the GTK+ library is compatible with the given version,
# or a string describing the version mismatch.
if gtkversion = Gtk.check_version(2,8,0)
	$stderr.puts "[*] The msfgui interface requires Gtk+ 2.8 or later"
	$stderr.puts "[*] Your Gtk+ version : #{gtkversion}"
	exit
end	

require 'msf/ui/gtk2'

# Declare the argument parser for msfgui
arguments = Rex::Parser::Arguments.new(
	"-v" => [ true,  "A number between 0 and 3 that controls log verbosity" ],
	"-d" => [ false, "Keep running in the foreground"                       ],
	"-D" => [ false, "Keep stdio and stderr open for debugging"             ],
	"-h" => [ false, "Help banner"                                          ])

opts = {}
foreground = true
debug = false

# Parse command line arguments.
arguments.parse(ARGV) { |opt, idx, val|
	case opt
		when "-v"
			opts['LogLevel'] = val
		when "-d"
			foreground = false
		when "-D"
			debug = true
		when "-h"
			print(
				"\nUsage: msfgui <options>\n" +
				arguments.usage)
			exit
	end
}

# Fork into the background if requested
begin
	if (not foreground)
		exit(0) if Process.fork()
	end
rescue ::NotImplementedError
	$stderr.puts "[*] Background mode is not available on this platform"
end

class FakeOut
	def write(buff)
		buff.length
	end
	
	def method_missing(meth, *args)
	end
end

if(not debug)
	fake = FakeOut.new
	$stdout = fake
	$stderr = fake
	
	if(Rex::Compat.is_windows)
		nul = Rex::Compat.win32_dev_null()
		Rex::Compat.win32_set_std_handle('stdout', nul)
		Rex::Compat.win32_set_std_handle('stderr', nul)
		Rex::Compat.win32_set_std_handle('stdin', nul)
		
		# The following will crash rubyw.exe without the above code
		GLib::Log.log("Gtk", GLib::Log::LEVEL_CRITICAL, "gtk + rubyw problems > /dev/null")
	end
end

$stderr.puts "[*] Debugging mode is enabled"

# Language is English
ENV['LANG'] = 'C'

# Create the driver instance and run it.
Msf::Ui::Gtk2::Driver.new(opts).run