import requests
import pandas as pd
from datetime import datetime, timedelta

# Mock vulnerability payload from scanners
VULN_DATA = [
    {"id": "CVE-2024-1001", "cvss": 9.8, "exploit_available": True, "asset_criticality": "high", "exposure": "internet", "component": "nginx:1.21"},
    {"id": "CVE-2024-1002", "cvss": 7.5, "exploit_available": False, "asset_criticality": "medium", "exposure": "internal", "component": "openssl:1.1.1"},
    {"id": "CVE-2024-1003", "cvss": 5.0, "exploit_available": True, "asset_criticality": "low", "exposure": "internet", "component": "log4j:2.14"}
]

def calculate_risk_score(vuln):
    """
    Dynamic risk scoring: CVSS + exploit maturity + asset criticality + exposure
    Returns a 0-100 score for prioritization
    """
    base = vuln["cvss"] * 10  # Scale CVSS to 0-100
    exploit_bonus = 15 if vuln["exploit_available"] else 0
    criticality_map = {"high": 20, "medium": 10, "low": 0}
    asset_bonus = criticality_map.get(vuln["asset_criticality"], 0)
    exposure_map = {"internet": 15, "dmz": 8, "internal": 0}
    exposure_bonus = exposure_map.get(vuln["exposure"], 0)
    
    raw_score = base + exploit_bonus + asset_bonus + exposure_bonus
    return min(raw_score, 100)  # Cap at 100

def prioritize_vulnerabilities(vulns):
    df = pd.DataFrame(vulns)
    df["risk_score"] = df.apply(calculate_risk_score, axis=1)
    df["priority"] = pd.cut(df["risk_score"], bins=[0, 40, 70, 100], labels=["low", "medium", "high"])
    return df.sort_values("risk_score", ascending=False)

# Execution
prioritized = prioritize_vulnerabilities(VULN_DATA)
print(prioritized[["id", "cvss", "risk_score", "priority", "component"]])

def create_remediation_ticket(vuln, priority):
    """
    Routes vulnerability to engineering ticketing system with SLA based on priority
    """
    sla_map = {
        "high": {"days": 7, "label": "CRITICAL-7D"},
        "medium": {"days": 30, "label": "HIGH-30D"},
        "low": {"days": 90, "label": "MEDIUM-90D"}
    }
    sla = sla_map.get(priority, {"days": 90, "label": "LOW-90D"})
    
    ticket_payload = {
        "fields": {
            "project": {"key": "SEC"},
            "summary": f"Remediate {vuln['id']} in {vuln['component']}",
            "description": f"Risk Score: {vuln['risk_score']}\nExposure: {vuln['exposure']}\nExploit Available: {vuln['exploit_available']}",
            "issuetype": {"name": "Vulnerability Remediation"},
            "labels": [sla["label"], "vuln-mgmt"],
            "customfield_10010": {"value": "Security"}  # Assign to security team initially
        }
    }
    # In production: replace with actual Jira/ServiceNow API call
    print(f"[TICKET CREATED] {vuln['id']} | Priority: {priority} | SLA: {sla['days']} days")
    return ticket_payload

# Route top findings
for _, row in prioritized.head(2).iterrows():
    create_remediation_ticket(row.to_dict(), row["priority"])

3. Verification & Drift Detection Hook

def verify_remediation(cve_id, component, scan_results):
    """
    Checks if a reported vulnerability persists in the latest scan
    """
    active = any(
        r["cve"] == cve_id and r["component"] == component and r["status"] == "open"
        for r in scan_results
    )
    return "REOPENED" if active else "RESOLVED"

# Example verification
latest_scans = [
    {"cve": "CVE-2024-1001", "component": "nginx:1.21", "status": "open"},
    {"cve": "CVE-2024-1002", "component": "openssl:1.1.1", "status": "fixed"}
]
print(f"Verification CVE-2024-1001: {verify_remediation('CVE-2024-1001', 'nginx:1.21', latest_scans)}")
print(f"Verification CVE-2024-1002: {verify_remediation('CVE-2024-1002', 'openssl:1.1.1', latest_scans)}")

Integration Notes:

• Replace mock data with API connectors to Qualys, Tenable, Trivy, Snyk, GitHub Advisory Database, or cloud provider security hubs.
• Store risk scores in a centralized database (PostgreSQL, DynamoDB) with versioning for audit trails.
• Use webhook triggers in CI/CD to block merges when risk_score > threshold and no remediation ticket exists.
• Implement rate limiting, retry logic, and credential rotation for production API calls.

---

Pitfall Guide

1. CVSS-Only Prioritization

Problem: Treating all high-CVSS vulnerabilities as equally urgent ignores exploit availability, asset context, and business impact. Mitigation: Implement dynamic risk scoring that layers CVSS with EPSS (Exploit Prediction Scoring System), asset criticality tags, network exposure, and data sensitivity. Tune weights quarterly based on remediation outcomes.

2. Lack of Asset Context & CMDB Sync

Problem: Scanners report vulnerabilities without knowing which systems are production, customer-facing, or decommissioned. Mitigation: Integrate VM data with your CMDB or cloud inventory. Tag assets with business units, data classification, and environment labels. Automatically suppress findings on non-production or archived assets.

3. Broken Remediation Ownership

Problem: Security creates tickets but engineering lacks clear ownership, leading to SLA drift and finger-pointing. Mitigation: Map vulnerabilities to code owners using repository metadata, IaC tags, or service mesh routing. Enforce automatic assignment to responsible teams. Track MTTR by team, not just organization-wide.

4. Tool Sprawl & Data Silos

Problem: Multiple scanners produce overlapping, conflicting, or uncorrelated findings. Teams waste time reconciling reports. Mitigation: Establish a single vulnerability ingestion pipeline with normalization (CVE, CWE, CPE, SBOM). Deduplicate using hash-based matching and component versioning. Retire redundant scanners where coverage overlaps.

5. No Closed-Loop Verification

Problem: Remediation is assumed complete after a ticket is closed, but drift or misconfiguration reintroduces risk. Mitigation: Automate post-remediation verification via scheduled scans, agent health checks, or pipeline re-scans. Flag reopened vulnerabilities for root-cause analysis. Maintain an audit trail of fix → verify → close.

6. Over-Automating Without Governance

Problem: Fully autonomous ticketing or patch deployment causes production outages or compliance violations. Mitigation: Implement approval gates for high-impact changes. Use canary deployments for patch rollouts. Maintain a change advisory board (CAB) workflow for critical infrastructure. Log all automated actions for compliance.

7. Ignoring Shift-Left & Developer Experience

Problem: VM is treated as a security team function, creating friction and delaying fixes until late in the lifecycle. Mitigation: Integrate vulnerability checks into PR/MR pipelines. Provide developers with clear remediation guidance, dependency upgrade commands, and bypass workflows for acceptable risk. Measure developer satisfaction and adoption rates.

---

Production Bundle

✅ VM Program Launch Checklist

Pre-Launch

• Define risk scoring model with business stakeholders
• Map all scanners, cloud security hubs, and SBOM tools to ingestion pipeline
• Establish CMDB sync or cloud inventory tagging strategy
• Configure ticketing system with custom fields, labels, and SLA workflows
• Draft remediation SOPs for critical, high, medium, and low tiers

Operational

• Deploy continuous scanning agents or API connectors
• Enable automated ticket creation with owner routing
• Set up verification hooks and drift detection schedules
• Configure dashboards for MTTR, SLA compliance, and risk trends
• Establish weekly triage sync between security, engineering, and ops

Compliance & Improvement

• Archive evidence chain for audits (scan → ticket → fix → verify)
• Review scoring weights quarterly based on false positive/negative rates
• Conduct tabletop exercises for zero-day response
• Track developer adoption and pipeline integration coverage
• Publish monthly risk posture report to executive leadership

📊 Decision Matrix

Decision Area	Option A	Option B	Option C	Recommended Path
Prioritization Framework	CVSS threshold	EPSS + CVSS	Dynamic risk scoring (CVSS + exploit + asset + exposure)	Option C
Ticketing Integration	Manual creation	Webhook-based API	CI/CD-native PR/MR gating	Option B + C hybrid
Scanner Architecture	Single enterprise scanner	Best-of-breed per layer	Unified ingestion with normalized data model	Option C
Team Ownership	Security-only	Shared security/dev	Security triage + dev remediation + ops verification	Option C
Automation Level	Manual triage	Semi-automated routing	Full pipeline integration with approval gates	Option C with governance

⚙️ Config Template (YAML)

vuln_management:
  ingestion:
    sources:
      - type: cloud_security_hub
        provider: aws
        regions: ["us-east-1", "eu-west-1"]
      - type: container_scanner
        tool: trivy
        scan_targets: ["registry", "filesystem"]
      - type: sast_sca
        platform: github
        repos: ["org/*"]
  
  risk_scoring:
    weights:
      cvss: 0.4
      exploit_maturity: 0.2
      asset_criticality: 0.25
      exposure: 0.15
    thresholds:
      critical: 85
      high: 70
      medium: 45
      low: 0
  
  sla:
    critical: { days: 7, escalation: true }
    high: { days: 30, escalation: false }
    medium: { days: 90, escalation: false }
    low: { days: 180, backlog: true }
  
  routing:
    ticketing: jira
    project_key: SEC
    assign_strategy: code_owner
    labels: ["vuln-mgmt", "sla-{priority}"]
  
  verification:
    schedule: "0 2 * * *"
    method: re-scan
    drift_alert: true
    close_condition: "status=fixed AND risk_score < 45"

🚀 Quick Start: 30-60-90 Day Plan

Days 1–30: Foundation & Ingestion

• Inventory all scanning tools, cloud security services, and dependency managers.
• Stand up centralized vulnerability database with normalized schema.
• Connect primary scanners via API; validate data flow and deduplication.
• Define initial risk scoring model with security and engineering leads.
• Deliver executive briefing on current exposure baseline.

Days 31–60: Automation & Routing

• Implement dynamic risk scoring engine with contextual weights.
• Configure ticketing automation with SLA mapping and owner routing.
• Integrate verification hooks for post-remediation scanning.
• Pilot pipeline gating for critical repositories; measure developer friction.
• Launch first MTTR and SLA compliance dashboard.

Days 61–90: Optimization & Governance

• Tune scoring weights based on 30-day remediation outcomes.
• Expand coverage to 80%+ of production workloads and CI/CD pipelines.
• Establish weekly cross-functional triage cadence.
• Draft zero-day response runbook and conduct tabletop exercise.
• Publish first quarterly risk posture report; align resourcing with trends.

---

Vulnerability management is no longer a periodic security exercise. It is a continuous engineering discipline that demands data normalization, contextual risk scoring, automated routing, and closed-loop verification. Organizations that treat vulnerability data as a real-time signal rather than a compliance artifact will dramatically reduce exposure windows, align security with product velocity, and build resilient, audit-ready defense postures. The code, configurations, and operational patterns outlined here provide a production-tested foundation. Scale them iteratively, measure relentlessly, and let risk—not fear—drive remediation priorities.