From 884a19b13d723285b42d918fd68b3c303bacbbd8 Mon Sep 17 00:00:00 2001
From: ethicalhack3r <ryandewhurst@gmail.com>
Date: Thu, 18 Sep 2014 10:24:55 +0200
Subject: [PATCH] Link to new vdb in references output

---
 lib/common/models/vulnerability.rb           | 4 ++--
 lib/common/models/vulnerability/urls.rb      | 5 ++++-
 spec/lib/common/models/vulnerability_spec.rb | 1 +
 spec/lib/common/models/wp_item_spec.rb       | 1 +
 spec/lib/common/models/wp_plugin_spec.rb     | 1 +
 spec/lib/common/models/wp_theme_spec.rb      | 1 +
 spec/lib/common/models/wp_version_spec.rb    | 1 +
 7 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/lib/common/models/vulnerability.rb b/lib/common/models/vulnerability.rb
index f7c4a3ce..0136692e 100755
--- a/lib/common/models/vulnerability.rb
+++ b/lib/common/models/vulnerability.rb
@@ -43,8 +43,8 @@ class Vulnerability
   def self.load_from_json_item(json_item)
     references = {}
 
-    [:url, :cve, :secunia, :osvdb, :metasploit, :exploitdb].each do |key|
-      references[key] = json_item[key.to_s].split(',') if json_item[key.to_s]
+    [: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]
     end
 
     new(
diff --git a/lib/common/models/vulnerability/urls.rb b/lib/common/models/vulnerability/urls.rb
index 112fb9e8..9b5b8ed6 100644
--- a/lib/common/models/vulnerability/urls.rb
+++ b/lib/common/models/vulnerability/urls.rb
@@ -22,12 +22,15 @@ class Vulnerability
     end
 
     def url_secunia(id)
-      "http://secunia.com/advisories/#{id}"
+      "https://secunia.com/advisories/#{id}"
     end
 
     def url_exploitdb(id)
       "http://www.exploit-db.com/exploits/#{id}/"
     end
 
+    def url_id(id)
+      "https://CHANGE_ME_BEFORE_MERGE/vulnerabilities/#{id}"
+    end
   end
 end
diff --git a/spec/lib/common/models/vulnerability_spec.rb b/spec/lib/common/models/vulnerability_spec.rb
index 8a2776d7..e0e05c57 100644
--- a/spec/lib/common/models/vulnerability_spec.rb
+++ b/spec/lib/common/models/vulnerability_spec.rb
@@ -37,6 +37,7 @@ describe Vulnerability do
     }
 
     expected_refs = {
+      :id         => ['3911'],
       :url        => ['Ref 1', 'Ref 2'],
       :cve        => ['2011-001'],
       :secunia    => ['secunia'],
diff --git a/spec/lib/common/models/wp_item_spec.rb b/spec/lib/common/models/wp_item_spec.rb
index 9b4342c7..f99d28ff 100644
--- a/spec/lib/common/models/wp_item_spec.rb
+++ b/spec/lib/common/models/wp_item_spec.rb
@@ -14,6 +14,7 @@ describe WpItem do
     let(:vulns_file)     { MODELS_FIXTURES + '/wp_item/vulnerable/items_vulns.json' }
     let(:identifier)    { 'neo' }
     let(:expected_refs)  { {
+        :id  => ['2993'],
         :url => ['Ref 1', 'Ref 2'],
         :cve => ['2011-001'],
         :secunia => ['secunia'],
diff --git a/spec/lib/common/models/wp_plugin_spec.rb b/spec/lib/common/models/wp_plugin_spec.rb
index 2e4837c9..4f7db2aa 100644
--- a/spec/lib/common/models/wp_plugin_spec.rb
+++ b/spec/lib/common/models/wp_plugin_spec.rb
@@ -8,6 +8,7 @@ describe WpPlugin do
     let(:options)        { { name: 'white-rabbit' } }
     let(:vulns_file)     { MODELS_FIXTURES + '/wp_plugin/vulnerable/plugins_vulns.json' }
     let(:expected_refs)  { {
+        :id  => ['2993'],
         :url => ['Ref 1', 'Ref 2'],
         :cve => ['2011-001'],
         :secunia => ['secunia'],
diff --git a/spec/lib/common/models/wp_theme_spec.rb b/spec/lib/common/models/wp_theme_spec.rb
index a88faaaf..f5b08668 100644
--- a/spec/lib/common/models/wp_theme_spec.rb
+++ b/spec/lib/common/models/wp_theme_spec.rb
@@ -9,6 +9,7 @@ describe WpTheme do
     let(:options)        { { name: 'the-oracle' } }
     let(:vulns_file)     { MODELS_FIXTURES + '/wp_theme/vulnerable/themes_vulns.json' }
     let(:expected_refs)  { {
+        :id => ['2993'],
         :url => ['Ref 1', 'Ref 2'],
         :cve => ['2011-001'],
         :secunia => ['secunia'],
diff --git a/spec/lib/common/models/wp_version_spec.rb b/spec/lib/common/models/wp_version_spec.rb
index cbd94410..6780285a 100644
--- a/spec/lib/common/models/wp_version_spec.rb
+++ b/spec/lib/common/models/wp_version_spec.rb
@@ -8,6 +8,7 @@ describe WpVersion do
     let(:options)        { { number: '3.2' } }
     let(:vulns_file)     { MODELS_FIXTURES + '/wp_version/vulnerable/versions_vulns.json' }
     let(:expected_refs)  { {
+        :id => ['2993'],
         :url => ['Ref 1', 'Ref 2'],
         :cve => ['2011-001'],
         :secunia => ['secunia'],