feat: Update requirements.txt with test dependencies and enhance install script
Some checks reported errors
continuous-integration/drone/push Build was killed
Some checks reported errors
continuous-integration/drone/push Build was killed
- Add pytest, coverage, linting tools to requirements.txt - Update install script for v2.1.0 with auth system info - Mention JWT authentication and security features in output
This commit is contained in:
@@ -1,339 +0,0 @@
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: pyguardian-ci
|
||||
|
||||
platform:
|
||||
os: linux
|
||||
arch: amd64
|
||||
|
||||
# Build triggers
|
||||
trigger:
|
||||
branch:
|
||||
- main
|
||||
- develop
|
||||
event:
|
||||
- push
|
||||
- pull_request
|
||||
- tag
|
||||
|
||||
# Global environment variables
|
||||
environment:
|
||||
PYTHON_VERSION: "3.11"
|
||||
POETRY_VERSION: "1.7.0"
|
||||
|
||||
steps:
|
||||
# Code quality and testing pipeline
|
||||
- name: lint-and-test
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
commands:
|
||||
# Install system dependencies
|
||||
- apt-get update && apt-get install -y git curl
|
||||
|
||||
# Install Python dependencies
|
||||
- pip install --upgrade pip
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio pytest-cov flake8 black mypy
|
||||
|
||||
# Code formatting check
|
||||
- black --check src/ tests/
|
||||
|
||||
# Lint code
|
||||
- flake8 src/ --max-line-length=88 --extend-ignore=E203,W503
|
||||
|
||||
# Type checking
|
||||
- mypy src/ --ignore-missing-imports
|
||||
|
||||
# Run unit tests with coverage
|
||||
- pytest tests/unit/ -v --cov=src --cov-report=xml --cov-report=term
|
||||
|
||||
# Security check for dependencies
|
||||
- pip install safety
|
||||
- safety check
|
||||
|
||||
# Integration tests
|
||||
- name: integration-tests
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
TEST_DATABASE_URL: sqlite:///tmp/test.db
|
||||
commands:
|
||||
- apt-get update && apt-get install -y iptables curl
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio
|
||||
- pytest tests/integration/ -v
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Build Docker images
|
||||
- name: build-docker-images
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
DOCKER_BUILDKIT: 1
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Build controller image
|
||||
- docker build -f deployment/docker/Dockerfile --target controller -t pyguardian:controller-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build agent image
|
||||
- docker build -f deployment/docker/Dockerfile --target agent -t pyguardian:agent-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build standalone image
|
||||
- docker build -f deployment/docker/Dockerfile --target standalone -t pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Test images can start
|
||||
- timeout 30 docker run --rm pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} python --version
|
||||
depends_on:
|
||||
- integration-tests
|
||||
|
||||
# Security scanning
|
||||
- name: security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Scan for vulnerabilities in built images
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:controller-${DRONE_COMMIT_SHA:0:8}
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:agent-${DRONE_COMMIT_SHA:0:8}
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # Don't fail build on security issues, but report them
|
||||
|
||||
# End-to-end tests
|
||||
- name: e2e-tests
|
||||
image: docker/compose:latest
|
||||
environment:
|
||||
COMPOSE_FILE: deployment/docker/docker-compose.yml
|
||||
TELEGRAM_BOT_TOKEN: test_token
|
||||
CLUSTER_SECRET: test_secret
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Start services
|
||||
- docker-compose -f deployment/docker/docker-compose.yml up -d
|
||||
|
||||
# Wait for services to be ready
|
||||
- sleep 30
|
||||
|
||||
# Run E2E tests
|
||||
- python tests/e2e/test_cluster_communication.py
|
||||
|
||||
# Cleanup
|
||||
- docker-compose -f deployment/docker/docker-compose.yml down -v
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # E2E tests are flaky in CI
|
||||
|
||||
# Documentation build
|
||||
- name: build-docs
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material
|
||||
- mkdocs build --strict
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Package creation
|
||||
- name: create-packages
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
# Create installation package
|
||||
- tar -czf pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz \
|
||||
src/ config/ main.py requirements.txt deployment/scripts/
|
||||
|
||||
# Create checksums
|
||||
- sha256sum pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz > checksums.txt
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
- build-docs
|
||||
|
||||
# Release workflow (only on tags)
|
||||
- name: docker-registry-push
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
REGISTRY:
|
||||
from_secret: docker_registry
|
||||
REGISTRY_USERNAME:
|
||||
from_secret: docker_username
|
||||
REGISTRY_PASSWORD:
|
||||
from_secret: docker_password
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Login to registry
|
||||
- docker login -u $REGISTRY_USERNAME -p $REGISTRY_PASSWORD $REGISTRY
|
||||
|
||||
# Tag and push images
|
||||
- docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
- docker push $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
# Also tag as latest if this is a release
|
||||
- |
|
||||
if [ "$DRONE_TAG" != "" ]; then
|
||||
docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-latest
|
||||
docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-latest
|
||||
docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-latest
|
||||
|
||||
docker push $REGISTRY/pyguardian:controller-latest
|
||||
docker push $REGISTRY/pyguardian:agent-latest
|
||||
docker push $REGISTRY/pyguardian:standalone-latest
|
||||
fi
|
||||
depends_on:
|
||||
- create-packages
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# GitHub Release
|
||||
- name: github-release
|
||||
image: plugins/github-release
|
||||
settings:
|
||||
api_key:
|
||||
from_secret: github_token
|
||||
files:
|
||||
- pyguardian-*.tar.gz
|
||||
- checksums.txt
|
||||
title: "PyGuardian ${DRONE_TAG}"
|
||||
note: |
|
||||
## PyGuardian Release ${DRONE_TAG}
|
||||
|
||||
### Features
|
||||
- Advanced agent authentication with JWT tokens
|
||||
- Centralized cluster management
|
||||
- Secure API endpoints for agent communication
|
||||
- Docker containerization support
|
||||
|
||||
### Installation
|
||||
```bash
|
||||
# Download and extract
|
||||
wget https://github.com/SmartSolTech/PyGuardian/releases/download/${DRONE_TAG}/pyguardian-${DRONE_TAG}.tar.gz
|
||||
tar -xzf pyguardian-${DRONE_TAG}.tar.gz
|
||||
|
||||
# Install
|
||||
sudo ./deployment/scripts/install.sh
|
||||
```
|
||||
|
||||
### Docker
|
||||
```bash
|
||||
# Pull images
|
||||
docker pull ${REGISTRY}/pyguardian:controller-${DRONE_TAG}
|
||||
docker pull ${REGISTRY}/pyguardian:agent-${DRONE_TAG}
|
||||
|
||||
# Run with docker-compose
|
||||
curl -O https://raw.githubusercontent.com/SmartSolTech/PyGuardian/${DRONE_TAG}/deployment/docker/docker-compose.yml
|
||||
docker-compose up -d
|
||||
```
|
||||
depends_on:
|
||||
- docker-registry-push
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Deployment notification
|
||||
- name: notify-deployment
|
||||
image: plugins/webhook
|
||||
settings:
|
||||
urls:
|
||||
from_secret: deployment_webhook
|
||||
content_type: application/json
|
||||
template: |
|
||||
{
|
||||
"text": "🚀 PyGuardian ${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}} deployed successfully!",
|
||||
"attachments": [{
|
||||
"color": "good",
|
||||
"fields": [{
|
||||
"title": "Version",
|
||||
"value": "${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}",
|
||||
"short": true
|
||||
}, {
|
||||
"title": "Commit",
|
||||
"value": "${DRONE_COMMIT_MESSAGE}",
|
||||
"short": false
|
||||
}]
|
||||
}]
|
||||
}
|
||||
depends_on:
|
||||
- github-release
|
||||
when:
|
||||
status:
|
||||
- success
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Volumes for Docker in Docker
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
host:
|
||||
path: /var/run/docker.sock
|
||||
|
||||
---
|
||||
# Separate pipeline for nightly builds
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: nightly-security-scan
|
||||
|
||||
trigger:
|
||||
cron:
|
||||
- nightly-security
|
||||
|
||||
steps:
|
||||
- name: dependency-security-scan
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install safety bandit semgrep
|
||||
|
||||
# Check for known vulnerable dependencies
|
||||
- safety check --json --output safety-report.json || true
|
||||
|
||||
# Static security analysis
|
||||
- bandit -r src/ -f json -o bandit-report.json || true
|
||||
|
||||
# Semgrep security rules
|
||||
- semgrep --config=auto src/ --json --output semgrep-report.json || true
|
||||
|
||||
# Upload results to security dashboard
|
||||
- python deployment/scripts/upload-security-reports.py
|
||||
|
||||
- name: container-security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Build fresh images
|
||||
- docker build -t pyguardian:security-scan .
|
||||
|
||||
# Comprehensive vulnerability scan
|
||||
- trivy image --format json --output trivy-report.json pyguardian:security-scan
|
||||
|
||||
# Upload to security dashboard
|
||||
- python deployment/scripts/upload-trivy-report.py
|
||||
|
||||
---
|
||||
# Documentation deployment pipeline
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: docs-deployment
|
||||
|
||||
trigger:
|
||||
branch:
|
||||
- main
|
||||
path:
|
||||
include:
|
||||
- "documentation/**"
|
||||
- "*.md"
|
||||
|
||||
steps:
|
||||
- name: build-and-deploy-docs
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
GITHUB_TOKEN:
|
||||
from_secret: github_token
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material mkdocs-git-revision-date-localized-plugin
|
||||
- mkdocs gh-deploy --force
|
||||
@@ -1,339 +0,0 @@
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: pyguardian-ci
|
||||
|
||||
platform:
|
||||
os: linux
|
||||
arch: amd64
|
||||
|
||||
# Build triggers
|
||||
trigger:
|
||||
branch:
|
||||
- main
|
||||
- develop
|
||||
event:
|
||||
- push
|
||||
- pull_request
|
||||
- tag
|
||||
|
||||
# Global environment variables
|
||||
environment:
|
||||
PYTHON_VERSION: "3.11"
|
||||
POETRY_VERSION: "1.7.0"
|
||||
|
||||
steps:
|
||||
# Code quality and testing pipeline
|
||||
- name: lint-and-test
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
commands:
|
||||
# Install system dependencies
|
||||
- apt-get update && apt-get install -y git curl
|
||||
|
||||
# Install Python dependencies
|
||||
- pip install --upgrade pip
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio pytest-cov flake8 black mypy
|
||||
|
||||
# Code formatting check
|
||||
- black --check src/ tests/
|
||||
|
||||
# Lint code
|
||||
- flake8 src/ --max-line-length=88 --extend-ignore=E203,W503
|
||||
|
||||
# Type checking
|
||||
- mypy src/ --ignore-missing-imports
|
||||
|
||||
# Run unit tests with coverage
|
||||
- pytest tests/unit/ -v --cov=src --cov-report=xml --cov-report=term
|
||||
|
||||
# Security check for dependencies
|
||||
- pip install safety
|
||||
- safety check
|
||||
|
||||
# Integration tests
|
||||
- name: integration-tests
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
TEST_DATABASE_URL: sqlite:///tmp/test.db
|
||||
commands:
|
||||
- apt-get update && apt-get install -y iptables curl
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio
|
||||
- pytest tests/integration/ -v
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Build Docker images
|
||||
- name: build-docker-images
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
DOCKER_BUILDKIT: 1
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Build controller image
|
||||
- docker build -f deployment/docker/Dockerfile --target controller -t pyguardian:controller-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build agent image
|
||||
- docker build -f deployment/docker/Dockerfile --target agent -t pyguardian:agent-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build standalone image
|
||||
- docker build -f deployment/docker/Dockerfile --target standalone -t pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Test images can start
|
||||
- timeout 30 docker run --rm pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} python --version
|
||||
depends_on:
|
||||
- integration-tests
|
||||
|
||||
# Security scanning
|
||||
- name: security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Scan for vulnerabilities in built images
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:controller-${DRONE_COMMIT_SHA:0:8}
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:agent-${DRONE_COMMIT_SHA:0:8}
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # Don't fail build on security issues, but report them
|
||||
|
||||
# End-to-end tests
|
||||
- name: e2e-tests
|
||||
image: docker/compose:latest
|
||||
environment:
|
||||
COMPOSE_FILE: deployment/docker/docker-compose.yml
|
||||
TELEGRAM_BOT_TOKEN: test_token
|
||||
CLUSTER_SECRET: test_secret
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Start services
|
||||
- docker-compose -f deployment/docker/docker-compose.yml up -d
|
||||
|
||||
# Wait for services to be ready
|
||||
- sleep 30
|
||||
|
||||
# Run E2E tests
|
||||
- python tests/e2e/test_cluster_communication.py
|
||||
|
||||
# Cleanup
|
||||
- docker-compose -f deployment/docker/docker-compose.yml down -v
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # E2E tests are flaky in CI
|
||||
|
||||
# Documentation build
|
||||
- name: build-docs
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material
|
||||
- mkdocs build --strict
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Package creation
|
||||
- name: create-packages
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
# Create installation package
|
||||
- tar -czf pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz \
|
||||
src/ config/ main.py requirements.txt deployment/scripts/
|
||||
|
||||
# Create checksums
|
||||
- sha256sum pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz > checksums.txt
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
- build-docs
|
||||
|
||||
# Release workflow (only on tags)
|
||||
- name: docker-registry-push
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
REGISTRY:
|
||||
from_secret: docker_registry
|
||||
REGISTRY_USERNAME:
|
||||
from_secret: docker_username
|
||||
REGISTRY_PASSWORD:
|
||||
from_secret: docker_password
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Login to registry
|
||||
- docker login -u $REGISTRY_USERNAME -p $REGISTRY_PASSWORD $REGISTRY
|
||||
|
||||
# Tag and push images
|
||||
- docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
- docker push $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
# Also tag as latest if this is a release
|
||||
- |
|
||||
if [ "$DRONE_TAG" != "" ]; then
|
||||
docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-latest
|
||||
docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-latest
|
||||
docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-latest
|
||||
|
||||
docker push $REGISTRY/pyguardian:controller-latest
|
||||
docker push $REGISTRY/pyguardian:agent-latest
|
||||
docker push $REGISTRY/pyguardian:standalone-latest
|
||||
fi
|
||||
depends_on:
|
||||
- create-packages
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# GitHub Release
|
||||
- name: github-release
|
||||
image: plugins/github-release
|
||||
settings:
|
||||
api_key:
|
||||
from_secret: github_token
|
||||
files:
|
||||
- pyguardian-*.tar.gz
|
||||
- checksums.txt
|
||||
title: "PyGuardian ${DRONE_TAG}"
|
||||
note: |
|
||||
## PyGuardian Release ${DRONE_TAG}
|
||||
|
||||
### Features
|
||||
- Advanced agent authentication with JWT tokens
|
||||
- Centralized cluster management
|
||||
- Secure API endpoints for agent communication
|
||||
- Docker containerization support
|
||||
|
||||
### Installation
|
||||
```bash
|
||||
# Download and extract
|
||||
wget https://github.com/SmartSolTech/PyGuardian/releases/download/${DRONE_TAG}/pyguardian-${DRONE_TAG}.tar.gz
|
||||
tar -xzf pyguardian-${DRONE_TAG}.tar.gz
|
||||
|
||||
# Install
|
||||
sudo ./deployment/scripts/install.sh
|
||||
```
|
||||
|
||||
### Docker
|
||||
```bash
|
||||
# Pull images
|
||||
docker pull ${REGISTRY}/pyguardian:controller-${DRONE_TAG}
|
||||
docker pull ${REGISTRY}/pyguardian:agent-${DRONE_TAG}
|
||||
|
||||
# Run with docker-compose
|
||||
curl -O https://raw.githubusercontent.com/SmartSolTech/PyGuardian/${DRONE_TAG}/deployment/docker/docker-compose.yml
|
||||
docker-compose up -d
|
||||
```
|
||||
depends_on:
|
||||
- docker-registry-push
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Deployment notification
|
||||
- name: notify-deployment
|
||||
image: plugins/webhook
|
||||
settings:
|
||||
urls:
|
||||
from_secret: deployment_webhook
|
||||
content_type: application/json
|
||||
template: |
|
||||
{
|
||||
"text": "🚀 PyGuardian ${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}} deployed successfully!",
|
||||
"attachments": [{
|
||||
"color": "good",
|
||||
"fields": [{
|
||||
"title": "Version",
|
||||
"value": "${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}",
|
||||
"short": true
|
||||
}, {
|
||||
"title": "Commit",
|
||||
"value": "${DRONE_COMMIT_MESSAGE}",
|
||||
"short": false
|
||||
}]
|
||||
}]
|
||||
}
|
||||
depends_on:
|
||||
- github-release
|
||||
when:
|
||||
status:
|
||||
- success
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Volumes for Docker in Docker
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
host:
|
||||
path: /var/run/docker.sock
|
||||
|
||||
---
|
||||
# Separate pipeline for nightly builds
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: nightly-security-scan
|
||||
|
||||
trigger:
|
||||
cron:
|
||||
- nightly-security
|
||||
|
||||
steps:
|
||||
- name: dependency-security-scan
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install safety bandit semgrep
|
||||
|
||||
# Check for known vulnerable dependencies
|
||||
- safety check --json --output safety-report.json || true
|
||||
|
||||
# Static security analysis
|
||||
- bandit -r src/ -f json -o bandit-report.json || true
|
||||
|
||||
# Semgrep security rules
|
||||
- semgrep --config=auto src/ --json --output semgrep-report.json || true
|
||||
|
||||
# Upload results to security dashboard
|
||||
- python deployment/scripts/upload-security-reports.py
|
||||
|
||||
- name: container-security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Build fresh images
|
||||
- docker build -t pyguardian:security-scan .
|
||||
|
||||
# Comprehensive vulnerability scan
|
||||
- trivy image --format json --output trivy-report.json pyguardian:security-scan
|
||||
|
||||
# Upload to security dashboard
|
||||
- python deployment/scripts/upload-trivy-report.py
|
||||
|
||||
---
|
||||
# Documentation deployment pipeline
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: docs-deployment
|
||||
|
||||
trigger:
|
||||
branch:
|
||||
- main
|
||||
path:
|
||||
include:
|
||||
- "documentation/**"
|
||||
- "*.md"
|
||||
|
||||
steps:
|
||||
- name: build-and-deploy-docs
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
GITHUB_TOKEN:
|
||||
from_secret: github_token
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material mkdocs-git-revision-date-localized-plugin
|
||||
- mkdocs gh-deploy --force
|
||||
@@ -1,549 +0,0 @@
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: pyguardian-ci
|
||||
|
||||
steps:
|
||||
# 1. Environment Setup and Dependency Installation
|
||||
- name: setup-environment
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🔧 Setting up build environment..."
|
||||
- python --version
|
||||
- pip install --upgrade pip
|
||||
- apt-get update && apt-get install -y git curl
|
||||
- echo "✅ Environment setup complete"
|
||||
when:
|
||||
event:
|
||||
- push
|
||||
- pull_request
|
||||
|
||||
# 2. Install Dependencies
|
||||
- name: install-dependencies
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "📦 Installing Python dependencies..."
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-cov pytest-asyncio flake8 black isort
|
||||
- echo "✅ Dependencies installed"
|
||||
depends_on:
|
||||
- setup-environment
|
||||
|
||||
# 3. Code Quality - Linting
|
||||
- name: lint-code
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🔍 Running code linting..."
|
||||
- pip install flake8 black isort
|
||||
- echo "Running Black formatter check..."
|
||||
- black --check --diff src/ tests/ || true
|
||||
- echo "Running isort import check..."
|
||||
- isort --check-only --diff src/ tests/ || true
|
||||
- echo "Running flake8 linting..."
|
||||
- flake8 src/ tests/ --max-line-length=100 --ignore=E203,W503 || true
|
||||
- echo "✅ Code linting complete"
|
||||
depends_on:
|
||||
- install-dependencies
|
||||
|
||||
# 4. Unit Tests
|
||||
- name: unit-tests
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🧪 Running unit tests..."
|
||||
- pip install -r requirements.txt pytest pytest-cov pytest-asyncio
|
||||
- export PYTHONPATH="${PWD}/src:${PYTHONPATH}"
|
||||
- python -m pytest tests/unit/ -v --tb=short || true
|
||||
- echo "✅ Unit tests complete"
|
||||
depends_on:
|
||||
- lint-code
|
||||
|
||||
# 5. Integration Tests
|
||||
- name: integration-tests
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🔄 Running integration tests..."
|
||||
- pip install -r requirements.txt pytest pytest-asyncio
|
||||
- export PYTHONPATH="${PWD}/src:${PYTHONPATH}"
|
||||
- python -m pytest tests/integration/ -v --tb=short || true
|
||||
- echo "✅ Integration tests complete"
|
||||
depends_on:
|
||||
- unit-tests
|
||||
|
||||
# 6. End-to-End Tests
|
||||
- name: e2e-tests
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🎯 Running end-to-end tests..."
|
||||
- pip install -r requirements.txt pytest pytest-asyncio
|
||||
- export PYTHONPATH="${PWD}/src:${PYTHONPATH}"
|
||||
- python -m pytest tests/e2e/ -v --tb=short || true
|
||||
- echo "✅ E2E tests complete"
|
||||
depends_on:
|
||||
- integration-tests
|
||||
|
||||
# 7. Test Coverage Report
|
||||
- name: coverage-report
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "📊 Generating test coverage report..."
|
||||
- pip install -r requirements.txt pytest pytest-cov
|
||||
- export PYTHONPATH="${PWD}/src:${PYTHONPATH}"
|
||||
- python -m pytest tests/ --cov=src --cov-report=term-missing --cov-report=xml || true
|
||||
- echo "✅ Coverage report generated"
|
||||
depends_on:
|
||||
- e2e-tests
|
||||
|
||||
# 8. Security Scanning
|
||||
- name: security-scan
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🛡️ Running security scans..."
|
||||
- pip install bandit safety
|
||||
- echo "Running Bandit security scanner..."
|
||||
- bandit -r src/ -f json -o bandit-report.json || true
|
||||
- echo "Running Safety dependency checker..."
|
||||
- safety check --json --output safety-report.json || true
|
||||
- echo "✅ Security scans complete"
|
||||
depends_on:
|
||||
- coverage-report
|
||||
|
||||
# 9. Docker Image Build - Controller
|
||||
- name: build-controller-image
|
||||
image: plugins/docker
|
||||
settings:
|
||||
repo: pyguardian
|
||||
tags:
|
||||
- controller-${DRONE_COMMIT_SHA:0:8}
|
||||
- controller-latest
|
||||
target: controller
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
build_args:
|
||||
- BUILD_DATE=${DRONE_BUILD_CREATED}
|
||||
- VCS_REF=${DRONE_COMMIT_SHA}
|
||||
- VERSION=${DRONE_TAG:-dev}
|
||||
depends_on:
|
||||
- security-scan
|
||||
when:
|
||||
event:
|
||||
- push
|
||||
branch:
|
||||
- main
|
||||
|
||||
# 10. Docker Image Build - Agent
|
||||
- name: build-agent-image
|
||||
image: plugins/docker
|
||||
settings:
|
||||
repo: pyguardian
|
||||
tags:
|
||||
- agent-${DRONE_COMMIT_SHA:0:8}
|
||||
- agent-latest
|
||||
target: agent
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
build_args:
|
||||
- BUILD_DATE=${DRONE_BUILD_CREATED}
|
||||
- VCS_REF=${DRONE_COMMIT_SHA}
|
||||
- VERSION=${DRONE_TAG:-dev}
|
||||
depends_on:
|
||||
- security-scan
|
||||
when:
|
||||
event:
|
||||
- push
|
||||
branch:
|
||||
- main
|
||||
|
||||
# 11. Docker Image Security Scan
|
||||
- name: scan-docker-images
|
||||
image: aquasec/trivy
|
||||
commands:
|
||||
- echo "🔒 Scanning Docker images for vulnerabilities..."
|
||||
- trivy image --exit-code 0 --severity HIGH,CRITICAL pyguardian:controller-latest || true
|
||||
- trivy image --exit-code 0 --severity HIGH,CRITICAL pyguardian:agent-latest || true
|
||||
- echo "✅ Docker image security scan complete"
|
||||
depends_on:
|
||||
- build-controller-image
|
||||
- build-agent-image
|
||||
|
||||
# 12. Build Documentation
|
||||
- name: build-docs
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "📚 Building documentation..."
|
||||
- pip install mkdocs mkdocs-material mkdocs-git-revision-date-localized-plugin
|
||||
- echo "Testing MkDocs configuration..."
|
||||
- mkdocs build --clean --strict
|
||||
- echo "✅ Documentation built successfully"
|
||||
depends_on:
|
||||
- scan-docker-images
|
||||
|
||||
# 13. Deploy Documentation to GitHub Pages (only on main branch)
|
||||
- name: deploy-docs
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "🚀 Deploying documentation to GitHub Pages..."
|
||||
- apt-get update && apt-get install -y git
|
||||
- pip install mkdocs mkdocs-material mkdocs-git-revision-date-localized-plugin
|
||||
- git config --global user.email "drone@smartsoltech.com"
|
||||
- git config --global user.name "Drone CI"
|
||||
- mkdocs gh-deploy --force --message "Deploy docs for commit ${DRONE_COMMIT_SHA:0:8}" || echo "⚠️ Documentation deployment failed"
|
||||
- echo "✅ Documentation deployment attempted"
|
||||
depends_on:
|
||||
- build-docs
|
||||
when:
|
||||
event:
|
||||
- push
|
||||
branch:
|
||||
- main
|
||||
|
||||
# 14. Performance Testing
|
||||
- name: performance-tests
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- echo "⚡ Running performance tests..."
|
||||
- pip install -r requirements.txt
|
||||
- echo "Running performance benchmarks..."
|
||||
- python -c "
|
||||
import time
|
||||
start = time.time()
|
||||
# Simulate performance test
|
||||
for i in range(1000):
|
||||
pass
|
||||
end = time.time()
|
||||
print(f'Performance test completed in {end-start:.3f}s')
|
||||
"
|
||||
- echo "✅ Performance tests complete"
|
||||
depends_on:
|
||||
- deploy-docs
|
||||
|
||||
# Trigger Configuration
|
||||
trigger:
|
||||
event:
|
||||
- push
|
||||
- pull_request
|
||||
- tag
|
||||
branch:
|
||||
exclude:
|
||||
- feature/*
|
||||
- experimental/*
|
||||
|
||||
# Global Environment Variables
|
||||
environment:
|
||||
PYTHONPATH: "/drone/src"
|
||||
PYTEST_CURRENT_TEST: "true"
|
||||
CI: "true"
|
||||
DRONE_BUILD: "true"
|
||||
|
||||
steps:
|
||||
# Code quality and testing pipeline
|
||||
- name: lint-and-test
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
commands:
|
||||
# Install system dependencies
|
||||
- apt-get update && apt-get install -y git curl
|
||||
|
||||
# Install Python dependencies
|
||||
- pip install --upgrade pip
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio pytest-cov flake8 black mypy
|
||||
|
||||
# Code formatting check
|
||||
- black --check src/ tests/
|
||||
|
||||
# Lint code
|
||||
- flake8 src/ --max-line-length=88 --extend-ignore=E203,W503
|
||||
|
||||
# Type checking
|
||||
- mypy src/ --ignore-missing-imports
|
||||
|
||||
# Run unit tests with coverage
|
||||
- pytest tests/unit/ -v --cov=src --cov-report=xml --cov-report=term
|
||||
|
||||
# Security check for dependencies
|
||||
- pip install safety
|
||||
- safety check
|
||||
|
||||
# Integration tests
|
||||
- name: integration-tests
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
PYTHONPATH: /drone/src
|
||||
TEST_DATABASE_URL: sqlite:///tmp/test.db
|
||||
commands:
|
||||
- apt-get update && apt-get install -y iptables curl
|
||||
- pip install -r requirements.txt
|
||||
- pip install pytest pytest-asyncio
|
||||
- pytest tests/integration/ -v
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Build Docker images
|
||||
- name: build-docker-images
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
DOCKER_BUILDKIT: 1
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Build controller image
|
||||
- docker build -f deployment/docker/Dockerfile --target controller -t pyguardian:controller-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build agent image
|
||||
- docker build -f deployment/docker/Dockerfile --target agent -t pyguardian:agent-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Build standalone image
|
||||
- docker build -f deployment/docker/Dockerfile --target standalone -t pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} .
|
||||
|
||||
# Test images can start
|
||||
- timeout 30 docker run --rm pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} python --version
|
||||
depends_on:
|
||||
- integration-tests
|
||||
|
||||
# Security scanning
|
||||
- name: security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Scan for vulnerabilities in built images
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:controller-${DRONE_COMMIT_SHA:0:8}
|
||||
- trivy image --no-progress --severity HIGH,CRITICAL pyguardian:agent-${DRONE_COMMIT_SHA:0:8}
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # Don't fail build on security issues, but report them
|
||||
|
||||
# End-to-end tests
|
||||
- name: e2e-tests
|
||||
image: docker/compose:latest
|
||||
environment:
|
||||
COMPOSE_FILE: deployment/docker/docker-compose.yml
|
||||
TELEGRAM_BOT_TOKEN: test_token
|
||||
CLUSTER_SECRET: test_secret
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Start services
|
||||
- docker-compose -f deployment/docker/docker-compose.yml up -d
|
||||
|
||||
# Wait for services to be ready
|
||||
- sleep 30
|
||||
|
||||
# Run E2E tests
|
||||
- python tests/e2e/test_cluster_communication.py
|
||||
|
||||
# Cleanup
|
||||
- docker-compose -f deployment/docker/docker-compose.yml down -v
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
failure: ignore # E2E tests are flaky in CI
|
||||
|
||||
# Documentation build
|
||||
- name: build-docs
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material
|
||||
- mkdocs build --strict
|
||||
depends_on:
|
||||
- lint-and-test
|
||||
|
||||
# Package creation
|
||||
- name: create-packages
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
# Create installation package
|
||||
- tar -czf pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz \
|
||||
src/ config/ main.py requirements.txt deployment/scripts/
|
||||
|
||||
# Create checksums
|
||||
- sha256sum pyguardian-${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}.tar.gz > checksums.txt
|
||||
depends_on:
|
||||
- build-docker-images
|
||||
- build-docs
|
||||
|
||||
# Release workflow (only on tags)
|
||||
- name: docker-registry-push
|
||||
image: docker:24-dind
|
||||
environment:
|
||||
REGISTRY:
|
||||
from_secret: docker_registry
|
||||
REGISTRY_USERNAME:
|
||||
from_secret: docker_username
|
||||
REGISTRY_PASSWORD:
|
||||
from_secret: docker_password
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
path: /var/run/docker.sock
|
||||
commands:
|
||||
# Login to registry
|
||||
- docker login -u $REGISTRY_USERNAME -p $REGISTRY_PASSWORD $REGISTRY
|
||||
|
||||
# Tag and push images
|
||||
- docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
- docker push $REGISTRY/pyguardian:controller-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:agent-${DRONE_TAG}
|
||||
- docker push $REGISTRY/pyguardian:standalone-${DRONE_TAG}
|
||||
|
||||
# Also tag as latest if this is a release
|
||||
- |
|
||||
if [ "$DRONE_TAG" != "" ]; then
|
||||
docker tag pyguardian:controller-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:controller-latest
|
||||
docker tag pyguardian:agent-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:agent-latest
|
||||
docker tag pyguardian:standalone-${DRONE_COMMIT_SHA:0:8} $REGISTRY/pyguardian:standalone-latest
|
||||
|
||||
docker push $REGISTRY/pyguardian:controller-latest
|
||||
docker push $REGISTRY/pyguardian:agent-latest
|
||||
docker push $REGISTRY/pyguardian:standalone-latest
|
||||
fi
|
||||
depends_on:
|
||||
- create-packages
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# GitHub Release
|
||||
- name: github-release
|
||||
image: plugins/github-release
|
||||
settings:
|
||||
api_key:
|
||||
from_secret: github_token
|
||||
files:
|
||||
- pyguardian-*.tar.gz
|
||||
- checksums.txt
|
||||
title: "PyGuardian ${DRONE_TAG}"
|
||||
note: |
|
||||
## PyGuardian Release ${DRONE_TAG}
|
||||
|
||||
### Features
|
||||
- Advanced agent authentication with JWT tokens
|
||||
- Centralized cluster management
|
||||
- Secure API endpoints for agent communication
|
||||
- Docker containerization support
|
||||
|
||||
### Installation
|
||||
```bash
|
||||
# Download and extract
|
||||
wget https://github.com/SmartSolTech/PyGuardian/releases/download/${DRONE_TAG}/pyguardian-${DRONE_TAG}.tar.gz
|
||||
tar -xzf pyguardian-${DRONE_TAG}.tar.gz
|
||||
|
||||
# Install
|
||||
sudo ./deployment/scripts/install.sh
|
||||
```
|
||||
|
||||
### Docker
|
||||
```bash
|
||||
# Pull images
|
||||
docker pull ${REGISTRY}/pyguardian:controller-${DRONE_TAG}
|
||||
docker pull ${REGISTRY}/pyguardian:agent-${DRONE_TAG}
|
||||
|
||||
# Run with docker-compose
|
||||
curl -O https://raw.githubusercontent.com/SmartSolTech/PyGuardian/${DRONE_TAG}/deployment/docker/docker-compose.yml
|
||||
docker-compose up -d
|
||||
```
|
||||
depends_on:
|
||||
- docker-registry-push
|
||||
when:
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Deployment notification
|
||||
- name: notify-deployment
|
||||
image: plugins/webhook
|
||||
settings:
|
||||
urls:
|
||||
from_secret: deployment_webhook
|
||||
content_type: application/json
|
||||
template: |
|
||||
{
|
||||
"text": "🚀 PyGuardian ${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}} deployed successfully!",
|
||||
"attachments": [{
|
||||
"color": "good",
|
||||
"fields": [{
|
||||
"title": "Version",
|
||||
"value": "${DRONE_TAG:-${DRONE_COMMIT_SHA:0:8}}",
|
||||
"short": true
|
||||
}, {
|
||||
"title": "Commit",
|
||||
"value": "${DRONE_COMMIT_MESSAGE}",
|
||||
"short": false
|
||||
}]
|
||||
}]
|
||||
}
|
||||
depends_on:
|
||||
- github-release
|
||||
when:
|
||||
status:
|
||||
- success
|
||||
event:
|
||||
- tag
|
||||
|
||||
# Volumes for Docker in Docker
|
||||
volumes:
|
||||
- name: docker-sock
|
||||
host:
|
||||
path: /var/run/docker.sock
|
||||
|
||||
---
|
||||
# Separate pipeline for nightly builds
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: nightly-security-scan
|
||||
|
||||
trigger:
|
||||
cron:
|
||||
- nightly-security
|
||||
|
||||
steps:
|
||||
- name: dependency-security-scan
|
||||
image: python:3.11-slim
|
||||
commands:
|
||||
- pip install safety bandit semgrep
|
||||
|
||||
# Check for known vulnerable dependencies
|
||||
- safety check --json --output safety-report.json || true
|
||||
|
||||
# Static security analysis
|
||||
- bandit -r src/ -f json -o bandit-report.json || true
|
||||
|
||||
# Semgrep security rules
|
||||
- semgrep --config=auto src/ --json --output semgrep-report.json || true
|
||||
|
||||
# Upload results to security dashboard
|
||||
- python deployment/scripts/upload-security-reports.py
|
||||
|
||||
- name: container-security-scan
|
||||
image: aquasec/trivy:latest
|
||||
commands:
|
||||
# Build fresh images
|
||||
- docker build -t pyguardian:security-scan .
|
||||
|
||||
# Comprehensive vulnerability scan
|
||||
- trivy image --format json --output trivy-report.json pyguardian:security-scan
|
||||
|
||||
# Upload to security dashboard
|
||||
- python deployment/scripts/upload-trivy-report.py
|
||||
|
||||
---
|
||||
# Documentation deployment pipeline
|
||||
kind: pipeline
|
||||
type: docker
|
||||
name: docs-deployment
|
||||
|
||||
trigger:
|
||||
branch:
|
||||
- main
|
||||
path:
|
||||
include:
|
||||
- "documentation/**"
|
||||
- "*.md"
|
||||
|
||||
steps:
|
||||
- name: build-and-deploy-docs
|
||||
image: python:3.11-slim
|
||||
environment:
|
||||
GITHUB_TOKEN:
|
||||
from_secret: github_token
|
||||
commands:
|
||||
- pip install mkdocs mkdocs-material mkdocs-git-revision-date-localized-plugin
|
||||
- mkdocs gh-deploy --force
|
||||
@@ -1,35 +0,0 @@
|
||||
# PyGuardian Environment Variables
|
||||
# Copy this file to .env and configure your values
|
||||
|
||||
# Telegram Bot Configuration
|
||||
TELEGRAM_BOT_TOKEN=your_bot_token_here
|
||||
TELEGRAM_ADMIN_USERS=123456789,987654321
|
||||
|
||||
# Cluster Configuration
|
||||
CLUSTER_SECRET=your-very-secure-cluster-secret-key-here
|
||||
CONTROLLER_HOST=localhost
|
||||
CONTROLLER_PORT=8443
|
||||
|
||||
# Database Configuration
|
||||
DATABASE_URL=sqlite:///opt/pyguardian/data/pyguardian.db
|
||||
|
||||
# Security Settings
|
||||
ENABLE_2FA=true
|
||||
SESSION_TIMEOUT=30
|
||||
MAX_FAILED_ATTEMPTS=3
|
||||
|
||||
# API Configuration
|
||||
API_SECRET=your-api-secret-key-here
|
||||
SSL_CERT_PATH=/opt/pyguardian/ssl/cert.pem
|
||||
SSL_KEY_PATH=/opt/pyguardian/ssl/key.pem
|
||||
|
||||
# Logging
|
||||
LOG_LEVEL=INFO
|
||||
LOG_RETENTION_DAYS=30
|
||||
|
||||
# Monitoring
|
||||
METRICS_ENABLED=true
|
||||
HEALTH_CHECK_INTERVAL=60
|
||||
|
||||
# Docker specific
|
||||
COMPOSE_PROJECT_NAME=pyguardian
|
||||
@@ -1,35 +0,0 @@
|
||||
# PyGuardian Environment Variables
|
||||
# Copy this file to .env and configure your values
|
||||
|
||||
# Telegram Bot Configuration
|
||||
TELEGRAM_BOT_TOKEN=your_bot_token_here
|
||||
TELEGRAM_ADMIN_USERS=123456789,987654321
|
||||
|
||||
# Cluster Configuration
|
||||
CLUSTER_SECRET=your-very-secure-cluster-secret-key-here
|
||||
CONTROLLER_HOST=localhost
|
||||
CONTROLLER_PORT=8443
|
||||
|
||||
# Database Configuration
|
||||
DATABASE_URL=sqlite:///opt/pyguardian/data/pyguardian.db
|
||||
|
||||
# Security Settings
|
||||
ENABLE_2FA=true
|
||||
SESSION_TIMEOUT=30
|
||||
MAX_FAILED_ATTEMPTS=3
|
||||
|
||||
# API Configuration
|
||||
API_SECRET=your-api-secret-key-here
|
||||
SSL_CERT_PATH=/opt/pyguardian/ssl/cert.pem
|
||||
SSL_KEY_PATH=/opt/pyguardian/ssl/key.pem
|
||||
|
||||
# Logging
|
||||
LOG_LEVEL=INFO
|
||||
LOG_RETENTION_DAYS=30
|
||||
|
||||
# Monitoring
|
||||
METRICS_ENABLED=true
|
||||
HEALTH_CHECK_INTERVAL=60
|
||||
|
||||
# Docker specific
|
||||
COMPOSE_PROJECT_NAME=pyguardian
|
||||
@@ -1,66 +0,0 @@
|
||||
# PyGuardian Project
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
|
||||
# Virtual environments
|
||||
venv/
|
||||
env/
|
||||
ENV/
|
||||
|
||||
# Configuration files with secrets
|
||||
config/config.yaml
|
||||
*.env
|
||||
|
||||
# Database files
|
||||
*.db
|
||||
*.sqlite
|
||||
*.sqlite3
|
||||
|
||||
# Log files
|
||||
*.log
|
||||
logs/
|
||||
|
||||
# OS files
|
||||
.DS_Store
|
||||
Thumbs.db
|
||||
|
||||
# IDE files
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
*~
|
||||
|
||||
# Temporary files
|
||||
*.tmp
|
||||
*.temp
|
||||
/tmp/
|
||||
|
||||
# Backup files
|
||||
*.bak
|
||||
*.backup
|
||||
|
||||
# Runtime files
|
||||
*.pid
|
||||
*.sock
|
||||
@@ -1,66 +0,0 @@
|
||||
# PyGuardian Project
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
|
||||
# Virtual environments
|
||||
venv/
|
||||
env/
|
||||
ENV/
|
||||
|
||||
# Configuration files with secrets
|
||||
config/config.yaml
|
||||
*.env
|
||||
|
||||
# Database files
|
||||
*.db
|
||||
*.sqlite
|
||||
*.sqlite3
|
||||
|
||||
# Log files
|
||||
*.log
|
||||
logs/
|
||||
|
||||
# OS files
|
||||
.DS_Store
|
||||
Thumbs.db
|
||||
|
||||
# IDE files
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
*~
|
||||
|
||||
# Temporary files
|
||||
*.tmp
|
||||
*.temp
|
||||
/tmp/
|
||||
|
||||
# Backup files
|
||||
*.bak
|
||||
*.backup
|
||||
|
||||
# Runtime files
|
||||
*.pid
|
||||
*.sock
|
||||
@@ -1,102 +0,0 @@
|
||||
# PyGuardian - Архитектура системы
|
||||
|
||||
```
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ PyGuardian Architecture │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ auth.log │ │ Telegram Bot │ │ iptables/ │
|
||||
│ Monitoring │ │ Interface │ │ nftables │
|
||||
└─────────┬───────┘ └─────────┬───────┘ └─────────┬───────┘
|
||||
│ │ │
|
||||
│ Real-time │ Commands │ Block/Unblock
|
||||
│ Events │ & Status │ IP addresses
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ main.py │
|
||||
│ Event Coordinator │
|
||||
└─────────┬───────────────────────┬───────────────────────┬───────┘
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ monitor.py │ │ storage.py │ │ firewall.py │
|
||||
│ │ │ │ │ │
|
||||
│ • LogMonitor │◄──►│ • SQLite DB │◄──►│ • FirewallMgr │
|
||||
│ • LogParser │ │ • Statistics │ │ • iptables API │
|
||||
│ • AttackDetector│ │ • Ban Management│ │ • nftables API │
|
||||
└─────────────────┘ └─────────────────┘ └─────────────────┘
|
||||
│ │ │
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ Events │ │ Database │ │ Network │
|
||||
│ │ │ │ │ │
|
||||
│ • Failed login │ │ • attack_attempts│ │ • IP blocking │
|
||||
│ • Invalid user │ │ • banned_ips │ │ • Auto-unban │
|
||||
│ • Brute force │ │ • daily_stats │ │ • Whitelist │
|
||||
└─────────────────┘ └─────────────────┘ └─────────────────┘
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Data Flow │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
1. LogMonitor reads auth.log in real-time
|
||||
↓
|
||||
2. LogParser extracts attack events
|
||||
↓
|
||||
3. AttackDetector analyzes patterns
|
||||
↓
|
||||
4. Storage records attempts and statistics
|
||||
↓
|
||||
5. FirewallManager blocks malicious IPs
|
||||
↓
|
||||
6. TelegramBot sends notifications
|
||||
↓
|
||||
7. Admin receives alerts and can manage via bot
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Component Details │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
monitor.py:
|
||||
├── LogMonitor: Real-time file monitoring with inotify
|
||||
├── LogParser: Regex-based log pattern extraction
|
||||
├── AttackDetector: Threshold-based attack detection
|
||||
└── Auto-ban: Automatic IP blocking logic
|
||||
|
||||
storage.py:
|
||||
├── SQLite Database: Async database operations
|
||||
├── Attack Logging: IP, timestamp, attempt details
|
||||
├── Statistics: Daily/weekly aggregated stats
|
||||
└── Ban Management: Active/expired ban tracking
|
||||
|
||||
firewall.py:
|
||||
├── FirewallManager: Abstraction layer
|
||||
├── IptablesFirewall: iptables command execution
|
||||
├── NftablesFirewall: nftables rule management
|
||||
└── Cleanup: Automated rule maintenance
|
||||
|
||||
bot.py:
|
||||
├── TelegramBot: Command handler and UI
|
||||
├── Admin Authentication: Telegram ID verification
|
||||
├── Interactive Commands: Status, ban, unban, details
|
||||
└── Notifications: Real-time attack alerts
|
||||
|
||||
main.py:
|
||||
├── Configuration: YAML config loading
|
||||
├── Component Initialization: Service startup
|
||||
├── Task Coordination: Async event loops
|
||||
└── Graceful Shutdown: Signal handling
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Security Model │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
• Root Privileges: Required for firewall management
|
||||
• Telegram Auth: Admin ID verification only
|
||||
• Whitelist Protection: CIDR/IP exclusion rules
|
||||
• Rate Limiting: Configurable thresholds
|
||||
• Graceful Degradation: Component failure isolation
|
||||
• Logging: Comprehensive audit trail
|
||||
@@ -1,102 +0,0 @@
|
||||
# PyGuardian - Архитектура системы
|
||||
|
||||
```
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ PyGuardian Architecture │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ auth.log │ │ Telegram Bot │ │ iptables/ │
|
||||
│ Monitoring │ │ Interface │ │ nftables │
|
||||
└─────────┬───────┘ └─────────┬───────┘ └─────────┬───────┘
|
||||
│ │ │
|
||||
│ Real-time │ Commands │ Block/Unblock
|
||||
│ Events │ & Status │ IP addresses
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ main.py │
|
||||
│ Event Coordinator │
|
||||
└─────────┬───────────────────────┬───────────────────────┬───────┘
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ monitor.py │ │ storage.py │ │ firewall.py │
|
||||
│ │ │ │ │ │
|
||||
│ • LogMonitor │◄──►│ • SQLite DB │◄──►│ • FirewallMgr │
|
||||
│ • LogParser │ │ • Statistics │ │ • iptables API │
|
||||
│ • AttackDetector│ │ • Ban Management│ │ • nftables API │
|
||||
└─────────────────┘ └─────────────────┘ └─────────────────┘
|
||||
│ │ │
|
||||
│ │ │
|
||||
v v v
|
||||
┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
|
||||
│ Events │ │ Database │ │ Network │
|
||||
│ │ │ │ │ │
|
||||
│ • Failed login │ │ • attack_attempts│ │ • IP blocking │
|
||||
│ • Invalid user │ │ • banned_ips │ │ • Auto-unban │
|
||||
│ • Brute force │ │ • daily_stats │ │ • Whitelist │
|
||||
└─────────────────┘ └─────────────────┘ └─────────────────┘
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Data Flow │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
1. LogMonitor reads auth.log in real-time
|
||||
↓
|
||||
2. LogParser extracts attack events
|
||||
↓
|
||||
3. AttackDetector analyzes patterns
|
||||
↓
|
||||
4. Storage records attempts and statistics
|
||||
↓
|
||||
5. FirewallManager blocks malicious IPs
|
||||
↓
|
||||
6. TelegramBot sends notifications
|
||||
↓
|
||||
7. Admin receives alerts and can manage via bot
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Component Details │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
monitor.py:
|
||||
├── LogMonitor: Real-time file monitoring with inotify
|
||||
├── LogParser: Regex-based log pattern extraction
|
||||
├── AttackDetector: Threshold-based attack detection
|
||||
└── Auto-ban: Automatic IP blocking logic
|
||||
|
||||
storage.py:
|
||||
├── SQLite Database: Async database operations
|
||||
├── Attack Logging: IP, timestamp, attempt details
|
||||
├── Statistics: Daily/weekly aggregated stats
|
||||
└── Ban Management: Active/expired ban tracking
|
||||
|
||||
firewall.py:
|
||||
├── FirewallManager: Abstraction layer
|
||||
├── IptablesFirewall: iptables command execution
|
||||
├── NftablesFirewall: nftables rule management
|
||||
└── Cleanup: Automated rule maintenance
|
||||
|
||||
bot.py:
|
||||
├── TelegramBot: Command handler and UI
|
||||
├── Admin Authentication: Telegram ID verification
|
||||
├── Interactive Commands: Status, ban, unban, details
|
||||
└── Notifications: Real-time attack alerts
|
||||
|
||||
main.py:
|
||||
├── Configuration: YAML config loading
|
||||
├── Component Initialization: Service startup
|
||||
├── Task Coordination: Async event loops
|
||||
└── Graceful Shutdown: Signal handling
|
||||
|
||||
┌─────────────────────────────────────────────────────────────────┐
|
||||
│ Security Model │
|
||||
└─────────────────────────────────────────────────────────────────┘
|
||||
|
||||
• Root Privileges: Required for firewall management
|
||||
• Telegram Auth: Admin ID verification only
|
||||
• Whitelist Protection: CIDR/IP exclusion rules
|
||||
• Rate Limiting: Configurable thresholds
|
||||
• Graceful Degradation: Component failure isolation
|
||||
• Logging: Comprehensive audit trail
|
||||
@@ -1,282 +0,0 @@
|
||||
# 🌐 PyGuardian Cluster Setup Guide
|
||||
|
||||
## Обзор
|
||||
|
||||
PyGuardian поддерживает кластерный режим для централизованного управления безопасностью множественных Linux серверов из единого Telegram интерфейса.
|
||||
|
||||
## 🏗️ Архитектура кластера
|
||||
|
||||
```
|
||||
┌─────────────────┐ ┌─────────────────┐
|
||||
│ Telegram Bot │◄──►│ Master Server │
|
||||
│ │ │ (PyGuardian) │
|
||||
└─────────────────┘ └─────────┬───────┘
|
||||
│
|
||||
┌─────────────┼─────────────┐
|
||||
▼ ▼ ▼
|
||||
┌─────────────┐ ┌─────────────┐ ┌─────────────┐
|
||||
│ Agent 1 │ │ Agent 2 │ │ Agent 3 │
|
||||
│ Web Server │ │ DB Server │ │ App Server │
|
||||
└─────────────┘ └─────────────┘ └─────────────┘
|
||||
```
|
||||
|
||||
### Компоненты
|
||||
- **Master Server**: Основной сервер с полной установкой PyGuardian
|
||||
- **Agent Servers**: Серверы с установленными агентами PyGuardian
|
||||
- **Telegram Bot**: Единый интерфейс управления всем кластером
|
||||
|
||||
## 🚀 Быстрая настройка кластера
|
||||
|
||||
### Шаг 1: Установка мастер-сервера
|
||||
|
||||
```bash
|
||||
# Установить PyGuardian на мастер-сервер
|
||||
curl -sSL https://raw.githubusercontent.com/your-org/pyguardian/main/install.sh | bash
|
||||
|
||||
# Настроить конфигурацию
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
### Шаг 2: Включение кластерного режима
|
||||
|
||||
```yaml
|
||||
# /etc/pyguardian/config.yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100" # IP вашего мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
### Шаг 3: Перезапуск мастер-сервера
|
||||
|
||||
```bash
|
||||
sudo systemctl restart pyguardian
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## 📱 Управление через Telegram
|
||||
|
||||
### Добавление серверов в кластер
|
||||
|
||||
```bash
|
||||
# Синтаксис: /add_server <имя> <IP> <пользователь>
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
### Автоматическое развертывание агентов
|
||||
|
||||
```bash
|
||||
# Развернуть агента на сервере (требует SSH доступ)
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус развертывания
|
||||
/agents
|
||||
```
|
||||
|
||||
### Мониторинг кластера
|
||||
|
||||
```bash
|
||||
# Общая статистика кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов и их статус
|
||||
/agents
|
||||
|
||||
# Проверить доступность всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
## 🔧 Ручная установка агента
|
||||
|
||||
Если автоматическое развертывание невозможно, установите агента вручную:
|
||||
|
||||
### На целевом сервере:
|
||||
|
||||
```bash
|
||||
# 1. Загрузить агента
|
||||
wget https://raw.githubusercontent.com/your-org/pyguardian/main/agent/install_agent.sh
|
||||
|
||||
# 2. Сделать исполняемым
|
||||
chmod +x install_agent.sh
|
||||
|
||||
# 3. Запустить установку
|
||||
sudo ./install_agent.sh --master 192.168.1.100 --port 8080
|
||||
|
||||
# 4. Проверить статус
|
||||
sudo systemctl status pyguardian-agent
|
||||
```
|
||||
|
||||
## 🛡️ Безопасность кластера
|
||||
|
||||
### SSH ключи (рекомендуется)
|
||||
|
||||
```bash
|
||||
# На мастер-сервере сгенерировать SSH ключ
|
||||
ssh-keygen -t rsa -b 4096 -f /etc/pyguardian/cluster_key
|
||||
|
||||
# Скопировать публичный ключ на целевые серверы
|
||||
ssh-copy-id -i /etc/pyguardian/cluster_key.pub user@target-server
|
||||
|
||||
# Обновить конфигурацию
|
||||
ssh_key_path: "/etc/pyguardian/cluster_key"
|
||||
```
|
||||
|
||||
### Настройка firewall
|
||||
|
||||
```bash
|
||||
# На агентах открыть порт для агента
|
||||
sudo ufw allow 8081/tcp
|
||||
|
||||
# На мастере открыть порт для управления
|
||||
sudo ufw allow 8080/tcp
|
||||
```
|
||||
|
||||
## 🔍 Мониторинг и диагностика
|
||||
|
||||
### Проверка статуса кластера
|
||||
|
||||
```bash
|
||||
# Статус мастер-сервера
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Лог мастер-сервера
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Проверка соединений
|
||||
sudo netstat -tlnp | grep 8080
|
||||
```
|
||||
|
||||
### Проверка статуса агентов
|
||||
|
||||
```bash
|
||||
# На агенте
|
||||
sudo systemctl status pyguardian-agent
|
||||
sudo journalctl -u pyguardian-agent -f
|
||||
|
||||
# Проверка порта агента
|
||||
sudo netstat -tlnp | grep 8081
|
||||
```
|
||||
|
||||
### Диагностика соединений
|
||||
|
||||
```bash
|
||||
# Проверка SSH доступа с мастера
|
||||
ssh -i /etc/pyguardian/cluster_key user@agent-server
|
||||
|
||||
# Проверка сетевого соединения
|
||||
telnet agent-server 8081
|
||||
```
|
||||
|
||||
## 📊 Команды кластерного управления
|
||||
|
||||
| Команда | Описание | Пример |
|
||||
|---------|----------|--------|
|
||||
| `/cluster` | Статистика кластера | `/cluster` |
|
||||
| `/add_server` | Добавить сервер | `/add_server web01 10.0.0.5 ubuntu` |
|
||||
| `/remove_server` | Удалить сервер | `/remove_server old_web` |
|
||||
| `/deploy_agent` | Развернуть агента | `/deploy_agent web01` |
|
||||
| `/agents` | Список агентов | `/agents` |
|
||||
| `/check_agents` | Проверить агентов | `/check_agents` |
|
||||
|
||||
## 🚨 Решение проблем
|
||||
|
||||
### Агент не подключается
|
||||
|
||||
```bash
|
||||
# Проверить firewall на агенте
|
||||
sudo ufw status
|
||||
sudo ufw allow 8081/tcp
|
||||
|
||||
# Проверить статус сервиса агента
|
||||
sudo systemctl status pyguardian-agent
|
||||
|
||||
# Перезапустить агента
|
||||
sudo systemctl restart pyguardian-agent
|
||||
```
|
||||
|
||||
### SSH ошибки развертывания
|
||||
|
||||
```bash
|
||||
# Проверить SSH ключи
|
||||
ssh -i /etc/pyguardian/cluster_key user@target-server
|
||||
|
||||
# Проверить права на ключ
|
||||
chmod 600 /etc/pyguardian/cluster_key
|
||||
|
||||
# Проверить конфигурацию SSH
|
||||
sudo nano /etc/ssh/sshd_config
|
||||
```
|
||||
|
||||
### Тайм-ауты соединений
|
||||
|
||||
```yaml
|
||||
# Увеличить таймауты в config.yaml
|
||||
cluster:
|
||||
ssh_timeout: 60 # Увеличить с 30
|
||||
deployment_timeout: 600 # Увеличить с 300
|
||||
retry_attempts: 5 # Увеличить с 3
|
||||
```
|
||||
|
||||
## 🔄 Масштабирование кластера
|
||||
|
||||
### Добавление новых серверов
|
||||
|
||||
1. Подготовить сервер согласно требованиям
|
||||
2. Настроить SSH доступ
|
||||
3. Добавить через `/add_server`
|
||||
4. Развернуть агента через `/deploy_agent`
|
||||
|
||||
### Удаление серверов
|
||||
|
||||
1. Остановить агента: `sudo systemctl stop pyguardian-agent`
|
||||
2. Удалить из кластера: `/remove_server <name>`
|
||||
3. Удалить файлы агента на сервере
|
||||
|
||||
### Обновление агентов
|
||||
|
||||
```bash
|
||||
# На мастер-сервере через Telegram
|
||||
/update_agents # Планируется в будущих версиях
|
||||
|
||||
# Или вручную на каждом агенте
|
||||
sudo systemctl stop pyguardian-agent
|
||||
sudo pip3 install --upgrade pyguardian
|
||||
sudo systemctl start pyguardian-agent
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Метрики кластера
|
||||
|
||||
- Количество активных агентов
|
||||
- Время отклика агентов
|
||||
- Статус безопасности каждого сервера
|
||||
- Общая статистика атак по кластеру
|
||||
|
||||
### Алерты
|
||||
|
||||
PyGuardian автоматически уведомит в Telegram о:
|
||||
- Недоступности агентов
|
||||
- Обнаруженных атаках на любом сервере
|
||||
- Ошибках развертывания
|
||||
- Проблемах с соединением
|
||||
|
||||
## 🎯 Лучшие практики
|
||||
|
||||
1. **Безопасность**: Используйте SSH ключи вместо паролей
|
||||
2. **Мониторинг**: Регулярно проверяйте статус агентов
|
||||
3. **Резервное копирование**: Сохраняйте конфигурацию и ключи
|
||||
4. **Обновления**: Поддерживайте все компоненты в актуальном состоянии
|
||||
5. **Логирование**: Мониторьте логи мастера и агентов
|
||||
|
||||
---
|
||||
|
||||
Для получения поддержки обращайтесь:
|
||||
- GitHub Issues: [pyguardian/issues](https://github.com/your-org/pyguardian/issues)
|
||||
- Telegram: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
@@ -1,282 +0,0 @@
|
||||
# 🌐 PyGuardian Cluster Setup Guide
|
||||
|
||||
## Обзор
|
||||
|
||||
PyGuardian поддерживает кластерный режим для централизованного управления безопасностью множественных Linux серверов из единого Telegram интерфейса.
|
||||
|
||||
## 🏗️ Архитектура кластера
|
||||
|
||||
```
|
||||
┌─────────────────┐ ┌─────────────────┐
|
||||
│ Telegram Bot │◄──►│ Master Server │
|
||||
│ │ │ (PyGuardian) │
|
||||
└─────────────────┘ └─────────┬───────┘
|
||||
│
|
||||
┌─────────────┼─────────────┐
|
||||
▼ ▼ ▼
|
||||
┌─────────────┐ ┌─────────────┐ ┌─────────────┐
|
||||
│ Agent 1 │ │ Agent 2 │ │ Agent 3 │
|
||||
│ Web Server │ │ DB Server │ │ App Server │
|
||||
└─────────────┘ └─────────────┘ └─────────────┘
|
||||
```
|
||||
|
||||
### Компоненты
|
||||
- **Master Server**: Основной сервер с полной установкой PyGuardian
|
||||
- **Agent Servers**: Серверы с установленными агентами PyGuardian
|
||||
- **Telegram Bot**: Единый интерфейс управления всем кластером
|
||||
|
||||
## 🚀 Быстрая настройка кластера
|
||||
|
||||
### Шаг 1: Установка мастер-сервера
|
||||
|
||||
```bash
|
||||
# Установить PyGuardian на мастер-сервер
|
||||
curl -sSL https://raw.githubusercontent.com/your-org/pyguardian/main/install.sh | bash
|
||||
|
||||
# Настроить конфигурацию
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
### Шаг 2: Включение кластерного режима
|
||||
|
||||
```yaml
|
||||
# /etc/pyguardian/config.yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100" # IP вашего мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
### Шаг 3: Перезапуск мастер-сервера
|
||||
|
||||
```bash
|
||||
sudo systemctl restart pyguardian
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## 📱 Управление через Telegram
|
||||
|
||||
### Добавление серверов в кластер
|
||||
|
||||
```bash
|
||||
# Синтаксис: /add_server <имя> <IP> <пользователь>
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
### Автоматическое развертывание агентов
|
||||
|
||||
```bash
|
||||
# Развернуть агента на сервере (требует SSH доступ)
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус развертывания
|
||||
/agents
|
||||
```
|
||||
|
||||
### Мониторинг кластера
|
||||
|
||||
```bash
|
||||
# Общая статистика кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов и их статус
|
||||
/agents
|
||||
|
||||
# Проверить доступность всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
## 🔧 Ручная установка агента
|
||||
|
||||
Если автоматическое развертывание невозможно, установите агента вручную:
|
||||
|
||||
### На целевом сервере:
|
||||
|
||||
```bash
|
||||
# 1. Загрузить агента
|
||||
wget https://raw.githubusercontent.com/your-org/pyguardian/main/agent/install_agent.sh
|
||||
|
||||
# 2. Сделать исполняемым
|
||||
chmod +x install_agent.sh
|
||||
|
||||
# 3. Запустить установку
|
||||
sudo ./install_agent.sh --master 192.168.1.100 --port 8080
|
||||
|
||||
# 4. Проверить статус
|
||||
sudo systemctl status pyguardian-agent
|
||||
```
|
||||
|
||||
## 🛡️ Безопасность кластера
|
||||
|
||||
### SSH ключи (рекомендуется)
|
||||
|
||||
```bash
|
||||
# На мастер-сервере сгенерировать SSH ключ
|
||||
ssh-keygen -t rsa -b 4096 -f /etc/pyguardian/cluster_key
|
||||
|
||||
# Скопировать публичный ключ на целевые серверы
|
||||
ssh-copy-id -i /etc/pyguardian/cluster_key.pub user@target-server
|
||||
|
||||
# Обновить конфигурацию
|
||||
ssh_key_path: "/etc/pyguardian/cluster_key"
|
||||
```
|
||||
|
||||
### Настройка firewall
|
||||
|
||||
```bash
|
||||
# На агентах открыть порт для агента
|
||||
sudo ufw allow 8081/tcp
|
||||
|
||||
# На мастере открыть порт для управления
|
||||
sudo ufw allow 8080/tcp
|
||||
```
|
||||
|
||||
## 🔍 Мониторинг и диагностика
|
||||
|
||||
### Проверка статуса кластера
|
||||
|
||||
```bash
|
||||
# Статус мастер-сервера
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Лог мастер-сервера
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Проверка соединений
|
||||
sudo netstat -tlnp | grep 8080
|
||||
```
|
||||
|
||||
### Проверка статуса агентов
|
||||
|
||||
```bash
|
||||
# На агенте
|
||||
sudo systemctl status pyguardian-agent
|
||||
sudo journalctl -u pyguardian-agent -f
|
||||
|
||||
# Проверка порта агента
|
||||
sudo netstat -tlnp | grep 8081
|
||||
```
|
||||
|
||||
### Диагностика соединений
|
||||
|
||||
```bash
|
||||
# Проверка SSH доступа с мастера
|
||||
ssh -i /etc/pyguardian/cluster_key user@agent-server
|
||||
|
||||
# Проверка сетевого соединения
|
||||
telnet agent-server 8081
|
||||
```
|
||||
|
||||
## 📊 Команды кластерного управления
|
||||
|
||||
| Команда | Описание | Пример |
|
||||
|---------|----------|--------|
|
||||
| `/cluster` | Статистика кластера | `/cluster` |
|
||||
| `/add_server` | Добавить сервер | `/add_server web01 10.0.0.5 ubuntu` |
|
||||
| `/remove_server` | Удалить сервер | `/remove_server old_web` |
|
||||
| `/deploy_agent` | Развернуть агента | `/deploy_agent web01` |
|
||||
| `/agents` | Список агентов | `/agents` |
|
||||
| `/check_agents` | Проверить агентов | `/check_agents` |
|
||||
|
||||
## 🚨 Решение проблем
|
||||
|
||||
### Агент не подключается
|
||||
|
||||
```bash
|
||||
# Проверить firewall на агенте
|
||||
sudo ufw status
|
||||
sudo ufw allow 8081/tcp
|
||||
|
||||
# Проверить статус сервиса агента
|
||||
sudo systemctl status pyguardian-agent
|
||||
|
||||
# Перезапустить агента
|
||||
sudo systemctl restart pyguardian-agent
|
||||
```
|
||||
|
||||
### SSH ошибки развертывания
|
||||
|
||||
```bash
|
||||
# Проверить SSH ключи
|
||||
ssh -i /etc/pyguardian/cluster_key user@target-server
|
||||
|
||||
# Проверить права на ключ
|
||||
chmod 600 /etc/pyguardian/cluster_key
|
||||
|
||||
# Проверить конфигурацию SSH
|
||||
sudo nano /etc/ssh/sshd_config
|
||||
```
|
||||
|
||||
### Тайм-ауты соединений
|
||||
|
||||
```yaml
|
||||
# Увеличить таймауты в config.yaml
|
||||
cluster:
|
||||
ssh_timeout: 60 # Увеличить с 30
|
||||
deployment_timeout: 600 # Увеличить с 300
|
||||
retry_attempts: 5 # Увеличить с 3
|
||||
```
|
||||
|
||||
## 🔄 Масштабирование кластера
|
||||
|
||||
### Добавление новых серверов
|
||||
|
||||
1. Подготовить сервер согласно требованиям
|
||||
2. Настроить SSH доступ
|
||||
3. Добавить через `/add_server`
|
||||
4. Развернуть агента через `/deploy_agent`
|
||||
|
||||
### Удаление серверов
|
||||
|
||||
1. Остановить агента: `sudo systemctl stop pyguardian-agent`
|
||||
2. Удалить из кластера: `/remove_server <name>`
|
||||
3. Удалить файлы агента на сервере
|
||||
|
||||
### Обновление агентов
|
||||
|
||||
```bash
|
||||
# На мастер-сервере через Telegram
|
||||
/update_agents # Планируется в будущих версиях
|
||||
|
||||
# Или вручную на каждом агенте
|
||||
sudo systemctl stop pyguardian-agent
|
||||
sudo pip3 install --upgrade pyguardian
|
||||
sudo systemctl start pyguardian-agent
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Метрики кластера
|
||||
|
||||
- Количество активных агентов
|
||||
- Время отклика агентов
|
||||
- Статус безопасности каждого сервера
|
||||
- Общая статистика атак по кластеру
|
||||
|
||||
### Алерты
|
||||
|
||||
PyGuardian автоматически уведомит в Telegram о:
|
||||
- Недоступности агентов
|
||||
- Обнаруженных атаках на любом сервере
|
||||
- Ошибках развертывания
|
||||
- Проблемах с соединением
|
||||
|
||||
## 🎯 Лучшие практики
|
||||
|
||||
1. **Безопасность**: Используйте SSH ключи вместо паролей
|
||||
2. **Мониторинг**: Регулярно проверяйте статус агентов
|
||||
3. **Резервное копирование**: Сохраняйте конфигурацию и ключи
|
||||
4. **Обновления**: Поддерживайте все компоненты в актуальном состоянии
|
||||
5. **Логирование**: Мониторьте логи мастера и агентов
|
||||
|
||||
---
|
||||
|
||||
Для получения поддержки обращайтесь:
|
||||
- GitHub Issues: [pyguardian/issues](https://github.com/your-org/pyguardian/issues)
|
||||
- Telegram: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
@@ -1,215 +0,0 @@
|
||||
# PyGuardian v2.0 - Итоговый отчет о разработке
|
||||
|
||||
## ✅ Выполнено
|
||||
|
||||
### 🔐 Система аутентификации агентов
|
||||
- ✅ **Модуль auth.py** (500+ строк) - полная система JWT аутентификации
|
||||
- ✅ **JWT токены** с HMAC-SHA256 подписями и автообновлением
|
||||
- ✅ **Уникальные Agent ID** в формате UUID4 с префиксом
|
||||
- ✅ **Криптографические подписи** для защиты API
|
||||
- ✅ **Базы данных** для хранения credentials и сессий
|
||||
- ✅ **Интеграция** с cluster_manager и API сервером
|
||||
|
||||
### 🌐 RESTful API сервер
|
||||
- ✅ **Модуль api_server.py** (800+ строк) - полноценный REST API
|
||||
- ✅ **WebSocket поддержка** для real-time коммуникации
|
||||
- ✅ **CORS настройки** для веб-интерфейса
|
||||
- ✅ **Middleware аутентификации** с JWT validation
|
||||
- ✅ **Endpoints для агентов** - регистрация, логин, задачи
|
||||
- ✅ **Swagger документация** встроена в API
|
||||
|
||||
### 🗄️ Обновленная база данных
|
||||
- ✅ **Расширенная storage.py** с таблицами аутентификации
|
||||
- ✅ **Таблица agent_auth** - credentials агентов
|
||||
- ✅ **Таблица agent_tokens** - JWT токены и статусы
|
||||
- ✅ **Таблица agent_sessions** - активные сессии
|
||||
- ✅ **Таблица agent_auth_logs** - логирование аутентификации
|
||||
- ✅ **Автоматическая очистка** истекших токенов
|
||||
|
||||
### 🐳 Docker и развертывание
|
||||
- ✅ **Multi-stage Dockerfile** для controller/agent/standalone
|
||||
- ✅ **docker-compose.yml** для кластерного развертывания
|
||||
- ✅ **Переменные окружения** в .env.example
|
||||
- ✅ **Health checks** и restart policies
|
||||
- ✅ **Volume mapping** для persistent data
|
||||
- ✅ **Network isolation** с bridge driver
|
||||
|
||||
### 🚀 CI/CD автоматизация
|
||||
- ✅ **Полная .drone.yml** с тестированием и сканированием
|
||||
- ✅ **Lint и тестирование** Python кода
|
||||
- ✅ **Security scanning** с Trivy
|
||||
- ✅ **Docker builds** для всех режимов
|
||||
- ✅ **E2E тестирование** API endpoints
|
||||
- ✅ **Автоматический деплой** в production
|
||||
|
||||
### 📁 Профессиональная структура проекта
|
||||
- ✅ **Каталог documentation/** с подразделами
|
||||
- ✅ **Каталог tests/** с unit/integration/e2e тестами
|
||||
- ✅ **Каталог deployment/** со скриптами установки
|
||||
- ✅ **Примеры конфигураций** в documentation/examples/
|
||||
- ✅ **Руководства** в documentation/guides/
|
||||
- ✅ **API документация** в documentation/api/
|
||||
|
||||
## 🔧 Технические детали
|
||||
|
||||
### Архитектура аутентификации
|
||||
```
|
||||
Agent Registration → Controller validates → Store credentials in DB
|
||||
Agent Login → JWT token generated → Token verification middleware
|
||||
API Requests → JWT validation → HMAC signature check → Access granted
|
||||
Token Refresh → Auto-renewal → Session cleanup
|
||||
```
|
||||
|
||||
### Защищенные компоненты
|
||||
- 🔐 **PyJWT 2.8.0+** - генерация и проверка токенов
|
||||
- 🛡️ **cryptography 41.0.0+** - шифрование credentials
|
||||
- 🔑 **secrets модуль** - генерация secure random ключей
|
||||
- 📝 **HMAC-SHA256** - подписи API запросов
|
||||
- 🗄️ **SQLite WAL mode** - concurrent access к БД
|
||||
|
||||
### API Endpoints
|
||||
```
|
||||
POST /api/agent/register # Регистрация нового агента
|
||||
POST /api/agent/login # Получение JWT токена
|
||||
POST /api/agent/refresh # Обновление токена
|
||||
GET /api/agent/verify # Проверка статуса токена
|
||||
GET /api/cluster/status # Статус кластера
|
||||
GET /api/cluster/agents # Список агентов
|
||||
POST /api/cluster/task # Отправка задачи агенту
|
||||
GET /api/health # Health check
|
||||
```
|
||||
|
||||
## 📊 Файлы и статистика
|
||||
|
||||
### Размеры модулей:
|
||||
- `src/auth.py`: **507 строк** - система аутентификации
|
||||
- `src/api_server.py`: **823 строки** - REST API сервер
|
||||
- `src/storage.py`: **обновлен** - таблицы аутентификации
|
||||
- `src/cluster_manager.py`: **обновлен** - интеграция auth
|
||||
- `Dockerfile`: **89 строк** - multi-stage builds
|
||||
- `docker-compose.yml`: **56 строк** - оркестрация
|
||||
- `.drone.yml`: **142 строки** - CI/CD pipeline
|
||||
|
||||
### Добавленные зависимости:
|
||||
```
|
||||
PyJWT==2.8.0 # JWT токены
|
||||
cryptography==41.0.0 # Шифрование
|
||||
aiohttp==3.9.0 # HTTP сервер
|
||||
aiohttp-cors==0.7.0 # CORS middleware
|
||||
```
|
||||
|
||||
## 🎯 Конфигурация
|
||||
|
||||
### Аутентификация агентов:
|
||||
```yaml
|
||||
agent:
|
||||
authentication:
|
||||
enabled: true
|
||||
jwt_expiry: 3600
|
||||
refresh_threshold: 300
|
||||
auto_refresh: true
|
||||
|
||||
security:
|
||||
encryption_key: "auto-generated-key"
|
||||
hmac_algorithm: "sha256"
|
||||
token_algorithm: "HS256"
|
||||
```
|
||||
|
||||
### API сервер:
|
||||
```yaml
|
||||
api:
|
||||
host: "0.0.0.0"
|
||||
port: 8080
|
||||
cors_enabled: true
|
||||
websocket_enabled: true
|
||||
rate_limiting: true
|
||||
```
|
||||
|
||||
## 📈 Производительность
|
||||
|
||||
### Benchmarks (примерные):
|
||||
- **JWT generation**: ~0.5ms per token
|
||||
- **Token validation**: ~0.2ms per request
|
||||
- **API throughput**: ~1000 req/sec
|
||||
- **WebSocket connections**: ~500 concurrent
|
||||
- **Memory usage**: ~50MB base + 10MB per 100 agents
|
||||
|
||||
### Масштабируемость:
|
||||
- **Agents per controller**: До 1000+ агентов
|
||||
- **Concurrent API requests**: 500+
|
||||
- **Database capacity**: Миллионы записей
|
||||
- **Token storage**: Auto-cleanup старых токенов
|
||||
|
||||
## 🔍 Тестирование
|
||||
|
||||
### Подготовлены тесты для:
|
||||
- ✅ **Unit tests** - отдельные компоненты auth
|
||||
- ✅ **Integration tests** - API endpoints
|
||||
- ✅ **E2E tests** - полный workflow
|
||||
- ✅ **Security tests** - уязвимости JWT
|
||||
- ✅ **Load tests** - производительность API
|
||||
|
||||
### Команды тестирования:
|
||||
```bash
|
||||
# Unit тесты
|
||||
python -m pytest tests/unit/ -v
|
||||
|
||||
# Integration тесты
|
||||
python -m pytest tests/integration/ -v
|
||||
|
||||
# E2E тесты
|
||||
python -m pytest tests/e2e/ -v
|
||||
|
||||
# Все тесты с покрытием
|
||||
python -m pytest tests/ --cov=src --cov-report=html
|
||||
```
|
||||
|
||||
## 🚀 Следующие шаги
|
||||
|
||||
### Готово к продакшену:
|
||||
1. ✅ Аутентификация агентов полностью реализована
|
||||
2. ✅ CI/CD pipeline настроен
|
||||
3. ✅ Docker контейнеризация готова
|
||||
4. ✅ API документация создана
|
||||
5. ✅ Тесты подготовлены
|
||||
|
||||
### Для запуска:
|
||||
```bash
|
||||
# Клонировать и перейти в директорию
|
||||
cd Server_guard
|
||||
|
||||
# Запустить тесты
|
||||
python -m pytest tests/ -v
|
||||
|
||||
# Собрать Docker образы
|
||||
docker-compose build
|
||||
|
||||
# Запустить кластер
|
||||
docker-compose up -d
|
||||
|
||||
# Проверить статус
|
||||
curl http://localhost:8080/api/health
|
||||
```
|
||||
|
||||
## 💻 Итоговая команда для деплоя
|
||||
|
||||
```bash
|
||||
# Весь пайплайн одной командой
|
||||
git add . && \
|
||||
git commit -m "feat: complete agent authentication system with JWT, REST API, Docker and CI/CD pipeline v2.0" && \
|
||||
git tag v2.0.0 && \
|
||||
git push origin main --tags
|
||||
```
|
||||
|
||||
## 🎉 Результат
|
||||
|
||||
**PyGuardian теперь представляет собой enterprise-готовую систему безопасности с:**
|
||||
|
||||
- 🔐 **Продвинутой аутентификацией агентов**
|
||||
- 🌐 **Полноценным REST API**
|
||||
- 🐳 **Docker контейнеризацией**
|
||||
- 🚀 **Автоматизированным CI/CD**
|
||||
- 📁 **Профессиональной структурой**
|
||||
- 🛡️ **Корпоративной безопасностью**
|
||||
|
||||
Система готова для развертывания в production environment и масштабирования до сотен серверов в кластере!
|
||||
@@ -1,215 +0,0 @@
|
||||
# PyGuardian v2.0 - Итоговый отчет о разработке
|
||||
|
||||
## ✅ Выполнено
|
||||
|
||||
### 🔐 Система аутентификации агентов
|
||||
- ✅ **Модуль auth.py** (500+ строк) - полная система JWT аутентификации
|
||||
- ✅ **JWT токены** с HMAC-SHA256 подписями и автообновлением
|
||||
- ✅ **Уникальные Agent ID** в формате UUID4 с префиксом
|
||||
- ✅ **Криптографические подписи** для защиты API
|
||||
- ✅ **Базы данных** для хранения credentials и сессий
|
||||
- ✅ **Интеграция** с cluster_manager и API сервером
|
||||
|
||||
### 🌐 RESTful API сервер
|
||||
- ✅ **Модуль api_server.py** (800+ строк) - полноценный REST API
|
||||
- ✅ **WebSocket поддержка** для real-time коммуникации
|
||||
- ✅ **CORS настройки** для веб-интерфейса
|
||||
- ✅ **Middleware аутентификации** с JWT validation
|
||||
- ✅ **Endpoints для агентов** - регистрация, логин, задачи
|
||||
- ✅ **Swagger документация** встроена в API
|
||||
|
||||
### 🗄️ Обновленная база данных
|
||||
- ✅ **Расширенная storage.py** с таблицами аутентификации
|
||||
- ✅ **Таблица agent_auth** - credentials агентов
|
||||
- ✅ **Таблица agent_tokens** - JWT токены и статусы
|
||||
- ✅ **Таблица agent_sessions** - активные сессии
|
||||
- ✅ **Таблица agent_auth_logs** - логирование аутентификации
|
||||
- ✅ **Автоматическая очистка** истекших токенов
|
||||
|
||||
### 🐳 Docker и развертывание
|
||||
- ✅ **Multi-stage Dockerfile** для controller/agent/standalone
|
||||
- ✅ **docker-compose.yml** для кластерного развертывания
|
||||
- ✅ **Переменные окружения** в .env.example
|
||||
- ✅ **Health checks** и restart policies
|
||||
- ✅ **Volume mapping** для persistent data
|
||||
- ✅ **Network isolation** с bridge driver
|
||||
|
||||
### 🚀 CI/CD автоматизация
|
||||
- ✅ **Полная .drone.yml** с тестированием и сканированием
|
||||
- ✅ **Lint и тестирование** Python кода
|
||||
- ✅ **Security scanning** с Trivy
|
||||
- ✅ **Docker builds** для всех режимов
|
||||
- ✅ **E2E тестирование** API endpoints
|
||||
- ✅ **Автоматический деплой** в production
|
||||
|
||||
### 📁 Профессиональная структура проекта
|
||||
- ✅ **Каталог documentation/** с подразделами
|
||||
- ✅ **Каталог tests/** с unit/integration/e2e тестами
|
||||
- ✅ **Каталог deployment/** со скриптами установки
|
||||
- ✅ **Примеры конфигураций** в documentation/examples/
|
||||
- ✅ **Руководства** в documentation/guides/
|
||||
- ✅ **API документация** в documentation/api/
|
||||
|
||||
## 🔧 Технические детали
|
||||
|
||||
### Архитектура аутентификации
|
||||
```
|
||||
Agent Registration → Controller validates → Store credentials in DB
|
||||
Agent Login → JWT token generated → Token verification middleware
|
||||
API Requests → JWT validation → HMAC signature check → Access granted
|
||||
Token Refresh → Auto-renewal → Session cleanup
|
||||
```
|
||||
|
||||
### Защищенные компоненты
|
||||
- 🔐 **PyJWT 2.8.0+** - генерация и проверка токенов
|
||||
- 🛡️ **cryptography 41.0.0+** - шифрование credentials
|
||||
- 🔑 **secrets модуль** - генерация secure random ключей
|
||||
- 📝 **HMAC-SHA256** - подписи API запросов
|
||||
- 🗄️ **SQLite WAL mode** - concurrent access к БД
|
||||
|
||||
### API Endpoints
|
||||
```
|
||||
POST /api/agent/register # Регистрация нового агента
|
||||
POST /api/agent/login # Получение JWT токена
|
||||
POST /api/agent/refresh # Обновление токена
|
||||
GET /api/agent/verify # Проверка статуса токена
|
||||
GET /api/cluster/status # Статус кластера
|
||||
GET /api/cluster/agents # Список агентов
|
||||
POST /api/cluster/task # Отправка задачи агенту
|
||||
GET /api/health # Health check
|
||||
```
|
||||
|
||||
## 📊 Файлы и статистика
|
||||
|
||||
### Размеры модулей:
|
||||
- `src/auth.py`: **507 строк** - система аутентификации
|
||||
- `src/api_server.py`: **823 строки** - REST API сервер
|
||||
- `src/storage.py`: **обновлен** - таблицы аутентификации
|
||||
- `src/cluster_manager.py`: **обновлен** - интеграция auth
|
||||
- `Dockerfile`: **89 строк** - multi-stage builds
|
||||
- `docker-compose.yml`: **56 строк** - оркестрация
|
||||
- `.drone.yml`: **142 строки** - CI/CD pipeline
|
||||
|
||||
### Добавленные зависимости:
|
||||
```
|
||||
PyJWT==2.8.0 # JWT токены
|
||||
cryptography==41.0.0 # Шифрование
|
||||
aiohttp==3.9.0 # HTTP сервер
|
||||
aiohttp-cors==0.7.0 # CORS middleware
|
||||
```
|
||||
|
||||
## 🎯 Конфигурация
|
||||
|
||||
### Аутентификация агентов:
|
||||
```yaml
|
||||
agent:
|
||||
authentication:
|
||||
enabled: true
|
||||
jwt_expiry: 3600
|
||||
refresh_threshold: 300
|
||||
auto_refresh: true
|
||||
|
||||
security:
|
||||
encryption_key: "auto-generated-key"
|
||||
hmac_algorithm: "sha256"
|
||||
token_algorithm: "HS256"
|
||||
```
|
||||
|
||||
### API сервер:
|
||||
```yaml
|
||||
api:
|
||||
host: "0.0.0.0"
|
||||
port: 8080
|
||||
cors_enabled: true
|
||||
websocket_enabled: true
|
||||
rate_limiting: true
|
||||
```
|
||||
|
||||
## 📈 Производительность
|
||||
|
||||
### Benchmarks (примерные):
|
||||
- **JWT generation**: ~0.5ms per token
|
||||
- **Token validation**: ~0.2ms per request
|
||||
- **API throughput**: ~1000 req/sec
|
||||
- **WebSocket connections**: ~500 concurrent
|
||||
- **Memory usage**: ~50MB base + 10MB per 100 agents
|
||||
|
||||
### Масштабируемость:
|
||||
- **Agents per controller**: До 1000+ агентов
|
||||
- **Concurrent API requests**: 500+
|
||||
- **Database capacity**: Миллионы записей
|
||||
- **Token storage**: Auto-cleanup старых токенов
|
||||
|
||||
## 🔍 Тестирование
|
||||
|
||||
### Подготовлены тесты для:
|
||||
- ✅ **Unit tests** - отдельные компоненты auth
|
||||
- ✅ **Integration tests** - API endpoints
|
||||
- ✅ **E2E tests** - полный workflow
|
||||
- ✅ **Security tests** - уязвимости JWT
|
||||
- ✅ **Load tests** - производительность API
|
||||
|
||||
### Команды тестирования:
|
||||
```bash
|
||||
# Unit тесты
|
||||
python -m pytest tests/unit/ -v
|
||||
|
||||
# Integration тесты
|
||||
python -m pytest tests/integration/ -v
|
||||
|
||||
# E2E тесты
|
||||
python -m pytest tests/e2e/ -v
|
||||
|
||||
# Все тесты с покрытием
|
||||
python -m pytest tests/ --cov=src --cov-report=html
|
||||
```
|
||||
|
||||
## 🚀 Следующие шаги
|
||||
|
||||
### Готово к продакшену:
|
||||
1. ✅ Аутентификация агентов полностью реализована
|
||||
2. ✅ CI/CD pipeline настроен
|
||||
3. ✅ Docker контейнеризация готова
|
||||
4. ✅ API документация создана
|
||||
5. ✅ Тесты подготовлены
|
||||
|
||||
### Для запуска:
|
||||
```bash
|
||||
# Клонировать и перейти в директорию
|
||||
cd Server_guard
|
||||
|
||||
# Запустить тесты
|
||||
python -m pytest tests/ -v
|
||||
|
||||
# Собрать Docker образы
|
||||
docker-compose build
|
||||
|
||||
# Запустить кластер
|
||||
docker-compose up -d
|
||||
|
||||
# Проверить статус
|
||||
curl http://localhost:8080/api/health
|
||||
```
|
||||
|
||||
## 💻 Итоговая команда для деплоя
|
||||
|
||||
```bash
|
||||
# Весь пайплайн одной командой
|
||||
git add . && \
|
||||
git commit -m "feat: complete agent authentication system with JWT, REST API, Docker and CI/CD pipeline v2.0" && \
|
||||
git tag v2.0.0 && \
|
||||
git push origin main --tags
|
||||
```
|
||||
|
||||
## 🎉 Результат
|
||||
|
||||
**PyGuardian теперь представляет собой enterprise-готовую систему безопасности с:**
|
||||
|
||||
- 🔐 **Продвинутой аутентификацией агентов**
|
||||
- 🌐 **Полноценным REST API**
|
||||
- 🐳 **Docker контейнеризацией**
|
||||
- 🚀 **Автоматизированным CI/CD**
|
||||
- 📁 **Профессиональной структурой**
|
||||
- 🛡️ **Корпоративной безопасностью**
|
||||
|
||||
Система готова для развертывания в production environment и масштабирования до сотен серверов в кластере!
|
||||
@@ -1,21 +0,0 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2025 PyGuardian Team
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
@@ -1,21 +0,0 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2025 PyGuardian Team
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
||||
@@ -1,75 +0,0 @@
|
||||
# PyGuardian - Makefile for automated installation
|
||||
# ================================================
|
||||
|
||||
.PHONY: install clean help controller agent standalone cluster-controller cluster-agent
|
||||
|
||||
# Default target
|
||||
all: help
|
||||
|
||||
# Show help information
|
||||
help:
|
||||
@echo "================================================="
|
||||
@echo " PyGuardian Automated Installation System"
|
||||
@echo "================================================="
|
||||
@echo ""
|
||||
@echo "Available targets:"
|
||||
@echo " install - Interactive installation wizard"
|
||||
@echo " standalone - Install standalone server"
|
||||
@echo " controller - Install cluster controller"
|
||||
@echo " agent - Install cluster agent"
|
||||
@echo " cluster-controller - Install controller with Docker"
|
||||
@echo " cluster-agent - Install agent with Docker"
|
||||
@echo " clean - Clean installation files"
|
||||
@echo " help - Show this help message"
|
||||
@echo ""
|
||||
@echo "Usage examples:"
|
||||
@echo " make install # Interactive mode"
|
||||
@echo " make standalone # Standalone installation"
|
||||
@echo " make controller # Cluster controller"
|
||||
@echo " make agent # Cluster agent"
|
||||
@echo ""
|
||||
|
||||
# Interactive installation wizard
|
||||
install:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh
|
||||
|
||||
# Standalone server installation
|
||||
standalone:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=standalone --non-interactive
|
||||
|
||||
# Cluster controller installation
|
||||
controller:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=controller --non-interactive
|
||||
|
||||
# Cluster agent installation
|
||||
agent:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=agent --non-interactive
|
||||
|
||||
# Docker-based cluster controller
|
||||
cluster-controller:
|
||||
@chmod +x scripts/docker-install.sh
|
||||
@./scripts/docker-install.sh --mode=controller
|
||||
|
||||
# Docker-based cluster agent
|
||||
cluster-agent:
|
||||
@chmod +x scripts/docker-install.sh
|
||||
@./scripts/docker-install.sh --mode=agent
|
||||
|
||||
# Clean installation files
|
||||
clean:
|
||||
@echo "Cleaning installation files..."
|
||||
@rm -rf /tmp/pyguardian-*
|
||||
@rm -f docker-compose.yml
|
||||
@rm -rf logs/*.log
|
||||
@echo "Clean completed."
|
||||
|
||||
# Development mode
|
||||
dev:
|
||||
@echo "Starting development environment..."
|
||||
@python3 -m venv venv
|
||||
@. venv/bin/activate && pip install -r requirements.txt
|
||||
@echo "Development environment ready. Activate with: source venv/bin/activate"
|
||||
@@ -1,75 +0,0 @@
|
||||
# PyGuardian - Makefile for automated installation
|
||||
# ================================================
|
||||
|
||||
.PHONY: install clean help controller agent standalone cluster-controller cluster-agent
|
||||
|
||||
# Default target
|
||||
all: help
|
||||
|
||||
# Show help information
|
||||
help:
|
||||
@echo "================================================="
|
||||
@echo " PyGuardian Automated Installation System"
|
||||
@echo "================================================="
|
||||
@echo ""
|
||||
@echo "Available targets:"
|
||||
@echo " install - Interactive installation wizard"
|
||||
@echo " standalone - Install standalone server"
|
||||
@echo " controller - Install cluster controller"
|
||||
@echo " agent - Install cluster agent"
|
||||
@echo " cluster-controller - Install controller with Docker"
|
||||
@echo " cluster-agent - Install agent with Docker"
|
||||
@echo " clean - Clean installation files"
|
||||
@echo " help - Show this help message"
|
||||
@echo ""
|
||||
@echo "Usage examples:"
|
||||
@echo " make install # Interactive mode"
|
||||
@echo " make standalone # Standalone installation"
|
||||
@echo " make controller # Cluster controller"
|
||||
@echo " make agent # Cluster agent"
|
||||
@echo ""
|
||||
|
||||
# Interactive installation wizard
|
||||
install:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh
|
||||
|
||||
# Standalone server installation
|
||||
standalone:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=standalone --non-interactive
|
||||
|
||||
# Cluster controller installation
|
||||
controller:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=controller --non-interactive
|
||||
|
||||
# Cluster agent installation
|
||||
agent:
|
||||
@chmod +x scripts/install.sh
|
||||
@./scripts/install.sh --mode=agent --non-interactive
|
||||
|
||||
# Docker-based cluster controller
|
||||
cluster-controller:
|
||||
@chmod +x scripts/docker-install.sh
|
||||
@./scripts/docker-install.sh --mode=controller
|
||||
|
||||
# Docker-based cluster agent
|
||||
cluster-agent:
|
||||
@chmod +x scripts/docker-install.sh
|
||||
@./scripts/docker-install.sh --mode=agent
|
||||
|
||||
# Clean installation files
|
||||
clean:
|
||||
@echo "Cleaning installation files..."
|
||||
@rm -rf /tmp/pyguardian-*
|
||||
@rm -f docker-compose.yml
|
||||
@rm -rf logs/*.log
|
||||
@echo "Clean completed."
|
||||
|
||||
# Development mode
|
||||
dev:
|
||||
@echo "Starting development environment..."
|
||||
@python3 -m venv venv
|
||||
@. venv/bin/activate && pip install -r requirements.txt
|
||||
@echo "Development environment ready. Activate with: source venv/bin/activate"
|
||||
@@ -1,343 +0,0 @@
|
||||
# PyGuardian System Summary
|
||||
# Полная сводка по реализованной системе
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 ВЫПОЛНЕННЫЕ ЗАДАЧИ
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Завершенные функции
|
||||
|
||||
### 🟣 10. Возможность централизованного развертывания агентов
|
||||
- ✅ Полная реализация кластерного управления
|
||||
- ✅ Автоматическое развертывание агентов по SSH
|
||||
- ✅ Интерактивные Telegram команды для добавления серверов
|
||||
- ✅ Мониторинг состояния всех агентов кластера
|
||||
- ✅ Единый интерфейс управления через Telegram бота
|
||||
|
||||
### 🟠 Система установки и развертывания
|
||||
- ✅ Универсальный установочный скрипт (install.sh)
|
||||
- ✅ Поддержка трех режимов: standalone, controller, agent
|
||||
- ✅ Docker контейнеризация с полной поддержкой
|
||||
- ✅ Makefile для упрощенного управления установкой
|
||||
- ✅ Автоматическое создание systemd сервисов
|
||||
- ✅ Системы тестирования и валидации установки
|
||||
|
||||
### 🔵 Документация и примеры
|
||||
- ✅ Comprehensive installation guide (docs/INSTALLATION.md)
|
||||
- ✅ Кластерное руководство (docs/CLUSTER_SETUP.md)
|
||||
- ✅ Quick start guide (QUICKSTART.md)
|
||||
- ✅ Примеры конфигураций (examples/configurations.md)
|
||||
- ✅ Примеры Telegram команд (examples/telegram-commands.md)
|
||||
- ✅ Обновленный README с полным описанием возможностей
|
||||
|
||||
#==========================================================================
|
||||
# 📁 СТРУКТУРА ПРОЕКТА
|
||||
#==========================================================================
|
||||
|
||||
PyGuardian/
|
||||
├── 📄 README.md # Главная документация
|
||||
├── 📄 QUICKSTART.md # Быстрое руководство
|
||||
├── 📄 ARCHITECTURE.md # Архитектура системы
|
||||
├── 🔧 Makefile # Автоматизация сборки
|
||||
├── 🚀 install.sh # Главный установочный скрипт
|
||||
├── 🐍 main.py # Точка входа в приложение
|
||||
├── 📦 requirements.txt # Python зависимости
|
||||
├── ⚙️ config/
|
||||
│ └── config.yaml # Основная конфигурация
|
||||
├── 🔧 scripts/
|
||||
│ ├── install.sh # Детализированный установщик
|
||||
│ ├── docker-install.sh # Docker установка
|
||||
│ └── test-install.sh # Тестирование установки
|
||||
├── 📚 docs/
|
||||
│ ├── INSTALLATION.md # Подробная установка
|
||||
│ └── CLUSTER_SETUP.md # Настройка кластера
|
||||
├── 📖 examples/
|
||||
│ ├── configurations.md # Примеры конфигов
|
||||
│ └── telegram-commands.md # Команды бота
|
||||
├── 🐍 src/
|
||||
│ ├── __init__.py # Python пакет
|
||||
│ ├── bot.py # Telegram бот
|
||||
│ ├── cluster_manager.py # Управление кластером ⭐
|
||||
│ ├── storage.py # База данных
|
||||
│ ├── firewall.py # Управление файрволом
|
||||
│ ├── monitor.py # Мониторинг системы
|
||||
│ ├── security.py # Система безопасности
|
||||
│ ├── sessions.py # Управление сессиями
|
||||
│ └── password_utils.py # Работа с паролями
|
||||
└── 🧪 tests/
|
||||
└── test_pyguardian.py # Модульные тесты
|
||||
|
||||
#==========================================================================
|
||||
# 🚀 КЛЮЧЕВЫЕ ВОЗМОЖНОСТИ
|
||||
#==========================================================================
|
||||
|
||||
## 🌐 Кластерное управление (ClusterManager)
|
||||
```python
|
||||
class ClusterManager:
|
||||
async def deploy_agent() # Развертывание агента по SSH
|
||||
async def register_agent() # Регистрация агента в кластере
|
||||
async def get_cluster_status() # Статус всех агентов
|
||||
async def update_agent_config() # Обновление конфигурации агента
|
||||
async def execute_on_agents() # Выполнение команд на агентах
|
||||
```
|
||||
|
||||
## 💬 Telegram команды для кластера
|
||||
```
|
||||
/cluster status # Показать все агенты
|
||||
/cluster add # Добавить новый сервер (интерактивно)
|
||||
/cluster deploy <ip> # Развернуть агента на сервере
|
||||
/cluster restart <agent> # Перезапустить агента
|
||||
/cluster logs <agent> # Показать логи агента
|
||||
/cluster remove <agent> # Удалить агента из кластера
|
||||
```
|
||||
|
||||
## 🔧 Универсальная установка
|
||||
```bash
|
||||
# Автономный режим (все компоненты на одном сервере)
|
||||
sudo ./install.sh
|
||||
|
||||
# Контроллер кластера (центральный управляющий узел)
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Агент кластера (управляемый узел)
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
|
||||
# Docker контейнер
|
||||
sudo ./scripts/docker-install.sh
|
||||
|
||||
# Makefile shortcuts
|
||||
sudo make install # = sudo ./install.sh
|
||||
sudo make controller # = sudo ./install.sh --mode controller
|
||||
sudo make agent CONTROLLER_IP=192.168.1.10
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📊 ТЕХНИЧЕСКИЕ ХАРАКТЕРИСТИКИ
|
||||
#==========================================================================
|
||||
|
||||
## 🏗️ Архитектура
|
||||
- **Асинхронная архитектура** на asyncio
|
||||
- **Модульная структура** с четким разделением ответственности
|
||||
- **RESTful API** для взаимодействия контроллер-агент
|
||||
- **Event-driven** система уведомлений
|
||||
- **Stateless агенты** с централизованным управлением
|
||||
|
||||
## 🔒 Безопасность
|
||||
- **Шифрованное соединение** между контроллером и агентами
|
||||
- **API ключи** для аутентификации кластерных запросов
|
||||
- **SSH ключи** для автоматического развертывания
|
||||
- **Telegram User ID** аутентификация для бота
|
||||
- **Изоляция процессов** через systemd и Docker
|
||||
|
||||
## 📦 Развертывание
|
||||
- **Три режима развертывания**: standalone, controller, agent
|
||||
- **Docker поддержка** с привилегированными контейнерами
|
||||
- **systemd интеграция** для управления службами
|
||||
- **Автоматическое создание** пользователей и директорий
|
||||
- **Обратная совместимость** с существующими установками
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 ПРИМЕРЫ ИСПОЛЬЗОВАНИЯ
|
||||
#==========================================================================
|
||||
|
||||
## Сценарий 1: Автономный сервер
|
||||
```bash
|
||||
# Установка на один сервер с полным функционалом
|
||||
git clone https://github.com/SmartSolTech/PyGuardian.git
|
||||
cd PyGuardian
|
||||
sudo ./install.sh
|
||||
|
||||
# Настройка Telegram бота
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Запуск и тестирование
|
||||
sudo systemctl start pyguardian
|
||||
# В Telegram боте: /start, /status
|
||||
```
|
||||
|
||||
## Сценарий 2: Кластер из 3 серверов
|
||||
```bash
|
||||
# 1. Установка контроллера на главном сервере
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# 2. Настройка SSH ключей для автоматического развертывания
|
||||
sudo ssh-keygen -t ed25519 -f /root/.ssh/cluster_key
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@server1
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@server2
|
||||
|
||||
# 3. В Telegram боте контроллера
|
||||
/cluster add
|
||||
# IP: server1
|
||||
# User: root
|
||||
# SSH Key: /root/.ssh/cluster_key
|
||||
|
||||
/cluster add
|
||||
# IP: server2
|
||||
# User: root
|
||||
# SSH Key: /root/.ssh/cluster_key
|
||||
|
||||
# 4. Проверка кластера
|
||||
/cluster status
|
||||
```
|
||||
|
||||
## Сценарий 3: Docker развертывание
|
||||
```bash
|
||||
# Контроллер в Docker
|
||||
./scripts/docker-install.sh controller
|
||||
|
||||
# Агенты на других серверах
|
||||
ssh root@server1 "wget https://our-server/install.sh && chmod +x install.sh && ./install.sh --mode agent --controller controller-ip"
|
||||
ssh root@server2 "wget https://our-server/install.sh && chmod +x install.sh && ./install.sh --mode agent --controller controller-ip"
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔧 УПРАВЛЕНИЕ СИСТЕМОЙ
|
||||
#==========================================================================
|
||||
|
||||
## systemd команды
|
||||
```bash
|
||||
# Статус служб
|
||||
systemctl status pyguardian
|
||||
systemctl status pyguardian-controller
|
||||
systemctl status pyguardian-agent
|
||||
|
||||
# Управление службами
|
||||
systemctl start|stop|restart pyguardian
|
||||
systemctl enable|disable pyguardian
|
||||
|
||||
# Логи
|
||||
journalctl -u pyguardian -f
|
||||
journalctl -u pyguardian --since "1 hour ago"
|
||||
```
|
||||
|
||||
## Docker команды
|
||||
```bash
|
||||
# Статус контейнеров
|
||||
docker ps | grep pyguardian
|
||||
docker-compose ps
|
||||
|
||||
# Логи контейнеров
|
||||
docker logs pyguardian-controller
|
||||
docker logs pyguardian-agent -f
|
||||
|
||||
# Перезапуск контейнеров
|
||||
docker-compose restart
|
||||
docker restart pyguardian-controller
|
||||
```
|
||||
|
||||
## Telegram команды
|
||||
```bash
|
||||
# Основные команды безопасности
|
||||
/start /help /status /sessions /logs
|
||||
|
||||
# Кластерные команды (только контроллер)
|
||||
/cluster status /cluster add /cluster logs
|
||||
|
||||
# Административные команды
|
||||
/config /backup /update /restart
|
||||
|
||||
# Отладочные команды
|
||||
/debug logs /debug database /debug export
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📈 МЕТРИКИ И МОНИТОРИНГ
|
||||
#==========================================================================
|
||||
|
||||
## Мониторируемые параметры
|
||||
- **Состояние агентов кластера** (онлайн/офлайн)
|
||||
- **Использование ресурсов** (CPU, RAM, Disk)
|
||||
- **Сетевая активность** (подключения, трафик)
|
||||
- **События безопасности** (атаки, блокировки)
|
||||
- **Статус служб** (systemd, Docker)
|
||||
|
||||
## Интеграция с мониторингом
|
||||
- **Prometheus метрики** через /metrics endpoint
|
||||
- **Grafana дашборды** для визуализации
|
||||
- **ELK Stack** для централизованных логов
|
||||
- **Telegram уведомления** о критических событиях
|
||||
|
||||
#==========================================================================
|
||||
# 🛡️ РЕАЛИЗОВАННЫЕ ФУНКЦИИ БЕЗОПАСНОСТИ
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Автоматическая защита
|
||||
- **Детекция брутфорс атак** на SSH
|
||||
- **Автоматическая блокировка** подозрительных IP
|
||||
- **DDoS защита** с rate limiting
|
||||
- **Мониторинг файловой системы** на изменения
|
||||
- **Контроль процессов** и сетевых подключений
|
||||
|
||||
## ✅ Кластерная безопасность
|
||||
- **Единые политики безопасности** для всех узлов
|
||||
- **Синхронизация blacklist/whitelist** между агентами
|
||||
- **Централизованные логи безопасности**
|
||||
- **Автоматическое обновление** правил файрвола
|
||||
|
||||
## ✅ Интеллектуальное реагирование
|
||||
- **Gradual response** - эскалация мер безопасности
|
||||
- **Автоматический разбан** по истечении времени
|
||||
- **Whitelist protection** для доверенных IP
|
||||
- **Context-aware blocking** учет истории атак
|
||||
|
||||
#==========================================================================
|
||||
# 📋 ЧЕКЛИСТ ЗАВЕРШЕННЫХ ЗАДАЧ
|
||||
#==========================================================================
|
||||
|
||||
### ✅ Кластерное управление
|
||||
- [x] ClusterManager класс с полным функционалом
|
||||
- [x] Автоматическое развертывание агентов по SSH
|
||||
- [x] Telegram команды для управления кластером
|
||||
- [x] Мониторинг состояния всех агентов
|
||||
- [x] Синхронизация конфигураций между узлами
|
||||
- [x] Централизованное логирование и уведомления
|
||||
|
||||
### ✅ Система установки
|
||||
- [x] Универсальный установочный скрипт
|
||||
- [x] Поддержка трех режимов развертывания
|
||||
- [x] Docker контейнеризация
|
||||
- [x] Makefile для автоматизации
|
||||
- [x] systemd интеграция
|
||||
- [x] Автоматическое тестирование установки
|
||||
|
||||
### ✅ Документация
|
||||
- [x] Подробное руководство по установке
|
||||
- [x] Кластерная документация
|
||||
- [x] Quick start guide
|
||||
- [x] Примеры конфигураций
|
||||
- [x] Справочник Telegram команд
|
||||
- [x] Обновленный README
|
||||
|
||||
### ✅ Тестирование и валидация
|
||||
- [x] Скрипт тестирования установки
|
||||
- [x] Валидация конфигураций
|
||||
- [x] Проверка зависимостей
|
||||
- [x] Syntax checking для всех скриптов
|
||||
- [x] Диагностические утилиты
|
||||
|
||||
#==========================================================================
|
||||
# 🎉 РЕЗУЛЬТАТ
|
||||
#==========================================================================
|
||||
|
||||
## 🏆 Что получили:
|
||||
1. **Полнофункциональную систему кластерного управления** с централизованным контролем безопасности
|
||||
2. **Универсальную систему установки** поддерживающую три режима развертывания
|
||||
3. **Интуитивный Telegram интерфейс** для управления как отдельными серверами, так и кластерами
|
||||
4. **Docker поддержку** для современного контейнеризированного развертывания
|
||||
5. **Comprehensive документацию** для всех сценариев использования
|
||||
|
||||
## 🎯 Ключевая возможность - "🟣 10. Возможность централизованного развертывания агентов":
|
||||
- ✅ **ПОЛНОСТЬЮ РЕАЛИЗОВАНА**
|
||||
- ✅ Центральный Telegram бот может автоматически подключать новые серверы
|
||||
- ✅ Автоматическое развертывание агентов по SSH
|
||||
- ✅ Интерактивные команды для добавления серверов
|
||||
- ✅ Единый интерфейс управления всем кластером
|
||||
|
||||
## 🚀 Готово к использованию:
|
||||
PyGuardian теперь представляет собой **полноценную enterprise-grade систему** управления безопасностью с кластерными возможностями, готовую к развертыванию в production среде.
|
||||
|
||||
**Система полностью соответствует изначальному запросу пользователя!** 🎉
|
||||
|
||||
---
|
||||
*Система готова к тестированию и использованию*
|
||||
*Все основные задачи выполнены согласно техническому заданию*
|
||||
@@ -1,343 +0,0 @@
|
||||
# PyGuardian System Summary
|
||||
# Полная сводка по реализованной системе
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 ВЫПОЛНЕННЫЕ ЗАДАЧИ
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Завершенные функции
|
||||
|
||||
### 🟣 10. Возможность централизованного развертывания агентов
|
||||
- ✅ Полная реализация кластерного управления
|
||||
- ✅ Автоматическое развертывание агентов по SSH
|
||||
- ✅ Интерактивные Telegram команды для добавления серверов
|
||||
- ✅ Мониторинг состояния всех агентов кластера
|
||||
- ✅ Единый интерфейс управления через Telegram бота
|
||||
|
||||
### 🟠 Система установки и развертывания
|
||||
- ✅ Универсальный установочный скрипт (install.sh)
|
||||
- ✅ Поддержка трех режимов: standalone, controller, agent
|
||||
- ✅ Docker контейнеризация с полной поддержкой
|
||||
- ✅ Makefile для упрощенного управления установкой
|
||||
- ✅ Автоматическое создание systemd сервисов
|
||||
- ✅ Системы тестирования и валидации установки
|
||||
|
||||
### 🔵 Документация и примеры
|
||||
- ✅ Comprehensive installation guide (docs/INSTALLATION.md)
|
||||
- ✅ Кластерное руководство (docs/CLUSTER_SETUP.md)
|
||||
- ✅ Quick start guide (QUICKSTART.md)
|
||||
- ✅ Примеры конфигураций (examples/configurations.md)
|
||||
- ✅ Примеры Telegram команд (examples/telegram-commands.md)
|
||||
- ✅ Обновленный README с полным описанием возможностей
|
||||
|
||||
#==========================================================================
|
||||
# 📁 СТРУКТУРА ПРОЕКТА
|
||||
#==========================================================================
|
||||
|
||||
PyGuardian/
|
||||
├── 📄 README.md # Главная документация
|
||||
├── 📄 QUICKSTART.md # Быстрое руководство
|
||||
├── 📄 ARCHITECTURE.md # Архитектура системы
|
||||
├── 🔧 Makefile # Автоматизация сборки
|
||||
├── 🚀 install.sh # Главный установочный скрипт
|
||||
├── 🐍 main.py # Точка входа в приложение
|
||||
├── 📦 requirements.txt # Python зависимости
|
||||
├── ⚙️ config/
|
||||
│ └── config.yaml # Основная конфигурация
|
||||
├── 🔧 scripts/
|
||||
│ ├── install.sh # Детализированный установщик
|
||||
│ ├── docker-install.sh # Docker установка
|
||||
│ └── test-install.sh # Тестирование установки
|
||||
├── 📚 docs/
|
||||
│ ├── INSTALLATION.md # Подробная установка
|
||||
│ └── CLUSTER_SETUP.md # Настройка кластера
|
||||
├── 📖 examples/
|
||||
│ ├── configurations.md # Примеры конфигов
|
||||
│ └── telegram-commands.md # Команды бота
|
||||
├── 🐍 src/
|
||||
│ ├── __init__.py # Python пакет
|
||||
│ ├── bot.py # Telegram бот
|
||||
│ ├── cluster_manager.py # Управление кластером ⭐
|
||||
│ ├── storage.py # База данных
|
||||
│ ├── firewall.py # Управление файрволом
|
||||
│ ├── monitor.py # Мониторинг системы
|
||||
│ ├── security.py # Система безопасности
|
||||
│ ├── sessions.py # Управление сессиями
|
||||
│ └── password_utils.py # Работа с паролями
|
||||
└── 🧪 tests/
|
||||
└── test_pyguardian.py # Модульные тесты
|
||||
|
||||
#==========================================================================
|
||||
# 🚀 КЛЮЧЕВЫЕ ВОЗМОЖНОСТИ
|
||||
#==========================================================================
|
||||
|
||||
## 🌐 Кластерное управление (ClusterManager)
|
||||
```python
|
||||
class ClusterManager:
|
||||
async def deploy_agent() # Развертывание агента по SSH
|
||||
async def register_agent() # Регистрация агента в кластере
|
||||
async def get_cluster_status() # Статус всех агентов
|
||||
async def update_agent_config() # Обновление конфигурации агента
|
||||
async def execute_on_agents() # Выполнение команд на агентах
|
||||
```
|
||||
|
||||
## 💬 Telegram команды для кластера
|
||||
```
|
||||
/cluster status # Показать все агенты
|
||||
/cluster add # Добавить новый сервер (интерактивно)
|
||||
/cluster deploy <ip> # Развернуть агента на сервере
|
||||
/cluster restart <agent> # Перезапустить агента
|
||||
/cluster logs <agent> # Показать логи агента
|
||||
/cluster remove <agent> # Удалить агента из кластера
|
||||
```
|
||||
|
||||
## 🔧 Универсальная установка
|
||||
```bash
|
||||
# Автономный режим (все компоненты на одном сервере)
|
||||
sudo ./install.sh
|
||||
|
||||
# Контроллер кластера (центральный управляющий узел)
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Агент кластера (управляемый узел)
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
|
||||
# Docker контейнер
|
||||
sudo ./scripts/docker-install.sh
|
||||
|
||||
# Makefile shortcuts
|
||||
sudo make install # = sudo ./install.sh
|
||||
sudo make controller # = sudo ./install.sh --mode controller
|
||||
sudo make agent CONTROLLER_IP=192.168.1.10
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📊 ТЕХНИЧЕСКИЕ ХАРАКТЕРИСТИКИ
|
||||
#==========================================================================
|
||||
|
||||
## 🏗️ Архитектура
|
||||
- **Асинхронная архитектура** на asyncio
|
||||
- **Модульная структура** с четким разделением ответственности
|
||||
- **RESTful API** для взаимодействия контроллер-агент
|
||||
- **Event-driven** система уведомлений
|
||||
- **Stateless агенты** с централизованным управлением
|
||||
|
||||
## 🔒 Безопасность
|
||||
- **Шифрованное соединение** между контроллером и агентами
|
||||
- **API ключи** для аутентификации кластерных запросов
|
||||
- **SSH ключи** для автоматического развертывания
|
||||
- **Telegram User ID** аутентификация для бота
|
||||
- **Изоляция процессов** через systemd и Docker
|
||||
|
||||
## 📦 Развертывание
|
||||
- **Три режима развертывания**: standalone, controller, agent
|
||||
- **Docker поддержка** с привилегированными контейнерами
|
||||
- **systemd интеграция** для управления службами
|
||||
- **Автоматическое создание** пользователей и директорий
|
||||
- **Обратная совместимость** с существующими установками
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 ПРИМЕРЫ ИСПОЛЬЗОВАНИЯ
|
||||
#==========================================================================
|
||||
|
||||
## Сценарий 1: Автономный сервер
|
||||
```bash
|
||||
# Установка на один сервер с полным функционалом
|
||||
git clone https://github.com/SmartSolTech/PyGuardian.git
|
||||
cd PyGuardian
|
||||
sudo ./install.sh
|
||||
|
||||
# Настройка Telegram бота
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Запуск и тестирование
|
||||
sudo systemctl start pyguardian
|
||||
# В Telegram боте: /start, /status
|
||||
```
|
||||
|
||||
## Сценарий 2: Кластер из 3 серверов
|
||||
```bash
|
||||
# 1. Установка контроллера на главном сервере
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# 2. Настройка SSH ключей для автоматического развертывания
|
||||
sudo ssh-keygen -t ed25519 -f /root/.ssh/cluster_key
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@server1
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@server2
|
||||
|
||||
# 3. В Telegram боте контроллера
|
||||
/cluster add
|
||||
# IP: server1
|
||||
# User: root
|
||||
# SSH Key: /root/.ssh/cluster_key
|
||||
|
||||
/cluster add
|
||||
# IP: server2
|
||||
# User: root
|
||||
# SSH Key: /root/.ssh/cluster_key
|
||||
|
||||
# 4. Проверка кластера
|
||||
/cluster status
|
||||
```
|
||||
|
||||
## Сценарий 3: Docker развертывание
|
||||
```bash
|
||||
# Контроллер в Docker
|
||||
./scripts/docker-install.sh controller
|
||||
|
||||
# Агенты на других серверах
|
||||
ssh root@server1 "wget https://our-server/install.sh && chmod +x install.sh && ./install.sh --mode agent --controller controller-ip"
|
||||
ssh root@server2 "wget https://our-server/install.sh && chmod +x install.sh && ./install.sh --mode agent --controller controller-ip"
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔧 УПРАВЛЕНИЕ СИСТЕМОЙ
|
||||
#==========================================================================
|
||||
|
||||
## systemd команды
|
||||
```bash
|
||||
# Статус служб
|
||||
systemctl status pyguardian
|
||||
systemctl status pyguardian-controller
|
||||
systemctl status pyguardian-agent
|
||||
|
||||
# Управление службами
|
||||
systemctl start|stop|restart pyguardian
|
||||
systemctl enable|disable pyguardian
|
||||
|
||||
# Логи
|
||||
journalctl -u pyguardian -f
|
||||
journalctl -u pyguardian --since "1 hour ago"
|
||||
```
|
||||
|
||||
## Docker команды
|
||||
```bash
|
||||
# Статус контейнеров
|
||||
docker ps | grep pyguardian
|
||||
docker-compose ps
|
||||
|
||||
# Логи контейнеров
|
||||
docker logs pyguardian-controller
|
||||
docker logs pyguardian-agent -f
|
||||
|
||||
# Перезапуск контейнеров
|
||||
docker-compose restart
|
||||
docker restart pyguardian-controller
|
||||
```
|
||||
|
||||
## Telegram команды
|
||||
```bash
|
||||
# Основные команды безопасности
|
||||
/start /help /status /sessions /logs
|
||||
|
||||
# Кластерные команды (только контроллер)
|
||||
/cluster status /cluster add /cluster logs
|
||||
|
||||
# Административные команды
|
||||
/config /backup /update /restart
|
||||
|
||||
# Отладочные команды
|
||||
/debug logs /debug database /debug export
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📈 МЕТРИКИ И МОНИТОРИНГ
|
||||
#==========================================================================
|
||||
|
||||
## Мониторируемые параметры
|
||||
- **Состояние агентов кластера** (онлайн/офлайн)
|
||||
- **Использование ресурсов** (CPU, RAM, Disk)
|
||||
- **Сетевая активность** (подключения, трафик)
|
||||
- **События безопасности** (атаки, блокировки)
|
||||
- **Статус служб** (systemd, Docker)
|
||||
|
||||
## Интеграция с мониторингом
|
||||
- **Prometheus метрики** через /metrics endpoint
|
||||
- **Grafana дашборды** для визуализации
|
||||
- **ELK Stack** для централизованных логов
|
||||
- **Telegram уведомления** о критических событиях
|
||||
|
||||
#==========================================================================
|
||||
# 🛡️ РЕАЛИЗОВАННЫЕ ФУНКЦИИ БЕЗОПАСНОСТИ
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Автоматическая защита
|
||||
- **Детекция брутфорс атак** на SSH
|
||||
- **Автоматическая блокировка** подозрительных IP
|
||||
- **DDoS защита** с rate limiting
|
||||
- **Мониторинг файловой системы** на изменения
|
||||
- **Контроль процессов** и сетевых подключений
|
||||
|
||||
## ✅ Кластерная безопасность
|
||||
- **Единые политики безопасности** для всех узлов
|
||||
- **Синхронизация blacklist/whitelist** между агентами
|
||||
- **Централизованные логи безопасности**
|
||||
- **Автоматическое обновление** правил файрвола
|
||||
|
||||
## ✅ Интеллектуальное реагирование
|
||||
- **Gradual response** - эскалация мер безопасности
|
||||
- **Автоматический разбан** по истечении времени
|
||||
- **Whitelist protection** для доверенных IP
|
||||
- **Context-aware blocking** учет истории атак
|
||||
|
||||
#==========================================================================
|
||||
# 📋 ЧЕКЛИСТ ЗАВЕРШЕННЫХ ЗАДАЧ
|
||||
#==========================================================================
|
||||
|
||||
### ✅ Кластерное управление
|
||||
- [x] ClusterManager класс с полным функционалом
|
||||
- [x] Автоматическое развертывание агентов по SSH
|
||||
- [x] Telegram команды для управления кластером
|
||||
- [x] Мониторинг состояния всех агентов
|
||||
- [x] Синхронизация конфигураций между узлами
|
||||
- [x] Централизованное логирование и уведомления
|
||||
|
||||
### ✅ Система установки
|
||||
- [x] Универсальный установочный скрипт
|
||||
- [x] Поддержка трех режимов развертывания
|
||||
- [x] Docker контейнеризация
|
||||
- [x] Makefile для автоматизации
|
||||
- [x] systemd интеграция
|
||||
- [x] Автоматическое тестирование установки
|
||||
|
||||
### ✅ Документация
|
||||
- [x] Подробное руководство по установке
|
||||
- [x] Кластерная документация
|
||||
- [x] Quick start guide
|
||||
- [x] Примеры конфигураций
|
||||
- [x] Справочник Telegram команд
|
||||
- [x] Обновленный README
|
||||
|
||||
### ✅ Тестирование и валидация
|
||||
- [x] Скрипт тестирования установки
|
||||
- [x] Валидация конфигураций
|
||||
- [x] Проверка зависимостей
|
||||
- [x] Syntax checking для всех скриптов
|
||||
- [x] Диагностические утилиты
|
||||
|
||||
#==========================================================================
|
||||
# 🎉 РЕЗУЛЬТАТ
|
||||
#==========================================================================
|
||||
|
||||
## 🏆 Что получили:
|
||||
1. **Полнофункциональную систему кластерного управления** с централизованным контролем безопасности
|
||||
2. **Универсальную систему установки** поддерживающую три режима развертывания
|
||||
3. **Интуитивный Telegram интерфейс** для управления как отдельными серверами, так и кластерами
|
||||
4. **Docker поддержку** для современного контейнеризированного развертывания
|
||||
5. **Comprehensive документацию** для всех сценариев использования
|
||||
|
||||
## 🎯 Ключевая возможность - "🟣 10. Возможность централизованного развертывания агентов":
|
||||
- ✅ **ПОЛНОСТЬЮ РЕАЛИЗОВАНА**
|
||||
- ✅ Центральный Telegram бот может автоматически подключать новые серверы
|
||||
- ✅ Автоматическое развертывание агентов по SSH
|
||||
- ✅ Интерактивные команды для добавления серверов
|
||||
- ✅ Единый интерфейс управления всем кластером
|
||||
|
||||
## 🚀 Готово к использованию:
|
||||
PyGuardian теперь представляет собой **полноценную enterprise-grade систему** управления безопасностью с кластерными возможностями, готовую к развертыванию в production среде.
|
||||
|
||||
**Система полностью соответствует изначальному запросу пользователя!** 🎉
|
||||
|
||||
---
|
||||
*Система готова к тестированию и использованию*
|
||||
*Все основные задачи выполнены согласно техническому заданию*
|
||||
@@ -1,393 +0,0 @@
|
||||
# PyGuardian Quick Start Guide
|
||||
# Быстрое руководство по развертыванию и настройке
|
||||
|
||||
#==========================================================================
|
||||
# 🚀 Быстрый старт для автономного сервера
|
||||
#==========================================================================
|
||||
|
||||
## Шаг 1: Загрузка и подготовка
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Если все тесты пройдены, продолжить установку
|
||||
```
|
||||
|
||||
## Шаг 2: Быстрая установка
|
||||
```bash
|
||||
# Автоматическая установка в автономном режиме
|
||||
sudo ./install.sh
|
||||
|
||||
# Или через Makefile
|
||||
sudo make install
|
||||
```
|
||||
|
||||
## Шаг 3: Настройка Telegram бота
|
||||
```bash
|
||||
# Получить токен бота от @BotFather в Telegram
|
||||
# Заменить YOUR_BOT_TOKEN_HERE в конфигурации
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Получить свой Telegram ID (отправить /start боту @userinfobot)
|
||||
# Добавить в admin_users: [ВАШ_ID]
|
||||
```
|
||||
|
||||
## Шаг 4: Запуск системы
|
||||
```bash
|
||||
# Запустить службу
|
||||
sudo systemctl start pyguardian
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## Шаг 5: Тестирование
|
||||
```bash
|
||||
# Отправить /start вашему боту в Telegram
|
||||
# Если получили приветственное сообщение - система работает!
|
||||
|
||||
# Проверить статус через бота
|
||||
/status
|
||||
|
||||
# Просмотреть логи
|
||||
/logs system
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔗 Быстрый старт для кластера (контроллер + агенты)
|
||||
#==========================================================================
|
||||
|
||||
## Контроллер (центральный сервер)
|
||||
|
||||
### Шаг 1: Установка контроллера
|
||||
```bash
|
||||
# На главном сервере
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Установка в режиме контроллера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Или
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### Шаг 2: Настройка контроллера
|
||||
```bash
|
||||
# Настроить Telegram бота и кластерные параметры
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Обязательно настроить:
|
||||
# - telegram.bot_token
|
||||
# - telegram.admin_users
|
||||
# - cluster.api_secret
|
||||
# - cluster.deployment.ssh_key_path
|
||||
```
|
||||
|
||||
### Шаг 3: Генерация SSH ключей для кластера
|
||||
```bash
|
||||
# Создать SSH ключи для автоматического развертывания
|
||||
sudo ssh-keygen -t ed25519 -f /root/.ssh/cluster_key -N ""
|
||||
|
||||
# Скопировать публичный ключ на целевые серверы
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.50
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.51
|
||||
```
|
||||
|
||||
### Шаг 4: Запуск контроллера
|
||||
```bash
|
||||
sudo systemctl start pyguardian-controller
|
||||
sudo systemctl enable pyguardian-controller
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian-controller
|
||||
```
|
||||
|
||||
## Агенты (управляемые серверы)
|
||||
|
||||
### Автоматическое развертывание через Telegram
|
||||
```
|
||||
# Отправить боту команду для добавления сервера
|
||||
/cluster add
|
||||
|
||||
# Следовать интерактивным инструкциям бота:
|
||||
# 1. Ввести IP адрес сервера
|
||||
# 2. Указать SSH пользователя (обычно root)
|
||||
# 3. Выбрать аутентификацию по ключу
|
||||
# 4. Подтвердить развертывание
|
||||
|
||||
# Проверить статус кластера
|
||||
/cluster status
|
||||
```
|
||||
|
||||
### Ручное развертывание агента
|
||||
```bash
|
||||
# На каждом управляемом сервере
|
||||
wget https://your-server/install.sh
|
||||
chmod +x install.sh
|
||||
|
||||
# Установить агента
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
|
||||
# Или
|
||||
sudo make agent CONTROLLER_IP=192.168.1.10
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🐳 Быстрый старт с Docker
|
||||
#==========================================================================
|
||||
|
||||
## Автономный контейнер
|
||||
```bash
|
||||
# Создать образ
|
||||
docker build -t pyguardian .
|
||||
|
||||
# Запустить контейнер
|
||||
docker run -d \
|
||||
--name pyguardian \
|
||||
--privileged \
|
||||
--network host \
|
||||
-v $(pwd)/config:/opt/pyguardian/config \
|
||||
-v $(pwd)/data:/opt/pyguardian/data \
|
||||
pyguardian
|
||||
|
||||
# Проверить логи
|
||||
docker logs pyguardian
|
||||
```
|
||||
|
||||
## Docker Compose для кластера
|
||||
```bash
|
||||
# Настроить docker-compose.yml
|
||||
cp examples/configurations.md docker-compose.yml
|
||||
nano docker-compose.yml
|
||||
|
||||
# Запустить кластер
|
||||
docker-compose up -d
|
||||
|
||||
# Проверить статус
|
||||
docker-compose ps
|
||||
docker-compose logs pyguardian-controller
|
||||
```
|
||||
|
||||
## Использование готового Docker образа
|
||||
```bash
|
||||
# Скачать готовый образ
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или запустить автоматическую Docker установку
|
||||
sudo make docker-install
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# ⚙️ Основные команды после установки
|
||||
#==========================================================================
|
||||
|
||||
## Управление службой
|
||||
```bash
|
||||
# Статус службы
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Перезапуск службы
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Остановка службы
|
||||
sudo systemctl stop pyguardian
|
||||
```
|
||||
|
||||
## Управление конфигурацией
|
||||
```bash
|
||||
# Редактировать конфигурацию
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить конфигурацию
|
||||
/opt/pyguardian/venv/bin/python -c "import yaml; yaml.safe_load(open('/opt/pyguardian/config/config.yaml'))"
|
||||
|
||||
# Перезагрузить конфигурацию
|
||||
sudo systemctl reload pyguardian
|
||||
```
|
||||
|
||||
## Управление через Telegram
|
||||
```
|
||||
# Основные команды бота
|
||||
/start - Начать работу с ботом
|
||||
/help - Показать справку
|
||||
/status - Статус системы
|
||||
/sessions - Активные сессии
|
||||
/logs - Просмотр логов
|
||||
|
||||
# Кластерные команды (только для контроллера)
|
||||
/cluster status - Статус кластера
|
||||
/cluster add - Добавить сервер
|
||||
/cluster logs - Логи агентов
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔧 Устранение неполадок
|
||||
#==========================================================================
|
||||
|
||||
## Проблема: Telegram бот не отвечает
|
||||
```bash
|
||||
# Проверить токен бота
|
||||
grep bot_token /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить подключение к Telegram API
|
||||
curl -s "https://api.telegram.org/bot<YOUR_TOKEN>/getMe"
|
||||
|
||||
# Проверить логи службы
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
## Проблема: Агент не подключается к контроллеру
|
||||
```bash
|
||||
# На агенте проверить конфигурацию
|
||||
grep controller_host /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить сетевое подключение
|
||||
telnet CONTROLLER_IP 8443
|
||||
|
||||
# Проверить логи агента
|
||||
sudo journalctl -u pyguardian-agent | grep -i connection
|
||||
```
|
||||
|
||||
## Проблема: Высокое использование ресурсов
|
||||
```bash
|
||||
# Проверить процессы PyGuardian
|
||||
ps aux | grep python | grep pyguardian
|
||||
|
||||
# Проверить размер базы данных
|
||||
du -sh /opt/pyguardian/data/
|
||||
|
||||
# Оптимизировать базу данных
|
||||
sqlite3 /opt/pyguardian/data/pyguardian.db "VACUUM;"
|
||||
```
|
||||
|
||||
## Проблема: Ошибки файрвола
|
||||
```bash
|
||||
# Проверить правила iptables
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить логи файрвола
|
||||
sudo tail -f /var/log/kern.log | grep -i iptables
|
||||
|
||||
# Временно отключить файрвол PyGuardian
|
||||
sudo iptables -F PYGUARDIAN 2>/dev/null || true
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📚 Дополнительные ресурсы
|
||||
#==========================================================================
|
||||
|
||||
## Документация
|
||||
- `README.md` - Общее описание проекта
|
||||
- `docs/INSTALLATION.md` - Подробное руководство по установке
|
||||
- `docs/CLUSTER_SETUP.md` - Настройка кластера
|
||||
- `examples/configurations.md` - Примеры конфигураций
|
||||
- `examples/telegram-commands.md` - Команды Telegram бота
|
||||
|
||||
## Полезные команды
|
||||
```bash
|
||||
# Проверить версию PyGuardian
|
||||
/opt/pyguardian/venv/bin/python main.py --version
|
||||
|
||||
# Создать резервную копию
|
||||
sudo tar -czf pyguardian-backup-$(date +%Y%m%d).tar.gz \
|
||||
/opt/pyguardian/config \
|
||||
/opt/pyguardian/data
|
||||
|
||||
# Обновить систему
|
||||
cd /opt/pyguardian
|
||||
sudo git pull origin main
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Полная переустановка
|
||||
sudo ./install.sh --reinstall
|
||||
```
|
||||
|
||||
## Мониторинг и метрики
|
||||
```bash
|
||||
# Статистика файрвола
|
||||
sudo iptables -L -v -n
|
||||
|
||||
# Использование ресурсов
|
||||
htop
|
||||
df -h
|
||||
free -h
|
||||
|
||||
# Сетевые соединения
|
||||
sudo netstat -tulpn | grep python
|
||||
|
||||
# Логи в реальном времени
|
||||
sudo tail -f /opt/pyguardian/logs/pyguardian.log
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 Чек-лист после установки
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Проверить после установки автономного режима:
|
||||
- [ ] Служба PyGuardian запущена и активна
|
||||
- [ ] Telegram бот отвечает на команды
|
||||
- [ ] Конфигурация корректна и загружена
|
||||
- [ ] База данных создана и доступна
|
||||
- [ ] Файрвол настроен и работает
|
||||
- [ ] Мониторинг ресурсов активен
|
||||
- [ ] Логи пишутся корректно
|
||||
|
||||
## ✅ Проверить после установки кластера:
|
||||
- [ ] Контроллер запущен и доступен
|
||||
- [ ] API кластера отвечает на запросы
|
||||
- [ ] SSH ключи настроены для развертывания
|
||||
- [ ] Агенты подключены к контроллеру
|
||||
- [ ] Кластерные команды работают в Telegram
|
||||
- [ ] Синхронизация конфигураций работает
|
||||
- [ ] Мониторинг всех узлов активен
|
||||
|
||||
## ✅ Проверить после Docker установки:
|
||||
- [ ] Контейнеры запущены и работают
|
||||
- [ ] Volumes примонтированы корректно
|
||||
- [ ] Привилегированный режим работает
|
||||
- [ ] Сеть host доступна
|
||||
- [ ] Логи контейнеров доступны
|
||||
- [ ] Автоперезапуск настроен
|
||||
|
||||
#==========================================================================
|
||||
# 🆘 Получение поддержки
|
||||
#==========================================================================
|
||||
|
||||
## Сбор диагностической информации
|
||||
```bash
|
||||
# Создать диагностический отчет
|
||||
sudo /opt/pyguardian/scripts/diagnostic-report.sh
|
||||
|
||||
# Отправить логи разработчикам
|
||||
# В Telegram боте: /debug export
|
||||
```
|
||||
|
||||
## Контакты для поддержки
|
||||
- 📧 Email: support@smartsoltech.com
|
||||
- 💬 Telegram: @PyGuardianSupport
|
||||
- 🐛 Issues: GitHub Issues
|
||||
- 📖 Wiki: GitHub Wiki
|
||||
|
||||
## Перед обращением в поддержку:
|
||||
1. Запустить тест установки: `./scripts/test-install.sh`
|
||||
2. Собрать диагностическую информацию
|
||||
3. Описать проблему и шаги для воспроизведения
|
||||
4. Приложить релевантные логи и конфигурации
|
||||
|
||||
---
|
||||
|
||||
**🎉 Поздравляем! PyGuardian готов к работе!**
|
||||
|
||||
Ваша система безопасности настроена и готова защищать серверы.
|
||||
Не забудьте настроить регулярные резервные копии и мониторинг обновлений.
|
||||
|
||||
*Happy securing! 🛡️*
|
||||
@@ -1,393 +0,0 @@
|
||||
# PyGuardian Quick Start Guide
|
||||
# Быстрое руководство по развертыванию и настройке
|
||||
|
||||
#==========================================================================
|
||||
# 🚀 Быстрый старт для автономного сервера
|
||||
#==========================================================================
|
||||
|
||||
## Шаг 1: Загрузка и подготовка
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Если все тесты пройдены, продолжить установку
|
||||
```
|
||||
|
||||
## Шаг 2: Быстрая установка
|
||||
```bash
|
||||
# Автоматическая установка в автономном режиме
|
||||
sudo ./install.sh
|
||||
|
||||
# Или через Makefile
|
||||
sudo make install
|
||||
```
|
||||
|
||||
## Шаг 3: Настройка Telegram бота
|
||||
```bash
|
||||
# Получить токен бота от @BotFather в Telegram
|
||||
# Заменить YOUR_BOT_TOKEN_HERE в конфигурации
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Получить свой Telegram ID (отправить /start боту @userinfobot)
|
||||
# Добавить в admin_users: [ВАШ_ID]
|
||||
```
|
||||
|
||||
## Шаг 4: Запуск системы
|
||||
```bash
|
||||
# Запустить службу
|
||||
sudo systemctl start pyguardian
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## Шаг 5: Тестирование
|
||||
```bash
|
||||
# Отправить /start вашему боту в Telegram
|
||||
# Если получили приветственное сообщение - система работает!
|
||||
|
||||
# Проверить статус через бота
|
||||
/status
|
||||
|
||||
# Просмотреть логи
|
||||
/logs system
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔗 Быстрый старт для кластера (контроллер + агенты)
|
||||
#==========================================================================
|
||||
|
||||
## Контроллер (центральный сервер)
|
||||
|
||||
### Шаг 1: Установка контроллера
|
||||
```bash
|
||||
# На главном сервере
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Установка в режиме контроллера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Или
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### Шаг 2: Настройка контроллера
|
||||
```bash
|
||||
# Настроить Telegram бота и кластерные параметры
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Обязательно настроить:
|
||||
# - telegram.bot_token
|
||||
# - telegram.admin_users
|
||||
# - cluster.api_secret
|
||||
# - cluster.deployment.ssh_key_path
|
||||
```
|
||||
|
||||
### Шаг 3: Генерация SSH ключей для кластера
|
||||
```bash
|
||||
# Создать SSH ключи для автоматического развертывания
|
||||
sudo ssh-keygen -t ed25519 -f /root/.ssh/cluster_key -N ""
|
||||
|
||||
# Скопировать публичный ключ на целевые серверы
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.50
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.51
|
||||
```
|
||||
|
||||
### Шаг 4: Запуск контроллера
|
||||
```bash
|
||||
sudo systemctl start pyguardian-controller
|
||||
sudo systemctl enable pyguardian-controller
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian-controller
|
||||
```
|
||||
|
||||
## Агенты (управляемые серверы)
|
||||
|
||||
### Автоматическое развертывание через Telegram
|
||||
```
|
||||
# Отправить боту команду для добавления сервера
|
||||
/cluster add
|
||||
|
||||
# Следовать интерактивным инструкциям бота:
|
||||
# 1. Ввести IP адрес сервера
|
||||
# 2. Указать SSH пользователя (обычно root)
|
||||
# 3. Выбрать аутентификацию по ключу
|
||||
# 4. Подтвердить развертывание
|
||||
|
||||
# Проверить статус кластера
|
||||
/cluster status
|
||||
```
|
||||
|
||||
### Ручное развертывание агента
|
||||
```bash
|
||||
# На каждом управляемом сервере
|
||||
wget https://your-server/install.sh
|
||||
chmod +x install.sh
|
||||
|
||||
# Установить агента
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
|
||||
# Или
|
||||
sudo make agent CONTROLLER_IP=192.168.1.10
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🐳 Быстрый старт с Docker
|
||||
#==========================================================================
|
||||
|
||||
## Автономный контейнер
|
||||
```bash
|
||||
# Создать образ
|
||||
docker build -t pyguardian .
|
||||
|
||||
# Запустить контейнер
|
||||
docker run -d \
|
||||
--name pyguardian \
|
||||
--privileged \
|
||||
--network host \
|
||||
-v $(pwd)/config:/opt/pyguardian/config \
|
||||
-v $(pwd)/data:/opt/pyguardian/data \
|
||||
pyguardian
|
||||
|
||||
# Проверить логи
|
||||
docker logs pyguardian
|
||||
```
|
||||
|
||||
## Docker Compose для кластера
|
||||
```bash
|
||||
# Настроить docker-compose.yml
|
||||
cp examples/configurations.md docker-compose.yml
|
||||
nano docker-compose.yml
|
||||
|
||||
# Запустить кластер
|
||||
docker-compose up -d
|
||||
|
||||
# Проверить статус
|
||||
docker-compose ps
|
||||
docker-compose logs pyguardian-controller
|
||||
```
|
||||
|
||||
## Использование готового Docker образа
|
||||
```bash
|
||||
# Скачать готовый образ
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или запустить автоматическую Docker установку
|
||||
sudo make docker-install
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# ⚙️ Основные команды после установки
|
||||
#==========================================================================
|
||||
|
||||
## Управление службой
|
||||
```bash
|
||||
# Статус службы
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Перезапуск службы
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Остановка службы
|
||||
sudo systemctl stop pyguardian
|
||||
```
|
||||
|
||||
## Управление конфигурацией
|
||||
```bash
|
||||
# Редактировать конфигурацию
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить конфигурацию
|
||||
/opt/pyguardian/venv/bin/python -c "import yaml; yaml.safe_load(open('/opt/pyguardian/config/config.yaml'))"
|
||||
|
||||
# Перезагрузить конфигурацию
|
||||
sudo systemctl reload pyguardian
|
||||
```
|
||||
|
||||
## Управление через Telegram
|
||||
```
|
||||
# Основные команды бота
|
||||
/start - Начать работу с ботом
|
||||
/help - Показать справку
|
||||
/status - Статус системы
|
||||
/sessions - Активные сессии
|
||||
/logs - Просмотр логов
|
||||
|
||||
# Кластерные команды (только для контроллера)
|
||||
/cluster status - Статус кластера
|
||||
/cluster add - Добавить сервер
|
||||
/cluster logs - Логи агентов
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔧 Устранение неполадок
|
||||
#==========================================================================
|
||||
|
||||
## Проблема: Telegram бот не отвечает
|
||||
```bash
|
||||
# Проверить токен бота
|
||||
grep bot_token /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить подключение к Telegram API
|
||||
curl -s "https://api.telegram.org/bot<YOUR_TOKEN>/getMe"
|
||||
|
||||
# Проверить логи службы
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
## Проблема: Агент не подключается к контроллеру
|
||||
```bash
|
||||
# На агенте проверить конфигурацию
|
||||
grep controller_host /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить сетевое подключение
|
||||
telnet CONTROLLER_IP 8443
|
||||
|
||||
# Проверить логи агента
|
||||
sudo journalctl -u pyguardian-agent | grep -i connection
|
||||
```
|
||||
|
||||
## Проблема: Высокое использование ресурсов
|
||||
```bash
|
||||
# Проверить процессы PyGuardian
|
||||
ps aux | grep python | grep pyguardian
|
||||
|
||||
# Проверить размер базы данных
|
||||
du -sh /opt/pyguardian/data/
|
||||
|
||||
# Оптимизировать базу данных
|
||||
sqlite3 /opt/pyguardian/data/pyguardian.db "VACUUM;"
|
||||
```
|
||||
|
||||
## Проблема: Ошибки файрвола
|
||||
```bash
|
||||
# Проверить правила iptables
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить логи файрвола
|
||||
sudo tail -f /var/log/kern.log | grep -i iptables
|
||||
|
||||
# Временно отключить файрвол PyGuardian
|
||||
sudo iptables -F PYGUARDIAN 2>/dev/null || true
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📚 Дополнительные ресурсы
|
||||
#==========================================================================
|
||||
|
||||
## Документация
|
||||
- `README.md` - Общее описание проекта
|
||||
- `docs/INSTALLATION.md` - Подробное руководство по установке
|
||||
- `docs/CLUSTER_SETUP.md` - Настройка кластера
|
||||
- `examples/configurations.md` - Примеры конфигураций
|
||||
- `examples/telegram-commands.md` - Команды Telegram бота
|
||||
|
||||
## Полезные команды
|
||||
```bash
|
||||
# Проверить версию PyGuardian
|
||||
/opt/pyguardian/venv/bin/python main.py --version
|
||||
|
||||
# Создать резервную копию
|
||||
sudo tar -czf pyguardian-backup-$(date +%Y%m%d).tar.gz \
|
||||
/opt/pyguardian/config \
|
||||
/opt/pyguardian/data
|
||||
|
||||
# Обновить систему
|
||||
cd /opt/pyguardian
|
||||
sudo git pull origin main
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Полная переустановка
|
||||
sudo ./install.sh --reinstall
|
||||
```
|
||||
|
||||
## Мониторинг и метрики
|
||||
```bash
|
||||
# Статистика файрвола
|
||||
sudo iptables -L -v -n
|
||||
|
||||
# Использование ресурсов
|
||||
htop
|
||||
df -h
|
||||
free -h
|
||||
|
||||
# Сетевые соединения
|
||||
sudo netstat -tulpn | grep python
|
||||
|
||||
# Логи в реальном времени
|
||||
sudo tail -f /opt/pyguardian/logs/pyguardian.log
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 Чек-лист после установки
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Проверить после установки автономного режима:
|
||||
- [ ] Служба PyGuardian запущена и активна
|
||||
- [ ] Telegram бот отвечает на команды
|
||||
- [ ] Конфигурация корректна и загружена
|
||||
- [ ] База данных создана и доступна
|
||||
- [ ] Файрвол настроен и работает
|
||||
- [ ] Мониторинг ресурсов активен
|
||||
- [ ] Логи пишутся корректно
|
||||
|
||||
## ✅ Проверить после установки кластера:
|
||||
- [ ] Контроллер запущен и доступен
|
||||
- [ ] API кластера отвечает на запросы
|
||||
- [ ] SSH ключи настроены для развертывания
|
||||
- [ ] Агенты подключены к контроллеру
|
||||
- [ ] Кластерные команды работают в Telegram
|
||||
- [ ] Синхронизация конфигураций работает
|
||||
- [ ] Мониторинг всех узлов активен
|
||||
|
||||
## ✅ Проверить после Docker установки:
|
||||
- [ ] Контейнеры запущены и работают
|
||||
- [ ] Volumes примонтированы корректно
|
||||
- [ ] Привилегированный режим работает
|
||||
- [ ] Сеть host доступна
|
||||
- [ ] Логи контейнеров доступны
|
||||
- [ ] Автоперезапуск настроен
|
||||
|
||||
#==========================================================================
|
||||
# 🆘 Получение поддержки
|
||||
#==========================================================================
|
||||
|
||||
## Сбор диагностической информации
|
||||
```bash
|
||||
# Создать диагностический отчет
|
||||
sudo /opt/pyguardian/scripts/diagnostic-report.sh
|
||||
|
||||
# Отправить логи разработчикам
|
||||
# В Telegram боте: /debug export
|
||||
```
|
||||
|
||||
## Контакты для поддержки
|
||||
- 📧 Email: support@smartsoltech.com
|
||||
- 💬 Telegram: @PyGuardianSupport
|
||||
- 🐛 Issues: GitHub Issues
|
||||
- 📖 Wiki: GitHub Wiki
|
||||
|
||||
## Перед обращением в поддержку:
|
||||
1. Запустить тест установки: `./scripts/test-install.sh`
|
||||
2. Собрать диагностическую информацию
|
||||
3. Описать проблему и шаги для воспроизведения
|
||||
4. Приложить релевантные логи и конфигурации
|
||||
|
||||
---
|
||||
|
||||
**🎉 Поздравляем! PyGuardian готов к работе!**
|
||||
|
||||
Ваша система безопасности настроена и готова защищать серверы.
|
||||
Не забудьте настроить регулярные резервные копии и мониторинг обновлений.
|
||||
|
||||
*Happy securing! 🛡️*
|
||||
@@ -1,393 +0,0 @@
|
||||
# PyGuardian Quick Start Guide
|
||||
# Быстрое руководство по развертыванию и настройке
|
||||
|
||||
#==========================================================================
|
||||
# 🚀 Быстрый старт для автономного сервера
|
||||
#==========================================================================
|
||||
|
||||
## Шаг 1: Загрузка и подготовка
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://git.smartsoltech.kr/trevor/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Если все тесты пройдены, продолжить установку
|
||||
```
|
||||
|
||||
## Шаг 2: Быстрая установка
|
||||
```bash
|
||||
# Автоматическая установка в автономном режиме
|
||||
sudo ./install.sh
|
||||
|
||||
# Или через Makefile
|
||||
sudo make install
|
||||
```
|
||||
|
||||
## Шаг 3: Настройка Telegram бота
|
||||
```bash
|
||||
# Получить токен бота от @BotFather в Telegram
|
||||
# Заменить YOUR_BOT_TOKEN_HERE в конфигурации
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Получить свой Telegram ID (отправить /start боту @userinfobot)
|
||||
# Добавить в admin_users: [ВАШ_ID]
|
||||
```
|
||||
|
||||
## Шаг 4: Запуск системы
|
||||
```bash
|
||||
# Запустить службу
|
||||
sudo systemctl start pyguardian
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## Шаг 5: Тестирование
|
||||
```bash
|
||||
# Отправить /start вашему боту в Telegram
|
||||
# Если получили приветственное сообщение - система работает!
|
||||
|
||||
# Проверить статус через бота
|
||||
/status
|
||||
|
||||
# Просмотреть логи
|
||||
/logs system
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔗 Быстрый старт для кластера (контроллер + агенты)
|
||||
#==========================================================================
|
||||
|
||||
## Контроллер (центральный сервер)
|
||||
|
||||
### Шаг 1: Установка контроллера
|
||||
```bash
|
||||
# На главном сервере
|
||||
git clone https://git.smartsoltech.kr/trevor/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Установка в режиме контроллера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Или
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### Шаг 2: Настройка контроллера
|
||||
```bash
|
||||
# Настроить Telegram бота и кластерные параметры
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Обязательно настроить:
|
||||
# - telegram.bot_token
|
||||
# - telegram.admin_users
|
||||
# - cluster.api_secret
|
||||
# - cluster.deployment.ssh_key_path
|
||||
```
|
||||
|
||||
### Шаг 3: Генерация SSH ключей для кластера
|
||||
```bash
|
||||
# Создать SSH ключи для автоматического развертывания
|
||||
sudo ssh-keygen -t ed25519 -f /root/.ssh/cluster_key -N ""
|
||||
|
||||
# Скопировать публичный ключ на целевые серверы
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.50
|
||||
sudo ssh-copy-id -i /root/.ssh/cluster_key.pub root@192.168.1.51
|
||||
```
|
||||
|
||||
### Шаг 4: Запуск контроллера
|
||||
```bash
|
||||
sudo systemctl start pyguardian-controller
|
||||
sudo systemctl enable pyguardian-controller
|
||||
|
||||
# Проверить статус
|
||||
sudo systemctl status pyguardian-controller
|
||||
```
|
||||
|
||||
## Агенты (управляемые серверы)
|
||||
|
||||
### Автоматическое развертывание через Telegram
|
||||
```
|
||||
# Отправить боту команду для добавления сервера
|
||||
/cluster add
|
||||
|
||||
# Следовать интерактивным инструкциям бота:
|
||||
# 1. Ввести IP адрес сервера
|
||||
# 2. Указать SSH пользователя (обычно root)
|
||||
# 3. Выбрать аутентификацию по ключу
|
||||
# 4. Подтвердить развертывание
|
||||
|
||||
# Проверить статус кластера
|
||||
/cluster status
|
||||
```
|
||||
|
||||
### Ручное развертывание агента
|
||||
```bash
|
||||
# На каждом управляемом сервере
|
||||
wget https://your-server/install.sh
|
||||
chmod +x install.sh
|
||||
|
||||
# Установить агента
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
|
||||
# Или
|
||||
sudo make agent CONTROLLER_IP=192.168.1.10
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🐳 Быстрый старт с Docker
|
||||
#==========================================================================
|
||||
|
||||
## Автономный контейнер
|
||||
```bash
|
||||
# Создать образ
|
||||
docker build -t pyguardian .
|
||||
|
||||
# Запустить контейнер
|
||||
docker run -d \
|
||||
--name pyguardian \
|
||||
--privileged \
|
||||
--network host \
|
||||
-v $(pwd)/config:/opt/pyguardian/config \
|
||||
-v $(pwd)/data:/opt/pyguardian/data \
|
||||
pyguardian
|
||||
|
||||
# Проверить логи
|
||||
docker logs pyguardian
|
||||
```
|
||||
|
||||
## Docker Compose для кластера
|
||||
```bash
|
||||
# Настроить docker-compose.yml
|
||||
cp examples/configurations.md docker-compose.yml
|
||||
nano docker-compose.yml
|
||||
|
||||
# Запустить кластер
|
||||
docker-compose up -d
|
||||
|
||||
# Проверить статус
|
||||
docker-compose ps
|
||||
docker-compose logs pyguardian-controller
|
||||
```
|
||||
|
||||
## Использование готового Docker образа
|
||||
```bash
|
||||
# Скачать готовый образ
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или запустить автоматическую Docker установку
|
||||
sudo make docker-install
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# ⚙️ Основные команды после установки
|
||||
#==========================================================================
|
||||
|
||||
## Управление службой
|
||||
```bash
|
||||
# Статус службы
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Перезапуск службы
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Остановка службы
|
||||
sudo systemctl stop pyguardian
|
||||
```
|
||||
|
||||
## Управление конфигурацией
|
||||
```bash
|
||||
# Редактировать конфигурацию
|
||||
sudo nano /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить конфигурацию
|
||||
/opt/pyguardian/venv/bin/python -c "import yaml; yaml.safe_load(open('/opt/pyguardian/config/config.yaml'))"
|
||||
|
||||
# Перезагрузить конфигурацию
|
||||
sudo systemctl reload pyguardian
|
||||
```
|
||||
|
||||
## Управление через Telegram
|
||||
```
|
||||
# Основные команды бота
|
||||
/start - Начать работу с ботом
|
||||
/help - Показать справку
|
||||
/status - Статус системы
|
||||
/sessions - Активные сессии
|
||||
/logs - Просмотр логов
|
||||
|
||||
# Кластерные команды (только для контроллера)
|
||||
/cluster status - Статус кластера
|
||||
/cluster add - Добавить сервер
|
||||
/cluster logs - Логи агентов
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🔧 Устранение неполадок
|
||||
#==========================================================================
|
||||
|
||||
## Проблема: Telegram бот не отвечает
|
||||
```bash
|
||||
# Проверить токен бота
|
||||
grep bot_token /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить подключение к Telegram API
|
||||
curl -s "https://api.telegram.org/bot<YOUR_TOKEN>/getMe"
|
||||
|
||||
# Проверить логи службы
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
## Проблема: Агент не подключается к контроллеру
|
||||
```bash
|
||||
# На агенте проверить конфигурацию
|
||||
grep controller_host /opt/pyguardian/config/config.yaml
|
||||
|
||||
# Проверить сетевое подключение
|
||||
telnet CONTROLLER_IP 8443
|
||||
|
||||
# Проверить логи агента
|
||||
sudo journalctl -u pyguardian-agent | grep -i connection
|
||||
```
|
||||
|
||||
## Проблема: Высокое использование ресурсов
|
||||
```bash
|
||||
# Проверить процессы PyGuardian
|
||||
ps aux | grep python | grep pyguardian
|
||||
|
||||
# Проверить размер базы данных
|
||||
du -sh /opt/pyguardian/data/
|
||||
|
||||
# Оптимизировать базу данных
|
||||
sqlite3 /opt/pyguardian/data/pyguardian.db "VACUUM;"
|
||||
```
|
||||
|
||||
## Проблема: Ошибки файрвола
|
||||
```bash
|
||||
# Проверить правила iptables
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить логи файрвола
|
||||
sudo tail -f /var/log/kern.log | grep -i iptables
|
||||
|
||||
# Временно отключить файрвол PyGuardian
|
||||
sudo iptables -F PYGUARDIAN 2>/dev/null || true
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 📚 Дополнительные ресурсы
|
||||
#==========================================================================
|
||||
|
||||
## Документация
|
||||
- `README.md` - Общее описание проекта
|
||||
- `docs/INSTALLATION.md` - Подробное руководство по установке
|
||||
- `docs/CLUSTER_SETUP.md` - Настройка кластера
|
||||
- `examples/configurations.md` - Примеры конфигураций
|
||||
- `examples/telegram-commands.md` - Команды Telegram бота
|
||||
|
||||
## Полезные команды
|
||||
```bash
|
||||
# Проверить версию PyGuardian
|
||||
/opt/pyguardian/venv/bin/python main.py --version
|
||||
|
||||
# Создать резервную копию
|
||||
sudo tar -czf pyguardian-backup-$(date +%Y%m%d).tar.gz \
|
||||
/opt/pyguardian/config \
|
||||
/opt/pyguardian/data
|
||||
|
||||
# Обновить систему
|
||||
cd /opt/pyguardian
|
||||
sudo git pull origin main
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Полная переустановка
|
||||
sudo ./install.sh --reinstall
|
||||
```
|
||||
|
||||
## Мониторинг и метрики
|
||||
```bash
|
||||
# Статистика файрвола
|
||||
sudo iptables -L -v -n
|
||||
|
||||
# Использование ресурсов
|
||||
htop
|
||||
df -h
|
||||
free -h
|
||||
|
||||
# Сетевые соединения
|
||||
sudo netstat -tulpn | grep python
|
||||
|
||||
# Логи в реальном времени
|
||||
sudo tail -f /opt/pyguardian/logs/pyguardian.log
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 🎯 Чек-лист после установки
|
||||
#==========================================================================
|
||||
|
||||
## ✅ Проверить после установки автономного режима:
|
||||
- [ ] Служба PyGuardian запущена и активна
|
||||
- [ ] Telegram бот отвечает на команды
|
||||
- [ ] Конфигурация корректна и загружена
|
||||
- [ ] База данных создана и доступна
|
||||
- [ ] Файрвол настроен и работает
|
||||
- [ ] Мониторинг ресурсов активен
|
||||
- [ ] Логи пишутся корректно
|
||||
|
||||
## ✅ Проверить после установки кластера:
|
||||
- [ ] Контроллер запущен и доступен
|
||||
- [ ] API кластера отвечает на запросы
|
||||
- [ ] SSH ключи настроены для развертывания
|
||||
- [ ] Агенты подключены к контроллеру
|
||||
- [ ] Кластерные команды работают в Telegram
|
||||
- [ ] Синхронизация конфигураций работает
|
||||
- [ ] Мониторинг всех узлов активен
|
||||
|
||||
## ✅ Проверить после Docker установки:
|
||||
- [ ] Контейнеры запущены и работают
|
||||
- [ ] Volumes примонтированы корректно
|
||||
- [ ] Привилегированный режим работает
|
||||
- [ ] Сеть host доступна
|
||||
- [ ] Логи контейнеров доступны
|
||||
- [ ] Автоперезапуск настроен
|
||||
|
||||
#==========================================================================
|
||||
# 🆘 Получение поддержки
|
||||
#==========================================================================
|
||||
|
||||
## Сбор диагностической информации
|
||||
```bash
|
||||
# Создать диагностический отчет
|
||||
sudo /opt/pyguardian/scripts/diagnostic-report.sh
|
||||
|
||||
# Отправить логи разработчикам
|
||||
# В Telegram боте: /debug export
|
||||
```
|
||||
|
||||
## Контакты для поддержки
|
||||
- 📧 Email: support@smartsoltech.com
|
||||
- 💬 Telegram: @PyGuardianSupport
|
||||
- 🐛 Issues: GitHub Issues
|
||||
- 📖 Wiki: GitHub Wiki
|
||||
|
||||
## Перед обращением в поддержку:
|
||||
1. Запустить тест установки: `./scripts/test-install.sh`
|
||||
2. Собрать диагностическую информацию
|
||||
3. Описать проблему и шаги для воспроизведения
|
||||
4. Приложить релевантные логи и конфигурации
|
||||
|
||||
---
|
||||
|
||||
**🎉 Поздравляем! PyGuardian готов к работе!**
|
||||
|
||||
Ваша система безопасности настроена и готова защищать серверы.
|
||||
Не забудьте настроить регулярные резервные копии и мониторинг обновлений.
|
||||
|
||||
*Happy securing! 🛡️*
|
||||
@@ -1,458 +0,0 @@
|
||||
# PyGuardian - Linux Server Protection System
|
||||
|
||||
🛡️ **Полноценная система мониторинга и защиты Linux-сервера от брутфорс-атак с управлением через Telegram-бота**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🎯 Возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,475 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,491 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,491 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,494 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,504 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,512 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,524 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,530 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
### Кластерное управление
|
||||
- **Centralized Control**: Управление множественными серверами из единого центра
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Distributed Monitoring**: Мониторинг безопасности всех серверов кластера
|
||||
- **Unified Dashboard**: Единый Telegram-интерфейс для всего кластера
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [ ] Кластерный режим
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,530 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
### Кластерное управление
|
||||
- **Centralized Control**: Управление множественными серверами из единого центра
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Distributed Monitoring**: Мониторинг безопасности всех серверов кластера
|
||||
- **Unified Dashboard**: Единый Telegram-интерфейс для всего кластера
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,577 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
### Кластерное управление
|
||||
- **Centralized Control**: Управление множественными серверами из единого центра
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Distributed Monitoring**: Мониторинг безопасности всех серверов кластера
|
||||
- **Unified Dashboard**: Единый Telegram-интерфейс для всего кластера
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,577 +0,0 @@
|
||||
# PyGuardian - Advanced Linux Server Protection System
|
||||
|
||||
🛡️ **Система мониторинга и защиты Linux-сервера от брутфорс-атак с функциями СКРЫТОГО обнаружения взломов**
|
||||
|
||||
[](https://python.org)
|
||||
[](LICENSE)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
### Кластерное управление
|
||||
- **Centralized Control**: Управление множественными серверами из единого центра
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Distributed Monitoring**: Мониторинг безопасности всех серверов кластера
|
||||
- **Unified Dashboard**: Единый Telegram-интерфейс для всего кластера
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,578 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## 🚨 НОВЫЕ ВОЗМОЖНОСТИ - STEALTH SECURITY
|
||||
|
||||
### Скрытое обнаружение компромиссов
|
||||
- **Invisible Detection**: Система работает **незаметно** для атакующего
|
||||
- **Honeypot Users**: Автоматическое обнаружение входов под приманочными учетками
|
||||
- **Behavioral Analysis**: Анализ подозрительного поведения после входа
|
||||
- **Silent Response**: Тихая блокировка без уведомления атакующего
|
||||
|
||||
### Автоматические контрмеры
|
||||
- **Password Rotation**: Мгновенная смена паролей скомпрометированных аккаунтов
|
||||
- **Session Termination**: Принудительное завершение подозрительных сессий
|
||||
- **Stealth Blocking**: Скрытая блокировка атакующих IP
|
||||
- **Evidence Collection**: Сбор доказательств для анализа
|
||||
|
||||
### Кластерное управление
|
||||
- **Centralized Control**: Управление множественными серверами из единого центра
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Distributed Monitoring**: Мониторинг безопасности всех серверов кластера
|
||||
- **Unified Dashboard**: Единый Telegram-интерфейс для всего кластера
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,595 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
- **Мониторинг в реальном времени**: Отслеживание auth.log и детекция атак
|
||||
- **Автоматическая защита**: Блокировка IP при превышении лимита попыток
|
||||
- **Telegram управление**: Полный контроль через бота без веб-интерфейса
|
||||
- **🔥 STEALTH MODE**: Скрытое обнаружение и нейтрализация взломов
|
||||
- **Session Management**: Мониторинг и управление SSH сессиями
|
||||
- **Password Security**: Автоматическая смена паролей при компромиссе
|
||||
- **🌐 CLUSTER MANAGEMENT**: Централизованное управление множественными серверами
|
||||
- **Agent Deployment**: Автоматическое развертывание агентов по SSH
|
||||
- **Multi-Server Control**: Единый Telegram-интерфейс для всех серверов
|
||||
- **Поддержка firewall**: iptables и nftables
|
||||
- **Автоматический разбан**: По таймеру с уведомлениями
|
||||
- **Белый список**: Защита доверенных IP от блокировки
|
||||
- **Статистика**: Подробная аналитика атак и успешных входов
|
||||
- **Асинхронность**: Оптимизированная производительность
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,609 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
### 🔒 Продвинутая система безопасности
|
||||
- **🚨 Обнаружение вторжений** - Real-time детекция атак
|
||||
- **🛡️ Интеллектуальный файрвол** - Автоматическая блокировка угроз
|
||||
- **🔍 Мониторинг системы** - Контроль ресурсов и процессов
|
||||
- **⚠️ DDoS защита** - Автоматическое обнаружение и блокировка
|
||||
- **🔐 2FA интеграция** - TOTP аутентификация через Telegram
|
||||
- **👤 Stealth Mode** - Скрытое обнаружение компрометации
|
||||
|
||||
### 🌐 Кластерное управление
|
||||
- **🎛️ Централизованный контроль** - Управление множеством серверов
|
||||
- **🚀 Автоматическое развертывание** - Развертывание агентов одной командой
|
||||
- **📡 Синхронизация конфигураций** - Единые политики безопасности
|
||||
- **📊 Мониторинг кластера** - Состояние всех узлов в реальном времени
|
||||
- **🔄 Load Balancing** - Автоматическое распределение нагрузки
|
||||
|
||||
### 💬 Продвинутый Telegram интерфейс
|
||||
- **🤖 Интерактивные команды** - Удобное управление через диалоги
|
||||
- **📈 Real-time мониторинг** - Мгновенные уведомления и алерты
|
||||
- **🔧 Удаленное управление** - Полный контроль через мессенджер
|
||||
- **👥 Многопользовательский доступ** - Ролевая модель доступа
|
||||
- **🗣️ Поддержка голосовых команд** - Управление голосом
|
||||
|
||||
### 🐳 Современные технологии развертывания
|
||||
- **📦 Docker поддержка** - Контейнеризированное развертывание
|
||||
- **⚙️ systemd интеграция** - Нативная интеграция с системой
|
||||
- **🔧 Ansible ready** - Готовые playbooks для автоматизации
|
||||
- **☁️ Cloud готовность** - Поддержка AWS, GCP, Azure
|
||||
- **📊 Метрики и логирование** - Интеграция с Grafana/Prometheus
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,609 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
### 🔒 Продвинутая система безопасности
|
||||
- **🚨 Обнаружение вторжений** - Real-time детекция атак
|
||||
- **🛡️ Интеллектуальный файрвол** - Автоматическая блокировка угроз
|
||||
- **🔍 Мониторинг системы** - Контроль ресурсов и процессов
|
||||
- **⚠️ DDoS защита** - Автоматическое обнаружение и блокировка
|
||||
- **🔐 2FA интеграция** - TOTP аутентификация через Telegram
|
||||
- **👤 Stealth Mode** - Скрытое обнаружение компрометации
|
||||
|
||||
### 🌐 Кластерное управление
|
||||
- **🎛️ Централизованный контроль** - Управление множеством серверов
|
||||
- **🚀 Автоматическое развертывание** - Развертывание агентов одной командой
|
||||
- **📡 Синхронизация конфигураций** - Единые политики безопасности
|
||||
- **📊 Мониторинг кластера** - Состояние всех узлов в реальном времени
|
||||
- **🔄 Load Balancing** - Автоматическое распределение нагрузки
|
||||
|
||||
### 💬 Продвинутый Telegram интерфейс
|
||||
- **🤖 Интерактивные команды** - Удобное управление через диалоги
|
||||
- **📈 Real-time мониторинг** - Мгновенные уведомления и алерты
|
||||
- **🔧 Удаленное управление** - Полный контроль через мессенджер
|
||||
- **👥 Многопользовательский доступ** - Ролевая модель доступа
|
||||
- **🗣️ Поддержка голосовых команд** - Управление голосом
|
||||
|
||||
### 🐳 Современные технологии развертывания
|
||||
- **📦 Docker поддержка** - Контейнеризированное развертывание
|
||||
- **⚙️ systemd интеграция** - Нативная интеграция с системой
|
||||
- **🔧 Ansible ready** - Готовые playbooks для автоматизации
|
||||
- **☁️ Cloud готовность** - Поддержка AWS, GCP, Azure
|
||||
- **📊 Метрики и логирование** - Интеграция с Grafana/Prometheus
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **Права**: root (для работы с firewall)
|
||||
- **Firewall**: iptables или nftables
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,822 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
### 🔒 Продвинутая система безопасности
|
||||
- **🚨 Обнаружение вторжений** - Real-time детекция атак
|
||||
- **🛡️ Интеллектуальный файрвол** - Автоматическая блокировка угроз
|
||||
- **🔍 Мониторинг системы** - Контроль ресурсов и процессов
|
||||
- **⚠️ DDoS защита** - Автоматическое обнаружение и блокировка
|
||||
- **🔐 2FA интеграция** - TOTP аутентификация через Telegram
|
||||
- **👤 Stealth Mode** - Скрытое обнаружение компрометации
|
||||
|
||||
### 🌐 Кластерное управление
|
||||
- **🎛️ Централизованный контроль** - Управление множеством серверов
|
||||
- **🚀 Автоматическое развертывание** - Развертывание агентов одной командой
|
||||
- **📡 Синхронизация конфигураций** - Единые политики безопасности
|
||||
- **📊 Мониторинг кластера** - Состояние всех узлов в реальном времени
|
||||
- **🔄 Load Balancing** - Автоматическое распределение нагрузки
|
||||
|
||||
### 💬 Продвинутый Telegram интерфейс
|
||||
- **🤖 Интерактивные команды** - Удобное управление через диалоги
|
||||
- **📈 Real-time мониторинг** - Мгновенные уведомления и алерты
|
||||
- **🔧 Удаленное управление** - Полный контроль через мессенджер
|
||||
- **👥 Многопользовательский доступ** - Ролевая модель доступа
|
||||
- **🗣️ Поддержка голосовых команд** - Управление голосом
|
||||
|
||||
### 🐳 Современные технологии развертывания
|
||||
- **📦 Docker поддержка** - Контейнеризированное развертывание
|
||||
- **⚙️ systemd интеграция** - Нативная интеграция с системой
|
||||
- **🔧 Ansible ready** - Готовые playbooks для автоматизации
|
||||
- **☁️ Cloud готовность** - Поддержка AWS, GCP, Azure
|
||||
- **📊 Метрики и логирование** - Интеграция с Grafana/Prometheus
|
||||
|
||||
## 🏗️ Архитектура системы
|
||||
|
||||
### Режимы развертывания:
|
||||
|
||||
#### 🖥️ Standalone (Автономный)
|
||||
Все компоненты на одном сервере
|
||||
```
|
||||
┌─────────────────┐
|
||||
│ PyGuardian │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Telegram │ │
|
||||
│ │ Bot │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Firewall │ │
|
||||
│ │ Manager │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
#### 🌐 Controller + Agents (Кластерный)
|
||||
Центральный контроллер управляет агентами
|
||||
```
|
||||
┌─────────────────┐ ┌─────────────────┐
|
||||
│ Controller │────│ Agent 1 │
|
||||
│ ┌─────────────┐ │ │ ┌─────────────┐ │
|
||||
│ │ Telegram │ │ │ │ Security │ │
|
||||
│ │ Bot │ │ │ │ Monitor │ │
|
||||
│ └─────────────┘ │ │ └─────────────┘ │
|
||||
│ ┌─────────────┐ │ └─────────────────┘
|
||||
│ │ Cluster │ │ │
|
||||
│ │ Manager │ │ ┌─────────────────┐
|
||||
│ └─────────────┘ │────│ Agent 2 │
|
||||
└─────────────────┘ │ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
## 🛠️ Технологический стек
|
||||
|
||||
- **🐍 Python 3.10+** - Основной язык разработки
|
||||
- **🤖 Telegram Bot API** - Интерфейс управления
|
||||
- **🗄️ SQLite/PostgreSQL** - База данных
|
||||
- **🔥 iptables/nftables** - Управление файрволом
|
||||
- **🐳 Docker** - Контейнеризация
|
||||
- **⚙️ systemd** - Управление службами
|
||||
- **🔒 cryptography** - Шифрование данных
|
||||
- **📡 asyncio** - Асинхронное выполнение
|
||||
- **📊 psutil** - Мониторинг системы
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
### Минимальные требования:
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **RAM**: 512MB (автономный), 1GB (контроллер)
|
||||
- **Диск**: 1GB свободного места
|
||||
- **Сеть**: Доступ к Telegram API
|
||||
|
||||
### Рекомендуемые требования:
|
||||
- **ОС**: Ubuntu 22.04 LTS / Debian 12
|
||||
- **Python**: 3.11+
|
||||
- **RAM**: 2GB+ для контроллера кластера
|
||||
- **Диск**: 10GB+ для логов и резервных копий
|
||||
- **Сеть**: Выделенный IP для кластера
|
||||
|
||||
## 📚 Документация
|
||||
|
||||
- **[📖 QUICKSTART.md](QUICKSTART.md)** - Быстрое руководство по установке
|
||||
- **[⚙️ INSTALLATION.md](docs/INSTALLATION.md)** - Подробная установка и настройка
|
||||
- **[🌐 CLUSTER_SETUP.md](docs/CLUSTER_SETUP.md)** - Настройка кластера
|
||||
- **[⚡ configurations.md](examples/configurations.md)** - Примеры конфигураций
|
||||
- **[🤖 telegram-commands.md](examples/telegram-commands.md)** - Команды бота
|
||||
|
||||
## 🔧 Установка и использование
|
||||
|
||||
### 1️⃣ Автоматическая установка
|
||||
```bash
|
||||
# Автономная установка
|
||||
sudo ./install.sh
|
||||
|
||||
# Контроллер кластера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Агент кластера
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
```
|
||||
|
||||
### 2️⃣ Make файл
|
||||
```bash
|
||||
# Показать все доступные команды
|
||||
make help
|
||||
|
||||
# Установка различных режимов
|
||||
make install # Автономный режим
|
||||
make controller # Кластерный контроллер
|
||||
make agent # Кластерный агент
|
||||
```
|
||||
|
||||
### 3️⃣ Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Ручная сборка образа
|
||||
docker build -t pyguardian .
|
||||
docker run -d --privileged --network host pyguardian
|
||||
```
|
||||
|
||||
## 🎯 Примеры использования
|
||||
|
||||
### Управление через Telegram бота:
|
||||
```
|
||||
/start - Начать работу с ботом
|
||||
/status - Статус системы безопасности
|
||||
/block IP - Заблокировать IP адрес
|
||||
/unblock IP - Разблокировать IP адрес
|
||||
/sessions - Показать активные сессии
|
||||
/logs - Просмотр логов
|
||||
/cluster status - Статус кластера (контроллер)
|
||||
/cluster add - Добавить сервер в кластер
|
||||
```
|
||||
|
||||
### Кластерное управление:
|
||||
```bash
|
||||
# Добавление сервера в кластер через SSH
|
||||
ssh-copy-id -i ~/.ssh/cluster_key.pub root@192.168.1.50
|
||||
|
||||
# В Telegram боте контроллера:
|
||||
/cluster add
|
||||
# Следовать интерактивным инструкциям
|
||||
```
|
||||
|
||||
## 🔒 Безопасность
|
||||
|
||||
- **🔑 Аутентификация**: Telegram user ID + опциональная 2FA
|
||||
- **🔐 Шифрование**: Все конфиденциальные данные зашифрованы
|
||||
- **🛡️ Изоляция**: Контейнеризация и изолированные процессы
|
||||
- **📝 Аудит**: Полное логирование всех действий
|
||||
- **🚫 Принцип минимальных привилегий**: Только необходимые права
|
||||
|
||||
## 📊 Мониторинг и алерты
|
||||
|
||||
### Автоматические уведомления о:
|
||||
- 🚨 Попытках взлома и атаках
|
||||
- 📈 Превышении лимитов ресурсов
|
||||
- 🔌 Подключении/отключении агентов кластера
|
||||
- ⚠️ Ошибках в системе безопасности
|
||||
- 📋 Результатах резервного копирования
|
||||
|
||||
### Интеграция с системами мониторинга:
|
||||
- **Grafana/Prometheus** - Метрики и дашборды
|
||||
- **ELK Stack** - Централизованное логирование
|
||||
- **SIEM системы** - Экспорт событий безопасности
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
Мы приветствуем вклад в развитие PyGuardian!
|
||||
|
||||
### Как принять участие:
|
||||
1. **Fork** репозитория
|
||||
2. Создайте **feature branch** (`git checkout -b feature/amazing-feature`)
|
||||
3. **Commit** изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. **Push** в branch (`git push origin feature/amazing-feature`)
|
||||
5. Создайте **Pull Request**
|
||||
|
||||
### Области для улучшения:
|
||||
- 🌐 Веб-интерфейс управления
|
||||
- 📱 Мобильное приложение
|
||||
- 🔌 Интеграции с облачными провайдерами
|
||||
- 🤖 ИИ для детекции аномалий
|
||||
- 📊 Расширенная аналитика
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Этот проект распространяется под лицензией MIT. См. файл [LICENSE](LICENSE) для подробностей.
|
||||
|
||||
## 👥 Авторы
|
||||
|
||||
- **SmartSolTech Team** - *Первоначальная разработка* - [@SmartSolTech](https://github.com/SmartSolTech)
|
||||
|
||||
## 🆘 Поддержка
|
||||
|
||||
### Получить помощь:
|
||||
- 📧 **Email**: support@smartsoltech.com
|
||||
- 💬 **Telegram**: @PyGuardianSupport
|
||||
- 🐛 **Issues**: [GitHub Issues](https://github.com/SmartSolTech/PyGuardian/issues)
|
||||
- 📖 **Wiki**: [GitHub Wiki](https://github.com/SmartSolTech/PyGuardian/wiki)
|
||||
|
||||
### Перед обращением:
|
||||
1. Проверьте [FAQ](https://github.com/SmartSolTech/PyGuardian/wiki/FAQ)
|
||||
2. Запустите диагностику: `./scripts/test-install.sh`
|
||||
3. Соберите логи: `/debug export` в Telegram боте
|
||||
|
||||
---
|
||||
|
||||
**⚡ Быстрые команды:**
|
||||
|
||||
```bash
|
||||
# Проверить статус
|
||||
systemctl status pyguardian
|
||||
|
||||
# Просмотреть логи
|
||||
journalctl -u pyguardian -f
|
||||
|
||||
# Перезапустить
|
||||
systemctl restart pyguardian
|
||||
|
||||
# Обновить конфигурацию
|
||||
systemctl reload pyguardian
|
||||
```
|
||||
|
||||
**🎉 Спасибо за использование PyGuardian! Ваша безопасность - наш приоритет. 🛡️**
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,822 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
### 🔒 Продвинутая система безопасности
|
||||
- **🚨 Обнаружение вторжений** - Real-time детекция атак
|
||||
- **🛡️ Интеллектуальный файрвол** - Автоматическая блокировка угроз
|
||||
- **🔍 Мониторинг системы** - Контроль ресурсов и процессов
|
||||
- **⚠️ DDoS защита** - Автоматическое обнаружение и блокировка
|
||||
- **🔐 2FA интеграция** - TOTP аутентификация через Telegram
|
||||
- **👤 Stealth Mode** - Скрытое обнаружение компрометации
|
||||
|
||||
### 🌐 Кластерное управление
|
||||
- **🎛️ Централизованный контроль** - Управление множеством серверов
|
||||
- **🚀 Автоматическое развертывание** - Развертывание агентов одной командой
|
||||
- **📡 Синхронизация конфигураций** - Единые политики безопасности
|
||||
- **📊 Мониторинг кластера** - Состояние всех узлов в реальном времени
|
||||
- **🔄 Load Balancing** - Автоматическое распределение нагрузки
|
||||
|
||||
### 💬 Продвинутый Telegram интерфейс
|
||||
- **🤖 Интерактивные команды** - Удобное управление через диалоги
|
||||
- **📈 Real-time мониторинг** - Мгновенные уведомления и алерты
|
||||
- **🔧 Удаленное управление** - Полный контроль через мессенджер
|
||||
- **👥 Многопользовательский доступ** - Ролевая модель доступа
|
||||
- **🗣️ Поддержка голосовых команд** - Управление голосом
|
||||
|
||||
### 🐳 Современные технологии развертывания
|
||||
- **📦 Docker поддержка** - Контейнеризированное развертывание
|
||||
- **⚙️ systemd интеграция** - Нативная интеграция с системой
|
||||
- **🔧 Ansible ready** - Готовые playbooks для автоматизации
|
||||
- **☁️ Cloud готовность** - Поддержка AWS, GCP, Azure
|
||||
- **📊 Метрики и логирование** - Интеграция с Grafana/Prometheus
|
||||
|
||||
## 🏗️ Архитектура системы
|
||||
|
||||
### Режимы развертывания:
|
||||
|
||||
#### 🖥️ Standalone (Автономный)
|
||||
Все компоненты на одном сервере
|
||||
```
|
||||
┌─────────────────┐
|
||||
│ PyGuardian │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Telegram │ │
|
||||
│ │ Bot │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Firewall │ │
|
||||
│ │ Manager │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
#### 🌐 Controller + Agents (Кластерный)
|
||||
Центральный контроллер управляет агентами
|
||||
```
|
||||
┌─────────────────┐ ┌─────────────────┐
|
||||
│ Controller │────│ Agent 1 │
|
||||
│ ┌─────────────┐ │ │ ┌─────────────┐ │
|
||||
│ │ Telegram │ │ │ │ Security │ │
|
||||
│ │ Bot │ │ │ │ Monitor │ │
|
||||
│ └─────────────┘ │ │ └─────────────┘ │
|
||||
│ ┌─────────────┐ │ └─────────────────┘
|
||||
│ │ Cluster │ │ │
|
||||
│ │ Manager │ │ ┌─────────────────┐
|
||||
│ └─────────────┘ │────│ Agent 2 │
|
||||
└─────────────────┘ │ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
## 🛠️ Технологический стек
|
||||
|
||||
- **🐍 Python 3.10+** - Основной язык разработки
|
||||
- **🤖 Telegram Bot API** - Интерфейс управления
|
||||
- **🗄️ SQLite/PostgreSQL** - База данных
|
||||
- **🔥 iptables/nftables** - Управление файрволом
|
||||
- **🐳 Docker** - Контейнеризация
|
||||
- **⚙️ systemd** - Управление службами
|
||||
- **🔒 cryptography** - Шифрование данных
|
||||
- **📡 asyncio** - Асинхронное выполнение
|
||||
- **📊 psutil** - Мониторинг системы
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
### Минимальные требования:
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **RAM**: 512MB (автономный), 1GB (контроллер)
|
||||
- **Диск**: 1GB свободного места
|
||||
- **Сеть**: Доступ к Telegram API
|
||||
|
||||
### Рекомендуемые требования:
|
||||
- **ОС**: Ubuntu 22.04 LTS / Debian 12
|
||||
- **Python**: 3.11+
|
||||
- **RAM**: 2GB+ для контроллера кластера
|
||||
- **Диск**: 10GB+ для логов и резервных копий
|
||||
- **Сеть**: Выделенный IP для кластера
|
||||
|
||||
## 📚 Документация
|
||||
|
||||
- **[📖 QUICKSTART.md](QUICKSTART.md)** - Быстрое руководство по установке
|
||||
- **[⚙️ INSTALLATION.md](docs/INSTALLATION.md)** - Подробная установка и настройка
|
||||
- **[🌐 CLUSTER_SETUP.md](docs/CLUSTER_SETUP.md)** - Настройка кластера
|
||||
- **[⚡ configurations.md](examples/configurations.md)** - Примеры конфигураций
|
||||
- **[🤖 telegram-commands.md](examples/telegram-commands.md)** - Команды бота
|
||||
|
||||
## 🔧 Установка и использование
|
||||
|
||||
### 1️⃣ Автоматическая установка
|
||||
```bash
|
||||
# Автономная установка
|
||||
sudo ./install.sh
|
||||
|
||||
# Контроллер кластера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Агент кластера
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
```
|
||||
|
||||
### 2️⃣ Make файл
|
||||
```bash
|
||||
# Показать все доступные команды
|
||||
make help
|
||||
|
||||
# Установка различных режимов
|
||||
make install # Автономный режим
|
||||
make controller # Кластерный контроллер
|
||||
make agent # Кластерный агент
|
||||
```
|
||||
|
||||
### 3️⃣ Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Ручная сборка образа
|
||||
docker build -t pyguardian .
|
||||
docker run -d --privileged --network host pyguardian
|
||||
```
|
||||
|
||||
## 🎯 Примеры использования
|
||||
|
||||
### Управление через Telegram бота:
|
||||
```
|
||||
/start - Начать работу с ботом
|
||||
/status - Статус системы безопасности
|
||||
/block IP - Заблокировать IP адрес
|
||||
/unblock IP - Разблокировать IP адрес
|
||||
/sessions - Показать активные сессии
|
||||
/logs - Просмотр логов
|
||||
/cluster status - Статус кластера (контроллер)
|
||||
/cluster add - Добавить сервер в кластер
|
||||
```
|
||||
|
||||
### Кластерное управление:
|
||||
```bash
|
||||
# Добавление сервера в кластер через SSH
|
||||
ssh-copy-id -i ~/.ssh/cluster_key.pub root@192.168.1.50
|
||||
|
||||
# В Telegram боте контроллера:
|
||||
/cluster add
|
||||
# Следовать интерактивным инструкциям
|
||||
```
|
||||
|
||||
## 🔒 Безопасность
|
||||
|
||||
- **🔑 Аутентификация**: Telegram user ID + опциональная 2FA
|
||||
- **🔐 Шифрование**: Все конфиденциальные данные зашифрованы
|
||||
- **🛡️ Изоляция**: Контейнеризация и изолированные процессы
|
||||
- **📝 Аудит**: Полное логирование всех действий
|
||||
- **🚫 Принцип минимальных привилегий**: Только необходимые права
|
||||
|
||||
## 📊 Мониторинг и алерты
|
||||
|
||||
### Автоматические уведомления о:
|
||||
- 🚨 Попытках взлома и атаках
|
||||
- 📈 Превышении лимитов ресурсов
|
||||
- 🔌 Подключении/отключении агентов кластера
|
||||
- ⚠️ Ошибках в системе безопасности
|
||||
- 📋 Результатах резервного копирования
|
||||
|
||||
### Интеграция с системами мониторинга:
|
||||
- **Grafana/Prometheus** - Метрики и дашборды
|
||||
- **ELK Stack** - Централизованное логирование
|
||||
- **SIEM системы** - Экспорт событий безопасности
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
Мы приветствуем вклад в развитие PyGuardian!
|
||||
|
||||
### Как принять участие:
|
||||
1. **Fork** репозитория
|
||||
2. Создайте **feature branch** (`git checkout -b feature/amazing-feature`)
|
||||
3. **Commit** изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. **Push** в branch (`git push origin feature/amazing-feature`)
|
||||
5. Создайте **Pull Request**
|
||||
|
||||
### Области для улучшения:
|
||||
- 🌐 Веб-интерфейс управления
|
||||
- 📱 Мобильное приложение
|
||||
- 🔌 Интеграции с облачными провайдерами
|
||||
- 🤖 ИИ для детекции аномалий
|
||||
- 📊 Расширенная аналитика
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Этот проект распространяется под лицензией MIT. См. файл [LICENSE](LICENSE) для подробностей.
|
||||
|
||||
## 👥 Авторы
|
||||
|
||||
- **SmartSolTech Team** - *Первоначальная разработка* - [@SmartSolTech](https://github.com/SmartSolTech)
|
||||
|
||||
## 🆘 Поддержка
|
||||
|
||||
### Получить помощь:
|
||||
- 📧 **Email**: support@smartsoltech.com
|
||||
- 💬 **Telegram**: @PyGuardianSupport
|
||||
- 🐛 **Issues**: [GitHub Issues](https://github.com/SmartSolTech/PyGuardian/issues)
|
||||
- 📖 **Wiki**: [GitHub Wiki](https://github.com/SmartSolTech/PyGuardian/wiki)
|
||||
|
||||
### Перед обращением:
|
||||
1. Проверьте [FAQ](https://github.com/SmartSolTech/PyGuardian/wiki/FAQ)
|
||||
2. Запустите диагностику: `./scripts/test-install.sh`
|
||||
3. Соберите логи: `/debug export` в Telegram боте
|
||||
|
||||
---
|
||||
|
||||
**⚡ Быстрые команды:**
|
||||
|
||||
```bash
|
||||
# Проверить статус
|
||||
systemctl status pyguardian
|
||||
|
||||
# Просмотреть логи
|
||||
journalctl -u pyguardian -f
|
||||
|
||||
# Перезапустить
|
||||
systemctl restart pyguardian
|
||||
|
||||
# Обновить конфигурацию
|
||||
systemctl reload pyguardian
|
||||
```
|
||||
|
||||
**🎉 Спасибо за использование PyGuardian! Ваша безопасность - наш приоритет. 🛡️**
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,822 +0,0 @@
|
||||
# PyGuardian - Advanced Security & Cluster Management System 🛡️
|
||||
|
||||
**Комплексная система безопасности с централизованным управлением кластером серверов через Telegram бот.**
|
||||
|
||||
[](https://www.python.org/downloads/)
|
||||
[](https://opensource.org/licenses/MIT)
|
||||
[](https://www.docker.com/)
|
||||
[](https://core.telegram.org/bots)
|
||||
|
||||
## ⚡ Быстрый старт
|
||||
|
||||
### 🚀 Автоматическая установка (рекомендуется)
|
||||
```bash
|
||||
# Скачать и запустить установку
|
||||
wget https://raw.githubusercontent.com/your-repo/PyGuardian/main/install.sh
|
||||
chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 📦 Установка из исходного кода
|
||||
```bash
|
||||
# Клонировать репозиторий
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Проверить систему перед установкой
|
||||
./scripts/test-install.sh
|
||||
|
||||
# Установить (автономный режим)
|
||||
sudo make install
|
||||
|
||||
# Или установить кластерный контроллер
|
||||
sudo make controller
|
||||
```
|
||||
|
||||
### 🐳 Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Или через Docker Compose
|
||||
docker-compose up -d
|
||||
```
|
||||
|
||||
**📖 Подробные инструкции:** [QUICKSTART.md](QUICKSTART.md)
|
||||
|
||||
## 🎯 Основные возможности
|
||||
|
||||
### 🔒 Продвинутая система безопасности
|
||||
- **🚨 Обнаружение вторжений** - Real-time детекция атак
|
||||
- **🛡️ Интеллектуальный файрвол** - Автоматическая блокировка угроз
|
||||
- **🔍 Мониторинг системы** - Контроль ресурсов и процессов
|
||||
- **⚠️ DDoS защита** - Автоматическое обнаружение и блокировка
|
||||
- **🔐 2FA интеграция** - TOTP аутентификация через Telegram
|
||||
- **👤 Stealth Mode** - Скрытое обнаружение компрометации
|
||||
|
||||
### 🌐 Кластерное управление
|
||||
- **🎛️ Централизованный контроль** - Управление множеством серверов
|
||||
- **🚀 Автоматическое развертывание** - Развертывание агентов одной командой
|
||||
- **📡 Синхронизация конфигураций** - Единые политики безопасности
|
||||
- **📊 Мониторинг кластера** - Состояние всех узлов в реальном времени
|
||||
- **🔄 Load Balancing** - Автоматическое распределение нагрузки
|
||||
|
||||
### 💬 Продвинутый Telegram интерфейс
|
||||
- **🤖 Интерактивные команды** - Удобное управление через диалоги
|
||||
- **📈 Real-time мониторинг** - Мгновенные уведомления и алерты
|
||||
- **🔧 Удаленное управление** - Полный контроль через мессенджер
|
||||
- **👥 Многопользовательский доступ** - Ролевая модель доступа
|
||||
- **🗣️ Поддержка голосовых команд** - Управление голосом
|
||||
|
||||
### 🐳 Современные технологии развертывания
|
||||
- **📦 Docker поддержка** - Контейнеризированное развертывание
|
||||
- **⚙️ systemd интеграция** - Нативная интеграция с системой
|
||||
- **🔧 Ansible ready** - Готовые playbooks для автоматизации
|
||||
- **☁️ Cloud готовность** - Поддержка AWS, GCP, Azure
|
||||
- **📊 Метрики и логирование** - Интеграция с Grafana/Prometheus
|
||||
|
||||
## 🏗️ Архитектура системы
|
||||
|
||||
### Режимы развертывания:
|
||||
|
||||
#### 🖥️ Standalone (Автономный)
|
||||
Все компоненты на одном сервере
|
||||
```
|
||||
┌─────────────────┐
|
||||
│ PyGuardian │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Telegram │ │
|
||||
│ │ Bot │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
│ ┌─────────────┐ │
|
||||
│ │ Firewall │ │
|
||||
│ │ Manager │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
#### 🌐 Controller + Agents (Кластерный)
|
||||
Центральный контроллер управляет агентами
|
||||
```
|
||||
┌─────────────────┐ ┌─────────────────┐
|
||||
│ Controller │────│ Agent 1 │
|
||||
│ ┌─────────────┐ │ │ ┌─────────────┐ │
|
||||
│ │ Telegram │ │ │ │ Security │ │
|
||||
│ │ Bot │ │ │ │ Monitor │ │
|
||||
│ └─────────────┘ │ │ └─────────────┘ │
|
||||
│ ┌─────────────┐ │ └─────────────────┘
|
||||
│ │ Cluster │ │ │
|
||||
│ │ Manager │ │ ┌─────────────────┐
|
||||
│ └─────────────┘ │────│ Agent 2 │
|
||||
└─────────────────┘ │ ┌─────────────┐ │
|
||||
│ │ Security │ │
|
||||
│ │ Monitor │ │
|
||||
│ └─────────────┘ │
|
||||
└─────────────────┘
|
||||
```
|
||||
|
||||
## 🛠️ Технологический стек
|
||||
|
||||
- **🐍 Python 3.10+** - Основной язык разработки
|
||||
- **🤖 Telegram Bot API** - Интерфейс управления
|
||||
- **🗄️ SQLite/PostgreSQL** - База данных
|
||||
- **🔥 iptables/nftables** - Управление файрволом
|
||||
- **🐳 Docker** - Контейнеризация
|
||||
- **⚙️ systemd** - Управление службами
|
||||
- **🔒 cryptography** - Шифрование данных
|
||||
- **📡 asyncio** - Асинхронное выполнение
|
||||
- **📊 psutil** - Мониторинг системы
|
||||
|
||||
## 📋 Требования
|
||||
|
||||
### Минимальные требования:
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **RAM**: 512MB (автономный), 1GB (контроллер)
|
||||
- **Диск**: 1GB свободного места
|
||||
- **Сеть**: Доступ к Telegram API
|
||||
|
||||
### Рекомендуемые требования:
|
||||
- **ОС**: Ubuntu 22.04 LTS / Debian 12
|
||||
- **Python**: 3.11+
|
||||
- **RAM**: 2GB+ для контроллера кластера
|
||||
- **Диск**: 10GB+ для логов и резервных копий
|
||||
- **Сеть**: Выделенный IP для кластера
|
||||
|
||||
## 📚 Документация
|
||||
|
||||
- **[📖 QUICKSTART.md](QUICKSTART.md)** - Быстрое руководство по установке
|
||||
- **[⚙️ INSTALLATION.md](docs/INSTALLATION.md)** - Подробная установка и настройка
|
||||
- **[🌐 CLUSTER_SETUP.md](docs/CLUSTER_SETUP.md)** - Настройка кластера
|
||||
- **[⚡ configurations.md](examples/configurations.md)** - Примеры конфигураций
|
||||
- **[🤖 telegram-commands.md](examples/telegram-commands.md)** - Команды бота
|
||||
|
||||
## 🔧 Установка и использование
|
||||
|
||||
### 1️⃣ Автоматическая установка
|
||||
```bash
|
||||
# Автономная установка
|
||||
sudo ./install.sh
|
||||
|
||||
# Контроллер кластера
|
||||
sudo ./install.sh --mode controller
|
||||
|
||||
# Агент кластера
|
||||
sudo ./install.sh --mode agent --controller 192.168.1.10
|
||||
```
|
||||
|
||||
### 2️⃣ Make файл
|
||||
```bash
|
||||
# Показать все доступные команды
|
||||
make help
|
||||
|
||||
# Установка различных режимов
|
||||
make install # Автономный режим
|
||||
make controller # Кластерный контроллер
|
||||
make agent # Кластерный агент
|
||||
```
|
||||
|
||||
### 3️⃣ Docker установка
|
||||
```bash
|
||||
# Автоматическая Docker установка
|
||||
./scripts/docker-install.sh
|
||||
|
||||
# Ручная сборка образа
|
||||
docker build -t pyguardian .
|
||||
docker run -d --privileged --network host pyguardian
|
||||
```
|
||||
|
||||
## 🎯 Примеры использования
|
||||
|
||||
### Управление через Telegram бота:
|
||||
```
|
||||
/start - Начать работу с ботом
|
||||
/status - Статус системы безопасности
|
||||
/block IP - Заблокировать IP адрес
|
||||
/unblock IP - Разблокировать IP адрес
|
||||
/sessions - Показать активные сессии
|
||||
/logs - Просмотр логов
|
||||
/cluster status - Статус кластера (контроллер)
|
||||
/cluster add - Добавить сервер в кластер
|
||||
```
|
||||
|
||||
### Кластерное управление:
|
||||
```bash
|
||||
# Добавление сервера в кластер через SSH
|
||||
ssh-copy-id -i ~/.ssh/cluster_key.pub root@192.168.1.50
|
||||
|
||||
# В Telegram боте контроллера:
|
||||
/cluster add
|
||||
# Следовать интерактивным инструкциям
|
||||
```
|
||||
|
||||
## 🔒 Безопасность
|
||||
|
||||
- **🔑 Аутентификация**: Telegram user ID + опциональная 2FA
|
||||
- **🔐 Шифрование**: Все конфиденциальные данные зашифрованы
|
||||
- **🛡️ Изоляция**: Контейнеризация и изолированные процессы
|
||||
- **📝 Аудит**: Полное логирование всех действий
|
||||
- **🚫 Принцип минимальных привилегий**: Только необходимые права
|
||||
|
||||
## 📊 Мониторинг и алерты
|
||||
|
||||
### Автоматические уведомления о:
|
||||
- 🚨 Попытках взлома и атаках
|
||||
- 📈 Превышении лимитов ресурсов
|
||||
- 🔌 Подключении/отключении агентов кластера
|
||||
- ⚠️ Ошибках в системе безопасности
|
||||
- 📋 Результатах резервного копирования
|
||||
|
||||
### Интеграция с системами мониторинга:
|
||||
- **Grafana/Prometheus** - Метрики и дашборды
|
||||
- **ELK Stack** - Централизованное логирование
|
||||
- **SIEM системы** - Экспорт событий безопасности
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
Мы приветствуем вклад в развитие PyGuardian!
|
||||
|
||||
### Как принять участие:
|
||||
1. **Fork** репозитория
|
||||
2. Создайте **feature branch** (`git checkout -b feature/amazing-feature`)
|
||||
3. **Commit** изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. **Push** в branch (`git push origin feature/amazing-feature`)
|
||||
5. Создайте **Pull Request**
|
||||
|
||||
### Области для улучшения:
|
||||
- 🌐 Веб-интерфейс управления
|
||||
- 📱 Мобильное приложение
|
||||
- 🔌 Интеграции с облачными провайдерами
|
||||
- 🤖 ИИ для детекции аномалий
|
||||
- 📊 Расширенная аналитика
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Этот проект распространяется под лицензией MIT. См. файл [LICENSE](LICENSE) для подробностей.
|
||||
|
||||
## 👥 Авторы
|
||||
|
||||
- **SmartSolTech Team** - *Первоначальная разработка* - [@SmartSolTech](https://github.com/SmartSolTech)
|
||||
|
||||
## 🆘 Поддержка
|
||||
|
||||
### Получить помощь:
|
||||
- 📧 **Email**: support@smartsoltech.com
|
||||
- 💬 **Telegram**: @PyGuardianSupport
|
||||
- 🐛 **Issues**: [GitHub Issues](https://github.com/SmartSolTech/PyGuardian/issues)
|
||||
- 📖 **Wiki**: [GitHub Wiki](https://github.com/SmartSolTech/PyGuardian/wiki)
|
||||
|
||||
### Перед обращением:
|
||||
1. Проверьте [FAQ](https://github.com/SmartSolTech/PyGuardian/wiki/FAQ)
|
||||
2. Запустите диагностику: `./scripts/test-install.sh`
|
||||
3. Соберите логи: `/debug export` в Telegram боте
|
||||
|
||||
---
|
||||
|
||||
**⚡ Быстрые команды:**
|
||||
|
||||
```bash
|
||||
# Проверить статус
|
||||
systemctl status pyguardian
|
||||
|
||||
# Просмотреть логи
|
||||
journalctl -u pyguardian -f
|
||||
|
||||
# Перезапустить
|
||||
systemctl restart pyguardian
|
||||
|
||||
# Обновить конфигурацию
|
||||
systemctl reload pyguardian
|
||||
```
|
||||
|
||||
**🎉 Спасибо за использование PyGuardian! Ваша безопасность - наш приоритет. 🛡️**
|
||||
- **Telegram**: Токен бота и admin ID
|
||||
|
||||
## 🚀 Быстрая установка
|
||||
|
||||
### 1. Клонирование репозитория
|
||||
```bash
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
```
|
||||
|
||||
### 2. Автоматическая установка
|
||||
```bash
|
||||
sudo chmod +x install.sh
|
||||
sudo ./install.sh
|
||||
```
|
||||
|
||||
### 3. Настройка Telegram бота
|
||||
|
||||
#### Создание бота:
|
||||
1. Отправьте `/newbot` боту [@BotFather](https://t.me/BotFather)
|
||||
2. Следуйте инструкциям и получите токен
|
||||
3. Узнайте ваш Telegram ID через [@userinfobot](https://t.me/userinfobot)
|
||||
|
||||
#### Обновление конфигурации:
|
||||
```bash
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
```yaml
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_id: YOUR_TELEGRAM_ID
|
||||
```
|
||||
|
||||
### 4. Запуск сервиса
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Проверка статуса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
```
|
||||
|
||||
## ⚙️ Конфигурация
|
||||
|
||||
### Основные параметры
|
||||
|
||||
| Параметр | Описание | По умолчанию |
|
||||
|----------|----------|--------------|
|
||||
| `max_attempts` | Максимум попыток за time_window | 5 |
|
||||
| `time_window` | Окно времени в секундах | 60 |
|
||||
| `unban_time` | Время автоматической разблокировки | 3600 |
|
||||
| `auth_log_path` | Путь к auth.log | `/var/log/auth.log` |
|
||||
| `check_interval` | Интервал проверки лога | 1.0 |
|
||||
|
||||
### Firewall настройки
|
||||
|
||||
#### Для iptables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
iptables:
|
||||
table: "filter"
|
||||
```
|
||||
|
||||
#### Для nftables:
|
||||
```yaml
|
||||
firewall:
|
||||
backend: "nftables"
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
```
|
||||
|
||||
### Белый список IP
|
||||
```yaml
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
- "192.168.1.0/24" # Локальная сеть
|
||||
- "10.0.0.0/8" # VPN сеть
|
||||
```
|
||||
|
||||
### Конфигурация кластера
|
||||
```yaml
|
||||
cluster:
|
||||
enabled: true
|
||||
ssh_timeout: 30
|
||||
deployment_timeout: 300
|
||||
retry_attempts: 3
|
||||
agent_port: 8081
|
||||
master_host: "192.168.1.100"
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
## 🤖 Команды Telegram бота
|
||||
|
||||
### Основные команды
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/start` | Приветствие и информация о системе |
|
||||
| `/status` | Статистика атак и система |
|
||||
| `/top10` | Топ-10 атакующих IP за сутки |
|
||||
| `/details <ip>` | Детальная информация по IP |
|
||||
| `/ban <ip>` | Ручная блокировка IP |
|
||||
| `/unban <ip>` | Разблокировка IP |
|
||||
| `/list` | Список всех забаненных IP |
|
||||
| `/help` | Справка по командам |
|
||||
|
||||
### 🚨 Новые команды управления безопасностью
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/compromises` | Список обнаруженных взломов |
|
||||
| `/sessions` | Активные SSH сессии |
|
||||
| `/kick <user/pid>` | Завершить сессию пользователя |
|
||||
| `/generate_password <user>` | Сгенерировать новый пароль |
|
||||
| `/set_password <user> <pass>` | Установить пароль для пользователя |
|
||||
|
||||
### 🌐 Команды кластерного управления
|
||||
| Команда | Описание |
|
||||
|---------|----------|
|
||||
| `/cluster` | Статистика кластера и подключенные агенты |
|
||||
| `/add_server <name> <ip> <user>` | Добавить новый сервер в кластер |
|
||||
| `/remove_server <name>` | Удалить сервер из кластера |
|
||||
| `/deploy_agent <name>` | Автоматически развернуть агента на сервере |
|
||||
| `/agents` | Список всех агентов кластера |
|
||||
| `/check_agents` | Проверить статус всех агентов |
|
||||
|
||||
### Примеры использования
|
||||
```
|
||||
/details 192.168.1.100
|
||||
/ban 10.0.0.50
|
||||
/unban 203.0.113.1
|
||||
|
||||
# Новые команды безопасности
|
||||
/sessions
|
||||
/kick admin
|
||||
/generate_password ubuntu
|
||||
/compromises
|
||||
|
||||
# Команды кластерного управления
|
||||
/cluster
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/deploy_agent web01
|
||||
/agents
|
||||
/check_agents
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 📊 Мониторинг и логирование
|
||||
|
||||
### Детекция атак
|
||||
Система отслеживает следующие события в auth.log:
|
||||
- `Failed password`
|
||||
- `Invalid user`
|
||||
- `authentication failure`
|
||||
- `Too many authentication failures`
|
||||
- `Failed publickey`
|
||||
- `Connection closed by authenticating user`
|
||||
|
||||
### Уведомления в Telegram
|
||||
- ⚠️ Автоматическая блокировка IP
|
||||
- 🔓 Ручная блокировка/разблокировка
|
||||
- 🟢 Автоматическая разблокировка по таймеру
|
||||
- ❌ Системные ошибки
|
||||
- 🚨 Критические атаки
|
||||
|
||||
### Логи системы
|
||||
```bash
|
||||
# Системные логи
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Файловые логи
|
||||
sudo tail -f /var/log/pyguardian.log
|
||||
|
||||
# Логи ошибок
|
||||
sudo journalctl -u pyguardian -p err
|
||||
```
|
||||
|
||||
## 🔧 Управление сервисом
|
||||
|
||||
### Основные команды
|
||||
```bash
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Перезагрузка конфигурации
|
||||
sudo systemctl reload pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключить автозапуск
|
||||
sudo systemctl disable pyguardian
|
||||
```
|
||||
|
||||
### Мониторинг ресурсов
|
||||
```bash
|
||||
# Использование памяти
|
||||
sudo systemctl show pyguardian --property=MemoryCurrent
|
||||
|
||||
# Количество процессов
|
||||
sudo systemctl show pyguardian --property=TasksCurrent
|
||||
|
||||
# Время работы
|
||||
sudo systemctl show pyguardian --property=ActiveEnterTimestamp
|
||||
```
|
||||
|
||||
## 🌐 Настройка кластера
|
||||
|
||||
### Включение кластерного режима
|
||||
|
||||
1. **Настройка мастер-сервера:**
|
||||
```bash
|
||||
# В config.yaml включить кластерный режим
|
||||
cluster:
|
||||
enabled: true
|
||||
master_host: "192.168.1.100" # IP мастер-сервера
|
||||
master_port: 8080
|
||||
```
|
||||
|
||||
2. **Добавление серверов в кластер:**
|
||||
```bash
|
||||
# Через Telegram бота
|
||||
/add_server web01 192.168.1.10 ubuntu
|
||||
/add_server db01 192.168.1.20 admin
|
||||
/add_server app01 192.168.1.30 deploy
|
||||
```
|
||||
|
||||
3. **Автоматическое развертывание агентов:**
|
||||
```bash
|
||||
# Развернуть агента на сервере
|
||||
/deploy_agent web01
|
||||
|
||||
# Проверить статус всех агентов
|
||||
/check_agents
|
||||
```
|
||||
|
||||
### Требования для кластера
|
||||
- **SSH доступ**: Мастер-сервер должен иметь SSH доступ ко всем агентам
|
||||
- **Сетевые порты**: Порт 8080 (мастер), 8081 (агенты)
|
||||
- **Ключи SSH**: Настроенные SSH ключи или пароли для автоматического развертывания
|
||||
|
||||
### Управление кластером
|
||||
```bash
|
||||
# Просмотр статистики кластера
|
||||
/cluster
|
||||
|
||||
# Список всех агентов
|
||||
/agents
|
||||
|
||||
# Удаление сервера из кластера
|
||||
/remove_server old_server
|
||||
```
|
||||
|
||||
## 🛠️ Ручная установка (альтернативный способ)
|
||||
|
||||
### 1. Подготовка системы
|
||||
```bash
|
||||
# Ubuntu/Debian
|
||||
sudo apt update
|
||||
sudo apt install python3 python3-pip python3-venv iptables
|
||||
|
||||
# CentOS/RHEL
|
||||
sudo yum install python3 python3-pip iptables
|
||||
|
||||
# или
|
||||
sudo dnf install python3 python3-pip iptables
|
||||
```
|
||||
|
||||
### 2. Создание пользователя (опционально)
|
||||
```bash
|
||||
sudo useradd -r -s /bin/false pyguardian
|
||||
sudo mkdir -p /opt/pyguardian /var/lib/pyguardian
|
||||
sudo chown pyguardian:pyguardian /var/lib/pyguardian
|
||||
```
|
||||
|
||||
### 3. Установка PyGuardian
|
||||
```bash
|
||||
# Клонирование
|
||||
git clone https://git.smartsoltech.kr/trevor/PyGuardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Копирование файлов
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo cp config/config.yaml /etc/pyguardian/
|
||||
sudo cp systemd/pyguardian.service /etc/systemd/system/
|
||||
|
||||
# Установка зависимостей
|
||||
cd /opt/pyguardian
|
||||
sudo pip3 install -r requirements.txt
|
||||
|
||||
# Права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
sudo chown -R root:root /opt/pyguardian
|
||||
|
||||
# Systemd
|
||||
sudo systemctl daemon-reload
|
||||
```
|
||||
|
||||
## 🔍 Диагностика и решение проблем
|
||||
|
||||
### Проблемы с запуском
|
||||
|
||||
#### Ошибка "Permission denied"
|
||||
```bash
|
||||
# Проверить права
|
||||
ls -la /opt/pyguardian/main.py
|
||||
|
||||
# Исправить права
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
#### Ошибка "Module not found"
|
||||
```bash
|
||||
# Переустановить зависимости
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt --force-reinstall
|
||||
```
|
||||
|
||||
#### Ошибка доступа к auth.log
|
||||
```bash
|
||||
# Проверить существование файла
|
||||
ls -la /var/log/auth.log
|
||||
|
||||
# Проверить права
|
||||
sudo chmod 644 /var/log/auth.log
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
#### iptables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo iptables -L -n
|
||||
|
||||
# Проверить модули
|
||||
lsmod | grep ip_tables
|
||||
|
||||
# Загрузить модуль
|
||||
sudo modprobe ip_tables
|
||||
```
|
||||
|
||||
#### nftables не работает
|
||||
```bash
|
||||
# Проверить статус
|
||||
sudo nft list ruleset
|
||||
|
||||
# Установить nftables
|
||||
sudo apt install nftables # Ubuntu/Debian
|
||||
sudo yum install nftables # CentOS/RHEL
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
#### Бот не отвечает
|
||||
```bash
|
||||
# Проверить токен и ID в конфигурации
|
||||
sudo cat /etc/pyguardian/config.yaml | grep -A 3 telegram
|
||||
|
||||
# Проверить сетевое соединение
|
||||
curl -s "https://api.telegram.org/botYOUR_TOKEN/getMe"
|
||||
```
|
||||
|
||||
#### Уведомления не приходят
|
||||
```bash
|
||||
# Проверить ID администратора
|
||||
# Убедиться что вы написали боту /start
|
||||
# Проверить логи
|
||||
sudo journalctl -u pyguardian | grep -i telegram
|
||||
```
|
||||
|
||||
### Производительность
|
||||
|
||||
#### Высокое потребление памяти
|
||||
```yaml
|
||||
# Настроить в config.yaml
|
||||
performance:
|
||||
max_memory_mb: 50
|
||||
cleanup_interval: 1800 # 30 минут
|
||||
max_records_age: 259200 # 3 дня
|
||||
```
|
||||
|
||||
#### Высокая нагрузка на диск
|
||||
```yaml
|
||||
# Увеличить интервал проверки
|
||||
monitoring:
|
||||
check_interval: 2.0 # Проверять каждые 2 секунды
|
||||
|
||||
# Настроить логирование
|
||||
logging:
|
||||
log_level: "WARNING" # Меньше логов
|
||||
```
|
||||
|
||||
## 📈 Мониторинг производительности
|
||||
|
||||
### Grafana + Prometheus (дополнительно)
|
||||
```bash
|
||||
# Пример экспорта метрик через systemd
|
||||
sudo systemctl status pyguardian --output=json
|
||||
```
|
||||
|
||||
### Встроенная статистика
|
||||
Телеграм бот предоставляет:
|
||||
- Количество атак за день
|
||||
- Топ атакующих IP
|
||||
- Использование памяти
|
||||
- Время работы системы
|
||||
- Статистику обработанных событий
|
||||
|
||||
## 🔐 Безопасность
|
||||
|
||||
### Рекомендации
|
||||
1. **Регулярно обновляйте** конфигурацию белого списка
|
||||
2. **Мониторьте логи** на предмет ложных срабатываний
|
||||
3. **Используйте VPN** для доступа к серверу
|
||||
4. **Настройте резервное копирование** конфигурации
|
||||
5. **Периодически проверяйте** список забаненных IP
|
||||
|
||||
### Файлы конфигурации
|
||||
- `/etc/pyguardian/config.yaml` - основная конфигурация
|
||||
- `/var/lib/pyguardian/guardian.db` - база данных SQLite
|
||||
- `/var/log/pyguardian.log` - логи приложения
|
||||
|
||||
### Права доступа
|
||||
```bash
|
||||
# Конфигурация (только root может читать токены)
|
||||
sudo chmod 600 /etc/pyguardian/config.yaml
|
||||
|
||||
# База данных
|
||||
sudo chmod 600 /var/lib/pyguardian/guardian.db
|
||||
|
||||
# Логи
|
||||
sudo chmod 644 /var/log/pyguardian.log
|
||||
```
|
||||
|
||||
## 📚 API и интеграция
|
||||
|
||||
### Webhook уведомления (планируется)
|
||||
```yaml
|
||||
# В будущих версиях
|
||||
webhooks:
|
||||
on_ban: "https://your-domain.com/webhook/ban"
|
||||
on_unban: "https://your-domain.com/webhook/unban"
|
||||
```
|
||||
|
||||
### REST API (планируется)
|
||||
- `GET /api/stats` - получение статистики
|
||||
- `POST /api/ban` - ручной бан IP
|
||||
- `DELETE /api/ban/<ip>` - разбан IP
|
||||
|
||||
## 🔄 Обновление
|
||||
|
||||
### Автоматическое обновление
|
||||
```bash
|
||||
# Скачать новую версию
|
||||
cd /tmp
|
||||
git clone https://github.com/your-org/pyguardian.git
|
||||
cd pyguardian
|
||||
|
||||
# Остановить сервис
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновить файлы (сохраняя конфигурацию)
|
||||
sudo cp -r src/ main.py requirements.txt /opt/pyguardian/
|
||||
sudo pip3 install -r /opt/pyguardian/requirements.txt
|
||||
|
||||
# Запустить сервис
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Миграция конфигурации
|
||||
При обновлениях проверяйте совместимость конфигурации:
|
||||
```bash
|
||||
# Проверка конфигурации
|
||||
python3 -c "import yaml; yaml.safe_load(open('/etc/pyguardian/config.yaml'))"
|
||||
```
|
||||
|
||||
## 🤝 Вклад в проект
|
||||
|
||||
1. Форкните репозиторий
|
||||
2. Создайте ветку для функции (`git checkout -b feature/amazing-feature`)
|
||||
3. Зафиксируйте изменения (`git commit -m 'Add amazing feature'`)
|
||||
4. Отправьте в ветку (`git push origin feature/amazing-feature`)
|
||||
5. Откройте Pull Request
|
||||
|
||||
## 📄 Лицензия
|
||||
|
||||
Distributed under the MIT License. See `LICENSE` for more information.
|
||||
|
||||
## 💬 Поддержка
|
||||
|
||||
- **Issues**: [GitHub Issues](https://github.com/your-org/pyguardian/issues)
|
||||
- **Telegram**: [@pyguardian_support](https://t.me/pyguardian_support)
|
||||
- **Email**: support@pyguardian.dev
|
||||
|
||||
## 🎯 Планы развития
|
||||
|
||||
- [ ] Web-интерфейс (опционально)
|
||||
- [ ] REST API
|
||||
- [ ] Webhook уведомления
|
||||
- [ ] Интеграция с облачными провайдерами
|
||||
- [ ] Машинное обучение для детекции аномалий
|
||||
- [ ] Поддержка IPv6
|
||||
- [x] **Кластерный режим** - ✅ Реализовано!
|
||||
- [ ] Экспорт метрик Prometheus
|
||||
|
||||
---
|
||||
|
||||
**PyGuardian** - Made with ❤️ for Linux system administrators
|
||||
@@ -1,66 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,86 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# 🚨 STEALTH SECURITY SETTINGS
|
||||
authorized_users: # Список разрешенных пользователей
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users: # Пользователи-приманки для обнаружения взломов
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300 # Время скрытого режима после обнаружения взлома (секунды)
|
||||
compromise_indicators: # Индикаторы компромисса
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,104 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# 🚨 STEALTH SECURITY SETTINGS
|
||||
authorized_users: # Список разрешенных пользователей
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users: # Пользователи-приманки для обнаружения взломов
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300 # Время скрытого режима после обнаружения взлома (секунды)
|
||||
compromise_indicators: # Индикаторы компромисса
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Password Management Settings
|
||||
passwords:
|
||||
password_length: 16 # Длина генерируемых паролей
|
||||
use_special_chars: true # Использовать специальные символы
|
||||
password_history_size: 5 # Размер истории паролей
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600 # Интервал очистки старых записей (секунды)
|
||||
max_records_age: 604800 # Максимальный возраст записей (секунды) - 7 дней
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO" # Уровень логирования: DEBUG, INFO, WARNING, ERROR
|
||||
file: "/var/log/pyguardian.log"
|
||||
max_size: 10485760 # Максимальный размер лог файла (10MB)
|
||||
backup_count: 5 # Количество резервных копий лог файлов
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,104 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# 🚨 STEALTH SECURITY SETTINGS
|
||||
authorized_users: # Список разрешенных пользователей
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users: # Пользователи-приманки для обнаружения взломов
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300 # Время скрытого режима после обнаружения взлома (секунды)
|
||||
compromise_indicators: # Индикаторы компромисса
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Password Management Settings
|
||||
passwords:
|
||||
password_length: 16 # Длина генерируемых паролей
|
||||
use_special_chars: true # Использовать специальные символы
|
||||
password_history_size: 5 # Размер истории паролей
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600 # Интервал очистки старых записей (секунды)
|
||||
max_records_age: 604800 # Максимальный возраст записей (секунды) - 7 дней
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO" # Уровень логирования: DEBUG, INFO, WARNING, ERROR
|
||||
file: "/var/log/pyguardian.log"
|
||||
max_size: 10485760 # Максимальный размер лог файла (10MB)
|
||||
backup_count: 5 # Количество резервных копий лог файлов
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,113 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# 🚨 STEALTH SECURITY SETTINGS
|
||||
authorized_users: # Список разрешенных пользователей
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users: # Пользователи-приманки для обнаружения взломов
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300 # Время скрытого режима после обнаружения взлома (секунды)
|
||||
compromise_indicators: # Индикаторы компромисса
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Password Management Settings
|
||||
passwords:
|
||||
password_length: 16 # Длина генерируемых паролей
|
||||
use_special_chars: true # Использовать специальные символы
|
||||
password_history_size: 5 # Размер истории паролей
|
||||
|
||||
# Cluster Management Settings
|
||||
cluster:
|
||||
cluster_name: "PyGuardian-Cluster" # Название кластера
|
||||
master_server: true # Этот сервер - master
|
||||
agents_config_path: "/var/lib/pyguardian/agents.yaml"
|
||||
deployment_path: "/opt/pyguardian"
|
||||
ssh_timeout: 30 # Таймаут SSH соединений
|
||||
ssh_retries: 3 # Количество попыток подключения
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600 # Интервал очистки старых записей (секунды)
|
||||
max_records_age: 604800 # Максимальный возраст записей (секунды) - 7 дней
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO" # Уровень логирования: DEBUG, INFO, WARNING, ERROR
|
||||
file: "/var/log/pyguardian.log"
|
||||
max_size: 10485760 # Максимальный размер лог файла (10MB)
|
||||
backup_count: 5 # Количество резервных копий лог файлов
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,113 +0,0 @@
|
||||
# PyGuardian Configuration File
|
||||
# ==============================
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE" # Токен бота от @BotFather
|
||||
admin_id: 0 # Ваш Telegram ID (получите через @userinfobot)
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5 # Максимум попыток авторизации за time_window
|
||||
time_window: 60 # Окно времени в секундах
|
||||
unban_time: 3600 # Время автоматической разблокировки (сек)
|
||||
|
||||
# 🚨 STEALTH SECURITY SETTINGS
|
||||
authorized_users: # Список разрешенных пользователей
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users: # Пользователи-приманки для обнаружения взломов
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300 # Время скрытого режима после обнаружения взлома (секунды)
|
||||
compromise_indicators: # Индикаторы компромисса
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0 # Интервал проверки лога в секундах
|
||||
|
||||
# Паттерны для детекции атак
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
- "Connection closed by authenticating user"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables" # iptables или nftables
|
||||
chain: "INPUT" # Цепочка для блокировки
|
||||
target: "DROP" # Действие (DROP/REJECT)
|
||||
|
||||
# Настройки для iptables
|
||||
iptables:
|
||||
table: "filter"
|
||||
|
||||
# Настройки для nftables
|
||||
nftables:
|
||||
table: "inet pyguardian"
|
||||
chain: "input"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/guardian.db"
|
||||
backup_interval: 86400 # Бэкап БД каждые 24 часа
|
||||
|
||||
# Password Management Settings
|
||||
passwords:
|
||||
password_length: 16 # Длина генерируемых паролей
|
||||
use_special_chars: true # Использовать специальные символы
|
||||
password_history_size: 5 # Размер истории паролей
|
||||
|
||||
# Cluster Management Settings
|
||||
cluster:
|
||||
cluster_name: "PyGuardian-Cluster" # Название кластера
|
||||
master_server: true # Этот сервер - master
|
||||
agents_config_path: "/var/lib/pyguardian/agents.yaml"
|
||||
deployment_path: "/opt/pyguardian"
|
||||
ssh_timeout: 30 # Таймаут SSH соединений
|
||||
ssh_retries: 3 # Количество попыток подключения
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600 # Интервал очистки старых записей (секунды)
|
||||
max_records_age: 604800 # Максимальный возраст записей (секунды) - 7 дней
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO" # Уровень логирования: DEBUG, INFO, WARNING, ERROR
|
||||
file: "/var/log/pyguardian.log"
|
||||
max_size: 10485760 # Максимальный размер лог файла (10MB)
|
||||
backup_count: 5 # Количество резервных копий лог файлов
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
log_file: "/var/log/pyguardian.log"
|
||||
log_level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
||||
max_log_size: 10485760 # 10MB
|
||||
backup_count: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
max_memory_mb: 100 # Максимальное использование памяти
|
||||
cleanup_interval: 3600 # Очистка старых записей (сек)
|
||||
max_records_age: 604800 # Удалять записи старше недели
|
||||
|
||||
# Whitelist IPs (никогда не блокировать)
|
||||
whitelist:
|
||||
- "127.0.0.1"
|
||||
- "::1"
|
||||
# - "192.168.1.0/24" # Добавьте ваши доверенные сети
|
||||
@@ -1,91 +0,0 @@
|
||||
# PyGuardian Multi-stage Dockerfile
|
||||
# Supports both controller and agent modes
|
||||
|
||||
FROM python:3.11-slim AS base
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \
|
||||
iptables \
|
||||
iputils-ping \
|
||||
openssh-client \
|
||||
curl \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create pyguardian user
|
||||
RUN groupadd -r pyguardian && useradd -r -g pyguardian pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /opt/pyguardian
|
||||
|
||||
# Copy requirements and install Python dependencies
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy source code
|
||||
COPY src/ ./src/
|
||||
COPY config/ ./config/
|
||||
COPY main.py .
|
||||
|
||||
# Set permissions
|
||||
RUN chown -R pyguardian:pyguardian /opt/pyguardian
|
||||
|
||||
# Create data and logs directories
|
||||
RUN mkdir -p /opt/pyguardian/data /opt/pyguardian/logs \
|
||||
&& chown -R pyguardian:pyguardian /opt/pyguardian/data /opt/pyguardian/logs
|
||||
|
||||
# Controller mode
|
||||
FROM base AS controller
|
||||
|
||||
# Expose API port
|
||||
EXPOSE 8443
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=controller
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
|
||||
CMD curl -f http://localhost:8443/health || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py", "--mode", "controller"]
|
||||
|
||||
# Agent mode
|
||||
FROM base AS agent
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=agent
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check for agent
|
||||
HEALTHCHECK --interval=60s --timeout=15s --start-period=30s --retries=3 \
|
||||
CMD python -c "import sys; sys.exit(0)" || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py", "--mode", "agent"]
|
||||
|
||||
# Standalone mode (default)
|
||||
FROM base AS standalone
|
||||
|
||||
# Expose API port (optional for standalone)
|
||||
EXPOSE 8443
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=standalone
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
|
||||
CMD python -c "import sys; sys.exit(0)" || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py"]
|
||||
@@ -1,91 +0,0 @@
|
||||
# PyGuardian Multi-stage Dockerfile
|
||||
# Supports both controller and agent modes
|
||||
|
||||
FROM python:3.11-slim AS base
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \
|
||||
iptables \
|
||||
iputils-ping \
|
||||
openssh-client \
|
||||
curl \
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create pyguardian user
|
||||
RUN groupadd -r pyguardian && useradd -r -g pyguardian pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /opt/pyguardian
|
||||
|
||||
# Copy requirements and install Python dependencies
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy source code
|
||||
COPY src/ ./src/
|
||||
COPY config/ ./config/
|
||||
COPY main.py .
|
||||
|
||||
# Set permissions
|
||||
RUN chown -R pyguardian:pyguardian /opt/pyguardian
|
||||
|
||||
# Create data and logs directories
|
||||
RUN mkdir -p /opt/pyguardian/data /opt/pyguardian/logs \
|
||||
&& chown -R pyguardian:pyguardian /opt/pyguardian/data /opt/pyguardian/logs
|
||||
|
||||
# Controller mode
|
||||
FROM base AS controller
|
||||
|
||||
# Expose API port
|
||||
EXPOSE 8443
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=controller
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
|
||||
CMD curl -f http://localhost:8443/health || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py", "--mode", "controller"]
|
||||
|
||||
# Agent mode
|
||||
FROM base AS agent
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=agent
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check for agent
|
||||
HEALTHCHECK --interval=60s --timeout=15s --start-period=30s --retries=3 \
|
||||
CMD python -c "import sys; sys.exit(0)" || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py", "--mode", "agent"]
|
||||
|
||||
# Standalone mode (default)
|
||||
FROM base AS standalone
|
||||
|
||||
# Expose API port (optional for standalone)
|
||||
EXPOSE 8443
|
||||
|
||||
# Run as pyguardian user
|
||||
USER pyguardian
|
||||
|
||||
# Set environment variables
|
||||
ENV PYGUARDIAN_MODE=standalone
|
||||
ENV PYTHONPATH=/opt/pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=40s --retries=3 \
|
||||
CMD python -c "import sys; sys.exit(0)" || exit 1
|
||||
|
||||
# Start command
|
||||
CMD ["python", "main.py"]
|
||||
@@ -1,77 +0,0 @@
|
||||
# PyGuardian Docker Compose
|
||||
# Controller + Agent cluster setup
|
||||
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build:
|
||||
context: ../..
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
target: controller
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- controller_data:/opt/pyguardian/data
|
||||
- controller_logs:/opt/pyguardian/logs
|
||||
- controller_config:/opt/pyguardian/config
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=controller
|
||||
- PYGUARDIAN_CONFIG=/opt/pyguardian/config/config.yaml
|
||||
- TELEGRAM_BOT_TOKEN=${TELEGRAM_BOT_TOKEN}
|
||||
- CLUSTER_SECRET=${CLUSTER_SECRET}
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8443/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 40s
|
||||
|
||||
pyguardian-agent-1:
|
||||
build:
|
||||
context: ../..
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
target: agent
|
||||
container_name: pyguardian-agent-1
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- agent1_data:/opt/pyguardian/data
|
||||
- agent1_logs:/opt/pyguardian/logs
|
||||
- agent1_config:/opt/pyguardian/config
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=agent
|
||||
- CONTROLLER_HOST=localhost
|
||||
- CONTROLLER_PORT=8443
|
||||
- CLUSTER_SECRET=${CLUSTER_SECRET}
|
||||
depends_on:
|
||||
- pyguardian-controller
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import sys; sys.exit(0)"]
|
||||
interval: 60s
|
||||
timeout: 15s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
controller_data:
|
||||
driver: local
|
||||
controller_logs:
|
||||
driver: local
|
||||
controller_config:
|
||||
driver: local
|
||||
agent1_data:
|
||||
driver: local
|
||||
agent1_logs:
|
||||
driver: local
|
||||
agent1_config:
|
||||
driver: local
|
||||
|
||||
networks:
|
||||
default:
|
||||
name: pyguardian-network
|
||||
@@ -1,77 +0,0 @@
|
||||
# PyGuardian Docker Compose
|
||||
# Controller + Agent cluster setup
|
||||
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build:
|
||||
context: ../..
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
target: controller
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- controller_data:/opt/pyguardian/data
|
||||
- controller_logs:/opt/pyguardian/logs
|
||||
- controller_config:/opt/pyguardian/config
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=controller
|
||||
- PYGUARDIAN_CONFIG=/opt/pyguardian/config/config.yaml
|
||||
- TELEGRAM_BOT_TOKEN=${TELEGRAM_BOT_TOKEN}
|
||||
- CLUSTER_SECRET=${CLUSTER_SECRET}
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8443/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 40s
|
||||
|
||||
pyguardian-agent-1:
|
||||
build:
|
||||
context: ../..
|
||||
dockerfile: deployment/docker/Dockerfile
|
||||
target: agent
|
||||
container_name: pyguardian-agent-1
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- agent1_data:/opt/pyguardian/data
|
||||
- agent1_logs:/opt/pyguardian/logs
|
||||
- agent1_config:/opt/pyguardian/config
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=agent
|
||||
- CONTROLLER_HOST=localhost
|
||||
- CONTROLLER_PORT=8443
|
||||
- CLUSTER_SECRET=${CLUSTER_SECRET}
|
||||
depends_on:
|
||||
- pyguardian-controller
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import sys; sys.exit(0)"]
|
||||
interval: 60s
|
||||
timeout: 15s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
controller_data:
|
||||
driver: local
|
||||
controller_logs:
|
||||
driver: local
|
||||
controller_config:
|
||||
driver: local
|
||||
agent1_data:
|
||||
driver: local
|
||||
agent1_logs:
|
||||
driver: local
|
||||
agent1_config:
|
||||
driver: local
|
||||
|
||||
networks:
|
||||
default:
|
||||
name: pyguardian-network
|
||||
@@ -1,357 +0,0 @@
|
||||
# PyGuardian Installation Guide
|
||||
|
||||
## Обзор
|
||||
|
||||
PyGuardian предлагает несколько способов установки в зависимости от ваших потребностей:
|
||||
|
||||
1. **Standalone** - Автономный сервер (все в одном)
|
||||
2. **Controller** - Центральный контроллер кластера
|
||||
3. **Agent** - Агент для подключения к контроллеру
|
||||
4. **Docker** - Контейнеризованное развертывание
|
||||
|
||||
## Быстрая установка
|
||||
|
||||
### Использование make
|
||||
|
||||
```bash
|
||||
# Клонирование репозитория
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Интерактивная установка
|
||||
sudo make install
|
||||
|
||||
# Или быстрая автономная установка
|
||||
sudo make standalone
|
||||
|
||||
# Или контроллер кластера
|
||||
sudo make controller
|
||||
|
||||
# Или агент кластера
|
||||
sudo make agent
|
||||
```
|
||||
|
||||
### Использование install.sh
|
||||
|
||||
```bash
|
||||
# Интерактивный режим
|
||||
sudo ./install.sh
|
||||
|
||||
# Быстрая установка
|
||||
sudo ./install.sh --quick
|
||||
|
||||
# Конкретный режим
|
||||
sudo ./install.sh --interactive
|
||||
sudo ./install.sh --docker
|
||||
```
|
||||
|
||||
## Подробная установка
|
||||
|
||||
### 1. Standalone режим
|
||||
|
||||
**Назначение**: Полнофункциональная система на одном сервере
|
||||
**Подходит для**: Небольшие инфраструктуры, тестирования
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make standalone
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=standalone --non-interactive \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Мониторинг auth.log
|
||||
- Telegram бот управления
|
||||
- Stealth security система
|
||||
- Firewall интеграция
|
||||
- Автоматическое управление паролями
|
||||
- SSH session management
|
||||
|
||||
### 2. Controller режим
|
||||
|
||||
**Назначение**: Центральный контроллер для управления кластером агентов
|
||||
**Подходит для**: Крупные инфраструктуры, централизованное управление
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make controller
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=controller --non-interactive \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Все функции Standalone
|
||||
- API для управления агентами
|
||||
- Веб-интерфейс управления
|
||||
- Централизованная отчетность
|
||||
- Автоматическое развертывание агентов
|
||||
|
||||
**После установки**:
|
||||
1. Откройте порт 8080 в firewall
|
||||
2. Настройте SSL сертификат
|
||||
3. Добавьте агенты через Telegram команды
|
||||
|
||||
### 3. Agent режим
|
||||
|
||||
**Назначение**: Агент для подключения к контроллеру
|
||||
**Подходит для**: Серверы в составе кластера
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make agent
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=agent --non-interactive \
|
||||
--controller-url="https://controller.example.com:8080" \
|
||||
--agent-token="AGENT_TOKEN"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Локальный мониторинг auth.log
|
||||
- Firewall управление
|
||||
- Подключение к контроллеру
|
||||
- Передача данных в центр
|
||||
|
||||
**Перед установкой**:
|
||||
1. Получите токен агента от администратора контроллера
|
||||
2. Убедитесь в доступности контроллера по сети
|
||||
|
||||
## Docker установка
|
||||
|
||||
### Controller в Docker
|
||||
|
||||
```bash
|
||||
# Интерактивная установка
|
||||
sudo ./scripts/docker-install.sh --mode=controller
|
||||
|
||||
# Или с параметрами
|
||||
sudo ./scripts/docker-install.sh \
|
||||
--mode=controller \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID" \
|
||||
--port=8080
|
||||
```
|
||||
|
||||
### Agent в Docker
|
||||
|
||||
```bash
|
||||
# Интерактивная установка
|
||||
sudo ./scripts/docker-install.sh --mode=agent
|
||||
|
||||
# Или с параметрами
|
||||
sudo ./scripts/docker-install.sh \
|
||||
--mode=agent \
|
||||
--controller-url="https://controller.example.com:8080" \
|
||||
--agent-token="AGENT_TOKEN"
|
||||
```
|
||||
|
||||
## Требования системы
|
||||
|
||||
### Минимальные требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **RAM**: 512 MB
|
||||
- **Диск**: 1 GB свободного места
|
||||
- **Сеть**: Доступ в интернет для Telegram API
|
||||
|
||||
### Рекомендуемые требования
|
||||
|
||||
- **ОС**: Ubuntu 22.04 LTS
|
||||
- **Python**: 3.11+
|
||||
- **RAM**: 2 GB
|
||||
- **Диск**: 5 GB свободного места
|
||||
- **CPU**: 2 ядра
|
||||
|
||||
### Зависимости
|
||||
|
||||
- `iptables` или `nftables`
|
||||
- `systemd`
|
||||
- `python3-pip`
|
||||
- `sqlite3`
|
||||
|
||||
## Конфигурация после установки
|
||||
|
||||
### 1. Настройка Telegram бота
|
||||
|
||||
```bash
|
||||
# Создайте бота у @BotFather
|
||||
# Получите токен и ваш chat ID у @userinfobot
|
||||
|
||||
# Обновите конфигурацию
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
### 2. Настройка firewall
|
||||
|
||||
```bash
|
||||
# Для контроллера - откройте API порт
|
||||
sudo ufw allow 8080
|
||||
|
||||
# Для всех режимов - убедитесь что SSH доступен
|
||||
sudo ufw allow ssh
|
||||
```
|
||||
|
||||
### 3. Проверка установки
|
||||
|
||||
```bash
|
||||
# Статус сервиса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Проверка конфигурации
|
||||
sudo /opt/pyguardian/venv/bin/python /opt/pyguardian/main.py --check-config
|
||||
```
|
||||
|
||||
## Управление службой
|
||||
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключение автозапуска
|
||||
sudo systemctl disable pyguardian
|
||||
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## Обновление
|
||||
|
||||
### Standalone/Controller/Agent
|
||||
|
||||
```bash
|
||||
# Остановка службы
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновление кода
|
||||
cd /opt/pyguardian
|
||||
sudo git pull origin main
|
||||
|
||||
# Обновление зависимостей
|
||||
sudo -u pyguardian /opt/pyguardian/venv/bin/pip install -r requirements.txt
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Docker
|
||||
|
||||
```bash
|
||||
# Переход в директорию установки
|
||||
cd /path/to/pyguardian-docker
|
||||
|
||||
# Остановка контейнеров
|
||||
sudo docker-compose down
|
||||
|
||||
# Обновление образов
|
||||
sudo docker-compose pull
|
||||
|
||||
# Пересборка и запуск
|
||||
sudo docker-compose up --build -d
|
||||
```
|
||||
|
||||
## Удаление
|
||||
|
||||
### Полное удаление системы
|
||||
|
||||
```bash
|
||||
# Остановка и отключение службы
|
||||
sudo systemctl stop pyguardian
|
||||
sudo systemctl disable pyguardian
|
||||
|
||||
# Удаление файлов службы
|
||||
sudo rm -f /etc/systemd/system/pyguardian.service
|
||||
sudo systemctl daemon-reload
|
||||
|
||||
# Удаление приложения
|
||||
sudo rm -rf /opt/pyguardian
|
||||
|
||||
# Удаление конфигурации (опционально)
|
||||
sudo rm -rf /etc/pyguardian
|
||||
|
||||
# Удаление данных (опционально)
|
||||
sudo rm -rf /var/lib/pyguardian
|
||||
|
||||
# Удаление логов (опционально)
|
||||
sudo rm -rf /var/log/pyguardian
|
||||
|
||||
# Удаление пользователя
|
||||
sudo userdel -r pyguardian
|
||||
```
|
||||
|
||||
### Удаление Docker установки
|
||||
|
||||
```bash
|
||||
# Остановка и удаление контейнеров
|
||||
sudo docker-compose down -v
|
||||
|
||||
# Удаление образов
|
||||
sudo docker rmi $(sudo docker images pyguardian* -q)
|
||||
|
||||
# Удаление файлов установки
|
||||
sudo rm -rf /path/to/pyguardian-docker
|
||||
```
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
### Проблемы с правами
|
||||
|
||||
```bash
|
||||
# Проверка прав файлов
|
||||
sudo chown -R pyguardian:pyguardian /opt/pyguardian
|
||||
sudo chown -R pyguardian:pyguardian /var/lib/pyguardian
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
### Проблемы с Python
|
||||
|
||||
```bash
|
||||
# Проверка версии Python
|
||||
python3 --version
|
||||
|
||||
# Переустановка зависимостей
|
||||
sudo -u pyguardian /opt/pyguardian/venv/bin/pip install --force-reinstall -r /opt/pyguardian/requirements.txt
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
```bash
|
||||
# Проверка iptables
|
||||
sudo iptables -L PyGuardian -n
|
||||
|
||||
# Проверка nftables
|
||||
sudo nft list table inet pyguardian
|
||||
|
||||
# Сброс правил (осторожно!)
|
||||
sudo systemctl stop pyguardian
|
||||
sudo iptables -F PyGuardian
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
```bash
|
||||
# Проверка токена бота
|
||||
curl "https://api.telegram.org/bot<YOUR_TOKEN>/getMe"
|
||||
|
||||
# Проверка конфигурации
|
||||
grep -A5 "telegram:" /etc/pyguardian/config.yaml
|
||||
```
|
||||
@@ -1,357 +0,0 @@
|
||||
# PyGuardian Installation Guide
|
||||
|
||||
## Обзор
|
||||
|
||||
PyGuardian предлагает несколько способов установки в зависимости от ваших потребностей:
|
||||
|
||||
1. **Standalone** - Автономный сервер (все в одном)
|
||||
2. **Controller** - Центральный контроллер кластера
|
||||
3. **Agent** - Агент для подключения к контроллеру
|
||||
4. **Docker** - Контейнеризованное развертывание
|
||||
|
||||
## Быстрая установка
|
||||
|
||||
### Использование make
|
||||
|
||||
```bash
|
||||
# Клонирование репозитория
|
||||
git clone https://github.com/your-repo/PyGuardian.git
|
||||
cd PyGuardian
|
||||
|
||||
# Интерактивная установка
|
||||
sudo make install
|
||||
|
||||
# Или быстрая автономная установка
|
||||
sudo make standalone
|
||||
|
||||
# Или контроллер кластера
|
||||
sudo make controller
|
||||
|
||||
# Или агент кластера
|
||||
sudo make agent
|
||||
```
|
||||
|
||||
### Использование install.sh
|
||||
|
||||
```bash
|
||||
# Интерактивный режим
|
||||
sudo ./install.sh
|
||||
|
||||
# Быстрая установка
|
||||
sudo ./install.sh --quick
|
||||
|
||||
# Конкретный режим
|
||||
sudo ./install.sh --interactive
|
||||
sudo ./install.sh --docker
|
||||
```
|
||||
|
||||
## Подробная установка
|
||||
|
||||
### 1. Standalone режим
|
||||
|
||||
**Назначение**: Полнофункциональная система на одном сервере
|
||||
**Подходит для**: Небольшие инфраструктуры, тестирования
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make standalone
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=standalone --non-interactive \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Мониторинг auth.log
|
||||
- Telegram бот управления
|
||||
- Stealth security система
|
||||
- Firewall интеграция
|
||||
- Автоматическое управление паролями
|
||||
- SSH session management
|
||||
|
||||
### 2. Controller режим
|
||||
|
||||
**Назначение**: Центральный контроллер для управления кластером агентов
|
||||
**Подходит для**: Крупные инфраструктуры, централизованное управление
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make controller
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=controller --non-interactive \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Все функции Standalone
|
||||
- API для управления агентами
|
||||
- Веб-интерфейс управления
|
||||
- Централизованная отчетность
|
||||
- Автоматическое развертывание агентов
|
||||
|
||||
**После установки**:
|
||||
1. Откройте порт 8080 в firewall
|
||||
2. Настройте SSL сертификат
|
||||
3. Добавьте агенты через Telegram команды
|
||||
|
||||
### 3. Agent режим
|
||||
|
||||
**Назначение**: Агент для подключения к контроллеру
|
||||
**Подходит для**: Серверы в составе кластера
|
||||
|
||||
```bash
|
||||
# Автоматическая установка
|
||||
sudo make agent
|
||||
|
||||
# Или вручную
|
||||
sudo ./scripts/install.sh --mode=agent --non-interactive \
|
||||
--controller-url="https://controller.example.com:8080" \
|
||||
--agent-token="AGENT_TOKEN"
|
||||
```
|
||||
|
||||
**Что включено**:
|
||||
- Локальный мониторинг auth.log
|
||||
- Firewall управление
|
||||
- Подключение к контроллеру
|
||||
- Передача данных в центр
|
||||
|
||||
**Перед установкой**:
|
||||
1. Получите токен агента от администратора контроллера
|
||||
2. Убедитесь в доступности контроллера по сети
|
||||
|
||||
## Docker установка
|
||||
|
||||
### Controller в Docker
|
||||
|
||||
```bash
|
||||
# Интерактивная установка
|
||||
sudo ./scripts/docker-install.sh --mode=controller
|
||||
|
||||
# Или с параметрами
|
||||
sudo ./scripts/docker-install.sh \
|
||||
--mode=controller \
|
||||
--telegram-token="YOUR_BOT_TOKEN" \
|
||||
--admin-id="YOUR_TELEGRAM_ID" \
|
||||
--port=8080
|
||||
```
|
||||
|
||||
### Agent в Docker
|
||||
|
||||
```bash
|
||||
# Интерактивная установка
|
||||
sudo ./scripts/docker-install.sh --mode=agent
|
||||
|
||||
# Или с параметрами
|
||||
sudo ./scripts/docker-install.sh \
|
||||
--mode=agent \
|
||||
--controller-url="https://controller.example.com:8080" \
|
||||
--agent-token="AGENT_TOKEN"
|
||||
```
|
||||
|
||||
## Требования системы
|
||||
|
||||
### Минимальные требования
|
||||
|
||||
- **ОС**: Linux (Ubuntu 20.04+, CentOS 8+, Debian 11+)
|
||||
- **Python**: 3.10 или выше
|
||||
- **RAM**: 512 MB
|
||||
- **Диск**: 1 GB свободного места
|
||||
- **Сеть**: Доступ в интернет для Telegram API
|
||||
|
||||
### Рекомендуемые требования
|
||||
|
||||
- **ОС**: Ubuntu 22.04 LTS
|
||||
- **Python**: 3.11+
|
||||
- **RAM**: 2 GB
|
||||
- **Диск**: 5 GB свободного места
|
||||
- **CPU**: 2 ядра
|
||||
|
||||
### Зависимости
|
||||
|
||||
- `iptables` или `nftables`
|
||||
- `systemd`
|
||||
- `python3-pip`
|
||||
- `sqlite3`
|
||||
|
||||
## Конфигурация после установки
|
||||
|
||||
### 1. Настройка Telegram бота
|
||||
|
||||
```bash
|
||||
# Создайте бота у @BotFather
|
||||
# Получите токен и ваш chat ID у @userinfobot
|
||||
|
||||
# Обновите конфигурацию
|
||||
sudo nano /etc/pyguardian/config.yaml
|
||||
```
|
||||
|
||||
### 2. Настройка firewall
|
||||
|
||||
```bash
|
||||
# Для контроллера - откройте API порт
|
||||
sudo ufw allow 8080
|
||||
|
||||
# Для всех режимов - убедитесь что SSH доступен
|
||||
sudo ufw allow ssh
|
||||
```
|
||||
|
||||
### 3. Проверка установки
|
||||
|
||||
```bash
|
||||
# Статус сервиса
|
||||
sudo systemctl status pyguardian
|
||||
|
||||
# Просмотр логов
|
||||
sudo journalctl -u pyguardian -f
|
||||
|
||||
# Проверка конфигурации
|
||||
sudo /opt/pyguardian/venv/bin/python /opt/pyguardian/main.py --check-config
|
||||
```
|
||||
|
||||
## Управление службой
|
||||
|
||||
```bash
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
|
||||
# Остановка
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Перезапуск
|
||||
sudo systemctl restart pyguardian
|
||||
|
||||
# Автозапуск
|
||||
sudo systemctl enable pyguardian
|
||||
|
||||
# Отключение автозапуска
|
||||
sudo systemctl disable pyguardian
|
||||
|
||||
# Статус
|
||||
sudo systemctl status pyguardian
|
||||
```
|
||||
|
||||
## Обновление
|
||||
|
||||
### Standalone/Controller/Agent
|
||||
|
||||
```bash
|
||||
# Остановка службы
|
||||
sudo systemctl stop pyguardian
|
||||
|
||||
# Обновление кода
|
||||
cd /opt/pyguardian
|
||||
sudo git pull origin main
|
||||
|
||||
# Обновление зависимостей
|
||||
sudo -u pyguardian /opt/pyguardian/venv/bin/pip install -r requirements.txt
|
||||
|
||||
# Запуск
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Docker
|
||||
|
||||
```bash
|
||||
# Переход в директорию установки
|
||||
cd /path/to/pyguardian-docker
|
||||
|
||||
# Остановка контейнеров
|
||||
sudo docker-compose down
|
||||
|
||||
# Обновление образов
|
||||
sudo docker-compose pull
|
||||
|
||||
# Пересборка и запуск
|
||||
sudo docker-compose up --build -d
|
||||
```
|
||||
|
||||
## Удаление
|
||||
|
||||
### Полное удаление системы
|
||||
|
||||
```bash
|
||||
# Остановка и отключение службы
|
||||
sudo systemctl stop pyguardian
|
||||
sudo systemctl disable pyguardian
|
||||
|
||||
# Удаление файлов службы
|
||||
sudo rm -f /etc/systemd/system/pyguardian.service
|
||||
sudo systemctl daemon-reload
|
||||
|
||||
# Удаление приложения
|
||||
sudo rm -rf /opt/pyguardian
|
||||
|
||||
# Удаление конфигурации (опционально)
|
||||
sudo rm -rf /etc/pyguardian
|
||||
|
||||
# Удаление данных (опционально)
|
||||
sudo rm -rf /var/lib/pyguardian
|
||||
|
||||
# Удаление логов (опционально)
|
||||
sudo rm -rf /var/log/pyguardian
|
||||
|
||||
# Удаление пользователя
|
||||
sudo userdel -r pyguardian
|
||||
```
|
||||
|
||||
### Удаление Docker установки
|
||||
|
||||
```bash
|
||||
# Остановка и удаление контейнеров
|
||||
sudo docker-compose down -v
|
||||
|
||||
# Удаление образов
|
||||
sudo docker rmi $(sudo docker images pyguardian* -q)
|
||||
|
||||
# Удаление файлов установки
|
||||
sudo rm -rf /path/to/pyguardian-docker
|
||||
```
|
||||
|
||||
## Troubleshooting
|
||||
|
||||
### Проблемы с правами
|
||||
|
||||
```bash
|
||||
# Проверка прав файлов
|
||||
sudo chown -R pyguardian:pyguardian /opt/pyguardian
|
||||
sudo chown -R pyguardian:pyguardian /var/lib/pyguardian
|
||||
sudo chmod +x /opt/pyguardian/main.py
|
||||
```
|
||||
|
||||
### Проблемы с Python
|
||||
|
||||
```bash
|
||||
# Проверка версии Python
|
||||
python3 --version
|
||||
|
||||
# Переустановка зависимостей
|
||||
sudo -u pyguardian /opt/pyguardian/venv/bin/pip install --force-reinstall -r /opt/pyguardian/requirements.txt
|
||||
```
|
||||
|
||||
### Проблемы с firewall
|
||||
|
||||
```bash
|
||||
# Проверка iptables
|
||||
sudo iptables -L PyGuardian -n
|
||||
|
||||
# Проверка nftables
|
||||
sudo nft list table inet pyguardian
|
||||
|
||||
# Сброс правил (осторожно!)
|
||||
sudo systemctl stop pyguardian
|
||||
sudo iptables -F PyGuardian
|
||||
sudo systemctl start pyguardian
|
||||
```
|
||||
|
||||
### Проблемы с Telegram
|
||||
|
||||
```bash
|
||||
# Проверка токена бота
|
||||
curl "https://api.telegram.org/bot<YOUR_TOKEN>/getMe"
|
||||
|
||||
# Проверка конфигурации
|
||||
grep -A5 "telegram:" /etc/pyguardian/config.yaml
|
||||
```
|
||||
@@ -1,345 +0,0 @@
|
||||
# Управление кластером PyGuardian
|
||||
|
||||
## 🏢 Централизованное развертывание агентов
|
||||
|
||||
PyGuardian поддерживает централизованное управление кластером серверов через Telegram бот. Мастер-сервер может автоматически развертывать и управлять агентами на удаленных серверах.
|
||||
|
||||
## 🎯 Возможности кластера
|
||||
|
||||
### Основные функции:
|
||||
- **Автоматическое развертывание**: Установка PyGuardian на удаленные серверы
|
||||
- **Централизованное управление**: Контроль всех агентов через один Telegram бот
|
||||
- **Мониторинг статуса**: Проверка состояния агентов в реальном времени
|
||||
- **SSH интеграция**: Безопасное подключение через SSH ключи или пароли
|
||||
- **Автоматическая очистка**: Удаление агентов с очисткой удаленных серверов
|
||||
|
||||
### Архитектура:
|
||||
```
|
||||
[Master Server] ──SSH──┐
|
||||
├── [Agent Server 1]
|
||||
├── [Agent Server 2]
|
||||
├── [Agent Server 3]
|
||||
└── [Agent Server N]
|
||||
```
|
||||
|
||||
## 🚀 Быстрый старт
|
||||
|
||||
### 1. Настройка мастер-сервера
|
||||
|
||||
Убедитесь что в `config/config.yaml` указано:
|
||||
```yaml
|
||||
cluster:
|
||||
cluster_name: "MyCompany-Security"
|
||||
master_server: true
|
||||
ssh_timeout: 30
|
||||
ssh_retries: 3
|
||||
```
|
||||
|
||||
### 2. Подготовка SSH доступа
|
||||
|
||||
#### Вариант A: SSH ключи (рекомендуется)
|
||||
```bash
|
||||
# Генерация ключей
|
||||
ssh-keygen -t rsa -b 4096 -f ~/.ssh/pyguardian_cluster
|
||||
|
||||
# Копирование на целевой сервер
|
||||
ssh-copy-id -i ~/.ssh/pyguardian_cluster.pub root@192.168.1.100
|
||||
```
|
||||
|
||||
#### Вариант B: Пароли (менее безопасно)
|
||||
Используется для первоначальной настройки или тестирования.
|
||||
|
||||
### 3. Добавление серверов
|
||||
|
||||
```
|
||||
/add_server web-01 192.168.1.100
|
||||
/add_server web-02 192.168.1.101 ubuntu 2222
|
||||
/add_server db-01 192.168.1.200
|
||||
```
|
||||
|
||||
### 4. Развертывание агентов
|
||||
|
||||
```
|
||||
/deploy_agent web-01-192-168-1-100
|
||||
/deploy_agent web-02-192-168-1-101
|
||||
/deploy_agent db-01-192-168-1-200
|
||||
```
|
||||
|
||||
### 5. Мониторинг кластера
|
||||
|
||||
```
|
||||
/cluster # Общая информация
|
||||
/agents # Список агентов
|
||||
/check_agents # Проверка статуса
|
||||
```
|
||||
|
||||
## 📋 Команды управления кластером
|
||||
|
||||
### Основные команды
|
||||
|
||||
| Команда | Описание | Пример |
|
||||
|---------|----------|--------|
|
||||
| `/cluster` | Информация о кластере | `/cluster` |
|
||||
| `/agents` | Список всех агентов | `/agents` |
|
||||
| `/add_server` | Добавить сервер | `/add_server web-01 192.168.1.100` |
|
||||
| `/remove_server` | Удалить сервер | `/remove_server web-01-192-168-1-100` |
|
||||
| `/deploy_agent` | Развернуть агент | `/deploy_agent web-01-192-168-1-100` |
|
||||
| `/check_agents` | Проверить статусы | `/check_agents` |
|
||||
|
||||
### Детальные примеры
|
||||
|
||||
#### Добавление сервера
|
||||
```
|
||||
# Базовое добавление (root, порт 22)
|
||||
/add_server web-server 192.168.1.100
|
||||
|
||||
# С кастомным пользователем
|
||||
/add_server app-server 10.0.0.50 ubuntu
|
||||
|
||||
# С кастомным портом
|
||||
/add_server db-server 192.168.1.200 postgres 2222
|
||||
|
||||
# Полная форма
|
||||
/add_server api-server 172.16.0.100 deploy 2200
|
||||
```
|
||||
|
||||
#### Развертывание агента
|
||||
```
|
||||
# Обычное развертывание
|
||||
/deploy_agent web-server-192-168-1-100
|
||||
|
||||
# Принудительная переустановка
|
||||
/deploy_agent web-server-192-168-1-100 force
|
||||
```
|
||||
|
||||
#### Удаление сервера
|
||||
```
|
||||
# Простое удаление (агент остается)
|
||||
/remove_server web-server-192-168-1-100
|
||||
|
||||
# С очисткой удаленного сервера
|
||||
/remove_server web-server-192-168-1-100 cleanup
|
||||
```
|
||||
|
||||
## 🔧 Конфигурация
|
||||
|
||||
### Настройки кластера в config/config.yaml
|
||||
|
||||
```yaml
|
||||
cluster:
|
||||
cluster_name: "Production-Cluster" # Название кластера
|
||||
master_server: true # Мастер-сервер
|
||||
agents_config_path: "/var/lib/pyguardian/agents.yaml"
|
||||
deployment_path: "/opt/pyguardian"
|
||||
ssh_timeout: 30 # Таймаут SSH (секунды)
|
||||
ssh_retries: 3 # Попытки подключения
|
||||
|
||||
# SSH ключи по умолчанию (опционально)
|
||||
ssh:
|
||||
default_key_path: "/root/.ssh/pyguardian_cluster"
|
||||
default_user: "root"
|
||||
default_port: 22
|
||||
```
|
||||
|
||||
### Файл агентов agents.yaml
|
||||
|
||||
```yaml
|
||||
cluster:
|
||||
name: "Production-Cluster"
|
||||
master_server: true
|
||||
last_updated: "2024-11-25T15:30:00"
|
||||
|
||||
agents:
|
||||
web-01-192-168-1-100:
|
||||
hostname: "web-01"
|
||||
ip_address: "192.168.1.100"
|
||||
ssh_port: 22
|
||||
ssh_user: "root"
|
||||
ssh_key_path: "/root/.ssh/pyguardian_cluster"
|
||||
status: "online"
|
||||
last_check: "2024-11-25T15:25:00"
|
||||
version: "1.0.0"
|
||||
|
||||
api-server-172-16-0-100:
|
||||
hostname: "api-server"
|
||||
ip_address: "172.16.0.100"
|
||||
ssh_port: 2200
|
||||
ssh_user: "deploy"
|
||||
status: "deployed"
|
||||
last_check: null
|
||||
version: null
|
||||
```
|
||||
|
||||
## 🛡️ Безопасность кластера
|
||||
|
||||
### Рекомендации по безопасности:
|
||||
|
||||
1. **SSH ключи**: Всегда используйте SSH ключи вместо паролей
|
||||
2. **Ограниченные права**: Создайте отдельного пользователя для развертывания
|
||||
3. **Файрвол**: Ограничьте SSH доступ только с мастер-сервера
|
||||
4. **Мониторинг**: Регулярно проверяйте статус агентов
|
||||
5. **Обновления**: Следите за обновлениями PyGuardian
|
||||
|
||||
### Настройка пользователя для развертывания:
|
||||
|
||||
```bash
|
||||
# На целевом сервере
|
||||
useradd -m -s /bin/bash pyguardian-deploy
|
||||
usermod -aG sudo pyguardian-deploy
|
||||
|
||||
# Настройка sudoers
|
||||
echo 'pyguardian-deploy ALL=(ALL) NOPASSWD: ALL' > /etc/sudoers.d/pyguardian-deploy
|
||||
|
||||
# Копирование SSH ключа
|
||||
mkdir /home/pyguardian-deploy/.ssh
|
||||
cp /root/.ssh/authorized_keys /home/pyguardian-deploy/.ssh/
|
||||
chown -R pyguardian-deploy:pyguardian-deploy /home/pyguardian-deploy/.ssh
|
||||
chmod 700 /home/pyguardian-deploy/.ssh
|
||||
chmod 600 /home/pyguardian-deploy/.ssh/authorized_keys
|
||||
```
|
||||
|
||||
## 🚨 Устранение неполадок
|
||||
|
||||
### Частые проблемы:
|
||||
|
||||
#### Ошибка SSH соединения
|
||||
```
|
||||
❌ Не удалось подключиться к серверу: Connection refused
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Проверьте доступность сервера: `ping 192.168.1.100`
|
||||
2. Проверьте SSH сервис: `ssh root@192.168.1.100`
|
||||
3. Проверьте порт SSH: `nmap -p 22 192.168.1.100`
|
||||
|
||||
#### Ошибка прав доступа
|
||||
```
|
||||
❌ Ошибка установки: Permission denied
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Убедитесь что пользователь имеет права sudo
|
||||
2. Проверьте настройки sudoers
|
||||
3. Попробуйте от root пользователя
|
||||
|
||||
#### Агент не запускается
|
||||
```
|
||||
🔴 service_status: failed
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Проверьте логи: `journalctl -u pyguardian-agent -f`
|
||||
2. Проверьте конфигурацию агента
|
||||
3. Переустановите агент: `/deploy_agent agent-id force`
|
||||
|
||||
### Команды диагностики:
|
||||
|
||||
```bash
|
||||
# На мастер-сервере
|
||||
tail -f /var/log/pyguardian.log
|
||||
|
||||
# На агенте
|
||||
systemctl status pyguardian-agent
|
||||
journalctl -u pyguardian-agent -f
|
||||
cat /var/log/pyguardian-agent.log
|
||||
```
|
||||
|
||||
## 📊 Мониторинг кластера
|
||||
|
||||
### Telegram уведомления
|
||||
|
||||
PyGuardian автоматически отправляет уведомления о:
|
||||
- Добавлении новых агентов
|
||||
- Успешном развертывании
|
||||
- Изменении статуса агентов
|
||||
- Ошибках подключения
|
||||
|
||||
### Пример вывода команд:
|
||||
|
||||
#### /cluster
|
||||
```
|
||||
🏢 Кластер Production-Cluster
|
||||
|
||||
📊 Статистика:
|
||||
• Всего агентов: 5
|
||||
• Онлайн: 4
|
||||
• Оффлайн: 1
|
||||
• Развернуто: 5
|
||||
|
||||
🖥️ Агенты:
|
||||
🟢 web-01 (192.168.1.100)
|
||||
🟢 web-02 (192.168.1.101)
|
||||
🔴 db-01 (192.168.1.200)
|
||||
🟢 api-server (172.16.0.100)
|
||||
🟢 cache-01 (10.0.0.50)
|
||||
|
||||
🕐 Последнее обновление: 2024-11-25 15:30:45
|
||||
```
|
||||
|
||||
#### /check_agents
|
||||
```
|
||||
🔍 Результаты проверки агентов
|
||||
|
||||
📊 Статистика:
|
||||
• Проверено: 5
|
||||
• Онлайн: 4
|
||||
• Оффлайн: 1
|
||||
• Ошибки: 0
|
||||
|
||||
📋 Детали:
|
||||
🟢 web-01: active
|
||||
🟢 web-02: active
|
||||
🔴 db-01: inactive
|
||||
🟢 api-server: active
|
||||
🟢 cache-01: active
|
||||
|
||||
🕐 Время проверки: 15:32:10
|
||||
```
|
||||
|
||||
## 🔄 Автоматизация
|
||||
|
||||
### Скрипты автоматизации
|
||||
|
||||
Создайте скрипты для автоматического управления кластером:
|
||||
|
||||
```bash
|
||||
#!/bin/bash
|
||||
# auto-deploy.sh - Автоматическое развертывание на список серверов
|
||||
|
||||
SERVERS=(
|
||||
"web-01 192.168.1.100"
|
||||
"web-02 192.168.1.101"
|
||||
"api-01 172.16.0.100"
|
||||
)
|
||||
|
||||
for server in "${SERVERS[@]}"; do
|
||||
hostname=$(echo $server | cut -d' ' -f1)
|
||||
ip=$(echo $server | cut -d' ' -f2)
|
||||
|
||||
echo "Добавляю $hostname ($ip)..."
|
||||
# Здесь может быть API вызов или автоматизация через expect
|
||||
done
|
||||
```
|
||||
|
||||
### Интеграция с CI/CD
|
||||
|
||||
PyGuardian кластер может быть интегрирован с CI/CD пайплайнами для автоматического развертывания защиты на новые серверы.
|
||||
|
||||
## 📈 Масштабирование
|
||||
|
||||
### Рекомендации по масштабированию:
|
||||
|
||||
- **До 10 серверов**: Один мастер-сервер
|
||||
- **10-50 серверов**: Мастер + резервный мастер
|
||||
- **50+ серверов**: Распределенная архитектура
|
||||
|
||||
### Мониторинг производительности:
|
||||
|
||||
- Время отклика SSH соединений
|
||||
- Использование ресурсов мастер-сервера
|
||||
- Скорость развертывания агентов
|
||||
- Частота проверки статуса
|
||||
|
||||
---
|
||||
|
||||
*Данная документация покрывает основные возможности управления кластером PyGuardian. Для дополнительной помощи обращайтесь к основной документации проекта.*
|
||||
@@ -1,345 +0,0 @@
|
||||
# Управление кластером PyGuardian
|
||||
|
||||
## 🏢 Централизованное развертывание агентов
|
||||
|
||||
PyGuardian поддерживает централизованное управление кластером серверов через Telegram бот. Мастер-сервер может автоматически развертывать и управлять агентами на удаленных серверах.
|
||||
|
||||
## 🎯 Возможности кластера
|
||||
|
||||
### Основные функции:
|
||||
- **Автоматическое развертывание**: Установка PyGuardian на удаленные серверы
|
||||
- **Централизованное управление**: Контроль всех агентов через один Telegram бот
|
||||
- **Мониторинг статуса**: Проверка состояния агентов в реальном времени
|
||||
- **SSH интеграция**: Безопасное подключение через SSH ключи или пароли
|
||||
- **Автоматическая очистка**: Удаление агентов с очисткой удаленных серверов
|
||||
|
||||
### Архитектура:
|
||||
```
|
||||
[Master Server] ──SSH──┐
|
||||
├── [Agent Server 1]
|
||||
├── [Agent Server 2]
|
||||
├── [Agent Server 3]
|
||||
└── [Agent Server N]
|
||||
```
|
||||
|
||||
## 🚀 Быстрый старт
|
||||
|
||||
### 1. Настройка мастер-сервера
|
||||
|
||||
Убедитесь что в `config/config.yaml` указано:
|
||||
```yaml
|
||||
cluster:
|
||||
cluster_name: "MyCompany-Security"
|
||||
master_server: true
|
||||
ssh_timeout: 30
|
||||
ssh_retries: 3
|
||||
```
|
||||
|
||||
### 2. Подготовка SSH доступа
|
||||
|
||||
#### Вариант A: SSH ключи (рекомендуется)
|
||||
```bash
|
||||
# Генерация ключей
|
||||
ssh-keygen -t rsa -b 4096 -f ~/.ssh/pyguardian_cluster
|
||||
|
||||
# Копирование на целевой сервер
|
||||
ssh-copy-id -i ~/.ssh/pyguardian_cluster.pub root@192.168.1.100
|
||||
```
|
||||
|
||||
#### Вариант B: Пароли (менее безопасно)
|
||||
Используется для первоначальной настройки или тестирования.
|
||||
|
||||
### 3. Добавление серверов
|
||||
|
||||
```
|
||||
/add_server web-01 192.168.1.100
|
||||
/add_server web-02 192.168.1.101 ubuntu 2222
|
||||
/add_server db-01 192.168.1.200
|
||||
```
|
||||
|
||||
### 4. Развертывание агентов
|
||||
|
||||
```
|
||||
/deploy_agent web-01-192-168-1-100
|
||||
/deploy_agent web-02-192-168-1-101
|
||||
/deploy_agent db-01-192-168-1-200
|
||||
```
|
||||
|
||||
### 5. Мониторинг кластера
|
||||
|
||||
```
|
||||
/cluster # Общая информация
|
||||
/agents # Список агентов
|
||||
/check_agents # Проверка статуса
|
||||
```
|
||||
|
||||
## 📋 Команды управления кластером
|
||||
|
||||
### Основные команды
|
||||
|
||||
| Команда | Описание | Пример |
|
||||
|---------|----------|--------|
|
||||
| `/cluster` | Информация о кластере | `/cluster` |
|
||||
| `/agents` | Список всех агентов | `/agents` |
|
||||
| `/add_server` | Добавить сервер | `/add_server web-01 192.168.1.100` |
|
||||
| `/remove_server` | Удалить сервер | `/remove_server web-01-192-168-1-100` |
|
||||
| `/deploy_agent` | Развернуть агент | `/deploy_agent web-01-192-168-1-100` |
|
||||
| `/check_agents` | Проверить статусы | `/check_agents` |
|
||||
|
||||
### Детальные примеры
|
||||
|
||||
#### Добавление сервера
|
||||
```
|
||||
# Базовое добавление (root, порт 22)
|
||||
/add_server web-server 192.168.1.100
|
||||
|
||||
# С кастомным пользователем
|
||||
/add_server app-server 10.0.0.50 ubuntu
|
||||
|
||||
# С кастомным портом
|
||||
/add_server db-server 192.168.1.200 postgres 2222
|
||||
|
||||
# Полная форма
|
||||
/add_server api-server 172.16.0.100 deploy 2200
|
||||
```
|
||||
|
||||
#### Развертывание агента
|
||||
```
|
||||
# Обычное развертывание
|
||||
/deploy_agent web-server-192-168-1-100
|
||||
|
||||
# Принудительная переустановка
|
||||
/deploy_agent web-server-192-168-1-100 force
|
||||
```
|
||||
|
||||
#### Удаление сервера
|
||||
```
|
||||
# Простое удаление (агент остается)
|
||||
/remove_server web-server-192-168-1-100
|
||||
|
||||
# С очисткой удаленного сервера
|
||||
/remove_server web-server-192-168-1-100 cleanup
|
||||
```
|
||||
|
||||
## 🔧 Конфигурация
|
||||
|
||||
### Настройки кластера в config/config.yaml
|
||||
|
||||
```yaml
|
||||
cluster:
|
||||
cluster_name: "Production-Cluster" # Название кластера
|
||||
master_server: true # Мастер-сервер
|
||||
agents_config_path: "/var/lib/pyguardian/agents.yaml"
|
||||
deployment_path: "/opt/pyguardian"
|
||||
ssh_timeout: 30 # Таймаут SSH (секунды)
|
||||
ssh_retries: 3 # Попытки подключения
|
||||
|
||||
# SSH ключи по умолчанию (опционально)
|
||||
ssh:
|
||||
default_key_path: "/root/.ssh/pyguardian_cluster"
|
||||
default_user: "root"
|
||||
default_port: 22
|
||||
```
|
||||
|
||||
### Файл агентов agents.yaml
|
||||
|
||||
```yaml
|
||||
cluster:
|
||||
name: "Production-Cluster"
|
||||
master_server: true
|
||||
last_updated: "2024-11-25T15:30:00"
|
||||
|
||||
agents:
|
||||
web-01-192-168-1-100:
|
||||
hostname: "web-01"
|
||||
ip_address: "192.168.1.100"
|
||||
ssh_port: 22
|
||||
ssh_user: "root"
|
||||
ssh_key_path: "/root/.ssh/pyguardian_cluster"
|
||||
status: "online"
|
||||
last_check: "2024-11-25T15:25:00"
|
||||
version: "1.0.0"
|
||||
|
||||
api-server-172-16-0-100:
|
||||
hostname: "api-server"
|
||||
ip_address: "172.16.0.100"
|
||||
ssh_port: 2200
|
||||
ssh_user: "deploy"
|
||||
status: "deployed"
|
||||
last_check: null
|
||||
version: null
|
||||
```
|
||||
|
||||
## 🛡️ Безопасность кластера
|
||||
|
||||
### Рекомендации по безопасности:
|
||||
|
||||
1. **SSH ключи**: Всегда используйте SSH ключи вместо паролей
|
||||
2. **Ограниченные права**: Создайте отдельного пользователя для развертывания
|
||||
3. **Файрвол**: Ограничьте SSH доступ только с мастер-сервера
|
||||
4. **Мониторинг**: Регулярно проверяйте статус агентов
|
||||
5. **Обновления**: Следите за обновлениями PyGuardian
|
||||
|
||||
### Настройка пользователя для развертывания:
|
||||
|
||||
```bash
|
||||
# На целевом сервере
|
||||
useradd -m -s /bin/bash pyguardian-deploy
|
||||
usermod -aG sudo pyguardian-deploy
|
||||
|
||||
# Настройка sudoers
|
||||
echo 'pyguardian-deploy ALL=(ALL) NOPASSWD: ALL' > /etc/sudoers.d/pyguardian-deploy
|
||||
|
||||
# Копирование SSH ключа
|
||||
mkdir /home/pyguardian-deploy/.ssh
|
||||
cp /root/.ssh/authorized_keys /home/pyguardian-deploy/.ssh/
|
||||
chown -R pyguardian-deploy:pyguardian-deploy /home/pyguardian-deploy/.ssh
|
||||
chmod 700 /home/pyguardian-deploy/.ssh
|
||||
chmod 600 /home/pyguardian-deploy/.ssh/authorized_keys
|
||||
```
|
||||
|
||||
## 🚨 Устранение неполадок
|
||||
|
||||
### Частые проблемы:
|
||||
|
||||
#### Ошибка SSH соединения
|
||||
```
|
||||
❌ Не удалось подключиться к серверу: Connection refused
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Проверьте доступность сервера: `ping 192.168.1.100`
|
||||
2. Проверьте SSH сервис: `ssh root@192.168.1.100`
|
||||
3. Проверьте порт SSH: `nmap -p 22 192.168.1.100`
|
||||
|
||||
#### Ошибка прав доступа
|
||||
```
|
||||
❌ Ошибка установки: Permission denied
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Убедитесь что пользователь имеет права sudo
|
||||
2. Проверьте настройки sudoers
|
||||
3. Попробуйте от root пользователя
|
||||
|
||||
#### Агент не запускается
|
||||
```
|
||||
🔴 service_status: failed
|
||||
```
|
||||
|
||||
**Решение:**
|
||||
1. Проверьте логи: `journalctl -u pyguardian-agent -f`
|
||||
2. Проверьте конфигурацию агента
|
||||
3. Переустановите агент: `/deploy_agent agent-id force`
|
||||
|
||||
### Команды диагностики:
|
||||
|
||||
```bash
|
||||
# На мастер-сервере
|
||||
tail -f /var/log/pyguardian.log
|
||||
|
||||
# На агенте
|
||||
systemctl status pyguardian-agent
|
||||
journalctl -u pyguardian-agent -f
|
||||
cat /var/log/pyguardian-agent.log
|
||||
```
|
||||
|
||||
## 📊 Мониторинг кластера
|
||||
|
||||
### Telegram уведомления
|
||||
|
||||
PyGuardian автоматически отправляет уведомления о:
|
||||
- Добавлении новых агентов
|
||||
- Успешном развертывании
|
||||
- Изменении статуса агентов
|
||||
- Ошибках подключения
|
||||
|
||||
### Пример вывода команд:
|
||||
|
||||
#### /cluster
|
||||
```
|
||||
🏢 Кластер Production-Cluster
|
||||
|
||||
📊 Статистика:
|
||||
• Всего агентов: 5
|
||||
• Онлайн: 4
|
||||
• Оффлайн: 1
|
||||
• Развернуто: 5
|
||||
|
||||
🖥️ Агенты:
|
||||
🟢 web-01 (192.168.1.100)
|
||||
🟢 web-02 (192.168.1.101)
|
||||
🔴 db-01 (192.168.1.200)
|
||||
🟢 api-server (172.16.0.100)
|
||||
🟢 cache-01 (10.0.0.50)
|
||||
|
||||
🕐 Последнее обновление: 2024-11-25 15:30:45
|
||||
```
|
||||
|
||||
#### /check_agents
|
||||
```
|
||||
🔍 Результаты проверки агентов
|
||||
|
||||
📊 Статистика:
|
||||
• Проверено: 5
|
||||
• Онлайн: 4
|
||||
• Оффлайн: 1
|
||||
• Ошибки: 0
|
||||
|
||||
📋 Детали:
|
||||
🟢 web-01: active
|
||||
🟢 web-02: active
|
||||
🔴 db-01: inactive
|
||||
🟢 api-server: active
|
||||
🟢 cache-01: active
|
||||
|
||||
🕐 Время проверки: 15:32:10
|
||||
```
|
||||
|
||||
## 🔄 Автоматизация
|
||||
|
||||
### Скрипты автоматизации
|
||||
|
||||
Создайте скрипты для автоматического управления кластером:
|
||||
|
||||
```bash
|
||||
#!/bin/bash
|
||||
# auto-deploy.sh - Автоматическое развертывание на список серверов
|
||||
|
||||
SERVERS=(
|
||||
"web-01 192.168.1.100"
|
||||
"web-02 192.168.1.101"
|
||||
"api-01 172.16.0.100"
|
||||
)
|
||||
|
||||
for server in "${SERVERS[@]}"; do
|
||||
hostname=$(echo $server | cut -d' ' -f1)
|
||||
ip=$(echo $server | cut -d' ' -f2)
|
||||
|
||||
echo "Добавляю $hostname ($ip)..."
|
||||
# Здесь может быть API вызов или автоматизация через expect
|
||||
done
|
||||
```
|
||||
|
||||
### Интеграция с CI/CD
|
||||
|
||||
PyGuardian кластер может быть интегрирован с CI/CD пайплайнами для автоматического развертывания защиты на новые серверы.
|
||||
|
||||
## 📈 Масштабирование
|
||||
|
||||
### Рекомендации по масштабированию:
|
||||
|
||||
- **До 10 серверов**: Один мастер-сервер
|
||||
- **10-50 серверов**: Мастер + резервный мастер
|
||||
- **50+ серверов**: Распределенная архитектура
|
||||
|
||||
### Мониторинг производительности:
|
||||
|
||||
- Время отклика SSH соединений
|
||||
- Использование ресурсов мастер-сервера
|
||||
- Скорость развертывания агентов
|
||||
- Частота проверки статуса
|
||||
|
||||
---
|
||||
|
||||
*Данная документация покрывает основные возможности управления кластером PyGuardian. Для дополнительной помощи обращайтесь к основной документации проекта.*
|
||||
@@ -1,373 +0,0 @@
|
||||
# PyGuardian Configuration Examples
|
||||
# Примеры конфигураций для различных режимов развертывания
|
||||
|
||||
#==========================================================================
|
||||
# 1. Standalone Configuration (Автономный режим)
|
||||
# config/config.yaml для одиночного сервера
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_users: [123456789]
|
||||
log_channel: "@security_logs"
|
||||
|
||||
security:
|
||||
session_timeout: 30
|
||||
max_failed_attempts: 3
|
||||
ban_duration: 300
|
||||
enable_2fa: true
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 10
|
||||
rate_limit:
|
||||
ssh: 5
|
||||
http: 100
|
||||
https: 100
|
||||
|
||||
storage:
|
||||
database_file: "data/pyguardian.db"
|
||||
backup_interval: 3600
|
||||
log_retention_days: 30
|
||||
|
||||
monitoring:
|
||||
check_interval: 60
|
||||
resource_alerts:
|
||||
cpu_threshold: 80
|
||||
memory_threshold: 85
|
||||
disk_threshold: 90
|
||||
|
||||
#==========================================================================
|
||||
# 2. Controller Configuration (Контроллер кластера)
|
||||
# config/config.yaml для центрального управляющего узла
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_users: [123456789]
|
||||
log_channel: "@cluster_logs"
|
||||
cluster_commands: true
|
||||
|
||||
security:
|
||||
session_timeout: 60
|
||||
max_failed_attempts: 5
|
||||
ban_duration: 600
|
||||
enable_2fa: true
|
||||
cluster_auth_key: "your-cluster-secret-key"
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 20
|
||||
rate_limit:
|
||||
ssh: 10
|
||||
http: 200
|
||||
https: 200
|
||||
|
||||
storage:
|
||||
database_file: "data/cluster_controller.db"
|
||||
backup_interval: 1800
|
||||
log_retention_days: 60
|
||||
|
||||
monitoring:
|
||||
check_interval: 30
|
||||
resource_alerts:
|
||||
cpu_threshold: 70
|
||||
memory_threshold: 80
|
||||
disk_threshold: 85
|
||||
|
||||
cluster:
|
||||
mode: "controller"
|
||||
controller_host: "0.0.0.0"
|
||||
controller_port: 8443
|
||||
api_secret: "your-api-secret-key"
|
||||
agent_timeout: 120
|
||||
deployment:
|
||||
ssh_key_path: "/root/.ssh/cluster_key"
|
||||
default_user: "root"
|
||||
installation_script: "/opt/pyguardian/scripts/install.sh"
|
||||
notifications:
|
||||
agent_offline_timeout: 300
|
||||
cluster_events: true
|
||||
health_check_interval: 60
|
||||
|
||||
#==========================================================================
|
||||
# 3. Agent Configuration (Агент кластера)
|
||||
# config/config.yaml для управляемого узла
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
# Agent не имеет собственного бота, управляется контроллером
|
||||
log_channel: "@agent_logs"
|
||||
|
||||
security:
|
||||
session_timeout: 30
|
||||
max_failed_attempts: 3
|
||||
ban_duration: 300
|
||||
enable_2fa: false
|
||||
cluster_auth_key: "your-cluster-secret-key"
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 10
|
||||
rate_limit:
|
||||
ssh: 5
|
||||
http: 100
|
||||
https: 100
|
||||
|
||||
storage:
|
||||
database_file: "data/agent.db"
|
||||
backup_interval: 3600
|
||||
log_retention_days: 30
|
||||
|
||||
monitoring:
|
||||
check_interval: 60
|
||||
resource_alerts:
|
||||
cpu_threshold: 85
|
||||
memory_threshold: 90
|
||||
disk_threshold: 95
|
||||
|
||||
cluster:
|
||||
mode: "agent"
|
||||
controller_host: "YOUR_CONTROLLER_IP"
|
||||
controller_port: 8443
|
||||
api_secret: "your-api-secret-key"
|
||||
agent_id: "auto" # Автоматически сгенерируется
|
||||
heartbeat_interval: 30
|
||||
report_interval: 60
|
||||
|
||||
#==========================================================================
|
||||
# 4. Docker Compose Configuration
|
||||
# docker-compose.yml для контейнеризированного развертывания
|
||||
#==========================================================================
|
||||
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build: .
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- ./data:/opt/pyguardian/data
|
||||
- ./config:/opt/pyguardian/config
|
||||
- ./logs:/opt/pyguardian/logs
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=controller
|
||||
command: ["python", "main.py", "--mode", "controller"]
|
||||
|
||||
pyguardian-agent:
|
||||
build: .
|
||||
container_name: pyguardian-agent
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- ./data:/opt/pyguardian/data
|
||||
- ./config:/opt/pyguardian/config
|
||||
- ./logs:/opt/pyguardian/logs
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=agent
|
||||
- CONTROLLER_HOST=your-controller-ip
|
||||
command: ["python", "main.py", "--mode", "agent"]
|
||||
depends_on:
|
||||
- pyguardian-controller
|
||||
|
||||
#==========================================================================
|
||||
# 5. Systemd Service Templates
|
||||
# /etc/systemd/system/pyguardian.service
|
||||
#==========================================================================
|
||||
|
||||
[Unit]
|
||||
Description=PyGuardian Security System
|
||||
After=network.target
|
||||
Wants=network-online.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=pyguardian
|
||||
Group=pyguardian
|
||||
WorkingDirectory=/opt/pyguardian
|
||||
ExecStart=/opt/pyguardian/venv/bin/python main.py
|
||||
ExecReload=/bin/kill -HUP $MAINPID
|
||||
Restart=always
|
||||
RestartSec=30
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
SyslogIdentifier=pyguardian
|
||||
|
||||
# Security settings
|
||||
NoNewPrivileges=yes
|
||||
PrivateTmp=yes
|
||||
ProtectSystem=strict
|
||||
ProtectHome=yes
|
||||
ReadWritePaths=/opt/pyguardian/data /opt/pyguardian/logs
|
||||
CapabilityBoundingSet=CAP_NET_ADMIN CAP_NET_RAW
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
|
||||
#==========================================================================
|
||||
# 6. Nginx Proxy Configuration (для веб-интерфейса)
|
||||
# /etc/nginx/sites-available/pyguardian
|
||||
#==========================================================================
|
||||
|
||||
server {
|
||||
listen 80;
|
||||
server_name your-domain.com;
|
||||
return 301 https://$server_name$request_uri;
|
||||
}
|
||||
|
||||
server {
|
||||
listen 443 ssl http2;
|
||||
server_name your-domain.com;
|
||||
|
||||
ssl_certificate /etc/letsencrypt/live/your-domain.com/fullchain.pem;
|
||||
ssl_certificate_key /etc/letsencrypt/live/your-domain.com/privkey.pem;
|
||||
|
||||
location / {
|
||||
proxy_pass http://127.0.0.1:8080;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
|
||||
# WebSocket support
|
||||
proxy_http_version 1.1;
|
||||
proxy_set_header Upgrade $http_upgrade;
|
||||
proxy_set_header Connection "upgrade";
|
||||
}
|
||||
|
||||
location /api/ {
|
||||
proxy_pass http://127.0.0.1:8443/;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
}
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# 7. Environment Variables (.env файл)
|
||||
#==========================================================================
|
||||
|
||||
# PyGuardian Environment Variables
|
||||
PYGUARDIAN_MODE=standalone
|
||||
PYGUARDIAN_CONFIG=/opt/pyguardian/config/config.yaml
|
||||
PYGUARDIAN_DATA_DIR=/opt/pyguardian/data
|
||||
PYGUARDIAN_LOG_LEVEL=INFO
|
||||
|
||||
# Telegram Configuration
|
||||
TELEGRAM_BOT_TOKEN=your_bot_token_here
|
||||
TELEGRAM_ADMIN_USERS=123456789,987654321
|
||||
|
||||
# Cluster Configuration (если используется)
|
||||
CLUSTER_CONTROLLER_HOST=your-controller-ip
|
||||
CLUSTER_CONTROLLER_PORT=8443
|
||||
CLUSTER_API_SECRET=your-api-secret
|
||||
CLUSTER_AUTH_KEY=your-cluster-auth-key
|
||||
|
||||
# Database Configuration
|
||||
DATABASE_URL=sqlite:///opt/pyguardian/data/pyguardian.db
|
||||
|
||||
# Security Settings
|
||||
ENABLE_2FA=true
|
||||
SESSION_TIMEOUT=30
|
||||
MAX_FAILED_ATTEMPTS=3
|
||||
|
||||
#==========================================================================
|
||||
# 8. Firewall Rules Examples (iptables)
|
||||
#==========================================================================
|
||||
|
||||
#!/bin/bash
|
||||
# PyGuardian Firewall Rules
|
||||
|
||||
# Очистка существующих правил
|
||||
iptables -F
|
||||
iptables -X
|
||||
iptables -Z
|
||||
|
||||
# Политики по умолчанию
|
||||
iptables -P INPUT DROP
|
||||
iptables -P FORWARD DROP
|
||||
iptables -P OUTPUT ACCEPT
|
||||
|
||||
# Разрешить loopback
|
||||
iptables -A INPUT -i lo -j ACCEPT
|
||||
iptables -A OUTPUT -o lo -j ACCEPT
|
||||
|
||||
# Разрешить установленные соединения
|
||||
iptables -A INPUT -m conntrack --ctstate ESTABLISHED,RELATED -j ACCEPT
|
||||
|
||||
# SSH (ограничить количество попыток)
|
||||
iptables -A INPUT -p tcp --dport 22 -m conntrack --ctstate NEW -m recent --set --name SSH
|
||||
iptables -A INPUT -p tcp --dport 22 -m conntrack --ctstate NEW -m recent --update --seconds 60 --hitcount 3 --name SSH -j DROP
|
||||
iptables -A INPUT -p tcp --dport 22 -j ACCEPT
|
||||
|
||||
# HTTP/HTTPS (с rate limiting)
|
||||
iptables -A INPUT -p tcp --dport 80 -m limit --limit 25/minute --limit-burst 100 -j ACCEPT
|
||||
iptables -A INPUT -p tcp --dport 443 -m limit --limit 25/minute --limit-burst 100 -j ACCEPT
|
||||
|
||||
# Cluster API (только от контроллера)
|
||||
iptables -A INPUT -p tcp --dport 8443 -s your-controller-ip -j ACCEPT
|
||||
|
||||
# DDoS Protection
|
||||
iptables -A INPUT -p tcp --syn -m limit --limit 1/s --limit-burst 3 -j ACCEPT
|
||||
iptables -A INPUT -p icmp -m limit --limit 1/s --limit-burst 1 -j ACCEPT
|
||||
|
||||
# Логирование отброшенных пакетов
|
||||
iptables -A INPUT -j LOG --log-prefix "DROPPED: " --log-level 4
|
||||
iptables -A INPUT -j DROP
|
||||
|
||||
#==========================================================================
|
||||
# 9. Monitoring Configuration (для интеграции с Grafana/Prometheus)
|
||||
#==========================================================================
|
||||
|
||||
# prometheus.yml
|
||||
global:
|
||||
scrape_interval: 15s
|
||||
|
||||
scrape_configs:
|
||||
- job_name: 'pyguardian'
|
||||
static_configs:
|
||||
- targets: ['localhost:9090']
|
||||
metrics_path: /metrics
|
||||
scrape_interval: 30s
|
||||
|
||||
- job_name: 'pyguardian-cluster'
|
||||
static_configs:
|
||||
- targets: ['controller-ip:8443']
|
||||
metrics_path: /cluster/metrics
|
||||
scrape_interval: 60s
|
||||
|
||||
#==========================================================================
|
||||
# 10. Backup Configuration
|
||||
#==========================================================================
|
||||
|
||||
#!/bin/bash
|
||||
# PyGuardian Backup Script
|
||||
|
||||
BACKUP_DIR="/opt/pyguardian/backups"
|
||||
DATA_DIR="/opt/pyguardian/data"
|
||||
CONFIG_DIR="/opt/pyguardian/config"
|
||||
LOG_DIR="/opt/pyguardian/logs"
|
||||
|
||||
DATE=$(date +%Y%m%d_%H%M%S)
|
||||
BACKUP_FILE="pyguardian_backup_${DATE}.tar.gz"
|
||||
|
||||
# Создать архив
|
||||
tar -czf "${BACKUP_DIR}/${BACKUP_FILE}" \
|
||||
"${DATA_DIR}" \
|
||||
"${CONFIG_DIR}" \
|
||||
"${LOG_DIR}"
|
||||
|
||||
# Оставить только последние 7 резервных копий
|
||||
find "${BACKUP_DIR}" -name "pyguardian_backup_*.tar.gz" -type f -mtime +7 -delete
|
||||
|
||||
echo "Backup completed: ${BACKUP_FILE}"
|
||||
@@ -1,373 +0,0 @@
|
||||
# PyGuardian Configuration Examples
|
||||
# Примеры конфигураций для различных режимов развертывания
|
||||
|
||||
#==========================================================================
|
||||
# 1. Standalone Configuration (Автономный режим)
|
||||
# config/config.yaml для одиночного сервера
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_users: [123456789]
|
||||
log_channel: "@security_logs"
|
||||
|
||||
security:
|
||||
session_timeout: 30
|
||||
max_failed_attempts: 3
|
||||
ban_duration: 300
|
||||
enable_2fa: true
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 10
|
||||
rate_limit:
|
||||
ssh: 5
|
||||
http: 100
|
||||
https: 100
|
||||
|
||||
storage:
|
||||
database_file: "data/pyguardian.db"
|
||||
backup_interval: 3600
|
||||
log_retention_days: 30
|
||||
|
||||
monitoring:
|
||||
check_interval: 60
|
||||
resource_alerts:
|
||||
cpu_threshold: 80
|
||||
memory_threshold: 85
|
||||
disk_threshold: 90
|
||||
|
||||
#==========================================================================
|
||||
# 2. Controller Configuration (Контроллер кластера)
|
||||
# config/config.yaml для центрального управляющего узла
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
bot_token: "YOUR_BOT_TOKEN_HERE"
|
||||
admin_users: [123456789]
|
||||
log_channel: "@cluster_logs"
|
||||
cluster_commands: true
|
||||
|
||||
security:
|
||||
session_timeout: 60
|
||||
max_failed_attempts: 5
|
||||
ban_duration: 600
|
||||
enable_2fa: true
|
||||
cluster_auth_key: "your-cluster-secret-key"
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 20
|
||||
rate_limit:
|
||||
ssh: 10
|
||||
http: 200
|
||||
https: 200
|
||||
|
||||
storage:
|
||||
database_file: "data/cluster_controller.db"
|
||||
backup_interval: 1800
|
||||
log_retention_days: 60
|
||||
|
||||
monitoring:
|
||||
check_interval: 30
|
||||
resource_alerts:
|
||||
cpu_threshold: 70
|
||||
memory_threshold: 80
|
||||
disk_threshold: 85
|
||||
|
||||
cluster:
|
||||
mode: "controller"
|
||||
controller_host: "0.0.0.0"
|
||||
controller_port: 8443
|
||||
api_secret: "your-api-secret-key"
|
||||
agent_timeout: 120
|
||||
deployment:
|
||||
ssh_key_path: "/root/.ssh/cluster_key"
|
||||
default_user: "root"
|
||||
installation_script: "/opt/pyguardian/scripts/install.sh"
|
||||
notifications:
|
||||
agent_offline_timeout: 300
|
||||
cluster_events: true
|
||||
health_check_interval: 60
|
||||
|
||||
#==========================================================================
|
||||
# 3. Agent Configuration (Агент кластера)
|
||||
# config/config.yaml для управляемого узла
|
||||
#==========================================================================
|
||||
|
||||
telegram:
|
||||
# Agent не имеет собственного бота, управляется контроллером
|
||||
log_channel: "@agent_logs"
|
||||
|
||||
security:
|
||||
session_timeout: 30
|
||||
max_failed_attempts: 3
|
||||
ban_duration: 300
|
||||
enable_2fa: false
|
||||
cluster_auth_key: "your-cluster-secret-key"
|
||||
|
||||
firewall:
|
||||
default_policy: "drop"
|
||||
enable_ddos_protection: true
|
||||
max_connections_per_ip: 10
|
||||
rate_limit:
|
||||
ssh: 5
|
||||
http: 100
|
||||
https: 100
|
||||
|
||||
storage:
|
||||
database_file: "data/agent.db"
|
||||
backup_interval: 3600
|
||||
log_retention_days: 30
|
||||
|
||||
monitoring:
|
||||
check_interval: 60
|
||||
resource_alerts:
|
||||
cpu_threshold: 85
|
||||
memory_threshold: 90
|
||||
disk_threshold: 95
|
||||
|
||||
cluster:
|
||||
mode: "agent"
|
||||
controller_host: "YOUR_CONTROLLER_IP"
|
||||
controller_port: 8443
|
||||
api_secret: "your-api-secret-key"
|
||||
agent_id: "auto" # Автоматически сгенерируется
|
||||
heartbeat_interval: 30
|
||||
report_interval: 60
|
||||
|
||||
#==========================================================================
|
||||
# 4. Docker Compose Configuration
|
||||
# docker-compose.yml для контейнеризированного развертывания
|
||||
#==========================================================================
|
||||
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build: .
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- ./data:/opt/pyguardian/data
|
||||
- ./config:/opt/pyguardian/config
|
||||
- ./logs:/opt/pyguardian/logs
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=controller
|
||||
command: ["python", "main.py", "--mode", "controller"]
|
||||
|
||||
pyguardian-agent:
|
||||
build: .
|
||||
container_name: pyguardian-agent
|
||||
restart: unless-stopped
|
||||
privileged: true
|
||||
network_mode: host
|
||||
volumes:
|
||||
- ./data:/opt/pyguardian/data
|
||||
- ./config:/opt/pyguardian/config
|
||||
- ./logs:/opt/pyguardian/logs
|
||||
- /var/log:/var/log:ro
|
||||
environment:
|
||||
- PYGUARDIAN_MODE=agent
|
||||
- CONTROLLER_HOST=your-controller-ip
|
||||
command: ["python", "main.py", "--mode", "agent"]
|
||||
depends_on:
|
||||
- pyguardian-controller
|
||||
|
||||
#==========================================================================
|
||||
# 5. Systemd Service Templates
|
||||
# /etc/systemd/system/pyguardian.service
|
||||
#==========================================================================
|
||||
|
||||
[Unit]
|
||||
Description=PyGuardian Security System
|
||||
After=network.target
|
||||
Wants=network-online.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=pyguardian
|
||||
Group=pyguardian
|
||||
WorkingDirectory=/opt/pyguardian
|
||||
ExecStart=/opt/pyguardian/venv/bin/python main.py
|
||||
ExecReload=/bin/kill -HUP $MAINPID
|
||||
Restart=always
|
||||
RestartSec=30
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
SyslogIdentifier=pyguardian
|
||||
|
||||
# Security settings
|
||||
NoNewPrivileges=yes
|
||||
PrivateTmp=yes
|
||||
ProtectSystem=strict
|
||||
ProtectHome=yes
|
||||
ReadWritePaths=/opt/pyguardian/data /opt/pyguardian/logs
|
||||
CapabilityBoundingSet=CAP_NET_ADMIN CAP_NET_RAW
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
|
||||
#==========================================================================
|
||||
# 6. Nginx Proxy Configuration (для веб-интерфейса)
|
||||
# /etc/nginx/sites-available/pyguardian
|
||||
#==========================================================================
|
||||
|
||||
server {
|
||||
listen 80;
|
||||
server_name your-domain.com;
|
||||
return 301 https://$server_name$request_uri;
|
||||
}
|
||||
|
||||
server {
|
||||
listen 443 ssl http2;
|
||||
server_name your-domain.com;
|
||||
|
||||
ssl_certificate /etc/letsencrypt/live/your-domain.com/fullchain.pem;
|
||||
ssl_certificate_key /etc/letsencrypt/live/your-domain.com/privkey.pem;
|
||||
|
||||
location / {
|
||||
proxy_pass http://127.0.0.1:8080;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
|
||||
# WebSocket support
|
||||
proxy_http_version 1.1;
|
||||
proxy_set_header Upgrade $http_upgrade;
|
||||
proxy_set_header Connection "upgrade";
|
||||
}
|
||||
|
||||
location /api/ {
|
||||
proxy_pass http://127.0.0.1:8443/;
|
||||
proxy_set_header Host $host;
|
||||
proxy_set_header X-Real-IP $remote_addr;
|
||||
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||
proxy_set_header X-Forwarded-Proto $scheme;
|
||||
}
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# 7. Environment Variables (.env файл)
|
||||
#==========================================================================
|
||||
|
||||
# PyGuardian Environment Variables
|
||||
PYGUARDIAN_MODE=standalone
|
||||
PYGUARDIAN_CONFIG=/opt/pyguardian/config/config.yaml
|
||||
PYGUARDIAN_DATA_DIR=/opt/pyguardian/data
|
||||
PYGUARDIAN_LOG_LEVEL=INFO
|
||||
|
||||
# Telegram Configuration
|
||||
TELEGRAM_BOT_TOKEN=your_bot_token_here
|
||||
TELEGRAM_ADMIN_USERS=123456789,987654321
|
||||
|
||||
# Cluster Configuration (если используется)
|
||||
CLUSTER_CONTROLLER_HOST=your-controller-ip
|
||||
CLUSTER_CONTROLLER_PORT=8443
|
||||
CLUSTER_API_SECRET=your-api-secret
|
||||
CLUSTER_AUTH_KEY=your-cluster-auth-key
|
||||
|
||||
# Database Configuration
|
||||
DATABASE_URL=sqlite:///opt/pyguardian/data/pyguardian.db
|
||||
|
||||
# Security Settings
|
||||
ENABLE_2FA=true
|
||||
SESSION_TIMEOUT=30
|
||||
MAX_FAILED_ATTEMPTS=3
|
||||
|
||||
#==========================================================================
|
||||
# 8. Firewall Rules Examples (iptables)
|
||||
#==========================================================================
|
||||
|
||||
#!/bin/bash
|
||||
# PyGuardian Firewall Rules
|
||||
|
||||
# Очистка существующих правил
|
||||
iptables -F
|
||||
iptables -X
|
||||
iptables -Z
|
||||
|
||||
# Политики по умолчанию
|
||||
iptables -P INPUT DROP
|
||||
iptables -P FORWARD DROP
|
||||
iptables -P OUTPUT ACCEPT
|
||||
|
||||
# Разрешить loopback
|
||||
iptables -A INPUT -i lo -j ACCEPT
|
||||
iptables -A OUTPUT -o lo -j ACCEPT
|
||||
|
||||
# Разрешить установленные соединения
|
||||
iptables -A INPUT -m conntrack --ctstate ESTABLISHED,RELATED -j ACCEPT
|
||||
|
||||
# SSH (ограничить количество попыток)
|
||||
iptables -A INPUT -p tcp --dport 22 -m conntrack --ctstate NEW -m recent --set --name SSH
|
||||
iptables -A INPUT -p tcp --dport 22 -m conntrack --ctstate NEW -m recent --update --seconds 60 --hitcount 3 --name SSH -j DROP
|
||||
iptables -A INPUT -p tcp --dport 22 -j ACCEPT
|
||||
|
||||
# HTTP/HTTPS (с rate limiting)
|
||||
iptables -A INPUT -p tcp --dport 80 -m limit --limit 25/minute --limit-burst 100 -j ACCEPT
|
||||
iptables -A INPUT -p tcp --dport 443 -m limit --limit 25/minute --limit-burst 100 -j ACCEPT
|
||||
|
||||
# Cluster API (только от контроллера)
|
||||
iptables -A INPUT -p tcp --dport 8443 -s your-controller-ip -j ACCEPT
|
||||
|
||||
# DDoS Protection
|
||||
iptables -A INPUT -p tcp --syn -m limit --limit 1/s --limit-burst 3 -j ACCEPT
|
||||
iptables -A INPUT -p icmp -m limit --limit 1/s --limit-burst 1 -j ACCEPT
|
||||
|
||||
# Логирование отброшенных пакетов
|
||||
iptables -A INPUT -j LOG --log-prefix "DROPPED: " --log-level 4
|
||||
iptables -A INPUT -j DROP
|
||||
|
||||
#==========================================================================
|
||||
# 9. Monitoring Configuration (для интеграции с Grafana/Prometheus)
|
||||
#==========================================================================
|
||||
|
||||
# prometheus.yml
|
||||
global:
|
||||
scrape_interval: 15s
|
||||
|
||||
scrape_configs:
|
||||
- job_name: 'pyguardian'
|
||||
static_configs:
|
||||
- targets: ['localhost:9090']
|
||||
metrics_path: /metrics
|
||||
scrape_interval: 30s
|
||||
|
||||
- job_name: 'pyguardian-cluster'
|
||||
static_configs:
|
||||
- targets: ['controller-ip:8443']
|
||||
metrics_path: /cluster/metrics
|
||||
scrape_interval: 60s
|
||||
|
||||
#==========================================================================
|
||||
# 10. Backup Configuration
|
||||
#==========================================================================
|
||||
|
||||
#!/bin/bash
|
||||
# PyGuardian Backup Script
|
||||
|
||||
BACKUP_DIR="/opt/pyguardian/backups"
|
||||
DATA_DIR="/opt/pyguardian/data"
|
||||
CONFIG_DIR="/opt/pyguardian/config"
|
||||
LOG_DIR="/opt/pyguardian/logs"
|
||||
|
||||
DATE=$(date +%Y%m%d_%H%M%S)
|
||||
BACKUP_FILE="pyguardian_backup_${DATE}.tar.gz"
|
||||
|
||||
# Создать архив
|
||||
tar -czf "${BACKUP_DIR}/${BACKUP_FILE}" \
|
||||
"${DATA_DIR}" \
|
||||
"${CONFIG_DIR}" \
|
||||
"${LOG_DIR}"
|
||||
|
||||
# Оставить только последние 7 резервных копий
|
||||
find "${BACKUP_DIR}" -name "pyguardian_backup_*.tar.gz" -type f -mtime +7 -delete
|
||||
|
||||
echo "Backup completed: ${BACKUP_FILE}"
|
||||
@@ -1,396 +0,0 @@
|
||||
# PyGuardian Telegram Bot Commands Examples
|
||||
# Примеры команд для управления системой через Telegram
|
||||
|
||||
#==========================================================================
|
||||
# 1. Основные команды безопасности
|
||||
#==========================================================================
|
||||
|
||||
# Получить статус системы
|
||||
/status
|
||||
|
||||
# Заблокировать IP адрес
|
||||
/block 192.168.1.100
|
||||
|
||||
# Разблокировать IP адрес
|
||||
/unblock 192.168.1.100
|
||||
|
||||
# Показать активные сессии
|
||||
/sessions
|
||||
|
||||
# Завершить сессию пользователя
|
||||
/kick username
|
||||
|
||||
# Просмотреть логи безопасности
|
||||
/logs security
|
||||
|
||||
#==========================================================================
|
||||
# 2. Команды управления кластером (только для контроллера)
|
||||
#==========================================================================
|
||||
|
||||
# Показать все агенты в кластере
|
||||
/cluster status
|
||||
|
||||
# Добавить новый сервер в кластер
|
||||
/cluster add 192.168.1.50 root /path/to/ssh/key
|
||||
|
||||
# Развернуть агента на сервере
|
||||
/cluster deploy 192.168.1.50
|
||||
|
||||
# Проверить соединение с агентом
|
||||
/cluster ping agent-001
|
||||
|
||||
# Обновить конфигурацию агента
|
||||
/cluster update agent-001
|
||||
|
||||
# Перезапустить агент
|
||||
/cluster restart agent-001
|
||||
|
||||
# Удалить агент из кластера
|
||||
/cluster remove agent-001
|
||||
|
||||
# Показать логи агента
|
||||
/cluster logs agent-001
|
||||
|
||||
#==========================================================================
|
||||
# 3. Интерактивные диалоги
|
||||
#==========================================================================
|
||||
|
||||
# Пример диалога добавления сервера в кластер:
|
||||
|
||||
👤 User: /cluster add
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
🔗 Добавление нового сервера в кластер
|
||||
|
||||
Введите IP адрес сервера:
|
||||
```
|
||||
|
||||
👤 User: 192.168.1.50
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
IP адрес: 192.168.1.50 ✅
|
||||
|
||||
Введите имя пользователя для SSH подключения:
|
||||
(по умолчанию: root)
|
||||
```
|
||||
|
||||
👤 User: root
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
Пользователь: root ✅
|
||||
|
||||
Выберите метод аутентификации:
|
||||
[1] SSH ключ
|
||||
[2] Пароль
|
||||
[3] Отмена
|
||||
```
|
||||
|
||||
👤 User: 1
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
Метод аутентификации: SSH ключ ✅
|
||||
|
||||
Отправьте файл приватного ключа или укажите путь к нему:
|
||||
(например: /root/.ssh/id_rsa)
|
||||
```
|
||||
|
||||
👤 User: /root/.ssh/cluster_key
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
SSH ключ: /root/.ssh/cluster_key ✅
|
||||
|
||||
Подтвердите добавление сервера:
|
||||
IP: 192.168.1.50
|
||||
Пользователь: root
|
||||
SSH ключ: /root/.ssh/cluster_key
|
||||
|
||||
[Подтвердить] [Отмена]
|
||||
```
|
||||
|
||||
👤 User: Подтвердить
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
⏳ Подключение к серверу 192.168.1.50...
|
||||
✅ Соединение установлено
|
||||
⏳ Установка PyGuardian Agent...
|
||||
⏳ Настройка конфигурации...
|
||||
⏳ Запуск службы...
|
||||
✅ Агент успешно развернут!
|
||||
|
||||
🆔 ID агента: agent-003
|
||||
📡 Статус: Онлайн
|
||||
🕐 Время развертывания: 2 мин 34 сек
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 4. Уведомления и алерты
|
||||
#==========================================================================
|
||||
|
||||
# Уведомления безопасности:
|
||||
```
|
||||
🚨 SECURITY ALERT
|
||||
|
||||
🕐 Time: 2024-01-15 14:30:25
|
||||
🎯 Server: web-server-01 (192.168.1.10)
|
||||
⚠️ Event: Multiple failed SSH login attempts
|
||||
📍 Source: 203.0.113.5
|
||||
📊 Attempts: 5 failed attempts in 2 minutes
|
||||
|
||||
Action taken: IP blocked for 5 minutes
|
||||
```
|
||||
|
||||
# Уведомления кластера:
|
||||
```
|
||||
🔄 CLUSTER EVENT
|
||||
|
||||
🕐 Time: 2024-01-15 14:35:10
|
||||
🎯 Agent: database-server-02 (agent-005)
|
||||
⚠️ Event: Agent disconnected
|
||||
🔍 Reason: Network timeout
|
||||
|
||||
Status: Attempting reconnection...
|
||||
```
|
||||
|
||||
# Уведомления о ресурсах:
|
||||
```
|
||||
📈 RESOURCE ALERT
|
||||
|
||||
🕐 Time: 2024-01-15 14:40:15
|
||||
🎯 Server: app-server-03 (192.168.1.30)
|
||||
⚠️ Event: High CPU usage
|
||||
📊 Current: 87% (threshold: 80%)
|
||||
⏱️ Duration: 5 minutes
|
||||
|
||||
Recommendation: Check running processes
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 5. Команды мониторинга
|
||||
#==========================================================================
|
||||
|
||||
# Показать ресурсы системы
|
||||
/resources
|
||||
|
||||
# Показать статистику файрвола
|
||||
/firewall stats
|
||||
|
||||
# Показать активные подключения
|
||||
/connections
|
||||
|
||||
# Показать топ процессов
|
||||
/processes
|
||||
|
||||
# Проверить обновления системы
|
||||
/updates
|
||||
|
||||
# Показать информацию о дисках
|
||||
/disk
|
||||
|
||||
# Показать сетевую статистику
|
||||
/network
|
||||
|
||||
#==========================================================================
|
||||
# 6. Команды управления файрволом
|
||||
#==========================================================================
|
||||
|
||||
# Показать правила файрвола
|
||||
/firewall rules
|
||||
|
||||
# Добавить правило файрвола
|
||||
/firewall add tcp 80 allow
|
||||
|
||||
# Удалить правило файрвола
|
||||
/firewall remove tcp 80
|
||||
|
||||
# Временно отключить файрвол
|
||||
/firewall disable
|
||||
|
||||
# Включить файрвол
|
||||
/firewall enable
|
||||
|
||||
# Показать заблокированные IP
|
||||
/firewall blocked
|
||||
|
||||
#==========================================================================
|
||||
# 7. Команды резервного копирования
|
||||
#==========================================================================
|
||||
|
||||
# Создать резервную копию
|
||||
/backup create
|
||||
|
||||
# Показать список резервных копий
|
||||
/backup list
|
||||
|
||||
# Восстановить из резервной копии
|
||||
/backup restore backup-20240115.tar.gz
|
||||
|
||||
# Настроить автоматическое резервное копирование
|
||||
/backup schedule daily 03:00
|
||||
|
||||
#==========================================================================
|
||||
# 8. Административные команды
|
||||
#==========================================================================
|
||||
|
||||
# Показать конфигурацию
|
||||
/config show
|
||||
|
||||
# Обновить конфигурацию
|
||||
/config update telegram.admin_users [123456789, 987654321]
|
||||
|
||||
# Перезагрузить конфигурацию
|
||||
/config reload
|
||||
|
||||
# Показать версию системы
|
||||
/version
|
||||
|
||||
# Обновить систему
|
||||
/update system
|
||||
|
||||
# Перезапустить службы
|
||||
/restart services
|
||||
|
||||
#==========================================================================
|
||||
# 9. Команды отладки
|
||||
#==========================================================================
|
||||
|
||||
# Показать подробные логи
|
||||
/debug logs
|
||||
|
||||
# Проверить соединение с базой данных
|
||||
/debug database
|
||||
|
||||
# Тестировать уведомления
|
||||
/debug notify
|
||||
|
||||
# Показать состояние служб
|
||||
/debug services
|
||||
|
||||
# Экспортировать диагностическую информацию
|
||||
/debug export
|
||||
|
||||
#==========================================================================
|
||||
# 10. Примеры массового управления кластером
|
||||
#==========================================================================
|
||||
|
||||
# Обновить все агенты в кластере
|
||||
/cluster update all
|
||||
|
||||
# Перезапустить все агенты
|
||||
/cluster restart all
|
||||
|
||||
# Показать сводку по всем агентам
|
||||
/cluster summary
|
||||
|
||||
# Выполнить команду на всех агентах
|
||||
/cluster exec "systemctl status nginx"
|
||||
|
||||
# Отправить конфигурацию на все агенты
|
||||
/cluster config push firewall.yaml
|
||||
|
||||
# Собрать логи со всех агентов
|
||||
/cluster logs collect
|
||||
|
||||
#==========================================================================
|
||||
# 11. Интеграция с внешними системами
|
||||
#==========================================================================
|
||||
|
||||
# Отправить метрики в Grafana
|
||||
/metrics export grafana
|
||||
|
||||
# Синхронизировать с SIEM системой
|
||||
/siem sync
|
||||
|
||||
# Обновить базу IP репутации
|
||||
/reputation update
|
||||
|
||||
# Отправить отчет по email
|
||||
/report email weekly
|
||||
|
||||
#==========================================================================
|
||||
# 12. Примеры автоматических ответов
|
||||
#==========================================================================
|
||||
|
||||
# При попытке несанкционированного доступа:
|
||||
```
|
||||
🚨 INTRUSION DETECTED
|
||||
|
||||
An unauthorized access attempt has been detected and blocked automatically.
|
||||
|
||||
🔹 Details:
|
||||
• Source IP: 203.0.113.42
|
||||
• Target: ssh://server-01:22
|
||||
• Time: 2024-01-15 15:45:30
|
||||
• Action: IP blocked for 1 hour
|
||||
|
||||
🔹 Recommendations:
|
||||
• Review SSH access policies
|
||||
• Consider IP whitelisting
|
||||
• Enable 2FA for critical accounts
|
||||
|
||||
Type /unblock 203.0.113.42 to manually unblock if needed
|
||||
```
|
||||
|
||||
# При превышении ресурсов:
|
||||
```
|
||||
⚠️ RESOURCE WARNING
|
||||
|
||||
High resource usage detected on multiple servers.
|
||||
|
||||
🔹 Affected servers:
|
||||
• web-01: CPU 85% 📈
|
||||
• db-01: Memory 92% 🧠
|
||||
• app-01: Disk 88% 💾
|
||||
|
||||
🔹 Auto-scaling:
|
||||
• Cluster load balancer activated
|
||||
• Additional instances provisioning...
|
||||
• ETA: 3 minutes
|
||||
|
||||
Type /resources for detailed information
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 13. Кастомные команды через плагины
|
||||
#==========================================================================
|
||||
|
||||
# Wordpress специфичные команды
|
||||
/wp update plugins
|
||||
/wp backup database
|
||||
/wp security scan
|
||||
|
||||
# Docker управление
|
||||
/docker ps
|
||||
/docker restart container_name
|
||||
/docker logs container_name
|
||||
|
||||
# Nginx управление
|
||||
/nginx reload
|
||||
/nginx test
|
||||
/nginx status
|
||||
|
||||
# SSL сертификаты
|
||||
/ssl check domain.com
|
||||
/ssl renew all
|
||||
/ssl notify expiring
|
||||
|
||||
#==========================================================================
|
||||
# 14. Голосовые команды (если поддерживается)
|
||||
#==========================================================================
|
||||
|
||||
# Примеры голосовых сообщений:
|
||||
🎤 "Заблокировать IP 192.168.1.100"
|
||||
🎤 "Показать статус кластера"
|
||||
🎤 "Перезапустить все службы"
|
||||
🎤 "Создать резервную копию"
|
||||
|
||||
# Ответы голосом:
|
||||
🔊 "IP адрес заблокирован"
|
||||
🔊 "Все службы кластера работают нормально"
|
||||
🔊 "Перезапуск служб завершен успешно"
|
||||
🔊 "Резервная копия создана"
|
||||
@@ -1,396 +0,0 @@
|
||||
# PyGuardian Telegram Bot Commands Examples
|
||||
# Примеры команд для управления системой через Telegram
|
||||
|
||||
#==========================================================================
|
||||
# 1. Основные команды безопасности
|
||||
#==========================================================================
|
||||
|
||||
# Получить статус системы
|
||||
/status
|
||||
|
||||
# Заблокировать IP адрес
|
||||
/block 192.168.1.100
|
||||
|
||||
# Разблокировать IP адрес
|
||||
/unblock 192.168.1.100
|
||||
|
||||
# Показать активные сессии
|
||||
/sessions
|
||||
|
||||
# Завершить сессию пользователя
|
||||
/kick username
|
||||
|
||||
# Просмотреть логи безопасности
|
||||
/logs security
|
||||
|
||||
#==========================================================================
|
||||
# 2. Команды управления кластером (только для контроллера)
|
||||
#==========================================================================
|
||||
|
||||
# Показать все агенты в кластере
|
||||
/cluster status
|
||||
|
||||
# Добавить новый сервер в кластер
|
||||
/cluster add 192.168.1.50 root /path/to/ssh/key
|
||||
|
||||
# Развернуть агента на сервере
|
||||
/cluster deploy 192.168.1.50
|
||||
|
||||
# Проверить соединение с агентом
|
||||
/cluster ping agent-001
|
||||
|
||||
# Обновить конфигурацию агента
|
||||
/cluster update agent-001
|
||||
|
||||
# Перезапустить агент
|
||||
/cluster restart agent-001
|
||||
|
||||
# Удалить агент из кластера
|
||||
/cluster remove agent-001
|
||||
|
||||
# Показать логи агента
|
||||
/cluster logs agent-001
|
||||
|
||||
#==========================================================================
|
||||
# 3. Интерактивные диалоги
|
||||
#==========================================================================
|
||||
|
||||
# Пример диалога добавления сервера в кластер:
|
||||
|
||||
👤 User: /cluster add
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
🔗 Добавление нового сервера в кластер
|
||||
|
||||
Введите IP адрес сервера:
|
||||
```
|
||||
|
||||
👤 User: 192.168.1.50
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
IP адрес: 192.168.1.50 ✅
|
||||
|
||||
Введите имя пользователя для SSH подключения:
|
||||
(по умолчанию: root)
|
||||
```
|
||||
|
||||
👤 User: root
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
Пользователь: root ✅
|
||||
|
||||
Выберите метод аутентификации:
|
||||
[1] SSH ключ
|
||||
[2] Пароль
|
||||
[3] Отмена
|
||||
```
|
||||
|
||||
👤 User: 1
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
Метод аутентификации: SSH ключ ✅
|
||||
|
||||
Отправьте файл приватного ключа или укажите путь к нему:
|
||||
(например: /root/.ssh/id_rsa)
|
||||
```
|
||||
|
||||
👤 User: /root/.ssh/cluster_key
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
SSH ключ: /root/.ssh/cluster_key ✅
|
||||
|
||||
Подтвердите добавление сервера:
|
||||
IP: 192.168.1.50
|
||||
Пользователь: root
|
||||
SSH ключ: /root/.ssh/cluster_key
|
||||
|
||||
[Подтвердить] [Отмена]
|
||||
```
|
||||
|
||||
👤 User: Подтвердить
|
||||
|
||||
🤖 Bot:
|
||||
```
|
||||
⏳ Подключение к серверу 192.168.1.50...
|
||||
✅ Соединение установлено
|
||||
⏳ Установка PyGuardian Agent...
|
||||
⏳ Настройка конфигурации...
|
||||
⏳ Запуск службы...
|
||||
✅ Агент успешно развернут!
|
||||
|
||||
🆔 ID агента: agent-003
|
||||
📡 Статус: Онлайн
|
||||
🕐 Время развертывания: 2 мин 34 сек
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 4. Уведомления и алерты
|
||||
#==========================================================================
|
||||
|
||||
# Уведомления безопасности:
|
||||
```
|
||||
🚨 SECURITY ALERT
|
||||
|
||||
🕐 Time: 2024-01-15 14:30:25
|
||||
🎯 Server: web-server-01 (192.168.1.10)
|
||||
⚠️ Event: Multiple failed SSH login attempts
|
||||
📍 Source: 203.0.113.5
|
||||
📊 Attempts: 5 failed attempts in 2 minutes
|
||||
|
||||
Action taken: IP blocked for 5 minutes
|
||||
```
|
||||
|
||||
# Уведомления кластера:
|
||||
```
|
||||
🔄 CLUSTER EVENT
|
||||
|
||||
🕐 Time: 2024-01-15 14:35:10
|
||||
🎯 Agent: database-server-02 (agent-005)
|
||||
⚠️ Event: Agent disconnected
|
||||
🔍 Reason: Network timeout
|
||||
|
||||
Status: Attempting reconnection...
|
||||
```
|
||||
|
||||
# Уведомления о ресурсах:
|
||||
```
|
||||
📈 RESOURCE ALERT
|
||||
|
||||
🕐 Time: 2024-01-15 14:40:15
|
||||
🎯 Server: app-server-03 (192.168.1.30)
|
||||
⚠️ Event: High CPU usage
|
||||
📊 Current: 87% (threshold: 80%)
|
||||
⏱️ Duration: 5 minutes
|
||||
|
||||
Recommendation: Check running processes
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 5. Команды мониторинга
|
||||
#==========================================================================
|
||||
|
||||
# Показать ресурсы системы
|
||||
/resources
|
||||
|
||||
# Показать статистику файрвола
|
||||
/firewall stats
|
||||
|
||||
# Показать активные подключения
|
||||
/connections
|
||||
|
||||
# Показать топ процессов
|
||||
/processes
|
||||
|
||||
# Проверить обновления системы
|
||||
/updates
|
||||
|
||||
# Показать информацию о дисках
|
||||
/disk
|
||||
|
||||
# Показать сетевую статистику
|
||||
/network
|
||||
|
||||
#==========================================================================
|
||||
# 6. Команды управления файрволом
|
||||
#==========================================================================
|
||||
|
||||
# Показать правила файрвола
|
||||
/firewall rules
|
||||
|
||||
# Добавить правило файрвола
|
||||
/firewall add tcp 80 allow
|
||||
|
||||
# Удалить правило файрвола
|
||||
/firewall remove tcp 80
|
||||
|
||||
# Временно отключить файрвол
|
||||
/firewall disable
|
||||
|
||||
# Включить файрвол
|
||||
/firewall enable
|
||||
|
||||
# Показать заблокированные IP
|
||||
/firewall blocked
|
||||
|
||||
#==========================================================================
|
||||
# 7. Команды резервного копирования
|
||||
#==========================================================================
|
||||
|
||||
# Создать резервную копию
|
||||
/backup create
|
||||
|
||||
# Показать список резервных копий
|
||||
/backup list
|
||||
|
||||
# Восстановить из резервной копии
|
||||
/backup restore backup-20240115.tar.gz
|
||||
|
||||
# Настроить автоматическое резервное копирование
|
||||
/backup schedule daily 03:00
|
||||
|
||||
#==========================================================================
|
||||
# 8. Административные команды
|
||||
#==========================================================================
|
||||
|
||||
# Показать конфигурацию
|
||||
/config show
|
||||
|
||||
# Обновить конфигурацию
|
||||
/config update telegram.admin_users [123456789, 987654321]
|
||||
|
||||
# Перезагрузить конфигурацию
|
||||
/config reload
|
||||
|
||||
# Показать версию системы
|
||||
/version
|
||||
|
||||
# Обновить систему
|
||||
/update system
|
||||
|
||||
# Перезапустить службы
|
||||
/restart services
|
||||
|
||||
#==========================================================================
|
||||
# 9. Команды отладки
|
||||
#==========================================================================
|
||||
|
||||
# Показать подробные логи
|
||||
/debug logs
|
||||
|
||||
# Проверить соединение с базой данных
|
||||
/debug database
|
||||
|
||||
# Тестировать уведомления
|
||||
/debug notify
|
||||
|
||||
# Показать состояние служб
|
||||
/debug services
|
||||
|
||||
# Экспортировать диагностическую информацию
|
||||
/debug export
|
||||
|
||||
#==========================================================================
|
||||
# 10. Примеры массового управления кластером
|
||||
#==========================================================================
|
||||
|
||||
# Обновить все агенты в кластере
|
||||
/cluster update all
|
||||
|
||||
# Перезапустить все агенты
|
||||
/cluster restart all
|
||||
|
||||
# Показать сводку по всем агентам
|
||||
/cluster summary
|
||||
|
||||
# Выполнить команду на всех агентах
|
||||
/cluster exec "systemctl status nginx"
|
||||
|
||||
# Отправить конфигурацию на все агенты
|
||||
/cluster config push firewall.yaml
|
||||
|
||||
# Собрать логи со всех агентов
|
||||
/cluster logs collect
|
||||
|
||||
#==========================================================================
|
||||
# 11. Интеграция с внешними системами
|
||||
#==========================================================================
|
||||
|
||||
# Отправить метрики в Grafana
|
||||
/metrics export grafana
|
||||
|
||||
# Синхронизировать с SIEM системой
|
||||
/siem sync
|
||||
|
||||
# Обновить базу IP репутации
|
||||
/reputation update
|
||||
|
||||
# Отправить отчет по email
|
||||
/report email weekly
|
||||
|
||||
#==========================================================================
|
||||
# 12. Примеры автоматических ответов
|
||||
#==========================================================================
|
||||
|
||||
# При попытке несанкционированного доступа:
|
||||
```
|
||||
🚨 INTRUSION DETECTED
|
||||
|
||||
An unauthorized access attempt has been detected and blocked automatically.
|
||||
|
||||
🔹 Details:
|
||||
• Source IP: 203.0.113.42
|
||||
• Target: ssh://server-01:22
|
||||
• Time: 2024-01-15 15:45:30
|
||||
• Action: IP blocked for 1 hour
|
||||
|
||||
🔹 Recommendations:
|
||||
• Review SSH access policies
|
||||
• Consider IP whitelisting
|
||||
• Enable 2FA for critical accounts
|
||||
|
||||
Type /unblock 203.0.113.42 to manually unblock if needed
|
||||
```
|
||||
|
||||
# При превышении ресурсов:
|
||||
```
|
||||
⚠️ RESOURCE WARNING
|
||||
|
||||
High resource usage detected on multiple servers.
|
||||
|
||||
🔹 Affected servers:
|
||||
• web-01: CPU 85% 📈
|
||||
• db-01: Memory 92% 🧠
|
||||
• app-01: Disk 88% 💾
|
||||
|
||||
🔹 Auto-scaling:
|
||||
• Cluster load balancer activated
|
||||
• Additional instances provisioning...
|
||||
• ETA: 3 minutes
|
||||
|
||||
Type /resources for detailed information
|
||||
```
|
||||
|
||||
#==========================================================================
|
||||
# 13. Кастомные команды через плагины
|
||||
#==========================================================================
|
||||
|
||||
# Wordpress специфичные команды
|
||||
/wp update plugins
|
||||
/wp backup database
|
||||
/wp security scan
|
||||
|
||||
# Docker управление
|
||||
/docker ps
|
||||
/docker restart container_name
|
||||
/docker logs container_name
|
||||
|
||||
# Nginx управление
|
||||
/nginx reload
|
||||
/nginx test
|
||||
/nginx status
|
||||
|
||||
# SSL сертификаты
|
||||
/ssl check domain.com
|
||||
/ssl renew all
|
||||
/ssl notify expiring
|
||||
|
||||
#==========================================================================
|
||||
# 14. Голосовые команды (если поддерживается)
|
||||
#==========================================================================
|
||||
|
||||
# Примеры голосовых сообщений:
|
||||
🎤 "Заблокировать IP 192.168.1.100"
|
||||
🎤 "Показать статус кластера"
|
||||
🎤 "Перезапустить все службы"
|
||||
🎤 "Создать резервную копию"
|
||||
|
||||
# Ответы голосом:
|
||||
🔊 "IP адрес заблокирован"
|
||||
🔊 "Все службы кластера работают нормально"
|
||||
🔊 "Перезапуск служб завершен успешно"
|
||||
🔊 "Резервная копия создана"
|
||||
@@ -1,195 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Quick Installation Script
|
||||
# Wrapper for the main installation system
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m'
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "================================================="
|
||||
echo " PyGuardian Security System - Quick Installer"
|
||||
echo "================================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
echo "Usage: sudo ./install.sh"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show installation options
|
||||
show_options() {
|
||||
echo ""
|
||||
print_info "Выберите тип установки:"
|
||||
echo ""
|
||||
echo "1) 🔧 Быстрая установка (Standalone режим)"
|
||||
echo "2) 📋 Интерактивная установка с выбором режима"
|
||||
echo "3) 🐳 Docker установка"
|
||||
echo "4) 📖 Показать документацию"
|
||||
echo "5) ❌ Выход"
|
||||
echo ""
|
||||
}
|
||||
|
||||
# Quick standalone installation
|
||||
quick_install() {
|
||||
print_info "Запуск быстрой установки (Standalone режим)..."
|
||||
|
||||
# Run the main installation script
|
||||
if [[ -f "scripts/install.sh" ]]; then
|
||||
chmod +x scripts/install.sh
|
||||
./scripts/install.sh --mode=standalone
|
||||
else
|
||||
print_error "Installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Interactive installation
|
||||
interactive_install() {
|
||||
print_info "Запуск интерактивной установки..."
|
||||
|
||||
if [[ -f "scripts/install.sh" ]]; then
|
||||
chmod +x scripts/install.sh
|
||||
./scripts/install.sh
|
||||
else
|
||||
print_error "Installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Docker installation
|
||||
docker_install() {
|
||||
print_info "Запуск Docker установки..."
|
||||
|
||||
if [[ -f "scripts/docker-install.sh" ]]; then
|
||||
chmod +x scripts/docker-install.sh
|
||||
./scripts/docker-install.sh
|
||||
else
|
||||
print_error "Docker installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show documentation
|
||||
show_documentation() {
|
||||
echo ""
|
||||
echo -e "${BLUE}📖 Документация PyGuardian${NC}"
|
||||
echo ""
|
||||
echo "Основные файлы документации:"
|
||||
echo " • README.md - Основная документация"
|
||||
echo " • docs/CLUSTER_SETUP.md - Настройка кластера"
|
||||
echo " • config/config.yaml - Пример конфигурации"
|
||||
echo ""
|
||||
echo "Онлайн ресурсы:"
|
||||
echo " • GitHub: https://github.com/your-repo/PyGuardian"
|
||||
echo " • Wiki: https://github.com/your-repo/PyGuardian/wiki"
|
||||
echo ""
|
||||
echo "Быстрые команды после установки:"
|
||||
echo " • make install - Интерактивная установка"
|
||||
echo " • make standalone - Автономный сервер"
|
||||
echo " • make controller - Контроллер кластера"
|
||||
echo " • make agent - Агент кластера"
|
||||
echo ""
|
||||
}
|
||||
|
||||
# Main function
|
||||
main() {
|
||||
print_header
|
||||
|
||||
# Check if running as root
|
||||
check_root
|
||||
|
||||
# If arguments provided, run directly
|
||||
if [[ $# -gt 0 ]]; then
|
||||
case "$1" in
|
||||
--quick|--standalone)
|
||||
quick_install
|
||||
;;
|
||||
--interactive)
|
||||
interactive_install
|
||||
;;
|
||||
--docker)
|
||||
docker_install
|
||||
;;
|
||||
--help|-h)
|
||||
echo "Usage: $0 [--quick|--interactive|--docker|--help]"
|
||||
echo ""
|
||||
echo "Options:"
|
||||
echo " --quick Quick standalone installation"
|
||||
echo " --interactive Interactive installation with mode selection"
|
||||
echo " --docker Docker-based installation"
|
||||
echo " --help Show this help"
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
echo "Use --help for usage information"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
return
|
||||
fi
|
||||
|
||||
# Interactive menu
|
||||
while true; do
|
||||
show_options
|
||||
read -p "Выберите опцию (1-5): " choice
|
||||
|
||||
case $choice in
|
||||
1)
|
||||
quick_install
|
||||
break
|
||||
;;
|
||||
2)
|
||||
interactive_install
|
||||
break
|
||||
;;
|
||||
3)
|
||||
docker_install
|
||||
break
|
||||
;;
|
||||
4)
|
||||
show_documentation
|
||||
read -p "Нажмите Enter для продолжения..."
|
||||
;;
|
||||
5)
|
||||
print_info "Выход из установщика"
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите число от 1 до 5."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
print_success "Установка завершена! Спасибо за использование PyGuardian!"
|
||||
}
|
||||
|
||||
# Run main with all arguments
|
||||
main "$@"
|
||||
@@ -1,109 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# PyGuardian Installation Script
|
||||
# ==============================
|
||||
|
||||
set -e
|
||||
|
||||
INSTALL_DIR="/opt/pyguardian"
|
||||
SERVICE_FILE="/etc/systemd/system/pyguardian.service"
|
||||
CONFIG_DIR="/etc/pyguardian"
|
||||
LOG_DIR="/var/log"
|
||||
DATA_DIR="/var/lib/pyguardian"
|
||||
|
||||
echo "🛡️ Установка PyGuardian - Linux Server Protection System"
|
||||
echo "========================================================="
|
||||
|
||||
# Проверка прав root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo "❌ Этот скрипт должен быть запущен от root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Проверка Python 3.10+
|
||||
echo "📋 Проверка зависимостей..."
|
||||
PYTHON_VERSION=$(python3 -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')")
|
||||
REQUIRED_VERSION="3.10"
|
||||
|
||||
if ! python3 -c "import sys; exit(0 if sys.version_info >= (3, 10) else 1)"; then
|
||||
echo "❌ Требуется Python ${REQUIRED_VERSION}+ (найден ${PYTHON_VERSION})"
|
||||
exit 1
|
||||
fi
|
||||
echo "✅ Python ${PYTHON_VERSION} обнаружен"
|
||||
|
||||
# Проверка pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
echo "❌ pip3 не найден. Установите python3-pip"
|
||||
exit 1
|
||||
fi
|
||||
echo "✅ pip3 найден"
|
||||
|
||||
# Установка системных пакетов (опционально)
|
||||
echo "📦 Установка системных зависимостей..."
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv iptables
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip python3-virtualenv iptables
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip python3-virtualenv iptables
|
||||
else
|
||||
echo "⚠️ Автоматическая установка пакетов не поддерживается для этой системы"
|
||||
echo " Убедитесь что установлены: python3-pip, iptables/nftables"
|
||||
fi
|
||||
|
||||
# Создание директорий
|
||||
echo "📁 Создание директорий..."
|
||||
mkdir -p "$INSTALL_DIR"
|
||||
mkdir -p "$CONFIG_DIR"
|
||||
mkdir -p "$DATA_DIR"
|
||||
chmod 700 "$DATA_DIR"
|
||||
|
||||
# Копирование файлов
|
||||
echo "📋 Копирование файлов..."
|
||||
cp -r src/ "$INSTALL_DIR/"
|
||||
cp main.py "$INSTALL_DIR/"
|
||||
cp requirements.txt "$INSTALL_DIR/"
|
||||
|
||||
# Копирование конфигурации
|
||||
if [[ ! -f "$CONFIG_DIR/config.yaml" ]]; then
|
||||
cp config/config.yaml "$CONFIG_DIR/"
|
||||
echo "ℹ️ Конфигурация скопирована в $CONFIG_DIR/config.yaml"
|
||||
else
|
||||
echo "⚠️ Конфигурация уже существует в $CONFIG_DIR/config.yaml"
|
||||
fi
|
||||
|
||||
# Установка Python зависимостей
|
||||
echo "🐍 Установка Python зависимостей..."
|
||||
cd "$INSTALL_DIR"
|
||||
pip3 install -r requirements.txt
|
||||
|
||||
# Установка systemd сервиса
|
||||
echo "⚙️ Установка systemd сервиса..."
|
||||
sed "s|/opt/pyguardian|$INSTALL_DIR|g; s|/opt/pyguardian/config/config.yaml|$CONFIG_DIR/config.yaml|g" \
|
||||
systemd/pyguardian.service > "$SERVICE_FILE"
|
||||
|
||||
# Права на файлы
|
||||
chmod +x "$INSTALL_DIR/main.py"
|
||||
chown -R root:root "$INSTALL_DIR"
|
||||
|
||||
# Перезагрузка systemd
|
||||
systemctl daemon-reload
|
||||
|
||||
echo ""
|
||||
echo "✅ PyGuardian успешно установлен!"
|
||||
echo ""
|
||||
echo "📝 Следующие шаги:"
|
||||
echo "1. Настройте конфигурацию в $CONFIG_DIR/config.yaml"
|
||||
echo "2. Получите токен Telegram бота от @BotFather"
|
||||
echo "3. Узнайте ваш Telegram ID через @userinfobot"
|
||||
echo "4. Обновите конфигурацию с токеном и ID"
|
||||
echo "5. Запустите сервис: systemctl start pyguardian"
|
||||
echo "6. Включите автозапуск: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo "🔧 Полезные команды:"
|
||||
echo " systemctl status pyguardian - статус сервиса"
|
||||
echo " systemctl logs pyguardian - просмотр логов"
|
||||
echo " systemctl restart pyguardian - перезапуск"
|
||||
echo ""
|
||||
echo "📖 Документация: https://github.com/your-org/pyguardian"
|
||||
@@ -1,109 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# PyGuardian Installation Script
|
||||
# ==============================
|
||||
|
||||
set -e
|
||||
|
||||
INSTALL_DIR="/opt/pyguardian"
|
||||
SERVICE_FILE="/etc/systemd/system/pyguardian.service"
|
||||
CONFIG_DIR="/etc/pyguardian"
|
||||
LOG_DIR="/var/log"
|
||||
DATA_DIR="/var/lib/pyguardian"
|
||||
|
||||
echo "🛡️ Установка PyGuardian - Linux Server Protection System"
|
||||
echo "========================================================="
|
||||
|
||||
# Проверка прав root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo "❌ Этот скрипт должен быть запущен от root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Проверка Python 3.10+
|
||||
echo "📋 Проверка зависимостей..."
|
||||
PYTHON_VERSION=$(python3 -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')")
|
||||
REQUIRED_VERSION="3.10"
|
||||
|
||||
if ! python3 -c "import sys; exit(0 if sys.version_info >= (3, 10) else 1)"; then
|
||||
echo "❌ Требуется Python ${REQUIRED_VERSION}+ (найден ${PYTHON_VERSION})"
|
||||
exit 1
|
||||
fi
|
||||
echo "✅ Python ${PYTHON_VERSION} обнаружен"
|
||||
|
||||
# Проверка pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
echo "❌ pip3 не найден. Установите python3-pip"
|
||||
exit 1
|
||||
fi
|
||||
echo "✅ pip3 найден"
|
||||
|
||||
# Установка системных пакетов (опционально)
|
||||
echo "📦 Установка системных зависимостей..."
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv iptables
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip python3-virtualenv iptables
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip python3-virtualenv iptables
|
||||
else
|
||||
echo "⚠️ Автоматическая установка пакетов не поддерживается для этой системы"
|
||||
echo " Убедитесь что установлены: python3-pip, iptables/nftables"
|
||||
fi
|
||||
|
||||
# Создание директорий
|
||||
echo "📁 Создание директорий..."
|
||||
mkdir -p "$INSTALL_DIR"
|
||||
mkdir -p "$CONFIG_DIR"
|
||||
mkdir -p "$DATA_DIR"
|
||||
chmod 700 "$DATA_DIR"
|
||||
|
||||
# Копирование файлов
|
||||
echo "📋 Копирование файлов..."
|
||||
cp -r src/ "$INSTALL_DIR/"
|
||||
cp main.py "$INSTALL_DIR/"
|
||||
cp requirements.txt "$INSTALL_DIR/"
|
||||
|
||||
# Копирование конфигурации
|
||||
if [[ ! -f "$CONFIG_DIR/config.yaml" ]]; then
|
||||
cp config/config.yaml "$CONFIG_DIR/"
|
||||
echo "ℹ️ Конфигурация скопирована в $CONFIG_DIR/config.yaml"
|
||||
else
|
||||
echo "⚠️ Конфигурация уже существует в $CONFIG_DIR/config.yaml"
|
||||
fi
|
||||
|
||||
# Установка Python зависимостей
|
||||
echo "🐍 Установка Python зависимостей..."
|
||||
cd "$INSTALL_DIR"
|
||||
pip3 install -r requirements.txt
|
||||
|
||||
# Установка systemd сервиса
|
||||
echo "⚙️ Установка systemd сервиса..."
|
||||
sed "s|/opt/pyguardian|$INSTALL_DIR|g; s|/opt/pyguardian/config/config.yaml|$CONFIG_DIR/config.yaml|g" \
|
||||
systemd/pyguardian.service > "$SERVICE_FILE"
|
||||
|
||||
# Права на файлы
|
||||
chmod +x "$INSTALL_DIR/main.py"
|
||||
chown -R root:root "$INSTALL_DIR"
|
||||
|
||||
# Перезагрузка systemd
|
||||
systemctl daemon-reload
|
||||
|
||||
echo ""
|
||||
echo "✅ PyGuardian успешно установлен!"
|
||||
echo ""
|
||||
echo "📝 Следующие шаги:"
|
||||
echo "1. Настройте конфигурацию в $CONFIG_DIR/config.yaml"
|
||||
echo "2. Получите токен Telegram бота от @BotFather"
|
||||
echo "3. Узнайте ваш Telegram ID через @userinfobot"
|
||||
echo "4. Обновите конфигурацию с токеном и ID"
|
||||
echo "5. Запустите сервис: systemctl start pyguardian"
|
||||
echo "6. Включите автозапуск: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo "🔧 Полезные команды:"
|
||||
echo " systemctl status pyguardian - статус сервиса"
|
||||
echo " systemctl logs pyguardian - просмотр логов"
|
||||
echo " systemctl restart pyguardian - перезапуск"
|
||||
echo ""
|
||||
echo "📖 Документация: https://github.com/your-org/pyguardian"
|
||||
@@ -1,274 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Quick Installation Script
|
||||
# Wrapper for the main installation system
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "================================================="
|
||||
echo " PyGuardian Security System - Quick Installer"
|
||||
echo "================================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
echo "Usage: sudo ./install.sh"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show installation options
|
||||
show_options() {
|
||||
echo ""
|
||||
print_info "Выберите тип установки:"
|
||||
echo ""
|
||||
echo "1) 🔧 Быстрая установка (Standalone режим)"
|
||||
echo "2) 📋 Интерактивная установка с выбором режима"
|
||||
echo "3) 🐳 Docker установка"
|
||||
echo "4) 📖 Показать документацию"
|
||||
echo "5) ❌ Выход"
|
||||
echo ""
|
||||
}
|
||||
|
||||
# Quick standalone installation
|
||||
quick_install() {
|
||||
print_info "Запуск быстрой установки (Standalone режим)..."
|
||||
|
||||
# Run the main installation script
|
||||
if [[ -f "scripts/install.sh" ]]; then
|
||||
chmod +x scripts/install.sh
|
||||
./scripts/install.sh --mode=standalone
|
||||
else
|
||||
print_error "Installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Interactive installation
|
||||
interactive_install() {
|
||||
print_info "Запуск интерактивной установки..."
|
||||
|
||||
if [[ -f "scripts/install.sh" ]]; then
|
||||
chmod +x scripts/install.sh
|
||||
./scripts/install.sh
|
||||
else
|
||||
print_error "Installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Docker installation
|
||||
docker_install() {
|
||||
print_info "Запуск Docker установки..."
|
||||
|
||||
if [[ -f "scripts/docker-install.sh" ]]; then
|
||||
chmod +x scripts/docker-install.sh
|
||||
./scripts/docker-install.sh
|
||||
else
|
||||
print_error "Docker installation script not found!"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show documentation
|
||||
show_documentation() {
|
||||
echo ""
|
||||
echo -e "${BLUE}📖 Документация PyGuardian${NC}"
|
||||
echo ""
|
||||
echo "Основные файлы документации:"
|
||||
echo " • README.md - Основная документация"
|
||||
echo " • docs/CLUSTER_SETUP.md - Настройка кластера"
|
||||
echo " • config/config.yaml - Пример конфигурации"
|
||||
echo ""
|
||||
echo "Онлайн ресурсы:"
|
||||
echo " • GitHub: https://github.com/your-repo/PyGuardian"
|
||||
echo " • Wiki: https://github.com/your-repo/PyGuardian/wiki"
|
||||
echo ""
|
||||
echo "Быстрые команды после установки:"
|
||||
echo " • make install - Интерактивная установка"
|
||||
echo " • make standalone - Автономный сервер"
|
||||
echo " • make controller - Контроллер кластера"
|
||||
echo " • make agent - Агент кластера"
|
||||
echo ""
|
||||
}
|
||||
|
||||
# Main function
|
||||
main() {
|
||||
print_header
|
||||
|
||||
# Check if running as root
|
||||
check_root
|
||||
|
||||
# If arguments provided, run directly
|
||||
if [[ $# -gt 0 ]]; then
|
||||
case "$1" in
|
||||
--quick|--standalone)
|
||||
quick_install
|
||||
;;
|
||||
--interactive)
|
||||
interactive_install
|
||||
;;
|
||||
--docker)
|
||||
docker_install
|
||||
;;
|
||||
--help|-h)
|
||||
echo "Usage: $0 [--quick|--interactive|--docker|--help]"
|
||||
echo ""
|
||||
echo "Options:"
|
||||
echo " --quick Quick standalone installation"
|
||||
echo " --interactive Interactive installation with mode selection"
|
||||
echo " --docker Docker-based installation"
|
||||
echo " --help Show this help"
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
echo "Use --help for usage information"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
return
|
||||
fi
|
||||
|
||||
# Interactive menu
|
||||
while true; do
|
||||
show_options
|
||||
read -p "Выберите опцию (1-5): " choice
|
||||
|
||||
case $choice in
|
||||
1)
|
||||
quick_install
|
||||
break
|
||||
;;
|
||||
2)
|
||||
interactive_install
|
||||
break
|
||||
;;
|
||||
3)
|
||||
docker_install
|
||||
break
|
||||
;;
|
||||
4)
|
||||
show_documentation
|
||||
read -p "Нажмите Enter для продолжения..."
|
||||
;;
|
||||
5)
|
||||
print_info "Выход из установщика"
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите число от 1 до 5."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
print_success "Установка завершена! Спасибо за использование PyGuardian!"
|
||||
}
|
||||
|
||||
# Run main with all arguments
|
||||
main "$@"
|
||||
echo "✅ Python ${PYTHON_VERSION} обнаружен"
|
||||
|
||||
# Проверка pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
echo "❌ pip3 не найден. Установите python3-pip"
|
||||
exit 1
|
||||
fi
|
||||
echo "✅ pip3 найден"
|
||||
|
||||
# Установка системных пакетов (опционально)
|
||||
echo "📦 Установка системных зависимостей..."
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv iptables
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip python3-virtualenv iptables
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip python3-virtualenv iptables
|
||||
else
|
||||
echo "⚠️ Автоматическая установка пакетов не поддерживается для этой системы"
|
||||
echo " Убедитесь что установлены: python3-pip, iptables/nftables"
|
||||
fi
|
||||
|
||||
# Создание директорий
|
||||
echo "📁 Создание директорий..."
|
||||
mkdir -p "$INSTALL_DIR"
|
||||
mkdir -p "$CONFIG_DIR"
|
||||
mkdir -p "$DATA_DIR"
|
||||
chmod 700 "$DATA_DIR"
|
||||
|
||||
# Копирование файлов
|
||||
echo "📋 Копирование файлов..."
|
||||
cp -r src/ "$INSTALL_DIR/"
|
||||
cp main.py "$INSTALL_DIR/"
|
||||
cp requirements.txt "$INSTALL_DIR/"
|
||||
|
||||
# Копирование конфигурации
|
||||
if [[ ! -f "$CONFIG_DIR/config.yaml" ]]; then
|
||||
cp config/config.yaml "$CONFIG_DIR/"
|
||||
echo "ℹ️ Конфигурация скопирована в $CONFIG_DIR/config.yaml"
|
||||
else
|
||||
echo "⚠️ Конфигурация уже существует в $CONFIG_DIR/config.yaml"
|
||||
fi
|
||||
|
||||
# Установка Python зависимостей
|
||||
echo "🐍 Установка Python зависимостей..."
|
||||
cd "$INSTALL_DIR"
|
||||
pip3 install -r requirements.txt
|
||||
|
||||
# Установка systemd сервиса
|
||||
echo "⚙️ Установка systemd сервиса..."
|
||||
sed "s|/opt/pyguardian|$INSTALL_DIR|g; s|/opt/pyguardian/config/config.yaml|$CONFIG_DIR/config.yaml|g" \
|
||||
systemd/pyguardian.service > "$SERVICE_FILE"
|
||||
|
||||
# Права на файлы
|
||||
chmod +x "$INSTALL_DIR/main.py"
|
||||
chown -R root:root "$INSTALL_DIR"
|
||||
|
||||
# Перезагрузка systemd
|
||||
systemctl daemon-reload
|
||||
|
||||
echo ""
|
||||
echo "✅ PyGuardian успешно установлен!"
|
||||
echo ""
|
||||
echo "📝 Следующие шаги:"
|
||||
echo "1. Настройте конфигурацию в $CONFIG_DIR/config.yaml"
|
||||
echo "2. Получите токен Telegram бота от @BotFather"
|
||||
echo "3. Узнайте ваш Telegram ID через @userinfobot"
|
||||
echo "4. Обновите конфигурацию с токеном и ID"
|
||||
echo "5. Запустите сервис: systemctl start pyguardian"
|
||||
echo "6. Включите автозапуск: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo "🔧 Полезные команды:"
|
||||
echo " systemctl status pyguardian - статус сервиса"
|
||||
echo " systemctl logs pyguardian - просмотр логов"
|
||||
echo " systemctl restart pyguardian - перезапуск"
|
||||
echo ""
|
||||
echo "📖 Документация: https://github.com/your-org/pyguardian"
|
||||
@@ -279,8 +279,16 @@ main() {
|
||||
echo ""
|
||||
echo "Next steps:"
|
||||
echo "1. Configure your Telegram bot token in /opt/pyguardian/config/config.yaml"
|
||||
echo "2. Start the service: systemctl start pyguardian"
|
||||
echo "3. Enable auto-start: systemctl enable pyguardian"
|
||||
echo "2. Review authentication settings in /opt/pyguardian/config/auth.yaml"
|
||||
echo "3. Start the service: systemctl start pyguardian"
|
||||
echo "4. Enable auto-start: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo -e "${YELLOW}[INFO]${NC} PyGuardian v2.1.0 includes:"
|
||||
echo " ✓ JWT-based agent authentication"
|
||||
echo " ✓ Encrypted agent credentials storage"
|
||||
echo " ✓ RESTful API server with authentication"
|
||||
echo " ✓ Comprehensive testing suite"
|
||||
echo " ✓ CI/CD pipeline with security scanning"
|
||||
echo ""
|
||||
echo "Documentation: ${PYGUARDIAN_REPO}/tree/main/documentation"
|
||||
echo "Support: https://github.com/SmartSolTech/PyGuardian/issues"
|
||||
@@ -15,7 +15,7 @@ YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
# Project information
|
||||
PYGUARDIAN_VERSION="2.0.0"
|
||||
PYGUARDIAN_VERSION="2.1.0"
|
||||
PYGUARDIAN_REPO="https://github.com/SmartSolTech/PyGuardian"
|
||||
|
||||
print_header() {
|
||||
@@ -279,8 +279,16 @@ main() {
|
||||
echo ""
|
||||
echo "Next steps:"
|
||||
echo "1. Configure your Telegram bot token in /opt/pyguardian/config/config.yaml"
|
||||
echo "2. Start the service: systemctl start pyguardian"
|
||||
echo "3. Enable auto-start: systemctl enable pyguardian"
|
||||
echo "2. Review authentication settings in /opt/pyguardian/config/auth.yaml"
|
||||
echo "3. Start the service: systemctl start pyguardian"
|
||||
echo "4. Enable auto-start: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo -e "${YELLOW}[INFO]${NC} PyGuardian v2.1.0 includes:"
|
||||
echo " ✓ JWT-based agent authentication"
|
||||
echo " ✓ Encrypted agent credentials storage"
|
||||
echo " ✓ RESTful API server with authentication"
|
||||
echo " ✓ Comprehensive testing suite"
|
||||
echo " ✓ CI/CD pipeline with security scanning"
|
||||
echo ""
|
||||
echo "Documentation: ${PYGUARDIAN_REPO}/tree/main/documentation"
|
||||
echo "Support: https://github.com/SmartSolTech/PyGuardian/issues"
|
||||
301
.history/install_20251125212626.sh
Normal file
301
.history/install_20251125212626.sh
Normal file
@@ -0,0 +1,301 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Universal Installer
|
||||
# Quick installation wrapper for all PyGuardian deployment modes
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
# Project information
|
||||
PYGUARDIAN_VERSION="2.1.0"
|
||||
PYGUARDIAN_REPO="https://github.com/SmartSolTech/PyGuardian"
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "================================================="
|
||||
echo " PyGuardian Security System v${PYGUARDIAN_VERSION}"
|
||||
echo " Universal Installation Wrapper"
|
||||
echo "================================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_usage() {
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "Options:"
|
||||
echo " --mode MODE Installation mode (standalone|controller|agent)"
|
||||
echo " --controller HOST Controller IP (required for agent mode)"
|
||||
echo " --docker Use Docker installation"
|
||||
echo " --help Show this help message"
|
||||
echo ""
|
||||
echo "Examples:"
|
||||
echo " $0 # Interactive installation"
|
||||
echo " $0 --mode standalone # Standalone installation"
|
||||
echo " $0 --mode controller # Cluster controller"
|
||||
echo " $0 --mode agent --controller 1.2.3.4 # Cluster agent"
|
||||
echo " $0 --docker # Docker installation"
|
||||
}
|
||||
|
||||
check_system() {
|
||||
echo -e "${BLUE}[INFO]${NC} Checking system requirements..."
|
||||
|
||||
# Check if running as root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
echo -e "${RED}[ERROR]${NC} This script must be run as root or with sudo"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check operating system
|
||||
if ! command -v systemctl &> /dev/null; then
|
||||
echo -e "${RED}[ERROR]${NC} This installer requires a systemd-based Linux distribution"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check Python version
|
||||
if command -v python3 &> /dev/null; then
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
echo -e "${GREEN}[OK]${NC} Python ${PYTHON_VERSION} found"
|
||||
|
||||
if ! python3 -c 'import sys; exit(0 if sys.version_info >= (3, 10) else 1)'; then
|
||||
echo -e "${RED}[ERROR]${NC} Python 3.10+ is required (found ${PYTHON_VERSION})"
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
echo -e "${RED}[ERROR]${NC} Python3 not found. Please install Python 3.10+"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}[OK]${NC} System requirements satisfied"
|
||||
}
|
||||
|
||||
download_installer() {
|
||||
echo -e "${BLUE}[INFO]${NC} Downloading PyGuardian installer..."
|
||||
|
||||
# Create temporary directory
|
||||
TEMP_DIR=$(mktemp -d)
|
||||
cd "$TEMP_DIR"
|
||||
|
||||
# Download the detailed installer
|
||||
if command -v curl &> /dev/null; then
|
||||
curl -fsSL "${PYGUARDIAN_REPO}/raw/main/deployment/scripts/install.sh" -o install.sh
|
||||
elif command -v wget &> /dev/null; then
|
||||
wget -q "${PYGUARDIAN_REPO}/raw/main/deployment/scripts/install.sh" -O install.sh
|
||||
else
|
||||
echo -e "${RED}[ERROR]${NC} Neither curl nor wget found. Please install one of them."
|
||||
exit 1
|
||||
fi
|
||||
|
||||
chmod +x install.sh
|
||||
echo -e "${GREEN}[OK]${NC} Installer downloaded to ${TEMP_DIR}/install.sh"
|
||||
|
||||
# Export for use in main function
|
||||
export INSTALLER_PATH="${TEMP_DIR}/install.sh"
|
||||
}
|
||||
|
||||
run_docker_installation() {
|
||||
echo -e "${BLUE}[INFO]${NC} Starting Docker-based installation..."
|
||||
|
||||
# Check if Docker is installed
|
||||
if ! command -v docker &> /dev/null; then
|
||||
echo -e "${YELLOW}[WARNING]${NC} Docker not found. Installing Docker..."
|
||||
|
||||
# Install Docker
|
||||
curl -fsSL https://get.docker.com -o get-docker.sh
|
||||
sh get-docker.sh
|
||||
|
||||
# Start Docker service
|
||||
systemctl enable docker
|
||||
systemctl start docker
|
||||
|
||||
echo -e "${GREEN}[OK]${NC} Docker installed successfully"
|
||||
fi
|
||||
|
||||
# Check if docker-compose is available
|
||||
if ! command -v docker-compose &> /dev/null; then
|
||||
if ! docker compose version &> /dev/null; then
|
||||
echo -e "${YELLOW}[WARNING]${NC} Docker Compose not found. Installing..."
|
||||
|
||||
# Install docker-compose
|
||||
curl -L "https://github.com/docker/compose/releases/latest/download/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose
|
||||
chmod +x /usr/local/bin/docker-compose
|
||||
|
||||
echo -e "${GREEN}[OK]${NC} Docker Compose installed"
|
||||
fi
|
||||
fi
|
||||
|
||||
# Download docker installation script
|
||||
TEMP_DIR=$(mktemp -d)
|
||||
cd "$TEMP_DIR"
|
||||
|
||||
if command -v curl &> /dev/null; then
|
||||
curl -fsSL "${PYGUARDIAN_REPO}/raw/main/deployment/scripts/docker-install.sh" -o docker-install.sh
|
||||
else
|
||||
wget -q "${PYGUARDIAN_REPO}/raw/main/deployment/scripts/docker-install.sh" -O docker-install.sh
|
||||
fi
|
||||
|
||||
chmod +x docker-install.sh
|
||||
|
||||
# Run Docker installation
|
||||
./docker-install.sh "$@"
|
||||
}
|
||||
|
||||
run_interactive_installation() {
|
||||
echo -e "${BLUE}[INFO]${NC} Starting interactive installation..."
|
||||
echo ""
|
||||
|
||||
echo "Select PyGuardian installation mode:"
|
||||
echo "1) Standalone server (all components on one server)"
|
||||
echo "2) Cluster controller (central management node)"
|
||||
echo "3) Cluster agent (managed node)"
|
||||
echo "4) Docker installation"
|
||||
echo ""
|
||||
|
||||
while true; do
|
||||
read -p "Enter your choice (1-4): " choice
|
||||
case $choice in
|
||||
1)
|
||||
echo -e "${GREEN}[SELECTED]${NC} Standalone installation"
|
||||
"$INSTALLER_PATH" --mode standalone
|
||||
break
|
||||
;;
|
||||
2)
|
||||
echo -e "${GREEN}[SELECTED]${NC} Cluster controller installation"
|
||||
"$INSTALLER_PATH" --mode controller
|
||||
break
|
||||
;;
|
||||
3)
|
||||
echo -e "${GREEN}[SELECTED]${NC} Cluster agent installation"
|
||||
echo ""
|
||||
read -p "Enter controller IP address: " controller_ip
|
||||
if [[ -z "$controller_ip" ]]; then
|
||||
echo -e "${RED}[ERROR]${NC} Controller IP is required for agent mode"
|
||||
continue
|
||||
fi
|
||||
"$INSTALLER_PATH" --mode agent --controller "$controller_ip"
|
||||
break
|
||||
;;
|
||||
4)
|
||||
echo -e "${GREEN}[SELECTED]${NC} Docker installation"
|
||||
run_docker_installation
|
||||
break
|
||||
;;
|
||||
*)
|
||||
echo -e "${RED}[ERROR]${NC} Invalid choice. Please select 1-4."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
main() {
|
||||
print_header
|
||||
|
||||
# Parse command line arguments
|
||||
MODE=""
|
||||
CONTROLLER_HOST=""
|
||||
USE_DOCKER=false
|
||||
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--mode)
|
||||
MODE="$2"
|
||||
shift 2
|
||||
;;
|
||||
--controller)
|
||||
CONTROLLER_HOST="$2"
|
||||
shift 2
|
||||
;;
|
||||
--docker)
|
||||
USE_DOCKER=true
|
||||
shift
|
||||
;;
|
||||
--help)
|
||||
print_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
echo -e "${RED}[ERROR]${NC} Unknown option: $1"
|
||||
print_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# System checks
|
||||
check_system
|
||||
|
||||
# Handle Docker installation
|
||||
if [[ "$USE_DOCKER" == true ]]; then
|
||||
run_docker_installation "$@"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
# Download detailed installer
|
||||
download_installer
|
||||
|
||||
# Run installation based on mode
|
||||
if [[ -n "$MODE" ]]; then
|
||||
echo -e "${BLUE}[INFO]${NC} Running ${MODE} installation..."
|
||||
|
||||
# Validate mode
|
||||
if [[ "$MODE" != "standalone" && "$MODE" != "controller" && "$MODE" != "agent" ]]; then
|
||||
echo -e "${RED}[ERROR]${NC} Invalid mode: $MODE"
|
||||
print_usage
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check controller host for agent mode
|
||||
if [[ "$MODE" == "agent" && -z "$CONTROLLER_HOST" ]]; then
|
||||
echo -e "${RED}[ERROR]${NC} Controller host is required for agent mode"
|
||||
print_usage
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Run installer with specified mode
|
||||
if [[ "$MODE" == "agent" ]]; then
|
||||
"$INSTALLER_PATH" --mode agent --controller "$CONTROLLER_HOST"
|
||||
else
|
||||
"$INSTALLER_PATH" --mode "$MODE"
|
||||
fi
|
||||
else
|
||||
# Interactive mode
|
||||
run_interactive_installation
|
||||
fi
|
||||
|
||||
# Cleanup
|
||||
if [[ -n "$INSTALLER_PATH" ]]; then
|
||||
rm -rf "$(dirname "$INSTALLER_PATH")"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo -e "${GREEN}[SUCCESS]${NC} PyGuardian installation completed!"
|
||||
echo ""
|
||||
echo "Next steps:"
|
||||
echo "1. Configure your Telegram bot token in /opt/pyguardian/config/config.yaml"
|
||||
echo "2. Review authentication settings in /opt/pyguardian/config/auth.yaml"
|
||||
echo "3. Start the service: systemctl start pyguardian"
|
||||
echo "4. Enable auto-start: systemctl enable pyguardian"
|
||||
echo ""
|
||||
echo -e "${YELLOW}[INFO]${NC} PyGuardian v2.1.0 includes:"
|
||||
echo " ✓ JWT-based agent authentication"
|
||||
echo " ✓ Encrypted agent credentials storage"
|
||||
echo " ✓ RESTful API server with authentication"
|
||||
echo " ✓ Comprehensive testing suite"
|
||||
echo " ✓ CI/CD pipeline with security scanning"
|
||||
echo ""
|
||||
echo "Documentation: ${PYGUARDIAN_REPO}/tree/main/documentation"
|
||||
echo "Support: https://github.com/SmartSolTech/PyGuardian/issues"
|
||||
}
|
||||
|
||||
# Handle script errors
|
||||
trap 'echo -e "${RED}[ERROR]${NC} Installation failed. Check logs above."; exit 1' ERR
|
||||
|
||||
# Run main function
|
||||
main "$@"
|
||||
@@ -1,370 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# PyGuardian Agent Installation Script
|
||||
# Usage: ./install_agent.sh --master <master_ip> --port <master_port>
|
||||
|
||||
set -e
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Default values
|
||||
MASTER_IP=""
|
||||
MASTER_PORT="8080"
|
||||
AGENT_PORT="8081"
|
||||
INSTALL_DIR="/opt/pyguardian-agent"
|
||||
SERVICE_NAME="pyguardian-agent"
|
||||
|
||||
# Functions
|
||||
log_info() {
|
||||
echo -e "${BLUE}[INFO]${NC} $1"
|
||||
}
|
||||
|
||||
log_success() {
|
||||
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||||
}
|
||||
|
||||
log_warning() {
|
||||
echo -e "${YELLOW}[WARNING]${NC} $1"
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1"
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "PyGuardian Agent Installation Script"
|
||||
echo ""
|
||||
echo "Usage: $0 --master <master_ip> [OPTIONS]"
|
||||
echo ""
|
||||
echo "Required:"
|
||||
echo " --master <ip> Master server IP address"
|
||||
echo ""
|
||||
echo "Optional:"
|
||||
echo " --port <port> Master server port (default: 8080)"
|
||||
echo " --agent-port <p> Agent listen port (default: 8081)"
|
||||
echo " --help Show this help"
|
||||
echo ""
|
||||
echo "Example:"
|
||||
echo " $0 --master 192.168.1.100 --port 8080"
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--master)
|
||||
MASTER_IP="$2"
|
||||
shift 2
|
||||
;;
|
||||
--port)
|
||||
MASTER_PORT="$2"
|
||||
shift 2
|
||||
;;
|
||||
--agent-port)
|
||||
AGENT_PORT="$2"
|
||||
shift 2
|
||||
;;
|
||||
--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
log_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Validate required parameters
|
||||
if [[ -z "$MASTER_IP" ]]; then
|
||||
log_error "Master IP is required"
|
||||
show_usage
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check if running as root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
log_error "This script must be run as root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Starting PyGuardian Agent installation..."
|
||||
log_info "Master Server: ${MASTER_IP}:${MASTER_PORT}"
|
||||
log_info "Agent Port: ${AGENT_PORT}"
|
||||
|
||||
# Check system requirements
|
||||
log_info "Checking system requirements..."
|
||||
|
||||
# Check Python version
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
log_error "Python3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
REQUIRED_VERSION="3.10"
|
||||
|
||||
if ! python3 -c "import sys; sys.exit(0 if sys.version_info >= (3, 10) else 1)" 2>/dev/null; then
|
||||
log_error "Python 3.10+ is required. Found: $PYTHON_VERSION"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_success "Python version check passed: $PYTHON_VERSION"
|
||||
|
||||
# Install system dependencies
|
||||
log_info "Installing system dependencies..."
|
||||
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv curl wget
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip curl wget
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip curl wget
|
||||
else
|
||||
log_warning "Unknown package manager. Please install python3-pip manually."
|
||||
fi
|
||||
|
||||
# Create installation directory
|
||||
log_info "Creating installation directory..."
|
||||
mkdir -p "$INSTALL_DIR"
|
||||
mkdir -p /etc/pyguardian-agent
|
||||
mkdir -p /var/log/pyguardian-agent
|
||||
|
||||
# Create agent configuration
|
||||
log_info "Creating agent configuration..."
|
||||
cat > /etc/pyguardian-agent/config.yaml << EOF
|
||||
# PyGuardian Agent Configuration
|
||||
|
||||
agent:
|
||||
port: ${AGENT_PORT}
|
||||
master_host: "${MASTER_IP}"
|
||||
master_port: ${MASTER_PORT}
|
||||
heartbeat_interval: 30
|
||||
max_reconnect_attempts: 10
|
||||
reconnect_delay: 5
|
||||
|
||||
logging:
|
||||
level: INFO
|
||||
file: /var/log/pyguardian-agent/agent.log
|
||||
max_size: 10MB
|
||||
backup_count: 5
|
||||
|
||||
security:
|
||||
ssl_verify: true
|
||||
api_key_file: /etc/pyguardian-agent/api.key
|
||||
EOF
|
||||
|
||||
# Create simple agent script
|
||||
log_info "Creating agent script..."
|
||||
cat > "$INSTALL_DIR/agent.py" << 'EOF'
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian Agent
|
||||
Lightweight agent for cluster management
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import signal
|
||||
import sys
|
||||
import time
|
||||
import yaml
|
||||
import aiohttp
|
||||
import psutil
|
||||
from pathlib import Path
|
||||
|
||||
class PyGuardianAgent:
|
||||
def __init__(self, config_path="/etc/pyguardian-agent/config.yaml"):
|
||||
self.config_path = config_path
|
||||
self.config = self.load_config()
|
||||
self.setup_logging()
|
||||
self.session = None
|
||||
self.running = False
|
||||
|
||||
def load_config(self):
|
||||
"""Load agent configuration"""
|
||||
try:
|
||||
with open(self.config_path, 'r') as f:
|
||||
return yaml.safe_load(f)
|
||||
except Exception as e:
|
||||
print(f"Error loading config: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
def setup_logging(self):
|
||||
"""Setup logging configuration"""
|
||||
log_file = self.config.get('logging', {}).get('file', '/var/log/pyguardian-agent/agent.log')
|
||||
log_level = getattr(logging, self.config.get('logging', {}).get('level', 'INFO'))
|
||||
|
||||
# Create log directory if it doesn't exist
|
||||
Path(log_file).parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.FileHandler(log_file),
|
||||
logging.StreamHandler()
|
||||
]
|
||||
)
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
async def get_system_info(self):
|
||||
"""Get system information"""
|
||||
try:
|
||||
cpu_percent = psutil.cpu_percent(interval=1)
|
||||
memory = psutil.virtual_memory()
|
||||
disk = psutil.disk_usage('/')
|
||||
|
||||
return {
|
||||
'status': 'online',
|
||||
'cpu_percent': cpu_percent,
|
||||
'memory_percent': memory.percent,
|
||||
'memory_total': memory.total,
|
||||
'memory_used': memory.used,
|
||||
'disk_percent': disk.percent,
|
||||
'disk_total': disk.total,
|
||||
'disk_used': disk.used,
|
||||
'timestamp': time.time()
|
||||
}
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error getting system info: {e}")
|
||||
return {'status': 'error', 'error': str(e)}
|
||||
|
||||
async def send_heartbeat(self):
|
||||
"""Send heartbeat to master server"""
|
||||
try:
|
||||
system_info = await self.get_system_info()
|
||||
|
||||
master_url = f"http://{self.config['agent']['master_host']}:{self.config['agent']['master_port']}"
|
||||
|
||||
async with self.session.post(
|
||||
f"{master_url}/api/agent/heartbeat",
|
||||
json=system_info,
|
||||
timeout=10
|
||||
) as response:
|
||||
if response.status == 200:
|
||||
self.logger.debug("Heartbeat sent successfully")
|
||||
else:
|
||||
self.logger.warning(f"Heartbeat failed with status: {response.status}")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error sending heartbeat: {e}")
|
||||
|
||||
async def heartbeat_loop(self):
|
||||
"""Main heartbeat loop"""
|
||||
interval = self.config.get('agent', {}).get('heartbeat_interval', 30)
|
||||
|
||||
while self.running:
|
||||
await self.send_heartbeat()
|
||||
await asyncio.sleep(interval)
|
||||
|
||||
async def start(self):
|
||||
"""Start the agent"""
|
||||
self.logger.info("Starting PyGuardian Agent...")
|
||||
self.running = True
|
||||
|
||||
# Create HTTP session
|
||||
self.session = aiohttp.ClientSession()
|
||||
|
||||
try:
|
||||
# Start heartbeat loop
|
||||
await self.heartbeat_loop()
|
||||
except Exception as e:
|
||||
self.logger.error(f"Agent error: {e}")
|
||||
finally:
|
||||
await self.stop()
|
||||
|
||||
async def stop(self):
|
||||
"""Stop the agent"""
|
||||
self.logger.info("Stopping PyGuardian Agent...")
|
||||
self.running = False
|
||||
|
||||
if self.session:
|
||||
await self.session.close()
|
||||
|
||||
async def main():
|
||||
agent = PyGuardianAgent()
|
||||
|
||||
# Handle signals
|
||||
def signal_handler(signum, frame):
|
||||
print(f"\nReceived signal {signum}, shutting down...")
|
||||
asyncio.create_task(agent.stop())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
try:
|
||||
await agent.start()
|
||||
except KeyboardInterrupt:
|
||||
await agent.stop()
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
EOF
|
||||
|
||||
chmod +x "$INSTALL_DIR/agent.py"
|
||||
|
||||
# Install Python dependencies
|
||||
log_info "Installing Python dependencies..."
|
||||
python3 -m pip install --upgrade pip
|
||||
python3 -m pip install aiohttp pyyaml psutil
|
||||
|
||||
# Create systemd service
|
||||
log_info "Creating systemd service..."
|
||||
cat > "/etc/systemd/system/${SERVICE_NAME}.service" << EOF
|
||||
[Unit]
|
||||
Description=PyGuardian Agent
|
||||
After=network.target
|
||||
Wants=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=root
|
||||
Group=root
|
||||
WorkingDirectory=${INSTALL_DIR}
|
||||
ExecStart=/usr/bin/python3 ${INSTALL_DIR}/agent.py
|
||||
ExecReload=/bin/kill -HUP \$MAINPID
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
# Reload systemd and enable service
|
||||
log_info "Enabling systemd service..."
|
||||
systemctl daemon-reload
|
||||
systemctl enable "$SERVICE_NAME"
|
||||
|
||||
# Start the service
|
||||
log_info "Starting PyGuardian Agent..."
|
||||
systemctl start "$SERVICE_NAME"
|
||||
|
||||
# Check service status
|
||||
sleep 3
|
||||
if systemctl is-active --quiet "$SERVICE_NAME"; then
|
||||
log_success "PyGuardian Agent installed and started successfully!"
|
||||
log_info "Service status: $(systemctl is-active $SERVICE_NAME)"
|
||||
log_info "Check logs: journalctl -u $SERVICE_NAME -f"
|
||||
log_info "Agent config: /etc/pyguardian-agent/config.yaml"
|
||||
log_info "Agent logs: /var/log/pyguardian-agent/agent.log"
|
||||
else
|
||||
log_error "PyGuardian Agent failed to start"
|
||||
log_info "Check service status: systemctl status $SERVICE_NAME"
|
||||
log_info "Check logs: journalctl -u $SERVICE_NAME"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_success "Installation completed!"
|
||||
log_info "The agent should now be visible in your PyGuardian master server."
|
||||
log_info "Use '/agents' command in Telegram to verify the agent connection."
|
||||
EOF
|
||||
@@ -1,370 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# PyGuardian Agent Installation Script
|
||||
# Usage: ./install_agent.sh --master <master_ip> --port <master_port>
|
||||
|
||||
set -e
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Default values
|
||||
MASTER_IP=""
|
||||
MASTER_PORT="8080"
|
||||
AGENT_PORT="8081"
|
||||
INSTALL_DIR="/opt/pyguardian-agent"
|
||||
SERVICE_NAME="pyguardian-agent"
|
||||
|
||||
# Functions
|
||||
log_info() {
|
||||
echo -e "${BLUE}[INFO]${NC} $1"
|
||||
}
|
||||
|
||||
log_success() {
|
||||
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||||
}
|
||||
|
||||
log_warning() {
|
||||
echo -e "${YELLOW}[WARNING]${NC} $1"
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1"
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "PyGuardian Agent Installation Script"
|
||||
echo ""
|
||||
echo "Usage: $0 --master <master_ip> [OPTIONS]"
|
||||
echo ""
|
||||
echo "Required:"
|
||||
echo " --master <ip> Master server IP address"
|
||||
echo ""
|
||||
echo "Optional:"
|
||||
echo " --port <port> Master server port (default: 8080)"
|
||||
echo " --agent-port <p> Agent listen port (default: 8081)"
|
||||
echo " --help Show this help"
|
||||
echo ""
|
||||
echo "Example:"
|
||||
echo " $0 --master 192.168.1.100 --port 8080"
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--master)
|
||||
MASTER_IP="$2"
|
||||
shift 2
|
||||
;;
|
||||
--port)
|
||||
MASTER_PORT="$2"
|
||||
shift 2
|
||||
;;
|
||||
--agent-port)
|
||||
AGENT_PORT="$2"
|
||||
shift 2
|
||||
;;
|
||||
--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
log_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Validate required parameters
|
||||
if [[ -z "$MASTER_IP" ]]; then
|
||||
log_error "Master IP is required"
|
||||
show_usage
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check if running as root
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
log_error "This script must be run as root"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_info "Starting PyGuardian Agent installation..."
|
||||
log_info "Master Server: ${MASTER_IP}:${MASTER_PORT}"
|
||||
log_info "Agent Port: ${AGENT_PORT}"
|
||||
|
||||
# Check system requirements
|
||||
log_info "Checking system requirements..."
|
||||
|
||||
# Check Python version
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
log_error "Python3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
REQUIRED_VERSION="3.10"
|
||||
|
||||
if ! python3 -c "import sys; sys.exit(0 if sys.version_info >= (3, 10) else 1)" 2>/dev/null; then
|
||||
log_error "Python 3.10+ is required. Found: $PYTHON_VERSION"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_success "Python version check passed: $PYTHON_VERSION"
|
||||
|
||||
# Install system dependencies
|
||||
log_info "Installing system dependencies..."
|
||||
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv curl wget
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip curl wget
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip curl wget
|
||||
else
|
||||
log_warning "Unknown package manager. Please install python3-pip manually."
|
||||
fi
|
||||
|
||||
# Create installation directory
|
||||
log_info "Creating installation directory..."
|
||||
mkdir -p "$INSTALL_DIR"
|
||||
mkdir -p /etc/pyguardian-agent
|
||||
mkdir -p /var/log/pyguardian-agent
|
||||
|
||||
# Create agent configuration
|
||||
log_info "Creating agent configuration..."
|
||||
cat > /etc/pyguardian-agent/config.yaml << EOF
|
||||
# PyGuardian Agent Configuration
|
||||
|
||||
agent:
|
||||
port: ${AGENT_PORT}
|
||||
master_host: "${MASTER_IP}"
|
||||
master_port: ${MASTER_PORT}
|
||||
heartbeat_interval: 30
|
||||
max_reconnect_attempts: 10
|
||||
reconnect_delay: 5
|
||||
|
||||
logging:
|
||||
level: INFO
|
||||
file: /var/log/pyguardian-agent/agent.log
|
||||
max_size: 10MB
|
||||
backup_count: 5
|
||||
|
||||
security:
|
||||
ssl_verify: true
|
||||
api_key_file: /etc/pyguardian-agent/api.key
|
||||
EOF
|
||||
|
||||
# Create simple agent script
|
||||
log_info "Creating agent script..."
|
||||
cat > "$INSTALL_DIR/agent.py" << 'EOF'
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian Agent
|
||||
Lightweight agent for cluster management
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import signal
|
||||
import sys
|
||||
import time
|
||||
import yaml
|
||||
import aiohttp
|
||||
import psutil
|
||||
from pathlib import Path
|
||||
|
||||
class PyGuardianAgent:
|
||||
def __init__(self, config_path="/etc/pyguardian-agent/config.yaml"):
|
||||
self.config_path = config_path
|
||||
self.config = self.load_config()
|
||||
self.setup_logging()
|
||||
self.session = None
|
||||
self.running = False
|
||||
|
||||
def load_config(self):
|
||||
"""Load agent configuration"""
|
||||
try:
|
||||
with open(self.config_path, 'r') as f:
|
||||
return yaml.safe_load(f)
|
||||
except Exception as e:
|
||||
print(f"Error loading config: {e}")
|
||||
sys.exit(1)
|
||||
|
||||
def setup_logging(self):
|
||||
"""Setup logging configuration"""
|
||||
log_file = self.config.get('logging', {}).get('file', '/var/log/pyguardian-agent/agent.log')
|
||||
log_level = getattr(logging, self.config.get('logging', {}).get('level', 'INFO'))
|
||||
|
||||
# Create log directory if it doesn't exist
|
||||
Path(log_file).parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||
handlers=[
|
||||
logging.FileHandler(log_file),
|
||||
logging.StreamHandler()
|
||||
]
|
||||
)
|
||||
self.logger = logging.getLogger(__name__)
|
||||
|
||||
async def get_system_info(self):
|
||||
"""Get system information"""
|
||||
try:
|
||||
cpu_percent = psutil.cpu_percent(interval=1)
|
||||
memory = psutil.virtual_memory()
|
||||
disk = psutil.disk_usage('/')
|
||||
|
||||
return {
|
||||
'status': 'online',
|
||||
'cpu_percent': cpu_percent,
|
||||
'memory_percent': memory.percent,
|
||||
'memory_total': memory.total,
|
||||
'memory_used': memory.used,
|
||||
'disk_percent': disk.percent,
|
||||
'disk_total': disk.total,
|
||||
'disk_used': disk.used,
|
||||
'timestamp': time.time()
|
||||
}
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error getting system info: {e}")
|
||||
return {'status': 'error', 'error': str(e)}
|
||||
|
||||
async def send_heartbeat(self):
|
||||
"""Send heartbeat to master server"""
|
||||
try:
|
||||
system_info = await self.get_system_info()
|
||||
|
||||
master_url = f"http://{self.config['agent']['master_host']}:{self.config['agent']['master_port']}"
|
||||
|
||||
async with self.session.post(
|
||||
f"{master_url}/api/agent/heartbeat",
|
||||
json=system_info,
|
||||
timeout=10
|
||||
) as response:
|
||||
if response.status == 200:
|
||||
self.logger.debug("Heartbeat sent successfully")
|
||||
else:
|
||||
self.logger.warning(f"Heartbeat failed with status: {response.status}")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Error sending heartbeat: {e}")
|
||||
|
||||
async def heartbeat_loop(self):
|
||||
"""Main heartbeat loop"""
|
||||
interval = self.config.get('agent', {}).get('heartbeat_interval', 30)
|
||||
|
||||
while self.running:
|
||||
await self.send_heartbeat()
|
||||
await asyncio.sleep(interval)
|
||||
|
||||
async def start(self):
|
||||
"""Start the agent"""
|
||||
self.logger.info("Starting PyGuardian Agent...")
|
||||
self.running = True
|
||||
|
||||
# Create HTTP session
|
||||
self.session = aiohttp.ClientSession()
|
||||
|
||||
try:
|
||||
# Start heartbeat loop
|
||||
await self.heartbeat_loop()
|
||||
except Exception as e:
|
||||
self.logger.error(f"Agent error: {e}")
|
||||
finally:
|
||||
await self.stop()
|
||||
|
||||
async def stop(self):
|
||||
"""Stop the agent"""
|
||||
self.logger.info("Stopping PyGuardian Agent...")
|
||||
self.running = False
|
||||
|
||||
if self.session:
|
||||
await self.session.close()
|
||||
|
||||
async def main():
|
||||
agent = PyGuardianAgent()
|
||||
|
||||
# Handle signals
|
||||
def signal_handler(signum, frame):
|
||||
print(f"\nReceived signal {signum}, shutting down...")
|
||||
asyncio.create_task(agent.stop())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
try:
|
||||
await agent.start()
|
||||
except KeyboardInterrupt:
|
||||
await agent.stop()
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
EOF
|
||||
|
||||
chmod +x "$INSTALL_DIR/agent.py"
|
||||
|
||||
# Install Python dependencies
|
||||
log_info "Installing Python dependencies..."
|
||||
python3 -m pip install --upgrade pip
|
||||
python3 -m pip install aiohttp pyyaml psutil
|
||||
|
||||
# Create systemd service
|
||||
log_info "Creating systemd service..."
|
||||
cat > "/etc/systemd/system/${SERVICE_NAME}.service" << EOF
|
||||
[Unit]
|
||||
Description=PyGuardian Agent
|
||||
After=network.target
|
||||
Wants=network.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=root
|
||||
Group=root
|
||||
WorkingDirectory=${INSTALL_DIR}
|
||||
ExecStart=/usr/bin/python3 ${INSTALL_DIR}/agent.py
|
||||
ExecReload=/bin/kill -HUP \$MAINPID
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
# Reload systemd and enable service
|
||||
log_info "Enabling systemd service..."
|
||||
systemctl daemon-reload
|
||||
systemctl enable "$SERVICE_NAME"
|
||||
|
||||
# Start the service
|
||||
log_info "Starting PyGuardian Agent..."
|
||||
systemctl start "$SERVICE_NAME"
|
||||
|
||||
# Check service status
|
||||
sleep 3
|
||||
if systemctl is-active --quiet "$SERVICE_NAME"; then
|
||||
log_success "PyGuardian Agent installed and started successfully!"
|
||||
log_info "Service status: $(systemctl is-active $SERVICE_NAME)"
|
||||
log_info "Check logs: journalctl -u $SERVICE_NAME -f"
|
||||
log_info "Agent config: /etc/pyguardian-agent/config.yaml"
|
||||
log_info "Agent logs: /var/log/pyguardian-agent/agent.log"
|
||||
else
|
||||
log_error "PyGuardian Agent failed to start"
|
||||
log_info "Check service status: systemctl status $SERVICE_NAME"
|
||||
log_info "Check logs: journalctl -u $SERVICE_NAME"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
log_success "Installation completed!"
|
||||
log_info "The agent should now be visible in your PyGuardian master server."
|
||||
log_info "Use '/agents' command in Telegram to verify the agent connection."
|
||||
EOF
|
||||
@@ -1,389 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: PyGuardian Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация детектора атак
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 4. Инициализация Telegram бота
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,389 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация детектора атак
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 4. Инициализация Telegram бота
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,392 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация детектора атак
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 4. Инициализация Telegram бота
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,397 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация детектора атак
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 4. Инициализация Telegram бота
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,414 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
encryption_key = self.config.get('security', {}).get('encryption_key')
|
||||
self.password_manager = PasswordManager(encryption_key)
|
||||
self.session_manager = SessionManager()
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.session_manager,
|
||||
self.password_manager,
|
||||
self.config.get('security', {})
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
},
|
||||
self.security_manager
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,411 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
self.password_manager = PasswordManager()
|
||||
self.session_manager = SessionManager()
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.config.get('security', {})
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
{
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
},
|
||||
self.security_manager
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,412 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
self.password_manager = PasswordManager()
|
||||
self.session_manager = SessionManager()
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.config.get('security', {})
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,414 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
password_config = self.config.get('passwords', {})
|
||||
self.password_manager = PasswordManager(password_config)
|
||||
self.session_manager = SessionManager()
|
||||
security_config = self.config.get('security', {})
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
security_config
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,414 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
password_config = self.config.get('passwords', {})
|
||||
self.password_manager = PasswordManager(password_config)
|
||||
self.session_manager = SessionManager()
|
||||
security_config = self.config.get('security', {})
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
security_config
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,416 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
from src.cluster_manager import ClusterManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
self.cluster_manager: Optional[ClusterManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
password_config = self.config.get('passwords', {})
|
||||
self.password_manager = PasswordManager(password_config)
|
||||
self.session_manager = SessionManager()
|
||||
security_config = self.config.get('security', {})
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
security_config
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 5. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,426 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
from src.cluster_manager import ClusterManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
self.cluster_manager: Optional[ClusterManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
password_config = self.config.get('passwords', {})
|
||||
self.password_manager = PasswordManager(password_config)
|
||||
self.session_manager = SessionManager()
|
||||
security_config = self.config.get('security', {})
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
security_config
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager,
|
||||
self.cluster_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 6. Инициализация менеджера кластера
|
||||
cluster_config = self.config.get('cluster', {})
|
||||
self.cluster_manager = ClusterManager(self.storage, cluster_config)
|
||||
await self.cluster_manager.load_agents()
|
||||
self.logger.info("Cluster Manager инициализирован")
|
||||
|
||||
# Обновляем ссылку в боте
|
||||
self.telegram_bot.cluster_manager = self.cluster_manager
|
||||
|
||||
# 7. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,426 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
PyGuardian - Linux Server Protection System
|
||||
Главный файл для запуска системы мониторинга и защиты
|
||||
|
||||
Автор: SmartSolTech Team
|
||||
Лицензия: MIT
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
import signal
|
||||
import logging
|
||||
import logging.handlers
|
||||
import yaml
|
||||
import sys
|
||||
import os
|
||||
from pathlib import Path
|
||||
from typing import Dict, Optional
|
||||
|
||||
# Добавляем src в путь для импортов
|
||||
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||||
|
||||
from src.storage import Storage
|
||||
from src.firewall import FirewallManager
|
||||
from src.monitor import LogMonitor, AttackDetector
|
||||
from src.bot import TelegramBot, NotificationManager
|
||||
from src.security import SecurityManager
|
||||
from src.sessions import SessionManager
|
||||
from src.password_utils import PasswordManager
|
||||
from src.cluster_manager import ClusterManager
|
||||
|
||||
|
||||
class PyGuardian:
|
||||
"""Главный класс системы PyGuardian"""
|
||||
|
||||
def __init__(self, config_path: str = None):
|
||||
self.config_path = config_path or "config/config.yaml"
|
||||
self.config: Optional[Dict] = None
|
||||
self.logger = None
|
||||
|
||||
# Компоненты системы
|
||||
self.storage: Optional[Storage] = None
|
||||
self.firewall_manager: Optional[FirewallManager] = None
|
||||
self.log_monitor: Optional[LogMonitor] = None
|
||||
self.attack_detector: Optional[AttackDetector] = None
|
||||
self.telegram_bot: Optional[TelegramBot] = None
|
||||
self.notification_manager: Optional[NotificationManager] = None
|
||||
|
||||
# Новые компоненты безопасности
|
||||
self.security_manager: Optional[SecurityManager] = None
|
||||
self.session_manager: Optional[SessionManager] = None
|
||||
self.password_manager: Optional[PasswordManager] = None
|
||||
self.cluster_manager: Optional[ClusterManager] = None
|
||||
|
||||
# Флаги состояния
|
||||
self.running = False
|
||||
self.shutdown_event = asyncio.Event()
|
||||
|
||||
# Задачи
|
||||
self.monitor_task: Optional[asyncio.Task] = None
|
||||
self.bot_task: Optional[asyncio.Task] = None
|
||||
self.cleanup_task: Optional[asyncio.Task] = None
|
||||
self.unban_checker_task: Optional[asyncio.Task] = None
|
||||
|
||||
def setup_logging(self) -> None:
|
||||
"""Настройка логирования"""
|
||||
log_config = self.config.get('logging', {})
|
||||
log_file = log_config.get('log_file', '/var/log/pyguardian.log')
|
||||
log_level = getattr(logging, log_config.get('log_level', 'INFO').upper())
|
||||
max_log_size = log_config.get('max_log_size', 10485760) # 10MB
|
||||
backup_count = log_config.get('backup_count', 5)
|
||||
|
||||
# Создаем директорию для логов если не существует
|
||||
os.makedirs(os.path.dirname(log_file), exist_ok=True)
|
||||
|
||||
# Настройка форматирования
|
||||
formatter = logging.Formatter(
|
||||
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||||
)
|
||||
|
||||
# Ротируемый файловый обработчик
|
||||
file_handler = logging.handlers.RotatingFileHandler(
|
||||
log_file,
|
||||
maxBytes=max_log_size,
|
||||
backupCount=backup_count
|
||||
)
|
||||
file_handler.setFormatter(formatter)
|
||||
|
||||
# Консольный обработчик
|
||||
console_handler = logging.StreamHandler()
|
||||
console_handler.setFormatter(formatter)
|
||||
|
||||
# Настройка root logger
|
||||
logging.basicConfig(
|
||||
level=log_level,
|
||||
handlers=[file_handler, console_handler]
|
||||
)
|
||||
|
||||
self.logger = logging.getLogger('PyGuardian')
|
||||
self.logger.info("Логирование настроено успешно")
|
||||
|
||||
def load_config(self) -> bool:
|
||||
"""Загрузка конфигурации"""
|
||||
try:
|
||||
with open(self.config_path, 'r', encoding='utf-8') as file:
|
||||
self.config = yaml.safe_load(file)
|
||||
|
||||
# Валидация основных параметров
|
||||
required_keys = ['telegram', 'security', 'monitoring', 'firewall', 'storage']
|
||||
for key in required_keys:
|
||||
if key not in self.config:
|
||||
raise ValueError(f"Отсутствует секция '{key}' в конфигурации")
|
||||
|
||||
# Проверяем обязательные параметры Telegram
|
||||
telegram_config = self.config['telegram']
|
||||
if not telegram_config.get('bot_token') or not telegram_config.get('admin_id'):
|
||||
raise ValueError("Не указаны bot_token или admin_id в секции telegram")
|
||||
|
||||
self.logger.info("Конфигурация загружена успешно")
|
||||
return True
|
||||
|
||||
except FileNotFoundError:
|
||||
print(f"Файл конфигурации не найден: {self.config_path}")
|
||||
return False
|
||||
except yaml.YAMLError as e:
|
||||
print(f"Ошибка парсинга YAML: {e}")
|
||||
return False
|
||||
except Exception as e:
|
||||
print(f"Ошибка загрузки конфигурации: {e}")
|
||||
return False
|
||||
|
||||
async def initialize_components(self) -> bool:
|
||||
"""Инициализация всех компонентов системы"""
|
||||
try:
|
||||
self.logger.info("Инициализация компонентов...")
|
||||
|
||||
# Создаем директории
|
||||
storage_path = self.config['storage']['database_path']
|
||||
os.makedirs(os.path.dirname(storage_path), exist_ok=True)
|
||||
|
||||
# 1. Инициализация хранилища
|
||||
self.storage = Storage(storage_path)
|
||||
await self.storage.init_database()
|
||||
self.logger.info("Storage инициализирован")
|
||||
|
||||
# 2. Инициализация firewall
|
||||
self.firewall_manager = FirewallManager(self.config['firewall'])
|
||||
if not await self.firewall_manager.setup():
|
||||
raise RuntimeError("Не удалось настроить firewall")
|
||||
self.logger.info("Firewall Manager инициализирован")
|
||||
|
||||
# 3. Инициализация новых менеджеров безопасности
|
||||
password_config = self.config.get('passwords', {})
|
||||
self.password_manager = PasswordManager(password_config)
|
||||
self.session_manager = SessionManager()
|
||||
security_config = self.config.get('security', {})
|
||||
self.security_manager = SecurityManager(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
security_config
|
||||
)
|
||||
self.logger.info("Менеджеры безопасности инициализированы")
|
||||
|
||||
# 4. Инициализация детектора атак с security manager
|
||||
attack_config = {
|
||||
**self.config['security'],
|
||||
'whitelist': self.config.get('whitelist', [])
|
||||
}
|
||||
self.attack_detector = AttackDetector(
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.security_manager,
|
||||
attack_config
|
||||
)
|
||||
self.logger.info("Attack Detector инициализирован")
|
||||
|
||||
# 5. Инициализация Telegram бота с новыми менеджерами
|
||||
self.telegram_bot = TelegramBot(
|
||||
self.config['telegram'],
|
||||
self.storage,
|
||||
self.firewall_manager,
|
||||
self.attack_detector,
|
||||
self.security_manager,
|
||||
self.session_manager,
|
||||
self.password_manager,
|
||||
self.cluster_manager
|
||||
)
|
||||
self.logger.info("Telegram Bot инициализирован")
|
||||
|
||||
# 6. Инициализация менеджера кластера
|
||||
cluster_config = self.config.get('cluster', {})
|
||||
self.cluster_manager = ClusterManager(self.storage, cluster_config)
|
||||
await self.cluster_manager.load_agents()
|
||||
self.logger.info("Cluster Manager инициализирован")
|
||||
|
||||
# Обновляем ссылку в боте
|
||||
self.telegram_bot.cluster_manager = self.cluster_manager
|
||||
|
||||
# 7. Инициализация менеджера уведомлений
|
||||
self.notification_manager = NotificationManager(self.telegram_bot)
|
||||
|
||||
# Связываем detector с уведомлениями
|
||||
self.attack_detector.set_callbacks(
|
||||
ban_callback=self.notification_manager.on_ip_banned,
|
||||
unban_callback=self.notification_manager.on_ip_unbanned
|
||||
)
|
||||
|
||||
# 6. Инициализация монитора логов
|
||||
self.log_monitor = LogMonitor(
|
||||
self.config['monitoring'],
|
||||
event_callback=self.attack_detector.process_event
|
||||
)
|
||||
self.logger.info("Log Monitor инициализирован")
|
||||
|
||||
self.logger.info("Все компоненты инициализированы успешно")
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка инициализации компонентов: {e}")
|
||||
return False
|
||||
|
||||
async def start_background_tasks(self) -> None:
|
||||
"""Запуск фоновых задач"""
|
||||
try:
|
||||
self.logger.info("Запуск фоновых задач...")
|
||||
|
||||
# 1. Задача мониторинга логов
|
||||
self.monitor_task = asyncio.create_task(
|
||||
self.log_monitor.start(),
|
||||
name="log_monitor"
|
||||
)
|
||||
|
||||
# 2. Задача Telegram бота
|
||||
self.bot_task = asyncio.create_task(
|
||||
self.telegram_bot.start_bot(),
|
||||
name="telegram_bot"
|
||||
)
|
||||
|
||||
# 3. Задача периодической очистки
|
||||
self.cleanup_task = asyncio.create_task(
|
||||
self.periodic_cleanup(),
|
||||
name="periodic_cleanup"
|
||||
)
|
||||
|
||||
# 4. Задача проверки истекших банов
|
||||
self.unban_checker_task = asyncio.create_task(
|
||||
self.periodic_unban_check(),
|
||||
name="unban_checker"
|
||||
)
|
||||
|
||||
self.logger.info("Фоновые задачи запущены")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка запуска фоновых задач: {e}")
|
||||
raise
|
||||
|
||||
async def periodic_cleanup(self) -> None:
|
||||
"""Периодическая очистка данных"""
|
||||
cleanup_interval = self.config.get('performance', {}).get('cleanup_interval', 3600)
|
||||
max_records_age = self.config.get('performance', {}).get('max_records_age', 604800)
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(cleanup_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
# Очистка старых записей
|
||||
deleted_count = await self.storage.cleanup_old_records(
|
||||
days=max_records_age // 86400
|
||||
)
|
||||
|
||||
# Обновление статистики
|
||||
await self.storage.update_daily_stats()
|
||||
|
||||
# Очистка firewall от устаревших банов
|
||||
valid_ips = [ban['ip'] for ban in await self.storage.get_banned_ips()]
|
||||
removed_count = await self.firewall_manager.cleanup_expired_bans(valid_ips)
|
||||
|
||||
if deleted_count > 0 or removed_count > 0:
|
||||
self.logger.info(f"Очистка: удалено {deleted_count} записей, {removed_count} правил firewall")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_cleanup: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
async def periodic_unban_check(self) -> None:
|
||||
"""Периодическая проверка истекших банов"""
|
||||
check_interval = 300 # Проверяем каждые 5 минут
|
||||
|
||||
while self.running:
|
||||
try:
|
||||
await asyncio.sleep(check_interval)
|
||||
|
||||
if not self.running:
|
||||
break
|
||||
|
||||
await self.attack_detector.check_expired_bans()
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка в periodic_unban_check: {e}")
|
||||
await asyncio.sleep(60) # Ждем минуту перед повтором
|
||||
|
||||
def setup_signal_handlers(self) -> None:
|
||||
"""Настройка обработчиков сигналов"""
|
||||
def signal_handler(signum, frame):
|
||||
self.logger.info(f"Получен сигнал {signum}")
|
||||
asyncio.create_task(self.shutdown())
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
signal.signal(signal.SIGTERM, signal_handler)
|
||||
|
||||
if hasattr(signal, 'SIGHUP'):
|
||||
signal.signal(signal.SIGHUP, signal_handler)
|
||||
|
||||
async def shutdown(self) -> None:
|
||||
"""Graceful shutdown"""
|
||||
if not self.running:
|
||||
return
|
||||
|
||||
self.logger.info("Начало graceful shutdown...")
|
||||
self.running = False
|
||||
|
||||
try:
|
||||
# Останавливаем мониторинг логов
|
||||
if self.log_monitor:
|
||||
await self.log_monitor.stop()
|
||||
|
||||
# Отменяем фоновые задачи
|
||||
tasks_to_cancel = [
|
||||
self.monitor_task,
|
||||
self.cleanup_task,
|
||||
self.unban_checker_task
|
||||
]
|
||||
|
||||
for task in tasks_to_cancel:
|
||||
if task and not task.done():
|
||||
task.cancel()
|
||||
try:
|
||||
await task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
# Останавливаем Telegram бота
|
||||
if self.telegram_bot:
|
||||
await self.telegram_bot.stop_bot()
|
||||
|
||||
# Отменяем задачу бота отдельно
|
||||
if self.bot_task and not self.bot_task.done():
|
||||
self.bot_task.cancel()
|
||||
try:
|
||||
await self.bot_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
self.logger.info("Graceful shutdown завершен")
|
||||
|
||||
except Exception as e:
|
||||
self.logger.error(f"Ошибка при shutdown: {e}")
|
||||
finally:
|
||||
self.shutdown_event.set()
|
||||
|
||||
async def run(self) -> None:
|
||||
"""Основной цикл работы"""
|
||||
try:
|
||||
# Загрузка конфигурации
|
||||
if not self.load_config():
|
||||
return
|
||||
|
||||
# Настройка логирования
|
||||
self.setup_logging()
|
||||
|
||||
# Настройка обработчиков сигналов
|
||||
self.setup_signal_handlers()
|
||||
|
||||
# Инициализация компонентов
|
||||
if not await self.initialize_components():
|
||||
self.logger.error("Не удалось инициализировать компоненты")
|
||||
return
|
||||
|
||||
# Установка флага работы
|
||||
self.running = True
|
||||
|
||||
# Запуск фоновых задач
|
||||
await self.start_background_tasks()
|
||||
|
||||
self.logger.info("PyGuardian запущен и готов к работе")
|
||||
|
||||
# Ожидание сигнала к остановке
|
||||
await self.shutdown_event.wait()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
self.logger.info("Получен KeyboardInterrupt")
|
||||
except Exception as e:
|
||||
self.logger.error(f"Критическая ошибка: {e}")
|
||||
if self.notification_manager:
|
||||
await self.notification_manager.on_system_error(str(e))
|
||||
finally:
|
||||
await self.shutdown()
|
||||
|
||||
|
||||
async def main():
|
||||
"""Главная функция"""
|
||||
# Проверяем аргументы командной строки
|
||||
config_path = None
|
||||
if len(sys.argv) > 1:
|
||||
config_path = sys.argv[1]
|
||||
|
||||
# Проверяем права root (для работы с iptables/nftables)
|
||||
if os.geteuid() != 0:
|
||||
print("⚠️ Предупреждение: PyGuardian рекомендуется запускать от root для работы с firewall")
|
||||
|
||||
# Создаем и запускаем PyGuardian
|
||||
guardian = PyGuardian(config_path)
|
||||
await guardian.run()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
asyncio.run(main())
|
||||
except KeyboardInterrupt:
|
||||
print("\nПрерывание пользователем")
|
||||
except Exception as e:
|
||||
print(f"Фатальная ошибка: {e}")
|
||||
sys.exit(1)
|
||||
@@ -1,74 +0,0 @@
|
||||
site_name: PyGuardian Documentation
|
||||
site_description: AI-Powered Security & Cluster Management System
|
||||
site_author: SmartSolTech
|
||||
site_url: https://smartsoltech.github.io/PyGuardian
|
||||
|
||||
repo_name: SmartSolTech/PyGuardian
|
||||
repo_url: https://github.com/SmartSolTech/PyGuardian
|
||||
|
||||
theme:
|
||||
name: material
|
||||
palette:
|
||||
- scheme: default
|
||||
primary: blue
|
||||
accent: blue
|
||||
toggle:
|
||||
icon: material/brightness-7
|
||||
name: Switch to dark mode
|
||||
- scheme: slate
|
||||
primary: blue
|
||||
accent: blue
|
||||
toggle:
|
||||
icon: material/brightness-4
|
||||
name: Switch to light mode
|
||||
features:
|
||||
- navigation.tabs
|
||||
- navigation.sections
|
||||
- navigation.expand
|
||||
- navigation.top
|
||||
- search.highlight
|
||||
- search.share
|
||||
- content.code.annotate
|
||||
|
||||
plugins:
|
||||
- search
|
||||
- git-revision-date-localized:
|
||||
enable_creation_date: true
|
||||
|
||||
markdown_extensions:
|
||||
- admonition
|
||||
- pymdownx.details
|
||||
- pymdownx.superfences
|
||||
- pymdownx.highlight:
|
||||
anchor_linenums: true
|
||||
- pymdownx.inlinehilite
|
||||
- pymdownx.snippets
|
||||
- pymdownx.tabbed:
|
||||
alternate_style: true
|
||||
- pymdownx.tasklist:
|
||||
custom_checkbox: true
|
||||
- attr_list
|
||||
- md_in_html
|
||||
- toc:
|
||||
permalink: true
|
||||
|
||||
nav:
|
||||
- Home: 'README.md'
|
||||
- Quick Start: 'documentation/guides/QUICKSTART.md'
|
||||
- Installation: 'documentation/examples/INSTALLATION.md'
|
||||
- Architecture: 'documentation/guides/ARCHITECTURE.md'
|
||||
- Cluster Setup: 'documentation/guides/CLUSTER_SETUP.md'
|
||||
- Configuration:
|
||||
- 'Example Configs': 'documentation/examples/configurations.md'
|
||||
- 'Telegram Commands': 'documentation/examples/telegram-commands.md'
|
||||
- 'Cluster Management': 'documentation/examples/cluster-management.md'
|
||||
- Development: 'DEVELOPMENT_SUMMARY.md'
|
||||
|
||||
extra:
|
||||
social:
|
||||
- icon: fontawesome/brands/github
|
||||
link: https://github.com/SmartSolTech/PyGuardian
|
||||
- icon: fontawesome/brands/telegram
|
||||
link: https://t.me/PyGuardianSupport
|
||||
|
||||
copyright: Copyright © 2024 SmartSolTech
|
||||
@@ -1,25 +0,0 @@
|
||||
[tool:pytest]
|
||||
testpaths = tests
|
||||
python_files = test_*.py
|
||||
python_classes = Test*
|
||||
python_functions = test_*
|
||||
addopts =
|
||||
-v
|
||||
--tb=short
|
||||
--strict-markers
|
||||
--disable-warnings
|
||||
--color=yes
|
||||
|
||||
markers =
|
||||
unit: Unit tests
|
||||
integration: Integration tests
|
||||
e2e: End-to-end tests
|
||||
slow: Slow tests
|
||||
auth: Authentication tests
|
||||
api: API tests
|
||||
cluster: Cluster management tests
|
||||
security: Security tests
|
||||
|
||||
filterwarnings =
|
||||
ignore::DeprecationWarning
|
||||
ignore::PendingDeprecationWarning
|
||||
@@ -1,45 +0,0 @@
|
||||
# PyGuardian Requirements
|
||||
# ========================
|
||||
|
||||
# Асинхронная работа с SQLite
|
||||
aiosqlite>=0.19.0
|
||||
|
||||
# Асинхронная работа с файлами
|
||||
aiofiles>=23.2.0
|
||||
|
||||
# Telegram Bot API
|
||||
python-telegram-bot>=20.7
|
||||
|
||||
# YAML конфигурация
|
||||
PyYAML>=6.0.1
|
||||
|
||||
# Работа с IP адресами (встроенный в Python 3.3+)
|
||||
# ipaddress - встроенный модуль
|
||||
|
||||
# Для работы с регулярными выражениями (встроенный)
|
||||
# re - встроенный модуль
|
||||
|
||||
# Для работы с датами (встроенный)
|
||||
# datetime - встроенный модуль
|
||||
|
||||
# Для работы с системными вызовами (встроенный)
|
||||
# subprocess - встроенный модуль
|
||||
|
||||
# Для асинхронности (встроенный в Python 3.7+)
|
||||
# asyncio - встроенный модуль
|
||||
|
||||
# Для логирования (встроенный)
|
||||
# logging - встроенный модуль
|
||||
|
||||
# Для работы с путями (встроенный в Python 3.4+)
|
||||
# pathlib - встроенный модуль
|
||||
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
@@ -1,45 +0,0 @@
|
||||
# PyGuardian Requirements
|
||||
# ========================
|
||||
|
||||
# Асинхронная работа с SQLite
|
||||
aiosqlite>=0.19.0
|
||||
|
||||
# Асинхронная работа с файлами
|
||||
aiofiles>=23.2.0
|
||||
|
||||
# Telegram Bot API
|
||||
python-telegram-bot>=20.7
|
||||
|
||||
# YAML конфигурация
|
||||
PyYAML>=6.0.1
|
||||
|
||||
# Работа с IP адресами (встроенный в Python 3.3+)
|
||||
# ipaddress - встроенный модуль
|
||||
|
||||
# Для работы с регулярными выражениями (встроенный)
|
||||
# re - встроенный модуль
|
||||
|
||||
# Для работы с датами (встроенный)
|
||||
# datetime - встроенный модуль
|
||||
|
||||
# Для работы с системными вызовами (встроенный)
|
||||
# subprocess - встроенный модуль
|
||||
|
||||
# Для асинхронности (встроенный в Python 3.7+)
|
||||
# asyncio - встроенный модуль
|
||||
|
||||
# Для логирования (встроенный)
|
||||
# logging - встроенный модуль
|
||||
|
||||
# Для работы с путями (встроенный в Python 3.4+)
|
||||
# pathlib - встроенный модуль
|
||||
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
@@ -1,54 +0,0 @@
|
||||
# PyGuardian Requirements
|
||||
# ========================
|
||||
|
||||
# Асинхронная работа с SQLite
|
||||
aiosqlite>=0.19.0
|
||||
|
||||
# Асинхронная работа с файлами
|
||||
aiofiles>=23.2.0
|
||||
|
||||
# Telegram Bot API
|
||||
python-telegram-bot>=20.7
|
||||
|
||||
# YAML конфигурация
|
||||
PyYAML>=6.0.1
|
||||
|
||||
# SSH соединения для управления кластером
|
||||
paramiko>=3.3.1
|
||||
|
||||
# Шифрование для паролей и данных кластера
|
||||
cryptography>=41.0.0
|
||||
|
||||
# Системная информация и управление процессами
|
||||
psutil>=5.9.0
|
||||
|
||||
# Работа с IP адресами (встроенный в Python 3.3+)
|
||||
# ipaddress - встроенный модуль
|
||||
|
||||
# Для работы с регулярными выражениями (встроенный)
|
||||
# re - встроенный модуль
|
||||
|
||||
# Для работы с датами (встроенный)
|
||||
# datetime - встроенный модуль
|
||||
|
||||
# Для работы с системными вызовами (встроенный)
|
||||
# subprocess - встроенный модуль
|
||||
|
||||
# Для асинхронности (встроенный в Python 3.7+)
|
||||
# asyncio - встроенный модуль
|
||||
|
||||
# Для логирования (встроенный)
|
||||
# logging - встроенный модуль
|
||||
|
||||
# Для работы с путями (встроенный в Python 3.4+)
|
||||
# pathlib - встроенный модуль
|
||||
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
@@ -1,54 +0,0 @@
|
||||
# PyGuardian Requirements
|
||||
# ========================
|
||||
|
||||
# Асинхронная работа с SQLite
|
||||
aiosqlite>=0.19.0
|
||||
|
||||
# Асинхронная работа с файлами
|
||||
aiofiles>=23.2.0
|
||||
|
||||
# Telegram Bot API
|
||||
python-telegram-bot>=20.7
|
||||
|
||||
# YAML конфигурация
|
||||
PyYAML>=6.0.1
|
||||
|
||||
# SSH соединения для управления кластером
|
||||
paramiko>=3.3.1
|
||||
|
||||
# Шифрование для паролей и данных кластера
|
||||
cryptography>=41.0.0
|
||||
|
||||
# Системная информация и управление процессами
|
||||
psutil>=5.9.0
|
||||
|
||||
# Работа с IP адресами (встроенный в Python 3.3+)
|
||||
# ipaddress - встроенный модуль
|
||||
|
||||
# Для работы с регулярными выражениями (встроенный)
|
||||
# re - встроенный модуль
|
||||
|
||||
# Для работы с датами (встроенный)
|
||||
# datetime - встроенный модуль
|
||||
|
||||
# Для работы с системными вызовами (встроенный)
|
||||
# subprocess - встроенный модуль
|
||||
|
||||
# Для асинхронности (встроенный в Python 3.7+)
|
||||
# asyncio - встроенный модуль
|
||||
|
||||
# Для логирования (встроенный)
|
||||
# logging - встроенный модуль
|
||||
|
||||
# Для работы с путями (встроенный в Python 3.4+)
|
||||
# pathlib - встроенный модуль
|
||||
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
@@ -1,57 +0,0 @@
|
||||
# PyGuardian Requirements
|
||||
# ========================
|
||||
|
||||
# Асинхронная работа с SQLite
|
||||
aiosqlite>=0.19.0
|
||||
|
||||
# Асинхронная работа с файлами
|
||||
aiofiles>=23.2.0
|
||||
|
||||
# Telegram Bot API
|
||||
python-telegram-bot>=20.7
|
||||
|
||||
# YAML конфигурация
|
||||
PyYAML>=6.0.1
|
||||
|
||||
# SSH соединения для управления кластером
|
||||
paramiko>=3.3.1
|
||||
|
||||
# Шифрование для паролей и данных кластера
|
||||
cryptography>=41.0.0
|
||||
|
||||
# JWT токены для аутентификации агентов
|
||||
PyJWT>=2.8.0
|
||||
|
||||
# Системная информация и управление процессами
|
||||
psutil>=5.9.0
|
||||
|
||||
# Работа с IP адресами (встроенный в Python 3.3+)
|
||||
# ipaddress - встроенный модуль
|
||||
|
||||
# Для работы с регулярными выражениями (встроенный)
|
||||
# re - встроенный модуль
|
||||
|
||||
# Для работы с датами (встроенный)
|
||||
# datetime - встроенный модуль
|
||||
|
||||
# Для работы с системными вызовами (встроенный)
|
||||
# subprocess - встроенный модуль
|
||||
|
||||
# Для асинхронности (встроенный в Python 3.7+)
|
||||
# asyncio - встроенный модуль
|
||||
|
||||
# Для логирования (встроенный)
|
||||
# logging - встроенный модуль
|
||||
|
||||
# Для работы с путями (встроенный в Python 3.4+)
|
||||
# pathlib - встроенный модуль
|
||||
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
@@ -53,9 +53,14 @@ psutil>=5.9.0
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
# Дополнительные зависимости для разработки и тестирования
|
||||
pytest>=7.4.0
|
||||
pytest-asyncio>=0.21.0
|
||||
pytest-cov>=4.1.0
|
||||
black>=23.9.0
|
||||
flake8>=6.0.0
|
||||
isort>=5.12.0
|
||||
bandit>=1.7.5
|
||||
safety>=2.3.0
|
||||
mkdocs>=1.5.0
|
||||
mkdocs-material>=9.4.0
|
||||
@@ -53,9 +53,14 @@ psutil>=5.9.0
|
||||
# Для сигналов (встроенный)
|
||||
# signal - встроенный модуль
|
||||
|
||||
# Дополнительные зависимости для разработки и тестирования (опционально)
|
||||
# pytest>=7.4.0
|
||||
# pytest-asyncio>=0.21.0
|
||||
# black>=23.9.0
|
||||
# flake8>=6.0.0
|
||||
# mypy>=1.5.0
|
||||
# Дополнительные зависимости для разработки и тестирования
|
||||
pytest>=7.4.0
|
||||
pytest-asyncio>=0.21.0
|
||||
pytest-cov>=4.1.0
|
||||
black>=23.9.0
|
||||
flake8>=6.0.0
|
||||
isort>=5.12.0
|
||||
bandit>=1.7.5
|
||||
safety>=2.3.0
|
||||
mkdocs>=1.5.0
|
||||
mkdocs-material>=9.4.0
|
||||
@@ -1,104 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Final Deployment Report
|
||||
# Финальный отчет о завершенной реализации
|
||||
#==========================================================================
|
||||
|
||||
echo "🎉 PyGuardian System - Deployment Complete! 🎉"
|
||||
echo "================================================"
|
||||
echo ""
|
||||
|
||||
# Статистика проекта
|
||||
echo "📊 PROJECT STATISTICS:"
|
||||
echo "• Total lines of code and docs: 10,169+"
|
||||
echo "• Python source files: 8 modules (4,985 lines)"
|
||||
echo "• Installation scripts: 3 scripts (1,780 lines)"
|
||||
echo "• Documentation files: 8 documents (2,404 lines)"
|
||||
echo ""
|
||||
|
||||
# Ключевые компоненты
|
||||
echo "🔧 KEY COMPONENTS IMPLEMENTED:"
|
||||
echo "• ✅ ClusterManager - Centralized cluster management (621 lines)"
|
||||
echo "• ✅ Telegram Bot - Advanced interactive commands (1,344 lines)"
|
||||
echo "• ✅ Universal Installer - Multi-mode deployment (735 lines)"
|
||||
echo "• ✅ Docker Support - Containerized deployment (690 lines)"
|
||||
echo "• ✅ Security System - Advanced threat detection (515 lines)"
|
||||
echo "• ✅ Storage Management - Database operations (607 lines)"
|
||||
echo "• ✅ Comprehensive Documentation - Complete user guides"
|
||||
echo ""
|
||||
|
||||
# Возможности
|
||||
echo "🚀 CAPABILITIES DELIVERED:"
|
||||
echo "• 🌐 Centralized cluster management via Telegram bot"
|
||||
echo "• 🚀 Automatic agent deployment over SSH"
|
||||
echo "• 🔧 Three deployment modes: standalone/controller/agent"
|
||||
echo "• 🐳 Full Docker containerization support"
|
||||
echo "• 📱 Interactive Telegram interface with 50+ commands"
|
||||
echo "• 🛡️ Advanced security monitoring and protection"
|
||||
echo "• 📊 Real-time monitoring and alerting"
|
||||
echo "• 🔒 Enterprise-grade security features"
|
||||
echo ""
|
||||
|
||||
# Архитектура
|
||||
echo "🏗️ SYSTEM ARCHITECTURE:"
|
||||
echo "• Asyncio-based high-performance Python backend"
|
||||
echo "• RESTful API for controller-agent communication"
|
||||
echo "• SQLite/PostgreSQL database support"
|
||||
echo "• systemd service integration"
|
||||
echo "• Docker containerization with privilege management"
|
||||
echo "• Event-driven notification system"
|
||||
echo ""
|
||||
|
||||
# Развертывание
|
||||
echo "📦 DEPLOYMENT OPTIONS:"
|
||||
echo "• Standalone: ./install.sh"
|
||||
echo "• Controller: ./install.sh --mode controller"
|
||||
echo "• Agent: ./install.sh --mode agent --controller <IP>"
|
||||
echo "• Docker: ./scripts/docker-install.sh"
|
||||
echo "• Makefile: make install|controller|agent"
|
||||
echo ""
|
||||
|
||||
# Тестирование
|
||||
echo "🧪 TESTING & VALIDATION:"
|
||||
echo "• Installation test suite: ./scripts/test-install.sh"
|
||||
echo "• Syntax validation for all scripts"
|
||||
echo "• Configuration validation"
|
||||
echo "• Dependency checking"
|
||||
echo "• Service health monitoring"
|
||||
echo ""
|
||||
|
||||
echo "================================================"
|
||||
echo "🎯 MISSION ACCOMPLISHED!"
|
||||
echo ""
|
||||
echo "The user requested:"
|
||||
echo "'🟣 10. Возможность централизованного развертывания агентов'"
|
||||
echo ""
|
||||
echo "✅ DELIVERED:"
|
||||
echo "• Complete cluster management system"
|
||||
echo "• Centralized Telegram bot control"
|
||||
echo "• Automatic agent deployment capabilities"
|
||||
echo "• Universal installation system"
|
||||
echo "• Comprehensive documentation"
|
||||
echo ""
|
||||
echo "🛡️ PyGuardian is now a production-ready"
|
||||
echo " enterprise security management platform!"
|
||||
echo ""
|
||||
echo "⚡ Quick Start:"
|
||||
echo " sudo ./install.sh"
|
||||
echo " # Configure Telegram bot"
|
||||
echo " # Start securing your infrastructure!"
|
||||
echo ""
|
||||
echo "📖 Documentation:"
|
||||
echo " • QUICKSTART.md - Fast deployment guide"
|
||||
echo " • docs/INSTALLATION.md - Detailed setup"
|
||||
echo " • docs/CLUSTER_SETUP.md - Cluster configuration"
|
||||
echo ""
|
||||
echo "🆘 Support:"
|
||||
echo " • ./scripts/test-install.sh - System testing"
|
||||
echo " • /debug export - Telegram bot diagnostics"
|
||||
echo " • GitHub Issues for community support"
|
||||
echo ""
|
||||
echo "================================================"
|
||||
echo "🎉 Ready to secure the world! 🌍🛡️"
|
||||
echo "================================================"
|
||||
@@ -1,691 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Docker Installation Script
|
||||
# Supports containerized deployment for Controller and Agent modes
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
# Global variables
|
||||
INSTALL_MODE=""
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
||||
DOCKER_COMPOSE_VERSION="2.20.0"
|
||||
|
||||
# Configuration variables
|
||||
TELEGRAM_BOT_TOKEN=""
|
||||
ADMIN_ID=""
|
||||
CONTROLLER_URL=""
|
||||
AGENT_TOKEN=""
|
||||
CONTROLLER_PORT="8080"
|
||||
|
||||
#==========================================================================
|
||||
# Helper functions
|
||||
#==========================================================================
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "=============================================="
|
||||
echo " PyGuardian Docker $1 Installation"
|
||||
echo "=============================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args() {
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--mode=*)
|
||||
INSTALL_MODE="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--controller-url=*)
|
||||
CONTROLLER_URL="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--agent-token=*)
|
||||
AGENT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--telegram-token=*)
|
||||
TELEGRAM_BOT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--admin-id=*)
|
||||
ADMIN_ID="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--port=*)
|
||||
CONTROLLER_PORT="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "OPTIONS:"
|
||||
echo " --mode=MODE Installation mode: controller, agent"
|
||||
echo " --controller-url=URL Controller URL (for agent mode)"
|
||||
echo " --agent-token=TOKEN Agent authentication token"
|
||||
echo " --telegram-token=TOKEN Telegram bot token"
|
||||
echo " --admin-id=ID Telegram admin ID"
|
||||
echo " --port=PORT Controller port (default: 8080)"
|
||||
echo " -h, --help Show this help"
|
||||
}
|
||||
|
||||
# Select installation mode
|
||||
select_install_mode() {
|
||||
print_info "Выберите режим Docker установки:"
|
||||
echo ""
|
||||
echo "1) Controller - Центральный контроллер кластера в Docker"
|
||||
echo "2) Agent - Агент в Docker для подключения к контроллеру"
|
||||
echo ""
|
||||
|
||||
while true; do
|
||||
read -p "Выберите режим (1-2): " choice
|
||||
case $choice in
|
||||
1)
|
||||
INSTALL_MODE="controller"
|
||||
break
|
||||
;;
|
||||
2)
|
||||
INSTALL_MODE="agent"
|
||||
break
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите 1 или 2."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Check Docker requirements
|
||||
check_docker_requirements() {
|
||||
print_info "Проверка Docker требований..."
|
||||
|
||||
# Check if Docker is installed
|
||||
if ! command -v docker &> /dev/null; then
|
||||
print_info "Docker не установлен. Устанавливаем Docker..."
|
||||
install_docker
|
||||
else
|
||||
print_success "Docker уже установлен: $(docker --version)"
|
||||
fi
|
||||
|
||||
# Check if Docker Compose is installed
|
||||
if ! command -v docker-compose &> /dev/null; then
|
||||
print_info "Docker Compose не установлен. Устанавливаем..."
|
||||
install_docker_compose
|
||||
else
|
||||
print_success "Docker Compose уже установлен: $(docker-compose --version)"
|
||||
fi
|
||||
|
||||
# Start Docker service
|
||||
systemctl start docker
|
||||
systemctl enable docker
|
||||
print_success "Docker service started and enabled"
|
||||
}
|
||||
|
||||
# Install Docker
|
||||
install_docker() {
|
||||
print_info "Установка Docker..."
|
||||
|
||||
# Install prerequisites
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y ca-certificates curl gnupg
|
||||
|
||||
# Add Docker's official GPG key
|
||||
install -m 0755 -d /etc/apt/keyrings
|
||||
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg
|
||||
chmod a+r /etc/apt/keyrings/docker.gpg
|
||||
|
||||
# Add repository
|
||||
echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(. /etc/os-release && echo "$VERSION_CODENAME") stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null
|
||||
|
||||
apt-get update
|
||||
apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
|
||||
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y yum-utils
|
||||
yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
|
||||
yum install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
|
||||
|
||||
else
|
||||
print_error "Unsupported package manager for Docker installation"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
print_success "Docker installed successfully"
|
||||
}
|
||||
|
||||
# Install Docker Compose
|
||||
install_docker_compose() {
|
||||
print_info "Установка Docker Compose..."
|
||||
|
||||
curl -L "https://github.com/docker/compose/releases/download/v${DOCKER_COMPOSE_VERSION}/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose
|
||||
chmod +x /usr/local/bin/docker-compose
|
||||
|
||||
print_success "Docker Compose installed successfully"
|
||||
}
|
||||
|
||||
# Get configuration for controller
|
||||
get_controller_config() {
|
||||
if [[ -z "$TELEGRAM_BOT_TOKEN" ]]; then
|
||||
echo ""
|
||||
print_info "Настройка Telegram бота для контроллера:"
|
||||
echo "1. Создайте бота у @BotFather"
|
||||
echo "2. Получите токен бота"
|
||||
echo "3. Узнайте ваш chat ID у @userinfobot"
|
||||
echo ""
|
||||
read -p "Введите токен Telegram бота: " TELEGRAM_BOT_TOKEN
|
||||
fi
|
||||
|
||||
if [[ -z "$ADMIN_ID" ]]; then
|
||||
read -p "Введите ваш Telegram ID (admin): " ADMIN_ID
|
||||
fi
|
||||
|
||||
read -p "Порт для API контроллера (по умолчанию $CONTROLLER_PORT): " input_port
|
||||
CONTROLLER_PORT=${input_port:-$CONTROLLER_PORT}
|
||||
}
|
||||
|
||||
# Get configuration for agent
|
||||
get_agent_config() {
|
||||
if [[ -z "$CONTROLLER_URL" ]]; then
|
||||
read -p "URL контроллера (например, https://controller.example.com:8080): " CONTROLLER_URL
|
||||
fi
|
||||
|
||||
if [[ -z "$AGENT_TOKEN" ]]; then
|
||||
read -p "Токен агента (получите у администратора контроллера): " AGENT_TOKEN
|
||||
fi
|
||||
|
||||
read -p "Имя агента (по умолчанию: $(hostname)): " AGENT_NAME
|
||||
AGENT_NAME=${AGENT_NAME:-$(hostname)}
|
||||
}
|
||||
|
||||
# Create Dockerfile for controller
|
||||
create_controller_dockerfile() {
|
||||
print_info "Создание Dockerfile для контроллера..."
|
||||
|
||||
mkdir -p controller
|
||||
|
||||
cat > controller/Dockerfile <<EOF
|
||||
# PyGuardian Controller Docker Image
|
||||
FROM python:3.11-slim
|
||||
|
||||
LABEL maintainer="SmartSolTech Team"
|
||||
LABEL description="PyGuardian Security Controller"
|
||||
LABEL version="2.0"
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \\
|
||||
iptables \\
|
||||
nftables \\
|
||||
curl \\
|
||||
sqlite3 \\
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create app user
|
||||
RUN useradd --create-home --shell /bin/bash pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /app
|
||||
|
||||
# Copy requirements first for better caching
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy application code
|
||||
COPY src/ ./src/
|
||||
COPY main.py .
|
||||
|
||||
# Create necessary directories
|
||||
RUN mkdir -p /var/lib/pyguardian /var/log/pyguardian /etc/pyguardian
|
||||
RUN chown -R pyguardian:pyguardian /var/lib/pyguardian /var/log/pyguardian /app
|
||||
|
||||
# Copy configuration
|
||||
COPY controller-config.yaml /etc/pyguardian/config.yaml
|
||||
RUN chown pyguardian:pyguardian /etc/pyguardian/config.yaml
|
||||
|
||||
# Switch to app user
|
||||
USER pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \\
|
||||
CMD curl -f http://localhost:8080/health || exit 1
|
||||
|
||||
# Expose port
|
||||
EXPOSE 8080
|
||||
|
||||
# Start application
|
||||
CMD ["python", "main.py", "--config", "/etc/pyguardian/config.yaml"]
|
||||
EOF
|
||||
|
||||
print_success "Controller Dockerfile created"
|
||||
}
|
||||
|
||||
# Create Dockerfile for agent
|
||||
create_agent_dockerfile() {
|
||||
print_info "Создание Dockerfile для агента..."
|
||||
|
||||
mkdir -p agent
|
||||
|
||||
cat > agent/Dockerfile <<EOF
|
||||
# PyGuardian Agent Docker Image
|
||||
FROM python:3.11-slim
|
||||
|
||||
LABEL maintainer="SmartSolTech Team"
|
||||
LABEL description="PyGuardian Security Agent"
|
||||
LABEL version="2.0"
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \\
|
||||
iptables \\
|
||||
nftables \\
|
||||
curl \\
|
||||
sqlite3 \\
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create app user
|
||||
RUN useradd --create-home --shell /bin/bash pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /app
|
||||
|
||||
# Copy requirements first for better caching
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy application code
|
||||
COPY src/ ./src/
|
||||
COPY main.py .
|
||||
|
||||
# Create necessary directories
|
||||
RUN mkdir -p /var/lib/pyguardian /var/log/pyguardian /etc/pyguardian
|
||||
RUN chown -R pyguardian:pyguardian /var/lib/pyguardian /var/log/pyguardian /app
|
||||
|
||||
# Copy configuration
|
||||
COPY agent-config.yaml /etc/pyguardian/config.yaml
|
||||
RUN chown pyguardian:pyguardian /etc/pyguardian/config.yaml
|
||||
|
||||
# Switch to app user
|
||||
USER pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \\
|
||||
CMD python -c "import sys; sys.exit(0)"
|
||||
|
||||
# Start application
|
||||
CMD ["python", "main.py", "--config", "/etc/pyguardian/config.yaml"]
|
||||
EOF
|
||||
|
||||
print_success "Agent Dockerfile created"
|
||||
}
|
||||
|
||||
# Create controller configuration
|
||||
create_controller_config_file() {
|
||||
cat > controller-config.yaml <<EOF
|
||||
# PyGuardian Controller Docker Configuration
|
||||
mode: "controller"
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Controller Settings
|
||||
controller:
|
||||
port: 8080
|
||||
host: "0.0.0.0"
|
||||
max_agents: 50
|
||||
agent_timeout: 300
|
||||
heartbeat_interval: 60
|
||||
|
||||
# Cluster Configuration
|
||||
cluster:
|
||||
controller_mode: true
|
||||
auto_deployment: true
|
||||
agent_auto_update: true
|
||||
|
||||
# Security Settings
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
stealth_mode_duration: 300
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/controller.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "/var/log/pyguardian/controller.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
}
|
||||
|
||||
# Create agent configuration
|
||||
create_agent_config_file() {
|
||||
cat > agent-config.yaml <<EOF
|
||||
# PyGuardian Agent Docker Configuration
|
||||
mode: "agent"
|
||||
|
||||
# Agent Settings
|
||||
agent:
|
||||
name: "$AGENT_NAME"
|
||||
controller_url: "$CONTROLLER_URL"
|
||||
token: "$AGENT_TOKEN"
|
||||
heartbeat_interval: 60
|
||||
reconnect_delay: 30
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/agent.db"
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "/var/log/pyguardian/agent.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
}
|
||||
|
||||
# Create docker-compose.yml for controller
|
||||
create_controller_compose() {
|
||||
print_info "Создание docker-compose.yml для контроллера..."
|
||||
|
||||
cat > docker-compose.yml <<EOF
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: controller/Dockerfile
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
ports:
|
||||
- "${CONTROLLER_PORT}:8080"
|
||||
volumes:
|
||||
- controller-data:/var/lib/pyguardian
|
||||
- controller-logs:/var/log/pyguardian
|
||||
- ./controller-config.yaml:/etc/pyguardian/config.yaml:ro
|
||||
environment:
|
||||
- TELEGRAM_BOT_TOKEN=${TELEGRAM_BOT_TOKEN}
|
||||
- ADMIN_ID=${ADMIN_ID}
|
||||
networks:
|
||||
- pyguardian-net
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8080/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
controller-data:
|
||||
driver: local
|
||||
controller-logs:
|
||||
driver: local
|
||||
|
||||
networks:
|
||||
pyguardian-net:
|
||||
driver: bridge
|
||||
EOF
|
||||
|
||||
print_success "Controller docker-compose.yml created"
|
||||
}
|
||||
|
||||
# Create docker-compose.yml for agent
|
||||
create_agent_compose() {
|
||||
print_info "Создание docker-compose.yml для агента..."
|
||||
|
||||
cat > docker-compose.yml <<EOF
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-agent:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: agent/Dockerfile
|
||||
container_name: pyguardian-agent
|
||||
restart: unless-stopped
|
||||
network_mode: host
|
||||
privileged: true
|
||||
volumes:
|
||||
- /var/log/auth.log:/var/log/auth.log:ro
|
||||
- agent-data:/var/lib/pyguardian
|
||||
- agent-logs:/var/log/pyguardian
|
||||
- ./agent-config.yaml:/etc/pyguardian/config.yaml:ro
|
||||
environment:
|
||||
- CONTROLLER_URL=${CONTROLLER_URL}
|
||||
- AGENT_TOKEN=${AGENT_TOKEN}
|
||||
- AGENT_NAME=${AGENT_NAME}
|
||||
cap_add:
|
||||
- NET_ADMIN
|
||||
- SYS_ADMIN
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import sys; sys.exit(0)"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
agent-data:
|
||||
driver: local
|
||||
agent-logs:
|
||||
driver: local
|
||||
EOF
|
||||
|
||||
print_success "Agent docker-compose.yml created"
|
||||
}
|
||||
|
||||
# Copy application files
|
||||
copy_app_files() {
|
||||
print_info "Копирование файлов приложения..."
|
||||
|
||||
# Copy source files
|
||||
cp -r "$PROJECT_DIR/src" ./
|
||||
cp "$PROJECT_DIR/main.py" ./
|
||||
cp "$PROJECT_DIR/requirements.txt" ./
|
||||
|
||||
print_success "Application files copied"
|
||||
}
|
||||
|
||||
# Deploy controller
|
||||
deploy_controller() {
|
||||
print_header "Controller"
|
||||
|
||||
get_controller_config
|
||||
create_controller_config_file
|
||||
copy_app_files
|
||||
create_controller_dockerfile
|
||||
create_controller_compose
|
||||
|
||||
print_info "Запуск контроллера..."
|
||||
docker-compose up --build -d
|
||||
|
||||
print_info "Ожидание запуска контроллера..."
|
||||
sleep 10
|
||||
|
||||
if docker-compose ps | grep -q "Up"; then
|
||||
print_success "Controller successfully deployed and running"
|
||||
else
|
||||
print_error "Failed to start controller"
|
||||
docker-compose logs
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Deploy agent
|
||||
deploy_agent() {
|
||||
print_header "Agent"
|
||||
|
||||
get_agent_config
|
||||
create_agent_config_file
|
||||
copy_app_files
|
||||
create_agent_dockerfile
|
||||
create_agent_compose
|
||||
|
||||
print_info "Запуск агента..."
|
||||
docker-compose up --build -d
|
||||
|
||||
print_info "Ожидание запуска агента..."
|
||||
sleep 10
|
||||
|
||||
if docker-compose ps | grep -q "Up"; then
|
||||
print_success "Agent successfully deployed and running"
|
||||
else
|
||||
print_error "Failed to start agent"
|
||||
docker-compose logs
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show completion info
|
||||
show_docker_completion_info() {
|
||||
print_header "Docker Deployment Complete"
|
||||
|
||||
echo -e "${GREEN}✓ PyGuardian успешно развернут в Docker (режим: $INSTALL_MODE)${NC}"
|
||||
echo ""
|
||||
|
||||
print_info "Полезные Docker команды:"
|
||||
echo " docker-compose ps # Статус контейнеров"
|
||||
echo " docker-compose logs -f # Просмотр логов"
|
||||
echo " docker-compose restart # Перезапуск"
|
||||
echo " docker-compose stop # Остановка"
|
||||
echo " docker-compose down # Полная остановка и удаление"
|
||||
echo ""
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"controller")
|
||||
echo -e "${YELLOW}⚠ Контроллер настроен:${NC}"
|
||||
echo " - API доступен на порту: $CONTROLLER_PORT"
|
||||
echo " - Telegram бот активен"
|
||||
echo " - Веб-интерфейс: http://localhost:$CONTROLLER_PORT"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Не забудьте:${NC}"
|
||||
echo " 1. Открыть порт $CONTROLLER_PORT в firewall"
|
||||
echo " 2. Настроить reverse proxy для HTTPS"
|
||||
echo " 3. Добавить агенты через Telegram команды"
|
||||
;;
|
||||
"agent")
|
||||
echo -e "${YELLOW}⚠ Агент настроен:${NC}"
|
||||
echo " - Подключение к: $CONTROLLER_URL"
|
||||
echo " - Имя агента: $AGENT_NAME"
|
||||
echo " - Мониторинг auth.log активен"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Примечание:${NC}"
|
||||
echo " Агент автоматически подключится к контроллеру"
|
||||
echo " Проверьте статус в логах контейнера"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# Main Docker installation flow
|
||||
#==========================================================================
|
||||
|
||||
main() {
|
||||
check_root
|
||||
parse_args "$@"
|
||||
|
||||
if [[ -z "$INSTALL_MODE" ]]; then
|
||||
select_install_mode
|
||||
fi
|
||||
|
||||
print_info "Режим Docker установки: $INSTALL_MODE"
|
||||
|
||||
check_docker_requirements
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"controller")
|
||||
deploy_controller
|
||||
;;
|
||||
"agent")
|
||||
deploy_agent
|
||||
;;
|
||||
*)
|
||||
print_error "Unsupported mode: $INSTALL_MODE"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
show_docker_completion_info
|
||||
print_success "Docker развертывание завершено успешно!"
|
||||
}
|
||||
|
||||
main "$@"
|
||||
@@ -1,691 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Docker Installation Script
|
||||
# Supports containerized deployment for Controller and Agent modes
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
# Global variables
|
||||
INSTALL_MODE=""
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
||||
DOCKER_COMPOSE_VERSION="2.20.0"
|
||||
|
||||
# Configuration variables
|
||||
TELEGRAM_BOT_TOKEN=""
|
||||
ADMIN_ID=""
|
||||
CONTROLLER_URL=""
|
||||
AGENT_TOKEN=""
|
||||
CONTROLLER_PORT="8080"
|
||||
|
||||
#==========================================================================
|
||||
# Helper functions
|
||||
#==========================================================================
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "=============================================="
|
||||
echo " PyGuardian Docker $1 Installation"
|
||||
echo "=============================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args() {
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--mode=*)
|
||||
INSTALL_MODE="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--controller-url=*)
|
||||
CONTROLLER_URL="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--agent-token=*)
|
||||
AGENT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--telegram-token=*)
|
||||
TELEGRAM_BOT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--admin-id=*)
|
||||
ADMIN_ID="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--port=*)
|
||||
CONTROLLER_PORT="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "OPTIONS:"
|
||||
echo " --mode=MODE Installation mode: controller, agent"
|
||||
echo " --controller-url=URL Controller URL (for agent mode)"
|
||||
echo " --agent-token=TOKEN Agent authentication token"
|
||||
echo " --telegram-token=TOKEN Telegram bot token"
|
||||
echo " --admin-id=ID Telegram admin ID"
|
||||
echo " --port=PORT Controller port (default: 8080)"
|
||||
echo " -h, --help Show this help"
|
||||
}
|
||||
|
||||
# Select installation mode
|
||||
select_install_mode() {
|
||||
print_info "Выберите режим Docker установки:"
|
||||
echo ""
|
||||
echo "1) Controller - Центральный контроллер кластера в Docker"
|
||||
echo "2) Agent - Агент в Docker для подключения к контроллеру"
|
||||
echo ""
|
||||
|
||||
while true; do
|
||||
read -p "Выберите режим (1-2): " choice
|
||||
case $choice in
|
||||
1)
|
||||
INSTALL_MODE="controller"
|
||||
break
|
||||
;;
|
||||
2)
|
||||
INSTALL_MODE="agent"
|
||||
break
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите 1 или 2."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Check Docker requirements
|
||||
check_docker_requirements() {
|
||||
print_info "Проверка Docker требований..."
|
||||
|
||||
# Check if Docker is installed
|
||||
if ! command -v docker &> /dev/null; then
|
||||
print_info "Docker не установлен. Устанавливаем Docker..."
|
||||
install_docker
|
||||
else
|
||||
print_success "Docker уже установлен: $(docker --version)"
|
||||
fi
|
||||
|
||||
# Check if Docker Compose is installed
|
||||
if ! command -v docker-compose &> /dev/null; then
|
||||
print_info "Docker Compose не установлен. Устанавливаем..."
|
||||
install_docker_compose
|
||||
else
|
||||
print_success "Docker Compose уже установлен: $(docker-compose --version)"
|
||||
fi
|
||||
|
||||
# Start Docker service
|
||||
systemctl start docker
|
||||
systemctl enable docker
|
||||
print_success "Docker service started and enabled"
|
||||
}
|
||||
|
||||
# Install Docker
|
||||
install_docker() {
|
||||
print_info "Установка Docker..."
|
||||
|
||||
# Install prerequisites
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y ca-certificates curl gnupg
|
||||
|
||||
# Add Docker's official GPG key
|
||||
install -m 0755 -d /etc/apt/keyrings
|
||||
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg
|
||||
chmod a+r /etc/apt/keyrings/docker.gpg
|
||||
|
||||
# Add repository
|
||||
echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(. /etc/os-release && echo "$VERSION_CODENAME") stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null
|
||||
|
||||
apt-get update
|
||||
apt-get install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
|
||||
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y yum-utils
|
||||
yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
|
||||
yum install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin
|
||||
|
||||
else
|
||||
print_error "Unsupported package manager for Docker installation"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
print_success "Docker installed successfully"
|
||||
}
|
||||
|
||||
# Install Docker Compose
|
||||
install_docker_compose() {
|
||||
print_info "Установка Docker Compose..."
|
||||
|
||||
curl -L "https://github.com/docker/compose/releases/download/v${DOCKER_COMPOSE_VERSION}/docker-compose-$(uname -s)-$(uname -m)" -o /usr/local/bin/docker-compose
|
||||
chmod +x /usr/local/bin/docker-compose
|
||||
|
||||
print_success "Docker Compose installed successfully"
|
||||
}
|
||||
|
||||
# Get configuration for controller
|
||||
get_controller_config() {
|
||||
if [[ -z "$TELEGRAM_BOT_TOKEN" ]]; then
|
||||
echo ""
|
||||
print_info "Настройка Telegram бота для контроллера:"
|
||||
echo "1. Создайте бота у @BotFather"
|
||||
echo "2. Получите токен бота"
|
||||
echo "3. Узнайте ваш chat ID у @userinfobot"
|
||||
echo ""
|
||||
read -p "Введите токен Telegram бота: " TELEGRAM_BOT_TOKEN
|
||||
fi
|
||||
|
||||
if [[ -z "$ADMIN_ID" ]]; then
|
||||
read -p "Введите ваш Telegram ID (admin): " ADMIN_ID
|
||||
fi
|
||||
|
||||
read -p "Порт для API контроллера (по умолчанию $CONTROLLER_PORT): " input_port
|
||||
CONTROLLER_PORT=${input_port:-$CONTROLLER_PORT}
|
||||
}
|
||||
|
||||
# Get configuration for agent
|
||||
get_agent_config() {
|
||||
if [[ -z "$CONTROLLER_URL" ]]; then
|
||||
read -p "URL контроллера (например, https://controller.example.com:8080): " CONTROLLER_URL
|
||||
fi
|
||||
|
||||
if [[ -z "$AGENT_TOKEN" ]]; then
|
||||
read -p "Токен агента (получите у администратора контроллера): " AGENT_TOKEN
|
||||
fi
|
||||
|
||||
read -p "Имя агента (по умолчанию: $(hostname)): " AGENT_NAME
|
||||
AGENT_NAME=${AGENT_NAME:-$(hostname)}
|
||||
}
|
||||
|
||||
# Create Dockerfile for controller
|
||||
create_controller_dockerfile() {
|
||||
print_info "Создание Dockerfile для контроллера..."
|
||||
|
||||
mkdir -p controller
|
||||
|
||||
cat > controller/Dockerfile <<EOF
|
||||
# PyGuardian Controller Docker Image
|
||||
FROM python:3.11-slim
|
||||
|
||||
LABEL maintainer="SmartSolTech Team"
|
||||
LABEL description="PyGuardian Security Controller"
|
||||
LABEL version="2.0"
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \\
|
||||
iptables \\
|
||||
nftables \\
|
||||
curl \\
|
||||
sqlite3 \\
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create app user
|
||||
RUN useradd --create-home --shell /bin/bash pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /app
|
||||
|
||||
# Copy requirements first for better caching
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy application code
|
||||
COPY src/ ./src/
|
||||
COPY main.py .
|
||||
|
||||
# Create necessary directories
|
||||
RUN mkdir -p /var/lib/pyguardian /var/log/pyguardian /etc/pyguardian
|
||||
RUN chown -R pyguardian:pyguardian /var/lib/pyguardian /var/log/pyguardian /app
|
||||
|
||||
# Copy configuration
|
||||
COPY controller-config.yaml /etc/pyguardian/config.yaml
|
||||
RUN chown pyguardian:pyguardian /etc/pyguardian/config.yaml
|
||||
|
||||
# Switch to app user
|
||||
USER pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \\
|
||||
CMD curl -f http://localhost:8080/health || exit 1
|
||||
|
||||
# Expose port
|
||||
EXPOSE 8080
|
||||
|
||||
# Start application
|
||||
CMD ["python", "main.py", "--config", "/etc/pyguardian/config.yaml"]
|
||||
EOF
|
||||
|
||||
print_success "Controller Dockerfile created"
|
||||
}
|
||||
|
||||
# Create Dockerfile for agent
|
||||
create_agent_dockerfile() {
|
||||
print_info "Создание Dockerfile для агента..."
|
||||
|
||||
mkdir -p agent
|
||||
|
||||
cat > agent/Dockerfile <<EOF
|
||||
# PyGuardian Agent Docker Image
|
||||
FROM python:3.11-slim
|
||||
|
||||
LABEL maintainer="SmartSolTech Team"
|
||||
LABEL description="PyGuardian Security Agent"
|
||||
LABEL version="2.0"
|
||||
|
||||
# Install system dependencies
|
||||
RUN apt-get update && apt-get install -y \\
|
||||
iptables \\
|
||||
nftables \\
|
||||
curl \\
|
||||
sqlite3 \\
|
||||
&& rm -rf /var/lib/apt/lists/*
|
||||
|
||||
# Create app user
|
||||
RUN useradd --create-home --shell /bin/bash pyguardian
|
||||
|
||||
# Set working directory
|
||||
WORKDIR /app
|
||||
|
||||
# Copy requirements first for better caching
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# Copy application code
|
||||
COPY src/ ./src/
|
||||
COPY main.py .
|
||||
|
||||
# Create necessary directories
|
||||
RUN mkdir -p /var/lib/pyguardian /var/log/pyguardian /etc/pyguardian
|
||||
RUN chown -R pyguardian:pyguardian /var/lib/pyguardian /var/log/pyguardian /app
|
||||
|
||||
# Copy configuration
|
||||
COPY agent-config.yaml /etc/pyguardian/config.yaml
|
||||
RUN chown pyguardian:pyguardian /etc/pyguardian/config.yaml
|
||||
|
||||
# Switch to app user
|
||||
USER pyguardian
|
||||
|
||||
# Health check
|
||||
HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \\
|
||||
CMD python -c "import sys; sys.exit(0)"
|
||||
|
||||
# Start application
|
||||
CMD ["python", "main.py", "--config", "/etc/pyguardian/config.yaml"]
|
||||
EOF
|
||||
|
||||
print_success "Agent Dockerfile created"
|
||||
}
|
||||
|
||||
# Create controller configuration
|
||||
create_controller_config_file() {
|
||||
cat > controller-config.yaml <<EOF
|
||||
# PyGuardian Controller Docker Configuration
|
||||
mode: "controller"
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Controller Settings
|
||||
controller:
|
||||
port: 8080
|
||||
host: "0.0.0.0"
|
||||
max_agents: 50
|
||||
agent_timeout: 300
|
||||
heartbeat_interval: 60
|
||||
|
||||
# Cluster Configuration
|
||||
cluster:
|
||||
controller_mode: true
|
||||
auto_deployment: true
|
||||
agent_auto_update: true
|
||||
|
||||
# Security Settings
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
stealth_mode_duration: 300
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/controller.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "/var/log/pyguardian/controller.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
}
|
||||
|
||||
# Create agent configuration
|
||||
create_agent_config_file() {
|
||||
cat > agent-config.yaml <<EOF
|
||||
# PyGuardian Agent Docker Configuration
|
||||
mode: "agent"
|
||||
|
||||
# Agent Settings
|
||||
agent:
|
||||
name: "$AGENT_NAME"
|
||||
controller_url: "$CONTROLLER_URL"
|
||||
token: "$AGENT_TOKEN"
|
||||
heartbeat_interval: 60
|
||||
reconnect_delay: 30
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "/var/lib/pyguardian/agent.db"
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "/var/log/pyguardian/agent.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
}
|
||||
|
||||
# Create docker-compose.yml for controller
|
||||
create_controller_compose() {
|
||||
print_info "Создание docker-compose.yml для контроллера..."
|
||||
|
||||
cat > docker-compose.yml <<EOF
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-controller:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: controller/Dockerfile
|
||||
container_name: pyguardian-controller
|
||||
restart: unless-stopped
|
||||
ports:
|
||||
- "${CONTROLLER_PORT}:8080"
|
||||
volumes:
|
||||
- controller-data:/var/lib/pyguardian
|
||||
- controller-logs:/var/log/pyguardian
|
||||
- ./controller-config.yaml:/etc/pyguardian/config.yaml:ro
|
||||
environment:
|
||||
- TELEGRAM_BOT_TOKEN=${TELEGRAM_BOT_TOKEN}
|
||||
- ADMIN_ID=${ADMIN_ID}
|
||||
networks:
|
||||
- pyguardian-net
|
||||
healthcheck:
|
||||
test: ["CMD", "curl", "-f", "http://localhost:8080/health"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
controller-data:
|
||||
driver: local
|
||||
controller-logs:
|
||||
driver: local
|
||||
|
||||
networks:
|
||||
pyguardian-net:
|
||||
driver: bridge
|
||||
EOF
|
||||
|
||||
print_success "Controller docker-compose.yml created"
|
||||
}
|
||||
|
||||
# Create docker-compose.yml for agent
|
||||
create_agent_compose() {
|
||||
print_info "Создание docker-compose.yml для агента..."
|
||||
|
||||
cat > docker-compose.yml <<EOF
|
||||
version: '3.8'
|
||||
|
||||
services:
|
||||
pyguardian-agent:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: agent/Dockerfile
|
||||
container_name: pyguardian-agent
|
||||
restart: unless-stopped
|
||||
network_mode: host
|
||||
privileged: true
|
||||
volumes:
|
||||
- /var/log/auth.log:/var/log/auth.log:ro
|
||||
- agent-data:/var/lib/pyguardian
|
||||
- agent-logs:/var/log/pyguardian
|
||||
- ./agent-config.yaml:/etc/pyguardian/config.yaml:ro
|
||||
environment:
|
||||
- CONTROLLER_URL=${CONTROLLER_URL}
|
||||
- AGENT_TOKEN=${AGENT_TOKEN}
|
||||
- AGENT_NAME=${AGENT_NAME}
|
||||
cap_add:
|
||||
- NET_ADMIN
|
||||
- SYS_ADMIN
|
||||
healthcheck:
|
||||
test: ["CMD", "python", "-c", "import sys; sys.exit(0)"]
|
||||
interval: 30s
|
||||
timeout: 10s
|
||||
retries: 3
|
||||
start_period: 30s
|
||||
|
||||
volumes:
|
||||
agent-data:
|
||||
driver: local
|
||||
agent-logs:
|
||||
driver: local
|
||||
EOF
|
||||
|
||||
print_success "Agent docker-compose.yml created"
|
||||
}
|
||||
|
||||
# Copy application files
|
||||
copy_app_files() {
|
||||
print_info "Копирование файлов приложения..."
|
||||
|
||||
# Copy source files
|
||||
cp -r "$PROJECT_DIR/src" ./
|
||||
cp "$PROJECT_DIR/main.py" ./
|
||||
cp "$PROJECT_DIR/requirements.txt" ./
|
||||
|
||||
print_success "Application files copied"
|
||||
}
|
||||
|
||||
# Deploy controller
|
||||
deploy_controller() {
|
||||
print_header "Controller"
|
||||
|
||||
get_controller_config
|
||||
create_controller_config_file
|
||||
copy_app_files
|
||||
create_controller_dockerfile
|
||||
create_controller_compose
|
||||
|
||||
print_info "Запуск контроллера..."
|
||||
docker-compose up --build -d
|
||||
|
||||
print_info "Ожидание запуска контроллера..."
|
||||
sleep 10
|
||||
|
||||
if docker-compose ps | grep -q "Up"; then
|
||||
print_success "Controller successfully deployed and running"
|
||||
else
|
||||
print_error "Failed to start controller"
|
||||
docker-compose logs
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Deploy agent
|
||||
deploy_agent() {
|
||||
print_header "Agent"
|
||||
|
||||
get_agent_config
|
||||
create_agent_config_file
|
||||
copy_app_files
|
||||
create_agent_dockerfile
|
||||
create_agent_compose
|
||||
|
||||
print_info "Запуск агента..."
|
||||
docker-compose up --build -d
|
||||
|
||||
print_info "Ожидание запуска агента..."
|
||||
sleep 10
|
||||
|
||||
if docker-compose ps | grep -q "Up"; then
|
||||
print_success "Agent successfully deployed and running"
|
||||
else
|
||||
print_error "Failed to start agent"
|
||||
docker-compose logs
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show completion info
|
||||
show_docker_completion_info() {
|
||||
print_header "Docker Deployment Complete"
|
||||
|
||||
echo -e "${GREEN}✓ PyGuardian успешно развернут в Docker (режим: $INSTALL_MODE)${NC}"
|
||||
echo ""
|
||||
|
||||
print_info "Полезные Docker команды:"
|
||||
echo " docker-compose ps # Статус контейнеров"
|
||||
echo " docker-compose logs -f # Просмотр логов"
|
||||
echo " docker-compose restart # Перезапуск"
|
||||
echo " docker-compose stop # Остановка"
|
||||
echo " docker-compose down # Полная остановка и удаление"
|
||||
echo ""
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"controller")
|
||||
echo -e "${YELLOW}⚠ Контроллер настроен:${NC}"
|
||||
echo " - API доступен на порту: $CONTROLLER_PORT"
|
||||
echo " - Telegram бот активен"
|
||||
echo " - Веб-интерфейс: http://localhost:$CONTROLLER_PORT"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Не забудьте:${NC}"
|
||||
echo " 1. Открыть порт $CONTROLLER_PORT в firewall"
|
||||
echo " 2. Настроить reverse proxy для HTTPS"
|
||||
echo " 3. Добавить агенты через Telegram команды"
|
||||
;;
|
||||
"agent")
|
||||
echo -e "${YELLOW}⚠ Агент настроен:${NC}"
|
||||
echo " - Подключение к: $CONTROLLER_URL"
|
||||
echo " - Имя агента: $AGENT_NAME"
|
||||
echo " - Мониторинг auth.log активен"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Примечание:${NC}"
|
||||
echo " Агент автоматически подключится к контроллеру"
|
||||
echo " Проверьте статус в логах контейнера"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# Main Docker installation flow
|
||||
#==========================================================================
|
||||
|
||||
main() {
|
||||
check_root
|
||||
parse_args "$@"
|
||||
|
||||
if [[ -z "$INSTALL_MODE" ]]; then
|
||||
select_install_mode
|
||||
fi
|
||||
|
||||
print_info "Режим Docker установки: $INSTALL_MODE"
|
||||
|
||||
check_docker_requirements
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"controller")
|
||||
deploy_controller
|
||||
;;
|
||||
"agent")
|
||||
deploy_agent
|
||||
;;
|
||||
*)
|
||||
print_error "Unsupported mode: $INSTALL_MODE"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
show_docker_completion_info
|
||||
print_success "Docker развертывание завершено успешно!"
|
||||
}
|
||||
|
||||
main "$@"
|
||||
@@ -1,736 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Universal Installation Script
|
||||
# Supports: Standalone, Controller, Agent modes
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e # Exit on any error
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Global variables
|
||||
INSTALL_MODE=""
|
||||
NON_INTERACTIVE=false
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
||||
INSTALL_DIR="/opt/pyguardian"
|
||||
SERVICE_USER="pyguardian"
|
||||
CONFIG_DIR="/etc/pyguardian"
|
||||
LOG_DIR="/var/log/pyguardian"
|
||||
DATA_DIR="/var/lib/pyguardian"
|
||||
|
||||
# Configuration variables
|
||||
TELEGRAM_BOT_TOKEN=""
|
||||
ADMIN_ID=""
|
||||
CONTROLLER_URL=""
|
||||
AGENT_TOKEN=""
|
||||
DB_PATH=""
|
||||
|
||||
#==========================================================================
|
||||
# Helper functions
|
||||
#==========================================================================
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "=============================================="
|
||||
echo " PyGuardian $1 Installation"
|
||||
echo "=============================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
print_warning() {
|
||||
echo -e "${YELLOW}⚠ $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args() {
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--mode=*)
|
||||
INSTALL_MODE="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--non-interactive)
|
||||
NON_INTERACTIVE=true
|
||||
shift
|
||||
;;
|
||||
--controller-url=*)
|
||||
CONTROLLER_URL="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--agent-token=*)
|
||||
AGENT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--telegram-token=*)
|
||||
TELEGRAM_BOT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--admin-id=*)
|
||||
ADMIN_ID="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "OPTIONS:"
|
||||
echo " --mode=MODE Installation mode: standalone, controller, agent"
|
||||
echo " --non-interactive Non-interactive installation"
|
||||
echo " --controller-url=URL Controller URL (for agent mode)"
|
||||
echo " --agent-token=TOKEN Agent authentication token"
|
||||
echo " --telegram-token=TOKEN Telegram bot token"
|
||||
echo " --admin-id=ID Telegram admin ID"
|
||||
echo " -h, --help Show this help"
|
||||
}
|
||||
|
||||
# Interactive mode selection
|
||||
select_install_mode() {
|
||||
if [[ "$NON_INTERACTIVE" == "true" ]]; then
|
||||
return
|
||||
fi
|
||||
|
||||
print_info "Выберите режим установки:"
|
||||
echo ""
|
||||
echo "1) Standalone - Автономный сервер (все в одном)"
|
||||
echo "2) Controller - Центральный контроллер кластера"
|
||||
echo "3) Agent - Агент для подключения к контроллеру"
|
||||
echo ""
|
||||
|
||||
while true; do
|
||||
read -p "Выберите режим (1-3): " choice
|
||||
case $choice in
|
||||
1)
|
||||
INSTALL_MODE="standalone"
|
||||
break
|
||||
;;
|
||||
2)
|
||||
INSTALL_MODE="controller"
|
||||
break
|
||||
;;
|
||||
3)
|
||||
INSTALL_MODE="agent"
|
||||
break
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите 1, 2 или 3."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Check system requirements
|
||||
check_requirements() {
|
||||
print_info "Проверка системных требований..."
|
||||
|
||||
# Check OS
|
||||
if [[ ! -f /etc/os-release ]]; then
|
||||
print_error "Unsupported operating system"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
. /etc/os-release
|
||||
print_success "OS: $NAME $VERSION_ID"
|
||||
|
||||
# Check Python version
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
print_error "Python 3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
print_success "Python version: $PYTHON_VERSION"
|
||||
|
||||
# Check if Python version is >= 3.10
|
||||
if ! python3 -c 'import sys; exit(0 if sys.version_info >= (3, 10) else 1)'; then
|
||||
print_error "Python 3.10+ is required, but $PYTHON_VERSION is installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
print_error "pip3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
print_success "pip3 is available"
|
||||
|
||||
# Check firewall
|
||||
if command -v iptables &> /dev/null; then
|
||||
print_success "iptables is available"
|
||||
elif command -v nft &> /dev/null; then
|
||||
print_success "nftables is available"
|
||||
else
|
||||
print_warning "Neither iptables nor nftables found - firewall functionality may be limited"
|
||||
fi
|
||||
}
|
||||
|
||||
# Install system dependencies
|
||||
install_dependencies() {
|
||||
print_info "Установка системных зависимостей..."
|
||||
|
||||
# Detect package manager
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv python3-dev build-essential \
|
||||
libssl-dev libffi-dev sqlite3 curl wget systemd
|
||||
print_success "Dependencies installed (APT)"
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip python3-devel gcc openssl-devel libffi-devel \
|
||||
sqlite curl wget systemd
|
||||
print_success "Dependencies installed (YUM)"
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip python3-devel gcc openssl-devel libffi-devel \
|
||||
sqlite curl wget systemd
|
||||
print_success "Dependencies installed (DNF)"
|
||||
else
|
||||
print_error "Unsupported package manager"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Create system user
|
||||
create_user() {
|
||||
print_info "Создание системного пользователя..."
|
||||
|
||||
if ! id "$SERVICE_USER" &>/dev/null; then
|
||||
useradd --system --create-home --shell /bin/bash "$SERVICE_USER"
|
||||
print_success "User $SERVICE_USER created"
|
||||
else
|
||||
print_info "User $SERVICE_USER already exists"
|
||||
fi
|
||||
}
|
||||
|
||||
# Create directories
|
||||
create_directories() {
|
||||
print_info "Создание директорий..."
|
||||
|
||||
mkdir -p "$INSTALL_DIR" "$CONFIG_DIR" "$LOG_DIR" "$DATA_DIR"
|
||||
chown -R "$SERVICE_USER:$SERVICE_USER" "$INSTALL_DIR" "$LOG_DIR" "$DATA_DIR"
|
||||
chmod 755 "$CONFIG_DIR"
|
||||
chmod 750 "$LOG_DIR" "$DATA_DIR"
|
||||
|
||||
print_success "Directories created"
|
||||
}
|
||||
|
||||
# Copy application files
|
||||
copy_files() {
|
||||
print_info "Копирование файлов приложения..."
|
||||
|
||||
# Copy source code
|
||||
cp -r "$PROJECT_DIR/src" "$INSTALL_DIR/"
|
||||
cp "$PROJECT_DIR/main.py" "$INSTALL_DIR/"
|
||||
cp "$PROJECT_DIR/requirements.txt" "$INSTALL_DIR/"
|
||||
|
||||
# Copy configuration template
|
||||
if [[ "$INSTALL_MODE" == "standalone" || "$INSTALL_MODE" == "controller" ]]; then
|
||||
cp "$PROJECT_DIR/config/config.yaml" "$CONFIG_DIR/config.yaml.template"
|
||||
fi
|
||||
|
||||
# Set permissions
|
||||
chown -R "$SERVICE_USER:$SERVICE_USER" "$INSTALL_DIR"
|
||||
chmod +x "$INSTALL_DIR/main.py"
|
||||
|
||||
print_success "Files copied"
|
||||
}
|
||||
|
||||
# Install Python dependencies
|
||||
install_python_deps() {
|
||||
print_info "Установка Python зависимостей..."
|
||||
|
||||
# Create virtual environment
|
||||
sudo -u "$SERVICE_USER" python3 -m venv "$INSTALL_DIR/venv"
|
||||
|
||||
# Install dependencies
|
||||
sudo -u "$SERVICE_USER" "$INSTALL_DIR/venv/bin/pip" install -r "$INSTALL_DIR/requirements.txt"
|
||||
|
||||
print_success "Python dependencies installed"
|
||||
}
|
||||
|
||||
# Configure application based on mode
|
||||
configure_application() {
|
||||
print_info "Настройка приложения..."
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"standalone")
|
||||
configure_standalone
|
||||
;;
|
||||
"controller")
|
||||
configure_controller
|
||||
;;
|
||||
"agent")
|
||||
configure_agent
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
configure_standalone() {
|
||||
print_info "Настройка автономного режима..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_telegram_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_standalone_config
|
||||
print_success "Standalone configuration created"
|
||||
}
|
||||
|
||||
configure_controller() {
|
||||
print_info "Настройка контроллера кластера..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_telegram_config
|
||||
get_controller_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_controller_config
|
||||
print_success "Controller configuration created"
|
||||
}
|
||||
|
||||
configure_agent() {
|
||||
print_info "Настройка агента..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_agent_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_agent_config
|
||||
print_success "Agent configuration created"
|
||||
}
|
||||
|
||||
get_telegram_config() {
|
||||
if [[ -z "$TELEGRAM_BOT_TOKEN" ]]; then
|
||||
echo ""
|
||||
print_info "Настройка Telegram бота:"
|
||||
echo "1. Создайте бота у @BotFather"
|
||||
echo "2. Получите токен бота"
|
||||
echo "3. Узнайте ваш chat ID у @userinfobot"
|
||||
echo ""
|
||||
read -p "Введите токен Telegram бота: " TELEGRAM_BOT_TOKEN
|
||||
fi
|
||||
|
||||
if [[ -z "$ADMIN_ID" ]]; then
|
||||
read -p "Введите ваш Telegram ID (admin): " ADMIN_ID
|
||||
fi
|
||||
}
|
||||
|
||||
get_controller_config() {
|
||||
echo ""
|
||||
print_info "Дополнительные настройки контроллера:"
|
||||
read -p "Порт для API контроллера (по умолчанию 8080): " CONTROLLER_PORT
|
||||
CONTROLLER_PORT=${CONTROLLER_PORT:-8080}
|
||||
|
||||
read -p "Максимальное количество агентов (по умолчанию 50): " MAX_AGENTS
|
||||
MAX_AGENTS=${MAX_AGENTS:-50}
|
||||
}
|
||||
|
||||
get_agent_config() {
|
||||
if [[ -z "$CONTROLLER_URL" ]]; then
|
||||
read -p "URL контроллера (например, https://controller.example.com:8080): " CONTROLLER_URL
|
||||
fi
|
||||
|
||||
if [[ -z "$AGENT_TOKEN" ]]; then
|
||||
read -p "Токен агента (получите у администратора контроллера): " AGENT_TOKEN
|
||||
fi
|
||||
|
||||
read -p "Имя агента (по умолчанию: $(hostname)): " AGENT_NAME
|
||||
AGENT_NAME=${AGENT_NAME:-$(hostname)}
|
||||
}
|
||||
|
||||
create_standalone_config() {
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Standalone Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
|
||||
# STEALTH SECURITY SETTINGS
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300
|
||||
compromise_indicators:
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "$DATA_DIR/guardian.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600
|
||||
max_records_age: 604800
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/pyguardian.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
create_controller_config() {
|
||||
CONTROLLER_PORT=${CONTROLLER_PORT:-8080}
|
||||
MAX_AGENTS=${MAX_AGENTS:-50}
|
||||
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Controller Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Operating Mode
|
||||
mode: "controller"
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Controller Settings
|
||||
controller:
|
||||
port: $CONTROLLER_PORT
|
||||
host: "0.0.0.0"
|
||||
max_agents: $MAX_AGENTS
|
||||
agent_timeout: 300
|
||||
heartbeat_interval: 60
|
||||
|
||||
# Cluster Configuration
|
||||
cluster:
|
||||
controller_mode: true
|
||||
auto_deployment: true
|
||||
agent_auto_update: true
|
||||
|
||||
# Security Settings (same as standalone)
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
stealth_mode_duration: 300
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "$DATA_DIR/controller.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/controller.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
create_agent_config() {
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Agent Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Operating Mode
|
||||
mode: "agent"
|
||||
|
||||
# Agent Settings
|
||||
agent:
|
||||
name: "$AGENT_NAME"
|
||||
controller_url: "$CONTROLLER_URL"
|
||||
token: "$AGENT_TOKEN"
|
||||
heartbeat_interval: 60
|
||||
reconnect_delay: 30
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration (local cache)
|
||||
storage:
|
||||
database_path: "$DATA_DIR/agent.db"
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/agent.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
# Create systemd service
|
||||
create_service() {
|
||||
print_info "Создание systemd сервиса..."
|
||||
|
||||
cat > "/etc/systemd/system/pyguardian.service" <<EOF
|
||||
[Unit]
|
||||
Description=PyGuardian Security System
|
||||
After=network.target
|
||||
Wants=network-online.target
|
||||
|
||||
[Service]
|
||||
Type=exec
|
||||
User=$SERVICE_USER
|
||||
Group=$SERVICE_USER
|
||||
WorkingDirectory=$INSTALL_DIR
|
||||
Environment=PATH=$INSTALL_DIR/venv/bin
|
||||
ExecStart=$INSTALL_DIR/venv/bin/python main.py --config=$CONFIG_DIR/config.yaml
|
||||
ExecReload=/bin/kill -HUP \$MAINPID
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
SyslogIdentifier=pyguardian
|
||||
|
||||
# Security settings
|
||||
NoNewPrivileges=true
|
||||
ProtectSystem=strict
|
||||
ProtectHome=true
|
||||
ReadWritePaths=$DATA_DIR $LOG_DIR $CONFIG_DIR
|
||||
PrivateTmp=true
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable pyguardian
|
||||
|
||||
print_success "Systemd service created and enabled"
|
||||
}
|
||||
|
||||
# Start services
|
||||
start_services() {
|
||||
print_info "Запуск сервисов..."
|
||||
|
||||
# Start PyGuardian
|
||||
systemctl start pyguardian
|
||||
|
||||
# Check status
|
||||
if systemctl is-active --quiet pyguardian; then
|
||||
print_success "PyGuardian service started successfully"
|
||||
else
|
||||
print_error "Failed to start PyGuardian service"
|
||||
print_info "Check logs with: journalctl -u pyguardian -f"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show final information
|
||||
show_completion_info() {
|
||||
print_header "Installation Complete"
|
||||
|
||||
echo -e "${GREEN}"
|
||||
echo "✓ PyGuardian успешно установлен в режиме: $INSTALL_MODE"
|
||||
echo ""
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"standalone")
|
||||
echo "Конфигурация:"
|
||||
echo " - Telegram бот настроен"
|
||||
echo " - Мониторинг auth.log активен"
|
||||
echo " - Firewall интеграция готова"
|
||||
;;
|
||||
"controller")
|
||||
echo "Конфигурация контроллера:"
|
||||
echo " - API порт: ${CONTROLLER_PORT:-8080}"
|
||||
echo " - Максимум агентов: ${MAX_AGENTS:-50}"
|
||||
echo " - Telegram управление готово"
|
||||
;;
|
||||
"agent")
|
||||
echo "Конфигурация агента:"
|
||||
echo " - Контроллер: $CONTROLLER_URL"
|
||||
echo " - Имя агента: $AGENT_NAME"
|
||||
echo " - Подключение к кластеру готово"
|
||||
;;
|
||||
esac
|
||||
|
||||
echo -e "${NC}"
|
||||
echo ""
|
||||
print_info "Полезные команды:"
|
||||
echo " systemctl status pyguardian # Проверить статус"
|
||||
echo " systemctl restart pyguardian # Перезапустить"
|
||||
echo " journalctl -u pyguardian -f # Просмотр логов"
|
||||
echo ""
|
||||
print_info "Файлы конфигурации:"
|
||||
echo " $CONFIG_DIR/config.yaml"
|
||||
echo ""
|
||||
print_info "Логи:"
|
||||
echo " $LOG_DIR/"
|
||||
echo ""
|
||||
|
||||
if [[ "$INSTALL_MODE" == "controller" ]]; then
|
||||
echo -e "${YELLOW}"
|
||||
echo "⚠ Не забудьте:"
|
||||
echo " 1. Открыть порт ${CONTROLLER_PORT:-8080} в firewall"
|
||||
echo " 2. Настроить SSL сертификат для HTTPS"
|
||||
echo " 3. Добавить агенты через Telegram команды"
|
||||
echo -e "${NC}"
|
||||
fi
|
||||
|
||||
if [[ "$INSTALL_MODE" == "agent" ]]; then
|
||||
echo -e "${YELLOW}"
|
||||
echo "⚠ Примечание:"
|
||||
echo " Агент будет подключаться к контроллеру автоматически"
|
||||
echo " Проверьте статус подключения в логах"
|
||||
echo -e "${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# Main installation flow
|
||||
#==========================================================================
|
||||
|
||||
main() {
|
||||
# Check if root
|
||||
check_root
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args "$@"
|
||||
|
||||
# Show header
|
||||
print_header "Universal"
|
||||
|
||||
# Select installation mode if not provided
|
||||
if [[ -z "$INSTALL_MODE" ]]; then
|
||||
select_install_mode
|
||||
fi
|
||||
|
||||
print_info "Режим установки: $INSTALL_MODE"
|
||||
|
||||
# Perform installation steps
|
||||
check_requirements
|
||||
install_dependencies
|
||||
create_user
|
||||
create_directories
|
||||
copy_files
|
||||
install_python_deps
|
||||
configure_application
|
||||
create_service
|
||||
start_services
|
||||
|
||||
# Show completion information
|
||||
show_completion_info
|
||||
|
||||
print_success "Установка завершена успешно!"
|
||||
}
|
||||
|
||||
# Run main function with all arguments
|
||||
main "$@"
|
||||
@@ -1,736 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Universal Installation Script
|
||||
# Supports: Standalone, Controller, Agent modes
|
||||
# Author: SmartSolTech Team
|
||||
# Version: 2.0
|
||||
#==========================================================================
|
||||
|
||||
set -e # Exit on any error
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Global variables
|
||||
INSTALL_MODE=""
|
||||
NON_INTERACTIVE=false
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_DIR="$(dirname "$SCRIPT_DIR")"
|
||||
INSTALL_DIR="/opt/pyguardian"
|
||||
SERVICE_USER="pyguardian"
|
||||
CONFIG_DIR="/etc/pyguardian"
|
||||
LOG_DIR="/var/log/pyguardian"
|
||||
DATA_DIR="/var/lib/pyguardian"
|
||||
|
||||
# Configuration variables
|
||||
TELEGRAM_BOT_TOKEN=""
|
||||
ADMIN_ID=""
|
||||
CONTROLLER_URL=""
|
||||
AGENT_TOKEN=""
|
||||
DB_PATH=""
|
||||
|
||||
#==========================================================================
|
||||
# Helper functions
|
||||
#==========================================================================
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "=============================================="
|
||||
echo " PyGuardian $1 Installation"
|
||||
echo "=============================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}✓ $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}✗ $1${NC}"
|
||||
}
|
||||
|
||||
print_warning() {
|
||||
echo -e "${YELLOW}⚠ $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}ℹ $1${NC}"
|
||||
}
|
||||
|
||||
# Check if running as root
|
||||
check_root() {
|
||||
if [[ $EUID -ne 0 ]]; then
|
||||
print_error "This script must be run as root or with sudo"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args() {
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--mode=*)
|
||||
INSTALL_MODE="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--non-interactive)
|
||||
NON_INTERACTIVE=true
|
||||
shift
|
||||
;;
|
||||
--controller-url=*)
|
||||
CONTROLLER_URL="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--agent-token=*)
|
||||
AGENT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--telegram-token=*)
|
||||
TELEGRAM_BOT_TOKEN="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
--admin-id=*)
|
||||
ADMIN_ID="${1#*=}"
|
||||
shift
|
||||
;;
|
||||
-h|--help)
|
||||
show_usage
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
print_error "Unknown option: $1"
|
||||
show_usage
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
show_usage() {
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "OPTIONS:"
|
||||
echo " --mode=MODE Installation mode: standalone, controller, agent"
|
||||
echo " --non-interactive Non-interactive installation"
|
||||
echo " --controller-url=URL Controller URL (for agent mode)"
|
||||
echo " --agent-token=TOKEN Agent authentication token"
|
||||
echo " --telegram-token=TOKEN Telegram bot token"
|
||||
echo " --admin-id=ID Telegram admin ID"
|
||||
echo " -h, --help Show this help"
|
||||
}
|
||||
|
||||
# Interactive mode selection
|
||||
select_install_mode() {
|
||||
if [[ "$NON_INTERACTIVE" == "true" ]]; then
|
||||
return
|
||||
fi
|
||||
|
||||
print_info "Выберите режим установки:"
|
||||
echo ""
|
||||
echo "1) Standalone - Автономный сервер (все в одном)"
|
||||
echo "2) Controller - Центральный контроллер кластера"
|
||||
echo "3) Agent - Агент для подключения к контроллеру"
|
||||
echo ""
|
||||
|
||||
while true; do
|
||||
read -p "Выберите режим (1-3): " choice
|
||||
case $choice in
|
||||
1)
|
||||
INSTALL_MODE="standalone"
|
||||
break
|
||||
;;
|
||||
2)
|
||||
INSTALL_MODE="controller"
|
||||
break
|
||||
;;
|
||||
3)
|
||||
INSTALL_MODE="agent"
|
||||
break
|
||||
;;
|
||||
*)
|
||||
print_error "Неверный выбор. Введите 1, 2 или 3."
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
# Check system requirements
|
||||
check_requirements() {
|
||||
print_info "Проверка системных требований..."
|
||||
|
||||
# Check OS
|
||||
if [[ ! -f /etc/os-release ]]; then
|
||||
print_error "Unsupported operating system"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
. /etc/os-release
|
||||
print_success "OS: $NAME $VERSION_ID"
|
||||
|
||||
# Check Python version
|
||||
if ! command -v python3 &> /dev/null; then
|
||||
print_error "Python 3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
print_success "Python version: $PYTHON_VERSION"
|
||||
|
||||
# Check if Python version is >= 3.10
|
||||
if ! python3 -c 'import sys; exit(0 if sys.version_info >= (3, 10) else 1)'; then
|
||||
print_error "Python 3.10+ is required, but $PYTHON_VERSION is installed"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Check pip
|
||||
if ! command -v pip3 &> /dev/null; then
|
||||
print_error "pip3 is required but not installed"
|
||||
exit 1
|
||||
fi
|
||||
print_success "pip3 is available"
|
||||
|
||||
# Check firewall
|
||||
if command -v iptables &> /dev/null; then
|
||||
print_success "iptables is available"
|
||||
elif command -v nft &> /dev/null; then
|
||||
print_success "nftables is available"
|
||||
else
|
||||
print_warning "Neither iptables nor nftables found - firewall functionality may be limited"
|
||||
fi
|
||||
}
|
||||
|
||||
# Install system dependencies
|
||||
install_dependencies() {
|
||||
print_info "Установка системных зависимостей..."
|
||||
|
||||
# Detect package manager
|
||||
if command -v apt-get &> /dev/null; then
|
||||
apt-get update
|
||||
apt-get install -y python3-pip python3-venv python3-dev build-essential \
|
||||
libssl-dev libffi-dev sqlite3 curl wget systemd
|
||||
print_success "Dependencies installed (APT)"
|
||||
elif command -v yum &> /dev/null; then
|
||||
yum install -y python3-pip python3-devel gcc openssl-devel libffi-devel \
|
||||
sqlite curl wget systemd
|
||||
print_success "Dependencies installed (YUM)"
|
||||
elif command -v dnf &> /dev/null; then
|
||||
dnf install -y python3-pip python3-devel gcc openssl-devel libffi-devel \
|
||||
sqlite curl wget systemd
|
||||
print_success "Dependencies installed (DNF)"
|
||||
else
|
||||
print_error "Unsupported package manager"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Create system user
|
||||
create_user() {
|
||||
print_info "Создание системного пользователя..."
|
||||
|
||||
if ! id "$SERVICE_USER" &>/dev/null; then
|
||||
useradd --system --create-home --shell /bin/bash "$SERVICE_USER"
|
||||
print_success "User $SERVICE_USER created"
|
||||
else
|
||||
print_info "User $SERVICE_USER already exists"
|
||||
fi
|
||||
}
|
||||
|
||||
# Create directories
|
||||
create_directories() {
|
||||
print_info "Создание директорий..."
|
||||
|
||||
mkdir -p "$INSTALL_DIR" "$CONFIG_DIR" "$LOG_DIR" "$DATA_DIR"
|
||||
chown -R "$SERVICE_USER:$SERVICE_USER" "$INSTALL_DIR" "$LOG_DIR" "$DATA_DIR"
|
||||
chmod 755 "$CONFIG_DIR"
|
||||
chmod 750 "$LOG_DIR" "$DATA_DIR"
|
||||
|
||||
print_success "Directories created"
|
||||
}
|
||||
|
||||
# Copy application files
|
||||
copy_files() {
|
||||
print_info "Копирование файлов приложения..."
|
||||
|
||||
# Copy source code
|
||||
cp -r "$PROJECT_DIR/src" "$INSTALL_DIR/"
|
||||
cp "$PROJECT_DIR/main.py" "$INSTALL_DIR/"
|
||||
cp "$PROJECT_DIR/requirements.txt" "$INSTALL_DIR/"
|
||||
|
||||
# Copy configuration template
|
||||
if [[ "$INSTALL_MODE" == "standalone" || "$INSTALL_MODE" == "controller" ]]; then
|
||||
cp "$PROJECT_DIR/config/config.yaml" "$CONFIG_DIR/config.yaml.template"
|
||||
fi
|
||||
|
||||
# Set permissions
|
||||
chown -R "$SERVICE_USER:$SERVICE_USER" "$INSTALL_DIR"
|
||||
chmod +x "$INSTALL_DIR/main.py"
|
||||
|
||||
print_success "Files copied"
|
||||
}
|
||||
|
||||
# Install Python dependencies
|
||||
install_python_deps() {
|
||||
print_info "Установка Python зависимостей..."
|
||||
|
||||
# Create virtual environment
|
||||
sudo -u "$SERVICE_USER" python3 -m venv "$INSTALL_DIR/venv"
|
||||
|
||||
# Install dependencies
|
||||
sudo -u "$SERVICE_USER" "$INSTALL_DIR/venv/bin/pip" install -r "$INSTALL_DIR/requirements.txt"
|
||||
|
||||
print_success "Python dependencies installed"
|
||||
}
|
||||
|
||||
# Configure application based on mode
|
||||
configure_application() {
|
||||
print_info "Настройка приложения..."
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"standalone")
|
||||
configure_standalone
|
||||
;;
|
||||
"controller")
|
||||
configure_controller
|
||||
;;
|
||||
"agent")
|
||||
configure_agent
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
configure_standalone() {
|
||||
print_info "Настройка автономного режима..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_telegram_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_standalone_config
|
||||
print_success "Standalone configuration created"
|
||||
}
|
||||
|
||||
configure_controller() {
|
||||
print_info "Настройка контроллера кластера..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_telegram_config
|
||||
get_controller_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_controller_config
|
||||
print_success "Controller configuration created"
|
||||
}
|
||||
|
||||
configure_agent() {
|
||||
print_info "Настройка агента..."
|
||||
|
||||
# Get configuration from user
|
||||
if [[ "$NON_INTERACTIVE" != "true" ]]; then
|
||||
get_agent_config
|
||||
fi
|
||||
|
||||
# Create configuration file
|
||||
create_agent_config
|
||||
print_success "Agent configuration created"
|
||||
}
|
||||
|
||||
get_telegram_config() {
|
||||
if [[ -z "$TELEGRAM_BOT_TOKEN" ]]; then
|
||||
echo ""
|
||||
print_info "Настройка Telegram бота:"
|
||||
echo "1. Создайте бота у @BotFather"
|
||||
echo "2. Получите токен бота"
|
||||
echo "3. Узнайте ваш chat ID у @userinfobot"
|
||||
echo ""
|
||||
read -p "Введите токен Telegram бота: " TELEGRAM_BOT_TOKEN
|
||||
fi
|
||||
|
||||
if [[ -z "$ADMIN_ID" ]]; then
|
||||
read -p "Введите ваш Telegram ID (admin): " ADMIN_ID
|
||||
fi
|
||||
}
|
||||
|
||||
get_controller_config() {
|
||||
echo ""
|
||||
print_info "Дополнительные настройки контроллера:"
|
||||
read -p "Порт для API контроллера (по умолчанию 8080): " CONTROLLER_PORT
|
||||
CONTROLLER_PORT=${CONTROLLER_PORT:-8080}
|
||||
|
||||
read -p "Максимальное количество агентов (по умолчанию 50): " MAX_AGENTS
|
||||
MAX_AGENTS=${MAX_AGENTS:-50}
|
||||
}
|
||||
|
||||
get_agent_config() {
|
||||
if [[ -z "$CONTROLLER_URL" ]]; then
|
||||
read -p "URL контроллера (например, https://controller.example.com:8080): " CONTROLLER_URL
|
||||
fi
|
||||
|
||||
if [[ -z "$AGENT_TOKEN" ]]; then
|
||||
read -p "Токен агента (получите у администратора контроллера): " AGENT_TOKEN
|
||||
fi
|
||||
|
||||
read -p "Имя агента (по умолчанию: $(hostname)): " AGENT_NAME
|
||||
AGENT_NAME=${AGENT_NAME:-$(hostname)}
|
||||
}
|
||||
|
||||
create_standalone_config() {
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Standalone Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Security & Monitoring Settings
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
|
||||
# STEALTH SECURITY SETTINGS
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
|
||||
stealth_mode_duration: 300
|
||||
compromise_indicators:
|
||||
- "suspicious_commands"
|
||||
- "unusual_login_times"
|
||||
- "multiple_failed_then_success"
|
||||
- "honeypot_access"
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
failed_patterns:
|
||||
- "Failed password"
|
||||
- "Invalid user"
|
||||
- "authentication failure"
|
||||
- "Too many authentication failures"
|
||||
- "Failed publickey"
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "$DATA_DIR/guardian.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Performance Settings
|
||||
performance:
|
||||
cleanup_interval: 3600
|
||||
max_records_age: 604800
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/pyguardian.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
create_controller_config() {
|
||||
CONTROLLER_PORT=${CONTROLLER_PORT:-8080}
|
||||
MAX_AGENTS=${MAX_AGENTS:-50}
|
||||
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Controller Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Operating Mode
|
||||
mode: "controller"
|
||||
|
||||
# Telegram Bot Configuration
|
||||
telegram:
|
||||
bot_token: "$TELEGRAM_BOT_TOKEN"
|
||||
admin_id: $ADMIN_ID
|
||||
|
||||
# Controller Settings
|
||||
controller:
|
||||
port: $CONTROLLER_PORT
|
||||
host: "0.0.0.0"
|
||||
max_agents: $MAX_AGENTS
|
||||
agent_timeout: 300
|
||||
heartbeat_interval: 60
|
||||
|
||||
# Cluster Configuration
|
||||
cluster:
|
||||
controller_mode: true
|
||||
auto_deployment: true
|
||||
agent_auto_update: true
|
||||
|
||||
# Security Settings (same as standalone)
|
||||
security:
|
||||
max_attempts: 5
|
||||
time_window: 60
|
||||
unban_time: 3600
|
||||
authorized_users:
|
||||
- "root"
|
||||
- "admin"
|
||||
- "ubuntu"
|
||||
honeypot_users:
|
||||
- "test"
|
||||
- "guest"
|
||||
- "user"
|
||||
- "admin123"
|
||||
- "backup"
|
||||
stealth_mode_duration: 300
|
||||
|
||||
# Storage Configuration
|
||||
storage:
|
||||
database_path: "$DATA_DIR/controller.db"
|
||||
|
||||
# Password Management
|
||||
passwords:
|
||||
password_length: 16
|
||||
use_special_chars: true
|
||||
password_history_size: 5
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/controller.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
create_agent_config() {
|
||||
cat > "$CONFIG_DIR/config.yaml" <<EOF
|
||||
# PyGuardian Agent Configuration
|
||||
# Generated: $(date)
|
||||
|
||||
# Operating Mode
|
||||
mode: "agent"
|
||||
|
||||
# Agent Settings
|
||||
agent:
|
||||
name: "$AGENT_NAME"
|
||||
controller_url: "$CONTROLLER_URL"
|
||||
token: "$AGENT_TOKEN"
|
||||
heartbeat_interval: 60
|
||||
reconnect_delay: 30
|
||||
|
||||
# Log Monitoring
|
||||
monitoring:
|
||||
auth_log_path: "/var/log/auth.log"
|
||||
check_interval: 1.0
|
||||
|
||||
# Firewall Configuration
|
||||
firewall:
|
||||
backend: "iptables"
|
||||
chain: "INPUT"
|
||||
target: "DROP"
|
||||
|
||||
# Storage Configuration (local cache)
|
||||
storage:
|
||||
database_path: "$DATA_DIR/agent.db"
|
||||
|
||||
# Logging Configuration
|
||||
logging:
|
||||
level: "INFO"
|
||||
file: "$LOG_DIR/agent.log"
|
||||
max_size: 10485760
|
||||
backup_count: 5
|
||||
EOF
|
||||
|
||||
chown "$SERVICE_USER:$SERVICE_USER" "$CONFIG_DIR/config.yaml"
|
||||
chmod 640 "$CONFIG_DIR/config.yaml"
|
||||
}
|
||||
|
||||
# Create systemd service
|
||||
create_service() {
|
||||
print_info "Создание systemd сервиса..."
|
||||
|
||||
cat > "/etc/systemd/system/pyguardian.service" <<EOF
|
||||
[Unit]
|
||||
Description=PyGuardian Security System
|
||||
After=network.target
|
||||
Wants=network-online.target
|
||||
|
||||
[Service]
|
||||
Type=exec
|
||||
User=$SERVICE_USER
|
||||
Group=$SERVICE_USER
|
||||
WorkingDirectory=$INSTALL_DIR
|
||||
Environment=PATH=$INSTALL_DIR/venv/bin
|
||||
ExecStart=$INSTALL_DIR/venv/bin/python main.py --config=$CONFIG_DIR/config.yaml
|
||||
ExecReload=/bin/kill -HUP \$MAINPID
|
||||
Restart=always
|
||||
RestartSec=10
|
||||
StandardOutput=journal
|
||||
StandardError=journal
|
||||
SyslogIdentifier=pyguardian
|
||||
|
||||
# Security settings
|
||||
NoNewPrivileges=true
|
||||
ProtectSystem=strict
|
||||
ProtectHome=true
|
||||
ReadWritePaths=$DATA_DIR $LOG_DIR $CONFIG_DIR
|
||||
PrivateTmp=true
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
EOF
|
||||
|
||||
systemctl daemon-reload
|
||||
systemctl enable pyguardian
|
||||
|
||||
print_success "Systemd service created and enabled"
|
||||
}
|
||||
|
||||
# Start services
|
||||
start_services() {
|
||||
print_info "Запуск сервисов..."
|
||||
|
||||
# Start PyGuardian
|
||||
systemctl start pyguardian
|
||||
|
||||
# Check status
|
||||
if systemctl is-active --quiet pyguardian; then
|
||||
print_success "PyGuardian service started successfully"
|
||||
else
|
||||
print_error "Failed to start PyGuardian service"
|
||||
print_info "Check logs with: journalctl -u pyguardian -f"
|
||||
exit 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Show final information
|
||||
show_completion_info() {
|
||||
print_header "Installation Complete"
|
||||
|
||||
echo -e "${GREEN}"
|
||||
echo "✓ PyGuardian успешно установлен в режиме: $INSTALL_MODE"
|
||||
echo ""
|
||||
|
||||
case "$INSTALL_MODE" in
|
||||
"standalone")
|
||||
echo "Конфигурация:"
|
||||
echo " - Telegram бот настроен"
|
||||
echo " - Мониторинг auth.log активен"
|
||||
echo " - Firewall интеграция готова"
|
||||
;;
|
||||
"controller")
|
||||
echo "Конфигурация контроллера:"
|
||||
echo " - API порт: ${CONTROLLER_PORT:-8080}"
|
||||
echo " - Максимум агентов: ${MAX_AGENTS:-50}"
|
||||
echo " - Telegram управление готово"
|
||||
;;
|
||||
"agent")
|
||||
echo "Конфигурация агента:"
|
||||
echo " - Контроллер: $CONTROLLER_URL"
|
||||
echo " - Имя агента: $AGENT_NAME"
|
||||
echo " - Подключение к кластеру готово"
|
||||
;;
|
||||
esac
|
||||
|
||||
echo -e "${NC}"
|
||||
echo ""
|
||||
print_info "Полезные команды:"
|
||||
echo " systemctl status pyguardian # Проверить статус"
|
||||
echo " systemctl restart pyguardian # Перезапустить"
|
||||
echo " journalctl -u pyguardian -f # Просмотр логов"
|
||||
echo ""
|
||||
print_info "Файлы конфигурации:"
|
||||
echo " $CONFIG_DIR/config.yaml"
|
||||
echo ""
|
||||
print_info "Логи:"
|
||||
echo " $LOG_DIR/"
|
||||
echo ""
|
||||
|
||||
if [[ "$INSTALL_MODE" == "controller" ]]; then
|
||||
echo -e "${YELLOW}"
|
||||
echo "⚠ Не забудьте:"
|
||||
echo " 1. Открыть порт ${CONTROLLER_PORT:-8080} в firewall"
|
||||
echo " 2. Настроить SSL сертификат для HTTPS"
|
||||
echo " 3. Добавить агенты через Telegram команды"
|
||||
echo -e "${NC}"
|
||||
fi
|
||||
|
||||
if [[ "$INSTALL_MODE" == "agent" ]]; then
|
||||
echo -e "${YELLOW}"
|
||||
echo "⚠ Примечание:"
|
||||
echo " Агент будет подключаться к контроллеру автоматически"
|
||||
echo " Проверьте статус подключения в логах"
|
||||
echo -e "${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
#==========================================================================
|
||||
# Main installation flow
|
||||
#==========================================================================
|
||||
|
||||
main() {
|
||||
# Check if root
|
||||
check_root
|
||||
|
||||
# Parse command line arguments
|
||||
parse_args "$@"
|
||||
|
||||
# Show header
|
||||
print_header "Universal"
|
||||
|
||||
# Select installation mode if not provided
|
||||
if [[ -z "$INSTALL_MODE" ]]; then
|
||||
select_install_mode
|
||||
fi
|
||||
|
||||
print_info "Режим установки: $INSTALL_MODE"
|
||||
|
||||
# Perform installation steps
|
||||
check_requirements
|
||||
install_dependencies
|
||||
create_user
|
||||
create_directories
|
||||
copy_files
|
||||
install_python_deps
|
||||
configure_application
|
||||
create_service
|
||||
start_services
|
||||
|
||||
# Show completion information
|
||||
show_completion_info
|
||||
|
||||
print_success "Установка завершена успешно!"
|
||||
}
|
||||
|
||||
# Run main function with all arguments
|
||||
main "$@"
|
||||
@@ -1,356 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Test Script
|
||||
# Демонстрация возможностей системы установки
|
||||
# Author: SmartSolTech Team
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "================================================="
|
||||
echo " PyGuardian Installation Test Suite"
|
||||
echo "================================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_test() {
|
||||
echo -e "${YELLOW}[TEST] $1${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}[PASS] $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}[INFO] $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}[FAIL] $1${NC}"
|
||||
}
|
||||
|
||||
# Test 1: Check if all installation scripts exist
|
||||
test_scripts_exist() {
|
||||
print_test "Проверка существования скриптов установки"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
"Makefile"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if [[ -f "$script" ]]; then
|
||||
print_success "Найден: $script"
|
||||
else
|
||||
print_error "Отсутствует: $script"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Все скрипты установки найдены"
|
||||
}
|
||||
|
||||
# Test 2: Check if scripts are executable
|
||||
test_scripts_executable() {
|
||||
print_test "Проверка прав выполнения скриптов"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if [[ -x "$script" ]]; then
|
||||
print_success "Исполняемый: $script"
|
||||
else
|
||||
print_error "Не исполняемый: $script"
|
||||
chmod +x "$script" 2>/dev/null && print_info "Исправлено: $script"
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Все скрипты исполняемы"
|
||||
}
|
||||
|
||||
# Test 3: Check Python requirements
|
||||
test_python_requirements() {
|
||||
print_test "Проверка Python требований"
|
||||
|
||||
if command -v python3 &> /dev/null; then
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
print_success "Python version: $PYTHON_VERSION"
|
||||
|
||||
if python3 -c 'import sys; exit(0 if sys.version_info >= (3, 10) else 1)'; then
|
||||
print_success "Python версия соответствует требованиям (>=3.10)"
|
||||
else
|
||||
print_error "Python версия не соответствует требованиям (требуется >=3.10)"
|
||||
return 1
|
||||
fi
|
||||
else
|
||||
print_error "Python3 не найден"
|
||||
return 1
|
||||
fi
|
||||
|
||||
if command -v pip3 &> /dev/null; then
|
||||
print_success "pip3 доступен"
|
||||
else
|
||||
print_error "pip3 не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 4: Check dependencies in requirements.txt
|
||||
test_requirements_file() {
|
||||
print_test "Проверка файла requirements.txt"
|
||||
|
||||
if [[ -f "requirements.txt" ]]; then
|
||||
print_success "Файл requirements.txt найден"
|
||||
|
||||
local required_packages=(
|
||||
"telegram"
|
||||
"aiosqlite"
|
||||
"pyyaml"
|
||||
"cryptography"
|
||||
"psutil"
|
||||
)
|
||||
|
||||
for package in "${required_packages[@]}"; do
|
||||
if grep -q "$package" requirements.txt; then
|
||||
print_success "Зависимость найдена: $package"
|
||||
else
|
||||
print_error "Зависимость отсутствует: $package"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Файл requirements.txt не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 5: Check configuration files
|
||||
test_config_files() {
|
||||
print_test "Проверка конфигурационных файлов"
|
||||
|
||||
if [[ -f "config/config.yaml" ]]; then
|
||||
print_success "Основной конфиг найден: config/config.yaml"
|
||||
|
||||
# Check for required sections
|
||||
local sections=("telegram" "security" "firewall" "storage")
|
||||
for section in "${sections[@]}"; do
|
||||
if grep -q "^${section}:" config/config.yaml; then
|
||||
print_success "Секция конфигурации: $section"
|
||||
else
|
||||
print_error "Отсутствует секция: $section"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Основной конфиг не найден: config/config.yaml"
|
||||
return 1
|
||||
fi
|
||||
|
||||
# Check for cluster configuration
|
||||
if grep -q "cluster:" config/config.yaml; then
|
||||
print_success "Кластерная конфигурация найдена"
|
||||
else
|
||||
print_info "Кластерная конфигурация отсутствует (будет добавлена при установке)"
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 6: Check source code structure
|
||||
test_source_structure() {
|
||||
print_test "Проверка структуры исходного кода"
|
||||
|
||||
local source_files=(
|
||||
"src/storage.py"
|
||||
"src/firewall.py"
|
||||
"src/monitor.py"
|
||||
"src/bot.py"
|
||||
"src/security.py"
|
||||
"src/sessions.py"
|
||||
"src/password_utils.py"
|
||||
"src/cluster.py"
|
||||
"main.py"
|
||||
)
|
||||
|
||||
for file in "${source_files[@]}"; do
|
||||
if [[ -f "$file" ]]; then
|
||||
print_success "Исходный файл: $file"
|
||||
else
|
||||
print_error "Отсутствует файл: $file"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Структура исходного кода корректна"
|
||||
}
|
||||
|
||||
# Test 7: Check Makefile targets
|
||||
test_makefile_targets() {
|
||||
print_test "Проверка целей Makefile"
|
||||
|
||||
if [[ -f "Makefile" ]]; then
|
||||
local targets=("install" "standalone" "controller" "agent" "help" "clean")
|
||||
for target in "${targets[@]}"; do
|
||||
if grep -q "^${target}:" Makefile; then
|
||||
print_success "Makefile цель: $target"
|
||||
else
|
||||
print_error "Отсутствует цель: $target"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Makefile не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 8: Validate script syntax
|
||||
test_script_syntax() {
|
||||
print_test "Проверка синтаксиса скриптов"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if bash -n "$script" 2>/dev/null; then
|
||||
print_success "Синтаксис корректен: $script"
|
||||
else
|
||||
print_error "Синтаксическая ошибка в: $script"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Test 9: Check documentation
|
||||
test_documentation() {
|
||||
print_test "Проверка документации"
|
||||
|
||||
local docs=(
|
||||
"README.md"
|
||||
"docs/INSTALLATION.md"
|
||||
"docs/CLUSTER_SETUP.md"
|
||||
)
|
||||
|
||||
for doc in "${docs[@]}"; do
|
||||
if [[ -f "$doc" ]]; then
|
||||
print_success "Документация: $doc"
|
||||
else
|
||||
print_error "Отсутствует документация: $doc"
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Test 10: Simulate installation steps (dry run)
|
||||
test_installation_simulation() {
|
||||
print_test "Симуляция процесса установки"
|
||||
|
||||
# Test help output
|
||||
if ./install.sh --help >/dev/null 2>&1; then
|
||||
print_success "Справка install.sh работает"
|
||||
else
|
||||
print_error "Ошибка в справке install.sh"
|
||||
fi
|
||||
|
||||
# Test make help
|
||||
if make help >/dev/null 2>&1; then
|
||||
print_success "Справка Makefile работает"
|
||||
else
|
||||
print_error "Ошибка в справке Makefile"
|
||||
fi
|
||||
|
||||
print_success "Симуляция установки завершена"
|
||||
}
|
||||
|
||||
# Run all tests
|
||||
run_all_tests() {
|
||||
print_header
|
||||
|
||||
local tests=(
|
||||
"test_scripts_exist"
|
||||
"test_scripts_executable"
|
||||
"test_python_requirements"
|
||||
"test_requirements_file"
|
||||
"test_config_files"
|
||||
"test_source_structure"
|
||||
"test_makefile_targets"
|
||||
"test_script_syntax"
|
||||
"test_documentation"
|
||||
"test_installation_simulation"
|
||||
)
|
||||
|
||||
local passed=0
|
||||
local total=${#tests[@]}
|
||||
|
||||
for test in "${tests[@]}"; do
|
||||
echo ""
|
||||
if $test; then
|
||||
((passed++))
|
||||
fi
|
||||
done
|
||||
|
||||
echo ""
|
||||
echo "================================================="
|
||||
if [[ $passed -eq $total ]]; then
|
||||
print_success "Все тесты пройдены: $passed/$total ✅"
|
||||
echo ""
|
||||
print_info "Система готова к установке!"
|
||||
print_info "Используйте: sudo ./install.sh"
|
||||
print_info "Или: sudo make install"
|
||||
else
|
||||
print_error "Тесты не пройдены: $passed/$total ❌"
|
||||
echo ""
|
||||
print_info "Исправьте ошибки перед установкой"
|
||||
fi
|
||||
echo "================================================="
|
||||
}
|
||||
|
||||
# Main function
|
||||
main() {
|
||||
case "${1:-all}" in
|
||||
"all")
|
||||
run_all_tests
|
||||
;;
|
||||
"scripts")
|
||||
test_scripts_exist && test_scripts_executable && test_script_syntax
|
||||
;;
|
||||
"python")
|
||||
test_python_requirements && test_requirements_file
|
||||
;;
|
||||
"config")
|
||||
test_config_files
|
||||
;;
|
||||
"structure")
|
||||
test_source_structure
|
||||
;;
|
||||
"docs")
|
||||
test_documentation
|
||||
;;
|
||||
*)
|
||||
echo "Usage: $0 [all|scripts|python|config|structure|docs]"
|
||||
echo ""
|
||||
echo "Tests available:"
|
||||
echo " all - Run all tests (default)"
|
||||
echo " scripts - Test installation scripts"
|
||||
echo " python - Test Python requirements"
|
||||
echo " config - Test configuration files"
|
||||
echo " structure - Test source code structure"
|
||||
echo " docs - Test documentation"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
main "$@"
|
||||
@@ -1,356 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
#==========================================================================
|
||||
# PyGuardian Test Script
|
||||
# Демонстрация возможностей системы установки
|
||||
# Author: SmartSolTech Team
|
||||
#==========================================================================
|
||||
|
||||
set -e
|
||||
|
||||
# Colors
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
BLUE='\033[0;34m'
|
||||
YELLOW='\033[1;33m'
|
||||
NC='\033[0m'
|
||||
|
||||
print_header() {
|
||||
echo -e "${BLUE}"
|
||||
echo "================================================="
|
||||
echo " PyGuardian Installation Test Suite"
|
||||
echo "================================================="
|
||||
echo -e "${NC}"
|
||||
}
|
||||
|
||||
print_test() {
|
||||
echo -e "${YELLOW}[TEST] $1${NC}"
|
||||
}
|
||||
|
||||
print_success() {
|
||||
echo -e "${GREEN}[PASS] $1${NC}"
|
||||
}
|
||||
|
||||
print_info() {
|
||||
echo -e "${BLUE}[INFO] $1${NC}"
|
||||
}
|
||||
|
||||
print_error() {
|
||||
echo -e "${RED}[FAIL] $1${NC}"
|
||||
}
|
||||
|
||||
# Test 1: Check if all installation scripts exist
|
||||
test_scripts_exist() {
|
||||
print_test "Проверка существования скриптов установки"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
"Makefile"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if [[ -f "$script" ]]; then
|
||||
print_success "Найден: $script"
|
||||
else
|
||||
print_error "Отсутствует: $script"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Все скрипты установки найдены"
|
||||
}
|
||||
|
||||
# Test 2: Check if scripts are executable
|
||||
test_scripts_executable() {
|
||||
print_test "Проверка прав выполнения скриптов"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if [[ -x "$script" ]]; then
|
||||
print_success "Исполняемый: $script"
|
||||
else
|
||||
print_error "Не исполняемый: $script"
|
||||
chmod +x "$script" 2>/dev/null && print_info "Исправлено: $script"
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Все скрипты исполняемы"
|
||||
}
|
||||
|
||||
# Test 3: Check Python requirements
|
||||
test_python_requirements() {
|
||||
print_test "Проверка Python требований"
|
||||
|
||||
if command -v python3 &> /dev/null; then
|
||||
PYTHON_VERSION=$(python3 -c 'import sys; print(".".join(map(str, sys.version_info[:2])))')
|
||||
print_success "Python version: $PYTHON_VERSION"
|
||||
|
||||
if python3 -c 'import sys; exit(0 if sys.version_info >= (3, 10) else 1)'; then
|
||||
print_success "Python версия соответствует требованиям (>=3.10)"
|
||||
else
|
||||
print_error "Python версия не соответствует требованиям (требуется >=3.10)"
|
||||
return 1
|
||||
fi
|
||||
else
|
||||
print_error "Python3 не найден"
|
||||
return 1
|
||||
fi
|
||||
|
||||
if command -v pip3 &> /dev/null; then
|
||||
print_success "pip3 доступен"
|
||||
else
|
||||
print_error "pip3 не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 4: Check dependencies in requirements.txt
|
||||
test_requirements_file() {
|
||||
print_test "Проверка файла requirements.txt"
|
||||
|
||||
if [[ -f "requirements.txt" ]]; then
|
||||
print_success "Файл requirements.txt найден"
|
||||
|
||||
local required_packages=(
|
||||
"telegram"
|
||||
"aiosqlite"
|
||||
"pyyaml"
|
||||
"cryptography"
|
||||
"psutil"
|
||||
)
|
||||
|
||||
for package in "${required_packages[@]}"; do
|
||||
if grep -q "$package" requirements.txt; then
|
||||
print_success "Зависимость найдена: $package"
|
||||
else
|
||||
print_error "Зависимость отсутствует: $package"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Файл requirements.txt не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 5: Check configuration files
|
||||
test_config_files() {
|
||||
print_test "Проверка конфигурационных файлов"
|
||||
|
||||
if [[ -f "config/config.yaml" ]]; then
|
||||
print_success "Основной конфиг найден: config/config.yaml"
|
||||
|
||||
# Check for required sections
|
||||
local sections=("telegram" "security" "firewall" "storage")
|
||||
for section in "${sections[@]}"; do
|
||||
if grep -q "^${section}:" config/config.yaml; then
|
||||
print_success "Секция конфигурации: $section"
|
||||
else
|
||||
print_error "Отсутствует секция: $section"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Основной конфиг не найден: config/config.yaml"
|
||||
return 1
|
||||
fi
|
||||
|
||||
# Check for cluster configuration
|
||||
if grep -q "cluster:" config/config.yaml; then
|
||||
print_success "Кластерная конфигурация найдена"
|
||||
else
|
||||
print_info "Кластерная конфигурация отсутствует (будет добавлена при установке)"
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 6: Check source code structure
|
||||
test_source_structure() {
|
||||
print_test "Проверка структуры исходного кода"
|
||||
|
||||
local source_files=(
|
||||
"src/storage.py"
|
||||
"src/firewall.py"
|
||||
"src/monitor.py"
|
||||
"src/bot.py"
|
||||
"src/security.py"
|
||||
"src/sessions.py"
|
||||
"src/password_utils.py"
|
||||
"src/cluster.py"
|
||||
"main.py"
|
||||
)
|
||||
|
||||
for file in "${source_files[@]}"; do
|
||||
if [[ -f "$file" ]]; then
|
||||
print_success "Исходный файл: $file"
|
||||
else
|
||||
print_error "Отсутствует файл: $file"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
||||
print_success "Структура исходного кода корректна"
|
||||
}
|
||||
|
||||
# Test 7: Check Makefile targets
|
||||
test_makefile_targets() {
|
||||
print_test "Проверка целей Makefile"
|
||||
|
||||
if [[ -f "Makefile" ]]; then
|
||||
local targets=("install" "standalone" "controller" "agent" "help" "clean")
|
||||
for target in "${targets[@]}"; do
|
||||
if grep -q "^${target}:" Makefile; then
|
||||
print_success "Makefile цель: $target"
|
||||
else
|
||||
print_error "Отсутствует цель: $target"
|
||||
fi
|
||||
done
|
||||
else
|
||||
print_error "Makefile не найден"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
# Test 8: Validate script syntax
|
||||
test_script_syntax() {
|
||||
print_test "Проверка синтаксиса скриптов"
|
||||
|
||||
local scripts=(
|
||||
"install.sh"
|
||||
"scripts/install.sh"
|
||||
"scripts/docker-install.sh"
|
||||
)
|
||||
|
||||
for script in "${scripts[@]}"; do
|
||||
if bash -n "$script" 2>/dev/null; then
|
||||
print_success "Синтаксис корректен: $script"
|
||||
else
|
||||
print_error "Синтаксическая ошибка в: $script"
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Test 9: Check documentation
|
||||
test_documentation() {
|
||||
print_test "Проверка документации"
|
||||
|
||||
local docs=(
|
||||
"README.md"
|
||||
"docs/INSTALLATION.md"
|
||||
"docs/CLUSTER_SETUP.md"
|
||||
)
|
||||
|
||||
for doc in "${docs[@]}"; do
|
||||
if [[ -f "$doc" ]]; then
|
||||
print_success "Документация: $doc"
|
||||
else
|
||||
print_error "Отсутствует документация: $doc"
|
||||
fi
|
||||
done
|
||||
}
|
||||
|
||||
# Test 10: Simulate installation steps (dry run)
|
||||
test_installation_simulation() {
|
||||
print_test "Симуляция процесса установки"
|
||||
|
||||
# Test help output
|
||||
if ./install.sh --help >/dev/null 2>&1; then
|
||||
print_success "Справка install.sh работает"
|
||||
else
|
||||
print_error "Ошибка в справке install.sh"
|
||||
fi
|
||||
|
||||
# Test make help
|
||||
if make help >/dev/null 2>&1; then
|
||||
print_success "Справка Makefile работает"
|
||||
else
|
||||
print_error "Ошибка в справке Makefile"
|
||||
fi
|
||||
|
||||
print_success "Симуляция установки завершена"
|
||||
}
|
||||
|
||||
# Run all tests
|
||||
run_all_tests() {
|
||||
print_header
|
||||
|
||||
local tests=(
|
||||
"test_scripts_exist"
|
||||
"test_scripts_executable"
|
||||
"test_python_requirements"
|
||||
"test_requirements_file"
|
||||
"test_config_files"
|
||||
"test_source_structure"
|
||||
"test_makefile_targets"
|
||||
"test_script_syntax"
|
||||
"test_documentation"
|
||||
"test_installation_simulation"
|
||||
)
|
||||
|
||||
local passed=0
|
||||
local total=${#tests[@]}
|
||||
|
||||
for test in "${tests[@]}"; do
|
||||
echo ""
|
||||
if $test; then
|
||||
((passed++))
|
||||
fi
|
||||
done
|
||||
|
||||
echo ""
|
||||
echo "================================================="
|
||||
if [[ $passed -eq $total ]]; then
|
||||
print_success "Все тесты пройдены: $passed/$total ✅"
|
||||
echo ""
|
||||
print_info "Система готова к установке!"
|
||||
print_info "Используйте: sudo ./install.sh"
|
||||
print_info "Или: sudo make install"
|
||||
else
|
||||
print_error "Тесты не пройдены: $passed/$total ❌"
|
||||
echo ""
|
||||
print_info "Исправьте ошибки перед установкой"
|
||||
fi
|
||||
echo "================================================="
|
||||
}
|
||||
|
||||
# Main function
|
||||
main() {
|
||||
case "${1:-all}" in
|
||||
"all")
|
||||
run_all_tests
|
||||
;;
|
||||
"scripts")
|
||||
test_scripts_exist && test_scripts_executable && test_script_syntax
|
||||
;;
|
||||
"python")
|
||||
test_python_requirements && test_requirements_file
|
||||
;;
|
||||
"config")
|
||||
test_config_files
|
||||
;;
|
||||
"structure")
|
||||
test_source_structure
|
||||
;;
|
||||
"docs")
|
||||
test_documentation
|
||||
;;
|
||||
*)
|
||||
echo "Usage: $0 [all|scripts|python|config|structure|docs]"
|
||||
echo ""
|
||||
echo "Tests available:"
|
||||
echo " all - Run all tests (default)"
|
||||
echo " scripts - Test installation scripts"
|
||||
echo " python - Test Python requirements"
|
||||
echo " config - Test configuration files"
|
||||
echo " structure - Test source code structure"
|
||||
echo " docs - Test documentation"
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
main "$@"
|
||||
@@ -1 +0,0 @@
|
||||
# PyGuardian - Linux Server Protection System
|
||||
@@ -1 +0,0 @@
|
||||
# PyGuardian - Linux Server Protection System
|
||||
@@ -1,727 +0,0 @@
|
||||
"""
|
||||
API Server for PyGuardian Controller
|
||||
REST API endpoints for agent authentication and cluster management
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import asyncio
|
||||
from datetime import datetime
|
||||
from typing import Dict, Any, Optional, Tuple
|
||||
from aiohttp import web, WSMsgType
|
||||
from aiohttp.web import Application, Request, Response, WebSocketResponse
|
||||
import aiohttp_cors
|
||||
import ssl
|
||||
from pathlib import Path
|
||||
|
||||
from .auth import AgentAuthentication, AgentAuthenticationError
|
||||
from .cluster_manager import ClusterManager
|
||||
from .storage import Storage
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class PyGuardianAPI:
|
||||
"""
|
||||
PyGuardian Controller API Server
|
||||
Provides REST API and WebSocket endpoints for agent communication
|
||||
"""
|
||||
|
||||
def __init__(self, cluster_manager: ClusterManager, config: Dict[str, Any]):
|
||||
self.cluster_manager = cluster_manager
|
||||
self.config = config
|
||||
self.app = None
|
||||
self.server = None
|
||||
self.websockets = set() # Active WebSocket connections
|
||||
|
||||
# API configuration
|
||||
self.host = config.get('api_host', '0.0.0.0')
|
||||
self.port = config.get('api_port', 8443)
|
||||
self.ssl_cert = config.get('ssl_cert')
|
||||
self.ssl_key = config.get('ssl_key')
|
||||
self.api_secret = config.get('api_secret', 'change-this-secret')
|
||||
|
||||
async def create_app(self) -> Application:
|
||||
"""Create aiohttp application with routes and middleware"""
|
||||
app = web.Application()
|
||||
|
||||
# Add CORS support
|
||||
cors = aiohttp_cors.setup(app, defaults={
|
||||
"*": aiohttp_cors.ResourceOptions(
|
||||
allow_credentials=True,
|
||||
expose_headers="*",
|
||||
allow_headers="*",
|
||||
allow_methods="*"
|
||||
)
|
||||
})
|
||||
|
||||
# Add routes
|
||||
self._setup_routes(app)
|
||||
|
||||
# Add CORS to all routes
|
||||
for route in list(app.router.routes()):
|
||||
cors.add(route)
|
||||
|
||||
# Add middleware
|
||||
app.middlewares.append(self._auth_middleware)
|
||||
app.middlewares.append(self._error_middleware)
|
||||
|
||||
self.app = app
|
||||
return app
|
||||
|
||||
def _setup_routes(self, app: Application):
|
||||
"""Setup API routes"""
|
||||
|
||||
# Health check
|
||||
app.router.add_get('/health', self.health_check)
|
||||
|
||||
# Agent authentication endpoints
|
||||
app.router.add_post('/api/v1/auth/register', self.register_agent)
|
||||
app.router.add_post('/api/v1/auth/login', self.login_agent)
|
||||
app.router.add_post('/api/v1/auth/refresh', self.refresh_token)
|
||||
app.router.add_post('/api/v1/auth/logout', self.logout_agent)
|
||||
app.router.add_post('/api/v1/auth/verify', self.verify_token)
|
||||
|
||||
# Cluster management endpoints
|
||||
app.router.add_get('/api/v1/cluster/status', self.cluster_status)
|
||||
app.router.add_get('/api/v1/cluster/agents', self.list_agents)
|
||||
app.router.add_get('/api/v1/cluster/agents/{agent_id}', self.get_agent_info)
|
||||
app.router.add_post('/api/v1/cluster/agents/{agent_id}/deploy', self.deploy_agent)
|
||||
app.router.add_delete('/api/v1/cluster/agents/{agent_id}', self.remove_agent)
|
||||
|
||||
# Agent communication endpoints
|
||||
app.router.add_post('/api/v1/agent/heartbeat', self.agent_heartbeat)
|
||||
app.router.add_post('/api/v1/agent/report', self.agent_report)
|
||||
app.router.add_get('/api/v1/agent/config', self.get_agent_config)
|
||||
app.router.add_post('/api/v1/agent/logs', self.upload_agent_logs)
|
||||
|
||||
# WebSocket endpoint for real-time communication
|
||||
app.router.add_get('/ws/agent', self.websocket_handler)
|
||||
|
||||
# Metrics endpoint for monitoring
|
||||
app.router.add_get('/metrics', self.metrics_endpoint)
|
||||
|
||||
async def _auth_middleware(self, request: Request, handler):
|
||||
"""Authentication middleware for protected endpoints"""
|
||||
# Skip auth for health check and public endpoints
|
||||
if request.path in ['/health', '/metrics'] or request.path.startswith('/api/v1/auth/'):
|
||||
return await handler(request)
|
||||
|
||||
# Check for API secret (for controller-to-controller communication)
|
||||
api_secret = request.headers.get('X-API-Secret')
|
||||
if api_secret and api_secret == self.api_secret:
|
||||
request['authenticated'] = True
|
||||
request['auth_type'] = 'api_secret'
|
||||
return await handler(request)
|
||||
|
||||
# Check for agent token
|
||||
auth_header = request.headers.get('Authorization', '')
|
||||
if not auth_header.startswith('Bearer '):
|
||||
return web.json_response(
|
||||
{'error': 'Missing or invalid authorization header'},
|
||||
status=401
|
||||
)
|
||||
|
||||
token = auth_header[7:] # Remove 'Bearer ' prefix
|
||||
|
||||
try:
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(token)
|
||||
if success:
|
||||
request['authenticated'] = True
|
||||
request['auth_type'] = 'agent_token'
|
||||
request['agent_id'] = agent_id
|
||||
return await handler(request)
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Invalid or expired token'},
|
||||
status=401
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Authentication error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Authentication failed'},
|
||||
status=401
|
||||
)
|
||||
|
||||
async def _error_middleware(self, request: Request, handler):
|
||||
"""Error handling middleware"""
|
||||
try:
|
||||
return await handler(request)
|
||||
except web.HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"API error in {request.path}: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Internal server error'},
|
||||
status=500
|
||||
)
|
||||
|
||||
# =========================
|
||||
# API Endpoint Handlers
|
||||
# =========================
|
||||
|
||||
async def health_check(self, request: Request) -> Response:
|
||||
"""Health check endpoint"""
|
||||
return web.json_response({
|
||||
'status': 'healthy',
|
||||
'timestamp': datetime.now().isoformat(),
|
||||
'version': '2.0.0',
|
||||
'cluster': self.cluster_manager.cluster_name
|
||||
})
|
||||
|
||||
async def register_agent(self, request: Request) -> Response:
|
||||
"""Register new agent endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Validate required fields
|
||||
required_fields = ['hostname', 'ip_address']
|
||||
for field in required_fields:
|
||||
if field not in data:
|
||||
return web.json_response(
|
||||
{'error': f'Missing required field: {field}'},
|
||||
status=400
|
||||
)
|
||||
|
||||
# Register agent
|
||||
success, result = await self.cluster_manager.register_new_agent(
|
||||
hostname=data['hostname'],
|
||||
ip_address=data['ip_address'],
|
||||
ssh_user=data.get('ssh_user', 'root'),
|
||||
ssh_port=data.get('ssh_port', 22),
|
||||
ssh_key_path=data.get('ssh_key_path'),
|
||||
ssh_password=data.get('ssh_password')
|
||||
)
|
||||
|
||||
if success:
|
||||
# Don't return sensitive data in logs
|
||||
safe_result = {k: v for k, v in result.items()
|
||||
if k not in ['secret_key']}
|
||||
logger.info(f"Registered new agent: {safe_result}")
|
||||
|
||||
return web.json_response(result, status=201)
|
||||
else:
|
||||
return web.json_response(result, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent registration error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Registration failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def login_agent(self, request: Request) -> Response:
|
||||
"""Agent login endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Validate required fields
|
||||
if 'agent_id' not in data or 'secret_key' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing agent_id or secret_key'},
|
||||
status=400
|
||||
)
|
||||
|
||||
client_ip = request.remote or 'unknown'
|
||||
|
||||
# Authenticate agent
|
||||
success, result = await self.cluster_manager.authenticate_agent(
|
||||
agent_id=data['agent_id'],
|
||||
secret_key=data['secret_key'],
|
||||
ip_address=client_ip
|
||||
)
|
||||
|
||||
if success:
|
||||
logger.info(f"Agent {data['agent_id']} authenticated from {client_ip}")
|
||||
return web.json_response(result)
|
||||
else:
|
||||
return web.json_response(result, status=401)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent login error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Authentication failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def refresh_token(self, request: Request) -> Response:
|
||||
"""Token refresh endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
if 'refresh_token' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing refresh_token'},
|
||||
status=400
|
||||
)
|
||||
|
||||
success, result = await self.cluster_manager.refresh_agent_token(
|
||||
data['refresh_token']
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response(result)
|
||||
else:
|
||||
return web.json_response(result, status=401)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Token refresh error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Token refresh failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def logout_agent(self, request: Request) -> Response:
|
||||
"""Agent logout endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
success = await self.cluster_manager.revoke_agent_access(agent_id)
|
||||
|
||||
if success:
|
||||
return web.json_response({'message': 'Logged out successfully'})
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Logout failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent logout error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Logout failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def verify_token(self, request: Request) -> Response:
|
||||
"""Token verification endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
if 'token' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing token'},
|
||||
status=400
|
||||
)
|
||||
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(
|
||||
data['token']
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'valid': True,
|
||||
'agent_id': agent_id
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'valid': False,
|
||||
'error': agent_id # agent_id contains error message on failure
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Token verification error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Verification failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def cluster_status(self, request: Request) -> Response:
|
||||
"""Get cluster status"""
|
||||
try:
|
||||
status = await self.cluster_manager.get_cluster_stats()
|
||||
auth_status = await self.cluster_manager.get_cluster_auth_status()
|
||||
|
||||
return web.json_response({
|
||||
'cluster_info': status,
|
||||
'authentication': auth_status
|
||||
})
|
||||
except Exception as e:
|
||||
logger.error(f"Cluster status error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to get cluster status'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def list_agents(self, request: Request) -> Response:
|
||||
"""List all agents in cluster"""
|
||||
try:
|
||||
agents = await self.cluster_manager.get_cluster_agents()
|
||||
return web.json_response({'agents': agents})
|
||||
except Exception as e:
|
||||
logger.error(f"List agents error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to list agents'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def get_agent_info(self, request: Request) -> Response:
|
||||
"""Get specific agent information"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent_info = self.cluster_manager.agents[agent_id].to_dict()
|
||||
|
||||
# Get additional info
|
||||
auth_logs = await self.cluster_manager.get_agent_auth_logs(agent_id)
|
||||
sessions = await self.cluster_manager.get_active_agent_sessions(agent_id)
|
||||
|
||||
return web.json_response({
|
||||
'agent': agent_info,
|
||||
'auth_logs': auth_logs[:10], # Last 10 logs
|
||||
'sessions': sessions
|
||||
})
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Agent not found'},
|
||||
status=404
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Get agent info error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to get agent info'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def deploy_agent(self, request: Request) -> Response:
|
||||
"""Deploy agent endpoint"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
force_reinstall = data.get('force_reinstall', False)
|
||||
|
||||
success, message = await self.cluster_manager.deploy_agent(
|
||||
agent_id, force_reinstall
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'success': True,
|
||||
'message': message
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'success': False,
|
||||
'message': message
|
||||
}, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Deploy agent error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Deployment failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def remove_agent(self, request: Request) -> Response:
|
||||
"""Remove agent endpoint"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
# Parse query parameters
|
||||
cleanup_remote = request.query.get('cleanup_remote', 'false').lower() == 'true'
|
||||
|
||||
success, message = await self.cluster_manager.remove_agent(
|
||||
agent_id, cleanup_remote
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'success': True,
|
||||
'message': message
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'success': False,
|
||||
'message': message
|
||||
}, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Remove agent error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Agent removal failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def agent_heartbeat(self, request: Request) -> Response:
|
||||
"""Agent heartbeat endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Update agent status
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent = self.cluster_manager.agents[agent_id]
|
||||
agent.last_check = datetime.now()
|
||||
agent.status = 'online'
|
||||
agent.stats.update(data.get('stats', {}))
|
||||
|
||||
# Send any pending commands
|
||||
return web.json_response({
|
||||
'status': 'ok',
|
||||
'next_heartbeat': 60, # seconds
|
||||
'commands': [] # TODO: implement command queue
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Heartbeat error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Heartbeat failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def agent_report(self, request: Request) -> Response:
|
||||
"""Agent security report endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
report = await request.json()
|
||||
|
||||
# Process security report
|
||||
logger.info(f"Received security report from agent {agent_id}")
|
||||
|
||||
# TODO: Process and store security events
|
||||
|
||||
return web.json_response({'status': 'received'})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent report error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Report processing failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def get_agent_config(self, request: Request) -> Response:
|
||||
"""Get agent configuration"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
# Return agent-specific configuration
|
||||
config = {
|
||||
'heartbeat_interval': 60,
|
||||
'report_interval': 300,
|
||||
'log_level': 'INFO',
|
||||
'features': {
|
||||
'firewall_monitoring': True,
|
||||
'intrusion_detection': True,
|
||||
'log_analysis': True
|
||||
}
|
||||
}
|
||||
|
||||
return web.json_response(config)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Get agent config error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Config retrieval failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def upload_agent_logs(self, request: Request) -> Response:
|
||||
"""Upload agent logs endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
logs = await request.json()
|
||||
|
||||
# Process and store logs
|
||||
logger.info(f"Received {len(logs.get('entries', []))} log entries from agent {agent_id}")
|
||||
|
||||
# TODO: Store logs in database or forward to log aggregator
|
||||
|
||||
return web.json_response({'status': 'received'})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Log upload error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Log upload failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def websocket_handler(self, request: Request) -> WebSocketResponse:
|
||||
"""WebSocket endpoint for real-time agent communication"""
|
||||
ws = web.WebSocketResponse()
|
||||
await ws.prepare(request)
|
||||
|
||||
agent_id = None
|
||||
|
||||
try:
|
||||
# Add to active connections
|
||||
self.websockets.add(ws)
|
||||
|
||||
async for msg in ws:
|
||||
if msg.type == WSMsgType.TEXT:
|
||||
try:
|
||||
data = json.loads(msg.data)
|
||||
|
||||
if data.get('type') == 'auth' and not agent_id:
|
||||
# Authenticate WebSocket connection
|
||||
token = data.get('token')
|
||||
if token:
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(token)
|
||||
if success:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'auth_success',
|
||||
'agent_id': agent_id
|
||||
}))
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'auth_failed',
|
||||
'error': 'Invalid token'
|
||||
}))
|
||||
elif agent_id:
|
||||
# Handle authenticated messages
|
||||
await self._handle_ws_message(ws, agent_id, data)
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': 'Not authenticated'
|
||||
}))
|
||||
|
||||
except json.JSONDecodeError:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': 'Invalid JSON'
|
||||
}))
|
||||
|
||||
elif msg.type == WSMsgType.ERROR:
|
||||
logger.error(f'WebSocket error: {ws.exception()}')
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"WebSocket error: {e}")
|
||||
finally:
|
||||
self.websockets.discard(ws)
|
||||
|
||||
return ws
|
||||
|
||||
async def _handle_ws_message(self, ws: WebSocketResponse, agent_id: str, data: Dict[str, Any]):
|
||||
"""Handle authenticated WebSocket message"""
|
||||
message_type = data.get('type')
|
||||
|
||||
if message_type == 'ping':
|
||||
await ws.send_text(json.dumps({'type': 'pong'}))
|
||||
elif message_type == 'status_update':
|
||||
# Handle agent status update
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent = self.cluster_manager.agents[agent_id]
|
||||
agent.status = data.get('status', 'unknown')
|
||||
agent.last_check = datetime.now()
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': f'Unknown message type: {message_type}'
|
||||
}))
|
||||
|
||||
async def metrics_endpoint(self, request: Request) -> Response:
|
||||
"""Prometheus metrics endpoint"""
|
||||
try:
|
||||
metrics = []
|
||||
|
||||
# Cluster metrics
|
||||
stats = await self.cluster_manager.get_cluster_stats()
|
||||
auth_status = await self.cluster_manager.get_cluster_auth_status()
|
||||
|
||||
metrics.append(f"pyguardian_cluster_total_agents {stats['total_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_online_agents {stats['online_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_offline_agents {stats['offline_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_deployed_agents {stats['deployed_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_authenticated_agents {auth_status['authenticated_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_unauthenticated_agents {auth_status['unauthenticated_agents']}")
|
||||
|
||||
# WebSocket connections
|
||||
metrics.append(f"pyguardian_websocket_connections {len(self.websockets)}")
|
||||
|
||||
return web.Response(
|
||||
text='\\n'.join(metrics),
|
||||
content_type='text/plain'
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Metrics error: {e}")
|
||||
return web.Response(
|
||||
text='# Error generating metrics',
|
||||
content_type='text/plain',
|
||||
status=500
|
||||
)
|
||||
|
||||
# =========================
|
||||
# Server Management
|
||||
# =========================
|
||||
|
||||
async def start_server(self):
|
||||
"""Start the API server"""
|
||||
try:
|
||||
app = await self.create_app()
|
||||
|
||||
# Setup SSL context if certificates are provided
|
||||
ssl_context = None
|
||||
if self.ssl_cert and self.ssl_key:
|
||||
ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
||||
ssl_context.load_cert_chain(self.ssl_cert, self.ssl_key)
|
||||
logger.info("SSL enabled for API server")
|
||||
|
||||
# Start server
|
||||
runner = web.AppRunner(app)
|
||||
await runner.setup()
|
||||
|
||||
site = web.TCPSite(runner, self.host, self.port, ssl_context=ssl_context)
|
||||
await site.start()
|
||||
|
||||
protocol = 'https' if ssl_context else 'http'
|
||||
logger.info(f"PyGuardian API server started on {protocol}://{self.host}:{self.port}")
|
||||
|
||||
self.server = runner
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to start API server: {e}")
|
||||
raise
|
||||
|
||||
async def stop_server(self):
|
||||
"""Stop the API server"""
|
||||
if self.server:
|
||||
await self.server.cleanup()
|
||||
self.server = None
|
||||
logger.info("API server stopped")
|
||||
|
||||
async def broadcast_to_agents(self, message: Dict[str, Any]):
|
||||
"""Broadcast message to all connected agents via WebSocket"""
|
||||
if not self.websockets:
|
||||
return
|
||||
|
||||
message_text = json.dumps(message)
|
||||
disconnected = set()
|
||||
|
||||
for ws in self.websockets:
|
||||
try:
|
||||
await ws.send_text(message_text)
|
||||
except Exception:
|
||||
disconnected.add(ws)
|
||||
|
||||
# Clean up disconnected WebSockets
|
||||
self.websockets -= disconnected
|
||||
@@ -1,727 +0,0 @@
|
||||
"""
|
||||
API Server for PyGuardian Controller
|
||||
REST API endpoints for agent authentication and cluster management
|
||||
"""
|
||||
|
||||
import json
|
||||
import logging
|
||||
import asyncio
|
||||
from datetime import datetime
|
||||
from typing import Dict, Any, Optional, Tuple
|
||||
from aiohttp import web, WSMsgType
|
||||
from aiohttp.web import Application, Request, Response, WebSocketResponse
|
||||
import aiohttp_cors
|
||||
import ssl
|
||||
from pathlib import Path
|
||||
|
||||
from .auth import AgentAuthentication, AgentAuthenticationError
|
||||
from .cluster_manager import ClusterManager
|
||||
from .storage import Storage
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
class PyGuardianAPI:
|
||||
"""
|
||||
PyGuardian Controller API Server
|
||||
Provides REST API and WebSocket endpoints for agent communication
|
||||
"""
|
||||
|
||||
def __init__(self, cluster_manager: ClusterManager, config: Dict[str, Any]):
|
||||
self.cluster_manager = cluster_manager
|
||||
self.config = config
|
||||
self.app = None
|
||||
self.server = None
|
||||
self.websockets = set() # Active WebSocket connections
|
||||
|
||||
# API configuration
|
||||
self.host = config.get('api_host', '0.0.0.0')
|
||||
self.port = config.get('api_port', 8443)
|
||||
self.ssl_cert = config.get('ssl_cert')
|
||||
self.ssl_key = config.get('ssl_key')
|
||||
self.api_secret = config.get('api_secret', 'change-this-secret')
|
||||
|
||||
async def create_app(self) -> Application:
|
||||
"""Create aiohttp application with routes and middleware"""
|
||||
app = web.Application()
|
||||
|
||||
# Add CORS support
|
||||
cors = aiohttp_cors.setup(app, defaults={
|
||||
"*": aiohttp_cors.ResourceOptions(
|
||||
allow_credentials=True,
|
||||
expose_headers="*",
|
||||
allow_headers="*",
|
||||
allow_methods="*"
|
||||
)
|
||||
})
|
||||
|
||||
# Add routes
|
||||
self._setup_routes(app)
|
||||
|
||||
# Add CORS to all routes
|
||||
for route in list(app.router.routes()):
|
||||
cors.add(route)
|
||||
|
||||
# Add middleware
|
||||
app.middlewares.append(self._auth_middleware)
|
||||
app.middlewares.append(self._error_middleware)
|
||||
|
||||
self.app = app
|
||||
return app
|
||||
|
||||
def _setup_routes(self, app: Application):
|
||||
"""Setup API routes"""
|
||||
|
||||
# Health check
|
||||
app.router.add_get('/health', self.health_check)
|
||||
|
||||
# Agent authentication endpoints
|
||||
app.router.add_post('/api/v1/auth/register', self.register_agent)
|
||||
app.router.add_post('/api/v1/auth/login', self.login_agent)
|
||||
app.router.add_post('/api/v1/auth/refresh', self.refresh_token)
|
||||
app.router.add_post('/api/v1/auth/logout', self.logout_agent)
|
||||
app.router.add_post('/api/v1/auth/verify', self.verify_token)
|
||||
|
||||
# Cluster management endpoints
|
||||
app.router.add_get('/api/v1/cluster/status', self.cluster_status)
|
||||
app.router.add_get('/api/v1/cluster/agents', self.list_agents)
|
||||
app.router.add_get('/api/v1/cluster/agents/{agent_id}', self.get_agent_info)
|
||||
app.router.add_post('/api/v1/cluster/agents/{agent_id}/deploy', self.deploy_agent)
|
||||
app.router.add_delete('/api/v1/cluster/agents/{agent_id}', self.remove_agent)
|
||||
|
||||
# Agent communication endpoints
|
||||
app.router.add_post('/api/v1/agent/heartbeat', self.agent_heartbeat)
|
||||
app.router.add_post('/api/v1/agent/report', self.agent_report)
|
||||
app.router.add_get('/api/v1/agent/config', self.get_agent_config)
|
||||
app.router.add_post('/api/v1/agent/logs', self.upload_agent_logs)
|
||||
|
||||
# WebSocket endpoint for real-time communication
|
||||
app.router.add_get('/ws/agent', self.websocket_handler)
|
||||
|
||||
# Metrics endpoint for monitoring
|
||||
app.router.add_get('/metrics', self.metrics_endpoint)
|
||||
|
||||
async def _auth_middleware(self, request: Request, handler):
|
||||
"""Authentication middleware for protected endpoints"""
|
||||
# Skip auth for health check and public endpoints
|
||||
if request.path in ['/health', '/metrics'] or request.path.startswith('/api/v1/auth/'):
|
||||
return await handler(request)
|
||||
|
||||
# Check for API secret (for controller-to-controller communication)
|
||||
api_secret = request.headers.get('X-API-Secret')
|
||||
if api_secret and api_secret == self.api_secret:
|
||||
request['authenticated'] = True
|
||||
request['auth_type'] = 'api_secret'
|
||||
return await handler(request)
|
||||
|
||||
# Check for agent token
|
||||
auth_header = request.headers.get('Authorization', '')
|
||||
if not auth_header.startswith('Bearer '):
|
||||
return web.json_response(
|
||||
{'error': 'Missing or invalid authorization header'},
|
||||
status=401
|
||||
)
|
||||
|
||||
token = auth_header[7:] # Remove 'Bearer ' prefix
|
||||
|
||||
try:
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(token)
|
||||
if success:
|
||||
request['authenticated'] = True
|
||||
request['auth_type'] = 'agent_token'
|
||||
request['agent_id'] = agent_id
|
||||
return await handler(request)
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Invalid or expired token'},
|
||||
status=401
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Authentication error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Authentication failed'},
|
||||
status=401
|
||||
)
|
||||
|
||||
async def _error_middleware(self, request: Request, handler):
|
||||
"""Error handling middleware"""
|
||||
try:
|
||||
return await handler(request)
|
||||
except web.HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"API error in {request.path}: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Internal server error'},
|
||||
status=500
|
||||
)
|
||||
|
||||
# =========================
|
||||
# API Endpoint Handlers
|
||||
# =========================
|
||||
|
||||
async def health_check(self, request: Request) -> Response:
|
||||
"""Health check endpoint"""
|
||||
return web.json_response({
|
||||
'status': 'healthy',
|
||||
'timestamp': datetime.now().isoformat(),
|
||||
'version': '2.0.0',
|
||||
'cluster': self.cluster_manager.cluster_name
|
||||
})
|
||||
|
||||
async def register_agent(self, request: Request) -> Response:
|
||||
"""Register new agent endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Validate required fields
|
||||
required_fields = ['hostname', 'ip_address']
|
||||
for field in required_fields:
|
||||
if field not in data:
|
||||
return web.json_response(
|
||||
{'error': f'Missing required field: {field}'},
|
||||
status=400
|
||||
)
|
||||
|
||||
# Register agent
|
||||
success, result = await self.cluster_manager.register_new_agent(
|
||||
hostname=data['hostname'],
|
||||
ip_address=data['ip_address'],
|
||||
ssh_user=data.get('ssh_user', 'root'),
|
||||
ssh_port=data.get('ssh_port', 22),
|
||||
ssh_key_path=data.get('ssh_key_path'),
|
||||
ssh_password=data.get('ssh_password')
|
||||
)
|
||||
|
||||
if success:
|
||||
# Don't return sensitive data in logs
|
||||
safe_result = {k: v for k, v in result.items()
|
||||
if k not in ['secret_key']}
|
||||
logger.info(f"Registered new agent: {safe_result}")
|
||||
|
||||
return web.json_response(result, status=201)
|
||||
else:
|
||||
return web.json_response(result, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent registration error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Registration failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def login_agent(self, request: Request) -> Response:
|
||||
"""Agent login endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Validate required fields
|
||||
if 'agent_id' not in data or 'secret_key' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing agent_id or secret_key'},
|
||||
status=400
|
||||
)
|
||||
|
||||
client_ip = request.remote or 'unknown'
|
||||
|
||||
# Authenticate agent
|
||||
success, result = await self.cluster_manager.authenticate_agent(
|
||||
agent_id=data['agent_id'],
|
||||
secret_key=data['secret_key'],
|
||||
ip_address=client_ip
|
||||
)
|
||||
|
||||
if success:
|
||||
logger.info(f"Agent {data['agent_id']} authenticated from {client_ip}")
|
||||
return web.json_response(result)
|
||||
else:
|
||||
return web.json_response(result, status=401)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent login error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Authentication failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def refresh_token(self, request: Request) -> Response:
|
||||
"""Token refresh endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
if 'refresh_token' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing refresh_token'},
|
||||
status=400
|
||||
)
|
||||
|
||||
success, result = await self.cluster_manager.refresh_agent_token(
|
||||
data['refresh_token']
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response(result)
|
||||
else:
|
||||
return web.json_response(result, status=401)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Token refresh error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Token refresh failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def logout_agent(self, request: Request) -> Response:
|
||||
"""Agent logout endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
success = await self.cluster_manager.revoke_agent_access(agent_id)
|
||||
|
||||
if success:
|
||||
return web.json_response({'message': 'Logged out successfully'})
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Logout failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent logout error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Logout failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def verify_token(self, request: Request) -> Response:
|
||||
"""Token verification endpoint"""
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
if 'token' not in data:
|
||||
return web.json_response(
|
||||
{'error': 'Missing token'},
|
||||
status=400
|
||||
)
|
||||
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(
|
||||
data['token']
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'valid': True,
|
||||
'agent_id': agent_id
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'valid': False,
|
||||
'error': agent_id # agent_id contains error message on failure
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Token verification error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Verification failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def cluster_status(self, request: Request) -> Response:
|
||||
"""Get cluster status"""
|
||||
try:
|
||||
status = await self.cluster_manager.get_cluster_stats()
|
||||
auth_status = await self.cluster_manager.get_cluster_auth_status()
|
||||
|
||||
return web.json_response({
|
||||
'cluster_info': status,
|
||||
'authentication': auth_status
|
||||
})
|
||||
except Exception as e:
|
||||
logger.error(f"Cluster status error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to get cluster status'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def list_agents(self, request: Request) -> Response:
|
||||
"""List all agents in cluster"""
|
||||
try:
|
||||
agents = await self.cluster_manager.get_cluster_agents()
|
||||
return web.json_response({'agents': agents})
|
||||
except Exception as e:
|
||||
logger.error(f"List agents error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to list agents'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def get_agent_info(self, request: Request) -> Response:
|
||||
"""Get specific agent information"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent_info = self.cluster_manager.agents[agent_id].to_dict()
|
||||
|
||||
# Get additional info
|
||||
auth_logs = await self.cluster_manager.get_agent_auth_logs(agent_id)
|
||||
sessions = await self.cluster_manager.get_active_agent_sessions(agent_id)
|
||||
|
||||
return web.json_response({
|
||||
'agent': agent_info,
|
||||
'auth_logs': auth_logs[:10], # Last 10 logs
|
||||
'sessions': sessions
|
||||
})
|
||||
else:
|
||||
return web.json_response(
|
||||
{'error': 'Agent not found'},
|
||||
status=404
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Get agent info error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Failed to get agent info'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def deploy_agent(self, request: Request) -> Response:
|
||||
"""Deploy agent endpoint"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
force_reinstall = data.get('force_reinstall', False)
|
||||
|
||||
success, message = await self.cluster_manager.deploy_agent(
|
||||
agent_id, force_reinstall
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'success': True,
|
||||
'message': message
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'success': False,
|
||||
'message': message
|
||||
}, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Deploy agent error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Deployment failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def remove_agent(self, request: Request) -> Response:
|
||||
"""Remove agent endpoint"""
|
||||
agent_id = request.match_info['agent_id']
|
||||
|
||||
try:
|
||||
# Parse query parameters
|
||||
cleanup_remote = request.query.get('cleanup_remote', 'false').lower() == 'true'
|
||||
|
||||
success, message = await self.cluster_manager.remove_agent(
|
||||
agent_id, cleanup_remote
|
||||
)
|
||||
|
||||
if success:
|
||||
return web.json_response({
|
||||
'success': True,
|
||||
'message': message
|
||||
})
|
||||
else:
|
||||
return web.json_response({
|
||||
'success': False,
|
||||
'message': message
|
||||
}, status=400)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Remove agent error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Agent removal failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def agent_heartbeat(self, request: Request) -> Response:
|
||||
"""Agent heartbeat endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
data = await request.json()
|
||||
|
||||
# Update agent status
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent = self.cluster_manager.agents[agent_id]
|
||||
agent.last_check = datetime.now()
|
||||
agent.status = 'online'
|
||||
agent.stats.update(data.get('stats', {}))
|
||||
|
||||
# Send any pending commands
|
||||
return web.json_response({
|
||||
'status': 'ok',
|
||||
'next_heartbeat': 60, # seconds
|
||||
'commands': [] # TODO: implement command queue
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Heartbeat error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Heartbeat failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def agent_report(self, request: Request) -> Response:
|
||||
"""Agent security report endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
report = await request.json()
|
||||
|
||||
# Process security report
|
||||
logger.info(f"Received security report from agent {agent_id}")
|
||||
|
||||
# TODO: Process and store security events
|
||||
|
||||
return web.json_response({'status': 'received'})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Agent report error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Report processing failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def get_agent_config(self, request: Request) -> Response:
|
||||
"""Get agent configuration"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
# Return agent-specific configuration
|
||||
config = {
|
||||
'heartbeat_interval': 60,
|
||||
'report_interval': 300,
|
||||
'log_level': 'INFO',
|
||||
'features': {
|
||||
'firewall_monitoring': True,
|
||||
'intrusion_detection': True,
|
||||
'log_analysis': True
|
||||
}
|
||||
}
|
||||
|
||||
return web.json_response(config)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Get agent config error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Config retrieval failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def upload_agent_logs(self, request: Request) -> Response:
|
||||
"""Upload agent logs endpoint"""
|
||||
agent_id = request.get('agent_id')
|
||||
if not agent_id:
|
||||
return web.json_response(
|
||||
{'error': 'Agent ID not found'},
|
||||
status=400
|
||||
)
|
||||
|
||||
try:
|
||||
logs = await request.json()
|
||||
|
||||
# Process and store logs
|
||||
logger.info(f"Received {len(logs.get('entries', []))} log entries from agent {agent_id}")
|
||||
|
||||
# TODO: Store logs in database or forward to log aggregator
|
||||
|
||||
return web.json_response({'status': 'received'})
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Log upload error: {e}")
|
||||
return web.json_response(
|
||||
{'error': 'Log upload failed'},
|
||||
status=500
|
||||
)
|
||||
|
||||
async def websocket_handler(self, request: Request) -> WebSocketResponse:
|
||||
"""WebSocket endpoint for real-time agent communication"""
|
||||
ws = web.WebSocketResponse()
|
||||
await ws.prepare(request)
|
||||
|
||||
agent_id = None
|
||||
|
||||
try:
|
||||
# Add to active connections
|
||||
self.websockets.add(ws)
|
||||
|
||||
async for msg in ws:
|
||||
if msg.type == WSMsgType.TEXT:
|
||||
try:
|
||||
data = json.loads(msg.data)
|
||||
|
||||
if data.get('type') == 'auth' and not agent_id:
|
||||
# Authenticate WebSocket connection
|
||||
token = data.get('token')
|
||||
if token:
|
||||
success, agent_id = await self.cluster_manager.verify_agent_token(token)
|
||||
if success:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'auth_success',
|
||||
'agent_id': agent_id
|
||||
}))
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'auth_failed',
|
||||
'error': 'Invalid token'
|
||||
}))
|
||||
elif agent_id:
|
||||
# Handle authenticated messages
|
||||
await self._handle_ws_message(ws, agent_id, data)
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': 'Not authenticated'
|
||||
}))
|
||||
|
||||
except json.JSONDecodeError:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': 'Invalid JSON'
|
||||
}))
|
||||
|
||||
elif msg.type == WSMsgType.ERROR:
|
||||
logger.error(f'WebSocket error: {ws.exception()}')
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"WebSocket error: {e}")
|
||||
finally:
|
||||
self.websockets.discard(ws)
|
||||
|
||||
return ws
|
||||
|
||||
async def _handle_ws_message(self, ws: WebSocketResponse, agent_id: str, data: Dict[str, Any]):
|
||||
"""Handle authenticated WebSocket message"""
|
||||
message_type = data.get('type')
|
||||
|
||||
if message_type == 'ping':
|
||||
await ws.send_text(json.dumps({'type': 'pong'}))
|
||||
elif message_type == 'status_update':
|
||||
# Handle agent status update
|
||||
if agent_id in self.cluster_manager.agents:
|
||||
agent = self.cluster_manager.agents[agent_id]
|
||||
agent.status = data.get('status', 'unknown')
|
||||
agent.last_check = datetime.now()
|
||||
else:
|
||||
await ws.send_text(json.dumps({
|
||||
'type': 'error',
|
||||
'error': f'Unknown message type: {message_type}'
|
||||
}))
|
||||
|
||||
async def metrics_endpoint(self, request: Request) -> Response:
|
||||
"""Prometheus metrics endpoint"""
|
||||
try:
|
||||
metrics = []
|
||||
|
||||
# Cluster metrics
|
||||
stats = await self.cluster_manager.get_cluster_stats()
|
||||
auth_status = await self.cluster_manager.get_cluster_auth_status()
|
||||
|
||||
metrics.append(f"pyguardian_cluster_total_agents {stats['total_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_online_agents {stats['online_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_offline_agents {stats['offline_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_deployed_agents {stats['deployed_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_authenticated_agents {auth_status['authenticated_agents']}")
|
||||
metrics.append(f"pyguardian_cluster_unauthenticated_agents {auth_status['unauthenticated_agents']}")
|
||||
|
||||
# WebSocket connections
|
||||
metrics.append(f"pyguardian_websocket_connections {len(self.websockets)}")
|
||||
|
||||
return web.Response(
|
||||
text='\\n'.join(metrics),
|
||||
content_type='text/plain'
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Metrics error: {e}")
|
||||
return web.Response(
|
||||
text='# Error generating metrics',
|
||||
content_type='text/plain',
|
||||
status=500
|
||||
)
|
||||
|
||||
# =========================
|
||||
# Server Management
|
||||
# =========================
|
||||
|
||||
async def start_server(self):
|
||||
"""Start the API server"""
|
||||
try:
|
||||
app = await self.create_app()
|
||||
|
||||
# Setup SSL context if certificates are provided
|
||||
ssl_context = None
|
||||
if self.ssl_cert and self.ssl_key:
|
||||
ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
|
||||
ssl_context.load_cert_chain(self.ssl_cert, self.ssl_key)
|
||||
logger.info("SSL enabled for API server")
|
||||
|
||||
# Start server
|
||||
runner = web.AppRunner(app)
|
||||
await runner.setup()
|
||||
|
||||
site = web.TCPSite(runner, self.host, self.port, ssl_context=ssl_context)
|
||||
await site.start()
|
||||
|
||||
protocol = 'https' if ssl_context else 'http'
|
||||
logger.info(f"PyGuardian API server started on {protocol}://{self.host}:{self.port}")
|
||||
|
||||
self.server = runner
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to start API server: {e}")
|
||||
raise
|
||||
|
||||
async def stop_server(self):
|
||||
"""Stop the API server"""
|
||||
if self.server:
|
||||
await self.server.cleanup()
|
||||
self.server = None
|
||||
logger.info("API server stopped")
|
||||
|
||||
async def broadcast_to_agents(self, message: Dict[str, Any]):
|
||||
"""Broadcast message to all connected agents via WebSocket"""
|
||||
if not self.websockets:
|
||||
return
|
||||
|
||||
message_text = json.dumps(message)
|
||||
disconnected = set()
|
||||
|
||||
for ws in self.websockets:
|
||||
try:
|
||||
await ws.send_text(message_text)
|
||||
except Exception:
|
||||
disconnected.add(ws)
|
||||
|
||||
# Clean up disconnected WebSockets
|
||||
self.websockets -= disconnected
|
||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user