lishang520 / autoscan Goto Github PK

感谢大佬们的工具，该工具我很喜欢，是一款针对重点资产指纹识别的工具，所以我的扫描器就是基于该工具来识别指纹

选该工具之前，我尝试过whatweb-plus 加强版的指纹识别工具，该指纹识别工具指纹内置8000多条，但是在使用过后，我还是放弃了，因为他的指纹多导致扫描时间长，而且很多指纹漏洞意义不大，所以最终放弃。,同时希望大家明白，指纹并不是越多越好，有的指纹识别出来了，但是它对应的资产少、对应的漏洞比较鸡肋的话，那么这种指纹也就没必要了，所以采用了EHole的指纹

    有2种模式，可自行选择。一种是对所有的资产都进行扫描，工具内置的策略是对没有识别出指纹的资产使用nuclei里的高危、严重的poc进行监测，对于识别出指纹的资产，首先判定nuclei中是否有该指纹，如果没有，则将其丢到高危、严重的那种扫描模式里去扫描，如果nuclei中存在该指纹，则对该指纹下的所哟目标进行扫描

应用场景：
    一般用于快速、大批量发现漏洞，比如edusrc、公益src

应用技巧：
    1、当我们对目标进行端口扫描，可将存在web服务的资产丢到该扫描器中
    2、当我们对目标进行子域名扫描，得到大量的子域名后，可将这些子域名统一丢到该扫描器中

 EHole3.0.exe finger -l targets.txt -o wait_check.xlsx

ProcessMonitor.py，监控上一步的指纹识别工具是否执行完毕，如果执行完毕才向下执行，否则就等待

#进程监控
import psutil
import subprocess
import platform,time
class ProcessMonitor():
    def __init__(self,processname):
        self.processname = processname
    def main(self):
        status = self.check_system()
        return status
    def check_system(self):
        plat = platform.system().lower()
        if plat == 'windows':
            return  self.win_process()
        elif plat == 'linux':
            return self.linux_process()

    def win_process(self):
        # print("[+]这是windows系统")
        while True:
            start_pl = psutil.pids()
            time.sleep(1)
            for pid in start_pl:
                if self.processname not in psutil.Process(pid).name():   #看我们监控的进程名是否在进程列表里
                    # print("[+]程序执行完毕")
                    return 1   #进程执行完毕


    # linux进程监控
    def linux_process(self):
        # print("[+]这是linux系统")
        while True:
            cmd = "ps -aux | grep {} ".format(self.processname)+"| grep -v grep | awk '{print $2}'"
            start_rsp = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
            time.sleep(1)
            end_rsp = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
            if len(start_rsp.stdout.read().strip()) != 0:
                # print("[+]程序正在执行中...")
                if len(end_rsp.stdout.read().strip()) == 0:
                    # print("[+]程序执行完毕")
                    return 1

    def extract_data(self):
        # print(self.all_message)
        temp_result = []
        for message in self.all_message:
            url = {}
            stat = 0  # 开关

            if message[1] == '':
                self.base_scan_target.append(message[0])    #该目标后续用高低中poc来进行批量扫描
            else :
                for t in self.nuclei_tags:
                    if str(t).lower() in str(message[1]).lower():
                        stat =1
                        url[t] = message[0]
                if stat == 0:     #一个目标，一个标签都没命中，就让其进行 高低中 扫描
                    self.base_scan_target.append(message[0])  # 该目标后续用高低中poc来进行批量扫描
            temp_result.append(url)
        result = {}

        for temp in temp_result:
            for tag in temp.keys():
                if result.get(tag) == None:
                    result.update({tag:[]})
                for t in temp.values():

                    result.get(tag).append(t)
        self.tags_scan_target = result