Factor out mettle configuration

Also cleans up some stuff: s/url/uri/ and base-64 encodes UUIDs

Author: acammack-r7

lib/msf/base/sessions/mettle_config.rb

@@ -0,0 +1,22 @@
+# -*- coding: binary -*-
+
+require 'msf/core/payload/transport_config'
+require 'base64'
+
+module Msf
+module Sessions
+module MettleConfig
+
+  include Msf::Payload::TransportConfig
+
+  def generate_config(opts={})
+    transport = transport_config_reverse_tcp(opts)
+    opts[:uuid] ||= generate_payload_uuid
+    opts[:uuid] = Base64.encode64(opts[:uuid].to_raw).strip
+    opts[:uri] ||= "#{transport[:scheme]}://#{transport[:lhost]}:#{transport[:lport]}"
+    opts.slice(:uuid, :uri, :debug, :log_file)
+  end
+
+end
+end
+end

modules/payloads/stages/linux/armle/mettle.rb

@@ -6,10 +6,12 @@
 require 'msf/core'
 require 'msf/base/sessions/meterpreter_armle_linux'
 require 'msf/base/sessions/meterpreter_options'
+require 'msf/base/sessions/mettle_config'
 require 'rex/elfparsey'
 
 module MetasploitModule
   include Msf::Sessions::MeterpreterOptions
+  include Msf::Sessions::MettleConfig
 
   def initialize(info = {})
     super(
@@ -80,8 +82,6 @@ def handle_intermediate_stage(conn, payload)
   end
 
   def generate_stage(opts = {})
-    opts[:uuid] ||= generate_payload_uuid
-    MetasploitPayloads::Mettle.new('armv5l-linux-musleabi', opts.slice(:uuid, :url, :debug, :log_file)).
-      to_binary :process_image
+    MetasploitPayloads::Mettle.new('armv5l-linux-musleabi', generate_config(opts)).to_binary :process_image
   end
 end

modules/payloads/stages/linux/mipsbe/mettle.rb

@@ -6,10 +6,12 @@
 require 'msf/core'
 require 'msf/base/sessions/meterpreter_mipsbe_linux'
 require 'msf/base/sessions/meterpreter_options'
+require 'msf/base/sessions/mettle_config'
 require 'rex/elfparsey'
 
 module MetasploitModule
   include Msf::Sessions::MeterpreterOptions
+  include Msf::Sessions::MettleConfig
 
   def initialize(info = {})
     super(
@@ -91,8 +93,6 @@ def handle_intermediate_stage(conn, payload)
   end
 
   def generate_stage(opts = {})
-    opts[:uuid] ||= generate_payload_uuid
-    MetasploitPayloads::Mettle.new('mips-linux-muslsf', opts.slice(:uuid, :url, :debug, :log_file)).
-      to_binary :process_image
+    MetasploitPayloads::Mettle.new('mips-linux-muslsf', generate_config(opts)).to_binary :process_image
   end
 end

modules/payloads/stages/linux/mipsle/mettle.rb

@@ -6,10 +6,12 @@
 require 'msf/core'
 require 'msf/base/sessions/meterpreter_mipsle_linux'
 require 'msf/base/sessions/meterpreter_options'
+require 'msf/base/sessions/mettle_config'
 require 'rex/elfparsey'
 
 module MetasploitModule
   include Msf::Sessions::MeterpreterOptions
+  include Msf::Sessions::MettleConfig
 
   def initialize(info = {})
     super(
@@ -91,8 +93,6 @@ def handle_intermediate_stage(conn, payload)
   end
 
   def generate_stage(opts = {})
-    opts[:uuid] ||= generate_payload_uuid
-    MetasploitPayloads::Mettle.new('mipsel-linux-muslsf', opts.slice(:uuid, :url, :debug, :log_file)).
-      to_binary :process_image
+    MetasploitPayloads::Mettle.new('mipsel-linux-muslsf', generate_config(opts)).to_binary :process_image
   end
 end

modules/payloads/stages/linux/x64/mettle.rb

@@ -6,10 +6,12 @@
 require 'msf/core'
 require 'msf/base/sessions/meterpreter_x64_mettle_linux'
 require 'msf/base/sessions/meterpreter_options'
+require 'msf/base/sessions/mettle_config'
 require 'rex/elfparsey'
 
 module MetasploitModule
   include Msf::Sessions::MeterpreterOptions
+  include Msf::Sessions::MettleConfig
 
   def initialize(info = {})
     super(
@@ -88,8 +90,6 @@ def handle_intermediate_stage(conn, payload)
   end
 
   def generate_stage(opts = {})
-    opts[:uuid] ||= generate_payload_uuid
-    MetasploitPayloads::Mettle.new('x86_64-linux-musl', opts.slice(:uuid, :url, :debug, :log_file)).
-      to_binary :process_image
+    MetasploitPayloads::Mettle.new('x86_64-linux-musl', generate_config(opts)).to_binary :process_image
   end
 end

modules/payloads/stages/linux/x86/mettle.rb

@@ -6,10 +6,12 @@
 require 'msf/core'
 require 'msf/base/sessions/meterpreter_x86_mettle_linux'
 require 'msf/base/sessions/meterpreter_options'
+require 'msf/base/sessions/mettle_config'
 require 'rex/elfparsey'
 
 module MetasploitModule
   include Msf::Sessions::MeterpreterOptions
+  include Msf::Sessions::MettleConfig
 
   def initialize(info = {})
     super(
@@ -91,8 +93,6 @@ def handle_intermediate_stage(conn, payload)
   end
 
   def generate_stage(opts = {})
-    opts[:uuid] ||= generate_payload_uuid
-    MetasploitPayloads::Mettle.new('i486-linux-musl', opts.slice(:uuid, :url, :debug, :log_file)).
-      to_binary :process_image
+    MetasploitPayloads::Mettle.new('i486-linux-musl', generate_config(opts)).to_binary :process_image
   end
 end