From effdb620c6da435fc7b1043da7761725990a4732 Mon Sep 17 00:00:00 2001 From: lanjelot Date: Fri, 29 Jun 2012 17:05:07 +1000 Subject: [PATCH] added a timeout option for mysql_login --- patator.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/patator.py b/patator.py index bb90925..ffc667a 100755 --- a/patator.py +++ b/patator.py @@ -2194,18 +2194,20 @@ class MySQL_login: ('port', 'ports to target [3306]'), ('user', 'usernames to test'), ('password', 'passwords to test'), + ('timeout', 'seconds to wait for a response [10]'), ) available_actions = () Response = Response_Base - def execute(self, host, port='3306', user='anony', password=''): + def execute(self, host, port='3306', user='anony', password='', timeout='10'): try: - fp = _mysql.connect(host=host, port=int(port), user=user, passwd=password) + fp = _mysql.connect(host=host, port=int(port), user=user, passwd=password, connect_timeout=int(timeout)) resp = '0', fp.get_server_info() - except _mysql.Error as resp: pass + except _mysql.Error as resp: + logger.debug('MysqlError: %s' % resp) code, mesg = resp return self.Response(code, mesg)