Should be able to pass argv to main()

import chipsec_main
chipsec_main.main(argv)
This commit is contained in:
c7zero
2016-08-23 22:33:44 -07:00
parent 0a6c933d13
commit 22e7366dc7
2 changed files with 8 additions and 4 deletions
+4 -2
View File
@@ -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__":
+4 -2
View File
@@ -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__":