source: monitoring/plugins/mpi/check_clarin_saml.py @ 3305

#!/usr/bin/python2

# This Nagios monitoring plugin is not as lean and clean as the other ones. Do not use this
# as an example.

import sys, getopt, httplib, subprocess, tempfile, os, re, pdb
import os.path
#import xml.etree.ElementTree


DESCRIPTION = "SAML"
command_line_parameters = [("-h", "host",),
                           ("-u", "URL"),
                           ("-d", "infra.clarin.eu root directory path",)]

nagios_codes = {
                'OK'        : 0,
                'WARNING'   : 1,
                'CRITICAL'  : 2,
                'UNKNOWN'   : 3,
                'DEPENDENT' : 4
                }

def nagios_return(code, response) :
    """ prints the response message
        and exits the script with one
        of the defined exit codes
        DOES NOT RETURN
    """
    print code + ": " + response
    sys.exit(nagios_codes[code])

def check_response_data_validity(data) :

    def interpret_for_well_formedness(stdout) :
        pattern                             = '^INFO : XML document parsed and is well-formed.'
        pattern_regex                       = re.compile(pattern, re.MULTILINE)
        results                             = pattern_regex.search(stdout)
    
        if results is not None :
            return(True)

        return(False)

    def interpret_for_validity(stdout) :
        pattern                             = '^INFO : XML document is schema valid'
        pattern_regex                       = re.compile(pattern, re.MULTILINE)
        results                             = pattern_regex.search(stdout)
    
        if results is not None :
            return(True)

        return(False)

    #pdb.set_trace()

    check_saml_metadata_file_path       = os.path.join(CLARIN_server_root_directory_path, "aai/check-saml-metadata/check_saml_metadata.sh")

    temporary_file                      = tempfile.NamedTemporaryFile(mode = 'wb')                                    
    temporary_file.write(data)
    temporary_file.flush()

    XML_response_metadata_file_path     = temporary_file.name

    command                             = [check_saml_metadata_file_path, XML_response_metadata_file_path]

    environment_variables               = { "JAVA_HOME" : '/usr/lib/jvm/java-6-openjdk-amd64/jre' } # X- hard coding
    process                             = subprocess.Popen(command, 
                                                           stdout = subprocess.PIPE, 
                                                           stderr = subprocess.PIPE,
                                                                                       env    = environment_variables)

    stdout, stderr                      = process.communicate()

    #pdb.set_trace()
    
    temporary_file.close()

    interpretation_tuple                = (interpret_for_well_formedness(stdout), interpret_for_validity(stdout),)

    #os.unlink(XML_response_metadata_file_path)

    return(interpretation_tuple)


def check_condition(host, URL, CLARIN_server_root_directory_path) :

    conn    = httplib.HTTPSConnection(host)
    
    # Use exception handling.
    conn.request("GET", URL)
    
    r1      = conn.getresponse()

    data    = r1.read()
    
    conn.close()

    #pdb.set_trace()
    
    if r1.status == 200 :
        interpretation_tuple = check_response_data_validity(data)

        well_formed = interpretation_tuple[0]
        valid       = interpretation_tuple[1]

        if well_formed :
            if valid :
                return {
                        "code"      : "OK", 
                        "message"   : 'Host %s, service %s is up and returns well-formed and valid XML data at %s.' % (host, DESCRIPTION, URL)
                       }
            else :
                return {
                        "code"      : "CRITICAL", 
                        "message"   : 'Host %s, service %s is up and returns well-formed, but invalid XML data at %s.' % (host, DESCRIPTION, URL)
                       }
        else :
            return {
                        "code"      : "CRITICAL", 
                        "message"   : 'Host %s, service %s is up but returns non-well-formed XML data %s.' % (host, DESCRIPTION, URL)
                    }
    else :
        return {
                "code"      : "CRITICAL", 
                "message"   : 'Host %s, service %s, location %s has a problem.' % (host, DESCRIPTION, URL)
               }

def test_case(host, URL, CLARIN_server_root_directory_path) :
    metadata_file       = open(os.path.join(CLARIN_server_root_directory_path, "aai/clarin-sp-metadata.xml"),
              mode      = "r+", 
              buffering = 0)
        
    data = metadata_file.read()

    metadata_file.close()       

    interpretation_tuple = check_response_data_validity(data)

    well_formed = interpretation_tuple[0]
    valid       = interpretation_tuple[1]

    if well_formed :
        if valid :
            return {
                    "code"      : "OK", 
                    "message"   : 'Host %s, service %s is up and returns well-formed and valid XML data at %s.' % (host, DESCRIPTION, URL)
                   }
        else :
            return {
                    "code"      : "CRITICAL", 
                    "message"   : 'Host %s, service %s is up and returns well-formed, but invalid XML data at %s.' % (host, DESCRIPTION, URL)
                   }
    else :
        return {
                "code"      : "CRITICAL", 
                "message"   : 'Host %s, service %s is up, but returns non-well-formed XML data at %s.' % (host, DESCRIPTION, URL)
               }

    Sys.exit(0)


def special_main_subroutine(host, 
                            URL, 
                            CLARIN_server_root_directory_path) :

    result = check_condition(host = host, 
                             URL = URL, 
                             CLARIN_server_root_directory_path = CLARIN_server_root_directory_path)
    
    nagios_return(result['code'], result['message'])

if __name__ == "__main__" :
    generic_tla_monitoring.main(special_main_subroutine, 
                                command_line_parameters)