Land rapid7#5344, print payload size from msfvenom

Author: Brent Cook

lib/msf/core/payload_generator.rb

@@ -276,12 +276,15 @@ def generate_java_payload
     # @return [String] A string containing the bytes of the payload in the format selected
     def generate_payload
       if platform == "java" or arch == "java" or payload.start_with? "java/"
-        generate_java_payload
+        p = generate_java_payload
+        cli_print "Payload size: #{p.length} bytes"
+        p
       else
         raw_payload = generate_raw_payload
         raw_payload = add_shellcode(raw_payload)
         encoded_payload = encode_payload(raw_payload)
         encoded_payload = prepend_nops(encoded_payload)
+        cli_print "Payload size: #{encoded_payload.length} bytes"
         format_payload(encoded_payload)
       end
     end

spec/lib/msf/core/payload_generator_spec.rb

@@ -606,7 +606,7 @@
       }
 
       it 'calls generate_java_payload' do
-        payload_generator.should_receive(:generate_java_payload)
+        payload_generator.should_receive(:generate_java_payload).and_call_original
         payload_generator.generate_payload
       end
     end