Whamcloud - gitweb
828874d3e817424c56d265c72d7bc194b6bb16d8
[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 "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 #ifdef HAVE_LIBMOUNT
243
244 # include <libmount/libmount.h>
245
246 /*
247  * The libmount is part of util-linux since 2.18.
248  * We use it to update utab to avoid umount would
249  * blocked in some rare case.
250  */
251 int update_utab_entry(struct mount_opts *mop)
252 {
253         struct libmnt_fs *fs = mnt_new_fs();
254         struct libmnt_update *upd;
255         int rc;
256
257         mnt_fs_set_source(fs, mop->mo_source);
258         mnt_fs_set_target(fs, mop->mo_target);
259         mnt_fs_set_fstype(fs, "lustre");
260         mnt_fs_set_attributes(fs, "lustre");
261
262         upd = mnt_new_update();
263         if (!upd)
264                 return -ENOMEM;
265
266         rc = mnt_update_set_fs(upd, mop->mo_nomtab ? MS_REMOUNT : 0, NULL, fs);
267         if (rc == 1) /* update is unnecessary */
268                 rc = 0;
269         if (rc) {
270                 fprintf(stderr,
271                         "error: failed to save utab entry: rc = %d\n", rc);
272         } else {
273                 rc = mnt_update_table(upd, NULL);
274         }
275
276         mnt_free_update(upd);
277         mnt_free_fs(fs);
278
279         return rc;
280 }
281 #else
282 int update_utab_entry(struct mount_opts *mop)
283 {
284         return 0;
285 }
286 #endif /* HAVE_LIBMOUNT */
287
288 int update_mtab_entry(char *spec, char *mtpt, char *type, char *opts,
289                 int flags, int freq, int pass)
290 {
291         FILE *fp;
292         struct mntent mnt;
293         int rc = 0;
294
295         /* Don't update mtab if it is linked to any file in /proc direcotry.*/
296         if (mtab_is_proc(MOUNTED))
297                 return 0;
298
299         mnt.mnt_fsname = spec;
300         mnt.mnt_dir = mtpt;
301         mnt.mnt_type = type;
302         mnt.mnt_opts = opts ? opts : "";
303         mnt.mnt_freq = freq;
304         mnt.mnt_passno = pass;
305
306         fp = setmntent(MOUNTED, "a+");
307         if (fp == NULL) {
308                 fprintf(stderr, "%s: setmntent(%s): %s:",
309                         progname, MOUNTED, strerror (errno));
310                 rc = 16;
311         } else {
312                 if ((addmntent(fp, &mnt)) == 1) {
313                         fprintf(stderr, "%s: addmntent: %s:",
314                                 progname, strerror (errno));
315                         rc = 16;
316                 }
317                 endmntent(fp);
318         }
319
320         return rc;
321 }
322
323 /* Search for opt in mntlist, returning true if found.
324  */
325 static int in_mntlist(char *opt, char *mntlist)
326 {
327         char *ml, *mlp, *item, *ctx = NULL;
328
329         if (!(ml = strdup(mntlist))) {
330                 fprintf(stderr, "%s: out of memory\n", progname);
331                 exit(1);
332         }
333         mlp = ml;
334         while ((item = strtok_r(mlp, ",", &ctx))) {
335                 if (!strcmp(opt, item))
336                         break;
337                 mlp = NULL;
338         }
339         free(ml);
340         return (item != NULL);
341 }
342
343 /* Issue a message on stderr for every item in wanted_mountopts that is not
344  * present in mountopts.  The justwarn boolean toggles between error and
345  * warning message.  Return an error count.
346  */
347 int check_mountfsoptions(char *mountopts, char *wanted_mountopts)
348 {
349         char *ml, *mlp, *item, *ctx = NULL;
350         int errors = 0;
351
352         if (!(ml = strdup(wanted_mountopts))) {
353                 fprintf(stderr, "%s: out of memory\n", progname);
354                 exit(1);
355         }
356         mlp = ml;
357         while ((item = strtok_r(mlp, ",", &ctx))) {
358                 if (!in_mntlist(item, mountopts)) {
359                         fprintf(stderr, "%s: Error: mandatory mount option"
360                                 " '%s' is missing\n", progname, item);
361                         errors++;
362                 }
363                 mlp = NULL;
364         }
365         free(ml);
366         return errors;
367 }
368
369 /* Trim embedded white space, leading and trailing commas from string s.
370  */
371 void trim_mountfsoptions(char *s)
372 {
373         char *p;
374
375         for (p = s; *p; ) {
376                 if (isspace(*p)) {
377                         memmove(p, p + 1, strlen(p + 1) + 1);
378                         continue;
379                 }
380                 p++;
381         }
382
383         while (s[0] == ',')
384                 memmove(&s[0], &s[1], strlen(&s[1]) + 1);
385
386         p = s + strlen(s) - 1;
387         while (p >= s && *p == ',')
388                 *p-- = '\0';
389 }
390
391 /* Setup a file in the first unused loop_device */
392 int loop_setup(struct mkfs_opts *mop)
393 {
394         char loop_base[20];
395         char l_device[64];
396         int i, ret = 0;
397
398         /* Figure out the loop device names */
399         if (!access("/dev/loop0", F_OK | R_OK) ||
400             !access("/dev/loop-control", F_OK | R_OK)) {
401                 strcpy(loop_base, "/dev/loop\0");
402         } else if (!access("/dev/loop/0", F_OK | R_OK)) {
403                 strcpy(loop_base, "/dev/loop/\0");
404         } else {
405                 fprintf(stderr, "%s: can't access loop devices\n", progname);
406                 return EACCES;
407         }
408
409         /* Find unused loop device */
410         for (i = 0; i < MAX_LOOP_DEVICES; i++) {
411                 char cmd[PATH_MAX];
412                 int cmdsz = sizeof(cmd);
413
414 #ifdef HAVE_LOOP_CTL_GET_FREE
415                 ret = open("/dev/loop-control", O_RDWR);
416                 if (ret < 0) {
417                         fprintf(stderr, "%s: can't access loop control\n", progname);
418                         return EACCES;
419                 }
420                 /* find or allocate a free loop device to use */
421                 i = ioctl(ret, LOOP_CTL_GET_FREE);
422                 if (i < 0) {
423                         fprintf(stderr, "%s: access loop control error\n", progname);
424                         return EACCES;
425                 }
426                 sprintf(l_device, "%s%d", loop_base, i);
427 #else
428                 sprintf(l_device, "%s%d", loop_base, i);
429                 if (access(l_device, F_OK | R_OK))
430                         break;
431 #endif
432                 snprintf(cmd, cmdsz, "losetup %s > /dev/null 2>&1", l_device);
433                 ret = system(cmd);
434
435                 /* losetup gets 1 (ret=256) for non-set-up device */
436                 if (ret) {
437                         /* Set up a loopback device to our file */
438                         snprintf(cmd, cmdsz, "losetup %s %s", l_device,
439                                  mop->mo_device);
440                         ret = run_command(cmd, cmdsz);
441                         if (ret == 256)
442                                 /* someone else picked up this loop device
443                                  * behind our back */
444                                 continue;
445                         if (ret) {
446                                 fprintf(stderr, "%s: error %d on losetup: %s\n",
447                                         progname, ret,
448                                         ret >= 0 ? strerror(ret) : "");
449                                 return ret;
450                         }
451                         strscpy(mop->mo_loopdev, l_device,
452                                 sizeof(mop->mo_loopdev));
453                         return ret;
454                 }
455         }
456
457         fprintf(stderr, "%s: out of loop devices!\n", progname);
458         return EMFILE;
459 }
460
461 int loop_cleanup(struct mkfs_opts *mop)
462 {
463         char cmd[150];
464         int ret = 0;
465
466         if ((mop->mo_flags & MO_IS_LOOP) && *mop->mo_loopdev) {
467                 int tries;
468
469                 sprintf(cmd, "losetup -d %s", mop->mo_loopdev);
470                 for (tries = 0; tries < 3; tries++) {
471                         ret = run_command(cmd, sizeof(cmd));
472                         if (ret == 0)
473                                 break;
474                         sleep(1);
475                 }
476         }
477
478         if (ret != 0)
479                 fprintf(stderr, "cannot cleanup %s: rc = %d\n",
480                         mop->mo_loopdev, ret);
481         return ret;
482 }
483
484 int loop_format(struct mkfs_opts *mop)
485 {
486         int fd;
487
488         if (mop->mo_device_kb == 0) {
489                 fatal();
490                 fprintf(stderr, "loop device requires a --device-size= "
491                         "param\n");
492                 return EINVAL;
493         }
494
495         fd = creat(mop->mo_device, S_IRUSR|S_IWUSR);
496         if (fd < 0) {
497                 fatal();
498                 fprintf(stderr, "%s: Unable to create backing store: %s\n",
499                         progname, strerror(errno));
500                 return errno;
501         }
502
503         if (ftruncate(fd, mop->mo_device_kb * 1024) != 0) {
504                 close(fd);
505                 fatal();
506                 fprintf(stderr, "%s: Unable to truncate backing store: %s\n",
507                         progname, strerror(errno));
508                 return errno;
509         }
510
511         close(fd);
512         return 0;
513 }
514
515 #define DLSYM(prefix, sym, func)                                        \
516         do {                                                            \
517                 char _fname[64];                                        \
518                 snprintf(_fname, sizeof(_fname), "%s_%s", prefix, #func); \
519                 sym->func = (typeof(sym->func))dlsym(sym->dl_handle, _fname); \
520         } while (0)
521
522 /**
523  * Load plugin for a given mount_type from ${pkglibdir}/mount_osd_FSTYPE.so and
524  * return struct of function pointers (will be freed in unloack_backfs_module).
525  *
526  * \param[in] mount_type        Mount type to load module for.
527  * \retval Value of backfs_ops struct
528  * \retval NULL if no module exists
529  */
530 struct module_backfs_ops *load_backfs_module(enum ldd_mount_type mount_type)
531 {
532         void *handle;
533         char *error, filename[512], fsname[512], *name;
534         struct module_backfs_ops *ops;
535
536         /* This deals with duplicate ldd_mount_types resolving to same OSD layer
537          * plugin (e.g. ext3/ldiskfs/ldiskfs2 all being ldiskfs) */
538         strncpy(fsname, mt_type(mount_type), sizeof(fsname));
539         name = fsname + sizeof("osd-") - 1;
540
541         /* change osd- to osd_ */
542         fsname[sizeof("osd-") - 2] = '_';
543
544         snprintf(filename, sizeof(filename), PLUGIN_DIR"/mount_%s.so", fsname);
545
546         handle = dlopen(filename, RTLD_LAZY);
547
548         /* Check for $LUSTRE environment variable from test-framework.
549          * This allows using locally built modules to be used.
550          */
551         if (handle == NULL) {
552                 char *dirname;
553                 dirname = getenv("LUSTRE");
554                 if (dirname) {
555                         snprintf(filename, sizeof(filename),
556                                  "%s/utils/.libs/mount_%s.so",
557                                  dirname, fsname);
558                         handle = dlopen(filename, RTLD_LAZY);
559                 }
560         }
561
562         /* Do not clutter up console with missing types */
563         if (handle == NULL)
564                 return NULL;
565
566         ops = malloc(sizeof(*ops));
567         if (ops == NULL) {
568                 dlclose(handle);
569                 return NULL;
570         }
571
572         ops->dl_handle = handle;
573         dlerror(); /* Clear any existing error */
574
575         DLSYM(name, ops, init);
576         DLSYM(name, ops, fini);
577         DLSYM(name, ops, read_ldd);
578         DLSYM(name, ops, write_ldd);
579         DLSYM(name, ops, is_lustre);
580         DLSYM(name, ops, make_lustre);
581         DLSYM(name, ops, prepare_lustre);
582         DLSYM(name, ops, tune_lustre);
583         DLSYM(name, ops, label_lustre);
584         DLSYM(name, ops, enable_quota);
585
586         error = dlerror();
587         if (error != NULL) {
588                 fatal();
589                 fprintf(stderr, "%s\n", error);
590                 dlclose(handle);
591                 free(ops);
592                 return NULL;
593         }
594
595         /* optional methods */
596         DLSYM(name, ops, fix_mountopts);
597
598         return ops;
599 }
600
601 /**
602  * Unload plugin and free backfs_ops structure. Must be called the same number
603  * of times as load_backfs_module is.
604  */
605 void unload_backfs_module(struct module_backfs_ops *ops)
606 {
607         if (ops == NULL)
608                 return;
609
610         dlclose(ops->dl_handle);
611         free(ops);
612 }
613
614 /* Return true if backfs_ops has operations for the given mount_type. */
615 int backfs_mount_type_okay(enum ldd_mount_type mount_type)
616 {
617         if (unlikely(mount_type >= LDD_MT_LAST || mount_type < 0)) {
618                 fatal();
619                 fprintf(stderr, "fs type out of range %d\n", mount_type);
620                 return 0;
621         }
622         if (backfs_ops[mount_type] == NULL) {
623                 fatal();
624                 fprintf(stderr, "unhandled/unloaded fs type %d '%s'\n",
625                         mount_type, mt_str(mount_type));
626                 return 0;
627         }
628         return 1;
629 }
630
631 /* Write the server config files */
632 int osd_write_ldd(struct mkfs_opts *mop)
633 {
634         struct lustre_disk_data *ldd = &mop->mo_ldd;
635         int ret;
636
637         if (backfs_mount_type_okay(ldd->ldd_mount_type))
638                 ret = backfs_ops[ldd->ldd_mount_type]->write_ldd(mop);
639
640         else
641                 ret = EINVAL;
642
643         return ret;
644 }
645
646 /* Read the server config files */
647 int osd_read_ldd(char *dev, struct lustre_disk_data *ldd)
648 {
649         int ret;
650
651         if (backfs_mount_type_okay(ldd->ldd_mount_type))
652                 ret = backfs_ops[ldd->ldd_mount_type]->read_ldd(dev, ldd);
653
654         else
655                 ret = EINVAL;
656
657         return ret;
658 }
659
660 /* Was this device formatted for Lustre */
661 int osd_is_lustre(char *dev, unsigned *mount_type)
662 {
663         int i;
664
665         vprint("checking for existing Lustre data: ");
666
667         for (i = 0; i < LDD_MT_LAST; ++i) {
668                 if (backfs_ops[i] != NULL &&
669                     backfs_ops[i]->is_lustre(dev, mount_type)) {
670                         vprint("found\n");
671                         return 1;
672                 }
673         }
674
675         vprint("not found\n");
676         return 0;
677 }
678
679 /* Build fs according to type */
680 int osd_make_lustre(struct mkfs_opts *mop)
681 {
682         struct lustre_disk_data *ldd = &mop->mo_ldd;
683         int ret;
684
685         if (backfs_mount_type_okay(ldd->ldd_mount_type))
686                 ret = backfs_ops[ldd->ldd_mount_type]->make_lustre(mop);
687
688         else
689                 ret = EINVAL;
690
691         return ret;
692 }
693
694 int osd_prepare_lustre(struct mkfs_opts *mop,
695                        char *wanted_mountopts, size_t len)
696 {
697         struct lustre_disk_data *ldd = &mop->mo_ldd;
698         int ret;
699
700         if (backfs_mount_type_okay(ldd->ldd_mount_type))
701                 ret = backfs_ops[ldd->ldd_mount_type]->prepare_lustre(mop,
702                                                         wanted_mountopts, len);
703
704         else
705                 ret = EINVAL;
706
707         return ret;
708 }
709
710 int osd_fix_mountopts(struct mkfs_opts *mop, char *mountopts, size_t len)
711 {
712         struct lustre_disk_data *ldd = &mop->mo_ldd;
713
714         if (!backfs_mount_type_okay(ldd->ldd_mount_type))
715                 return EINVAL;
716
717         if (backfs_ops[ldd->ldd_mount_type]->fix_mountopts == NULL)
718                 return 0;
719
720         return backfs_ops[ldd->ldd_mount_type]->fix_mountopts(mop, mountopts,
721                                                               len);
722 }
723
724 int osd_tune_lustre(char *dev, struct mount_opts *mop)
725 {
726         struct lustre_disk_data *ldd = &mop->mo_ldd;
727         int ret;
728
729         if (backfs_mount_type_okay(ldd->ldd_mount_type))
730                 ret = backfs_ops[ldd->ldd_mount_type]->tune_lustre(dev, mop);
731
732         else
733                 ret = EINVAL;
734
735         return ret;
736 }
737
738 int osd_label_lustre(struct mount_opts *mop)
739 {
740         struct lustre_disk_data *ldd = &mop->mo_ldd;
741         int ret;
742
743         if (backfs_mount_type_okay(ldd->ldd_mount_type))
744                 ret = backfs_ops[ldd->ldd_mount_type]->label_lustre(mop);
745
746         else
747                 ret = EINVAL;
748
749         return ret;
750 }
751
752 /* Enable quota accounting */
753 int osd_enable_quota(struct mkfs_opts *mop)
754 {
755         struct lustre_disk_data *ldd = &mop->mo_ldd;
756         int ret;
757
758         if (backfs_mount_type_okay(ldd->ldd_mount_type))
759                 ret = backfs_ops[ldd->ldd_mount_type]->enable_quota(mop);
760
761         else
762                 ret = EINVAL;
763
764         return ret;
765 }
766
767 int osd_init(void)
768 {
769         int i, rc, ret = EINVAL;
770
771         for (i = 0; i < LDD_MT_LAST; ++i) {
772                 rc = 0;
773                 backfs_ops[i] = load_backfs_module(i);
774                 if (backfs_ops[i] != NULL)
775                         rc = backfs_ops[i]->init();
776                 if (rc != 0) {
777                         backfs_ops[i]->fini();
778                         unload_backfs_module(backfs_ops[i]);
779                         backfs_ops[i] = NULL;
780                 } else
781                         ret = 0;
782         }
783
784         return ret;
785 }
786
787 void osd_fini(void)
788 {
789         int i;
790
791         for (i = 0; i < LDD_MT_LAST; ++i) {
792                 if (backfs_ops[i] != NULL) {
793                         backfs_ops[i]->fini();
794                         unload_backfs_module(backfs_ops[i]);
795                         backfs_ops[i] = NULL;
796                 }
797         }
798 }
799
800 __u64 get_device_size(char* device)
801 {
802         int ret, fd;
803         __u64 size = 0;
804
805         fd = open(device, O_RDONLY);
806         if (fd < 0) {
807                 fprintf(stderr, "%s: cannot open %s: %s\n",
808                         progname, device, strerror(errno));
809                 return 0;
810         }
811
812 #ifdef BLKGETSIZE64
813         /* size in bytes. bz5831 */
814         ret = ioctl(fd, BLKGETSIZE64, (void*)&size);
815 #else
816         {
817                 __u32 lsize = 0;
818                 /* size in blocks */
819                 ret = ioctl(fd, BLKGETSIZE, (void*)&lsize);
820                 size = (__u64)lsize * 512;
821         }
822 #endif
823         close(fd);
824         if (ret < 0) {
825                 fprintf(stderr, "%s: size ioctl failed: %s\n",
826                         progname, strerror(errno));
827                 return 0;
828         }
829
830         vprint("device size = %juMB\n", (uintmax_t)(size >> 20));
831         /* return value in KB */
832         return size >> 10;
833 }
834
835 int file_create(char *path, __u64 size)
836 {
837         __u64 size_max;
838         int ret;
839         int fd;
840
841         /*
842          * Since "size" is in KB, the file offset it represents could overflow
843          * off_t.
844          */
845         size_max = (off_t)1 << (_FILE_OFFSET_BITS - 1 - 10);
846         if (size >= size_max) {
847                 fprintf(stderr, "%s: %ju KB: Backing store size must be "
848                         "smaller than %ju KB\n", progname, (uintmax_t) size,
849                         (uintmax_t)size_max);
850                 return EFBIG;
851         }
852
853         ret = access(path, F_OK);
854         if (ret == 0) {
855                 ret = unlink(path);
856                 if (ret != 0)
857                         return errno;
858         }
859
860         fd = creat(path, S_IRUSR|S_IWUSR);
861         if (fd < 0) {
862                 fatal();
863                 fprintf(stderr, "%s: Unable to create backing store: %s\n",
864                         progname, strerror(errno));
865                 return errno;
866         }
867
868         ret = ftruncate(fd, size * 1024);
869         close(fd);
870         if (ret != 0) {
871                 fatal();
872                 fprintf(stderr, "%s: Unable to truncate backing store: %s\n",
873                         progname, strerror(errno));
874                 return errno;
875         }
876
877         return 0;
878 }