(mar) Don't expire timers on users currently over quota.

[moira.git] / gen / setquota.c

/*
 *	$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.
 * 	
 *	$Log$
 *	Revision 1.3  1988-10-06 10:46:08  raeburn
 *	(mar) Don't expire timers on users currently over quota.
 *
 * Revision 1.1  87/09/04  21:32:47  wesommer
 * Initial revision
 * 
 */

#ifndef lint
static char *rcsid_setquota_c = "$Header$";
#endif lint

#include <stdio.h>
#include <ctype.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;
{
	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;
		}
	}
	
	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 (!isdigit(*argv[2]) || !isdigit(*argv[3])) {
		fprintf(stderr, "setquota: uid and quota must be numeric\n");
		goto usage;
	}
	
	device = argv[1];
	uid = atoi(argv[2]);
	
	if (quotactl(Q_GETQUOTA, device, uid, &odb) != 0) {
	    perror("Can't get current quota info");
	    exit(1);
	}

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