from operator import itemgetter
from typing import Dict, Set
from urllib.parse import urlparse, urlunparse

from httpobs.scanner.analyzer.decorators import scored_test
from httpobs.scanner.analyzer.utils import is_hsts_preloaded, only_if_worse
from httpobs.scanner.retriever import get_duplicate_header_values

# Ignore the CloudFlare __cfduid tracking cookies. They *are* actually bad, but it is out of a site's
# control.  See https://github.com/mozilla/http-observatory/issues/121 for additional details. Hopefully
# this will eventually be fixed on CloudFlare's end.

# Also ignore the Heroku sticky session cookie, see:
# https://github.com/mozilla/http-observatory/issues/282
COOKIES_TO_DELETE = ['__cfduid', 'heroku-session-affinity']

# CSP settings
SHORTEST_DIRECTIVE = 'img-src'
SHORTEST_DIRECTIVE_LENGTH = len(SHORTEST_DIRECTIVE) - 1  # the shortest policy accepted by the CSP test


def __parse_csp(csp_strings: list) -> Dict[str, Set]:
    """
    Decompose the CSP; could probably do this in one step, but it's complicated enough
    Should look like:
    {
      'default-src': {'\'none\''},
      'object-src': {'\'none\''},
      'script-src': {'https://mozilla.org', '\'unsafe-inline\''},
      'style-src': {'\'self\', 'https://mozilla.org'},
      'upgrade-insecure-requests': {},
    }
    """

    # Clean out all the junk on each possible entry
    csp_strings = [csp_string.replace('\r', '').replace('\n', '').strip() for csp_string in csp_strings]

    # So technically the shortest directive is img-src, so lets just assume that
    # anything super short is invalid
    if not csp_strings:
        return {}

    for csp_string in csp_strings:
        if len(csp_string) < SHORTEST_DIRECTIVE_LENGTH or csp_string.isspace():
            raise ValueError('CSP policy does not meet minimum length requirements')

    csp = {}

    # since we can have multiple policies, we need to iterate through each policy
    for policy_index, policy in enumerate(csp_strings):
        directive_seen_before_in_this_policy = {}

        for entry in [directive.strip().split(maxsplit=1) for directive in policy.split(';') if directive]:
            if not entry:  # Catch errant semi-colons
                continue

            # Using lower due to directives being case insensitive after CSP3
            directive = entry[0].lower()

            # While technically valid in that you just use the first entry, we are saying that repeated
            # directives are invalid so that people notice it
            if directive in directive_seen_before_in_this_policy:
                raise ValueError('Repeated policy directives are invalid')
            else:
                directive_seen_before_in_this_policy[directive] = True

            # Technically the path part of any source is case-sensitive, but since we don't test
            # any paths, we can cheat a little bit here, TODO: Fix this?

            # each value that gets to the set is a tuple consisting of:
            # (value, policy index, whether it should kept from the list later in the function)
            if len(entry) > 1:
                values = []

                for source in entry[-1].split():
                    if '://' in source:
                        # we have to do this to make the domain lowercase for comparisons later
                        url = urlparse(source)
                        url = url._replace(netloc=url.netloc.lower())
                        values.append(
                            {
                                'source': urlunparse(url),
                                'index': policy_index,
                                'keep': True if policy_index == 0 else False,
                            }
                        )
                    else:
                        values.append(
                            {
                                'source': source.lower(),
                                'index': policy_index,
                                'keep': True if policy_index == 0 else False,
                            }
                        )
            elif len(entry) == 1 and directive.endswith("-src"):  # if it's a source list with no values, it's 'none'
                values = [
                    {
                        'source': "'none'",
                        'index': policy_index,
                        'keep': True if policy_index == 0 else False,
                    }
                ]
            else:
                values = []

            if policy_index == 0:
                combined_sources = sorted(values, key=itemgetter('source'))
            else:
                combined_sources = sorted(csp.get(directive, []) + list(values), key=itemgetter('source'))

            # for the first pass through, we simply remove sources where the previous item in the source
            # list starts with the current item in the source list, making it redundant

            # If you have multiple CSP policies, a directive has to be in _both_ sets of policies or
            # it doesn't apply. This is super hard to do in "reverse", as the Observatory does.
            # To do this, we combine the directive from multiple policies

            if len(combined_sources) > 1:
                for index, source in enumerate(combined_sources[1:], start=1):
                    # convenience variable pointing to previous entry in the combined list
                    prev = combined_sources[index - 1]

                    # if it's from the same policy and they start with the same thing, the longer one is
                    # superfluous, e.g. https://example.com/foo and https://example.com/foobar
                    if source['index'] == prev['index'] and source['source'].startswith(prev['source']):
                        source['keep'] = False

                    # a source _has_ to exist in both policies for it to count
                    if source['index'] != prev['index'] and source['source'].startswith(prev['source']):
                        source['keep'] = True

            # now we need to purge anything that's not necessary and store it into the policy
            csp[directive] = [source for source in combined_sources if source['keep'] is True]

            # the first time through the loop is special case -- everything is marked as True to keep,
            # and only purged if it has a shorter match. however, if we are going to have more loops through
            # due to having multiple CSP policies, then everything needs to be marked False to keep and
            # then forcibly kept in future loops
            if policy_index == 0 and len(csp_strings) > 1:
                for source in csp[directive]:
                    source['keep'] = False

    # now we need to flatten out all the CSP directives (e.g. (source, index, False) back into actual values
    # if they had defined a directive and didn't have a value remaining, then force it to none
    for directive, sources in csp.items():
        csp[directive] = set([source['source'] for source in sources]) if sources else {"'none'"}

    return csp


@scored_test
def content_security_policy(reqs: dict, expectation='csp-implemented-with-no-unsafe') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        csp-implemented-with-no-unsafe: CSP implemented with no unsafe inline keywords [default]
        csp-implemented-with-unsafe-in-style-src-only: Allow the 'unsafe' keyword in style-src only
        csp-implemented-with-insecure-scheme-in-passive-content-only:
          CSP implemented with insecure schemes (http, ftp) in img/media-src
        csp-implemented-with-unsafe-inline: CSP implemented with unsafe-inline
        csp-implemented-with-unsafe-eval: CSP implemented with unsafe-eval
        csp-implemented-with-insecure-scheme: CSP implemented with having sources over http:
        csp-invalid-header: Invalid CSP header
        csp-not-implemented: CSP not implemented
    :return: dictionary with:
        data: the CSP lookup dictionary
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
    """

    output = {
        'data': None,
        'expectation': expectation,
        'http': False,  # whether an HTTP header was available
        'meta': False,  # whether an HTTP meta-equiv was available
        'pass': False,
        'policy': None,
        'result': None,
    }
    response = reqs['responses']['auto']

    # TODO: check for CSP meta tags
    # TODO: try to parse when there are multiple CSP headers

    # Obviously you can get around it with things like https://*.org, but you're only hurting yourself
    DANGEROUSLY_BROAD = ('ftp:', 'http:', 'https:', '*', 'http://*', 'http://*.*', 'https://*', 'https://*.*')
    UNSAFE_INLINE = ('\'unsafe-inline\'', 'data:')

    # Passive content check
    PASSIVE_DIRECTIVES = ('img-src', 'media-src')

    # What do nonces and hashes start with?
    NONCES_HASHES = ('\'sha256-', '\'sha384-', '\'sha512-', '\'nonce-')

    # First, let's grab the CSP values from both the HTTP headers and the meta tags
    http_csp_header = get_duplicate_header_values(response, 'Content-Security-Policy')
    equiv_csp_header = response.http_equiv.get('Content-Security-Policy', [])
    output['numPolicies'] = len(http_csp_header) + len(equiv_csp_header)  # TODO: add tests

    # First we need to combine the HTTP headers and HTTP Equiv "headers"
    try:
        csp = __parse_csp(http_csp_header + equiv_csp_header)
    except:
        output['result'] = 'csp-header-invalid'
        return output

    # If we have neither HTTP header nor meta, then there isn't any CSP
    if not csp:
        output['result'] = 'csp-not-implemented'
        return output

    # we need the CSP from headers alone, for things like frame-ancestors
    try:
        http_header_only_csp = __parse_csp(http_csp_header)
    except ValueError:
        http_header_only_csp = {}

    # If we make it this far, we have a policy object
    output['policy'] = {
        'antiClickjacking': False,
        'defaultNone': False,
        'insecureBaseUri': False,
        'insecureFormAction': False,
        'insecureSchemeActive': False,
        'insecureSchemePassive': False,
        'strictDynamic': False,
        'unsafeEval': False,
        'unsafeInline': False,
        'unsafeInlineStyle': False,
        'unsafeObjects': False,
    }

    # mark whether we saw csp there or not
    output['http'] = True if http_csp_header else False
    output['meta'] = True if equiv_csp_header else False

    # Get the various directives we look at
    base_uri = csp.get('base-uri') or {'*'}
    frame_ancestors = http_header_only_csp.get('frame-ancestors', {'*'})
    form_action = csp.get('form-action') or {'*'}
    object_src = csp.get('object-src') or csp.get('default-src') or {'*'}
    script_src = csp.get('script-src') or csp.get('default-src') or {'*'}
    style_src = csp.get('style-src') or csp.get('default-src') or {'*'}

    # Remove 'unsafe-inline' if nonce or hash are used in script-src or style-src
    # See: https://github.com/mozilla/http-observatory/issues/88
    #      https://github.com/mozilla/http-observatory/issues/277
    for source_list in (script_src, style_src):
        if any(source.startswith(NONCES_HASHES) for source in source_list) and '\'unsafe-inline\'' in source_list:
            source_list.remove('\'unsafe-inline\'')

    # If a script-src uses 'strict-dynamic', we need to:
    # 1. Check to make sure there's a valid nonce/hash source
    # 2. Remove any source that starts with as scheme
    # 3. Remove 'self' and 'unsafe-inline'
    if any(source.startswith(NONCES_HASHES) for source in script_src) and '\'strict-dynamic\'' in script_src:
        for source in set(script_src):
            if source.startswith(DANGEROUSLY_BROAD) or source == '\'self\'' or source == '\'unsafe-inline\'':
                script_src.remove(source)
        output['policy']['strictDynamic'] = True
    # 'strict-dynamic' in script-src without hash or nonce
    elif '\'strict-dynamic\'' in script_src:
        output['result'] = 'csp-header-invalid' if output['result'] is None else output['result']

    # Some checks look only at active/passive CSP directives
    # This could be inlined, but the code is quite hard to read at that point
    active_csp_sources = [
        source
        for directive, source_list in csp.items()
        for source in source_list
        if directive not in PASSIVE_DIRECTIVES and directive not in 'script-src'
    ] + list(script_src)
    passive_csp_sources = [
        source
        for source_list in [csp.get(directive, csp.get('default-src', [])) for directive in PASSIVE_DIRECTIVES]
        for source in source_list
    ]

    # No 'unsafe-inline' or data: in script-src
    # Also don't allow overly broad schemes such as https: in either object-src or script-src
    # Likewise, if you don't have object-src or script-src defined, then all sources are allowed
    if script_src.intersection(DANGEROUSLY_BROAD + UNSAFE_INLINE) or object_src.intersection(DANGEROUSLY_BROAD):
        output['result'] = 'csp-implemented-with-unsafe-inline' if output['result'] is None else output['result']
        output['policy']['unsafeInline'] = True

    # If the site is https, it shouldn't allow any http: as a source (active content)
    if (
        urlparse(response.url).scheme == 'https'
        and [source for source in active_csp_sources if 'http:' in source or 'ftp:' in source]
        and not output['policy']['strictDynamic']
    ):
        output['result'] = 'csp-implemented-with-insecure-scheme' if output['result'] is None else output['result']
        output['policy']['insecureSchemeActive'] = True

    # Don't allow 'unsafe-eval' in script-src or style-src
    if script_src.union(style_src).intersection({'\'unsafe-eval\''}):
        output['result'] = 'csp-implemented-with-unsafe-eval' if output['result'] is None else output['result']
        output['policy']['unsafeEval'] = True

    # If the site is https, it shouldn't allow any http: as a source (passive content)
    if urlparse(response.url).scheme == 'https' and [
        source for source in passive_csp_sources if 'http:' in source or 'ftp:' in source
    ]:
        output['result'] = (
            'csp-implemented-with-insecure-scheme-in-passive-content-only'
            if output['result'] is None
            else output['result']
        )
        output['policy']['insecureSchemePassive'] = True

    # Don't allow 'unsafe-inline', data:, or overly broad sources in style-src
    if style_src.intersection(DANGEROUSLY_BROAD + UNSAFE_INLINE):
        output['result'] = (
            'csp-implemented-with-unsafe-inline-in-style-src-only' if output['result'] is None else output['result']
        )
        output['policy']['unsafeInlineStyle'] = True

    # Only if default-src is 'none' and 'none' alone, since additional uris override 'none'
    if csp.get('default-src') == {'\'none\''}:
        output['result'] = (
            'csp-implemented-with-no-unsafe-default-src-none' if output['result'] is None else output['result']
        )
        output['policy']['defaultNone'] = True
    else:
        output['result'] = 'csp-implemented-with-no-unsafe' if output['result'] is None else output['result']

    # Some other checks for the CSP analyzer
    output['policy']['antiClickjacking'] = not bool(frame_ancestors.intersection(DANGEROUSLY_BROAD))
    output['policy']['insecureBaseUri'] = bool(base_uri.intersection(DANGEROUSLY_BROAD + UNSAFE_INLINE))
    output['policy']['insecureFormAction'] = bool(form_action.intersection(DANGEROUSLY_BROAD))
    output['policy']['unsafeObjects'] = bool(object_src.intersection(DANGEROUSLY_BROAD))

    # Once we're done, convert every set() in csp to an array
    csp = {k: list(v) for k, v in csp.items()}

    # TODO: allow a small bonus for upgrade-insecure-requests?

    # Code defensively on the size of the data
    output['data'] = csp if len(str(csp)) < 32768 else {}

    # Check to see if the test passed or failed
    if output['result'] in (
        expectation,
        'csp-implemented-with-no-unsafe-default-src-none',
        'csp-implemented-with-unsafe-inline-in-style-src-only',
        'csp-implemented-with-insecure-scheme-in-passive-content-only',
    ):
        output['pass'] = True

    return output


@scored_test
def cookies(reqs: dict, expectation='cookies-secure-with-httponly-sessions') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        cookies-secure-with-httponly-sessions-and-samesite: All cookies are secure,
          use HttpOnly if needed, and SameSite
        cookies-secure-with-httponly-sessions: All cookies have secure flag set, all session cookies are HttpOnly
        cookies-without-secure-flag-but-protected-by-hsts: Cookies don't have secure, but site uses HSTS
        cookies-session-without-secure-flag-but-protected-by-hsts: Same, but session cookie
        cookies-without-secure-flag: Cookies set without secure flag
        cookies-samesite-flag-invalid: Cookies set with invalid SameSite value
          (must be either unset, Strict, Lax or None)
        cookies-session-without-secure-flag: Session cookies lack the Secure flag
        cookies-session-without-httponly-flag: Session cookies lack the HttpOnly flag
        cookies-not-found: No cookies found in HTTP requests
    :return: dictionary with:
        data: the cookie jar
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
        sameSite: True if all session cookies have a valid SameSite attribute
                  False if any session cookie has an invalid or missing SameSite attribute
                  None if there are no session cookies
    """

    output = {
        'data': None,
        'expectation': expectation,
        'pass': False,
        'result': None,
        'sameSite': None,
    }
    session = reqs['session']  # all requests and their associated cookies

    # The order of how bad the various results are
    goodness = [
        'cookies-without-secure-flag-but-protected-by-hsts',
        'cookies-without-secure-flag',
        'cookies-session-without-secure-flag-but-protected-by-hsts',
        'cookies-samesite-flag-invalid',
        'cookies-anticsrf-without-samesite-flag',
        'cookies-session-without-httponly-flag',
        'cookies-session-without-secure-flag',
    ]

    # TODO: Support cookies set over http-equiv (ugh)
    # https://github.com/mozilla/http-observatory/issues/265

    # Get their HTTP Strict Transport Security status, which can help when cookies are set without Secure
    hsts = strict_transport_security(reqs)['pass']

    # If there are no cookies
    if not session.cookies:
        output['result'] = 'cookies-not-found'

    else:
        jar = {}

        # There are certain cookies we ignore, because they are set by service providers and sites have
        # no control over them.
        for cookie in COOKIES_TO_DELETE:
            del session.cookies[cookie]

        for cookie in session.cookies:
            # The HttpOnly and SameSite functionality is a bit broken
            cookie.httponly = cookie.samesite = False
            for key in cookie._rest:
                if key.lower() == 'httponly' and getattr(cookie, 'httponly') is False:
                    cookie.httponly = True
                elif key.lower() == 'samesite' and getattr(cookie, 'samesite') is False:
                    samesiteVal = '' if cookie._rest[key] is None else str(cookie._rest[key])
                    if samesiteVal.strip().lower() == 'lax':
                        cookie.samesite = 'Lax'
                    elif samesiteVal.strip().lower() == 'strict':
                        cookie.samesite = 'Strict'
                    elif samesiteVal.strip().lower() == 'none':
                        cookie.samesite = 'None'
                    else:
                        output['result'] = only_if_worse('cookies-samesite-flag-invalid', output['result'], goodness)

            # Add it to the jar
            jar[cookie.name] = {
                i: getattr(cookie, i, None)
                for i in ['domain', 'expires', 'httponly', 'max-age', 'path', 'port', 'samesite', 'secure']
            }

            # Is it a session identifier or an anti-csrf token?
            sessionid = any(i in cookie.name.lower() for i in ('login', 'sess'))
            anticsrf = True if 'csrf' in cookie.name.lower() else False

            if not cookie.secure and cookie.samesite == 'None':
                output['result'] = only_if_worse('cookies-samesite-flag-invalid', output['result'], goodness)

            if not cookie.secure and hsts:
                output['result'] = only_if_worse(
                    'cookies-without-secure-flag-but-protected-by-hsts', output['result'], goodness
                )

            elif not cookie.secure:
                output['result'] = only_if_worse('cookies-without-secure-flag', output['result'], goodness)

            # Anti-CSRF tokens should be set using the SameSite option
            if anticsrf and not cookie.samesite:
                output['result'] = only_if_worse('cookies-anticsrf-without-samesite-flag', output['result'], goodness)

            # Login and session cookies should be set with Secure
            if sessionid and not cookie.secure and hsts:
                output['result'] = only_if_worse(
                    'cookies-session-without-secure-flag-but-protected-by-hsts', output['result'], goodness
                )
            elif sessionid and not cookie.secure:
                output['result'] = only_if_worse('cookies-session-without-secure-flag', output['result'], goodness)

            # Login and session cookies should be set with HttpOnly
            if sessionid and not cookie.httponly:
                output['result'] = only_if_worse('cookies-session-without-httponly-flag', output['result'], goodness)

        # Store whether or not we saw SameSite cookies, if cookies were set
        if output['result'] is None:
            if any(c for c in session.cookies if c.samesite is False):
                output['result'] = 'cookies-secure-with-httponly-sessions'
                output['sameSite'] = False
            else:
                output['result'] = 'cookies-secure-with-httponly-sessions-and-samesite'
                output['sameSite'] = True

        # Save the cookie jar
        output['data'] = jar if len(str(jar)) < 32768 else {}

    # Check to see if the test passed or failed
    if output['result'] in ('cookies-not-found', 'cookies-secure-with-httponly-sessions-and-samesite', expectation):
        output['pass'] = True

    return output


@scored_test
def referrer_policy(reqs: dict, expectation='referrer-policy-private') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        referrer-policy-private: Referrer-Policy header set to "no-referrer" or "same-origin", "strict-origin"
          or "strict-origin-when-origin"
        referrer-policy-no-referrer-when-downgrade: Referrer-Policy header set to "no-referrer-when-downgrade"
        referrer-policy-origin: Referrer-Policy header set to "origin"
        referrer-policy-origin-when-cross-origin: Referrer-Policy header set to "origin-when-cross-origin"
        referrer-policy-unsafe-url: Referrer-Policy header set to "unsafe-url"
        referrer-policy-not-implemented: Referrer-Policy header not implemented
        referrer-policy-header-invalid
    :return: dictionary with:
        data: the raw HTTP Referrer-Policy header
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
    """

    output = {
        'data': None,
        'expectation': expectation,
        'http': False,  # whether an HTTP header was available
        'meta': False,  # whether an HTTP meta-equiv was available
        'pass': False,
        'result': None,
    }

    goodness = ['no-referrer', 'same-origin', 'strict-origin', 'strict-origin-when-cross-origin']

    badness = ['origin', 'origin-when-cross-origin', 'unsafe-url']

    valid = goodness + badness + ['no-referrer-when-downgrade']

    response = reqs['responses']['auto']

    # Store whether the header or meta were present
    output['http'] = True if 'Referrer-Policy' in response.headers else False
    output['meta'] = True if 'Referrer-Policy' in response.http_equiv else False

    # If it's in both a header and http-equiv, http-equiv gets precedence (aka comes last)
    if 'Referrer-Policy' in response.headers and 'Referrer-Policy' in response.http_equiv:
        output['data'] = ', '.join([response.headers['Referrer-Policy'], response.http_equiv['Referrer-Policy']])[
            0:256
        ]  # Code defensively
    elif 'Referrer-Policy' in response.headers or 'Referrer-Policy' in response.http_equiv:
        output['data'] = (response.http_equiv.get('Referrer-Policy') or response.headers.get('Referrer-Policy'))[0:256]
    else:
        output['result'] = 'referrer-policy-not-implemented'
        output['pass'] = True
        return output

    # Find the last known valid policy value in the Referer Policy
    policy = [token.strip() for token in output['data'].lower().split(',') if token.strip() in valid]
    policy = policy.pop() if policy else None

    if policy in goodness:
        output['result'] = 'referrer-policy-private'
    elif policy == 'no-referrer-when-downgrade':
        output['result'] = 'referrer-policy-no-referrer-when-downgrade'
    elif policy in badness:
        output['result'] = 'referrer-policy-unsafe'
    else:
        output['result'] = 'referrer-policy-header-invalid'

    # Test passed or failed
    if output['result'] in (
        'referrer-policy-private',
        'referrer-policy-not-implemented',
        'referrer-policy-no-referrer-when-downgrade',
        expectation,
    ):
        output['pass'] = True

    return output


@scored_test
def strict_transport_security(reqs: dict, expectation='hsts-implemented-max-age-at-least-six-months') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        hsts-implemented-max-age-at-least-six-months: HSTS implemented with a max age of at least six months (15768000)
        hsts-implemented-max-age-less-than-six-months: HSTS implemented with a max age of less than six months
        hsts-not-implemented-no-https: HSTS can't be implemented on http only sites
        hsts-not-implemented: HSTS not implemented
        hsts-header-invalid: HSTS header isn't parsable
        hsts-invalid-cert: Invalid certificate chain
    :return: dictionary with:
        data: the raw HSTS header
        expectation: test expectation
        includesubdomains: whether the includeSubDomains directive is set
        pass: whether the site's configuration met its expectation
        preload: whether the preload flag is set
        result: short string describing the result of the test
    """
    SIX_MONTHS = 15552000  # 15768000 is six months, but a lot of sites use 15552000, so a white lie is in order

    output = {
        'data': None,
        'expectation': expectation,
        'includeSubDomains': False,
        'max-age': None,
        'pass': False,
        'preload': False,
        'preloaded': False,
        'result': 'hsts-not-implemented',
    }
    response = reqs['responses']['https']

    # If there's no HTTPS, we can't have HSTS
    if response is None:
        output['result'] = 'hsts-not-implemented-no-https'

    # Also need a valid certificate chain for HSTS
    elif not response.verified:
        output['result'] = 'hsts-invalid-cert'

    elif 'Strict-Transport-Security' in response.headers:
        output['data'] = response.headers['Strict-Transport-Security'][0:1024]  # code against malicious headers

        try:
            sts = [i.lower().strip() for i in output['data'].split(';')]

            # Throw an error if the header is set twice
            if ',' in output['data']:
                raise ValueError

            for parameter in sts:
                if parameter.startswith('max-age='):
                    output['max-age'] = int(parameter[8:128])  # defense
                elif parameter == 'includesubdomains':
                    output['includeSubDomains'] = True
                elif parameter == 'preload':
                    output['preload'] = True

            if output['max-age']:
                if output['max-age'] < SIX_MONTHS:  # must be at least six months
                    output['result'] = 'hsts-implemented-max-age-less-than-six-months'
                else:
                    output['result'] = 'hsts-implemented-max-age-at-least-six-months'
            else:
                raise ValueError

        except:
            output['result'] = 'hsts-header-invalid'

    # If they're in the preloaded list, this overrides most anything else
    # TODO: Check to see if all redirect domains are preloaded
    # TODO: Check every redirect along the way for HSTS
    if response is not None:
        preloaded = is_hsts_preloaded(urlparse(response.url).netloc)
        if preloaded:
            output['result'] = 'hsts-preloaded'
            output['includeSubDomains'] = preloaded['includeSubDomains']
            output['preloaded'] = True

    # Check to see if the test passed or failed
    if output['result'] in ('hsts-implemented-max-age-at-least-six-months', 'hsts-preloaded', expectation):
        output['pass'] = True

    return output


@scored_test
def x_content_type_options(reqs: dict, expectation='x-content-type-options-nosniff') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        x-content-type-options-nosniff: X-Content-Type-Options set to "nosniff" [default]
        x-content-type-options-not-implemented: X-Content-Type-Options header missing
        x-content-type-options-header-invalid
    :return: dictionary with:
        data: the raw X-Content-Type-Options header
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
    """

    output = {
        'data': None,
        'expectation': expectation,
        'pass': False,
        'result': None,
    }
    response = reqs['responses']['auto']

    if 'X-Content-Type-Options' in response.headers:
        output['data'] = response.headers['X-Content-Type-Options'][0:256]  # code defensively

        if output['data'].strip().lower() == 'nosniff':
            output['result'] = 'x-content-type-options-nosniff'
        else:
            output['result'] = 'x-content-type-options-header-invalid'
    else:
        output['result'] = 'x-content-type-options-not-implemented'

    # Check to see if the test passed or failed
    if expectation == output['result']:
        output['pass'] = True

    return output


@scored_test
def x_frame_options(reqs: dict, expectation='x-frame-options-sameorigin-or-deny') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        x-frame-options-sameorigin-or-deny: X-Frame-Options set to "sameorigin" or "deny" [default]
        x-frame-options-allow-from-origin: X-Frame-Options set to ALLOW-FROM uri
        x-frame-options-implemented-via-csp: X-Frame-Options implemented via CSP frame-ancestors directive
        x-frame-options-not-implemented: X-Frame-Options header missing
        x-frame-options-header-invalid: Invalid X-Frame-Options header
    :return: dictionary with:
        data: the raw X-Frame-Options header
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
    """

    output = {
        'data': None,
        'expectation': expectation,
        'pass': False,
        'result': None,
    }
    response = reqs['responses']['auto']

    if 'X-Frame-Options' in response.headers:
        output['data'] = response.headers['X-Frame-Options'][0:1024]  # code defensively
        xfo = output['data'].strip().lower()

        if xfo in ('deny', 'sameorigin'):
            output['result'] = 'x-frame-options-sameorigin-or-deny'
        elif xfo.startswith('allow-from '):
            output['result'] = 'x-frame-options-allow-from-origin'
        else:
            output['result'] = 'x-frame-options-header-invalid'
    else:
        output['result'] = 'x-frame-options-not-implemented'

    # Check to see if frame-ancestors is implemented in CSP; if it is, then it isn't needed
    csp = content_security_policy(reqs)
    if csp['data']:
        if 'frame-ancestors' in csp['data']:  # specifically not checking for * in frame-ancestors
            output['result'] = 'x-frame-options-implemented-via-csp'

    # Check to see if the test passed or failed
    if output['result'] in (
        'x-frame-options-allow-from-origin',
        'x-frame-options-sameorigin-or-deny',
        'x-frame-options-implemented-via-csp',
        expectation,
    ):
        output['pass'] = True

    return output


@scored_test
def x_xss_protection(reqs: dict, expectation='x-xss-protection-disabled') -> dict:
    """
    :param reqs: dictionary containing all the request and response objects
    :param expectation: test expectation
        x-xss-protection-enabled-mode-block: X-XSS-Protection set to "1; block"
        x-xss-protection-enabled: X-XSS-Protection set to "1"
        x-xss-protection-disabled: X-XSS-Protection set to "0" (disabled) [default]
        x-xss-protection-not-implemented: X-XSS-Protection header missing
        x-xss-protection-header-invalid
    :return: dictionary with:
        data: the raw X-XSS-Protection header
        expectation: test expectation
        pass: whether the site's configuration met its expectation
        result: short string describing the result of the test
    """
    VALID_DIRECTIVES = ('0', '1', 'mode', 'report')
    VALID_MODES = ('block',)

    output = {
        'data': None,
        'expectation': expectation,
        'pass': False,
        'result': None,
    }

    enabled = False  # XXSSP enabled or not
    valid = True  # XXSSP header valid or not
    response = reqs['responses']['auto']
    header = response.headers.get('X-XSS-Protection', '').strip()
    xxssp = {}

    if header:
        output['data'] = header[0:256]  # code defensively

        # Parse out the X-XSS-Protection header
        try:
            if header[0] not in ('0', '1'):
                raise ValueError

            if header[0] == '1':
                enabled = True

            # {'1': None, 'mode': 'block', 'report': 'https://www.example.com/__reporturi__'}
            for directive in header.lower().split(';'):
                k, v = [d.strip() for d in directive.split('=')] if '=' in directive else (directive.strip(), None)

                # An invalid directive, like foo=bar
                if k not in VALID_DIRECTIVES:
                    raise ValueError

                # An invalid mode, like mode=allow
                if k == 'mode' and v not in VALID_MODES:
                    raise ValueError

                # A repeated directive, such as 1; mode=block; mode=block
                if k in xxssp:
                    raise ValueError

                xxssp[k] = v
        except:
            output['result'] = 'x-xss-protection-header-invalid'
            valid = False

        if valid and enabled and xxssp.get('mode') == 'block':
            output['result'] = 'x-xss-protection-enabled-mode-block'
            output['pass'] = True
        elif valid and enabled:
            output['result'] = 'x-xss-protection-enabled'
            output['pass'] = True
        elif valid and not enabled:
            output['result'] = 'x-xss-protection-disabled'
            output['pass'] = True

    else:
        output['result'] = 'x-xss-protection-not-implemented'
        output['pass'] = True

    return output