Upload 5 files

src/utils/dns_lookup.py

@@ -0,0 +1,13 @@
+import socket
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
+def dns_lookup(domain):
+    try:
+        ip = socket.gethostbyname(domain)
+        logging.info(f"[DNS] {domain} resolved to {ip}")
+        return {"domain": domain, "ip_address": ip}
+    except socket.gaierror:
+        logging.warning(f"[DNS] Failed to resolve {domain}")
+        return {"domain": domain, "ip_address": None}

src/utils/email_alert.py

@@ -0,0 +1,18 @@
+import smtplib
+from email.mime.text import MIMEText
+import logging
+
+def send_email_alert(subject, body, recipient_email, sender_email, smtp_server, smtp_port, smtp_username, smtp_password):
+    try:
+        msg = MIMEText(body)
+        msg["Subject"] = subject
+        msg["From"] = sender_email
+        msg["To"] = recipient_email
+
+        with smtplib.SMTP_SSL(smtp_server, smtp_port) as server:
+            server.login(smtp_username, smtp_password)
+            server.sendmail(sender_email, recipient_email, msg.as_string())
+
+        logging.info(f"Email alert sent to {recipient_email}")
+    except Exception as e:
+        logging.error(f"Failed to send email alert: {e}")

src/utils/violation_logger.py

@@ -0,0 +1,80 @@
+import json
+import uuid
+import logging
+from datetime import datetime
+from pathlib import Path
+
+# Optional integrations
+try:
+    from src.protocol.security.alert_webhook import WebhookAlerter
+    from src.protocol.permanent_memory import PermanentMemory
+    memory = PermanentMemory()
+except ImportError:
+    WebhookAlerter = None
+    memory = None
+
+# === CONFIG ===
+VIOLATIONS_JSON = Path("violations.json")
+VIOLATIONS_STREAM = Path("violations.jsonl")  # JSON Lines format
+ENABLE_DISCORD_ALERTS = True
+WEBHOOK_URL = "https://discord.com/api/webhooks/xxx/yyy"  # Replace with your real one
+
+# === LOGGER SETUP ===
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+
+def log_violation(
+    violation_type,
+    description,
+    source_url=None,
+    severity="medium",
+    detected_by="AutoScanner",
+    context=None
+):
+    """
+    Logs a violation to both JSON and streaming JSONL files.
+    Optionally triggers alerts and writes to permanent memory.
+    """
+    entry = {
+        "id": str(uuid.uuid4()),
+        "timestamp": datetime.utcnow().isoformat() + "Z",
+        "violation_type": violation_type,
+        "description": description,
+        "source_url": source_url,
+        "severity": severity,
+        "detected_by": detected_by,
+        "context": context or {}
+    }
+
+    # === Write to violations.json ===
+    try:
+        if VIOLATIONS_JSON.exists():
+            with open(VIOLATIONS_JSON, "r") as f:
+                data = json.load(f)
+        else:
+            data = {"log_created": datetime.utcnow().isoformat() + "Z", "entries": []}
+    except Exception:
+        data = {"log_created": datetime.utcnow().isoformat() + "Z", "entries": []}
+
+    data["entries"].append(entry)
+    with open(VIOLATIONS_JSON, "w") as f:
+        json.dump(data, f, indent=2)
+
+    # === Write to violations.jsonl ===
+    with open(VIOLATIONS_STREAM, "a") as f:
+        f.write(json.dumps(entry) + "\n")
+
+    logging.warning(f"[{severity.upper()}] Violation logged: {violation_type} – {description}")
+
+    # === Alert Webhook ===
+    if ENABLE_DISCORD_ALERTS and WebhookAlerter:
+        try:
+            alerter = WebhookAlerter(WEBHOOK_URL)
+            alerter.send_alert(f"🚨 Violation Detected: `{violation_type}`\n> {description}")
+        except Exception as e:
+            logging.error(f"Failed to send webhook alert: {e}")
+
+    # === Permanent Memory Sync ===
+    if memory:
+        memory.write("violation_log", entry)
+
+    return entry

src/utils/webhook_alert.py

@@ -0,0 +1,12 @@
+import requests
+import logging
+
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+
+def send_webhook_alert(webhook_url: str, payload: dict):
+    try:
+        response = requests.post(webhook_url, json=payload)
+        response.raise_for_status()
+        logging.info(f"Webhook alert sent. Status: {response.status_code}")
+    except Exception as e:
+        logging.error(f"Webhook alert failed: {e}")

src/utils/whois_lookup.py

@@ -0,0 +1,25 @@
+import whois
+import logging
+
+logging.basicConfig(level=logging.INFO)
+
+def whois_lookup(domain):
+    try:
+        data = whois.whois(domain)
+        logging.info(f"[WHOIS] Data fetched for {domain}")
+        return {
+            "domain": domain,
+            "registrar": data.registrar,
+            "creation_date": str(data.creation_date),
+            "expiration_date": str(data.expiration_date),
+            "name_servers": data.name_servers
+        }
+    except Exception as e:
+        logging.error(f"[WHOIS] Failed for {domain}: {e}")
+        return {
+            "domain": domain,
+            "registrar": None,
+            "creation_date": None,
+            "expiration_date": None,
+            "name_servers": None
+        }