Set the arch while generating

Author: zeroSteiner

lib/msf/core/payload/windows/encrypted_reverse_tcp.rb

@@ -76,7 +76,7 @@ def generate(opts={})
       keep_exe:      datastore['KeepExe'],
       show_compile_cmd: datastore['ShowCompileCMD'],
       f_name:        Tempfile.new(staged? ? 'reverse_pic_stager' : 'reverse_pic_stageless').path,
-      arch:          self.arch_to_s
+      arch:          opts.fetch(:arch, self.arch_to_s)
     }
 
     comp_code = get_compiled_shellcode(src, compile_opts)

modules/payloads/adapters/cmd/windows/powershell.rb

@@ -34,7 +34,8 @@ def compatible?(mod)
     super
   end
 
-  def generate
+  def generate(opts = {})
+    opts[:arch] ||= module_info['AdaptedArch']
     payload = super
 
     cmd_psh_payload(payload, ARCH_X86, remove_comspec: true)

modules/payloads/adapters/cmd/windows/powershell/x64.rb

@@ -34,7 +34,8 @@ def compatible?(mod)
     super
   end
 
-  def generate
+  def generate(opts = {})
+    opts[:arch] ||= module_info['AdaptedArch']
     payload = super
 
     cmd_psh_payload(payload, ARCH_X64, remove_comspec: true)