HD Moore
6a06a95f2f
git-svn-id: file:///home/svn/framework3/trunk@10532 4d416f70-5f16-0410-b530-b9f4589650da
82 lines
1.6 KiB
Ruby
82 lines
1.6 KiB
Ruby
module Msf
|
|
class DBManager
|
|
|
|
class Workspace < ActiveRecord::Base
|
|
include DBSave
|
|
|
|
DEFAULT = "default"
|
|
|
|
has_many :hosts, :dependent => :destroy
|
|
has_many :services, :through => :hosts
|
|
has_many :notes, :dependent => :destroy
|
|
has_many :loots, :dependent => :destroy
|
|
has_many :events,:dependent => :destroy
|
|
has_many :reports, :dependent => :destroy
|
|
has_many :report_templates, :dependent => :destroy
|
|
has_many :tasks, :dependent => :destroy
|
|
has_many :clients, :through => :hosts
|
|
has_many :vulns, :through => :hosts
|
|
has_many :creds, :dependent => :destroy
|
|
has_many :exploited_hosts, :through => :hosts
|
|
|
|
|
|
validates_uniqueness_of :name
|
|
validates_presence_of :name
|
|
|
|
def self.default
|
|
Workspace.find_or_create_by_name(DEFAULT)
|
|
end
|
|
|
|
def default?
|
|
name == DEFAULT
|
|
end
|
|
|
|
def creds
|
|
Cred.find(
|
|
:all,
|
|
:include => {:service => :host}, # That's some magic right there.
|
|
:conditions => ["hosts.workspace_id = ?", self.id]
|
|
)
|
|
end
|
|
|
|
#
|
|
# This method iterates the creds table calling the supplied block with the
|
|
# cred instance of each entry.
|
|
#
|
|
def each_cred(&block)
|
|
creds.each do |cred|
|
|
block.call(cred)
|
|
end
|
|
end
|
|
|
|
def web_sites
|
|
hosts.map{|host| host.web_sites}.flatten
|
|
end
|
|
|
|
def web_pages
|
|
web_sites.map{|w| w.web_pages}.flatten
|
|
end
|
|
|
|
def web_forms
|
|
web_sites.map{|w| w.web_forms}.flatten
|
|
end
|
|
|
|
def web_vulns
|
|
web_sites.map{|w| w.web_vulns}.flatten
|
|
end
|
|
|
|
def web_unique_forms(addrs=nil)
|
|
xhosts = addrs ? hosts.select{|host| addrs.include?(host.address) } : hosts
|
|
xhosts.map { |host|
|
|
host.web_sites.map{|site|
|
|
site.web_forms.find(:all, :select => 'DISTINCT web_site_id, path, method, query')
|
|
}
|
|
}.flatten
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
|