360 lines
12 KiB
Python
360 lines
12 KiB
Python
"""
|
||
SkillSpector Web – Wrapper um die skillspector-CLI, geeignet für öffentlichen Betrieb.
|
||
|
||
Sicherheitsprinzipien:
|
||
- Führt NIEMALS den gescannten Skill aus (das macht skillspector selbst auch nicht).
|
||
- Nur http/https-URLs werden akzeptiert; Ziele, die auf private/interne Adressen
|
||
auflösen (SSRF-Schutz: localhost, 127.x, 10.x, 192.168.x, 172.16-31.x, link-local,
|
||
IPv6-Loopback/ULA), werden abgelehnt.
|
||
- Läuft standardmäßig mit --no-llm; LLM-Analyse ist nur nutzbar, wenn der Admin sie
|
||
explizit über ENABLE_LLM_OPTION freischaltet UND ein API-Key konfiguriert ist.
|
||
- Zugriff optional per Basic-Auth (REQUIRE_AUTH); bei öffentlichem Betrieb ohne Auth
|
||
greifen automatisch Rate-Limit und Concurrency-Limit.
|
||
- Jeder Scan läuft in einem eigenen, danach gelöschten Temp-Verzeichnis.
|
||
- Harte Zeit- und Upload-Größenbegrenzung.
|
||
"""
|
||
|
||
import ipaddress
|
||
import json
|
||
import os
|
||
import re
|
||
import shutil
|
||
import socket
|
||
import subprocess
|
||
import tempfile
|
||
import threading
|
||
import time
|
||
from functools import wraps
|
||
from pathlib import Path
|
||
from urllib.parse import urlparse
|
||
|
||
from flask import Flask, Response, render_template_string, request
|
||
|
||
app = Flask(__name__)
|
||
app.config["MAX_CONTENT_LENGTH"] = 50 * 1024 * 1024 # 50 MB Upload-Limit
|
||
|
||
SCAN_TMP_DIR = os.environ.get("SCAN_TMP_DIR", "/tmp/skillspector")
|
||
Path(SCAN_TMP_DIR).mkdir(parents=True, exist_ok=True)
|
||
|
||
REQUIRE_AUTH = os.environ.get("REQUIRE_AUTH", "true").lower() == "true"
|
||
WEBUI_USER = os.environ.get("WEBUI_USER", "admin")
|
||
WEBUI_PASSWORD = os.environ.get("WEBUI_PASSWORD")
|
||
|
||
SCAN_TIMEOUT = int(os.environ.get("SCAN_TIMEOUT_SECONDS", "300"))
|
||
MAX_CONCURRENT_SCANS = int(os.environ.get("MAX_CONCURRENT_SCANS", "2"))
|
||
MAX_SCANS_PER_HOUR_PER_IP = int(os.environ.get("MAX_SCANS_PER_HOUR_PER_IP", "10"))
|
||
|
||
LLM_KEY_PRESENT = bool(os.environ.get("OPENAI_API_KEY") or os.environ.get("ANTHROPIC_API_KEY"))
|
||
ENABLE_LLM_OPTION = os.environ.get("ENABLE_LLM_OPTION", "false").lower() == "true"
|
||
ALLOW_LLM = LLM_KEY_PRESENT and ENABLE_LLM_OPTION
|
||
|
||
GIT_URL_RE = re.compile(r"^https?://[^\s]+$")
|
||
|
||
_scan_semaphore = threading.Semaphore(MAX_CONCURRENT_SCANS)
|
||
_rate_lock = threading.Lock()
|
||
_rate_log: dict[str, list[float]] = {}
|
||
|
||
|
||
# --------------------------------------------------------------------------- #
|
||
# Auth
|
||
# --------------------------------------------------------------------------- #
|
||
def check_auth(username: str, password: str) -> bool:
|
||
return bool(WEBUI_PASSWORD) and username == WEBUI_USER and password == WEBUI_PASSWORD
|
||
|
||
|
||
def authenticate():
|
||
return Response(
|
||
"Zugriff verweigert.",
|
||
401,
|
||
{"WWW-Authenticate": 'Basic realm="SkillSpector"'},
|
||
)
|
||
|
||
|
||
def requires_auth(f):
|
||
@wraps(f)
|
||
def decorated(*args, **kwargs):
|
||
if not REQUIRE_AUTH:
|
||
return f(*args, **kwargs)
|
||
if not WEBUI_PASSWORD:
|
||
return Response(
|
||
"REQUIRE_AUTH=true, aber WEBUI_PASSWORD ist nicht gesetzt – "
|
||
"Zugriff aus Sicherheitsgründen gesperrt.",
|
||
500,
|
||
)
|
||
auth = request.authorization
|
||
if not auth or not check_auth(auth.username, auth.password):
|
||
return authenticate()
|
||
return f(*args, **kwargs)
|
||
|
||
return decorated
|
||
|
||
|
||
# --------------------------------------------------------------------------- #
|
||
# Rate-Limiting (pro Client-IP, in-memory – reicht für einen einzelnen Prozess)
|
||
# --------------------------------------------------------------------------- #
|
||
def client_ip() -> str:
|
||
fwd = request.headers.get("X-Forwarded-For")
|
||
return fwd.split(",")[0].strip() if fwd else request.remote_addr or "unknown"
|
||
|
||
|
||
def rate_limited(ip: str) -> bool:
|
||
now = time.time()
|
||
with _rate_lock:
|
||
entries = [t for t in _rate_log.get(ip, []) if now - t < 3600]
|
||
if len(entries) >= MAX_SCANS_PER_HOUR_PER_IP:
|
||
_rate_log[ip] = entries
|
||
return True
|
||
entries.append(now)
|
||
_rate_log[ip] = entries
|
||
return False
|
||
|
||
|
||
# --------------------------------------------------------------------------- #
|
||
# SSRF-Schutz: nur öffentliche http(s)-Ziele erlauben
|
||
# --------------------------------------------------------------------------- #
|
||
def is_public_target(url: str) -> bool:
|
||
if not GIT_URL_RE.match(url):
|
||
return False
|
||
host = urlparse(url).hostname
|
||
if not host:
|
||
return False
|
||
if host.lower() == "localhost":
|
||
return False
|
||
try:
|
||
infos = socket.getaddrinfo(host, None)
|
||
except socket.gaierror:
|
||
return False
|
||
for info in infos:
|
||
ip = ipaddress.ip_address(info[4][0])
|
||
if (
|
||
ip.is_private
|
||
or ip.is_loopback
|
||
or ip.is_link_local
|
||
or ip.is_reserved
|
||
or ip.is_multicast
|
||
or ip.is_unspecified
|
||
):
|
||
return False
|
||
return True
|
||
|
||
|
||
# --------------------------------------------------------------------------- #
|
||
# Design im Stil von software.obraasch.de: dunkler Header, orange Akzente,
|
||
# helle Cards.
|
||
# --------------------------------------------------------------------------- #
|
||
PAGE = """
|
||
<!doctype html>
|
||
<html lang="de">
|
||
<head>
|
||
<meta charset="utf-8">
|
||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||
<title>SkillSpector</title>
|
||
<style>
|
||
:root{
|
||
--orange:#f2731e;
|
||
--orange-dark:#d9601a;
|
||
--bg:#eef1f4;
|
||
--card:#ffffff;
|
||
--ink:#1a1d21;
|
||
--ink-soft:#5b6470;
|
||
--border:#e2e6ea;
|
||
}
|
||
*{box-sizing:border-box}
|
||
body{
|
||
margin:0;font-family:-apple-system,Segoe UI,Roboto,Helvetica,Arial,sans-serif;
|
||
background:var(--bg);color:var(--ink);
|
||
}
|
||
header{
|
||
background:#0d0e10;padding:28px 24px;
|
||
}
|
||
.header-inner{max-width:760px;margin:0 auto}
|
||
.logo{font-size:1.9em;font-weight:800;letter-spacing:.5px;margin:0}
|
||
.logo .accent{color:var(--orange)}
|
||
.subtitle{color:#9aa2ab;margin:6px 0 0;font-size:.95em}
|
||
main{max-width:760px;margin:32px auto;padding:0 24px 48px}
|
||
.card{
|
||
background:var(--card);border:1px solid var(--border);border-radius:14px;
|
||
padding:28px;margin-bottom:24px;box-shadow:0 1px 2px rgba(0,0,0,.03);
|
||
}
|
||
label{display:block;margin-top:16px;font-size:.85em;color:var(--ink-soft);font-weight:600}
|
||
input[type=text],input[type=file]{
|
||
width:100%;padding:11px 12px;margin-top:6px;border:1px solid var(--border);
|
||
border-radius:9px;font-size:.95em;background:#fbfcfd;
|
||
}
|
||
input[type=text]:focus{outline:2px solid var(--orange);border-color:var(--orange)}
|
||
.checkbox-row{display:flex;align-items:flex-start;gap:8px;margin-top:16px}
|
||
.checkbox-row input{width:auto;margin-top:2px}
|
||
.checkbox-row label{margin:0;font-weight:400;color:var(--ink-soft)}
|
||
button{
|
||
margin-top:20px;padding:12px 20px;border:none;border-radius:9px;
|
||
background:var(--orange);color:#fff;font-weight:700;font-size:.95em;
|
||
cursor:pointer;transition:background .15s;
|
||
}
|
||
button:hover{background:var(--orange-dark)}
|
||
.badge{
|
||
display:inline-block;padding:4px 12px;border-radius:999px;font-size:.8em;
|
||
font-weight:700;border:1px solid transparent;
|
||
}
|
||
.badge.SAFE{background:#e8f7ee;color:#1e8e4a;border-color:#c6ebd3}
|
||
.badge.CAUTION{background:#fff4e0;color:#b5720a;border-color:#ffe3ad}
|
||
.badge.DO_NOT_INSTALL{background:#fdecec;color:#c62828;border-color:#f9cccc}
|
||
.score-row{display:flex;align-items:center;gap:12px;margin-bottom:16px}
|
||
.score-num{font-size:1.9em;font-weight:800}
|
||
pre{
|
||
background:#14161a;color:#d7dbe0;padding:16px;border-radius:10px;
|
||
overflow-x:auto;white-space:pre-wrap;font-size:.82em;line-height:1.5;
|
||
}
|
||
.err{
|
||
background:#fdecec;color:#c62828;border:1px solid #f9cccc;border-radius:10px;
|
||
padding:14px 16px;font-weight:600;
|
||
}
|
||
.notice{color:var(--ink-soft);font-size:.85em;margin-top:8px}
|
||
footer{
|
||
max-width:760px;margin:0 auto;padding:0 24px 40px;color:#9aa2ab;font-size:.85em;
|
||
}
|
||
a{color:var(--orange)}
|
||
</style>
|
||
</head>
|
||
<body>
|
||
<header>
|
||
<div class="header-inner">
|
||
<p class="logo">SKILL<span class="accent">SPECTOR</span></p>
|
||
<p class="subtitle">Sicherheits-Scanner für AI-Agent-Skills – vor dem Installieren prüfen</p>
|
||
</div>
|
||
</header>
|
||
<main>
|
||
<div class="card">
|
||
<form method="post" enctype="multipart/form-data">
|
||
<label>Git-Repository-URL</label>
|
||
<input type="text" name="git_url" placeholder="https://git.obraasch.de/olli1986/mein-skill">
|
||
<label>...oder ZIP-Datei hochladen (max. 50 MB)</label>
|
||
<input type="file" name="zip_file" accept=".zip">
|
||
{% if allow_llm %}
|
||
<div class="checkbox-row">
|
||
<input type="checkbox" name="use_llm" id="use_llm">
|
||
<label for="use_llm">LLM-Analyse aktivieren (sendet Dateiinhalte an den konfigurierten KI-Anbieter)</label>
|
||
</div>
|
||
{% endif %}
|
||
<button type="submit">Scan starten</button>
|
||
</form>
|
||
</div>
|
||
|
||
{% if result %}
|
||
<div class="card">
|
||
<div class="score-row">
|
||
<span class="score-num">{{ result.risk_assessment.score }}/100</span>
|
||
<span class="badge {{ result.risk_assessment.recommendation }}">{{ result.risk_assessment.recommendation }}</span>
|
||
</div>
|
||
<pre>{{ result_json }}</pre>
|
||
</div>
|
||
{% elif error %}
|
||
<div class="card">
|
||
<p class="err">{{ error }}</p>
|
||
</div>
|
||
{% endif %}
|
||
</main>
|
||
<footer>
|
||
Läuft auf <a href="https://github.com/NVIDIA/SkillSpector" target="_blank">NVIDIA SkillSpector</a>
|
||
(Apache-2.0). Statische Analyse läuft immer lokal
|
||
{% if allow_llm %}– bei aktivierter LLM-Analyse werden Dateiinhalte extern übertragen{% endif %}.
|
||
Der gescannte Skill wird nie ausgeführt, kein Ergebnis wird dauerhaft gespeichert.
|
||
</footer>
|
||
</body>
|
||
</html>
|
||
"""
|
||
|
||
|
||
@app.route("/", methods=["GET", "POST"])
|
||
@requires_auth
|
||
def index():
|
||
if request.method == "GET":
|
||
return render_template_string(PAGE, allow_llm=ALLOW_LLM, result=None, error=None)
|
||
|
||
ip = client_ip()
|
||
if rate_limited(ip):
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error=f"Limit erreicht: maximal {MAX_SCANS_PER_HOUR_PER_IP} Scans pro Stunde und IP. Bitte später erneut versuchen.",
|
||
)
|
||
|
||
use_llm = ALLOW_LLM and request.form.get("use_llm") == "on"
|
||
git_url = (request.form.get("git_url") or "").strip()
|
||
zip_file = request.files.get("zip_file")
|
||
|
||
if not _scan_semaphore.acquire(blocking=False):
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error="Gerade laufen bereits zu viele Scans gleichzeitig. Bitte kurz warten und erneut versuchen.",
|
||
)
|
||
|
||
work_dir = tempfile.mkdtemp(dir=SCAN_TMP_DIR)
|
||
try:
|
||
if zip_file and zip_file.filename:
|
||
zip_path = Path(work_dir) / "upload.zip"
|
||
zip_file.save(zip_path)
|
||
target = str(zip_path)
|
||
elif git_url:
|
||
if not is_public_target(git_url):
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error="Diese URL ist nicht erlaubt (nur öffentlich erreichbare http/https-Ziele).",
|
||
)
|
||
target = git_url
|
||
else:
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error="Bitte eine Git-URL angeben oder eine ZIP-Datei hochladen.",
|
||
)
|
||
|
||
report_path = Path(work_dir) / "report.json"
|
||
cmd = ["skillspector", "scan", target, "--format", "json", "-o", str(report_path)]
|
||
if not use_llm:
|
||
cmd.append("--no-llm")
|
||
|
||
proc = subprocess.run(
|
||
cmd,
|
||
cwd=work_dir,
|
||
timeout=SCAN_TIMEOUT,
|
||
capture_output=True,
|
||
text=True,
|
||
)
|
||
|
||
if report_path.exists():
|
||
result = json.loads(report_path.read_text())
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=result,
|
||
result_json=json.dumps(result, indent=2, ensure_ascii=False),
|
||
error=None,
|
||
)
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error=f"Scan fehlgeschlagen (Exit {proc.returncode}):\n{proc.stderr[-2000:]}",
|
||
)
|
||
except subprocess.TimeoutExpired:
|
||
return render_template_string(
|
||
PAGE,
|
||
allow_llm=ALLOW_LLM,
|
||
result=None,
|
||
error=f"Scan abgebrochen: Zeitlimit von {SCAN_TIMEOUT}s überschritten.",
|
||
)
|
||
finally:
|
||
shutil.rmtree(work_dir, ignore_errors=True)
|
||
_scan_semaphore.release()
|
||
|
||
|
||
@app.route("/healthz")
|
||
def healthz():
|
||
return {"status": "ok"}
|