blob: 16ee589e959c24e6019a2c4afba09e772e15e94f [file] [log] [blame]
Simon Glass89b86b82011-07-17 23:49:49 -07001# 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
7This modules uses a few rudimentary other libraries for its activity.
8
9Here are the names we give to the various files we deal with. It is important
10to 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 Glassceff3ff2012-04-04 11:23:45 -070019import glob
Simon Glass89b86b82011-07-17 23:49:49 -070020import os
21import re
22
23import cros_output
24from fdt import Fdt
25from pack_firmware import PackFirmware
26import shutil
Simon Glass7c2d5572011-11-15 14:47:08 -080027import struct
Simon Glass89b86b82011-07-17 23:49:49 -070028import tempfile
Simon Glass439fe7a2012-03-09 16:19:34 -080029from tools import CmdError
Simon Glass89b86b82011-07-17 23:49:49 -070030from tools import Tools
31from 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.
36yaml_data = '''
37bmpblock: 1.0
38
39images:
40 devmode: DeveloperBmp/DeveloperBmp.bmp
41 recovery: RecoveryBmp/RecoveryBmp.bmp
42 rec_yuck: RecoveryNoOSBmp/RecoveryNoOSBmp.bmp
43 rec_insert: RecoveryMissingOSBmp/RecoveryMissingOSBmp.bmp
44
45screens:
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
55localizations:
56 - [ dev_en, rec_en, yuck_en, ins_en ]
57'''
58
59class Bundle:
Simon Glass290a1802011-07-17 13:54:32 -070060 """This class encapsulates the entire bundle firmware logic.
Simon Glass89b86b82011-07-17 23:49:49 -070061
Simon Glass290a1802011-07-17 13:54:32 -070062 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 Glassa4934b72012-05-09 13:35:02 -070068 .. can call bundle.AddConfigList(), AddEnableList() if required
Simon Glass290a1802011-07-17 13:54:32 -070069 bundle.Start(...)
Simon Glass89b86b82011-07-17 23:49:49 -070070
Simon Glass290a1802011-07-17 13:54:32 -070071 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 Glass559b6612012-05-23 13:28:45 -070078 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 Glass290a1802011-07-17 13:54:32 -070082 """
Simon Glass89b86b82011-07-17 23:49:49 -070083
Simon Glass290a1802011-07-17 13:54:32 -070084 def __init__(self, tools, output):
85 """Set up a new Bundle object.
Simon Glass89b86b82011-07-17 23:49:49 -070086
Simon Glass290a1802011-07-17 13:54:32 -070087 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 Glass89b86b82011-07-17 23:49:49 -070090 """
Simon Glass290a1802011-07-17 13:54:32 -070091 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 Lin5b649382011-08-03 15:01:16 +0800100 self.bmpblk_fname = None # Filename of our Bitmap Block
Stefan Reinauer8d79d362011-08-16 14:20:43 -0700101 self.coreboot_fname = None # Filename of our coreboot binary.
Vincent Palatinf7286772011-10-12 14:31:53 -0700102 self.seabios_fname = None # Filename of our SeaBIOS payload.
Simon Glass7e199222012-03-13 15:51:18 -0700103 self.exynos_bl1 = None # Filename of Exynos BL1 (pre-boot)
104 self.exynos_bl2 = None # Filename of Exynos BL2 (SPL)
Simon Glass559b6612012-05-23 13:28:45 -0700105 self.spl_source = 'straps' # SPL boot according to board settings
Simon Glass07267952012-06-08 12:45:13 -0700106 self.skeleton_fname = None # Filename of Coreboot skeleton file
Simon Glass290a1802011-07-17 13:54:32 -0700107
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 Glass6dcc2f22011-07-28 15:26:49 +1200116 def SetFiles(self, board, bct, uboot=None, bmpblk=None, coreboot=None,
Simon Glass07267952012-06-08 12:45:13 -0700117 postload=None, seabios=None, exynos_bl1=None, exynos_bl2=None,
118 skeleton=None):
Simon Glass290a1802011-07-17 13:54:32 -0700119 """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 Lin5b649382011-08-03 15:01:16 +0800125 bmpblk: The filename of bitmap block file to use.
Stefan Reinauer8d79d362011-08-16 14:20:43 -0700126 coreboot: The filename of the coreboot image to use (on x86)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200127 postload: The filename of the u-boot-post.bin image to use.
Vincent Palatinf7286772011-10-12 14:31:53 -0700128 seabios: The filename of the SeaBIOS payload to use if any.
Simon Glass07267952012-06-08 12:45:13 -0700129 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 Glass290a1802011-07-17 13:54:32 -0700132 """
133 self._board = board
134 self.uboot_fname = uboot
135 self.bct_fname = bct
Hung-Te Lin5b649382011-08-03 15:01:16 +0800136 self.bmpblk_fname = bmpblk
Stefan Reinauer8d79d362011-08-16 14:20:43 -0700137 self.coreboot_fname = coreboot
Simon Glass6dcc2f22011-07-28 15:26:49 +1200138 self.postload_fname = postload
Vincent Palatinf7286772011-10-12 14:31:53 -0700139 self.seabios_fname = seabios
Simon Glass7e199222012-03-13 15:51:18 -0700140 self.exynos_bl1 = exynos_bl1
141 self.exynos_bl2 = exynos_bl2
Simon Glass07267952012-06-08 12:45:13 -0700142 self.skeleton_fname = skeleton
Simon Glass290a1802011-07-17 13:54:32 -0700143
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 Glass493163b2011-09-14 11:19:57 -0700158 build_root = os.path.join('##', 'build', self._board, 'firmware')
Simon Glass881964d2012-04-04 11:34:09 -0700159 dir_name = os.path.join(build_root, 'dts')
Simon Glass290a1802011-07-17 13:54:32 -0700160 if not self._fdt_fname:
Simon Glassceff3ff2012-04-04 11:23:45 -0700161 # Figure out where the file should be, and the name we expect.
Simon Glassceff3ff2012-04-04 11:23:45 -0700162 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 Glass881964d2012-04-04 11:34:09 -0700173 # 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 Glass290a1802011-07-17 13:54:32 -0700178 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 Glass2a7f0b32011-08-26 11:25:17 -0700182 if not self.bmpblk_fname:
183 self.bmpblk_fname = os.path.join(build_root, 'default.bmpblk')
Simon Glass1d376832012-03-15 20:50:54 -0700184 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 Glass07267952012-06-08 12:45:13 -0700188 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 Glass89b86b82011-07-17 23:49:49 -0700192
Simon Glass75759302012-03-15 20:26:53 -0700193 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 Glass56577572011-07-19 11:08:06 +1200210 def _CreateGoogleBinaryBlock(self, hardware_id):
Simon Glass89b86b82011-07-17 23:49:49 -0700211 """Create a GBB for the image.
212
Simon Glass56577572011-07-19 11:08:06 +1200213 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 Glass89b86b82011-07-17 23:49:49 -0700217 Returns:
218 Path of the created GBB file.
219
220 Raises:
221 CmdError if a command fails.
222 """
Simon Glass56577572011-07-19 11:08:06 +1200223 if not hardware_id:
Simon Glass02d124a2012-03-02 14:47:20 -0800224 hardware_id = self.fdt.GetString('/config', 'hwid')
Simon Glass89b86b82011-07-17 23:49:49 -0700225 gbb_size = self.fdt.GetFlashPartSize('ro', 'gbb')
Simon Glass290a1802011-07-17 13:54:32 -0700226 odir = self._tools.outdir
Simon Glass89b86b82011-07-17 23:49:49 -0700227
Stefan Reinauer975e68f2012-02-27 13:27:08 -0800228 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 Glass89b86b82011-07-17 23:49:49 -0700235 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 Glass290a1802011-07-17 13:54:32 -0700239 keydir = self._tools.Filename(self._keydir)
240 self._tools.Run('gbb_utility', ['-c', ','.join(sizes), gbb], cwd=odir)
Simon Glass89b86b82011-07-17 23:49:49 -0700241 self._tools.Run('gbb_utility', ['-s',
Simon Glass56577572011-07-19 11:08:06 +1200242 '--hwid=%s' % hardware_id,
Simon Glass89b86b82011-07-17 23:49:49 -0700243 '--rootkey=%s/root_key.vbpubk' % keydir,
244 '--recoverykey=%s/recovery_key.vbpubk' % keydir,
Simon Glass2a7f0b32011-08-26 11:25:17 -0700245 '--bmpfv=%s' % self._tools.Filename(self.bmpblk_fname),
Stefan Reinauer975e68f2012-02-27 13:27:08 -0800246 '--flags=%d' % gbb_flags,
Simon Glass89b86b82011-07-17 23:49:49 -0700247 gbb],
Simon Glass290a1802011-07-17 13:54:32 -0700248 cwd=odir)
249 return os.path.join(odir, gbb)
Simon Glass89b86b82011-07-17 23:49:49 -0700250
Simon Glasse13ee2c2011-07-28 08:12:28 +1200251 def _SignBootstub(self, bct, bootstub, text_base):
Simon Glass89b86b82011-07-17 23:49:49 -0700252 """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 Glass89b86b82011-07-17 23:49:49 -0700258
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 Glasse13ee2c2011-07-28 08:12:28 +1200266 signed = os.path.join(self._tools.outdir, 'signed.bin')
Simon Glass89b86b82011-07-17 23:49:49 -0700267 self._out.Progress('Signing Bootstub')
Simon Glasse13ee2c2011-07-28 08:12:28 +1200268 config = os.path.join(self._tools.outdir, 'boot.cfg')
Simon Glass89b86b82011-07-17 23:49:49 -0700269 fd = open(config, 'w')
270 fd.write('Version = 1;\n')
271 fd.write('Redundancy = 1;\n')
272 fd.write('Bctfile = %s;\n' % bct)
Doug Anderson0eeb0742011-09-15 18:11:40 -0700273
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 Glass89b86b82011-07-17 23:49:49 -0700281 fd.write('BootLoader = %s,%#x,%#x,Complete;\n' % (bootstub, text_base,
282 text_base))
Doug Anderson0eeb0742011-09-15 18:11:40 -0700283
Simon Glass89b86b82011-07-17 23:49:49 -0700284 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 Anderson86ce5f42011-07-27 10:40:18 -0700291 def SetBootcmd(self, bootcmd, bootsecure):
Simon Glass290a1802011-07-17 13:54:32 -0700292 """Set the boot command for U-Boot.
Simon Glass89b86b82011-07-17 23:49:49 -0700293
294 Args:
Simon Glass290a1802011-07-17 13:54:32 -0700295 bootcmd: Boot command to use, as a string (if None this this is a nop).
Doug Anderson86ce5f42011-07-27 10:40:18 -0700296 bootsecure: We'll set '/config/bootsecure' to 1 if True and 0 if False.
Simon Glass89b86b82011-07-17 23:49:49 -0700297 """
Simon Glass290a1802011-07-17 13:54:32 -0700298 if bootcmd:
Simon Glass02d124a2012-03-02 14:47:20 -0800299 self.fdt.PutString('/config', 'bootcmd', bootcmd)
300 self.fdt.PutInteger('/config', 'bootsecure', int(bootsecure))
Simon Glass290a1802011-07-17 13:54:32 -0700301 self._out.Info('Boot command: %s' % bootcmd)
Simon Glass89b86b82011-07-17 23:49:49 -0700302
Simon Glassa4934b72012-05-09 13:35:02 -0700303 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 Glass290a1802011-07-17 13:54:32 -0700367 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 Glass02d124a2012-03-02 14:47:20 -0800394 self.fdt.PutInteger('/config', '%s' % config[0], value)
Simon Glass290a1802011-07-17 13:54:32 -0700395 else:
Simon Glass02d124a2012-03-02 14:47:20 -0800396 self.fdt.PutString('/config', '%s' % config[0], value)
Simon Glass290a1802011-07-17 13:54:32 -0700397
Simon Glass7c2d5572011-11-15 14:47:08 -0800398 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 Glass02d124a2012-03-02 14:47:20 -0800437 fdt_text_base = fdt.GetInt('/chromeos-config', 'textbase')
Simon Glass7c2d5572011-11-15 14:47:08 -0800438 text_base = self.DecodeTextBase(data)
Simon Glass60a40af2012-06-07 11:54:17 -0700439 self._out.Info('TEXT_BASE: fdt says %#x, %s says %#x' % (fdt_text_base,
440 fname, text_base))
Simon Glass7c2d5572011-11-15 14:47:08 -0800441
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 Glass6dcc2f22011-07-28 15:26:49 +1200450 def _CreateBootStub(self, uboot, base_fdt, postload):
Simon Glass89b86b82011-07-17 23:49:49 -0700451 """Create a boot stub and a signed boot stub.
452
Simon Glass6dcc2f22011-07-28 15:26:49 +1200453 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 Glass89b86b82011-07-17 23:49:49 -0700464 Args:
465 uboot: Path to u-boot.bin (may be chroot-relative)
Simon Glass29b96ad2012-03-09 15:34:33 -0800466 base_fdt: Fdt object containing the flat device tree.
Simon Glass6dcc2f22011-07-28 15:26:49 +1200467 postload: Path to u-boot-post.bin, or None if none.
Simon Glass89b86b82011-07-17 23:49:49 -0700468
469 Returns:
470 Tuple containing:
Simon Glass6dcc2f22011-07-28 15:26:49 +1200471 Full path to bootstub (uboot + fdt(-1) + postload).
472 Full path to signed (uboot + fdt(flash pos) + bct) + postload.
Simon Glass89b86b82011-07-17 23:49:49 -0700473
474 Raises:
475 CmdError if a command fails.
476 """
Simon Glasse13ee2c2011-07-28 08:12:28 +1200477 bootstub = os.path.join(self._tools.outdir, 'u-boot-fdt.bin')
Simon Glass7c2d5572011-11-15 14:47:08 -0800478 text_base = self.CalcTextBase('', self.fdt, uboot)
Simon Glass89b86b82011-07-17 23:49:49 -0700479 uboot_data = self._tools.ReadFile(uboot)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200480
481 # Make a copy of the fdt for the bootstub
482 fdt = base_fdt.Copy(os.path.join(self._tools.outdir, 'bootstub.dtb'))
Simon Glass02d124a2012-03-02 14:47:20 -0800483 fdt.PutInteger('/config', 'postload-text-offset', 0xffffffff);
Simon Glass290a1802011-07-17 13:54:32 -0700484 fdt_data = self._tools.ReadFile(fdt.fname)
Simon Glasse13ee2c2011-07-28 08:12:28 +1200485
Simon Glass89b86b82011-07-17 23:49:49 -0700486 self._tools.WriteFile(bootstub, uboot_data + fdt_data)
Simon Glass290a1802011-07-17 13:54:32 -0700487 self._tools.OutputSize('U-Boot binary', self.uboot_fname)
488 self._tools.OutputSize('U-Boot fdt', self._fdt_fname)
Simon Glass89b86b82011-07-17 23:49:49 -0700489 self._tools.OutputSize('Combined binary', bootstub)
490
Simon Glasse13ee2c2011-07-28 08:12:28 +1200491 # Sign the bootstub; this is a combination of the board specific
Simon Glass89b86b82011-07-17 23:49:49 -0700492 # bct and the stub u-boot image.
Simon Glass290a1802011-07-17 13:54:32 -0700493 signed = self._SignBootstub(self._tools.Filename(self.bct_fname),
Simon Glasse13ee2c2011-07-28 08:12:28 +1200494 bootstub, text_base)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200495
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 Glass02d124a2012-03-02 14:47:20 -0800510 fdt.PutInteger('/config', 'postload-text-offset', len(data))
Simon Glass6dcc2f22011-07-28 15:26:49 +1200511 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 Glass89b86b82011-07-17 23:49:49 -0700528
Vincent Palatinf7286772011-10-12 14:31:53 -0700529 def _CreateCorebootStub(self, uboot, coreboot, fdt, seabios):
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700530 """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 Palatinf7286772011-10-12 14:31:53 -0700536 seabios: Path to SeaBIOS payload binary or None
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700537
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 Reinauerc2e1e4d2011-08-23 14:50:59 -0700545 uboot_elf = uboot.replace(".bin", ".elf")
546 shutil.copyfile(coreboot, bootstub)
Vincent Palatinf7286772011-10-12 14:31:53 -0700547 if seabios:
Simon Glass146cb8e2012-03-09 15:51:24 -0800548 self._tools.Run('cbfstool', [bootstub, 'add-payload', seabios,
Vincent Palatinf7286772011-10-12 14:31:53 -0700549 'fallback/payload', 'lzma'])
Simon Glass146cb8e2012-03-09 15:51:24 -0800550 self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf,
Vincent Palatinf7286772011-10-12 14:31:53 -0700551 'img/U-Boot', 'lzma'])
552 else:
Simon Glass146cb8e2012-03-09 15:51:24 -0800553 self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf,
Vincent Palatinf7286772011-10-12 14:31:53 -0700554 'fallback/payload', 'lzma'])
Simon Glass146cb8e2012-03-09 15:51:24 -0800555 self._tools.Run('cbfstool', [bootstub, 'add', fdt.fname, 'u-boot.dtb',
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700556 '0xac'])
557 return bootstub
558
Simon Glass3b404092012-05-23 13:10:36 -0700559 def _UpdateBl2Parameters(self, fdt, spl_load_size, data, pos):
Simon Glassdf95dd22012-03-13 15:46:16 -0700560 """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 Glass3b404092012-05-23 13:10:36 -0700567 spl_load_size: Size of U-Boot image that SPL must load
Simon Glassdf95dd22012-03-13 15:46:16 -0700568 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 Glassdf95dd22012-03-13 15:46:16 -0700574 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 Glass7dbdf5b2012-03-15 20:58:04 -0700579 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 Glassdf95dd22012-03-13 15:46:16 -0700582
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 Glass66c1a9f2012-03-22 19:15:53 -0700589 pos += (param_len + 4) & ~3
Simon Glassdf95dd22012-03-13 15:46:16 -0700590
591 # Work through the parameters one at a time, adding each value
592 new_data = ''
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700593 upto = 0
Simon Glassdf95dd22012-03-13 15:46:16 -0700594 for param in param_list:
Simon Glass2c48ddf2012-03-23 16:55:22 -0700595 value = struct.unpack('<1L', data[pos + upto:pos + upto + 4])[0]
Simon Glassdf95dd22012-03-13 15:46:16 -0700596 if param == 'm' :
Simon Glass1a77ded2012-03-15 21:00:49 -0700597 mem_type = fdt.GetString('/dmc', 'mem-type')
Simon Glassdf95dd22012-03-13 15:46:16 -0700598 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 Andersonee46cfe2012-05-18 09:53:08 -0700603 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 Glassdf95dd22012-03-13 15:46:16 -0700610 elif param == 'v':
611 value = 31
612 self._out.Info(' Memory interleave: %#0x' % value)
Simon Glass8e1fdb22012-03-15 21:02:10 -0700613 elif param == 'u':
Simon Glass3b404092012-05-23 13:10:36 -0700614 value = (spl_load_size + 0xfff) & ~0xfff
615 self._out.Info(' U-Boot size: %#0x (rounded up from %#0x)' %
616 (value, spl_load_size))
Simon Glass559b6612012-05-23 13:28:45 -0700617 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 Glassdf95dd22012-03-13 15:46:16 -0700630 else:
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700631 self._out.Warning("Unknown machine parameter type '%s'" % param)
Simon Glass2c48ddf2012-03-23 16:55:22 -0700632 self._out.Info(' Unknown value: %#0x' % value)
Simon Glassdf95dd22012-03-13 15:46:16 -0700633 new_data += struct.pack('<L', value)
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700634 upto += 4
Simon Glassdf95dd22012-03-13 15:46:16 -0700635
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 Glasse5e8afb2012-05-23 11:19:23 -0700641 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 Glass559b6612012-05-23 13:28:45 -0700658 def ConfigureExynosBl2(self, fdt, spl_load_size, orig_bl2, name=''):
Simon Glass7e199222012-03-13 15:51:18 -0700659 """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 Glass3b404092012-05-23 13:10:36 -0700665 spl_load_size: Size of U-Boot image that SPL must load
Simon Glass7e199222012-03-13 15:51:18 -0700666 orig_bl2: Filename of original BL2 file to modify.
667 """
Simon Glass66c1a9f2012-03-22 19:15:53 -0700668 self._out.Info('Configuring BL2')
Simon Glass559b6612012-05-23 13:28:45 -0700669 bl2 = os.path.join(self._tools.outdir, 'updated-spl%s.bin' % name)
Simon Glass66c1a9f2012-03-22 19:15:53 -0700670 data = self._tools.ReadFile(orig_bl2)
671 self._tools.WriteFile(bl2, data)
Simon Glass7e199222012-03-13 15:51:18 -0700672
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 Glass3b404092012-05-23 13:10:36 -0700680 data = self._UpdateBl2Parameters(fdt, spl_load_size, data, pos)
Simon Glasse5e8afb2012-05-23 11:19:23 -0700681 data = self._UpdateChecksum(data)
Simon Glass7e199222012-03-13 15:51:18 -0700682 self._tools.WriteFile(bl2, data)
683 return bl2
684
Simon Glass89b86b82011-07-17 23:49:49 -0700685 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 Glass439fe7a2012-03-09 16:19:34 -0800695 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 Glass7e199222012-03-13 15:51:18 -0700714 elif blob_type == 'exynos-bl1':
715 pack.AddProperty(blob_type, self.exynos_bl1)
716 elif blob_type == 'exynos-bl2':
Simon Glass3b404092012-05-23 13:10:36 -0700717 spl_load_size = os.stat(pack.GetProperty('boot+dtb')).st_size
718 bl2 = self.ConfigureExynosBl2(fdt, spl_load_size, self.exynos_bl2)
Simon Glass7e199222012-03-13 15:51:18 -0700719 pack.AddProperty(blob_type, bl2)
Simon Glass439fe7a2012-03-09 16:19:34 -0800720 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 Glass290a1802011-07-17 13:54:32 -0700726 def _CreateImage(self, gbb, fdt):
Simon Glass89b86b82011-07-17 23:49:49 -0700727 """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 Glasse13ee2c2011-07-28 08:12:28 +1200734 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 Glass89b86b82011-07-17 23:49:49 -0700739
740 Raises:
741 CmdError if a command fails.
742 """
Simon Glass02d124a2012-03-02 14:47:20 -0800743 self._out.Notice("Model: %s" % fdt.GetString('/', 'model'))
Simon Glass89b86b82011-07-17 23:49:49 -0700744
Simon Glass439fe7a2012-03-09 16:19:34 -0800745 # 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 Glass07267952012-06-08 12:45:13 -0700751 pack.AddProperty('skeleton', self.skeleton_fname)
Simon Glass50f74602012-03-15 21:04:25 -0700752
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 Glass439fe7a2012-03-09 16:19:34 -0800763 pack.AddProperty('gbb', self.uboot_fname)
Simon Glass07267952012-06-08 12:45:13 -0700764 for blob_type in pack.GetBlobList():
Simon Glass439fe7a2012-03-09 16:19:34 -0800765 self._BuildBlob(pack, fdt, blob_type)
Simon Glass89b86b82011-07-17 23:49:49 -0700766
767 if gbb:
Simon Glasse76bf7b2012-03-13 15:34:41 -0700768 pack.RequireAllEntries()
Hung-Te Lina7462e72011-07-27 19:17:10 +0800769 fwid = '.'.join([
Simon Glass02d124a2012-03-02 14:47:20 -0800770 re.sub('[ ,]+', '_', fdt.GetString('/', 'model')),
Hung-Te Lina7462e72011-07-27 19:17:10 +0800771 self._tools.GetChromeosVersion()])
Simon Glass89b86b82011-07-17 23:49:49 -0700772 self._out.Notice('Firmware ID: %s' % fwid)
Simon Glass439fe7a2012-03-09 16:19:34 -0800773 pack.AddProperty('fwid', fwid)
774 pack.AddProperty('gbb', gbb)
775 pack.AddProperty('keydir', self._keydir)
Simon Glassc90cf582012-03-13 15:40:47 -0700776
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 Glass89b86b82011-07-17 23:49:49 -0700781
Simon Glass439fe7a2012-03-09 16:19:34 -0800782 image = pack.GetProperty('image')
Simon Glass89b86b82011-07-17 23:49:49 -0700783 self._tools.OutputSize('Final image', image)
Simon Glassc90cf582012-03-13 15:40:47 -0700784 return image, pack
Simon Glass89b86b82011-07-17 23:49:49 -0700785
Simon Glass290a1802011-07-17 13:54:32 -0700786 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 Glassc0f3dc62011-08-09 14:19:05 -0700792 Returns:
793 The Fdt object of the original fdt file, which we will not modify.
794
Simon Glass290a1802011-07-17 13:54:32 -0700795 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 Glass29b96ad2012-03-09 15:34:33 -0800801 fdt.Compile()
Simon Glass290a1802011-07-17 13:54:32 -0700802 self.fdt = fdt.Copy(os.path.join(self._tools.outdir, 'updated.dtb'))
Simon Glassc0f3dc62011-08-09 14:19:05 -0700803 return fdt
Simon Glass290a1802011-07-17 13:54:32 -0700804
Simon Glassc90cf582012-03-13 15:40:47 -0700805 def Start(self, hardware_id, output_fname, show_map):
Simon Glass290a1802011-07-17 13:54:32 -0700806 """This creates a firmware bundle according to settings provided.
Simon Glass89b86b82011-07-17 23:49:49 -0700807
808 - Checks options, tools, output directory, fdt.
809 - Creates GBB and image.
Simon Glass290a1802011-07-17 13:54:32 -0700810
811 Args:
Simon Glass56577572011-07-19 11:08:06 +1200812 hardware_id: Hardware ID to use for this board. If None, then the
813 default from the Fdt will be used
Simon Glass290a1802011-07-17 13:54:32 -0700814 output_fname: Output filename for the image. If this is not None, then
815 the final image will be copied here.
Simon Glassc90cf582012-03-13 15:40:47 -0700816 show_map: Show a flash map, with each area's name and position
Simon Glass290a1802011-07-17 13:54:32 -0700817
818 Returns:
819 Filename of the resulting image (not the output_fname copy).
Simon Glass89b86b82011-07-17 23:49:49 -0700820 """
Simon Glass89b86b82011-07-17 23:49:49 -0700821 gbb = ''
Simon Glass290a1802011-07-17 13:54:32 -0700822 if not self._small:
Simon Glass56577572011-07-19 11:08:06 +1200823 gbb = self._CreateGoogleBinaryBlock(hardware_id)
Simon Glass89b86b82011-07-17 23:49:49 -0700824
825 # This creates the actual image.
Simon Glassc90cf582012-03-13 15:40:47 -0700826 image, pack = self._CreateImage(gbb, self.fdt)
827 if show_map:
828 pack.ShowMap()
Simon Glass290a1802011-07-17 13:54:32 -0700829 if output_fname:
830 shutil.copyfile(image, output_fname)
831 self._out.Notice("Output image '%s'" % output_fname)
Simon Glass794217e2012-06-07 11:40:37 -0700832 return image, pack.props