eswyft/suddenly-opened-ports-checker
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

if script fails, report to telegram too

Browse Source
This commit is contained in:
Evgeny Zinoviev 2022-03-18 04:48:55 +03:00
parent 37ec4be9af
commit 5644c92731
1 changed files with 45 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
suddenly-opened-ports-checker
View File

@ -2,10 +2,12 @@
import logging
import yaml
import math
import html
from argparse import ArgumentParser
from lib.worker import Worker
from lib.results import Results
from ch1p import telegram_notify
logger = logging.getLogger(__name__)
@ -40,47 +42,56 @@ def main():
if not args.no_telegram:
assert 'telegram' in config
# let's go
results = Results()
max_threads = math.inf if args.threads_limit == 0 else args.threads_limit
active_threads = 1
try:
# let's go
results = Results()
max_threads = math.inf if args.threads_limit == 0 else args.threads_limit
active_threads = 1
def get_active_threads():
n = active_threads
if workers:
n += workers[0].concurrency
return n
def get_active_threads():
n = active_threads
if workers:
n += workers[0].concurrency
return n
workers = []
for name, data in config['servers'].items():
w = Worker(name, data['host'], data['opened'], data['ignore'],
concurrency=int(data['concurrency']) if 'concurrency' in data else args.concurrency,
timeout=int(data['timeout']) if 'timeout' in data else args.timeout)
workers.append(w)
workers = []
for name, data in config['servers'].items():
w = Worker(name, data['host'], data['opened'], data['ignore'],
concurrency=int(data['concurrency']) if 'concurrency' in data else args.concurrency,
timeout=int(data['timeout']) if 'timeout' in data else args.timeout)
workers.append(w)
current_workers = []
while workers:
w = workers.pop(0)
active_threads += w.concurrency+1
current_workers = []
while workers:
w = workers.pop(0)
active_threads += w.concurrency+1
current_workers.append(w)
w.start()
current_workers.append(w)
w.start()
while current_workers and get_active_threads() >= max_threads:
for cw in current_workers:
cw.join(timeout=0.1)
if not cw.is_alive():
results.add(cw)
current_workers.remove(cw)
active_threads -= cw.concurrency+1
while current_workers and get_active_threads() >= max_threads:
for cw in current_workers:
cw.join(timeout=0.1)
if not cw.is_alive():
results.add(cw)
current_workers.remove(cw)
active_threads -= cw.concurrency+1
for cw in current_workers:
cw.join()
results.add(cw)
for cw in current_workers:
cw.join()
results.add(cw)
if results.has_warnings() and not args.no_telegram:
results.notify(chat_id=config['telegram']['chat-id'],
token=config['telegram']['token'])
if results.has_warnings() and not args.no_telegram:
results.notify(chat_id=config['telegram']['chat-id'],
token=config['telegram']['token'])
except KeyboardInterrupt:
pass
except Exception as e:
if not args.no_telegram:
telegram_notify(html.escape(str(e)),
parse_mode='html',
chat_id=config['telegram']['chat-id'],
token=config['telegram']['token'])
if __name__ == '__main__':