diff --git a/module_generator/casper-fs-gen.py b/module_generator/casper-fs-gen.py index a9f17e9..7033880 100644 --- a/module_generator/casper-fs-gen.py +++ b/module_generator/casper-fs-gen.py @@ -1,24 +1,28 @@ #!/usr/bin/python3 -# Casper-fs v0.1'/' +# Casper-fs v0.2'/' #You nd yaml, pyyaml modules... from util import parser -rules_filename="" - +def main(): + rules_filename="" # Get argvs of user's input -rules_filename = parser.arguments() + rules_filename = parser.arguments() # load rules of firewall at directory rules -try: - rules_casper=parser.Get_config(rules_filename) -except Exception as e: - traceback.print_exc() - print(" log error in config parser rules: "+str(e)) - exit(0) + try: + rules_casper=parser.Get_config(rules_filename) + except Exception as e: + traceback.print_exc() + print(" log error in config parser rules: "+str(e)) + exit(0) +# Load templates and generate LKM + try: + parser.start_generator(rules_casper) + except Exception as e: + print(" log error in rule generator: "+str(e)) + exit(0) + + +if __name__=="__main__": + main() -# Load templates and generate -try: - parser.start_generator(rules_casper) -except Exception as e: - print(" log error in rule generator: "+str(e)) - exit(0) diff --git a/module_generator/util/parser.py b/module_generator/util/parser.py index 6f3a6c8..b1822cd 100644 --- a/module_generator/util/parser.py +++ b/module_generator/util/parser.py @@ -14,7 +14,7 @@ def banner(): print(colored(logo,'white',attrs=['blink']) ) file.close() print(colored("\tCasper-fs is a Custom Hidden Linux Kernel Module generator. Each module works in the file system to protect and hide secret files.",'yellow')) - print(colored("\tVersion 0.1 coded by CoolerVoid - github.com/CoolerVoid/casper-fs",'cyan')) + print(colored("\tVersion 0.2 coded by CoolerVoid - github.com/CoolerVoid/casper-fs",'cyan')) print ("\tExample to use:") print ("\tpython3 Casper-fs.py --rules rules/my_secret_files.yaml\n")