root/framework3/trunk/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'))
$:.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_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 packages"
        $stderr.puts "[*] Dependencies include ruby-pango, ruby-glib2, ruby-gdkpixbuf2, and ruby-atk"
        $stderr.puts "[*] Error: #{e.class} #{e}"
        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" ],
        "-r" => [ true,  "Execute the specified resource file"                  ],
        "-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 "-r"
                        opts['Resource'] = 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)

        begin; $stdout.close; rescue; end
        begin; $stderr.close; rescue; end
        
        fake = FakeOut.new
        $stdout = fake
        $stderr = fake
        
        # Only treat super nasty Gtk errors as fatal    
        GLib::Log.set_fatal_mask("GLib", GLib::Log::LEVEL_ERROR)
        GLib::Log.set_fatal_mask("Gtk", GLib::Log::LEVEL_ERROR)
        GLib::Log.set_fatal_mask("Gdk", GLib::Log::LEVEL_ERROR)
        GLib::Log.set_fatal_mask(nil, GLib::Log::LEVEL_ERROR)
        
        # GLib::Log.log("Gtk", GLib::Log::LEVEL_CRITICAL, "TESTING > /dev/null")
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