/*
 * vmfs-tools - Tools to access VMFS filesystems
 * Copyright (C) 2009 Christophe Fillot <cf@utc.fr>
 * Copyright (C) 2009,2012 Mike Hommey <mh@glandium.org>
 *
 * 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 2 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
#define FUSE_USE_VERSION 26

#include <fuse_lowlevel.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include "vmfs.h"

static vmfs_fs_t *fs;

static inline uint32_t ino2blkid(fuse_ino_t ino)
{
   if (ino == FUSE_ROOT_ID)
      return(VMFS_BLK_FD_BUILD(0, 0, 0));
   return((uint32_t)ino);
}

static inline fuse_ino_t blkid2ino(uint32_t blk_id)
{
   if (blk_id == VMFS_BLK_FD_BUILD(0, 0, 0))
      return(FUSE_ROOT_ID);
   return((fuse_ino_t)blk_id);
}

static void vmfs_fuse_getattr(fuse_req_t req, fuse_ino_t ino,
                             struct fuse_file_info *fi)
{
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);
   struct stat stbuf = { 0, };

   if (!vmfs_inode_stat_from_blkid(fs, ino2blkid(ino), &stbuf)) {
      stbuf.st_ino = ino;
      fuse_reply_attr(req, &stbuf, 1.0);
   } else
      fuse_reply_err(req, ENOENT);
}

static void vmfs_fuse_setattr(fuse_req_t req, fuse_ino_t ino, 
                              struct stat *attr, int to_set, 
                              struct fuse_file_info *fi) 
{
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);
   struct stat stbuf = { 0, };
   vmfs_inode_t *inode;

   if (!(inode = vmfs_inode_acquire(fs,ino2blkid(ino)))) {
      fuse_reply_err(req, ENOENT);
      return;
   }

   if (to_set & FUSE_SET_ATTR_MODE)
      inode->mode = attr->st_mode;

   if (to_set & FUSE_SET_ATTR_UID)
      inode->uid = attr->st_uid;

   if (to_set & FUSE_SET_ATTR_GID)
      inode->gid = attr->st_gid;

   if (to_set & FUSE_SET_ATTR_ATIME)
      inode->atime = attr->st_atime;

   if (to_set & FUSE_SET_ATTR_MTIME)
      inode->atime = attr->st_mtime;

   if (to_set & FUSE_SET_ATTR_SIZE)
      vmfs_inode_truncate(inode,attr->st_size);

   vmfs_inode_stat(inode,&stbuf);
   stbuf.st_ino = blkid2ino(inode->id);

   fuse_reply_attr(req, &stbuf, 1.0);
   vmfs_inode_release(inode);
}

static void vmfs_fuse_readlink(fuse_req_t req,fuse_ino_t ino)
{ 
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);
   vmfs_file_t *f;
   size_t str_len;
   char *str;

   if (!(f = vmfs_file_open_from_blkid(fs, ino2blkid(ino)))) {
      fuse_reply_err(req, ENOENT);
      return;
   }

   str_len = vmfs_file_get_size(f);

   if (!(str = malloc(str_len+1))) {
      vmfs_file_close(f);
      fuse_reply_err(req, ENOMEM);
      return;
   }

   if (vmfs_file_pread(f,(u_char *)str,str_len,0) != str_len) {
      vmfs_file_close(f);
      free(str);
      fuse_reply_err(req, EIO);
      return;
   }

   str[str_len] = 0;

   fuse_reply_readlink(req,str);
   free(str);
}

static void vmfs_fuse_mknod(fuse_req_t req,fuse_ino_t parent,const char *name,
                            mode_t mode, dev_t rdev)
{   
   struct fuse_entry_param entry = { 0, };
   vmfs_inode_t *inode;
   vmfs_dir_t *dir;
   int res;

   if (!(dir = vmfs_dir_open_from_blkid(fs, ino2blkid(parent)))) {
      fuse_reply_err(req, ENOENT);
      return;
   }        

   if ((res = vmfs_file_create(dir,name,mode,&inode)) < 0) {
      fuse_reply_err(req, -res);
      return;
   }

   vmfs_dir_close(dir);

   vmfs_inode_stat(inode,&entry.attr);
   entry.ino = entry.attr.st_ino = blkid2ino(inode->id);
   entry.generation = 1;
   entry.attr_timeout = 1.0;
   entry.entry_timeout = 1.0;
   fuse_reply_entry(req, &entry);

   vmfs_inode_release(inode);
}

static void vmfs_fuse_mkdir(fuse_req_t req, fuse_ino_t parent,
                            const char *name, mode_t mode) 
{
   struct fuse_entry_param entry = { 0, };
   vmfs_inode_t *inode;
   vmfs_dir_t *dir;
   int res;

   if (!(dir = vmfs_dir_open_from_blkid(fs, ino2blkid(parent)))) {
      fuse_reply_err(req, ENOENT);
      return;
   }        

   if ((res = vmfs_dir_create(dir,name,mode,&inode)) < 0) {
      fuse_reply_err(req, -res);
      return;
   }

   vmfs_dir_close(dir);

   vmfs_inode_stat(inode,&entry.attr);
   entry.ino = entry.attr.st_ino = blkid2ino(inode->id);
   entry.generation = 1;
   entry.attr_timeout = 1.0;
   entry.entry_timeout = 1.0;

   fuse_reply_entry(req, &entry);

   vmfs_inode_release(inode);
}

static void vmfs_fuse_unlink(fuse_req_t req,fuse_ino_t parent,const char *name) 
{
   vmfs_dir_t *dir;
   int res;

   if (!(dir = vmfs_dir_open_from_blkid(fs, ino2blkid(parent)))) {
      fuse_reply_err(req, ENOENT);
      return;
   } 

   res = vmfs_file_delete(dir,name);
   vmfs_dir_close(dir);

   fuse_reply_err(req,-res);
}

static void vmfs_fuse_rmdir(fuse_req_t req,fuse_ino_t parent,const char *name) 
{
   vmfs_dir_t *dir;
   int res;

   if (!(dir = vmfs_dir_open_from_blkid(fs, ino2blkid(parent)))) {
      fuse_reply_err(req, ENOENT);
      return;
   } 

   res = vmfs_dir_delete(dir,name);
   vmfs_dir_close(dir);

   fuse_reply_err(req,-res);
}

static void vmfs_fuse_opendir(fuse_req_t req, fuse_ino_t ino,
                              struct fuse_file_info *fi)
{
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);

   fi->fh = (uint64_t)(unsigned long)
            vmfs_dir_open_from_blkid(fs, ino2blkid(ino));
   if (fi->fh)
      fuse_reply_open(req, fi);
   else
      fuse_reply_err(req, ENOTDIR);
}

static void vmfs_fuse_readdir(fuse_req_t req, fuse_ino_t ino, size_t size,
                              off_t off, struct fuse_file_info *fi)
{
   char buf[size];
   const vmfs_dirent_t *entry;
   struct stat st = {0, };
   size_t sz;

   if (!fi->fh) {
      fuse_reply_err(req, EBADF);
      return;
   }
   if ((entry = vmfs_dir_read((vmfs_dir_t *)(unsigned long)fi->fh))) {
      st.st_mode = vmfs_file_type2mode(entry->type);
      st.st_ino = blkid2ino(entry->block_id);
      sz = fuse_add_direntry(req, buf, size, entry->name, &st, off + 1);
      fuse_reply_buf(req, buf, sz);
   } else
      fuse_reply_buf(req, NULL, 0);
}

static void vmfs_fuse_releasedir(fuse_req_t req, fuse_ino_t ino,
                                 struct fuse_file_info *fi)
{
   if (!fi->fh) {
      fuse_reply_err(req, EBADF);
      return;
   }
   vmfs_dir_close((vmfs_dir_t *)(unsigned long)fi->fh);
   fuse_reply_err(req, 0);
}

static void vmfs_fuse_statfs(fuse_req_t req, fuse_ino_t ino)
{  
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);
   struct statvfs st;
   u_int alloc_count;

   memset(&st,0,sizeof(st));

   /* Blocks */
   alloc_count = vmfs_bitmap_allocated_items(fs->fbb);
   st.f_bsize = st.f_frsize = vmfs_fs_get_blocksize(fs);
   st.f_blocks = fs->fbb->bmh.total_items;
   st.f_bfree = st.f_bavail = st.f_blocks - alloc_count;
   
   /* Inodes */
   alloc_count = vmfs_bitmap_allocated_items(fs->fdc);
   st.f_files = fs->fdc->bmh.total_items;
   st.f_ffree = st.f_favail = st.f_files - alloc_count;

   fuse_reply_statfs(req,&st);
}

static void vmfs_fuse_lookup(fuse_req_t req, fuse_ino_t parent,
                             const char *name)
{
   struct fuse_entry_param entry = { 0, };
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);
   vmfs_dir_t *d = vmfs_dir_open_from_blkid(fs, ino2blkid(parent));
   const vmfs_dirent_t *rec;

   if (!d) {
      fuse_reply_err(req, ENOENT);
      return;
   }

   rec = vmfs_dir_lookup(d, name);

   if (rec && !vmfs_inode_stat_from_blkid(fs, rec->block_id, &entry.attr)) {
      entry.ino = entry.attr.st_ino = blkid2ino(rec->block_id);
      entry.generation = 1;
      entry.attr_timeout = 1.0;
      entry.entry_timeout = 1.0;
      fuse_reply_entry(req, &entry);
   } else
      fuse_reply_err(req, ENOENT);

   vmfs_dir_close(d);
}

static void vmfs_fuse_open(fuse_req_t req, fuse_ino_t ino,
                           struct fuse_file_info *fi)
{
   vmfs_fs_t *fs = (vmfs_fs_t *) fuse_req_userdata(req);

   fi->fh = (uint64_t)(unsigned long)
            vmfs_file_open_from_blkid(fs, ino2blkid(ino));
   if (fi->fh)
      fuse_reply_open(req, fi);
   else
      fuse_reply_err(req, ENOTDIR);
}

static void vmfs_fuse_create(fuse_req_t req, fuse_ino_t parent,
                             const char *name, mode_t mode,
                             struct fuse_file_info *fi)
{   
   struct fuse_entry_param entry = { 0, };
   vmfs_dir_t *dir;
   vmfs_inode_t *inode;
   vmfs_file_t *f;
   int res;

   if (!(dir = vmfs_dir_open_from_blkid(fs, ino2blkid(parent)))) {
      fuse_reply_err(req, ENOENT);
      return;
   }      

   if ((res = vmfs_file_create(dir,name,mode,&inode)) < 0) {
      fuse_reply_err(req, -res);
      return;
   }

   vmfs_dir_close(dir);

   if (!(f = vmfs_file_open_from_inode(inode))) {
      vmfs_inode_release(inode);
      fuse_reply_err(req,ENOMEM);
   }

   fi->fh = (uint64_t)(unsigned long)f;

   vmfs_inode_stat(inode,&entry.attr);
   entry.ino = entry.attr.st_ino = blkid2ino(inode->id);
   entry.generation = 1;
   entry.attr_timeout = 1.0;
   entry.entry_timeout = 1.0;
   fuse_reply_create(req,&entry,fi);
}

static void vmfs_fuse_read(fuse_req_t req, fuse_ino_t ino, size_t size,
                           off_t off, struct fuse_file_info *fi)
{
   char buf[size];
   ssize_t sz;

   if (!fi->fh) {
      fuse_reply_err(req, EBADF);
      return;
   }

   sz = vmfs_file_pread((vmfs_file_t *)(unsigned long)fi->fh,
                        (u_char *)buf, size, off);

   if (sz < 0) {
      fuse_reply_err(req, -sz);
      return;
   }

   fuse_reply_buf(req, buf, sz);
}

static void vmfs_fuse_write(fuse_req_t req, fuse_ino_t ino, 
                            const char *buf, size_t size, off_t off, 
                            struct fuse_file_info *fi) 
{
   ssize_t sz;

   if (!fi->fh) {
      fuse_reply_err(req, EBADF);
      return;
   }

   sz = vmfs_file_pwrite((vmfs_file_t *)(unsigned long)fi->fh,
                         (u_char *)buf, size, off);

   if (sz < 0) {
      fuse_reply_err(req, -sz);
      return;
   }

   fuse_reply_write(req,sz);
}

static void vmfs_fuse_release(fuse_req_t req, fuse_ino_t ino,
                              struct fuse_file_info *fi)
{
   if (!fi->fh) {
      fuse_reply_err(req, EBADF);
      return;
   }
   vmfs_file_close((vmfs_file_t *)(unsigned long)fi->fh);
   fuse_reply_err(req, 0);
}

const static struct fuse_lowlevel_ops vmfs_oper = {
   .getattr = vmfs_fuse_getattr,
   .setattr = vmfs_fuse_setattr,
   .readlink = vmfs_fuse_readlink,
   .mknod = vmfs_fuse_mknod,
   .mkdir = vmfs_fuse_mkdir,
   .unlink = vmfs_fuse_unlink,
   .rmdir = vmfs_fuse_rmdir,
   .opendir = vmfs_fuse_opendir,
   .readdir = vmfs_fuse_readdir,
   .releasedir = vmfs_fuse_releasedir,
   .statfs = vmfs_fuse_statfs,
   .lookup = vmfs_fuse_lookup,
   .open = vmfs_fuse_open,
   .create = vmfs_fuse_create,
   .read = vmfs_fuse_read,
   .write = vmfs_fuse_write,
   .release = vmfs_fuse_release,
};

struct vmfs_fuse_opts {
   char **path;
   char *paths[VMFS_LVM_MAX_EXTENTS + 1];
   char *mountpoint;
   int foreground;
};

static const struct fuse_opt vmfs_fuse_args[] = {
  { "-d", offsetof(struct vmfs_fuse_opts, foreground), 1 },
  { "-f", offsetof(struct vmfs_fuse_opts, foreground), 1 },
  FUSE_OPT_KEY("-d", FUSE_OPT_KEY_KEEP),
};

static int vmfs_fuse_opts_func(void *data, const char *arg, int key,
                struct fuse_args *outargs)
{
   struct vmfs_fuse_opts *opts = (struct vmfs_fuse_opts *) data;
   struct stat st;
   if (key == FUSE_OPT_KEY_NONOPT) {
      if (opts->mountpoint) {
         fprintf(stderr, "'%s' is not allowed here\n", arg);
         return -1;
      }
      if (stat(arg, &st)) {
         fprintf(stderr, "Error stat()ing '%s'\n", arg);
         return -1;
      }
      if (S_ISDIR(st.st_mode)) {
         opts->mountpoint = strdup(arg);
      } else if (S_ISREG(st.st_mode) || S_ISBLK(st.st_mode)) {
         *(opts->path++) = strdup(arg);
      }
      return 0;
   }
   return 1;
}

int main(int argc, char *argv[])
{
   struct fuse_args args = FUSE_ARGS_INIT(argc, argv);
   struct vmfs_fuse_opts opts = { 0, };
   struct fuse_chan *chan;
   vmfs_flags_t flags;
   int err = -1;

   flags.packed = 0;

#ifdef VMFS_WRITE
   flags.read_write = 1;
#endif

   flags.allow_missing_extents = 1;

   opts.path = &opts.paths[0];
   if ((fuse_opt_parse(&args, &opts, vmfs_fuse_args,
                       &vmfs_fuse_opts_func) == -1) ||
       (fuse_opt_add_arg(&args, "-odefault_permissions"))) {
      goto cleanup;
   }

   if (!(fs = vmfs_fs_open(opts.paths, flags))) {
      fprintf(stderr,"Unable to open filesystem\n");
      goto cleanup;
   }

   if ((chan = fuse_mount(opts.mountpoint, &args)) != NULL) {
   struct fuse_session *session;
      session = fuse_lowlevel_new(&args, &vmfs_oper,
                                  sizeof(vmfs_oper), fs);
      if (session != NULL) {
         fuse_daemonize(opts.foreground);
         if (fuse_set_signal_handlers(session) != -1) {
            fuse_session_add_chan(session, chan);
            err = fuse_session_loop(session);
            fuse_remove_signal_handlers(session);
            fuse_session_remove_chan(chan);
         }
         fuse_session_destroy(session);
      }
      fuse_unmount(opts.mountpoint, chan);
   }

cleanup:
   vmfs_fs_close(fs);
   fuse_opt_free_args(&args);
   opts.path = &opts.paths[0];
   for(;*opts.path;opts.path++)
     free(*opts.path);
   free(opts.mountpoint);

   return err ? 1 : 0;
}
