123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124 |
- # SPDX-License-Identifier: GPL-2.0+
- # Copyright (c) 2016 Google, Inc
- # Written by Simon Glass <sjg@chromium.org>
- #
- # Entry-type module for testing purposes. Not used in real images.
- #
- from collections import OrderedDict
- from binman.entry import Entry, EntryArg
- from dtoc import fdt_util
- from patman import tools
- class Entry__testing(Entry):
- """A fake entry used for testing
- This entry should not be used in normal images. It is a special entry with
- strange features used for testing.
- Properties / Entry arguments
- test-str-fdt: Test string, normally in the node
- test-int-fdt: Test integer, normally in the node
- test-str-arg: Test string, normally in the entry arguments
- test-int-arg: Test integer, normally in the entry arguments
- The entry has a single 'a' byte as its contents. Operation is controlled by
- a number of properties in the node, as follows:
- Properties:
- return-invalid-entry: Return an invalid entry from GetOffsets()
- return-unknown-contents: Refuse to provide any contents (to cause a
- failure)
- bad-update-contents: Return a larger size in ProcessContents
- bad-shrink-contents: Return a larger size in ProcessContents
- never-complete-process-fdt: Refund to process the FDT (to cause a
- failure)
- require-args: Require that all used args are present (generating an
- error if not)
- force-bad-datatype: Force a call to GetEntryArgsOrProps() with a bad
- data type (generating an error)
- """
- def __init__(self, section, etype, node):
- Entry.__init__(self, section, etype, node)
- def ReadNode(self):
- Entry.ReadNode(self)
- self.return_invalid_entry = fdt_util.GetBool(self._node,
- 'return-invalid-entry')
- self.return_unknown_contents = fdt_util.GetBool(self._node,
- 'return-unknown-contents')
- self.bad_update_contents = fdt_util.GetBool(self._node,
- 'bad-update-contents')
- self.bad_shrink_contents = fdt_util.GetBool(self._node,
- 'bad-shrink-contents')
- self.return_contents_once = fdt_util.GetBool(self._node,
- 'return-contents-once')
- self.bad_update_contents_twice = fdt_util.GetBool(self._node,
- 'bad-update-contents-twice')
- # Set to True when the entry is ready to process the FDT.
- self.process_fdt_ready = False
- self.never_complete_process_fdt = fdt_util.GetBool(self._node,
- 'never-complete-process-fdt')
- self.require_args = fdt_util.GetBool(self._node, 'require-args')
- # This should be picked up by GetEntryArgsOrProps()
- self.test_existing_prop = 'existing'
- self.force_bad_datatype = fdt_util.GetBool(self._node,
- 'force-bad-datatype')
- (self.test_str_fdt, self.test_str_arg, self.test_int_fdt,
- self.test_int_arg, existing) = self.GetEntryArgsOrProps([
- EntryArg('test-str-fdt', str),
- EntryArg('test-str-arg', str),
- EntryArg('test-int-fdt', int),
- EntryArg('test-int-arg', int),
- EntryArg('test-existing-prop', str)], self.require_args)
- if self.force_bad_datatype:
- self.GetEntryArgsOrProps([EntryArg('test-bad-datatype-arg', bool)])
- self.return_contents = True
- self.contents = b'aa'
- def ObtainContents(self):
- if self.return_unknown_contents or not self.return_contents:
- return False
- self.data = self.contents
- self.contents_size = len(self.data)
- if self.return_contents_once:
- self.return_contents = False
- return True
- def GetOffsets(self):
- if self.return_invalid_entry :
- return {'invalid-entry': [1, 2]}
- return {}
- def ProcessContents(self):
- data = self.contents
- if self.bad_update_contents:
- # Request to update the contents with something larger, to cause a
- # failure.
- if self.bad_update_contents_twice:
- data = self.data + b'a'
- else:
- data = b'aaa'
- return self.ProcessContentsUpdate(data)
- if self.bad_shrink_contents:
- # Request to update the contents with something smaller, to cause a
- # failure.
- data = b'a'
- return self.ProcessContentsUpdate(data)
- if self.bad_shrink_contents:
- # Request to update the contents with something smaller, to cause a
- # failure.
- data = b'a'
- return self.ProcessContentsUpdate(data)
- return True
- def ProcessFdt(self, fdt):
- """Force reprocessing the first time"""
- ready = self.process_fdt_ready
- if not self.never_complete_process_fdt:
- self.process_fdt_ready = True
- return ready
|