VIGIL: Defending LLM Agents Against Tool Stream Injection via Verify-Before-Commit
Junda Lin, Zhaomeng Zhou, Zhi Zheng, Shuochen Liu, Tong Xu, Yong Chen, Enhong Chen
TL;DR
This work addresses the dual threat of indirect prompt injection in LLM agents, particularly via tool streams exploited in open environments. It introduces VIGIL, a verify-before-commit framework that grounds agent safety in intent-driven constraints, perception sanitization, speculative hypothesis generation, and rigorous verification, enabling adaptive recovery without sacrificing reasoning flexibility. The SIREN benchmark provides a unified assessment of dual-stream threats across five tool-stream vectors and data-stream bases, demonstrating that VIGIL outperforms static and dynamic defenses by significantly reducing attack success rates while maintaining or improving utility. The results show that decoupling speculative exploration from irreversible actions preserves task completion under attack, suggesting practical implications for deploying secure, capable agents in open, untrusted environments.
Abstract
LLM agents operating in open environments face escalating risks from indirect prompt injection, particularly within the tool stream where manipulated metadata and runtime feedback hijack execution flow. Existing defenses encounter a critical dilemma as advanced models prioritize injected rules due to strict alignment while static protection mechanisms sever the feedback loop required for adaptive reasoning. To reconcile this conflict, we propose \textbf{VIGIL}, a framework that shifts the paradigm from restrictive isolation to a verify-before-commit protocol. By facilitating speculative hypothesis generation and enforcing safety through intent-grounded verification, \textbf{VIGIL} preserves reasoning flexibility while ensuring robust control. We further introduce \textbf{SIREN}, a benchmark comprising 959 tool stream injection cases designed to simulate pervasive threats characterized by dynamic dependencies. Extensive experiments demonstrate that \textbf{VIGIL} outperforms state-of-the-art dynamic defenses by reducing the attack success rate by over 22\% while more than doubling the utility under attack compared to static baselines, thereby achieving an optimal balance between security and utility. Code is available at https://anonymous.4open.science/r/VIGIL-378B/.
