Merge pull request #135 from DomainTools/domaintools-patch-1

Added code to allow 3rd party modules
pull/139/head
Raphaël Vinot 2017-09-06 16:34:06 +02:00 committed by GitHub
commit 2f6406b1e5
1 changed files with 6 additions and 0 deletions

View File

@ -217,6 +217,7 @@ def main():
argParser.add_argument('-d', default=False, action='store_true', help='Enable debugging') argParser.add_argument('-d', default=False, action='store_true', help='Enable debugging')
argParser.add_argument('-p', default=6666, help='misp-modules TCP port (default 6666)') argParser.add_argument('-p', default=6666, help='misp-modules TCP port (default 6666)')
argParser.add_argument('-l', default='localhost', help='misp-modules listen address (default localhost)') argParser.add_argument('-l', default='localhost', help='misp-modules listen address (default localhost)')
argParser.add_argument('-m', default=[], action='append', help='Register a custom module')
args = argParser.parse_args() args = argParser.parse_args()
port = args.p port = args.p
listen = args.l listen = args.l
@ -232,6 +233,11 @@ def main():
helpersdir = 'helpers' helpersdir = 'helpers'
load_helpers(helpersdir=helpersdir) load_helpers(helpersdir=helpersdir)
mhandlers, loaded_modules = load_modules(modulesdir) mhandlers, loaded_modules = load_modules(modulesdir)
for module in args.m:
mispmod = importlib.import_module(module)
mispmod.register(mhandlers, loaded_modules)
service = [(r'/modules', ListModules), (r'/query', QueryModule)] service = [(r'/modules', ListModules), (r'/query', QueryModule)]
application = tornado.web.Application(service) application = tornado.web.Application(service)