Error checking for setupHEADmaster

1 files changed, 2 insertions, 1 deletions

diff --git a/main.c b/main.c
index 23b6104..9b28b53 100644
--- a/main.c
+++ b/main.c
@@ -150,7 +150,8 @@ static int setup_gsr_ui(const char *user_homepath) {
         success &= (copy_file_atomic_set_capabilities(GSR_GLOBAL_HOTKEYS_FILEPATH, gsr_global_hotkeys_local_filepath, (const cap_value_t[]){ CAP_SETUID }, 1) == 1);
         return success ? 0 : 1;
     } else {
-        create_local_kms_server_proxy_directory(user_homepath);
+        if(create_local_kms_server_proxy_directory(user_homepath) != 1)
+            return 1;
 
         const char *args[] = { "pkexec", self_path, "setup-gsr-ui", user_homepath, NULL };
         return execvp(args[0], (char *const*)args);