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