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