Handle exceptions in mysql_login

Author: sinn3r

lib/msf/core/exploit/mysql.rb

@@ -40,16 +40,32 @@ def mysql_login(user='root', pass='', db=nil)
 		disconnect if self.sock
 		connect
 
-		@mysql_handle = ::RbMysql.connect({
-			:host           => rhost,
-			:port           => rport,
-			:read_timeout   => 300,
-			:write_timeout  => 300,
-			:socket         => sock,
-			:user           => user,
-			:password       => pass,
-			:db             => db
-		})
+		begin
+			@mysql_handle = ::RbMysql.connect({
+				:host           => rhost,
+				:port           => rport,
+				:read_timeout   => 300,
+				:write_timeout  => 300,
+				:socket         => sock,
+				:user           => user,
+				:password       => pass,
+				:db             => db
+			})
+		rescue Errno::ECONNREFUSED
+			vprint_error("Connection refused")
+			return false
+		rescue RbMysql::ClientError
+			vprint_error("Connection timedout")
+			return false
+		rescue Errno::ETIMEDOUT
+			vprint_error("Operation timedout")
+			return false
+		rescue RbMysql::AccessDeniedError
+			vprint_error("Access denied")
+			return false
+		end
+
+		return true
 	end
 
 	def mysql_logoff