Class MySAMLParser

python3_cron_scripts/libs3/Rapid7.py:119–147

This sub-class searches an HTML response for the SAML data

class MySAMLParser(HTMLParser):
    """
    This sub-class searches an HTML response for the SAML data
    """

    saml_response = ""
    relay_state = ""

    def handle_starttag(self, tag, attrs):
        if tag == "input":
            found_saml = False
            found_relay = False
            for attr in attrs:
                if (
                    self.saml_response == ""
                    and attr[0] == "name"
                    and attr[1] == "SAMLResponse"
                ):
                    found_saml = True
                elif (
                    self.relay_state == ""
                    and attr[0] == "name"
                    and attr[1] == "RelayState"
                ):
                    found_relay = True
                elif found_saml and attr[0] == "value":
                    self.saml_response = attr[1]
                elif found_relay and attr[0] == "value":
                    self.relay_state = attr[1]


class Rapid7(object):