Class: TrackSuccessfulExploits

Inherits:
ActiveRecord::Migration
  • Object
show all
Defined in:
db/migrate/20110513143900_track_successful_exploits.rb

Defined Under Namespace

Classes: ExploitedHost, Vuln

Class Method Summary collapse

Class Method Details

.down ⇒ Object 



27
28
29
 
# File 'db/migrate/20110513143900_track_successful_exploits.rb', line 27

def self.down
  remove_column :vulns, :exploited_at
end

.up ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
 
# File 'db/migrate/20110513143900_track_successful_exploits.rb', line 10

def self.up
  add_column :vulns, :exploited_at, :timestamp

  # Migrate existing exploited_hosts entries

  ExploitedHost.find(:all).select {|x| x.name}.each do |exploited_host|
    next unless(exploited_host.name =~ /^(exploit|auxiliary)\//)
    vulns = Vuln.find_all_by_name_and_host_id(exploited_host.name, exploited_host.host_id)
    next if vulns.empty?
    vulns.each do |vuln|
      vuln.exploited_at = exploited_host.updated_at
      vuln.save
    end
  end
  
end