5 from PLC.Faults import *
6 from PLC.Method import Method
7 from PLC.Parameter import Parameter, Mixed
8 from PLC.Nodes import Node, Nodes
9 from PLC.Interfaces import Interface, Interfaces
10 from PLC.Auth import Auth
12 class GenerateNodeConfFile(Method):
14 Creates a new node configuration file if all network settings are
15 present. This function will generate a new node key for the
16 specified node, effectively invalidating any old configuration
19 Non-admins can only generate files for nodes at their sites.
21 Returns the contents of the file if successful, faults otherwise.
24 roles = ['admin', 'pi', 'tech']
28 Mixed(Node.fields['node_id'],
29 Node.fields['hostname']),
30 Parameter(bool, "True if you want to regenerate node key")
33 returns = Parameter(str, "Node configuration file")
35 def call(self, auth, node_id_or_hostname, regenerate_node_key = True):
36 # Get node information
37 nodes = Nodes(self.api, [node_id_or_hostname])
39 raise PLCInvalidArgument, "No such node"
42 if node['peer_id'] is not None:
43 raise PLCInvalidArgument, "Not a local node"
45 # If we are not an admin, make sure that the caller is a
46 # member of the site at which the node is located.
47 if 'admin' not in self.caller['roles']:
48 if node['site_id'] not in self.caller['site_ids']:
49 raise PLCPermissionDenied, "Not allowed to generate a configuration file for that node"
51 # Get interfaces for this node
53 interfaces = Interfaces(self.api, node['interface_ids'])
54 for interface in interfaces:
55 if interface['is_primary']:
59 raise PLCInvalidArgument, "No primary network configured"
61 # Split hostname into host and domain parts
62 parts = node['hostname'].split(".", 1)
64 raise PLCInvalidArgument, "Node hostname is invalid"
68 if regenerate_node_key:
69 # Generate 32 random bytes
70 bytes = random.sample(xrange(0, 256), 32)
71 # Base64 encode their string representation
72 node['key'] = base64.b64encode("".join(map(chr, bytes)))
73 # XXX Boot Manager cannot handle = in the key
74 node['key'] = node['key'].replace("=", "")
78 # Generate node configuration file suitable for BootCD
81 file += 'NODE_ID="%d"\n' % node['node_id']
82 file += 'NODE_KEY="%s"\n' % node['key']
85 file += 'NET_DEVICE="%s"\n' % primary['mac'].lower()
87 file += 'IP_METHOD="%s"\n' % primary['method']
89 if primary['method'] == 'static':
90 file += 'IP_ADDRESS="%s"\n' % primary['ip']
91 file += 'IP_GATEWAY="%s"\n' % primary['gateway']
92 file += 'IP_NETMASK="%s"\n' % primary['netmask']
93 file += 'IP_NETADDR="%s"\n' % primary['network']
94 file += 'IP_BROADCASTADDR="%s"\n' % primary['broadcast']
95 file += 'IP_DNS1="%s"\n' % primary['dns1']
96 file += 'IP_DNS2="%s"\n' % (primary['dns2'] or "")
98 file += 'HOST_NAME="%s"\n' % host
99 file += 'DOMAIN_NAME="%s"\n' % domain
101 for interface in interfaces:
102 if interface['method'] == 'ipmi':
103 file += 'IPMI_ADDRESS="%s"\n' % interface['ip']
105 file += 'IPMI_MAC="%s"\n' % interface['mac'].lower()