Report on TaskSession associations

add TaskSession objects so when we report
on a session, we know what Task created it, if there
was a task

Author: David Maloney

Gemfile

@@ -15,7 +15,7 @@ group :db do
 	# Needed for Msf::DbManager
 	gem 'activerecord'
 	# Database models shared between framework and Pro.
-	gem 'metasploit_data_models', '~> 0.15.2'
+	gem 'metasploit_data_models', '~> 0.16.1'
 	# Needed for module caching in Mdm::ModuleDetails
 	gem 'pg', '>= 0.11'
 end

Gemfile.lock

@@ -23,7 +23,7 @@ GEM
     i18n (0.6.1)
     json (1.7.7)
     metaclass (0.0.1)
-    metasploit_data_models (0.15.2)
+    metasploit_data_models (0.16.1)
       activerecord (>= 3.2.13)
       activesupport
       pg
@@ -65,7 +65,7 @@ DEPENDENCIES
   database_cleaner
   factory_girl (>= 4.1.0)
   json
-  metasploit_data_models (~> 0.15.2)
+  metasploit_data_models (~> 0.16.1)
   msgpack
   nokogiri
   pcaprub

lib/msf/core/db.rb

@@ -786,6 +786,14 @@ def report_session(opts)
 		s = ::Mdm::Session.new(sess_data)
 		s.save!
 
+    if session.exploit_task and session.exploit_task.record
+      session_task =  session.exploit_task.record
+      if session_task.class == Mdm::Task
+        Mdm::TaskSession.create(:task => session_task, :session => s )
+      end
+    end
+
+
 		if opts[:session]
 			session.db_record = s
 		end

lib/msf/core/session.rb

@@ -270,6 +270,9 @@ def set_from_exploit(m)
 		self.user_output = m.user_output if m.user_output
 		self.exploit_uuid = m.uuid
 		self.exploit = m
+    if m[:task]
+      self.exploit_task = m[:task]
+    end
 	end
 
 	#
@@ -365,6 +368,10 @@ def alive?
 	# The datastore of the exploit that created this session
 	#
 	attr_accessor :exploit_datastore
+  #
+  # The task that ran the exploit that got the session (that swallowed the fly)
+  #
+  attr_accessor :exploit_task
 	#
 	# The specific identified session info
 	#