3 # Copyright (c) 2003 Intel Corporation
6 # Copyright (c) 2004-2006 The Trustees of Princeton University
11 from Exceptions import *
19 See if a node installation is valid. More checks should certainly be
20 done in the future, but for now, make sure that the sym links kernel-boot
23 Expect the following variables to be set:
24 SYSIMG_PATH the path where the system image will be mounted
25 (always starts with TEMP_PATH)
26 ROOT_MOUNTED the node root file system is mounted
27 NODE_ID The db node_id for this machine
28 PLCONF_DIR The directory to store the configuration file in
30 Set the following variables upon successfully running:
31 ROOT_MOUNTED the node root file system is mounted
34 log.write("\n\nStep: Validating node installation.\n")
36 # make sure we have the variables we need
38 SYSIMG_PATH = vars["SYSIMG_PATH"]
40 raise ValueError("SYSIMG_PATH")
42 NODE_ID = vars["NODE_ID"]
44 raise ValueError("NODE_ID")
46 PLCONF_DIR = vars["PLCONF_DIR"]
48 raise ValueError("PLCONF_DIR")
50 NODE_MODEL_OPTIONS = vars["NODE_MODEL_OPTIONS"]
52 PARTITIONS = vars["PARTITIONS"]
53 if PARTITIONS == None:
54 raise ValueError("PARTITIONS")
56 except KeyError as var:
57 raise BootManagerException("Missing variable in vars: {}\n".format(var))
58 except ValueError as var:
59 raise BootManagerException("Variable in vars, shouldn't be: {}\n".format(var))
63 if vars.has_key('ROOT_MOUNTED'):
64 ROOT_MOUNTED = vars['ROOT_MOUNTED']
66 # mount the root system image if we haven't already.
67 # capture BootManagerExceptions during the vgscan/change and mount
68 # calls, so we can return 0 instead
71 # simply creating an instance of this class and listing the system
72 # block devices will make them show up so vgscan can find the planetlab
74 systeminfo.get_block_devices_dict(vars, log)
77 utils.sysexec("vgscan", log)
78 utils.sysexec("vgchange -ay planetlab", log)
79 except BootManagerException as e:
80 log.write("BootManagerException during vgscan/vgchange: {}\n".format(e))
83 utils.makedirs(SYSIMG_PATH)
85 # xxx - TODO - need to fsck the btrfs partition
86 if vars['virt'] == 'vs':
87 filesystems_tocheck = ['root', 'vservers']
89 filesystems_tocheck = ['root']
91 for filesystem in filesystems_tocheck:
93 # first run fsck to prevent fs corruption from hanging mount...
94 log.write("fsck {} file system\n".format(filesystem))
95 utils.sysexec("e2fsck -v -p {}".format(PARTITIONS[filesystem]), log, fsck=True)
96 except BootManagerException as e:
97 log.write("BootManagerException during fsck of {} ({}) filesystem : {}\n"\
98 .format(filesystem, PARTITIONS[filesystem], str(e)))
100 log.write("Trying to recover filesystem errors on {}\n".format(filesystem))
101 utils.sysexec("e2fsck -v -y {}".format(PARTITIONS[filesystem]), log, fsck=True)
102 except BootManagerException as e:
103 log.write("BootManagerException while trying to recover"
104 "filesystem errors on {} ({}) filesystem : {}\n"
105 .format(filesystem, PARTITIONS[filesystem], str(e)))
108 # disable time/count based filesystems checks
109 utils.sysexec_noerr("tune2fs -c -1 -i 0 {}".format(PARTITIONS[filesystem]), log)
112 # then attempt to mount them
113 log.write("mounting root file system\n")
114 utils.sysexec("mount -t ext3 {} {}".format(PARTITIONS["root"], SYSIMG_PATH),log)
115 except BootManagerException as e:
116 log.write("BootManagerException during mount of /root: {}\n".format(str(e)))
120 PROC_PATH = "{}/proc".format(SYSIMG_PATH)
121 utils.makedirs(PROC_PATH)
122 log.write("mounting /proc\n")
123 utils.sysexec("mount -t proc none {}".format(PROC_PATH), log)
124 except BootManagerException as e:
125 log.write("BootManagerException during mount of /proc: {}\n".format(str(e)))
129 one_partition = vars['ONE_PARTITION']=='1'
131 if (not one_partition):
133 VSERVERS_PATH = "{}/vservers".format(SYSIMG_PATH)
134 utils.makedirs(VSERVERS_PATH)
135 log.write("mounting vservers partition in root file system\n")
136 if vars['virt'] == 'vs':
137 utils.sysexec("mount -t ext3 {} {}".format(PARTITIONS["vservers"], VSERVERS_PATH), log)
139 utils.sysexec("mount -t btrfs {} {}".format(PARTITIONS["vservers"], VSERVERS_PATH), log)
140 except BootManagerException as e:
141 log.write("BootManagerException while mounting /vservers: {}\n".format(str(e)))
145 vars['ROOT_MOUNTED'] = 1
147 # check if the base kernel is installed
148 # these 2 links are created by our kernel's post-install scriplet
149 log.write("Checking for a custom kernel\n")
151 if vars['virt'] == 'vs':
152 os.stat("{}/boot/kernel-boot".format(SYSIMG_PATH))
155 kversion = os.popen("chroot {} rpm -qa kernel | tail -1 | cut -c 8-"\
156 .format(SYSIMG_PATH)).read().rstrip()
157 os.stat("{}/boot/vmlinuz-{}".format(SYSIMG_PATH, kversion))
158 major_version = int(kversion[0]) # Check if the string looks like a kernel version
160 kversion = os.popen("ls -lrt {}/lib/modules | tail -1 | awk '{print $9;}'"\
161 .format(SYSIMG_PATH)).read().rstrip()
163 log.write("Couldn't locate base kernel (you might be using the stock kernel).\n")
166 # check if the model specified kernel is installed
168 if NODE_MODEL_OPTIONS & ModelOptions.SMP:
171 os.stat("{}/boot/kernel-boot{}".format(SYSIMG_PATH, option))
173 # smp kernel is not there; remove option from modeloptions
174 # such that the rest of the code base thinks we are just
175 # using the base kernel.
176 NODE_MODEL_OPTIONS = NODE_MODEL_OPTIONS & ~ModelOptions.SMP
177 vars["NODE_MODEL_OPTIONS"] = NODE_MODEL_OPTIONS
178 log.write("WARNING: Couldn't locate smp kernel.\n")
180 # write out the node id to /etc/planetlab/node_id. if this fails, return
181 # 0, indicating the node isn't a valid install.
183 node_id_file_path = "{}/{}/node_id".format(SYSIMG_PATH, PLCONF_DIR)
184 node_id_file = file(node_id_file_path, "w")
185 node_id_file.write(str(NODE_ID))
188 log.write("Updated /etc/planetlab/node_id\n")
190 log.write("Unable to write out /etc/planetlab/node_id\n")
193 log.write("Node installation appears to be ok\n")