setup.sh

#!/bin/bash

# Cobalt Strike Web Client Setup Script

echo "=== Cobalt Strike Web Client Setup ==="

# Check Python version
python_version=$(python3 --version 2>&1)
if [[ $? -ne 0 ]]; then
    echo "Error: Python 3 is not installed or not in PATH"
    exit 1
fi

echo "Using $python_version"

# Create virtual environment
echo "Creating virtual environment..."
python3 -m venv venv

# Activate virtual environment
echo "Activating virtual environment..."
if [[ "$OSTYPE" == "msys" || "$OSTYPE" == "win32" ]]; then
    source venv/Scripts/activate
else
    source venv/bin/activate
fi

# Upgrade pip
echo "Upgrading pip..."
pip install --upgrade pip

# Install dependencies
echo "Installing dependencies..."
pip install -r requirements.txt

# Create necessary directories
echo "Creating directories..."
mkdir -p logs
mkdir -p certs

# Copy environment file
if [[ ! -f .env ]]; then
    echo "Creating .env file from template..."
    cp .env.example .env
    echo "Please edit .env file with your configuration"
fi

# Generate self-signed certificate for development
if [[ ! -f certs/cert.pem ]]; then
    echo "Generating self-signed certificate for development..."
    openssl req -x509 -newkey rsa:4096 -keyout certs/key.pem -out certs/cert.pem -days 365 -nodes \
        -subj "/C=US/ST=State/L=City/O=Organization/CN=localhost"
fi

# Set file permissions
chmod 600 certs/key.pem
chmod 644 certs/cert.pem

echo ""
echo "=== Setup Complete ==="
echo ""
echo "Next steps:"
echo "1. Edit the .env file with your Cobalt Strike server details"
echo "2. Ensure your Cobalt Strike team server has REST API enabled"
echo "3. Run the application:"
echo "   python run.py"
echo ""
echo "The application will be available at: https://127.0.0.1:5000"
echo ""
echo "Security Notes:"
echo "- Change the SECRET_KEY in .env before production use"
echo "- Use proper SSL certificates in production"
echo "- Ensure firewall rules are properly configured"
echo "- Monitor logs for security events"
echo ""