edquota.c revision 180187
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 180187 2008-07-02 15:51:59Z des $");
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(void)
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(const char *name, int quotatype)
303{
304	struct passwd *pw;
305	struct group *gr;
306
307	if (alldigits(name))
308		return (atoi(name));
309	switch(quotatype) {
310	case USRQUOTA:
311		if ((pw = getpwnam(name)))
312			return (pw->pw_uid);
313		warnx("%s: no such user", name);
314		break;
315	case GRPQUOTA:
316		if ((gr = getgrnam(name)))
317			return (gr->gr_gid);
318		warnx("%s: no such group", name);
319		break;
320	default:
321		warnx("%d: unknown quota type", quotatype);
322		break;
323	}
324	sleep(1);
325	return (-1);
326}
327
328/*
329 * Collect the requested quota information.
330 */
331struct quotause *
332getprivs(long id, int quotatype, char *fspath)
333{
334	struct fstab *fs;
335	struct quotause *qup, *quptail;
336	struct quotause *quphead;
337	int qcmd, qupsize, fd;
338	char *qfpathname;
339	static int warned = 0;
340
341	setfsent();
342	quphead = quptail = NULL;
343	qcmd = QCMD(Q_GETQUOTA, quotatype);
344	while ((fs = getfsent())) {
345		if (fspath && *fspath && strcmp(fspath, fs->fs_spec) &&
346		    strcmp(fspath, fs->fs_file))
347			continue;
348		if (strcmp(fs->fs_vfstype, "ufs"))
349			continue;
350		if (!hasquota(fs, quotatype, &qfpathname))
351			continue;
352		qupsize = sizeof(*qup) + strlen(qfpathname);
353		if ((qup = (struct quotause *)malloc(qupsize)) == NULL)
354			errx(2, "out of memory");
355		if (quotactl(fs->fs_file, qcmd, id, &qup->dqblk) != 0) {
356			if (errno == EOPNOTSUPP && !warned) {
357				warned++;
358		warnx("warning: quotas are not compiled into this kernel");
359				sleep(3);
360			}
361			if ((fd = open(qfpathname, O_RDONLY)) < 0) {
362				fd = open(qfpathname, O_RDWR|O_CREAT, 0640);
363				if (fd < 0 && errno != ENOENT) {
364					warn("%s", qfpathname);
365					free(qup);
366					continue;
367				}
368				warnx("creating quota file %s", qfpathname);
369				sleep(3);
370				(void) fchown(fd, getuid(),
371				    getentry(quotagroup, GRPQUOTA));
372				(void) fchmod(fd, 0640);
373			}
374			if (lseek(fd, (off_t)id * sizeof(struct dqblk),
375			    L_SET) < 0) {
376				warn("seek error on %s", qfpathname);
377				close(fd);
378				free(qup);
379				continue;
380			}
381			switch (read(fd, &qup->dqblk, sizeof(struct dqblk))) {
382			case 0:			/* EOF */
383				/*
384				 * Convert implicit 0 quota (EOF)
385				 * into an explicit one (zero'ed dqblk)
386				 */
387				bzero((caddr_t)&qup->dqblk,
388				    sizeof(struct dqblk));
389				break;
390
391			case sizeof(struct dqblk):	/* OK */
392				break;
393
394			default:		/* ERROR */
395				warn("read error in %s", qfpathname);
396				close(fd);
397				free(qup);
398				continue;
399			}
400			close(fd);
401		}
402		strcpy(qup->qfname, qfpathname);
403		strcpy(qup->fsname, fs->fs_file);
404		if (quphead == NULL)
405			quphead = qup;
406		else
407			quptail->next = qup;
408		quptail = qup;
409		qup->next = 0;
410	}
411	endfsent();
412	return (quphead);
413}
414
415/*
416 * Store the requested quota information.
417 */
418void
419putprivs(long id, int quotatype, struct quotause *quplist)
420{
421	struct quotause *qup;
422	int qcmd, fd;
423	struct dqblk dqbuf;
424
425	qcmd = QCMD(Q_SETQUOTA, quotatype);
426	for (qup = quplist; qup; qup = qup->next) {
427		if (quotactl(qup->fsname, qcmd, id, &qup->dqblk) == 0)
428			continue;
429		if ((fd = open(qup->qfname, O_RDWR)) < 0) {
430			warn("%s", qup->qfname);
431			continue;
432		}
433		if (lseek(fd, (off_t)id * sizeof(struct dqblk), L_SET) < 0) {
434			warn("seek error on %s", qup->qfname);
435			close(fd);
436			continue;
437		}
438		switch (read(fd, &dqbuf, sizeof(struct dqblk))) {
439		case 0:			/* EOF */
440			/*
441			 * Convert implicit 0 quota (EOF)
442			 * into an explicit one (zero'ed dqblk)
443			 */
444			bzero(&dqbuf, sizeof(struct dqblk));
445			break;
446
447		case sizeof(struct dqblk):	/* OK */
448			break;
449
450		default:		/* ERROR */
451			warn("read error in %s", qup->qfname);
452			close(fd);
453			continue;
454		}
455		/*
456		 * Reset time limit if have a soft limit and were
457		 * previously under it, but are now over it
458		 * or if there previously was no soft limit, but
459		 * now have one and are over it.
460		 */
461		if (dqbuf.dqb_bsoftlimit && id != 0 &&
462		    dqbuf.dqb_curblocks < dqbuf.dqb_bsoftlimit &&
463		    dqbuf.dqb_curblocks >= qup->dqblk.dqb_bsoftlimit)
464			qup->dqblk.dqb_btime = 0;
465		if (dqbuf.dqb_bsoftlimit == 0 && id != 0 &&
466		    dqbuf.dqb_curblocks >= qup->dqblk.dqb_bsoftlimit)
467			qup->dqblk.dqb_btime = 0;
468		if (dqbuf.dqb_isoftlimit && id != 0 &&
469		    dqbuf.dqb_curinodes < dqbuf.dqb_isoftlimit &&
470		    dqbuf.dqb_curinodes >= qup->dqblk.dqb_isoftlimit)
471			qup->dqblk.dqb_itime = 0;
472		if (dqbuf.dqb_isoftlimit == 0 && id !=0 &&
473		    dqbuf.dqb_curinodes >= qup->dqblk.dqb_isoftlimit)
474			qup->dqblk.dqb_itime = 0;
475		qup->dqblk.dqb_curinodes = dqbuf.dqb_curinodes;
476		qup->dqblk.dqb_curblocks = dqbuf.dqb_curblocks;
477		if (lseek(fd, (off_t)id * sizeof(struct dqblk), L_SET) < 0) {
478			warn("seek error on %s", qup->qfname);
479			close(fd);
480			continue;
481		}
482		if (write(fd, &qup->dqblk, sizeof (struct dqblk)) !=
483		    sizeof (struct dqblk)) {
484			warn("%s", qup->qfname);
485			}
486		close(fd);
487	}
488}
489
490/*
491 * Take a list of priviledges and get it edited.
492 */
493int
494editit(char *tmpf)
495{
496	long omask;
497	int pid, status;
498
499	omask = sigblock(sigmask(SIGINT)|sigmask(SIGQUIT)|sigmask(SIGHUP));
500 top:
501	if ((pid = fork()) < 0) {
502
503		if (errno == EPROCLIM) {
504			warnx("you have too many processes");
505			return(0);
506		}
507		if (errno == EAGAIN) {
508			sleep(1);
509			goto top;
510		}
511		warn("fork");
512		return (0);
513	}
514	if (pid == 0) {
515		const char *ed;
516
517		sigsetmask(omask);
518		setgid(getgid());
519		setuid(getuid());
520		if ((ed = getenv("EDITOR")) == (char *)0)
521			ed = _PATH_VI;
522		execlp(ed, ed, tmpf, (char *)0);
523		err(1, "%s", ed);
524	}
525	waitpid(pid, &status, 0);
526	sigsetmask(omask);
527	if (!WIFEXITED(status) || WEXITSTATUS(status) != 0)
528		return (0);
529	return (1);
530}
531
532/*
533 * Convert a quotause list to an ASCII file.
534 */
535int
536writeprivs(struct quotause *quplist, int outfd, char *name, int quotatype)
537{
538	struct quotause *qup;
539	FILE *fd;
540
541	ftruncate(outfd, 0);
542	lseek(outfd, 0, L_SET);
543	if ((fd = fdopen(dup(outfd), "w")) == NULL)
544		err(1, "%s", tmpfil);
545	fprintf(fd, "Quotas for %s %s:\n", qfextension[quotatype], name);
546	for (qup = quplist; qup; qup = qup->next) {
547		fprintf(fd, "%s: %s %lu, limits (soft = %lu, hard = %lu)\n",
548		    qup->fsname, "kbytes in use:",
549		    (unsigned long)(dbtob(qup->dqblk.dqb_curblocks) / 1024),
550		    (unsigned long)(dbtob(qup->dqblk.dqb_bsoftlimit) / 1024),
551		    (unsigned long)(dbtob(qup->dqblk.dqb_bhardlimit) / 1024));
552		fprintf(fd, "%s %lu, limits (soft = %lu, hard = %lu)\n",
553		    "\tinodes in use:",
554		    (unsigned long)qup->dqblk.dqb_curinodes,
555		    (unsigned long)qup->dqblk.dqb_isoftlimit,
556		    (unsigned long)qup->dqblk.dqb_ihardlimit);
557	}
558	fclose(fd);
559	return (1);
560}
561
562/*
563 * Merge changes to an ASCII file into a quotause list.
564 */
565int
566readprivs(struct quotause *quplist, char *inname)
567{
568	struct quotause *qup;
569	FILE *fd;
570	unsigned long bhardlimit, bsoftlimit, curblocks;
571	unsigned long ihardlimit, isoftlimit, curinodes;
572	int cnt;
573	char *cp;
574	struct dqblk dqblk;
575	char *fsp, line1[BUFSIZ], line2[BUFSIZ];
576
577	fd = fopen(inname, "r");
578	if (fd == NULL) {
579		warnx("can't re-read temp file!!");
580		return (0);
581	}
582	/*
583	 * Discard title line, then read pairs of lines to process.
584	 */
585	(void) fgets(line1, sizeof (line1), fd);
586	while (fgets(line1, sizeof (line1), fd) != NULL &&
587	       fgets(line2, sizeof (line2), fd) != NULL) {
588		if ((fsp = strtok(line1, " \t:")) == NULL) {
589			warnx("%s: bad format", line1);
590			return (0);
591		}
592		if ((cp = strtok((char *)0, "\n")) == NULL) {
593			warnx("%s: %s: bad format", fsp, &fsp[strlen(fsp) + 1]);
594			return (0);
595		}
596		cnt = sscanf(cp,
597		    " kbytes in use: %lu, limits (soft = %lu, hard = %lu)",
598		    &curblocks, &bsoftlimit, &bhardlimit);
599		if (cnt != 3) {
600			warnx("%s:%s: bad format", fsp, cp);
601			return (0);
602		}
603		dqblk.dqb_curblocks = btodb((off_t)curblocks * 1024);
604		dqblk.dqb_bsoftlimit = btodb((off_t)bsoftlimit * 1024);
605		dqblk.dqb_bhardlimit = btodb((off_t)bhardlimit * 1024);
606		if ((cp = strtok(line2, "\n")) == NULL) {
607			warnx("%s: %s: bad format", fsp, line2);
608			return (0);
609		}
610		cnt = sscanf(cp,
611		    "\tinodes in use: %lu, limits (soft = %lu, hard = %lu)",
612		    &curinodes, &isoftlimit, &ihardlimit);
613		if (cnt != 3) {
614			warnx("%s: %s: bad format", fsp, line2);
615			return (0);
616		}
617		dqblk.dqb_curinodes = curinodes;
618		dqblk.dqb_isoftlimit = isoftlimit;
619		dqblk.dqb_ihardlimit = ihardlimit;
620		for (qup = quplist; qup; qup = qup->next) {
621			if (strcmp(fsp, qup->fsname))
622				continue;
623			/*
624			 * Cause time limit to be reset when the quota
625			 * is next used if previously had no soft limit
626			 * or were under it, but now have a soft limit
627			 * and are over it.
628			 */
629			if (dqblk.dqb_bsoftlimit &&
630			    qup->dqblk.dqb_curblocks >= dqblk.dqb_bsoftlimit &&
631			    (qup->dqblk.dqb_bsoftlimit == 0 ||
632			     qup->dqblk.dqb_curblocks <
633			     qup->dqblk.dqb_bsoftlimit))
634				qup->dqblk.dqb_btime = 0;
635			if (dqblk.dqb_isoftlimit &&
636			    qup->dqblk.dqb_curinodes >= dqblk.dqb_isoftlimit &&
637			    (qup->dqblk.dqb_isoftlimit == 0 ||
638			     qup->dqblk.dqb_curinodes <
639			     qup->dqblk.dqb_isoftlimit))
640				qup->dqblk.dqb_itime = 0;
641			qup->dqblk.dqb_bsoftlimit = dqblk.dqb_bsoftlimit;
642			qup->dqblk.dqb_bhardlimit = dqblk.dqb_bhardlimit;
643			qup->dqblk.dqb_isoftlimit = dqblk.dqb_isoftlimit;
644			qup->dqblk.dqb_ihardlimit = dqblk.dqb_ihardlimit;
645			qup->flags |= FOUND;
646			if (dqblk.dqb_curblocks == qup->dqblk.dqb_curblocks &&
647			    dqblk.dqb_curinodes == qup->dqblk.dqb_curinodes)
648				break;
649			warnx("%s: cannot change current allocation", fsp);
650			break;
651		}
652	}
653	fclose(fd);
654	/*
655	 * Disable quotas for any filesystems that have not been found.
656	 */
657	for (qup = quplist; qup; qup = qup->next) {
658		if (qup->flags & FOUND) {
659			qup->flags &= ~FOUND;
660			continue;
661		}
662		qup->dqblk.dqb_bsoftlimit = 0;
663		qup->dqblk.dqb_bhardlimit = 0;
664		qup->dqblk.dqb_isoftlimit = 0;
665		qup->dqblk.dqb_ihardlimit = 0;
666	}
667	return (1);
668}
669
670/*
671 * Convert a quotause list to an ASCII file of grace times.
672 */
673int
674writetimes(struct quotause *quplist, int outfd, int quotatype)
675{
676	struct quotause *qup;
677	FILE *fd;
678
679	ftruncate(outfd, 0);
680	lseek(outfd, 0, L_SET);
681	if ((fd = fdopen(dup(outfd), "w")) == NULL)
682		err(1, "%s", tmpfil);
683	fprintf(fd, "Time units may be: days, hours, minutes, or seconds\n");
684	fprintf(fd, "Grace period before enforcing soft limits for %ss:\n",
685	    qfextension[quotatype]);
686	for (qup = quplist; qup; qup = qup->next) {
687		fprintf(fd, "%s: block grace period: %s, ",
688		    qup->fsname, cvtstoa(qup->dqblk.dqb_btime));
689		fprintf(fd, "file grace period: %s\n",
690		    cvtstoa(qup->dqblk.dqb_itime));
691	}
692	fclose(fd);
693	return (1);
694}
695
696/*
697 * Merge changes of grace times in an ASCII file into a quotause list.
698 */
699int
700readtimes(struct quotause *quplist, char *inname)
701{
702	struct quotause *qup;
703	FILE *fd;
704	int cnt;
705	char *cp;
706	time_t itime, btime, iseconds, bseconds;
707	long l_itime, l_btime;
708	char *fsp, bunits[10], iunits[10], line1[BUFSIZ];
709
710	fd = fopen(inname, "r");
711	if (fd == NULL) {
712		warnx("can't re-read temp file!!");
713		return (0);
714	}
715	/*
716	 * Discard two title lines, then read lines to process.
717	 */
718	(void) fgets(line1, sizeof (line1), fd);
719	(void) fgets(line1, sizeof (line1), fd);
720	while (fgets(line1, sizeof (line1), fd) != NULL) {
721		if ((fsp = strtok(line1, " \t:")) == NULL) {
722			warnx("%s: bad format", line1);
723			return (0);
724		}
725		if ((cp = strtok((char *)0, "\n")) == NULL) {
726			warnx("%s: %s: bad format", fsp, &fsp[strlen(fsp) + 1]);
727			return (0);
728		}
729		cnt = sscanf(cp,
730		    " block grace period: %ld %s file grace period: %ld %s",
731		    &l_btime, bunits, &l_itime, iunits);
732		if (cnt != 4) {
733			warnx("%s:%s: bad format", fsp, cp);
734			return (0);
735		}
736		btime = l_btime;
737		itime = l_itime;
738		if (cvtatos(btime, bunits, &bseconds) == 0)
739			return (0);
740		if (cvtatos(itime, iunits, &iseconds) == 0)
741			return (0);
742		for (qup = quplist; qup; qup = qup->next) {
743			if (strcmp(fsp, qup->fsname))
744				continue;
745			qup->dqblk.dqb_btime = bseconds;
746			qup->dqblk.dqb_itime = iseconds;
747			qup->flags |= FOUND;
748			break;
749		}
750	}
751	fclose(fd);
752	/*
753	 * reset default grace periods for any filesystems
754	 * that have not been found.
755	 */
756	for (qup = quplist; qup; qup = qup->next) {
757		if (qup->flags & FOUND) {
758			qup->flags &= ~FOUND;
759			continue;
760		}
761		qup->dqblk.dqb_btime = 0;
762		qup->dqblk.dqb_itime = 0;
763	}
764	return (1);
765}
766
767/*
768 * Convert seconds to ASCII times.
769 */
770char *
771cvtstoa(time_t secs)
772{
773	static char buf[20];
774
775	if (secs % (24 * 60 * 60) == 0) {
776		secs /= 24 * 60 * 60;
777		sprintf(buf, "%ld day%s", (long)secs, secs == 1 ? "" : "s");
778	} else if (secs % (60 * 60) == 0) {
779		secs /= 60 * 60;
780		sprintf(buf, "%ld hour%s", (long)secs, secs == 1 ? "" : "s");
781	} else if (secs % 60 == 0) {
782		secs /= 60;
783		sprintf(buf, "%ld minute%s", (long)secs, secs == 1 ? "" : "s");
784	} else
785		sprintf(buf, "%ld second%s", (long)secs, secs == 1 ? "" : "s");
786	return (buf);
787}
788
789/*
790 * Convert ASCII input times to seconds.
791 */
792int
793cvtatos(time_t period, char *units, time_t *seconds)
794{
795
796	if (bcmp(units, "second", 6) == 0)
797		*seconds = period;
798	else if (bcmp(units, "minute", 6) == 0)
799		*seconds = period * 60;
800	else if (bcmp(units, "hour", 4) == 0)
801		*seconds = period * 60 * 60;
802	else if (bcmp(units, "day", 3) == 0)
803		*seconds = period * 24 * 60 * 60;
804	else {
805		printf("%s: bad units, specify %s\n", units,
806		    "days, hours, minutes, or seconds");
807		return (0);
808	}
809	return (1);
810}
811
812/*
813 * Free a list of quotause structures.
814 */
815void
816freeprivs(struct quotause *quplist)
817{
818	struct quotause *qup, *nextqup;
819
820	for (qup = quplist; qup; qup = nextqup) {
821		nextqup = qup->next;
822		free(qup);
823	}
824}
825
826/*
827 * Check whether a string is completely composed of digits.
828 */
829int
830alldigits(const char *s)
831{
832	int c;
833
834	c = *s++;
835	do {
836		if (!isdigit(c))
837			return (0);
838	} while ((c = *s++));
839	return (1);
840}
841
842/*
843 * Check to see if a particular quota is to be enabled.
844 */
845int
846hasquota(struct fstab *fs, int type, char **qfnamep)
847{
848	char *opt;
849	char *cp;
850	struct statfs sfb;
851	static char initname, usrname[100], grpname[100];
852	static char buf[BUFSIZ];
853
854	if (!initname) {
855		(void)snprintf(usrname, sizeof(usrname), "%s%s",
856		    qfextension[USRQUOTA], qfname);
857		(void)snprintf(grpname, sizeof(grpname), "%s%s",
858		    qfextension[GRPQUOTA], qfname);
859		initname = 1;
860	}
861	strcpy(buf, fs->fs_mntops);
862	for (opt = strtok(buf, ","); opt; opt = strtok(NULL, ",")) {
863		if ((cp = index(opt, '=')))
864			*cp++ = '\0';
865		if (type == USRQUOTA && strcmp(opt, usrname) == 0)
866			break;
867		if (type == GRPQUOTA && strcmp(opt, grpname) == 0)
868			break;
869	}
870	if (!opt)
871		return (0);
872	if (cp)
873		*qfnamep = cp;
874	else {
875		(void)snprintf(buf, sizeof(buf), "%s/%s.%s", fs->fs_file,
876		    qfname, qfextension[type]);
877		*qfnamep = buf;
878	}
879	if (statfs(fs->fs_file, &sfb) != 0) {
880		warn("cannot statfs mount point %s", fs->fs_file);
881		return (0);
882	}
883	if (strcmp(fs->fs_file, sfb.f_mntonname)) {
884		warnx("%s not mounted for %s quotas", fs->fs_file,
885		    type == USRQUOTA ? "user" : "group");
886		sleep(3);
887		return (0);
888	}
889	return (1);
890}
891