metasploit_data_models 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 25eceec43c9e5fb37c7b184bee21d950d1ead779
-  data.tar.gz: 4504f3ddd2f4379685a963e70911844364c548bf
+  metadata.gz: b4abe639facb4bde408a5a4df77e10b684868bee
+  data.tar.gz: 9cee9a8a732c5a49a5725c117c074f51df88138a
 SHA512:
-  metadata.gz: 01e136d527c911b4feba3ee3a0050455eafed2953295407760f6410a5cd4a266c4b4f10a6099b6388c6a5d2161e36496ef7193f13179071d86abc31bc23eb582
-  data.tar.gz: a7e9e86609d95a1b95b2c14ada58a86e169ac7b06eb05dec426a949e7f0490af2b8335ddbb32537a1eb5236767403a69e5440e2610504e144d3b2d3776ab2e5d
+  metadata.gz: 7456df1528a09430fcc9ed93e240ba97ab33b653d7422942a24b1b86315a2439a93854c0815aa1fcad18690bd574b513997ee37c8bbe553267073ec4e89ae778
+  data.tar.gz: c669eea4b2105a2367b6c121fb4727d0d8eea0eb4ae50ac34aa45eca37e94b3d27e3a1529cfad54c8324523c571a7c29ecdc83edae2a9779320f209554301e6c

data/app/models/mdm/vuln.rb

@@ -30,6 +30,14 @@ class Mdm::Vuln < ActiveRecord::Base
              class_name: 'Mdm::Service',
              inverse_of: :vulns
 
+  # @!attribute [rw] origin
+  #   A polymorphic association to the origin that found
+  #   the vulnerability.
+  #
+  #   @return [ActiveRecord::Relation<origin>]
+  belongs_to :origin,
+             polymorphic: true
+
   # @!attribute [rw] vuln_attempts
   #   Attempts to exploit this vulnerability.
   #

data/db/migrate/20150514182921_add_origin_to_mdm_vuln.rb

@@ -0,0 +1,13 @@
+class AddOriginToMdmVuln < ActiveRecord::Migration
+  def up
+    add_column :vulns, :origin_id,   :integer
+    add_column :vulns, :origin_type, :string
+
+    add_index :vulns, :origin_id
+  end
+
+  def down
+    remove_column :vulns, :origin_id
+    remove_column :vulns, :origin_type
+  end
+end

data/lib/metasploit_data_models/version.rb

@@ -8,9 +8,11 @@ module MetasploitDataModels
     # The major version number.
     MAJOR = 1
     # The minor version number, scoped to the {MAJOR} version number.
-    MINOR = 0
+    MINOR = 1
     # The patch version number, scoped to the {MAJOR} and {MINOR} version numbers.
-    PATCH = 1
+    PATCH = 0
+
+    # The prerelease version, scoped to the {MAJOR}, {MINOR}, and {PATCH} version numbers.
 
     #
     # Module Methods

data/spec/app/models/mdm/vuln_spec.rb

@@ -32,6 +32,7 @@ RSpec.describe Mdm::Vuln, type: :model do
 
   context 'associations' do
     it { is_expected.to belong_to(:host).class_name('Mdm::Host') }
+    it { is_expected.to belong_to(:origin) }
     it { is_expected.to belong_to(:service).class_name('Mdm::Service') }
     it { is_expected.to have_many(:module_refs).class_name('Mdm::Module::Ref').through(:refs) }
     it { is_expected.to have_many(:module_runs).class_name('MetasploitDataModels::ModuleRun') }
@@ -128,6 +129,8 @@ RSpec.describe Mdm::Vuln, type: :model do
       it { is_expected.to have_db_column(:host_id).of_type(:integer) }
       it { is_expected.to have_db_column(:info).of_type(:string) }
       it { is_expected.to have_db_column(:name).of_type(:string) }
+      it { is_expected.to have_db_column(:origin_id).of_type(:integer) }
+      it { is_expected.to have_db_column(:origin_type).of_type(:string) }
       it { is_expected.to have_db_column(:service_id).of_type(:integer) }
 
       context 'counter caches' do

data/spec/dummy/db/structure.sql

@@ -1659,7 +1659,9 @@ CREATE TABLE vulns (
     info character varying(65536),
     exploited_at timestamp without time zone,
     vuln_detail_count integer DEFAULT 0,
-    vuln_attempt_count integer DEFAULT 0
+    vuln_attempt_count integer DEFAULT 0,
+    origin_id integer,
+    origin_type character varying(255)
 );
 
 
@@ -3081,6 +3083,13 @@ CREATE INDEX index_sessions_on_module_run_id ON sessions USING btree (module_run
 CREATE INDEX index_vulns_on_name ON vulns USING btree (name);
 
 
+--
+-- Name: index_vulns_on_origin_id; Type: INDEX; Schema: public; Owner: -; Tablespace: 
+--
+
+CREATE INDEX index_vulns_on_origin_id ON vulns USING btree (origin_id);
+
+
 --
 -- Name: index_web_forms_on_path; Type: INDEX; Schema: public; Owner: -; Tablespace: 
 --
@@ -3381,6 +3390,8 @@ INSERT INTO schema_migrations (version) VALUES ('20150326183742');
 
 INSERT INTO schema_migrations (version) VALUES ('20150421211719');
 
+INSERT INTO schema_migrations (version) VALUES ('20150514182921');
+
 INSERT INTO schema_migrations (version) VALUES ('21');
 
 INSERT INTO schema_migrations (version) VALUES ('22');

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: metasploit_data_models
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Samuel Huckins
@@ -11,7 +11,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-05-12 00:00:00.000000000 Z
+date: 2015-05-21 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: metasploit-version
@@ -501,6 +501,7 @@ files:
 - db/migrate/20150317145455_rename_module_indices.rb
 - db/migrate/20150326183742_add_missing_ae_indices.rb
 - db/migrate/20150421211719_rename_automatic_exploitation_index.rb
+- db/migrate/20150514182921_add_origin_to_mdm_vuln.rb
 - lib/mdm.rb
 - lib/mdm/host/operating_system_normalization.rb
 - lib/mdm/module.rb