catch exception for patched Vista

Author: zerosum0x0

lib/msf/core/exploit/smb/client/psexec_ms17_010.rb

@@ -358,7 +358,7 @@ def exploit_matched_pairs(pipe_handle)
     begin
       leak_frag_size(pipe_handle.file_id)
     rescue TypeError => e
-      raise MS17_010_Error 'TypeError leaking initial Frag size, is the target patched?'
+      raise MS17_010_Error, 'TypeError leaking initial Frag size, is the target patched?'
     end
 
     # we have all info for offsets now
@@ -604,7 +604,11 @@ def exploit_fish_barrel(pipe_handle)
 
     if @ctx['os'] == 'WIN7' and !(@ctx.key? 'arch')
       # leak_frag_size() can be used against Windows Vista/2008 to determine target architecture
-      leak_frag_size(pipe_handle.file_id)
+      begin
+        leak_frag_size(pipe_handle.file_id)
+      rescue Timeout::Error
+        raise MS17_010_Error, 'Timeout::Error leaking initial Frag size, is the target patched?'
+      end
     end
 
     attempt_list = []