Land #4876, @hmoore-r7 give encoders and payloads space available

lib/msf/base/simple/payload.rb

@@ -51,12 +51,13 @@ module Payload
 
     # Generate the payload
     e = EncodedPayload.create(payload,
-        'BadChars' => opts['BadChars'],
-        'MinNops'  => opts['NopSledSize'],
-        'Encoder'  => opts['Encoder'],
+        'BadChars'    => opts['BadChars'],
+        'MinNops'     => opts['NopSledSize'],
+        'Encoder'     => opts['Encoder'],
         'Iterations'  => opts['Iterations'],
         'ForceEncode' => opts['ForceEncode'],
-        'Space'    => opts['MaxSize'])
+        'DisableNops' => opts['DisableNops'],
+        'Space'       => opts['MaxSize'])
 
     fmt = opts['Format'] || 'raw'
 

lib/msf/core/encoded_payload.rb

@@ -34,6 +34,7 @@ class EncodedPayload
     self.framework = framework
     self.pinst     = pinst
     self.reqs      = reqs
+    self.space     = reqs['Space']
   end
 
   #
@@ -64,6 +65,9 @@ class EncodedPayload
       # First, validate
       pinst.validate()
 
+      # Tell the payload how much space is available
+      pinst.available_space = self.space
+
       # Generate the raw version of the payload first
       generate_raw() if self.raw.nil?
 
@@ -191,6 +195,9 @@ class EncodedPayload
           next
         end
 
+        # Tell the encoder how much space is available
+        self.encoder.available_space = self.space
+
         eout = self.raw.dup
 
         next_encoder = false
@@ -456,7 +463,10 @@ class EncodedPayload
   # The number of encoding iterations used
   #
   attr_reader :iterations
-
+  #
+  # The maximum number of bytes acceptable for the encoded payload
+  #
+  attr_reader :space
 protected
 
   attr_writer :raw # :nodoc:
@@ -467,6 +477,7 @@ protected
   attr_writer :encoder # :nodoc:
   attr_writer :nop # :nodoc:
   attr_writer :iterations # :nodoc:
+  attr_writer :space # :nodoc
 
   #
   # The payload instance used to generate the payload

lib/msf/core/encoder.rb

@@ -434,6 +434,12 @@ class Encoder < Module
     false
   end
 
+  #
+  # The amount of space available to the encoder, which may be nil,
+  # indicating that the smallest possible encoding should be used.
+  #
+  attr_accessor :available_space
+
 protected
 
   #

lib/msf/core/payload.rb

@@ -500,6 +500,12 @@ class Payload < Msf::Module
   #
   attr_accessor :assoc_exploit
 
+  #
+  # The amount of space available to the payload, which may be nil,
+  # indicating that the smallest possible payload should be used.
+  #
+  attr_accessor :available_space
+
 protected
 
   #

lib/msf/core/payload_generator.rb

@@ -184,6 +184,7 @@ module Msf
         encoder_list.each do |encoder_mod|
           cli_print "Attempting to encode payload with #{iterations} iterations of #{encoder_mod.refname}"
           begin
+            encoder_mod.available_space = @space
             return run_encoder(encoder_mod, shellcode.dup)
           rescue ::Msf::EncoderSpaceViolation => e
             cli_print "#{encoder_mod.refname} failed with #{e.message}"
@@ -298,9 +299,11 @@ module Msf
         end
 
         payload_module.generate_simple(
-            'Format'   => 'raw',
-            'Options'  => datastore,
-            'Encoder'  => nil
+            'Format'      => 'raw',
+            'Options'     => datastore,
+            'Encoder'     => nil,
+            'MaxSize'     => @space,
+            'DisableNops' => true
         )
       end
     end