blob: 2a2f5cdd82aaa153cc82e25de29fa87179b44a07 [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')
68 .. can call bundle.AddConfigList() if required
69 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.
78 """
Simon Glass89b86b82011-07-17 23:49:49 -070079
Simon Glass290a1802011-07-17 13:54:32 -070080 def __init__(self, tools, output):
81 """Set up a new Bundle object.
Simon Glass89b86b82011-07-17 23:49:49 -070082
Simon Glass290a1802011-07-17 13:54:32 -070083 Args:
84 tools: A tools.Tools object to use for external tools.
85 output: A cros_output.Output object to use for program output.
Simon Glass89b86b82011-07-17 23:49:49 -070086 """
Simon Glass290a1802011-07-17 13:54:32 -070087 self._tools = tools
88 self._out = output
89
90 # Set up the things we need to know in order to operate.
91 self._board = None # Board name, e.g. tegra2_seaboard.
92 self._fdt_fname = None # Filename of our FDT.
93 self.uboot_fname = None # Filename of our U-Boot binary.
94 self.bct_fname = None # Filename of our BCT file.
95 self.fdt = None # Our Fdt object.
Hung-Te Lin5b649382011-08-03 15:01:16 +080096 self.bmpblk_fname = None # Filename of our Bitmap Block
Stefan Reinauer8d79d362011-08-16 14:20:43 -070097 self.coreboot_fname = None # Filename of our coreboot binary.
Vincent Palatinf7286772011-10-12 14:31:53 -070098 self.seabios_fname = None # Filename of our SeaBIOS payload.
Simon Glass7e199222012-03-13 15:51:18 -070099 self.exynos_bl1 = None # Filename of Exynos BL1 (pre-boot)
100 self.exynos_bl2 = None # Filename of Exynos BL2 (SPL)
Simon Glass290a1802011-07-17 13:54:32 -0700101
102 def SetDirs(self, keydir):
103 """Set up directories required for Bundle.
104
105 Args:
106 keydir: Directory containing keys to use for signing firmware.
107 """
108 self._keydir = keydir
109
Simon Glass6dcc2f22011-07-28 15:26:49 +1200110 def SetFiles(self, board, bct, uboot=None, bmpblk=None, coreboot=None,
Simon Glass7e199222012-03-13 15:51:18 -0700111 postload=None, seabios=None, exynos_bl1=None, exynos_bl2=None):
Simon Glass290a1802011-07-17 13:54:32 -0700112 """Set up files required for Bundle.
113
114 Args:
115 board: The name of the board to target (e.g. tegra2_seaboard).
116 uboot: The filename of the u-boot.bin image to use.
117 bct: The filename of the binary BCT file to use.
Hung-Te Lin5b649382011-08-03 15:01:16 +0800118 bmpblk: The filename of bitmap block file to use.
Stefan Reinauer8d79d362011-08-16 14:20:43 -0700119 coreboot: The filename of the coreboot image to use (on x86)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200120 postload: The filename of the u-boot-post.bin image to use.
Vincent Palatinf7286772011-10-12 14:31:53 -0700121 seabios: The filename of the SeaBIOS payload to use if any.
Simon Glass290a1802011-07-17 13:54:32 -0700122 """
123 self._board = board
124 self.uboot_fname = uboot
125 self.bct_fname = bct
Hung-Te Lin5b649382011-08-03 15:01:16 +0800126 self.bmpblk_fname = bmpblk
Stefan Reinauer8d79d362011-08-16 14:20:43 -0700127 self.coreboot_fname = coreboot
Simon Glass6dcc2f22011-07-28 15:26:49 +1200128 self.postload_fname = postload
Vincent Palatinf7286772011-10-12 14:31:53 -0700129 self.seabios_fname = seabios
Simon Glass7e199222012-03-13 15:51:18 -0700130 self.exynos_bl1 = exynos_bl1
131 self.exynos_bl2 = exynos_bl2
Simon Glass290a1802011-07-17 13:54:32 -0700132
133 def SetOptions(self, small):
134 """Set up options supported by Bundle.
135
136 Args:
137 small: Only create a signed U-Boot - don't produce the full packed
138 firmware image. This is useful for devs who want to replace just the
139 U-Boot part while keeping the keys, gbb, etc. the same.
140 """
141 self._small = small
142
143 def CheckOptions(self):
144 """Check provided options and select defaults."""
145 if not self._board:
146 raise ValueError('No board defined - please define a board to use')
Simon Glass493163b2011-09-14 11:19:57 -0700147 build_root = os.path.join('##', 'build', self._board, 'firmware')
Simon Glass881964d2012-04-04 11:34:09 -0700148 dir_name = os.path.join(build_root, 'dts')
Simon Glass290a1802011-07-17 13:54:32 -0700149 if not self._fdt_fname:
Simon Glassceff3ff2012-04-04 11:23:45 -0700150 # Figure out where the file should be, and the name we expect.
Simon Glassceff3ff2012-04-04 11:23:45 -0700151 base_name = re.sub('_', '-', self._board)
152
153 # In case the name exists with a prefix or suffix, find it.
154 wildcard = os.path.join(dir_name, '*%s*.dts' % base_name)
155 found_list = glob.glob(self._tools.Filename(wildcard))
156 if len(found_list) == 1:
157 self._fdt_fname = found_list[0]
158 else:
159 # We didn't find anything definite, so set up our expected name.
160 self._fdt_fname = os.path.join(dir_name, '%s.dts' % base_name)
161
Simon Glass881964d2012-04-04 11:34:09 -0700162 # Convert things like 'exynos5250-daisy' into a full path.
163 root, ext = os.path.splitext(self._fdt_fname)
164 if not ext and not os.path.dirname(root):
165 self._fdt_fname = os.path.join(dir_name, '%s.dts' % root)
166
Simon Glass290a1802011-07-17 13:54:32 -0700167 if not self.uboot_fname:
168 self.uboot_fname = os.path.join(build_root, 'u-boot.bin')
169 if not self.bct_fname:
170 self.bct_fname = os.path.join(build_root, 'bct', 'board.bct')
Simon Glass2a7f0b32011-08-26 11:25:17 -0700171 if not self.bmpblk_fname:
172 self.bmpblk_fname = os.path.join(build_root, 'default.bmpblk')
Simon Glass1d376832012-03-15 20:50:54 -0700173 if not self.exynos_bl1:
174 self.exynos_bl1 = os.path.join(build_root, 'E5250.nbl1.bin')
175 if not self.exynos_bl2:
176 self.exynos_bl2 = os.path.join(build_root, 'smdk5250-spl.bin')
Simon Glass89b86b82011-07-17 23:49:49 -0700177
Simon Glass75759302012-03-15 20:26:53 -0700178 def GetFiles(self):
179 """Get a list of files that we know about.
180
181 This is the opposite of SetFiles except that we may have put in some
182 default names. It returns a dictionary containing the filename for
183 each of a number of pre-defined files.
184
185 Returns:
186 Dictionary, with one entry for each file.
187 """
188 file_list = {
189 'bct' : self.bct_fname,
190 'exynos-bl1' : self.exynos_bl1,
191 'exynos-bl2' : self.exynos_bl2,
192 }
193 return file_list
194
Simon Glass56577572011-07-19 11:08:06 +1200195 def _CreateGoogleBinaryBlock(self, hardware_id):
Simon Glass89b86b82011-07-17 23:49:49 -0700196 """Create a GBB for the image.
197
Simon Glass56577572011-07-19 11:08:06 +1200198 Args:
199 hardware_id: Hardware ID to use for this board. If None, then the
200 default from the Fdt will be used
201
Simon Glass89b86b82011-07-17 23:49:49 -0700202 Returns:
203 Path of the created GBB file.
204
205 Raises:
206 CmdError if a command fails.
207 """
Simon Glass56577572011-07-19 11:08:06 +1200208 if not hardware_id:
Simon Glass02d124a2012-03-02 14:47:20 -0800209 hardware_id = self.fdt.GetString('/config', 'hwid')
Simon Glass89b86b82011-07-17 23:49:49 -0700210 gbb_size = self.fdt.GetFlashPartSize('ro', 'gbb')
Simon Glass290a1802011-07-17 13:54:32 -0700211 odir = self._tools.outdir
Simon Glass89b86b82011-07-17 23:49:49 -0700212
Stefan Reinauer975e68f2012-02-27 13:27:08 -0800213 chromeos_config = self.fdt.GetProps("/chromeos-config")
214 if 'fast-developer-mode' not in chromeos_config:
215 gbb_flags = 0
216 else:
217 self._out.Notice("Enabling fast-developer-mode.")
218 gbb_flags = 1
219
Simon Glass89b86b82011-07-17 23:49:49 -0700220 self._out.Progress('Creating GBB')
221 sizes = [0x100, 0x1000, gbb_size - 0x2180, 0x1000]
222 sizes = ['%#x' % size for size in sizes]
223 gbb = 'gbb.bin'
Simon Glass290a1802011-07-17 13:54:32 -0700224 keydir = self._tools.Filename(self._keydir)
225 self._tools.Run('gbb_utility', ['-c', ','.join(sizes), gbb], cwd=odir)
Simon Glass89b86b82011-07-17 23:49:49 -0700226 self._tools.Run('gbb_utility', ['-s',
Simon Glass56577572011-07-19 11:08:06 +1200227 '--hwid=%s' % hardware_id,
Simon Glass89b86b82011-07-17 23:49:49 -0700228 '--rootkey=%s/root_key.vbpubk' % keydir,
229 '--recoverykey=%s/recovery_key.vbpubk' % keydir,
Simon Glass2a7f0b32011-08-26 11:25:17 -0700230 '--bmpfv=%s' % self._tools.Filename(self.bmpblk_fname),
Stefan Reinauer975e68f2012-02-27 13:27:08 -0800231 '--flags=%d' % gbb_flags,
Simon Glass89b86b82011-07-17 23:49:49 -0700232 gbb],
Simon Glass290a1802011-07-17 13:54:32 -0700233 cwd=odir)
234 return os.path.join(odir, gbb)
Simon Glass89b86b82011-07-17 23:49:49 -0700235
Simon Glasse13ee2c2011-07-28 08:12:28 +1200236 def _SignBootstub(self, bct, bootstub, text_base):
Simon Glass89b86b82011-07-17 23:49:49 -0700237 """Sign an image so that the Tegra SOC will boot it.
238
239 Args:
240 bct: BCT file to use.
241 bootstub: Boot stub (U-Boot + fdt) file to sign.
242 text_base: Address of text base for image.
Simon Glass89b86b82011-07-17 23:49:49 -0700243
244 Returns:
245 filename of signed image.
246
247 Raises:
248 CmdError if a command fails.
249 """
250 # First create a config file - this is how we instruct cbootimage
Simon Glasse13ee2c2011-07-28 08:12:28 +1200251 signed = os.path.join(self._tools.outdir, 'signed.bin')
Simon Glass89b86b82011-07-17 23:49:49 -0700252 self._out.Progress('Signing Bootstub')
Simon Glasse13ee2c2011-07-28 08:12:28 +1200253 config = os.path.join(self._tools.outdir, 'boot.cfg')
Simon Glass89b86b82011-07-17 23:49:49 -0700254 fd = open(config, 'w')
255 fd.write('Version = 1;\n')
256 fd.write('Redundancy = 1;\n')
257 fd.write('Bctfile = %s;\n' % bct)
Doug Anderson0eeb0742011-09-15 18:11:40 -0700258
259 # TODO(dianders): Right now, we don't have enough space in our flash map
260 # for two copies of the BCT when we're using NAND, so hack it to 1. Not
261 # sure what this does for reliability, but at least things will fit...
262 is_nand = "NvBootDevType_Nand" in self._tools.Run('bct_dump', [bct])
263 if is_nand:
264 fd.write('Bctcopy = 1;\n')
265
Simon Glass89b86b82011-07-17 23:49:49 -0700266 fd.write('BootLoader = %s,%#x,%#x,Complete;\n' % (bootstub, text_base,
267 text_base))
Doug Anderson0eeb0742011-09-15 18:11:40 -0700268
Simon Glass89b86b82011-07-17 23:49:49 -0700269 fd.close()
270
271 self._tools.Run('cbootimage', [config, signed])
272 self._tools.OutputSize('BCT', bct)
273 self._tools.OutputSize('Signed image', signed)
274 return signed
275
Doug Anderson86ce5f42011-07-27 10:40:18 -0700276 def SetBootcmd(self, bootcmd, bootsecure):
Simon Glass290a1802011-07-17 13:54:32 -0700277 """Set the boot command for U-Boot.
Simon Glass89b86b82011-07-17 23:49:49 -0700278
279 Args:
Simon Glass290a1802011-07-17 13:54:32 -0700280 bootcmd: Boot command to use, as a string (if None this this is a nop).
Doug Anderson86ce5f42011-07-27 10:40:18 -0700281 bootsecure: We'll set '/config/bootsecure' to 1 if True and 0 if False.
Simon Glass89b86b82011-07-17 23:49:49 -0700282 """
Simon Glass290a1802011-07-17 13:54:32 -0700283 if bootcmd:
Simon Glass02d124a2012-03-02 14:47:20 -0800284 self.fdt.PutString('/config', 'bootcmd', bootcmd)
285 self.fdt.PutInteger('/config', 'bootsecure', int(bootsecure))
Simon Glass290a1802011-07-17 13:54:32 -0700286 self._out.Info('Boot command: %s' % bootcmd)
Simon Glass89b86b82011-07-17 23:49:49 -0700287
Simon Glass290a1802011-07-17 13:54:32 -0700288 def AddConfigList(self, config_list, use_int=False):
289 """Add a list of config items to the fdt.
290
291 Normally these values are written to the fdt as strings, but integers
292 are also supported, in which case the values will be converted to integers
293 (if necessary) before being stored.
294
295 Args:
296 config_list: List of (config, value) tuples to add to the fdt. For each
297 tuple:
298 config: The fdt node to write to will be /config/<config>.
299 value: An integer or string value to write.
300 use_int: True to only write integer values.
301
302 Raises:
303 CmdError: if a value is required to be converted to integer but can't be.
304 """
305 if config_list:
306 for config in config_list:
307 value = config[1]
308 if use_int:
309 try:
310 value = int(value)
311 except ValueError as str:
312 raise CmdError("Cannot convert config option '%s' to integer" %
313 value)
314 if type(value) == type(1):
Simon Glass02d124a2012-03-02 14:47:20 -0800315 self.fdt.PutInteger('/config', '%s' % config[0], value)
Simon Glass290a1802011-07-17 13:54:32 -0700316 else:
Simon Glass02d124a2012-03-02 14:47:20 -0800317 self.fdt.PutString('/config', '%s' % config[0], value)
Simon Glass290a1802011-07-17 13:54:32 -0700318
Simon Glass7c2d5572011-11-15 14:47:08 -0800319 def DecodeTextBase(self, data):
320 """Look at a U-Boot image and try to decode its TEXT_BASE.
321
322 This works because U-Boot has a header with the value 0x12345678
323 immediately followed by the TEXT_BASE value. We can therefore read this
324 from the image with some certainty. We check only the first 40 words
325 since the header should be within that region.
326
327 Args:
328 data: U-Boot binary data
329
330 Returns:
331 Text base (integer) or None if none was found
332 """
333 found = False
334 for i in range(0, 160, 4):
335 word = data[i:i + 4]
336
337 # TODO(sjg): This does not cope with a big-endian target
338 value = struct.unpack('<I', word)[0]
339 if found:
340 return value
341 if value == 0x12345678:
342 found = True
343
344 return None
345
346 def CalcTextBase(self, name, fdt, fname):
347 """Calculate the TEXT_BASE to use for U-Boot.
348
349 Normally this value is in the fdt, so we just read it from there. But as
350 a second check we look at the image itself in case this is different, and
351 switch to that if it is.
352
353 This allows us to flash any U-Boot even if its TEXT_BASE is different.
354 This is particularly useful with upstream U-Boot which uses a different
355 value (which we will move to).
356 """
357 data = self._tools.ReadFile(fname)
Simon Glass02d124a2012-03-02 14:47:20 -0800358 fdt_text_base = fdt.GetInt('/chromeos-config', 'textbase')
Simon Glass7c2d5572011-11-15 14:47:08 -0800359 text_base = self.DecodeTextBase(data)
360
361 # If they are different, issue a warning and switch over.
362 if text_base and text_base != fdt_text_base:
363 self._out.Warning("TEXT_BASE %x in %sU-Boot doesn't match "
364 "fdt value of %x. Using %x" % (text_base, name,
365 fdt_text_base, text_base))
366 fdt_text_base = text_base
367 return fdt_text_base
368
Simon Glass6dcc2f22011-07-28 15:26:49 +1200369 def _CreateBootStub(self, uboot, base_fdt, postload):
Simon Glass89b86b82011-07-17 23:49:49 -0700370 """Create a boot stub and a signed boot stub.
371
Simon Glass6dcc2f22011-07-28 15:26:49 +1200372 For postload:
373 We add a /config/postload-text-offset entry to the signed bootstub's
374 fdt so that U-Boot can find the postload code.
375
376 The raw (unsigned) bootstub will have a value of -1 for this since we will
377 simply append the postload code to the bootstub and it can find it there.
378 This will be used for RW A/B firmware.
379
380 For the signed case this value will specify where in the flash to find
381 the postload code. This will be used for RO firmware.
382
Simon Glass89b86b82011-07-17 23:49:49 -0700383 Args:
384 uboot: Path to u-boot.bin (may be chroot-relative)
Simon Glass29b96ad2012-03-09 15:34:33 -0800385 base_fdt: Fdt object containing the flat device tree.
Simon Glass6dcc2f22011-07-28 15:26:49 +1200386 postload: Path to u-boot-post.bin, or None if none.
Simon Glass89b86b82011-07-17 23:49:49 -0700387
388 Returns:
389 Tuple containing:
Simon Glass6dcc2f22011-07-28 15:26:49 +1200390 Full path to bootstub (uboot + fdt(-1) + postload).
391 Full path to signed (uboot + fdt(flash pos) + bct) + postload.
Simon Glass89b86b82011-07-17 23:49:49 -0700392
393 Raises:
394 CmdError if a command fails.
395 """
Simon Glasse13ee2c2011-07-28 08:12:28 +1200396 bootstub = os.path.join(self._tools.outdir, 'u-boot-fdt.bin')
Simon Glass7c2d5572011-11-15 14:47:08 -0800397 text_base = self.CalcTextBase('', self.fdt, uboot)
Simon Glass89b86b82011-07-17 23:49:49 -0700398 uboot_data = self._tools.ReadFile(uboot)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200399
400 # Make a copy of the fdt for the bootstub
401 fdt = base_fdt.Copy(os.path.join(self._tools.outdir, 'bootstub.dtb'))
Simon Glass02d124a2012-03-02 14:47:20 -0800402 fdt.PutInteger('/config', 'postload-text-offset', 0xffffffff);
Simon Glass290a1802011-07-17 13:54:32 -0700403 fdt_data = self._tools.ReadFile(fdt.fname)
Simon Glasse13ee2c2011-07-28 08:12:28 +1200404
Simon Glass89b86b82011-07-17 23:49:49 -0700405 self._tools.WriteFile(bootstub, uboot_data + fdt_data)
Simon Glass290a1802011-07-17 13:54:32 -0700406 self._tools.OutputSize('U-Boot binary', self.uboot_fname)
407 self._tools.OutputSize('U-Boot fdt', self._fdt_fname)
Simon Glass89b86b82011-07-17 23:49:49 -0700408 self._tools.OutputSize('Combined binary', bootstub)
409
Simon Glasse13ee2c2011-07-28 08:12:28 +1200410 # Sign the bootstub; this is a combination of the board specific
Simon Glass89b86b82011-07-17 23:49:49 -0700411 # bct and the stub u-boot image.
Simon Glass290a1802011-07-17 13:54:32 -0700412 signed = self._SignBootstub(self._tools.Filename(self.bct_fname),
Simon Glasse13ee2c2011-07-28 08:12:28 +1200413 bootstub, text_base)
Simon Glass6dcc2f22011-07-28 15:26:49 +1200414
415 signed_postload = os.path.join(self._tools.outdir, 'signed-postload.bin')
416 data = self._tools.ReadFile(signed)
417
418 if postload:
419 # We must add postload to the bootstub since A and B will need to
420 # be able to find it without the /config/postload-text-offset mechanism.
421 bs_data = self._tools.ReadFile(bootstub)
422 bs_data += self._tools.ReadFile(postload)
423 bootstub = os.path.join(self._tools.outdir, 'u-boot-fdt-postload.bin')
424 self._tools.WriteFile(bootstub, bs_data)
425 self._tools.OutputSize('Combined binary with postload', bootstub)
426
427 # Now that we know the file size, adjust the fdt and re-sign
428 postload_bootstub = os.path.join(self._tools.outdir, 'postload.bin')
Simon Glass02d124a2012-03-02 14:47:20 -0800429 fdt.PutInteger('/config', 'postload-text-offset', len(data))
Simon Glass6dcc2f22011-07-28 15:26:49 +1200430 fdt_data = self._tools.ReadFile(fdt.fname)
431 self._tools.WriteFile(postload_bootstub, uboot_data + fdt_data)
432 signed = self._SignBootstub(self._tools.Filename(self.bct_fname),
433 postload_bootstub, text_base)
434 if len(data) != os.path.getsize(signed):
435 raise CmdError('Signed file size changed from %d to %d after updating '
436 'fdt' % (len(data), os.path.getsize(signed)))
437
438 # Re-read the signed image, and add the post-load binary.
439 data = self._tools.ReadFile(signed)
440 data += self._tools.ReadFile(postload)
441 self._tools.OutputSize('Post-load binary', postload)
442
443 self._tools.WriteFile(signed_postload, data)
444 self._tools.OutputSize('Final bootstub with postload', signed_postload)
445
446 return bootstub, signed_postload
Simon Glass89b86b82011-07-17 23:49:49 -0700447
Vincent Palatinf7286772011-10-12 14:31:53 -0700448 def _CreateCorebootStub(self, uboot, coreboot, fdt, seabios):
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700449 """Create a coreboot boot stub.
450
451 Args:
452 uboot: Path to u-boot.bin (may be chroot-relative)
453 coreboot: Path to coreboot.rom
454 fdt: Device Tree
Vincent Palatinf7286772011-10-12 14:31:53 -0700455 seabios: Path to SeaBIOS payload binary or None
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700456
457 Returns:
458 Full path to bootstub (coreboot + uboot + fdt).
459
460 Raises:
461 CmdError if a command fails.
462 """
463 bootstub = os.path.join(self._tools.outdir, 'coreboot-full.rom')
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700464 uboot_elf = uboot.replace(".bin", ".elf")
465 shutil.copyfile(coreboot, bootstub)
Vincent Palatinf7286772011-10-12 14:31:53 -0700466 if seabios:
Simon Glass146cb8e2012-03-09 15:51:24 -0800467 self._tools.Run('cbfstool', [bootstub, 'add-payload', seabios,
Vincent Palatinf7286772011-10-12 14:31:53 -0700468 'fallback/payload', 'lzma'])
Simon Glass146cb8e2012-03-09 15:51:24 -0800469 self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf,
Vincent Palatinf7286772011-10-12 14:31:53 -0700470 'img/U-Boot', 'lzma'])
471 else:
Simon Glass146cb8e2012-03-09 15:51:24 -0800472 self._tools.Run('cbfstool', [bootstub, 'add-payload', uboot_elf,
Vincent Palatinf7286772011-10-12 14:31:53 -0700473 'fallback/payload', 'lzma'])
Simon Glass146cb8e2012-03-09 15:51:24 -0800474 self._tools.Run('cbfstool', [bootstub, 'add', fdt.fname, 'u-boot.dtb',
Stefan Reinauerc2e1e4d2011-08-23 14:50:59 -0700475 '0xac'])
476 return bootstub
477
Simon Glass8e1fdb22012-03-15 21:02:10 -0700478 def _UpdateBl2Parameters(self, fdt, pack, data, pos):
Simon Glassdf95dd22012-03-13 15:46:16 -0700479 """Update the parameters in a BL2 blob.
480
481 We look at the list in the parameter block, extract the value of each
482 from the device tree, and write that value to the parameter block.
483
484 Args:
485 fdt: Device tree containing the parameter values.
Simon Glass8e1fdb22012-03-15 21:02:10 -0700486 pack: The firmware packer object
Simon Glassdf95dd22012-03-13 15:46:16 -0700487 data: The BL2 data.
488 pos: The position of the start of the parameter block.
489
490 Returns:
491 The new contents of the parameter block, after updating.
492 """
Simon Glassdf95dd22012-03-13 15:46:16 -0700493 version, size = struct.unpack('<2L', data[pos + 4:pos + 12])
494 if version != 1:
495 raise CmdError("Cannot update machine parameter block version '%d'" %
496 version)
497 if size < 0 or pos + size > len(data):
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700498 raise CmdError("Machine parameter block size %d is invalid: "
499 "pos=%d, size=%d, space=%d, len=%d" %
500 (size, pos, size, len(data) - pos, len(data)))
Simon Glassdf95dd22012-03-13 15:46:16 -0700501
502 # Move past the header and read the parameter list, which is terminated
503 # with \0.
504 pos += 12
505 param_list = struct.unpack('<%ds' % (len(data) - pos), data[pos:])[0]
506 param_len = param_list.find('\0')
507 param_list = param_list[:param_len]
Simon Glass66c1a9f2012-03-22 19:15:53 -0700508 pos += (param_len + 4) & ~3
Simon Glassdf95dd22012-03-13 15:46:16 -0700509
510 # Work through the parameters one at a time, adding each value
511 new_data = ''
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700512 upto = 0
Simon Glassdf95dd22012-03-13 15:46:16 -0700513 for param in param_list:
Simon Glass2c48ddf2012-03-23 16:55:22 -0700514 value = struct.unpack('<1L', data[pos + upto:pos + upto + 4])[0]
Simon Glassdf95dd22012-03-13 15:46:16 -0700515 if param == 'm' :
Simon Glass1a77ded2012-03-15 21:00:49 -0700516 mem_type = fdt.GetString('/dmc', 'mem-type')
Simon Glassdf95dd22012-03-13 15:46:16 -0700517 mem_types = ['ddr2', 'ddr3', 'lpddr2', 'lpddr3']
518 if not mem_type in mem_types:
519 raise CmdError("Unknown memory type '%s'" % mem_type)
520 value = mem_types.index(mem_type)
521 self._out.Info(' Memory type: %s (%d)' % (mem_type, value))
522 elif param == 'v':
523 value = 31
524 self._out.Info(' Memory interleave: %#0x' % value)
Simon Glass8e1fdb22012-03-15 21:02:10 -0700525 elif param == 'u':
Simon Glass2c48ddf2012-03-23 16:55:22 -0700526 # TODO(sjg): Seems to not work unless set to the same value as in the
527 # existing image. Need to find root cause.
528 #value = os.stat(pack.GetProperty('boot+dtb')).st_size
529 #value = (value + 0xfff) & ~0xfff
530 self._out.Warning("Leaving U-Boot size unchanged")
Simon Glass8e1fdb22012-03-15 21:02:10 -0700531 self._out.Info(' U-Boot size: %#0x' % value)
Simon Glassdf95dd22012-03-13 15:46:16 -0700532 else:
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700533 self._out.Warning("Unknown machine parameter type '%s'" % param)
Simon Glass2c48ddf2012-03-23 16:55:22 -0700534 self._out.Info(' Unknown value: %#0x' % value)
Simon Glassdf95dd22012-03-13 15:46:16 -0700535 new_data += struct.pack('<L', value)
Simon Glass7dbdf5b2012-03-15 20:58:04 -0700536 upto += 4
Simon Glassdf95dd22012-03-13 15:46:16 -0700537
538 # Put the data into our block.
539 data = data[:pos] + new_data + data[pos + len(new_data):]
540 self._out.Info('BL2 configuration complete')
541 return data
542
Simon Glass8e1fdb22012-03-15 21:02:10 -0700543 def _ConfigureExynosBl2(self, fdt, pack, orig_bl2):
Simon Glass7e199222012-03-13 15:51:18 -0700544 """Configure an Exynos BL2 binary for our needs.
545
546 We create a new modified BL2 and return its filename.
547
548 Args:
549 fdt: Device tree containing the parameter values.
Simon Glass8e1fdb22012-03-15 21:02:10 -0700550 pack: The firmware packer object
Simon Glass7e199222012-03-13 15:51:18 -0700551 orig_bl2: Filename of original BL2 file to modify.
552 """
Simon Glass66c1a9f2012-03-22 19:15:53 -0700553 self._out.Info('Configuring BL2')
Simon Glass7e199222012-03-13 15:51:18 -0700554 bl2 = os.path.join(self._tools.outdir, 'updated-spl.bin')
Simon Glass66c1a9f2012-03-22 19:15:53 -0700555 data = self._tools.ReadFile(orig_bl2)
556 self._tools.WriteFile(bl2, data)
Simon Glass7e199222012-03-13 15:51:18 -0700557
558 # Locate the parameter block
559 data = self._tools.ReadFile(bl2)
560 marker = struct.pack('<L', 0xdeadbeef)
561 pos = data.rfind(marker)
562 if not pos:
563 raise CmdError("Could not find machine parameter block in '%s'" %
564 orig_bl2)
Simon Glass8e1fdb22012-03-15 21:02:10 -0700565 data = self._UpdateBl2Parameters(fdt, pack, data, pos)
Simon Glass7e199222012-03-13 15:51:18 -0700566 self._tools.WriteFile(bl2, data)
567 return bl2
568
Simon Glass89b86b82011-07-17 23:49:49 -0700569 def _PackOutput(self, msg):
570 """Helper function to write output from PackFirmware (verbose level 2).
571
572 This is passed to PackFirmware for it to use to write output.
573
574 Args:
575 msg: Message to display.
576 """
577 self._out.Notice(msg)
578
Simon Glass439fe7a2012-03-09 16:19:34 -0800579 def _BuildBlob(self, pack, fdt, blob_type):
580 """Build the blob data for a particular blob type.
581
582 Args:
583 blob_type: The type of blob to create data for. Supported types are:
584 coreboot A coreboot image (ROM plus U-boot and .dtb payloads).
585 signed Nvidia T20/T30 signed image (BCT, U-Boot, .dtb).
586 """
587 if blob_type == 'coreboot':
588 coreboot = self._CreateCorebootStub(self.uboot_fname,
589 self.coreboot_fname, fdt, self.seabios_fname)
590 pack.AddProperty('coreboot', coreboot)
591 pack.AddProperty('image', coreboot)
592 elif blob_type == 'signed':
593 bootstub, signed = self._CreateBootStub(self.uboot_fname, fdt,
594 self.postload_fname)
595 pack.AddProperty('bootstub', bootstub)
596 pack.AddProperty('signed', signed)
597 pack.AddProperty('image', signed)
Simon Glass7e199222012-03-13 15:51:18 -0700598 elif blob_type == 'exynos-bl1':
599 pack.AddProperty(blob_type, self.exynos_bl1)
600 elif blob_type == 'exynos-bl2':
Simon Glass8e1fdb22012-03-15 21:02:10 -0700601 bl2 = self._ConfigureExynosBl2(fdt, pack, self.exynos_bl2)
Simon Glass7e199222012-03-13 15:51:18 -0700602 pack.AddProperty(blob_type, bl2)
Simon Glass439fe7a2012-03-09 16:19:34 -0800603 elif pack.GetProperty(blob_type):
604 pass
605 else:
606 raise CmdError("Unknown blob type '%s' required in flash map" %
607 blob_type)
608
Simon Glass290a1802011-07-17 13:54:32 -0700609 def _CreateImage(self, gbb, fdt):
Simon Glass89b86b82011-07-17 23:49:49 -0700610 """Create a full firmware image, along with various by-products.
611
612 This uses the provided u-boot.bin, fdt and bct to create a firmware
613 image containing all the required parts. If the GBB is not supplied
614 then this will just return a signed U-Boot as the image.
615
616 Args:
Simon Glasse13ee2c2011-07-28 08:12:28 +1200617 gbb: Full path to the GBB file, or empty if a GBB is not required.
618 fdt: Fdt object containing required information.
619
620 Returns:
621 Path to image file
Simon Glass89b86b82011-07-17 23:49:49 -0700622
623 Raises:
624 CmdError if a command fails.
625 """
Simon Glass02d124a2012-03-02 14:47:20 -0800626 self._out.Notice("Model: %s" % fdt.GetString('/', 'model'))
Simon Glass89b86b82011-07-17 23:49:49 -0700627
Simon Glass439fe7a2012-03-09 16:19:34 -0800628 # Get the flashmap so we know what to build
629 pack = PackFirmware(self._tools, self._out)
630 pack.SelectFdt(fdt)
631
632 # Get all our blobs ready
633 pack.AddProperty('boot', self.uboot_fname)
Simon Glass50f74602012-03-15 21:04:25 -0700634
635 # Make a copy of the fdt for the bootstub
636 fdt_data = self._tools.ReadFile(fdt.fname)
637 uboot_data = self._tools.ReadFile(self.uboot_fname)
638 uboot_copy = os.path.join(self._tools.outdir, 'u-boot.bin')
639 self._tools.WriteFile(uboot_copy, uboot_data)
640
641 bootstub = os.path.join(self._tools.outdir, 'u-boot-dtb.bin')
642 self._tools.WriteFile(bootstub, uboot_data + fdt_data)
643 pack.AddProperty('boot+dtb', bootstub)
644
Simon Glass439fe7a2012-03-09 16:19:34 -0800645 pack.AddProperty('gbb', self.uboot_fname)
646 for blob_type in pack.GetBlobList(self.coreboot_fname is not None):
647 self._BuildBlob(pack, fdt, blob_type)
Simon Glass89b86b82011-07-17 23:49:49 -0700648
649 if gbb:
Simon Glasse76bf7b2012-03-13 15:34:41 -0700650 pack.RequireAllEntries()
Hung-Te Lina7462e72011-07-27 19:17:10 +0800651 fwid = '.'.join([
Simon Glass02d124a2012-03-02 14:47:20 -0800652 re.sub('[ ,]+', '_', fdt.GetString('/', 'model')),
Hung-Te Lina7462e72011-07-27 19:17:10 +0800653 self._tools.GetChromeosVersion()])
Simon Glass89b86b82011-07-17 23:49:49 -0700654 self._out.Notice('Firmware ID: %s' % fwid)
Simon Glass439fe7a2012-03-09 16:19:34 -0800655 pack.AddProperty('fwid', fwid)
656 pack.AddProperty('gbb', gbb)
657 pack.AddProperty('keydir', self._keydir)
Simon Glassc90cf582012-03-13 15:40:47 -0700658
659 pack.CheckProperties()
660 image = os.path.join(self._tools.outdir, 'image.bin')
661 pack.PackImage(self._tools.outdir, image)
662 pack.AddProperty('image', image)
Simon Glass89b86b82011-07-17 23:49:49 -0700663
Simon Glass439fe7a2012-03-09 16:19:34 -0800664 image = pack.GetProperty('image')
Simon Glass89b86b82011-07-17 23:49:49 -0700665 self._tools.OutputSize('Final image', image)
Simon Glassc90cf582012-03-13 15:40:47 -0700666 return image, pack
Simon Glass89b86b82011-07-17 23:49:49 -0700667
Simon Glass290a1802011-07-17 13:54:32 -0700668 def SelectFdt(self, fdt_fname):
669 """Select an FDT to control the firmware bundling
670
671 Args:
672 fdt_fname: The filename of the fdt to use.
673
Simon Glassc0f3dc62011-08-09 14:19:05 -0700674 Returns:
675 The Fdt object of the original fdt file, which we will not modify.
676
Simon Glass290a1802011-07-17 13:54:32 -0700677 We make a copy of this which will include any on-the-fly changes we want
678 to make.
679 """
680 self._fdt_fname = fdt_fname
681 self.CheckOptions()
682 fdt = Fdt(self._tools, self._fdt_fname)
Simon Glass29b96ad2012-03-09 15:34:33 -0800683 fdt.Compile()
Simon Glass290a1802011-07-17 13:54:32 -0700684 self.fdt = fdt.Copy(os.path.join(self._tools.outdir, 'updated.dtb'))
Simon Glassc0f3dc62011-08-09 14:19:05 -0700685 return fdt
Simon Glass290a1802011-07-17 13:54:32 -0700686
Simon Glassc90cf582012-03-13 15:40:47 -0700687 def Start(self, hardware_id, output_fname, show_map):
Simon Glass290a1802011-07-17 13:54:32 -0700688 """This creates a firmware bundle according to settings provided.
Simon Glass89b86b82011-07-17 23:49:49 -0700689
690 - Checks options, tools, output directory, fdt.
691 - Creates GBB and image.
Simon Glass290a1802011-07-17 13:54:32 -0700692
693 Args:
Simon Glass56577572011-07-19 11:08:06 +1200694 hardware_id: Hardware ID to use for this board. If None, then the
695 default from the Fdt will be used
Simon Glass290a1802011-07-17 13:54:32 -0700696 output_fname: Output filename for the image. If this is not None, then
697 the final image will be copied here.
Simon Glassc90cf582012-03-13 15:40:47 -0700698 show_map: Show a flash map, with each area's name and position
Simon Glass290a1802011-07-17 13:54:32 -0700699
700 Returns:
701 Filename of the resulting image (not the output_fname copy).
Simon Glass89b86b82011-07-17 23:49:49 -0700702 """
Simon Glass89b86b82011-07-17 23:49:49 -0700703 gbb = ''
Simon Glass290a1802011-07-17 13:54:32 -0700704 if not self._small:
Simon Glass56577572011-07-19 11:08:06 +1200705 gbb = self._CreateGoogleBinaryBlock(hardware_id)
Simon Glass89b86b82011-07-17 23:49:49 -0700706
707 # This creates the actual image.
Simon Glassc90cf582012-03-13 15:40:47 -0700708 image, pack = self._CreateImage(gbb, self.fdt)
709 if show_map:
710 pack.ShowMap()
Simon Glass290a1802011-07-17 13:54:32 -0700711 if output_fname:
712 shutil.copyfile(image, output_fname)
713 self._out.Notice("Output image '%s'" % output_fname)
714 return image