handle no datastore

rapid7 · Dec 29, 2017 · a444bdb · a444bdb
1 parent 198aeda
commit a444bdb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/msf/base/sessions/command_shell.rb b/lib/msf/base/sessions/command_shell.rb
@@ -51,7 +51,8 @@ def initialize(conn, opts = {})
     self.platform ||= ""
     self.arch     ||= ""
     self.max_threads = 1
-    if !opts[:datastore]["CommandShellCleanupCommand"].blank?
+    datastore = opts[:datastore]
+    if datastore && !datastore["CommandShellCleanupCommand"].blank?
       @cleanup_command = opts[:datastore]["CommandShellCleanupCommand"]
     end
     super