Modify to use find_or_create_host

2018-05-14 17:29:57 -04:00 · 2018-05-14 17:29:57 -04:00 · 25fdc02902
parent 09a7396ad5
commit 25fdc02902
1 changed files with 2 additions and 2 deletions
--- a/lib/msf/core/db_manager/event.rb
+++ b/lib/msf/core/db_manager/event.rb
@ -12,8 +12,8 @@ module Msf::DBManager::Event
    return if not wspace # Temp fix?
    uname  = opts.delete(:username)

-    if ! opts[:host].kind_of? ::Mdm::Host and opts[:host]
-      opts[:host] = report_host(:workspace => wspace, :host => opts[:host])
+    if !opts[:host].nil? && !opts[:host].kind_of?(::Mdm::Host)
+      opts[:host] = find_or_create_host(workspace: wspace, host: opts[:host])
    end

    ::Mdm::Event.create(opts.merge(:workspace_id => wspace[:id], :username => uname))