import os
import signal
from gevent import monkey
monkey.patch_all(threaded=False)
from settings import banner
from lib.log import logger as log
from lib.core.prepare import prepare
from core.PocManager import PoCManager
from lib.core.cmdparser import cmdline_parse
import checkKernel

def main():
    init_signal()
    options, args = cmdline_parse()
    try :
        if options.Vuls:
            val=options.Vuls
            if val =='all':
                fb=checkKernel.kernelcheck.dispose(self='')
            else:
                print('暂时只支持all参数')
    except:
        try:
            if options.argus:
                if options.argus=='all':
                    os.system('python3 argus/argus.py')
                else:
                    print('暂时只支持all参数')
        except:
            try:
                if options.fuzz:
                    if options.fuzz == 'dbus':
                        print('fuzz 正在完善')
                    else:
                        print('暂时只支持all参数')
            except:
                fb = prepare(options)
                start(fb)

def init_signal():
    os.setpgrp()
    signal.signal(
        signal.SIGTERM,
        lambda *_: os.kill(0, signal.SIGKILL))

def start(fb):
    print(banner)
    poc_manager = PoCManager(fb)
    results = poc_manager.run()
    if fb.console:
        return results if results else None
    else:
        if results: log.result(results)


if __name__ == "__main__":
    main()