metasploit-framework/lib/msf/core/db_objects.rb

module Msf

##
#
# This module defines all of the DB database tables
# and creates ActiveRecord objects for each one of them
#
##

class DBManager

class Lock
	@@mutex = Mutex.new
	def self.mutex
		@@mutex
	end
end


# ActiveRecord/sqlite3 has locking issues when you update a table with a pending select
# This set of instance/class wrappers should prevent a table lock
# Straight up gangsta shit from spoon (ripped from BION)
module DBSave

	def save(*args)
		Lock.mutex.synchronize do
			super(*args)
		end
	end	

	def self.included(mod)
		class << mod
			def find(*args)
				Lock.mutex.synchronize do
					super(*args)
				end
			end			
		end
	end
end

# Host object definition
class Host < ActiveRecord::Base
	include DBSave
	has_many :services
	has_many :vulns, :through => :services
end

# Service object definition
class Service < ActiveRecord::Base
	include DBSave
	has_many :vulns
	belongs_to :host

	def host
		Host.find(:first, :conditions => [ "id = ?", host_id ])
	end	
end

# Vuln object definition
class Vuln < ActiveRecord::Base
	include DBSave
	belongs_to :service
	has_and_belongs_to_many :refs, :join_table => :vulns_refs

	def service
		Service.find(:first, :conditions => [ "id = ?", service_id ])
	end

	def host
		Host.find(:first, :conditions => [ "id = ?", service.host_id ])
	end
end

# Reference object definition
class Ref < ActiveRecord::Base
	include DBSave
	has_and_belongs_to_many :vulns, :join_table => :vulns_refs
end

# Reference object definition
class VulnRefs < ActiveRecord::Base
	set_table_name 'vulns_refs'
	include DBSave
end


# Service object definition
class Note < ActiveRecord::Base
	include DBSave
	belongs_to :host

	def host
		Host.find(:first, :conditions => [ "id = ?", host_id ])
	end	
end

end
end