Tried making the perm permissions faster by perparing queries

Makefile

@@ -21,14 +21,14 @@ endif
 
 # set up cflags and libs
 
-CFLAGS := -D_FILE_OFFSET_BITS=64
+CFLAGS := -D_FILE_OFFSET_BITS=64 -g
 LDFLAGS :=
 
 CFLAGS += $(shell pkg-config --cflags $(PACKAGE_NAMES))
 LDFLAGS += $(shell pkg-config --libs $(PACKAGE_NAMES))
 
 ifeq ($(DEBUG),1)
-	CFLAGS += -O0 -pedantic -g -Wall -Wextra -Wcast-align \
+	CFLAGS += -O0 -pedantic -Wall -Wextra -Wcast-align \
 						-Wcast-qual -Wdisabled-optimization -Wformat=2 \
 						-Winit-self -Wlogical-op -Wmissing-declarations \
 						-Wmissing-include-dirs -Wredundant-decls -Wshadow \

src/access_t.h

@@ -1,6 +0,0 @@
-#ifndef ACCESS_T_H
-#define ACCESS_T_H
-
-typedef enum { DENY, ALLOW, ALLOW_TEMP, NDEF } access_t;
-
-#endif // !ACCESS_T_H

src/perm_permissions_table.c

@@ -7,7 +7,6 @@
 */
 
 #include "perm_permissions_table.h"
-#include "access_t.h"
 #include "process_info.h"
 #include <fcntl.h>
 #include <pthread.h>
@@ -24,10 +23,11 @@
 sqlite3 *perm_database = NULL;
 const char *const table_name = "permissions";
 // one row corresponds to a permission to access one file for one executable
-const int column_count = 3;
-const char *const schema[] = {"executable", "filename", "mode"};
-const char *const types[] = {"TEXT", "TEXT", "INTEGER"};
+const int column_count = 2;
+const char *const schema[] = {"executable", "filename"};
+const char *const types[] = {"TEXT", "TEXT"};
 uid_t ruid, euid, current_pid;
+sqlite3_stmt *perm_check_statement = NULL;
 pthread_mutex_t uid_switch = PTHREAD_MUTEX_INITIALIZER;
 
 void set_db_fsuid() {
@@ -70,10 +70,9 @@ static int check_table_col_schema(void *notused, int argc, char **argv,
   }
   int column_num = atoi(argv[0]);
   if (column_num >= column_count) {
-    fprintf(stderr, "Table contains unexpected amount of columns.\n");
+    fprintf(stderr, "Table contains more columns than expected.\n");
     return 1;
   }
-
   if (strcmp(schema[column_num], argv[1]) == 0 &&
       strcmp(types[column_num], argv[2]) == 0) {
     return 0;
@@ -92,9 +91,8 @@ static int set_flag(void *flag, int argc, char **argv, char **colname) {
 
 int create_database_schema() {
   fprintf(stderr, "Creating table 'permissions'.\n");
-  const char *create_query =
-      "CREATE TABLE permissions(executable TEXT NOT "
-      "NULL, filename TEXT NOT NULL, mode INTEGER NOT NULL);";
+  const char *create_query = "CREATE TABLE permissions(executable TEXT NOT "
+                             "NULL, filename TEXT NOT NULL);";
   char *err = NULL;
   int ret = sqlite3_exec(perm_database, create_query, NULL, NULL, &err);
 
@@ -144,6 +142,45 @@ int ensure_database_schema() {
   return 0;
 }
 
+int prepare_sql_queries() {
+  const char *query_template =
+      "SELECT * FROM %s WHERE executable = ? AND filename = ?;";
+  char *query_string = NULL;
+  int query_len = snprintf(NULL, 0, query_template, table_name) + 1;
+
+  if (query_len < 0) {
+    fprintf(stderr, "Failed to prepare statement");
+    perror("");
+    return 1;
+  }
+
+  query_string = malloc(query_len);
+  if (query_string == NULL) {
+    fprintf(stderr, "Failed to allocate memory for the query");
+    perror("");
+    return 1;
+  }
+
+  int ret = snprintf(query_string, query_len, query_template, table_name);
+  if (ret < 0) {
+    fprintf(stderr, "Failed to prepare statement");
+    perror("");
+    free(query_string);
+    return 1;
+  }
+
+  if (sqlite3_prepare_v2(perm_database, query_string, -1, &perm_check_statement,
+                         NULL) != SQLITE_OK) {
+    fprintf(stderr, "Failed to prepare statement: %s\n",
+            sqlite3_errmsg(perm_database));
+    free(query_string);
+    return 1;
+  }
+  free(query_string);
+  return 0;
+}
+
+void free_sql_queries(void) { sqlite3_finalize(perm_check_statement); }
 /**
  * Initializes the permanent permissions table.
  *
@@ -172,58 +209,61 @@ int init_perm_permissions_table(const char *db_filename) {
 
   int status = seteuid(ruid);
   if (status < 0) {
-    fprintf(stderr, "Couldn't set euid to ruid during database setup.\n");
+    fprintf(stderr, "Couldn't set euid to ruid.\n");
     exit(status);
   }
 
+  if (prepare_sql_queries()) {
+    fprintf(stderr, "Couldn't prepare sql queries.\n");
+    exit(status);
+  }
   return 0;
 }
 
 /**
  * Destroys the permanent permissions table.
  */
-void destroy_perm_permissions_table() { sqlite3_close(perm_database); }
+void destroy_perm_permissions_table(void) {
+  free_sql_queries();
+  sqlite3_close(perm_database);
+}
 
 /**
  * Checks if the process has a permanent access to the file.
  *
  * @param filename: The file that the process is trying to access
  * @pram pi: The process information
- * @return: access status - ALLOW, DENY or NDEF in case if no information was
- * found
+ * @return: 0 if access is denied, 1 if access is allowed
  */
-access_t check_perm_access(const char *filename, struct process_info pi) {
-
-  char *query = NULL;
-  int ret = asprintf(&query,
-                     "SELECT * FROM %s WHERE executable = \'%s\' "
-                     "AND filename = \'%s\' AND mode = TRUE;",
+int check_perm_access(const char *filename, struct process_info pi) {
+  size_t query_len =
+      56 + strlen(table_name) + strlen(filename) + strlen(pi.name);
+  const char *query = malloc(query_len);
+  size_t should_be_written = snprintf(
+      query, query_len,
+      "SELECT * FROM %s WHERE executable = \'%s\' AND filename = \'%s\';",
       table_name, pi.name, filename);
-
-  if (ret < 0) {
-    // If asprintf fails, the contents of query are undefined (see man
-    // asprintf). That does not explicitly rule out that query will be a valid
-    // pointer. But the risk of freeing a non-allocated pointer is too much to
-    // justify preparing for this.
-    fprintf(stderr, "Could not create query on access check");
-    perror("");
-    return NDEF;
+  // -1 for the \0
+  if (should_be_written != query_len - 1) {
+    fprintf(stderr,
+            "Unexpected query size while permanent access rule check: "
+            "Expected %lu, but snprintf returned %lu. The query: %s\n",
+            query_len, should_be_written, query);
+    return 0;
   }
 
   char *sqlite_error = NULL;
   int flag = 0;
-  ret = sqlite3_exec(perm_database, query, set_flag, &flag, &sqlite_error);
-  free((void *)query);
+  int ret = sqlite3_exec(perm_database, query, set_flag, &flag, &sqlite_error);
   if (ret != SQLITE_OK) {
     fprintf(stderr, "SQLite returned an error: %s\n", sqlite_error);
     sqlite3_free(sqlite_error);
-    return NDEF;
+    free(query);
+    return 0;
   }
 
-  if (flag) {
-    return ALLOW;
-  }
-  return NDEF;
+  free(query);
+  return flag;
 }
 
 /**
@@ -234,23 +274,23 @@ access_t check_perm_access(const char *filename, struct process_info pi) {
  * @return: 0 on success, 1 on failure
  */
 int give_perm_access(const char *filename, struct process_info pi) {
-  char *query = NULL;
-  int ret = asprintf(&query, "INSERT INTO %s VALUES (\'%s\', \'%s\', TRUE);",
+  size_t query_len =
+      30 + strlen(table_name) + strlen(filename) + strlen(pi.name);
+  const char *query = malloc(query_len);
+  size_t should_be_written =
+      snprintf(query, query_len, "INSERT INTO %s VALUES (\'%s\', \'%s\');",
                table_name, pi.name, filename);
-
-  if (ret < 0) {
-    // If asprintf fails, the contents of query are undefined (see man
-    // asprintf). That does not explicitly rule out that query will be a valid
-    // pointer. But the risk of freeing a non-allocated pointer is too much to
-    // justify preparing for this.
-    fprintf(stderr, "Could not create query on rule insertion");
-    perror("");
+  // -1 for the \0
+  if (should_be_written != query_len - 1) {
+    fprintf(stderr,
+            "Unexpected query size while permanent access rule insertion: "
+            "Expected %lu, but snprintf returned %lu\n",
+            query_len, should_be_written);
     return 1;
   }
 
   char *sqlite_error = NULL;
-  ret = sqlite3_exec(perm_database, query, NULL, NULL, &sqlite_error);
-  free(query);
+  int ret = sqlite3_exec(perm_database, query, NULL, NULL, &sqlite_error);
   if (ret != SQLITE_OK) {
     fprintf(stderr, "SQLite returned an error: %s\n", sqlite_error);
     sqlite3_free(sqlite_error);
@@ -258,5 +298,6 @@ int give_perm_access(const char *filename, struct process_info pi) {
     return 1;
   }
 
+  free(query);
   return 0;
 }

src/perm_permissions_table.h

@@ -9,7 +9,6 @@
 #ifndef PERM_PERMISSION_TABLE_H
 #define PERM_PERMISSION_TABLE_H
 
-#include "access_t.h"
 #include "process_info.h"
 
 /**
@@ -30,10 +29,9 @@ void destroy_perm_permissions_table();
  *
  * @param filename: The file that the process is trying to access
  * @pram pi: The process information
- * @return: access status - ALLOW, DENY or NDEF in case if no information was
- * found
+ * @return: 0 if access is denied, 1 if access is allowed
  */
-access_t check_perm_access(const char *filename, struct process_info pi);
+int check_perm_access(const char *filename, struct process_info pi);
 
 /**
  * Gives permanent access to the process to the file.

src/temp_permissions_table.c

@@ -7,7 +7,6 @@
 */
 
 #include "temp_permissions_table.h"
-#include "access_t.h"
 #include "cc.h"
 #include "process_info.h"
 #include <pthread.h>
@@ -19,7 +18,6 @@ struct temp_process_permissions {
   // proc_pid_stat(5))
   unsigned long long creation_time;
   vec(char *) allowed_files;
-  vec(char *) denied_files;
 };
 
 map(pid_t, struct temp_process_permissions) temp_permissions_table;
@@ -35,8 +33,8 @@ pthread_mutex_t temp_permissions_table_lock;
  * never really equal to 0, it exceptionally unlikely.
  */
 unsigned long long get_process_creation_time(pid_t pid) {
-  char path[256];
-  FILE *fp = NULL;
+  char path[32];
+  FILE *fp;
   unsigned long long creation_time = 0;
 
   // Construct the path to the process's status file
@@ -74,7 +72,7 @@ unsigned long long get_process_creation_time(pid_t pid) {
  *
  * @return: 0 on success, -1 on failure (e.g. ENOMEM)
  */
-int init_temp_permissions_table(void) {
+int init_temp_permissions_table() {
   pthread_mutex_init(&temp_permissions_table_lock, PTHREAD_MUTEX_DEFAULT);
   init(&temp_permissions_table);
   return 0;
@@ -83,21 +81,14 @@ int init_temp_permissions_table(void) {
 /**
  * Destroys the temporary permissions table.
  *
- * @note: the table is guranteed to be destroyed if it is already initialized.
- * It does not indicate any errors whatsoever. If something goes wrong - you are
- * screwed.
+ * @note: the table is guranteed to be destroyed if it is already initialized
  */
-void destroy_temp_permissions_table(void) {
+void destroy_temp_permissions_table() {
   // free the memory allocated for the table
   for_each(&temp_permissions_table, entry) {
     for_each(&entry->allowed_files, allowed_file) { free(*allowed_file); }
     cleanup(&entry->allowed_files);
   }
-  for_each(&temp_permissions_table, entry) {
-    for_each(&entry->denied_files, denied_file) { free(*denied_file); }
-    cleanup(&entry->denied_files);
-  }
-
   cleanup(&temp_permissions_table);
   pthread_mutex_destroy(&temp_permissions_table_lock);
 }
@@ -107,10 +98,10 @@ void destroy_temp_permissions_table(void) {
  *
  * @param filename: The file that the process is trying to access
  * @pram pid: PID of the process
- * @return: access status - ALLOW, DENY or NDEF in case if no information was
- * found is avaliable
+ * @return: 0 if access is denied, 1 if access is allowed
  */
-access_t check_temp_access_noparent(const char *filename, pid_t pid) {
+
+int check_temp_access_noparent(const char *filename, pid_t pid) {
   // TODO: more efficient locking
   pthread_mutex_lock(&temp_permissions_table_lock);
   struct temp_process_permissions *permission_entry =
@@ -120,28 +111,22 @@ access_t check_temp_access_noparent(const char *filename, pid_t pid) {
     if (process_creation_time == 0) {
       perror("Could not retrieve process creation time");
       pthread_mutex_unlock(&temp_permissions_table_lock);
-      return NDEF;
+      return 0;
     }
 
     if (process_creation_time == permission_entry->creation_time) {
       // the process is the same as the one that was granted temporary access
       // to the file
-      for_each(&permission_entry->denied_files, denied_file) {
-        if (strncmp(*denied_file, filename, strlen(filename)) == 0) {
-          pthread_mutex_unlock(&temp_permissions_table_lock);
-          return DENY;
-        }
-      }
       for_each(&permission_entry->allowed_files, allowed_file) {
         if (strncmp(*allowed_file, filename, strlen(filename)) == 0) {
           pthread_mutex_unlock(&temp_permissions_table_lock);
-          return ALLOW;
+          return 1;
         }
       }
     }
   }
   pthread_mutex_unlock(&temp_permissions_table_lock);
-  return NDEF;
+  return 0;
 }
 
 /**
@@ -180,36 +165,31 @@ pid_t get_parent_pid(pid_t pid) {
  *
  * @param filename: The file that the process is trying to access
  * @pram pi: The process information
- * @return: access status - ALLOW, DENY or NDEF in case if no information was
- * found. Does not return ALLOW_TEMP.
+ * @return: 0 if access is denied, 1 if access is allowed
  * @note: In case one of the parent processes is killed while this function
  * execution the result is not guranteed to be correct. It should only lead to
  * false negatives, though.
  */
-access_t check_temp_access(const char *filename, struct process_info pi) {
+int check_temp_access(const char *filename, struct process_info pi) {
   pid_t current_pid = pi.PID;
   while (current_pid != 0) {
-    access_t access = check_temp_access_noparent(filename, current_pid);
-    if (access != NDEF) {
-      return access;
+    if (check_temp_access_noparent(filename, current_pid)) {
+      return 1;
     }
     current_pid = get_parent_pid(current_pid);
   }
 
-  return NDEF;
+  return 0;
 }
 
 /**
- * Sets temporary access mode of the process to the file.
+ * Gives temporary access to the process to the file.
  *
  * @param filename: The file that the process is trying to access
  * @param pi: The process information
- * @param mode: Kind of access rule to be set - SET_DENY to deny access, and
- * SET_ALLOW to allow access.
- * @return: 0 on success, -1 on failure.
+ * @return: 0 on success, -1 on failure (e.g. ENOMEM)
  */
-int set_temp_access(const char *filename, struct process_info pi,
-                    set_mode_t mode) {
+int give_temp_access(const char *filename, struct process_info pi) {
   pthread_mutex_lock(&temp_permissions_table_lock);
   struct temp_process_permissions *permission_entry =
       get(&temp_permissions_table, pi.PID);
@@ -227,13 +207,7 @@ int set_temp_access(const char *filename, struct process_info pi,
     if (process_creation_time == permission_entry->creation_time) {
       // the process is the same as the one that was granted temporary access
       // to the file
-      if (mode == SET_ALLOW) {
       push(&permission_entry->allowed_files, strdup(filename));
-      }
-      if (mode == SET_DENY) {
-        push(&permission_entry->denied_files, strdup(filename));
-      }
-
       pthread_mutex_unlock(&temp_permissions_table_lock);
       return 0;
     }
@@ -249,13 +223,7 @@ int set_temp_access(const char *filename, struct process_info pi,
 
   new_permission_entry.creation_time = get_process_creation_time(pi.PID);
   init(&new_permission_entry.allowed_files);
-  init(&new_permission_entry.denied_files);
-  if (mode == SET_ALLOW) {
   push(&new_permission_entry.allowed_files, strdup(filename));
-  }
-  if (mode == SET_DENY) {
-    push(&new_permission_entry.denied_files, strdup(filename));
-  }
 
   insert(&temp_permissions_table, pi.PID, new_permission_entry);
 

src/temp_permissions_table.h

@@ -2,7 +2,6 @@
 #ifndef TEMP_PERMISSIONS_TABLE_H
 #define TEMP_PERMISSIONS_TABLE_H
 
-#include "access_t.h"
 #include "process_info.h"
 
 /**
@@ -10,43 +9,31 @@
  *
  * @return: 0 on success, -1 on failure (e.g. ENOMEM)
  */
-int init_temp_permissions_table(void);
+int init_temp_permissions_table();
 
 /**
  * Destroys the temporary permissions table.
  *
- * @note: the table is guranteed to be destroyed if it is already initialized.
- * It does not indicate any errors whatsoever. If something goes wrong - you are
- * screwed.
+ * @note: the table is guranteed to be destroyed if it is already initialized
  */
-void destroy_temp_permissions_table(void);
+void destroy_temp_permissions_table();
 
 /**
- * Checks if the process or any of it's parents have temporary access to the
- * file.
- *
- * @param filename: The file that the process is trying to access
- * @pram pi: The process information
- * @return: access status - ALLOW, DENY or NDEF in case if no information was
- * found. Does not return ALLOW_TEMP.
- * @note: In case one of the parent processes is killed while this function
- * execution the result is not guranteed to be correct. It should only lead to
- * false negatives, though.
- */
-access_t check_temp_access(const char *filename, struct process_info pi);
-
-typedef enum { SET_DENY, SET_ALLOW } set_mode_t;
-
-/**
- * Sets temporary access mode of the process to the file.
+ * Checks if the process has a temporary access to the file.
  *
  * @param filename: The file that the process is trying to access
  * @param pi: The process information
- * @param mode: Kind of access rule to be set - SET_DENY to deny access, and
- * SET_ALLOW to allow access.
- * @return: 0 on success, -1 on failure.
+ * @return: 0 if access is denied, 1 if access is allowed
  */
-int set_temp_access(const char *filename, struct process_info pi,
-                    set_mode_t mode);
+int check_temp_access(const char *filename, struct process_info pi);
+
+/**
+ * Gives temporary access to the process to the file.
+ *
+ * @param filename: The file that the process is trying to access
+ * @param pi: The process information
+ * @return: 0 on success, -1 on failure (e.g. ENOMEM)
+ */
+int give_temp_access(const char *filename, struct process_info pi);
 
 #endif // !TEMP_PERMISSIONS_TABLE_H

src/ui-socket.c

@@ -6,7 +6,6 @@
   See the file LICENSE.
 */
 
-#include "access_t.h"
 #include <stddef.h>
 #include <sys/types.h>
 #include <time.h>
@@ -59,36 +58,27 @@ void destroy_ui_socket(void) {
  *
  * @param filename: The file that the process is trying to access
  * @param pi: The process information
- * @return: access status - ALLOW, DENY or ALLOW_TEMP
+ * @return: 0 if access is denied, 1 if access is allowed, 2 if access is
  * allowed for the runtime of the process
  */
-access_t ask_access(const char *filename, struct process_info proc_info) {
+int ask_access(const char *filename, struct process_info proc_info) {
   FILE *fp = NULL;
-  char *command = NULL;
-  int ret =
-      asprintf(&command,
+  size_t command_len =
+      139 + sizeof(pid_t) * 8 + strlen(proc_info.name) + strlen(filename);
+  char *command = (char *)malloc(command_len);
+  snprintf(command, command_len,
            "zenity --question --extra-button \"Allow this time\" --title "
            "\"Allow Access?\" --text \"Allow process "
            "<tt>%s</tt> with PID <tt>%d</tt> to access <tt>%s</tt>\"",
            proc_info.name, proc_info.PID, filename);
 
-  if (ret < 0) {
-    // If asprintf fails, the contents of command are undefined (see man
-    // asprintf). That does not explicitly rule out that command will be a valid
-    // pointer. But the risk of freeing a non-allocated pointer is too much to
-    // justify preparing for this.
-    fprintf(stderr, "Could not create query on rule insertion");
-    perror("");
-    return 1;
-  }
-
   // Zenity Question Message Popup
   fp = popen(command, "r");
   free(command);
 
   if (fp == NULL) {
     perror("Pipe returned a error");
-    return DENY;
+    return -1;
   }
 
   // if the user clicks the "Allow this time" button, `zenity` will only
@@ -99,7 +89,7 @@ access_t ask_access(const char *filename, struct process_info proc_info) {
     printf("%s", buffer);
     if (strcmp(buffer, ZENITY_TEMP_ALLOW_MESSAGE) == 0) {
       pclose(fp);
-      return ALLOW_TEMP;
+      return 2;
     }
   }
 
@@ -107,10 +97,10 @@ access_t ask_access(const char *filename, struct process_info proc_info) {
   fprintf(stderr, "zenity returned %d\n", zenity_exit_code);
   // zenity returns 1 on "No" >:(
   if (zenity_exit_code == 0) {
-    return ALLOW;
+    return 1;
   }
 
-  return DENY;
+  return 0;
 }
 
 /**
@@ -127,21 +117,11 @@ access_t ask_access(const char *filename, struct process_info proc_info) {
 int interactive_access(const char *filename, struct process_info proc_info,
                        int opts) {
 
-  access_t access = check_temp_access(filename, proc_info);
-  if (access == ALLOW) {
+  if (check_temp_access(filename, proc_info) ||
+      check_perm_access(filename, proc_info)) {
+    // access was already granted before
     return 1;
   }
-  if (access == DENY) {
-    return 0;
-  }
-
-  access = check_perm_access(filename, proc_info);
-  if (access == ALLOW) {
-    return 1;
-  }
-  if (access == DENY) {
-    return 0;
-  }
 
   // if noth GRANT_TEMP and GRANT_PERM are selected, then only permanent
   // permissions are granted
@@ -151,26 +131,23 @@ int interactive_access(const char *filename, struct process_info proc_info,
     return 1;
   }
   if (opts & GRANT_TEMP) {
-    set_temp_access(filename, proc_info, SET_ALLOW);
+    give_temp_access(filename, proc_info);
     return 1;
   }
 
-  access_t user_response = ask_access(filename, proc_info);
-  if (user_response == ALLOW) {
+  int user_response = ask_access(filename, proc_info);
+  if (user_response == 1) {
+    // user said "yes"
     give_perm_access(filename, proc_info);
     return 1;
   }
 
-  if (user_response == ALLOW_TEMP) {
-    set_temp_access(filename, proc_info, SET_ALLOW);
+  if (user_response == 2) {
+    // user said "yes, but only this time"
+    give_temp_access(filename, proc_info);
     return 1;
   }
 
-  if (user_response == DENY) {
-    set_temp_access(filename, proc_info, SET_DENY);
-    return 0;
-  }
-
-  // deny on unknown options.
+  // otherwise "no"
   return 0;
 }

test/mock/zenity

@@ -12,10 +12,17 @@ else
     if [[ $FAKE_ZENITY_RESPONSE == "yes_tmp" ]]; then
       printf "Allow this time\n"
       exit 1
+    elif [[ $FAKE_ZENITY_RESPONSE == "yes_tmp_alt" ]]; then
+      printf "Allow this time\n"
+      echo "yes_alt" >~/.fake_zenity_response
+      exit 1
     elif [[ $FAKE_ZENITY_RESPONSE == "no" ]]; then
       exit 1
     elif [[ $FAKE_ZENITY_RESPONSE == "yes" ]]; then
       exit 0
+    elif [[ $FAKE_ZENITY_RESPONSE == "yes_alt" ]]; then
+      echo "yes_tmp_alt" >~/.fake_zenity_response
+      exit 0
     fi
   fi
 fi

test/test.bash

@@ -25,10 +25,18 @@ if [[ $1 == "--setuid" ]]; then
   echo "Valgrind will not be used due to setuid compatibility issues."
   ../build/icfs -o default_permissions ./protected ./.pt.db &
   sleep 1
+elif [[ $1 == "--perf" ]]; then
+  echo "Profiling with perf..."
+  ../build/icfs -o default_permissions ./protected ./.pt.db &
+  echo "Profiling will require root privilieges."
+  sleep 3
+  echo "Attaching to $(pgrep icfs)"
+  sudo perf record -g -e cycles:u --call-graph dwarf -p $(pgrep icfs) &
+  sleep 10
 else
   echo "Database protection will not be tested due to the lack of setuid capabilites."
   echo "To test it, run this script with '--setuid'."
-  valgrind -s ../build/icfs -o default_permissions ./protected ./.pt.db &
+  valgrind --leak-check=full -s ../build/icfs -o default_permissions ./protected ./.pt.db &
   sleep 5
 fi
 
@@ -126,9 +134,29 @@ else
   echo "[ICFS-TEST]: OK"
 fi
 
+if [[ $1 == "--perf" ]]; then
+  zenity --set-fake-response yes_tmp
+  rm -rf ./protected/*
+  zenity --set-fake-response yes_alt
+  bonnie++ -p 4
+  bonnie++ -d ./protected -c 4 -r 256 -y s >/dev/null &
+  bonnie++ -d ./protected -c 4 -r 256 -y s >/dev/null &
+  bonnie++ -d ./protected -c 4 -r 256 -y s >/dev/null &
+  bonnie++ -d ./protected -c 4 -r 256 -y s >/dev/null
+  bonnie++ -p -1
+fi
+
 # unmount
 
 sleep 0.5
 #lsof +f -- $(realpath ./protected)
 umount $(realpath ./protected)
 sleep 0.5
+
+if [[ $1 == "--perf" ]]; then
+  mv ./callgraph.png ./callgraph_old.png
+  real_user=$USER
+  sudo chown "$real_user" ./perf.data
+  perf script --dsos=icfs | gprof2dot -f perf | dot -Tpng -o callgraph.png
+  echo "Profile graph was written to \"callgraph.png\""
+fi