X-Git-Url: http://andersk.mit.edu/gitweb/moira.git/blobdiff_plain/5dc7ec18c05c6b2596f76fdea04f3f1399dbde35..0041862dcee4646f8bc7fea981e1fb7007cc066f:/gen/setquota.c

diff --git a/gen/setquota.c b/gen/setquota.c
index 3190685f..5bcf8100 100644
--- a/gen/setquota.c
+++ b/gen/setquota.c
@@ -1,18 +1,14 @@
 /*
- *	$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.1  1987-09-04 21:32:47  wesommer
- *	Initial revision
  *
  */
 
@@ -22,55 +18,156 @@ static char *rcsid_setquota_c = "$Header$";
 
 #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;
-	
-	if (argc != 4) {
-	usage:
-		fprintf(stderr, "usage: setquota special uid quota\n\
-special is a mounted filesystem special device\n\
-quota is in 1KB units\n");
-		exit(1);
+  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 (!isdigit(*argv[2]) || !isdigit(*argv[3])) {
-		fprintf(stderr, "setquota: uid and quota must be numeric\n");
-		goto usage;
+  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);
 	}
-	
-	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 (quotactl(Q_SETQLIM, device, uid, &db) < 0) {
-		fprintf (stderr, "quotactl: %d on ", uid);
-		perror (device);
-		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);
+	    }
+
+	  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);
+	    }
 	}
-	if (quotactl(Q_SYNC, device, 0, 0) < 0) {
-		perror ("can't sync disk quota");
-		exit (1);
+
+      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_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);
 }
-