From 22e7366dc7d79dc8d501b89d620a2cc89a75d734 Mon Sep 17 00:00:00 2001
From: c7zero <c7zero@gmail.com>
Date: Tue, 23 Aug 2016 22:33:44 -0700
Subject: [PATCH] Should be able to pass argv to main()

import chipsec_main
chipsec_main.main(argv)
---
 source/tool/chipsec_main.py | 6 ++++--
 source/tool/chipsec_util.py | 6 ++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/source/tool/chipsec_main.py b/source/tool/chipsec_main.py
index 0dfb2327..f5f63736 100644
--- a/source/tool/chipsec_main.py
+++ b/source/tool/chipsec_main.py
@@ -656,8 +656,10 @@ class ChipsecMain:
         logger().disable()
         return modules_failed
 
-def main():
-    chipsecMain = ChipsecMain(sys.argv[1:])
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv[1:]
+    chipsecMain = ChipsecMain(argv)
     return chipsecMain.main()
 
 if __name__ == "__main__":
diff --git a/source/tool/chipsec_util.py b/source/tool/chipsec_util.py
index 8976df26..fef4e4df 100644
--- a/source/tool/chipsec_util.py
+++ b/source/tool/chipsec_util.py
@@ -207,10 +207,12 @@ class ChipsecUtil:
                       "################################################################" )
         logger().log( "[CHIPSEC] Version %s" % __version__ )
 
-def main():
+def main(argv=None):
+    if argv is None:
+        argv = sys.argv
     chipsecUtil = ChipsecUtil()
     chipsecUtil.print_banner()
-    return chipsecUtil.main(sys.argv)
+    return chipsecUtil.main(argv)
 
        
 if __name__ == "__main__":