1
mirror of https://github.com/rapid7/metasploit-framework synced 2024-10-29 18:07:27 +01:00

Land #9606, Add some tests for cmd_exec

Merge branch 'for-the-windows' into upstream-master
This commit is contained in:
bwatters-r7 2018-03-06 15:13:07 -06:00
commit ccbc7fcefe
No known key found for this signature in database
GPG Key ID: ECC0F0A52E65F268

View File

@ -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