Mutliple Security Header Check.js

// Multiple Security Header checker by freakyclown@gmail.com

var ScanRuleMetadata = Java.type(
  "org.zaproxy.addon.commonlib.scanrules.ScanRuleMetadata"
);

function getMetadata() {
  return ScanRuleMetadata.fromYaml(`
id: 100016
name: Missing Security Headers
description: >
  Some of the following security headers are missing from the HTTP response:
  Strict-Transport-Security, Content-Security-Policy,
  X-XSS-Protection, X-Content-Type-Options, X-Frame-Options, Referrer-Policy.
solution: >
  Ensure that your web server, application server, load balancer, etc.
  is configured to set the missing security headers.
risk: low
confidence: high
cweId: 693  # CWE-693: Protection Mechanism Failure
wascId: 15  # WASC-15: Application Misconfiguration
status: alpha
codeLink: https://github.com/zaproxy/community-scripts/blob/main/passive/Mutliple%20Security%20Header%20Check.js
helpLink: https://www.zaproxy.org/docs/desktop/addons/community-scripts/
`);
}

function scan(helper, msg, src) {
var alertTitle = [
    "Strict-Transport-Security (HSTS) Header Not Set (script)",
    "Content-Security-Policy (script)",
    "Web Browser XSS Protection Not Enabled (script)",
    "X-Content-Type-Options Header Missing (script)",
    "X-Frame-Options Header Not Set (script)",
    "Referrer-Policy Header Not Set (script)",
    "",
  ];
  var alertDesc = [
    "HTTP Strict Transport Security (HSTS) is a web security policy mechanism whereby a web server declares that complying user agents (such as a web browser) are to interact with it using only secure HTTPS  connections (i.e. HTTP layered over TLS/SSL). HSTS is an IETF standards track protocol and is specified in RFC 6797.",
    "Content Security Policy (CSP) is an added layer of security that helps to detect and mitigate certain types of attacks, including Cross Site Scripting (XSS) and data injection attacks. These attacks are used for everything from data theft to site defacement or distribution of malware. CSP provides a standard HTTP header that allows website owners to declare approved sources of content that browsers should be allowed to load on that page — covered types are JavaScript, CSS, HTML frames, fonts, images and embeddable objects such as Java applets, ActiveX, audio and video files.",
    "Web Browser XSS Protection is not enabled, or is disabled by the configuration of the 'X-XSS-Protection' HTTP response header on the web server",
    "The Anti-MIME-Sniffing header X-Content-Type-Options was not set to 'nosniff'. This allows older versions of Internet Explorer and Chrome to perform MIME-sniffing on the response body, potentially causing the response body to be interpreted and displayed as a content type other than the declared content type. Current (early 2014) and legacy versions of Firefox will use the declared content type (if one is set), rather than performing MIME-sniffing.",
    "X-Frame-Options header is not included in the HTTP response to protect against 'ClickJacking' attacks.",
    "The Referrer-Policy header is not set. This header controls how much referrer information is included with requests, helping to protect user privacy and prevent information leakage.",
    "",
  ];
  var alertSolution = [
    "Ensure that your web server, application server, load balancer, etc. is configured to set Strict Transport Security headers.",
    "Ensure that your web server, application server, load balancer, etc. is configured to set Content Security Policy headers.",
    "Ensure that the web browser's XSS filter is enabled, by setting the X-XSS-Protection HTTP response header to '1'.",
    "Ensure that the application/web server sets the Content-Type header appropriately, and that it sets the X-Content-Type-Options header to 'nosniff' for all web pages. If possible, ensure that the end user uses a standards-compliant and modern web browser that does not perform MIME-sniffing at all, or that can be directed by the web application/web server to not perform MIME-sniffing.",
    "Most modern Web browsers support the X-Frame-Options HTTP header. Ensure it's set on all web pages returned by your site (if you expect the page to be framed only by pages on your server (e.g. it's part of a FRAMESET) then you'll want to use SAMEORIGIN, otherwise if you never expect the page to be framed, you should use DENY.  ALLOW-FROM allows specific websites to frame the web page in supported web browsers).",
    "Ensure that your web server, application server, load balancer, etc. is configured to set the Referrer-Policy header. Recommended value: 'strict-origin-when-cross-origin'.",
    "",
  ];
  var responseHeader = msg.getResponseHeader().toString();

  // test sts
  if (msg.getRequestHeader().isSecure()) {
    if (
      msg.getResponseHeader().getHeaders("Strict-Transport-Security") == null
    ) {
      helper
        .newAlert()
        .setName(alertTitle[0])
        .setDescription(alertDesc[0])
        .setSolution(alertSolution[0])
        .setMessage(msg)
        .raise();
    }
  }
  // test csp
  if (
    !hasAnyHeader(msg.getResponseHeader(), [
      "Content-Security-Policy",
      "X-Content-Security-Policy",
      "X-WebKit-CSP",
    ])
  ) {
    helper
      .newAlert()
      .setName(alertTitle[1])
      .setDescription(alertDesc[1])
      .setSolution(alertSolution[1])
      .setMessage(msg)
      .raise();
  }

  // test xxs protection
  var re_xss = /(X\-XSS\-Protection\:.+1)/g;
  if (!re_xss.test(responseHeader)) {
    //if its false
    helper
      .newAlert()
      .setName(alertTitle[2])
      .setDescription(alertDesc[2])
      .setSolution(alertSolution[2])
      .setMessage(msg)
      .raise();
  }

  // test xcontent no sniff protection
  var re_nosniff = /(X\-Content\-Type\-Options\:.*nosniff.*)/g;
  if (!re_nosniff.test(responseHeader)) {
    //if its false
    helper
      .newAlert()
      .setRisk(2)
      .setConfidence(2)
      .setName(alertTitle[3])
      .setDescription(alertDesc[3])
      .setSolution(alertSolution[3])
      .setMessage(msg)
      .raise();
  }

  // test xcontent no sniff protection
  var re_clickjack = /(X\-Frame\-Options\:.+[Dd][Ee][Nn][Yy])/g;
  if (!re_clickjack.test(responseHeader)) {
    //if its false
    helper
      .newAlert()
      .setName(alertTitle[4])
      .setDescription(alertDesc[4])
      .setSolution(alertSolution[4])
      .setMessage(msg)
      .raise();
  }
}

// test Referrer-Policy
  if (msg.getResponseHeader().getHeaders("Referrer-Policy") == null) {
    helper
      .newAlert()
      .setName(alertTitle[5])
      .setDescription(alertDesc[5])
      .setSolution(alertSolution[5])
      .setMessage(msg)
      .raise();
  }

function hasAnyHeader(header, headers) {
  for (var i in headers) {
    if (header.getHeader(headers[i])) {
      return true;
    }
  }
  return false;
}