catch exception for patched Vista

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

@@ -358,7 +358,7 @@ module Exploit::Remote::SMB::Client::Psexec_MS17_010
     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 @@ module Exploit::Remote::SMB::Client::Psexec_MS17_010
 
     if @ctx['os'] == 'WIN7' and !(@ctx.key? 'arch')
       # leak_frag_size() can be used against Windows Vista/2008 to determine target architecture
+      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 = []