Change StageEncoder behaviour

Following rapid7#3770, some
change have been done as multiple encoders is managed by encoded_payload
to be used by Encoder and StageEncoder in the same time.

Author: agix

lib/msf/core/payload/stager.rb

@@ -17,7 +17,7 @@ def initialize(info={})
         Msf::OptBool.new("EnableStageEncoding", [ false, "Encode the second stage payload", false ]),
         Msf::OptString.new("StageEncoder", [ false, "Encoder to use if EnableStageEncoding is set", nil ]),
         Msf::OptString.new("StageEncoderSaveRegisters", [ false, "Additional registers to preserve in the staged payload if EnableStageEncoding is set", "" ]),
-        Msf::OptBool.new("StageEncodingFallback", [ false, "Fallback to default encoders or no encoding if the selected StageEncoder is not compatible", true ])
+        Msf::OptBool.new("StageEncodingFallback", [ false, "Fallback to no encoding if the selected StageEncoder is not compatible", true ])
       ], Msf::Payload::Stager)
 
   end
@@ -221,22 +221,11 @@ def encode_stage_preserved_registers
   # @return [String] Encoded version of +stg+
   def encode_stage(stg)
     return stg unless encode_stage?
-    stage_enc_mod = []
+    stage_enc_mod = nil
 
     # Handle StageEncoder if specified by the user
     if datastore['StageEncoder'].to_s.length > 0
-      # Allow multiple encoders separated by commas
-      stage_enc_mod = datastore["StageEncoder"].split(',').map(&:strip).select{|x| x.to_s.length > 0}.uniq
-    end
-
-    # Add automatic encoding as a fallback if needed
-    if datastore['StageEncodingFallback']
-      stage_enc_mod << nil
-    end
-
-    # If fallback has been disabled and no encoder was parsed, exit early and rop the session
-    if stage_enc_mod.length == 0
-      raise RuntimeError, "StageEncoder is invalid and StageEncodingFallback is disabled"
+      stage_enc_mod = datastore["StageEncoder"]
     end
 
     # Allow the user to specify additional registers to preserve
@@ -247,34 +236,32 @@ def encode_stage(stg)
     saved_registers.strip!
 
     estg = nil
-
-    stage_enc_mod.each do |encoder_refname_from_user|
-
+    begin
       # Generate an encoded version of the stage.  We tell the encoding system
       # to save certain registers to ensure that it does not get clobbered.
       encp = Msf::EncodedPayload.create(
         self,
         'Raw'                => stg,
-        'Encoder'            => encoder_refname_from_user,
+        'Encoder'            => stage_enc_mod,
         'EncoderOptions'     => { 'SaveRegisters' => saved_registers },
         'ForceSaveRegisters' => true,
         'ForceEncode'        => true)
 
       if encp.encoder
-        print_status("Encoded stage with #{encp.encoder.refname}")
+        if stage_enc_mod
+          print_status("Encoded stage with #{stage_enc_mod}")
+        else
+          print_status("Encoded stage with #{encp.encoder.refname}")
+        end
         estg = encp.encoded
-
-        break
       end
-    end
-
-    if datastore['StageEncodingFallback'] && estg.nil?
-      print_warning("StageEncoder failed, falling back to no encoding")
-      estg = stg
-    end
-
-    unless estg
-      raise RuntimeError, "Stage encoding failed and StageEncodingFallback is disabled"
+    rescue
+      if datastore['StageEncodingFallback'] && estg.nil?
+        print_warning("StageEncoder failed, falling back to no encoding")
+        estg = stg
+      else
+        raise RuntimeError, "Stage encoding failed and StageEncodingFallback is disabled"
+      end
     end
 
     estg