Paths updated

Author: ramirak

lss.py

@@ -1,6 +1,7 @@
 
 from main_frame import create_main_window
-from json_handler import save_to_file 
+from json_handler import save_to_file
+from utils import *
 import os
 
 
@@ -15,16 +16,15 @@ def print_logo():
 
 
 def check_data_folder():
-    data_folder = "data"
+    data_folder = get_data_dir()
     if not os.path.exists(data_folder):
         os.mkdir(data_folder)
-    return data_folder
 
 
 def create_default_policy():
-    data_folder = check_data_folder()
+    check_data_folder()
 
-    policies_file = data_folder + "/policies.json"
+    policies_file = get_data_dir() + "/policies.json"
     if not os.path.isfile(policies_file):
         policy = {"Accept-all": [{"dir": "OUTPUT", "src": "any", "dst": "any", "sport":"any", "dport": "any", "proto": "any", "action": "ACCEPT"}, 
                                  {"dir": "INPUT", "src": "any", "dst": "any", "sport":"any", "dport": "any", "proto": "any", "action": "ACCEPT"}]}
@@ -34,7 +34,7 @@ def create_default_policy():
 def create_default_config():
     data_folder = check_data_folder()
 
-    conf_file = "data/config.json"
+    conf_file = get_data_dir() + "/config.json"
     if not os.path.isfile(conf_file):
         config = {"active_policy":"Accept-all"}
         save_to_file(conf_file, config)

main_frame.py

@@ -4,14 +4,15 @@
 from json_handler import *
 from ui_tools import *
 from policy_editor import *
+from utils import *
 from policy import apply_current_policy, apply_blocklist
 from iptable_tools import clear_chains, set_rule_return, save_all
 
 title = "Linux Security Suite"
 
 def set_main_buttons(root, text, LEFT_FRAME, RIGHT_FRAME):
     sec_status = '''echo -e "~~~ My Configurations: ~~~\n" ; 
-                cat data/config.json ; echo -e "\n" ; 
+                cat ''' + get_data_dir() + '''/config.json ; echo -e "\n" ; 
                 systemctl status iptables.service ; 
                 echo -e "\n" ;systemctl status apparmor ; 
                 echo -e "\n~~~ Listening ports ~~~ \n"; 

policy.py

@@ -4,6 +4,7 @@
 from iptable_tools import *
 from json_handler import *
 from ip_checks import *
+from utils import *
 
 def load_policy(policy):
     print("Creating blocklist chain if it does not exists already..")
@@ -47,8 +48,8 @@ def load_policy(policy):
 
 
 def apply_current_policy():
-    my_policy_name = retrieve_from_file("data/config.json")["active_policy"]
-    policy = retrieve_from_file("data/policies.json")[my_policy_name]
+    my_policy_name = retrieve_from_file(get_data_dir() + "/config.json")["active_policy"]
+    policy = retrieve_from_file(get_data_dir() + "/policies.json")[my_policy_name]
     load_policy(policy)
 
 
@@ -58,7 +59,7 @@ def apply_blocklist():
     print("Clearing previous rules if any..")
     clear_chains("BLOCKLIST")
     ip_list = []
-    with open("data/addresses.list") as file:
+    with open(get_data_dir() + "/addresses.list") as file:
         for line in file:
             ip_list.append(line.rstrip())
     for addr in ip_list:

policy_editor.py

@@ -1,10 +1,10 @@
 from ui_tools import *
-
+from utils import *
 
 def create_editor_buttons(root, frame1, frame2, treeview):
     # Get data from jsons
-    all_policies = retrieve_from_file("data/policies.json")
-    my_config = retrieve_from_file("data/config.json")
+    all_policies = retrieve_from_file(get_data_dir() + "/policies.json")
+    my_config = retrieve_from_file(get_data_dir() + "/config.json")
 
     # Change policy menu
     drop_down, menu_var = create_dropdown(frame1, list(all_policies))
@@ -27,14 +27,14 @@ def save_polciy():
         policy = []
         for child in treeview.get_children():
             policy.append(list_to_json(policy_keys,treeview.item(child)["values"]))
-        replace_val_from_key(menu_var.get(), policy, "data/policies.json")
+        replace_val_from_key(menu_var.get(), policy, get_data_dir() + "/policies.json")
         nonlocal all_policies
-        all_policies = retrieve_from_file("data/policies.json")
+        all_policies = retrieve_from_file(get_data_dir() + "/policies.json")
 
     def mark_active():
         nonlocal my_config
-        replace_val_from_key('active_policy', menu_var.get(), "data/config.json")
-        my_config = retrieve_from_file("data/config.json")
+        replace_val_from_key('active_policy', menu_var.get(), get_data_dir() + "/config.json")
+        my_config = retrieve_from_file(get_data_dir() + "/config.json")
 
     def update_table(tree):
         clear_all_tree_vals(tree)

utils.py

@@ -0,0 +1,5 @@
+import os
+
+def get_data_dir():
+    return os.path.expanduser('~') + "/.lss"
+