infosecn1nja
/
metasploit-framework
mirror of https://github.com/infosecn1nja/metasploit-framework.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

use the new database api. fixes 753 

git-svn-id: file:///home/svn/framework3/trunk@8086 4d416f70-5f16-0410-b530-b9f4589650da
unstable
James Lee 2010-01-07 19:14:44 +00:00
parent 30b897b6cd
commit 9cdef831bb
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
plugins/db_tracker.rb
View File

@ -30,15 +30,10 @@ class Plugin::DB_Tracer < Msf::Plugin
# to hold the framework in 'Msf'
return if not param.context['Msf']
host = param.context['Msf'].db.get_host(param.context, sock.peerhost)
host = param.context['Msf'].db.find_or_create_host(:host => sock.peerhost, :state => Msf::HostState::Alive)
return if not host
port = param.context['Msf'].db.get_service(param.context, host, param.proto, sock.peerport)
return if not port
if host.state != Msf::HostState::Alive
param.context['Msf'].db.report_host_state(self, sock.peerhost, Msf::HostState::Alive)
end
param.context['Msf'].db.report_service(:host => host, :proto => param.proto, :port => sock.peerport)
end
end
end