mirror of https://github.com/dirtbags/moth.git
54 lines
1.1 KiB
Python
Executable File
54 lines
1.1 KiB
Python
Executable File
#! /usr/bin/env python3
|
|
|
|
import asyncore
|
|
import os
|
|
import sys
|
|
import optparse
|
|
import signal
|
|
from ctf import pointsd
|
|
from ctf import flagd
|
|
from ctf import histogram
|
|
from ctf import config
|
|
|
|
do_reap = False
|
|
|
|
def chart(s):
|
|
if not os.fork():
|
|
histogram.main(s)
|
|
sys.exit(0)
|
|
|
|
def reap():
|
|
try:
|
|
while True:
|
|
os.waitpid(0, os.WNOHANG)
|
|
except OSError:
|
|
pass
|
|
|
|
def sigchld(signum, frame):
|
|
do_reap = True
|
|
|
|
def main():
|
|
p = optparse.OptionParser()
|
|
p.add_option('-p', '--genpass', dest='cat', default=None,
|
|
help='Generate a flagger password for the given category')
|
|
opts, args = p.parse_args()
|
|
if opts.cat:
|
|
print('%s:::%s' % (opts.cat, flagd.hexdigest(opts.cat.encode('utf-8'))))
|
|
return
|
|
|
|
pointsrv = pointsd.start()
|
|
flagsrv = flagd.start()
|
|
|
|
signal.signal(signal.SIGCHLD, sigchld)
|
|
s = pointsrv.store
|
|
slen = 0
|
|
while True:
|
|
if do_reap:
|
|
reap()
|
|
asyncore.loop(timeout=30, use_poll=True, count=1)
|
|
if len(s) > slen:
|
|
slen = len(s)
|
|
chart(s)
|
|
|
|
main()
|