tinybrain

module
v3.0.1 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Jun 12, 2026 License: MIT

README

TinyBrain 🧠

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

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:

# If you get "main redeclared" errors, ensure you're using v1.2.1 or later
# Older versions (v1.2.0) had duplicate files that caused build errors
go install github.com/rainmana/tinybrain/v3/cmd/server@latest

# If you get authentication errors, use direct clone method
git clone https://github.com/rainmana/tinybrain.git
cd tinybrain
go build -o server ./cmd/server

# If repository is private, ensure you have access
git config --global url."git@github.com:".insteadOf "https://github.com/"

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

# For checksum database errors (temporary issue with new releases)
GOSUMDB=off go install github.com/rainmana/tinybrain/v3/cmd/server@latest

Common Issues:

  • "main redeclared" error: You're using an old version. Use @v1.2.1 or @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

{
  "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
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. 🧠🔒

Directories

Path Synopsis
cmd
server command
Command server is an alternate entry point for TinyBrain kept for compatibility with the documented install path:
 Command server is an alternate entry point for TinyBrain kept for compatibility with the documented install path:
tinybrain command
Command tinybrain is the TinyBrain security-focused memory storage MCP server.
 Command tinybrain is the TinyBrain security-focused memory storage MCP server.
internal
app
database
mcp
models
repository
services

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.