garfield/wpscan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix bug in output

Browse Source
This commit is contained in:
ethicalhack3r
2014-09-21 21:05:49 +02:00
parent 604299a1ac
commit d7488bd402
6 changed files with 40 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/common/models/vulnerability.rb
View File

@@ -43,8 +43,11 @@ class Vulnerability
def self.load_from_json_item(json_item) def self.load_from_json_item(json_item)
references = {} references = {}
[:id, :url, :cve, :secunia, :osvdb, :metasploit, :exploitdb].each do |key| %w(id url cve secunia osvdb metasploit exploitdb).each do |key|
references[key] = json_item[key.to_s].to_s.split(',') if json_item[key.to_s] if json_item[key]
json_item[key] = [json_item[key]] if json_item[key].class != Array
references[key] = json_item[key]
end
end end
new( new(

14
spec/lib/common/models/vulnerability_spec.rb
View File

@@ -37,13 +37,13 @@ describe Vulnerability do
} }
expected_refs = { expected_refs = {
:id => ['3911'], 'id' => ['3911'],
:url => ['Ref 1', 'Ref 2'], 'url' => ['Ref 1,Ref 2'],
:cve => ['2011-001'], 'cve' => ['2011-001'],
:secunia => ['secunia'], 'secunia' => ['secunia'],
:osvdb => ['osvdb'], 'osvdb' => ['osvdb'],
:metasploit => ['exploit/ex1'], 'metasploit' => ['exploit/ex1'],
:exploitdb => ['exploitdb'] 'exploitdb' => ['exploitdb']
} }
its(:title) { should == 'Vuln Title' } its(:title) { should == 'Vuln Title' }

14
spec/lib/common/models/wp_item_spec.rb
View File

@@ -14,13 +14,13 @@ describe WpItem do
let(:vulns_file) { MODELS_FIXTURES + '/wp_item/vulnerable/items_vulns.json' } let(:vulns_file) { MODELS_FIXTURES + '/wp_item/vulnerable/items_vulns.json' }
let(:identifier) { 'neo' } let(:identifier) { 'neo' }
let(:expected_refs) { { let(:expected_refs) { {
:id => ['2993'], 'id' => [2993],
:url => ['Ref 1', 'Ref 2'], 'url' => ['Ref 1,Ref 2'],
:cve => ['2011-001'], 'cve' => ['2011-001'],
:secunia => ['secunia'], 'secunia' => ['secunia'],
:osvdb => ['osvdb'], 'osvdb' => ['osvdb'],
:metasploit => ['exploit/ex1'], 'metasploit' => ['exploit/ex1'],
:exploitdb => ['exploitdb'] 'exploitdb' => ['exploitdb']
} } } }
let(:expected_vulns) { Vulnerabilities.new(1, Vulnerability.new("I'm the one", 'XSS', expected_refs)) } let(:expected_vulns) { Vulnerabilities.new(1, Vulnerability.new("I'm the one", 'XSS', expected_refs)) }
end end

14
spec/lib/common/models/wp_plugin_spec.rb
View File

@@ -8,13 +8,13 @@ describe WpPlugin do
let(:options) { { name: 'white-rabbit' } } let(:options) { { name: 'white-rabbit' } }
let(:vulns_file) { MODELS_FIXTURES + '/wp_plugin/vulnerable/plugins_vulns.json' } let(:vulns_file) { MODELS_FIXTURES + '/wp_plugin/vulnerable/plugins_vulns.json' }
let(:expected_refs) { { let(:expected_refs) { {
:id => ['2993'], 'id' => [2993],
:url => ['Ref 1', 'Ref 2'], 'url' => ['Ref 1,Ref 2'],
:cve => ['2011-001'], 'cve' => ['2011-001'],
:secunia => ['secunia'], 'secunia' => ['secunia'],
:osvdb => ['osvdb'], 'osvdb' => ['osvdb'],
:metasploit => ['exploit/ex1'], 'metasploit' => ['exploit/ex1'],
:exploitdb => ['exploitdb'] 'exploitdb' => ['exploitdb']
} } } }
let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('Follow me!', 'REDIRECT', expected_refs) } let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('Follow me!', 'REDIRECT', expected_refs) }
end end

14
spec/lib/common/models/wp_theme_spec.rb
View File

@@ -9,13 +9,13 @@ describe WpTheme do
let(:options) { { name: 'the-oracle' } } let(:options) { { name: 'the-oracle' } }
let(:vulns_file) { MODELS_FIXTURES + '/wp_theme/vulnerable/themes_vulns.json' } let(:vulns_file) { MODELS_FIXTURES + '/wp_theme/vulnerable/themes_vulns.json' }
let(:expected_refs) { { let(:expected_refs) { {
:id => ['2993'], 'id' => [2993],
:url => ['Ref 1', 'Ref 2'], 'url' => ['Ref 1,Ref 2'],
:cve => ['2011-001'], 'cve' => ['2011-001'],
:secunia => ['secunia'], 'secunia' => ['secunia'],
:osvdb => ['osvdb'], 'osvdb' => ['osvdb'],
:metasploit => ['exploit/ex1'], 'metasploit' => ['exploit/ex1'],
:exploitdb => ['exploitdb'] 'exploitdb' => ['exploitdb']
} } } }
let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('I see you', 'FPD', expected_refs) } let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('I see you', 'FPD', expected_refs) }
end end

14
spec/lib/common/models/wp_version_spec.rb
View File

@@ -8,13 +8,13 @@ describe WpVersion do
let(:options) { { number: '3.2' } } let(:options) { { number: '3.2' } }
let(:vulns_file) { MODELS_FIXTURES + '/wp_version/vulnerable/versions_vulns.json' } let(:vulns_file) { MODELS_FIXTURES + '/wp_version/vulnerable/versions_vulns.json' }
let(:expected_refs) { { let(:expected_refs) { {
:id => ['2993'], 'id' => [2993],
:url => ['Ref 1', 'Ref 2'], 'url' => ['Ref 1,Ref 2'],
:cve => ['2011-001'], 'cve' => ['2011-001'],
:secunia => ['secunia'], 'secunia' => ['secunia'],
:osvdb => ['osvdb'], 'osvdb' => ['osvdb'],
:metasploit => ['exploit/ex1'], 'metasploit' => ['exploit/ex1'],
:exploitdb => ['exploitdb'] 'exploitdb' => ['exploitdb']
} } } }
let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('Here I Am', 'SQLI', expected_refs) } let(:expected_vulns) { Vulnerabilities.new << Vulnerability.new('Here I Am', 'SQLI', expected_refs) }
end end