diff --git a/dislocker_client.py b/dislocker_client.py index 5b5b589..f2933f9 100644 --- a/dislocker_client.py +++ b/dislocker_client.py @@ -39,12 +39,16 @@ elif os.path.isfile(client_config_path): client_config = json.load(r) def init(**kwargs): - #sp_startupinfo = subprocess.STARTUPINFO() - #sp_startupinfo.dwFlags = subprocess.STARTF_USESHOWWINDOW - #sp_startupinfo.wShowWindow = subprocess.SW_HIDE - #task_exist = subprocess.run('tasklist /fi "IMAGENAME eq dislocker_client.exe"', startupinfo=sp_startupinfo, stdout=subprocess.PIPE, text=True) - #if 'dislocker_client.exe' in task_exist.stdout: - #return 1 + sp_startupinfo = subprocess.STARTUPINFO() + sp_startupinfo.dwFlags = subprocess.STARTF_USESHOWWINDOW + sp_startupinfo.wShowWindow = subprocess.SW_HIDE + task_exist = subprocess.run('tasklist /fi "IMAGENAME eq dislocker_client.exe"', startupinfo=sp_startupinfo, stdout=subprocess.PIPE, text=True) + if 'dislocker_client.exe' in task_exist.stdout: + task_count = task_exist.stdout.count("dislocker_client.exe") + if task_count == 1: + pass + else: + return 1 if client_config["initial"] == 1: master_password = master_password_gen() @@ -60,7 +64,6 @@ def init(**kwargs): return 2 else: return 0 - class App(customtkinter.CTk):