Fix lurking bugs in the shell's prompt_yesno

2018-08-16 09:05:22 -05:00 · 2018-08-16 09:05:22 -05:00 · 4375a3fbfb
parent c045f70e80
commit 4375a3fbfb
1 changed files with 2 additions and 2 deletions
--- a/lib/rex/ui/text/shell.rb
+++ b/lib/rex/ui/text/shell.rb
@ -434,9 +434,9 @@ protected
  #
  def prompt_yesno(query)
    p = "#{query} [y/N]"
-    old_p = [self.prompt.sub(/#{self.prompt_char} $/, ''), self.prompt_char]
+    old_p = [self.prompt.sub(/#{Regexp.escape(self.prompt_char)} $/, ''), self.prompt_char]
    update_prompt p, ' ', true
-    /^y/ === get_input_line
+    /^y/i === get_input_line
  ensure
    update_prompt *old_p, true
  end