Home | History | Annotate | Line # | Download | only in edquota
edquota.c revision 1.23
      1   1.1      cgd /*
      2   1.5  mycroft  * Copyright (c) 1980, 1990, 1993
      3   1.5  mycroft  *	The Regents of the University of California.  All rights reserved.
      4   1.1      cgd  *
      5   1.1      cgd  * This code is derived from software contributed to Berkeley by
      6   1.1      cgd  * Robert Elz at The University of Melbourne.
      7   1.1      cgd  *
      8   1.1      cgd  * Redistribution and use in source and binary forms, with or without
      9   1.1      cgd  * modification, are permitted provided that the following conditions
     10   1.1      cgd  * are met:
     11   1.1      cgd  * 1. Redistributions of source code must retain the above copyright
     12   1.1      cgd  *    notice, this list of conditions and the following disclaimer.
     13   1.1      cgd  * 2. Redistributions in binary form must reproduce the above copyright
     14   1.1      cgd  *    notice, this list of conditions and the following disclaimer in the
     15   1.1      cgd  *    documentation and/or other materials provided with the distribution.
     16   1.1      cgd  * 3. All advertising materials mentioning features or use of this software
     17   1.1      cgd  *    must display the following acknowledgement:
     18   1.1      cgd  *	This product includes software developed by the University of
     19   1.1      cgd  *	California, Berkeley and its contributors.
     20   1.1      cgd  * 4. Neither the name of the University nor the names of its contributors
     21   1.1      cgd  *    may be used to endorse or promote products derived from this software
     22   1.1      cgd  *    without specific prior written permission.
     23   1.1      cgd  *
     24   1.1      cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     25   1.1      cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     26   1.1      cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     27   1.1      cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     28   1.1      cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     29   1.1      cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     30   1.1      cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     31   1.1      cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     32   1.1      cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     33   1.1      cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     34   1.1      cgd  * SUCH DAMAGE.
     35   1.1      cgd  */
     36   1.1      cgd 
     37  1.15    lukem #include <sys/cdefs.h>
     38   1.1      cgd #ifndef lint
     39  1.15    lukem __COPYRIGHT("@(#) Copyright (c) 1980, 1990, 1993\n\
     40  1.15    lukem 	The Regents of the University of California.  All rights reserved.\n");
     41   1.1      cgd #endif /* not lint */
     42   1.1      cgd 
     43   1.1      cgd #ifndef lint
     44  1.15    lukem #if 0
     45  1.15    lukem static char sccsid[] = "from: @(#)edquota.c	8.3 (Berkeley) 4/27/95";
     46  1.15    lukem #else
     47  1.23     jonb __RCSID("$NetBSD: edquota.c,v 1.23 2002/12/05 22:54:45 jonb Exp $");
     48  1.15    lukem #endif
     49   1.1      cgd #endif /* not lint */
     50   1.1      cgd 
     51   1.1      cgd /*
     52   1.1      cgd  * Disk quota editor.
     53   1.1      cgd  */
     54   1.1      cgd #include <sys/param.h>
     55   1.1      cgd #include <sys/stat.h>
     56   1.1      cgd #include <sys/file.h>
     57   1.1      cgd #include <sys/wait.h>
     58  1.12    mikel #include <sys/queue.h>
     59   1.4      cgd #include <ufs/ufs/quota.h>
     60  1.15    lukem #include <err.h>
     61   1.1      cgd #include <errno.h>
     62   1.1      cgd #include <fstab.h>
     63   1.1      cgd #include <pwd.h>
     64   1.1      cgd #include <grp.h>
     65   1.1      cgd #include <ctype.h>
     66  1.15    lukem #include <signal.h>
     67   1.1      cgd #include <stdio.h>
     68   1.9      cgd #include <stdlib.h>
     69   1.1      cgd #include <string.h>
     70   1.5  mycroft #include <unistd.h>
     71   1.1      cgd #include "pathnames.h"
     72   1.1      cgd 
     73   1.1      cgd char *qfname = QUOTAFILENAME;
     74   1.1      cgd char *qfextension[] = INITQFNAMES;
     75   1.1      cgd char *quotagroup = QUOTAGROUP;
     76   1.1      cgd char tmpfil[] = _PATH_TMP;
     77   1.1      cgd 
     78   1.1      cgd struct quotause {
     79   1.1      cgd 	struct	quotause *next;
     80   1.1      cgd 	long	flags;
     81   1.1      cgd 	struct	dqblk dqblk;
     82   1.1      cgd 	char	fsname[MAXPATHLEN + 1];
     83   1.1      cgd 	char	qfname[1];	/* actually longer */
     84  1.15    lukem };
     85   1.1      cgd #define	FOUND	0x01
     86  1.13    mikel 
     87  1.23     jonb #define MAX_TMPSTR	(100+MAXPATHLEN)
     88  1.23     jonb 
     89  1.15    lukem int	main __P((int, char **));
     90  1.13    mikel void	usage __P((void));
     91  1.13    mikel int	getentry __P((char *, int));
     92  1.13    mikel struct quotause *
     93  1.22   bouyer 	getprivs __P((long, int, char *));
     94  1.13    mikel void	putprivs __P((long, int, struct quotause *));
     95  1.13    mikel int	editit __P((char *));
     96  1.13    mikel int	writeprivs __P((struct quotause *, int, char *, int));
     97  1.13    mikel int	readprivs __P((struct quotause *, int));
     98  1.13    mikel int	writetimes __P((struct quotause *, int, int));
     99  1.13    mikel int	readtimes __P((struct quotause *, int));
    100  1.13    mikel char *	cvtstoa __P((time_t));
    101  1.13    mikel int	cvtatos __P((time_t, char *, time_t *));
    102  1.13    mikel void	freeprivs __P((struct quotause *));
    103  1.13    mikel int	alldigits __P((char *));
    104  1.13    mikel int	hasquota __P((struct fstab *, int, char **));
    105   1.1      cgd 
    106  1.12    mikel int
    107   1.1      cgd main(argc, argv)
    108   1.1      cgd 	int argc;
    109  1.15    lukem 	char **argv;
    110   1.1      cgd {
    111  1.15    lukem 	struct quotause *qup, *protoprivs, *curprivs;
    112  1.15    lukem 	long id, protoid;
    113  1.15    lukem 	int quotatype, tmpfd;
    114  1.11     mark 	char *protoname;
    115  1.22   bouyer 	char *soft = NULL, *hard = NULL;
    116  1.22   bouyer 	char *fs = NULL;
    117  1.11     mark 	int ch;
    118   1.1      cgd 	int tflag = 0, pflag = 0;
    119   1.1      cgd 
    120   1.1      cgd 	if (argc < 2)
    121   1.1      cgd 		usage();
    122  1.15    lukem 	if (getuid())
    123  1.15    lukem 		errx(1, "permission denied");
    124  1.15    lukem 	protoname = NULL;
    125   1.1      cgd 	quotatype = USRQUOTA;
    126  1.22   bouyer 	while ((ch = getopt(argc, argv, "ugtp:s:h:f:")) != -1) {
    127   1.1      cgd 		switch(ch) {
    128   1.1      cgd 		case 'p':
    129   1.1      cgd 			protoname = optarg;
    130   1.1      cgd 			pflag++;
    131   1.1      cgd 			break;
    132   1.1      cgd 		case 'g':
    133   1.1      cgd 			quotatype = GRPQUOTA;
    134   1.1      cgd 			break;
    135   1.1      cgd 		case 'u':
    136   1.1      cgd 			quotatype = USRQUOTA;
    137   1.1      cgd 			break;
    138   1.1      cgd 		case 't':
    139   1.1      cgd 			tflag++;
    140   1.1      cgd 			break;
    141  1.22   bouyer 		case 's':
    142  1.22   bouyer 			soft = optarg;
    143  1.22   bouyer 			break;
    144  1.22   bouyer 		case 'h':
    145  1.22   bouyer 			hard = optarg;
    146  1.22   bouyer 			break;
    147  1.22   bouyer 		case 'f':
    148  1.22   bouyer 			fs = optarg;
    149  1.22   bouyer 			break;
    150   1.1      cgd 		default:
    151   1.1      cgd 			usage();
    152   1.1      cgd 		}
    153   1.1      cgd 	}
    154   1.1      cgd 	argc -= optind;
    155   1.1      cgd 	argv += optind;
    156   1.1      cgd 	if (pflag) {
    157  1.22   bouyer 		if (soft || hard)
    158  1.22   bouyer 			usage();
    159   1.1      cgd 		if ((protoid = getentry(protoname, quotatype)) == -1)
    160   1.1      cgd 			exit(1);
    161  1.22   bouyer 		protoprivs = getprivs(protoid, quotatype, fs);
    162   1.1      cgd 		for (qup = protoprivs; qup; qup = qup->next) {
    163   1.1      cgd 			qup->dqblk.dqb_btime = 0;
    164   1.1      cgd 			qup->dqblk.dqb_itime = 0;
    165   1.1      cgd 		}
    166   1.1      cgd 		while (argc-- > 0) {
    167   1.1      cgd 			if ((id = getentry(*argv++, quotatype)) < 0)
    168   1.1      cgd 				continue;
    169   1.1      cgd 			putprivs(id, quotatype, protoprivs);
    170   1.1      cgd 		}
    171   1.1      cgd 		exit(0);
    172   1.1      cgd 	}
    173  1.22   bouyer 	if (soft || hard) {
    174  1.22   bouyer 		struct quotause *qup;
    175  1.22   bouyer 		u_int32_t softb, hardb, softi, hardi;
    176  1.22   bouyer 		if (tflag)
    177  1.22   bouyer 			usage();
    178  1.22   bouyer 		if (soft) {
    179  1.22   bouyer 			if (sscanf(soft, "%d/%d", &softb, &softi) != 2)
    180  1.22   bouyer 				usage();
    181  1.22   bouyer 			softb = btodb((u_quad_t)softb * 1024);
    182  1.22   bouyer 		}
    183  1.22   bouyer 		if (hard) {
    184  1.22   bouyer 			if (sscanf(hard, "%d/%d", &hardb, &hardi) != 2)
    185  1.22   bouyer 				usage();
    186  1.22   bouyer 			hardb = btodb((u_quad_t)hardb * 1024);
    187  1.22   bouyer 		}
    188  1.22   bouyer 		for ( ; argc > 0; argc--, argv++) {
    189  1.22   bouyer 			if ((id = getentry(*argv, quotatype)) == -1)
    190  1.22   bouyer 				continue;
    191  1.22   bouyer 			curprivs = getprivs(id, quotatype, fs);
    192  1.22   bouyer 			for (qup = curprivs; qup; qup = qup->next) {
    193  1.22   bouyer 				if (soft) {
    194  1.22   bouyer 					if (softb &&
    195  1.22   bouyer 					    qup->dqblk.dqb_curblocks >= softb &&
    196  1.22   bouyer 					    (qup->dqblk.dqb_bsoftlimit == 0 ||
    197  1.22   bouyer 					    qup->dqblk.dqb_curblocks <
    198  1.22   bouyer 					    qup->dqblk.dqb_bsoftlimit))
    199  1.22   bouyer 						qup->dqblk.dqb_btime = 0;
    200  1.22   bouyer 					if (softi &&
    201  1.22   bouyer 					    qup->dqblk.dqb_curinodes >= softi &&
    202  1.22   bouyer 					    (qup->dqblk.dqb_isoftlimit == 0 ||
    203  1.22   bouyer 					    qup->dqblk.dqb_curinodes <
    204  1.22   bouyer 					    qup->dqblk.dqb_isoftlimit))
    205  1.22   bouyer 						qup->dqblk.dqb_itime = 0;
    206  1.22   bouyer 					qup->dqblk.dqb_bsoftlimit = softb;
    207  1.22   bouyer 					qup->dqblk.dqb_isoftlimit = softi;
    208  1.22   bouyer 				}
    209  1.22   bouyer 				if (hard) {
    210  1.22   bouyer 					qup->dqblk.dqb_bhardlimit = hardb;
    211  1.22   bouyer 					qup->dqblk.dqb_ihardlimit = hardi;
    212  1.22   bouyer 				}
    213  1.22   bouyer 			}
    214  1.22   bouyer 			putprivs(id, quotatype, curprivs);
    215  1.22   bouyer 			freeprivs(curprivs);
    216  1.22   bouyer 		}
    217  1.22   bouyer 		exit(0);
    218  1.22   bouyer 	}
    219   1.1      cgd 	tmpfd = mkstemp(tmpfil);
    220   1.1      cgd 	fchown(tmpfd, getuid(), getgid());
    221   1.1      cgd 	if (tflag) {
    222  1.22   bouyer 		if (soft || hard)
    223  1.22   bouyer 			usage();
    224  1.22   bouyer 		protoprivs = getprivs(0, quotatype, fs);
    225   1.1      cgd 		if (writetimes(protoprivs, tmpfd, quotatype) == 0)
    226   1.1      cgd 			exit(1);
    227   1.1      cgd 		if (editit(tmpfil) && readtimes(protoprivs, tmpfd))
    228   1.1      cgd 			putprivs(0, quotatype, protoprivs);
    229   1.1      cgd 		freeprivs(protoprivs);
    230   1.1      cgd 		exit(0);
    231   1.1      cgd 	}
    232   1.1      cgd 	for ( ; argc > 0; argc--, argv++) {
    233   1.1      cgd 		if ((id = getentry(*argv, quotatype)) == -1)
    234   1.1      cgd 			continue;
    235  1.22   bouyer 		curprivs = getprivs(id, quotatype, fs);
    236   1.1      cgd 		if (writeprivs(curprivs, tmpfd, *argv, quotatype) == 0)
    237   1.1      cgd 			continue;
    238   1.1      cgd 		if (editit(tmpfil) && readprivs(curprivs, tmpfd))
    239   1.1      cgd 			putprivs(id, quotatype, curprivs);
    240   1.1      cgd 		freeprivs(curprivs);
    241   1.1      cgd 	}
    242   1.1      cgd 	close(tmpfd);
    243   1.1      cgd 	unlink(tmpfil);
    244   1.1      cgd 	exit(0);
    245   1.1      cgd }
    246   1.1      cgd 
    247  1.12    mikel void
    248   1.1      cgd usage()
    249   1.1      cgd {
    250  1.22   bouyer 	fprintf(stderr,
    251  1.22   bouyer 	    "Usage: edquota [-u] [-p username] [-f filesystem] username ...\n"
    252  1.22   bouyer 	    "\tedquota -g [-p groupname] [-f filesystem] groupname ...\n"
    253  1.22   bouyer 	    "\tedquota [-u] [-f filesystem] [-s b#/i#] [-h b#/i#] username ...\n"
    254  1.22   bouyer 	    "\tedquota -g [-f filesystem] [-s b#/i#] [-h b#/i#] groupname ...\n"
    255  1.22   bouyer 	    "\tedquota [-u] [-f filesystem] -t\n"
    256  1.22   bouyer 	    "\tedquota -g [-f filesystem] -t\n"
    257  1.22   bouyer 	    );
    258   1.1      cgd 	exit(1);
    259   1.1      cgd }
    260   1.1      cgd 
    261   1.1      cgd /*
    262   1.1      cgd  * This routine converts a name for a particular quota type to
    263   1.1      cgd  * an identifier. This routine must agree with the kernel routine
    264   1.1      cgd  * getinoquota as to the interpretation of quota types.
    265   1.1      cgd  */
    266  1.12    mikel int
    267   1.1      cgd getentry(name, quotatype)
    268   1.1      cgd 	char *name;
    269   1.1      cgd 	int quotatype;
    270   1.1      cgd {
    271   1.1      cgd 	struct passwd *pw;
    272   1.1      cgd 	struct group *gr;
    273   1.1      cgd 
    274   1.1      cgd 	if (alldigits(name))
    275   1.1      cgd 		return (atoi(name));
    276   1.1      cgd 	switch(quotatype) {
    277   1.1      cgd 	case USRQUOTA:
    278  1.15    lukem 		if ((pw = getpwnam(name)) != NULL)
    279   1.1      cgd 			return (pw->pw_uid);
    280  1.15    lukem 		warnx("%s: no such user", name);
    281   1.1      cgd 		break;
    282   1.1      cgd 	case GRPQUOTA:
    283  1.15    lukem 		if ((gr = getgrnam(name)) != NULL)
    284   1.1      cgd 			return (gr->gr_gid);
    285  1.15    lukem 		warnx("%s: no such group", name);
    286   1.1      cgd 		break;
    287   1.1      cgd 	default:
    288  1.15    lukem 		warnx("%d: unknown quota type", quotatype);
    289   1.1      cgd 		break;
    290   1.1      cgd 	}
    291   1.1      cgd 	sleep(1);
    292   1.1      cgd 	return (-1);
    293   1.1      cgd }
    294   1.1      cgd 
    295   1.1      cgd /*
    296   1.1      cgd  * Collect the requested quota information.
    297   1.1      cgd  */
    298   1.1      cgd struct quotause *
    299  1.22   bouyer getprivs(id, quotatype, filesys)
    300  1.15    lukem 	long id;
    301   1.1      cgd 	int quotatype;
    302  1.22   bouyer 	char *filesys;
    303   1.1      cgd {
    304  1.15    lukem 	struct fstab *fs;
    305  1.15    lukem 	struct quotause *qup, *quptail;
    306   1.1      cgd 	struct quotause *quphead;
    307   1.1      cgd 	int qcmd, qupsize, fd;
    308   1.1      cgd 	char *qfpathname;
    309   1.1      cgd 	static int warned = 0;
    310   1.1      cgd 
    311   1.1      cgd 	setfsent();
    312  1.15    lukem 	quptail = NULL;
    313   1.1      cgd 	quphead = (struct quotause *)0;
    314   1.1      cgd 	qcmd = QCMD(Q_GETQUOTA, quotatype);
    315  1.15    lukem 	while ((fs = getfsent()) != NULL) {
    316  1.10      jtc 		if (strcmp(fs->fs_vfstype, "ffs"))
    317  1.22   bouyer 			continue;
    318  1.22   bouyer 		if (filesys && strcmp(fs->fs_spec, filesys) != 0 &&
    319  1.22   bouyer 		    strcmp(fs->fs_file, filesys) != 0)
    320   1.1      cgd 			continue;
    321   1.1      cgd 		if (!hasquota(fs, quotatype, &qfpathname))
    322   1.1      cgd 			continue;
    323   1.1      cgd 		qupsize = sizeof(*qup) + strlen(qfpathname);
    324  1.15    lukem 		if ((qup = (struct quotause *)malloc(qupsize)) == NULL)
    325  1.15    lukem 			errx(2, "out of memory");
    326   1.1      cgd 		if (quotactl(fs->fs_file, qcmd, id, &qup->dqblk) != 0) {
    327   1.1      cgd 	    		if (errno == EOPNOTSUPP && !warned) {
    328   1.1      cgd 				warned++;
    329  1.15    lukem 				warnx(
    330   1.1      cgd 				    "Quotas are not compiled into this kernel");
    331   1.1      cgd 				sleep(3);
    332   1.1      cgd 			}
    333   1.1      cgd 			if ((fd = open(qfpathname, O_RDONLY)) < 0) {
    334   1.1      cgd 				fd = open(qfpathname, O_RDWR|O_CREAT, 0640);
    335   1.1      cgd 				if (fd < 0 && errno != ENOENT) {
    336  1.15    lukem 					warnx("open `%s'", qfpathname);
    337   1.1      cgd 					free(qup);
    338   1.1      cgd 					continue;
    339   1.1      cgd 				}
    340  1.15    lukem 				warnx("Creating quota file %s", qfpathname);
    341   1.1      cgd 				sleep(3);
    342   1.1      cgd 				(void) fchown(fd, getuid(),
    343   1.1      cgd 				    getentry(quotagroup, GRPQUOTA));
    344   1.1      cgd 				(void) fchmod(fd, 0640);
    345   1.1      cgd 			}
    346  1.14   kleink 			(void)lseek(fd, (off_t)(id * sizeof(struct dqblk)),
    347  1.14   kleink 			    SEEK_SET);
    348   1.1      cgd 			switch (read(fd, &qup->dqblk, sizeof(struct dqblk))) {
    349   1.1      cgd 			case 0:			/* EOF */
    350   1.1      cgd 				/*
    351   1.1      cgd 				 * Convert implicit 0 quota (EOF)
    352   1.1      cgd 				 * into an explicit one (zero'ed dqblk)
    353   1.1      cgd 				 */
    354  1.15    lukem 				memset((caddr_t)&qup->dqblk, 0,
    355   1.1      cgd 				    sizeof(struct dqblk));
    356   1.1      cgd 				break;
    357   1.1      cgd 
    358   1.1      cgd 			case sizeof(struct dqblk):	/* OK */
    359   1.1      cgd 				break;
    360   1.1      cgd 
    361   1.1      cgd 			default:		/* ERROR */
    362  1.15    lukem 				warn("read error in `%s'", qfpathname);
    363   1.1      cgd 				close(fd);
    364   1.1      cgd 				free(qup);
    365   1.1      cgd 				continue;
    366   1.1      cgd 			}
    367   1.1      cgd 			close(fd);
    368   1.1      cgd 		}
    369   1.1      cgd 		strcpy(qup->qfname, qfpathname);
    370   1.1      cgd 		strcpy(qup->fsname, fs->fs_file);
    371   1.1      cgd 		if (quphead == NULL)
    372   1.1      cgd 			quphead = qup;
    373   1.1      cgd 		else
    374   1.1      cgd 			quptail->next = qup;
    375   1.1      cgd 		quptail = qup;
    376   1.1      cgd 		qup->next = 0;
    377   1.1      cgd 	}
    378   1.1      cgd 	endfsent();
    379   1.1      cgd 	return (quphead);
    380   1.1      cgd }
    381   1.1      cgd 
    382   1.1      cgd /*
    383   1.1      cgd  * Store the requested quota information.
    384   1.1      cgd  */
    385  1.12    mikel void
    386   1.1      cgd putprivs(id, quotatype, quplist)
    387   1.1      cgd 	long id;
    388   1.1      cgd 	int quotatype;
    389   1.1      cgd 	struct quotause *quplist;
    390   1.1      cgd {
    391  1.15    lukem 	struct quotause *qup;
    392   1.1      cgd 	int qcmd, fd;
    393   1.1      cgd 
    394   1.1      cgd 	qcmd = QCMD(Q_SETQUOTA, quotatype);
    395   1.1      cgd 	for (qup = quplist; qup; qup = qup->next) {
    396   1.1      cgd 		if (quotactl(qup->fsname, qcmd, id, &qup->dqblk) == 0)
    397   1.1      cgd 			continue;
    398   1.1      cgd 		if ((fd = open(qup->qfname, O_WRONLY)) < 0) {
    399  1.15    lukem 			warnx("open `%s'", qup->qfname);
    400   1.1      cgd 		} else {
    401  1.14   kleink 			(void)lseek(fd,
    402  1.14   kleink 			    (off_t)(id * (long)sizeof (struct dqblk)),
    403  1.14   kleink 			    SEEK_SET);
    404   1.1      cgd 			if (write(fd, &qup->dqblk, sizeof (struct dqblk)) !=
    405  1.15    lukem 			    sizeof (struct dqblk))
    406  1.15    lukem 				warnx("writing `%s'", qup->qfname);
    407   1.1      cgd 			close(fd);
    408   1.1      cgd 		}
    409   1.1      cgd 	}
    410   1.1      cgd }
    411   1.1      cgd 
    412   1.1      cgd /*
    413  1.18   simonb  * Take a list of privileges and get it edited.
    414   1.1      cgd  */
    415  1.12    mikel int
    416   1.1      cgd editit(tmpfile)
    417   1.1      cgd 	char *tmpfile;
    418   1.1      cgd {
    419   1.1      cgd 	long omask;
    420   1.1      cgd 	int pid, stat;
    421  1.23     jonb 	char p[MAX_TMPSTR];
    422   1.1      cgd 
    423   1.1      cgd 	omask = sigblock(sigmask(SIGINT)|sigmask(SIGQUIT)|sigmask(SIGHUP));
    424   1.1      cgd  top:
    425   1.1      cgd 	if ((pid = fork()) < 0) {
    426   1.1      cgd 
    427   1.1      cgd 		if (errno == EPROCLIM) {
    428  1.15    lukem 			warnx("You have too many processes");
    429   1.1      cgd 			return(0);
    430   1.1      cgd 		}
    431   1.1      cgd 		if (errno == EAGAIN) {
    432   1.1      cgd 			sleep(1);
    433   1.1      cgd 			goto top;
    434   1.1      cgd 		}
    435  1.15    lukem 		warn("fork");
    436   1.1      cgd 		return (0);
    437   1.1      cgd 	}
    438   1.1      cgd 	if (pid == 0) {
    439  1.15    lukem 		char *ed;
    440   1.1      cgd 
    441   1.1      cgd 		sigsetmask(omask);
    442   1.1      cgd 		setgid(getgid());
    443   1.1      cgd 		setuid(getuid());
    444   1.1      cgd 		if ((ed = getenv("EDITOR")) == (char *)0)
    445   1.1      cgd 			ed = _PATH_VI;
    446  1.23     jonb 		if (strlen(ed) + strlen(tmpfile) + 2 >= MAX_TMPSTR) {
    447  1.23     jonb 			err (1, "%s", "editor or filename too long");
    448  1.23     jonb 		}
    449  1.23     jonb 		snprintf (p, MAX_TMPSTR, "%s %s", ed, tmpfile);
    450  1.23     jonb 		execlp(_PATH_BSHELL, _PATH_BSHELL, "-c", p, NULL);
    451  1.15    lukem 		err(1, "%s", ed);
    452   1.1      cgd 	}
    453   1.1      cgd 	waitpid(pid, &stat, 0);
    454   1.1      cgd 	sigsetmask(omask);
    455   1.1      cgd 	if (!WIFEXITED(stat) || WEXITSTATUS(stat) != 0)
    456   1.1      cgd 		return (0);
    457   1.1      cgd 	return (1);
    458   1.1      cgd }
    459   1.1      cgd 
    460   1.1      cgd /*
    461   1.1      cgd  * Convert a quotause list to an ASCII file.
    462   1.1      cgd  */
    463  1.12    mikel int
    464   1.1      cgd writeprivs(quplist, outfd, name, quotatype)
    465   1.1      cgd 	struct quotause *quplist;
    466   1.1      cgd 	int outfd;
    467   1.1      cgd 	char *name;
    468   1.1      cgd 	int quotatype;
    469   1.1      cgd {
    470  1.15    lukem 	struct quotause *qup;
    471   1.1      cgd 	FILE *fd;
    472   1.1      cgd 
    473   1.1      cgd 	ftruncate(outfd, 0);
    474  1.14   kleink 	(void)lseek(outfd, (off_t)0, SEEK_SET);
    475  1.15    lukem 	if ((fd = fdopen(dup(outfd), "w")) == NULL)
    476  1.15    lukem 		errx(1, "fdopen `%s'", tmpfil);
    477   1.1      cgd 	fprintf(fd, "Quotas for %s %s:\n", qfextension[quotatype], name);
    478   1.1      cgd 	for (qup = quplist; qup; qup = qup->next) {
    479   1.1      cgd 		fprintf(fd, "%s: %s %d, limits (soft = %d, hard = %d)\n",
    480   1.1      cgd 		    qup->fsname, "blocks in use:",
    481  1.19   bouyer 		    (int)(dbtob((u_quad_t)qup->dqblk.dqb_curblocks) / 1024),
    482  1.19   bouyer 		    (int)(dbtob((u_quad_t)qup->dqblk.dqb_bsoftlimit) / 1024),
    483  1.19   bouyer 		    (int)(dbtob((u_quad_t)qup->dqblk.dqb_bhardlimit) / 1024));
    484   1.1      cgd 		fprintf(fd, "%s %d, limits (soft = %d, hard = %d)\n",
    485   1.1      cgd 		    "\tinodes in use:", qup->dqblk.dqb_curinodes,
    486   1.1      cgd 		    qup->dqblk.dqb_isoftlimit, qup->dqblk.dqb_ihardlimit);
    487   1.1      cgd 	}
    488   1.1      cgd 	fclose(fd);
    489   1.1      cgd 	return (1);
    490   1.1      cgd }
    491   1.1      cgd 
    492   1.1      cgd /*
    493   1.1      cgd  * Merge changes to an ASCII file into a quotause list.
    494   1.1      cgd  */
    495  1.12    mikel int
    496   1.1      cgd readprivs(quplist, infd)
    497   1.1      cgd 	struct quotause *quplist;
    498   1.1      cgd 	int infd;
    499   1.1      cgd {
    500  1.15    lukem 	struct quotause *qup;
    501   1.1      cgd 	FILE *fd;
    502   1.1      cgd 	int cnt;
    503  1.15    lukem 	char *cp;
    504   1.1      cgd 	struct dqblk dqblk;
    505   1.1      cgd 	char *fsp, line1[BUFSIZ], line2[BUFSIZ];
    506   1.1      cgd 
    507  1.14   kleink 	(void)lseek(infd, (off_t)0, SEEK_SET);
    508   1.1      cgd 	fd = fdopen(dup(infd), "r");
    509   1.1      cgd 	if (fd == NULL) {
    510  1.15    lukem 		warn("Can't re-read temp file");
    511   1.1      cgd 		return (0);
    512   1.1      cgd 	}
    513   1.1      cgd 	/*
    514   1.1      cgd 	 * Discard title line, then read pairs of lines to process.
    515   1.1      cgd 	 */
    516   1.1      cgd 	(void) fgets(line1, sizeof (line1), fd);
    517   1.1      cgd 	while (fgets(line1, sizeof (line1), fd) != NULL &&
    518   1.1      cgd 	       fgets(line2, sizeof (line2), fd) != NULL) {
    519   1.1      cgd 		if ((fsp = strtok(line1, " \t:")) == NULL) {
    520  1.15    lukem 			warnx("%s: bad format", line1);
    521   1.1      cgd 			return (0);
    522   1.1      cgd 		}
    523   1.1      cgd 		if ((cp = strtok((char *)0, "\n")) == NULL) {
    524  1.15    lukem 			warnx("%s: %s: bad format", fsp,
    525   1.1      cgd 			    &fsp[strlen(fsp) + 1]);
    526   1.1      cgd 			return (0);
    527   1.1      cgd 		}
    528   1.1      cgd 		cnt = sscanf(cp,
    529   1.1      cgd 		    " blocks in use: %d, limits (soft = %d, hard = %d)",
    530   1.1      cgd 		    &dqblk.dqb_curblocks, &dqblk.dqb_bsoftlimit,
    531   1.1      cgd 		    &dqblk.dqb_bhardlimit);
    532   1.1      cgd 		if (cnt != 3) {
    533  1.15    lukem 			warnx("%s:%s: bad format", fsp, cp);
    534   1.1      cgd 			return (0);
    535   1.1      cgd 		}
    536  1.19   bouyer 		dqblk.dqb_curblocks = btodb((u_quad_t)
    537  1.19   bouyer 		    dqblk.dqb_curblocks * 1024);
    538  1.19   bouyer 		dqblk.dqb_bsoftlimit = btodb((u_quad_t)
    539  1.19   bouyer 		    dqblk.dqb_bsoftlimit * 1024);
    540  1.19   bouyer 		dqblk.dqb_bhardlimit = btodb((u_quad_t)
    541  1.19   bouyer 		    dqblk.dqb_bhardlimit * 1024);
    542   1.1      cgd 		if ((cp = strtok(line2, "\n")) == NULL) {
    543  1.15    lukem 			warnx("%s: %s: bad format", fsp, line2);
    544   1.1      cgd 			return (0);
    545   1.1      cgd 		}
    546   1.1      cgd 		cnt = sscanf(cp,
    547   1.1      cgd 		    "\tinodes in use: %d, limits (soft = %d, hard = %d)",
    548   1.1      cgd 		    &dqblk.dqb_curinodes, &dqblk.dqb_isoftlimit,
    549   1.1      cgd 		    &dqblk.dqb_ihardlimit);
    550   1.1      cgd 		if (cnt != 3) {
    551  1.15    lukem 			warnx("%s: %s: bad format", fsp, line2);
    552   1.1      cgd 			return (0);
    553   1.1      cgd 		}
    554   1.1      cgd 		for (qup = quplist; qup; qup = qup->next) {
    555   1.1      cgd 			if (strcmp(fsp, qup->fsname))
    556   1.1      cgd 				continue;
    557   1.1      cgd 			/*
    558   1.1      cgd 			 * Cause time limit to be reset when the quota
    559   1.1      cgd 			 * is next used if previously had no soft limit
    560   1.1      cgd 			 * or were under it, but now have a soft limit
    561   1.1      cgd 			 * and are over it.
    562   1.1      cgd 			 */
    563   1.1      cgd 			if (dqblk.dqb_bsoftlimit &&
    564   1.1      cgd 			    qup->dqblk.dqb_curblocks >= dqblk.dqb_bsoftlimit &&
    565   1.1      cgd 			    (qup->dqblk.dqb_bsoftlimit == 0 ||
    566   1.1      cgd 			     qup->dqblk.dqb_curblocks <
    567   1.1      cgd 			     qup->dqblk.dqb_bsoftlimit))
    568   1.1      cgd 				qup->dqblk.dqb_btime = 0;
    569   1.1      cgd 			if (dqblk.dqb_isoftlimit &&
    570   1.1      cgd 			    qup->dqblk.dqb_curinodes >= dqblk.dqb_isoftlimit &&
    571   1.1      cgd 			    (qup->dqblk.dqb_isoftlimit == 0 ||
    572   1.1      cgd 			     qup->dqblk.dqb_curinodes <
    573   1.1      cgd 			     qup->dqblk.dqb_isoftlimit))
    574   1.1      cgd 				qup->dqblk.dqb_itime = 0;
    575   1.1      cgd 			qup->dqblk.dqb_bsoftlimit = dqblk.dqb_bsoftlimit;
    576   1.1      cgd 			qup->dqblk.dqb_bhardlimit = dqblk.dqb_bhardlimit;
    577   1.1      cgd 			qup->dqblk.dqb_isoftlimit = dqblk.dqb_isoftlimit;
    578   1.1      cgd 			qup->dqblk.dqb_ihardlimit = dqblk.dqb_ihardlimit;
    579   1.1      cgd 			qup->flags |= FOUND;
    580   1.1      cgd 			if (dqblk.dqb_curblocks == qup->dqblk.dqb_curblocks &&
    581   1.1      cgd 			    dqblk.dqb_curinodes == qup->dqblk.dqb_curinodes)
    582   1.1      cgd 				break;
    583  1.15    lukem 			warnx("%s: cannot change current allocation", fsp);
    584   1.1      cgd 			break;
    585   1.1      cgd 		}
    586   1.1      cgd 	}
    587   1.1      cgd 	fclose(fd);
    588   1.1      cgd 	/*
    589   1.1      cgd 	 * Disable quotas for any filesystems that have not been found.
    590   1.1      cgd 	 */
    591   1.1      cgd 	for (qup = quplist; qup; qup = qup->next) {
    592   1.1      cgd 		if (qup->flags & FOUND) {
    593   1.1      cgd 			qup->flags &= ~FOUND;
    594   1.1      cgd 			continue;
    595   1.1      cgd 		}
    596   1.1      cgd 		qup->dqblk.dqb_bsoftlimit = 0;
    597   1.1      cgd 		qup->dqblk.dqb_bhardlimit = 0;
    598   1.1      cgd 		qup->dqblk.dqb_isoftlimit = 0;
    599   1.1      cgd 		qup->dqblk.dqb_ihardlimit = 0;
    600   1.1      cgd 	}
    601   1.1      cgd 	return (1);
    602   1.1      cgd }
    603   1.1      cgd 
    604   1.1      cgd /*
    605   1.1      cgd  * Convert a quotause list to an ASCII file of grace times.
    606   1.1      cgd  */
    607  1.12    mikel int
    608   1.1      cgd writetimes(quplist, outfd, quotatype)
    609   1.1      cgd 	struct quotause *quplist;
    610   1.1      cgd 	int outfd;
    611   1.1      cgd 	int quotatype;
    612   1.1      cgd {
    613  1.15    lukem 	struct quotause *qup;
    614   1.1      cgd 	FILE *fd;
    615   1.1      cgd 
    616   1.1      cgd 	ftruncate(outfd, 0);
    617  1.14   kleink 	(void)lseek(outfd, (off_t)0, SEEK_SET);
    618  1.15    lukem 	if ((fd = fdopen(dup(outfd), "w")) == NULL)
    619  1.15    lukem 		err(1, "fdopen `%s'", tmpfil);
    620   1.1      cgd 	fprintf(fd, "Time units may be: days, hours, minutes, or seconds\n");
    621   1.1      cgd 	fprintf(fd, "Grace period before enforcing soft limits for %ss:\n",
    622   1.1      cgd 	    qfextension[quotatype]);
    623   1.1      cgd 	for (qup = quplist; qup; qup = qup->next) {
    624   1.1      cgd 		fprintf(fd, "%s: block grace period: %s, ",
    625   1.1      cgd 		    qup->fsname, cvtstoa(qup->dqblk.dqb_btime));
    626   1.1      cgd 		fprintf(fd, "file grace period: %s\n",
    627   1.1      cgd 		    cvtstoa(qup->dqblk.dqb_itime));
    628   1.1      cgd 	}
    629   1.1      cgd 	fclose(fd);
    630   1.1      cgd 	return (1);
    631   1.1      cgd }
    632   1.1      cgd 
    633   1.1      cgd /*
    634   1.1      cgd  * Merge changes of grace times in an ASCII file into a quotause list.
    635   1.1      cgd  */
    636  1.12    mikel int
    637   1.1      cgd readtimes(quplist, infd)
    638   1.1      cgd 	struct quotause *quplist;
    639   1.1      cgd 	int infd;
    640   1.1      cgd {
    641  1.15    lukem 	struct quotause *qup;
    642   1.1      cgd 	FILE *fd;
    643   1.1      cgd 	int cnt;
    644  1.15    lukem 	char *cp;
    645  1.15    lukem 	long litime, lbtime;
    646   1.1      cgd 	time_t itime, btime, iseconds, bseconds;
    647   1.1      cgd 	char *fsp, bunits[10], iunits[10], line1[BUFSIZ];
    648   1.1      cgd 
    649  1.14   kleink 	(void)lseek(infd, (off_t)0, SEEK_SET);
    650   1.1      cgd 	fd = fdopen(dup(infd), "r");
    651   1.1      cgd 	if (fd == NULL) {
    652  1.15    lukem 		warnx("Can't re-read temp file!!");
    653   1.1      cgd 		return (0);
    654   1.1      cgd 	}
    655   1.1      cgd 	/*
    656   1.1      cgd 	 * Discard two title lines, then read lines to process.
    657   1.1      cgd 	 */
    658   1.1      cgd 	(void) fgets(line1, sizeof (line1), fd);
    659   1.1      cgd 	(void) fgets(line1, sizeof (line1), fd);
    660   1.1      cgd 	while (fgets(line1, sizeof (line1), fd) != NULL) {
    661   1.1      cgd 		if ((fsp = strtok(line1, " \t:")) == NULL) {
    662  1.15    lukem 			warnx("%s: bad format", line1);
    663   1.1      cgd 			return (0);
    664   1.1      cgd 		}
    665   1.1      cgd 		if ((cp = strtok((char *)0, "\n")) == NULL) {
    666  1.15    lukem 			warnx("%s: %s: bad format", fsp,
    667   1.1      cgd 			    &fsp[strlen(fsp) + 1]);
    668   1.1      cgd 			return (0);
    669   1.1      cgd 		}
    670   1.1      cgd 		cnt = sscanf(cp,
    671  1.15    lukem 		    " block grace period: %ld %s file grace period: %ld %s",
    672  1.15    lukem 		    &lbtime, bunits, &litime, iunits);
    673   1.1      cgd 		if (cnt != 4) {
    674  1.15    lukem 			warnx("%s:%s: bad format", fsp, cp);
    675   1.1      cgd 			return (0);
    676   1.1      cgd 		}
    677  1.15    lukem 		itime = (time_t)litime;
    678  1.15    lukem 		btime = (time_t)lbtime;
    679   1.1      cgd 		if (cvtatos(btime, bunits, &bseconds) == 0)
    680   1.1      cgd 			return (0);
    681   1.1      cgd 		if (cvtatos(itime, iunits, &iseconds) == 0)
    682   1.1      cgd 			return (0);
    683   1.1      cgd 		for (qup = quplist; qup; qup = qup->next) {
    684   1.1      cgd 			if (strcmp(fsp, qup->fsname))
    685   1.1      cgd 				continue;
    686   1.1      cgd 			qup->dqblk.dqb_btime = bseconds;
    687   1.1      cgd 			qup->dqblk.dqb_itime = iseconds;
    688   1.1      cgd 			qup->flags |= FOUND;
    689   1.1      cgd 			break;
    690   1.1      cgd 		}
    691   1.1      cgd 	}
    692   1.1      cgd 	fclose(fd);
    693   1.1      cgd 	/*
    694   1.1      cgd 	 * reset default grace periods for any filesystems
    695   1.1      cgd 	 * that have not been found.
    696   1.1      cgd 	 */
    697   1.1      cgd 	for (qup = quplist; qup; qup = qup->next) {
    698   1.1      cgd 		if (qup->flags & FOUND) {
    699   1.1      cgd 			qup->flags &= ~FOUND;
    700   1.1      cgd 			continue;
    701   1.1      cgd 		}
    702   1.1      cgd 		qup->dqblk.dqb_btime = 0;
    703   1.1      cgd 		qup->dqblk.dqb_itime = 0;
    704   1.1      cgd 	}
    705   1.1      cgd 	return (1);
    706   1.1      cgd }
    707   1.1      cgd 
    708   1.1      cgd /*
    709   1.1      cgd  * Convert seconds to ASCII times.
    710   1.1      cgd  */
    711   1.1      cgd char *
    712   1.1      cgd cvtstoa(time)
    713   1.1      cgd 	time_t time;
    714   1.1      cgd {
    715   1.1      cgd 	static char buf[20];
    716   1.1      cgd 
    717   1.1      cgd 	if (time % (24 * 60 * 60) == 0) {
    718   1.1      cgd 		time /= 24 * 60 * 60;
    719  1.15    lukem 		snprintf(buf, sizeof buf, "%ld day%s", (long)time,
    720  1.15    lukem 		    time == 1 ? "" : "s");
    721   1.1      cgd 	} else if (time % (60 * 60) == 0) {
    722   1.1      cgd 		time /= 60 * 60;
    723  1.15    lukem 		sprintf(buf, "%ld hour%s", (long)time, time == 1 ? "" : "s");
    724   1.1      cgd 	} else if (time % 60 == 0) {
    725   1.1      cgd 		time /= 60;
    726  1.15    lukem 		sprintf(buf, "%ld minute%s", (long)time, time == 1 ? "" : "s");
    727   1.1      cgd 	} else
    728  1.15    lukem 		sprintf(buf, "%ld second%s", (long)time, time == 1 ? "" : "s");
    729   1.1      cgd 	return (buf);
    730   1.1      cgd }
    731   1.1      cgd 
    732   1.1      cgd /*
    733   1.1      cgd  * Convert ASCII input times to seconds.
    734   1.1      cgd  */
    735  1.12    mikel int
    736   1.1      cgd cvtatos(time, units, seconds)
    737   1.1      cgd 	time_t time;
    738   1.1      cgd 	char *units;
    739   1.1      cgd 	time_t *seconds;
    740   1.1      cgd {
    741   1.1      cgd 
    742  1.15    lukem 	if (memcmp(units, "second", 6) == 0)
    743   1.1      cgd 		*seconds = time;
    744  1.15    lukem 	else if (memcmp(units, "minute", 6) == 0)
    745   1.1      cgd 		*seconds = time * 60;
    746  1.15    lukem 	else if (memcmp(units, "hour", 4) == 0)
    747   1.1      cgd 		*seconds = time * 60 * 60;
    748  1.15    lukem 	else if (memcmp(units, "day", 3) == 0)
    749   1.1      cgd 		*seconds = time * 24 * 60 * 60;
    750   1.1      cgd 	else {
    751   1.1      cgd 		printf("%s: bad units, specify %s\n", units,
    752   1.1      cgd 		    "days, hours, minutes, or seconds");
    753   1.1      cgd 		return (0);
    754   1.1      cgd 	}
    755   1.1      cgd 	return (1);
    756   1.1      cgd }
    757   1.1      cgd 
    758   1.1      cgd /*
    759   1.1      cgd  * Free a list of quotause structures.
    760   1.1      cgd  */
    761  1.12    mikel void
    762   1.1      cgd freeprivs(quplist)
    763   1.1      cgd 	struct quotause *quplist;
    764   1.1      cgd {
    765  1.15    lukem 	struct quotause *qup, *nextqup;
    766   1.1      cgd 
    767   1.1      cgd 	for (qup = quplist; qup; qup = nextqup) {
    768   1.1      cgd 		nextqup = qup->next;
    769   1.1      cgd 		free(qup);
    770   1.1      cgd 	}
    771   1.1      cgd }
    772   1.1      cgd 
    773   1.1      cgd /*
    774   1.1      cgd  * Check whether a string is completely composed of digits.
    775   1.1      cgd  */
    776  1.12    mikel int
    777   1.1      cgd alldigits(s)
    778  1.15    lukem 	char *s;
    779   1.1      cgd {
    780  1.15    lukem 	int c;
    781   1.1      cgd 
    782   1.1      cgd 	c = *s++;
    783   1.1      cgd 	do {
    784   1.1      cgd 		if (!isdigit(c))
    785   1.1      cgd 			return (0);
    786  1.15    lukem 	} while ((c = *s++) != 0);
    787   1.1      cgd 	return (1);
    788   1.1      cgd }
    789   1.1      cgd 
    790   1.1      cgd /*
    791   1.1      cgd  * Check to see if a particular quota is to be enabled.
    792   1.1      cgd  */
    793  1.12    mikel int
    794   1.1      cgd hasquota(fs, type, qfnamep)
    795  1.15    lukem 	struct fstab *fs;
    796   1.1      cgd 	int type;
    797   1.1      cgd 	char **qfnamep;
    798   1.1      cgd {
    799  1.15    lukem 	char *opt;
    800  1.15    lukem 	char *cp;
    801   1.1      cgd 	static char initname, usrname[100], grpname[100];
    802   1.1      cgd 	static char buf[BUFSIZ];
    803   1.1      cgd 
    804   1.1      cgd 	if (!initname) {
    805   1.1      cgd 		sprintf(usrname, "%s%s", qfextension[USRQUOTA], qfname);
    806   1.1      cgd 		sprintf(grpname, "%s%s", qfextension[GRPQUOTA], qfname);
    807   1.1      cgd 		initname = 1;
    808   1.1      cgd 	}
    809   1.1      cgd 	strcpy(buf, fs->fs_mntops);
    810  1.16      mrg 	cp = NULL;
    811   1.1      cgd 	for (opt = strtok(buf, ","); opt; opt = strtok(NULL, ",")) {
    812  1.15    lukem 		if ((cp = strchr(opt, '=')) != NULL)
    813   1.1      cgd 			*cp++ = '\0';
    814   1.1      cgd 		if (type == USRQUOTA && strcmp(opt, usrname) == 0)
    815   1.1      cgd 			break;
    816   1.1      cgd 		if (type == GRPQUOTA && strcmp(opt, grpname) == 0)
    817   1.1      cgd 			break;
    818   1.1      cgd 	}
    819   1.1      cgd 	if (!opt)
    820   1.1      cgd 		return (0);
    821   1.1      cgd 	if (cp) {
    822   1.1      cgd 		*qfnamep = cp;
    823   1.1      cgd 		return (1);
    824   1.1      cgd 	}
    825   1.1      cgd 	(void) sprintf(buf, "%s/%s.%s", fs->fs_file, qfname, qfextension[type]);
    826   1.1      cgd 	*qfnamep = buf;
    827   1.1      cgd 	return (1);
    828   1.1      cgd }
    829