handle sessions still open

Author: busterb

lib/rex/ui/text/shell.rb

@@ -128,7 +128,7 @@ def run(&block)
 
       while true
         # If the stop flag was set or we've hit EOF, break out
-        break if (self.stop_flag or self.stop_count > 1)
+        break if self.stop_flag || self.stop_count > 1
 
         init_tab_complete
 
@@ -378,6 +378,8 @@ def get_input_line
         output.input = nil
         log_output(input.prompt)
       else
+        self.stop_count += 1
+        next if self.stop_count > 1
         run_single("quit")
       end
     end