1 # SPDX-License-Identifier: GPL-2.0+
2 # Copyright (c) 2016 Google, Inc
3 # Written by Simon Glass <sjg@chromium.org>
5 # Class for an image, the output of binman
8 from __future__ import print_function
10 from collections import OrderedDict
11 from operator import attrgetter
20 """A Image, representing an output from binman
22 An image is comprised of a collection of entries each containing binary
23 data. The image size must be large enough to hold all of this data.
25 This class implements the various operations needed for images.
28 _node: Node object that contains the image definition in device tree
30 _size: Image size in bytes, or None if not known yet
31 _filename: Output filename for image
32 _sections: Sections present in this image (may be one or more)
35 test: True if this is being called from a test of Images. This this case
36 there is no device tree defining the structure of the section, so
37 we create a section manually.
39 def __init__(self, name, node, test=False):
43 self._filename = '%s.bin' % self._name
45 self._section = bsection.Section('main-section', self._node, True)
50 """Read properties from the image node"""
51 self._size = fdt_util.GetInt(self._node, 'size')
52 filename = fdt_util.GetString(self._node, 'filename')
54 self._filename = filename
55 self._section = bsection.Section('main-section', self._node)
58 """Get the set of device tree files used by this image"""
59 return self._section.GetFdtSet()
61 def AddMissingProperties(self):
62 """Add properties that are not present in the device tree
64 When binman has completed packing the entries the offset and size of
65 each entry are known. But before this the device tree may not specify
66 these. Add any missing properties, with a dummy value, so that the
67 size of the entry is correct. That way we can insert the correct values
70 self._section.AddMissingProperties()
72 def ProcessFdt(self, fdt):
73 """Allow entries to adjust the device tree
75 Some entries need to adjust the device tree for their purposes. This
76 may involve adding or deleting properties.
78 return self._section.ProcessFdt(fdt)
80 def GetEntryContents(self):
81 """Call ObtainContents() for the section
83 self._section.GetEntryContents()
85 def GetEntryOffsets(self):
86 """Handle entries that want to set the offset/size of other entries
88 This calls each entry's GetOffsets() method. If it returns a list
89 of entries to update, it updates them.
91 self._section.GetEntryOffsets()
93 def PackEntries(self):
94 """Pack all entries into the image"""
95 self._section.PackEntries()
98 """Check that the image contents does not exceed its size, etc."""
99 self._size = self._section.CheckSize()
101 def CheckEntries(self):
102 """Check that entries do not overlap or extend outside the image"""
103 self._section.CheckEntries()
105 def SetCalculatedProperties(self):
106 self._section.SetCalculatedProperties()
108 def SetImagePos(self):
109 self._section.SetImagePos(0)
111 def ProcessEntryContents(self):
112 """Call the ProcessContents() method for each entry
114 This is intended to adjust the contents as needed by the entry type.
116 self._section.ProcessEntryContents()
118 def WriteSymbols(self):
119 """Write symbol values into binary files for access at run time"""
120 self._section.WriteSymbols()
122 def BuildImage(self):
123 """Write the image to a file"""
124 fname = tools.GetOutputFilename(self._filename)
125 with open(fname, 'wb') as fd:
126 self._section.BuildSection(fd, 0)
128 def GetEntries(self):
129 return self._section.GetEntries()
132 """Write a map of the image to a .map file"""
133 filename = '%s.map' % self._name
134 fname = tools.GetOutputFilename(filename)
135 with open(fname, 'w') as fd:
136 print('%8s %8s %8s %s' % ('ImagePos', 'Offset', 'Size', 'Name'),
138 self._section.WriteMap(fd, 0)