tools/binman/entry.py

   1 # SPDX-License-Identifier: GPL-2.0+
   2 # Copyright (c) 2016 Google, Inc
   3 #
   4 # Base class for all entries
   5 #
   6
   7 from collections import namedtuple
   8 import importlib
   9 import os
  10 import sys
  11
  12 from dtoc import fdt_util
  13 from patman import tools
  14 from patman.tools import ToHex, ToHexSize
  15 from patman import tout
  16
  17 modules = {}
  18
  19
  20 # An argument which can be passed to entries on the command line, in lieu of
  21 # device-tree properties.
  22 EntryArg = namedtuple('EntryArg', ['name', 'datatype'])
  23
  24 # Information about an entry for use when displaying summaries
  25 EntryInfo = namedtuple('EntryInfo', ['indent', 'name', 'etype', 'size',
  26                                      'image_pos', 'uncomp_size', 'offset',
  27                                      'entry'])
  28
  29 class Entry(object):
  30     """An Entry in the section
  31
  32     An entry corresponds to a single node in the device-tree description
  33     of the section. Each entry ends up being a part of the final section.
  34     Entries can be placed either right next to each other, or with padding
  35     between them. The type of the entry determines the data that is in it.
  36
  37     This class is not used by itself. All entry objects are subclasses of
  38     Entry.
  39
  40     Attributes:
  41         section: Section object containing this entry
  42         node: The node that created this entry
  43         offset: Offset of entry within the section, None if not known yet (in
  44             which case it will be calculated by Pack())
  45         size: Entry size in bytes, None if not known
  46         pre_reset_size: size as it was before ResetForPack(). This allows us to
  47             keep track of the size we started with and detect size changes
  48         uncomp_size: Size of uncompressed data in bytes, if the entry is
  49             compressed, else None
  50         contents_size: Size of contents in bytes, 0 by default
  51         align: Entry start offset alignment relative to the start of the
  52             containing section, or None
  53         align_size: Entry size alignment, or None
  54         align_end: Entry end offset alignment relative to the start of the
  55             containing section, or None
  56         pad_before: Number of pad bytes before the contents when it is placed
  57             in the containing section, 0 if none. The pad bytes become part of
  58             the entry.
  59         pad_after: Number of pad bytes after the contents when it is placed in
  60             the containing section, 0 if none. The pad bytes become part of
  61             the entry.
  62         data: Contents of entry (string of bytes). This does not include
  63             padding created by pad_before or pad_after. If the entry is
  64             compressed, this contains the compressed data.
  65         uncomp_data: Original uncompressed data, if this entry is compressed,
  66             else None
  67         compress: Compression algoithm used (e.g. 'lz4'), 'none' if none
  68         orig_offset: Original offset value read from node
  69         orig_size: Original size value read from node
  70         missing: True if this entry is missing its contents
  71         allow_missing: Allow children of this entry to be missing (used by
  72             subclasses such as Entry_section)
  73         external: True if this entry contains an external binary blob
  74     """
  75     def __init__(self, section, etype, node, name_prefix=''):
  76         # Put this here to allow entry-docs and help to work without libfdt
  77         global state
  78         from binman import state
  79
  80         self.section = section
  81         self.etype = etype
  82         self._node = node
  83         self.name = node and (name_prefix + node.name) or 'none'
  84         self.offset = None
  85         self.size = None
  86         self.pre_reset_size = None
  87         self.uncomp_size = None
  88         self.data = None
  89         self.uncomp_data = None
  90         self.contents_size = 0
  91         self.align = None
  92         self.align_size = None
  93         self.align_end = None
  94         self.pad_before = 0
  95         self.pad_after = 0
  96         self.offset_unset = False
  97         self.image_pos = None
  98         self._expand_size = False
  99         self.compress = 'none'
 100         self.missing = False
 101         self.external = False
 102         self.allow_missing = False
 103
 104     @staticmethod
 105     def Lookup(node_path, etype):
 106         """Look up the entry class for a node.
 107
 108         Args:
 109             node_node: Path name of Node object containing information about
 110                        the entry to create (used for errors)
 111             etype:   Entry type to use
 112
 113         Returns:
 114             The entry class object if found, else None
 115         """
 116         # Convert something like 'u-boot@0' to 'u_boot' since we are only
 117         # interested in the type.
 118         module_name = etype.replace('-', '_')
 119         if '@' in module_name:
 120             module_name = module_name.split('@')[0]
 121         module = modules.get(module_name)
 122
 123         # Also allow entry-type modules to be brought in from the etype directory.
 124
 125         # Import the module if we have not already done so.
 126         if not module:
 127             try:
 128                 module = importlib.import_module('binman.etype.' + module_name)
 129             except ImportError as e:
 130                 raise ValueError("Unknown entry type '%s' in node '%s' (expected etype/%s.py, error '%s'" %
 131                                  (etype, node_path, module_name, e))
 132             modules[module_name] = module
 133
 134         # Look up the expected class name
 135         return getattr(module, 'Entry_%s' % module_name)
 136
 137     @staticmethod
 138     def Create(section, node, etype=None):
 139         """Create a new entry for a node.
 140
 141         Args:
 142             section: Section object containing this node
 143             node:    Node object containing information about the entry to
 144                      create
 145             etype:   Entry type to use, or None to work it out (used for tests)
 146
 147         Returns:
 148             A new Entry object of the correct type (a subclass of Entry)
 149         """
 150         if not etype:
 151             etype = fdt_util.GetString(node, 'type', node.name)
 152         obj = Entry.Lookup(node.path, etype)
 153
 154         # Call its constructor to get the object we want.
 155         return obj(section, etype, node)
 156
 157     def ReadNode(self):
 158         """Read entry information from the node
 159
 160         This must be called as the first thing after the Entry is created.
 161
 162         This reads all the fields we recognise from the node, ready for use.
 163         """
 164         if 'pos' in self._node.props:
 165             self.Raise("Please use 'offset' instead of 'pos'")
 166         self.offset = fdt_util.GetInt(self._node, 'offset')
 167         self.size = fdt_util.GetInt(self._node, 'size')
 168         self.orig_offset = fdt_util.GetInt(self._node, 'orig-offset')
 169         self.orig_size = fdt_util.GetInt(self._node, 'orig-size')
 170         if self.GetImage().copy_to_orig:
 171             self.orig_offset = self.offset
 172             self.orig_size = self.size
 173
 174         # These should not be set in input files, but are set in an FDT map,
 175         # which is also read by this code.
 176         self.image_pos = fdt_util.GetInt(self._node, 'image-pos')
 177         self.uncomp_size = fdt_util.GetInt(self._node, 'uncomp-size')
 178
 179         self.align = fdt_util.GetInt(self._node, 'align')
 180         if tools.NotPowerOfTwo(self.align):
 181             raise ValueError("Node '%s': Alignment %s must be a power of two" %
 182                              (self._node.path, self.align))
 183         self.pad_before = fdt_util.GetInt(self._node, 'pad-before', 0)
 184         self.pad_after = fdt_util.GetInt(self._node, 'pad-after', 0)
 185         self.align_size = fdt_util.GetInt(self._node, 'align-size')
 186         if tools.NotPowerOfTwo(self.align_size):
 187             self.Raise("Alignment size %s must be a power of two" %
 188                        self.align_size)
 189         self.align_end = fdt_util.GetInt(self._node, 'align-end')
 190         self.offset_unset = fdt_util.GetBool(self._node, 'offset-unset')
 191         self.expand_size = fdt_util.GetBool(self._node, 'expand-size')
 192         self.missing_msg = fdt_util.GetString(self._node, 'missing-msg')
 193
 194         # This is only supported by blobs and sections at present
 195         self.compress = fdt_util.GetString(self._node, 'compress', 'none')
 196
 197     def GetDefaultFilename(self):
 198         return None
 199
 200     def GetFdts(self):
 201         """Get the device trees used by this entry
 202
 203         Returns:
 204             Empty dict, if this entry is not a .dtb, otherwise:
 205             Dict:
 206                 key: Filename from this entry (without the path)
 207                 value: Tuple:
 208                     Fdt object for this dtb, or None if not available
 209                     Filename of file containing this dtb
 210         """
 211         return {}
 212
 213     def ExpandEntries(self):
 214         pass
 215
 216     def AddMissingProperties(self, have_image_pos):
 217         """Add new properties to the device tree as needed for this entry
 218
 219         Args:
 220             have_image_pos: True if this entry has an image position. This can
 221                 be False if its parent section is compressed, since compression
 222                 groups all entries together into a compressed block of data,
 223                 obscuring the start of each individual child entry
 224         """
 225         for prop in ['offset', 'size']:
 226             if not prop in self._node.props:
 227                 state.AddZeroProp(self._node, prop)
 228         if have_image_pos and 'image-pos' not in self._node.props:
 229             state.AddZeroProp(self._node, 'image-pos')
 230         if self.GetImage().allow_repack:
 231             if self.orig_offset is not None:
 232                 state.AddZeroProp(self._node, 'orig-offset', True)
 233             if self.orig_size is not None:
 234                 state.AddZeroProp(self._node, 'orig-size', True)
 235
 236         if self.compress != 'none':
 237             state.AddZeroProp(self._node, 'uncomp-size')
 238         err = state.CheckAddHashProp(self._node)
 239         if err:
 240             self.Raise(err)
 241
 242     def SetCalculatedProperties(self):
 243         """Set the value of device-tree properties calculated by binman"""
 244         state.SetInt(self._node, 'offset', self.offset)
 245         state.SetInt(self._node, 'size', self.size)
 246         base = self.section.GetRootSkipAtStart() if self.section else 0
 247         if self.image_pos is not None:
 248             state.SetInt(self._node, 'image-pos', self.image_pos)
 249         if self.GetImage().allow_repack:
 250             if self.orig_offset is not None:
 251                 state.SetInt(self._node, 'orig-offset', self.orig_offset, True)
 252             if self.orig_size is not None:
 253                 state.SetInt(self._node, 'orig-size', self.orig_size, True)
 254         if self.uncomp_size is not None:
 255             state.SetInt(self._node, 'uncomp-size', self.uncomp_size)
 256         state.CheckSetHashValue(self._node, self.GetData)
 257
 258     def ProcessFdt(self, fdt):
 259         """Allow entries to adjust the device tree
 260
 261         Some entries need to adjust the device tree for their purposes. This
 262         may involve adding or deleting properties.
 263
 264         Returns:
 265             True if processing is complete
 266             False if processing could not be completed due to a dependency.
 267                 This will cause the entry to be retried after others have been
 268                 called
 269         """
 270         return True
 271
 272     def SetPrefix(self, prefix):
 273         """Set the name prefix for a node
 274
 275         Args:
 276             prefix: Prefix to set, or '' to not use a prefix
 277         """
 278         if prefix:
 279             self.name = prefix + self.name
 280
 281     def SetContents(self, data):
 282         """Set the contents of an entry
 283
 284         This sets both the data and content_size properties
 285
 286         Args:
 287             data: Data to set to the contents (bytes)
 288         """
 289         self.data = data
 290         self.contents_size = len(self.data)
 291
 292     def ProcessContentsUpdate(self, data):
 293         """Update the contents of an entry, after the size is fixed
 294
 295         This checks that the new data is the same size as the old. If the size
 296         has changed, this triggers a re-run of the packing algorithm.
 297
 298         Args:
 299             data: Data to set to the contents (bytes)
 300
 301         Raises:
 302             ValueError if the new data size is not the same as the old
 303         """
 304         size_ok = True
 305         new_size = len(data)
 306         if state.AllowEntryExpansion() and new_size > self.contents_size:
 307             # self.data will indicate the new size needed
 308             size_ok = False
 309         elif state.AllowEntryContraction() and new_size < self.contents_size:
 310             size_ok = False
 311
 312         # If not allowed to change, try to deal with it or give up
 313         if size_ok:
 314             if new_size > self.contents_size:
 315                 self.Raise('Cannot update entry size from %d to %d' %
 316                         (self.contents_size, new_size))
 317
 318             # Don't let the data shrink. Pad it if necessary
 319             if size_ok and new_size < self.contents_size:
 320                 data += tools.GetBytes(0, self.contents_size - new_size)
 321
 322         if not size_ok:
 323             tout.Debug("Entry '%s' size change from %s to %s" % (
 324                 self._node.path, ToHex(self.contents_size),
 325                 ToHex(new_size)))
 326         self.SetContents(data)
 327         return size_ok
 328
 329     def ObtainContents(self):
 330         """Figure out the contents of an entry.
 331
 332         Returns:
 333             True if the contents were found, False if another call is needed
 334             after the other entries are processed.
 335         """
 336         # No contents by default: subclasses can implement this
 337         return True
 338
 339     def ResetForPack(self):
 340         """Reset offset/size fields so that packing can be done again"""
 341         self.Detail('ResetForPack: offset %s->%s, size %s->%s' %
 342                     (ToHex(self.offset), ToHex(self.orig_offset),
 343                      ToHex(self.size), ToHex(self.orig_size)))
 344         self.pre_reset_size = self.size
 345         self.offset = self.orig_offset
 346         self.size = self.orig_size
 347
 348     def Pack(self, offset):
 349         """Figure out how to pack the entry into the section
 350
 351         Most of the time the entries are not fully specified. There may be
 352         an alignment but no size. In that case we take the size from the
 353         contents of the entry.
 354
 355         If an entry has no hard-coded offset, it will be placed at @offset.
 356
 357         Once this function is complete, both the offset and size of the
 358         entry will be know.
 359
 360         Args:
 361             Current section offset pointer
 362
 363         Returns:
 364             New section offset pointer (after this entry)
 365         """
 366         self.Detail('Packing: offset=%s, size=%s, content_size=%x' %
 367                     (ToHex(self.offset), ToHex(self.size),
 368                      self.contents_size))
 369         if self.offset is None:
 370             if self.offset_unset:
 371                 self.Raise('No offset set with offset-unset: should another '
 372                            'entry provide this correct offset?')
 373             self.offset = tools.Align(offset, self.align)
 374         needed = self.pad_before + self.contents_size + self.pad_after
 375         needed = tools.Align(needed, self.align_size)
 376         size = self.size
 377         if not size:
 378             size = needed
 379         new_offset = self.offset + size
 380         aligned_offset = tools.Align(new_offset, self.align_end)
 381         if aligned_offset != new_offset:
 382             size = aligned_offset - self.offset
 383             new_offset = aligned_offset
 384
 385         if not self.size:
 386             self.size = size
 387
 388         if self.size < needed:
 389             self.Raise("Entry contents size is %#x (%d) but entry size is "
 390                        "%#x (%d)" % (needed, needed, self.size, self.size))
 391         # Check that the alignment is correct. It could be wrong if the
 392         # and offset or size values were provided (i.e. not calculated), but
 393         # conflict with the provided alignment values
 394         if self.size != tools.Align(self.size, self.align_size):
 395             self.Raise("Size %#x (%d) does not match align-size %#x (%d)" %
 396                   (self.size, self.size, self.align_size, self.align_size))
 397         if self.offset != tools.Align(self.offset, self.align):
 398             self.Raise("Offset %#x (%d) does not match align %#x (%d)" %
 399                   (self.offset, self.offset, self.align, self.align))
 400         self.Detail('   - packed: offset=%#x, size=%#x, content_size=%#x, next_offset=%x' %
 401                     (self.offset, self.size, self.contents_size, new_offset))
 402
 403         return new_offset
 404
 405     def Raise(self, msg):
 406         """Convenience function to raise an error referencing a node"""
 407         raise ValueError("Node '%s': %s" % (self._node.path, msg))
 408
 409     def Detail(self, msg):
 410         """Convenience function to log detail referencing a node"""
 411         tag = "Node '%s'" % self._node.path
 412         tout.Detail('%30s: %s' % (tag, msg))
 413
 414     def GetEntryArgsOrProps(self, props, required=False):
 415         """Return the values of a set of properties
 416
 417         Args:
 418             props: List of EntryArg objects
 419
 420         Raises:
 421             ValueError if a property is not found
 422         """
 423         values = []
 424         missing = []
 425         for prop in props:
 426             python_prop = prop.name.replace('-', '_')
 427             if hasattr(self, python_prop):
 428                 value = getattr(self, python_prop)
 429             else:
 430                 value = None
 431             if value is None:
 432                 value = self.GetArg(prop.name, prop.datatype)
 433             if value is None and required:
 434                 missing.append(prop.name)
 435             values.append(value)
 436         if missing:
 437             self.Raise('Missing required properties/entry args: %s' %
 438                        (', '.join(missing)))
 439         return values
 440
 441     def GetPath(self):
 442         """Get the path of a node
 443
 444         Returns:
 445             Full path of the node for this entry
 446         """
 447         return self._node.path
 448
 449     def GetData(self):
 450         """Get the contents of an entry
 451
 452         Returns:
 453             bytes content of the entry, excluding any padding. If the entry is
 454                 compressed, the compressed data is returned
 455         """
 456         self.Detail('GetData: size %s' % ToHexSize(self.data))
 457         return self.data
 458
 459     def GetOffsets(self):
 460         """Get the offsets for siblings
 461
 462         Some entry types can contain information about the position or size of
 463         other entries. An example of this is the Intel Flash Descriptor, which
 464         knows where the Intel Management Engine section should go.
 465
 466         If this entry knows about the position of other entries, it can specify
 467         this by returning values here
 468
 469         Returns:
 470             Dict:
 471                 key: Entry type
 472                 value: List containing position and size of the given entry
 473                     type. Either can be None if not known
 474         """
 475         return {}
 476
 477     def SetOffsetSize(self, offset, size):
 478         """Set the offset and/or size of an entry
 479
 480         Args:
 481             offset: New offset, or None to leave alone
 482             size: New size, or None to leave alone
 483         """
 484         if offset is not None:
 485             self.offset = offset
 486         if size is not None:
 487             self.size = size
 488
 489     def SetImagePos(self, image_pos):
 490         """Set the position in the image
 491
 492         Args:
 493             image_pos: Position of this entry in the image
 494         """
 495         self.image_pos = image_pos + self.offset
 496
 497     def ProcessContents(self):
 498         """Do any post-packing updates of entry contents
 499
 500         This function should call ProcessContentsUpdate() to update the entry
 501         contents, if necessary, returning its return value here.
 502
 503         Args:
 504             data: Data to set to the contents (bytes)
 505
 506         Returns:
 507             True if the new data size is OK, False if expansion is needed
 508
 509         Raises:
 510             ValueError if the new data size is not the same as the old and
 511                 state.AllowEntryExpansion() is False
 512         """
 513         return True
 514
 515     def WriteSymbols(self, section):
 516         """Write symbol values into binary files for access at run time
 517
 518         Args:
 519           section: Section containing the entry
 520         """
 521         pass
 522
 523     def CheckOffset(self):
 524         """Check that the entry offsets are correct
 525
 526         This is used for entries which have extra offset requirements (other
 527         than having to be fully inside their section). Sub-classes can implement
 528         this function and raise if there is a problem.
 529         """
 530         pass
 531
 532     @staticmethod
 533     def GetStr(value):
 534         if value is None:
 535             return '<none>  '
 536         return '%08x' % value
 537
 538     @staticmethod
 539     def WriteMapLine(fd, indent, name, offset, size, image_pos):
 540         print('%s  %s%s  %s  %s' % (Entry.GetStr(image_pos), ' ' * indent,
 541                                     Entry.GetStr(offset), Entry.GetStr(size),
 542                                     name), file=fd)
 543
 544     def WriteMap(self, fd, indent):
 545         """Write a map of the entry to a .map file
 546
 547         Args:
 548             fd: File to write the map to
 549             indent: Curent indent level of map (0=none, 1=one level, etc.)
 550         """
 551         self.WriteMapLine(fd, indent, self.name, self.offset, self.size,
 552                           self.image_pos)
 553
 554     def GetEntries(self):
 555         """Return a list of entries contained by this entry
 556
 557         Returns:
 558             List of entries, or None if none. A normal entry has no entries
 559                 within it so will return None
 560         """
 561         return None
 562
 563     def GetArg(self, name, datatype=str):
 564         """Get the value of an entry argument or device-tree-node property
 565
 566         Some node properties can be provided as arguments to binman. First check
 567         the entry arguments, and fall back to the device tree if not found
 568
 569         Args:
 570             name: Argument name
 571             datatype: Data type (str or int)
 572
 573         Returns:
 574             Value of argument as a string or int, or None if no value
 575
 576         Raises:
 577             ValueError if the argument cannot be converted to in
 578         """
 579         value = state.GetEntryArg(name)
 580         if value is not None:
 581             if datatype == int:
 582                 try:
 583                     value = int(value)
 584                 except ValueError:
 585                     self.Raise("Cannot convert entry arg '%s' (value '%s') to integer" %
 586                                (name, value))
 587             elif datatype == str:
 588                 pass
 589             else:
 590                 raise ValueError("GetArg() internal error: Unknown data type '%s'" %
 591                                  datatype)
 592         else:
 593             value = fdt_util.GetDatatype(self._node, name, datatype)
 594         return value
 595
 596     @staticmethod
 597     def WriteDocs(modules, test_missing=None):
 598         """Write out documentation about the various entry types to stdout
 599
 600         Args:
 601             modules: List of modules to include
 602             test_missing: Used for testing. This is a module to report
 603                 as missing
 604         """
 605         print('''Binman Entry Documentation
 606 ===========================
 607
 608 This file describes the entry types supported by binman. These entry types can
 609 be placed in an image one by one to build up a final firmware image. It is
 610 fairly easy to create new entry types. Just add a new file to the 'etype'
 611 directory. You can use the existing entries as examples.
 612
 613 Note that some entries are subclasses of others, using and extending their
 614 features to produce new behaviours.
 615
 616
 617 ''')
 618         modules = sorted(modules)
 619
 620         # Don't show the test entry
 621         if '_testing' in modules:
 622             modules.remove('_testing')
 623         missing = []
 624         for name in modules:
 625             module = Entry.Lookup('WriteDocs', name)
 626             docs = getattr(module, '__doc__')
 627             if test_missing == name:
 628                 docs = None
 629             if docs:
 630                 lines = docs.splitlines()
 631                 first_line = lines[0]
 632                 rest = [line[4:] for line in lines[1:]]
 633                 hdr = 'Entry: %s: %s' % (name.replace('_', '-'), first_line)
 634                 print(hdr)
 635                 print('-' * len(hdr))
 636                 print('\n'.join(rest))
 637                 print()
 638                 print()
 639             else:
 640                 missing.append(name)
 641
 642         if missing:
 643             raise ValueError('Documentation is missing for modules: %s' %
 644                              ', '.join(missing))
 645
 646     def GetUniqueName(self):
 647         """Get a unique name for a node
 648
 649         Returns:
 650             String containing a unique name for a node, consisting of the name
 651             of all ancestors (starting from within the 'binman' node) separated
 652             by a dot ('.'). This can be useful for generating unique filesnames
 653             in the output directory.
 654         """
 655         name = self.name
 656         node = self._node
 657         while node.parent:
 658             node = node.parent
 659             if node.name == 'binman':
 660                 break
 661             name = '%s.%s' % (node.name, name)
 662         return name
 663
 664     def ExpandToLimit(self, limit):
 665         """Expand an entry so that it ends at the given offset limit"""
 666         if self.offset + self.size < limit:
 667             self.size = limit - self.offset
 668             # Request the contents again, since changing the size requires that
 669             # the data grows. This should not fail, but check it to be sure.
 670             if not self.ObtainContents():
 671                 self.Raise('Cannot obtain contents when expanding entry')
 672
 673     def HasSibling(self, name):
 674         """Check if there is a sibling of a given name
 675
 676         Returns:
 677             True if there is an entry with this name in the the same section,
 678                 else False
 679         """
 680         return name in self.section.GetEntries()
 681
 682     def GetSiblingImagePos(self, name):
 683         """Return the image position of the given sibling
 684
 685         Returns:
 686             Image position of sibling, or None if the sibling has no position,
 687                 or False if there is no such sibling
 688         """
 689         if not self.HasSibling(name):
 690             return False
 691         return self.section.GetEntries()[name].image_pos
 692
 693     @staticmethod
 694     def AddEntryInfo(entries, indent, name, etype, size, image_pos,
 695                      uncomp_size, offset, entry):
 696         """Add a new entry to the entries list
 697
 698         Args:
 699             entries: List (of EntryInfo objects) to add to
 700             indent: Current indent level to add to list
 701             name: Entry name (string)
 702             etype: Entry type (string)
 703             size: Entry size in bytes (int)
 704             image_pos: Position within image in bytes (int)
 705             uncomp_size: Uncompressed size if the entry uses compression, else
 706                 None
 707             offset: Entry offset within parent in bytes (int)
 708             entry: Entry object
 709         """
 710         entries.append(EntryInfo(indent, name, etype, size, image_pos,
 711                                  uncomp_size, offset, entry))
 712
 713     def ListEntries(self, entries, indent):
 714         """Add files in this entry to the list of entries
 715
 716         This can be overridden by subclasses which need different behaviour.
 717
 718         Args:
 719             entries: List (of EntryInfo objects) to add to
 720             indent: Current indent level to add to list
 721         """
 722         self.AddEntryInfo(entries, indent, self.name, self.etype, self.size,
 723                           self.image_pos, self.uncomp_size, self.offset, self)
 724
 725     def ReadData(self, decomp=True):
 726         """Read the data for an entry from the image
 727
 728         This is used when the image has been read in and we want to extract the
 729         data for a particular entry from that image.
 730
 731         Args:
 732             decomp: True to decompress any compressed data before returning it;
 733                 False to return the raw, uncompressed data
 734
 735         Returns:
 736             Entry data (bytes)
 737         """
 738         # Use True here so that we get an uncompressed section to work from,
 739         # although compressed sections are currently not supported
 740         tout.Debug("ReadChildData section '%s', entry '%s'" %
 741                    (self.section.GetPath(), self.GetPath()))
 742         data = self.section.ReadChildData(self, decomp)
 743         return data
 744
 745     def ReadChildData(self, child, decomp=True):
 746         """Read the data for a particular child entry
 747
 748         This reads data from the parent and extracts the piece that relates to
 749         the given child.
 750
 751         Args:
 752             child: Child entry to read data for (must be valid)
 753             decomp: True to decompress any compressed data before returning it;
 754                 False to return the raw, uncompressed data
 755
 756         Returns:
 757             Data for the child (bytes)
 758         """
 759         pass
 760
 761     def LoadData(self, decomp=True):
 762         data = self.ReadData(decomp)
 763         self.contents_size = len(data)
 764         self.ProcessContentsUpdate(data)
 765         self.Detail('Loaded data size %x' % len(data))
 766
 767     def GetImage(self):
 768         """Get the image containing this entry
 769
 770         Returns:
 771             Image object containing this entry
 772         """
 773         return self.section.GetImage()
 774
 775     def WriteData(self, data, decomp=True):
 776         """Write the data to an entry in the image
 777
 778         This is used when the image has been read in and we want to replace the
 779         data for a particular entry in that image.
 780
 781         The image must be re-packed and written out afterwards.
 782
 783         Args:
 784             data: Data to replace it with
 785             decomp: True to compress the data if needed, False if data is
 786                 already compressed so should be used as is
 787
 788         Returns:
 789             True if the data did not result in a resize of this entry, False if
 790                  the entry must be resized
 791         """
 792         if self.size is not None:
 793             self.contents_size = self.size
 794         else:
 795             self.contents_size = self.pre_reset_size
 796         ok = self.ProcessContentsUpdate(data)
 797         self.Detail('WriteData: size=%x, ok=%s' % (len(data), ok))
 798         section_ok = self.section.WriteChildData(self)
 799         return ok and section_ok
 800
 801     def WriteChildData(self, child):
 802         """Handle writing the data in a child entry
 803
 804         This should be called on the child's parent section after the child's
 805         data has been updated. It
 806
 807         This base-class implementation does nothing, since the base Entry object
 808         does not have any children.
 809
 810         Args:
 811             child: Child Entry that was written
 812
 813         Returns:
 814             True if the section could be updated successfully, False if the
 815                 data is such that the section could not updat
 816         """
 817         return True
 818
 819     def GetSiblingOrder(self):
 820         """Get the relative order of an entry amoung its siblings
 821
 822         Returns:
 823             'start' if this entry is first among siblings, 'end' if last,
 824                 otherwise None
 825         """
 826         entries = list(self.section.GetEntries().values())
 827         if entries:
 828             if self == entries[0]:
 829                 return 'start'
 830             elif self == entries[-1]:
 831                 return 'end'
 832         return 'middle'
 833
 834     def SetAllowMissing(self, allow_missing):
 835         """Set whether a section allows missing external blobs
 836
 837         Args:
 838             allow_missing: True if allowed, False if not allowed
 839         """
 840         # This is meaningless for anything other than sections
 841         pass
 842
 843     def CheckMissing(self, missing_list):
 844         """Check if any entries in this section have missing external blobs
 845
 846         If there are missing blobs, the entries are added to the list
 847
 848         Args:
 849             missing_list: List of Entry objects to be added to
 850         """
 851         if self.missing:
 852             missing_list.append(self)
 853
 854     def GetAllowMissing(self):
 855         """Get whether a section allows missing external blobs
 856
 857         Returns:
 858             True if allowed, False if not allowed
 859         """
 860         return self.allow_missing
 861
 862     def GetHelpTags(self):
 863         """Get the tags use for missing-blob help
 864
 865         Returns:
 866             list of possible tags, most desirable first
 867         """
 868         return list(filter(None, [self.missing_msg, self.name, self.etype]))
 869
 870     def CompressData(self, indata):
 871         """Compress data according to the entry's compression method
 872
 873         Args:
 874             indata: Data to compress
 875
 876         Returns:
 877             Compressed data (first word is the compressed size)
 878         """
 879         self.uncomp_data = indata
 880         if self.compress != 'none':
 881             self.uncomp_size = len(indata)
 882         data = tools.Compress(indata, self.compress)
 883         return data