From 8c1a597a25aa5f94ad2de524fa5672989d0481e3 Mon Sep 17 00:00:00 2001
From: James Lee <egypt@metasploit.com>
Date: Wed, 1 Apr 2015 13:12:28 -0500
Subject: [PATCH] Make a Session record before using it

How about that.
---
 lib/msf/core/db_manager/session.rb | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

diff --git a/lib/msf/core/db_manager/session.rb b/lib/msf/core/db_manager/session.rb
index 4d45d9d1ac..811728d892 100644
--- a/lib/msf/core/db_manager/session.rb
+++ b/lib/msf/core/db_manager/session.rb
@@ -187,13 +187,6 @@ module Msf::DBManager::Session
         via_payload: session.via_payload,
       }
 
-      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
-
       # In the case of multi handler we cannot yet determine the true
       # exploit responsible. But we can at least show the parent versus
       # just the generic handler:
@@ -202,6 +195,14 @@ module Msf::DBManager::Session
       end
 
       s = ::Mdm::Session.create!(sess_data)
+
+      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
+
       s
     }
   end