96 lines
2.9 KiB
Ruby
96 lines
2.9 KiB
Ruby
|
module Msf
|
||
|
|
module Ui
|
||
|
|
module Gtk2
|
||
|
|
|
||
|
|
class Stdapi
|
||
|
|
|
||
|
|
###
|
||
|
|
#
|
||
|
|
# The file system portion of the standard API extension.
|
||
|
|
#
|
||
|
|
###
|
||
|
|
class Fs < Msf::Ui::Gtk2::SkeletonBasic
|
||
|
|
COL_PATH, COL_DISPLAY_NAME, COL_IS_DIR, COL_PIXBUF = (0..4).to_a
|
||
|
|
|
||
|
|
include Msf::Ui::Gtk2::MyControls
|
||
|
|
|
||
|
|
def initialize(client)
|
||
|
|
@client = client
|
||
|
|
super("MsfBrowser on #{@client.via_session}")
|
||
|
|
@model = Gtk::ListStore.new(String, String, TrueClass, Gdk::Pixbuf)
|
||
|
|
@parent = "/"
|
||
|
|
|
||
|
|
@file_pixbuf = Gdk::Pixbuf.new(driver.get_image("gnome-file-c.png"))
|
||
|
|
@folder_pixbuf = Gdk::Pixbuf.new(driver.get_image("gnome-folder.png"))
|
||
|
|
|
||
|
|
@model.set_default_sort_func do |a, b|
|
||
|
|
if !a[COL_IS_DIR] and b[COL_IS_DIR]
|
||
|
|
1
|
||
|
|
elsif a[COL_IS_DIR] and !b[COL_IS_DIR]
|
||
|
|
-1
|
||
|
|
else
|
||
|
|
a[COL_DISPLAY_NAME] <=> b[COL_DISPLAY_NAME]
|
||
|
|
end
|
||
|
|
end
|
||
|
|
@model.set_sort_column_id(Gtk::TreeSortable::DEFAULT_SORT_COLUMN_ID, Gtk::SORT_ASCENDING)
|
||
|
|
|
||
|
|
cmd_ls
|
||
|
|
set_default_size(650, 400)
|
||
|
|
set_border_width(8)
|
||
|
|
|
||
|
|
vbox = Gtk::VBox.new(false, 0)
|
||
|
|
add(vbox)
|
||
|
|
|
||
|
|
sw = Gtk::ScrolledWindow.new
|
||
|
|
sw.shadow_type = Gtk::SHADOW_ETCHED_IN
|
||
|
|
sw.set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_AUTOMATIC)
|
||
|
|
vbox.pack_start(sw, true, true, 0)
|
||
|
|
|
||
|
|
iconview = Gtk::IconView.new(@model)
|
||
|
|
iconview.selection_mode = Gtk::SELECTION_MULTIPLE
|
||
|
|
iconview.text_column = COL_DISPLAY_NAME
|
||
|
|
iconview.pixbuf_column = COL_PIXBUF
|
||
|
|
iconview.signal_connect("item_activated") do |iview, path|
|
||
|
|
iter = @model.get_iter(path)
|
||
|
|
if iter[COL_DISPLAY_NAME]
|
||
|
|
@parent = iter[COL_PATH]
|
||
|
|
cmd_ls
|
||
|
|
# up_button.sensitive = true
|
||
|
|
end
|
||
|
|
end
|
||
|
|
sw.add(iconview)
|
||
|
|
iconview.grab_focus
|
||
|
|
|
||
|
|
show_all
|
||
|
|
end
|
||
|
|
|
||
|
|
def cmd_ls(*args)
|
||
|
|
@model.clear
|
||
|
|
path = args[0] || @client.fs.dir.getwd
|
||
|
|
items = 0
|
||
|
|
|
||
|
|
# Enumerate each item...
|
||
|
|
@client.fs.dir.entries_with_info(path).sort { |a,b| a['FileName'] <=> b['FileName'] }.each do |p|
|
||
|
|
if p['StatBuf'].ftype[0,3] == "dir"
|
||
|
|
is_dir = true
|
||
|
|
elsif p['StatBuf'].ftype[0,3] == "fil"
|
||
|
|
is_dir = false
|
||
|
|
end
|
||
|
|
iter = @model.append
|
||
|
|
iter[COL_DISPLAY_NAME] = p['FileName'] || 'unknown'
|
||
|
|
iter[COL_PATH] = path
|
||
|
|
iter[COL_IS_DIR] = is_dir
|
||
|
|
iter[COL_PIXBUF] = is_dir ? @folder_pixbuf : @file_pixbuf
|
||
|
|
items += 1
|
||
|
|
end
|
||
|
|
|
||
|
|
end # cmd_ls
|
||
|
|
|
||
|
|
end # Fs
|
||
|
|
|
||
|
|
end
|
||
|
|
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|