module Msf
module RPC
class Db < Base

private
	def db 
		@framework.db.active
	end

	def workspace(wspace = nil)
	 	if(wspace and wspace != "")
			return @framework.db.find_workspace(wspace) 
		end
		@framework.db.workspace
	end

	def fixOpts(opts)
		newopts = {}
		opts.each do |k,v|
			newopts[k.to_sym] = v
		end
		newopts
	end

	def opts2Hosts(opts)
		wspace = workspace(opts[:workspace]) 
		hosts  = []
		if opts[:host] or opts[:address]
			host = opts[:host] || opts[:address]
			hent = wspace.hosts.find_by_address(host)
			return hosts if hent == nil
			hosts << hent if hent.class == Msf::DBManager::Host
			hosts |= hent if hent.class == Array
		elsif opts[:addresses]
			return hosts if opts[:addresses].class != Array
			conditions = {}
			conditions[:address] = opts[:addresses]
			hent = wspace.hosts.all(:conditions => conditions)
			hosts |= hent if hent.class == Array
		end
		return hosts
	end

	def opts2Services(hosts,opts)
		wspace = workspace(opts[:workspace]) 
		services = []
		if opts[:host] or opts[:address] or opts[:addresses]
			return services if hosts.count < 1
			hosts.each do |h|
				if opts[:port] or opts[:proto]
					conditions = {}
					conditions[:port] = opts[:port] if opts[:port]
					conditions[:proto] = opts[:proto] if opts[:proto]
					sret = h.services.all(:conditions => conditions)
					next if sret == nil
					services |= sret if sret.class == Array
					services << sret if sret.class == Msf::DBManager::Service
				else
					services |= h.services
				end
			end
		elsif opts[:port] or opts[:proto]
			conditions = {}
			conditions[:port] = opts[:port] if opts[:port]
			conditions[:proto] = opts[:proto] if opts[:proto]
			sret = wspace.services.all(:conditions => conditions)
			services |= sret if sret.class == Array
			services << sret if sret.class == Msf::DBManager::Service
		end
		return services
	end

public

	def hosts(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)

		conditions = {}
		conditions[:state] = [Msf::HostState::Alive, Msf::HostState::Unknown] if opts[:only_up]
		conditions[:address] = opts[:addresses] if opts[:addresses]

		wspace = workspace(opts[:workspace])

		ret = {}
		ret[:hosts] = []
		wspace.hosts.all(:conditions => conditions, :order => :address).each do |h|
			host = {}
			host[:created_at] = h.created_at.to_s
			host[:address] = h.address.to_s
			host[:address6] = h.address6.to_s
			host[:mac] = h.mac.to_s
			host[:name] = h.name.to_s
			host[:state] = h.state.to_s
			host[:os_name] = h.os_name.to_s
			host[:os_flavor] = h.os_flavor.to_s
			host[:os_sp] = h.os_sp.to_s
			host[:os_lang] = h.os_lang.to_s
			host[:updated_at] = h.updated_at.to_s
			host[:purpose] = h.purpose.to_s
			host[:info] = h.info.to_s
			ret[:hosts]  << host
		end
		ret
	end

	def services(token, xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace])
		opts[:workspace] = wspace if opts[:workspace]
		hosts = []

		if opts[:addresses]
			conditions = {}
			conditions[:address] = opts[:addresses] if opts[:addresses]
			hosts = wspace.hosts.all(:conditions => conditions, :order => :address)
		elsif opts[:host] || opts[:address]
			host = @framework.db.get_host(opts)
			hosts << host
		end

		ret = {}
		ret[:services] = []

		a = @framework.db.get_host(opts)

		services = []
		if opts[:host] || opts[:address] || opts[:addresses]
			hosts.each do |host|
				sret = nil
				if(opts[:proto] && opts[:port])
					sret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
				else
					sret = host.services
				end
				next if sret == nil
				services << sret if sret.class == Msf::DBManager::Service
				services |= sret if sret.class == Array
			end
		else
			services = wspace.services
		end

		return ret if (not services)
		
		services.each do |s|
			service = {}
			host = s.host
			service[:host] = host.address || host.address6 || "unknown"
			service[:created_at] = s[:created_at].to_s
			service[:updated_at] = s[:updated_at].to_s
			service[:port] = s[:port]
			service[:proto] = s[:proto].to_s
			service[:state] = s[:state].to_s
			service[:name] = s[:name].to_s
			service[:info] = s[:info].to_s
			ret[:services] << service
		end
		ret
	end


	def vulns(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace])
		opts[:workspace] = wspace if opts[:workspace]

		ret = {}
		ret[:vulns] = []
		hosts = []
		services = []
		vulns = []
		# Get Matching Hosts
		if opts[:addresses]
			conditions = {}
			conditions[:address] = opts[:addresses] if opts[:addresses]
			hosts = wspace.hosts.all(:conditions => conditions, :order => :address)
		elsif opts[:host] || opts[:address]
			host = @framework.db.get_host(opts)
			hosts << host
		end

		#Get Matching Services
		if opts[:host] || opts[:address] || opts[:addresses]
			hosts.each do |host|
				sret = nil
				if(opts[:proto] && opts[:port])
					sret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
				else
					sret = host.services
				end
				next if sret == nil
				services << sret if sret.class == Msf::DBManager::Service
				services |= sret if sret.class == Array
			end
		elsif opts[:port] && opts[:proto]
			sret = wspace.services.find_by_proto_and_port(opts[:proto],opts[:port])
			services << sret if sret.class == Msf::DBManager::Service
			services |= sret if sret.class == Array
		end
		
		#get list of vulns
		if services.count > 0
			services.each do |s|
				if opts[:name]
					nret = s.vulns.find_by_name(opts[:name])
				else
					nret = s.vulns
				end
				next if nret == nil
				vulns << nret if nret.class == Msf::DBManager::Vuln
				vulns |= nret if nret.class == Array
			end
		elsif hosts.count > 0
			hosts.each do |h|
				if opts[:name]
					nret = h.vulns.find_by_name(opts[:name])
				else
					nret = h.vulns
				end
				next if nret == nil
				vulns << nret if nret.class == Msf::DBManager::Vuln
				vulns |= nret if nret.class == Array
			end
		else
			nret = wspace.vulns
			vulns << nret if nret.class == Msf::DBManager::Vuln
			vulns |= nret if nret.class == Array
		end

		vulns.each do |v|
			vuln = {}
			reflist = v.refs.map { |r| r.name }
			if(v.service)	
				vuln[:port] = v.service.port
				vuln[:proto] = v.service.proto
			else
				vuln[:port] = nil
				vuln[:proto] = nil
			end
			vuln[:time] = v.created_at
			vuln[:host] = v.host.address || v.host.address6 || nil	
			vuln[:name] = v.name
			vuln[:refs] = reflist.join(',')
			ret[:vulns] << vuln
		end
		ret
	end

	def workspaces(token)
		authenticate(token)
		if(not db)
			raise ::XMLRPC::FaultException.new(404, "database not loaded")
		end
		res = {}
		res[:workspaces] = []
		@framework.db.workspaces.each do |j|
			ws = {}
			ws[:name] = j.name
			ws[:created_at] = j.created_at.to_s
			ws[:updated_at] = j.updated_at.to_s
			res[:workspaces] << ws
		end
		res
	end

	def current_workspace(token)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		{ "workspace" => @framework.db.workspace.name }

	end

	def get_workspace(token,wspace)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		wspace = workspace(wspace)
		ret = {}
		ret[:workspace] = []
		if(wspace)
			w = {}
			w[:name] = wspace.name
			w[:created_at] = wspace.created_at.to_s
			w[:modified_at] = wspace.modified_at.to_s
			ret[:workspace] << w
		end
		ret
	end

	def set_workspace(token,wspace)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		workspace = @framework.db.find_workspace(wspace)
		if(workspace)
			@framework.db.workspace = workspace
			return { 'result' => "success" }
		end
		{ 'result' => 'failed' }
	end

	def del_workspace(token,wspace)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		# Delete workspace
		workspace = @framework.db.find_workspace(wspace)
		if workspace.nil?
			raise ::XMLRPC::FaultException.new(404, "Workspace not found: #{wspace}")
		elsif workspace.default?
			workspace.destroy
			workspace = @framework.db.add_workspace(name)
		else
			# switch to the default workspace if we're about to delete the current one
			@framework.db.workspace = @framework.db.default_workspace if @framework.db.workspace.name == workspace.name
			# now destroy the named workspace
			workspace.destroy
		end
		{ 'result' => "success" }
	end

	def add_workspace(token,wspace)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		wspace = @framework.db.add_workspace(wspace)
		return { 'result' => 'success' } if(wspace)
		{ 'result' => 'failed' }
	end

	def get_host(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		ret = {}
		ret[:host] = []
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		h = @framework.db.get_host(opts)
		if(h)
			host = {}
			host[:created_at] = h.created_at.to_s
			host[:address] = h.address.to_s
			host[:address6] = h.address6.to_s
			host[:mac] = h.mac.to_s
			host[:name] = h.name.to_s
			host[:state] = h.state.to_s
			host[:os_name] = h.os_name.to_s
			host[:os_flavor] = h.os_flavor.to_s
			host[:os_sp] = h.os_sp.to_s
			host[:os_lang] = h.os_lang.to_s
			host[:updated_at] = h.updated_at.to_s
			host[:purpose] = h.purpose.to_s
			host[:info] = h.info.to_s
			ret[:host] << host
		end
		ret	
	end

	def report_host(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]

		res = @framework.db.report_host(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
		
	end

	def report_service(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		res = @framework.db.report_service(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
	end

	def get_service(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace])
		opts[:workspace] = wspace if opts[:workspace]

		ret = {}
		ret[:service] = []

		host = @framework.db.get_host(opts)

		services = []
		sret = nil

		if(host && opts[:proto] && opts[:port])
			sret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
		elsif(opts[:proto] && opts[:port])
			conditions = {}
			conditions[:state] = [ServiceState::Open] if opts[:up]
			conditions[:proto] = opts[:proto] if opts[:proto]
			conditions[:port] = opts[:port] if opts[:port]
			conditions[:name] = opts[:names] if opts[:names]
			sret = wspace.services.all(:conditions => conditions, :order => "hosts.address, port")
		else
			sret = host.services
		end
		return ret if sret == nil
		services << sret if sret.class == Msf::DBManager::Service
		services |= sret if sret.class == Array

		
		services.each do |s|
			service = {}
			host = s.host
			service[:host] = host.address || host.address6 || "unknown"
			service[:created_at] = s[:created_at].to_s
			service[:updated_at] = s[:updated_at].to_s
			service[:port] = s[:port]
			service[:proto] = s[:proto].to_s
			service[:state] = s[:state].to_s
			service[:name] = s[:name].to_s
			service[:info] = s[:info].to_s
			ret[:service] << service
		end
		ret
	end

	def get_note(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]

		ret = {}
		ret[:note] = []

		host = @framework.db.get_host(opts)

		return ret if( not host)
		notes = []
		if(opts[:proto] && opts[:port])
			services = []
			nret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
			return ret if nret == nil
			services << nret if nret.class == Msf::DBManager::Service
			services |= nret if nret.class == Array

			services.each do |s|
				nret = nil
				if opts[:ntype]
					nret = s.notes.find_by_ntype(opts[:ntype])
				else
					nret = s.notes
				end
				next if nret == nil
				notes << nret if nret.class == Msf::DBManager::Note
				notes |= nret if nret.class == Array
			end
		else
			notes = host.notes
		end
		notes.each do |n|
			note = {}
			host = n.host
			note[:host] = host.address || host.address6 || "unknown"
			if n.service
				note[:port] = n.service.port
				note[:proto] = n.service.proto
			end
			note[:created_at] = n[:created_at].to_s
			note[:updated_at] = n[:updated_at].to_s
			note[:ntype] = n[:ntype].to_s
			note[:data] = n[:data]
			note[:critical] = n[:critical].to_s
			note[:seen] = n[:seen].to_s
			ret[:note] << note
		end
		ret
	end

	def get_client(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		ret = {}
		ret[:client] = []
		c = @framework.db.get_client(opts)
		if(c)
			client = {}
			host = c.host
			client[:host] = host.address
			client[:created_at] = c.created_at.to_s
			client[:updated_at] = c.updated_at.to_s
			client[:ua_string] = c.ua_string.to_s
			client[:ua_name] = c.ua_name.to_s
			client[:ua_ver] = c.ua_ver.to_s
			ret[:client] << client
		end
		ret
	end

	def report_client(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		res = @framework.db.report_client(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
	end

	#DOC NOTE: :data and :ntype are REQUIRED
	def report_note(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		if (opts[:host] or opts[:address]) and opts[:port] and opts[:proto]
			addr = opts[:host] || opts[:address]
			wspace = opts[:workspace] || @framework.db.workspace
			host = wspace.hosts.find_by_address(addr)
			service = host.services.find_by_proto_and_port(opts[:proto],opts[:port]) if host.services.count > 0
			opts[:service] = service if service
		end
			
		res = @framework.db.report_note(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
	end

	def notes(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) if opts[:workspace]
		opts[:workspace] = wspace

		ret = {}
		ret[:notes] = []
		hosts = []
		services = []
		notes = []

		# Get Matching Hosts
		if opts[:addresses]
			conditions = {}
			conditions[:address] = opts[:addresses] if opts[:addresses]
			hosts = wspace.hosts.all(:conditions => conditions, :order => :address)
		elsif opts[:host] || opts[:address]
			host = @framework.db.get_host(opts)
			hosts << host
		end

		#Get Matching Services
		if opts[:host] || opts[:address] || opts[:addresses]
			hosts.each do |host|
				sret = nil
				if(opts[:proto] && opts[:port])
					sret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
				else
					sret = host.services
				end
				next if sret == nil
				services << sret if sret.class == Msf::DBManager::Service
				services |= sret if sret.class == Array
			end
		elsif opts[:port] && opts[:proto]
			sret = wspace.services.find_by_proto_and_port(opts[:proto],opts[:port])
			services << sret if sret.class == Msf::DBManager::Service
			services |= sret if sret.class == Array
		end
		
		#get list of notes
		if services.count > 0
			services.each do |s|
				if opts[:ntype]
					nret = s.notes.find_by_ntype(opts[:ntype])
				else
					nret = s.notes
				end
				next if nret == nil
				notes << nret if nret.class == Msf::DBManager::Note
				notes |= nret if nret.class == Array
			end
		elsif hosts.count > 0
			hosts.each do |h|
				if opts[:ntype]
					nret = h.notes.find_by_ntype(opts[:ntype])
				else
					nret = h.notes
				end
				next if nret == nil
				notes << nret if nret.class == Msf::DBManager::Note
				notes |= nret if nret.class == Array
			end
		else
			nret = wspace.notes
			notes << nret if nret.class == Msf::DBManager::Note
			notes |= nret if nret.class == Array
		end

		notes.each do |n|
			note = {}
			note[:time] = n.created_at.to_s
			note[:host] = ""
			note[:service] = ""
			note[:host] = n.host.address || n.host.address6 if(n.host)
			note[:service] = n.service.name || n.service.port  if(n.service)
			note[:type ] = n.ntype.to_s
			note[:data] = n.data.inspect
			ret[:notes] << note
		end
		ret
	end

	def report_auth_info(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		res = @framework.db.report_auth_info(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
	end

	def get_auth_info(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		ret = {}
		ret[:auth_info] = []
		ai = @framework.db.get_auth_info(opts)
		ai.each do |i|
			info = {}
			i.each do |k,v|
				info[k.to_sym] = v
			end
			ret[:auth_info] << info	
		end
		ret
	end

	def get_ref(token,name)
		authenticate(token)
		return @framework.db.get_ref(name)
	end

	def del_vuln(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts  = []
		services = []
		vulns = []
			
		if opts[:host] or opts[:address] or opts[:addresses]
			hosts = opts2Hosts(opts)
		end

		if opts[:port] or opts[:proto]
			if opts[:host] or opts[:address] or opts[:addresses]
				services = opts2Services(hosts,opts)
			else
				services = opts2Services([],opts)
			end
		end

		if opts[:port] or opts[:proto]
			services.each do |s|
				vret = nil
				if opts[:name]
					vret = s.vulns.find_by_name(opts[:name])
				else
					vret = s.vulns
				end
				next if vret == nil
				vulns << vret if vret.class == Msf::DBManager::Vuln
				vulns |= vret if vret.class == Array
			end
		elsif opts[:address] or opts[:host] or opts[:addresses]
			hosts.each do |h|
				vret = nil
				if opts[:name]
					vret = h.vulns.find_by_name(opts[:name])
				else
					vret = h.vulns
				end
				next if vret == nil
				vulns << vret if vret.class == Msf::DBManager::Vuln
				vulns |= vret if vret.class == Array
			end
		else
			vret = nil
			if opts[:name]
				vret = wspace.vulns.find_by_name(opts[:name])
			else
				vret = wspace.vulns
			end
			vulns << vret if vret.class == Msf::DBManager::Vuln
			vulns |= vret if vret.class == Array
		end

		deleted = []
		vulns.each do |v|
			dent = {}
			dent[:address] = v.host.address.to_s if v.host
			dent[:port] = v.service.port if v.service
			dent[:proto] = v.service.proto if v.service
			dent[:name] = v.name
			deleted << dent
			v.destroy	
		end
			
		return { :result => 'success', :deleted => deleted } 
	end

	def del_note(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts  = []
		services = []
		notes = []
			
		if opts[:host] or opts[:address] or opts[:addresses]
			hosts = opts2Hosts(opts)
		end

		if opts[:port] or opts[:proto]
			if opts[:host] or opts[:address] or opts[:addresses]
				services = opts2Services(hosts,opts)
			else
				services = opts2Services([],opts)
			end
		end

		if opts[:port] or opts[:proto]
			services.each do |s|
				nret = nil
				if opts[:ntype]
					nret = s.notes.find_by_ntype(opts[:ntype])
				else
					nret = s.notes
				end
				next if nret == nil
				notes << nret if nret.class == Msf::DBManager::Note
				notes |= nret if nret.class == Array
			end
		elsif opts[:address] or opts[:host] or opts[:addresses]
			hosts.each do |h|
				nret = nil
				if opts[:ntype]
					nret = h.notes.find_by_ntype(opts[:ntype])
				else
					nret = h.notes
				end
				next if nret == nil
				notes << nret if nret.class == Msf::DBManager::Note
				notes |= nret if nret.class == Array
			end
		else
			nret = nil
			if opts[:ntype]
				nret = wspace.notes.find_by_ntype(opts[:ntype])
			else
				nret = wspace.notes
			end
			notes << nret if nret.class == Msf::DBManager::Note
			notes |= nret if nret.class == Array
		end
		deleted = []
		notes.each do |n|
			dent = {}
			dent[:address] = n.host.address.to_s if n.host
			dent[:port] = n.service.port if n.service
			dent[:proto] = n.service.proto if n.service
			dent[:ntype] = n.ntype
			deleted << dent
			n.destroy	
		end
			
		return { :result => 'success', :deleted => deleted } 
	end

	def del_service(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts  = []
		services = []
		if opts[:host] or opts[:address]
			host = opts[:host] || opts[:address]
			hent = wspace.hosts.find_by_address(host)
			return { :result => 'failed' } if hent == nil or hent.class != Msf::DBManager::Host
			hosts << hent
		elsif opts[:addresses]
			return { :result => 'failed' } if opts[:addresses].class != Array
			conditions = { :address => opts[:addresses] }
			hent = wspace.hosts.all(:conditions => conditions)
			return { :result => 'failed' } if hent == nil
			hosts |= hent if hent.class == Array
			hosts << hent if hent.class == Msf::DBManager::Host
		end
		if opts[:addresses] or opts[:address] or opts[:host]
			hosts.each do |h|
				sret = nil
				if opts[:port] or opts[:proto]
					conditions = {}
					conditions[:port] = opts[:port] if opts[:port]
					conditions[:proto] = opts[:proto] if opts[:proto]
					sret = h.services.all(:conditions => conditions)
					next if sret == nil
					services << sret if sret.class == Msf::DBManager::Service
					services |= sret if sret.class == Array
				else
					services |= h.services
				end
			end
		elsif opts[:port] or opts[:proto]
			conditions = {}
			conditions[:port] = opts[:port] if opts[:port]
			conditions[:proto] = opts[:proto] if opts[:proto]
			sret = wspace.services.all(:conditions => conditions)
			services << sret if sret and sret.class == Msf::DBManager::Service
			services |= sret if sret and sret.class == Array
		end
					
				
				
		deleted = []
		services.each do |s|
			dent = {}
			dent[:address] = s.host.address.to_s
			dent[:port] = s.port
			dent[:proto] = s.proto
			deleted << dent
			s.destroy	
		end
			
		return { :result => 'success', :deleted => deleted } 
	end

	def del_host(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts  = []
		if opts[:host] or opts[:address]
			host = opts[:host] || opts[:address]
			hent = wspace.hosts.find_by_address(host)
			return { :result => 'failed' } if hent == nil or hent.class != Msf::DBManager::Host
			hosts << hent
		elsif opts[:addresses]
			return { :result => 'failed' } if opts[:addresses].class != Array
			conditions = { :address => opts[:addresses] }
			hent = wspace.hosts.all(:conditions => conditions)
			return { :result => 'failed' } if hent == nil
			hosts |= hent if hent.class == Array
			hosts << hent if hent.class == Msf::DBManager::Host
		end
		deleted = []
		hosts.each do |h|
			deleted << h.address.to_s
			h.destroy	
		end
			
		return { :result => 'success', :deleted => deleted } 
	end


	def report_vuln(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		res = @framework.db.report_vuln(opts)
		return { :result => 'success' } if(res)
		{ :result => 'failed' }
	end


	def events(token,wspace = nil)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		wspace = workspace(wspace)
		raise ::XMLRPC::FaultException.new(404, "unknown workspace") if(not wspace)
		ret = {}
		ret[:events] = []

		@framework.db.events(wspace).each do |e|
			event = {}
			event[:host] = e.host.address || e.host.address6 if(e.host)
			event[:created_at] = e.created_at
			event[:updated_at] = e.updated_at
			event[:name] = e.name
			event[:critical] = e.critical if(e.critical)	
			event[:username] = e.critical if(e.username)	
			event[:info] = e.info
			ret[:events] << event
		end
		ret
	end
	def report_event(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		res = @framework.db.report_event(opts)
		return { :result => 'success' } if(res)
	end

	#NOTE Path is required
	#NOTE To match a service need host, port, proto
	def report_loot(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		if opts[:host] && opts[:port] && opts[:proto]
			opts[:service] = @framework.db.find_or_create_service(opts)
		end

		ret = @framework.db.report_loot(opts)
		return { :result => 'success' } if(res)
	end

	def loots(token,wspace=nil)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		wspace = workspace(wspace)
		raise ::XMLRPC::FaultException.new(404, "unknown workspace") if(not wspace)
		ret = {}
		ret[:loots] = []
		@framework.db.loots(wspace).each do |l|
			loot = {}
			loot[:host] = l.host.address || l.host.address6 if(l.host)
			loot[:service] = l.service.name || n.service.port  if(n.service)
			loot[:ltype] = l.ltype
			loot[:ctype] = l.ctype
			loot[:data] = l.data
			loot[:created_at] = l.created_at
			loot[:updated_at] = l.updated_at
			loot[:name] = l.name
			loot[:info] = l.info
			ret[:loots] << loot
		end
		ret
	end
	def import_data(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import(opts)
		return { :result => 'success' }
	end
	def import_msfe_xml(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_msfe_xml(opts)
		return { :result => 'success' }
	end
	def import_nexpose_simplexml(args={})
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nexpose_simplexml(opts)
		return { :result => 'success' }
	end
	def import_nexpose_rawxml(args={})
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nexpose_rawxml(opts)
		return { :result => 'success' }
	end
	def import_nmap_xml(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nmap_xml(opts)
		return { :result => 'success' }
	end
	def import_nessus_nbe(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nessus_nbe(opts)
		return { :result => 'success' }
	end
	def import_nessus_xml(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nessus_xml(opts)
		return { :result => 'success' }
	end
	def import_nessus_xml_v2(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nessus_xml_v2(opts)
		return { :result => 'success' }
	end
	def import_qualys_xml(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_qualys_xml(opts)
		return { :result => 'success' }
	end
	def import_ip_list(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_ip_list(opts)
		return { :result => 'success' }
	end
	def import_amap_log(args={})
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_nexpose_rawxml(opts)
		return { :result => 'success' }
	end
	def import_amap_mlog(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)
		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]
		opts[:data] = Rex::Text.decode_base64(opts[:data])
		@framework.db.import_amap_mlog(opts)
		return { :result => 'success' }
	end

	def get_vuln(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		opts[:workspace] = workspace(opts[:workspace]) if opts[:workspace]

		ret = {}
		ret[:vuln] = []

		host = @framework.db.get_host(opts)

		return ret if( not host)
		vulns = []

		if(opts[:proto] && opts[:port])
			services = []
			sret = host.services.find_by_proto_and_port(opts[:proto], opts[:port])
			return ret if sret == nil
			services << sret if sret.class == Msf::DBManager::Service
			services |= sret if sret.class == Array
			
			services.each do |s|
				vulns |= s.vulns
			end
		else
			vulns = host.vulns
		end

		return ret if (not vulns)
		
		vulns.each do |v|
			vuln= {}
			host= v.host
			vuln[:host] = host.address || host.address6 || "unknown"
			if v.service
				vuln[:port] = v.service.port
				vuln[:proto] = v.service.proto
			end
			vuln[:created_at] = v[:created_at].to_s
			vuln[:updated_at] = v[:updated_at].to_s
			vuln[:name] = v[:name].to_s
			vuln[:info] = v[:info].to_s
			vuln[:refs] = []
			v.refs.each do |r|
				vuln[:refs] << r.name
			end	
			ret[:vuln] << vuln
		end
		ret
	end
	def clients(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts = []
		clients = []
		ret = {}
		ret[:clients] = []

		if opts[:host] or opts[:address] or opts[:addresses]
			hosts = opts2Hosts(opts)
		else
			hosts = wspace.hosts
		end

		hosts.each do |h|
			cret = nil
			if opts[:ua_name] or opts[:ua_ver]	
				conditions = {}
				conditions[:ua_name] = opts[:ua_name] if opts[:ua_name]
				conditions[:ua_ver] = opts[:ua_ver] if opts[:ua_ver]
				cret = h.clients.all(:conditions => conditions)
			else
				cret = h.clients
			end
			next if cret == nil
			clients << cret if cret.class == Msf::DBManager::Client
			clients |= cret if cret.class == Array
		end
		clients.each do |c|
			client = {}
			client[:host] = c.host.address.to_s if c.host
			client[:ua_string] = c.ua_string
			client[:ua_name] = c.ua_name
			client[:ua_ver] = c.ua_ver
			client[:created_at] = c.created_at.to_s
			client[:updated_at] = c.updated_at.to_s
			ret[:clients] << client
		end
		ret
	end

	def del_client(token,xopts)
		authenticate(token)
		raise ::XMLRPC::FaultException.new(404, "database not loaded") if(not db)

		opts = fixOpts(xopts)
		wspace = workspace(opts[:workspace]) 
		hosts = []
		clients = []

		if opts[:host] or opts[:address] or opts[:addresses]
			hosts = opts2Hosts(opts)
		else
			hosts = wspace.hosts
		end

		hosts.each do |h|
			cret = nil
			if opts[:ua_name] or opts[:ua_ver]	
				conditions = {}
				conditions[:ua_name] = opts[:ua_name] if opts[:ua_name]
				conditions[:ua_ver] = opts[:ua_ver] if opts[:ua_ver]
				cret = h.clients.all(:conditions => conditions)
			else
				cret = h.clients
			end
			next if cret == nil
			clients << cret if cret.class == Msf::DBManager::Client
			clients |= cret if cret.class == Array
		end

		deleted = []
		clients.each do |c|
			dent = {}
			dent[:address] = c.host.address.to_s
			dent[:ua_string] = c.ua_string
			deleted << dent
			c.destroy	
		end

		return { :result => 'success', :deleted => deleted } 
			
	end

	def driver(token,xopts)
		authenticate(token)
		opts = fixOpts(xopts)
		if opts[:driver]
			if @framework.db.drivers.include?(opts[:driver])
				@framework.db.driver = opts[:driver]
				return { :result => 'success' }
			else
				return { :result => 'failed' }

			end
		else
			return { :driver => @framework.db.driver.to_s }
		end
		return { :result => 'failed' }
	end

	def connect(token,xopts)
		authenticate(token)
		opts = fixOpts(xopts)
		if(not @framework.db.driver and not opts[:driver])
			return { :result => 'failed' }
		end

		if opts[:driver]
			if @framework.db.drivers.include?(opts[:driver])
				@framework.db.driver = opts[:driver]
			else
				return { :result => 'failed' }
			end
		end
		
		driver = @framework.db.driver

		case driver
		when 'mysql'
			opts['adapter'] = 'mysql'
		when 'postgresql'
			opts['adapter'] = 'postgresql'
		when 'sqlite3'
			opts['adapter'] = 'sqlite3'
		else
			return { :result => 'failed' }
		end
	
		if (not @framework.db.connect(opts))		
			return { :result => 'failed' }
		end
		return { :result => 'success' }
		
	end

	def status(token)
		authenticate(token)
		if (not @framework.db.driver)
			return {:driver => 'None' }
		end
		cdb = ""
                if ActiveRecord::Base.connected? and ActiveRecord::Base.connection.active?
                                        if ActiveRecord::Base.connection.respond_to? :current_database
                                                cdb = ActiveRecord::Base.connection.current_database
                                        else
						cdb = ActiveRecord::Base.connection.instance_variable_get(:@config)[:database]
					end
					return {:driver => @framework.db.driver.to_s , :db => cdb }
		else
			return {:driver => @framework.db.driver.to_s}
		end
		return {:driver => 'None' }
	end
	def disconnect(token)
		authenticate(token)
		if (@framework.db)
			@framework.db.disconnect()
			return { :result => 'success' }
		else
			return { :result => 'failed' }
		end
	end


end
end
end