IMCP - Insecure Model Context Protocol
An educational framework for understanding AI security vulnerabilities
⚠️ Educational Purposes Only
IMCP (Insecure Model Context Protocol) is a deliberately vulnerable application designed exclusively for educational and research purposes. It demonstrates critical AI security vulnerabilities. DO NOT deploy in production environments or use with sensitive data.
🔍 Overview
IMCP is an educational framework that exposes 16 critical security vulnerabilities in AI/ML model serving systems. It serves as a controlled, "vulnerable by design" platform for security researchers, developers, and educators to learn about and mitigate emerging AI threats.
Think of IMCP as the “DVWA for AI” — a safe environment where you can explore:
- Model Poisoning
- Prompt Injection
- Embedding Vector Exploits
- RAG System Weaknesses
- And many more…
🛡️ Vulnerabilities Demonstrated
Core AI Manipulation
- Model Poisoning: Malicious training data injection.
- Token Prediction Attacks: Exploiting token probability for sensitive data extraction.
- Multimodal Vulnerabilities: Cross-modal prompt leakage and metadata manipulation.
- Credential Vulnerabilities: Insecure authentication mechanisms in AI systems.
- Embedding Vector Attacks: Poisoning vector stores for unauthorized access.
- RAG Vulnerabilities: Exploiting document stores for cross-user data leakage.
- User Data Leakage: Unintended exposure of conversation histories.
- Model Capability Enumeration: Over-disclosure of internal model details.
Control Manipulation
- Context Manipulation: Unrestricted modifications to model contexts and system prompts.
- Prompt Injection: Techniques to bypass AI safety filters.
- Model Access Control Bypass: Elevation of privileges to access restricted functionalities.
- Model Chain Attacks: Exploiting chained model interactions.
📜 Test Suite
The test suite in test_vulnerabilities.py demonstrates each vulnerability with detailed explanations and examples. It includes:
- Model Poisoning: Injecting malicious data into model responses.
- Token Prediction: Extracting sensitive information character by character.
- Embedding Vector Attacks: Unauthorized access to sensitive embeddings.
- Context Manipulation: Modifying system prompts and configurations.
- Function Calling Vulnerabilities: Registering functions for remote code execution.
- RAG Vulnerabilities: Cross-user document access and manipulation.
📜 API Endpoints
/imcp: Main JSON-RPC endpoint for IMCP functionality./v1/chat/completions: OpenAI API-compatible endpoint./v1/models: List available models./v1/embeddings: Generate embeddings./v1/auth/token: Authentication endpoint./.well-known/imcp-configuration: Service discovery endpoint.
🚀 Getting Started
Prerequisites
- Python 3.8+
- OpenAI API Key (required for live examples)
Installation
Clone the repository and set up your environment:
# Clone the repository
git clone https://github.com/nav33n25/IMCP.git
cd imcp
# Create a virtual environment
python -m venv venv
source venv/bin/activate # For Windows: venv\Scripts\activate
# Install dependencies
pip install -r requirements.txt
# Configure the environment
cp .env.example .env
# Edit .env to include your OpenAI API key
Running IMCP
Start the server and run the test suite:
# Start the IMCP server
python -m flask run --host=0.0.0.0 --port=5000
# In another terminal, run the test suite
python test_vulnerabilities.py
📚 Documentation
All the comprehensive guides are located in the documentation/ directory:
- Vulnerability Guide: Detailed explanations of each vulnerability.
- Exploitation Guide: Step-by-step instructions to reproduce each vulnerability.
- Mitigation Guide: Strategies and best practices to secure AI systems.
🌟 Key Features
- Realistic AI Service Implementation
- 16 Unique AI-Specific Security Vulnerabilities
- Comprehensive Test Suite for Demonstrations
- Detailed Documentation for In-Depth Learning
- Compatibility with Modern LLM APIs (e.g., OpenAI)
- Mock Mode for Cost-Free Testing
🤝 Contributing
We welcome contributions from the community! Areas where you can help include:
- Additional Vulnerability Demonstrations: New scenarios or enhancements.
- Improved Documentation: Detailed educational materials and guides.
- Integration: Support for other LLM providers.
- UI Enhancements: Better visualizations and user experience improvements.
Please check out our CONTRIBUTING.md for more details on how to get started.
📜 License
This project is licensed under the MIT License. See the LICENSE file for details.
⚠️ Disclaimer
IMCP is intentionally vulnerable software for educational purposes only. The creators are not liable for any misuse or damage caused by the use of this software.