funfs/funfsmain.cc

   1 #include "wvfuse.h"
   2 #include "funfs.h"
   3 #include "cachefs.h"
   4
   5 #include <wvlogrcv.h>
   6 #include <uniconfroot.h>
   7 #include <wvistreamlist.h>
   8 #include <wvstringlist.h>
   9 #include <fileutils.h>
  10
  11 #include <signal.h>
  12 #include <getopt.h>
  13
  14 static bool want_to_die = false;
  15 static WvFuse* fuse;
  16 #define FUSE_UMOUNT_CMD_ENV  "_FUSE_UNMOUNT_CMD"
  17
  18 static void fuse_unmount()
  19 {
  20     if (!fuse || !fuse->fs_unmounted)
  21         system(getenv(FUSE_UMOUNT_CMD_ENV));
  22     close(0);
  23     delete fuse;
  24 }
  25
  26 void sighandler_die(int signum)
  27 {
  28     want_to_die = true;
  29 }
  30
  31 void sighandler_ignore(int signum)
  32 {
  33 }
  34
  35 void usage()
  36 {
  37     wvcon->print("\n");
  38     wvcon->print("Usage: funfs <remote mount point> <local mount point> [options]\n"
  39                  "Supported options:\n"
  40                  "\t-d/--debug <1-9>\n"
  41                  "\t-h/--help\n"
  42                  "\t-k/--kernel-cache[=no]  \tdefault: no\n"
  43                  "\t-m/--unmount[=no]       \tdefault: yes\n"
  44                  "\t-p/--port <int>\n"
  45                  );
  46     exit(99);
  47 }
  48
  49 int main(int argc, char *argv[])
  50 {
  51     WvLog::LogLevel lvl = WvLog::Error;
  52     signal(SIGTERM, sighandler_die);
  53     signal(SIGHUP, sighandler_die);
  54     signal(SIGINT, sighandler_die);
  55     signal(SIGUSR1, sighandler_ignore);
  56     signal(SIGPIPE, SIG_IGN);
  57
  58     fuse = NULL;
  59     bool slave = false;
  60     bool cache = false;
  61     bool umount = true;
  62     int port = 1337;
  63     atexit(fuse_unmount);
  64
  65     char *remotempoint = NULL;
  66     char *localmpoint = NULL;
  67
  68     while (1)
  69     {
  70         int option_index = 0;
  71         static struct option long_options[] = {
  72             {"help", 0, 0, 'h'},
  73             {"debug", 1, 0, 'd'},
  74             {"slave", 0, 0, 's'},
  75             {"kernel-cache", 2, 0, 'k'},
  76             {"port", 1, 0, 'p'},
  77             {"unmount", 2, 0, 'm'},
  78             {0, 0, 0, 0}
  79         };
  80
  81         int opt = getopt_long(argc, argv, "d:hk::m::p:s", long_options, &option_index);
  82
  83         if (opt == -1)
  84             break;
  85
  86         if (opt == '?' || opt == ':')
  87             usage();
  88
  89         switch (opt)
  90         {
  91             case 'd':
  92                 {
  93                     int loglevel = atoi(optarg);
  94                     if (loglevel < 1 || loglevel > 9)
  95                     {
  96                         wvcon->print("%s: Invalid debug level.\n", argv[0]);
  97                         usage();
  98                     }
  99                     lvl = (WvLog::LogLevel)(loglevel - 1);
 100                 }
 101                 break;
 102
 103             case 'h':
 104                 usage();
 105                 break;
 106
 107             case 'k':
 108                 if (optarg)
 109                 {
 110                     cache = !strstr(optarg, "no");
 111                 }
 112                 else
 113                     cache = true;
 114                 break;
 115
 116
 117             case 'p':
 118                 port = atoi(optarg);
 119                 if (!port || port < 1 || port > 65536)
 120                 {
 121                     wvcon->print("%s: Invalid port %s\n", argv[0], optarg);
 122                     usage();
 123                 }
 124
 125             case 's':
 126                 slave = true;
 127                 break;
 128
 129             case 'u':
 130                 if (optarg)
 131                     umount = !strstr(optarg, "no");
 132                 else
 133                     umount = true;
 134                 break;
 135
 136             default:
 137                 break;
 138         }
 139     }
 140
 141     if (argc - optind < 2 - !!slave)
 142     {
 143         usage();
 144     }
 145     else if (argc - optind > 2)
 146     {
 147         optind += 2;
 148         wvcon->print("%s: Unrecognized paramater(s): ", argv[0]);
 149         while (optind < argc)
 150             wvcon->print("%s ", argv[optind++]);
 151         wvcon->print("\n");
 152
 153         usage();
 154     }
 155
 156     remotempoint = argv[optind++];
 157
 158     if (!slave)
 159     {
 160         localmpoint = argv[optind++];
 161
 162         if (umount)
 163             system(WvString("fusermount -u %s 2>/dev/null", localmpoint));
 164
 165         const char *params[11];
 166         int next = 0;
 167
 168         WvString strlvl(lvl);
 169         WvString strport(port);
 170
 171         params[next++] = "fusermount";
 172         if (cache)
 173             params[next++] = "-c";
 174         params[next++] = localmpoint;
 175         params[next++] = argv[0];
 176         params[next++] = remotempoint;
 177         params[next++] = "--slave";
 178         params[next++] = "--debug";
 179         params[next++] = strlvl;
 180         params[next++] = "--port";
 181         params[next++] = strport;
 182         params[next++] = NULL;
 183
 184         execvp("fusermount", (char* const*)params);
 185
 186         fprintf(stderr, "%s: fusermount: %s\n", argv[0], strerror(errno));
 187         return errno;
 188     }
 189
 190     wvcon->print("\n");
 191
 192     WvLogConsole logdisp(2, lvl);
 193
 194     WvStringList lst;
 195     lst.split(remotempoint, ":", 2);
 196
 197     if (lst.count() != 2)
 198     {
 199         wvcon->print("funfs: Bad mount point %s: expected format 'host:/path'\n",
 200                 remotempoint);
 201         return 10;
 202     }
 203
 204     UniConfRoot cfg("temp");
 205
 206     UniConf funfscfg = cfg["FunFS"];
 207     funfscfg["Hostname"].set(lst.popstr());
 208     funfscfg["Remote Mount Point"].set(lst.popstr());
 209     funfscfg["Port"].setint(port);
 210     funfscfg["Local Mount Point"].set(localmpoint);
 211
 212     UniConf cachefscfg = cfg["CacheFS"];
 213     cachefscfg["Cache Root"].set("cache");
 214     cachefscfg["Cache Size"].setint(5 * 1024); // Megabytes
 215
 216     cachefscfg["DB Location"].set("%s/%s", cachefscfg["Cache Root"].get(),
 217             getpid());
 218
 219     mkdirp(cachefscfg["DB Location"].get());
 220     system(WvString("rm -f %s/*", cachefscfg["DB Location"].get()));
 221
 222     cachefscfg["Meta Root"].set("%s/meta/%s/%s", cachefscfg["Cache Root"].get(),
 223             funfscfg["Hostname"].get(), funfscfg["Remote Mount Point"].get());
 224     cachefscfg["Data Root"].set("%s/data/%s/%s", cachefscfg["Cache Root"].get(),
 225             funfscfg["Hostname"].get(), funfscfg["Remote Mount Point"].get());
 226
 227     UniConf fusecfg = cfg["WvFuse"];
 228     fusecfg["DB Location"].set(cachefscfg["DB Location"].get());
 229
 230     FunFS *funfs = new FunFS(funfscfg);
 231
 232     fuse = new WvFuse(new CacheFS(funfs, cachefscfg), fusecfg);
 233     WvIStreamList::globallist.append(fuse, false);
 234
 235     while (fuse->isok() && funfs->isok() && !want_to_die)
 236     {
 237         WvIStreamList::globallist.select(-1);
 238     }
 239
 240     WvIStreamList::globallist.zap();
 241     system(WvString("rm -f %s/*", cachefscfg["DB Location"].get()));
 242     rmdir(cachefscfg["DB Location"].get());
 243
 244     return 0;
 245 }