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