Added a command line for the 'red' module.

This commit is contained in:
Cédric Bonhomme 2017-02-06 09:31:26 +01:00
parent 4a45689298
commit 086c3898b7
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
5 changed files with 94 additions and 2 deletions

View file

@ -72,6 +72,12 @@ parser_reveal.add_argument("-o", dest="secret_binary",
arguments = parser.parse_args()
try:
arguments.generator_function
except AttributeError:
print('You must specify the name of a generator.')
parser.print_help()
exit(1)
try:
generator = getattr(generators, arguments.generator_function)()