mirror of
https://github.com/jorhelp/Ingram.git
synced 2026-10-16 21:20:40 +08:00
fff122
0c4953f4e3
monkey.patch_all(thread=False) still patches the queue module, which
replaces queue.SimpleQueue used internally by ThreadPoolExecutor with
a gevent queue. Worker threads in ThreadPoolExecutor have no gevent
hub, causing LoopExit ("This operation would block forever") and
AssertionError ("Can only use Waiter.switch method from the Hub
greenlet") exceptions during scanning.
Adding queue=False preserves the standard queue for ThreadPoolExecutor
while keeping gevent's socket/ssl patching for cooperative I/O.
67 lines
1.9 KiB
Python
Executable File
67 lines
1.9 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
# coding : utf-8
|
|
# @Author : Jor<jorhelp@qq.com>
|
|
# @Date : Wed Apr 20 00:17:30 HKT 2022
|
|
# @Desc : Webcam vulnerability scanning tool
|
|
|
|
#=================== 需放置于最开头 ====================
|
|
import warnings; warnings.filterwarnings("ignore")
|
|
from gevent import monkey; monkey.patch_all(thread=False, queue=False)
|
|
#======================================================
|
|
|
|
import os
|
|
import sys
|
|
from multiprocessing import Process
|
|
|
|
from loguru import logger
|
|
|
|
from Ingram import get_config
|
|
from Ingram import Core
|
|
from Ingram.utils import color
|
|
from Ingram.utils import common
|
|
from Ingram.utils import get_parse
|
|
from Ingram.utils import log
|
|
from Ingram.utils import logo
|
|
|
|
|
|
def run():
|
|
try:
|
|
# logo
|
|
for icon, font in zip(*logo):
|
|
print(f"{color.yellow(icon, 'bright')} {color.magenta(font, 'bright')}")
|
|
|
|
# config
|
|
config = get_config(get_parse())
|
|
if not os.path.isdir(config.out_dir):
|
|
os.mkdir(config.out_dir)
|
|
os.mkdir(os.path.join(config.out_dir, config.snapshots))
|
|
if not os.path.isfile(config.in_file):
|
|
print(f"{color.red('the input file')} {color.yellow(config.in_file)} {color.red('does not exists!')}")
|
|
sys.exit()
|
|
|
|
# log 配置
|
|
log.config_logger(os.path.join(config.out_dir, config.log), config.debug)
|
|
|
|
# 任务进程
|
|
p = Process(target=Core(config).run)
|
|
if common.os_check() == 'windows':
|
|
p.run()
|
|
else:
|
|
p.start()
|
|
p.join()
|
|
|
|
except KeyboardInterrupt:
|
|
logger.warning('Ctrl + c was pressed')
|
|
p.kill()
|
|
sys.exit()
|
|
|
|
except Exception as e:
|
|
logger.error(e)
|
|
print(f"{color.red('error occurred, see the')} {color.yellow(config.log)} "
|
|
f"{color.red('for more information.')}")
|
|
sys.exit()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
run()
|