metasploit_data_models 3.0.10 → 4.0.0

data/db/migrate/20130412154159_change_foreign_key_in_module_actions.rb

@@ -1,6 +1,6 @@
 # Changes `module_actions.module_detail_id` to `module_actions.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleActions < ActiveRecord::Migration
+class ChangeForeignKeyInModuleActions < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412171844_change_foreign_key_in_module_archs.rb

@@ -1,6 +1,6 @@
 # Changes `module_archs.module_detail_id` to `module_archs.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleArchs < ActiveRecord::Migration
+class ChangeForeignKeyInModuleArchs < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412173121_change_foreign_key_in_module_authors.rb

@@ -1,6 +1,6 @@
 # Changes `module_authors.module_detail_id` to `module_authors.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleAuthors < ActiveRecord::Migration
+class ChangeForeignKeyInModuleAuthors < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412173640_change_foreign_key_in_module_mixins.rb

@@ -1,6 +1,6 @@
 # Changes `module_mixins.module_detail_id` to `module_mixins.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleMixins < ActiveRecord::Migration
+class ChangeForeignKeyInModuleMixins < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412174254_change_foreign_key_in_module_platforms.rb

@@ -1,6 +1,6 @@
 # Changes `module_platforms.module_detail_id` to `module_platforms.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModulePlatforms < ActiveRecord::Migration
+class ChangeForeignKeyInModulePlatforms < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412174719_change_foreign_key_in_module_refs.rb

@@ -1,6 +1,6 @@
 # Changes `module_refs.module_detail_id` to `module_refs.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleRefs < ActiveRecord::Migration
+class ChangeForeignKeyInModuleRefs < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130412175040_change_foreign_key_in_module_targets.rb

@@ -1,6 +1,6 @@
 # Changes `module_targets.module_detail_id` to `module_targets.detail_id` so that foreign key matches the conventional
 # name when `Mdm::ModuleDetail` became {Mdm::Module::Detail}.
-class ChangeForeignKeyInModuleTargets < ActiveRecord::Migration
+class ChangeForeignKeyInModuleTargets < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130423211152_add_creds_counter_cache.rb

@@ -1,4 +1,4 @@
-class AddCredsCounterCache < ActiveRecord::Migration
+class AddCredsCounterCache < ActiveRecord::Migration[4.2]
   def up
     add_column :hosts, :cred_count, :integer, :default => 0
     Mdm::Host.reset_column_information

data/db/migrate/20130430162145_enforce_address_uniqueness_in_workspace_in_hosts.rb

@@ -1,6 +1,6 @@
 # Changes index on address so it scoped to workspace_id and is unique to match the validation in {Mdm::Host} on
 # {Mdm::Host#address}.
-class EnforceAddressUniquenessInWorkspaceInHosts < ActiveRecord::Migration
+class EnforceAddressUniquenessInWorkspaceInHosts < ActiveRecord::Migration[4.2]
   TABLE_NAME = :hosts
 
   # maps Table -> Association Column for models that "belong to" a Host
@@ -33,7 +33,7 @@ class EnforceAddressUniquenessInWorkspaceInHosts < ActiveRecord::Migration
   #  idea what version of the code the user has checked out. So we fall back to SQL :(
   def find_and_merge_duplicate_hosts!
     # find all duplicate addresses within the same workspace currently in the db
-    dupe_addresses_and_workspaces = ActiveRecord::Base.connection.execute(%Q{
+    dupe_addresses_and_workspaces = ApplicationRecord.connection.execute(%Q{
       SELECT workspace_id, address, count_addr
         FROM (
           SELECT workspace_id, address, COUNT(address) AS count_addr
@@ -49,10 +49,10 @@ class EnforceAddressUniquenessInWorkspaceInHosts < ActiveRecord::Migration
       # iterate through the duped IPs
       dupe_addresses_and_workspaces.each do |result|
         # so its come to this
-        address      = ActiveRecord::Base.connection.quote(result['address'])
+        address      = ApplicationRecord.connection.quote(result['address'])
         workspace_id = result['workspace_id'].to_i
         # look up the duplicate Host table entries to find all IDs of the duped Hosts
-        hosts = ActiveRecord::Base.connection.execute(%Q|
+        hosts = ApplicationRecord.connection.execute(%Q|
           SELECT id
             FROM hosts
             WHERE address=#{address} AND workspace_id=#{workspace_id}
@@ -65,13 +65,13 @@ class EnforceAddressUniquenessInWorkspaceInHosts < ActiveRecord::Migration
         dupe_host_ids = hosts[1..-1]
         # update associations to these duplicate Hosts
         HOST_ASSOCIATION_MAP.each do |table, column|
-          ActiveRecord::Base.connection.execute(%Q|
+          ApplicationRecord.connection.execute(%Q|
             UPDATE #{table} SET #{column}=#{first_host_id}
               WHERE #{column} IN (#{dupe_host_ids.join(',')})
           |)
         end
         # destroy the duplicate host rows
-        ActiveRecord::Base.connection.execute(%Q|
+        ApplicationRecord.connection.execute(%Q|
           DELETE FROM hosts WHERE id IN (#{dupe_host_ids.join(',')})
         |)
       end

data/db/migrate/20130510021637_remove_campaigns.rb

@@ -1,4 +1,4 @@
-class RemoveCampaigns < ActiveRecord::Migration
+class RemoveCampaigns < ActiveRecord::Migration[4.2]
   def up
     drop_table :attachments
     drop_table :attachments_email_templates

data/db/migrate/20130515164311_change_web_vulns_confidence_to_integer.rb

@@ -1,5 +1,5 @@
 # Changes web_vulns.confidence from text to integer as it is populated with integers.
-class ChangeWebVulnsConfidenceToInteger < ActiveRecord::Migration
+class ChangeWebVulnsConfidenceToInteger < ActiveRecord::Migration[4.2]
   #
   # CONSTANTS
   #

data/db/migrate/20130515172727_valid_mdm_web_vuln_params.rb

@@ -1,4 +1,4 @@
-class ValidMdmWebVulnParams < ActiveRecord::Migration
+class ValidMdmWebVulnParams < ActiveRecord::Migration[4.2]
   # Don't put back the bad format because there's not way to figure our which of the [] were '' before {#up} and
   # which were `[]` before {#up}.
   #

data/db/migrate/20130516204810_making_vulns_refs_a_real_ar_model.rb

@@ -1,5 +1,5 @@
-class MakingVulnsRefsARealArModel < ActiveRecord::Migration
+class MakingVulnsRefsARealArModel < ActiveRecord::Migration[4.2]
   def change
     add_column :vulns_refs, :id, :primary_key
   end
-end
+end

data/db/migrate/20130522001343_create_task_creds.rb

@@ -1,4 +1,4 @@
-class CreateTaskCreds < ActiveRecord::Migration
+class CreateTaskCreds < ActiveRecord::Migration[4.2]
   def change
     create_table :task_creds do |t|
       t.references :task, :null => false

data/db/migrate/20130522032517_create_task_hosts.rb

@@ -1,4 +1,4 @@
-class CreateTaskHosts < ActiveRecord::Migration
+class CreateTaskHosts < ActiveRecord::Migration[4.2]
   def change
     create_table :task_hosts do |t|
       t.references :task, :null => false

data/db/migrate/20130522041110_create_task_services.rb

@@ -1,4 +1,4 @@
-class CreateTaskServices < ActiveRecord::Migration
+class CreateTaskServices < ActiveRecord::Migration[4.2]
   def change
     create_table :task_services do |t|
       t.references :task, :null => false

data/db/migrate/20130525015035_remove_campaign_id_from_clients.rb

@@ -1,4 +1,4 @@
-class RemoveCampaignIdFromClients < ActiveRecord::Migration
+class RemoveCampaignIdFromClients < ActiveRecord::Migration[4.2]
   def up
     remove_column :clients, :campaign_id
   end

data/db/migrate/20130525212420_drop_table_imported_creds.rb

@@ -1,4 +1,4 @@
-class DropTableImportedCreds < ActiveRecord::Migration
+class DropTableImportedCreds < ActiveRecord::Migration[4.2]
   def up
     drop_table :imported_creds
   end

data/db/migrate/20130531144949_making_host_tags_a_real_ar_model.rb

@@ -1,4 +1,4 @@
-class MakingHostTagsARealArModel < ActiveRecord::Migration
+class MakingHostTagsARealArModel < ActiveRecord::Migration[4.2]
   def change
     add_column :hosts_tags, :id, :primary_key
   end

data/db/migrate/20130604145732_create_task_sessions.rb

@@ -1,4 +1,4 @@
-class CreateTaskSessions < ActiveRecord::Migration
+class CreateTaskSessions < ActiveRecord::Migration[4.2]
   def change
     create_table :task_sessions do |t|
       t.references :task, :null => false

data/db/migrate/20130717150737_remove_pname_validation.rb

@@ -1,4 +1,4 @@
-class RemovePnameValidation < ActiveRecord::Migration
+class RemovePnameValidation < ActiveRecord::Migration[4.2]
 
   def change
 		change_column :web_vulns, :pname, :text, :null => true

data/db/migrate/20131002004641_create_automatic_exploitation_matches.rb

@@ -1,4 +1,4 @@
-class CreateAutomaticExploitationMatches < ActiveRecord::Migration
+class CreateAutomaticExploitationMatches < ActiveRecord::Migration[4.2]
   def change
     create_table :automatic_exploitation_matches do |t|
       t.integer :ref_id

data/db/migrate/20131002164449_create_automatic_exploitation_match_sets.rb

@@ -1,4 +1,4 @@
-class CreateAutomaticExploitationMatchSets < ActiveRecord::Migration
+class CreateAutomaticExploitationMatchSets < ActiveRecord::Migration[4.2]
   def change
     create_table :automatic_exploitation_match_sets do |t|
       t.integer :workspace_id

data/db/migrate/20131008213344_create_automatic_exploitation_runs.rb

@@ -1,4 +1,4 @@
-class CreateAutomaticExploitationRuns < ActiveRecord::Migration
+class CreateAutomaticExploitationRuns < ActiveRecord::Migration[4.2]
   def change
     create_table :automatic_exploitation_runs do |t|
       t.integer :workspace_id

data/db/migrate/20131011184338_module_detail_on_automatic_exploitation_match.rb

@@ -1,4 +1,4 @@
-class ModuleDetailOnAutomaticExploitationMatch < ActiveRecord::Migration
+class ModuleDetailOnAutomaticExploitationMatch < ActiveRecord::Migration[4.2]
   def up
     rename_column :automatic_exploitation_matches, :ref_id, :module_detail_id
     add_column :automatic_exploitation_matches, :match_set_id, :integer

data/db/migrate/20131017150735_create_automatic_exploitation_match_results.rb

@@ -1,4 +1,4 @@
-class CreateAutomaticExploitationMatchResults < ActiveRecord::Migration
+class CreateAutomaticExploitationMatchResults < ActiveRecord::Migration[4.2]
   def change
     create_table :automatic_exploitation_match_results do |t|
       t.integer :match_id

data/db/migrate/20131021185657_make_match_polymorphic.rb

@@ -1,4 +1,4 @@
-class MakeMatchPolymorphic < ActiveRecord::Migration
+class MakeMatchPolymorphic < ActiveRecord::Migration[4.2]
   def up
     add_column :automatic_exploitation_matches, :matchable_type, :string
     add_column :automatic_exploitation_matches, :matchable_id, :integer

data/db/migrate/20140905031549_add_detected_arch_to_host.rb

@@ -1,4 +1,4 @@
-class AddDetectedArchToHost < ActiveRecord::Migration
+class AddDetectedArchToHost < ActiveRecord::Migration[4.2]
   def change
     add_column :hosts, :detected_arch, :string, { :null => true }
   end

data/db/migrate/20150112203945_remove_duplicate_services.rb

@@ -1,4 +1,4 @@
-class RemoveDuplicateServices < ActiveRecord::Migration
+class RemoveDuplicateServices < ActiveRecord::Migration[4.2]
   def change
     select_mgr = Mdm::Service.arel_table.project(
       Mdm::Service[:host_id],

data/db/migrate/20150205192745_drop_service_uniqueness_index.rb

@@ -1,4 +1,4 @@
-class DropServiceUniquenessIndex < ActiveRecord::Migration
+class DropServiceUniquenessIndex < ActiveRecord::Migration[4.2]
   def change
     remove_index(:services, :host_id_and_port_and_proto)
   end

data/db/migrate/20150209195939_add_vuln_id_to_note.rb

@@ -1,4 +1,4 @@
-class AddVulnIdToNote < ActiveRecord::Migration
+class AddVulnIdToNote < ActiveRecord::Migration[4.2]
   def change
     add_column :notes, :vuln_id, :integer
     add_index :notes, :vuln_id

data/db/migrate/20150212214222_remove_duplicate_services2.rb

@@ -1,4 +1,4 @@
-class RemoveDuplicateServices2 < ActiveRecord::Migration
+class RemoveDuplicateServices2 < ActiveRecord::Migration[4.2]
   def change
     select_mgr = Mdm::Service.arel_table.project(
       Mdm::Service[:host_id],

data/db/migrate/20150219173821_create_module_runs.rb

@@ -1,4 +1,4 @@
-class CreateModuleRuns < ActiveRecord::Migration
+class CreateModuleRuns < ActiveRecord::Migration[4.2]
   def change
     create_table :module_runs do |t|
       t.datetime :attempted_at

data/db/migrate/20150219215039_add_module_run_to_session.rb

@@ -1,4 +1,4 @@
-class AddModuleRunToSession < ActiveRecord::Migration
+class AddModuleRunToSession < ActiveRecord::Migration[4.2]
   def change
     change_table :sessions do |t|
       t.integer :module_run_id

data/db/migrate/20150226151459_add_module_run_fk_to_loot.rb

@@ -1,4 +1,4 @@
-class AddModuleRunFkToLoot < ActiveRecord::Migration
+class AddModuleRunFkToLoot < ActiveRecord::Migration[4.2]
   def change
     change_table(:loots) do |t|
       t.integer :module_run_id

data/db/migrate/20150312155312_add_module_full_name_to_match.rb

@@ -1,4 +1,4 @@
-class AddModuleFullNameToMatch < ActiveRecord::Migration
+class AddModuleFullNameToMatch < ActiveRecord::Migration[4.2]
   def change
     add_column :automatic_exploitation_matches, :module_fullname, :text
     add_index :automatic_exploitation_matches, :module_fullname

data/db/migrate/20150317145455_rename_module_indices.rb

@@ -1,8 +1,8 @@
-class RenameModuleIndices < ActiveRecord::Migration
+class RenameModuleIndices < ActiveRecord::Migration[4.2]
   def change
     # This migration will only need to be run if the indices were originally
     # generated before Rails 4. Fresh builds can safely skip this.
-    return if ActiveRecord::Base.connection.indexes(:module_actions).first.name != 'index_module_actions_on_module_detail_id'
+    return if ApplicationRecord.connection.indexes(:module_actions).first.name != 'index_module_actions_on_module_detail_id'
 
     rename_index :module_actions,
                  'index_module_actions_on_module_detail_id',

data/db/migrate/20150326183742_add_missing_ae_indices.rb

@@ -1,4 +1,4 @@
-class AddMissingAeIndices < ActiveRecord::Migration
+class AddMissingAeIndices < ActiveRecord::Migration[4.2]
   def up
     add_index :automatic_exploitation_match_results, :match_id
     add_index :automatic_exploitation_match_results, :run_id

data/db/migrate/20150421211719_rename_automatic_exploitation_index.rb

@@ -1,13 +1,13 @@
-class RenameAutomaticExploitationIndex < ActiveRecord::Migration
+class RenameAutomaticExploitationIndex < ActiveRecord::Migration[4.2]
   def up
-    if index_name_exists? :automatic_exploitation_matches, :index_automatic_exploitation_matches_on_ref_id, false
+    if index_name_exists? :automatic_exploitation_matches, :index_automatic_exploitation_matches_on_ref_id
       rename_index :automatic_exploitation_matches,
       :index_automatic_exploitation_matches_on_ref_id,
       :index_automatic_exploitation_matches_on_module_detail_id
     end
   end
   def down
-    if index_name_exists? :automatic_exploitation_matches, :index_automatic_exploitation_matches_on_module_detail_id, false
+    if index_name_exists? :automatic_exploitation_matches, :index_automatic_exploitation_matches_on_module_detail_id
       rename_index :automatic_exploitation_matches,
       :index_automatic_exploitation_matches_on_module_detail_id,
       :index_automatic_exploitation_matches_on_ref_id

data/db/migrate/20150514182921_add_origin_to_mdm_vuln.rb

@@ -1,4 +1,4 @@
-class AddOriginToMdmVuln < ActiveRecord::Migration
+class AddOriginToMdmVuln < ActiveRecord::Migration[4.2]
   def up
     add_column :vulns, :origin_id,   :integer
     add_column :vulns, :origin_type, :string

data/db/migrate/20160415153312_remove_not_null_from_web_vuln_p_arams.rb

@@ -1,4 +1,4 @@
-class RemoveNotNullFromWebVulnPArams < ActiveRecord::Migration
+class RemoveNotNullFromWebVulnPArams < ActiveRecord::Migration[4.2]
   def change
     change_column_null(:web_vulns, :params, true)
   end

data/db/migrate/20161004165612_add_fingerprinted_to_workspace.rb

@@ -1,4 +1,4 @@
-class AddFingerprintedToWorkspace < ActiveRecord::Migration
+class AddFingerprintedToWorkspace < ActiveRecord::Migration[4.2]
   def change
     add_column :workspaces, :import_fingerprint, :boolean, default: false
   end

data/db/migrate/20161227212223_add_os_family_to_hosts.rb

@@ -1,4 +1,4 @@
-class AddOsFamilyToHosts < ActiveRecord::Migration
+class AddOsFamilyToHosts < ActiveRecord::Migration[4.2]
   def change
     add_column :hosts, :os_family, :string
   end

data/db/migrate/20180904120211_create_payloads.rb

@@ -1,4 +1,4 @@
-class CreatePayloads < ActiveRecord::Migration
+class CreatePayloads < ActiveRecord::Migration[4.2]
   def change
     create_table :payloads do |t|
       t.string :name

data/db/migrate/20190308134512_create_async_callbacks.rb

@@ -1,4 +1,4 @@
-class CreateAsyncCallbacks < ActiveRecord::Migration
+class CreateAsyncCallbacks < ActiveRecord::Migration[4.2]
   def change
     create_table :async_callbacks do |t|
       t.string :uuid, :null => false

data/db/migrate/20190507120211_remove_payload_workspaces.rb

@@ -1,4 +1,4 @@
-class RemovePayloadWorkspaces < ActiveRecord::Migration
+class RemovePayloadWorkspaces < ActiveRecord::Migration[4.2]
   def change
     remove_column :payloads, :workspace_id, :references
   end

data/lib/metasploit_data_models.rb

@@ -18,7 +18,6 @@ require 'active_support/all'
 require 'metasploit/concern'
 require 'metasploit/model'
 require 'arel-helpers'
-require 'postgres_ext'
 
 #
 # Project

data/lib/metasploit_data_models/automatic_exploitation.rb

@@ -16,7 +16,7 @@ module MetasploitDataModels::AutomaticExploitation
   # Module Methods
   #
 
-  # The prefix of the `ActiveRecord::Base#table_name` of subclasses in this namespace.
+  # The prefix of the `ApplicationRecord#table_name` of subclasses in this namespace.
   #
   # @return [String]
   def self.table_name_prefix

data/lib/metasploit_data_models/base64_serializer.rb

@@ -1,5 +1,5 @@
 # Provides ActiveRecord 3.1x-friendly serialization for descendants of
-# ActiveRecord::Base. Backwards compatible with older YAML methods and
+# ApplicationRecord. Backwards compatible with older YAML methods and
 # will fall back to string decoding in the worst case
 #
 # @example Using default default of {}

data/lib/metasploit_data_models/change_required_columns_to_null_false.rb

@@ -2,7 +2,7 @@
 # `:null => true`, to `:null => false`.
 #
 #  @abstract Subclass and define COLUMNS as Array<Symbol> and TABLE_NAME as Symbol.
-class MetasploitDataModels::ChangeRequiredColumnsToNullFalse < ActiveRecord::Migration
+class MetasploitDataModels::ChangeRequiredColumnsToNullFalse < ActiveRecord::Migration[4.2]
   # Marks all the COLUMNS as `:null => true`
   def down
     # Use self.class:: so constants are resolved in subclasses instead of this class.
@@ -18,4 +18,4 @@ class MetasploitDataModels::ChangeRequiredColumnsToNullFalse < ActiveRecord::Mig
       change_column_null(self.class::TABLE_NAME, column, false)
     end
   end
-end
+end