/*
 *      $Source$
 *      $Author$
 *      $Header$
 *
 *      Copyright (C) 1987 by the Massachusetts Institute of Technology
 *
 *      Set quota on specified device for specified user to specified value.
 *
 *      Uses the NFS style quota system/quotactl rather than the Melbourne
 * quota system.
 *
 */

#ifndef lint
static char *rcsid_setquota_c = "$Header$";
#endif 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))

static char device[20];
static char quotafilename[30];
static struct dqblk zblk = {0};

int main(int argc, char **argv)
{
  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 && !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);
	    }

	  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);
	    }
	}

      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;
	}
    }
  fprintf(stderr, "%s not mounted.\n", device_or_dir);
  exit(1);
}