From 39a932090b804a70bc707dad7f4c939eb43f854a Mon Sep 17 00:00:00 2001
From: Gregor Haas <gregorhaas1997@gmail.com>
Date: Mon, 13 Jan 2020 21:48:06 -0500
Subject: [PATCH] Make headerize and librarize scripts more reliable with
 arguments

---
 c8_libpayload/scripts/headerize.py | 14 +++++++-------
 c8_libpayload/scripts/librarize.py | 14 +++++++-------
 2 files changed, 14 insertions(+), 14 deletions(-)

diff --git a/c8_libpayload/scripts/headerize.py b/c8_libpayload/scripts/headerize.py
index fa9ec78..507263f 100644
--- a/c8_libpayload/scripts/headerize.py
+++ b/c8_libpayload/scripts/headerize.py
@@ -9,13 +9,13 @@ if __name__ == '__main__':
     lib_names = []
     hdr_dir = os.path.abspath(sys.argv[-1])
 
-    if os.path.isdir(sys.argv[1]):
-        lib_folder = os.path.abspath(sys.argv[1])
-        for lib_fname in os.listdir(lib_folder):
-            lib_names.append(lib_folder + '/' + lib_fname)
-    else:
-        for n in sys.argv[1:-1]:
-            lib_names.append(os.path.abspath(n))
+    for libarg in sys.argv[1:-1]:
+        if os.path.isdir(libarg):
+            lib_folder = os.path.abspath(libarg)
+            for lib_fname in os.listdir(lib_folder):
+                lib_names.append(lib_folder + '/' + lib_fname)
+        else:
+            lib_names.append(os.path.abspath(libarg))
 
     header_lines = ['#ifndef CHECKM8_TOOL_LIBPAYLOAD_H\n',
                     '#define CHECKM8_TOOL_LIBPAYLOAD_H\n',
diff --git a/c8_libpayload/scripts/librarize.py b/c8_libpayload/scripts/librarize.py
index 78d62b4..25e987f 100644
--- a/c8_libpayload/scripts/librarize.py
+++ b/c8_libpayload/scripts/librarize.py
@@ -12,13 +12,13 @@ if __name__ == '__main__':
     bin_names = []
     lib_dir = os.path.abspath(sys.argv[-1])
 
-    if os.path.isdir(sys.argv[1]):
-        bin_folder = os.path.abspath(sys.argv[1])
-        for bin_fname in os.listdir(bin_folder):
-            bin_names.append(bin_folder + '/' + bin_fname)
-    else:
-        for n in sys.argv[1:-1]:
-            bin_names.append(os.path.abspath(n))
+    for binarg in sys.argv[1:-1]:
+        if os.path.isdir(binarg):
+            bin_folder = os.path.abspath(binarg)
+            for bin_fname in os.listdir(bin_folder):
+                bin_names.append(bin_folder + '/' + bin_fname)
+        else:
+            bin_names.append(os.path.abspath(binarg))
 
     source_lines = defaultdict(list)
     for n in bin_names: