![[tokkee]](http://tokkee.org/images/avatar.png){kind=avatar}
diff --git a/nagixsc_conf2http.py b/nagixsc_conf2http.py
index 995fa453f85e288958390823fbda4a2be7883c41..6b063d6d4b15dc58b12e3f8525efe95f29860e37 100755 (executable)
--- a/nagixsc_conf2http.py
+++ b/nagixsc_conf2http.py
#!/usr/bin/python
-
-import BaseHTTPServer
+#
+# Nag(ix)SC -- nagixsc_conf2http.py
+#
+# Copyright (C) 2009-2010 Sven Velt <sv@teamix.net>
+#
+# This program is free software; you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by the
+# Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This program is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+
+import ConfigParser
import base64
+import optparse
import os
import re
import subprocess
+import sys
try:
from hashlib import md5
except ImportError:
from md5 import md5
-config = { 'ip': '',
- 'port': 15666,
- }
+##############################################################################
+
+from nagixsc import *
+
+##############################################################################
+
+parser = optparse.OptionParser()
+
+parser.add_option('-c', '', dest='cfgfile', help='Config file')
+parser.add_option('-d', '--daemon', action='store_true', dest='daemon', help='Daemonize, go to background')
+parser.add_option('', '--nossl', action='store_true', dest='nossl', help='Disable SSL (overwrites config file)')
+
+parser.set_defaults(cfgfile='conf2http.cfg')
+
+(options, args) = parser.parse_args()
+
+cfgread = ConfigParser.SafeConfigParser()
+cfgread.optionxform = str # We need case-sensitive options
+cfg_list = cfgread.read(options.cfgfile)
-users = { 'nagixsc': '019b0966d98fb71d1a4bc4ca0c81d5cc', # PW: nagixsc
+if cfg_list == []:
+ print 'Config file "%s" could not be read!' % options.cfgfile
+ sys.exit(1)
+
+config = {
+ 'ip': '0.0.0.0',
+ 'port': '15666',
+ 'ssl': False,
+ 'sslcert': None,
+ 'conf_dir': '',
+ 'pidfile': '/var/run/nagixsc_conf2http.pid',
+ 'livestatus_socket' : None,
}
-CONFDIR='./examples'
-C2X='./nagixsc_conf2xml.py'
+if 'ip' in cfgread.options('server'):
+ config['ip'] = cfgread.get('server', 'ip')
+
+if 'port' in cfgread.options('server'):
+ config['port'] = cfgread.get('server', 'port')
+try:
+ config['port'] = int(config['port'])
+except ValueError:
+ print 'Port "%s" not an integer!' % config['port']
+ sys.exit(127)
+
+if 'ssl' in cfgread.options('server'):
+ try:
+ config['ssl'] = cfgread.getboolean('server', 'ssl')
+ except ValueError:
+ print 'Value for "ssl" ("%s") not boolean!' % config['ssl']
+ sys.exit(127)
+
+if config['ssl']:
+ if 'sslcert' in cfgread.options('server'):
+ config['sslcert'] = cfgread.get('server', 'sslcert')
+ else:
+ print 'SSL but no certificate file specified!'
+ sys.exit(127)
+
+try:
+ config['conf_dir'] = cfgread.get('server', 'conf_dir')
+except ConfigParser.NoOptionError:
+ print 'No "conf_dir" specified!'
+ sys.exit(127)
+
+if 'pidfile' in cfgread.options('server'):
+ config['pidfile'] = cfgread.get('server', 'pidfile')
-class Conf2HTTPHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+if 'livestatus_socket' in cfgread.options('server'):
+ config['livestatus_socket'] = prepare_socket(cfgread.get('server', 'livestatus_socket'))
+
+
+users = {}
+for u in cfgread.options('users'):
+ users[u] = cfgread.get('users', u)
+
+##############################################################################
+
+class Conf2HTTPHandler(MyHTTPRequestHandler):
def http_error(self, code, output):
self.send_response(code)
def do_GET(self):
- cmdline = C2X
-
path = self.path.split('/')
# Check Basic Auth
if len(path) >= 4:
service = path[3]
else:
- service = ''
+ service = None
if len(path) >= 3:
host = path[2]
else:
- host = ''
+ host = None
if len(path) >= 2:
- configfile = path[1]
+ configfile = path[1] + '.conf'
else:
- configfile =''
+ self.http_error(500, 'No config file specified')
+ return
if re.search('\.\.', configfile):
self.http_error(500, 'Found ".." in config file name')
return
- if configfile and not re.search('^[a-zA-Z0-9-_\.]+$', configfile):
+ if not re.search('^[a-zA-Z0-9-_]+.conf$', configfile):
self.http_error(500, 'Config file name contains invalid characters')
return
- if configfile:
- configfile += '.conf'
- cmdline += ' -c ' + os.path.join(CONFDIR, configfile)
+ # Just be sure it exists
+ checks = None
- if host:
- cmdline += ' -H %s' % host
- if service:
- cmdline += ' -D %s' % service
+ # If config file name starts with "_" it's something special
+ if not configfile.startswith('_'):
+ # Try to read config file, execute checks
+ check_config = read_inifile(os.path.join(config['conf_dir'], configfile))
+ if not check_config:
+ self.http_error(500, 'Could not read config file "%s"' % configfile)
+ return
+ checks = conf2dict(check_config, host, service)
- try:
- cmd = subprocess.Popen(cmdline.split(' '), stdout=subprocess.PIPE)
- output = cmd.communicate()[0].rstrip()
- retcode = cmd.returncode
- except OSError:
- self.http_error(500, 'Could not execute "%s"' % cmdline)
+ elif configfile=='_livestatus.conf' and config['livestatus_socket']:
+ # Read mk-livestatus and translate into XML
+ checks = livestatus2dict(config['livestatus_socket'], host, service)
+
+
+ # No check results? No (good) answer...
+ if not checks:
+ self.http_error(500, 'No check results')
return
- if retcode == 0:
- self.send_response(200)
- self.send_header('Content-Type', 'text/xml')
- self.end_headers()
- self.wfile.write(output)
- else:
- self.http_error(500, output)
+ self.send_response(200)
+ self.send_header('Content-Type', 'text/xml')
+ self.end_headers()
+ self.wfile.write(xml_from_dict(checks))
return
def main():
+ if options.nossl:
+ config['ssl'] = False
+
+ if config['ssl'] and not os.path.isfile(config['sslcert']):
+ print 'SSL certificate "%s" not found!' % config['sslcert']
+ sys.exit(127)
+
+ if not os.path.isdir(config['conf_dir']):
+ print 'Not a config file directory: "%s"' % config['conf_dir']
+ sys.exit(127)
+
+ if options.daemon:
+ daemonize(pidfile=config['pidfile'])
+
+ server = MyHTTPServer((config['ip'], config['port']), Conf2HTTPHandler, ssl=config['ssl'], sslpemfile=config['sslcert'])
try:
- server = BaseHTTPServer.HTTPServer((config['ip'], config['port']), Conf2HTTPHandler)
server.serve_forever()
except:
server.socket.close()