Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 1 | # Copyright (c) 2011 The Chromium OS Authors. All rights reserved. |
| 2 | # Use of this source code is governed by a BSD-style license that can be |
| 3 | # found in the LICENSE file. |
| 4 | |
| 5 | """This module builds a firmware image for a tegra-based board. |
| 6 | |
| 7 | This modules uses a few rudimentary other libraries for its activity. |
| 8 | |
| 9 | Here are the names we give to the various files we deal with. It is important |
| 10 | to keep these consistent! |
| 11 | |
| 12 | uboot u-boot.bin (with no device tree) |
| 13 | fdt the fdt blob |
| 14 | bct the BCT file |
| 15 | bootstub uboot + fdt |
| 16 | signed (uboot + fdt + bct) signed blob |
| 17 | """ |
| 18 | |
Simon Glass | ceff3ff | 2012-04-04 11:23:45 -0700 | [diff] [blame] | 19 | import glob |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 20 | import os |
| 21 | import re |
| 22 | |
| 23 | import cros_output |
| 24 | from fdt import Fdt |
| 25 | from pack_firmware import PackFirmware |
| 26 | import shutil |
Simon Glass | 7c2d557 | 2011-11-15 14:47:08 -0800 | [diff] [blame] | 27 | import struct |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 28 | import tempfile |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 29 | from tools import CmdError |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 30 | from tools import Tools |
| 31 | from write_firmware import WriteFirmware |
| 32 | |
| 33 | # This data is required by bmpblk_utility. Does it ever change? |
| 34 | # It was stored with the chromeos-bootimage ebuild, but we want |
| 35 | # this utility to work outside the chroot. |
| 36 | yaml_data = ''' |
| 37 | bmpblock: 1.0 |
| 38 | |
| 39 | images: |
| 40 | devmode: DeveloperBmp/DeveloperBmp.bmp |
| 41 | recovery: RecoveryBmp/RecoveryBmp.bmp |
| 42 | rec_yuck: RecoveryNoOSBmp/RecoveryNoOSBmp.bmp |
| 43 | rec_insert: RecoveryMissingOSBmp/RecoveryMissingOSBmp.bmp |
| 44 | |
| 45 | screens: |
| 46 | dev_en: |
| 47 | - [0, 0, devmode] |
| 48 | rec_en: |
| 49 | - [0, 0, recovery] |
| 50 | yuck_en: |
| 51 | - [0, 0, rec_yuck] |
| 52 | ins_en: |
| 53 | - [0, 0, rec_insert] |
| 54 | |
| 55 | localizations: |
| 56 | - [ dev_en, rec_en, yuck_en, ins_en ] |
| 57 | ''' |
| 58 | |
| 59 | class Bundle: |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 60 | """This class encapsulates the entire bundle firmware logic. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 61 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 62 | Sequence of events: |
| 63 | bundle = Bundle(tools.Tools(), cros_output.Output()) |
| 64 | bundle.SetDirs(...) |
| 65 | bundle.SetFiles(...) |
| 66 | bundle.SetOptions(...) |
| 67 | bundle.SelectFdt(fdt.Fdt('filename.dtb') |
Simon Glass | a4934b7 | 2012-05-09 13:35:02 -0700 | [diff] [blame] | 68 | .. can call bundle.AddConfigList(), AddEnableList() if required |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 69 | bundle.Start(...) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 70 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 71 | Public properties: |
| 72 | fdt: The fdt object that we use for building our image. This wil be the |
| 73 | one specified by the user, except that we might add config options |
| 74 | to it. This is set up by SelectFdt() which must be called before |
| 75 | bundling starts. |
| 76 | uboot_fname: Full filename of the U-Boot binary we use. |
| 77 | bct_fname: Full filename of the BCT file we use. |
Simon Glass | 559b661 | 2012-05-23 13:28:45 -0700 | [diff] [blame] | 78 | spl_source: Source device to load U-Boot from, in SPL: |
| 79 | straps: Select device according to CPU strap pins |
| 80 | spi: Boot from SPI |
| 81 | emmc: Boot from eMMC |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 82 | """ |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 83 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 84 | def __init__(self, tools, output): |
| 85 | """Set up a new Bundle object. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 86 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 87 | Args: |
| 88 | tools: A tools.Tools object to use for external tools. |
| 89 | output: A cros_output.Output object to use for program output. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 90 | """ |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 91 | self._tools = tools |
| 92 | self._out = output |
| 93 | |
| 94 | # Set up the things we need to know in order to operate. |
| 95 | self._board = None # Board name, e.g. tegra2_seaboard. |
| 96 | self._fdt_fname = None # Filename of our FDT. |
| 97 | self.uboot_fname = None # Filename of our U-Boot binary. |
| 98 | self.bct_fname = None # Filename of our BCT file. |
| 99 | self.fdt = None # Our Fdt object. |
Hung-Te Lin | 5b64938 | 2011-08-03 15:01:16 +0800 | [diff] [blame] | 100 | self.bmpblk_fname = None # Filename of our Bitmap Block |
Stefan Reinauer | 8d79d36 | 2011-08-16 14:20:43 -0700 | [diff] [blame] | 101 | self.coreboot_fname = None # Filename of our coreboot binary. |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 102 | self.seabios_fname = None # Filename of our SeaBIOS payload. |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 103 | self.exynos_bl1 = None # Filename of Exynos BL1 (pre-boot) |
| 104 | self.exynos_bl2 = None # Filename of Exynos BL2 (SPL) |
Simon Glass | 559b661 | 2012-05-23 13:28:45 -0700 | [diff] [blame] | 105 | self.spl_source = 'straps' # SPL boot according to board settings |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 106 | self.skeleton_fname = None # Filename of Coreboot skeleton file |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 107 | |
| 108 | def SetDirs(self, keydir): |
| 109 | """Set up directories required for Bundle. |
| 110 | |
| 111 | Args: |
| 112 | keydir: Directory containing keys to use for signing firmware. |
| 113 | """ |
| 114 | self._keydir = keydir |
| 115 | |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 116 | def SetFiles(self, board, bct, uboot=None, bmpblk=None, coreboot=None, |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 117 | postload=None, seabios=None, exynos_bl1=None, exynos_bl2=None, |
| 118 | skeleton=None): |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 119 | """Set up files required for Bundle. |
| 120 | |
| 121 | Args: |
| 122 | board: The name of the board to target (e.g. tegra2_seaboard). |
| 123 | uboot: The filename of the u-boot.bin image to use. |
| 124 | bct: The filename of the binary BCT file to use. |
Hung-Te Lin | 5b64938 | 2011-08-03 15:01:16 +0800 | [diff] [blame] | 125 | bmpblk: The filename of bitmap block file to use. |
Stefan Reinauer | 8d79d36 | 2011-08-16 14:20:43 -0700 | [diff] [blame] | 126 | coreboot: The filename of the coreboot image to use (on x86) |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 127 | postload: The filename of the u-boot-post.bin image to use. |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 128 | seabios: The filename of the SeaBIOS payload to use if any. |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 129 | exynos_bl1: The filename of the exynos BL1 file |
| 130 | exynos_bl2: The filename of the exynos BL2 file (U-Boot spl) |
| 131 | skeleton: The filename of the coreboot skeleton file. |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 132 | """ |
| 133 | self._board = board |
| 134 | self.uboot_fname = uboot |
| 135 | self.bct_fname = bct |
Hung-Te Lin | 5b64938 | 2011-08-03 15:01:16 +0800 | [diff] [blame] | 136 | self.bmpblk_fname = bmpblk |
Stefan Reinauer | 8d79d36 | 2011-08-16 14:20:43 -0700 | [diff] [blame] | 137 | self.coreboot_fname = coreboot |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 138 | self.postload_fname = postload |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 139 | self.seabios_fname = seabios |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 140 | self.exynos_bl1 = exynos_bl1 |
| 141 | self.exynos_bl2 = exynos_bl2 |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 142 | self.skeleton_fname = skeleton |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 143 | |
| 144 | def SetOptions(self, small): |
| 145 | """Set up options supported by Bundle. |
| 146 | |
| 147 | Args: |
| 148 | small: Only create a signed U-Boot - don't produce the full packed |
| 149 | firmware image. This is useful for devs who want to replace just the |
| 150 | U-Boot part while keeping the keys, gbb, etc. the same. |
| 151 | """ |
| 152 | self._small = small |
| 153 | |
| 154 | def CheckOptions(self): |
| 155 | """Check provided options and select defaults.""" |
| 156 | if not self._board: |
| 157 | raise ValueError('No board defined - please define a board to use') |
Simon Glass | 493163b | 2011-09-14 11:19:57 -0700 | [diff] [blame] | 158 | build_root = os.path.join('##', 'build', self._board, 'firmware') |
Simon Glass | 881964d | 2012-04-04 11:34:09 -0700 | [diff] [blame] | 159 | dir_name = os.path.join(build_root, 'dts') |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 160 | if not self._fdt_fname: |
Simon Glass | ceff3ff | 2012-04-04 11:23:45 -0700 | [diff] [blame] | 161 | # Figure out where the file should be, and the name we expect. |
Simon Glass | ceff3ff | 2012-04-04 11:23:45 -0700 | [diff] [blame] | 162 | base_name = re.sub('_', '-', self._board) |
| 163 | |
| 164 | # In case the name exists with a prefix or suffix, find it. |
| 165 | wildcard = os.path.join(dir_name, '*%s*.dts' % base_name) |
| 166 | found_list = glob.glob(self._tools.Filename(wildcard)) |
| 167 | if len(found_list) == 1: |
| 168 | self._fdt_fname = found_list[0] |
| 169 | else: |
| 170 | # We didn't find anything definite, so set up our expected name. |
| 171 | self._fdt_fname = os.path.join(dir_name, '%s.dts' % base_name) |
| 172 | |
Simon Glass | 881964d | 2012-04-04 11:34:09 -0700 | [diff] [blame] | 173 | # Convert things like 'exynos5250-daisy' into a full path. |
| 174 | root, ext = os.path.splitext(self._fdt_fname) |
| 175 | if not ext and not os.path.dirname(root): |
| 176 | self._fdt_fname = os.path.join(dir_name, '%s.dts' % root) |
| 177 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 178 | if not self.uboot_fname: |
| 179 | self.uboot_fname = os.path.join(build_root, 'u-boot.bin') |
| 180 | if not self.bct_fname: |
| 181 | self.bct_fname = os.path.join(build_root, 'bct', 'board.bct') |
Simon Glass | 2a7f0b3 | 2011-08-26 11:25:17 -0700 | [diff] [blame] | 182 | if not self.bmpblk_fname: |
| 183 | self.bmpblk_fname = os.path.join(build_root, 'default.bmpblk') |
Simon Glass | 1d37683 | 2012-03-15 20:50:54 -0700 | [diff] [blame] | 184 | if not self.exynos_bl1: |
| 185 | self.exynos_bl1 = os.path.join(build_root, 'E5250.nbl1.bin') |
| 186 | if not self.exynos_bl2: |
| 187 | self.exynos_bl2 = os.path.join(build_root, 'smdk5250-spl.bin') |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 188 | if not self.coreboot_fname: |
| 189 | self.coreboot_fname = os.path.join(build_root, 'coreboot.rom') |
| 190 | if not self.skeleton_fname: |
| 191 | self.skeleton_fname = os.path.join(build_root, 'skeleton.bin') |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 192 | |
Simon Glass | 7575930 | 2012-03-15 20:26:53 -0700 | [diff] [blame] | 193 | def GetFiles(self): |
| 194 | """Get a list of files that we know about. |
| 195 | |
| 196 | This is the opposite of SetFiles except that we may have put in some |
| 197 | default names. It returns a dictionary containing the filename for |
| 198 | each of a number of pre-defined files. |
| 199 | |
| 200 | Returns: |
| 201 | Dictionary, with one entry for each file. |
| 202 | """ |
| 203 | file_list = { |
| 204 | 'bct' : self.bct_fname, |
| 205 | 'exynos-bl1' : self.exynos_bl1, |
| 206 | 'exynos-bl2' : self.exynos_bl2, |
| 207 | } |
| 208 | return file_list |
| 209 | |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 210 | def _CreateGoogleBinaryBlock(self, hardware_id): |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 211 | """Create a GBB for the image. |
| 212 | |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 213 | Args: |
| 214 | hardware_id: Hardware ID to use for this board. If None, then the |
| 215 | default from the Fdt will be used |
| 216 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 217 | Returns: |
| 218 | Path of the created GBB file. |
| 219 | |
| 220 | Raises: |
| 221 | CmdError if a command fails. |
| 222 | """ |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 223 | if not hardware_id: |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 224 | hardware_id = self.fdt.GetString('/config', 'hwid') |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 225 | gbb_size = self.fdt.GetFlashPartSize('ro', 'gbb') |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 226 | odir = self._tools.outdir |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 227 | |
Stefan Reinauer | 975e68f | 2012-02-27 13:27:08 -0800 | [diff] [blame] | 228 | chromeos_config = self.fdt.GetProps("/chromeos-config") |
| 229 | if 'fast-developer-mode' not in chromeos_config: |
| 230 | gbb_flags = 0 |
| 231 | else: |
| 232 | self._out.Notice("Enabling fast-developer-mode.") |
| 233 | gbb_flags = 1 |
| 234 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 235 | self._out.Progress('Creating GBB') |
| 236 | sizes = [0x100, 0x1000, gbb_size - 0x2180, 0x1000] |
| 237 | sizes = ['%#x' % size for size in sizes] |
| 238 | gbb = 'gbb.bin' |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 239 | keydir = self._tools.Filename(self._keydir) |
| 240 | self._tools.Run('gbb_utility', ['-c', ','.join(sizes), gbb], cwd=odir) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 241 | self._tools.Run('gbb_utility', ['-s', |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 242 | '--hwid=%s' % hardware_id, |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 243 | '--rootkey=%s/root_key.vbpubk' % keydir, |
| 244 | '--recoverykey=%s/recovery_key.vbpubk' % keydir, |
Simon Glass | 2a7f0b3 | 2011-08-26 11:25:17 -0700 | [diff] [blame] | 245 | '--bmpfv=%s' % self._tools.Filename(self.bmpblk_fname), |
Stefan Reinauer | 975e68f | 2012-02-27 13:27:08 -0800 | [diff] [blame] | 246 | '--flags=%d' % gbb_flags, |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 247 | gbb], |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 248 | cwd=odir) |
| 249 | return os.path.join(odir, gbb) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 250 | |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 251 | def _SignBootstub(self, bct, bootstub, text_base): |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 252 | """Sign an image so that the Tegra SOC will boot it. |
| 253 | |
| 254 | Args: |
| 255 | bct: BCT file to use. |
| 256 | bootstub: Boot stub (U-Boot + fdt) file to sign. |
| 257 | text_base: Address of text base for image. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 258 | |
| 259 | Returns: |
| 260 | filename of signed image. |
| 261 | |
| 262 | Raises: |
| 263 | CmdError if a command fails. |
| 264 | """ |
| 265 | # First create a config file - this is how we instruct cbootimage |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 266 | signed = os.path.join(self._tools.outdir, 'signed.bin') |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 267 | self._out.Progress('Signing Bootstub') |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 268 | config = os.path.join(self._tools.outdir, 'boot.cfg') |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 269 | fd = open(config, 'w') |
| 270 | fd.write('Version = 1;\n') |
| 271 | fd.write('Redundancy = 1;\n') |
| 272 | fd.write('Bctfile = %s;\n' % bct) |
Doug Anderson | 0eeb074 | 2011-09-15 18:11:40 -0700 | [diff] [blame] | 273 | |
| 274 | # TODO(dianders): Right now, we don't have enough space in our flash map |
| 275 | # for two copies of the BCT when we're using NAND, so hack it to 1. Not |
| 276 | # sure what this does for reliability, but at least things will fit... |
| 277 | is_nand = "NvBootDevType_Nand" in self._tools.Run('bct_dump', [bct]) |
| 278 | if is_nand: |
| 279 | fd.write('Bctcopy = 1;\n') |
| 280 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 281 | fd.write('BootLoader = %s,%#x,%#x,Complete;\n' % (bootstub, text_base, |
| 282 | text_base)) |
Doug Anderson | 0eeb074 | 2011-09-15 18:11:40 -0700 | [diff] [blame] | 283 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 284 | fd.close() |
| 285 | |
| 286 | self._tools.Run('cbootimage', [config, signed]) |
| 287 | self._tools.OutputSize('BCT', bct) |
| 288 | self._tools.OutputSize('Signed image', signed) |
| 289 | return signed |
| 290 | |
Doug Anderson | 86ce5f4 | 2011-07-27 10:40:18 -0700 | [diff] [blame] | 291 | def SetBootcmd(self, bootcmd, bootsecure): |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 292 | """Set the boot command for U-Boot. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 293 | |
| 294 | Args: |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 295 | bootcmd: Boot command to use, as a string (if None this this is a nop). |
Doug Anderson | 86ce5f4 | 2011-07-27 10:40:18 -0700 | [diff] [blame] | 296 | bootsecure: We'll set '/config/bootsecure' to 1 if True and 0 if False. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 297 | """ |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 298 | if bootcmd: |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 299 | self.fdt.PutString('/config', 'bootcmd', bootcmd) |
| 300 | self.fdt.PutInteger('/config', 'bootsecure', int(bootsecure)) |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 301 | self._out.Info('Boot command: %s' % bootcmd) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 302 | |
Simon Glass | a4934b7 | 2012-05-09 13:35:02 -0700 | [diff] [blame] | 303 | def SetNodeEnabled(self, node_name, enabled): |
| 304 | """Set whether an node is enabled or disabled. |
| 305 | |
| 306 | This simply sets the 'status' property of a node to "ok", or "disabled". |
| 307 | |
| 308 | The node should either be a full path to the node (like '/uart@10200000') |
| 309 | or an alias property. |
| 310 | |
| 311 | Aliases are supported like this: |
| 312 | |
| 313 | aliases { |
| 314 | console = "/uart@10200000"; |
| 315 | }; |
| 316 | |
| 317 | pointing to a node: |
| 318 | |
| 319 | uart@10200000 { |
| 320 | status = "ok"; |
| 321 | }; |
| 322 | |
| 323 | In this case, this function takes the name of the alias ('console' in |
| 324 | this case) and updates the status of the node that is pointed to, to |
| 325 | either ok or disabled. If the alias does not exist, a warning is |
| 326 | displayed. |
| 327 | |
| 328 | Args: |
| 329 | node_name: Name of node (e.g. '/uart@10200000') or alias alias |
| 330 | (e.g. 'console') to adjust |
| 331 | enabled: True to enable, False to disable |
| 332 | """ |
| 333 | # Look up the alias if this is an alias reference |
| 334 | if not node_name.startswith('/'): |
| 335 | lookup = self.fdt.GetString('/aliases', node_name, '') |
| 336 | if not lookup: |
| 337 | self._out.Warning("Cannot find alias '%s' - ignoring" % node_name) |
| 338 | return |
| 339 | node_name = lookup |
| 340 | if enabled: |
| 341 | status = 'ok' |
| 342 | else: |
| 343 | status = 'disabled' |
| 344 | self.fdt.PutString(node_name, 'status', status) |
| 345 | |
| 346 | def AddEnableList(self, enable_list): |
| 347 | """Process a list of nodes to enable/disable. |
| 348 | |
| 349 | Args: |
| 350 | config_list: List of (node, value) tuples to add to the fdt. For each |
| 351 | tuple: |
| 352 | node: The fdt node to write to will be <node> or pointed to by |
| 353 | /aliases/<node>. We can tell which |
| 354 | value: 0 to disable the node, 1 to enable it |
| 355 | """ |
| 356 | if enable_list: |
| 357 | for node_name, enabled in enable_list: |
| 358 | try: |
| 359 | enabled = int(enabled) |
| 360 | if enabled not in (0, 1): |
| 361 | raise ValueError |
| 362 | except ValueError as str: |
| 363 | raise CmdError("Invalid enable option value '%s' " |
| 364 | "(should be 0 or 1)" % enabled) |
| 365 | self.SetNodeEnabled(node_name, enabled) |
| 366 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 367 | def AddConfigList(self, config_list, use_int=False): |
| 368 | """Add a list of config items to the fdt. |
| 369 | |
| 370 | Normally these values are written to the fdt as strings, but integers |
| 371 | are also supported, in which case the values will be converted to integers |
| 372 | (if necessary) before being stored. |
| 373 | |
| 374 | Args: |
| 375 | config_list: List of (config, value) tuples to add to the fdt. For each |
| 376 | tuple: |
| 377 | config: The fdt node to write to will be /config/<config>. |
| 378 | value: An integer or string value to write. |
| 379 | use_int: True to only write integer values. |
| 380 | |
| 381 | Raises: |
| 382 | CmdError: if a value is required to be converted to integer but can't be. |
| 383 | """ |
| 384 | if config_list: |
| 385 | for config in config_list: |
| 386 | value = config[1] |
| 387 | if use_int: |
| 388 | try: |
| 389 | value = int(value) |
| 390 | except ValueError as str: |
| 391 | raise CmdError("Cannot convert config option '%s' to integer" % |
| 392 | value) |
| 393 | if type(value) == type(1): |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 394 | self.fdt.PutInteger('/config', '%s' % config[0], value) |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 395 | else: |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 396 | self.fdt.PutString('/config', '%s' % config[0], value) |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 397 | |
Simon Glass | 7c2d557 | 2011-11-15 14:47:08 -0800 | [diff] [blame] | 398 | def DecodeTextBase(self, data): |
| 399 | """Look at a U-Boot image and try to decode its TEXT_BASE. |
| 400 | |
| 401 | This works because U-Boot has a header with the value 0x12345678 |
| 402 | immediately followed by the TEXT_BASE value. We can therefore read this |
| 403 | from the image with some certainty. We check only the first 40 words |
| 404 | since the header should be within that region. |
| 405 | |
| 406 | Args: |
| 407 | data: U-Boot binary data |
| 408 | |
| 409 | Returns: |
| 410 | Text base (integer) or None if none was found |
| 411 | """ |
| 412 | found = False |
| 413 | for i in range(0, 160, 4): |
| 414 | word = data[i:i + 4] |
| 415 | |
| 416 | # TODO(sjg): This does not cope with a big-endian target |
| 417 | value = struct.unpack('<I', word)[0] |
| 418 | if found: |
| 419 | return value |
| 420 | if value == 0x12345678: |
| 421 | found = True |
| 422 | |
| 423 | return None |
| 424 | |
| 425 | def CalcTextBase(self, name, fdt, fname): |
| 426 | """Calculate the TEXT_BASE to use for U-Boot. |
| 427 | |
| 428 | Normally this value is in the fdt, so we just read it from there. But as |
| 429 | a second check we look at the image itself in case this is different, and |
| 430 | switch to that if it is. |
| 431 | |
| 432 | This allows us to flash any U-Boot even if its TEXT_BASE is different. |
| 433 | This is particularly useful with upstream U-Boot which uses a different |
| 434 | value (which we will move to). |
| 435 | """ |
| 436 | data = self._tools.ReadFile(fname) |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 437 | fdt_text_base = fdt.GetInt('/chromeos-config', 'textbase') |
Simon Glass | 7c2d557 | 2011-11-15 14:47:08 -0800 | [diff] [blame] | 438 | text_base = self.DecodeTextBase(data) |
Simon Glass | 60a40af | 2012-06-07 11:54:17 -0700 | [diff] [blame] | 439 | self._out.Info('TEXT_BASE: fdt says %#x, %s says %#x' % (fdt_text_base, |
| 440 | fname, text_base)) |
Simon Glass | 7c2d557 | 2011-11-15 14:47:08 -0800 | [diff] [blame] | 441 | |
| 442 | # If they are different, issue a warning and switch over. |
| 443 | if text_base and text_base != fdt_text_base: |
| 444 | self._out.Warning("TEXT_BASE %x in %sU-Boot doesn't match " |
| 445 | "fdt value of %x. Using %x" % (text_base, name, |
| 446 | fdt_text_base, text_base)) |
| 447 | fdt_text_base = text_base |
| 448 | return fdt_text_base |
| 449 | |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 450 | def _CreateBootStub(self, uboot, base_fdt, postload): |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 451 | """Create a boot stub and a signed boot stub. |
| 452 | |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 453 | For postload: |
| 454 | We add a /config/postload-text-offset entry to the signed bootstub's |
| 455 | fdt so that U-Boot can find the postload code. |
| 456 | |
| 457 | The raw (unsigned) bootstub will have a value of -1 for this since we will |
| 458 | simply append the postload code to the bootstub and it can find it there. |
| 459 | This will be used for RW A/B firmware. |
| 460 | |
| 461 | For the signed case this value will specify where in the flash to find |
| 462 | the postload code. This will be used for RO firmware. |
| 463 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 464 | Args: |
| 465 | uboot: Path to u-boot.bin (may be chroot-relative) |
Simon Glass | 29b96ad | 2012-03-09 15:34:33 -0800 | [diff] [blame] | 466 | base_fdt: Fdt object containing the flat device tree. |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 467 | postload: Path to u-boot-post.bin, or None if none. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 468 | |
| 469 | Returns: |
| 470 | Tuple containing: |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 471 | Full path to bootstub (uboot + fdt(-1) + postload). |
| 472 | Full path to signed (uboot + fdt(flash pos) + bct) + postload. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 473 | |
| 474 | Raises: |
| 475 | CmdError if a command fails. |
| 476 | """ |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 477 | bootstub = os.path.join(self._tools.outdir, 'u-boot-fdt.bin') |
Simon Glass | 7c2d557 | 2011-11-15 14:47:08 -0800 | [diff] [blame] | 478 | text_base = self.CalcTextBase('', self.fdt, uboot) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 479 | uboot_data = self._tools.ReadFile(uboot) |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 480 | |
| 481 | # Make a copy of the fdt for the bootstub |
| 482 | fdt = base_fdt.Copy(os.path.join(self._tools.outdir, 'bootstub.dtb')) |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 483 | fdt.PutInteger('/config', 'postload-text-offset', 0xffffffff); |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 484 | fdt_data = self._tools.ReadFile(fdt.fname) |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 485 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 486 | self._tools.WriteFile(bootstub, uboot_data + fdt_data) |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 487 | self._tools.OutputSize('U-Boot binary', self.uboot_fname) |
| 488 | self._tools.OutputSize('U-Boot fdt', self._fdt_fname) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 489 | self._tools.OutputSize('Combined binary', bootstub) |
| 490 | |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 491 | # Sign the bootstub; this is a combination of the board specific |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 492 | # bct and the stub u-boot image. |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 493 | signed = self._SignBootstub(self._tools.Filename(self.bct_fname), |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 494 | bootstub, text_base) |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 495 | |
| 496 | signed_postload = os.path.join(self._tools.outdir, 'signed-postload.bin') |
| 497 | data = self._tools.ReadFile(signed) |
| 498 | |
| 499 | if postload: |
| 500 | # We must add postload to the bootstub since A and B will need to |
| 501 | # be able to find it without the /config/postload-text-offset mechanism. |
| 502 | bs_data = self._tools.ReadFile(bootstub) |
| 503 | bs_data += self._tools.ReadFile(postload) |
| 504 | bootstub = os.path.join(self._tools.outdir, 'u-boot-fdt-postload.bin') |
| 505 | self._tools.WriteFile(bootstub, bs_data) |
| 506 | self._tools.OutputSize('Combined binary with postload', bootstub) |
| 507 | |
| 508 | # Now that we know the file size, adjust the fdt and re-sign |
| 509 | postload_bootstub = os.path.join(self._tools.outdir, 'postload.bin') |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 510 | fdt.PutInteger('/config', 'postload-text-offset', len(data)) |
Simon Glass | 6dcc2f2 | 2011-07-28 15:26:49 +1200 | [diff] [blame] | 511 | fdt_data = self._tools.ReadFile(fdt.fname) |
| 512 | self._tools.WriteFile(postload_bootstub, uboot_data + fdt_data) |
| 513 | signed = self._SignBootstub(self._tools.Filename(self.bct_fname), |
| 514 | postload_bootstub, text_base) |
| 515 | if len(data) != os.path.getsize(signed): |
| 516 | raise CmdError('Signed file size changed from %d to %d after updating ' |
| 517 | 'fdt' % (len(data), os.path.getsize(signed))) |
| 518 | |
| 519 | # Re-read the signed image, and add the post-load binary. |
| 520 | data = self._tools.ReadFile(signed) |
| 521 | data += self._tools.ReadFile(postload) |
| 522 | self._tools.OutputSize('Post-load binary', postload) |
| 523 | |
| 524 | self._tools.WriteFile(signed_postload, data) |
| 525 | self._tools.OutputSize('Final bootstub with postload', signed_postload) |
| 526 | |
| 527 | return bootstub, signed_postload |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 528 | |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 529 | def _CreateCorebootStub(self, uboot, coreboot, fdt, seabios): |
Stefan Reinauer | c2e1e4d | 2011-08-23 14:50:59 -0700 | [diff] [blame] | 530 | """Create a coreboot boot stub. |
| 531 | |
| 532 | Args: |
| 533 | uboot: Path to u-boot.bin (may be chroot-relative) |
| 534 | coreboot: Path to coreboot.rom |
| 535 | fdt: Device Tree |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 536 | seabios: Path to SeaBIOS payload binary or None |
Stefan Reinauer | c2e1e4d | 2011-08-23 14:50:59 -0700 | [diff] [blame] | 537 | |
| 538 | Returns: |
| 539 | Full path to bootstub (coreboot + uboot + fdt). |
| 540 | |
| 541 | Raises: |
| 542 | CmdError if a command fails. |
| 543 | """ |
| 544 | bootstub = os.path.join(self._tools.outdir, 'coreboot-full.rom') |
Stefan Reinauer | c2e1e4d | 2011-08-23 14:50:59 -0700 | [diff] [blame] | 545 | uboot_elf = uboot.replace(".bin", ".elf") |
| 546 | shutil.copyfile(coreboot, bootstub) |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 547 | if seabios: |
Simon Glass | 146cb8e | 2012-03-09 15:51:24 -0800 | [diff] [blame] | 548 | self._tools.Run('cbfstool', [bootstub, 'add-payload', seabios, |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 549 | 'fallback/payload', 'lzma']) |
Simon Glass | 146cb8e | 2012-03-09 15:51:24 -0800 | [diff] [blame] | 550 | self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf, |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 551 | 'img/U-Boot', 'lzma']) |
| 552 | else: |
Simon Glass | 146cb8e | 2012-03-09 15:51:24 -0800 | [diff] [blame] | 553 | self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf, |
Vincent Palatin | f728677 | 2011-10-12 14:31:53 -0700 | [diff] [blame] | 554 | 'fallback/payload', 'lzma']) |
Simon Glass | 146cb8e | 2012-03-09 15:51:24 -0800 | [diff] [blame] | 555 | self._tools.Run('cbfstool', [bootstub, 'add', fdt.fname, 'u-boot.dtb', |
Stefan Reinauer | c2e1e4d | 2011-08-23 14:50:59 -0700 | [diff] [blame] | 556 | '0xac']) |
| 557 | return bootstub |
| 558 | |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 559 | def _UpdateBl2Parameters(self, fdt, spl_load_size, data, pos): |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 560 | """Update the parameters in a BL2 blob. |
| 561 | |
| 562 | We look at the list in the parameter block, extract the value of each |
| 563 | from the device tree, and write that value to the parameter block. |
| 564 | |
| 565 | Args: |
| 566 | fdt: Device tree containing the parameter values. |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 567 | spl_load_size: Size of U-Boot image that SPL must load |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 568 | data: The BL2 data. |
| 569 | pos: The position of the start of the parameter block. |
| 570 | |
| 571 | Returns: |
| 572 | The new contents of the parameter block, after updating. |
| 573 | """ |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 574 | version, size = struct.unpack('<2L', data[pos + 4:pos + 12]) |
| 575 | if version != 1: |
| 576 | raise CmdError("Cannot update machine parameter block version '%d'" % |
| 577 | version) |
| 578 | if size < 0 or pos + size > len(data): |
Simon Glass | 7dbdf5b | 2012-03-15 20:58:04 -0700 | [diff] [blame] | 579 | raise CmdError("Machine parameter block size %d is invalid: " |
| 580 | "pos=%d, size=%d, space=%d, len=%d" % |
| 581 | (size, pos, size, len(data) - pos, len(data))) |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 582 | |
| 583 | # Move past the header and read the parameter list, which is terminated |
| 584 | # with \0. |
| 585 | pos += 12 |
| 586 | param_list = struct.unpack('<%ds' % (len(data) - pos), data[pos:])[0] |
| 587 | param_len = param_list.find('\0') |
| 588 | param_list = param_list[:param_len] |
Simon Glass | 66c1a9f | 2012-03-22 19:15:53 -0700 | [diff] [blame] | 589 | pos += (param_len + 4) & ~3 |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 590 | |
| 591 | # Work through the parameters one at a time, adding each value |
| 592 | new_data = '' |
Simon Glass | 7dbdf5b | 2012-03-15 20:58:04 -0700 | [diff] [blame] | 593 | upto = 0 |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 594 | for param in param_list: |
Simon Glass | 2c48ddf | 2012-03-23 16:55:22 -0700 | [diff] [blame] | 595 | value = struct.unpack('<1L', data[pos + upto:pos + upto + 4])[0] |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 596 | if param == 'm' : |
Simon Glass | 1a77ded | 2012-03-15 21:00:49 -0700 | [diff] [blame] | 597 | mem_type = fdt.GetString('/dmc', 'mem-type') |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 598 | mem_types = ['ddr2', 'ddr3', 'lpddr2', 'lpddr3'] |
| 599 | if not mem_type in mem_types: |
| 600 | raise CmdError("Unknown memory type '%s'" % mem_type) |
| 601 | value = mem_types.index(mem_type) |
| 602 | self._out.Info(' Memory type: %s (%d)' % (mem_type, value)) |
Doug Anderson | ee46cfe | 2012-05-18 09:53:08 -0700 | [diff] [blame] | 603 | elif param == 'M' : |
| 604 | mem_manuf = fdt.GetString('/dmc', 'mem-manuf') |
| 605 | mem_manufs = ['autodetect', 'elpida', 'samsung'] |
| 606 | if not mem_manuf in mem_manufs: |
| 607 | raise CmdError("Unknown memory manufacturer: '%s'" % mem_manuf) |
| 608 | value = mem_manufs.index(mem_manuf) |
| 609 | self._out.Info(' Memory manufacturer: %s (%d)' % (mem_manuf, value)) |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 610 | elif param == 'v': |
| 611 | value = 31 |
| 612 | self._out.Info(' Memory interleave: %#0x' % value) |
Simon Glass | 8e1fdb2 | 2012-03-15 21:02:10 -0700 | [diff] [blame] | 613 | elif param == 'u': |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 614 | value = (spl_load_size + 0xfff) & ~0xfff |
| 615 | self._out.Info(' U-Boot size: %#0x (rounded up from %#0x)' % |
| 616 | (value, spl_load_size)) |
Simon Glass | 559b661 | 2012-05-23 13:28:45 -0700 | [diff] [blame] | 617 | elif param == 'b': |
| 618 | # These values come from enum boot_mode in U-Boot's cpu.h |
| 619 | if self.spl_source == 'straps': |
| 620 | value = 32 |
| 621 | elif self.spl_source == 'emmc': |
| 622 | value = 4 |
| 623 | elif self.spl_source == 'spi': |
| 624 | value = 20 |
| 625 | elif self.spl_source == 'usb': |
| 626 | value = 33 |
| 627 | else: |
| 628 | raise CmdError("Invalid boot source '%s'" % self.spl_source) |
| 629 | self._out.Info(' Boot source: %#0x' % value) |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 630 | else: |
Simon Glass | 7dbdf5b | 2012-03-15 20:58:04 -0700 | [diff] [blame] | 631 | self._out.Warning("Unknown machine parameter type '%s'" % param) |
Simon Glass | 2c48ddf | 2012-03-23 16:55:22 -0700 | [diff] [blame] | 632 | self._out.Info(' Unknown value: %#0x' % value) |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 633 | new_data += struct.pack('<L', value) |
Simon Glass | 7dbdf5b | 2012-03-15 20:58:04 -0700 | [diff] [blame] | 634 | upto += 4 |
Simon Glass | df95dd2 | 2012-03-13 15:46:16 -0700 | [diff] [blame] | 635 | |
| 636 | # Put the data into our block. |
| 637 | data = data[:pos] + new_data + data[pos + len(new_data):] |
| 638 | self._out.Info('BL2 configuration complete') |
| 639 | return data |
| 640 | |
Simon Glass | e5e8afb | 2012-05-23 11:19:23 -0700 | [diff] [blame] | 641 | def _UpdateChecksum(self, data): |
| 642 | """Update the BL2 checksum. |
| 643 | |
| 644 | The checksum is a 4 byte sum of all the bytes in the image before the |
| 645 | last 4 bytes (which hold the checksum). |
| 646 | |
| 647 | Args: |
| 648 | data: The BL2 data to update. |
| 649 | |
| 650 | Returns: |
| 651 | The new contents of the BL2 data, after updating the checksum. |
| 652 | """ |
| 653 | checksum = 0 |
| 654 | for ch in data[:-4]: |
| 655 | checksum += ord(ch) |
| 656 | return data[:-4] + struct.pack('<L', checksum & 0xffffffff) |
| 657 | |
Simon Glass | 559b661 | 2012-05-23 13:28:45 -0700 | [diff] [blame] | 658 | def ConfigureExynosBl2(self, fdt, spl_load_size, orig_bl2, name=''): |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 659 | """Configure an Exynos BL2 binary for our needs. |
| 660 | |
| 661 | We create a new modified BL2 and return its filename. |
| 662 | |
| 663 | Args: |
| 664 | fdt: Device tree containing the parameter values. |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 665 | spl_load_size: Size of U-Boot image that SPL must load |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 666 | orig_bl2: Filename of original BL2 file to modify. |
| 667 | """ |
Simon Glass | 66c1a9f | 2012-03-22 19:15:53 -0700 | [diff] [blame] | 668 | self._out.Info('Configuring BL2') |
Simon Glass | 559b661 | 2012-05-23 13:28:45 -0700 | [diff] [blame] | 669 | bl2 = os.path.join(self._tools.outdir, 'updated-spl%s.bin' % name) |
Simon Glass | 66c1a9f | 2012-03-22 19:15:53 -0700 | [diff] [blame] | 670 | data = self._tools.ReadFile(orig_bl2) |
| 671 | self._tools.WriteFile(bl2, data) |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 672 | |
| 673 | # Locate the parameter block |
| 674 | data = self._tools.ReadFile(bl2) |
| 675 | marker = struct.pack('<L', 0xdeadbeef) |
| 676 | pos = data.rfind(marker) |
| 677 | if not pos: |
| 678 | raise CmdError("Could not find machine parameter block in '%s'" % |
| 679 | orig_bl2) |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 680 | data = self._UpdateBl2Parameters(fdt, spl_load_size, data, pos) |
Simon Glass | e5e8afb | 2012-05-23 11:19:23 -0700 | [diff] [blame] | 681 | data = self._UpdateChecksum(data) |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 682 | self._tools.WriteFile(bl2, data) |
| 683 | return bl2 |
| 684 | |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 685 | def _PackOutput(self, msg): |
| 686 | """Helper function to write output from PackFirmware (verbose level 2). |
| 687 | |
| 688 | This is passed to PackFirmware for it to use to write output. |
| 689 | |
| 690 | Args: |
| 691 | msg: Message to display. |
| 692 | """ |
| 693 | self._out.Notice(msg) |
| 694 | |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 695 | def _BuildBlob(self, pack, fdt, blob_type): |
| 696 | """Build the blob data for a particular blob type. |
| 697 | |
| 698 | Args: |
| 699 | blob_type: The type of blob to create data for. Supported types are: |
| 700 | coreboot A coreboot image (ROM plus U-boot and .dtb payloads). |
| 701 | signed Nvidia T20/T30 signed image (BCT, U-Boot, .dtb). |
| 702 | """ |
| 703 | if blob_type == 'coreboot': |
| 704 | coreboot = self._CreateCorebootStub(self.uboot_fname, |
| 705 | self.coreboot_fname, fdt, self.seabios_fname) |
| 706 | pack.AddProperty('coreboot', coreboot) |
| 707 | pack.AddProperty('image', coreboot) |
| 708 | elif blob_type == 'signed': |
| 709 | bootstub, signed = self._CreateBootStub(self.uboot_fname, fdt, |
| 710 | self.postload_fname) |
| 711 | pack.AddProperty('bootstub', bootstub) |
| 712 | pack.AddProperty('signed', signed) |
| 713 | pack.AddProperty('image', signed) |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 714 | elif blob_type == 'exynos-bl1': |
| 715 | pack.AddProperty(blob_type, self.exynos_bl1) |
| 716 | elif blob_type == 'exynos-bl2': |
Simon Glass | 3b40409 | 2012-05-23 13:10:36 -0700 | [diff] [blame] | 717 | spl_load_size = os.stat(pack.GetProperty('boot+dtb')).st_size |
| 718 | bl2 = self.ConfigureExynosBl2(fdt, spl_load_size, self.exynos_bl2) |
Simon Glass | 7e19922 | 2012-03-13 15:51:18 -0700 | [diff] [blame] | 719 | pack.AddProperty(blob_type, bl2) |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 720 | elif pack.GetProperty(blob_type): |
| 721 | pass |
| 722 | else: |
| 723 | raise CmdError("Unknown blob type '%s' required in flash map" % |
| 724 | blob_type) |
| 725 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 726 | def _CreateImage(self, gbb, fdt): |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 727 | """Create a full firmware image, along with various by-products. |
| 728 | |
| 729 | This uses the provided u-boot.bin, fdt and bct to create a firmware |
| 730 | image containing all the required parts. If the GBB is not supplied |
| 731 | then this will just return a signed U-Boot as the image. |
| 732 | |
| 733 | Args: |
Simon Glass | e13ee2c | 2011-07-28 08:12:28 +1200 | [diff] [blame] | 734 | gbb: Full path to the GBB file, or empty if a GBB is not required. |
| 735 | fdt: Fdt object containing required information. |
| 736 | |
| 737 | Returns: |
| 738 | Path to image file |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 739 | |
| 740 | Raises: |
| 741 | CmdError if a command fails. |
| 742 | """ |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 743 | self._out.Notice("Model: %s" % fdt.GetString('/', 'model')) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 744 | |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 745 | # Get the flashmap so we know what to build |
| 746 | pack = PackFirmware(self._tools, self._out) |
| 747 | pack.SelectFdt(fdt) |
| 748 | |
| 749 | # Get all our blobs ready |
| 750 | pack.AddProperty('boot', self.uboot_fname) |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 751 | pack.AddProperty('skeleton', self.skeleton_fname) |
Simon Glass | 50f7460 | 2012-03-15 21:04:25 -0700 | [diff] [blame] | 752 | |
| 753 | # Make a copy of the fdt for the bootstub |
| 754 | fdt_data = self._tools.ReadFile(fdt.fname) |
| 755 | uboot_data = self._tools.ReadFile(self.uboot_fname) |
| 756 | uboot_copy = os.path.join(self._tools.outdir, 'u-boot.bin') |
| 757 | self._tools.WriteFile(uboot_copy, uboot_data) |
| 758 | |
| 759 | bootstub = os.path.join(self._tools.outdir, 'u-boot-dtb.bin') |
| 760 | self._tools.WriteFile(bootstub, uboot_data + fdt_data) |
| 761 | pack.AddProperty('boot+dtb', bootstub) |
| 762 | |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 763 | pack.AddProperty('gbb', self.uboot_fname) |
Simon Glass | 0726795 | 2012-06-08 12:45:13 -0700 | [diff] [blame] | 764 | for blob_type in pack.GetBlobList(): |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 765 | self._BuildBlob(pack, fdt, blob_type) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 766 | |
| 767 | if gbb: |
Simon Glass | e76bf7b | 2012-03-13 15:34:41 -0700 | [diff] [blame] | 768 | pack.RequireAllEntries() |
Hung-Te Lin | a7462e7 | 2011-07-27 19:17:10 +0800 | [diff] [blame] | 769 | fwid = '.'.join([ |
Simon Glass | 02d124a | 2012-03-02 14:47:20 -0800 | [diff] [blame] | 770 | re.sub('[ ,]+', '_', fdt.GetString('/', 'model')), |
Hung-Te Lin | a7462e7 | 2011-07-27 19:17:10 +0800 | [diff] [blame] | 771 | self._tools.GetChromeosVersion()]) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 772 | self._out.Notice('Firmware ID: %s' % fwid) |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 773 | pack.AddProperty('fwid', fwid) |
| 774 | pack.AddProperty('gbb', gbb) |
| 775 | pack.AddProperty('keydir', self._keydir) |
Simon Glass | c90cf58 | 2012-03-13 15:40:47 -0700 | [diff] [blame] | 776 | |
| 777 | pack.CheckProperties() |
| 778 | image = os.path.join(self._tools.outdir, 'image.bin') |
| 779 | pack.PackImage(self._tools.outdir, image) |
| 780 | pack.AddProperty('image', image) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 781 | |
Simon Glass | 439fe7a | 2012-03-09 16:19:34 -0800 | [diff] [blame] | 782 | image = pack.GetProperty('image') |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 783 | self._tools.OutputSize('Final image', image) |
Simon Glass | c90cf58 | 2012-03-13 15:40:47 -0700 | [diff] [blame] | 784 | return image, pack |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 785 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 786 | def SelectFdt(self, fdt_fname): |
| 787 | """Select an FDT to control the firmware bundling |
| 788 | |
| 789 | Args: |
| 790 | fdt_fname: The filename of the fdt to use. |
| 791 | |
Simon Glass | c0f3dc6 | 2011-08-09 14:19:05 -0700 | [diff] [blame] | 792 | Returns: |
| 793 | The Fdt object of the original fdt file, which we will not modify. |
| 794 | |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 795 | We make a copy of this which will include any on-the-fly changes we want |
| 796 | to make. |
| 797 | """ |
| 798 | self._fdt_fname = fdt_fname |
| 799 | self.CheckOptions() |
| 800 | fdt = Fdt(self._tools, self._fdt_fname) |
Simon Glass | 29b96ad | 2012-03-09 15:34:33 -0800 | [diff] [blame] | 801 | fdt.Compile() |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 802 | self.fdt = fdt.Copy(os.path.join(self._tools.outdir, 'updated.dtb')) |
Simon Glass | c0f3dc6 | 2011-08-09 14:19:05 -0700 | [diff] [blame] | 803 | return fdt |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 804 | |
Simon Glass | c90cf58 | 2012-03-13 15:40:47 -0700 | [diff] [blame] | 805 | def Start(self, hardware_id, output_fname, show_map): |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 806 | """This creates a firmware bundle according to settings provided. |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 807 | |
| 808 | - Checks options, tools, output directory, fdt. |
| 809 | - Creates GBB and image. |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 810 | |
| 811 | Args: |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 812 | hardware_id: Hardware ID to use for this board. If None, then the |
| 813 | default from the Fdt will be used |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 814 | output_fname: Output filename for the image. If this is not None, then |
| 815 | the final image will be copied here. |
Simon Glass | c90cf58 | 2012-03-13 15:40:47 -0700 | [diff] [blame] | 816 | show_map: Show a flash map, with each area's name and position |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 817 | |
| 818 | Returns: |
| 819 | Filename of the resulting image (not the output_fname copy). |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 820 | """ |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 821 | gbb = '' |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 822 | if not self._small: |
Simon Glass | 5657757 | 2011-07-19 11:08:06 +1200 | [diff] [blame] | 823 | gbb = self._CreateGoogleBinaryBlock(hardware_id) |
Simon Glass | 89b86b8 | 2011-07-17 23:49:49 -0700 | [diff] [blame] | 824 | |
| 825 | # This creates the actual image. |
Simon Glass | c90cf58 | 2012-03-13 15:40:47 -0700 | [diff] [blame] | 826 | image, pack = self._CreateImage(gbb, self.fdt) |
| 827 | if show_map: |
| 828 | pack.ShowMap() |
Simon Glass | 290a180 | 2011-07-17 13:54:32 -0700 | [diff] [blame] | 829 | if output_fname: |
| 830 | shutil.copyfile(image, output_fname) |
| 831 | self._out.Notice("Output image '%s'" % output_fname) |
Simon Glass | 794217e | 2012-06-07 11:40:37 -0700 | [diff] [blame] | 832 | return image, pack.props |