Skip to content

Commit

Permalink
Fix pastis-benchmark plot command for the proxy
Browse files Browse the repository at this point in the history
  • Loading branch information
cnheitman committed Aug 26, 2023
1 parent 2f202a0 commit f504f68
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion bin/pastis-benchmark
Original file line number Diff line number Diff line change
Expand Up @@ -318,15 +318,21 @@ def replay(workspace: str, type: str, injloc: str, live: bool, stream: bool, tim
@click.option('--union/--no-union', type=bool, is_flag=True, default=True, help="Show union of analyses")
@click.option('-t', "--timeout", type=int, default=86400, help="Duration of the campaign (timeout)")
@click.option('-n', '--name', type=str, required=True, help="Name to give to the plotting")
@click.option('--is-proxy', type=bool, is_flag=True, default=False, help="Fuzzer is a proxy")
@click.argument('workspace', type=click.Path(exists=True, file_okay=False, dir_okay=True, readable=True), nargs=-1)
def plot(show: bool, show_tt_inputs: bool, union: bool, timeout: int, name: str, workspace: Tuple[str]):
def plot(show: bool, show_tt_inputs: bool, union: bool, timeout: int, name: str, is_proxy: bool, workspace: Tuple[str]):

configure_logging(logging.INFO)

plotter = Plotter(name, timeout)
for ws in workspace:
logging.info(f"Load workspace: {ws}")
campaign = CampaignResult(Path(ws))

# FIXME: Rework this.
if is_proxy:
campaign.SEED_FUZZER = 'PROXY'

campaign.load() # by default load in qbdi

if campaign.replay_ok():
Expand Down

0 comments on commit f504f68

Please sign in to comment.