Fix #9827

rapid7 · Apr 6, 2018 · c303859 · c303859
1 parent 81c78a5
commit c303859
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lib/msf/core/payload/transport_config.rb b/lib/msf/core/payload/transport_config.rb
@@ -51,9 +51,16 @@ def transport_config_reverse_https(opts={})
 
   def transport_uri_components(opts={})
     ds = opts[:datastore] || datastore
-    scheme = opts[:scheme]
-    lhost = ds['LHOST']
-    lport = ds['LPORT']
+    if opts[:url]
+      u = URI(opts[:url])
+      scheme = u.scheme
+      lhost = u.host
+      lport = u.port
+    else
+      scheme = opts[:scheme]
+      lhost = ds['LHOST']
+      lport = ds['LPORT']
+    end
     if ds['OverrideRequestHost']
       scheme = ds['OverrideScheme'] || scheme
       lhost = ds['OverrideLHOST'] || lhost