binman: bintool: remove btool_ prefix from btool names

The binary is looked on the system by the suffix of the packer class.
This means binman was looking for btool_gzip on the system and not gzip.

Since a btool can have its btool_ prefix missing but its module and
binary presence on the system appropriately found, there's no need to
actually keep this prefix after listing all possible btools, so let's
remove it.

This fixes gzip btool by letting Bintool.find_bintool_class handle the
missing prefix and still return the correct class which is then init
with gzip name instead of btool_gzip.

Additionally, there was an issue with the cached module global variable.
The variable only stores the module and not the associated class name
when calling find_bintool_class.
This means that when caching the module on the first call to
find_bintool_class, class_name would be set to Bintoolbtool_gzip but the
module_name gzip only, adding the module in the gzip key in the module
dictionary. When hitting the cache on next calls, the gzip key would be
found, so its value (the module) is used. However the default class_name
(Bintoolgzip) is used, failing the getattr call.

Instead, let's enforce the same class name: Bintool<packer>, whatever
the filename it is contained in.

Cc: Quentin Schulz <foss+uboot@0leil.net>
Signed-off-by: Quentin Schulz <quentin.schulz@theobroma-systems.com>
Reviewed-by: Simon Glass <sjg@chromium.org>
This commit is contained in:
Quentin Schulz 2022-11-07 13:54:54 +01:00 committed by Simon Glass
parent 521277ec15
commit 478332a345
2 changed files with 3 additions and 2 deletions

View File

@ -85,7 +85,6 @@ class Bintool:
try: try:
# Deal with classes which must be renamed due to conflicts # Deal with classes which must be renamed due to conflicts
# with Python libraries # with Python libraries
class_name = f'Bintoolbtool_{module_name}'
module = importlib.import_module('binman.btool.btool_' + module = importlib.import_module('binman.btool.btool_' +
module_name) module_name)
except ImportError: except ImportError:
@ -137,6 +136,8 @@ class Bintool:
names = [os.path.splitext(os.path.basename(fname))[0] names = [os.path.splitext(os.path.basename(fname))[0]
for fname in files] for fname in files]
names = [name for name in names if name[0] != '_'] names = [name for name in names if name[0] != '_']
names = [name[6:] if name.startswith('btool_') else name
for name in names]
if include_testing: if include_testing:
names.append('_testing') names.append('_testing')
return sorted(names) return sorted(names)

View File

@ -14,7 +14,7 @@ Documentation is available via::
from binman import bintool from binman import bintool
# pylint: disable=C0103 # pylint: disable=C0103
class Bintoolbtool_gzip(bintool.BintoolPacker): class Bintoolgzip(bintool.BintoolPacker):
"""Compression/decompression using the gzip algorithm """Compression/decompression using the gzip algorithm
This bintool supports running `gzip` to compress and decompress data, as This bintool supports running `gzip` to compress and decompress data, as