From 98aba35b295219e91f1d7a993eea8e7df3c2f6a1 Mon Sep 17 00:00:00 2001
From: William Vu <William_Vu@rapid7.com>
Date: Thu, 23 Aug 2018 21:53:49 -0500
Subject: [PATCH] Add msfvenom -l arches like -l platforms

---
 msfvenom | 30 ++++++++++++++++++++++++++----
 1 file changed, 26 insertions(+), 4 deletions(-)

diff --git a/msfvenom b/msfvenom
index 4eee811a84..ce7ec4fd7d 100755
--- a/msfvenom
+++ b/msfvenom
@@ -69,7 +69,7 @@ def parse_args(args)
   opt.separator('')
   opt.separator('Options:')
 
-  opt.on('-l', '--list            <type>', Array, 'List all modules for [type]. Types are: payloads, encoders, nops, platforms, encrypt, formats, all') do |l|
+  opt.on('-l', '--list            <type>', Array, 'List all modules for [type]. Types are: payloads, encoders, nops, platforms, arches, encrypt, formats, all') do |l|
     if l.to_s.empty?
       l = ["all"]
     end
@@ -113,7 +113,7 @@ def parse_args(args)
     opts[:encryption_iv] = e
   end
 
-  opt.on('-a', '--arch            <arch>', String, 'The architecture to use for --payload and --encoders') do |a|
+  opt.on('-a', '--arch            <arch>', String, 'The architecture to use for --payload and --encoders (use --list arches to list)') do |a|
     opts[:arch] = a
   end
 
@@ -239,7 +239,26 @@ def dump_platforms
     ])
 
   supported_platforms.sort.each do |name|
-    tbl << [ name]
+    tbl << [name]
+  end
+
+  "\n" + tbl.to_s + "\n"
+end
+
+def dump_arches
+  init_framework(:module_types => [])
+  supported_arches = ARCH_ALL.dup
+
+  tbl = Rex::Text::Table.new(
+    'Indent'  => 4,
+    'Header'  => "Framework Architectures [--arch <value>]",
+    'Columns' =>
+    [
+      "Name",
+    ])
+
+  supported_arches.sort.each do |name|
+    tbl << [name]
   end
 
   "\n" + tbl.to_s + "\n"
@@ -371,6 +390,8 @@ if generator_opts[:list]
       $stdout.puts dump_nops
     when "platforms", "dump_platform"
       $stdout.puts dump_platforms
+    when "arches", "dump_arch"
+      $stdout.puts dump_arches
     when "encrypts", "encrypt", "encryption"
       $stdout.puts dump_encrypt
     when "formats", "format", "f"
@@ -383,10 +404,11 @@ if generator_opts[:list]
       $stdout.puts dump_encoders
       $stdout.puts dump_nops
       $stdout.puts dump_platforms
+      $stdout.puts dump_arches
       $stdout.puts dump_encrypt
       $stdout.puts dump_formats
     else
-      $stderr.puts "Invalid type (#{mod}). These are valid: payloads, encoders, nops, platforms, encrypt, formats, all"
+      $stderr.puts "Invalid type (#{mod}). These are valid: payloads, encoders, nops, platforms, arches, encrypt, formats, all"
     end
   end
   exit(0)