binman: Allow unit addresses for binaries

[oweals/u-boot.git] / tools / binman / etype / entry.py

diff --git a/tools/binman/etype/entry.py b/tools/binman/etype/entry.py
index c331312c491ab34c83dbb159e00be0ded4ffd2dd..23e436a2e9ddc057ecf26ef72f6cd73987bd9ee3 100644 (file)
--- a/tools/binman/etype/entry.py
+++ b/tools/binman/etype/entry.py
@@ -72,7 +72,12 @@ class Entry(object):
         """
         if not etype:
             etype = fdt_util.GetString(node, 'type', node.name)
+
+        # Convert something like 'u-boot@0' to 'u_boot' since we are only
+        # interested in the type.
         module_name = etype.replace('-', '_')
+        if '@' in module_name:
+            module_name = module_name.split('@')[0]
         module = modules.get(module_name)
 
         # Import the module if we have not already done so.