insert main() function

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)

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")