修改错误和完善插件框架

app/model/exts.py

@@ -0,0 +1,4 @@
+from flask_sqlalchemy import SQLAlchemy
+#专门存放db，防止循环引用问题的出现
+
+db=SQLAlchemy()

app/scan/scanIndex.py

@@ -25,6 +25,9 @@
 from pocsuite3.api import get_results
 import os
 import json
+from app.utils.beforeScan import getPluginDepends
+
+
 
 def saveVul(result,tid,poc):
     with app.app_context():
@@ -116,6 +119,11 @@ def scanConsole(url,poclist,tid,pluginlist):
         db.session.add(basemsgdb)
         db.session.commit()
 
+    # 预处理
+    getPluginDepends()
+
+    time.sleep(5)
+
     # 前置扫描
     scanPocs(target,poclist,tid)
     scanPlugins(target,pluginlist,tid)

app/utils/beforeScan.py

@@ -0,0 +1,58 @@
+#!/usr/bin/env python 
+# -*- coding: utf-8 -*-
+# @Time    : 2022/3/26 23:27
+# @Author  : Cl0udG0d
+# @File    : beforeScan.py
+# @Github: https://github.com/Cl0udG0d
+import importlib
+import os
+import re
+import logging
+from pip._internal import main
+import sys
+
+def getPluginDepends():
+    pattern = re.compile("^import (.*?)$")
+    moduleKeys=list(sys.modules.keys())
+    currdir = os.path.join(os.path.dirname(os.path.dirname(__file__)), "../plugins/")
+    for files in os.listdir(currdir):
+        if os.path.splitext(files)[1] == '.py' and not files.startswith("_"):
+            filename = os.path.splitext(files)[0]
+            filepath=currdir+"/"+filename+".py"
+            logging.info("{} is Checking".format(filepath))
+            with open(filepath, 'r') as f:
+                for line in f.readlines():
+                    result=pattern.findall(line.strip())
+                    if result:
+                        name=result[0]
+                        if name and checkLib(name,moduleKeys):
+                            logging.info("{} Lib is Loading".format(name))
+                            install(name)
+                        else:
+                            print("{} Lib is Loaded".format(name))
+    return
+
+
+def install(package,source="https://pypi.tuna.tsinghua.edu.cn/simple"):
+    main(['install', package,'-i',source])
+
+
+def checkLib(libName,moduleKeys):
+    try:
+        if libName in moduleKeys:
+            return False
+        importlib.import_module(libName)
+        return False
+    except Exception as e:
+        logging.warning(e)
+        return True
+
+
+def test():
+    name="import re dd"
+    pattern=re.compile("^import (.*?)$")
+    print(pattern.findall(name))
+
+
+if __name__ == '__main__':
+    getDepends("../plugins")

docs/index.md

@@ -10,7 +10,11 @@
 
 在这些情况下原有的大型扫描器可能不太适用，所以我重构了原来的碎遮扫描器
 
-扫描器砍掉了非必要和耗时比较长的信息收集部分，用户可以自定义POC和插件来进行扫描，POC和插件都可以从本地进行上传，相应的编写文档为：xxx。
+扫描器砍掉了非必要和耗时比较长的信息收集部分，用户可以自定义POC和插件来进行扫描，POC和插件支持本地上传，相关文档如下:
+
++ [Pocsuite3 开发文档及 PoC 编写规范及要求说明](./other/CODING.md)
++ [python实现插件框架]()
++ [插件编写]()
 
 
 项目使用了以下技术：
@@ -35,8 +39,6 @@ docker-compose up -d
 
 选择需要的POC和插件，添加任务进行扫描即可
 
-## 进阶使用
-
 
 ## 开发团队