def lackey_worker(self):
# TODO: Fix the bug that this doesn't exit on KeyBoardExit.
self.lackey = Lackey(self.server_ip, self.server_port, self.seed_dir,
self.samples_dir, self.crash_dir)
self.lackey.run()
class Fizzled():
def __init__(self):
# Parse Args
parser = ArgumentParser()
parser.add_argument('--tool', choices=AVAILABLE_TOOLS, required=True)
parser.add_argument('--settings', default='settings')
parser.add_argument('--daemonize', action='store_true')
parser.add_argument('--no-banner', action='store_true')
# TODO: More and less verbose
args = parser.parse_args()
self.tool = args.tool
self.settings_file = args.settings
if args.daemonize:
raise NotImplemented()
# Parse Settings
settings = __import__(self.settings_file)
self.seed_dir = settings.SEED_DIRECTORY
self.work_dir = settings.WORK_DIRECTORY
self.samples_dir = settings.SAMPLES_DIRECTORY
self.crash_dir = settings.CRASH_DIRECTORY
self.binary = settings.BINARY
self.fuzz_args = settings.ARGUMENTS
self.ttl = settings.TIME_TO_LIVE
self.recover_time = settings.RECOVER_TIME
self.destructive = settings.DESTRUCTIVE
self.debugger = settings.DEBUGGER
self.strategy = settings.STRATEGY
self.max_total_mutations = settings.MAX_TOTAL_MUTATIONS
self.radamsa_bin = settings.RADAMSA_BIN
self.run_lackey = settings.RUN_LACKEY
self.server_ip = settings.SERVER_IP
self.server_port = settings.SERVER_PORT
if not args.no_banner:
print(BANNER)
def run(self):
if self.tool == 'mutilator':
mutilator.run(self.seed_dir, self.samples_dir,
self.strategy, self.max_total_mutations)
elif self.tool == 'taskmaster':
taskmaster.run(self.binary,
self.fuzz_args,
self.ttl,
self.samples_dir,
self.work_dir,
self.crash_dir,
self.recover_time,
self.destructive,
self.debugger)
elif self.tool == 'autopsy':
autopsy.run(self.binary, self.fuzz_args, self.ttl)
elif self.tool == 'auto':
if self.run_lackey:
lackey_thread = Thread(target=self.lackey_worker)
lackey_thread.start()
worker = Thread(target=self.mutilator_worker)
worker.start()
taskmaster.run(self.binary,
self.fuzz_args,
self.ttl,
self.samples_dir,
self.work_dir,
self.crash_dir,
self.recover_time,
self.destructive,
self.debugger)
def exit(self):
try:
self.lackey.exit()
except AttributeError:
pass
sys.exit(0)
def lackey_worker(self):
# TODO: Fix the bug that this doesn't exit on KeyBoardExit.
self.lackey = Lackey(self.server_ip, self.server_port,
self.seed_dir, self.samples_dir,
self.crash_dir)
self.lackey.run()
def mutilator_worker(self):
strategy_args = dict()
strategy_args['radamsa_bin'] = self.radamsa_bin
while True:
mutilator.run(self.seed_dir, self.samples_dir,
self.strategy, self.max_total_mutations,
strategy_args)
sleep(self.recover_time)
class Fizzled():
def __init__(self):
# Parse Args
parser = ArgumentParser()
parser.add_argument('--tool', choices=AVAILABLE_TOOLS, required=True)
parser.add_argument('--settings', default='settings')
parser.add_argument('--daemonize', action='store_true')
parser.add_argument('--no-banner', action='store_true')
# TODO: More and less verbose
args = parser.parse_args()
self.tool = args.tool
self.settings_file = args.settings
if args.daemonize:
raise NotImplemented()
# Parse Settings
settings = __import__(self.settings_file)
self.seed_dir = settings.SEED_DIRECTORY
self.work_dir = settings.WORK_DIRECTORY
self.samples_dir = settings.SAMPLES_DIRECTORY
self.crash_dir = settings.CRASH_DIRECTORY
self.binary = settings.BINARY
self.fuzz_args = settings.ARGUMENTS
self.ttl = settings.TIME_TO_LIVE
self.recover_time = settings.RECOVER_TIME
self.destructive = settings.DESTRUCTIVE
self.debugger = settings.DEBUGGER
self.strategy = settings.STRATEGY
self.max_total_mutations = settings.MAX_TOTAL_MUTATIONS
self.radamsa_bin = settings.RADAMSA_BIN
self.run_lackey = settings.RUN_LACKEY
self.server_ip = settings.SERVER_IP
self.server_port = settings.SERVER_PORT
if not args.no_banner:
print(BANNER)
def run(self):
if self.tool == 'mutilator':
mutilator.run(self.seed_dir, self.samples_dir, self.strategy,
self.max_total_mutations)
elif self.tool == 'taskmaster':
taskmaster.run(self.binary, self.fuzz_args, self.ttl,
self.samples_dir, self.work_dir, self.crash_dir,
self.recover_time, self.destructive, self.debugger)
elif self.tool == 'autopsy':
autopsy.run(self.binary, self.fuzz_args, self.ttl)
elif self.tool == 'auto':
if self.run_lackey:
lackey_thread = Thread(target=self.lackey_worker)
lackey_thread.start()
worker = Thread(target=self.mutilator_worker)
worker.start()
taskmaster.run(self.binary, self.fuzz_args, self.ttl,
self.samples_dir, self.work_dir, self.crash_dir,
self.recover_time, self.destructive, self.debugger)
def exit(self):
try:
self.lackey.exit()
except AttributeError:
pass
sys.exit(0)
def lackey_worker(self):
# TODO: Fix the bug that this doesn't exit on KeyBoardExit.
self.lackey = Lackey(self.server_ip, self.server_port, self.seed_dir,
self.samples_dir, self.crash_dir)
self.lackey.run()
def mutilator_worker(self):
strategy_args = dict()
strategy_args['radamsa_bin'] = self.radamsa_bin
while True:
mutilator.run(self.seed_dir, self.samples_dir, self.strategy,
self.max_total_mutations, strategy_args)
sleep(self.recover_time)
def lackey_worker(self):
# TODO: Fix the bug that this doesn't exit on KeyBoardExit.
self.lackey = Lackey(self.server_ip, self.server_port,
self.seed_dir, self.samples_dir,
self.crash_dir)
self.lackey.run()
