def main():
try:
paths['ROOT_PATH'] = os.path.dirname(
os.path.dirname(os.path.realpath(__file__)))
setPaths()
parse_args()
if IS_WIN:
win_color_init()
banner()
if conf['DEBUG']:
showDebugData()
if conf['UPDATE']:
update()
load_payloads()
if conf['ENGINE'] is 't':
from lib.controller.threads import ThreadsEngine
ThreadsEngine().run()
elif conf['ENGINE'] is 'c':
from lib.controller.coroutine import CoroutineEngine
CoroutineEngine().run()
except KeyboardInterrupt, e:
logger.log(CUSTOM_LOGGING.ERROR, 'User quit!')
def main():
try:
paths['ROOT_PATH'] = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
setPaths()
parse_args()
if IS_WIN:
win_color_init()
banner()
if conf['DEBUG']:
showDebugData()
if conf['UPDATE']:
update()
load_payloads()
if conf['ENGINE'] is 't':
from lib.controller.threads import ThreadsEngine
ThreadsEngine().run()
elif conf['ENGINE'] is 'c':
from lib.controller.coroutine import CoroutineEngine
CoroutineEngine().run()
except KeyboardInterrupt, e:
logger.log(CUSTOM_LOGGING.ERROR, 'User quit!')
def setArgs(args):
conf['MODULE_NAME'] = args.m
conf['MODULE_FILE_PATH'] = os.path.join(paths['MODULES_PATH'],
conf['MODULE_NAME'] + ".py")
conf['THREADS_NUM'] = args.t
conf['SCREEN_OUTPUT'] = args.nS
conf['FILE_OUTPUT'] = args.nF
conf['SINGLE_MODE'] = args.single
conf['DEBUG'] = args.debug
conf['NETWORK_STR'] = args.n
conf['OPEN_BROWSER'] = args.browser
# TODO
th['THREADS_NUM'] = conf['THREADS_NUM']
if args.update:
conf['UPDATE'] = args.update
update()
if args.T:
conf['ENGINE'] = 't'
elif args.C:
conf['ENGINE'] = 'c'
if args.f:
conf['MODULE_MODE'] = 'f'
conf['INPUT_FILE_PATH'] = args.f
elif args.i:
conf['MODULE_MODE'] = 'i'
conf['I_NUM2'] = args.i
conf['INPUT_FILE_PATH'] = None
elif args.n:
conf['MODULE_MODE'] = 'n'
conf['INPUT_FILE_PATH'] = None
conf['OUTPUT_FILE_PATH'] = os.path.abspath(args.o) if args.o else \
os.path.abspath(
os.path.join(
paths['OUTPUT_PATH'], time.strftime(
'[%Y%m%d-%H%M%S]', time.localtime(
time.time())) + conf['MODULE_NAME'] + '.txt'))
conf['SCREEN_OUTPUT'] = args.nS
conf['FILE_OUTPUT'] = args.nF
def main():
"""
Main function of POC-T when running from command line.
"""
try:
paths['ROOT_PATH'] = os.path.dirname(
os.path.dirname(os.path.realpath(__file__)))
try:
os.path.isdir(paths['ROOT_PATH'])
except UnicodeEncodeError:
errMsg = "your system does not properly handle non-ASCII paths. "
errMsg += "Please move the project root directory to another location"
logger.error(errMsg)
raise SystemExit
setPaths()
parse_args()
if IS_WIN:
win_color_init()
banner()
if conf['DEBUG']:
showDebugData()
if conf['UPDATE']:
update()
load_payloads()
if conf['ENGINE'] is 't':
from lib.controller.threads import ThreadsEngine
ThreadsEngine().run()
elif conf['ENGINE'] is 'c':
from lib.controller.coroutine import CoroutineEngine
CoroutineEngine().run()
sysquit(0)
except KeyboardInterrupt, e:
sysquit(1)
def setArgs(args):
conf['MODULE_NAME'] = args['-m']
conf['MODULE_FILE_PATH'] = os.path.join(paths['MODULES_PATH'], conf['MODULE_NAME'] + ".py")
conf['THREADS_NUM'] = int(args['-t'])
conf['SCREEN_OUTPUT'] = not args['--nS']
conf['FILE_OUTPUT'] = not args['--nF']
conf['SINGLE_MODE'] = args['--single']
conf['DEBUG'] = args['--debug']
conf['NETWORK_STR'] = args['-n']
conf['OPEN_BROWSER'] = args['--browser']
# TODO
th['THREADS_NUM'] = conf['THREADS_NUM']
if args['--update']:
conf['UPDATE'] = args['--update']
update()
if args['-T']:
conf['ENGINE'] = 't'
elif args['-C']:
conf['ENGINE'] = 'c'
if args['-f']:
conf['MODULE_MODE'] = 'f'
conf['INPUT_FILE_PATH'] = args['-f']
elif args['-i']:
conf['MODULE_MODE'] = 'i'
conf['I_NUM2'] = args['-i']
conf['INPUT_FILE_PATH'] = None
elif args['-n']:
conf['MODULE_MODE'] = 'n'
conf['INPUT_FILE_PATH'] = None
conf['OUTPUT_FILE_PATH'] = os.path.abspath(args['-o']) if args['-o'] else \
os.path.abspath(
os.path.join(
paths['OUTPUT_PATH'], time.strftime(
'[%Y%m%d-%H%M%S]', time.localtime(
time.time())) + conf['MODULE_NAME'] + '.txt'))
def _checkUpdate(args):
# conflict with args.update(),so we use args['update'] here
if args['update']:
conf.UPDATE = args['update']
update()
def checkUpdate(args):
if args.sys_update:
update()
def _checkUpdate(args):
# conflict with args.update(),so we use args['update'] here
if args['update']:
raw_input('update?')
conf.UPDATE = args['update']
update()
