Files
aethas-free/modules/security.py
T

62 lines
2.3 KiB
Python

import json
import os
import time
import requests
from datetime import datetime
from modules.logger import log_event
LOG_FILE = "aethas_ip_logs.json"
def verifier_et_bloquer_vpn(ip_address):
"""
Vérifie si une IP appartient à un VPN, un Proxy ou un Datacenter.
Utilise l'API publique gratuite blackbox.ipinfo.app (Renvoie 'Y' si VPN, 'N' si clean).
"""
# On ne bloque jamais les requêtes locales du serveur (localhost)
if ip_address in ["127.0.0.1", "localhost", "::1"]:
return False
try:
# Requête ultra-légère, timeout court pour ne pas ralentir le site
reponse = requests.get(f"https://blackbox.ipinfo.app/lookup/{ip_address}", timeout=3)
if reponse.text.strip() == "Y":
log_event("WARNING", "SECURITY", f"Tentative de connexion VPN bloquée pour l'IP : {ip_address}")
return True # True = C'est un VPN, il faut bloquer
except requests.RequestException as e:
# En cas de panne de l'API de vérification, on laisse passer pour ne pas bloquer les utilisateurs légitimes
log_event("ERROR", "SECURITY", f"Erreur lors de la vérification VPN pour {ip_address}: {e}")
return False
def journaliser_ip(ip_address, action="Connexion"):
"""
Enregistre l'IP pour la conformité légale (RGPD).
Purge automatique des logs vieux de plus de 365 jours.
"""
logs = []
# 1. Chargement de l'historique existant
if os.path.exists(LOG_FILE):
with open(LOG_FILE, 'r', encoding='utf-8') as f:
try:
logs = json.load(f)
except json.JSONDecodeError:
logs = []
# 2. Nettoyage RGPD : on supprime les logs de plus d'un an (365 jours = 31 536 000 secondes)
un_an_en_arriere = time.time() - 31536000
logs = [log for log in logs if log.get("timestamp", 0) > un_an_en_arriere]
# 3. Ajout de la nouvelle entrée
logs.append({
"ip": ip_address,
"action": action,
"date": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
"timestamp": time.time()
})
# 4. Sauvegarde
with open(LOG_FILE, 'w', encoding='utf-8') as f:
json.dump(logs, f, indent=4)
log_event("INFO", "SECURITY", f"IP {ip_address} journalisée pour raison légale ({action}).")