1 files changed, 217 insertions, 0 deletions
diff --git a/minion/server.py b/minion/server.py
new file mode 100755
index 0000000..73ef114
--- /dev/null
+++ b/minion/server.py
@@ -0,0 +1,217 @@
+#!/usr/bin/python
+
+"""
+func
+
+Copyright 2007, Red Hat, Inc
+see AUTHORS
+
+This software may be freely redistributed under the terms of the GNU
+general public license.
+
+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., 675 Mass Ave, Cambridge, MA 02139, USA.
+"""
+
+# standard modules
+import SimpleXMLRPCServer
+import string
+import socket
+import sys
+import traceback
+
+from rhpl.translate import _, N_, textdomain, utf8
+I18N_DOMAIN = "func"
+
+# our modules
+import codes
+import config_data
+import logger
+import module_loader
+import utils
+
+# ======================================================================================
+
+class XmlRpcInterface(object):
+
+    def __init__(self, modules={}, server=None):
+
+        """
+        Constructor.
+        """
+
+        config_obj = config_data.Config()
+        self.config = config_obj.get()
+        self.modules = modules
+        self.logger = logger.Logger().logger
+        self.audit_logger = logger.AuditLogger()
+        self.__setup_handlers()
+
+        # need a reference so we can log ip's, certs, etc
+        self.server = server
+        
+    def __setup_handlers(self):
+
+        """
+        Add RPC functions from each class to the global list so they can be called.
+        """
+
+        self.handlers = {}
+        for x in self.modules.keys():
+            try:
+                self.modules[x].register_rpc(self.handlers, x)
+                self.logger.debug("adding %s" % x)
+            except AttributeError, e:
+                self.logger.warning("module %s not loaded, missing register_rpc method" % self.modules[x])
+
+
+        # internal methods that we do instead of spreading internal goo
+        # all over the modules. For now, at lest -akl
+
+        self.handlers["system.listMethods"] = self.list_methods
+
+    
+    def list_methods(self):
+        return self.handlers.keys()
+
+
+
+    def get_dispatch_method(self, method):
+
+        if method in self.handlers:
+            return FuncApiMethod(self.logger, method, self.handlers[method])
+      
+        else:
+            self.logger.info("Unhandled method call for method: %s " % method)
+            raise codes.InvalidMethodException
+
+    def _dispatch(self, method, params):
+
+        """
+        the SimpleXMLRPCServer class will call _dispatch if it doesn't
+        find a handler method 
+        """
+
+        # Recognize ipython's tab completion calls
+        if method == 'trait_names' or method == '_getAttributeNames':
+            return self.handlers.keys()
+
+        # XXX FIXME - need to figure out how to dig into the server base classes
+        # so we can get client ip, and eventually cert id info -akl
+        self.audit_logger.log_call(method, params)
+
+        return self.get_dispatch_method(method)(*params)
+
+# ======================================================================================
+
+class FuncApiMethod:
+
+    """
+    Used to hold a reference to all of the registered functions.
+    """
+
+    def __init__(self, logger, name, method):
+
+        self.logger = logger
+        self.__method = method
+        self.__name = name
+        
+    def __log_exc(self):
+
+        """
+        Log an exception.
+        """
+
+        (t, v, tb) = sys.exc_info()
+        self.logger.info("Exception occured: %s" % t )
+        self.logger.info("Exception value: %s" % v)
+        self.logger.info("Exception Info:\n%s" % string.join(traceback.format_list(traceback.extract_tb(tb))))
+
+    def __call__(self, *args):
+
+        self.logger.debug("(X) -------------------------------------------")
+
+        try:
+            rc = self.__method(*args)
+        except codes.FuncException, e:
+            self.__log_exc()
+            rc = e
+        except:
+            self.logger.debug("Not a Func-specific exception")
+            self.__log_exc()
+            raise
+        self.logger.debug("Return code for %s: %s" % (self.__name, rc))
+
+        return rc
+
+# ======================================================================================
+
+def serve():
+
+     """
+     Code for starting the XMLRPC service. 
+     FIXME:  make this HTTPS (see RRS code) and make accompanying Rails changes..
+     """
+
+     modules = module_loader.load_modules()
+
+     server =FuncXMLRPCServer(('', 51234))
+     server.logRequests = 0 # don't print stuff to console
+
+     websvc = XmlRpcInterface(modules=modules,server=server)
+     
+     server.register_instance(websvc)
+     server.serve_forever()
+
+# ======================================================================================
+
+class FuncXMLRPCServer(SimpleXMLRPCServer.SimpleXMLRPCServer):
+
+    def __init__(self, args):
+
+       self.allow_reuse_address = True
+       SimpleXMLRPCServer.SimpleXMLRPCServer.__init__(self, args)
+
+# ======================================================================================
+      
+def main(argv):
+
+    """
+    Start things up.
+    """
+
+    modules = module_loader.load_modules()
+
+    print "\n\n\n\n\n"
+    print " WARNING WARNING WARNING"
+    print "DANGER DANGER DANGER"
+    print "\n\n\n\n"
+    print "THERE IS NO AUTHENTICATION IN THIS VERSION"
+    print "DO NOT RUN ON A MACHINE EXPOSED TO ANYONE YOU DO NOT TRUST"
+    print " THEY CAN DO VERY BAD THINGS"
+    print "\n\n\n\n\n"
+    print "Really, don't do that. It is not at all secure at the moment"
+    print "like, at all."
+    print ""
+    print "Seriously.\n\n"
+
+    try:
+        serve()
+    except codes.FuncException, e:
+        print >> sys.stderr, 'error: %s' % e
+        sys.exit(1)
+
+    if "daemon" in sys.argv or "--daemon" in sys.argv:
+        utils.daemonize("/var/run/vf_server.pid")
+    else:
+        print "serving...\n"
+
+       
+# ======================================================================================
+
+if __name__ == "__main__":
+    textdomain(I18N_DOMAIN)
+    main(sys.argv)
+
+