forked from koutto/jok3r
-
Notifications
You must be signed in to change notification settings - Fork 0
/
jok3r.py
64 lines (52 loc) · 1.78 KB
/
jok3r.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
###
### Jok3r main function
###
import sys
import traceback
from lib.core.ArgumentsParser import ArgumentsParser
from lib.core.Config import *
from lib.core.Exceptions import *
from lib.core.Settings import Settings
from lib.controller.MainController import MainController
from lib.db.Mission import Mission
from lib.db.Session import *
from lib.output.Logger import logger
class Program:
def __init__(self):
try:
print(BANNER)
# Parse settings files
settings = Settings()
# Parse command-line arguments
arguments = ArgumentsParser(settings)
# Create db if needed and initialize sqlalchemy session
Base.metadata.create_all(engine)
session = Session()
# Create "default" mission if necessary
mission = session.query(Mission).filter(Mission.name == 'default').first()
if not mission:
mission = Mission(name='default', comment='Default scope')
session.add(mission)
session.commit()
# Controller
controller = MainController(arguments, settings, session)
controller.run()
except KeyboardInterrupt:
print()
logger.error('Ctrl+C received ! User aborted')
sys.exit(0)
except (SettingsException, AttackException) as e:
logger.error(e)
sys.exit(1)
except (ValueError, ArgumentsException):
print
sys.exit(1)
except Exception as e:
print
logger.error('Unexpected error occured: {0}'.format(str(e)))
traceback.print_exc()
sys.exit(1)
if __name__ == '__main__':
main = Program()