Пример #1
0
def __test():
    db = Database()
    db.db_init()
    dbsession = db.db_session()

    testCVE = Spider(dbsession)
    result, e = testCVE.get_vulns()
    if not result:
        print(e)
Пример #2
0
from lib.common.banner import banner
from config.config import *
import config.config as config
from lib.database import Database
from lib.Modles import Vuln
from sqlalchemy import and_, or_
import ipdb

import logging

# 原计划实现对Python版本判断,但疑似python2在运行前便对整个脚本进行了语法校验,如此导致大量错误。
# if not check_python_version(3):
#     sys.exit()

db = Database()
db.db_init()


def spider(spider_plugin):
    global db
    # plugin_path = os.path.join(configs.get('plugin_dir'), spider_plugin)
    try:
        print()  # 换行
        print("[*] Loading spider plugin :: < %s >" % spider_plugin, end="")
        plugin = importlib.import_module("lib.plugins.%s"%spider_plugin[:-3])
        print("\tSuccess.")
        Spider = getattr(plugin, "Spider")
        # print(Spider)
    except Exception as e:
        print("[*] Something wrong with run < %s >"%spider_plugin)
        print("[*] Error - %s" % str(e))