+
+static int mntdf(char *mntdir, int ishow, int cooked)
+{
+ struct obd_statfs stat_buf, sum = { .os_bsize = 1 };
+ struct obd_uuid uuid_buf;
+ __u32 index;
+ int rc;
+
+ if (ishow)
+ printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
+ "UUID", "Inodes", "IUsed", "IFree",
+ "IUse%", "Mounted on");
+ else
+ printf(UUF" "CSF" "CSF" "CSF" "RSF" %-s\n",
+ "UUID", cooked ? "bytes" : "1K-blocks",
+ "Used", "Available", "Use%", "Mounted on");
+
+ for (index = 0; ; index++) {
+ memset(&stat_buf, 0, sizeof(struct obd_statfs));
+ memset(&uuid_buf, 0, sizeof(struct obd_uuid));
+ rc = llapi_obd_statfs(mntdir, LL_STATFS_MDC, index,
+ &stat_buf, &uuid_buf);
+ if (rc == -ENODEV)
+ break;
+
+ if (rc == -EAGAIN)
+ continue;
+
+ if (rc == -ENOTCONN || rc == -ETIMEDOUT || rc == -EIO ||
+ rc == -ENODATA || rc == 0) {
+ showdf(mntdir, &stat_buf, obd_uuid2str(&uuid_buf),
+ ishow, cooked, "MDT", index, rc);
+ } else {
+ fprintf(stderr,
+ "error: llapi_obd_statfs(%s): %s (%d)\n",
+ obd_uuid2str(&uuid_buf), strerror(-rc), rc);
+ return rc;
+ }
+ if (rc == 0) {
+ sum.os_ffree += stat_buf.os_ffree;
+ sum.os_files += stat_buf.os_files;
+ }
+ }
+
+ for (index = 0; ; index++) {
+ memset(&stat_buf, 0, sizeof(struct obd_statfs));
+ memset(&uuid_buf, 0, sizeof(struct obd_uuid));
+ rc = llapi_obd_statfs(mntdir, LL_STATFS_LOV, index,
+ &stat_buf, &uuid_buf);
+ if (rc == -ENODEV)
+ break;
+
+ if (rc == -EAGAIN)
+ continue;
+
+ if (rc == -ENOTCONN || rc == -ETIMEDOUT || rc == -EIO ||
+ rc == -ENODATA || rc == 0) {
+ showdf(mntdir, &stat_buf, obd_uuid2str(&uuid_buf),
+ ishow, cooked, "OST", index, rc);
+ } else {
+ fprintf(stderr,
+ "error: llapi_obd_statfs failed: %s (%d)\n",
+ strerror(-rc), rc);
+ return rc;
+ }
+ if (rc == 0) {
+ sum.os_blocks += stat_buf.os_blocks * stat_buf.os_bsize;
+ sum.os_bfree += stat_buf.os_bfree * stat_buf.os_bsize;
+ sum.os_bavail += stat_buf.os_bavail * stat_buf.os_bsize;
+ }
+ }
+
+ printf("\n");
+ showdf(mntdir, &sum, "filesystem summary:", ishow, cooked, NULL, 0,0);
+
+ return 0;
+}
+
+static int lfs_df(int argc, char **argv)
+{
+ FILE *fp;
+ char *path = NULL;
+ struct mntent *mnt = NULL;
+ char *mntdir = NULL;
+ int ishow = 0, cooked = 0;
+ int c, rc = 0;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ih")) != -1) {
+ switch (c) {
+ case 'i':
+ ishow = 1;
+ break;
+ case 'h':
+ cooked = 1;
+ break;
+ default:
+ return CMD_HELP;
+ }
+ }
+ if (optind < argc )
+ path = argv[optind];
+
+ fp = setmntent(MOUNTED, "r");
+ if (fp == NULL) {
+ rc = -errno;
+ fprintf(stderr, "error: %s: open %s failed( %s )\n",
+ argv[0], MOUNTED, strerror(errno));
+ return rc;
+ }
+
+ if ((mntdir = malloc(PATH_MAX)) == NULL) {
+ fprintf(stderr, "error: cannot allocate %d bytes\n",
+ PATH_MAX);
+ return -ENOMEM;
+ }
+ memset(mntdir, 0, PATH_MAX);
+
+ if (path) {
+ rc = path2mnt(path, fp, mntdir, PATH_MAX);
+ if (rc) {
+ endmntent(fp);
+ free(mntdir);
+ return rc;
+ }
+
+ rc = mntdf(mntdir, ishow, cooked);
+ printf("\n");
+ endmntent(fp);
+ } else {
+ mnt = getmntent(fp);
+ while (feof(fp) == 0 && ferror(fp) == 0) {
+ if (llapi_is_lustre_mnt(mnt)) {
+ rc = mntdf(mnt->mnt_dir, ishow, cooked);
+ if (rc)
+ break;
+ printf("\n");
+ }
+ mnt = getmntent(fp);
+ }
+ endmntent(fp);
+ }
+
+ free(mntdir);
+ return rc;
+}
+
+static int lfs_check(int argc, char **argv)
+{
+ int rc;
+ FILE *fp;
+ struct mntent *mnt = NULL;
+ int num_types = 1;
+ char *obd_types[2];
+ char obd_type1[4];
+ char obd_type2[4];
+
+ if (argc != 2)
+ return CMD_HELP;
+
+ obd_types[0] = obd_type1;
+ obd_types[1] = obd_type2;
+
+ if (strcmp(argv[1], "osts") == 0) {
+ strcpy(obd_types[0], "osc");
+ } else if (strcmp(argv[1], "mds") == 0) {
+ strcpy(obd_types[0], "mdc");
+ } else if (strcmp(argv[1], "servers") == 0) {
+ num_types = 2;
+ strcpy(obd_types[0], "osc");
+ strcpy(obd_types[1], "mdc");
+ } else {
+ fprintf(stderr, "error: %s: option '%s' unrecognized\n",
+ argv[0], argv[1]);
+ return CMD_HELP;
+ }
+
+ fp = setmntent(MOUNTED, "r");
+ if (fp == NULL) {
+ fprintf(stderr, "setmntent(%s): %s:", MOUNTED,
+ strerror (errno));
+ } else {
+ mnt = getmntent(fp);
+ while (feof(fp) == 0 && ferror(fp) ==0) {
+ if (llapi_is_lustre_mnt(mnt))
+ break;
+ mnt = getmntent(fp);
+ }
+ endmntent(fp);
+ }
+
+ if (!mnt) {
+ fprintf(stderr, "No suitable Lustre mount found\n");
+ return -1;
+ }
+
+ rc = llapi_target_check(num_types, obd_types, mnt->mnt_dir);
+
+ if (rc)
+ fprintf(stderr, "error: %s: %s status failed\n",
+ argv[0],argv[1]);
+
+ return rc;
+
+}
+
+static int lfs_catinfo(int argc, char **argv)
+{
+ FILE *fp;
+ struct mntent *mnt = NULL;
+ int rc;
+
+ if (argc < 2 || (!strcmp(argv[1],"config") && argc < 3))
+ return CMD_HELP;
+
+ if (strcmp(argv[1], "config") && strcmp(argv[1], "deletions"))
+ return CMD_HELP;
+
+ fp = setmntent(MOUNTED, "r");
+ if (fp == NULL) {
+ fprintf(stderr, "setmntent(%s): %s:", MOUNTED,
+ strerror(errno));
+ } else {
+ mnt = getmntent(fp);
+ while (feof(fp) == 0 && ferror(fp) == 0) {
+ if (llapi_is_lustre_mnt(mnt))
+ break;
+ mnt = getmntent(fp);
+ }
+ endmntent(fp);
+ }
+
+ if (mnt) {
+ if (argc == 3)
+ rc = llapi_catinfo(mnt->mnt_dir, argv[1], argv[2]);
+ else
+ rc = llapi_catinfo(mnt->mnt_dir, argv[1], NULL);
+ } else {
+ fprintf(stderr, "no lustre_lite mounted.\n");
+ rc = -1;
+ }
+
+ return rc;
+}
+
+int lfs_join(int argc, char **argv)
+{
+ char *name_head, *name_tail;
+ int fd, rc;
+ loff_t size;
+
+ if (argc != 3)
+ return CMD_HELP;
+ name_head = argv[1];
+ fd = open(name_head, O_WRONLY);
+ if (fd < 0) {
+ fprintf(stderr, "Can not open name_head %s rc=%d\n",
+ name_head, fd);
+ return fd;
+ }
+ size = lseek(fd, 0, SEEK_END);
+ if (size % JOIN_FILE_ALIGN) {
+ fprintf(stderr,"head file %s size %llu must be mutiple of %d\n",
+ name_head, (long long)size, JOIN_FILE_ALIGN);
+ rc = -EINVAL;
+ goto out;
+ }
+ name_tail = argv[2];
+ rc = ioctl(fd, LL_IOC_JOIN, name_tail);
+out:
+ close(fd);
+ if (rc) {
+ fprintf(stderr, "Lustre joining files: %s, %s, failed\n",
+ argv[1], argv[2]);
+ }
+ return rc;
+}
+
+#ifdef HAVE_SYS_QUOTA_H
+static int lfs_quotachown(int argc, char **argv)
+{
+
+ int c,rc;
+ int flag = 0;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "i")) != -1) {
+ switch (c) {
+ case 'i':
+ flag++;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+ if (optind == argc)
+ return CMD_HELP;
+ rc = llapi_quotachown(argv[optind], flag);
+ if(rc)
+ fprintf(stderr,"error: change file owner/group failed.\n");
+ return rc;
+}
+
+static int lfs_quotacheck(int argc, char **argv)
+{
+ int c, check_type = 0;
+ char *mnt;
+ struct if_quotacheck qchk;
+ struct if_quotactl qctl;
+ char *obd_type = (char *)qchk.obd_type;
+ int rc;
+
+ memset(&qchk, 0, sizeof(qchk));
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ug")) != -1) {
+ switch (c) {
+ case 'u':
+ check_type |= 0x01;
+ break;
+ case 'g':
+ check_type |= 0x02;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ if (check_type)
+ check_type--;
+ else /* do quotacheck for both user & group quota by default */
+ check_type = 0x02;
+
+ if (argc == optind)
+ return CMD_HELP;
+
+ mnt = argv[optind];
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
+ qctl.qc_type = check_type;
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ fprintf(stderr, "quota off failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ rc = llapi_quotacheck(mnt, check_type);
+ if (rc) {
+ fprintf(stderr, "quotacheck failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ rc = llapi_poll_quotacheck(mnt, &qchk);
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", obd_type,
+ obd_uuid2str(&qchk.obd_uuid));
+ fprintf(stderr, "quota check failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_QUOTAON;
+ qctl.qc_type = check_type;
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", (char *)qctl.obd_type,
+ obd_uuid2str(&qctl.obd_uuid));
+ fprintf(stderr, "%s turn on quota failed: %s\n",
+ argv[0], strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+static int lfs_quotaon(int argc, char **argv)
+{
+ int c;
+ char *mnt;
+ struct if_quotactl qctl;
+ char *obd_type = (char *)qctl.obd_type;
+ int rc;
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_QUOTAON;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ugf")) != -1) {
+ switch (c) {
+ case 'u':
+ qctl.qc_type |= 0x01;
+ break;
+ case 'g':
+ qctl.qc_type |= 0x02;
+ break;
+ case 'f':
+ qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ if (qctl.qc_type)
+ qctl.qc_type--;
+ else /* by default, enable quota for both user & group */
+ qctl.qc_type = 0x02;
+
+ if (argc == optind)
+ return CMD_HELP;
+
+ mnt = argv[optind];
+
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", obd_type,
+ obd_uuid2str(&qctl.obd_uuid));
+ fprintf(stderr, "%s failed: %s\n", argv[0], strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+static int lfs_quotaoff(int argc, char **argv)
+{
+ int c;
+ char *mnt;
+ struct if_quotactl qctl;
+ char *obd_type = (char *)qctl.obd_type;
+ int rc;
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_QUOTAOFF;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ug")) != -1) {
+ switch (c) {
+ case 'u':
+ qctl.qc_type |= 0x01;
+ break;
+ case 'g':
+ qctl.qc_type |= 0x02;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ if (qctl.qc_type)
+ qctl.qc_type--;
+ else /* by default, disable quota for both user & group */
+ qctl.qc_type = 0x02;
+
+ if (argc == optind)
+ return CMD_HELP;
+
+ mnt = argv[optind];
+
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc == -1 && errno == ESRCH) {
+ fprintf(stderr, "\n%s quotas are not enabled.\n",
+ qctl.qc_type == 0x00 ? "user" : "group");
+ return 0;
+ }
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", obd_type,
+ obd_uuid2str(&qctl.obd_uuid));
+ fprintf(stderr, "quotaoff failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+static int lfs_quotainv(int argc, char **argv)
+{
+ int c;
+ char *mnt;
+ struct if_quotactl qctl;
+ int rc;
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_INVALIDATE;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ugf")) != -1) {
+ switch (c) {
+ case 'u':
+ qctl.qc_type |= 0x01;
+ break;
+ case 'g':
+ qctl.qc_type |= 0x02;
+ break;
+ case 'f':
+ qctl.qc_cmd = LUSTRE_Q_FINVALIDATE;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ if (qctl.qc_type)
+ qctl.qc_type--;
+ else /* by default, invalidate quota for both user & group */
+ qctl.qc_type = 0x02;
+
+ if (argc == optind)
+ return CMD_HELP;
+
+ mnt = argv[optind];
+
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ fprintf(stderr, "quotainv failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+#define ARG2INT(nr, str, msg) \
+do { \
+ char *endp; \
+ nr = strtol(str, &endp, 0); \
+ if (*endp) { \
+ fprintf(stderr, "error: bad %s: %s\n", msg, str); \
+ return CMD_HELP; \
+ } \
+} while (0)
+
+#define ADD_OVERFLOW(a,b) ((a + b) < a) ? (a = ULONG_MAX) : (a = a + b)
+
+/* Convert format time string "XXwXXdXXhXXmXXs" into seconds value
+ * returns the value or ULONG_MAX on integer overflow or incorrect format
+ * Notes:
+ * 1. the order of specifiers is arbitrary (may be: 5w3s or 3s5w)
+ * 2. specifiers may be encountered multiple times (2s3s is 5 seconds)
+ * 3. empty integer value is interpreted as 0
+ */
+
+static unsigned long str2sec(const char* timestr) {
+ const char spec[] = "smhdw";
+ const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
+ unsigned long val = 0;
+ char *tail;
+
+ if (strpbrk(timestr, spec) == NULL) {
+ /* no specifiers inside the time string,
+ should treat it as an integer value */
+ val = strtoul(timestr, &tail, 10);
+ return *tail ? ULONG_MAX : val;
+ }
+
+ /* format string is XXwXXdXXhXXmXXs */
+ while (*timestr) {
+ unsigned long v;
+ int ind;
+ char* ptr;
+
+ v = strtoul(timestr, &tail, 10);
+ if (v == ULONG_MAX || *tail == '\0')
+ /* value too large (ULONG_MAX or more)
+ or missing specifier */
+ goto error;
+
+ ptr = strchr(spec, *tail);
+ if (ptr == NULL)
+ /* unknown specifier */
+ goto error;
+
+ ind = ptr - spec;
+
+ /* check if product will overflow the type */
+ if (!(v < ULONG_MAX / mult[ind]))
+ goto error;
+
+ ADD_OVERFLOW(val, mult[ind] * v);
+ if (val == ULONG_MAX)
+ goto error;
+
+ timestr = tail + 1;
+ }
+
+ return val;
+
+error:
+ return ULONG_MAX;
+}
+
+#define ARG2ULL(nr, str, defscale) \
+do { \
+ unsigned long long limit, units = 0; \
+ int rc; \
+ \
+ rc = parse_size(str, &limit, &units, 1); \
+ if (rc < 0) { \
+ fprintf(stderr, "error: bad limit value %s\n", str); \
+ return CMD_HELP; \
+ } \
+ nr = ((units == 0) ? (defscale) : 1) * limit; \
+} while (0)
+
+static inline int has_times_option(int argc, char **argv)
+{
+ int i;
+
+ for (i = 1; i < argc; i++)
+ if (!strcmp(argv[i], "-t"))
+ return 1;
+
+ return 0;
+}
+
+int lfs_setquota_times(int argc, char **argv)
+{
+ int c, rc;
+ struct if_quotactl qctl;
+ char *mnt, *obd_type = (char *)qctl.obd_type;
+ struct obd_dqblk *dqb = &qctl.qc_dqblk;
+ struct obd_dqinfo *dqi = &qctl.qc_dqinfo;
+ struct option long_opts[] = {
+ {"user", no_argument, 0, 'u'},
+ {"group", no_argument, 0, 'g'},
+ {"block-grace", required_argument, 0, 'b'},
+ {"inode-grace", required_argument, 0, 'i'},
+ {"times", no_argument, 0, 't'},
+ {0, 0, 0, 0}
+ };
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_SETINFO;
+ qctl.qc_type = UGQUOTA;
+
+ optind = 0;
+ while ((c = getopt_long(argc, argv, "ugb:i:t", long_opts, NULL)) != -1) {
+ switch (c) {
+ case 'u':
+ case 'g':
+ if (qctl.qc_type != UGQUOTA) {
+ fprintf(stderr, "error: -u and -g can't be used "
+ "more than once\n");
+ return CMD_HELP;
+ }
+ qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
+ break;
+ case 'b':
+ if ((dqi->dqi_bgrace = str2sec(optarg)) == ULONG_MAX) {
+ fprintf(stderr, "error: bad block-grace: %s\n",
+ optarg);
+ return CMD_HELP;
+ }
+ dqb->dqb_valid |= QIF_BTIME;
+ break;
+ case 'i':
+ if ((dqi->dqi_igrace = str2sec(optarg)) == ULONG_MAX) {
+ fprintf(stderr, "error: bad inode-grace: %s\n",
+ optarg);
+ return CMD_HELP;
+ }
+ dqb->dqb_valid |= QIF_ITIME;
+ break;
+ case 't': /* Yes, of course! */
+ break;
+ default: /* getopt prints error message for us when opterr != 0 */
+ return CMD_HELP;
+ }
+ }
+
+ if (qctl.qc_type == UGQUOTA) {
+ fprintf(stderr, "error: neither -u nor -g specified\n");
+ return CMD_HELP;
+ }
+
+ if (optind != argc - 1) {
+ fprintf(stderr, "error: unexpected parameters encountered\n");
+ return CMD_HELP;
+ }
+
+ mnt = argv[optind];
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", obd_type,
+ obd_uuid2str(&qctl.obd_uuid));
+ fprintf(stderr, "setquota failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+#define BSLIMIT (1 << 0)
+#define BHLIMIT (1 << 1)
+#define ISLIMIT (1 << 2)
+#define IHLIMIT (1 << 3)
+
+int lfs_setquota(int argc, char **argv)
+{
+ int c, rc;
+ struct if_quotactl qctl;
+ char *mnt, *obd_type = (char *)qctl.obd_type;
+ struct obd_dqblk *dqb = &qctl.qc_dqblk;
+ struct option long_opts[] = {
+ {"user", required_argument, 0, 'u'},
+ {"group", required_argument, 0, 'g'},
+ {"block-softlimit", required_argument, 0, 'b'},
+ {"block-hardlimit", required_argument, 0, 'B'},
+ {"inode-softlimit", required_argument, 0, 'i'},
+ {"inode-hardlimit", required_argument, 0, 'I'},
+ {0, 0, 0, 0}
+ };
+ unsigned limit_mask = 0;
+
+ if (has_times_option(argc, argv))
+ return lfs_setquota_times(argc, argv);
+
+ memset(&qctl, 0, sizeof(qctl));
+ qctl.qc_cmd = LUSTRE_Q_SETQUOTA;
+ qctl.qc_type = UGQUOTA; /* UGQUOTA makes no sense for setquota,
+ * so it can be used as a marker that qc_type
+ * isn't reinitialized from command line */
+
+ optind = 0;
+ while ((c = getopt_long(argc, argv, "u:g:b:B:i:I:", long_opts, NULL)) != -1) {
+ switch (c) {
+ case 'u':
+ case 'g':
+ if (qctl.qc_type != UGQUOTA) {
+ fprintf(stderr, "error: -u and -g can't be used"
+ " more than once\n");
+ return CMD_HELP;
+ }
+ qctl.qc_type = (c == 'u') ? USRQUOTA : GRPQUOTA;
+ rc = name2id(&qctl.qc_id, optarg,
+ (qctl.qc_type == USRQUOTA) ? USER : GROUP);
+ if (rc) {
+ fprintf(stderr, "error: unknown id %s\n",
+ optarg);
+ return CMD_HELP;
+ }
+ break;
+ case 'b':
+ ARG2ULL(dqb->dqb_bsoftlimit, optarg, 1024);
+ dqb->dqb_bsoftlimit >>= 10;
+ limit_mask |= BSLIMIT;
+ break;
+ case 'B':
+ ARG2ULL(dqb->dqb_bhardlimit, optarg, 1024);
+ dqb->dqb_bhardlimit >>= 10;
+ limit_mask |= BHLIMIT;
+ break;
+ case 'i':
+ ARG2ULL(dqb->dqb_isoftlimit, optarg, 1);
+ limit_mask |= ISLIMIT;
+ break;
+ case 'I':
+ ARG2ULL(dqb->dqb_ihardlimit, optarg, 1);
+ limit_mask |= IHLIMIT;
+ break;
+ default: /* getopt prints error message for us when opterr != 0 */
+ return CMD_HELP;
+ }
+ }
+
+ if (qctl.qc_type == UGQUOTA) {
+ fprintf(stderr, "error: neither -u nor -g are specified\n");
+ return CMD_HELP;
+ }
+
+ if (optind != argc - 1) {
+ fprintf(stderr, "error: unexpected parameters encountered\n");
+ return CMD_HELP;
+ }
+
+ mnt = argv[optind];
+
+ if ((!(limit_mask & BHLIMIT) ^ !(limit_mask & BSLIMIT)) ||
+ (!(limit_mask & IHLIMIT) ^ !(limit_mask & ISLIMIT))) {
+ /* sigh, we can't just set blimits/ilimits */
+ struct if_quotactl tmp_qctl = {.qc_cmd = LUSTRE_Q_GETQUOTA,
+ .qc_type = qctl.qc_type,
+ .qc_id = qctl.qc_id};
+
+ rc = llapi_quotactl(mnt, &tmp_qctl);
+ if (rc < 0) {
+ fprintf(stderr, "error: getquota failed\n");
+ return CMD_HELP;
+ }
+
+ if (!(limit_mask & BHLIMIT))
+ dqb->dqb_bhardlimit = tmp_qctl.qc_dqblk.dqb_bhardlimit;
+ if (!(limit_mask & BSLIMIT))
+ dqb->dqb_bsoftlimit = tmp_qctl.qc_dqblk.dqb_bsoftlimit;
+ if (!(limit_mask & IHLIMIT))
+ dqb->dqb_ihardlimit = tmp_qctl.qc_dqblk.dqb_ihardlimit;
+ if (!(limit_mask & ISLIMIT))
+ dqb->dqb_isoftlimit = tmp_qctl.qc_dqblk.dqb_isoftlimit;
+ }
+
+ dqb->dqb_valid |= (limit_mask & (BHLIMIT | BSLIMIT)) ? QIF_BLIMITS : 0;
+ dqb->dqb_valid |= (limit_mask & (IHLIMIT | ISLIMIT)) ? QIF_ILIMITS : 0;
+
+ rc = llapi_quotactl(mnt, &qctl);
+ if (rc) {
+ if (*obd_type)
+ fprintf(stderr, "%s %s ", obd_type,
+ obd_uuid2str(&qctl.obd_uuid));
+ fprintf(stderr, "setquota failed: %s\n", strerror(errno));
+ return rc;
+ }
+
+ return 0;
+}
+
+static inline char *type2name(int check_type)
+{
+ if (check_type == USRQUOTA)
+ return "user";
+ else if (check_type == GRPQUOTA)
+ return "group";
+ else
+ return "unknown";
+}
+
+
+/* Converts seconds value into format string
+ * result is returned in buf
+ * Notes:
+ * 1. result is in descenting order: 1w2d3h4m5s
+ * 2. zero fields are not filled (except for p. 3): 5d1s
+ * 3. zero seconds value is presented as "0s"
+ */
+static void sec2str(time_t seconds, char *buf)
+{
+ const char spec[] = "smhdw";
+ const unsigned long mult[] = {1, 60, 60*60, 24*60*60, 7*24*60*60};
+ unsigned long c;
+ char* tail = buf;
+ int i;
+
+ for (i = sizeof(mult) / sizeof(mult[0]) - 1 ; i >= 0; i--) {
+ c = seconds / mult[i];
+
+ if (c > 0 || (i == 0 && buf == tail))
+ tail += snprintf(tail, 40-(tail-buf), "%lu%c", c, spec[i]);
+
+ seconds %= mult[i];
+ }
+}
+
+
+static void diff2str(time_t seconds, char *buf, time_t now)
+{
+
+ buf[0] = 0;
+ if (!seconds)
+ return;
+ if (seconds <= now) {
+ strcpy(buf, "none");
+ return;
+ }
+ sec2str(seconds - now, buf);
+}
+
+static void print_quota_title(char *name, struct if_quotactl *qctl)
+{
+ printf("Disk quotas for %s %s (%cid %u):\n",
+ type2name(qctl->qc_type), name,
+ *type2name(qctl->qc_type), qctl->qc_id);
+ printf("%15s%8s %7s%8s%8s%8s %7s%8s%8s\n",
+ "Filesystem",
+ "kbytes", "quota", "limit", "grace",
+ "files", "quota", "limit", "grace");
+}
+
+static void print_quota(char *mnt, struct if_quotactl *qctl)
+{
+ time_t now;
+
+ time(&now);
+
+ if (qctl->qc_cmd == LUSTRE_Q_GETQUOTA || qctl->qc_cmd == Q_GETOQUOTA) {
+ int bover = 0, iover = 0;
+ struct obd_dqblk *dqb = &qctl->qc_dqblk;
+
+ if (dqb->dqb_bhardlimit &&
+ toqb(dqb->dqb_curspace) >= dqb->dqb_bhardlimit) {
+ bover = 1;
+ } else if (dqb->dqb_bsoftlimit &&
+ toqb(dqb->dqb_curspace) >= dqb->dqb_bsoftlimit) {
+ if (dqb->dqb_btime > now) {
+ bover = 2;
+ } else {
+ bover = 3;
+ }
+ }
+
+ if (dqb->dqb_ihardlimit &&
+ dqb->dqb_curinodes >= dqb->dqb_ihardlimit) {
+ iover = 1;
+ } else if (dqb->dqb_isoftlimit &&
+ dqb->dqb_curinodes >= dqb->dqb_isoftlimit) {
+ if (dqb->dqb_btime > now) {
+ iover = 2;
+ } else {
+ iover = 3;
+ }
+ }
+
+#if 0 /* XXX: always print quotas even when no usages */
+ if (dqb->dqb_curspace || dqb->dqb_curinodes)
+#endif
+ {
+ char numbuf[3][32];
+ char timebuf[40];
+
+ if (strlen(mnt) > 15)
+ printf("%s\n%15s", mnt, "");
+ else
+ printf("%15s", mnt);
+
+ if (bover)
+ diff2str(dqb->dqb_btime, timebuf, now);
+ sprintf(numbuf[0], (dqb->dqb_valid & QIF_SPACE) ?
+ LPU64 : "["LPU64"]", toqb(dqb->dqb_curspace));
+ if (qctl->qc_valid == QC_GENERAL)
+ sprintf(numbuf[1], (dqb->dqb_valid & QIF_BLIMITS)
+ ? LPU64 : "["LPU64"]",
+ dqb->dqb_bsoftlimit);
+ else
+ sprintf(numbuf[1], "%s", "");
+ sprintf(numbuf[2], (dqb->dqb_valid & QIF_BLIMITS)
+ ? LPU64 : "["LPU64"]", dqb->dqb_bhardlimit);
+ printf(" %7s%c %6s %7s %7s",
+ numbuf[0], bover ? '*' : ' ', numbuf[1],
+ numbuf[2], bover > 1 ? timebuf : "");
+
+ if (iover)
+ diff2str(dqb->dqb_itime, timebuf, now);
+
+ sprintf(numbuf[0], (dqb->dqb_valid & QIF_INODES) ?
+ LPU64 : "["LPU64"]", dqb->dqb_curinodes);
+ if (qctl->qc_valid == QC_GENERAL)
+ sprintf(numbuf[1], (dqb->dqb_valid & QIF_ILIMITS)
+ ? LPU64 : "["LPU64"]",
+ dqb->dqb_isoftlimit);
+ else
+ sprintf(numbuf[1], "%s", "");
+ sprintf(numbuf[2], (dqb->dqb_valid & QIF_ILIMITS) ?
+ LPU64 : "["LPU64"]", dqb->dqb_ihardlimit);
+ if (qctl->qc_valid != QC_OSTIDX)
+ printf(" %7s%c %6s %7s %7s",
+ numbuf[0], iover ? '*' : ' ', numbuf[1],
+ numbuf[2], iover > 1 ? timebuf : "");
+ printf("\n");
+ }
+ } else if (qctl->qc_cmd == LUSTRE_Q_GETINFO ||
+ qctl->qc_cmd == Q_GETOINFO) {
+ char bgtimebuf[40];
+ char igtimebuf[40];
+
+ sec2str(qctl->qc_dqinfo.dqi_bgrace, bgtimebuf);
+ sec2str(qctl->qc_dqinfo.dqi_igrace, igtimebuf);
+ printf("Block grace time: %s; Inode grace time: %s\n",
+ bgtimebuf, igtimebuf);
+ }
+}
+
+static int print_obd_quota(char *mnt, struct if_quotactl *qctl, int is_mdt)
+{
+ int rc = 0, rc1 = 0, count = 0;
+ __u32 valid = qctl->qc_valid;
+
+ rc = llapi_get_obd_count(mnt, &count, is_mdt);
+ if (rc) {
+ fprintf(stderr, "can not get %s count: %s\n",
+ is_mdt ? "mdt": "ost", strerror(errno));
+ return rc;
+ }
+
+ for (qctl->qc_idx = 0; qctl->qc_idx < count; qctl->qc_idx++) {
+ qctl->qc_valid = is_mdt ? QC_MDTIDX : QC_OSTIDX;
+ rc = llapi_quotactl(mnt, qctl);
+ if (rc) {
+ /* It is remote client case. */
+ if (errno == EOPNOTSUPP) {
+ rc = 0;
+ goto out;
+ }
+
+ if (!rc1)
+ rc1 = rc;
+ fprintf(stderr, "quotactl %s%d failed.\n",
+ is_mdt ? "mdt": "ost", qctl->qc_idx);
+ continue;
+ }
+
+ print_quota(obd_uuid2str(&qctl->obd_uuid), qctl);
+ }
+
+out:
+ qctl->qc_valid = valid;
+ return rc ? : rc1;
+}
+
+static int lfs_quota(int argc, char **argv)
+{
+ int c;
+ char *mnt, *name = NULL;
+ struct if_quotactl qctl = { .qc_cmd = LUSTRE_Q_GETQUOTA,
+ .qc_type = UGQUOTA };
+ char *obd_type = (char *)qctl.obd_type;
+ char *obd_uuid = (char *)qctl.obd_uuid.uuid;
+ int rc, rc1 = 0, rc2 = 0, rc3 = 0, verbose = 0, pass = 0;
+ __u32 valid = QC_GENERAL, idx = 0;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "ugto:i:I:v")) != -1) {
+ switch (c) {
+ case 'u':
+ if (qctl.qc_type != UGQUOTA) {
+ fprintf(stderr, "error: use either -u or -g\n");
+ return CMD_HELP;
+ }
+ qctl.qc_type = USRQUOTA;
+ break;
+ case 'g':
+ if (qctl.qc_type != UGQUOTA) {
+ fprintf(stderr, "error: use either -u or -g\n");
+ return CMD_HELP;
+ }
+ qctl.qc_type = GRPQUOTA;
+ break;
+ case 't':
+ qctl.qc_cmd = LUSTRE_Q_GETINFO;
+ break;
+ case 'o':
+ valid = qctl.qc_valid = QC_UUID;
+ strncpy(obd_uuid, optarg, sizeof(qctl.obd_uuid));
+ break;
+ case 'i':
+ valid = qctl.qc_valid = QC_MDTIDX;
+ idx = qctl.qc_idx = atoi(optarg);
+ break;
+ case 'I':
+ valid = qctl.qc_valid = QC_OSTIDX;
+ idx = qctl.qc_idx = atoi(optarg);
+ break;
+ case 'v':
+ verbose = 1;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ /* current uid/gid info for "lfs quota /path/to/lustre/mount" */
+ if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA && qctl.qc_type == UGQUOTA &&
+ optind == argc - 1) {
+ug_output:
+ memset(&qctl, 0, sizeof(qctl)); /* spoiled by print_*_quota */
+ qctl.qc_cmd = LUSTRE_Q_GETQUOTA;
+ qctl.qc_valid = valid;
+ qctl.qc_idx = idx;
+ if (pass++ == 0) {
+ qctl.qc_type = USRQUOTA;
+ qctl.qc_id = geteuid();
+ } else {
+ qctl.qc_type = GRPQUOTA;
+ qctl.qc_id = getegid();
+ }
+ rc = id2name(&name, qctl.qc_id,
+ (qctl.qc_type == USRQUOTA) ? USER : GROUP);
+ if (rc)
+ name = "<unknown>";
+ } else if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA) {
+ if (optind + 2 != argc) {
+ fprintf(stderr, "error: missing quota argument(s)\n");
+ return CMD_HELP;
+ }
+
+ name = argv[optind++];
+ rc = name2id(&qctl.qc_id, name,
+ (qctl.qc_type == USRQUOTA) ? USER : GROUP);
+ if (rc) {
+ fprintf(stderr,"error: can't find id for name %s: %s\n",
+ name, strerror(errno));
+ return CMD_HELP;
+ }
+ } else if (optind + 1 != argc) {
+ fprintf(stderr, "error: missing quota info argument(s)\n");
+ return CMD_HELP;
+ }
+
+ if (qctl.qc_cmd == LUSTRE_Q_GETQUOTA)
+ print_quota_title(name, &qctl);
+
+ mnt = argv[optind];
+
+ rc1 = llapi_quotactl(mnt, &qctl);
+ if (rc1 == -1 && errno == ESRCH) {
+ fprintf(stderr, "\n%s quotas are not enabled.\n",
+ qctl.qc_type == USRQUOTA ? "user" : "group");
+ goto out;
+ }
+ if (rc1 && *obd_type)
+ fprintf(stderr, "%s %s ", obd_type, obd_uuid);
+
+ if (qctl.qc_valid != QC_GENERAL)
+ mnt = obd_uuid2str(&qctl.obd_uuid);
+
+ print_quota(mnt, &qctl);
+
+ if (qctl.qc_valid == QC_GENERAL && qctl.qc_cmd != LUSTRE_Q_GETINFO && verbose) {
+ rc2 = print_obd_quota(mnt, &qctl, 1);
+ rc3 = print_obd_quota(mnt, &qctl, 0);
+ }
+
+ if (rc1 || rc2 || rc3)
+ printf("Some errors happened when getting quota info. "
+ "Some devices may be not working or deactivated. "
+ "The data in \"[]\" is inaccurate.\n");
+
+out:
+ if (pass == 1)
+ goto ug_output;
+
+ return 0;
+}
+#endif /* HAVE_SYS_QUOTA_H! */
+
+static int flushctx_ioctl(char *mp)
+{
+ int fd, rc;
+
+ fd = open(mp, O_RDONLY);
+ if (fd == -1) {
+ fprintf(stderr, "flushctx: error open %s: %s\n",
+ mp, strerror(errno));
+ return -1;
+ }
+
+ rc = ioctl(fd, LL_IOC_FLUSHCTX);
+ if (rc == -1)
+ fprintf(stderr, "flushctx: error ioctl %s: %s\n",
+ mp, strerror(errno));
+
+ close(fd);
+ return rc;
+}
+
+static int lfs_flushctx(int argc, char **argv)
+{
+ int kdestroy = 0, c;
+ FILE *proc;
+ char procline[PATH_MAX], *line;
+ int rc = 0;
+
+ optind = 0;
+ while ((c = getopt(argc, argv, "k")) != -1) {
+ switch (c) {
+ case 'k':
+ kdestroy = 1;
+ break;
+ default:
+ fprintf(stderr, "error: %s: option '-%c' "
+ "unrecognized\n", argv[0], c);
+ return CMD_HELP;
+ }
+ }
+
+ if (kdestroy)
+ system("kdestroy > /dev/null");
+
+ if (optind >= argc) {
+ /* flush for all mounted lustre fs. */
+ proc = fopen("/proc/mounts", "r");
+ if (!proc) {
+ fprintf(stderr, "error: %s: can't open /proc/mounts\n",
+ argv[0]);
+ return -1;
+ }
+
+ while ((line = fgets(procline, PATH_MAX, proc)) != NULL) {
+ char dev[PATH_MAX];
+ char mp[PATH_MAX];
+ char fs[PATH_MAX];
+
+ if (sscanf(line, "%s %s %s", dev, mp, fs) != 3) {
+ fprintf(stderr, "%s: unexpected format in "
+ "/proc/mounts\n",
+ argv[0]);
+ return -1;
+ }
+
+ if (strcmp(fs, "lustre") != 0)
+ continue;
+ /* we use '@' to determine it's a client. are there
+ * any other better way?
+ */
+ if (strchr(dev, '@') == NULL)
+ continue;
+
+ if (flushctx_ioctl(mp))
+ rc = -1;
+ }
+ } else {
+ /* flush fs as specified */
+ while (optind < argc) {
+ if (flushctx_ioctl(argv[optind++]))
+ rc = -1;
+ }
+ }
+
+ return rc;
+}
+
+static int lfs_lsetfacl(int argc, char **argv)
+{
+ argv[0]++;
+ return(llapi_lsetfacl(argc, argv));
+}
+
+static int lfs_lgetfacl(int argc, char **argv)
+{
+ argv[0]++;
+ return(llapi_lgetfacl(argc, argv));
+}
+
+static int lfs_rsetfacl(int argc, char **argv)
+{
+ argv[0]++;
+ return(llapi_rsetfacl(argc, argv));
+}
+
+static int lfs_rgetfacl(int argc, char **argv)
+{
+ argv[0]++;
+ return(llapi_rgetfacl(argc, argv));
+}
+
+static int lfs_cp(int argc, char **argv)
+{
+ return(llapi_cp(argc, argv));
+}
+
+static int lfs_ls(int argc, char **argv)
+{
+ return(llapi_ls(argc, argv));
+}
+
+/* A helper function to return single, whole lines delimited by newline.
+ Returns length of line. Not reentrant! */
+static int get_next_full_line(int fd, char **ptr)
+{
+ static char buf[8192]; /* bigger than MAX_PATH_LENGTH */
+ static char *sptr = buf, *eptr = buf;
+ static int len, rem;
+
+ if ((*ptr == NULL) /* first time */
+ || (eptr >= buf + len) /* buffer empty */) {
+ sptr = eptr = buf;
+ len = read(fd, buf, sizeof(buf));
+ if (len <= 0)
+ return len;
+ } else {
+ sptr = eptr + 1;
+ }
+
+full_line:
+ while (eptr < buf + len) {
+ eptr++;
+ /* parse full lines */
+ if (*eptr == '\n') {
+ *eptr = '\0';
+ *ptr = sptr;
+ return (eptr - sptr);
+ }
+ }
+
+ /* partial line; move to front of buf */
+ rem = buf + len - sptr;
+ memcpy(buf, sptr, rem);
+ sptr = buf;
+ eptr = buf + rem;
+ len = read(fd, eptr, sizeof(buf) - rem);
+ if (len <= 0)
+ return len;
+ len += rem;
+ goto full_line;
+}
+
+static int lfs_changelog(int argc, char **argv)
+{
+ long long startrec = 0, endrec = 0, recnum;
+ int fd, len;
+ char c, *mdd, *ptr = NULL;
+ struct option long_opts[] = {
+ {"follow", 0, 0, 'f'},
+ {0, 0, 0, 0}
+ };
+ char short_opts[] = "f";
+ int follow = 0;
+
+ optind = 0;
+ while ((c = getopt_long(argc, argv, short_opts,
+ long_opts, NULL)) != -1) {
+ switch (c) {
+ case 'f':
+ follow++;
+ break;
+ case '?':
+ return CMD_HELP;
+ default:
+ fprintf(stderr, "error: %s: option '%s' unrecognized\n",
+ argv[0], argv[optind - 1]);
+ return CMD_HELP;
+ }
+ }
+ if (optind >= argc)
+ return CMD_HELP;
+
+ mdd = argv[optind++];
+ if (argc > optind)
+ startrec = strtoll(argv[optind++], NULL, 10);
+ if (argc > optind)
+ endrec = strtoll(argv[optind++], NULL, 10);
+
+ fd = llapi_changelog_open(mdd, startrec);
+ if (fd < 0)
+ return fd;
+
+ while ((len = get_next_full_line(fd, &ptr)) >= 0) {
+ if (len == 0) {
+ struct pollfd pfds[1];
+ int rc;
+
+ if (!follow)
+ break;
+ pfds[0].fd = fd;
+ pfds[0].events = POLLIN;
+ rc = poll(pfds, 1, -1);
+ if (rc < 0)
+ break;
+ continue;
+ }
+ /* eg. 2 02MKDIR 4405821890 t=[0x100000400/0x5] p=[0x100000400/0x4] pics */
+ sscanf(ptr, "%lld *", &recnum);
+ if (endrec && recnum > endrec)
+ break;
+ if (recnum < startrec)
+ continue;
+ printf("%.*s\n", len, ptr);
+ }
+
+ close(fd);
+
+ if (len < 0) {
+ printf("read err %d\n", errno);
+ return -errno;
+ }
+
+ return 0;
+}
+
+static int lfs_changelog_clear(int argc, char **argv)
+{
+ long long endrec;
+
+ if (argc != 3)
+ return CMD_HELP;
+
+ endrec = strtoll(argv[2], NULL, 10);
+
+ return(llapi_changelog_clear(argv[1], endrec));
+}
+
+static int lfs_fid2path(int argc, char **argv)
+{
+ struct option long_opts[] = {
+ {"link", 1, 0, 'l'},
+ {"rec", 1, 0, 'r'},
+ {0, 0, 0, 0}
+ };
+ char c, short_opts[] = "l:r:";
+ char *device, *fid, *path;
+ long long recno = -1;
+ int linkno = -1;
+ int lnktmp;
+ int rc;
+
+ optind = 0;
+ while ((c = getopt_long(argc, argv, short_opts,
+ long_opts, NULL)) != -1) {
+ switch (c) {
+ case 'l':
+ linkno = strtol(optarg, NULL, 10);
+ break;
+ case 'r':
+ recno = strtoll(optarg, NULL, 10);
+ break;
+ case '?':
+ return CMD_HELP;
+ default:
+ fprintf(stderr, "error: %s: option '%s' unrecognized\n",
+ argv[0], argv[optind - 1]);
+ return CMD_HELP;
+ }
+ }
+
+ device = argv[optind++];
+ fid = argv[optind++];
+ if (optind != argc)
+ return CMD_HELP;
+
+ path = calloc(1, PATH_MAX);
+
+ lnktmp = (linkno >= 0) ? linkno : 0;
+ while (1) {
+ int oldtmp = lnktmp;
+ rc = llapi_fid2path(device, fid, path, PATH_MAX, recno,
+ &lnktmp);
+ if (rc < 0) {
+ fprintf(stderr, "%s error: %s\n", argv[0],
+ strerror(errno = -rc));
+ break;
+ } else {
+ fprintf(stdout, "%s\n", path);
+ }
+ if (linkno >= 0)
+ /* specified linkno */
+ break;
+ if (oldtmp == lnktmp)
+ /* no more links */
+ break;
+ }
+
+ free(path);
+ return rc;
+}
+
+static int lfs_path2fid(int argc, char **argv)
+{
+ char *path;
+ unsigned long long seq;
+ unsigned long oid, ver;
+ int rc;
+
+ if (argc != 2)
+ return CMD_HELP;
+
+ path = argv[1];
+ rc = llapi_path2fid(path, &seq, &oid, &ver);
+ if (rc) {
+ fprintf(stderr, "can't get fid for %s: %s\n", path,
+ strerror(errno = -rc));
+ return rc;
+ }
+
+ printf(DFID"\n", seq, (unsigned int)oid, (unsigned int)ver);
+
+ return 0;
+}
+
+int main(int argc, char **argv)
+{
+ int rc;
+
+ setlinebuf(stdout);
+
+ ptl_initialize(argc, argv);
+ if (obd_initialize(argc, argv) < 0)
+ exit(2);
+ if (dbg_initialize(argc, argv) < 0)
+ exit(3);
+
+ Parser_init("lfs > ", cmdlist);
+
+ if (argc > 1) {
+ rc = Parser_execarg(argc - 1, argv + 1, cmdlist);
+ } else {
+ rc = Parser_commands();
+ }
+
+ obd_finalize(argc, argv);
+ return rc;
+}
+