def test_e2e(self):
# Set up hermetic environment.
host = Host()
host.fd_out = self._stdout
host.fd_err = self._stderr
with host.temp_dir() as temp_dir:
# (Re-)parse the command line arguments, a la main.py.
factory = Factory(host=host)
parser = factory.parser
args = parser.parse_args()
# Ensure exactly 1 fuzzer is selected.
fuzzer = factory.create_fuzzer(args)
self.assertNoErrors()
args.name = str(fuzzer)
list_args = parser.parse_args(['list', args.name])
list_args.command(list_args, factory)
self.assertOut(
['Found 1 matching fuzzer for "{}":'.format(str(fuzzer))], n=1)
self.assertNoErrors()
start_args = parser.parse_args(
['start', '-o', temp_dir.pathname, args.name])
proc = command.start_fuzzer(start_args, factory)
self.assertNoErrors()
stop_args = parser.parse_args(['stop', args.name])
command.stop_fuzzer(stop_args, factory)
self.assertNoErrors()
if proc:
proc.wait()
check_args = parser.parse_args(['check', args.name])
command.check_fuzzer(check_args, factory)
self.assertOut(['{}: STOPPED'.format(args.name)], n=1)
self.assertNoErrors()
unit = os.path.join(temp_dir.pathname, 'unit')
with open(unit, 'w') as opened:
opened.write('hello world')
repro_args = parser.parse_args(['repro', args.name, unit])
command.repro_units(repro_args, factory)
self.assertNoErrors()
analyze_args = ['analyze', '-max_total_time=10', args.name]
if args.local:
analyze_args.append('--local')
analyze_args = parser.parse_args(analyze_args)
command.analyze_fuzzer(analyze_args, factory)
self.assertNoErrors()
def test_single_fuzzer(self):
# (Re-)parse the command line arguments, a la main.py.
args = self.parser.parse_args()
# Ensure exactly 1 fuzzer is selected.
fuzzer = self.factory.create_fuzzer(args)
self.assertNoErrors()
args.name = str(fuzzer)
list_args = self.parser.parse_args(['list', args.name])
list_args.command(list_args, self.factory)
self.assertOut(
['Found 1 matching fuzzer for "{}":'.format(str(fuzzer))], n=1)
self.assertNoErrors()
start_args = self.parser.parse_args(
['start', '-o', self.temp_dir, args.name])
proc = command.start_fuzzer(start_args, self.factory)
self.assertNoErrors()
stop_args = self.parser.parse_args(['stop', args.name])
command.stop_fuzzer(stop_args, self.factory)
self.assertNoErrors()
if proc:
proc.wait()
check_args = self.parser.parse_args(['check', args.name])
command.check_fuzzer(check_args, self.factory)
self.assertOut(['{}: STOPPED'.format(args.name)], n=1)
self.assertNoErrors()
unit = os.path.join(self.temp_dir, 'unit')
with open(unit, 'w') as opened:
opened.write('hello world')
repro_args = self.parser.parse_args(['repro', args.name, unit])
command.repro_units(repro_args, self.factory)
self.assertNoErrors()
analyze_args = ['analyze', '-max_total_time=10', args.name]
if args.local:
analyze_args.append('--local')
analyze_args = self.parser.parse_args(analyze_args)
command.analyze_fuzzer(analyze_args, self.factory)
self.assertNoErrors()
def test_repro_units(self):
unit = 'crash-deadbeef'
self.host.touch(unit)
args = self.parse_args('repro', str(self.fuzzer), unit)
command.repro_units(args, self.factory)