![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
diff --git a/nagixsc.py b/nagixsc.py
index bf204c042c5ce0cf0e4e9082aefba679eed710df..56c4cf3ead117ca047a818085912232a5ad514d4 100644 (file)
--- a/nagixsc.py
+++ b/nagixsc.py
+import BaseHTTPServer
+import ConfigParser
+import SocketServer
import base64
import datetime
import libxml2
+import mimetools
+import os
+import random
+import shlex
+import socket
+import string
+import subprocess
+import sys
def debug(level, verb, string):
if level <= verb:
return ['base64', 'plain',]
+def check_encoding(enc):
+ if enc in available_encodings():
+ return True
+ else:
+ return False
+
+
def decode(data, encoding):
if encoding == 'plain':
return data
return base64.b64encode(data)
+##############################################################################
+
+def read_inifile(inifile):
+ config = ConfigParser.RawConfigParser()
+ config.optionxform = str # We need case-sensitive options
+ ini_list = config.read(inifile)
+
+ if ini_list:
+ return config
+ else:
+ return False
+
+
+##############################################################################
+
+def exec_check(host_name, service_descr, cmdline):
+ try:
+ cmd = subprocess.Popen(shlex.split(cmdline), stdout=subprocess.PIPE)
+ output = cmd.communicate()[0].rstrip()
+ retcode = cmd.returncode
+ except OSError:
+ output = 'Could not execute "%s"' % cmdline
+ retcode = 127
+
+ return {'host_name':host_name, 'service_description':service_descr, 'returncode':retcode, 'output':output, 'timestamp':datetime.datetime.now().strftime('%s')}
+
+
+##############################################################################
+
+def conf2dict(config, opt_host=None, opt_service=None):
+ checks = []
+
+ # Sections are Hosts (not 'nagixsc'), options in sections are Services
+ hosts = config.sections()
+ if 'nagixsc' in hosts:
+ hosts.remove('nagixsc')
+
+ # Filter out host/section if it exists
+ if opt_host:
+ if opt_host in hosts:
+ hosts = [opt_host,]
+ else:
+ hosts = []
+
+ for host in hosts:
+ # Overwrite section/host name with '_host_name'
+ if config.has_option(host,'_host_name'):
+ host_name = config.get(host,'_host_name')
+ else:
+ host_name = host
+
+
+ services = config.options(host)
+ # Look for host check
+ if '_host_check' in services and not opt_service:
+ cmdline = config.get(host, '_host_check')
+ check = exec_check(host_name, None, cmdline)
+ checks.append(check)
+
+
+ # Filter out service if given in cmd line options
+ if opt_service:
+ if opt_service in services:
+ services = [opt_service,]
+ else:
+ services = []
+
+ for service in services:
+ # If option starts with '_' it may be a NagixSC option in the future
+ if service[0] != '_':
+ cmdline = config.get(host, service)
+
+ check = exec_check(host_name, service, cmdline)
+ checks.append(check)
+
+ return checks
+
+
+##############################################################################
+
+def dict2out_passive(checks, xmltimestamp, opt_pipe, opt_verb=0):
+ FORMAT_HOST = '[%s] PROCESS_HOST_CHECK_RESULT;%s;%s;%s'
+ FORMAT_SERVICE = '[%s] PROCESS_SERVICE_CHECK_RESULT;%s;%s;%s;%s'
+ count_services = 0
+ now = datetime.datetime.now().strftime('%s')
+
+ # Prepare
+ if opt_verb <= 2:
+ pipe = open(opt_pipe, "w")
+ else:
+ pipe = None
+
+ # Output
+ for check in checks:
+ count_services += 1
+ if check.has_key('timestamp'):
+ timestamp = check['timestamp']
+ else:
+ timestamp = xmltimestamp
+ count_services += 1
+
+ if check['service_description'] == None or check['service_description'] == '':
+ # Host check
+ line = FORMAT_HOST % (now, check['host_name'], check['returncode'], check['output'].replace('\n', '\\n'))
+ else:
+ # Service check
+ line = FORMAT_SERVICE % (now, check['host_name'], check['service_description'], check['returncode'], check['output'].replace('\n', '\\n'))
+
+ if pipe:
+ pipe.write(line + '\n')
+ debug(2, opt_verb, line)
+
+ # Close
+ if pipe:
+ pipe.close()
+ else:
+ print "Passive check results NOT written to Nagios pipe due to -vvv!"
+
+ return count_services
+
+
+def dict2out_checkresult(checks, xmltimestamp, opt_checkresultdir, opt_verb):
+ count_services = 0
+ count_failed = 0
+ list_failed = []
+ chars = string.letters + string.digits
+ ctimestamp = datetime.datetime.now().ctime()
+
+ for check in checks:
+ count_services += 1
+ if check.has_key('timestamp'):
+ timestamp = check['timestamp']
+ else:
+ timestamp = xmltimestamp
+
+ filename = os.path.join(opt_checkresultdir, 'c' + ''.join([random.choice(chars) for i in range(6)]))
+ try:
+ crfile = open(filename, "w")
+ if check['service_description'] == None or check['service_description'] == '':
+ # Host check
+ crfile.write('### Active Check Result File ###\nfile_time=%s\n\n### Nagios Service Check Result ###\n# Time: %s\nhost_name=%s\ncheck_type=0\ncheck_options=0\nscheduled_check=1\nreschedule_check=1\nlatency=0.0\nstart_time=%s.00\nfinish_time=%s.05\nearly_timeout=0\nexited_ok=1\nreturn_code=%s\noutput=%s\n' % (timestamp, ctimestamp, check['host_name'], timestamp, timestamp, check['returncode'], check['output'].replace('\n', '\\n') ) )
+ else:
+ # Service check
+ crfile.write('### Active Check Result File ###\nfile_time=%s\n\n### Nagios Service Check Result ###\n# Time: %s\nhost_name=%s\nservice_description=%s\ncheck_type=0\ncheck_options=0\nscheduled_check=1\nreschedule_check=1\nlatency=0.0\nstart_time=%s.00\nfinish_time=%s.05\nearly_timeout=0\nexited_ok=1\nreturn_code=%s\noutput=%s\n' % (timestamp, ctimestamp, check['host_name'], check['service_description'], timestamp, timestamp, check['returncode'], check['output'].replace('\n', '\\n') ) )
+ crfile.close()
+
+ # Create OK file
+ open(filename + '.ok', 'w').close()
+ except:
+ count_failed += 1
+ list_failed.append([filename, check['host_name'], check['service_description']])
+
+ return (count_services, count_failed, list_failed)
+
+
+##############################################################################
+
+def read_xml(options):
+ if options.url != None:
+ import urllib2
+
+ if options.httpuser and options.httppasswd:
+ passman = urllib2.HTTPPasswordMgrWithDefaultRealm()
+ passman.add_password(None, options.url, options.httpuser, options.httppasswd)
+ authhandler = urllib2.HTTPBasicAuthHandler(passman)
+ opener = urllib2.build_opener(authhandler)
+ urllib2.install_opener(opener)
+
+ try:
+ response = urllib2.urlopen(options.url)
+ except urllib2.HTTPError, error:
+ print error
+ sys.exit(0)
+ except urllib2.URLError, error:
+ print error.reason[1]
+ sys.exit(0)
+
+ doc = libxml2.parseDoc(response.read())
+ response.close()
+
+ else:
+ doc = libxml2.parseFile(options.file)
+
+ return doc
+
+
+def read_xml_from_string(content):
+ return libxml2.parseDoc(content)
+
+
##############################################################################
def xml_check_version(xmldoc):
def xml_to_dict(xmldoc, verb=0, hostfilter=None, servicefilter=None):
checks = []
- filetimestamp = xml_get_timestamp(xmldoc)
+ now = int(datetime.datetime.now().strftime('%s'))
+ filetimestamp = reset_future_timestamp(xml_get_timestamp(xmldoc), now)
if hostfilter:
hosts = xmldoc.xpathNewContext().xpathEval('/nagixsc/host[name="%s"] | /nagixsc/host[name="%s"]' % (hostfilter, encode(hostfilter)))
output = None
if host.xpathEval('timestamp'):
- timestamp = int(host.xpathEval('timestamp')[0].get_content())
+ timestamp = reset_future_timestamp(int(host.xpathEval('timestamp')[0].get_content()), now)
else:
timestamp = filetimestamp
output = decode(xmloutput.get_content(), xmloutput.prop('encoding')).rstrip()
try:
- timestamp = int(service.xpathEval('timestamp')[0].get_content())
+ timestamp = reset_future_timestamp(int(service.xpathEval('timestamp')[0].get_content()), now)
except:
timestamp = filetimestamp
check['returncode'] = 3
return check
+
+def reset_future_timestamp(timestamp, now):
+ if timestamp <= now:
+ return timestamp
+ else:
+ return now
+
+##############################################################################
+
+def encode_multipart(xmldoc, httpuser, httppasswd):
+ BOUNDARY = mimetools.choose_boundary()
+ CRLF = '\r\n'
+ L = []
+ L.append('--' + BOUNDARY)
+ L.append('Content-Disposition: form-data; name="xmlfile"; filename="xmlfile"')
+ L.append('Content-Type: application/xml')
+ L.append('')
+ L.append(xmldoc.serialize())
+ L.append('--' + BOUNDARY + '--')
+ L.append('')
+ body = CRLF.join(L)
+ content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
+ headers = {'Content-Type': content_type, 'Content-Length': str(len(body))}
+
+ if httpuser and httppasswd:
+ headers['Authorization'] = 'Basic %s' % base64.b64encode(':'.join([httpuser, httppasswd]))
+
+ return (headers, body)
+
+##############################################################################
+
+class MyHTTPServer(BaseHTTPServer.HTTPServer):
+ def __init__(self, server_address, HandlerClass, ssl=False, sslpemfile=None):
+ if ssl:
+ # FIXME: SSL is in Py2.6
+ try:
+ from OpenSSL import SSL
+ except:
+ print 'No Python OpenSSL wrapper/bindings found!'
+ sys.exit(127)
+
+ SocketServer.BaseServer.__init__(self, server_address, HandlerClass)
+ context = SSL.Context(SSL.SSLv23_METHOD)
+ context.use_privatekey_file (sslpemfile)
+ context.use_certificate_file(sslpemfile)
+ self.socket = SSL.Connection(context, socket.socket(self.address_family, self.socket_type))
+ else:
+ SocketServer.BaseServer.__init__(self, server_address, HandlerClass)
+ self.socket = socket.socket(self.address_family, self.socket_type)
+
+ self.server_bind()
+ self.server_activate()
+
+
+class MyHTTPRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+ def setup(self):
+ self.connection = self.request
+ self.rfile = socket._fileobject(self.request, "rb", self.rbufsize)
+ self.wfile = socket._fileobject(self.request, "wb", self.wbufsize)
+
+##############################################################################
+