Code

Config files and cmd line options for daemons
[nagixsc.git] / nagixsc_conf2http.py
index 96d86ac99a5e9153f77958e58b758cbcbf13acf6..941fa1ccb122438ac01fd8b68df70dbed6d442bd 100755 (executable)
 #!/usr/bin/python
 
-import cherrypy
+import BaseHTTPServer
+import ConfigParser
+import base64
+import optparse
 import os
 import re
 import subprocess
+import sys
 
-config = {     'server.socket_host':           '0.0.0.0',
-                       'server.socket_port':           15666,
-                       'log.screen':                           False,
-                       'log.access_file':                      None,
-                       'log.error_file':                       None,
-               }
+try:
+       from hashlib import md5
+except ImportError:
+       from md5 import md5
 
-users = {      'nagixsc':              '019b0966d98fb71d1a4bc4ca0c81d5cc',             # PW: nagixsc
-               }
+##############################################################################
 
-CONFDIR='./examples'
-C2X='./nagixsc_conf2xml.py'
-class Conf2CGI:
-       def default(*args, **kwargs):
-               cmdline = C2X
+parser = optparse.OptionParser()
 
-               if len(args) >= 5:
-                       print 'Ignoring arguments: ', args[4:]
+parser.add_option('-c', '', dest='cfgfile', help='Config file')
 
-               if len(args) >= 4:
-                       c_service = args[3]
+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)
+
+if cfg_list == []:
+       print 'Config file "%s" could not be read!' % options.cfgfile
+       sys.exit(1)
+
+config = {}
+try:
+       config['ip']   = cfgread.get('server', 'ip')
+       config['port'] = cfgread.getint('server', 'port')
+
+       config['conf_dir']         = cfgread.get('server', 'conf_dir')
+       config['conf2xml_cmdline'] = cfgread.get('server', 'conf2xml_cmdline')
+
+except ConfigParser.NoOptionError, e:
+       print 'Config file error: %s ' % e
+       sys.exit(1)
+
+users = {}
+for u in cfgread.options('users'):
+       users[u] = cfgread.get('users', u)
+
+##############################################################################
+
+class Conf2HTTPHandler(BaseHTTPServer.BaseHTTPRequestHandler):
+
+       def http_error(self, code, output):
+               self.send_response(code)
+               self.send_header('Content-Type', 'text/plain')
+               self.end_headers()
+               self.wfile.write(output)
+               return
+
+
+       def do_GET(self):
+               cmdline = config['conf2xml_cmdline']
+
+               path = self.path.split('/')
+
+               # Check Basic Auth
+               try:
+                       authdata = base64.b64decode(self.headers['Authorization'].split(' ')[1]).split(':')
+                       if not users[authdata[0]] == md5(authdata[1]).hexdigest():
+                               raise Exception
+               except:
+                       self.send_response(401)
+                       self.send_header('WWW-Authenticate', 'Basic realm="Nag(ix)SC Pull"')
+                       self.send_header('Content-Type', 'text/plain')
+                       self.end_headers()
+                       self.wfile.write('Sorry! No action without login!')
+                       return
+
+
+               if len(path) >= 4:
+                       service = path[3]
                else:
-                       c_service = ''
+                       service = ''
 
-               if len(args) >= 3:
-                       c_host = args[2]
+               if len(path) >= 3:
+                       host = path[2]
                else:
-                       c_host = ''
+                       host = ''
 
-               if len(args) >= 2:
-                       c_configfile = args[1]
+               if len(path) >= 2:
+                       configfile = path[1]
                else:
-                       c_configfile = ''
-                       print 'No config file specified!'
-
-               if c_configfile:
-                       cherrypy.lib.auth.basic_auth('Nag(ix)SC HTTP', users)
-
-                       if re.search('\.\.', c_configfile):
-                               return 'Found ".." in config file name'
-                       if not re.search('^[a-zA-Z0-9-_\.]+$', c_configfile):
-                               return 'Config file name contains invalid characters'
-                       cmdline += ' -c ' + os.path.join(CONFDIR, c_configfile)
-
-                       if c_host:
-                               cmdline += ' -H %s' % c_host
-                               if c_service:
-                                       cmdline += ' -D %s' % c_service
-                       try:
-                               cmd     = subprocess.Popen(cmdline.split(' '), stdout=subprocess.PIPE)
-                               output  = cmd.communicate()[0].rstrip()
-                       except OSError:
-                               return 'Could not execute "%s"' % cmdline
-
-                       cherrypy.response.headers['Content-Type'] = 'text/xml'
-                       return output
+                       configfile =''
+
+               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):
+                       self.http_error(500, 'Config file name contains invalid characters')
+                       return
+
+               if configfile:
+                       configfile += '.conf'
+                       cmdline    += ' -c ' + os.path.join(config['conf_dir'], configfile)
+
+               if host:
+                       cmdline += ' -H %s' % host
+                       if service:
+                               cmdline += ' -D %s' % 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)
+                       return
+
+               if retcode == 0:
+                       self.send_response(200)
+                       self.send_header('Content-Type', 'text/xml')
+                       self.end_headers()
+                       self.wfile.write(output)
                else:
-                       return '42'
+                       self.http_error(500, output)
+
+               return
+
+
 
-       default.exposed = True
+def main():
+       try:
+               server = BaseHTTPServer.HTTPServer((config['ip'], config['port']), Conf2HTTPHandler)
+               server.serve_forever()
+       except:
+               server.socket.close()
 
-cherrypy.config.update(config)
-cherrypy.tree.mount(Conf2CGI(), '')
-cherrypy.quickstart(config=config)
+if __name__ == '__main__':
+       main()