Use a continuation flag to disable tab completion

Author: zeroSteiner

lib/rex/ui/text/shell.rb

@@ -47,6 +47,7 @@ def initialize(prompt, prompt_char = '>', histfile = nil, framework = nil)
     # Initialize the prompt
     self.init_prompt = prompt
     self.cont_prompt = ' > '
+    self.cont_flag = false
     self.prompt_char = prompt_char
 
     self.histfile = histfile
@@ -57,7 +58,8 @@ def initialize(prompt, prompt_char = '>', histfile = nil, framework = nil)
 
   def init_tab_complete
     if (self.input and self.input.supports_readline)
-      self.input = Input::Readline.new(lambda { |str| tab_complete(str) })
+      # Unless cont_flag because there's no tab complete for continuation lines
+      self.input = Input::Readline.new(lambda { |str| tab_complete(str) unless cont_flag })
       if Readline::HISTORY.length == 0 and histfile and File.exist?(histfile)
         File.readlines(histfile).each { |e|
           Readline::HISTORY << e.chomp
@@ -355,13 +357,15 @@ def get_input_line
     line = "\\\n"
     prompt_needs_reset = false
 
+    self.cont_flag = false
     while line =~ /(^|[^\\])\\\s*$/
       # Strip \ and all the trailing whitespace
       line.sub!(/\\\s*/, '')
 
       if line.length > 0
         # Using update_prompt will overwrite the primary prompt
         input.prompt = output.update_prompt(self.cont_prompt)
+        self.cont_flag = true
         prompt_needs_reset = true
       end
 
@@ -370,6 +374,7 @@ def get_input_line
       output.input = nil
       log_output(input.prompt)
     end
+    self.cont_flag = false
 
     if prompt_needs_reset
       # The continuation prompt was used so reset the prompt
@@ -424,6 +429,11 @@ def log_output(buf)
   attr_accessor :histfile # :nodoc:
   attr_accessor :hist_last_saved # the number of history lines when last saved/loaded
   attr_accessor :log_source, :stop_count # :nodoc:
+  attr_reader   :cont_flag # :nodoc:
+
+private
+
+  attr_writer   :cont_flag # :nodoc:
 
 end