lustre/utils/mount_utils.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  *
  30  * Copyright (c) 2012, 2014, Intel Corporation.
  31  */
  32 /*
  33  * This file is part of Lustre, http://www.lustre.org/
  34  * Lustre is a trademark of Sun Microsystems, Inc.
  35  */
  36
  37 #if HAVE_CONFIG_H
  38 #  include "config.h"
  39 #endif /* HAVE_CONFIG_H */
  40
  41 #include "mount_utils.h"
  42 #include <mntent.h>
  43 #include <stdio.h>
  44 #include <errno.h>
  45 #include <string.h>
  46 #include <unistd.h>
  47 #include <config.h>
  48 #include <lustre_disk.h>
  49 #include <lustre_ver.h>
  50 #include <sys/mount.h>
  51 #include <sys/stat.h>
  52 #include <sys/utsname.h>
  53 #include <linux/loop.h>
  54 #include <dlfcn.h>
  55
  56 extern char *progname;
  57 extern int verbose;
  58
  59 #define vprint(fmt, arg...) if (verbose > 0) printf(fmt, ##arg)
  60 #define verrprint(fmt, arg...) if (verbose >= 0) fprintf(stderr, fmt, ##arg)
  61
  62 static struct module_backfs_ops *backfs_ops[LDD_MT_LAST];
  63
  64 void fatal(void)
  65 {
  66         verbose = 0;
  67         fprintf(stderr, "\n%s FATAL: ", progname);
  68 }
  69
  70 int run_command(char *cmd, int cmdsz)
  71 {
  72         char log[] = "/tmp/run_command_logXXXXXX";
  73         int fd = -1, rc;
  74
  75         if ((cmdsz - strlen(cmd)) < 6) {
  76                 fatal();
  77                 fprintf(stderr, "Command buffer overflow: %.*s...\n",
  78                         cmdsz, cmd);
  79                 return ENOMEM;
  80         }
  81
  82         if (verbose > 1) {
  83                 printf("cmd: %s\n", cmd);
  84         } else {
  85                 if ((fd = mkstemp(log)) >= 0) {
  86                         close(fd);
  87                         strcat(cmd, " >");
  88                         strcat(cmd, log);
  89                 }
  90         }
  91         strcat(cmd, " 2>&1");
  92
  93         /* Can't use popen because we need the rv of the command */
  94         rc = system(cmd);
  95         if (rc && (fd >= 0)) {
  96                 char buf[128];
  97                 FILE *fp;
  98                 fp = fopen(log, "r");
  99                 if (fp) {
 100                         while (fgets(buf, sizeof(buf), fp) != NULL) {
 101                                 printf("   %s", buf);
 102                         }
 103                         fclose(fp);
 104                 }
 105         }
 106         if (fd >= 0)
 107                 remove(log);
 108         return rc;
 109 }
 110
 111 int add_param(char *buf, char *key, char *val)
 112 {
 113         int end = sizeof(((struct lustre_disk_data *)0)->ldd_params);
 114         int start = strlen(buf);
 115         int keylen = 0;
 116
 117         if (key)
 118                 keylen = strlen(key);
 119         if (start + 1 + keylen + strlen(val) >= end) {
 120                 fprintf(stderr, "%s: params are too long-\n%s %s%s\n",
 121                         progname, buf, key ? key : "", val);
 122                 return 1;
 123         }
 124
 125         sprintf(buf + start, " %s%s", key ? key : "", val);
 126         return 0;
 127 }
 128
 129 int get_param(char *buf, char *key, char **val)
 130 {
 131         int i, key_len = strlen(key);
 132         char *ptr;
 133
 134         ptr = strstr(buf, key);
 135         if (ptr) {
 136                 *val = strdup(ptr + key_len);
 137                 if (*val == NULL)
 138                         return ENOMEM;
 139
 140                 for (i = 0; i < strlen(*val); i++)
 141                         if (((*val)[i] == ' ') || ((*val)[i] == '\0'))
 142                                 break;
 143
 144                 (*val)[i] = '\0';
 145                 return 0;
 146         }
 147
 148         return ENOENT;
 149 }
 150
 151 int append_param(char *buf, char *key, char *val, char sep)
 152 {
 153         int key_len, i, offset, old_val_len;
 154         char *ptr = NULL, str[1024];
 155
 156         if (key)
 157                 ptr = strstr(buf, key);
 158
 159         /* key doesn't exist yet, so just add it */
 160         if (ptr == NULL)
 161                 return add_param(buf, key, val);
 162
 163         key_len = strlen(key);
 164
 165         /* Copy previous values to str */
 166         for (i = 0; i < sizeof(str); ++i) {
 167                 if ((ptr[i+key_len] == ' ') || (ptr[i+key_len] == '\0'))
 168                         break;
 169                 str[i] = ptr[i+key_len];
 170         }
 171         if (i == sizeof(str))
 172                 return E2BIG;
 173         old_val_len = i;
 174
 175         offset = old_val_len+key_len;
 176
 177         /* Move rest of buf to overwrite previous key and value */
 178         for (i = 0; ptr[i+offset] != '\0'; ++i)
 179                 ptr[i] = ptr[i+offset];
 180
 181         ptr[i] = '\0';
 182
 183         snprintf(str+old_val_len, sizeof(str)-old_val_len, "%c%s", sep, val);
 184
 185         return add_param(buf, key, str);
 186 }
 187
 188 char *strscat(char *dst, char *src, int buflen)
 189 {
 190         dst[buflen - 1] = 0;
 191         if (strlen(dst) + strlen(src) >= buflen) {
 192                 fprintf(stderr, "string buffer overflow (max %d): '%s' + '%s'"
 193                         "\n", buflen, dst, src);
 194                 exit(EOVERFLOW);
 195         }
 196         return strcat(dst, src);
 197 }
 198
 199 char *strscpy(char *dst, char *src, int buflen)
 200 {
 201         dst[0] = 0;
 202         return strscat(dst, src, buflen);
 203 }
 204
 205 int check_mtab_entry(char *spec1, char *spec2, char *mtpt, char *type)
 206 {
 207         FILE *fp;
 208         struct mntent *mnt;
 209
 210         fp = setmntent(MOUNTED, "r");
 211         if (fp == NULL)
 212                 return 0;
 213
 214         while ((mnt = getmntent(fp)) != NULL) {
 215                 if ((strcmp(mnt->mnt_fsname, spec1) == 0 ||
 216                      strcmp(mnt->mnt_fsname, spec2) == 0) &&
 217                     (mtpt == NULL || strcmp(mnt->mnt_dir, mtpt) == 0) &&
 218                     (type == NULL || strcmp(mnt->mnt_type, type) == 0)) {
 219                         endmntent(fp);
 220                         return(EEXIST);
 221                 }
 222         }
 223         endmntent(fp);
 224
 225         return 0;
 226 }
 227
 228 #define PROC_DIR        "/proc/"
 229 static int mtab_is_proc(const char *mtab)
 230 {
 231         char path[16];
 232
 233         if (readlink(mtab, path, sizeof(path)) < 0)
 234                 return 0;
 235
 236         if (strncmp(path, PROC_DIR, strlen(PROC_DIR)))
 237                 return 0;
 238
 239         return 1;
 240 }
 241
 242 int update_mtab_entry(char *spec, char *mtpt, char *type, char *opts,
 243                 int flags, int freq, int pass)
 244 {
 245         FILE *fp;
 246         struct mntent mnt;
 247         int rc = 0;
 248
 249         /* Don't update mtab if it is linked to any file in /proc direcotry.*/
 250         if (mtab_is_proc(MOUNTED))
 251                 return 0;
 252
 253         mnt.mnt_fsname = spec;
 254         mnt.mnt_dir = mtpt;
 255         mnt.mnt_type = type;
 256         mnt.mnt_opts = opts ? opts : "";
 257         mnt.mnt_freq = freq;
 258         mnt.mnt_passno = pass;
 259
 260         fp = setmntent(MOUNTED, "a+");
 261         if (fp == NULL) {
 262                 fprintf(stderr, "%s: setmntent(%s): %s:",
 263                         progname, MOUNTED, strerror (errno));
 264                 rc = 16;
 265         } else {
 266                 if ((addmntent(fp, &mnt)) == 1) {
 267                         fprintf(stderr, "%s: addmntent: %s:",
 268                                 progname, strerror (errno));
 269                         rc = 16;
 270                 }
 271                 endmntent(fp);
 272         }
 273
 274         return rc;
 275 }
 276
 277 /* Search for opt in mntlist, returning true if found.
 278  */
 279 static int in_mntlist(char *opt, char *mntlist)
 280 {
 281         char *ml, *mlp, *item, *ctx = NULL;
 282
 283         if (!(ml = strdup(mntlist))) {
 284                 fprintf(stderr, "%s: out of memory\n", progname);
 285                 exit(1);
 286         }
 287         mlp = ml;
 288         while ((item = strtok_r(mlp, ",", &ctx))) {
 289                 if (!strcmp(opt, item))
 290                         break;
 291                 mlp = NULL;
 292         }
 293         free(ml);
 294         return (item != NULL);
 295 }
 296
 297 /* Issue a message on stderr for every item in wanted_mountopts that is not
 298  * present in mountopts.  The justwarn boolean toggles between error and
 299  * warning message.  Return an error count.
 300  */
 301 int check_mountfsoptions(char *mountopts, char *wanted_mountopts)
 302 {
 303         char *ml, *mlp, *item, *ctx = NULL;
 304         int errors = 0;
 305
 306         if (!(ml = strdup(wanted_mountopts))) {
 307                 fprintf(stderr, "%s: out of memory\n", progname);
 308                 exit(1);
 309         }
 310         mlp = ml;
 311         while ((item = strtok_r(mlp, ",", &ctx))) {
 312                 if (!in_mntlist(item, mountopts)) {
 313                         fprintf(stderr, "%s: Error: mandatory mount option"
 314                                 " '%s' is missing\n", progname, item);
 315                         errors++;
 316                 }
 317                 mlp = NULL;
 318         }
 319         free(ml);
 320         return errors;
 321 }
 322
 323 /* Trim embedded white space, leading and trailing commas from string s.
 324  */
 325 void trim_mountfsoptions(char *s)
 326 {
 327         char *p;
 328
 329         for (p = s; *p; ) {
 330                 if (isspace(*p)) {
 331                         memmove(p, p + 1, strlen(p + 1) + 1);
 332                         continue;
 333                 }
 334                 p++;
 335         }
 336
 337         while (s[0] == ',')
 338                 memmove(&s[0], &s[1], strlen(&s[1]) + 1);
 339
 340         p = s + strlen(s) - 1;
 341         while (p >= s && *p == ',')
 342                 *p-- = '\0';
 343 }
 344
 345 /* Setup a file in the first unused loop_device */
 346 int loop_setup(struct mkfs_opts *mop)
 347 {
 348         char loop_base[20];
 349         char l_device[64];
 350         int i, ret = 0;
 351
 352         /* Figure out the loop device names */
 353         if (!access("/dev/loop0", F_OK | R_OK) ||
 354             !access("/dev/loop-control", F_OK | R_OK)) {
 355                 strcpy(loop_base, "/dev/loop\0");
 356         } else if (!access("/dev/loop/0", F_OK | R_OK)) {
 357                 strcpy(loop_base, "/dev/loop/\0");
 358         } else {
 359                 fprintf(stderr, "%s: can't access loop devices\n", progname);
 360                 return EACCES;
 361         }
 362
 363         /* Find unused loop device */
 364         for (i = 0; i < MAX_LOOP_DEVICES; i++) {
 365                 char cmd[PATH_MAX];
 366                 int cmdsz = sizeof(cmd);
 367
 368 #ifdef LOOP_CTL_GET_FREE
 369                 ret = open("/dev/loop-control", O_RDWR);
 370                 if (ret < 0) {
 371                         fprintf(stderr, "%s: can't access loop control\n", progname);
 372                         return EACCES;
 373                 }
 374                 /* find or allocate a free loop device to use */
 375                 i = ioctl(ret, LOOP_CTL_GET_FREE);
 376                 if (i < 0) {
 377                         fprintf(stderr, "%s: access loop control error\n", progname);
 378                         return EACCES;
 379                 }
 380                 sprintf(l_device, "%s%d", loop_base, i);
 381 #else
 382                 sprintf(l_device, "%s%d", loop_base, i);
 383                 if (access(l_device, F_OK | R_OK))
 384                         break;
 385 #endif
 386                 snprintf(cmd, cmdsz, "losetup %s > /dev/null 2>&1", l_device);
 387                 ret = system(cmd);
 388
 389                 /* losetup gets 1 (ret=256) for non-set-up device */
 390                 if (ret) {
 391                         /* Set up a loopback device to our file */
 392                         snprintf(cmd, cmdsz, "losetup %s %s", l_device,
 393                                  mop->mo_device);
 394                         ret = run_command(cmd, cmdsz);
 395                         if (ret == 256)
 396                                 /* someone else picked up this loop device
 397                                  * behind our back */
 398                                 continue;
 399                         if (ret) {
 400                                 fprintf(stderr, "%s: error %d on losetup: %s\n",
 401                                         progname, ret,
 402                                         ret >= 0 ? strerror(ret) : "");
 403                                 return ret;
 404                         }
 405                         strscpy(mop->mo_loopdev, l_device,
 406                                 sizeof(mop->mo_loopdev));
 407                         return ret;
 408                 }
 409         }
 410
 411         fprintf(stderr, "%s: out of loop devices!\n", progname);
 412         return EMFILE;
 413 }
 414
 415 int loop_cleanup(struct mkfs_opts *mop)
 416 {
 417         char cmd[150];
 418         int ret = 0;
 419
 420         if ((mop->mo_flags & MO_IS_LOOP) && *mop->mo_loopdev) {
 421                 int tries;
 422
 423                 sprintf(cmd, "losetup -d %s", mop->mo_loopdev);
 424                 for (tries = 0; tries < 3; tries++) {
 425                         ret = run_command(cmd, sizeof(cmd));
 426                         if (ret == 0)
 427                                 break;
 428                         sleep(1);
 429                 }
 430         }
 431
 432         if (ret != 0)
 433                 fprintf(stderr, "cannot cleanup %s: rc = %d\n",
 434                         mop->mo_loopdev, ret);
 435         return ret;
 436 }
 437
 438 int loop_format(struct mkfs_opts *mop)
 439 {
 440         int fd;
 441
 442         if (mop->mo_device_kb == 0) {
 443                 fatal();
 444                 fprintf(stderr, "loop device requires a --device-size= "
 445                         "param\n");
 446                 return EINVAL;
 447         }
 448
 449         fd = creat(mop->mo_device, S_IRUSR|S_IWUSR);
 450         if (fd < 0) {
 451                 fatal();
 452                 fprintf(stderr, "%s: Unable to create backing store: %s\n",
 453                         progname, strerror(errno));
 454                 return errno;
 455         }
 456
 457         if (ftruncate(fd, mop->mo_device_kb * 1024) != 0) {
 458                 close(fd);
 459                 fatal();
 460                 fprintf(stderr, "%s: Unable to truncate backing store: %s\n",
 461                         progname, strerror(errno));
 462                 return errno;
 463         }
 464
 465         close(fd);
 466         return 0;
 467 }
 468
 469 #define DLSYM(prefix, sym, func)                                        \
 470         do {                                                            \
 471                 char _fname[64];                                        \
 472                 snprintf(_fname, sizeof(_fname), "%s_%s", prefix, #func); \
 473                 sym->func = (typeof(sym->func))dlsym(sym->dl_handle, _fname); \
 474         } while (0)
 475
 476 /**
 477  * Load plugin for a given mount_type from ${pkglibdir}/mount_osd_FSTYPE.so and
 478  * return struct of function pointers (will be freed in unloack_backfs_module).
 479  *
 480  * \param[in] mount_type        Mount type to load module for.
 481  * \retval Value of backfs_ops struct
 482  * \retval NULL if no module exists
 483  */
 484 struct module_backfs_ops *load_backfs_module(enum ldd_mount_type mount_type)
 485 {
 486         void *handle;
 487         char *error, filename[512], fsname[512], *name;
 488         struct module_backfs_ops *ops;
 489
 490         /* This deals with duplicate ldd_mount_types resolving to same OSD layer
 491          * plugin (e.g. ext3/ldiskfs/ldiskfs2 all being ldiskfs) */
 492         strncpy(fsname, mt_type(mount_type), sizeof(fsname));
 493         name = fsname + sizeof("osd-") - 1;
 494
 495         /* change osd- to osd_ */
 496         fsname[sizeof("osd-") - 2] = '_';
 497
 498         snprintf(filename, sizeof(filename), PLUGIN_DIR"/mount_%s.so", fsname);
 499
 500         handle = dlopen(filename, RTLD_LAZY);
 501
 502         /* Check for $LUSTRE environment variable from test-framework.
 503          * This allows using locally built modules to be used.
 504          */
 505         if (handle == NULL) {
 506                 char *dirname;
 507                 dirname = getenv("LUSTRE");
 508                 if (dirname) {
 509                         snprintf(filename, sizeof(filename),
 510                                  "%s/utils/.libs/mount_%s.so",
 511                                  dirname, fsname);
 512                         handle = dlopen(filename, RTLD_LAZY);
 513                 }
 514         }
 515
 516         /* Do not clutter up console with missing types */
 517         if (handle == NULL)
 518                 return NULL;
 519
 520         ops = malloc(sizeof(*ops));
 521         if (ops == NULL) {
 522                 dlclose(handle);
 523                 return NULL;
 524         }
 525
 526         ops->dl_handle = handle;
 527         dlerror(); /* Clear any existing error */
 528
 529         DLSYM(name, ops, init);
 530         DLSYM(name, ops, fini);
 531         DLSYM(name, ops, read_ldd);
 532         DLSYM(name, ops, write_ldd);
 533         DLSYM(name, ops, is_lustre);
 534         DLSYM(name, ops, make_lustre);
 535         DLSYM(name, ops, prepare_lustre);
 536         DLSYM(name, ops, tune_lustre);
 537         DLSYM(name, ops, label_lustre);
 538         DLSYM(name, ops, enable_quota);
 539
 540         error = dlerror();
 541         if (error != NULL) {
 542                 fatal();
 543                 fprintf(stderr, "%s\n", error);
 544                 dlclose(handle);
 545                 free(ops);
 546                 return NULL;
 547         }
 548
 549         /* optional methods */
 550         DLSYM(name, ops, fix_mountopts);
 551
 552         return ops;
 553 }
 554
 555 /**
 556  * Unload plugin and free backfs_ops structure. Must be called the same number
 557  * of times as load_backfs_module is.
 558  */
 559 void unload_backfs_module(struct module_backfs_ops *ops)
 560 {
 561         if (ops == NULL)
 562                 return;
 563
 564         dlclose(ops->dl_handle);
 565         free(ops);
 566 }
 567
 568 /* Return true if backfs_ops has operations for the given mount_type. */
 569 int backfs_mount_type_okay(enum ldd_mount_type mount_type)
 570 {
 571         if (unlikely(mount_type >= LDD_MT_LAST || mount_type < 0)) {
 572                 fatal();
 573                 fprintf(stderr, "fs type out of range %d\n", mount_type);
 574                 return 0;
 575         }
 576         if (backfs_ops[mount_type] == NULL) {
 577                 fatal();
 578                 fprintf(stderr, "unhandled/unloaded fs type %d '%s'\n",
 579                         mount_type, mt_str(mount_type));
 580                 return 0;
 581         }
 582         return 1;
 583 }
 584
 585 /* Write the server config files */
 586 int osd_write_ldd(struct mkfs_opts *mop)
 587 {
 588         struct lustre_disk_data *ldd = &mop->mo_ldd;
 589         int ret;
 590
 591         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 592                 ret = backfs_ops[ldd->ldd_mount_type]->write_ldd(mop);
 593
 594         else
 595                 ret = EINVAL;
 596
 597         return ret;
 598 }
 599
 600 /* Read the server config files */
 601 int osd_read_ldd(char *dev, struct lustre_disk_data *ldd)
 602 {
 603         int ret;
 604
 605         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 606                 ret = backfs_ops[ldd->ldd_mount_type]->read_ldd(dev, ldd);
 607
 608         else
 609                 ret = EINVAL;
 610
 611         return ret;
 612 }
 613
 614 /* Was this device formatted for Lustre */
 615 int osd_is_lustre(char *dev, unsigned *mount_type)
 616 {
 617         int i;
 618
 619         vprint("checking for existing Lustre data: ");
 620
 621         for (i = 0; i < LDD_MT_LAST; ++i) {
 622                 if (backfs_ops[i] != NULL &&
 623                     backfs_ops[i]->is_lustre(dev, mount_type)) {
 624                         vprint("found\n");
 625                         return 1;
 626                 }
 627         }
 628
 629         vprint("not found\n");
 630         return 0;
 631 }
 632
 633 /* Build fs according to type */
 634 int osd_make_lustre(struct mkfs_opts *mop)
 635 {
 636         struct lustre_disk_data *ldd = &mop->mo_ldd;
 637         int ret;
 638
 639         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 640                 ret = backfs_ops[ldd->ldd_mount_type]->make_lustre(mop);
 641
 642         else
 643                 ret = EINVAL;
 644
 645         return ret;
 646 }
 647
 648 int osd_prepare_lustre(struct mkfs_opts *mop,
 649                        char *wanted_mountopts, size_t len)
 650 {
 651         struct lustre_disk_data *ldd = &mop->mo_ldd;
 652         int ret;
 653
 654         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 655                 ret = backfs_ops[ldd->ldd_mount_type]->prepare_lustre(mop,
 656                                                         wanted_mountopts, len);
 657
 658         else
 659                 ret = EINVAL;
 660
 661         return ret;
 662 }
 663
 664 int osd_fix_mountopts(struct mkfs_opts *mop, char *mountopts, size_t len)
 665 {
 666         struct lustre_disk_data *ldd = &mop->mo_ldd;
 667
 668         if (!backfs_mount_type_okay(ldd->ldd_mount_type))
 669                 return EINVAL;
 670
 671         if (backfs_ops[ldd->ldd_mount_type]->fix_mountopts == NULL)
 672                 return 0;
 673
 674         return backfs_ops[ldd->ldd_mount_type]->fix_mountopts(mop, mountopts,
 675                                                               len);
 676 }
 677
 678 int osd_tune_lustre(char *dev, struct mount_opts *mop)
 679 {
 680         struct lustre_disk_data *ldd = &mop->mo_ldd;
 681         int ret;
 682
 683         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 684                 ret = backfs_ops[ldd->ldd_mount_type]->tune_lustre(dev, mop);
 685
 686         else
 687                 ret = EINVAL;
 688
 689         return ret;
 690 }
 691
 692 int osd_label_lustre(struct mount_opts *mop)
 693 {
 694         struct lustre_disk_data *ldd = &mop->mo_ldd;
 695         int ret;
 696
 697         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 698                 ret = backfs_ops[ldd->ldd_mount_type]->label_lustre(mop);
 699
 700         else
 701                 ret = EINVAL;
 702
 703         return ret;
 704 }
 705
 706 /* Enable quota accounting */
 707 int osd_enable_quota(struct mkfs_opts *mop)
 708 {
 709         struct lustre_disk_data *ldd = &mop->mo_ldd;
 710         int ret;
 711
 712         if (backfs_mount_type_okay(ldd->ldd_mount_type))
 713                 ret = backfs_ops[ldd->ldd_mount_type]->enable_quota(mop);
 714
 715         else
 716                 ret = EINVAL;
 717
 718         return ret;
 719 }
 720
 721 int osd_init(void)
 722 {
 723         int i, rc, ret = EINVAL;
 724
 725         for (i = 0; i < LDD_MT_LAST; ++i) {
 726                 rc = 0;
 727                 backfs_ops[i] = load_backfs_module(i);
 728                 if (backfs_ops[i] != NULL)
 729                         rc = backfs_ops[i]->init();
 730                 if (rc != 0) {
 731                         backfs_ops[i]->fini();
 732                         unload_backfs_module(backfs_ops[i]);
 733                         backfs_ops[i] = NULL;
 734                 } else
 735                         ret = 0;
 736         }
 737
 738         return ret;
 739 }
 740
 741 void osd_fini(void)
 742 {
 743         int i;
 744
 745         for (i = 0; i < LDD_MT_LAST; ++i) {
 746                 if (backfs_ops[i] != NULL) {
 747                         backfs_ops[i]->fini();
 748                         unload_backfs_module(backfs_ops[i]);
 749                         backfs_ops[i] = NULL;
 750                 }
 751         }
 752 }
 753
 754 __u64 get_device_size(char* device)
 755 {
 756         int ret, fd;
 757         __u64 size = 0;
 758
 759         fd = open(device, O_RDONLY);
 760         if (fd < 0) {
 761                 fprintf(stderr, "%s: cannot open %s: %s\n",
 762                         progname, device, strerror(errno));
 763                 return 0;
 764         }
 765
 766 #ifdef BLKGETSIZE64
 767         /* size in bytes. bz5831 */
 768         ret = ioctl(fd, BLKGETSIZE64, (void*)&size);
 769 #else
 770         {
 771                 __u32 lsize = 0;
 772                 /* size in blocks */
 773                 ret = ioctl(fd, BLKGETSIZE, (void*)&lsize);
 774                 size = (__u64)lsize * 512;
 775         }
 776 #endif
 777         close(fd);
 778         if (ret < 0) {
 779                 fprintf(stderr, "%s: size ioctl failed: %s\n",
 780                         progname, strerror(errno));
 781                 return 0;
 782         }
 783
 784         vprint("device size = "LPU64"MB\n", size >> 20);
 785         /* return value in KB */
 786         return size >> 10;
 787 }
 788
 789 int file_create(char *path, __u64 size)
 790 {
 791         __u64 size_max;
 792         int ret;
 793         int fd;
 794
 795         /*
 796          * Since "size" is in KB, the file offset it represents could overflow
 797          * off_t.
 798          */
 799         size_max = (off_t)1 << (_FILE_OFFSET_BITS - 1 - 10);
 800         if (size >= size_max) {
 801                 fprintf(stderr, "%s: "LPU64" KB: Backing store size must be "
 802                         "smaller than "LPU64" KB\n", progname, size, size_max);
 803                 return EFBIG;
 804         }
 805
 806         ret = access(path, F_OK);
 807         if (ret == 0) {
 808                 ret = unlink(path);
 809                 if (ret != 0)
 810                         return errno;
 811         }
 812
 813         fd = creat(path, S_IRUSR|S_IWUSR);
 814         if (fd < 0) {
 815                 fatal();
 816                 fprintf(stderr, "%s: Unable to create backing store: %s\n",
 817                         progname, strerror(errno));
 818                 return errno;
 819         }
 820
 821         ret = ftruncate(fd, size * 1024);
 822         close(fd);
 823         if (ret != 0) {
 824                 fatal();
 825                 fprintf(stderr, "%s: Unable to truncate backing store: %s\n",
 826                         progname, strerror(errno));
 827                 return errno;
 828         }
 829
 830         return 0;
 831 }