mirror of
https://github.com/ail-project/ail-framework.git
synced 2024-11-14 02:28:23 +00:00
1379ef705a
AIL is a modular framework to analyse potential information leak from unstructured data source like pastes from Past ebin or similar services. AIL framework is flexible and can be extended to support other functionalities to mine sen sitive information
45 lines
1.6 KiB
Python
Executable file
45 lines
1.6 KiB
Python
Executable file
#!/usr/bin/env python2
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import argparse
|
|
import signal
|
|
from pubsublogger import subscriber
|
|
|
|
|
|
def signal_handler(signal, frame):
|
|
if subscriber.pubsub is not None:
|
|
subscriber.stop()
|
|
print "Subscriber closed."
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
if __name__ == '__main__':
|
|
|
|
parser = argparse.ArgumentParser(description='Configure a logging subscriber.')
|
|
|
|
parser.add_argument("-H", "--hostname", default='localhost',
|
|
type=str, help='Set the hostname of the server.')
|
|
parser.add_argument("-p", "--port", default=6379,
|
|
type=int, help='Set the server port.')
|
|
parser.add_argument('-s', '--use_unix_socket', action='store_true',
|
|
help='Use a unix socket path instead of a tcp socket.')
|
|
parser.add_argument("--unix_socket_path", default='/tmp/redis.sock',
|
|
type=str, help='Unix socket path.')
|
|
parser.add_argument("-c", "--channel",
|
|
type=str, required=True, help='Channel to subscribe to.')
|
|
parser.add_argument("-l", "--log_path",
|
|
required=True, help='Path where the logs will be written')
|
|
parser.add_argument("-d", "--debug", action="store_true",
|
|
help='Also log debug messages.')
|
|
parser.add_argument("-m", "--mail", type=file, default=None,
|
|
help='Path to the config file used to send errors by email.')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.use_unix_socket:
|
|
subscriber.unix_socket = args.unix_socket_path
|
|
else:
|
|
subscriber.hostname = args.hostname
|
|
subscriber.port = args.port
|
|
subscriber.run(args.channel, args.log_path, args.debug, args.mail)
|
|
|