features: Now it will add xattrs support to an specific directory (instead of the...
[rrq/fuse_xattrs.git] / fuse_xattrs.c
index b94182ef836136d65e15259a3a7b8bf5db817189..84007f7fe7990482c012a89f11d9172d19f6ab2c 100644 (file)
 /* For pread()/pwrite()/utimensat() */
 #define _XOPEN_SOURCE 700
 
+/* For get_current_dir_name */
+#define _GNU_SOURCE
+
 #include <errno.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <unistd.h>
 
 #include <fuse.h>
 #include <sys/xattr.h>
 
 #include "utils.h"
 #include "passthrough.h"
+#include "fuse_xattrs_config.h"
 
 #include "binary_storage.h"
 #include "const.h"
@@ -42,14 +47,20 @@ static int xmp_setxattr(const char *path, const char *name, const char *value, s
         return -ENOSPC;
     }
 
+    char *_path = prepend_source_directory(xattrs_config.source_dir, path);
+
 #ifdef DEBUG
     char *sanitized_value = sanitize_value(value, size);
     debug_print("path=%s name=%s value=%s size=%zu XATTR_CREATE=%d XATTR_REPLACE=%d\n",
-                path, name, sanitized_value, size, flags & XATTR_CREATE, flags & XATTR_REPLACE);
+                _path, name, sanitized_value, size, flags & XATTR_CREATE, flags & XATTR_REPLACE);
 
     free(sanitized_value);
 #endif
-     return binary_storage_write_key(path, name, value, size, flags);
+
+    int rtval = binary_storage_write_key(_path, name, value, size, flags);
+    free(_path);
+
+    return rtval;
 }
 
 static int xmp_getxattr(const char *path, const char *name, char *value, size_t size)
@@ -63,8 +74,12 @@ static int xmp_getxattr(const char *path, const char *name, char *value, size_t
         return -ERANGE;
     }
 
-    debug_print("path=%s name=%s size=%zu\n", path, name, size);
-    return binary_storage_read_key(path, name, value, size);
+    char *_path = prepend_source_directory(xattrs_config.source_dir, path);
+    debug_print("path=%s name=%s size=%zu\n", _path, name, size);
+    int rtval = binary_storage_read_key(_path, name, value, size);
+    free(_path);
+
+    return rtval;
 }
 
 static int xmp_listxattr(const char *path, char *list, size_t size)
@@ -74,8 +89,12 @@ static int xmp_listxattr(const char *path, char *list, size_t size)
         return -E2BIG;
     }
 
-    debug_print("path=%s size=%zu\n", path, size);
-    return binary_storage_list_keys(path, list, size);
+    char *_path = prepend_source_directory(xattrs_config.source_dir, path);
+    debug_print("path=%s size=%zu\n", _path, size);
+    int rtval = binary_storage_list_keys(_path, list, size);
+    free(_path);
+
+    return rtval;
 }
 
 static int xmp_removexattr(const char *path, const char *name)
@@ -89,8 +108,12 @@ static int xmp_removexattr(const char *path, const char *name)
         return -ERANGE;
     }
 
-    debug_print("path=%s name=%s\n", path, name);
-    return binary_storage_remove_key(path, name);
+    char *_path = prepend_source_directory(xattrs_config.source_dir, path);
+    debug_print("path=%s name=%s\n", _path, name);
+    int rtval = binary_storage_remove_key(_path, name);
+    free(_path);
+
+    return rtval;
 }
 
 static struct fuse_operations xmp_oper = {
@@ -126,9 +149,127 @@ static struct fuse_operations xmp_oper = {
         .removexattr = xmp_removexattr,
 };
 
-int main(int argc, char *argv[])
-{
-    // TODO: parse options...
+
+enum {
+    KEY_HELP,
+    KEY_VERSION,
+};
+
+
+static struct fuse_opt xattrs_opts[] = {
+        FUSE_OPT_KEY("-V", KEY_VERSION),
+        FUSE_OPT_KEY("--version", KEY_VERSION),
+        FUSE_OPT_KEY("-h", KEY_HELP),
+        FUSE_OPT_KEY("--help", KEY_HELP),
+        FUSE_OPT_END
+};
+
+int is_directory(const char *path) {
+    struct stat statbuf;
+    if (stat(path, &statbuf) != 0) {
+        fprintf(stderr, "cannot get source directory status: %s\n", path);
+        return -1;
+    }
+
+    if (!S_ISDIR(statbuf.st_mode)) {
+        fprintf(stderr, "source directory must be a directory: %s\n", path);
+        return -1;
+    }
+
+    return 1;
+}
+
+/**
+ * Check if the path is valid. If it's a relative path,
+ * prepend the working path.
+ * @param path relative or absolute path to eval.
+ * @return new string with absolute path
+ */
+const char *sanitized_source_directory(const char *path) {
+    char *absolute_path;
+    if (strlen(path) == 0) {
+        return NULL;
+    }
+
+    /* absolute path, we don't do anything */
+    if (path[0] == '/') {
+        if (is_directory(path) == -1) {
+            return NULL;
+        }
+        absolute_path = strdup(path);
+        return absolute_path;
+    }
+
+    char *pwd = get_current_dir_name();
+    size_t len = strlen(pwd) + 1 + strlen(path) + 1;
+    int has_trailing_backslash = (path[strlen(path)-1] == '/');
+    if (!has_trailing_backslash)
+        len++;
+
+    absolute_path = (char*) malloc(sizeof(char) * len);
+    memset(absolute_path, '\0', len);
+    sprintf(absolute_path, "%s/%s", pwd, path);
+
+    if(!has_trailing_backslash)
+        absolute_path[len-2] = '/';
+
+    if (is_directory(absolute_path) == -1) {
+        free(absolute_path);
+        return NULL;
+    }
+
+    return absolute_path;
+}
+
+static int xattrs_opt_proc(void *data, const char *arg, int key,
+                           struct fuse_args *outargs) {
+    (void) data;
+    switch (key) {
+        case FUSE_OPT_KEY_NONOPT:
+            if (!xattrs_config.source_dir) {
+                xattrs_config.source_dir = sanitized_source_directory(arg);
+                return 0;
+            }
+            break;
+
+        case KEY_HELP:
+            fprintf(stderr,
+                    "usage: %s source_dir mountpoint [options]\n"
+                            "\n"
+                            "general options:\n"
+                            "    -o opt,[opt...]  mount options\n"
+                            "    -h   --help      print help\n"
+                            "    -V   --version   print version\n"
+                            "\n"
+                            "FUSE XATTRS options:\n"
+                            "\n", outargs->argv[0]);
+
+            fuse_opt_add_arg(outargs, "-ho");
+            fuse_main(outargs->argc, outargs->argv, &xmp_oper, NULL);
+            exit(1);
+
+        case KEY_VERSION:
+            fuse_opt_add_arg(outargs, "--version");
+            fuse_main(outargs->argc, outargs->argv, &xmp_oper, NULL);
+            exit(0);
+    }
+    return 1;
+}
+
+
+
+int main(int argc, char *argv[]) {
+    struct fuse_args args = FUSE_ARGS_INIT(argc, argv);
+    if (fuse_opt_parse(&args, &xattrs_config, xattrs_opts, xattrs_opt_proc) == -1) {
+        exit(1);
+    }
+
+    if (!xattrs_config.source_dir) {
+        fprintf(stderr, "missing source directory\n");
+        fprintf(stderr, "see `%s -h' for usage\n", argv[0]);
+        exit(1);
+    }
+
     umask(0);
-    return fuse_main(argc, argv, &xmp_oper, NULL);
+    return fuse_main(args.argc, args.argv, &xmp_oper, NULL);
 }