add debugging info to connection
parent
6a1742dc20
commit
d7aa98b4d8
|
@ -124,31 +124,48 @@ class connection(object):
|
|||
|
||||
def call_modules(self):
|
||||
for module in self.module:
|
||||
self.logger.debug(f"Loading module {module}")
|
||||
module_logger = CMEAdapter(extra={
|
||||
'module': module.name.upper(),
|
||||
'host': self.host,
|
||||
'port': self.args.port,
|
||||
'hostname': self.hostname
|
||||
})
|
||||
self.logger.debug(f"Loading module {module.name} - {module}")
|
||||
module_logger = CMEAdapter(
|
||||
extra={
|
||||
'module': module.name.upper(),
|
||||
'host': self.host,
|
||||
'port': self.args.port,
|
||||
'hostname': self.hostname
|
||||
},
|
||||
)
|
||||
|
||||
self.logger.debug(f"Loading context for module {module.name} - {module}")
|
||||
context = Context(self.db, module_logger, self.args)
|
||||
context.localip = self.local_ip
|
||||
|
||||
try:
|
||||
if hasattr(module, 'on_request') or hasattr(module, 'has_response'):
|
||||
self.logger.debug(f"Module {module.name} has on_request or has_response methods")
|
||||
self.server.connection = self
|
||||
self.server.context.localip = self.local_ip
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error while calling {module.name}'s on_request or has_response methods: {e}")
|
||||
|
||||
try:
|
||||
if hasattr(module, 'on_login'):
|
||||
self.logger.debug(f"Module {module.name} has on_login method")
|
||||
module.on_login(context, self)
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error while calling {module.name}'s on_login method: {e}")
|
||||
|
||||
try:
|
||||
if self.admin_privs and hasattr(module, 'on_admin_login'):
|
||||
self.logger.debug(f"Module {module.name} has on_admin_login method")
|
||||
module.on_admin_login(context, self)
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error while calling {module.name}'s on_admin_login method: {e}")
|
||||
|
||||
try:
|
||||
if (not hasattr(module, 'on_request') and not hasattr(module, 'has_response')) and hasattr(module,'on_shutdown'):
|
||||
self.logger.debug(f"Module {module.name} has on_shutdown method")
|
||||
module.on_shutdown(context, self)
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error while loading module {module}: {e}")
|
||||
self.logger.error(f"Error while calling {module.name}'s on_shutdown method: {e}")
|
||||
pass
|
||||
|
||||
def inc_failed_login(self, username):
|
||||
|
|
Loading…
Reference in New Issue