new release for msfassistant

git-svn-id: file:///home/svn/framework3/trunk@4965 4d416f70-5f16-0410-b530-b9f4589650da

lib/msf/ui/gtk2/assistant.rb

@@ -1,3 +1,4 @@
 require 'msf/ui/gtk2/assistant/skeleton'
+require 'msf/ui/gtk2/assistant/types'
 require 'msf/ui/gtk2/assistant/tips'
 require 'msf/ui/gtk2/assistant/exploit'

lib/msf/ui/gtk2/assistant/exploit.rb

@@ -65,8 +65,13 @@ module Msf
 
 
             # Main interface
-            @model_required = Gtk::ListStore.new(String, String, String, String)
+            @frame_advanced = Gtk::Expander.new("Advanced")
-            @model_advanced = Gtk::ListStore.new(String, String, String, String)
+            @frame_evasion = Gtk::Expander.new("Evasion")
+            @options_required = Gtk::VBox.new(false, 0)
+            @options_advanced = Gtk::VBox.new(false, 0)
+            @options_evasion = Gtk::VBox.new(false, 0)
+
+            # Begin the wizard
             target_completion()
 
             # Build the left frame
@@ -95,14 +100,14 @@ module Msf
           #
           def save
             @mydriver.exploit.datastore.import_options_from_hash(@hash, imported = false)
-            
+
             # If payload
             begin
               @mydriver.payload.share_datastore(@mydriver.exploit.datastore)
             rescue
               nil
             end
-            
+
             # TODO: choose the $gtk2driver or @mydriver.exploit ?
             $gtk2driver.active_module = @mydriver.exploit
             $gtk2driver.save_config
@@ -132,43 +137,29 @@ module Msf
               [@label_options], 			# actual
               [@label_review]				# next
               )
-              button_forward.set_sensitive(false)
+              #button_forward.set_sensitive(false)
               display()
               options_completion()
             elsif (self.page == "options")
-              self.page = "end"
+              if not validate()
-              refresh_label(	[@label_target, @label_payload, @label_options],
+                self.page = "options"
-              [@label_review],
+                refresh_label(	[@label_target, @label_payload],	# historic
-              nil
+                [@label_options], 			# actual
-              )
+                [@label_review]				# next
-              display()
+                )
-              review_completion()
+                #button_forward.set_sensitive(false)
-            end
+                display()
-          end
+                options_completion()
-
+              else
-          #
+                self.page = "end"
-          # Validate options in datastore
+                refresh_label(	[@label_target, @label_payload, @label_options],
-          #
+                [@label_review],
-          def validate
+                nil
-            errors = []
+                )
-            @mydriver.exploit.datastore.import_options_from_hash(@hash)
+                display()
-
+                review_completion()
-            @mydriver.exploit.options.each_pair do |name, option|
-              if (!option.valid?(@mydriver.exploit.datastore[name]))
-                errors << name
-
-                # If the option is valid, normalize its format to the correct type.
-              elsif ((val = option.normalize(@mydriver.exploit.datastore[name])) != nil)
-                @mydriver.exploit.datastore.update_value(name, val)
               end
             end
-
-            if (errors.empty? == false)
-              button_forward.set_sensitive(false)
-              # MsfDialog::Error.new(self, "Failed to validate : #{errors.join(', ')}")
-            else
-              button_forward.set_sensitive(true)
-            end
           end
 
           #
@@ -312,32 +303,43 @@ module Msf
           # Display options view
           #
           def options_completion
+            
+            # Expanded frame
+            @frame_advanced.each do |widget|
+              @frame_advanced.remove(widget)
+            end
+            @frame_evasion.each do |widget|
+              @frame_evasion.remove(widget)
+            end
 
-            #
+            # Required
-            @button_forward.set_sensitive(false)
+            @options_required.each do |widget|
+              @options_required.remove(widget)
+            end
 
-            # Clear treeview
+            # Advanced
-            @model_required.clear
+            @options_advanced.each do |widget|
-            @model_advanced.clear
+              @options_advanced.remove(widget)
+            end
+            
+            # Evasion
+            @options_evasion.each do |widget|
+              @options_evasion.remove(widget)
+            end
 
-            # An expander view for advanced options
+            # Pack
-            frame_advanced = Gtk::Expander.new('Advanced')
+            self.main.pack_start(@options_required, false, false, 10)
-
+            self.main.pack_start(@frame_advanced, false, false, 10)
-            # TreeView
+            self.main.pack_start(@frame_evasion, false, false, 10)
-            @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)
 
             # Payload options
             @mydriver.payload = framework.payloads.create(@hash["PAYLOAD"])
             @mydriver.payload.options.each do |key, opt|
               if (opt.required?)
-                pack(@model_required, key, opt, @mydriver.exploit.datastore[key])
+                @options_required.pack_start(add_option(key, opt, @mydriver.exploit.datastore[key]), false, false, 10)
               else
-                pack(@model_advanced, key, opt, @mydriver.exploit.datastore[key])
+                @options_advanced.pack_start(add_option(key, opt, @mydriver.exploit.datastore[key]), false, false, 10)
               end
             end
 
@@ -345,18 +347,82 @@ module Msf
             @mydriver.exploit.options.sorted.each do |key, opt|
               next if (opt.evasion?)
               if (opt.required?)
-                pack(@model_required, key, opt, @mydriver.exploit.datastore[key])
+                @options_required.pack_start(add_option(key, opt, @mydriver.exploit.datastore[key]), false, false, 10)
               else
-                pack(@model_advanced, key, opt, @mydriver.exploit.datastore[key])
+                @options_advanced.pack_start(add_option(key, opt, @mydriver.exploit.datastore[key]), false, false, 10)
+              end
+            end
+            
+            # Evasion options
+            @mydriver.exploit.options.sorted.each do |key, opt|
+              if (opt.evasion?)
+                @options_evasion.pack_start(add_option(key, opt, @mydriver.exploit.datastore[key]), false, false, 10)
+              end
+            end
+            
+            
+            # If backup or save ?
+            #@options.required
+
+            @frame_advanced.set_label("Advanced")
+            @frame_advanced.add(@options_advanced)
+            @frame_evasion.set_label("Evasion")
+            @frame_evasion.add(@options_evasion)
+
+            self.main.show_all
+          end
+
+          #
+          # Validate options in datastore
+          #
+          def validate
+            @options_required.each do |widget|
+              name, value = widget.get_pair
+              begin
+                if (@mydriver.exploit.options[name].default.to_s == value)
+                  nil
+                else
+                  @hash[name] = value
+                end
+              rescue
+                nil
               end
             end
 
-            # add treeview to frame
+            @options_advanced.each do |widget|
-            self.main.pack_start(@treeview_required, false, false, 0)
+              name, value = widget.get_pair
-            self.main.pack_start(frame_advanced, false, false, 10)
+              begin
-            frame_advanced.add(@treeview_advanced)
+                if (@mydriver.exploit.options[name].default.to_s == value)
+                  nil
+                else
+                  @hash[name] = value
+                end
+              rescue
+                nil
+              end
+            end
 
-            self.main.show_all
+            errors = []
+            @mydriver.exploit.datastore.import_options_from_hash(@hash)
+
+            @mydriver.exploit.options.each_pair do |name, option|
+              if (!option.valid?(@mydriver.exploit.datastore[name]))
+                errors << name
+
+                # If the option is valid, normalize its format to the correct type.
+              elsif ((val = option.normalize(@mydriver.exploit.datastore[name])) != nil)
+                @mydriver.exploit.datastore.update_value(name, val)
+              end
+            end
+
+            if (errors.empty? == false)
+              #button_forward.set_sensitive(false)
+              MsfDialog::Error.new(self, "Failed to validate : #{errors.join(', ')}")
+              false
+            else
+              true
+              #button_forward.set_sensitive(true)
+            end
           end
 
           #
@@ -379,82 +445,6 @@ module Msf
             self.main.show_all
           end
 
-          #
-          # Add column and tips
-          #
-          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)
-            column.set_min_width(100)
-            treeview.append_column(column)
-
-            # Init tips on the treeview
-            tips = AssistantTips.new(column)
-            tips.add_view(@treeview_required)
-            tips.add_view(@treeview_advanced)
-          end
-
-          #
-          # Display values on the treeview
-          #
-          def pack(model, key, opt, value = "")
-            default = ""
-            if (opt.default.to_s.length > 50)
-              size = opt.default.to_s.length
-              default = "[...]" + opt.default.to_s[size - 27, size]
-            else
-              default = opt.default.to_s
-            end
-            iter = model.append
-            iter[KEY] = key
-            iter[DEFAULT] = default
-            iter[DESC] = opt.desc.to_s
-            if (key == "LHOST" and value == "")
-              iter[VALUE] = Rex::Socket.source_address
-              @hash['LHOST'] = Rex::Socket.source_address
-            else
-              iter[VALUE] = value.to_s
-              @hash['LHOST'] = value
-            end
-            validate()
-          end
-
-          #
-          # Update changing value
-          #
-          def updatevalue(model, column, path, text)
-            iter = model.get_iter(path)
-            iter[column] = text
-            @hash[iter.get_value(KEY)] = text
-            validate()
-          end
-
           #
           # Fire !!
           #

lib/msf/ui/gtk2/assistant/skeleton.rb

@@ -11,7 +11,7 @@ module Msf
         include Msf::Ui::Gtk2::MyControls
 
         attr_accessor :vbox, :hbox, :main, :page, :bbox, :vbox_left, :vbox_label, :save_button
-        attr_accessor :button_forward
+        attr_accessor :button_back, :button_forward, :option_state
 
         def initialize(title)
           super
@@ -32,6 +32,9 @@ module Msf
           # HBox
           @hbox = Gtk::HBox.new(false, 10)
           @vbox.pack_start(@hbox, true, false, 0)
+          
+          # Option state
+          @option_state = true
 
           # Left frame
           @vbox_left = Gtk::VBox.new(false, 5)
@@ -120,7 +123,7 @@ module Msf
           if state
             label.set_markup("<span foreground=\"black\"><b>#{text}</b></span>")
           else
-            label.set_markup("<span foreground=\"#C0C0C0\">#{text}</span>")
+            label.set_markup("<span foreground=\"white\">#{text}</span>")
           end
           return label
         end
@@ -191,7 +194,7 @@ module Msf
         def refresh_label(hist, actual , nex)
           if not (hist == nil)
             hist.each do |label|
-              label.set_markup("<span foreground=\"white\"><i>#{label.text}</i></span>")
+              label.set_markup("<span foreground=\"black\"><i>#{label.text}</i></span>")
             end
           end
 
@@ -201,11 +204,46 @@ module Msf
 
           if not (nex == nil)
             nex.each do |label|
-              label.set_markup("<span foreground=\"#C0C0C0\"><b>#{label.text}</b></span>")
+              label.set_markup("<span foreground=\"white\">#{label.text}</span>")
             end
           end
         end
 
+        #
+        # Options stuff
+        #
+        def add_option(key, opt, store="")
+          type = opt.type
+          
+          if (type == "string")
+            MsfTypes::String.new(key, opt, store)
+          elsif (type == "raw")
+            MsfTypes::Raw.new(key, opt)
+          elsif (type == "bool")
+            MsfTypes::Bool.new(key, opt)
+          elsif (type == "port")
+            MsfTypes::Port.new(key, opt)
+          elsif (type == "address")
+            MsfTypes::Address.new(key, opt, store)
+          elsif (type == "path")
+            MsfTypes::String.new(key, opt, store)
+          elsif (type == "integer")
+            MsfTypes::Integer.new(key, opt)
+          elsif (type == "enum")
+            MsfTypes::String.new(key, opt, store)
+          elsif (type == "addressrange")
+            MsfTypes::String.new(key, opt, store)
+          end
+        end
+        
+        #
+        # Dummy function
+        #
+        def options_validate
+          raise NotImplementedError, "Subclass must implement options_validate"
+        end
+
+
         #########
         private #
         #########

lib/msf/ui/gtk2/assistant/types.rb

@@ -0,0 +1,27 @@
+###
+#
+# Core option types.  The core supported option types are:
+#
+# OptString  - Multi-byte character string
+# OptRaw     - Multi-byte raw string
+# OptBool    - Boolean true or false indication
+# OptPort    - TCP/UDP service port
+# OptAddress - IP address or hostname
+# OptPath    - Path name on disk
+# OptInt     - An integer value
+# OptEnum    - Select from a set of valid values
+# OptAddressRange - A subnet or range of addresses
+#
+###
+
+require 'msf/ui/gtk2/assistant/types/skeleton'
+
+require 'msf/ui/gtk2/assistant/types/string'
+require 'msf/ui/gtk2/assistant/types/raw'
+require 'msf/ui/gtk2/assistant/types/bool'
+require 'msf/ui/gtk2/assistant/types/port'
+require 'msf/ui/gtk2/assistant/types/address'
+#require 'msf/ui/gtk2/assistant/types/path'
+require 'msf/ui/gtk2/assistant/types/integer'
+#require 'msf/ui/gtk2/assistant/types/enum'
+#require 'msf/ui/gtk2/assistant/types/addressrange'

lib/msf/ui/gtk2/assistant/types/address.rb

@@ -0,0 +1,72 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptRaw     - Multi-byte raw string
+        #
+        class Address < Msf::Ui::Gtk2::SkeletonType
+
+          def initialize(key, opt, store)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_address(key, opt.default, store)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_address(name, value, store)
+            hbox = Gtk::HBox.new(false, 10)
+            self.pack_start(hbox, false, false, 0)
+
+            @name = name
+
+            label = Gtk::Label.new
+            label.set_markup("<span foreground=\"black\">#{@name} :</span>")
+            hbox.pack_start(label, false, false, 0)
+
+            @entry = Gtk::Entry.new
+            if (name == "LHOST" and store == "")
+              @entry.set_text(Rex::Socket.source_address)
+            elsif (not store == "")
+              @entry.set_text(store)
+            else
+              @entry.set_text(value)
+            end
+            @entry.set_width_chars(15)
+            @entry.set_max_length(15)
+            hbox.pack_start(@entry, false, false, 0)
+          end
+
+          #
+          #
+          #
+          def check?
+            if (@entry.text == "")
+              false
+            else
+              true
+            end
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @entry.text
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

lib/msf/ui/gtk2/assistant/types/bool.rb

@@ -0,0 +1,57 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptBool    - Boolean true or false indication
+        #
+        class Bool < Msf::Ui::Gtk2::SkeletonType
+
+          def initialize(key, opt)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_bool(key, opt.default)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_bool(name, value)
+            hbox = Gtk::HBox.new(false, 0)
+            self.pack_start(hbox, false, false, 0)
+            
+            @name = name
+
+            @checkbutton = Gtk::CheckButton.new(@name, true)
+            hbox.pack_start(@checkbutton, true, true, 0)
+
+            @checkbutton.set_active(value)
+          end
+
+          #
+          #
+          #
+          def check?
+            return @checkbutton.active?
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @checkbutton.active?
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

lib/msf/ui/gtk2/assistant/types/integer.rb

@@ -0,0 +1,68 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptString  - Multi-byte character string
+        #
+        class Integer < Msf::Ui::Gtk2::SkeletonType
+
+          include Msf::Ui::Gtk2::MyControls
+
+          def initialize(key, opt)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_integer(key, opt.default.to_s)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_integer(name, value)
+            hbox = Gtk::HBox.new(false, 10)
+            self.pack_start(hbox, false, false, 0)
+            
+            @name = name
+
+            label = Gtk::Label.new
+            label.set_markup("<span foreground=\"black\">#{@name} :</span>")
+            hbox.pack_start(label, false, false, 0)
+
+            @entry = Gtk::Entry.new
+            @entry.set_text(value.to_s)
+            @entry.set_width_chars(15)
+            @entry.set_max_length(15)
+            hbox.pack_start(@entry, false, false, 0)
+          end
+
+          #
+          #
+          #
+          def check?
+            if (@entry.text == "")
+              false
+            else
+              true
+            end
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @entry.text
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

lib/msf/ui/gtk2/assistant/types/port.rb

@@ -0,0 +1,66 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptRaw     - Multi-byte raw string
+        #
+        class Port < Msf::Ui::Gtk2::SkeletonType
+
+          def initialize(key, opt)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_port(key, opt.default)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_port(name, value)
+            hbox = Gtk::HBox.new(false, 10)
+            self.pack_start(hbox, false, false, 0)
+            
+            @name = name            
+
+            label = Gtk::Label.new
+            label.set_markup("<span foreground=\"black\">#{@name} :</span>")
+            hbox.pack_start(label, false, false, 0)
+
+            @entry = Gtk::Entry.new
+            @entry.set_text(value.to_s)
+            @entry.set_width_chars(5)
+            @entry.set_max_length(5)
+            hbox.pack_start(@entry, false, false, 0)
+          end
+
+          #
+          #
+          #
+          def check?
+            if (@entry.text == "")
+              false
+            else
+              true
+            end
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @entry.text
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

lib/msf/ui/gtk2/assistant/types/raw.rb

@@ -0,0 +1,67 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptRaw     - Multi-byte raw string
+        #
+        class Raw < Msf::Ui::Gtk2::SkeletonType
+
+          def initialize(key, opt)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_raw(key, opt.default)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_raw(name, value)
+            hbox = Gtk::HBox.new(false, 10)
+            self.pack_start(hbox, false, false, 0)
+
+            @name = name
+
+            label = Gtk::Label.new
+            label.set_markup("<span foreground=\"black\">#{@name} :</span>")
+            hbox.pack_start(label, false, false, 0)
+
+            @combo = Gtk::ComboBox.new(true)
+            exitfunc = ["seh", "thread", "process"]
+            exitfunc.each_with_index do |val, idx|
+              @combo.prepend_text(val)
+              if (val == value)
+                @combo.set_active(idx)
+              end
+            end
+
+            hbox.pack_start(@combo, false, false, 0)
+          end
+
+          #
+          #
+          #
+          def check?
+            return @combo.active
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @combo.active_text
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+end

lib/msf/ui/gtk2/assistant/types/skeleton.rb

@@ -0,0 +1,28 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      #
+      #
+      #
+      class SkeletonType < Gtk::VBox
+
+        def initialize
+          super(false, 0)
+        end
+
+        #
+        #
+        #
+        def pack_description(description)
+          label = Gtk::Label.new
+          label.set_alignment(0, 1)
+          label.set_markup("<i>#{description}</i>")
+          self.pack_start(label, false, false, 0)
+        end
+
+      end
+
+    end
+  end
+end

lib/msf/ui/gtk2/assistant/types/string.rb

@@ -0,0 +1,76 @@
+module Msf
+  module Ui
+    module Gtk2
+
+      class MsfTypes
+
+        #
+        # OptString  - Multi-byte character string
+        #
+        class String < Msf::Ui::Gtk2::SkeletonType
+
+          include Msf::Ui::Gtk2::MyControls
+
+          def initialize(key, opt, store)
+            super()
+
+            pack_description(opt.desc.to_s + " :")
+            pack_string(key, opt.default.to_s, store)
+
+            return self
+          end
+
+          #
+          #
+          #
+          def pack_string(name, value, store)
+            hbox = Gtk::HBox.new(false, 10)
+            self.pack_start(hbox, false, false, 0)
+
+            @name = name
+
+            label = Gtk::Label.new
+            label.set_markup("<span foreground=\"black\">#{@name} :</span>")
+            hbox.pack_start(label, false, false, 0)
+
+            @entry = Gtk::Entry.new
+            if (store == "")
+              @entry.set_text(value)
+            else
+              @entry.set_text(store)
+            end
+            if (name == "Locale")
+              @entry.set_width_chars(15)
+              @entry.set_max_length(15)
+              hbox.pack_start(@entry, false, false, 0)
+            else
+              hbox.pack_start(@entry, true, true, 0)
+            end
+          end
+
+          #
+          #
+          #
+          def check?
+            if (@entry.text == "")
+              false
+            else
+              true
+            end
+          end
+
+          #
+          #
+          #
+          def get_pair
+            return @name, @entry.text
+          end
+
+        end
+
+
+      end
+
+    end
+  end
+end

lib/msf/ui/gtk2/driver.rb

@@ -29,10 +29,11 @@ module Msf
       #
       ###
       class Driver < Msf::Ui::Driver
-
+        
+        # main view
         attr_accessor :session_tree, :module_tree, :job_tree, :log_text, :module_model
         attr_accessor :module_completion, :main, :tips
-
+        
         include Msf::Ui::Gtk2::FrameworkEventManager
 
         ConfigCore  = "framework/core"

lib/msf/ui/gtk2/frame/modules_tree.rb

@@ -117,7 +117,7 @@ module Msf
               if (type == "Standard")
                 MsfAssistant::Exploit.new(active_module.get_value(MODULE))
               elsif (type ==  "Payloads")
-                MsfAssistant::Payload.new(active_module.get_value(MODULE))
+                MsfDialog::Error.new($gtk2driver.main, "Not available")
               else
                 MsfDialog::Error.new($gtk2driver.main, "Not available")
               end

lib/msf/ui/gtk2/meterpreter/stdapi/fs.rb

@@ -20,18 +20,21 @@ module Msf
             @client = client
 
             # call the parent
-            super("MsfBrowser on #{@client.tunnel_peer}")
+            super("File Browser on #{@client.tunnel_peer}")
 
             # Define the models (navigation, view)
             # TODO: model for navigation
             @model_view = Gtk::ListStore.new(String, String, TrueClass, Gdk::Pixbuf)
+            @model_local = Gtk::TreeStore.new(String, String, TrueClass, Gdk::Pixbuf)
 
-            # Define the parent the parent
+            # Define thes parents
             @parent = "/"
+            @parent_local = "/"
 
             # Define the icons for folders and files
             @file_pixbuf = Gdk::Pixbuf.new(driver.get_image("msf_file.png"))
             @folder_pixbuf = Gdk::Pixbuf.new(driver.get_image("msf_folder.png"))
+            @local_folder_pixbuf = Gdk::Pixbuf.new(driver.get_image("msf_local_folder.png"))
 
             @model_view.set_default_sort_func do |a, b|
               if !a[COL_IS_DIR] and b[COL_IS_DIR]
@@ -58,6 +61,25 @@ module Msf
             hbox.pack_start(vbox_left, false, false, 0)
             hbox.pack_start(vbox_right, true, true, 0)
 
+            # Local
+            @treeview_local = Gtk::TreeView.new(@model_local)
+            vbox_left.pack_start(@treeview_local, true, true, 0)
+            renderer_pix = Gtk::CellRendererPixbuf.new
+            renderer_name = Gtk::CellRendererText.new
+
+            column_name = Gtk::TreeViewColumn.new
+            column_name.pack_start(renderer_pix, false)
+            column_name.set_cell_data_func(renderer_pix) do |column, cell, model, iter|
+              cell.pixbuf = iter[COL_PIXBUF]
+            end
+            column_name.pack_start(renderer_name, true)
+            column_name.set_cell_data_func(renderer_name) do |column, cell, model, iter|
+              cell.text = iter[COL_DISPLAY_NAME]
+            end
+            @treeview_local.append_column(column_name)
+
+            # Remote
+
             # Label, Entry and Signal for the path selection
             hbox_path = Gtk::HBox.new(false, 0)
             vbox_right.pack_start(hbox_path, false, true, 0)
@@ -88,6 +110,7 @@ module Msf
             end
 
             # Populate the view
+            local_ls
             cmd_ls
 
             sw.add(iconview)
@@ -100,6 +123,7 @@ module Msf
           # Lists file on the remote machine
           #
           def cmd_ls(*args)
+            # Try to list the remote path
             begin
               # Just ignore the invalid UTF8
               # Don't know why GLib.filename_to_utf8() don't work ;-(
@@ -123,12 +147,25 @@ module Msf
                 iter[COL_IS_DIR] = is_dir
                 iter[COL_PIXBUF] = is_dir ? @folder_pixbuf : @file_pixbuf
               end
+              # If not possible return a *warning***
             rescue
               MsfDialog::Warning.new(self, "No entries exist in #{path}")
               cmd_ls
             end
           end # cmd_ls
 
+          def local_ls(*args)
+            Dir.glob(File.join(@parent_local, "*")).each do |path|
+              if FileTest.directory?(path)
+                iter = @model_local.append(nil)
+                iter[COL_DISPLAY_NAME] = GLib.filename_to_utf8(File.basename(path))
+                iter[COL_PATH] = path
+                iter[COL_IS_DIR] = true
+                iter[COL_PIXBUF] = @local_folder_pixbuf
+              end
+            end
+          end
+
         end # Fs
 
       end