47 lines
1.3 KiB
Ruby
47 lines
1.3 KiB
Ruby
module MsfTest
|
|
module MeterpreterSpecHelper
|
|
|
|
def self.included(base)
|
|
base.class_eval do
|
|
|
|
|
|
def hlp_run_command_check_output(name,command,success_strings=[],fail_strings=[], fail_exception_strings=[])
|
|
|
|
temp_command_file = "#{@output_directory}/#{name}"
|
|
|
|
command_output = Rex::Ui::Text::Output::File.new(temp_command_file)
|
|
@session.init_ui(@input, command_output)
|
|
|
|
command_output.print_line("meterpreter_functional_test_start")
|
|
puts "Running Command: " + command
|
|
@session.run_cmd(command)
|
|
command_output.print_line("meterpreter_functional_test_end")
|
|
data = hlp_file_to_string(temp_command_file)
|
|
|
|
## Ugh, this is ghetto. test unit sucks so baddd.
|
|
x = MsfTestCaseHelper.new
|
|
x.complete?(data,"meterpreter_functional_test_start","meterpreter_functional_test_end").should be_true
|
|
x.all_successes_exist?(data, success_strings).should be_true
|
|
x.no_failures_exist?(data, fail_strings, fail_exception_strings ).should be_true
|
|
end
|
|
|
|
def hlp_file_to_string(filename)
|
|
data = ""
|
|
f = File.open(filename, "r")
|
|
f.each_line do |line|
|
|
data += line
|
|
end
|
|
return data
|
|
end
|
|
|
|
def hlp_string_to_file(string, filepath)
|
|
# Create a new file and write to it
|
|
File.open(filepath, 'w') do |f2|
|
|
f2.puts string
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|