AIMap

AI एजेंट इंफ्रास्ट्रक्चर के लिए इंटरनेट-स्केल डिस्कवरी और सुरक्षा परीक्षण प्लेटफ़ॉर्म।

AIMap सार्वजनिक रूप से एक्सपोज़्ड AI एंडपॉइंट्स — MCP सर्वर, Ollama इंस्टेंस, vLLM/LiteLLM प्रॉक्सी, LangServe चेन, Gradio ऐप्स, ComfyUI नोड्स, और बहुत कुछ — को ढूंढता है, फिंगरप्रिंट करता है, और सुरक्षा-परीक्षण करता है। Shodan की तरह सोचें, लेकिन AI एजेंट अटैक सरफ़ेस के लिए विशेष रूप से बनाया गया।

Bishop Fox द्वारा निर्मित।

चेतावनी यह उपकरण केवल अधिकृत पेनिट्रेशन टेस्टिंग और सुरक्षा अनुसंधान के लिए है। आप AIMap का उपयोग केवल उन सिस्टमों के विरुद्ध कर सकते हैं जिनके आप स्वामी हैं या जिनके परीक्षण के लिए आपके पास स्पष्ट लिखित अनुमति है। कंप्यूटर सिस्टमों तक अनधिकृत पहुंच अवैध है। Bishop Fox कोई देयता नहीं लेता है और इस उपकरण के किसी भी दुरुपयोग या क्षति के लिए जिम्मेदार नहीं है। जिम्मेदारी से उपयोग करें।

---

यह क्या करता है

1. खोज — Shodan पर 32+ क्यूरेटेड सर्च क्वेरी के साथ इंटरनेट पर एक्सपोज़्ड AI/ML एंडपॉइंट खोजने के लिए क्वेरी करता है
2. फिंगरप्रिंट — प्रत्येक एंडपॉइंट को Nuclei टेम्पलेट और लाइव HTTP जांच के साथ प्रोब करता है ताकि प्रोटोकॉल, फ्रेमवर्क, ऑथ स्थिति, टूल, मॉडल और सिस्टम प्रॉम्प्ट की पहचान हो सके
3. स्कोर — प्रमाणीकरण, टूल एक्सपोज़र, CORS नीति, TLS, सिस्टम प्रॉम्प्ट लीक और खतरनाक क्षमता संयोजनों के आधार पर 0–10 का जोखिम स्कोर की गणना करता है
4. परीक्षण — प्रोटोकॉल-विशिष्ट अटैक सूट (MCP टूल दुरुपयोग, Ollama मॉडल निष्कर्षण, प्रॉम्प्ट इंजेक्शन) को रीयल-टाइम स्ट्रीमिंग परिणामों के साथ लॉन्च करता है
5. विज़ुअलाइज़ — 3D ग्लोब जो हर खोजे गए एंडपॉइंट को दिखाता है, Shodan-शैली क्वेरी भाषा के साथ खोजा जा सकता है

---

आर्किटेक्चर

┌─────────────┐     ┌──────────────┐     ┌───────────┐
│  React SPA  │────▶│  FastAPI      │────▶│  MongoDB  │
│  (Vite)     │ WS  │  Backend      │     │           │
└─────────────┘     └──────┬───────┘     └───────────┘
                           │
              ┌────────────┼────────────┐
              ▼            ▼            ▼
        ┌──────────┐ ┌──────────┐ ┌──────────┐
        │  Shodan  │ │  Nuclei  │ │  Redis   │
        │  API     │ │  Scanner │ │  Streams │
        └──────────┘ └──────────┘ └──────────┘

बैकएंड — Python/FastAPI एसिंक MongoDB (Motor) के साथ, अटैक लॉग स्ट्रीमिंग के लिए Redis स्ट्रीम्स, और एक डिस्कवरी इंजन जो Shodan क्वेरी → httpx लाइवनेस चेक → Nuclei टेम्पलेट स्कैन → एनरिचमेंट पाइपलाइन को ऑर्केस्ट्रेट करता है।

फ्रंटएंड — React 18 + TypeScript + Tailwind CSS + shadcn/ui। इसमें 3D ग्लोब (globe.gl), WebSocket के माध्यम से रीयल-टाइम अटैक स्ट्रीमिंग और Shodan-शैली सर्च इंटरफ़ेस है।

स्कैनिंग — MCP सर्वर डिटेक्शन, MCP टूल एनुमरेशन, OpenAI-संगत API डिटेक्शन, LangServe डिटेक्शन और प्रॉम्प्ट लीक परीक्षण के लिए 5 कस्टम Nuclei YAML टेम्पलेट।

---

समर्थित प्रोटोकॉल

प्रोटोकॉल	डिटेक्शन विधि	Shodan क्वेरी
MCP (मॉडल कॉन्टेक्स्ट प्रोटोकॉल)	SSE ट्रांसपोर्ट, JSON-RPC, /mcp/sse पथ	4 queries
Ollama	डिफ़ॉल्ट पोर्ट 11434, उत्पाद फिंगरप्रिंट	3 queries
vLLM / LiteLLM / LocalAI	/v1/models, /v1/chat/completions एंडपॉइंट	4 queries
LangServe / LangChain	प्लेग्राउंड एंडपॉइंट, langserve मार्कर	2 queries
OpenClaw / Clawdbot	कंट्रोल डैशबोर्ड, पोर्ट 18789	3 queries
Open WebUI / LibreChat	शीर्षक-आधारित डिटेक्शन	2 queries
Gradio	शीर्षक, फुटर वॉटरमार्क, फ़ेविकॉन हैश	3 queries
Streamlit	शीर्षक, फ़ेविकॉन हैश	2 queries
ComfyUI / Stable Diffusion	शीर्षक, पोर्ट-आधारित डिटेक्शन	4 queries
HuggingFace TGI	HTML मार्कर	1 query
जेनेरिक इनफ़रेंस	/api/generate, /api/tags पथ	2 queries

---

जोखिम स्कोरिंग

प्रत्येक एंडपॉइंट को 0–10 का जोखिम स्कोर मिलता है जिसकी गणना निम्न से की जाती है:

कारक	स्कोर प्रभाव
कोई प्रमाणीकरण नहीं	+4.0
अज्ञात ऑथ स्थिति	+1.0
10+ टूल एक्सपोज़्ड	+2.0
5+ टूल एक्सपोज़्ड	+1.0
क्रिटिकल-जोखिम टूल (जैसे, exec_code, run_shell)	+1.0 प्रत्येक
उच्च-जोखिम टूल (जैसे, query_db, file_read)	+0.5 प्रत्येक
ओपन CORS (*)	+1.0
कोई TLS नहीं	+0.5
सिस्टम प्रॉम्प्ट लीक हुआ	+0.5
मॉडल एक्सपोज़्ड	+1.0
अनसेंसर्ड मॉडल पाया गया	+2.0
साइनअप सक्षम (कोई आमंत्रण आवश्यक नहीं)	+1.5
खतरनाक संयोजन (जैसे, कोई ऑथ नहीं + कोड निष्पादन टूल)	+1.0 प्रत्येक

---

सेटअप

पूर्वापेक्षाएँ

• Python 3.12+
• Node.js 18+
• MongoDB 7+
• Redis 7+ (वैकल्पिक — स्थानीय देव के लिए इन-मेमोरी पर वापस आ जाता है)
• Nuclei (वैकल्पिक — सक्रिय स्कैनिंग के लिए आवश्यक)
• Shodan API कुंजी (डिस्कवरी स्कैन के लिए आवश्यक)

त्वरित प्रारंभ (Docker Compose)

# क्लोन करें
git clone [email protected]:BishopFox/aimap.git
cd aimap

# कॉन्फ़िगर करें
cp .env.example .env
# .env संपादित करें — कम से कम SHODAN_API_KEY सेट करें

# लॉन्च करें
docker compose up --build

यह 4 सेवाएँ शुरू करता है:

• MongoDB पोर्ट 27017 पर
• Redis पोर्ट 6379 पर
• बैकएंड पोर्ट 8000 पर
• फ्रंटएंड पोर्ट 80 पर

UI तक पहुँचने के लिए http://localhost खोलें।

स्थानीय विकास (Docker के बिना)

# बैकएंड
cd backend
python -m venv .venv
source .venv/bin/activate
pip install -r requirements.txt
uvicorn app.main:app --reload --port 8000

# फ्रंटएंड (अलग टर्मिनल)
cd frontend
npm install
npm run dev   # http://localhost:5173 पर शुरू होता है

सुनिश्चित करें कि MongoDB स्थानीय रूप से पोर्ट 27017 पर चल रहा है। Redis वैकल्पिक है — जब Redis अनुपलब्ध होता है तो बैकएंड इन-मेमोरी बफर पर वापस आ जाता है।

पर्यावरण चर

प्रोजेक्ट रूट में .env फ़ाइल बनाएँ:

# आवश्यक
SHODAN_API_KEY=your_shodan_api_key

# वैकल्पिक — अतिरिक्त डिस्कवरी स्रोत के रूप में Censys
CENSYS_API_ID=
CENSYS_API_SECRET=

# वैकल्पिक — AI-संचालित अटैक विश्लेषण को सक्षम करता है
ANTHROPIC_API_KEY=

# MongoDB (डिफ़ॉल्ट स्थानीय देव के लिए काम करते हैं)
MONGODB_URI=mongodb://localhost:27017
MONGODB_DB=aimap

# Redis (डिफ़ॉल्ट स्थानीय देव के लिए काम करते हैं; वैकल्पिक)
REDIS_URL=redis://localhost:6379/0

# CORS (डिफ़ॉल्ट सभी ऑरिजिन की अनुमति देता है)
CORS_ORIGINS=*

# Modal सर्वरलेस (स्कैन/अटैक को Modal कंटेनरों में भेजता है)
MODAL_ENABLED=false

# Clerk ऑथ — नीचे देखें
CLERK_ISSUER=

प्रमाणीकरण (Clerk)

AIMap प्रमाणीकरण के लिए Clerk का उपयोग करता है। सक्षम करने के लिए:

1. clerk.com पर एक Clerk एप्लिकेशन बनाएँ
2. बैकएंड जारीकर्ता URL सेट करें:

   # .env (प्रोजेक्ट रूट)
   CLERK_ISSUER=https://your-app.clerk.accounts.dev
   

3. फ्रंटएंड प्रकाशनीय कुंजी सेट करें:

   # frontend/.env.local
   VITE_CLERK_PUBLISHABLE_KEY=pk_test_...
   

प्रमाणीकरण अक्षम करने के लिए (स्थानीय देव, डेमो): CLERK_ISSUER को खाली या अनसेट छोड़ दें। बैकएंड सभी अनुरोधों को एक सिंथेटिक local उपयोगकर्ता पहचान के साथ स्वीकार करेगा।

---

उपयोग

डिस्कवरी स्कैन चलाना

1. साइडबार में Scans पर जाएँ
2. New Scan पर क्लिक करें
3. क्वेरी प्रीसेट चुनें (जैसे, ollama, mcp_protocol, vllm) या कस्टम Shodan क्वेरी दर्ज करें
4. वैकल्पिक रूप से CIDR रेंज में स्कोप करें (ऑर्केस्ट्रेटर प्रत्येक क्वेरी में net:<cidr> जोड़ता है)
5. Run पर क्लिक करें — स्कैन पाइपलाइन निष्पादित होती है:
   • Shodan खोज — मिलान करने वाले होस्ट खींचता है
   • httpx स्वीप — होस्ट के जीवित होने की पुष्टि करता है
   • Nuclei स्कैन — जीवित होस्ट के विरुद्ध कस्टम टेम्पलेट चलाता है
   • एनरिचमेंट — फ्रेमवर्क डिटेक्शन, ऑथ प्रोबिंग, जोखिम स्कोरिंग
6. रीयल-टाइम WebSocket स्थिति बार या स्कैन विवरण पृष्ठ को पोल करके प्रगति की निगरानी करें

एंडपॉइंट खोजना

Shodan-शैली क्वेरी सिंटैक्स के साथ सर्च बार का उपयोग करें: