Add tab completion to the payload generate command

Author: zeroSteiner

lib/msf/ui/console/command_dispatcher/payload.rb

@@ -13,7 +13,7 @@ class Payload
           include Msf::Ui::Console::ModuleCommandDispatcher
 
           # Load supported formats
-          supported_formats = \
+          @@supported_formats = \
             Msf::Simple::Buffer.transform_formats + \
             Msf::Util::EXE.to_executable_fmt_formats
 
@@ -25,7 +25,7 @@ class Payload
             "-o" => [ true,  "A comma separated list of options in VAR=VAL format." ],
             "-s" => [ true,  "NOP sled length."                                     ],
             "-f" => [ true,  "The output file name (otherwise stdout)"              ],
-            "-t" => [ true,  "The output format: #{supported_formats.join(',')}"    ],
+            "-t" => [ true,  "The output format: #{@@supported_formats.join(',')}"    ],
             "-p" => [ true,  "The Platform for output."                             ],
             "-k" => [ false, "Keep the template executable functional"              ],
             "-x" => [ true,  "The executable template to use"                       ],
@@ -151,6 +151,25 @@ def cmd_generate(*args)
             end
             true
           end
+
+          def cmd_generate_tabs(str, words)
+            last_word = words[-1]
+            fmt = @@generate_opts.fmt
+            fmt = fmt.select { |key, value| last_word == key || !words.include?(key) }
+
+            option = fmt[last_word]
+            return fmt.keys if !option || !option[0]
+
+            tabs = []
+            case last_word
+              when '-e'
+                tabs = framework.encoders.map { |refname, mod| refname }
+              when '-f'
+                tabs = tab_complete_filenames(str, words)
+              when '-t'
+                tabs = @@supported_formats
+            end
+          end
         end
       end
     end