entrypoint.sh

#!/bin/bash
set -e

# ReconFlow Docker Entrypoint Script
# Handles initialization and tool validation

echo "=========================================="
echo "  ReconFlow Docker Container"
echo "=========================================="
echo ""

# Color codes for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Function to print colored output
print_status() {
    echo -e "${GREEN}[+]${NC} $1"
}

print_warning() {
    echo -e "${YELLOW}[!]${NC} $1"
}

print_error() {
    echo -e "${RED}[-]${NC} $1"
}

# Check if required tools are installed
check_tools() {
    print_status "Checking required tools..."
    
    local required_tools=(
        "subfinder"
        "httpx"
        "ffuf"
        "masscan"
        "nmap"
        "katana"
        "dnsx"
        "gau"
        "waybackurls"
    )
    
    local missing_tools=()
    
    for tool in "${required_tools[@]}"; do
        if command -v "$tool" &> /dev/null; then
            version=$($tool -version 2>/dev/null || echo "installed")
            echo "  $tool: $version"
        else
            print_warning "  $tool: not found"
            missing_tools+=("$tool")
        fi
    done
    
    if [ ${#missing_tools[@]} -gt 0 ]; then
        print_warning "Some tools are missing. The container may not function properly."
        print_warning "Missing tools: ${missing_tools[*]}"
    else
        print_status "All required tools are installed"
    fi
}

# Initialize subfinder API configuration
init_subfinder() {
    print_status "Initializing subfinder configuration..."
    mkdir -p "$HOME/.config/subfinder"
}

# Create output directories
setup_output() {
    print_status "Setting up output directories..."
    mkdir -p /app/output
    chmod 755 /app/output
}

# Check capabilities
check_capabilities() {
    print_status "Checking container capabilities..."
    
    # Check for NET_RAW capability (needed for masscan)
    if capsh --print 2>/dev/null | grep -q "cap_net_raw"; then
        print_status "NET_RAW capability: available"
    else
        print_warning "NET_RAW capability: not available (needed for masscan)"
        print_warning "Run with: docker run --cap-add=NET_RAW ..."
    fi
    
    # Check for NET_ADMIN capability (needed for masscan)
    if capsh --print 2>/dev/null | grep -q "cap_net_admin"; then
        print_status "NET_ADMIN capability: available"
    else
        print_warning "NET_ADMIN capability: not available (needed for masscan)"
        print_warning "Run with: docker run --cap-add=NET_ADMIN ..."
    fi
    
    # Check for SYS_ADMIN capability (needed for some scanning)
    if capsh --print 2>/dev/null | grep -q "cap_sys_admin"; then
        print_status "SYS_ADMIN capability: available"
    else
        print_warning "SYS_ADMIN capability: not available"
    fi
}

# Show usage information
show_usage() {
    echo ""
    echo "ReconFlow - Automated Reconnaissance Framework"
    echo ""
    echo "Usage:"
    echo "  reconflow <command> [options]"
    echo ""
    echo "Commands:"
    echo "  scan     Start a new reconnaissance scan"
    echo "  resume   Resume an interrupted scan"
    echo "  install  Install required tools"
    echo "  version  Show version"
    echo "  help     Show this help message"
    echo ""
    echo "Examples:"
    echo "  reconflow scan -d example.com --full"
    echo "  reconflow resume -d example.com"
    echo "  reconflow install --check"
    echo ""
    echo "Docker Examples:"
    echo "  docker run --rm -it \\"
    echo "    --cap-add NET_RAW --cap-add NET_ADMIN \\"
    echo "    -v \$(pwd)/output:/app/output \\"
    echo "    reconflow scan -d example.com --full"
    echo ""
}

# Handle commands
case "${1:-}" in
    --help|-h|"")
        show_usage
        check_tools
        check_capabilities
        setup_output
        exec reconflow "$@"
        ;;
    scan|resume|install|version|help)
        check_tools
        check_capabilities
        setup_output
        exec reconflow "$@"
        ;;
    *)
        print_error "Unknown command: $1"
        show_usage
        exit 1
        ;;
esac