Security Headers Analyzer — Kotlin source

// HTTP security-header analysis.
//
// Language: Kotlin 1.9+ (JVM), standard library only.
// Ported from src/lib/security-headers.ts — display source, part of CosmoDev's
// polyglot tool pages. Functionally equivalent to the TS reference: same
// parsing (obs-fold continuation, duplicate joining), same per-header A/B/C/F
// graders and messages, same weighted overall score. Never throws.
//
// Parses a raw response-header block (the "key: value" lines copied from
// DevTools' Network panel), grades each of 8 security headers A/B/C/F, and
// rolls a weighted overall score. Header names match case-insensitively,
// folded (indented) continuation lines join their parent value, and duplicate
// names join with ", ". Malformed lines and the HTTP status line are ignored.

/** Per-header grade: A best practice, B present but weak, C missing or dangerous, F actively harmful. */
enum class HeaderGrade { A, B, C, F }

/** One security header as analyzed: the parsed value plus its grade and guidance. */
data class ParsedHeader(
    val name: String, // canonical name, e.g. 'Strict-Transport-Security'
    val value: String?, // raw value as parsed; null when the header is absent
    val grade: HeaderGrade,
    val explanation: String,
    val recommendation: String,
)

/** The full analysis result: per-header rows, the overall grade, and fix list. */
data class HeaderAnalysis(
    val headers: List<ParsedHeader>,
    val grade: HeaderGrade,
    val score: Int, // 0-100 weighted average of per-header grades
    val recommendations: List<String>,
)

/** Catalog entry: canonical name, why the header matters, and its score weight. */
data class HeaderSpec(
    val name: String,
    val description: String,
    val weight: Double,
)

/** The 8 security headers this analyzer grades, in display order. */
val SECURITY_HEADERS = listOf(
    HeaderSpec(
        "Strict-Transport-Security",
        "HSTS tells the browser 'only ever reach this site over HTTPS' for max-age seconds. It defeats SSL-strip attacks, which downgrade the very first plain-HTTP visit. includeSubDomains extends the guarantee to every subdomain; preload lets the site join the browser-built HSTS preload list so even the first visit is HTTPS.",
        1.5,
    ),
    HeaderSpec(
        "Content-Security-Policy",
        "CSP is the strongest anti-XSS control available in a header: it declares which sources scripts, styles, and other resources may load from. 'unsafe-inline' and 'unsafe-eval' punch holes straight through it - they re-allow inline injection and string-to-code evaluation. A default-src directive is the baseline so resource types you forgot to list inherit a restrictive fallback instead of the browser's permissive one.",
        2.0,
    ),
    HeaderSpec(
        "X-Frame-Options",
        "Stops other sites from embedding this page in an iframe, which is the basis of clickjacking: an attacker overlays invisible UI on your framed page and harvests the victim's clicks. DENY blocks all framing; SAMEORIGIN allows only your own origin. The modern replacement is the CSP frame-ancestors directive.",
        1.0,
    ),
    HeaderSpec(
        "X-Content-Type-Options",
        "With the value nosniff, the browser must respect the declared Content-Type instead of 'helpfully' sniffing the payload. Sniffing turns an uploaded text file into executable JavaScript when its bytes look script-shaped - the classic content-type-confusion attack.",
        1.0,
    ),
    HeaderSpec(
        "Referrer-Policy",
        "Controls how much of the URL the browser leaks in the Referer header when the user navigates to another site. Full URLs can carry IDs, tokens, or search queries to third parties. strict-origin-when-cross-origin sends only the origin cross-origin (the browser default since 2020); no-referrer and same-origin leak even less. unsafe-url sends the full URL everywhere, including on plain-HTTP requests.",
        1.0,
    ),
    HeaderSpec(
        "Permissions-Policy",
        "Declares which browser features (camera, microphone, geolocation, payment, USB...) the page and its embedded iframes may use. An empty allowlist - feature=() - switches the feature off entirely; an =* wildcard re-allows it everywhere, defeating the point of declaring it.",
        1.0,
    ),
    HeaderSpec(
        "X-XSS-Protection",
        "The 2010-era XSS Auditor: browsers inspected reflected input in the page and blocked obviously injected scripts. Chrome removed it in 2019 (its heuristics introduced cross-site leaks of their own) and every modern browser now ignores it. The header is deprecated - CSP is the real defense. Sites either omit it or send '0' to switch old browsers off.",
        0.5,
    ),
    HeaderSpec(
        "Cross-Origin-Opener-Policy",
        "Isolates the page's window handle from other origins. Without COOP, a site that opened your page in a popup (or was opened by it) keeps a JS reference to it, enabling some cross-window attacks and Spectre-class side channels. same-origin cuts that shared handle; unsafe-none restores the old shared browsing context.",
        1.0,
    ),
)

/** Grade points used for the weighted overall score. */
private val GRADE_POINTS = mapOf(
    HeaderGrade.A to 100.0, HeaderGrade.B to 70.0, HeaderGrade.C to 40.0, HeaderGrade.F to 0.0,
)

// One year in seconds - the minimum HSTS max-age worth calling strong.
private const val ONE_YEAR = 31536000

/**
 * Parse a raw header block into a lowercased-name map. Lines are `name: value`;
 * the `HTTP/1.1 200 OK` status line and any line without a colon are ignored.
 * A line starting with space/tab continues the previous header (RFC 7230
 * obs-fold); a repeated name joins its values with ", ".
 */
fun parseHeaders(raw: String): Map<String, String> {
    val headers = linkedMapOf<String, String>()
    var lastName: String? = null
    for (line in (raw).split(Regex("\\r?\\n"))) {
        if (line.startsWith(" ") || line.startsWith("\t")) {
            // Folded continuation - append to the previous header, if any.
            val name = lastName
            if (name != null) {
                val prev = headers[name] ?: ""
                headers[name] = "$prev ${line.trim()}".trim()
            }
            continue
        }
        val colon = line.indexOf(':')
        if (colon == -1) continue // status line, blank line, or junk
        val name = line.substring(0, colon).trim().lowercase()
        if (name.isEmpty()) continue
        val value = line.substring(colon + 1).trim()
        val prev = headers[name]
        headers[name] = if (prev != null) "$prev, $value" else value
        lastName = name
    }
    return headers
}

// --- Per-header graders: value (null when absent) -> grade + guidance ---

/** The verdict fields every grader returns; folded into ParsedHeader. */
data class Grading(val grade: HeaderGrade, val explanation: String, val recommendation: String)

private fun gradeHsts(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - the browser accepts plain HTTP, so the first visit (and SSL-strip attacks) can downgrade the connection.",
            "Add: strict-transport-security: max-age=31536000; includeSubDomains; preload",
        )
    }
    val maxAgeMatch = Regex("max-age\\s*=\\s*(\\d+)", RegexOption.IGNORE_CASE).find(value)
        ?: return Grading(
            HeaderGrade.B,
            "Present but malformed - no readable max-age directive ($value).",
            "Use: strict-transport-security: max-age=31536000; includeSubDomains",
        )
    val maxAge = maxAgeMatch.groupValues[1].toLong()
    if (maxAge == 0L) {
        return Grading(
            HeaderGrade.F,
            "max-age=0 actively deletes the HSTS policy - the site opts out of HTTPS-only enforcement.",
            "Raise max-age to at least 31536000 (one year).",
        )
    }
    val hasSubs = Regex(";\\s*includesubdomains", RegexOption.IGNORE_CASE).containsMatchIn(value)
    if (maxAge < ONE_YEAR) {
        return Grading(
            HeaderGrade.B,
            "Present with max-age=${maxAge}s (< 1 year)${if (hasSubs) " and includeSubDomains" else ""} - a shorter window weakens the guarantee.",
            "Raise max-age to 31536000 (one year) or more.",
        )
    }
    if (!hasSubs) {
        return Grading(
            HeaderGrade.B,
            "Strong max-age (>= 1 year), but includeSubDomains is missing - subdomains stay downgradable.",
            "Append ; includeSubDomains once every subdomain serves HTTPS.",
        )
    }
    val hasPreload = Regex(";\\s*preload", RegexOption.IGNORE_CASE).containsMatchIn(value)
    return Grading(
        HeaderGrade.A,
        "Strong: max-age >= 1 year + includeSubDomains${if (hasPreload) " + preload - eligible for the browser preload list." else ". Add preload to join the browser preload list."}",
        "Best practice met - nothing to change.",
    )
}

private fun gradeCsp(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - no restriction on where scripts and resources load from, so injected markup runs at full power.",
            "Add a policy, e.g.: content-security-policy: default-src 'none'; script-src 'self'; ...",
        )
    }
    val inline = Regex("'unsafe-inline'", RegexOption.IGNORE_CASE).containsMatchIn(value)
    val evaluates = Regex("'unsafe-eval'", RegexOption.IGNORE_CASE).containsMatchIn(value)
    if (inline && evaluates) {
        return Grading(
            HeaderGrade.F,
            "Actively harmful - 'unsafe-inline' + 'unsafe-eval' together re-allow inline injection and string evaluation, gutting the policy while it still looks present.",
            "Remove 'unsafe-inline' and 'unsafe-eval'; use nonces or hashes for inline scripts.",
        )
    }
    if (inline || evaluates) {
        val culprit = if (inline) "'unsafe-inline'" else "'unsafe-eval'"
        return Grading(
            HeaderGrade.B,
            "Present but weak - $culprit punches a hole through the anti-XSS guarantee.",
            "Remove $culprit; use nonces or hashes for legitimate inline code.",
        )
    }
    if (!Regex("(^|\\s)default-src\\s", RegexOption.IGNORE_CASE).containsMatchIn(value)) {
        return Grading(
            HeaderGrade.B,
            "Present with no default-src - resource types not listed explicitly fall back to the browser permissive default.",
            "Start the policy with default-src 'none' or 'self', then allow what you need.",
        )
    }
    return Grading(
        HeaderGrade.A,
        "Strong - a default-src fallback with no unsafe-inline / unsafe-eval escape hatches.",
        "Best practice met - nothing to change.",
    )
}

private fun gradeFrameOptions(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - any site can embed this page in an iframe and overlay it (clickjacking).",
            "Add: x-frame-options: DENY (or SAMEORIGIN if you embed yourself)",
        )
    }
    val v = value.trim().uppercase()
    if (v == "DENY" || v == "SAMEORIGIN") {
        return Grading(
            HeaderGrade.A,
            "Strong - $v blocks framing by other origins.",
            "Best practice met - nothing to change.",
        )
    }
    if (v.startsWith("ALLOW-FROM")) {
        return Grading(
            HeaderGrade.B,
            "ALLOW-FROM is deprecated and modern browsers ignore it - the page is effectively frameable.",
            "Replace with a CSP frame-ancestors directive (x-frame-options: DENY as fallback).",
        )
    }
    return Grading(
        HeaderGrade.B,
        "Unrecognized value ($value) - browsers ignore it, leaving the page frameable.",
        "Use DENY or SAMEORIGIN.",
    )
}

private fun gradeContentTypeOptions(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - browsers may sniff uploaded content into executable types.",
            "Add: x-content-type-options: nosniff",
        )
    }
    if (value.trim().lowercase() == "nosniff") {
        return Grading(
            HeaderGrade.A,
            "Strong - MIME sniffing is off; the declared Content-Type is final.",
            "Best practice met - nothing to change.",
        )
    }
    return Grading(
        HeaderGrade.B,
        "Present but the value ($value) is not nosniff, so browsers ignore the header.",
        "Send exactly: x-content-type-options: nosniff",
    )
}

private val REFERRER_STRICT = listOf("no-referrer", "same-origin", "strict-origin", "strict-origin-when-cross-origin")
private val REFERRER_WEAK = listOf("origin", "origin-when-cross-origin", "no-referrer-when-downgrade")

private fun gradeReferrerPolicy(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - older browsers default to leaking the full URL to whatever site the user navigates to.",
            "Add: referrer-policy: strict-origin-when-cross-origin (or no-referrer)",
        )
    }
    val v = value.trim().lowercase()
    if (v == "unsafe-url") {
        return Grading(
            HeaderGrade.F,
            "Actively harmful - unsafe-url sends the full URL (including IDs and tokens in query strings) to every destination, even on plain HTTP.",
            "Use strict-origin-when-cross-origin or no-referrer instead.",
        )
    }
    if (v in REFERRER_STRICT) {
        return Grading(
            HeaderGrade.A,
            "Strong - $v leaks at most the origin cross-origin.",
            "Best practice met - nothing to change.",
        )
    }
    if (v in REFERRER_WEAK) {
        return Grading(
            HeaderGrade.B,
            "Present but leaky - $v still shares the full URL in some same- or cross-origin cases.",
            "Tighten to strict-origin-when-cross-origin, strict-origin, same-origin, or no-referrer.",
        )
    }
    return Grading(
        HeaderGrade.B,
        "Unrecognized value ($value) - browsers fall back to their default policy.",
        "Use one of the standard directives, e.g. strict-origin-when-cross-origin.",
    )
}

private fun gradePermissionsPolicy(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - powerful features (camera, geolocation, payment...) default to the browser policy, not yours.",
            "Add: permissions-policy: camera=(), microphone=(), geolocation=()",
        )
    }
    val v = value.replace(Regex("\\s+"), "").lowercase()
    if (v.isEmpty()) {
        return Grading(
            HeaderGrade.B,
            "Present but empty - no feature is restricted, so it declares nothing.",
            "List the features to switch off, e.g. camera=(), microphone=().",
        )
    }
    if (v == "*" || v.contains("=*")) {
        return Grading(
            HeaderGrade.B,
            "Present but permissive - a =* wildcard re-allows the listed feature(s) everywhere.",
            "Use an empty allowlist () or a specific origin instead of *.",
        )
    }
    return Grading(
        HeaderGrade.A,
        "Strong - the policy restricts at least one powerful feature.",
        "Best practice met - nothing to change.",
    )
}

private fun gradeXssProtection(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.A,
            "Correctly omitted - the header is deprecated; modern browsers removed the XSS Auditor it drove.",
            "Nothing to change - rely on Content-Security-Policy instead.",
        )
    }
    val v = value.trim()
    if (v == "0" || v.startsWith("0;")) {
        return Grading(
            HeaderGrade.A,
            "Explicitly disabled ('0') - the right call for a deprecated header old browsers still honor.",
            "Nothing to change - or remove the header entirely.",
        )
    }
    if (v.startsWith("1")) {
        return Grading(
            HeaderGrade.B,
            "Deprecated - '1; mode=block' is ignored by modern browsers and the old Auditor had cross-site leaks of its own.",
            "Send '0' or drop the header; use Content-Security-Policy for XSS defense.",
        )
    }
    return Grading(
        HeaderGrade.B,
        "Present with an unrecognized value ($value) - remove the deprecated header.",
        "Send '0' or drop the header entirely.",
    )
}

private fun gradeCoop(value: String?): Grading {
    if (value == null) {
        return Grading(
            HeaderGrade.C,
            "Missing - pages that open or are opened by other origins share a browsing context with them.",
            "Add: cross-origin-opener-policy: same-origin",
        )
    }
    val v = value.trim().lowercase()
    return when (v) {
        "same-origin" -> Grading(
            HeaderGrade.A,
            "Strong - the window handle is isolated from cross-origin openers.",
            "Best practice met - nothing to change.",
        )
        "same-origin-allow-popups" -> Grading(
            HeaderGrade.B,
            "Partially isolated - popups you open keep a reference to this window.",
            "Use same-origin unless you genuinely need cross-origin popup handles.",
        )
        "unsafe-none" -> Grading(
            HeaderGrade.B,
            "Explicit opt-out - unsafe-none restores the shared browsing context COOP exists to cut.",
            "Use same-origin unless a documented integration requires shared handles.",
        )
        else -> Grading(
            HeaderGrade.B,
            "Unrecognized value ($value) - browsers treat it as no COOP.",
            "Use same-origin.",
        )
    }
}

// Compile-time-complete grader table: every catalog name has a grader.
private val GRADERS: Map<String, (String?) -> Grading> = mapOf(
    "Strict-Transport-Security" to ::gradeHsts,
    "Content-Security-Policy" to ::gradeCsp,
    "X-Frame-Options" to ::gradeFrameOptions,
    "X-Content-Type-Options" to ::gradeContentTypeOptions,
    "Referrer-Policy" to ::gradeReferrerPolicy,
    "Permissions-Policy" to ::gradePermissionsPolicy,
    "X-XSS-Protection" to ::gradeXssProtection,
    "Cross-Origin-Opener-Policy" to ::gradeCoop,
)

/** Map a weighted 0-100 score to the overall A/B/C/F grade. */
private fun overallGrade(score: Int): HeaderGrade = when {
    score >= 90 -> HeaderGrade.A
    score >= 70 -> HeaderGrade.B
    score >= 45 -> HeaderGrade.C
    else -> HeaderGrade.F
}

/**
 * Analyze a raw HTTP response-header block. Every catalog header gets a
 * ParsedHeader row (in catalog order); the overall score is the weight-adjusted
 * average of per-header grade points, and recommendations list one fix per
 * header that is not at grade A.
 */
fun analyzeHeaders(raw: String): HeaderAnalysis {
    val parsed = parseHeaders(raw)
    val headers = SECURITY_HEADERS.map { spec ->
        val value = parsed[spec.name.lowercase()]
        val grading = GRADERS.getValue(spec.name)(value)
        ParsedHeader(spec.name, value, grading.grade, grading.explanation, grading.recommendation)
    }
    val totalWeight = SECURITY_HEADERS.sumOf { it.weight }
    val score = Math.round(
        headers.zip(SECURITY_HEADERS).sumOf { (h, spec) -> GRADE_POINTS.getValue(h.grade) * spec.weight } / totalWeight
    ).toInt()
    val recommendations = headers.filter { it.grade != HeaderGrade.A }.map { it.recommendation }
    return HeaderAnalysis(headers, overallGrade(score), score, recommendations)
}