# $Id$
+# $URL$
import random
import base64
import os
from PLC.Nodes import Node, Nodes
from PLC.Interfaces import Interface, Interfaces
from PLC.InterfaceTags import InterfaceTag, InterfaceTags
+from PLC.NodeTags import NodeTag, NodeTags
# could not define this in the class..
# create a dict with the allowed actions for each type of node
allowed_actions = {
- 'regular' : [ 'node-preview',
+ 'regular' : [ 'node-preview',
'node-floppy',
'node-iso',
'node-usb',
'dummynet-iso',
'dummynet-usb',
],
- }
+ }
# compute a new key
-# xxx used by GetDummyBoxMedium
def compute_key():
# Generate 32 random bytes
bytes = random.sample(xrange(0, 256), 32)
This method is a redesign based on former, supposedly dedicated,
AdmGenerateNodeConfFile
- As compared with its ancestor, this method provides a much more detailed
+ As compared with its ancestor, this method provides a much more
detailed interface, that allows to
(*) either just preview the node config file -- in which case
the node key is NOT recomputed, and NOT provided in the output
(*) or just provide the generic ISO or USB boot images
in which case of course the node_id_or_hostname parameter is not used
- action is expected among the following string constants
+ action is expected among the following string constants according the
+ node type value:
+
+ for a 'regular' node:
(*) node-preview
(*) node-floppy
(*) node-iso
(*) generic-iso
(*) generic-usb
+ for a 'dummynet' node:
+ (*) node-preview
+ (*) dummynet-iso
+ (*) dummynet-usb
+
Apart for the preview mode, this method generates a new node key for the
specified node, effectively invalidating any old boot medium.
Options: an optional array of keywords.
options are not supported for generic images
- Currently supported are
+ options are not supported for dummynet boxes
+ Currently supported are
- 'partition' - for USB actions only
- 'cramfs'
- 'serial' or 'serial:<console_spec>'
- - 'no-hangcheck'
console_spec (or 'default') is passed as-is to bootcd/build.sh
it is expected to be a colon separated string denoting
tty - baudrate - parity - bits
e.g. ttyS0:115200:n:8
-
+ - 'variant:<variantname>'
+ passed to build.sh as -V <variant>
+ variants are used to run a different kernel on the bootCD
+ see kvariant.sh for how to create a variant
+ - 'no-hangcheck' - disable hangcheck
+
+ Tags: the following tags are taken into account when attached to the node:
+ 'serial', 'cramfs', 'kvariant', 'kargs', 'no-hangcheck'
+
Security:
- Non-admins can only generate files for nodes at their sites.
- Non-admins, when they provide a filename, *must* specify it in the %d area
returns = Parameter(str, "Node boot medium, either inlined, or filename, depending on the filename parameter")
+ # define globals for regular nodes, override later for other types
BOOTCDDIR = "/usr/share/bootcd-@NODEFAMILY@/"
BOOTCDBUILD = "/usr/share/bootcd-@NODEFAMILY@/build.sh"
GENERICDIR = "/var/www/html/download-@NODEFAMILY@/"
# This function will create the configuration file a node
# composed by:
# - a common part, regardless of the 'node_type' tag
- # - XXX a special part, depending on the 'snode_type' tag value.
+ # - XXX a special part, depending on the 'node_type' tag value.
def floppy_contents (self, node, renew_key):
# Do basic checks
( host, domain ) = self.split_hostname (node)
- # renew the key and save it on the database
+ # renew the key and save it on the database
if renew_key:
node['key'] = compute_key()
node.sync()
if category_settings:
file += '### Category : %s\n'%category
for setting in category_settings:
- file += '%s_%s="%s"\n'%(category.upper(),setting['name'].upper(),setting['value'])
+ file += '%s_%s="%s"\n'%(category.upper(),setting['tagname'].upper(),setting['value'])
for interface in interfaces:
if interface['method'] == 'ipmi':
return filename
+ # Build the command line to be executed
+ # according the node type
+ def build_command(self, node_type, build_sh_spec, node_image, type, floppy_file, log_file):
+
+ command = ""
+
+ # regular node, make build's arguments
+ # and build the full command line to be called
+ if node_type == 'regular':
+
+ build_sh_options=""
+ if "cramfs" in build_sh_spec:
+ type += "_cramfs"
+ if "serial" in build_sh_spec:
+ build_sh_options += " -s %s"%build_sh_spec['serial']
+ if "variant" in build_sh_spec:
+ build_sh_options += " -V %s"%build_sh_spec['variant']
+
+ for karg in build_sh_spec['kargs']:
+ build_sh_options += ' -k "%s"'%karg
+
+ log_file="%s.log"%node_image
+
+ command = '%s -f "%s" -o "%s" -t "%s" %s &> %s' % (self.BOOTCDBUILD,
+ floppy_file,
+ node_image,
+ type,
+ build_sh_options,
+ log_file)
+ # dummynet node
+ elif node_type == 'dummynet':
+ # the build script expect the following parameters:
+ # the package base directory
+ # the working directory
+ # the full path of the configuration file
+ # the name of the resulting image file
+ # the type of the generated image
+ # the name of the log file
+ command = "%s -b %s -w %s -f %s -o %s -t %s -l %s" \
+ % (self.BOOTCDBUILD, self.BOOTCDDIR, self.WORKDIR,
+ floppy_file, node_image, type, log_file)
+ command = "touch %s %s; echo 'dummynet build script not yet supported'" \
+ % (log_file, node_image)
+
+ if self.DEBUG:
+ print "The build command line is %s" % command
+
+ return command
+
def call(self, auth, node_id_or_hostname, action, filename, options = []):
self.trash=[]
suffix=".txt"
type = "txt"
- # handle / caconicalize options
+ # check for node existence and get node_type
+ nodes = Nodes(self.api, [node_id_or_hostname])
+ if not nodes:
+ raise PLCInvalidArgument, "No such node %r"%node_id_or_hostname
+ node = nodes[0]
+
+ if self.DEBUG: print "%s required on node %s. Node type is: %s" \
+ % (action, node['node_id'], node['node_type'])
+
+ # check the required action against the node type
+ node_type = node['node_type']
+ if action not in allowed_actions[node_type]:
+ raise PLCInvalidArgument, "Action %s not valid for %s nodes, valid actions are %s" \
+ % (action, node_type, "|".join(allowed_actions[node_type]))
+
+ # handle / canonicalize options
if type == "txt":
if options:
raise PLCInvalidArgument, "Options are not supported for node configs"
else:
# create a dict for build.sh
build_sh_spec={'kargs':[]}
+ # use node tags as defaults
+ # check for node tag equivalents
+ tags = NodeTags(self.api,
+ {'node_id': node['node_id'],
+ 'tagname': ['serial', 'cramfs', 'kvariant', 'kargs', 'no-hangcheck']},
+ ['tagname', 'value'])
+ if tags:
+ for tag in tags:
+ if tag['tagname'] == 'serial':
+ build_sh_spec['serial'] = tag['value']
+ if tag['tagname'] == 'cramfs':
+ build_sh_spec['cramfs'] = True
+ if tag['tagname'] == 'kvariant':
+ build_sh_spec['variant'] = tag['value']
+ if tag['tagname'] == 'kargs':
+ build_sh_spec['kargs'].append(tag['value'].split())
+ if tag['tagname'] == 'no-hangcheck':
+ build_sh_spec['kargs'].append('hcheck_reboot0')
+ # then options can override tags
for option in options:
if option == "cramfs":
build_sh_spec['cramfs']=True
build_sh_spec['serial']='default'
elif option.find("serial:") == 0:
build_sh_spec['serial']=option.replace("serial:","")
+ elif option.find("variant:") == 0:
+ build_sh_spec['variant']=option.replace("variant:","")
elif option == "no-hangcheck":
build_sh_spec['kargs'].append('hcheck_reboot0')
else:
raise PLCInvalidArgument, "unknown option %s"%option
- ### check for node existence and get node_type
- nodes = Nodes(self.api, [node_id_or_hostname])
- if not nodes:
- raise PLCInvalidArgument, "No such node %r"%node_id_or_hostname
- node = nodes[0]
-
- if self.DEBUG: print "%s required on node %s. Node type is: %s" \
- % (action, node['node_id'], node['node_type'])
-
- # checks required action against the node type
- node_type = node['node_type']
- if action not in allowed_actions[node_type]:
- raise PLCInvalidArgument, "Action %s not valid for %s nodes, valid actions are %s" \
- % (action, node_type, "|".join(allowed_actions[node_type]))
-
- # compute nodename
+ # compute nodename according the action
if action.find("node-") == 0 or action.find("dummynet-") == 0:
nodename = node['hostname']
else:
- node = None # XXX
+ node = None
# compute a 8 bytes random number
tempbytes = random.sample (xrange(0,256), 8);
def hexa2 (c): return chr((c>>4)+65) + chr ((c&16)+65)
nodename = "".join(map(hexa2,tempbytes))
+ # override some global definition, according node_type
+ if node_type == 'dummynet':
+ self.BOOTCDDIR = "/usr/share/dummynet" # the base installation dir
+ self.BOOTCDBUILD = "/usr/share/dummynet/build.sh" # dummynet build script
+ self.WORKDIR = "/var/tmp/DummynetBoxMedium" # temporary working dir
+
# get nodefamily
(pldistro,arch) = self.get_nodefamily(node)
self.nodefamily="%s-%s"%(pldistro,arch)
for attr in [ "BOOTCDDIR", "BOOTCDBUILD", "GENERICDIR" ]:
setattr(self,attr,getattr(self,attr).replace("@NODEFAMILY@",self.nodefamily))
- filename = self.handle_filename(filename, nodename, suffix, arch)
+ filename = self.handle_filename(filename, nodename, suffix, arch)
# log call
if node:
generic_path = "%s/%s" % (self.GENERICDIR,generic_name)
if filename:
- ret=os.system ("cp %s %s"%(generic_path,filename))
+ ret=os.system ('cp "%s" "%s"'%(generic_path,filename))
if ret==0:
return filename
else:
- raise PLCPermissionDenied, "Could not copy %s into"%(generic_path,filename)
+ raise PLCPermissionDenied, "Could not copy %s into %s"%(generic_path,filename)
else:
### return the generic medium content as-is, just base64 encoded
return base64.b64encode(file(generic_path).read())
- ### config file preview or regenerated
- if action == 'node-preview' or action == 'node-floppy':
+ ### config file preview or regenerated
+ if action == 'node-preview' or action == 'node-floppy':
renew_key = (action == 'node-floppy')
floppy = self.floppy_contents (node,renew_key)
- if filename:
- try:
- file(filename,'w').write(floppy)
- except:
- raise PLCPermissionDenied, "Could not write into %s"%filename
- return filename
- else:
- return floppy
+ if filename:
+ try:
+ file(filename,'w').write(floppy)
+ except:
+ raise PLCPermissionDenied, "Could not write into %s"%filename
+ return filename
+ else:
+ return floppy
### we're left with node-iso and node-usb
- if action == 'node-iso' or action == 'node-usb':
+ # the steps involved in the image creation are:
+ # - create and test the working environment
+ # - generate the configuration file
+ # - build and invoke the build command
+ # - delivery the resulting image file
+
+ if action == 'node-iso' or action == 'node-usb' \
+ or action == 'dummynet-iso' or action == 'dummynet-usb':
### check we've got required material
version = self.bootcd_version()
self.trash.append(floppy_file)
node_image = "%s/%s%s"%(self.WORKDIR,nodename,suffix)
+ log_file="%s.log"%node_image
- # make build's arguments
- build_sh_options=""
- if "cramfs" in build_sh_spec:
- type += "_cramfs"
- if "serial" in build_sh_spec:
- build_sh_options += " -s %s"%build_sh_spec['serial']
-
- for karg in build_sh_spec['kargs']:
- build_sh_options += ' -k "%s"'%karg
+ command = self.build_command(node_type, build_sh_spec, node_image, type, floppy_file, log_file)
+
+ # invoke the image build script
+ if command != "":
+ ret=os.system(command)
- log_file="%s.log"%node_image
- # invoke build.sh
- build_command = '%s -f "%s" -o "%s" -t "%s" %s &> %s' % (self.BOOTCDBUILD,
- floppy_file,
- node_image,
- type,
- build_sh_options,
- log_file)
- if self.DEBUG:
- print 'build command:',build_command
- ret=os.system(build_command)
if ret != 0:
- raise PLCAPIError,"bootcd/build.sh failed\n%s\n%s"%(
- build_command,file(log_file).read())
+ raise PLCAPIError, "%s failed Command line was: %s Error logs: %s" % \
+ (self.BOOTCDBUILD, command, file(log_file).read())
self.trash.append(log_file)
+
if not os.path.isfile (node_image):
raise PLCAPIError,"Unexpected location of build.sh output - %s"%node_image
# handle result
if filename:
- ret=os.system("mv %s %s"%(node_image,filename))
+ ret=os.system('mv "%s" "%s"'%(node_image,filename))
if ret != 0:
self.trash.append(node_image)
self.cleantrash()
# we're done here, or we missed something
raise PLCAPIError,'Unhandled action %s'%action
+
git://git.onelab.eu / plcapi.git / blobdiff