OSDN Git Service

Added blksize option that should solve "Invalid argument" error while mounting a...
[android-x86/external-exfat.git] / fuse / main.c
index c0f1474..05ee725 100644 (file)
@@ -1,11 +1,22 @@
 /*
- *  main.c
- *  FUSE-based exFAT implementation. Requires FUSE 2.6 or later.
- *
- *  Created by Andrew Nayenko on 01.09.09.
- *  This software is distributed under the GNU General Public License 
- *  version 3 or any later.
- */
+       main.c (01.09.09)
+       FUSE-based exFAT implementation. Requires FUSE 2.6 or later.
+
+       Copyright (C) 2009, 2010  Andrew Nayenko
+
+       This program is free software: you can redistribute it and/or modify
+       it under the terms of the GNU General Public License as published by
+       the Free Software Foundation, either version 3 of the License, or
+       (at your option) any later version.
+
+       This program is distributed in the hope that it will be useful,
+       but WITHOUT ANY WARRANTY; without even the implied warranty of
+       MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+       GNU General Public License for more details.
+
+       You should have received a copy of the GNU General Public License
+       along with this program.  If not, see <http://www.gnu.org/licenses/>.
+*/
 
 #include <fuse.h>
 #include <errno.h>
 #include <stdlib.h>
 #include <string.h>
 #include <exfat.h>
+#include <inttypes.h>
+#include <limits.h>
+#include <sys/types.h>
+#include <pwd.h>
+#include <unistd.h>
 
 #define exfat_debug(format, ...)
 
        #error FUSE 2.6 or later is required
 #endif
 
+const char* default_options = "allow_other,blkdev";
+
 struct exfat ef;
 
-static struct exfat_node* get_node(const struct fuse_file_info *fi)
+static struct exfat_node* get_node(const struct fuse_file_infofi)
 {
        return (struct exfat_node*) (size_t) fi->fh;
 }
 
-static void set_node(struct fuse_file_info *fi, struct exfat_node* node)
+static void set_node(struct fuse_file_infofi, struct exfat_node* node)
 {
        fi->fh = (uint64_t) (size_t) node;
 }
 
-static int fuse_exfat_getattr(const char *path, struct stat *stbuf)
+static int fuse_exfat_getattr(const char* path, struct stat* stbuf)
 {
        struct exfat_node* node;
        int rc;
@@ -44,29 +62,29 @@ static int fuse_exfat_getattr(const char *path, struct stat *stbuf)
        if (rc != 0)
                return rc;
 
-       exfat_stat(node, stbuf);
+       exfat_stat(&ef, node, stbuf);
        exfat_put_node(&ef, node);
        return 0;
 }
 
-static int fuse_exfat_truncate(const char *path, off_t size)
+static int fuse_exfat_truncate(const charpath, off_t size)
 {
        struct exfat_node* node;
        int rc;
 
-       exfat_debug("[fuse_exfat_truncate] %s, %llu", path, size);
+       exfat_debug("[fuse_exfat_truncate] %s, %"PRIu64, path, (uint64_t) size);
 
        rc = exfat_lookup(&ef, &node, path);
        if (rc != 0)
                return rc;
 
-       exfat_truncate(&ef, node, size);
+       rc = exfat_truncate(&ef, node, size);
        exfat_put_node(&ef, node);
-       return 0;
+       return rc;
 }
 
-static int fuse_exfat_readdir(const char *path, void *buffer,
-               fuse_fill_dir_t filler, off_t offset, struct fuse_file_info *fi)
+static int fuse_exfat_readdir(const char* path, void* buffer,
+               fuse_fill_dir_t filler, off_t offset, struct fuse_file_infofi)
 {
        struct exfat_node* parent;
        struct exfat_node* node;
@@ -99,9 +117,9 @@ static int fuse_exfat_readdir(const char *path, void *buffer,
        while ((node = exfat_readdir(&ef, &it)))
        {
                exfat_get_name(node, name, EXFAT_NAME_MAX);
-               exfat_debug("[fuse_exfat_readdir] %s: %s, %llu bytes, cluster %u",
+               exfat_debug("[fuse_exfat_readdir] %s: %s, %"PRIu64" bytes, cluster %u",
                                name, IS_CONTIGUOUS(*node) ? "contiguous" : "fragmented",
-                               node->size, node->start_cluster);
+                               (uint64_t) node->size, node->start_cluster);
                filler(buffer, name, NULL, 0);
                exfat_put_node(&ef, node);
        }
@@ -110,7 +128,7 @@ static int fuse_exfat_readdir(const char *path, void *buffer,
        return 0;
 }
 
-static int fuse_exfat_open(const char *path, struct fuse_file_info *fi)
+static int fuse_exfat_open(const char* path, struct fuse_file_info* fi)
 {
        struct exfat_node* node;
        int rc;
@@ -124,27 +142,27 @@ static int fuse_exfat_open(const char *path, struct fuse_file_info *fi)
        return 0;
 }
 
-static int fuse_exfat_release(const char *path, struct fuse_file_info *fi)
+static int fuse_exfat_release(const char* path, struct fuse_file_info* fi)
 {
        exfat_put_node(&ef, get_node(fi));
        return 0;
 }
 
-static int fuse_exfat_read(const char *path, char *buffer, size_t size,
-               off_t offset, struct fuse_file_info *fi)
+static int fuse_exfat_read(const char* path, char* buffer, size_t size,
+               off_t offset, struct fuse_file_infofi)
 {
        exfat_debug("[fuse_exfat_read] %s (%zu bytes)", path, size);
        return exfat_read(&ef, get_node(fi), buffer, size, offset);
 }
 
-static int fuse_exfat_write(const char *path, const char *buffer, size_t size,
-               off_t offset, struct fuse_file_info *fi)
+static int fuse_exfat_write(const char* path, const char* buffer, size_t size,
+               off_t offset, struct fuse_file_infofi)
 {
        exfat_debug("[fuse_exfat_write] %s (%zu bytes)", path, size);
        return exfat_write(&ef, get_node(fi), buffer, size, offset);
 }
 
-int fuse_exfat_unlink(const char *path)
+static int fuse_exfat_unlink(const char* path)
 {
        struct exfat_node* node;
        int rc;
@@ -160,7 +178,7 @@ int fuse_exfat_unlink(const char *path)
        return rc;
 }
 
-int fuse_exfat_rmdir(const char *path)
+static int fuse_exfat_rmdir(const char* path)
 {
        struct exfat_node* node;
        int rc;
@@ -188,14 +206,34 @@ static int fuse_exfat_mkdir(const char* path, mode_t mode)
        return exfat_mkdir(&ef, path);
 }
 
-static int fuse_exfat_statfs(const char *path, struct statvfs *sfs)
+static int fuse_exfat_rename(const char* old_path, const char* new_path)
 {
-       const uint64_t block_count = le64_to_cpu(ef.sb->block_count);
+       exfat_debug("[fuse_exfat_rename] %s => %s", old_path, new_path);
+       return exfat_rename(&ef, old_path, new_path);
+}
+
+static int fuse_exfat_utimens(const char* path, const struct timespec tv[2])
+{
+       struct exfat_node* node;
+       int rc;
+
+       exfat_debug("[fuse_exfat_utimens] %s", path);
+
+       rc = exfat_lookup(&ef, &node, path);
+       if (rc != 0)
+               return rc;
+
+       exfat_utimes(node, tv);
+       exfat_put_node(&ef, node);
+       return 0;
+}
 
-       sfs->f_bsize = BLOCK_SIZE(*ef.sb);
+static int fuse_exfat_statfs(const char* path, struct statvfs* sfs)
+{
+       sfs->f_bsize = CLUSTER_SIZE(*ef.sb);
        sfs->f_frsize = CLUSTER_SIZE(*ef.sb);
-       sfs->f_blocks = block_count;
-       sfs->f_bavail = block_count - ef.sb->allocated_percent * block_count / 100;
+       sfs->f_blocks = le64_to_cpu(ef.sb->block_count) >> ef.sb->bpc_bits;
+       sfs->f_bavail = exfat_count_free_clusters(&ef);
        sfs->f_bfree = sfs->f_bavail;
        sfs->f_namemax = EXFAT_NAME_MAX;
 
@@ -212,14 +250,14 @@ static int fuse_exfat_statfs(const char *path, struct statvfs *sfs)
        return 0;
 }
 
-void fuse_exfat_destroy(void* unused)
+static void fuse_exfat_destroy(void* unused)
 {
        exfat_unmount(&ef);
 }
 
 static void usage(const char* prog)
 {
-       fprintf(stderr, "Usage: %s <spec> <mountpoint> [-o options]\n", prog);
+       fprintf(stderr, "Usage: %s [-d] [-o options] <device> <dir>\n", prog);
        exit(1);
 }
 
@@ -236,23 +274,100 @@ static struct fuse_operations fuse_exfat_ops =
        .rmdir          = fuse_exfat_rmdir,
        .mknod          = fuse_exfat_mknod,
        .mkdir          = fuse_exfat_mkdir,
+       .rename         = fuse_exfat_rename,
+       .utimens        = fuse_exfat_utimens,
        .statfs         = fuse_exfat_statfs,
        .destroy        = fuse_exfat_destroy,
 };
 
+static char* add_option(char* options, const char* name, const char* value)
+{
+       size_t size;
+
+       if (value)
+               size = strlen(options) + strlen(name) + strlen(value) + 3;
+       else
+               size = strlen(options) + strlen(name) + 2;
+
+       options = realloc(options, size);
+       if (options == NULL)
+       {
+               exfat_error("failed to reallocate options string");
+               return NULL;
+       }
+       strcat(options, ",");
+       strcat(options, name);
+       if (value)
+       {
+               strcat(options, "=");
+               strcat(options, value);
+       }
+       return options;
+}
+
+static char* add_fsname_option(char* options, const char* spec)
+{
+       char spec_abs[PATH_MAX];
+
+       if (realpath(spec, spec_abs) == NULL)
+       {
+               free(options);
+               exfat_error("failed to get absolute path for `%s'", spec);
+               return NULL;
+       }
+       return add_option(options, "fsname", spec_abs);
+}
+
+static char* add_user_option(char* options)
+{
+       struct passwd* pw;
+
+       if (getuid() == 0)
+               return options;
+
+       pw = getpwuid(getuid());
+       if (pw == NULL || pw->pw_name == NULL)
+       {
+               free(options);
+               exfat_error("failed to determine username");
+               return NULL;
+       }
+       return add_option(options, "user", pw->pw_name);
+}
+
+static char* add_blksize_option(char* options, long cluster_size)
+{
+       long page_size = sysconf(_SC_PAGESIZE);
+       char blksize[20];
+
+       if (page_size < 1)
+               page_size = 0x1000;
+
+       snprintf(blksize, sizeof(blksize), "%ld", MIN(page_size, cluster_size));
+       return add_option(options, "blksize", blksize);
+}
+
 int main(int argc, char* argv[])
 {
        struct fuse_args mount_args = FUSE_ARGS_INIT(0, NULL);
        struct fuse_args newfs_args = FUSE_ARGS_INIT(0, NULL);
        const char* spec = NULL;
        const char* mount_point = NULL;
-       const char* mount_options = "";
+       char* mount_options;
        int debug = 0;
        struct fuse_chan* fc = NULL;
        struct fuse* fh = NULL;
        char** pp;
 
-       printf("FUSE exfat %u.%u\n", EXFAT_VERSION_MAJOR, EXFAT_VERSION_MINOR);
+       printf("FUSE exfat %u.%u.%u\n",
+                       EXFAT_VERSION_MAJOR, EXFAT_VERSION_MINOR, EXFAT_VERSION_PATCH);
+
+       mount_options = strdup(default_options);
+       if (mount_options == NULL)
+       {
+               exfat_error("failed to allocate options string");
+               return 1;
+       }
 
        for (pp = argv + 1; *pp; pp++)
        {
@@ -261,7 +376,9 @@ int main(int argc, char* argv[])
                        pp++;
                        if (*pp == NULL)
                                usage(argv[0]);
-                       mount_options = *pp;
+                       mount_options = add_option(mount_options, *pp, NULL);
+                       if (mount_options == NULL)
+                               return 1;
                }
                else if (strcmp(*pp, "-d") == 0)
                        debug = 1;
@@ -270,28 +387,69 @@ int main(int argc, char* argv[])
                else if (mount_point == NULL)
                        mount_point = *pp;
                else
+               {
+                       free(mount_options);
                        usage(argv[0]);
+               }
        }
        if (spec == NULL || mount_point == NULL)
+       {
+               free(mount_options);
                usage(argv[0]);
+       }
+
+       if (exfat_mount(&ef, spec, mount_options) != 0)
+       {
+               free(mount_options);
+               return 1;
+       }
+
+       mount_options = add_fsname_option(mount_options, spec);
+       if (mount_options == NULL)
+       {
+               exfat_unmount(&ef);
+               return 1;
+       }
+       mount_options = add_user_option(mount_options);
+       if (mount_options == NULL)
+       {
+               exfat_unmount(&ef);
+               return 1;
+       }
+       mount_options = add_blksize_option(mount_options, CLUSTER_SIZE(*ef.sb));
+       if (mount_options == NULL)
+       {
+               exfat_unmount(&ef);
+               return 1;
+       }
 
        /* create arguments for fuse_mount() */
        if (fuse_opt_add_arg(&mount_args, "exfat") != 0 ||
                fuse_opt_add_arg(&mount_args, "-o") != 0 ||
                fuse_opt_add_arg(&mount_args, mount_options) != 0)
+       {
+               exfat_unmount(&ef);
+               free(mount_options);
                return 1;
+       }
+
+       free(mount_options);
 
        /* create FUSE mount point */
        fc = fuse_mount(mount_point, &mount_args);
        fuse_opt_free_args(&mount_args);
        if (fc == NULL)
+       {
+               exfat_unmount(&ef);
                return 1;
+       }
 
        /* create arguments for fuse_new() */
        if (fuse_opt_add_arg(&newfs_args, "") != 0 ||
                (debug && fuse_opt_add_arg(&newfs_args, "-d") != 0))
        {
                fuse_unmount(mount_point, fc);
+               exfat_unmount(&ef);
                return 1;
        }
 
@@ -302,6 +460,7 @@ int main(int argc, char* argv[])
        if (fh == NULL)
        {
                fuse_unmount(mount_point, fc);
+               exfat_unmount(&ef);
                return 1;
        }
 
@@ -310,13 +469,7 @@ int main(int argc, char* argv[])
        {
                fuse_unmount(mount_point, fc);
                fuse_destroy(fh);
-               return 1;
-       }
-
-       if (exfat_mount(&ef, spec) != 0)
-       {
-               fuse_unmount(mount_point, fc);
-               fuse_destroy(fh);
+               exfat_unmount(&ef);
                return 1;
        }