/*
- * $Source$
- * $Author$
- * $Header$
+ * $Source$
+ * $Author$
+ * $Header$
*
- * Copyright (C) 1987 by the Massachusetts Institute of Technology
+ * Copyright (C) 1987 by the Massachusetts Institute of Technology
*
- * Set quota on specified device for specified user to specified value.
+ * Set quota on specified device for specified user to specified value.
*
- * Uses the NFS style quota system/quotactl rather than the Melbourne
+ * Uses the NFS style quota system/quotactl rather than the Melbourne
* quota system.
- *
- * $Log$
- * Revision 1.2 1988-10-06 10:41:55 raeburn
- * (wesommer?) implemented -u flag to update quota values with increment
- * from current.
*
- * Revision 1.1 87/09/04 21:32:47 wesommer
- * Initial revision
- *
*/
#ifndef lint
#include <stdio.h>
#include <ctype.h>
+#include <mntent.h>
+
+#include <sys/file.h>
#include <sys/param.h>
#include <sys/time.h>
#include <ufs/quota.h>
#define kb(n) (howmany(dbtob(n), 1024))
-main(argc, argv)
- int argc;
- char **argv;
+static char device[20];
+static char quotafilename[30];
+static struct dqblk zblk = {0};
+
+int main(int argc, char **argv)
{
- char *device;
- int uid;
- struct dqblk db, odb;
- int uflag = 0;
-
- while (argc > 4 && *argv[1] == '-') {
- switch(argv[1][1]) {
- case 'u':
- uflag = 1;
- --argc;
- ++argv;
- break;
- default:
- goto usage;
- }
+ int uid, uid_low, uid_high, soft_quota, qfd;
+ struct dqblk db, odb;
+ int uflag = 0;
+ int range_mode = 0;
+
+ while (argc > 4 && *argv[1] == '-')
+ {
+ switch (argv[1][1])
+ {
+ case 'u':
+ uflag = 1;
+ --argc;
+ ++argv;
+ break;
+ case 'r':
+ range_mode = 1;
+ --argc;
+ ++argv;
+ break;
+ default:
+ goto usage;
}
-
- if (argc != 4) {
- usage:
- fprintf(stderr, "usage: setquota [-u] special uid quota\n\
--u means set limit to <quota> + cur usage\n\
-special is a mounted filesystem special device\n\
-quota is in 1KB units\n");
- exit(1);
+ }
+
+ if ((argc != 4 && !range_mode) || (argc != 5 && range_mode))
+ {
+ usage:
+ fprintf(stderr, "usage: setquota [-u] special uid quota\n"
+ "setquota -r [-u] special uid_low uid_high quota\n"
+ "-u means set limit to <quota> + cur usage\n"
+ "special is a mounted filesystem special device\n"
+ "quota is in 1KB units\n");
+ exit(1);
+ }
+
+ if ((!range_mode && (!isdigit(*argv[2]) || !isdigit(*argv[3]))) ||
+ (range_mode && (!isdigit(*argv[2]) || !isdigit(*argv[3]) ||
+ !isdigit(*argv[4]))))
+ {
+ fprintf(stderr, "setquota: uid and quota must be numeric\n");
+ goto usage;
+ }
+
+ if (range_mode)
+ {
+ uid_low = atoi(argv[2]);
+ uid_high = atoi(argv[3]);
+ soft_quota = atoi(argv[4]);
+ if (uid_low > uid_high)
+ {
+ fprintf(stderr, "setquota: range error\n");
+ exit(1);
}
+ }
+ else
+ {
+ uid_low = uid_high = atoi(argv[2]);
+ soft_quota = atoi(argv[3]);
+ }
+
+ get_device(argv[1]);
+
+ for (uid = uid_low; uid <= uid_high; uid++)
+ {
+ if (quotactl(Q_GETQUOTA, device, uid, &odb))
+ {
+ if (!(qfd = open(quotafilename, O_RDWR)))
+ {
+ perror("No quota file");
+ exit(1);
+ }
- if (!isdigit(*argv[2]) || !isdigit(*argv[3])) {
- fprintf(stderr, "setquota: uid and quota must be numeric\n");
- goto usage;
+ lseek(qfd, 32767 * sizeof(struct dqblk), L_SET);
+ write(qfd, &zblk, sizeof(struct dqblk));
+ close(qfd);
+
+ if (quotactl(Q_GETQUOTA, device, uid, &odb) != 0)
+ {
+ perror("Can't get current quota info");
+ exit(1);
+ }
}
-
- device = argv[1];
- uid = atoi(argv[2]);
-
- db.dqb_bsoftlimit = atoi(argv[3]);
- db.dqb_bhardlimit = db.dqb_bsoftlimit * 1.2;
- db.dqb_fsoftlimit = atoi(argv[3]) * .5;
- db.dqb_fhardlimit = db.dqb_fsoftlimit * 1.2;
- db.dqb_btimelimit = DQ_FTIMELIMIT;
- db.dqb_btimelimit = DQ_BTIMELIMIT;
-
- db.dqb_bsoftlimit *= btodb(1024);
- db.dqb_bhardlimit *= btodb(1024);
-
- if (uflag) {
- if (quotactl(Q_GETQUOTA, device, uid, &odb) == 0) {
- db.dqb_bhardlimit += odb.dqb_curblocks;
- db.dqb_bsoftlimit += odb.dqb_curblocks;
- db.dqb_fhardlimit += odb.dqb_curfiles;
- db.dqb_fsoftlimit += odb.dqb_curfiles;
- }
+
+ db.dqb_bsoftlimit = soft_quota;
+ db.dqb_bhardlimit = (db.dqb_bsoftlimit * 6) / 5;
+ db.dqb_fsoftlimit = soft_quota / 2;
+ db.dqb_fhardlimit = (db.dqb_fsoftlimit * 6) / 5;
+ db.dqb_btimelimit = odb.dqb_btimelimit;
+ db.dqb_ftimelimit = odb.dqb_ftimelimit;
+
+ db.dqb_bsoftlimit *= btodb(1024);
+ db.dqb_bhardlimit *= btodb(1024);
+
+ if (uflag)
+ {
+ db.dqb_bhardlimit += odb.dqb_curblocks;
+ db.dqb_bsoftlimit += odb.dqb_curblocks;
+ db.dqb_fhardlimit += odb.dqb_curfiles;
+ db.dqb_fsoftlimit += odb.dqb_curfiles;
}
-
- if (quotactl(Q_SETQLIM, device, uid, &db) < 0) {
- fprintf (stderr, "quotactl: %d on ", uid);
- perror (device);
- exit (1);
+
+ if (quotactl(Q_SETQLIM, device, uid, &db) < 0)
+ {
+ fprintf(stderr, "quotactl: %d on ", uid);
+ perror(device);
+ exit(1);
}
- if (quotactl(Q_SYNC, device, 0, 0) < 0) {
- perror ("can't sync disk quota");
- exit (1);
+ }
+
+ if (quotactl(Q_SYNC, device, 0, 0) < 0)
+ {
+ perror("can't sync disk quota");
+ exit(1);
+ }
+
+ exit(0);
+}
+
+get_device(char *device_or_dir)
+{
+ struct mntent *mntp;
+ FILE *fstab;
+
+ fstab = setmntent(MNTTAB, "r");
+ while (mntp = getmntent(fstab))
+ {
+ if (!strcmp(mntp->mnt_fsname, device_or_dir) ||
+ !strcmp(mntp->mnt_dir, device_or_dir))
+ {
+ strcpy(device, mntp->mnt_fsname);
+ sprintf(quotafilename, "%s/quotas", mntp->mnt_dir);
+ endmntent(fstab);
+ return;
}
- exit (0);
+ }
+ fprintf(stderr, "%s not mounted.\n", device_or_dir);
+ exit(1);
}
-
andersk / moira.git / blobdiff