metasploit-framework/spec/tools/msu_finder_spec.rb

279 lines
5.7 KiB
Ruby
Raw Normal View History

2015-09-29 19:50:56 +00:00
load Metasploit::Framework.root.join('tools/msu_finder.rb').to_path
2015-09-29 22:20:30 +00:00
require 'nokogiri'
2015-09-29 19:50:56 +00:00
describe MicrosoftPatch do
2015-09-29 22:20:30 +00:00
before(:each) do
cli = Rex::Proto::Http::Client.new('127.0.0.1')
allow(cli).to receive(:connect)
allow(cli).to receive(:request_cgi)
allow(cli).to receive(:send_recv).and_return(Rex::Proto::Http::Response.new)
allow(Rex::Proto::Http::Client).to receive(:new).and_return(cli)
end
2015-09-29 19:50:56 +00:00
describe MicrosoftPatch::SiteInfo do
context 'Constants' do
context 'TECHNET' do
let(:technet) do
MicrosoftPatch::SiteInfo::TECHNET
end
it 'returns 157.56.148.23 as the IP' do
expect(technet[:ip]).to eq('157.56.148.23')
end
it 'returns technet.microsoft.com as the vhost' do
expect(technet[:vhost]).to eq('technet.microsoft.com')
end
end
context 'MICROSOFT' do
let(:microsoft) do
MicrosoftPatch::SiteInfo::MICROSOFT
end
it 'returns 104.72.230.162 as the IP' do
expect(microsoft[:ip]).to eq('104.72.230.162')
end
it 'returns www.microsoft.com as the vhost' do
expect(microsoft[:vhost]).to eq('www.microsoft.com')
end
end
context 'GOOGLEAPIS' do
let(:googleapis) do
MicrosoftPatch::SiteInfo::GOOGLEAPIS
end
it 'returns 74.125.28.95 as the IP' do
expect(googleapis[:ip]).to eq('74.125.28.95')
end
it 'returns www.googleapis.com as the vhost' do
expect(googleapis[:vhost]).to eq('www.googleapis.com')
end
end
end
end
describe MicrosoftPatch::Base do
2015-09-29 22:20:30 +00:00
def get_stdout(&block)
out = $stdout
$stdout = fake = StringIO.new
begin
yield
ensure
$stdout = out
end
fake.string
end
def get_stderr(&block)
out = $stderr
$stderr = fake = StringIO.new
begin
yield
ensure
$stderr = out
end
fake.string
end
subject do
MicrosoftPatch::Base.new
end
describe '#print_debug' do
it 'prints a [DEBUG] message' do
output = get_stderr { subject.print_debug }
expect(output).to include('[DEBUG]')
end
end
describe '#print_status' do
it 'prints a [*] message' do
output = get_stderr { subject.print_status }
expect(output).to include('[*]')
end
end
describe '#print_error' do
it 'prints an [ERROR] message' do
output = get_stderr { subject.print_error }
expect(output).to include('[ERROR]')
end
end
describe '#print_line' do
it 'prints a regular message' do
msg = 'TEST'
output = get_stdout { subject.print_line(msg) }
expect(output).to eq("#{msg}\n")
end
end
describe '#send_http_request' do
it 'returns a Rex::Proto::Http::Response object' do
allow(subject).to receive(:print_debug)
res = subject.send_http_request(MicrosoftPatch::SiteInfo::TECHNET)
expect(res).to be_kind_of(Rex::Proto::Http::Response)
end
end
2015-09-29 19:50:56 +00:00
end
describe MicrosoftPatch::PatchLinkCollector do
2015-09-29 22:20:30 +00:00
subject do
MicrosoftPatch::PatchLinkCollector.new
end
before(:each) do
allow(subject).to receive(:print_debug)
end
describe '#download_advisory' do
it 'returns a Rex::Proto::Http::Response object' do
res = subject.download_advisory('ms15-100')
expect(res).to be_kind_of(Rex::Proto::Http::Response)
end
end
describe '#get_appropriate_pattern' do
let(:ms15_100_html) do
%Q|
<html>
<div id="mainBody">
<div>
<h2>
<div>
<span>Affected Software</span>
<div class="sectionblock">
<table>
<tr><td><a href="http://technet.microsoft.com">fake link</a></td></tr>
</table>
</div>
</div>
</h2>
</div>
</div>
</html>
|
end
let(:ms07_029_html) do
%Q|
<html>
</html>
|
end
let(:ms03_039_html) do
%Q|
<html>
</html>
|
end
let(:ms07_030_html) do
%Q|
<html>
</html>
|
end
it 'returns a pattern for ms15-100' do
expected_pattern = '//div[@id="mainBody"]//div//div[@class="sectionblock"]//table//a'
p = subject.get_appropriate_pattern(::Nokogiri::HTML(ms15_100_html))
expect(p).to eq(expected_pattern)
end
it 'returns a pattern for ms07-029' do
end
it 'returns a pattern for ms03-039' do
end
it 'returns a pattern for ms07-030' do
end
end
describe '#get_details_aspx' do
end
describe '#follow_redirect' do
end
describe '#get_download_page' do
end
describe '#get_download_links' do
end
describe '#has_advisory?' do
end
describe '#is_valid_msb?' do
end
2015-09-29 19:50:56 +00:00
end
describe MicrosoftPatch::TechnetMsbSearch do
2015-09-29 22:20:30 +00:00
describe '#find_msb_numbers' do
end
describe '#search' do
end
describe '#search_by_product_ids' do
end
describe '#search_by_keyword' do
end
describe '#get_product_dropdown_list' do
end
2015-09-29 19:50:56 +00:00
end
describe MicrosoftPatch::GoogleMsbSearch do
2015-09-29 22:20:30 +00:00
describe '#find_msb_numbers' do
end
describe '#search' do
end
describe '#parse_results' do
end
describe '#get_total_results' do
end
describe '#get_next_index' do
end
2015-09-29 19:50:56 +00:00
end
describe MicrosoftPatch::Module do
2015-09-29 22:20:30 +00:00
describe '#get_download_links' do
end
describe '#google_search' do
end
describe '#technet_search' do
end
describe '#run' do
end
2015-09-29 19:50:56 +00:00
end
2015-09-29 22:20:30 +00:00
end