X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=api.py;h=96fafe7942cd64d2b2cc952b6eefd56f371fe1d1;hb=b6e806fc73f0fa8f2526cc7d1e62245db814d725;hp=6baaf3e0953d1a0fcc0bd684f07bbe211e392126;hpb=a04bb0c32b2a03b943eb0785cd0cdaf8b2c88d90;p=nodemanager.git diff --git a/api.py b/api.py index 6baaf3e..96fafe7 100644 --- a/api.py +++ b/api.py @@ -54,6 +54,7 @@ def Ticket(tkt): data = ticket.verify(tkt) if data != None: deliver_ticket(data) + logger.log('Got ticket') except Exception, err: raise xmlrpclib.Fault(102, 'Ticket error: ' + str(err)) @@ -134,7 +135,9 @@ class APIRequestHandler(SimpleXMLRPCServer.SimpleXMLRPCRequestHandler): api_method_list.sort() raise xmlrpclib.Fault(100, 'Invalid API method %s. Valid choices are %s' % (method_name, ', '.join(api_method_list))) expected_nargs = nargs_dict[method_name] - if len(args) != expected_nargs: raise xmlrpclib.Fault(101, 'Invalid argument count: got %d, expecting %d.' % (len(args), expected_nargs)) + if len(args) != expected_nargs: + raise xmlrpclib.Fault(101, 'Invalid argument count: got %d, expecting %d.' % (len(args), + expected_nargs)) else: # Figure out who's calling. # XXX - these ought to be imported directly from some .h file @@ -143,11 +146,15 @@ class APIRequestHandler(SimpleXMLRPCServer.SimpleXMLRPCRequestHandler): ucred = self.request.getsockopt(socket.SOL_SOCKET, SO_PEERCRED, sizeof_struct_ucred) xid = struct.unpack('3i', ucred)[2] caller_name = pwd.getpwuid(xid)[0] - if method_name not in ('Help', 'Ticket', 'GetXIDs', 'GetSSHKeys'): + if method_name not in ('ReCreate', 'Help', 'Ticket', 'GetXIDs', 'GetSSHKeys'): target_name = args[0] target_rec = database.db.get(target_name) - if not (target_rec and target_rec['type'].startswith('sliver.')): raise xmlrpclib.Fault(102, 'Invalid argument: the first argument must be a sliver name.') - if not (caller_name in (args[0], 'root') or (caller_name, method_name) in target_rec['delegations'] or (caller_name == 'utah_elab_delegate' and target_name.startswith('utah_'))): raise xmlrpclib.Fault(108, 'Permission denied.') + if not (target_rec and target_rec['type'].startswith('sliver.')): + raise xmlrpclib.Fault(102, 'Invalid argument: the first argument must be a sliver name.') + if not (caller_name, method_name) in target_rec['delegations']: + # or (caller_name == 'utah_elab_delegate' and target_name.startswith('utah_'))): + raise xmlrpclib.Fault(108, 'Permission denied.') + result = method(target_rec, *args[1:]) else: result = method(*args) if result == None: result = 1