Archives/patator
2
0
Fork 0
mirror of https://github.com/lanjelot/patator synced 2024-11-10 01:13:31 +00:00
Code Issues Projects Releases Wiki Activity

added a timeout option for pgsql_login

Browse Source
This commit is contained in:
lanjelot 2012-07-04 17:49:38 +10:00
parent 3c3539feeb
commit 7600bd3ffe
1 changed files with 11 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
patator.py
View File

@ -124,7 +124,7 @@ cx_Oracle | Oracle | http://cx-oracle.sourceforge.net/
-------------------------------------------------------------------------------------------------- --------------------------------------------------------------------------------------------------
mysql-python | MySQL | http://sourceforge.net/projects/mysql-python/ | 1.2.3 | mysql-python | MySQL | http://sourceforge.net/projects/mysql-python/ | 1.2.3 |
-------------------------------------------------------------------------------------------------- --------------------------------------------------------------------------------------------------
psycopg | PostgreSQL | http://initd.org/psycopg/ | 2.4.1 | psycopg | PostgreSQL | http://initd.org/psycopg/ | 2.4.5 |
-------------------------------------------------------------------------------------------------- --------------------------------------------------------------------------------------------------
pycrypto | VNC | http://www.dlitz.net/software/pycrypto/ | 2.3 | pycrypto | VNC | http://www.dlitz.net/software/pycrypto/ | 2.3 |
-------------------------------------------------------------------------------------------------- --------------------------------------------------------------------------------------------------
@ -1068,7 +1068,7 @@ Please read the README inside for more examples and usage information.
self.fail_count = 0 self.fail_count = 0
self.seconds = [1]*25 # avoid division by zero early bug condition self.seconds = [1]*25 # avoid division by zero early bug condition
gqueues = [Queue(maxsize=10000) for i in range(self.num_threads)] gqueues = [Queue(maxsize=10000) for _ in range(self.num_threads)]
# consumers # consumers
for num in range(self.num_threads): for num in range(self.num_threads):
@ -1506,7 +1506,7 @@ class FTP_login(TCP_Cache):
('port', 'ports to target [21]'), ('port', 'ports to target [21]'),
('user', 'usernames to test'), ('user', 'usernames to test'),
('password', 'passwords to test'), ('password', 'passwords to test'),
('timeout', 'seconds to wait for a FTP response [10]'), ('timeout', 'seconds to wait for a response [10]'),
) )
available_options += TCP_Cache.available_options available_options += TCP_Cache.available_options
@ -1876,7 +1876,7 @@ class Finger_lookup:
('host', 'hostnames or subnets to target'), ('host', 'hostnames or subnets to target'),
('port', 'ports to target [79]'), ('port', 'ports to target [79]'),
('user', 'usernames to test'), ('user', 'usernames to test'),
('timeout', 'seconds to wait on socket operations [5]'), ('timeout', 'seconds to wait for a response [5]'),
) )
available_actions = () available_actions = ()
@ -2387,16 +2387,18 @@ class Pgsql_login:
('user', 'usernames to test'), ('user', 'usernames to test'),
('password', 'passwords to test'), ('password', 'passwords to test'),
('database', 'databases to test [postgres]'), ('database', 'databases to test [postgres]'),
('timeout', 'seconds to wait for a response [10]'),
) )
available_actions = () available_actions = ()
Response = Response_Base Response = Response_Base
def execute(self, host, port='5432', user=None, password=None, database='postgres', ssl='disable'): def execute(self, host, port='5432', user=None, password=None, database='postgres', ssl='disable', timeout='10'):
try: try:
psycopg2.connect(host=host, port=int(port), user=user, password=password, database=database, sslmode=ssl) psycopg2.connect(host=host, port=int(port), user=user, password=password, database=database, sslmode=ssl, connect_timeout=int(timeout))
code, mesg = '0', 'OK' code, mesg = '0', 'OK'
except psycopg2.OperationalError as e: except psycopg2.OperationalError as e:
logger.debug('OperationalError: %s' % e)
code, mesg = '1', str(e)[:-1] code, mesg = '1', str(e)[:-1]
return self.Response(code, mesg) return self.Response(code, mesg)
@ -2872,7 +2874,7 @@ class Controller_DNS(Controller):
print('Records ' + '-'*42) print('Records ' + '-'*42)
for name, infos in sorted(self.records.items()): for name, infos in sorted(self.records.items()):
for qclass, qtype, rdata in infos: for qclass, qtype, rdata in infos:
print('%34s %8s %-8s %s' % (name, qclass, qtype, rdata)) print('%34s %4s %-7s %s' % (name, qclass, qtype, rdata))
''' Expected output: ''' Expected output:
Hostmap ------ Hostmap ------
@ -3008,7 +3010,7 @@ class DNS_reverse:
available_options = ( available_options = (
('host', 'IP addresses to reverse lookup'), ('host', 'IP addresses to reverse lookup'),
('server', 'name server to query (directly asking a zone authoritative NS may return more results) [8.8.8.8]'), ('server', 'name server to query (directly asking a zone authoritative NS may return more results) [8.8.8.8]'),
('timeout', 'seconds to wait for a DNS response [5]'), ('timeout', 'seconds to wait for a response [5]'),
('protocol', 'send queries over udp or tcp [udp]'), ('protocol', 'send queries over udp or tcp [udp]'),
) )
available_actions = () available_actions = ()
@ -3042,7 +3044,7 @@ class DNS_forward:
available_options = ( available_options = (
('name', 'domain names to lookup'), ('name', 'domain names to lookup'),
('server', 'name server to query (directly asking the zone authoritative NS may return more results) [8.8.8.8]'), ('server', 'name server to query (directly asking the zone authoritative NS may return more results) [8.8.8.8]'),
('timeout', 'seconds to wait for a DNS response [5]'), ('timeout', 'seconds to wait for a response [5]'),
('protocol', 'send queries over udp or tcp [udp]'), ('protocol', 'send queries over udp or tcp [udp]'),
('qtype', 'type to query [ANY]'), ('qtype', 'type to query [ANY]'),
('qclass', 'class to query [IN]'), ('qclass', 'class to query [IN]'),