Support expressions for meterpreter's irb too

Author: zeroSteiner

lib/rex/post/meterpreter/ui/console/command_dispatcher/core.rb

@@ -30,6 +30,10 @@ def initialize(shell)
     self.bgjob_id   = 0
   end
 
+  @@irb_opts = Rex::Parser::Arguments.new(
+    "-h" => [ false, "Help banner."                  ],
+    "-e" => [ true,  "Expression to evaluate."       ])
+
   @@load_opts = Rex::Parser::Arguments.new(
     "-l" => [ false, "List all available extensions" ],
     "-h" => [ false, "Help menu."                    ])
@@ -322,15 +326,43 @@ def cmd_interact(*args)
 
   alias cmd_interact_tabs cmd_close_tabs
 
+  def cmd_irb_help
+    print_line "Usage: irb"
+    print_line
+    print_line "Execute commands in a Ruby environment"
+    print @@irb_opts.usage()
+  end
+
   #
   # Runs the IRB scripting shell
   #
   def cmd_irb(*args)
-    print_status("Starting IRB shell")
-    print_status("The 'client' variable holds the meterpreter client\n")
+    expressions = []
+
+    # Parse the command options
+    @@irb_opts.parse(args) do |opt, idx, val|
+      case opt
+        when '-e'
+          expressions << val
+        when '-h'
+          cmd_irb_help
+          return
+      end
+    end
 
     session = client
     framework = client.framework
+
+    if !expressions.empty?
+      expressions.each do |expression|
+        eval(expression, binding)
+      end
+      return
+    end
+
+    print_status("Starting IRB shell")
+    print_status("The 'client' variable holds the meterpreter client\n")
+
     Rex::Ui::Text::IrbShell.new(binding).run
   end