xref: /freebsd/usr.bin/quota/quota.c (revision afe61c15161c324a7af299a9b8457aba5afc92db)
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  * 3. All advertising materials mentioning features or use of this software
17  *    must display the following acknowledgement:
18  *	This product includes software developed by the University of
19  *	California, Berkeley and its contributors.
20  * 4. Neither the name of the University nor the names of its contributors
21  *    may be used to endorse or promote products derived from this software
22  *    without specific prior written permission.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34  * SUCH DAMAGE.
35  */
36 
37 #ifndef lint
38 static char copyright[] =
39 "@(#) Copyright (c) 1980, 1990, 1993\n\
40 	The Regents of the University of California.  All rights reserved.\n";
41 #endif /* not lint */
42 
43 #ifndef lint
44 static char sccsid[] = "@(#)quota.c	8.1 (Berkeley) 6/6/93";
45 #endif /* not lint */
46 
47 /*
48  * Disk quota reporting program.
49  */
50 #include <sys/param.h>
51 #include <sys/file.h>
52 #include <sys/stat.h>
53 #include <ufs/ufs/quota.h>
54 #include <stdio.h>
55 #include <fstab.h>
56 #include <ctype.h>
57 #include <pwd.h>
58 #include <grp.h>
59 #include <errno.h>
60 
61 char *qfname = QUOTAFILENAME;
62 char *qfextension[] = INITQFNAMES;
63 
64 struct quotause {
65 	struct	quotause *next;
66 	long	flags;
67 	struct	dqblk dqblk;
68 	char	fsname[MAXPATHLEN + 1];
69 } *getprivs();
70 #define	FOUND	0x01
71 
72 int	qflag;
73 int	vflag;
74 
75 main(argc, argv)
76 	char *argv[];
77 {
78 	int ngroups, gidset[NGROUPS];
79 	int i, gflag = 0, uflag = 0;
80 	char ch;
81 	extern char *optarg;
82 	extern int optind, errno;
83 
84 	if (quotactl("/", 0, 0, (caddr_t)0) < 0 && errno == EOPNOTSUPP) {
85 		fprintf(stderr, "There are no quotas on this system\n");
86 		exit(0);
87 	}
88 	while ((ch = getopt(argc, argv, "ugvq")) != EOF) {
89 		switch(ch) {
90 		case 'g':
91 			gflag++;
92 			break;
93 		case 'u':
94 			uflag++;
95 			break;
96 		case 'v':
97 			vflag++;
98 			break;
99 		case 'q':
100 			qflag++;
101 			break;
102 		default:
103 			usage();
104 		}
105 	}
106 	argc -= optind;
107 	argv += optind;
108 	if (!uflag && !gflag)
109 		uflag++;
110 	if (argc == 0) {
111 		if (uflag)
112 			showuid(getuid());
113 		if (gflag) {
114 			ngroups = getgroups(NGROUPS, gidset);
115 			if (ngroups < 0) {
116 				perror("quota: getgroups");
117 				exit(1);
118 			}
119 			for (i = 1; i < ngroups; i++)
120 				showgid(gidset[i]);
121 		}
122 		exit(0);
123 	}
124 	if (uflag && gflag)
125 		usage();
126 	if (uflag) {
127 		for (; argc > 0; argc--, argv++) {
128 			if (alldigits(*argv))
129 				showuid(atoi(*argv));
130 			else
131 				showusrname(*argv);
132 		}
133 		exit(0);
134 	}
135 	if (gflag) {
136 		for (; argc > 0; argc--, argv++) {
137 			if (alldigits(*argv))
138 				showgid(atoi(*argv));
139 			else
140 				showgrpname(*argv);
141 		}
142 		exit(0);
143 	}
144 }
145 
146 usage()
147 {
148 
149 	fprintf(stderr, "%s\n%s\n%s\n",
150 		"Usage: quota [-guqv]",
151 		"\tquota [-qv] -u username ...",
152 		"\tquota [-qv] -g groupname ...");
153 	exit(1);
154 }
155 
156 /*
157  * Print out quotas for a specified user identifier.
158  */
159 showuid(uid)
160 	u_long uid;
161 {
162 	struct passwd *pwd = getpwuid(uid);
163 	u_long myuid;
164 	char *name;
165 
166 	if (pwd == NULL)
167 		name = "(no account)";
168 	else
169 		name = pwd->pw_name;
170 	myuid = getuid();
171 	if (uid != myuid && myuid != 0) {
172 		printf("quota: %s (uid %d): permission denied\n", name, uid);
173 		return;
174 	}
175 	showquotas(USRQUOTA, uid, name);
176 }
177 
178 /*
179  * Print out quotas for a specifed user name.
180  */
181 showusrname(name)
182 	char *name;
183 {
184 	struct passwd *pwd = getpwnam(name);
185 	u_long myuid;
186 
187 	if (pwd == NULL) {
188 		fprintf(stderr, "quota: %s: unknown user\n", name);
189 		return;
190 	}
191 	myuid = getuid();
192 	if (pwd->pw_uid != myuid && myuid != 0) {
193 		fprintf(stderr, "quota: %s (uid %d): permission denied\n",
194 		    name, pwd->pw_uid);
195 		return;
196 	}
197 	showquotas(USRQUOTA, pwd->pw_uid, name);
198 }
199 
200 /*
201  * Print out quotas for a specified group identifier.
202  */
203 showgid(gid)
204 	u_long gid;
205 {
206 	struct group *grp = getgrgid(gid);
207 	int ngroups, gidset[NGROUPS];
208 	register int i;
209 	char *name;
210 
211 	if (grp == NULL)
212 		name = "(no entry)";
213 	else
214 		name = grp->gr_name;
215 	ngroups = getgroups(NGROUPS, gidset);
216 	if (ngroups < 0) {
217 		perror("quota: getgroups");
218 		return;
219 	}
220 	for (i = 1; i < ngroups; i++)
221 		if (gid == gidset[i])
222 			break;
223 	if (i >= ngroups && getuid() != 0) {
224 		fprintf(stderr, "quota: %s (gid %d): permission denied\n",
225 		    name, gid);
226 		return;
227 	}
228 	showquotas(GRPQUOTA, gid, name);
229 }
230 
231 /*
232  * Print out quotas for a specifed group name.
233  */
234 showgrpname(name)
235 	char *name;
236 {
237 	struct group *grp = getgrnam(name);
238 	int ngroups, gidset[NGROUPS];
239 	register int i;
240 
241 	if (grp == NULL) {
242 		fprintf(stderr, "quota: %s: unknown group\n", name);
243 		return;
244 	}
245 	ngroups = getgroups(NGROUPS, gidset);
246 	if (ngroups < 0) {
247 		perror("quota: getgroups");
248 		return;
249 	}
250 	for (i = 1; i < ngroups; i++)
251 		if (grp->gr_gid == gidset[i])
252 			break;
253 	if (i >= ngroups && getuid() != 0) {
254 		fprintf(stderr, "quota: %s (gid %d): permission denied\n",
255 		    name, grp->gr_gid);
256 		return;
257 	}
258 	showquotas(GRPQUOTA, grp->gr_gid, name);
259 }
260 
261 showquotas(type, id, name)
262 	int type;
263 	u_long id;
264 	char *name;
265 {
266 	register struct quotause *qup;
267 	struct quotause *quplist, *getprivs();
268 	char *msgi, *msgb, *timeprt();
269 	int myuid, fd, lines = 0;
270 	static int first;
271 	static time_t now;
272 
273 	if (now == 0)
274 		time(&now);
275 	quplist = getprivs(id, type);
276 	for (qup = quplist; qup; qup = qup->next) {
277 		if (!vflag &&
278 		    qup->dqblk.dqb_isoftlimit == 0 &&
279 		    qup->dqblk.dqb_ihardlimit == 0 &&
280 		    qup->dqblk.dqb_bsoftlimit == 0 &&
281 		    qup->dqblk.dqb_bhardlimit == 0)
282 			continue;
283 		msgi = (char *)0;
284 		if (qup->dqblk.dqb_ihardlimit &&
285 		    qup->dqblk.dqb_curinodes >= qup->dqblk.dqb_ihardlimit)
286 			msgi = "File limit reached on";
287 		else if (qup->dqblk.dqb_isoftlimit &&
288 		    qup->dqblk.dqb_curinodes >= qup->dqblk.dqb_isoftlimit)
289 			if (qup->dqblk.dqb_itime > now)
290 				msgi = "In file grace period on";
291 			else
292 				msgi = "Over file quota on";
293 		msgb = (char *)0;
294 		if (qup->dqblk.dqb_bhardlimit &&
295 		    qup->dqblk.dqb_curblocks >= qup->dqblk.dqb_bhardlimit)
296 			msgb = "Block limit reached on";
297 		else if (qup->dqblk.dqb_bsoftlimit &&
298 		    qup->dqblk.dqb_curblocks >= qup->dqblk.dqb_bsoftlimit)
299 			if (qup->dqblk.dqb_btime > now)
300 				msgb = "In block grace period on";
301 			else
302 				msgb = "Over block quota on";
303 		if (qflag) {
304 			if ((msgi != (char *)0 || msgb != (char *)0) &&
305 			    lines++ == 0)
306 				heading(type, id, name, "");
307 			if (msgi != (char *)0)
308 				printf("\t%s %s\n", msgi, qup->fsname);
309 			if (msgb != (char *)0)
310 				printf("\t%s %s\n", msgb, qup->fsname);
311 			continue;
312 		}
313 		if (vflag ||
314 		    qup->dqblk.dqb_curblocks ||
315 		    qup->dqblk.dqb_curinodes) {
316 			if (lines++ == 0)
317 				heading(type, id, name, "");
318 			printf("%15s%8d%c%7d%8d%8s"
319 				, qup->fsname
320 				, dbtob(qup->dqblk.dqb_curblocks) / 1024
321 				, (msgb == (char *)0) ? ' ' : '*'
322 				, dbtob(qup->dqblk.dqb_bsoftlimit) / 1024
323 				, dbtob(qup->dqblk.dqb_bhardlimit) / 1024
324 				, (msgb == (char *)0) ? ""
325 				    : timeprt(qup->dqblk.dqb_btime));
326 			printf("%8d%c%7d%8d%8s\n"
327 				, qup->dqblk.dqb_curinodes
328 				, (msgi == (char *)0) ? ' ' : '*'
329 				, qup->dqblk.dqb_isoftlimit
330 				, qup->dqblk.dqb_ihardlimit
331 				, (msgi == (char *)0) ? ""
332 				    : timeprt(qup->dqblk.dqb_itime)
333 			);
334 			continue;
335 		}
336 	}
337 	if (!qflag && lines == 0)
338 		heading(type, id, name, "none");
339 }
340 
341 heading(type, id, name, tag)
342 	int type;
343 	u_long id;
344 	char *name, *tag;
345 {
346 
347 	printf("Disk quotas for %s %s (%cid %d): %s\n", qfextension[type],
348 	    name, *qfextension[type], id, tag);
349 	if (!qflag && tag[0] == '\0') {
350 		printf("%15s%8s %7s%8s%8s%8s %7s%8s%8s\n"
351 			, "Filesystem"
352 			, "blocks"
353 			, "quota"
354 			, "limit"
355 			, "grace"
356 			, "files"
357 			, "quota"
358 			, "limit"
359 			, "grace"
360 		);
361 	}
362 }
363 
364 /*
365  * Calculate the grace period and return a printable string for it.
366  */
367 char *
368 timeprt(seconds)
369 	time_t seconds;
370 {
371 	time_t hours, minutes;
372 	static char buf[20];
373 	static time_t now;
374 
375 	if (now == 0)
376 		time(&now);
377 	if (now > seconds)
378 		return ("none");
379 	seconds -= now;
380 	minutes = (seconds + 30) / 60;
381 	hours = (minutes + 30) / 60;
382 	if (hours >= 36) {
383 		sprintf(buf, "%ddays", (hours + 12) / 24);
384 		return (buf);
385 	}
386 	if (minutes >= 60) {
387 		sprintf(buf, "%2d:%d", minutes / 60, minutes % 60);
388 		return (buf);
389 	}
390 	sprintf(buf, "%2d", minutes);
391 	return (buf);
392 }
393 
394 /*
395  * Collect the requested quota information.
396  */
397 struct quotause *
398 getprivs(id, quotatype)
399 	register long id;
400 	int quotatype;
401 {
402 	register struct fstab *fs;
403 	register struct quotause *qup, *quptail;
404 	struct quotause *quphead;
405 	char *qfpathname;
406 	int qcmd, fd;
407 
408 	setfsent();
409 	quphead = (struct quotause *)0;
410 	qcmd = QCMD(Q_GETQUOTA, quotatype);
411 	while (fs = getfsent()) {
412 		if (strcmp(fs->fs_vfstype, "ufs"))
413 			continue;
414 		if (!hasquota(fs, quotatype, &qfpathname))
415 			continue;
416 		if ((qup = (struct quotause *)malloc(sizeof *qup)) == NULL) {
417 			fprintf(stderr, "quota: out of memory\n");
418 			exit(2);
419 		}
420 		if (quotactl(fs->fs_file, qcmd, id, &qup->dqblk) != 0) {
421 			if ((fd = open(qfpathname, O_RDONLY)) < 0) {
422 				perror(qfpathname);
423 				free(qup);
424 				continue;
425 			}
426 			lseek(fd, (long)(id * sizeof(struct dqblk)), L_SET);
427 			switch (read(fd, &qup->dqblk, sizeof(struct dqblk))) {
428 			case 0:			/* EOF */
429 				/*
430 				 * Convert implicit 0 quota (EOF)
431 				 * into an explicit one (zero'ed dqblk)
432 				 */
433 				bzero((caddr_t)&qup->dqblk,
434 				    sizeof(struct dqblk));
435 				break;
436 
437 			case sizeof(struct dqblk):	/* OK */
438 				break;
439 
440 			default:		/* ERROR */
441 				fprintf(stderr, "quota: read error");
442 				perror(qfpathname);
443 				close(fd);
444 				free(qup);
445 				continue;
446 			}
447 			close(fd);
448 		}
449 		strcpy(qup->fsname, fs->fs_file);
450 		if (quphead == NULL)
451 			quphead = qup;
452 		else
453 			quptail->next = qup;
454 		quptail = qup;
455 		qup->next = 0;
456 	}
457 	endfsent();
458 	return (quphead);
459 }
460 
461 /*
462  * Check to see if a particular quota is to be enabled.
463  */
464 hasquota(fs, type, qfnamep)
465 	register struct fstab *fs;
466 	int type;
467 	char **qfnamep;
468 {
469 	register char *opt;
470 	char *cp, *index(), *strtok();
471 	static char initname, usrname[100], grpname[100];
472 	static char buf[BUFSIZ];
473 
474 	if (!initname) {
475 		sprintf(usrname, "%s%s", qfextension[USRQUOTA], qfname);
476 		sprintf(grpname, "%s%s", qfextension[GRPQUOTA], qfname);
477 		initname = 1;
478 	}
479 	strcpy(buf, fs->fs_mntops);
480 	for (opt = strtok(buf, ","); opt; opt = strtok(NULL, ",")) {
481 		if (cp = index(opt, '='))
482 			*cp++ = '\0';
483 		if (type == USRQUOTA && strcmp(opt, usrname) == 0)
484 			break;
485 		if (type == GRPQUOTA && strcmp(opt, grpname) == 0)
486 			break;
487 	}
488 	if (!opt)
489 		return (0);
490 	if (cp) {
491 		*qfnamep = cp;
492 		return (1);
493 	}
494 	(void) sprintf(buf, "%s/%s.%s", fs->fs_file, qfname, qfextension[type]);
495 	*qfnamep = buf;
496 	return (1);
497 }
498 
499 alldigits(s)
500 	register char *s;
501 {
502 	register c;
503 
504 	c = *s++;
505 	do {
506 		if (!isdigit(c))
507 			return (0);
508 	} while (c = *s++);
509 	return (1);
510 }
511