Skip to content

rainmana/tinybrain

BranchesTags

Repository files navigation

TinyBrain 🧠

License: MIT Go Version MCP Protocol Security Focused Version

Security-Focused LLM Memory Storage with Intelligence Gathering, Reverse Engineering, and MITRE ATT&CK Integration

TinyBrain is a comprehensive memory storage system designed specifically for security professionals, penetration testers, and AI assistants working on offensive security tasks. It provides intelligent memory management, pattern recognition, and comprehensive intelligence gathering capabilities through the Model Context Protocol (MCP).

📖 View Complete Documentation | 🐛 Report Issues | 💬 Discussions

✨ Key Features

🧠 Intelligence Gathering

  • OSINT: Open Source Intelligence collection and analysis
  • HUMINT: Human Intelligence gathering and social engineering assessment
  • SIGINT: Signals Intelligence and communications analysis
  • GEOINT: Geospatial Intelligence and location-based analysis
  • MASINT: Measurement and Signature Intelligence
  • TECHINT: Technical Intelligence and technology assessment
  • FININT: Financial Intelligence and cryptocurrency tracking
  • CYBINT: Cyber Intelligence and threat analysis

🔍 Reverse Engineering

  • Malware Analysis: Static and dynamic malware analysis capabilities
  • Binary Analysis: PE, ELF, Mach-O file format analysis
  • Vulnerability Research: Fuzzing, exploit development, and vulnerability analysis
  • Protocol Analysis: Network and application protocol reverse engineering
  • Code Analysis: Source code and assembly analysis tools

🎯 MITRE ATT&CK Integration

  • Complete Framework: All 14 Enterprise tactics and 200+ techniques
  • TTP Mapping: Map findings to specific tactics, techniques, and procedures
  • Attack Chain Analysis: Complete attack chain mapping and analysis
  • Threat Hunting: Hunt for specific TTPs and attack patterns
  • Campaign Tracking: Track attack campaigns and threat actor activities
  • Real-Time Data: Live MITRE ATT&CK dataset with 823+ techniques and 14 tactics
  • Intelligent Querying: Semantic search across attack techniques and procedures

🛡️ Security Patterns & Standards

  • CWE Integration: Common Weakness Enumeration patterns and classifications
  • OWASP Compliance: OWASP Top 10 2021 and testing guide integration
  • Multi-Language Support: Security patterns for 10+ programming languages
  • Authorization Templates: RBAC, ABAC, and DAC access control patterns
  • Standards Compliance: NIST, ISO 27001, PTES, and industry standards
  • NVD Integration: National Vulnerability Database with 314,835+ CVEs
  • OWASP Testing Guide: Complete web application security testing procedures

🔬 Security Knowledge Hub

  • NVD Integration: National Vulnerability Database with 314,835+ CVEs
  • MITRE ATT&CK: Complete framework with 823+ techniques and 14 tactics
  • OWASP Testing Guide: Comprehensive web application security testing procedures
  • Intelligent Retrieval: Context-aware querying with summarization
  • Real-Time Updates: Incremental data updates from official sources
  • Rate Limiting: Respectful API usage with proper rate limiting

📊 Memory Management

  • 30+ Memory Categories: Comprehensive categorization for intelligence, reconnaissance, and analysis data
  • Intelligence Objects: Threat actors, attack campaigns, IOCs, TTPs, patterns, and correlations
  • Context-Aware Storage: Automatically categorizes and prioritizes information
  • Advanced Search: Semantic, exact, fuzzy, tag-based, and relationship-based search
  • Access Tracking: Monitors which memories are most relevant and frequently accessed
  • Context Summaries: Provides relevant memory summaries for current tasks

High Performance & Reliability

  • Embedded SQLite Backend: Single binary with a pure-Go SQLite engine (no cgo) and FTS5 full-text search
  • Dashboard: Web-based status dashboard at http://127.0.0.1:8090/_/ (in serve mode)
  • REST API: Full REST API at http://127.0.0.1:8090/api/ for external integrations
  • MCP Protocol: JSON-RPC 2.0 over stdio for AI assistant integration
  • Optimized Queries: Indexed searches and efficient relationship traversal
  • Transaction Safety: ACID compliance for data integrity
  • Concurrent Access: Thread-safe operations for multiple LLM interactions
  • Zero Configuration: Works out of the box with minimal setup

AI-Enhanced Search & Intelligence

  • Semantic Search: AI-powered memory search using embeddings for conceptual similarity
  • Embedding Generation: Generate embeddings for text (placeholder for AI integration)
  • Similarity Calculation: Calculate semantic similarity between embeddings
  • Future-Ready: Complete foundation for OpenAI, Cohere, or local model integration

Real-Time Notifications & Alerts

  • Memory Notifications: Real-time alerts for memory events and system activities
  • High Priority Alerts: Notifications for high-priority memories (priority ≥8, confidence ≥0.8)
  • Duplicate Detection: Alerts for potential duplicate memories with similarity scores
  • Cleanup Notifications: Notifications for automated cleanup operations
  • Notification Management: Mark notifications as read, filter by session, priority-based sorting

Developer Experience

  • Simple Installation: go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest
  • Comprehensive Logging: Detailed logging with structured output
  • Extensive Testing: Full test coverage for all MCP tool handlers
  • Docker Support: Containerized deployment ready
  • 50+ MCP Tools: Complete API for all memory management operations
  • Single Binary: MCP server, REST API, and dashboard in one executable

🛠️ Complete MCP Tool Set (50+ Tools)

TinyBrain provides a comprehensive set of more than 50 MCP tools for complete LLM memory management:

Core Memory Operations (8 tools)

  • store_memory - Store new memory entries
  • get_memory - Retrieve memory by ID
  • search_memories - Advanced search with multiple strategies
  • update_memory - Update existing memory entries
  • delete_memory - Delete memory entries
  • find_similar_memories - Find similar memories by content
  • check_duplicates - Check for duplicate memories
  • get_memory_stats - Get comprehensive memory statistics

Session & Task Management (6 tools)

  • create_session - Create new security assessment sessions
  • get_session - Retrieve session information
  • list_sessions - List all sessions with filtering
  • create_task_progress - Create task progress entries
  • update_task_progress - Update task progress
  • list_task_progress - List task progress entries

Advanced Memory Features (8 tools)

  • create_relationship - Create memory relationships
  • get_related_memories - Get related memories
  • create_context_snapshot - Create context snapshots
  • get_context_snapshot - Retrieve context snapshots
  • list_context_snapshots - List context snapshots
  • get_context_summary - Get memory summaries for context
  • export_session_data - Export session data
  • import_session_data - Import session data

Security Templates & Batch Operations (6 tools)

  • get_security_templates - Get predefined security templates
  • create_memory_from_template - Create memories from templates
  • batch_create_memories - Bulk create memory entries
  • batch_update_memories - Bulk update memory entries
  • batch_delete_memories - Bulk delete memory entries
  • get_detailed_memory_info - Get detailed memory debugging info

Memory Lifecycle & Cleanup (4 tools)

  • cleanup_old_memories - Age-based memory cleanup
  • cleanup_low_priority_memories - Priority-based cleanup
  • cleanup_unused_memories - Access-based cleanup
  • get_system_diagnostics - Comprehensive system diagnostics

AI-Enhanced Search (3 tools)

  • semantic_search - AI-powered semantic search
  • generate_embedding - Generate embeddings for text
  • calculate_similarity - Calculate embedding similarity

Real-Time Notifications (4 tools)

  • get_notifications - Get notifications and alerts
  • mark_notification_read - Mark notifications as read
  • check_high_priority_memories - Check for high-priority alerts
  • check_duplicate_memories - Check for duplicate alerts

System Monitoring (1 tool)

  • health_check - Perform system health checks

🛡️ Security Standards & Source Attribution

Standards Compliance

TinyBrain's security patterns and vulnerability datasets are aligned with industry-standard security frameworks:

Source Attribution

Our security patterns and vulnerability datasets are based on authoritative sources:

Multi-Language Coverage

Our security patterns cover 10 major programming languages with language-specific vulnerability patterns:

  • JavaScript/Node.js - Web application security patterns
  • Python - Backend and automation security patterns
  • Java - Enterprise application security patterns
  • C#/.NET - Microsoft ecosystem security patterns
  • PHP - Web application security patterns
  • Ruby - Web framework security patterns
  • Go - System and API security patterns
  • C/C++ - System-level security patterns
  • TypeScript - Type-safe web application patterns
  • Rust - Memory-safe system programming patterns

🚀 Quick Start

Installation

Option 1: Pre-built Binaries (Recommended)

Download the latest release for your platform from Releases:

  • macOS (Apple Silicon): tinybrain_*_Darwin_arm64.tar.gz
  • macOS (Intel): tinybrain_*_Darwin_x86_64.tar.gz
  • Linux (x86_64): tinybrain_*_Linux_x86_64.tar.gz
  • Linux (ARM64): tinybrain_*_Linux_arm64.tar.gz
  • Windows: tinybrain_*_Windows_x86_64.zip

Extract and run:

# Extract (Linux/macOS)
tar -xzf tinybrain_*_*.tar.gz
cd tinybrain_*

# Make executable
chmod +x tinybrain

# Run
./tinybrain serve

Option 2: Install from Source with go install

# Install latest version
go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

# Install specific version
go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

# The binary will be installed as 'tinybrain' in your $GOPATH/bin or $GOBIN
# Make sure $GOPATH/bin or $GOBIN is in your PATH

Option 3: Clone and Build Locally

git clone https://github.com/rainmana/tinybrain.git
cd tinybrain
go build -o tinybrain ./cmd/tinybrain

Option 4: Docker

docker pull rainmana/tinybrain:latest
docker run -p 8090:8090 rainmana/tinybrain

Claude Code Plugin (MCP server + skills in one install)

TinyBrain ships as a Claude Code plugin that registers the MCP server and installs skills teaching Claude how to use it well (capture findings as you go, build an engagement knowledge graph, recall it across sessions). The plugin includes a core using-tinybrain skill plus domain workflows for code review, CTFs, and threat intelligence.

First install the binary (the plugin's MCP config runs tinybrain from your PATH):

go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

Then add the marketplace and install the plugin from inside Claude Code:

/plugin marketplace add rainmana/tinybrain
/plugin install tinybrain@tinybrain-marketplace

That's it — the tinybrain MCP server is registered (pointed at ~/.tinybrain/memory.db) and the skills activate automatically when you do security work. To point at a different database, set TINYBRAIN_DB_PATH in your environment before launching Claude Code.

Basic Usage

# Start the server (data stored in ~/.tinybrain by default)
tinybrain serve

# Or specify a custom data directory
tinybrain serve --dir /path/to/your/data

# Customize the HTTP host/port (default is 127.0.0.1:8090)
tinybrain serve --port 9000
tinybrain serve --host 0.0.0.0 --port 8090

# Or use environment variables
TINYBRAIN_HOST=0.0.0.0 TINYBRAIN_PORT=9000 tinybrain serve

# Access the dashboard
open http://127.0.0.1:8090/_/

# The REST API is available at:
# http://127.0.0.1:8090/api/

In serve mode TinyBrain speaks the MCP protocol over stdio and serves the REST API/dashboard over HTTP. Running plain tinybrain (no subcommand) starts a stdio-only MCP server, which is what most MCP client configs use.

Server Features

  • Single Binary: Everything in one executable with zero configuration
  • Dashboard: Web interface at http://127.0.0.1:8090/_/ for stats and links
  • REST API: Sessions and memory endpoints at http://127.0.0.1:8090/api/
  • Optional Auth: Set TINYBRAIN_API_TOKEN to require a bearer token on the REST API
  • Data Persistence: SQLite database at <dir>/memory.db (default ~/.tinybrain/memory.db)
  • MCP over stdio: JSON-RPC 2.0; works with Claude Desktop, Cursor, Cline, and any MCP client

Intelligence Gathering Example

// Create an OSINT intelligence session
session := &Session{
    Name: "OSINT Intelligence Gathering",
    TaskType: "intelligence_analysis",
    IntelligenceType: "osint",
    Classification: "unclassified",
    ThreatLevel: "medium",
}

// Store intelligence findings
finding := &IntelligenceFinding{
    Title: "Social Media Intelligence",
    IntelligenceType: "osint",
    ThreatLevel: "medium",
    MITRETactic: "TA0001",
    MITRETechnique: "T1591",
    KillChainPhase: "reconnaissance",
}

MCP Integration

{
  "jsonrpc": "2.0",
  "id": 1,
  "method": "tools/call",
  "params": {
    "name": "mcp_tinybrain-mcp-server_create_session",
    "arguments": {
      "name": "Security Assessment",
      "task_type": "penetration_test",
      "intelligence_type": "osint"
    }
  }
}

Troubleshooting Installation

If you encounter issues with go install, try these solutions:

# Standard install (note the /v3 in the module path — required for v3.x)
go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

# Build from a clone instead
git clone https://github.com/rainmana/tinybrain.git
cd tinybrain
go build -o tinybrain ./cmd/tinybrain

# For Go module proxy issues, use direct mode
GOPROXY=direct go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

# For checksum database errors right after a new release
GOSUMDB=off go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest

Common Issues:

  • "module path must match major version" error: Include the /v3 in the path — go install github.com/rainmana/tinybrain/v3/cmd/tinybrain@latest
  • Binary not found: Ensure $GOPATH/bin or $GOBIN is in your PATH
  • Checksum errors: Wait a few minutes after a new release, or temporarily use GOSUMDB=off

MCP Client Configuration

Add to your MCP client configuration (e.g., Claude Desktop):

{
  "mcpServers": {
    "tinybrain": {
      "command": "tinybrain",
      "args": ["serve"],
      "env": {
        "TINYBRAIN_DB_PATH": "~/.tinybrain/memory.db"
      }
    }
  }
}

Note: The binary name is tinybrain (from cmd/tinybrain); go install github.com/rainmana/tinybrain/v3/cmd/server@latest installs the same program as server. The SQLite database is created at ~/.tinybrain/memory.db by default.

📚 Documentation

For complete documentation, API reference, and detailed guides, visit our comprehensive documentation site:

📖 Complete Documentation

The documentation includes:

Quick API Reference

Session Management

Task Types: security_review, penetration_test, exploit_dev, vulnerability_analysis, threat_modeling, incident_response, intelligence_analysis

Intelligence Types: osint, humint, sigint, geoint, masint, techint, finint, cybint

List Sessions

{
  "name": "list_sessions",
  "arguments": {
    "task_type": "security_review",
    "status": "active",
    "limit": 10
  }
}

Memory Operations

Store Memory

{
  "name": "store_memory",
  "arguments": {
    "session_id": "session_123",
    "title": "SQL Injection in Login Form",
    "content": "Found SQL injection vulnerability in username parameter of login form. Payload: ' OR 1=1--",
    "category": "vulnerability",
    "content_type": "text",
    "priority": 8,
    "confidence": 0.9,
    "tags": "[\"sql-injection\", \"authentication\", \"critical\"]",
    "source": "manual-testing"
  }
}

Categories: finding, vulnerability, exploit, payload, technique, tool, reference, context, hypothesis, evidence, recommendation, note

Search Memories

{
  "name": "search_memories",
  "arguments": {
    "query": "SQL injection authentication",
    "session_id": "session_123",
    "search_type": "semantic",
    "categories": "[\"vulnerability\", \"exploit\"]",
    "min_priority": 7,
    "limit": 20
  }
}

Search Types: semantic, exact, fuzzy, tag, category, relationship

Get Related Memories

{
  "name": "get_related_memories",
  "arguments": {
    "memory_id": "memory_456",
    "relationship_type": "exploits",
    "limit": 10
  }
}

Relationship Management

Create Relationship

{
  "name": "create_relationship",
  "arguments": {
    "source_memory_id": "memory_123",
    "target_memory_id": "memory_456",
    "relationship_type": "exploits",
    "strength": 0.8,
    "description": "SQL injection can be used to bypass authentication"
  }
}

Relationship Types: depends_on, causes, mitigates, exploits, references, contradicts, supports, related_to, parent_of, child_of

Context Management

Get Context Summary

{
  "name": "get_context_summary",
  "arguments": {
    "session_id": "session_123",
    "current_task": "Analyzing authentication vulnerabilities",
    "max_memories": 20
  }
}

🏗️ Architecture

TinyBrain is built with:

  • Go 1.24+ - High-performance backend
  • modernc.org/sqlite - Pure-Go embedded SQLite with FTS5 full-text search (no cgo)
  • MCP Protocol - LLM integration standard with 50+ tools over stdio (JSON-RPC 2.0)
  • MITRE ATT&CK - Security framework integration
  • Jekyll - Documentation site with Minimal theme

Single-Binary Benefits

  • No External Dependencies: Works anywhere Go runs; CGO-free cross-compilation
  • Embedded Database: SQLite database file managed by the binary
  • Dashboard: Built-in status dashboard at /_/
  • REST API: Full REST API for external integrations and automation
  • Zero Configuration: Works out of the box with sensible defaults
  • Data Persistence: All data automatically persisted across restarts

Key Design Principles

  1. Security-First: All data structures and operations designed for security tasks
  2. Intelligence-Focused: Comprehensive intelligence gathering and analysis capabilities
  3. Performance: Optimized queries and indexes for fast retrieval
  4. Flexibility: Extensible schema and relationship system
  5. Reliability: ACID transactions and data integrity checks
  6. Usability: Simple API with comprehensive documentation

🧪 Testing

# Run all tests
make test

# Run tests with coverage
go test -v -race -coverprofile=coverage.out ./...
go tool cover -html=coverage.out -o coverage.html

# Run benchmarks
make bench

# Run specific test
go test -v ./internal/database -run TestNewDatabase

🐳 Docker Deployment

# Build Docker image
make docker-build

# Run container
docker run --rm -it \
  -v ~/.tinybrain:/app/data \
  tinybrain:latest

📊 Performance

Benchmarks

  • Memory Entry Creation: ~1000 entries/second
  • Search Operations: ~100 searches/second
  • Relationship Queries: ~500 queries/second
  • Database Size: ~1MB per 10,000 memory entries

Optimization Features

  • Connection Pooling: Optimized for SQLite single-writer model
  • Index Strategy: Comprehensive indexing for all query patterns
  • Full-Text Search: FTS5 virtual tables for semantic search
  • Caching: Access pattern tracking for intelligent caching

🔧 Configuration

Environment Variables

  • TINYBRAIN_DB_PATH: Path to the SQLite database (default: ~/.tinybrain/memory.db)
  • TINYBRAIN_HOST: HTTP listen host in serve mode (default: 127.0.0.1)
  • TINYBRAIN_PORT: HTTP listen port in serve mode (default: 8090)
  • TINYBRAIN_HTTP: Set to false to disable the HTTP API in serve mode
  • TINYBRAIN_API_TOKEN: If set, the REST API requires this bearer token

Server Endpoints (serve mode)

Data Directory Structure

~/.tinybrain/
├── memory.db          # SQLite database
├── memory.db-wal      # Write-ahead log
└── memory.db-shm      # Shared memory file

🛡️ Security Datasets & Templates

Comprehensive Security Patterns

The security pattern libraries, CWE references, and templates are published on the documentation site:

  • Security Patterns - OWASP Top 10 2021 patterns, CWE Top 25, multi-language vulnerability patterns, and exploitation techniques
  • CWE LLM Dataset - LLM-optimized CWE dataset in structured JSON format for efficient consumption (ships in this repo)

AI Assistant Configurations

Standards-Based Approach

All security patterns are derived from authoritative sources and aligned with industry standards:

🤝 Contributing

  1. Fork the repository
  2. Create a feature branch
  3. Add tests for new functionality
  4. Ensure all tests pass
  5. Submit a pull request

Development Setup

# Clone repository
git clone https://github.com/rainmana/tinybrain.git
cd tinybrain

# Setup development environment
make dev-setup

# Run tests
make test

# Build
make build

📄 License

MIT License - see LICENSE file for details.

🙏 Acknowledgments

Technical Dependencies

Security Standards & Sources

📈 Roadmap

  • Intelligence gathering frameworks (OSINT, HUMINT, SIGINT, etc.)
  • MITRE ATT&CK integration
  • Reverse engineering capabilities
  • Enhanced memory categories
  • Comprehensive documentation site
  • HTTP transport support
  • Memory compression and archiving
  • Advanced analytics and insights
  • Multi-user support with access controls
  • Plugin system for custom memory types
  • Integration with popular security tools
  • Web dashboard for memory visualization

TinyBrain - Making LLM memory storage intelligent, fast, and security-focused. 🧠🔒

About

LLM local persistent storage for security nerds

rainmana.github.io/tinybrain/

Topics

mcp knowledge-graph penetration-testing penetration-testing-tools llm mcp-server

Resources

Readme

License

MIT license
Activity

Stars

3 stars

Watchers

0 watching

Forks

1 fork
Report repository

Releases 4

tinybrain 3.0.1 Latest
Jun 12, 2026
+ 3 releases

Packages

 
 
 

Contributors

Languages