wg-metasploit_data_models 4.1.4.01 → 4.1.4.02

data/db/migrate/20130515164311_change_web_vulns_confidence_to_integer.rb

@@ -0,0 +1,48 @@
+# Changes web_vulns.confidence from text to integer as it is populated with integers.
+class ChangeWebVulnsConfidenceToInteger < ActiveRecord::Migration[4.2]
+  #
+  # CONSTANTS
+  #
+
+  # Columns in {TABLE} whose type needs to be change.
+  COLUMN = :confidence
+  # The correct type for {COLUMN}.
+  NEW_TYPE = :integer
+  # The incorrect type for {COLUMN}.
+  OLD_TYPE = :text
+  # The table in which {COLUMN} is defined.
+  TABLE = :web_vulns
+
+  #
+  # Methods
+  #
+
+  # Changes web_vulns.confidence back to text
+  #
+  # @return [void]
+  def down
+    alter_type(:to => OLD_TYPE)
+  end
+
+  # Changes web_vulns.confidence to integer
+  #
+  # @return [void]
+  def up
+    alter_type(:to => NEW_TYPE)
+  end
+
+  private
+
+  # Alters {COLUMN} type in {TABLE} from old to new type
+  #
+  # @param options [Hash{Symbol => #to_s}]
+  # @option options [#to_s] :from The old type name.
+  # @option options [#to_s] :to The new type name.
+  def alter_type(options={})
+    options.assert_valid_keys(:to)
+
+    new = options.fetch(:to)
+
+    execute "ALTER TABLE #{TABLE} ALTER COLUMN #{COLUMN} TYPE #{new} USING confidence::#{new}"
+  end
+end

data/db/migrate/20130515172727_valid_mdm_web_vuln_params.rb

@@ -0,0 +1,30 @@
+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}.
+  #
+  # @return [void]
+  def down
+  end
+
+  # Changes any Mdm::WebVuln#params with value `''` to value `[]`.
+  #
+  # @return [void]
+  def up
+    # Can't search serialized columns, so have to load all the Mdm::WebVulns in memory
+    Mdm::WebVuln.find_each do |web_vuln|
+      if web_vuln.invalid?
+        # cast nil, '' and {} to correct [].
+        if web_vuln.params.blank?
+          web_vuln.params = []
+        end
+
+        # If its still invalid have to destroy the Mdm::WebVuln or a different export error could occur.
+        if web_vuln.invalid?
+          web_vuln.destroy
+        else
+          web_vuln.save!
+        end
+      end
+    end
+  end
+end

data/db/migrate/20130516204810_making_vulns_refs_a_real_ar_model.rb

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

data/db/migrate/20130522001343_create_task_creds.rb

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

data/db/migrate/20130522032517_create_task_hosts.rb

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

data/db/migrate/20130522041110_create_task_services.rb

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

data/db/migrate/20130525015035_remove_campaign_id_from_clients.rb

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

data/db/migrate/20130525212420_drop_table_imported_creds.rb

@@ -0,0 +1,14 @@
+class DropTableImportedCreds < ActiveRecord::Migration[4.2]
+  def up
+    drop_table :imported_creds
+  end
+
+  def down
+    create_table :imported_creds do |t|
+      t.integer   :workspace_id, :null => false, :default => 1
+      t.string    :user, :limit  => 512
+      t.string    :pass, :limit  => 512
+      t.string    :ptype, :limit  => 16, :default => "password"
+    end
+  end
+end

data/db/migrate/20130531144949_making_host_tags_a_real_ar_model.rb

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

data/db/migrate/20130604145732_create_task_sessions.rb

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

data/db/migrate/20130717150737_remove_pname_validation.rb

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

data/db/migrate/20131002004641_create_automatic_exploitation_matches.rb

@@ -0,0 +1,13 @@
+class CreateAutomaticExploitationMatches < ActiveRecord::Migration[4.2]
+  def change
+    create_table :automatic_exploitation_matches do |t|
+      t.integer :ref_id
+      t.string :state
+      t.integer :nexpose_data_vulnerability_definition_id
+
+      t.timestamps null: false
+    end
+
+    add_index :automatic_exploitation_matches, :ref_id
+  end
+end

data/db/migrate/20131002164449_create_automatic_exploitation_match_sets.rb

@@ -0,0 +1,12 @@
+class CreateAutomaticExploitationMatchSets < ActiveRecord::Migration[4.2]
+  def change
+    create_table :automatic_exploitation_match_sets do |t|
+      t.integer :workspace_id
+      t.integer :user_id
+
+      t.timestamps null: false
+    end
+    add_index :automatic_exploitation_match_sets, :user_id
+    add_index :automatic_exploitation_match_sets, :workspace_id
+  end
+end

data/db/migrate/20131008213344_create_automatic_exploitation_runs.rb

@@ -0,0 +1,11 @@
+class CreateAutomaticExploitationRuns < ActiveRecord::Migration[4.2]
+  def change
+    create_table :automatic_exploitation_runs do |t|
+      t.integer :workspace_id
+      t.integer :user_id
+      t.integer :match_set_id
+
+      t.timestamps null: false
+    end
+  end
+end

data/db/migrate/20131011184338_module_detail_on_automatic_exploitation_match.rb

@@ -0,0 +1,10 @@
+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
+  end
+
+  def down
+    rename_column :automatic_exploitation_matches, :module_detail_id, :ref_id
+  end
+end

data/db/migrate/20131017150735_create_automatic_exploitation_match_results.rb

@@ -0,0 +1,11 @@
+class CreateAutomaticExploitationMatchResults < ActiveRecord::Migration[4.2]
+  def change
+    create_table :automatic_exploitation_match_results do |t|
+      t.integer :match_id
+      t.integer :run_id
+      t.string :state, null: false
+
+      t.timestamps null: false
+    end
+  end
+end

data/db/migrate/20131021185657_make_match_polymorphic.rb

@@ -0,0 +1,11 @@
+class MakeMatchPolymorphic < ActiveRecord::Migration[4.2]
+  def up
+    add_column :automatic_exploitation_matches, :matchable_type, :string
+    add_column :automatic_exploitation_matches, :matchable_id, :integer
+  end
+
+  def down
+    remove_column :automatic_exploitation_matches, :matchable_type
+    remove_column :automatic_exploitation_matches, :matchable_id
+  end
+end

data/db/migrate/20140905031549_add_detected_arch_to_host.rb

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

data/db/migrate/20150112203945_remove_duplicate_services.rb

@@ -0,0 +1,17 @@
+class RemoveDuplicateServices < ActiveRecord::Migration[4.2]
+  def change
+    select_mgr = Mdm::Service.arel_table.project(
+      Mdm::Service[:host_id],
+      Mdm::Service[:proto],
+      Mdm::Service[:port].count
+    ).group(
+      'host_id',
+      'port',
+      'proto'
+    ).having(Mdm::Service[:port].count.gt(1))
+
+    Mdm::Service.find_by_sql(select_mgr).each(&:destroy)
+
+    add_index :services, [:host_id, :port, :proto], unique: true
+  end
+end

data/db/migrate/20150205192745_drop_service_uniqueness_index.rb

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

data/db/migrate/20150209195939_add_vuln_id_to_note.rb

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

data/db/migrate/20150212214222_remove_duplicate_services2.rb

@@ -0,0 +1,17 @@
+class RemoveDuplicateServices2 < ActiveRecord::Migration[4.2]
+  def change
+    select_mgr = Mdm::Service.arel_table.project(
+      Mdm::Service[:host_id],
+      Mdm::Service[:proto],
+      Mdm::Service[:port].count
+    ).group(
+      'host_id',
+      'port',
+      'proto'
+    ).having(Mdm::Service[:port].count.gt(1))
+
+    Mdm::Service.find_by_sql(select_mgr).each(&:destroy)
+
+    add_index :services, [:host_id, :port, :proto], unique: true
+  end
+end

data/db/migrate/20150219173821_create_module_runs.rb

@@ -0,0 +1,23 @@
+class CreateModuleRuns < ActiveRecord::Migration[4.2]
+  def change
+    create_table :module_runs do |t|
+      t.datetime :attempted_at
+      t.text :fail_detail
+      t.string :fail_reason
+      t.text :module_fullname
+      t.integer :port
+      t.string :proto
+      t.integer :session_id
+      t.string :status
+      t.integer :trackable_id
+      t.string :trackable_type
+      t.integer :user_id
+      t.string :username
+
+      t.timestamps null: false
+    end
+
+    add_index :module_runs, :session_id
+    add_index :module_runs, :user_id
+  end
+end

data/db/migrate/20150219215039_add_module_run_to_session.rb

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

data/db/migrate/20150226151459_add_module_run_fk_to_loot.rb

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

data/db/migrate/20150312155312_add_module_full_name_to_match.rb

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

data/db/migrate/20150317145455_rename_module_indices.rb

@@ -0,0 +1,29 @@
+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 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',
+                 'index_module_actions_on_detail_id'
+    rename_index :module_archs,
+                 'index_module_archs_on_module_detail_id',
+                 'index_module_archs_on_detail_id'
+    rename_index :module_authors,
+                 'index_module_authors_on_module_detail_id',
+                 'index_module_authors_on_detail_id'
+    rename_index :module_mixins,
+                 'index_module_mixins_on_module_detail_id',
+                 'index_module_mixins_on_detail_id'
+    rename_index :module_platforms,
+                 'index_module_platforms_on_module_detail_id',
+                 'index_module_platforms_on_detail_id'
+    rename_index :module_refs,
+                 'index_module_refs_on_module_detail_id',
+                 'index_module_refs_on_detail_id'
+    rename_index :module_targets,
+                 'index_module_targets_on_module_detail_id',
+                 'index_module_targets_on_detail_id'
+  end
+end

data/db/migrate/20150326183742_add_missing_ae_indices.rb

@@ -0,0 +1,13 @@
+class AddMissingAeIndices < ActiveRecord::Migration[4.2]
+  def up
+    add_index :automatic_exploitation_match_results, :match_id
+    add_index :automatic_exploitation_match_results, :run_id
+
+    add_index :automatic_exploitation_runs, :match_set_id
+    add_index :automatic_exploitation_runs, :user_id
+    add_index :automatic_exploitation_runs, :workspace_id
+  end
+
+  def down
+  end
+end

data/db/migrate/20150421211719_rename_automatic_exploitation_index.rb

@@ -0,0 +1,16 @@
+class RenameAutomaticExploitationIndex < ActiveRecord::Migration[4.2]
+  def up
+    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
+      rename_index :automatic_exploitation_matches,
+      :index_automatic_exploitation_matches_on_module_detail_id,
+      :index_automatic_exploitation_matches_on_ref_id
+    end
+  end
+end

data/db/migrate/20150514182921_add_origin_to_mdm_vuln.rb

@@ -0,0 +1,13 @@
+class AddOriginToMdmVuln < ActiveRecord::Migration[4.2]
+  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/db/migrate/20160415153312_remove_not_null_from_web_vuln_p_arams.rb

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

data/db/migrate/20161004165612_add_fingerprinted_to_workspace.rb

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

data/db/migrate/20161227212223_add_os_family_to_hosts.rb

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

data/db/migrate/20180904120211_create_payloads.rb

@@ -0,0 +1,21 @@
+class CreatePayloads < ActiveRecord::Migration[4.2]
+  def change
+    create_table :payloads do |t|
+      t.string :name
+      t.string :uuid
+      t.integer :uuid_mask
+      t.integer :timestamp
+      t.string :arch
+      t.string :platform
+      t.string :urls
+      t.string :description
+      t.references :workspace
+      t.string :raw_payload
+      t.string :raw_payload_hash
+      t.string :build_status
+      t.string :build_opts
+
+      t.timestamps null: false
+    end
+  end
+end

data/db/migrate/20190308134512_create_async_callbacks.rb

@@ -0,0 +1,13 @@
+class CreateAsyncCallbacks < ActiveRecord::Migration[4.2]
+  def change
+    create_table :async_callbacks do |t|
+      t.string :uuid, :null => false
+      t.integer :timestamp, :null => false
+      t.string :listener_uri
+      t.string :target_host
+      t.string :target_port
+
+      t.timestamps null: false
+    end
+  end
+end

data/db/migrate/20190507120211_remove_payload_workspaces.rb

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