Rooba
/
cowrie
mirror of https://github.com/cowrie/cowrie.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

wip

This commit is contained in:
Michel Oosterhof 2016-10-18 15:07:59 +04:00
parent d65f7ac7aa
commit 414569c1ea
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
twisted/plugins/cowrie_plugin.py
View File

@ -37,7 +37,7 @@ from zope.interface import implementer
import os import os
import sys import sys
from twisted.python import usage from twisted.python import log, usage
from twisted.plugin import IPlugin from twisted.plugin import IPlugin
from twisted.application.service import IServiceMaker from twisted.application.service import IServiceMaker
from twisted.application import internet, service from twisted.application import internet, service
@ -72,6 +72,7 @@ class CowrieServiceMaker(object):
options = Options options = Options
dbloggers = None dbloggers = None
output_plugins = None output_plugins = None
cfg = None
def makeService(self, options): def makeService(self, options):
""" """
@ -105,13 +106,13 @@ class CowrieServiceMaker(object):
# Load db loggers # Load db loggers
self.dbloggers = [] self.dbloggers = []
for x in self.cfg.sections(): for x in cfg.sections():
if not x.startswith('database_'): if not x.startswith('database_'):
continue continue
engine = x.split('_')[1] engine = x.split('_')[1]
try: try:
dblogger = __import__( 'cowrie.dblog.{}'.format(engine), dblogger = __import__( 'cowrie.dblog.{}'.format(engine),
globals(), locals(), ['dblog']).DBLogger(self.cfg) globals(), locals(), ['dblog']).DBLogger(cfg)
log.addObserver(dblogger.emit) log.addObserver(dblogger.emit)
self.dbloggers.append(dblogger) self.dbloggers.append(dblogger)
log.msg("Loaded dblog engine: {}".format(engine)) log.msg("Loaded dblog engine: {}".format(engine))
@ -121,13 +122,13 @@ class CowrieServiceMaker(object):
# Load output modules # Load output modules
self.output_plugins = [] self.output_plugins = []
for x in self.cfg.sections(): for x in cfg.sections():
if not x.startswith('output_'): if not x.startswith('output_'):
continue continue
engine = x.split('_')[1] engine = x.split('_')[1]
try: try:
output = __import__( 'cowrie.output.{}'.format(engine), output = __import__( 'cowrie.output.{}'.format(engine),
globals(), locals(), ['output']).Output(self.cfg) globals(), locals(), ['output']).Output(cfg)
log.addObserver(output.emit) log.addObserver(output.emit)
self.output_plugins.append(output) self.output_plugins.append(output)
log.msg("Loaded output engine: {}".format(engine)) log.msg("Loaded output engine: {}".format(engine))