doomarena-browsergym 0.0.4

BrowserGym extensions for DoomArena

BrowserGym+ AgentLab Threat Models

This repository contains tools and scripts for defining and testing threat models in the BrowserGym + AgentLab agentic framework. It allows researchers and developers to evaluate the security posture of LLM-powered web agents against various attack vectors such as security warning popups and malicious banner attacks.

Overview

The framework provides a structured way to:

• Simulate various attack scenarios against web agents
• Measure attack success rate (ASR), task success rate (TSR), and attack stealth rate
• Evaluate different LLM models' robustness against these attacks
• Generate comprehensive reports of attack effectiveness

This toolkit specifically focuses on testing agents in the BrowserGym setting.

Installation

1. Run pip install -e doomarena/browsergym from the root repo or do pip install doomarena-browsergym for the latest release
2. Run your instance of webarena and set an environment variable DOOMARENA_WEBARENA_BASE_URL="http://XXXX.com"
3. Run the tests pytest doomarena/browsergym/tests (to exclude webarena use -m 'non local' flag)

Usage

You can run any of the following experiment scripts to test different attack scenarios:

• run_banner_attack_webarena_reddit_notext.py - Test banner attacks without text
• run_banner_with_alt_text_attack_webarena_reddit.py - Test banner attacks with alt text
• run_popup_attack_webarena_reddit.py - Test security warning popup attacks
• run_bgym_experiment.py - Run a general BrowserGym experiment with customizable parameters

For example:

python -m doomarena.browsergym.scripts.run_bgym_experiment

Experiment Configuration Parameters

When running experiments, you can customize various parameters to configure how the tests are executed. Here's an explanation of the key parameters that can be set:

# Example usage:
run_bgym_experiment(
    base_url="<path to your webarena instance>",
    bgym_experiments=bgym_experiments,
    relaunch=False,
    n_jobs=0,  # set to 0 to display browser, 1 for headless, >=1 for parallel headless with "ray" (incompatible with debugger)
    max_steps=15,  # lower for faster testing, use 15 for full task
    skip_reset_and_massage=True,  # skipping is faster, but set False for reproducing numbers
)

• base_url: The URL of the BrowserGym server where the experiments will run.

• bgym_experiments: A list of experiment configurations defining which benchmarks, tasks, agents, and attacks to run.

• relaunch: When set to True, forces rerunning of experiments even if results already exist but were aborted. Default is False to avoid duplicate runs.

• n_jobs: Controls parallelization of experiment execution:

  • 0: Runs with visible browser UI (non-headless mode); useful for demo/debugging
  • 1: Runs in headless mode (no UI) with a single process
  • >1: Runs in parallel headless mode using Ray with the specified number of workers

• max_steps: Maximum number of steps each agent is allowed to take per task. Lower values (5-10) are useful for quick testing, while higher values (30) are recommended for full task completion evaluation.

• skip_reset_and_massage: When set to True, skips environment reset to speed up debugging. This should always be set to False when making experiments.

Results

Experiment results are stored in the results/browsergym directory, organized by the datetime when they were created. Each results directory contains:

• Detailed information about benchmarks, attacks, and tasks
• CSV files with metrics including:
  • Attack Success Rate (ASR)
  • Task Success Rate (TSR)
  • Attack Stealth Rate
  • Input/output token usage
  • Step counts
  • Agent model information

You can drill down into individual tasks using standard BrowserGym tools to further analyze agent behavior during the attacks.

To aggregate the results to csv, run:

from doomarena.browsergym.scripts import collect_results

collect_results('<Path to your experiments>')  # recursive

This will produce one attack_df.csv per task, as well as a globally aggregated one.

Attack Types

We have implemented several attacks which can be found in doomarena/browsergym/src/doomarena/browsergym/attacks:

1. Security Warning Popup Attacks - Simulates malicious popups that attempt to deceive agents
2. Banner Attacks - Tests agents against misleading banner elements
3. SVG Attacks - Evaluates agent behavior with potentially malicious SVG elements