80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import logging
|
||
|
|
import traceback
|
||
|
|
import logging.config
|
||
|
|
from config import WormConfiguration
|
||
|
|
from model import MONKEY_ARG, DROPPER_ARG
|
||
|
|
from dropper import MonkeyDrops
|
||
|
|
from monkey import ChaosMonkey
|
||
|
|
|
||
|
|
__author__ = 'itamar'
|
||
|
|
|
||
|
|
LOG = None
|
||
|
|
|
||
|
|
LOG_CONFIG = {'version': 1,
|
||
|
|
'disable_existing_loggers': False,
|
||
|
|
'formatters': {'standard': {'format': '%(asctime)s [%(process)d:%(levelname)s] %(module)s.%(funcName)s.%(lineno)d: %(message)s'},
|
||
|
|
},
|
||
|
|
'handlers': {'console': {'class': 'logging.StreamHandler',
|
||
|
|
'level': 'DEBUG',
|
||
|
|
'formatter': 'standard'},
|
||
|
|
'file': {'class': 'logging.FileHandler',
|
||
|
|
'level': 'DEBUG',
|
||
|
|
'formatter': 'standard',
|
||
|
|
'filename': None}
|
||
|
|
},
|
||
|
|
'root': {'level': 'DEBUG',
|
||
|
|
'handlers': ['console']},
|
||
|
|
}
|
||
|
|
|
||
|
|
def main():
|
||
|
|
global LOG
|
||
|
|
|
||
|
|
if 2 > len(sys.argv):
|
||
|
|
return True
|
||
|
|
|
||
|
|
monkey_mode = sys.argv[1]
|
||
|
|
monkey_args = sys.argv[2:]
|
||
|
|
|
||
|
|
try:
|
||
|
|
if MONKEY_ARG == monkey_mode:
|
||
|
|
log_path = WormConfiguration.monkey_log_path
|
||
|
|
monkey_cls = ChaosMonkey
|
||
|
|
elif DROPPER_ARG == monkey_mode:
|
||
|
|
log_path = WormConfiguration.dropper_log_path
|
||
|
|
monkey_cls = MonkeyDrops
|
||
|
|
else:
|
||
|
|
return True
|
||
|
|
except ValueError:
|
||
|
|
return True
|
||
|
|
|
||
|
|
if WormConfiguration.use_file_logging:
|
||
|
|
LOG_CONFIG['handlers']['file']['filename'] = log_path
|
||
|
|
LOG_CONFIG['root']['handlers'].append('file')
|
||
|
|
|
||
|
|
logging.config.dictConfig(LOG_CONFIG)
|
||
|
|
LOG = logging.getLogger()
|
||
|
|
|
||
|
|
def log_uncaught_exceptions(ex_cls, ex, tb):
|
||
|
|
LOG.critical(''.join(traceback.format_tb(tb)))
|
||
|
|
LOG.critical('{0}: {1}'.format(ex_cls, ex))
|
||
|
|
|
||
|
|
sys.excepthook = log_uncaught_exceptions
|
||
|
|
|
||
|
|
LOG.info(">>>>>>>>>> Initializing monkey (%s): PID %s <<<<<<<<<<",
|
||
|
|
monkey_cls.__name__, os.getpid())
|
||
|
|
|
||
|
|
monkey = monkey_cls(monkey_args)
|
||
|
|
monkey.initialize()
|
||
|
|
|
||
|
|
try:
|
||
|
|
monkey.start()
|
||
|
|
return True
|
||
|
|
finally:
|
||
|
|
monkey.cleanup()
|
||
|
|
|
||
|
|
if "__main__" == __name__:
|
||
|
|
if not main():
|
||
|
|
sys.exit(1)
|