showard | 4cfdce1 | 2009-06-15 20:23:29 +0000 | [diff] [blame] | 1 | import os, shutil, copy, pickle, re, glob, time, logging |
mbligh | c61fb36 | 2008-06-05 16:22:15 +0000 | [diff] [blame] | 2 | from autotest_lib.client.bin import kernel_config, os_dep, kernelexpand, test |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 3 | from autotest_lib.client.bin import utils |
| 4 | from autotest_lib.client.common_lib import log, error, packages |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 5 | |
mbligh | b8e0a11 | 2007-11-05 20:27:36 +0000 | [diff] [blame] | 6 | |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 7 | def tee_output_logdir_mark(fn): |
| 8 | def tee_logdir_mark_wrapper(self, *args, **dargs): |
| 9 | mark = self.__class__.__name__ + "." + fn.__name__ |
showard | 4cfdce1 | 2009-06-15 20:23:29 +0000 | [diff] [blame] | 10 | logging.info("--- START %s ---", mark) |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 11 | self.job.logging.tee_redirect_debug_dir(self.log_dir) |
| 12 | try: |
| 13 | result = fn(self, *args, **dargs) |
| 14 | finally: |
| 15 | self.job.logging.restore() |
showard | 4cfdce1 | 2009-06-15 20:23:29 +0000 | [diff] [blame] | 16 | logging.info("--- END %s ---", mark) |
showard | 75cdfee | 2009-06-10 17:40:41 +0000 | [diff] [blame] | 17 | |
| 18 | return result |
| 19 | |
| 20 | tee_logdir_mark_wrapper.__name__ = fn.__name__ |
| 21 | return tee_logdir_mark_wrapper |
| 22 | |
| 23 | |
mbligh | 516c8df | 2010-02-12 18:48:52 +0000 | [diff] [blame^] | 24 | def _add_kernel_to_bootloader(bootloader, base_args, tag, args, image, initrd): |
| 25 | """ |
| 26 | Add a kernel with the specified tag to the boot config using the given |
| 27 | bootloader object. Also process the base_args and args kernel arguments |
| 28 | by removing all root= options and give the last root= option value to |
| 29 | the bootloader as a root device. |
| 30 | |
| 31 | @param bootloader: bootloader object |
| 32 | @param base_args: base cmdline kernel arguments |
| 33 | @param tag: kernel tag |
| 34 | @param args: kernel cmdline arguments that are merged with base_args; a |
| 35 | root= option in "args" will override any from base_args |
| 36 | @param image: kernel image file |
| 37 | @param initrd: initrd file |
| 38 | """ |
| 39 | # remove existing entry if present |
| 40 | bootloader.remove_kernel(tag) |
| 41 | |
| 42 | if base_args: |
| 43 | args = ' '.join((base_args, args)) |
| 44 | |
| 45 | root_prefix = 'root=' |
| 46 | # stores the last root= value |
| 47 | root = None |
| 48 | # a list with all arguments that don't start with root= so we give them |
| 49 | # later to bootloader.add_kernel() |
| 50 | arglist = [] |
| 51 | |
| 52 | for arg in args.split(): |
| 53 | if arg.startswith(root_prefix): |
| 54 | # set the current root value with the one from the argument |
| 55 | # thus after processing all the arguments we keep the last |
| 56 | # root value (so root= options from args overrides any from |
| 57 | # base_args) |
| 58 | root = arg[len(root_prefix):] |
| 59 | else: |
| 60 | arglist.append(arg) |
| 61 | |
| 62 | # add the kernel entry |
| 63 | bootloader.add_kernel(image, tag, initrd=initrd, args=' '.join(arglist), |
| 64 | root=root) |
| 65 | |
| 66 | |
jadmanski | 6ca37b6 | 2008-06-30 21:17:07 +0000 | [diff] [blame] | 67 | class kernel(object): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 68 | """ Class for compiling kernels. |
mbligh | c86b0b4 | 2006-07-28 17:35:28 +0000 | [diff] [blame] | 69 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 70 | Data for the object includes the src files |
| 71 | used to create the kernel, patches applied, config (base + changes), |
| 72 | the build directory itself, and logged output |
mbligh | c86b0b4 | 2006-07-28 17:35:28 +0000 | [diff] [blame] | 73 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 74 | Properties: |
| 75 | job |
| 76 | Backpointer to the job object we're part of |
| 77 | autodir |
| 78 | Path to the top level autotest dir (/usr/local/autotest) |
| 79 | src_dir |
| 80 | <tmp_dir>/src/ |
| 81 | build_dir |
| 82 | <tmp_dir>/linux/ |
| 83 | config_dir |
| 84 | <results_dir>/config/ |
| 85 | log_dir |
| 86 | <results_dir>/debug/ |
| 87 | results_dir |
| 88 | <results_dir>/results/ |
| 89 | """ |
mbligh | c86b0b4 | 2006-07-28 17:35:28 +0000 | [diff] [blame] | 90 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 91 | autodir = '' |
mbligh | 8baa2ea | 2006-12-17 23:01:24 +0000 | [diff] [blame] | 92 | |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 93 | def __init__(self, job, base_tree, subdir, tmp_dir, build_dir, leave=False): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 94 | """Initialize the kernel build environment |
mbligh | c86b0b4 | 2006-07-28 17:35:28 +0000 | [diff] [blame] | 95 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 96 | job |
| 97 | which job this build is part of |
| 98 | base_tree |
| 99 | base kernel tree. Can be one of the following: |
| 100 | 1. A local tarball |
| 101 | 2. A URL to a tarball |
| 102 | 3. A local directory (will symlink it) |
| 103 | 4. A shorthand expandable (eg '2.6.11-git3') |
| 104 | subdir |
| 105 | subdir in the results directory (eg "build") |
| 106 | (holds config/, debug/, results/) |
| 107 | tmp_dir |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 108 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 109 | leave |
| 110 | Boolean, whether to leave existing tmpdir or not |
| 111 | """ |
| 112 | self.job = job |
| 113 | self.autodir = job.autodir |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 114 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 115 | self.src_dir = os.path.join(tmp_dir, 'src') |
| 116 | self.build_dir = os.path.join(tmp_dir, build_dir) |
| 117 | # created by get_kernel_tree |
| 118 | self.config_dir = os.path.join(subdir, 'config') |
| 119 | self.log_dir = os.path.join(subdir, 'debug') |
| 120 | self.results_dir = os.path.join(subdir, 'results') |
| 121 | self.subdir = os.path.basename(subdir) |
mbligh | 1e8858e | 2006-11-24 22:18:35 +0000 | [diff] [blame] | 122 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 123 | self.installed_as = None |
apw | 87c65c1 | 2007-09-27 17:19:37 +0000 | [diff] [blame] | 124 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 125 | if not leave: |
| 126 | if os.path.isdir(self.src_dir): |
| 127 | utils.system('rm -rf ' + self.src_dir) |
| 128 | if os.path.isdir(self.build_dir): |
| 129 | utils.system('rm -rf ' + self.build_dir) |
mbligh | 1e8858e | 2006-11-24 22:18:35 +0000 | [diff] [blame] | 130 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 131 | if not os.path.exists(self.src_dir): |
| 132 | os.mkdir(self.src_dir) |
| 133 | for path in [self.config_dir, self.log_dir, self.results_dir]: |
| 134 | if os.path.exists(path): |
| 135 | utils.system('rm -rf ' + path) |
| 136 | os.mkdir(path) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 137 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 138 | logpath = os.path.join(self.log_dir, 'build_log') |
| 139 | self.logfile = open(logpath, 'w+') |
| 140 | self.applied_patches = [] |
mbligh | 4426de0 | 2006-10-10 07:18:28 +0000 | [diff] [blame] | 141 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 142 | self.target_arch = None |
| 143 | self.build_target = 'bzImage' |
| 144 | self.build_image = None |
mbligh | fdbcaec | 2006-10-01 23:28:57 +0000 | [diff] [blame] | 145 | |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 146 | arch = utils.get_current_kernel_arch() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 147 | if arch == 's390' or arch == 's390x': |
| 148 | self.build_target = 'image' |
| 149 | elif arch == 'ia64': |
| 150 | self.build_target = 'all' |
| 151 | self.build_image = 'vmlinux.gz' |
mbligh | cac347a | 2007-06-02 17:21:48 +0000 | [diff] [blame] | 152 | |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 153 | if not leave: |
| 154 | self.logfile.write('BASE: %s\n' % base_tree) |
mbligh | 534015f | 2006-09-15 03:28:56 +0000 | [diff] [blame] | 155 | |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 156 | # Where we have direct version hint record that |
| 157 | # for later configuration selection. |
| 158 | shorthand = re.compile(r'^\d+\.\d+\.\d+') |
| 159 | if shorthand.match(base_tree): |
| 160 | self.base_tree_version = base_tree |
| 161 | else: |
| 162 | self.base_tree_version = None |
apw | 2366d99 | 2007-03-12 20:35:57 +0000 | [diff] [blame] | 163 | |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 164 | # Actually extract the tree. Make sure we know it occured |
| 165 | self.extract(base_tree) |
apw | 040dcaa | 2007-11-21 19:36:55 +0000 | [diff] [blame] | 166 | |
apw | 7bae90e | 2008-03-05 12:18:11 +0000 | [diff] [blame] | 167 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 168 | def kernelexpand(self, kernel): |
| 169 | # If we have something like a path, just use it as it is |
| 170 | if '/' in kernel: |
| 171 | return [kernel] |
apw | 7bae90e | 2008-03-05 12:18:11 +0000 | [diff] [blame] | 172 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 173 | # Find the configured mirror list. |
| 174 | mirrors = self.job.config_get('mirror.mirrors') |
| 175 | if not mirrors: |
| 176 | # LEGACY: convert the kernel.org mirror |
| 177 | mirror = self.job.config_get('mirror.ftp_kernel_org') |
| 178 | if mirror: |
| 179 | korg = 'http://www.kernel.org/pub/linux/kernel' |
| 180 | mirrors = [ |
| 181 | [ korg + '/v2.6', mirror + '/v2.6' ], |
mbligh | 9e6a4f1 | 2008-06-06 21:55:12 +0000 | [diff] [blame] | 182 | [ korg + '/people/akpm/patches/2.6', mirror + '/akpm' ], |
| 183 | [ korg + '/people/mbligh', mirror + '/mbligh' ], |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 184 | ] |
apw | 7bae90e | 2008-03-05 12:18:11 +0000 | [diff] [blame] | 185 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 186 | patches = kernelexpand.expand_classic(kernel, mirrors) |
| 187 | print patches |
apw | 7bae90e | 2008-03-05 12:18:11 +0000 | [diff] [blame] | 188 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 189 | return patches |
apw | 7bae90e | 2008-03-05 12:18:11 +0000 | [diff] [blame] | 190 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 191 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 192 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 193 | @tee_output_logdir_mark |
| 194 | def extract(self, base_tree): |
| 195 | if os.path.exists(base_tree): |
| 196 | self.get_kernel_tree(base_tree) |
| 197 | else: |
| 198 | base_components = self.kernelexpand(base_tree) |
| 199 | print 'kernelexpand: ' |
| 200 | print base_components |
| 201 | self.get_kernel_tree(base_components.pop(0)) |
| 202 | if base_components: # apply remaining patches |
| 203 | self.patch(*base_components) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 204 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 205 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 206 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 207 | @tee_output_logdir_mark |
| 208 | def patch(self, *patches): |
| 209 | """Apply a list of patches (in order)""" |
| 210 | if not patches: |
| 211 | return |
| 212 | print 'Applying patches: ', patches |
| 213 | self.apply_patches(self.get_patches(patches)) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 214 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 215 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 216 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 217 | @tee_output_logdir_mark |
mbligh | b3400e0 | 2008-11-06 15:44:25 +0000 | [diff] [blame] | 218 | def config(self, config_file = '', config_list = None, defconfig = False, make = None): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 219 | self.set_cross_cc() |
| 220 | config = kernel_config.kernel_config(self.job, self.build_dir, |
| 221 | self.config_dir, config_file, config_list, |
mbligh | b3400e0 | 2008-11-06 15:44:25 +0000 | [diff] [blame] | 222 | defconfig, self.base_tree_version, make) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 223 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 224 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 225 | def get_patches(self, patches): |
| 226 | """fetch the patches to the local src_dir""" |
| 227 | local_patches = [] |
| 228 | for patch in patches: |
mbligh | 1c9b1d2 | 2008-06-12 17:46:12 +0000 | [diff] [blame] | 229 | dest = os.path.join(self.src_dir, os.path.basename(patch)) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 230 | # FIXME: this isn't unique. Append something to it |
| 231 | # like wget does if it's not there? |
jadmanski | e3f2f71 | 2008-06-12 17:54:56 +0000 | [diff] [blame] | 232 | print "get_file %s %s %s %s" % (patch, dest, self.src_dir, |
| 233 | os.path.basename(patch)) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 234 | utils.get_file(patch, dest) |
| 235 | # probably safer to use the command, not python library |
| 236 | md5sum = utils.system_output('md5sum ' + dest).split()[0] |
| 237 | local_patches.append((patch, dest, md5sum)) |
| 238 | return local_patches |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 239 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 240 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 241 | def apply_patches(self, local_patches): |
| 242 | """apply the list of patches, in order""" |
| 243 | builddir = self.build_dir |
| 244 | os.chdir(builddir) |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 245 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 246 | if not local_patches: |
| 247 | return None |
| 248 | for (spec, local, md5sum) in local_patches: |
| 249 | if local.endswith('.bz2') or local.endswith('.gz'): |
| 250 | ref = spec |
| 251 | else: |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 252 | ref = utils.force_copy(local, self.results_dir) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 253 | ref = self.job.relative_path(ref) |
| 254 | patch_id = "%s %s %s" % (spec, ref, md5sum) |
| 255 | log = "PATCH: " + patch_id + "\n" |
| 256 | print log |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 257 | utils.cat_file_to_cmd(local, 'patch -p1 > /dev/null') |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 258 | self.logfile.write(log) |
| 259 | self.applied_patches.append(patch_id) |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 260 | |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 261 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 262 | def get_kernel_tree(self, base_tree): |
| 263 | """Extract/link base_tree to self.build_dir""" |
mbligh | 5970cf0 | 2006-08-06 15:39:22 +0000 | [diff] [blame] | 264 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 265 | # if base_tree is a dir, assume uncompressed kernel |
| 266 | if os.path.isdir(base_tree): |
| 267 | print 'Symlinking existing kernel source' |
| 268 | os.symlink(base_tree, self.build_dir) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 269 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 270 | # otherwise, extract tarball |
| 271 | else: |
| 272 | os.chdir(os.path.dirname(self.src_dir)) |
| 273 | # Figure out local destination for tarball |
| 274 | tarball = os.path.join(self.src_dir, os.path.basename(base_tree)) |
| 275 | utils.get_file(base_tree, tarball) |
| 276 | print 'Extracting kernel tarball:', tarball, '...' |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 277 | utils.extract_tarball_to_dir(tarball, self.build_dir) |
mbligh | fdbcaec | 2006-10-01 23:28:57 +0000 | [diff] [blame] | 278 | |
| 279 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 280 | def extraversion(self, tag, append=1): |
| 281 | os.chdir(self.build_dir) |
| 282 | extraversion_sub = r's/^EXTRAVERSION =\s*\(.*\)/EXTRAVERSION = ' |
| 283 | if append: |
| 284 | p = extraversion_sub + '\\1-%s/' % tag |
| 285 | else: |
| 286 | p = extraversion_sub + '-%s/' % tag |
| 287 | utils.system('mv Makefile Makefile.old') |
| 288 | utils.system('sed "%s" < Makefile.old > Makefile' % p) |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 289 | |
apw | c784610 | 2006-04-06 18:22:13 +0000 | [diff] [blame] | 290 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 291 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 292 | @tee_output_logdir_mark |
| 293 | def build(self, make_opts = '', logfile = '', extraversion='autotest'): |
| 294 | """build the kernel |
apw | c784610 | 2006-04-06 18:22:13 +0000 | [diff] [blame] | 295 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 296 | make_opts |
| 297 | additional options to make, if any |
| 298 | """ |
| 299 | os_dep.commands('gcc', 'make') |
| 300 | if logfile == '': |
| 301 | logfile = os.path.join(self.log_dir, 'kernel_build') |
| 302 | os.chdir(self.build_dir) |
| 303 | if extraversion: |
| 304 | self.extraversion(extraversion) |
| 305 | self.set_cross_cc() |
| 306 | # setup_config_file(config_file, config_overrides) |
mbligh | 1e8858e | 2006-11-24 22:18:35 +0000 | [diff] [blame] | 307 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 308 | # Not needed on 2.6, but hard to tell -- handle failure |
| 309 | utils.system('make dep', ignore_status=True) |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 310 | threads = 2 * utils.count_cpus() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 311 | build_string = 'make -j %d %s %s' % (threads, make_opts, |
| 312 | self.build_target) |
| 313 | # eg make bzImage, or make zImage |
| 314 | print build_string |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 315 | utils.system(build_string) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 316 | if kernel_config.modules_needed('.config'): |
| 317 | utils.system('make -j %d modules' % (threads)) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 318 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 319 | kernel_version = self.get_kernel_build_ver() |
| 320 | kernel_version = re.sub('-autotest', '', kernel_version) |
| 321 | self.logfile.write('BUILD VERSION: %s\n' % kernel_version) |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 322 | |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 323 | utils.force_copy(self.build_dir+'/System.map', |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 324 | self.results_dir) |
mbligh | 30f28c5 | 2007-10-11 18:35:35 +0000 | [diff] [blame] | 325 | |
mbligh | 30f28c5 | 2007-10-11 18:35:35 +0000 | [diff] [blame] | 326 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 327 | def build_timed(self, threads, timefile = '/dev/null', make_opts = '', |
| 328 | output = '/dev/null'): |
| 329 | """time the bulding of the kernel""" |
| 330 | os.chdir(self.build_dir) |
| 331 | self.set_cross_cc() |
| 332 | |
| 333 | self.clean(logged=False) |
| 334 | build_string = "/usr/bin/time -o %s make %s -j %s vmlinux" \ |
| 335 | % (timefile, make_opts, threads) |
| 336 | build_string += ' > %s 2>&1' % output |
| 337 | print build_string |
| 338 | utils.system(build_string) |
| 339 | |
| 340 | if (not os.path.isfile('vmlinux')): |
| 341 | errmsg = "no vmlinux found, kernel build failed" |
| 342 | raise error.TestError(errmsg) |
| 343 | |
| 344 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 345 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 346 | @tee_output_logdir_mark |
| 347 | def clean(self): |
| 348 | """make clean in the kernel tree""" |
| 349 | os.chdir(self.build_dir) |
| 350 | print "make clean" |
| 351 | utils.system('make clean > /dev/null 2> /dev/null') |
| 352 | |
| 353 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 354 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 355 | @tee_output_logdir_mark |
| 356 | def mkinitrd(self, version, image, system_map, initrd): |
| 357 | """Build kernel initrd image. |
| 358 | Try to use distro specific way to build initrd image. |
| 359 | Parameters: |
| 360 | version |
| 361 | new kernel version |
| 362 | image |
| 363 | new kernel image file |
| 364 | system_map |
| 365 | System.map file |
| 366 | initrd |
| 367 | initrd image file to build |
| 368 | """ |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 369 | vendor = utils.get_os_vendor() |
mbligh | b8a14e3 | 2006-05-06 00:17:35 +0000 | [diff] [blame] | 370 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 371 | if os.path.isfile(initrd): |
| 372 | print "Existing %s file, will remove it." % initrd |
| 373 | os.remove(initrd) |
mbligh | b8a14e3 | 2006-05-06 00:17:35 +0000 | [diff] [blame] | 374 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 375 | args = self.job.config_get('kernel.mkinitrd_extra_args') |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 376 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 377 | # don't leak 'None' into mkinitrd command |
| 378 | if not args: |
| 379 | args = '' |
mbligh | 50f42ea | 2006-09-30 22:22:21 +0000 | [diff] [blame] | 380 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 381 | if vendor in ['Red Hat', 'Fedora Core']: |
| 382 | utils.system('mkinitrd %s %s %s' % (args, initrd, version)) |
| 383 | elif vendor in ['SUSE']: |
jadmanski | d524b0e | 2008-09-15 14:28:20 +0000 | [diff] [blame] | 384 | utils.system('mkinitrd %s -k %s -i %s -M %s' % |
| 385 | (args, image, initrd, system_map)) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 386 | elif vendor in ['Debian', 'Ubuntu']: |
| 387 | if os.path.isfile('/usr/sbin/mkinitrd'): |
| 388 | cmd = '/usr/sbin/mkinitrd' |
| 389 | elif os.path.isfile('/usr/sbin/mkinitramfs'): |
| 390 | cmd = '/usr/sbin/mkinitramfs' |
| 391 | else: |
| 392 | raise error.TestError('No Debian initrd builder') |
| 393 | utils.system('%s %s -o %s %s' % (cmd, args, initrd, version)) |
| 394 | else: |
| 395 | raise error.TestError('Unsupported vendor %s' % vendor) |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 396 | |
apw | e43a30b | 2007-09-25 16:51:30 +0000 | [diff] [blame] | 397 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 398 | def set_build_image(self, image): |
| 399 | self.build_image = image |
mbligh | 3d515d4 | 2007-11-09 17:00:36 +0000 | [diff] [blame] | 400 | |
mbligh | 50f42ea | 2006-09-30 22:22:21 +0000 | [diff] [blame] | 401 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 402 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 403 | @tee_output_logdir_mark |
| 404 | def install(self, tag='autotest', prefix = '/'): |
| 405 | """make install in the kernel tree""" |
mbligh | 50f42ea | 2006-09-30 22:22:21 +0000 | [diff] [blame] | 406 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 407 | # Record that we have installed the kernel, and |
| 408 | # the tag under which we installed it. |
| 409 | self.installed_as = tag |
mbligh | 8baa2ea | 2006-12-17 23:01:24 +0000 | [diff] [blame] | 410 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 411 | os.chdir(self.build_dir) |
mbligh | 8baa2ea | 2006-12-17 23:01:24 +0000 | [diff] [blame] | 412 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 413 | if not os.path.isdir(prefix): |
| 414 | os.mkdir(prefix) |
| 415 | self.boot_dir = os.path.join(prefix, 'boot') |
| 416 | if not os.path.isdir(self.boot_dir): |
| 417 | os.mkdir(self.boot_dir) |
apw | 87c65c1 | 2007-09-27 17:19:37 +0000 | [diff] [blame] | 418 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 419 | if not self.build_image: |
| 420 | images = glob.glob('arch/*/boot/' + self.build_target) |
| 421 | if len(images): |
| 422 | self.build_image = images[0] |
| 423 | else: |
| 424 | self.build_image = self.build_target |
apw | 87c65c1 | 2007-09-27 17:19:37 +0000 | [diff] [blame] | 425 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 426 | # remember installed files |
| 427 | self.vmlinux = self.boot_dir + '/vmlinux-' + tag |
| 428 | if (self.build_image != 'vmlinux'): |
| 429 | self.image = self.boot_dir + '/vmlinuz-' + tag |
| 430 | else: |
| 431 | self.image = self.vmlinux |
| 432 | self.system_map = self.boot_dir + '/System.map-' + tag |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 433 | self.config_file = self.boot_dir + '/config-' + tag |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 434 | self.initrd = '' |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 435 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 436 | # copy to boot dir |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 437 | utils.force_copy('vmlinux', self.vmlinux) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 438 | if (self.build_image != 'vmlinux'): |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 439 | utils.force_copy(self.build_image, self.image) |
| 440 | utils.force_copy('System.map', self.system_map) |
| 441 | utils.force_copy('.config', self.config_file) |
mbligh | 0ad6558 | 2006-10-06 04:16:36 +0000 | [diff] [blame] | 442 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 443 | if not kernel_config.modules_needed('.config'): |
| 444 | return |
mbligh | a87116f | 2006-10-10 02:47:08 +0000 | [diff] [blame] | 445 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 446 | utils.system('make modules_install INSTALL_MOD_PATH=%s' % prefix) |
| 447 | if prefix == '/': |
| 448 | self.initrd = self.boot_dir + '/initrd-' + tag |
| 449 | self.mkinitrd(self.get_kernel_build_ver(), self.image, |
| 450 | self.system_map, self.initrd) |
mbligh | a87116f | 2006-10-10 02:47:08 +0000 | [diff] [blame] | 451 | |
mbligh | a87116f | 2006-10-10 02:47:08 +0000 | [diff] [blame] | 452 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 453 | def add_to_bootloader(self, tag='autotest', args=''): |
| 454 | """ add this kernel to bootloader, taking an |
| 455 | optional parameter of space separated parameters |
| 456 | e.g.: kernel.add_to_bootloader('mykernel', 'ro acpi=off') |
| 457 | """ |
mbligh | 516c8df | 2010-02-12 18:48:52 +0000 | [diff] [blame^] | 458 | _add_kernel_to_bootloader(self.job.bootloader, |
| 459 | self.job.config_get('boot.default_args'), |
| 460 | tag, args, self.image, self.initrd) |
apw | cbe3257 | 2006-11-28 10:00:23 +0000 | [diff] [blame] | 461 | |
mbligh | a87116f | 2006-10-10 02:47:08 +0000 | [diff] [blame] | 462 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 463 | def get_kernel_build_arch(self, arch=None): |
| 464 | """ |
| 465 | Work out the current kernel architecture (as a kernel arch) |
| 466 | """ |
| 467 | if not arch: |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 468 | arch = utils.get_current_kernel_arch() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 469 | if re.match('i.86', arch): |
| 470 | return 'i386' |
| 471 | elif re.match('sun4u', arch): |
| 472 | return 'sparc64' |
| 473 | elif re.match('arm.*', arch): |
| 474 | return 'arm' |
| 475 | elif re.match('sa110', arch): |
| 476 | return 'arm' |
| 477 | elif re.match('s390x', arch): |
| 478 | return 's390' |
| 479 | elif re.match('parisc64', arch): |
| 480 | return 'parisc' |
| 481 | elif re.match('ppc.*', arch): |
| 482 | return 'powerpc' |
| 483 | elif re.match('mips.*', arch): |
| 484 | return 'mips' |
| 485 | else: |
| 486 | return arch |
mbligh | 6a1d4db | 2006-10-06 04:30:16 +0000 | [diff] [blame] | 487 | |
mbligh | 201aa89 | 2006-10-29 04:02:05 +0000 | [diff] [blame] | 488 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 489 | def get_kernel_build_release(self): |
| 490 | releasem = re.compile(r'.*UTS_RELEASE\s+"([^"]+)".*'); |
| 491 | versionm = re.compile(r'.*UTS_VERSION\s+"([^"]+)".*'); |
mbligh | 548f29a | 2006-10-17 04:55:12 +0000 | [diff] [blame] | 492 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 493 | release = None |
| 494 | version = None |
mbligh | 6a1d4db | 2006-10-06 04:30:16 +0000 | [diff] [blame] | 495 | |
jadmanski | d524b0e | 2008-09-15 14:28:20 +0000 | [diff] [blame] | 496 | for f in [self.build_dir + "/include/linux/version.h", |
| 497 | self.build_dir + "/include/linux/utsrelease.h", |
mbligh | 508cbf6 | 2009-11-06 03:01:50 +0000 | [diff] [blame] | 498 | self.build_dir + "/include/linux/compile.h", |
| 499 | self.build_dir + "/include/generated/utsrelease.h", |
| 500 | self.build_dir + "/include/generated/compile.h"]: |
jadmanski | d524b0e | 2008-09-15 14:28:20 +0000 | [diff] [blame] | 501 | if os.path.exists(f): |
| 502 | fd = open(f, 'r') |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 503 | for line in fd.readlines(): |
| 504 | m = releasem.match(line) |
| 505 | if m: |
| 506 | release = m.groups()[0] |
| 507 | m = versionm.match(line) |
| 508 | if m: |
| 509 | version = m.groups()[0] |
| 510 | fd.close() |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 511 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 512 | return (release, version) |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 513 | |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 514 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 515 | def get_kernel_build_ident(self): |
| 516 | (release, version) = self.get_kernel_build_release() |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 517 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 518 | if not release or not version: |
| 519 | raise error.JobError('kernel has no identity') |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 520 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 521 | return release + '::' + version |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 522 | |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 523 | |
jadmanski | 067b26c | 2008-09-25 19:46:56 +0000 | [diff] [blame] | 524 | def boot(self, args='', ident=True): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 525 | """ install and boot this kernel, do not care how |
| 526 | just make it happen. |
| 527 | """ |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 528 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 529 | # If we can check the kernel identity do so. |
jadmanski | 067b26c | 2008-09-25 19:46:56 +0000 | [diff] [blame] | 530 | expected_ident = self.get_kernel_build_ident() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 531 | if ident: |
| 532 | when = int(time.time()) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 533 | args += " IDENT=%d" % (when) |
jadmanski | 067b26c | 2008-09-25 19:46:56 +0000 | [diff] [blame] | 534 | self.job.next_step_prepend(["job.end_reboot_and_verify", when, |
| 535 | expected_ident, self.subdir, |
| 536 | self.applied_patches]) |
| 537 | else: |
| 538 | self.job.next_step_prepend(["job.end_reboot", self.subdir, |
| 539 | expected_ident, self.applied_patches]) |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 540 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 541 | # Check if the kernel has been installed, if not install |
| 542 | # as the default tag and boot that. |
| 543 | if not self.installed_as: |
| 544 | self.install() |
mbligh | 237bed3 | 2007-09-05 13:05:57 +0000 | [diff] [blame] | 545 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 546 | # Boot the selected tag. |
| 547 | self.add_to_bootloader(args=args, tag=self.installed_as) |
apw | 87c65c1 | 2007-09-27 17:19:37 +0000 | [diff] [blame] | 548 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 549 | # Boot it. |
jadmanski | 02c0e45 | 2008-10-21 16:33:44 +0000 | [diff] [blame] | 550 | self.job.start_reboot() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 551 | self.job.reboot(tag=self.installed_as) |
apw | 1b5dc36 | 2006-10-31 11:24:26 +0000 | [diff] [blame] | 552 | |
apw | 1b5dc36 | 2006-10-31 11:24:26 +0000 | [diff] [blame] | 553 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 554 | def get_kernel_build_ver(self): |
| 555 | """Check Makefile and .config to return kernel version""" |
| 556 | version = patchlevel = sublevel = extraversion = localversion = '' |
apw | 1b5dc36 | 2006-10-31 11:24:26 +0000 | [diff] [blame] | 557 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 558 | for line in open(self.build_dir + '/Makefile', 'r').readlines(): |
| 559 | if line.startswith('VERSION'): |
| 560 | version = line[line.index('=') + 1:].strip() |
| 561 | if line.startswith('PATCHLEVEL'): |
| 562 | patchlevel = line[line.index('=') + 1:].strip() |
| 563 | if line.startswith('SUBLEVEL'): |
| 564 | sublevel = line[line.index('=') + 1:].strip() |
| 565 | if line.startswith('EXTRAVERSION'): |
| 566 | extraversion = line[line.index('=') + 1:].strip() |
mbligh | e11f5fc | 2006-10-04 04:42:22 +0000 | [diff] [blame] | 567 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 568 | for line in open(self.build_dir + '/.config', 'r').readlines(): |
| 569 | if line.startswith('CONFIG_LOCALVERSION='): |
| 570 | localversion = line.rstrip().split('"')[1] |
mbligh | e11f5fc | 2006-10-04 04:42:22 +0000 | [diff] [blame] | 571 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 572 | return "%s.%s.%s%s%s" %(version, patchlevel, sublevel, extraversion, localversion) |
mbligh | e11f5fc | 2006-10-04 04:42:22 +0000 | [diff] [blame] | 573 | |
mbligh | fdbcaec | 2006-10-01 23:28:57 +0000 | [diff] [blame] | 574 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 575 | def set_build_target(self, build_target): |
| 576 | if build_target: |
| 577 | self.build_target = build_target |
| 578 | print 'BUILD TARGET: %s' % self.build_target |
mbligh | fdbcaec | 2006-10-01 23:28:57 +0000 | [diff] [blame] | 579 | |
mbligh | 8baa2ea | 2006-12-17 23:01:24 +0000 | [diff] [blame] | 580 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 581 | def set_cross_cc(self, target_arch=None, cross_compile=None, |
| 582 | build_target='bzImage'): |
| 583 | """Set up to cross-compile. |
| 584 | This is broken. We need to work out what the default |
| 585 | compile produces, and if not, THEN set the cross |
| 586 | compiler. |
| 587 | """ |
mbligh | 8baa2ea | 2006-12-17 23:01:24 +0000 | [diff] [blame] | 588 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 589 | if self.target_arch: |
| 590 | return |
mbligh | cc2e666 | 2006-09-14 01:24:07 +0000 | [diff] [blame] | 591 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 592 | # if someone has set build_target, don't clobber in set_cross_cc |
| 593 | # run set_build_target before calling set_cross_cc |
| 594 | if not self.build_target: |
| 595 | self.set_build_target(build_target) |
mbligh | 678823f | 2006-12-07 18:49:00 +0000 | [diff] [blame] | 596 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 597 | # If no 'target_arch' given assume native compilation |
mbligh | d876f45 | 2008-12-03 15:09:17 +0000 | [diff] [blame] | 598 | if target_arch is None: |
mbligh | 53da18e | 2009-01-05 21:13:26 +0000 | [diff] [blame] | 599 | target_arch = utils.get_current_kernel_arch() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 600 | if target_arch == 'ppc64': |
| 601 | if self.build_target == 'bzImage': |
| 602 | self.build_target = 'vmlinux' |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 603 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 604 | if not cross_compile: |
| 605 | cross_compile = self.job.config_get('kernel.cross_cc') |
mbligh | 678823f | 2006-12-07 18:49:00 +0000 | [diff] [blame] | 606 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 607 | if cross_compile: |
| 608 | os.environ['CROSS_COMPILE'] = cross_compile |
| 609 | else: |
| 610 | if os.environ.has_key('CROSS_COMPILE'): |
| 611 | del os.environ['CROSS_COMPILE'] |
mbligh | 678823f | 2006-12-07 18:49:00 +0000 | [diff] [blame] | 612 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 613 | return # HACK. Crap out for now. |
mbligh | f4c3532 | 2006-03-13 01:01:10 +0000 | [diff] [blame] | 614 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 615 | # At this point I know what arch I *want* to build for |
| 616 | # but have no way of working out what arch the default |
| 617 | # compiler DOES build for. |
mbligh | cc2e666 | 2006-09-14 01:24:07 +0000 | [diff] [blame] | 618 | |
mbligh | 925e1b1 | 2008-06-12 17:48:38 +0000 | [diff] [blame] | 619 | def install_package(package): |
| 620 | raise NotImplementedError("I don't exist yet!") |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 621 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 622 | if target_arch == 'ppc64': |
| 623 | install_package('ppc64-cross') |
| 624 | cross_compile = os.path.join(self.autodir, 'sources/ppc64-cross/bin') |
mbligh | cc2e666 | 2006-09-14 01:24:07 +0000 | [diff] [blame] | 625 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 626 | elif target_arch == 'x86_64': |
| 627 | install_package('x86_64-cross') |
| 628 | cross_compile = os.path.join(self.autodir, 'sources/x86_64-cross/bin') |
mbligh | b8a14e3 | 2006-05-06 00:17:35 +0000 | [diff] [blame] | 629 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 630 | os.environ['ARCH'] = self.target_arch = target_arch |
mbligh | 5970cf0 | 2006-08-06 15:39:22 +0000 | [diff] [blame] | 631 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 632 | self.cross_compile = cross_compile |
| 633 | if self.cross_compile: |
| 634 | os.environ['CROSS_COMPILE'] = self.cross_compile |
mbligh | cc2e666 | 2006-09-14 01:24:07 +0000 | [diff] [blame] | 635 | |
mbligh | 72b88fc | 2006-12-16 18:41:35 +0000 | [diff] [blame] | 636 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 637 | def pickle_dump(self, filename): |
| 638 | """dump a pickle of ourself out to the specified filename |
mbligh | c86b0b4 | 2006-07-28 17:35:28 +0000 | [diff] [blame] | 639 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 640 | we can't pickle the backreference to job (it contains fd's), |
| 641 | nor would we want to. Same for logfile (fd's). |
| 642 | """ |
| 643 | temp = copy.copy(self) |
| 644 | temp.job = None |
| 645 | temp.logfile = None |
| 646 | pickle.dump(temp, open(filename, 'w')) |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 647 | |
| 648 | |
jadmanski | 6ca37b6 | 2008-06-30 21:17:07 +0000 | [diff] [blame] | 649 | class rpm_kernel(object): |
mbligh | eaa75e5 | 2009-11-06 03:08:08 +0000 | [diff] [blame] | 650 | """ |
| 651 | Class for installing a binary rpm kernel package |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 652 | """ |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 653 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 654 | def __init__(self, job, rpm_package, subdir): |
| 655 | self.job = job |
| 656 | self.rpm_package = rpm_package |
| 657 | self.log_dir = os.path.join(subdir, 'debug') |
| 658 | self.subdir = os.path.basename(subdir) |
| 659 | if os.path.exists(self.log_dir): |
| 660 | utils.system('rm -rf ' + self.log_dir) |
| 661 | os.mkdir(self.log_dir) |
| 662 | self.installed_as = None |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 663 | |
| 664 | |
mbligh | eaa75e5 | 2009-11-06 03:08:08 +0000 | [diff] [blame] | 665 | def build(self, *args, **dargs): |
| 666 | """ |
| 667 | Dummy function, binary kernel so nothing to build. |
| 668 | """ |
| 669 | pass |
| 670 | |
| 671 | |
mbligh | 1b3b376 | 2008-09-25 02:46:34 +0000 | [diff] [blame] | 672 | @log.record |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 673 | @tee_output_logdir_mark |
mbligh | a25e8c3 | 2009-06-15 21:27:23 +0000 | [diff] [blame] | 674 | def install(self, tag='autotest', install_vmlinux=True): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 675 | self.installed_as = tag |
mbligh | da0311e | 2007-10-25 16:03:33 +0000 | [diff] [blame] | 676 | |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 677 | self.image = None |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 678 | self.initrd = '' |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 679 | for rpm_pack in self.rpm_package: |
| 680 | rpm_name = utils.system_output('rpm -qp ' + rpm_pack) |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 681 | |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 682 | # install |
| 683 | utils.system('rpm -i --force ' + rpm_pack) |
| 684 | |
| 685 | # get file list |
| 686 | files = utils.system_output('rpm -ql ' + rpm_name).splitlines() |
| 687 | |
| 688 | # search for vmlinuz |
| 689 | for file in files: |
| 690 | if file.startswith('/boot/vmlinuz'): |
| 691 | self.full_version = file[len('/boot/vmlinuz-'):] |
| 692 | self.image = file |
| 693 | self.rpm_flavour = rpm_name.split('-')[1] |
| 694 | |
| 695 | # get version and release number |
| 696 | self.version, self.release = utils.system_output( |
| 697 | 'rpm --queryformat="%{VERSION}\\n%{RELEASE}\\n" -q ' |
| 698 | + rpm_name).splitlines()[0:2] |
| 699 | |
| 700 | # prefer /boot/kernel-version before /boot/kernel |
| 701 | if self.full_version: |
| 702 | break |
| 703 | |
| 704 | # search for initrd |
| 705 | for file in files: |
| 706 | if file.startswith('/boot/initrd'): |
| 707 | self.initrd = file |
| 708 | # prefer /boot/initrd-version before /boot/initrd |
| 709 | if len(file) > len('/boot/initrd'): |
| 710 | break |
| 711 | |
| 712 | if self.image == None: |
| 713 | errmsg = "specified rpm file(s) don't contain /boot/vmlinuz" |
| 714 | raise error.TestError(errmsg) |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 715 | |
mbligh | a25e8c3 | 2009-06-15 21:27:23 +0000 | [diff] [blame] | 716 | # install vmlinux |
| 717 | if install_vmlinux: |
| 718 | for rpm_pack in self.rpm_package: |
| 719 | vmlinux = utils.system_output( |
| 720 | 'rpm -q -l -p %s | grep /boot/vmlinux' % rpm_pack) |
jadmanski | 19426ea | 2009-07-28 20:19:40 +0000 | [diff] [blame] | 721 | utils.system('cd /; rpm2cpio %s | cpio -imuv .%s 2>&1' |
mbligh | a25e8c3 | 2009-06-15 21:27:23 +0000 | [diff] [blame] | 722 | % (rpm_pack, vmlinux)) |
| 723 | if not os.path.exists(vmlinux): |
| 724 | raise error.TestError('%s does not exist after installing %s' |
| 725 | % (vmlinux, rpm_pack)) |
| 726 | |
mbligh | 736adc9 | 2007-10-18 03:23:22 +0000 | [diff] [blame] | 727 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 728 | def add_to_bootloader(self, tag='autotest', args=''): |
| 729 | """ Add this kernel to bootloader |
| 730 | """ |
mbligh | 516c8df | 2010-02-12 18:48:52 +0000 | [diff] [blame^] | 731 | _add_kernel_to_bootloader(self.job.bootloader, |
| 732 | self.job.config_get('boot.default_args'), |
| 733 | tag, args, self.image, self.initrd) |
mbligh | 10a24a7 | 2007-10-24 21:02:53 +0000 | [diff] [blame] | 734 | |
| 735 | |
jadmanski | 067b26c | 2008-09-25 19:46:56 +0000 | [diff] [blame] | 736 | def boot(self, args='', ident=True): |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 737 | """ install and boot this kernel |
| 738 | """ |
mbligh | 73e82a3 | 2007-11-08 21:35:29 +0000 | [diff] [blame] | 739 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 740 | # Check if the kernel has been installed, if not install |
| 741 | # as the default tag and boot that. |
| 742 | if not self.installed_as: |
| 743 | self.install() |
mbligh | da0311e | 2007-10-25 16:03:33 +0000 | [diff] [blame] | 744 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 745 | # If we can check the kernel identity do so. |
mbligh | b1887c8 | 2009-03-12 00:25:48 +0000 | [diff] [blame] | 746 | expected_ident = self.full_version |
| 747 | if not expected_ident: |
| 748 | expected_ident = '-'.join([self.version, |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 749 | self.rpm_flavour, |
mbligh | b1887c8 | 2009-03-12 00:25:48 +0000 | [diff] [blame] | 750 | self.release]) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 751 | if ident: |
| 752 | when = int(time.time()) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 753 | args += " IDENT=%d" % (when) |
jadmanski | 067b26c | 2008-09-25 19:46:56 +0000 | [diff] [blame] | 754 | self.job.next_step_prepend(["job.end_reboot_and_verify", |
| 755 | when, expected_ident, None, 'rpm']) |
| 756 | else: |
| 757 | self.job.next_step_prepend(["job.end_reboot", None, |
| 758 | expected_ident, []]) |
mbligh | 10a24a7 | 2007-10-24 21:02:53 +0000 | [diff] [blame] | 759 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 760 | # Boot the selected tag. |
| 761 | self.add_to_bootloader(args=args, tag=self.installed_as) |
mbligh | 10a24a7 | 2007-10-24 21:02:53 +0000 | [diff] [blame] | 762 | |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 763 | # Boot it. |
jadmanski | 02c0e45 | 2008-10-21 16:33:44 +0000 | [diff] [blame] | 764 | self.job.start_reboot() |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 765 | self.job.reboot(tag=self.installed_as) |
mbligh | 6ee7ee0 | 2007-11-13 23:49:05 +0000 | [diff] [blame] | 766 | |
| 767 | |
mbligh | e7785cc | 2009-03-17 17:32:47 +0000 | [diff] [blame] | 768 | class rpm_kernel_suse(rpm_kernel): |
| 769 | """ Class for installing openSUSE/SLE rpm kernel package |
| 770 | """ |
| 771 | |
| 772 | def install(self): |
| 773 | # do not set the new kernel as the default one |
| 774 | os.environ['PBL_AUTOTEST'] = '1' |
| 775 | |
| 776 | rpm_kernel.install(self, 'dummy') |
| 777 | self.installed_as = self.job.bootloader.get_title_for_kernel(self.image) |
| 778 | if not self.installed_as: |
| 779 | errmsg = "cannot find installed kernel in bootloader configuration" |
| 780 | raise error.TestError(errmsg) |
| 781 | |
| 782 | |
| 783 | def add_to_bootloader(self, tag='dummy', args=''): |
| 784 | """ Set parameters of this kernel in bootloader |
| 785 | """ |
| 786 | |
| 787 | # pull the base argument set from the job config |
| 788 | baseargs = self.job.config_get('boot.default_args') |
| 789 | if baseargs: |
| 790 | args = baseargs + ' ' + args |
| 791 | |
| 792 | self.job.bootloader.add_args(tag, args) |
| 793 | |
| 794 | |
| 795 | def rpm_kernel_vendor(job, rpm_package, subdir): |
mbligh | 1ef218d | 2009-08-03 16:57:56 +0000 | [diff] [blame] | 796 | vendor = utils.get_os_vendor() |
| 797 | if vendor == "SUSE": |
| 798 | return rpm_kernel_suse(job, rpm_package, subdir) |
| 799 | else: |
| 800 | return rpm_kernel(job, rpm_package, subdir) |
mbligh | e7785cc | 2009-03-17 17:32:47 +0000 | [diff] [blame] | 801 | |
| 802 | |
mbligh | 062ed15 | 2009-01-13 00:57:14 +0000 | [diff] [blame] | 803 | # just make the preprocessor a nop |
| 804 | def _preprocess_path_dummy(path): |
| 805 | return path.strip() |
| 806 | |
| 807 | |
mbligh | 6ee7ee0 | 2007-11-13 23:49:05 +0000 | [diff] [blame] | 808 | # pull in some optional site-specific path pre-processing |
jadmanski | 19426ea | 2009-07-28 20:19:40 +0000 | [diff] [blame] | 809 | preprocess_path = utils.import_site_function(__file__, |
mbligh | 062ed15 | 2009-01-13 00:57:14 +0000 | [diff] [blame] | 810 | "autotest_lib.client.bin.site_kernel", "preprocess_path", |
| 811 | _preprocess_path_dummy) |
mbligh | 6ee7ee0 | 2007-11-13 23:49:05 +0000 | [diff] [blame] | 812 | |
mbligh | c5ddfd1 | 2008-08-04 17:15:00 +0000 | [diff] [blame] | 813 | |
mbligh | 6ee7ee0 | 2007-11-13 23:49:05 +0000 | [diff] [blame] | 814 | def auto_kernel(job, path, subdir, tmp_dir, build_dir, leave=False): |
mbligh | 7aeda67 | 2009-01-30 00:35:59 +0000 | [diff] [blame] | 815 | """ |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 816 | Create a kernel object, dynamically selecting the appropriate class to use |
| 817 | based on the path provided. |
| 818 | """ |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 819 | kernel_paths = [preprocess_path(path)] |
| 820 | if kernel_paths[0].endswith('.list'): |
mbligh | 1ef218d | 2009-08-03 16:57:56 +0000 | [diff] [blame] | 821 | # Fetch the list of packages to install |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 822 | kernel_list = os.path.join(tmp_dir, 'kernel.list') |
| 823 | utils.get_file(kernel_paths[0], kernel_list) |
| 824 | kernel_paths = [p.strip() for p in open(kernel_list).readlines()] |
mbligh | c5ddfd1 | 2008-08-04 17:15:00 +0000 | [diff] [blame] | 825 | |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 826 | if kernel_paths[0].endswith('.rpm'): |
| 827 | rpm_paths = [] |
| 828 | for kernel_path in kernel_paths: |
jadmanski | 19426ea | 2009-07-28 20:19:40 +0000 | [diff] [blame] | 829 | if os.path.exists(kernel_path): |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 830 | rpm_paths.append(kernel_path) |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 831 | else: |
| 832 | # Fetch the rpm into the job's packages directory and pass it to |
| 833 | # rpm_kernel |
| 834 | rpm_name = os.path.basename(kernel_path) |
mbligh | 7aeda67 | 2009-01-30 00:35:59 +0000 | [diff] [blame] | 835 | |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 836 | # If the preprocessed path (kernel_path) is only a name then |
| 837 | # search for the kernel in all the repositories, else fetch the |
| 838 | # kernel from that specific path. |
| 839 | job.pkgmgr.fetch_pkg(rpm_name, os.path.join(job.pkgdir, rpm_name), |
| 840 | repo_url=os.path.dirname(kernel_path)) |
| 841 | |
| 842 | rpm_paths.append(os.path.join(job.pkgdir, rpm_name)) |
| 843 | return rpm_kernel_vendor(job, rpm_paths, subdir) |
jadmanski | 0afbb63 | 2008-06-06 21:10:57 +0000 | [diff] [blame] | 844 | else: |
mbligh | 1b160a0 | 2009-05-21 01:27:10 +0000 | [diff] [blame] | 845 | if len(kernel_paths) > 1: |
| 846 | raise error.TestError("don't know what to do with more than one non-rpm kernel file") |
| 847 | return kernel(job,kernel_paths[0], subdir, tmp_dir, build_dir, leave) |