Add UUIDs to mettle stages

Author: acammack-r7

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

@@ -79,7 +79,9 @@ def handle_intermediate_stage(conn, payload)
     conn.put(midstager) == midstager.length
   end
 
-  def generate_stage(_opts = {})
-    MetasploitPayloads::Mettle.read('armv5l-linux-musleabi', 'mettle.bin')
+  def generate_stage(opts = {})
+    opts[:uuid] ||= generate_payload_uuid
+    MetasploitPayloads::Mettle.new('armv5l-linux-musleabi', opts.slice(:uuid, :url, :debug, :log_file)).
+      to_bininary :process_image
   end
 end

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

@@ -90,7 +90,9 @@ def handle_intermediate_stage(conn, payload)
     conn.put(midstager) == midstager.length
   end
 
-  def generate_stage(_opts = {})
-    MetasploitPayloads::Mettle.read('mips-linux-muslsf', 'mettle.bin')
+  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
   end
 end

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

@@ -90,7 +90,9 @@ def handle_intermediate_stage(conn, payload)
     conn.put(midstager) == midstager.length
   end
 
-  def generate_stage(_opts = {})
-    MetasploitPayloads::Mettle.read('mipsel-linux-muslsf', 'mettle.bin')
+  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
   end
 end

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

@@ -87,7 +87,9 @@ def handle_intermediate_stage(conn, payload)
     conn.put(midstager) == midstager.length
   end
 
-  def generate_stage(_opts = {})
-    MetasploitPayloads::Mettle.read('x86_64-linux-musl', 'mettle.bin')
+  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
   end
 end

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

@@ -90,7 +90,9 @@ def handle_intermediate_stage(conn, payload)
     conn.put(midstager) == midstager.length
   end
 
-  def generate_stage(_opts = {})
-    MetasploitPayloads::Mettle.read('i486-linux-musl', 'mettle.bin')
+  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
   end
 end