Bye bye
git-svn-id: file:///home/svn/framework3/trunk@4872 4d416f70-5f16-0410-b530-b9f4589650daunstable
fab
parent
6eeec269d0
commit
92d61c505c
|
|
@ -1,576 +0,0 @@
|
|||
module Msf
|
||||
module Ui
|
||||
module Gtk2
|
||||
|
||||
|
||||
##
|
||||
# This class perform an assistant to configure module
|
||||
#
|
||||
##
|
||||
class MsfAssistant
|
||||
KEY, DEFAULT, VALUE, DESC = *(0..5).to_a
|
||||
|
||||
##
|
||||
# Describe an assistant for standard module
|
||||
##
|
||||
class Standard
|
||||
|
||||
include Msf::Ui::Gtk2::MyControls
|
||||
|
||||
def initialize(active_module)
|
||||
@session_tree = $gtk2driver.session_tree
|
||||
@job_tree = $gtk2driver.job_tree
|
||||
@active_module = active_module
|
||||
|
||||
# initialize exploit driver's exploit instance
|
||||
@mydriver = Msf::ExploitDriver.new(framework)
|
||||
@mydriver.exploit = framework.exploits.create(@active_module.refname)
|
||||
|
||||
@myassistant = Gtk::Assistant.new
|
||||
@myassistant.set_default_size(400, 300)
|
||||
@myassistant.set_title(@active_module.refname)
|
||||
|
||||
# Hash to store entry assistant
|
||||
@hash = {}
|
||||
|
||||
# signals assistant
|
||||
@myassistant.signal_connect('cancel') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('close') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('apply') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('prepare') { |assistant, page|
|
||||
update_page(page)
|
||||
}
|
||||
|
||||
@model_required = Gtk::ListStore.new(String, String, String, String)
|
||||
@model_advanced = Gtk::ListStore.new(String, String, String, String)
|
||||
|
||||
# add target payload
|
||||
@target_page = target_completion()
|
||||
@myassistant.append_page(@target_page)
|
||||
@myassistant.set_page_title(@target_page, "Target")
|
||||
#@myassistant.set_page_type(page, Gtk::Assistant::PAGE_INTRO)
|
||||
@myassistant.set_page_complete(@target_page, false)
|
||||
|
||||
# add payload frame
|
||||
@payload_page = payload_completion()
|
||||
@myassistant.append_page(@payload_page)
|
||||
@myassistant.set_page_title(@payload_page, "Payload")
|
||||
#@myassistant.set_page_type(page, Gtk::Assistant::PAGE_CONFIRM)
|
||||
@myassistant.set_page_complete(@payload_page, false)
|
||||
|
||||
# add options frame
|
||||
@options_page = options_completion()
|
||||
@myassistant.append_page(@options_page)
|
||||
@myassistant.set_page_title(@options_page, "Options")
|
||||
@myassistant.set_page_type(@options_page, Gtk::Assistant::PAGE_CONFIRM)
|
||||
@myassistant.set_page_complete(@options_page, true)
|
||||
|
||||
# add summary frame
|
||||
@summary_page = summary_completion()
|
||||
@myassistant.append_page(@summary_page)
|
||||
@myassistant.set_page_title(@summary_page, "Summary")
|
||||
@myassistant.set_page_type(@summary_page, Gtk::Assistant::PAGE_SUMMARY)
|
||||
@myassistant.set_page_complete(@summary_page, true)
|
||||
|
||||
@myassistant.show_all
|
||||
end # def initialize
|
||||
|
||||
def target_completion
|
||||
page = Gtk::VBox.new(false, 2)
|
||||
|
||||
# Gtk::Frame for combo target entry
|
||||
frame_target = Gtk::Frame.new("Select :")
|
||||
page.pack_start(frame_target, false, false, 10)
|
||||
|
||||
# Model for Gtk::Combo
|
||||
model_target = Gtk::ListStore.new(String, Object)
|
||||
|
||||
# Add iter to Gtk::Combo
|
||||
@active_module.targets.each_with_index do |target, idx|
|
||||
iter = model_target.append
|
||||
iter[0] = target.name
|
||||
iter[1] = idx
|
||||
end
|
||||
|
||||
# Gtk::ComboBox
|
||||
combo_target = Gtk::ComboBox.new(model_target)
|
||||
|
||||
frame_target.add(combo_target)
|
||||
|
||||
# Gtk::Frame for selected target
|
||||
selected = Gtk::Frame.new("Selected TARGET")
|
||||
page.pack_start(selected, false, false, 10)
|
||||
|
||||
# Pack & renderer combo_all
|
||||
renderer = Gtk::CellRendererText.new
|
||||
combo_target.pack_start(renderer, true)
|
||||
combo_target.set_attributes(renderer, :text => 0)
|
||||
|
||||
# Gtk::Label for selected payload
|
||||
label = Gtk::Label.new
|
||||
selected.add(label)
|
||||
|
||||
# Signal for combo payload
|
||||
combo_target.signal_connect('changed') do ||
|
||||
text = "<span foreground='blue' size='xx-large'>#{combo_target.active_iter[0]}</span>"
|
||||
label.set_markup(text)
|
||||
@hash["TARGET"] = combo_target.active_iter[1]
|
||||
@myassistant.set_page_complete(@target_page, true)
|
||||
end
|
||||
page.show_all
|
||||
end # def target_completion
|
||||
|
||||
def payload_completion
|
||||
page = Gtk::VBox.new(false, 4)
|
||||
|
||||
# Gtk::Frame for combo payload entry
|
||||
frame_all = Gtk::Frame.new("Payloads")
|
||||
|
||||
page.pack_start(frame_all, false, false, 10)
|
||||
|
||||
|
||||
# Model for Gtk::Combox
|
||||
model_all = Gtk::ListStore.new(String, Object)
|
||||
|
||||
# Add iter to Model
|
||||
@active_module.compatible_payloads.each do |refname, payload|
|
||||
iter = model_all.append
|
||||
iter[0] = refname
|
||||
iter[1] = payload
|
||||
end
|
||||
|
||||
# Gtk::ComboBox
|
||||
combo_all = Gtk::ComboBox.new(model_all)
|
||||
|
||||
# Pack & renderer combo_all
|
||||
renderer = Gtk::CellRendererText.new
|
||||
combo_all.pack_start(renderer, true)
|
||||
combo_all.set_attributes(renderer, :text => 0)
|
||||
|
||||
frame_all.add(combo_all)
|
||||
|
||||
# Gtk::Frame for description selected payload
|
||||
description = Gtk::Frame.new("Description")
|
||||
page.pack_start(description, false, false, 10)
|
||||
|
||||
# Stuff for description payload
|
||||
textscroll = Gtk::ScrolledWindow.new
|
||||
textscroll.shadow_type = Gtk::SHADOW_IN
|
||||
textscroll.hscrollbar_policy = Gtk::POLICY_AUTOMATIC
|
||||
textscroll.vscrollbar_policy = Gtk::POLICY_AUTOMATIC
|
||||
buffer = Gtk::TextBuffer.new
|
||||
textview = Gtk::TextView.new(buffer)
|
||||
textview.set_editable(false)
|
||||
textview.set_cursor_visible(false)
|
||||
textscroll.add(textview)
|
||||
description.add(textscroll)
|
||||
|
||||
# Gtk::Frame for selected payload
|
||||
selected = Gtk::Frame.new("Selected PAYLOAD")
|
||||
page.pack_start(selected, false, false, 10)
|
||||
|
||||
# Gtk::Label for selected payload
|
||||
label = Gtk::Label.new
|
||||
selected.add(label)
|
||||
|
||||
# Signal for combo payload
|
||||
combo_all.signal_connect('changed') do
|
||||
text = "<span foreground='blue' size='xx-large'>#{combo_all.active_iter[0]}</span>"
|
||||
buffer.set_text(combo_all.active_iter[1].new.description)
|
||||
label.set_markup(text)
|
||||
@hash["PAYLOAD"] = combo_all.active_iter[0]
|
||||
@myassistant.set_page_complete(@payload_page, true)
|
||||
end
|
||||
|
||||
page.show_all
|
||||
end # def payload_completion
|
||||
|
||||
def options_completion
|
||||
page = Gtk::VBox.new(false, 2)
|
||||
|
||||
# Gtk::Frame for required options
|
||||
frame_required = Gtk::Frame.new("Required")
|
||||
page.pack_start(frame_required, false, false, 10)
|
||||
|
||||
# Gtk::Frame for advanced options
|
||||
frame_advanced = Gtk::Expander.new('Advanced')
|
||||
page.pack_start(frame_advanced, false, false, 10)
|
||||
|
||||
# TreeView
|
||||
treeview_required = Gtk::TreeView.new(@model_required)
|
||||
treeview_advanced = Gtk::TreeView.new(@model_advanced)
|
||||
|
||||
# Column
|
||||
add_columns(treeview_required, @model_required)
|
||||
add_columns(treeview_advanced, @model_advanced)
|
||||
|
||||
# add treeview to frame
|
||||
frame_required.add(treeview_required)
|
||||
frame_advanced.add(treeview_advanced)
|
||||
|
||||
page.show_all
|
||||
end # def options_completion
|
||||
|
||||
def summary_completion
|
||||
return Gtk::Label.new("Summary").show
|
||||
end # def summary_completion
|
||||
|
||||
def add_columns(treeview, model)
|
||||
# column for KEY
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Name',
|
||||
renderer,
|
||||
'text' => KEY)
|
||||
column.set_sort_column_id(KEY)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for DEFAULT
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Default',
|
||||
renderer,
|
||||
'text' => DEFAULT)
|
||||
column.set_sort_column_id(DEFAULT)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for VALUE
|
||||
renderer = Gtk::CellRendererText.new
|
||||
renderer.background = "#f6ffd6"
|
||||
renderer.editable = true
|
||||
renderer.signal_connect("edited") do |rend, path, text|
|
||||
updatevalue(model, VALUE, path, text)
|
||||
end
|
||||
column = Gtk::TreeViewColumn.new('Value',
|
||||
renderer,
|
||||
'text' => VALUE)
|
||||
column.set_sort_column_id(VALUE)
|
||||
column.set_resizable(true)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for DESC
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Description',
|
||||
renderer,
|
||||
'text' => DESC)
|
||||
column.set_sort_column_id(DESC)
|
||||
treeview.append_column(column)
|
||||
end # def add_columns
|
||||
|
||||
def pack(model, key, opt)
|
||||
iter = model.append
|
||||
iter[KEY] = key
|
||||
iter[DEFAULT] = opt.default.to_s
|
||||
iter[DESC] = opt.desc.to_s
|
||||
end # def pack
|
||||
|
||||
def updatevalue(model, column, path, text)
|
||||
iter = model.get_iter(path)
|
||||
iter[column] = text
|
||||
@hash[iter.get_value(KEY)] = text
|
||||
end # def updatevalue
|
||||
|
||||
def update_page(page)
|
||||
if @myassistant.get_page_title(page) == "Options"
|
||||
|
||||
# Clear treeview
|
||||
@model_required.clear
|
||||
@model_advanced.clear
|
||||
|
||||
# Exploits options
|
||||
@mydriver.exploit.options.sorted.each do |key, opt|
|
||||
next if (opt.evasion?)
|
||||
if (opt.required?)
|
||||
pack(@model_required, key, opt)
|
||||
else
|
||||
pack(@model_advanced, key, opt)
|
||||
end
|
||||
end
|
||||
|
||||
# Payload options
|
||||
@mydriver.payload = framework.payloads.create(@hash["PAYLOAD"])
|
||||
@mydriver.payload.options.each do |key, opt|
|
||||
if (opt.required?)
|
||||
pack(@model_required, key, opt)
|
||||
else
|
||||
pack(@model_advanced, key, opt)
|
||||
end
|
||||
end
|
||||
|
||||
elsif @myassistant.get_page_title(page) == "Summary"
|
||||
|
||||
# Import options from the supplied assistant
|
||||
@mydriver.exploit.datastore.import_options_from_hash(@hash)
|
||||
|
||||
# Share the exploit's datastore with the payload
|
||||
@mydriver.payload.share_datastore(@mydriver.exploit.datastore)
|
||||
|
||||
@mydriver.target_idx = (@mydriver.exploit.datastore['TARGET']).to_i
|
||||
|
||||
@pipe = Msf::Ui::Gtk2::GtkConsolePipe.new
|
||||
|
||||
@mydriver.exploit.init_ui(@pipe, @pipe)
|
||||
@mydriver.payload.init_ui(@pipe, @pipe)
|
||||
|
||||
# Session registration is done by event handler
|
||||
# XXX: No output from the exploit when this is set!
|
||||
@mydriver.use_job = true
|
||||
|
||||
@pipe.create_subscriber_proc() do |msg|
|
||||
$stderr.puts "MSG: #{msg}"
|
||||
$gtk2driver.append_log_view(msg)
|
||||
end
|
||||
|
||||
Thread.new do
|
||||
0.upto(20) do |i|
|
||||
$pipe.print_status("I am alive at #{i}")
|
||||
select(nil, nil, nil, 1.0)
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
@pipe.print_status("Launching exploit #{@mydriver.exploit.refname}...")
|
||||
|
||||
begin
|
||||
@mydriver.run
|
||||
@job_tree.add_oneshot(@active_module, @hash["RHOST"])
|
||||
rescue ::Exception => e
|
||||
@pipe.print_error("Exploit failed: #{e}")
|
||||
end
|
||||
end
|
||||
end # def update_page
|
||||
|
||||
end # class Standard
|
||||
|
||||
##
|
||||
# Describe an assistant for payload module
|
||||
##
|
||||
class Payload
|
||||
include Msf::Ui::Gtk2::MyControls
|
||||
|
||||
def initialize(active_module)
|
||||
@active_module = active_module
|
||||
|
||||
# Create the payload instance
|
||||
@payload = framework.payloads.create(@active_module.refname)
|
||||
|
||||
@myassistant = Gtk::Assistant.new
|
||||
@myassistant.set_default_size(400, 300)
|
||||
@myassistant.set_title(@active_module.refname)
|
||||
|
||||
# Store options payloads
|
||||
@fmt = nil
|
||||
@hash = {}
|
||||
|
||||
# signals assistant
|
||||
@myassistant.signal_connect('cancel') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('close') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('apply') {
|
||||
@myassistant.hide
|
||||
}
|
||||
|
||||
@myassistant.signal_connect('prepare') { |assistant, page|
|
||||
update_page(page)
|
||||
}
|
||||
|
||||
@model_required = Gtk::ListStore.new(String, String, String, String)
|
||||
|
||||
# add options frame
|
||||
@options_page = options_completion()
|
||||
@myassistant.append_page(@options_page)
|
||||
@myassistant.set_page_title(@options_page, "Options")
|
||||
#@myassistant.set_page_type(@options_page, Gtk::Assistant::PAGE_CONFIRM)
|
||||
@myassistant.set_page_complete(@options_page, true)
|
||||
|
||||
# add output frame
|
||||
@output_page = output_completion()
|
||||
@myassistant.append_page(@output_page)
|
||||
@myassistant.set_page_title(@output_page, "Output")
|
||||
@myassistant.set_page_type(@output_page, Gtk::Assistant::PAGE_CONFIRM)
|
||||
@myassistant.set_page_complete(@output_page, false)
|
||||
|
||||
# add summary frame
|
||||
@summary_page = summary_completion()
|
||||
@myassistant.append_page(@summary_page)
|
||||
@myassistant.set_page_title(@summary_page, "Summary")
|
||||
@myassistant.set_page_type(@summary_page, Gtk::Assistant::PAGE_SUMMARY)
|
||||
@myassistant.set_page_complete(@summary_page, true)
|
||||
|
||||
@myassistant.show_all
|
||||
end # def initialize
|
||||
|
||||
def add_columns(treeview, model)
|
||||
# column for KEY
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Name',
|
||||
renderer,
|
||||
'text' => KEY)
|
||||
column.set_sort_column_id(KEY)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for DEFAULT
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Default',
|
||||
renderer,
|
||||
'text' => DEFAULT)
|
||||
column.set_sort_column_id(DEFAULT)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for VALUE
|
||||
renderer = Gtk::CellRendererText.new
|
||||
renderer.background = "#f6ffd6"
|
||||
renderer.editable = true
|
||||
renderer.signal_connect("edited") do |rend, path, text|
|
||||
updatevalue(model, VALUE, path, text)
|
||||
end
|
||||
column = Gtk::TreeViewColumn.new('Value',
|
||||
renderer,
|
||||
'text' => VALUE)
|
||||
column.set_sort_column_id(VALUE)
|
||||
column.set_resizable(true)
|
||||
treeview.append_column(column)
|
||||
|
||||
# column for DESC
|
||||
renderer = Gtk::CellRendererText.new
|
||||
column = Gtk::TreeViewColumn.new('Description',
|
||||
renderer,
|
||||
'text' => DESC)
|
||||
column.set_sort_column_id(DESC)
|
||||
treeview.append_column(column)
|
||||
end # def add_columns
|
||||
|
||||
def pack(model, key, opt)
|
||||
iter = model.append
|
||||
iter[KEY] = key
|
||||
iter[DEFAULT] = opt.default.to_s
|
||||
iter[DESC] = opt.desc.to_s
|
||||
end # def pack
|
||||
|
||||
def options_completion
|
||||
page = Gtk::VBox.new(false, 2)
|
||||
|
||||
# Gtk::Frame for required options
|
||||
frame_required = Gtk::Frame.new("Required")
|
||||
page.pack_start(frame_required, false, false, 10)
|
||||
|
||||
# TreeView
|
||||
treeview_required = Gtk::TreeView.new(@model_required)
|
||||
|
||||
# Column
|
||||
add_columns(treeview_required, @model_required)
|
||||
|
||||
# add treeview to frame
|
||||
frame_required.add(treeview_required)
|
||||
|
||||
page.show_all
|
||||
end # def options_completion
|
||||
|
||||
def output_completion
|
||||
|
||||
selection = ['Raw', 'Perl', 'C', 'Javascript']
|
||||
page = Gtk::VBox.new(false, 2)
|
||||
|
||||
# Gtk::Frame for output selection
|
||||
frame_output = Gtk::Frame.new("Output :")
|
||||
page.pack_start(frame_output, false, false, 10)
|
||||
|
||||
# Model for Gtk::Combo
|
||||
model_output = Gtk::ListStore.new(String)
|
||||
|
||||
# Add iter to Gtk::Combo
|
||||
selection.each do |output|
|
||||
iter = model_output.append
|
||||
iter[0] = output
|
||||
end
|
||||
|
||||
# Gtk::ComboBox
|
||||
combo_output = Gtk::ComboBox.new(model_output)
|
||||
|
||||
frame_output.add(combo_output)
|
||||
|
||||
# Gtk::Frame for selected target
|
||||
selected = Gtk::Frame.new("Selected")
|
||||
page.pack_start(selected, false, false, 10)
|
||||
|
||||
# Pack & renderer combo_all
|
||||
renderer = Gtk::CellRendererText.new
|
||||
combo_output.pack_start(renderer, true)
|
||||
combo_output.set_attributes(renderer, :text => 0)
|
||||
|
||||
# Gtk::Label for selected payload
|
||||
label = Gtk::Label.new
|
||||
selected.add(label)
|
||||
|
||||
# Signal for combo payload
|
||||
combo_output.signal_connect('changed') do ||
|
||||
text = "<span foreground='blue' size='xx-large'>#{combo_output.active_iter[0]}</span>"
|
||||
label.set_markup(text)
|
||||
choice = combo_output.active_iter[0]
|
||||
@fmt = 'perl' if (choice == 'Perl')
|
||||
@fmt = 'raw' if (choice == 'Raw')
|
||||
@fmt = 'c' if (choice == 'C')
|
||||
@fmt = 'js_be' if (choice == 'Javascript' and Rex::Arch.endian(@payload.arch) == ENDIAN_BIG)
|
||||
@fmt = 'js_le' if (choice == 'Javascript' and ! @fmt)
|
||||
|
||||
@myassistant.set_page_complete(@output_page, true)
|
||||
end
|
||||
page.show_all
|
||||
end
|
||||
|
||||
def updatevalue(model, column, path, text)
|
||||
iter = model.get_iter(path)
|
||||
iter[column] = text
|
||||
@hash[iter.get_value(KEY)] = text
|
||||
end # def updatevalue
|
||||
|
||||
def summary_completion
|
||||
return Gtk::Label.new("Summary").show
|
||||
end # def summary_completion
|
||||
|
||||
def update_page(page)
|
||||
if @myassistant.get_page_title(page) == "Options"
|
||||
|
||||
# Clear treeview
|
||||
@model_required.clear
|
||||
|
||||
# Exploits options
|
||||
if (@payload.options.has_options?)
|
||||
@payload.options.each do |key, opt|
|
||||
next if (opt.advanced?)
|
||||
next if (opt.evasion?)
|
||||
pack(@model_required, key, opt)
|
||||
end
|
||||
end
|
||||
|
||||
elsif @myassistant.get_page_title(page) == "Summary"
|
||||
begin
|
||||
buf = @payload.generate_simple(
|
||||
'Format' => @fmt,
|
||||
'Options' => @hash)
|
||||
MsfDialog::Payload.new(@active_module.refname, buf)
|
||||
rescue
|
||||
MsfDialog::Error.new($gtk2driver.main, "Error generating payload: #{$!}")
|
||||
end
|
||||
end
|
||||
end # def update_page
|
||||
|
||||
end # class Payload
|
||||
|
||||
end # class MsfAssistant
|
||||
|
||||
end
|
||||
end
|
||||
end
|
||||
Loading…
Reference in New Issue