byt3bl33d3r 2017-04-14 15:26:17 -06:00
parent 0390529cf1
commit e8def0ad4c
1 changed files with 7 additions and 1 deletions

View File

@ -1,4 +1,5 @@
import logging
from socket import gethostbyname
from traceback import format_exc
from gevent.lock import BoundedSemaphore
from functools import wraps
@ -20,7 +21,6 @@ class connection(object):
def __init__(self, args, db, host):
self.args = args
self.db = db
self.host = host
self.conn = None
self.admin_privs = False
self.hostname = None
@ -30,6 +30,12 @@ class connection(object):
self.failed_logins = 0
self.local_ip = None
try:
self.host = gethostbyname(host)
except Exception as e:
logging.debug('Error resolving hostname {}: {}'.format(host, e))
return
self.proto_flow()
@staticmethod