mirror of
https://github.com/jhd3197/ServerKit.git
synced 2026-11-10 05:53:05 +09:00
137 lines
4.9 KiB
Python
137 lines
4.9 KiB
Python
"""Proving tests for the panel's security headers.
|
|
|
|
HTTPS is optional in ServerKit, so the panel must NOT advertise HSTS/preload
|
|
(a hard-to-reverse browser commitment) unless the deployment actually
|
|
terminates HTTPS. The installer records that via SSL_MODE/HSTS_ENABLED; behind
|
|
a proxy Flask can't tell real TLS from a Cloudflare-Flexible edge, so we trust
|
|
the recorded choice rather than the request scheme.
|
|
"""
|
|
|
|
from pathlib import Path
|
|
|
|
import config as cfg
|
|
import pytest
|
|
|
|
|
|
PANEL_NGINX_CONFIGS = [
|
|
Path(__file__).resolve().parents[2] / 'nginx' / 'nginx.conf',
|
|
Path(__file__).resolve().parents[2]
|
|
/ 'nginx'
|
|
/ 'sites-available'
|
|
/ 'serverkit.conf',
|
|
]
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
ALL_CSP_CONFIGS = PANEL_NGINX_CONFIGS + [
|
|
ROOT / 'nginx' / 'sites-available' / 'example.conf.template',
|
|
]
|
|
|
|
|
|
def _script_src(csp):
|
|
return next(d.strip() for d in csp.split(';') if d.strip().startswith('script-src '))
|
|
|
|
|
|
def _make_app(hsts_enabled, debug):
|
|
from flask import Flask
|
|
from app.middleware.security import register_security_headers
|
|
app = Flask(__name__)
|
|
app.debug = debug
|
|
app.config['HSTS_ENABLED'] = hsts_enabled
|
|
register_security_headers(app)
|
|
|
|
@app.route('/ping')
|
|
def ping():
|
|
return 'ok'
|
|
|
|
return app
|
|
|
|
|
|
def test_hsts_sent_in_secure_production():
|
|
r = _make_app(hsts_enabled=True, debug=False).test_client().get('/ping')
|
|
hsts = r.headers.get('Strict-Transport-Security')
|
|
assert hsts and 'preload' in hsts and 'includeSubDomains' in hsts
|
|
|
|
|
|
def test_hsts_absent_in_insecure_mode():
|
|
r = _make_app(hsts_enabled=False, debug=False).test_client().get('/ping')
|
|
assert 'Strict-Transport-Security' not in r.headers
|
|
# The non-committal hardening headers are still applied.
|
|
assert r.headers.get('X-Content-Type-Options') == 'nosniff'
|
|
assert 'Content-Security-Policy' in r.headers
|
|
|
|
|
|
def test_hsts_absent_in_debug_even_if_secure():
|
|
r = _make_app(hsts_enabled=True, debug=True).test_client().get('/ping')
|
|
assert 'Strict-Transport-Security' not in r.headers
|
|
|
|
|
|
@pytest.mark.parametrize('debug', [False, True])
|
|
def test_csp_allows_only_github_as_an_external_form_target(debug):
|
|
r = _make_app(hsts_enabled=False, debug=debug).test_client().get('/ping')
|
|
csp = r.headers['Content-Security-Policy']
|
|
form_action = next(
|
|
directive.strip()
|
|
for directive in csp.split(';')
|
|
if directive.strip().startswith('form-action ')
|
|
)
|
|
|
|
assert form_action == "form-action 'self' https://github.com"
|
|
|
|
|
|
@pytest.mark.parametrize('config_path', PANEL_NGINX_CONFIGS)
|
|
def test_panel_nginx_csp_allows_github_manifest_submission(config_path):
|
|
config = config_path.read_text(encoding='utf-8')
|
|
|
|
assert "form-action 'self' https://github.com;" in config
|
|
assert "form-action 'self';" not in config
|
|
|
|
|
|
def test_resolve_ssl_mode(monkeypatch):
|
|
monkeypatch.setenv('SERVERKIT_SSL_MODE', 'secure')
|
|
assert cfg._resolve_ssl_mode() == 'secure'
|
|
# Unrecognised value falls through; no /etc/serverkit/ssl-mode on the test
|
|
# box, so it lands on the safe default.
|
|
monkeypatch.setenv('SERVERKIT_SSL_MODE', 'bogus')
|
|
assert cfg._resolve_ssl_mode() == 'insecure'
|
|
monkeypatch.delenv('SERVERKIT_SSL_MODE', raising=False)
|
|
assert cfg._resolve_ssl_mode() == 'insecure'
|
|
|
|
|
|
def test_prod_csp_script_src_is_strict():
|
|
r = _make_app(hsts_enabled=False, debug=False).test_client().get('/ping')
|
|
script_src = _script_src(r.headers['Content-Security-Policy'])
|
|
assert 'unsafe-eval' not in script_src
|
|
assert 'unsafe-inline' not in script_src
|
|
|
|
|
|
@pytest.mark.parametrize('config_path', ALL_CSP_CONFIGS)
|
|
def test_nginx_csp_script_src_is_strict(config_path):
|
|
line = next(
|
|
l for l in config_path.read_text(encoding='utf-8').splitlines()
|
|
if 'Content-Security-Policy' in l
|
|
)
|
|
script_src = _script_src(line.split('"')[1])
|
|
assert 'unsafe-eval' not in script_src
|
|
assert 'unsafe-inline' not in script_src
|
|
|
|
|
|
def test_index_html_has_no_inline_script():
|
|
html = (ROOT / 'frontend' / 'index.html').read_text(encoding='utf-8')
|
|
assert '<script>' not in html
|
|
|
|
|
|
def test_csp_importmap_hash_matches_vendor_manifest():
|
|
import base64, hashlib, json, re
|
|
from app.middleware.security import VENDOR_IMPORTMAP_HASH
|
|
src = (ROOT / 'frontend/src/plugins/runtime/vendorManifest.js').read_text(encoding='utf-8')
|
|
pairs = re.findall(r"specifier: '([^']+)', file: '([^']+)'", src)
|
|
body = json.dumps({'imports': {s: '/serverkit-vendor/' + f for s, f in pairs}}, separators=(',', ':'))
|
|
digest = base64.b64encode(hashlib.sha256(body.encode()).digest()).decode()
|
|
assert VENDOR_IMPORTMAP_HASH == 'sha256-' + digest
|
|
for path in ALL_CSP_CONFIGS:
|
|
assert VENDOR_IMPORTMAP_HASH in path.read_text(encoding='utf-8')
|
|
|
|
|
|
def test_managed_app_nginx_template_has_no_unsafe_eval():
|
|
from app.services.nginx import templates
|
|
assert 'unsafe-eval' not in Path(templates.__file__).read_text(encoding='utf-8')
|