James Lee
d25756976c
Also some tests for doing so, all of which pass on Linux, Windows, and Java meterpreter, as well as shell sessions on Linux and Solaris. They will fail miserably on Windows shell sessions.
67 lines
1.2 KiB
Ruby
67 lines
1.2 KiB
Ruby
|
|
|
|
module Msf
|
|
|
|
module ModuleTest
|
|
attr_accessor :tests
|
|
attr_accessor :failures
|
|
|
|
def initialize(info={})
|
|
@tests = 0
|
|
@failures = 0
|
|
super
|
|
end
|
|
|
|
def run_all_tests
|
|
tests = self.methods.select { |m| m.to_s =~ /^test_/ }
|
|
tests.each { |test_method|
|
|
self.send(test_method)
|
|
}
|
|
|
|
end
|
|
|
|
def it(msg="", &block)
|
|
@tests += 1
|
|
begin
|
|
result = block.call
|
|
unless result
|
|
print_error("FAILED: #{msg}")
|
|
print_error("FAILED: #{error}") if error
|
|
@failures += 1
|
|
return
|
|
end
|
|
rescue ::Exception => e
|
|
print_error("FAILED: #{msg}")
|
|
print_error("Exception: #{e.class} : #{e}")
|
|
return
|
|
end
|
|
|
|
print_good("#{msg}")
|
|
end
|
|
|
|
def pending(msg="", &block)
|
|
print_status("PENDING: #{msg}")
|
|
end
|
|
end
|
|
|
|
module ModuleTest::PostTest
|
|
include ModuleTest
|
|
def run
|
|
print_status("Running against session #{datastore["SESSION"]}")
|
|
print_status("Session type is #{session.type} and platform is #{session.platform}")
|
|
|
|
t = Time.now
|
|
@tests = 0; @failures = 0
|
|
run_all_tests
|
|
|
|
vprint_status("Testing complete in #{Time.now - t}")
|
|
if (@failures > 0)
|
|
print_error("Passed: #{@tests - @failures}; Failed: #{@failures}")
|
|
else
|
|
print_status("Passed: #{@tests - @failures}; Failed: #{@failures}")
|
|
end
|
|
end
|
|
end
|
|
|
|
end
|