Added some security restrictions to the server.

Author: Eta06

main.py

@@ -1,6 +1,7 @@
 # Description: Main file for the LockDown project
 import os
 from flask import Flask, render_template, request, redirect, url_for
+from werkzeug.exceptions import Forbidden
 from modules.config import readAppConfig, updateAppConfig, loadLanguageFiles, tryLoad
 import sys
 
@@ -12,6 +13,14 @@
 author = tryLoad("app_author")
 port = tryLoad("app_port")
 
+restricted_routes = ["/set_language", "/reset_everything", "/set_port", "/"]  # Add the routes you want to restrict
+
+
+@app.before_request
+def restrict_access():
+    if request.path in restricted_routes and request.remote_addr != "127.0.0.1":
+        raise Forbidden("Access to this route is limited to localhost.")
+
 
 def restart_server():
     """Restarts the current Python script."""

screenloc.py

@@ -1,5 +1,4 @@
 from PyQt5 import uic, QtCore, QtWidgets
-import cryptography.fernet
 import requests
 import platform
 import socket
@@ -25,6 +24,7 @@ def save_new_client():
     # Generate a key for the client
 
 
+
 def report_client_to_server(client_token):
     # Get the IP address, Mac address, Username, CPU Info, Memory Info, Disk Info, OS Info
     try: