Procházet zdrojové kódy

ldgen: Fix crash if --sections argument not supplied

Angus Gratton před 7 roky
rodič
revize
9f8587360c
1 změnil soubory, kde provedl 6 přidání a 3 odebrání
  1. 6 3
      tools/ldgen/ldgen.py

+ 6 - 3
tools/ldgen/ldgen.py

@@ -73,14 +73,17 @@ def main():
     fragment_files = [] if not args.fragments else args.fragments
     config_file = args.config
     output_path = args.output
-    sections_info_files = [] if not args.sections else args.sections
     kconfig_file = args.kconfig
+    sections = args.sections
 
     try:
         sections_infos = SectionsInfo()
 
-        section_info_contents = [s.strip() for s in sections_info_files.read().split("\n")]
-        section_info_contents = [s for s in section_info_contents if s]
+        if sections:
+            section_info_contents = [s.strip() for s in sections.read().split("\n")]
+            section_info_contents = [s for s in section_info_contents if s]
+        else:
+            section_info_contents = []
 
         for sections_info_file in section_info_contents:
             with open(sections_info_file) as sections_info_file_obj: