blob: 2f00279232b87dea4c85428aba487d2668b6346a [file] [log] [blame]
Tom Rini83d290c2018-05-06 17:58:06 -04001# SPDX-License-Identifier: GPL-2.0+
Simon Glassbf7fd502016-11-25 20:15:51 -07002# Copyright (c) 2016 Google, Inc
3# Written by Simon Glass <sjg@chromium.org>
4#
Simon Glassbf7fd502016-11-25 20:15:51 -07005# Creates binary images from input files controlled by a description
6#
7
8from collections import OrderedDict
Simon Glass87d43322020-08-05 13:27:46 -06009import glob
Jan Kiszkade65b122023-04-22 16:42:48 +020010try:
11 import importlib.resources
Simon Glasse2259612023-06-01 10:22:25 -060012except ImportError: # pragma: no cover
Jan Kiszkade65b122023-04-22 16:42:48 +020013 # for Python 3.6
14 import importlib_resources
Simon Glassbf7fd502016-11-25 20:15:51 -070015import os
Simon Glass9fbfaba2020-08-29 11:36:14 -060016import pkg_resources
Simon Glassb2381432020-09-06 10:39:09 -060017import re
Simon Glass9fbfaba2020-08-29 11:36:14 -060018
Simon Glassbf7fd502016-11-25 20:15:51 -070019import sys
Simon Glassbf7fd502016-11-25 20:15:51 -070020
Simon Glass386c63c2022-01-09 20:13:50 -070021from binman import bintool
Simon Glass16287932020-04-17 18:09:03 -060022from binman import cbfs_util
Simon Glass7960a0a2022-08-07 09:46:46 -060023from binman import elf
24from binman import entry
Simon Glassf6abd522023-07-18 07:24:04 -060025from dtoc import fdt_util
Simon Glass4583c002023-02-23 18:18:04 -070026from u_boot_pylib import command
27from u_boot_pylib import tools
28from u_boot_pylib import tout
Simon Glassbf7fd502016-11-25 20:15:51 -070029
Simon Glass8d2ef3e2022-02-11 13:23:21 -070030# These are imported if needed since they import libfdt
31state = None
32Image = None
33
Simon Glassbf7fd502016-11-25 20:15:51 -070034# List of images we plan to create
35# Make this global so that it can be referenced from tests
36images = OrderedDict()
37
Simon Glassb2381432020-09-06 10:39:09 -060038# Help text for each type of missing blob, dict:
39# key: Value of the entry's 'missing-msg' or entry name
40# value: Text for the help
41missing_blob_help = {}
42
Simon Glass0b6023e2021-03-18 20:25:06 +130043def _ReadImageDesc(binman_node, use_expanded):
Simon Glassbf7fd502016-11-25 20:15:51 -070044 """Read the image descriptions from the /binman node
45
46 This normally produces a single Image object called 'image'. But if
47 multiple images are present, they will all be returned.
48
49 Args:
50 binman_node: Node object of the /binman node
Simon Glass0b6023e2021-03-18 20:25:06 +130051 use_expanded: True if the FDT will be updated with the entry information
Simon Glassbf7fd502016-11-25 20:15:51 -070052 Returns:
53 OrderedDict of Image objects, each of which describes an image
54 """
Simon Glass8d2ef3e2022-02-11 13:23:21 -070055 # For Image()
56 # pylint: disable=E1102
Simon Glassbf7fd502016-11-25 20:15:51 -070057 images = OrderedDict()
58 if 'multiple-images' in binman_node.props:
59 for node in binman_node.subnodes:
Simon Glassaf41b242023-07-22 21:43:56 -060060 if not node.name.startswith('template'):
Simon Glass35f72fb2023-07-18 07:24:05 -060061 images[node.name] = Image(node.name, node,
62 use_expanded=use_expanded)
Simon Glassbf7fd502016-11-25 20:15:51 -070063 else:
Simon Glass0b6023e2021-03-18 20:25:06 +130064 images['image'] = Image('image', binman_node, use_expanded=use_expanded)
Simon Glassbf7fd502016-11-25 20:15:51 -070065 return images
66
Simon Glassec3f3782017-05-27 07:38:29 -060067def _FindBinmanNode(dtb):
Simon Glassbf7fd502016-11-25 20:15:51 -070068 """Find the 'binman' node in the device tree
69
70 Args:
Simon Glassec3f3782017-05-27 07:38:29 -060071 dtb: Fdt object to scan
Simon Glassbf7fd502016-11-25 20:15:51 -070072 Returns:
73 Node object of /binman node, or None if not found
74 """
Simon Glassec3f3782017-05-27 07:38:29 -060075 for node in dtb.GetRoot().subnodes:
Simon Glassbf7fd502016-11-25 20:15:51 -070076 if node.name == 'binman':
77 return node
78 return None
79
Simon Glassb2381432020-09-06 10:39:09 -060080def _ReadMissingBlobHelp():
81 """Read the missing-blob-help file
82
83 This file containins help messages explaining what to do when external blobs
84 are missing.
85
86 Returns:
87 Dict:
88 key: Message tag (str)
89 value: Message text (str)
90 """
91
92 def _FinishTag(tag, msg, result):
93 if tag:
94 result[tag] = msg.rstrip()
95 tag = None
96 msg = ''
97 return tag, msg
98
99 my_data = pkg_resources.resource_string(__name__, 'missing-blob-help')
100 re_tag = re.compile('^([-a-z0-9]+):$')
101 result = {}
102 tag = None
103 msg = ''
104 for line in my_data.decode('utf-8').splitlines():
105 if not line.startswith('#'):
106 m_tag = re_tag.match(line)
107 if m_tag:
108 _, msg = _FinishTag(tag, msg, result)
109 tag = m_tag.group(1)
110 elif tag:
111 msg += line + '\n'
112 _FinishTag(tag, msg, result)
113 return result
114
Jonas Karlman8f452bc2023-07-18 20:34:39 +0000115def _ShowBlobHelp(level, path, text, fname):
116 tout.do_output(level, '%s (%s):' % (path, fname))
Simon Glassb2381432020-09-06 10:39:09 -0600117 for line in text.splitlines():
Jonas Karlmand92c4dd2023-07-18 20:34:35 +0000118 tout.do_output(level, ' %s' % line)
Jonas Karlmanc2600152023-07-18 20:34:37 +0000119 tout.do_output(level, '')
Simon Glassb2381432020-09-06 10:39:09 -0600120
Jonas Karlmand92c4dd2023-07-18 20:34:35 +0000121def _ShowHelpForMissingBlobs(level, missing_list):
Simon Glassb2381432020-09-06 10:39:09 -0600122 """Show help for each missing blob to help the user take action
123
124 Args:
125 missing_list: List of Entry objects to show help for
126 """
127 global missing_blob_help
128
129 if not missing_blob_help:
130 missing_blob_help = _ReadMissingBlobHelp()
131
132 for entry in missing_list:
133 tags = entry.GetHelpTags()
134
135 # Show the first match help message
Jonas Karlman8f452bc2023-07-18 20:34:39 +0000136 shown_help = False
Simon Glassb2381432020-09-06 10:39:09 -0600137 for tag in tags:
138 if tag in missing_blob_help:
Jonas Karlman8f452bc2023-07-18 20:34:39 +0000139 _ShowBlobHelp(level, entry._node.path, missing_blob_help[tag],
140 entry.GetDefaultFilename())
141 shown_help = True
Simon Glassb2381432020-09-06 10:39:09 -0600142 break
Jonas Karlman8f452bc2023-07-18 20:34:39 +0000143 # Or a generic help message
144 if not shown_help:
145 _ShowBlobHelp(level, entry._node.path, "Missing blob",
146 entry.GetDefaultFilename())
Simon Glassb2381432020-09-06 10:39:09 -0600147
Simon Glass87d43322020-08-05 13:27:46 -0600148def GetEntryModules(include_testing=True):
149 """Get a set of entry class implementations
150
151 Returns:
152 Set of paths to entry class filenames
153 """
Simon Glass9fbfaba2020-08-29 11:36:14 -0600154 glob_list = pkg_resources.resource_listdir(__name__, 'etype')
155 glob_list = [fname for fname in glob_list if fname.endswith('.py')]
Simon Glass87d43322020-08-05 13:27:46 -0600156 return set([os.path.splitext(os.path.basename(item))[0]
157 for item in glob_list
158 if include_testing or '_testing' not in item])
159
Simon Glassc55a50f2018-09-14 04:57:19 -0600160def WriteEntryDocs(modules, test_missing=None):
161 """Write out documentation for all entries
Simon Glassecab8972018-07-06 10:27:40 -0600162
163 Args:
Simon Glassc55a50f2018-09-14 04:57:19 -0600164 modules: List of Module objects to get docs for
Simon Glassbc570642022-01-09 20:14:11 -0700165 test_missing: Used for testing only, to force an entry's documentation
Simon Glassc55a50f2018-09-14 04:57:19 -0600166 to show as missing even if it is present. Should be set to None in
167 normal use.
Simon Glassecab8972018-07-06 10:27:40 -0600168 """
Simon Glass16287932020-04-17 18:09:03 -0600169 from binman.entry import Entry
Simon Glassfd8d1f72018-07-17 13:25:36 -0600170 Entry.WriteDocs(modules, test_missing)
171
Simon Glass61f564d2019-07-08 14:25:48 -0600172
Simon Glassbc570642022-01-09 20:14:11 -0700173def write_bintool_docs(modules, test_missing=None):
174 """Write out documentation for all bintools
175
176 Args:
177 modules: List of Module objects to get docs for
178 test_missing: Used for testing only, to force an entry's documentation
179 to show as missing even if it is present. Should be set to None in
180 normal use.
181 """
182 bintool.Bintool.WriteDocs(modules, test_missing)
183
184
Simon Glass61f564d2019-07-08 14:25:48 -0600185def ListEntries(image_fname, entry_paths):
186 """List the entries in an image
187
188 This decodes the supplied image and displays a table of entries from that
189 image, preceded by a header.
190
191 Args:
192 image_fname: Image filename to process
193 entry_paths: List of wildcarded paths (e.g. ['*dtb*', 'u-boot*',
194 'section/u-boot'])
195 """
196 image = Image.FromFile(image_fname)
197
198 entries, lines, widths = image.GetListEntries(entry_paths)
199
200 num_columns = len(widths)
201 for linenum, line in enumerate(lines):
202 if linenum == 1:
203 # Print header line
204 print('-' * (sum(widths) + num_columns * 2))
205 out = ''
206 for i, item in enumerate(line):
207 width = -widths[i]
208 if item.startswith('>'):
209 width = -width
210 item = item[1:]
211 txt = '%*s ' % (width, item)
212 out += txt
213 print(out.rstrip())
214
Simon Glassf667e452019-07-08 14:25:50 -0600215
216def ReadEntry(image_fname, entry_path, decomp=True):
217 """Extract an entry from an image
218
219 This extracts the data from a particular entry in an image
220
221 Args:
222 image_fname: Image filename to process
223 entry_path: Path to entry to extract
224 decomp: True to return uncompressed data, if the data is compress
225 False to return the raw data
226
227 Returns:
228 data extracted from the entry
229 """
Simon Glass8dbb7442019-08-24 07:22:44 -0600230 global Image
Simon Glass07237982020-08-05 13:27:47 -0600231 from binman.image import Image
Simon Glass8dbb7442019-08-24 07:22:44 -0600232
Simon Glassf667e452019-07-08 14:25:50 -0600233 image = Image.FromFile(image_fname)
Stefan Herbrechtsmeier917b3c32022-08-19 16:25:22 +0200234 image.CollectBintools()
Simon Glassf667e452019-07-08 14:25:50 -0600235 entry = image.FindEntryPath(entry_path)
236 return entry.ReadData(decomp)
237
238
Simon Glass943bf782021-11-23 21:09:50 -0700239def ShowAltFormats(image):
240 """Show alternative formats available for entries in the image
241
242 This shows a list of formats available.
243
244 Args:
245 image (Image): Image to check
246 """
247 alt_formats = {}
248 image.CheckAltFormats(alt_formats)
249 print('%-10s %-20s %s' % ('Flag (-F)', 'Entry type', 'Description'))
250 for name, val in alt_formats.items():
251 entry, helptext = val
252 print('%-10s %-20s %s' % (name, entry.etype, helptext))
253
254
Simon Glass71ce0ba2019-07-08 14:25:52 -0600255def ExtractEntries(image_fname, output_fname, outdir, entry_paths,
Simon Glass943bf782021-11-23 21:09:50 -0700256 decomp=True, alt_format=None):
Simon Glass71ce0ba2019-07-08 14:25:52 -0600257 """Extract the data from one or more entries and write it to files
258
259 Args:
260 image_fname: Image filename to process
261 output_fname: Single output filename to use if extracting one file, None
262 otherwise
263 outdir: Output directory to use (for any number of files), else None
264 entry_paths: List of entry paths to extract
Simon Glass3ad804e2019-07-20 12:24:12 -0600265 decomp: True to decompress the entry data
Simon Glass71ce0ba2019-07-08 14:25:52 -0600266
267 Returns:
268 List of EntryInfo records that were written
269 """
270 image = Image.FromFile(image_fname)
Stefan Herbrechtsmeier917b3c32022-08-19 16:25:22 +0200271 image.CollectBintools()
Simon Glass71ce0ba2019-07-08 14:25:52 -0600272
Simon Glass943bf782021-11-23 21:09:50 -0700273 if alt_format == 'list':
274 ShowAltFormats(image)
275 return
276
Simon Glass71ce0ba2019-07-08 14:25:52 -0600277 # Output an entry to a single file, as a special case
278 if output_fname:
279 if not entry_paths:
Simon Glassbb5edc12019-07-20 12:24:14 -0600280 raise ValueError('Must specify an entry path to write with -f')
Simon Glass71ce0ba2019-07-08 14:25:52 -0600281 if len(entry_paths) != 1:
Simon Glassbb5edc12019-07-20 12:24:14 -0600282 raise ValueError('Must specify exactly one entry path to write with -f')
Simon Glass71ce0ba2019-07-08 14:25:52 -0600283 entry = image.FindEntryPath(entry_paths[0])
Simon Glass943bf782021-11-23 21:09:50 -0700284 data = entry.ReadData(decomp, alt_format)
Simon Glassc1aa66e2022-01-29 14:14:04 -0700285 tools.write_file(output_fname, data)
Simon Glassf3385a52022-01-29 14:14:15 -0700286 tout.notice("Wrote %#x bytes to file '%s'" % (len(data), output_fname))
Simon Glass71ce0ba2019-07-08 14:25:52 -0600287 return
288
289 # Otherwise we will output to a path given by the entry path of each entry.
290 # This means that entries will appear in subdirectories if they are part of
291 # a sub-section.
292 einfos = image.GetListEntries(entry_paths)[0]
Simon Glassf3385a52022-01-29 14:14:15 -0700293 tout.notice('%d entries match and will be written' % len(einfos))
Simon Glass71ce0ba2019-07-08 14:25:52 -0600294 for einfo in einfos:
295 entry = einfo.entry
Simon Glass943bf782021-11-23 21:09:50 -0700296 data = entry.ReadData(decomp, alt_format)
Simon Glass71ce0ba2019-07-08 14:25:52 -0600297 path = entry.GetPath()[1:]
298 fname = os.path.join(outdir, path)
299
300 # If this entry has children, create a directory for it and put its
301 # data in a file called 'root' in that directory
302 if entry.GetEntries():
Simon Glass862ddf92021-03-18 20:24:51 +1300303 if fname and not os.path.exists(fname):
Simon Glass71ce0ba2019-07-08 14:25:52 -0600304 os.makedirs(fname)
305 fname = os.path.join(fname, 'root')
Simon Glassf3385a52022-01-29 14:14:15 -0700306 tout.notice("Write entry '%s' size %x to '%s'" %
Simon Glass5b378e42021-01-06 21:35:13 -0700307 (entry.GetPath(), len(data), fname))
Simon Glassc1aa66e2022-01-29 14:14:04 -0700308 tools.write_file(fname, data)
Simon Glass71ce0ba2019-07-08 14:25:52 -0600309 return einfos
310
311
Simon Glassd7fa4e42019-07-20 12:24:13 -0600312def BeforeReplace(image, allow_resize):
313 """Handle getting an image ready for replacing entries in it
314
315 Args:
316 image: Image to prepare
317 """
318 state.PrepareFromLoadedData(image)
Alper Nebi Yasak8ee4ec92022-03-27 18:31:45 +0300319 image.CollectBintools()
Lukas Funke7a52a452023-07-18 13:53:10 +0200320 image.LoadData(decomp=False)
Simon Glassd7fa4e42019-07-20 12:24:13 -0600321
322 # If repacking, drop the old offset/size values except for the original
323 # ones, so we are only left with the constraints.
Alper Nebi Yasake2ce4fb2022-03-27 18:31:46 +0300324 if image.allow_repack and allow_resize:
Simon Glassd7fa4e42019-07-20 12:24:13 -0600325 image.ResetForPack()
326
327
328def ReplaceOneEntry(image, entry, data, do_compress, allow_resize):
329 """Handle replacing a single entry an an image
330
331 Args:
332 image: Image to update
333 entry: Entry to write
334 data: Data to replace with
335 do_compress: True to compress the data if needed, False if data is
336 already compressed so should be used as is
337 allow_resize: True to allow entries to change size (this does a re-pack
338 of the entries), False to raise an exception
339 """
340 if not entry.WriteData(data, do_compress):
341 if not image.allow_repack:
342 entry.Raise('Entry data size does not match, but allow-repack is not present for this image')
343 if not allow_resize:
344 entry.Raise('Entry data size does not match, but resize is disabled')
345
346
347def AfterReplace(image, allow_resize, write_map):
348 """Handle write out an image after replacing entries in it
349
350 Args:
351 image: Image to write
352 allow_resize: True to allow entries to change size (this does a re-pack
353 of the entries), False to raise an exception
354 write_map: True to write a map file
355 """
Simon Glassf3385a52022-01-29 14:14:15 -0700356 tout.info('Processing image')
Simon Glassd7fa4e42019-07-20 12:24:13 -0600357 ProcessImage(image, update_fdt=True, write_map=write_map,
358 get_contents=False, allow_resize=allow_resize)
359
360
361def WriteEntryToImage(image, entry, data, do_compress=True, allow_resize=True,
362 write_map=False):
363 BeforeReplace(image, allow_resize)
Simon Glassf3385a52022-01-29 14:14:15 -0700364 tout.info('Writing data to %s' % entry.GetPath())
Simon Glassd7fa4e42019-07-20 12:24:13 -0600365 ReplaceOneEntry(image, entry, data, do_compress, allow_resize)
366 AfterReplace(image, allow_resize=allow_resize, write_map=write_map)
367
368
Simon Glass3ad804e2019-07-20 12:24:12 -0600369def WriteEntry(image_fname, entry_path, data, do_compress=True,
370 allow_resize=True, write_map=False):
Simon Glass22a76b72019-07-20 12:24:11 -0600371 """Replace an entry in an image
372
373 This replaces the data in a particular entry in an image. This size of the
374 new data must match the size of the old data unless allow_resize is True.
375
376 Args:
377 image_fname: Image filename to process
378 entry_path: Path to entry to extract
379 data: Data to replace with
Simon Glass3ad804e2019-07-20 12:24:12 -0600380 do_compress: True to compress the data if needed, False if data is
Simon Glass22a76b72019-07-20 12:24:11 -0600381 already compressed so should be used as is
382 allow_resize: True to allow entries to change size (this does a re-pack
383 of the entries), False to raise an exception
Simon Glass3ad804e2019-07-20 12:24:12 -0600384 write_map: True to write a map file
Simon Glass22a76b72019-07-20 12:24:11 -0600385
386 Returns:
387 Image object that was updated
388 """
Simon Glassf3385a52022-01-29 14:14:15 -0700389 tout.info("Write entry '%s', file '%s'" % (entry_path, image_fname))
Simon Glass22a76b72019-07-20 12:24:11 -0600390 image = Image.FromFile(image_fname)
Stefan Herbrechtsmeier917b3c32022-08-19 16:25:22 +0200391 image.CollectBintools()
Simon Glass22a76b72019-07-20 12:24:11 -0600392 entry = image.FindEntryPath(entry_path)
Simon Glassd7fa4e42019-07-20 12:24:13 -0600393 WriteEntryToImage(image, entry, data, do_compress=do_compress,
394 allow_resize=allow_resize, write_map=write_map)
Simon Glass22a76b72019-07-20 12:24:11 -0600395
Simon Glass22a76b72019-07-20 12:24:11 -0600396 return image
397
Simon Glassa6cb9952019-07-20 12:24:15 -0600398
399def ReplaceEntries(image_fname, input_fname, indir, entry_paths,
400 do_compress=True, allow_resize=True, write_map=False):
401 """Replace the data from one or more entries from input files
402
403 Args:
404 image_fname: Image filename to process
Jan Kiszka89cc0522021-11-11 08:13:30 +0100405 input_fname: Single input filename to use if replacing one file, None
Simon Glassa6cb9952019-07-20 12:24:15 -0600406 otherwise
407 indir: Input directory to use (for any number of files), else None
Jan Kiszka89cc0522021-11-11 08:13:30 +0100408 entry_paths: List of entry paths to replace
Simon Glassa6cb9952019-07-20 12:24:15 -0600409 do_compress: True if the input data is uncompressed and may need to be
410 compressed if the entry requires it, False if the data is already
411 compressed.
412 write_map: True to write a map file
413
414 Returns:
415 List of EntryInfo records that were written
416 """
Jan Kiszkac700f102021-11-11 08:14:18 +0100417 image_fname = os.path.abspath(image_fname)
Simon Glassa6cb9952019-07-20 12:24:15 -0600418 image = Image.FromFile(image_fname)
419
Simon Glass7caa3722023-03-02 17:02:44 -0700420 image.mark_build_done()
421
Simon Glassa6cb9952019-07-20 12:24:15 -0600422 # Replace an entry from a single file, as a special case
423 if input_fname:
424 if not entry_paths:
425 raise ValueError('Must specify an entry path to read with -f')
426 if len(entry_paths) != 1:
427 raise ValueError('Must specify exactly one entry path to write with -f')
428 entry = image.FindEntryPath(entry_paths[0])
Simon Glassc1aa66e2022-01-29 14:14:04 -0700429 data = tools.read_file(input_fname)
Simon Glassf3385a52022-01-29 14:14:15 -0700430 tout.notice("Read %#x bytes from file '%s'" % (len(data), input_fname))
Simon Glassa6cb9952019-07-20 12:24:15 -0600431 WriteEntryToImage(image, entry, data, do_compress=do_compress,
432 allow_resize=allow_resize, write_map=write_map)
433 return
434
435 # Otherwise we will input from a path given by the entry path of each entry.
436 # This means that files must appear in subdirectories if they are part of
437 # a sub-section.
438 einfos = image.GetListEntries(entry_paths)[0]
Simon Glassf3385a52022-01-29 14:14:15 -0700439 tout.notice("Replacing %d matching entries in image '%s'" %
Simon Glassa6cb9952019-07-20 12:24:15 -0600440 (len(einfos), image_fname))
441
442 BeforeReplace(image, allow_resize)
443
444 for einfo in einfos:
445 entry = einfo.entry
446 if entry.GetEntries():
Simon Glassf3385a52022-01-29 14:14:15 -0700447 tout.info("Skipping section entry '%s'" % entry.GetPath())
Simon Glassa6cb9952019-07-20 12:24:15 -0600448 continue
449
450 path = entry.GetPath()[1:]
451 fname = os.path.join(indir, path)
452
453 if os.path.exists(fname):
Simon Glassf3385a52022-01-29 14:14:15 -0700454 tout.notice("Write entry '%s' from file '%s'" %
Simon Glassa6cb9952019-07-20 12:24:15 -0600455 (entry.GetPath(), fname))
Simon Glassc1aa66e2022-01-29 14:14:04 -0700456 data = tools.read_file(fname)
Simon Glassa6cb9952019-07-20 12:24:15 -0600457 ReplaceOneEntry(image, entry, data, do_compress, allow_resize)
458 else:
Simon Glassf3385a52022-01-29 14:14:15 -0700459 tout.warning("Skipping entry '%s' from missing file '%s'" %
Simon Glassa6cb9952019-07-20 12:24:15 -0600460 (entry.GetPath(), fname))
461
462 AfterReplace(image, allow_resize=allow_resize, write_map=write_map)
463 return image
464
Ivan Mikhaylov4023dc92023-03-08 01:13:39 +0000465def SignEntries(image_fname, input_fname, privatekey_fname, algo, entry_paths,
466 write_map=False):
467 """Sign and replace the data from one or more entries from input files
468
469 Args:
470 image_fname: Image filename to process
471 input_fname: Single input filename to use if replacing one file, None
472 otherwise
473 algo: Hashing algorithm
474 entry_paths: List of entry paths to sign
475 privatekey_fname: Private key filename
476 write_map (bool): True to write the map file
477 """
478 image_fname = os.path.abspath(image_fname)
479 image = Image.FromFile(image_fname)
480
Ivan Mikhaylov5b34efe2023-03-08 01:13:40 +0000481 image.mark_build_done()
482
Ivan Mikhaylov4023dc92023-03-08 01:13:39 +0000483 BeforeReplace(image, allow_resize=True)
484
485 for entry_path in entry_paths:
486 entry = image.FindEntryPath(entry_path)
487 entry.UpdateSignatures(privatekey_fname, algo, input_fname)
488
489 AfterReplace(image, allow_resize=True, write_map=write_map)
Simon Glassa6cb9952019-07-20 12:24:15 -0600490
Simon Glassf6abd522023-07-18 07:24:04 -0600491def _ProcessTemplates(parent):
492 """Handle any templates in the binman description
493
494 Args:
495 parent: Binman node to process (typically /binman)
496
Simon Glassb2f47a52023-07-22 21:43:52 -0600497 Returns:
498 bool: True if any templates were processed
499
Simon Glassf6abd522023-07-18 07:24:04 -0600500 Search though each target node looking for those with an 'insert-template'
501 property. Use that as a list of references to template nodes to use to
502 adjust the target node.
503
504 Processing involves copying each subnode of the template node into the
505 target node.
506
Simon Glass696f2b72023-07-18 07:24:07 -0600507 This is done recursively, so templates can be at any level of the binman
508 image, e.g. inside a section.
Simon Glassf6abd522023-07-18 07:24:04 -0600509
510 See 'Templates' in the Binman documnentation for details.
511 """
Simon Glassb2f47a52023-07-22 21:43:52 -0600512 found = False
Simon Glassf6abd522023-07-18 07:24:04 -0600513 for node in parent.subnodes:
514 tmpl = fdt_util.GetPhandleList(node, 'insert-template')
515 if tmpl:
516 node.copy_subnodes_from_phandles(tmpl)
Simon Glassb2f47a52023-07-22 21:43:52 -0600517 found = True
518
519 found |= _ProcessTemplates(node)
520 return found
Simon Glassf6abd522023-07-18 07:24:04 -0600521
Simon Glassaf41b242023-07-22 21:43:56 -0600522def _RemoveTemplates(parent):
523 """Remove any templates in the binman description
524 """
525 for node in parent.subnodes:
526 if node.name.startswith('template'):
527 node.Delete()
528
Simon Glass0b6023e2021-03-18 20:25:06 +1300529def PrepareImagesAndDtbs(dtb_fname, select_images, update_fdt, use_expanded):
Simon Glassa8573c42019-07-20 12:23:27 -0600530 """Prepare the images to be processed and select the device tree
531
532 This function:
533 - reads in the device tree
534 - finds and scans the binman node to create all entries
535 - selects which images to build
536 - Updates the device tress with placeholder properties for offset,
537 image-pos, etc.
538
539 Args:
540 dtb_fname: Filename of the device tree file to use (.dts or .dtb)
541 selected_images: List of images to output, or None for all
542 update_fdt: True to update the FDT wth entry offsets, etc.
Simon Glass0b6023e2021-03-18 20:25:06 +1300543 use_expanded: True to use expanded versions of entries, if available.
544 So if 'u-boot' is called for, we use 'u-boot-expanded' instead. This
545 is needed if update_fdt is True (although tests may disable it)
Simon Glasse9d336d2020-09-01 05:13:55 -0600546
547 Returns:
548 OrderedDict of images:
549 key: Image name (str)
550 value: Image object
Simon Glassa8573c42019-07-20 12:23:27 -0600551 """
552 # Import these here in case libfdt.py is not available, in which case
553 # the above help option still works.
Simon Glass16287932020-04-17 18:09:03 -0600554 from dtoc import fdt
555 from dtoc import fdt_util
Simon Glassa8573c42019-07-20 12:23:27 -0600556 global images
557
558 # Get the device tree ready by compiling it and copying the compiled
559 # output into a file in our output directly. Then scan it for use
560 # in binman.
561 dtb_fname = fdt_util.EnsureCompiled(dtb_fname)
Simon Glassc1aa66e2022-01-29 14:14:04 -0700562 fname = tools.get_output_filename('u-boot.dtb.out')
563 tools.write_file(fname, tools.read_file(dtb_fname))
Simon Glassa8573c42019-07-20 12:23:27 -0600564 dtb = fdt.FdtScan(fname)
565
566 node = _FindBinmanNode(dtb)
567 if not node:
568 raise ValueError("Device tree '%s' does not have a 'binman' "
569 "node" % dtb_fname)
570
Simon Glassb2f47a52023-07-22 21:43:52 -0600571 if _ProcessTemplates(node):
572 dtb.Sync(True)
573 fname = tools.get_output_filename('u-boot.dtb.tmpl1')
574 tools.write_file(fname, dtb.GetContents())
Simon Glassf6abd522023-07-18 07:24:04 -0600575
Simon Glassaf41b242023-07-22 21:43:56 -0600576 _RemoveTemplates(node)
577 dtb.Sync(True)
Simon Glassd4d97662023-07-22 21:43:57 -0600578
579 # Rescan the dtb to pick up the new phandles
580 dtb.Scan()
581 node = _FindBinmanNode(dtb)
Simon Glassaf41b242023-07-22 21:43:56 -0600582 fname = tools.get_output_filename('u-boot.dtb.tmpl2')
583 tools.write_file(fname, dtb.GetContents())
584
Simon Glass0b6023e2021-03-18 20:25:06 +1300585 images = _ReadImageDesc(node, use_expanded)
Simon Glassa8573c42019-07-20 12:23:27 -0600586
587 if select_images:
588 skip = []
589 new_images = OrderedDict()
590 for name, image in images.items():
591 if name in select_images:
592 new_images[name] = image
593 else:
594 skip.append(name)
595 images = new_images
Simon Glassf3385a52022-01-29 14:14:15 -0700596 tout.notice('Skipping images: %s' % ', '.join(skip))
Simon Glassa8573c42019-07-20 12:23:27 -0600597
598 state.Prepare(images, dtb)
599
600 # Prepare the device tree by making sure that any missing
601 # properties are added (e.g. 'pos' and 'size'). The values of these
602 # may not be correct yet, but we add placeholders so that the
603 # size of the device tree is correct. Later, in
604 # SetCalculatedProperties() we will insert the correct values
605 # without changing the device-tree size, thus ensuring that our
606 # entry offsets remain the same.
607 for image in images.values():
Simon Glassc9ee33a2022-03-05 20:19:00 -0700608 image.gen_entries()
Stefan Herbrechtsmeier917b3c32022-08-19 16:25:22 +0200609 image.CollectBintools()
Simon Glassa8573c42019-07-20 12:23:27 -0600610 if update_fdt:
Simon Glassa9fad072020-10-26 17:40:17 -0600611 image.AddMissingProperties(True)
Simon Glassa8573c42019-07-20 12:23:27 -0600612 image.ProcessFdt(dtb)
613
Simon Glass4bdd1152019-07-20 12:23:29 -0600614 for dtb_item in state.GetAllFdts():
Simon Glassa8573c42019-07-20 12:23:27 -0600615 dtb_item.Sync(auto_resize=True)
616 dtb_item.Pack()
617 dtb_item.Flush()
618 return images
619
620
Simon Glass51014aa2019-07-20 12:23:56 -0600621def ProcessImage(image, update_fdt, write_map, get_contents=True,
Heiko Thierya89c8f22022-01-06 11:49:41 +0100622 allow_resize=True, allow_missing=False,
623 allow_fake_blobs=False):
Simon Glassb88e81c2019-07-20 12:23:24 -0600624 """Perform all steps for this image, including checking and # writing it.
625
626 This means that errors found with a later image will be reported after
627 earlier images are already completed and written, but that does not seem
628 important.
629
630 Args:
631 image: Image to process
632 update_fdt: True to update the FDT wth entry offsets, etc.
633 write_map: True to write a map file
Simon Glass10f9d002019-07-20 12:23:50 -0600634 get_contents: True to get the image contents from files, etc., False if
635 the contents is already present
Simon Glass51014aa2019-07-20 12:23:56 -0600636 allow_resize: True to allow entries to change size (this does a re-pack
637 of the entries), False to raise an exception
Simon Glass4f9f1052020-07-09 18:39:38 -0600638 allow_missing: Allow blob_ext objects to be missing
Heiko Thierya89c8f22022-01-06 11:49:41 +0100639 allow_fake_blobs: Allow blob_ext objects to be faked with dummy files
Simon Glassb1cca952020-07-09 18:39:40 -0600640
641 Returns:
Heiko Thierya89c8f22022-01-06 11:49:41 +0100642 True if one or more external blobs are missing or faked,
643 False if all are present
Simon Glassb88e81c2019-07-20 12:23:24 -0600644 """
Simon Glass10f9d002019-07-20 12:23:50 -0600645 if get_contents:
Simon Glass4f9f1052020-07-09 18:39:38 -0600646 image.SetAllowMissing(allow_missing)
Heiko Thierya89c8f22022-01-06 11:49:41 +0100647 image.SetAllowFakeBlob(allow_fake_blobs)
Simon Glass10f9d002019-07-20 12:23:50 -0600648 image.GetEntryContents()
Simon Glassc8c9f312023-01-07 14:07:12 -0700649 image.drop_absent()
Simon Glassb88e81c2019-07-20 12:23:24 -0600650 image.GetEntryOffsets()
651
652 # We need to pack the entries to figure out where everything
653 # should be placed. This sets the offset/size of each entry.
654 # However, after packing we call ProcessEntryContents() which
655 # may result in an entry changing size. In that case we need to
656 # do another pass. Since the device tree often contains the
657 # final offset/size information we try to make space for this in
658 # AddMissingProperties() above. However, if the device is
659 # compressed we cannot know this compressed size in advance,
660 # since changing an offset from 0x100 to 0x104 (for example) can
661 # alter the compressed size of the device tree. So we need a
662 # third pass for this.
Simon Glasseb0f4a42019-07-20 12:24:06 -0600663 passes = 5
Simon Glassb88e81c2019-07-20 12:23:24 -0600664 for pack_pass in range(passes):
665 try:
666 image.PackEntries()
Simon Glassb88e81c2019-07-20 12:23:24 -0600667 except Exception as e:
668 if write_map:
669 fname = image.WriteMap()
670 print("Wrote map file '%s' to show errors" % fname)
671 raise
672 image.SetImagePos()
673 if update_fdt:
674 image.SetCalculatedProperties()
Simon Glass4bdd1152019-07-20 12:23:29 -0600675 for dtb_item in state.GetAllFdts():
Simon Glassb88e81c2019-07-20 12:23:24 -0600676 dtb_item.Sync()
Simon Glass51014aa2019-07-20 12:23:56 -0600677 dtb_item.Flush()
Simon Glass261cbe02019-08-24 07:23:12 -0600678 image.WriteSymbols()
Simon Glassb88e81c2019-07-20 12:23:24 -0600679 sizes_ok = image.ProcessEntryContents()
680 if sizes_ok:
681 break
682 image.ResetForPack()
Simon Glassf3385a52022-01-29 14:14:15 -0700683 tout.info('Pack completed after %d pass(es)' % (pack_pass + 1))
Simon Glassb88e81c2019-07-20 12:23:24 -0600684 if not sizes_ok:
Simon Glass61ec04f2019-07-20 12:23:58 -0600685 image.Raise('Entries changed size after packing (tried %s passes)' %
Simon Glassb88e81c2019-07-20 12:23:24 -0600686 passes)
687
Simon Glassb88e81c2019-07-20 12:23:24 -0600688 image.BuildImage()
689 if write_map:
690 image.WriteMap()
Simon Glass67a05012023-01-07 14:07:15 -0700691
Simon Glassb1cca952020-07-09 18:39:40 -0600692 missing_list = []
693 image.CheckMissing(missing_list)
694 if missing_list:
Jonas Karlmanc2600152023-07-18 20:34:37 +0000695 tout.error("Image '%s' is missing external blobs and is non-functional: %s\n" %
Jonas Karlmand92c4dd2023-07-18 20:34:35 +0000696 (image.name, ' '.join([e.name for e in missing_list])))
697 _ShowHelpForMissingBlobs(tout.ERROR, missing_list)
Simon Glass67a05012023-01-07 14:07:15 -0700698
Heiko Thierya89c8f22022-01-06 11:49:41 +0100699 faked_list = []
700 image.CheckFakedBlobs(faked_list)
701 if faked_list:
Simon Glassf3385a52022-01-29 14:14:15 -0700702 tout.warning(
Jonas Karlmanc2600152023-07-18 20:34:37 +0000703 "Image '%s' has faked external blobs and is non-functional: %s\n" %
Simon Glass2cc8c1f2022-01-09 20:13:45 -0700704 (image.name, ' '.join([os.path.basename(e.GetDefaultFilename())
705 for e in faked_list])))
Simon Glass67a05012023-01-07 14:07:15 -0700706
707 optional_list = []
708 image.CheckOptional(optional_list)
709 if optional_list:
710 tout.warning(
Jonas Karlmanc2600152023-07-18 20:34:37 +0000711 "Image '%s' is missing optional external blobs but is still functional: %s\n" %
Simon Glass67a05012023-01-07 14:07:15 -0700712 (image.name, ' '.join([e.name for e in optional_list])))
Jonas Karlmand92c4dd2023-07-18 20:34:35 +0000713 _ShowHelpForMissingBlobs(tout.WARNING, optional_list)
Simon Glass67a05012023-01-07 14:07:15 -0700714
Simon Glass4f9ee832022-01-09 20:14:09 -0700715 missing_bintool_list = []
716 image.check_missing_bintools(missing_bintool_list)
717 if missing_bintool_list:
Simon Glassf3385a52022-01-29 14:14:15 -0700718 tout.warning(
Jonas Karlmanc2600152023-07-18 20:34:37 +0000719 "Image '%s' has missing bintools and is non-functional: %s\n" %
Simon Glass4f9ee832022-01-09 20:14:09 -0700720 (image.name, ' '.join([os.path.basename(bintool.name)
721 for bintool in missing_bintool_list])))
722 return any([missing_list, faked_list, missing_bintool_list])
Simon Glassb88e81c2019-07-20 12:23:24 -0600723
724
Simon Glass53cd5d92019-07-08 14:25:29 -0600725def Binman(args):
Simon Glassbf7fd502016-11-25 20:15:51 -0700726 """The main control code for binman
727
728 This assumes that help and test options have already been dealt with. It
729 deals with the core task of building images.
730
731 Args:
Simon Glass53cd5d92019-07-08 14:25:29 -0600732 args: Command line arguments Namespace object
Simon Glassbf7fd502016-11-25 20:15:51 -0700733 """
Simon Glass8dbb7442019-08-24 07:22:44 -0600734 global Image
735 global state
736
Simon Glass53cd5d92019-07-08 14:25:29 -0600737 if args.full_help:
Simon Glass8de6adb2023-02-23 18:18:20 -0700738 with importlib.resources.path('binman', 'README.rst') as readme:
739 tools.print_full_help(str(readme))
Simon Glassbf7fd502016-11-25 20:15:51 -0700740 return 0
741
Simon Glass8dbb7442019-08-24 07:22:44 -0600742 # Put these here so that we can import this module without libfdt
Simon Glass07237982020-08-05 13:27:47 -0600743 from binman.image import Image
Simon Glass16287932020-04-17 18:09:03 -0600744 from binman import state
Simon Glass8dbb7442019-08-24 07:22:44 -0600745
Simon Glassfe7e9242023-02-22 12:14:49 -0700746 tool_paths = []
747 if args.toolpath:
748 tool_paths += args.toolpath
749 if args.tooldir:
750 tool_paths.append(args.tooldir)
751 tools.set_tool_paths(tool_paths or None)
752 bintool.Bintool.set_tool_dir(args.tooldir)
753
Ivan Mikhaylov4023dc92023-03-08 01:13:39 +0000754 if args.cmd in ['ls', 'extract', 'replace', 'tool', 'sign']:
Simon Glass96b6c502019-07-20 12:23:53 -0600755 try:
Simon Glassf3385a52022-01-29 14:14:15 -0700756 tout.init(args.verbosity)
Simon Glasse00197f2023-03-02 17:02:42 -0700757 if args.cmd == 'replace':
758 tools.prepare_output_dir(args.outdir, args.preserve)
759 else:
760 tools.prepare_output_dir(None)
Simon Glass7bc4f0f2019-09-15 18:10:36 -0600761 if args.cmd == 'ls':
762 ListEntries(args.image, args.paths)
Simon Glass61f564d2019-07-08 14:25:48 -0600763
Simon Glass7bc4f0f2019-09-15 18:10:36 -0600764 if args.cmd == 'extract':
765 ExtractEntries(args.image, args.filename, args.outdir, args.paths,
Simon Glass943bf782021-11-23 21:09:50 -0700766 not args.uncompressed, args.format)
Simon Glass71ce0ba2019-07-08 14:25:52 -0600767
Simon Glass7bc4f0f2019-09-15 18:10:36 -0600768 if args.cmd == 'replace':
769 ReplaceEntries(args.image, args.filename, args.indir, args.paths,
770 do_compress=not args.compressed,
771 allow_resize=not args.fix_size, write_map=args.map)
Simon Glass386c63c2022-01-09 20:13:50 -0700772
Ivan Mikhaylov4023dc92023-03-08 01:13:39 +0000773 if args.cmd == 'sign':
774 SignEntries(args.image, args.file, args.key, args.algo, args.paths)
775
Simon Glass386c63c2022-01-09 20:13:50 -0700776 if args.cmd == 'tool':
Simon Glass386c63c2022-01-09 20:13:50 -0700777 if args.list:
778 bintool.Bintool.list_all()
779 elif args.fetch:
780 if not args.bintools:
781 raise ValueError(
782 "Please specify bintools to fetch or 'all' or 'missing'")
783 bintool.Bintool.fetch_tools(bintool.FETCH_ANY,
784 args.bintools)
785 else:
786 raise ValueError("Invalid arguments to 'tool' subcommand")
Simon Glass7bc4f0f2019-09-15 18:10:36 -0600787 except:
788 raise
Simon Glassa6cb9952019-07-20 12:24:15 -0600789 finally:
Simon Glassc1aa66e2022-01-29 14:14:04 -0700790 tools.finalise_output_dir()
Simon Glassa6cb9952019-07-20 12:24:15 -0600791 return 0
792
Simon Glass0427bed2021-11-03 21:09:18 -0600793 elf_params = None
794 if args.update_fdt_in_elf:
795 elf_params = args.update_fdt_in_elf.split(',')
796 if len(elf_params) != 4:
797 raise ValueError('Invalid args %s to --update-fdt-in-elf: expected infile,outfile,begin_sym,end_sym' %
798 elf_params)
799
Simon Glassbf7fd502016-11-25 20:15:51 -0700800 # Try to figure out which device tree contains our image description
Simon Glass53cd5d92019-07-08 14:25:29 -0600801 if args.dt:
802 dtb_fname = args.dt
Simon Glassbf7fd502016-11-25 20:15:51 -0700803 else:
Simon Glass53cd5d92019-07-08 14:25:29 -0600804 board = args.board
Simon Glassbf7fd502016-11-25 20:15:51 -0700805 if not board:
806 raise ValueError('Must provide a board to process (use -b <board>)')
Simon Glass53cd5d92019-07-08 14:25:29 -0600807 board_pathname = os.path.join(args.build_dir, board)
Simon Glassbf7fd502016-11-25 20:15:51 -0700808 dtb_fname = os.path.join(board_pathname, 'u-boot.dtb')
Simon Glass53cd5d92019-07-08 14:25:29 -0600809 if not args.indir:
810 args.indir = ['.']
811 args.indir.append(board_pathname)
Simon Glassbf7fd502016-11-25 20:15:51 -0700812
813 try:
Simon Glassf3385a52022-01-29 14:14:15 -0700814 tout.init(args.verbosity)
Simon Glass53cd5d92019-07-08 14:25:29 -0600815 elf.debug = args.debug
816 cbfs_util.VERBOSE = args.verbosity > 2
817 state.use_fake_dtb = args.fake_dtb
Simon Glass0b6023e2021-03-18 20:25:06 +1300818
819 # Normally we replace the 'u-boot' etype with 'u-boot-expanded', etc.
820 # When running tests this can be disabled using this flag. When not
821 # updating the FDT in image, it is not needed by binman, but we use it
822 # for consistency, so that the images look the same to U-Boot at
823 # runtime.
824 use_expanded = not args.no_expanded
Simon Glassbf7fd502016-11-25 20:15:51 -0700825 try:
Simon Glassc1aa66e2022-01-29 14:14:04 -0700826 tools.set_input_dirs(args.indir)
827 tools.prepare_output_dir(args.outdir, args.preserve)
Simon Glass53cd5d92019-07-08 14:25:29 -0600828 state.SetEntryArgs(args.entry_arg)
Simon Glassc69d19c2021-07-06 10:36:37 -0600829 state.SetThreads(args.threads)
Simon Glassecab8972018-07-06 10:27:40 -0600830
Simon Glassa8573c42019-07-20 12:23:27 -0600831 images = PrepareImagesAndDtbs(dtb_fname, args.image,
Simon Glass0b6023e2021-03-18 20:25:06 +1300832 args.update_fdt, use_expanded)
Heiko Thierya89c8f22022-01-06 11:49:41 +0100833
Simon Glassc69d19c2021-07-06 10:36:37 -0600834 if args.test_section_timeout:
835 # Set the first image to timeout, used in testThreadTimeout()
836 images[list(images.keys())[0]].test_section_timeout = True
Heiko Thierya89c8f22022-01-06 11:49:41 +0100837 invalid = False
Simon Glass4f9ee832022-01-09 20:14:09 -0700838 bintool.Bintool.set_missing_list(
839 args.force_missing_bintools.split(',') if
840 args.force_missing_bintools else None)
Simon Glass7960a0a2022-08-07 09:46:46 -0600841
842 # Create the directory here instead of Entry.check_fake_fname()
843 # since that is called from a threaded context so different threads
844 # may race to create the directory
845 if args.fake_ext_blobs:
846 entry.Entry.create_fake_dir()
847
Simon Glassbf7fd502016-11-25 20:15:51 -0700848 for image in images.values():
Heiko Thierya89c8f22022-01-06 11:49:41 +0100849 invalid |= ProcessImage(image, args.update_fdt, args.map,
850 allow_missing=args.allow_missing,
851 allow_fake_blobs=args.fake_ext_blobs)
Simon Glass2a72cc72018-09-14 04:57:20 -0600852
853 # Write the updated FDTs to our output files
Simon Glass4bdd1152019-07-20 12:23:29 -0600854 for dtb_item in state.GetAllFdts():
Simon Glassc1aa66e2022-01-29 14:14:04 -0700855 tools.write_file(dtb_item._fname, dtb_item.GetContents())
Simon Glass2a72cc72018-09-14 04:57:20 -0600856
Simon Glass0427bed2021-11-03 21:09:18 -0600857 if elf_params:
858 data = state.GetFdtForEtype('u-boot-dtb').GetContents()
859 elf.UpdateFile(*elf_params, data)
860
Simon Glass823f5c32023-10-14 14:40:25 -0600861 bintool.Bintool.set_missing_list(None)
862
Simon Glassb38da152022-11-09 19:14:42 -0700863 # This can only be True if -M is provided, since otherwise binman
864 # would have raised an error already
Heiko Thierya89c8f22022-01-06 11:49:41 +0100865 if invalid:
Jonas Karlmanc2600152023-07-18 20:34:37 +0000866 msg = 'Some images are invalid'
Simon Glassb38da152022-11-09 19:14:42 -0700867 if args.ignore_missing:
868 tout.warning(msg)
869 else:
870 tout.error(msg)
871 return 103
Simon Glass03ebc202021-07-06 10:36:41 -0600872
873 # Use this to debug the time take to pack the image
874 #state.TimingShow()
Simon Glassbf7fd502016-11-25 20:15:51 -0700875 finally:
Simon Glassc1aa66e2022-01-29 14:14:04 -0700876 tools.finalise_output_dir()
Simon Glassbf7fd502016-11-25 20:15:51 -0700877 finally:
Simon Glassf3385a52022-01-29 14:14:15 -0700878 tout.uninit()
Simon Glassbf7fd502016-11-25 20:15:51 -0700879
880 return 0