Set file permissions and ownership for all gsr files needed

1 files changed, 20 insertions, 12 deletions

diff --git a/main.c b/main.c
index 8d0c1a3..a963be6 100644
--- a/main.c
+++ b/main.c
@@ -64,10 +64,6 @@ static bool file_set_capabilities(const char *filepath, const cap_value_t *caps_
     return res == 0;
 }
 
-static bool file_set_permissions(const char *filepath, unsigned int permissions) {
-    return chmod(filepath, permissions) == 0;
-}
-
 static bool create_local_kms_server_proxy_directory(const char *home) {
     char path[PATH_MAX];
     int err;
@@ -127,14 +123,26 @@ static void usage(void) {
     exit(1);
 }
 
-static bool set_kms_server_proxy_permissions_and_capabilities(const char *kms_server_proxy_filepath) {
-    /* owner: read/write/execute, group: read/execute, public: read/execute */
-    if(!file_set_permissions(kms_server_proxy_filepath, 0755)) {
-        fprintf(stderr, "Error: failed to set kms-server-proxy permissions\n");
-        return false;
+/* |gsr_kms_server_local_filepath| and |gsr_global_hotkeys_local_filepath| can be NULL */
+static bool set_gsr_files_set_permissions_and_capabilities(const char *kms_server_proxy_local_filepath, const char *gsr_kms_server_local_filepath, const char *gsr_global_hotkeys_local_filepath) {
+    const char *filepaths[3] = {kms_server_proxy_local_filepath, gsr_kms_server_local_filepath, gsr_global_hotkeys_local_filepath};
+    for(int i = 0; i < 3; ++i) {
+        if(!filepaths[i])
+            continue;
+
+        /* owner: read/write/execute, group: read/execute, public: read/execute */
+        if(chmod(filepaths[i], 0755) != 0) {
+            fprintf(stderr, "Error: failed to set %s permissions\n", filepaths[i]);
+            return false;
+        }
+
+        if(chown(filepaths[i], 0, 0) != 0) {
+            fprintf(stderr, "Error: failed to set %s ownership\n", filepaths[i]);
+            return false;
+        }
     }
 
-    if(!file_set_capabilities(kms_server_proxy_filepath, (const cap_value_t[]){ CAP_SYS_ADMIN, CAP_SETFCAP, CAP_SETUID }, 3)) {
+    if(!file_set_capabilities(kms_server_proxy_local_filepath, (const cap_value_t[]){ CAP_SYS_ADMIN, CAP_SETFCAP, CAP_SETUID }, 3)) {
         fprintf(stderr, "Error: failed to set kms-server-proxy capabilities\n");
         return false;
     }
@@ -193,7 +201,7 @@ static int setup_gsr_ui(const char *user_homepath) {
     snprintf(gsr_global_hotkeys_local_filepath, sizeof(gsr_global_hotkeys_local_filepath), "%s/.local/share/gpu-screen-recorder/gsr-global-hotkeys", user_homepath);
 
     if(geteuid() == 0) { /* is current user root? */
-        if(!set_kms_server_proxy_permissions_and_capabilities(kms_server_proxy_local_filepath))
+        if(!set_gsr_files_set_permissions_and_capabilities(kms_server_proxy_local_filepath, gsr_kms_server_local_filepath, gsr_global_hotkeys_local_filepath))
             return 1;
 
         return 0;
@@ -240,7 +248,7 @@ static int launch_gsr_kms_server(const char *initial_socket_path, const char *ca
         const char *args[] = { "pkexec", gsr_kms_server_local_filepath, initial_socket_path, card_path, NULL };
         return execvp(args[0], (char *const*)args);
     } else if(geteuid() == 0) { /* is current user root? */
-        if(!set_kms_server_proxy_permissions_and_capabilities(kms_server_proxy_local_filepath))
+        if(!set_gsr_files_set_permissions_and_capabilities(kms_server_proxy_local_filepath, gsr_kms_server_local_filepath, NULL))
             return 1;
 
         const char *args[] = { gsr_kms_server_local_filepath, initial_socket_path, card_path, NULL };