git.sur5r.net Git - u-boot/blob - tools/binman/etype/u_boot_dtb_with_ucode.py

   1 # Copyright (c) 2016 Google, Inc
   2 ## Written by Simon Glass <sjg@chromium.org>
   3
   4 # SPDX-License-Identifier:      GPL-2.0+
   5 #
   6 # Entry-type module for U-Boot device tree with the microcode removed
   7 #
   8
   9 import fdt_select
  10 from entry import Entry
  11 from blob import Entry_blob
  12 import tools
  13
  14 class Entry_u_boot_dtb_with_ucode(Entry_blob):
  15     """A U-Boot device tree file, with the microcode removed
  16
  17     See Entry_u_boot_ucode for full details of the 3 entries involved in this
  18     process.
  19     """
  20     def __init__(self, image, etype, node):
  21         Entry_blob.__init__(self, image, etype, node)
  22         self.ucode_data = ''
  23         self.collate = False
  24         self.ucode_offset = None
  25         self.ucode_size = None
  26
  27     def GetDefaultFilename(self):
  28         return 'u-boot.dtb'
  29
  30     def ObtainContents(self):
  31         Entry_blob.ObtainContents(self)
  32
  33         # If the image does not need microcode, there is nothing to do
  34         ucode_dest_entry = self.image.FindEntryType('u-boot-spl-with-ucode-ptr')
  35         if not ucode_dest_entry or not ucode_dest_entry.target_pos:
  36             ucode_dest_entry = self.image.FindEntryType('u-boot-with-ucode-ptr')
  37         if not ucode_dest_entry or not ucode_dest_entry.target_pos:
  38             return True
  39
  40         # Create a new file to hold the copied device tree
  41         dtb_name = 'u-boot-dtb-with-ucode.dtb'
  42         fname = tools.GetOutputFilename(dtb_name)
  43         with open(fname, 'wb') as fd:
  44             fd.write(self.data)
  45
  46         # Remove the microcode
  47         fdt = fdt_select.FdtScan(fname)
  48         fdt.Scan()
  49         ucode = fdt.GetNode('/microcode')
  50         if not ucode:
  51             raise self.Raise("No /microcode node found in '%s'" % fname)
  52
  53         # There's no need to collate it (move all microcode into one place)
  54         # if we only have one chunk of microcode.
  55         self.collate = len(ucode.subnodes) > 1
  56         for node in ucode.subnodes:
  57             data_prop = node.props.get('data')
  58             if data_prop:
  59                 self.ucode_data += ''.join(data_prop.bytes)
  60                 if not self.collate:
  61                     poffset = data_prop.GetOffset()
  62                     if poffset is None:
  63                         # We cannot obtain a property offset. Collate instead.
  64                         self.collate = True
  65                     else:
  66                         # Find the offset in the device tree of the ucode data
  67                         self.ucode_offset = poffset + 12
  68                         self.ucode_size = len(data_prop.bytes)
  69                 if self.collate:
  70                     prop = node.DeleteProp('data')
  71         if self.collate:
  72             fdt.Pack()
  73             fdt.Flush()
  74
  75             # Make this file the contents of this entry
  76             self._pathname = fname
  77             self.ReadContents()
  78         return True