X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=cloudeebus%2Fcloudeebus.py;h=7f1517e87721cf3ab17e064fe51e0e1a7c024615;hb=ab1b6308a612abcc835673c9874b845c94827054;hp=517b05773c87364fb82bc84a764c0fdb1665c6e0;hpb=d308109132243b9549e2cf6a5ede11609d611c2d;p=contrib%2Fcloudeebus.git diff --git a/cloudeebus/cloudeebus.py b/cloudeebus/cloudeebus.py index 517b057..7f1517e 100755 --- a/cloudeebus/cloudeebus.py +++ b/cloudeebus/cloudeebus.py @@ -36,6 +36,8 @@ from autobahn.wamp import exportRpc, WampServerFactory, WampCraServerProtocol from dbus.mainloop.glib import DBusGMainLoop import gobject +import re +import dbus.service gobject.threads_init() from dbus import glib @@ -44,14 +46,16 @@ glib.init_threads() # enable debug log from twisted.python import log - +# XML parser module +from xml.etree.ElementTree import XMLParser ############################################################################### -VERSION = "0.3.1" +VERSION = "0.6.0" OPENDOOR = False CREDENTIALS = {} WHITELIST = [] +SERVICELIST = [] NETMASK = [] ############################################################################### @@ -176,6 +180,253 @@ class DbusCallHandler: +################################################################################ +class ExecCode: + ''' + Execute DynDBusClass generated code + ''' + def __init__(self, globalCtx, localCtx) : + self.exec_string = "" + self.exec_code = None + self.exec_code_valid = 1 + self.indent_level = 0 + self.indent_increment = 1 + self.line = 0 + self.localCtx = localCtx + self.globalCtx = globalCtx + + + def append_stmt(self, stmt) : + self.exec_code_valid = 0 + self.line += 1 + for x in range(0,self.indent_level): + self.exec_string = self.exec_string + ' ' + self.exec_string = self.exec_string + stmt + '\n' + + def indent(self) : + self.indent_level = self.indent_level + self.indent_increment + + def dedent(self) : + self.indent_level = self.indent_level - self.indent_increment + + # compile : Compile exec_string into exec_code using the builtin + # compile function. Skip if already in sync. + def compile(self) : + if not self.exec_code_valid : + self.exec_code = compile(self.exec_string, "", "exec") + self.exec_code_valid = True + + def execute(self) : + if not self.exec_code_valid : + self.compile() + exec(self.exec_code, self.globalCtx, self.localCtx) + + + +################################################################################ +class XmlCbParser: # The target object of the parser + maxDepth = 0 + depth = 0 + def __init__(self, dynDBusClass): + self.dynDBusClass = dynDBusClass + + def start(self, tag, attrib): # Called for each opening tag. + if (tag == 'node'): + return + # Set interface name + if (tag == 'interface'): + self.dynDBusClass.set_interface(attrib['name']) + return + # Set method name + if (tag == 'method'): + self.current = tag + self.dynDBusClass.def_method(attrib['name']) + return + if (tag == 'signal'): + self.current = tag + self.dynDBusClass.def_signal(attrib['name']) + return + + # Set signature (in/out & name) for method + if (tag == 'arg'): + if (self.current == 'method'): + if (attrib.has_key('direction') == False): + attrib['direction'] = "in" + self.dynDBusClass.add_signature(attrib['name'], + attrib['direction'], + attrib['type']) + return + if (self.current == 'signal'): + if (attrib.has_key('name') == False): + attrib['name'] = 'value' + self.dynDBusClass.add_signature(attrib['name'], 'in', + attrib['type']) + return + def end(self, tag): # Called for each closing tag. + if (tag == 'method'): + self.dynDBusClass.add_dbus_method() + self.dynDBusClass.add_body_method() + self.dynDBusClass.end_method() + if (tag == 'signal'): + self.dynDBusClass.add_dbus_signal() + self.dynDBusClass.add_body_signal() + self.dynDBusClass.end_method() + + def data(self, data): + pass # We do not need to do anything with data. + def close(self): # Called when all data has been parsed. + return self.maxDepth + + + +############################################################################### +def createClassName(objectPath): + return re.sub('/', '_', objectPath[1:]) + +################################################################################ +class DynDBusClass(): + def __init__(self, className, globalCtx, localCtx): + self.xmlCB = XmlCbParser(self) + self.signature = {} + self.class_code = ExecCode(globalCtx, localCtx) + self.class_code.indent_increment = 4 + self.class_code.append_stmt("import dbus") + self.class_code.append_stmt("\n") + self.class_code.append_stmt("class " + className + "(dbus.service.Object):") + self.class_code.indent() + + ## Overload of __init__ method + self.def_method("__init__") + self.add_method("bus, callback=None, objPath='/sample', srvName='org.cloudeebus'") + self.add_stmt("self.bus = bus") + self.add_stmt("self.objPath = objPath") + self.add_stmt("self.srvName = srvName") + self.add_stmt("self.callback = callback") + self.add_stmt("dbus.service.Object.__init__(self, conn=bus, bus_name=srvName)") + self.end_method() + + ## Create 'add_to_connection' method + self.def_method("add_to_connection") + self.add_method("connection=None, path=None") + self.add_stmt("dbus.service.Object.add_to_connection(self, connection=self.bus, path=self.objPath)") + self.end_method() + + ## Create 'remove_from_connection' method + self.def_method("remove_from_connection") + self.add_method("connection=None, path=None") + self.add_stmt("dbus.service.Object.remove_from_connection(self, connection=None, path=self.objPath)") + self.end_method() + + def createDBusServiceFromXML(self, xml): + self.parser = XMLParser(target=self.xmlCB) + self.parser.feed(xml) + self.parser.close() + + def set_interface(self, ifName): + self.ifName = ifName + + def def_method(self, methodName): + self.methodToAdd = methodName + self.signalToAdd = None + self.args_str = str() + self.signature = {} + self.signature['name'] = str() + self.signature['in'] = str() + self.signature['out'] = str() + + def def_signal(self, signalName): + self.methodToAdd = None + self.signalToAdd = signalName + self.args_str = str() + self.signature = {} + self.signature['name'] = str() + self.signature['in'] = str() + self.signature['out'] = str() + + def add_signature(self, name, direction, signature): + if (direction == 'in'): + self.signature['in'] += signature + if (self.signature['name'] != str()): + self.signature['name'] += ", " + self.signature['name'] += name + if (direction == 'out'): + self.signature['out'] = signature + + def add_method(self, args = None, async_success_cb = None, async_err_cb = None): + async_cb_str = str() + if (self.methodToAdd != None): + name = self.methodToAdd + else: + name = self.signalToAdd + if (args != None): + self.args_str = args + if (async_success_cb != None): + async_cb_str = async_success_cb + if (async_err_cb != None): + if (async_cb_str != str()): + async_cb_str += ", " + async_cb_str += async_err_cb + + parameters = self.args_str + if (async_cb_str != str()): + if (parameters != str()): + parameters += ", " + parameters +=async_cb_str + + if (parameters != str()): + self.class_code.append_stmt("def " + name + "(self, %s):" % parameters) + else: + self.class_code.append_stmt("def " + name + "(self):") + self.class_code.indent() + + def end_method(self): + self.class_code.append_stmt("\n") + self.class_code.dedent() + + def add_dbus_method(self): + decorator = '@dbus.service.method("' + self.ifName + '"' + if (self.signature.has_key('in') and self.signature['in'] != str()): + decorator += ", in_signature='" + self.signature['in'] + "'" + if (self.signature.has_key('out') and self.signature['out'] != str()): + decorator += ", out_signature='" + self.signature['out'] + "'" + decorator += ", async_callbacks=('dbus_async_cb', 'dbus_async_err_cb')" + decorator += ")" + self.class_code.append_stmt(decorator) + if (self.signature.has_key('name') and self.signature['name'] != str()): + self.add_method(self.signature['name'], async_success_cb='dbus_async_cb', async_err_cb='dbus_async_err_cb') + else: + self.add_method(async_success_cb='dbus_async_cb', async_err_cb='dbus_async_err_cb') + + def add_dbus_signal(self): + decorator = '@dbus.service.signal("' + self.ifName + '"' + if (self.signature.has_key('in') and self.signature['in'] != str()): + decorator += ", signature='" + self.signature['in'] + "'" + decorator += ")" + self.class_code.append_stmt(decorator) + if (self.signature.has_key('name') and self.signature['name'] != str()): + self.add_method(self.signature['name']) + else: + self.add_method() + + def add_body_method(self): + if (self.methodToAdd != None): + if (self.args_str != str()): + self.class_code.append_stmt("self.callback(self.srvName,'" + self.methodToAdd + "', self.objPath, '" + self.ifName + "', " + "dbus_async_cb, dbus_async_err_cb, %s)" % self.args_str) + else: + self.class_code.append_stmt("self.callback(self.srvName,'" + self.methodToAdd + "', self.objPath, '" + self.ifName + "', " + "dbus_async_cb, dbus_async_err_cb)") + + def add_body_signal(self): + self.class_code.append_stmt("return") ## TODO: Remove and fix with code ad hoc + self.class_code.append_stmt("\n") + + def add_stmt(self, stmt) : + self.class_code.append_stmt(stmt) + + def declare(self) : + self.class_code.execute() + + + ############################################################################### class CloudeebusService: ''' @@ -185,10 +436,27 @@ class CloudeebusService: self.permissions = {}; self.permissions['permissions'] = permissions['permissions'] self.permissions['authextra'] = permissions['authextra'] + self.permissions['services'] = permissions['services'] self.proxyObjects = {} self.proxyMethods = {} self.pendingCalls = [] - + self.dynDBusClasses = {} # DBus class source code generated dynamically (a list because one by classname) + self.services = {} # DBus service created + self.serviceAgents = {} # Instantiated DBus class previously generated dynamically, for now, one by classname + self.servicePendingCalls = {} # JS methods called (and waiting for a Success/error response), containing 'methodId', (successCB, errorCB) + self.localCtx = locals() + self.globalCtx = globals() + + self.patternDbus = re.compile('^dbus\.(\w+)') + self.patternDbusBoolean = re.compile('^dbus.Boolean\((\w+)\)$') + self.patternDbusByte = re.compile('^dbus.Byte\((\d+)\)$') + self.patternDbusInt16 = re.compile('^dbus.Int16\((\d+)\)$') + self.patternDbusInt32 = re.compile('^dbus.Int32\((\d+)\)$') + self.patternDbusInt64 = re.compile('^dbus.Int64\((\d+)\)$') + self.patternDbusUInt16 = re.compile('^dbus.UInt16\((\d+)\)$') + self.patternDbusUInt32 = re.compile('^dbus.UInt32\((\d+)\)$') + self.patternDbusUInt64 = re.compile('^dbus.UInt64\((\d+)\)$') + self.patternDbusDouble = re.compile('^dbus.Double\((\d+\.\d+)\)$') def proxyObject(self, busName, serviceName, objectName): ''' @@ -214,6 +482,45 @@ class CloudeebusService: self.proxyMethods[id] = obj.get_dbus_method(methodName, interfaceName) return self.proxyMethods[id] + def decodeArgs( self, args ): + if isinstance( args, list ): + newArgs = [] + for arg in args: + newArgs.append( self.decodeArgs( arg )) + return newArgs + elif isinstance( args, dict ): + newDict = {} + for key, value in args.iteritems(): + key = self.decodeArgs( key ) + newValue = self.decodeArgs( value ) + newDict[key] = newValue + return newDict + elif isinstance( args, basestring ): + newArg = self.decodeDbusString( args ) + return newArg + else: + return args + + def decodeDbusString( self, dbusString ): + matchDbus = self.patternDbus.match( dbusString ) + + if not matchDbus: + return dbusString + + + result = { + "Boolean" : lambda x : dbus.Boolean( self.patternDbusBoolean.match( x ).group( 1 ).lower() in ("yes", "true", "t", "1")), + "Byte" : lambda x : dbus.Byte( int( self.patternDbusByte.match( x ).group( 1 ))), + "Int16" : lambda x : dbus.Int16( self.patternDbusInt16.match( x ).group( 1 )), + "Int32" : lambda x : dbus.Int32( self.patternDbusInt32.match( x ).group( 1 )), + "Int64" : lambda x : dbus.Int64( self.patternDbusInt64.match( x ).group( 1 )), + "UInt16" : lambda x : dbus.UInt16( self.patternDbusUInt16.match( x ).group( 1 )), + "UInt32" : lambda x : dbus.UInt32( self.patternDbusUInt32.match( x ).group( 1 )), + "UInt64" : lambda x : dbus.UInt64( self.patternDbusUInt64.match( x ).group( 1 )), + "Double" : lambda x : dbus.Double( self.patternDbusDouble.match( x ).group( 1 )) + }[matchDbus.group(1)](dbusString) + + return result @exportRpc def dbusRegister(self, list): @@ -255,7 +562,9 @@ class CloudeebusService: # parse JSON arg list args = [] if len(list) == 6: - args = json.loads(list[5]) + jsonArgs = json.loads(list[5]) + if jsonArgs: + args = self.decodeArgs( jsonArgs ) # get dbus proxy method method = self.proxyMethod(*list[0:5]) @@ -267,6 +576,145 @@ class CloudeebusService: @exportRpc + def emitSignal(self, list): + ''' + arguments: agentObjectPath, signalName, args (to emit) + ''' + objectPath = list[0] + className = re.sub('/', '_', objectPath[1:]) + signalName = list[1] + args = [] + jsonArgs = json.loads(list[2]) + print "JSON Arguments:", jsonArgs + if jsonArgs: + args = self.decodeArgs( jsonArgs ) + print "Decoded Arguments: ", args + + if (self.serviceAgents.has_key(className) == True): + exe_str = "self.serviceAgents['"+ className +"']."+ signalName + "(" + if len(args) > 0: + exe_str += json.dumps(args[0]) + for idx in args[1:]: + exe_str += "," + json.dumps(idx) + exe_str += ")" + eval(exe_str, self.globalCtx, self.localCtx) + else: + raise Exception("No object path " + objectPath) + + @exportRpc + def returnMethod(self, list): + ''' + arguments: methodId, callIndex, success (=true, error otherwise), result (to return) + ''' + methodId = list[0] + callIndex = list[1] + success = list[2] + result = list[3] + if (self.servicePendingCalls.has_key(methodId)): + cb = self.servicePendingCalls[methodId]['calls'][callIndex] + if cb is None: + raise Exception("No pending call " + str(callIndex) + " for methodID " + methodId) + if (success): + successCB = cb["successCB"] + if (result != None): + successCB(result) + else: + successCB() + else: + errorCB = cb["errorCB"] + if (result != None): + errorCB(result) + else: + errorCB() + self.servicePendingCalls[methodId]['calls'][callIndex] = None + self.servicePendingCalls[methodId]['count'] = self.servicePendingCalls[methodId]['count'] - 1 + if self.servicePendingCalls[methodId]['count'] == 0: + del self.servicePendingCalls[methodId] + else: + raise Exception("No methodID " + methodId) + + def srvCB(self, srvName, name, objPath, ifName, async_succes_cb, async_error_cb, *args): + methodId = srvName + "#" + objPath + "#" + ifName + "#" + name + cb = { 'successCB': async_succes_cb, + 'errorCB': async_error_cb} + if methodId not in self.servicePendingCalls: + self.servicePendingCalls[methodId] = {'count': 0, 'calls': []} + + try: + pendingCallStr = json.dumps({'callIndex': len(self.servicePendingCalls[methodId]['calls']), 'args': args}) + except Exception, e: + args = eval( str(args).replace("dbus.Byte", "dbus.Int16") ) + pendingCallStr = json.dumps({'callIndex': len(self.servicePendingCalls[methodId]['calls']), 'args': args}) + + self.servicePendingCalls[methodId]['calls'].append(cb) + self.servicePendingCalls[methodId]['count'] = self.servicePendingCalls[methodId]['count'] + 1 + factory.dispatch(methodId, pendingCallStr) + + @exportRpc + def serviceAdd(self, list): + ''' + arguments: busName, srvName + ''' + busName = list[0] + self.bus = cache.dbusConnexion( busName ) + srvName = list[1] + if not OPENDOOR and (SERVICELIST == [] or SERVICELIST != [] and self.permissions['services'] == None): + SERVICELIST.index(srvName) + + if (self.services.has_key(srvName) == False): + self.services[srvName] = dbus.service.BusName(name = srvName, bus = self.bus) + return srvName + + @exportRpc + def serviceRelease(self, list): + ''' + arguments: busName, srvName + ''' + srvName = list[0] + if (self.services.has_key(srvName) == True): + self.services.pop(srvName) + return srvName + else: + raise Exception(srvName + " does not exist") + + @exportRpc + def serviceAddAgent(self, list): + ''' + arguments: objectPath, xmlTemplate + ''' + srvName = list[0] + agentObjectPath = list[1] + xmlTemplate = list[2] + className = createClassName(agentObjectPath) + if (self.dynDBusClasses.has_key(className) == False): + self.dynDBusClasses[className] = DynDBusClass(className, self.globalCtx, self.localCtx) + self.dynDBusClasses[className].createDBusServiceFromXML(xmlTemplate) + self.dynDBusClasses[className].declare() + + ## Class already exist, instanciate it if not already instanciated + if (self.serviceAgents.has_key(className) == False): + self.serviceAgents[className] = eval(className + "(self.bus, callback=self.srvCB, objPath='" + agentObjectPath + "', srvName='" + srvName + "')", self.globalCtx, self.localCtx) + + self.serviceAgents[className].add_to_connection() + return (agentObjectPath) + + @exportRpc + def serviceDelAgent(self, list): + ''' + arguments: objectPath, xmlTemplate + ''' + agentObjectPath = list[0] + className = createClassName(agentObjectPath) + + if (self.serviceAgents.has_key(className)): + self.serviceAgents[className].remove_from_connection() + self.serviceAgents.pop(className) + else: + raise Exception(agentObjectPath + " doesn't exist!") + + return (agentObjectPath) + + @exportRpc def getVersion(self): ''' return current version string @@ -291,7 +739,8 @@ class CloudeebusServerProtocol(WampCraServerProtocol): def getAuthPermissions(self, key, extra): return {'permissions': extra.get("permissions", None), - 'authextra': extra.get("authextra", None)} + 'authextra': extra.get("authextra", None), + 'services': extra.get("services", None)} def getAuthSecret(self, key): secret = CREDENTIALS.get(key, None) @@ -317,8 +766,13 @@ class CloudeebusServerProtocol(WampCraServerProtocol): if key is None: raise Exception("Authentication failed") # check permissions, array.index throws exception - for req in permissions['permissions']: + if (permissions['permissions'] != None): + for req in permissions['permissions']: WHITELIST.index(req); + # check allowed service creation, array.index throws exception + if (permissions['services'] != None): + for req in permissions['services']: + SERVICELIST.index(req); # create cloudeebus service instance self.cloudeebusService = CloudeebusService(permissions) # register it for RPC @@ -352,7 +806,9 @@ if __name__ == '__main__': parser.add_argument('-c', '--credentials', help='path to credentials file') parser.add_argument('-w', '--whitelist', - help='path to whitelist file') + help='path to whitelist file (DBus services to use)') + parser.add_argument('-s', '--servicelist', + help='path to servicelist file (DBus services to export)') parser.add_argument('-n', '--netmask', help='netmask,IP filter (comma separated.) eg. : -n 127.0.0.1,192.168.2.0/24,10.12.16.0/255.255.255.0') @@ -377,6 +833,11 @@ if __name__ == '__main__': WHITELIST = json.load(jfile) jfile.close() + if args.servicelist: + jfile = open(args.servicelist) + SERVICELIST = json.load(jfile) + jfile.close() + if args.netmask: iplist = args.netmask.split(",") for ip in iplist: @@ -388,7 +849,7 @@ if __name__ == '__main__': ipAllowed = ip mask = "255.255.255.255" NETMASK.append( {'ipAllowed': ipV4ToHex(ipAllowed), 'mask' : ipV4ToHex(mask)} ) - + uri = "ws://localhost:" + args.port factory = WampServerFactory(uri, debugWamp = args.debug)