6 from sfa.util.faults import *
7 from sfa.util.misc import *
8 from sfa.util.method import Method
9 from sfa.util.parameter import Parameter, Mixed
10 from sfa.trust.auth import Auth
11 from sfa.util.genitable import *
12 from sfa.trust.certificate import Keypair
14 class get_key(Method):
16 Generate a new keypair and gid for requesting caller (component).
17 @return 1 If successful
20 interfaces = ['registry']
24 returns = Parameter(int, "1 if successful, faults otherwise")
27 # verify that the callers's ip address exist in the db and is an inteface
28 # for a node in the db
29 (ip, port) = self.api.remote_addr
30 interfaces = self.api.plshell.GetInterfaces(self.api.plauth, {'ip': ip}, ['node_id'])
32 raise NonExistingRecord("no such ip %(ip)s" % locals())
33 nodes = self.api.plshell.GetNodes(self.api.plauth, [interfaces[0]['node_id']], ['node_id', 'hostname'])
35 raise NonExistingRecord("no such node using ip %(ip)s" % locals())
38 # look up the sfa record
40 records = table.findObjects({'type': 'node', 'pointer': node['node_id']})
42 raise RecordNotFound("pointer:" + str(node['node_id']))
45 # generate a new keypair and gid
47 pkey = Keypair(create=True)
48 gid_object = self.api.auth.hierarchy.create_gid(record['hrn'], uuid, pkey)
49 gid = gid_object.save_to_string(save_parents=True)
56 # attempt the scp the key
57 # and gid onto the node
58 # this will only work for planetlab based components
59 (kfd, key_filename) = tempfile.mkstemp()
60 (gfd, gid_filename) = tempfile.mkstemp()
61 pkey.save_to_file(key_filename)
62 gid_object.save_to_file(gid_filename, save_parents=True)
63 host = node['hostname']
64 key_dest="/etc/sfa/node.key"
65 gid_dest="/etc/sfa/node.gid"
67 identity = "/etc/sfa/root_ssh_key"
68 scp_options=" -i %(identity)s " % locals()
69 scp_options+="-o StrictHostKeyChecking=no " % locals()
70 scp_key_command="%(scp)s %(scp_options)s %(key_filename)s root@%(host)s:%(key_dest)s" %\
72 scp_gid_command="%(scp)s %(scp_options)s %(gid_filename)s root@%(host)s:%(gid_dest)s" %\
75 all_commands = [scp_key_command, scp_gid_command]
77 for command in all_commands:
78 (status, output) = commands.getstatusoutput(command)
80 raise Exception, output
82 for filename in [key_filename, gid_filename]: