rescue NoCompatiblePayloadError inside size method, fixes #1604

git-svn-id: file:///home/svn/framework3/trunk@10051 4d416f70-5f16-0410-b530-b9f4589650da
2010-08-19 00:10:06 +00:00 · 2010-08-19 00:10:06 +00:00 · 7d560e9c18
parent 48e91af25d
commit 7d560e9c18
1 changed files with 6 additions and 1 deletions
--- a/lib/msf/core/payload.rb
+++ b/lib/msf/core/payload.rb
@ -160,7 +160,12 @@ class Payload < Msf::Module
 	# first stage is returned.
 	#
 	def size
-		return (generate() || '').length
+		pl = ''
+		begin
+			pl = generate()
+		rescue NoCompatiblePayloadError
+		end
+		pl.length
 	end

 	#