Land #9606, Add some tests for cmd_exec

Merge branch 'for-the-windows' into upstream-master

test/modules/post/test/cmd_exec.rb

@@ -0,0 +1,94 @@
+require 'rex'
+
+lib = File.join(Msf::Config.install_root, "test", "lib")
+$LOAD_PATH.push(lib) unless $LOAD_PATH.include?(lib)
+require 'module_test'
+
+class MetasploitModule < Msf::Post
+  include Msf::ModuleTest::PostTest
+  include Msf::Post::File
+
+  def initialize(info = {})
+    super(update_info(info,
+        'Name'          => 'Meterpreter cmd_exec test',
+        'Description'   => %q( This module will test the meterpreter cmd_exec API ),
+        'License'       => MSF_LICENSE,
+        'Platform'      => ['windows', 'linux', 'unix'],
+        'SessionTypes'  => ['meterpreter']
+      ))
+  end
+
+  def test_cmd_exec
+    vprint_status("Starting cmd_exec tests")
+
+    it "should return the result of echo" do
+      test_string = Rex::Text.rand_text_alpha(4)
+      if session.platform.eql? 'windows'
+        output = cmd_exec('cmd.exe', "/c echo #{test_string}")
+      else
+        output = cmd_exec("echo #{test_string}")
+      end
+      output == test_string
+    end
+
+    # trying to do a sleep in windows without trashing stdout is hard
+    unless session.platform.eql? 'windows'
+      it "should return the result after sleeping" do
+        test_string = Rex::Text.rand_text_alpha(4)
+        output = cmd_exec("sleep 1; echo #{test_string}")
+        output == test_string
+      end
+
+      it "should return the full response after sleeping" do
+        test_string = Rex::Text.rand_text_alpha(4)
+        test_string2 = Rex::Text.rand_text_alpha(4)
+        if session.platform.eql? 'windows'
+          output = cmd_exec('cmd.exe', "/c echo #{test_string} & timeout 1 > null & echo #{test_string2}")
+        else
+          output = cmd_exec("echo #{test_string}; sleep 1; echo #{test_string2}")
+        end
+        output.delete("\r") == "#{test_string}\n#{test_string2}"
+      end
+    end
+
+    it "should return the result of echo 10 times" do
+      10.times do
+        test_string = Rex::Text.rand_text_alpha(4)
+        if session.platform.eql? 'windows'
+          output = cmd_exec("cmd.exe", "/c echo #{test_string}")
+        else
+          output = cmd_exec("echo #{test_string}")
+        end
+        return false unless output == test_string
+      end
+      true
+    end
+    vprint_status("Finished cmd_exec tests")
+  end
+
+  def test_cmd_exec_quotes
+    vprint_status("Starting cmd_exec quote tests")
+
+    it "should return the result of echo with single quotes" do
+      test_string = Rex::Text.rand_text_alpha(4)
+      if session.platform.eql? 'windows'
+        output = cmd_exec("cmd.exe", "/c echo '#{test_string}'")
+        output == "'" + test_string + "'"
+      else
+        output = cmd_exec("echo '#{test_string}'")
+        output == test_string
+      end
+    end
+
+    it "should return the result of echo with double quotes" do
+      test_string = Rex::Text.rand_text_alpha(4)
+      if session.platform.eql? 'windows'
+        output = cmd_exec("cmd.exe", "/c echo \"#{test_string}\"")
+        output == "\"" + test_string + "\""
+      else
+        output = cmd_exec("echo \"#{test_string}\"")
+        output == test_string
+      end
+    end
+  end
+end