quotaon.c revision 330897 
1/*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1980, 1990, 1993
5 *	The Regents of the University of California.  All rights reserved.
6 *
7 * This code is derived from software contributed to Berkeley by
8 * Robert Elz at The University of Melbourne.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 *    notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 *    notice, this list of conditions and the following disclaimer in the
17 *    documentation and/or other materials provided with the distribution.
18 * 4. Neither the name of the University nor the names of its contributors
19 *    may be used to endorse or promote products derived from this software
20 *    without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 */
34
35#if 0
36#ifndef lint
37static const char copyright[] =
38"@(#) Copyright (c) 1980, 1990, 1993\n\
39	The Regents of the University of California.  All rights reserved.\n";
40#endif /* not lint */
41
42#ifndef lint
43static char sccsid[] = "@(#)quotaon.c	8.1 (Berkeley) 6/6/93";
44#endif /* not lint */
45#endif
46#include <sys/cdefs.h>
47__FBSDID("$FreeBSD: stable/11/usr.sbin/quotaon/quotaon.c 330897 2018-03-14 03:19:51Z eadler $");
48
49/*
50 * Turn quota on/off for a filesystem.
51 */
52#include <sys/param.h>
53#include <sys/file.h>
54#include <sys/mount.h>
55#include <ufs/ufs/quota.h>
56#include <err.h>
57#include <fstab.h>
58#include <libutil.h>
59#include <stdio.h>
60#include <stdlib.h>
61#include <string.h>
62#include <unistd.h>
63
64static const char *qfextension[] = INITQFNAMES;
65
66static int	aflag;		/* all filesystems */
67static int	gflag;		/* operate on group quotas */
68static int	uflag;		/* operate on user quotas */
69static int	vflag;		/* verbose */
70
71static int oneof(char *, char *[], int);
72static int quotaonoff(struct fstab *fs, int, int);
73static void usage(void);
74
75int
76main(int argc, char **argv)
77{
78	struct fstab *fs;
79	const char *whoami;
80	long argnum, done = 0;
81	int ch, i, offmode = 0, errs = 0;
82
83	whoami = getprogname();
84	if (strcmp(whoami, "quotaoff") == 0)
85		offmode++;
86	else if (strcmp(whoami, "quotaon") != 0)
87		errx(1, "name must be quotaon or quotaoff");
88	while ((ch = getopt(argc, argv, "avug")) != -1) {
89		switch(ch) {
90		case 'a':
91			aflag++;
92			break;
93		case 'g':
94			gflag++;
95			break;
96		case 'u':
97			uflag++;
98			break;
99		case 'v':
100			vflag++;
101			break;
102		default:
103			usage();
104		}
105	}
106	argc -= optind;
107	argv += optind;
108	if (argc <= 0 && !aflag)
109		usage();
110	if (!gflag && !uflag) {
111		gflag++;
112		uflag++;
113	}
114	setfsent();
115	while ((fs = getfsent()) != NULL) {
116		if (strcmp(fs->fs_vfstype, "ufs") ||
117		    strcmp(fs->fs_type, FSTAB_RW))
118			continue;
119		if (aflag) {
120			if (gflag)
121				errs += quotaonoff(fs, offmode, GRPQUOTA);
122			if (uflag)
123				errs += quotaonoff(fs, offmode, USRQUOTA);
124			continue;
125		}
126		if ((argnum = oneof(fs->fs_file, argv, argc)) >= 0 ||
127		    (argnum = oneof(fs->fs_spec, argv, argc)) >= 0) {
128			done |= 1 << argnum;
129			if (gflag)
130				errs += quotaonoff(fs, offmode, GRPQUOTA);
131			if (uflag)
132				errs += quotaonoff(fs, offmode, USRQUOTA);
133		}
134	}
135	endfsent();
136	for (i = 0; i < argc; i++)
137		if ((done & (1 << i)) == 0)
138			warnx("%s not found in fstab", argv[i]);
139	exit(errs);
140}
141
142static void
143usage(void)
144{
145
146	fprintf(stderr, "%s\n%s\n%s\n%s\n",
147		"usage: quotaon [-g] [-u] [-v] -a",
148		"       quotaon [-g] [-u] [-v] filesystem ...",
149		"       quotaoff [-g] [-u] [-v] -a",
150		"       quotaoff [-g] [-u] [-v] filesystem ...");
151	exit(1);
152}
153
154static int
155quotaonoff(struct fstab *fs, int offmode, int type)
156{
157	struct quotafile *qf;
158
159	if ((qf = quota_open(fs, type, O_RDONLY)) == NULL)
160		return (0);
161	if (offmode) {
162		if (quota_off(qf) != 0) {
163			warn("%s", quota_fsname(qf));
164			return (1);
165		}
166		if (vflag)
167			printf("%s: quotas turned off\n", quota_fsname(qf));
168		quota_close(qf);
169		return(0);
170	}
171	if (quota_on(qf) != 0) {
172		warn("using %s on %s", quota_qfname(qf), quota_fsname(qf));
173		return (1);
174	}
175	if (vflag)
176		printf("%s: %s quotas turned on with data file %s\n",
177		    quota_fsname(qf), qfextension[type], quota_qfname(qf));
178	quota_close(qf);
179	return(0);
180}
181
182/*
183 * Check to see if target appears in list of size cnt.
184 */
185static int
186oneof(char *target, char *list[], int cnt)
187{
188	int i;
189
190	for (i = 0; i < cnt; i++)
191		if (strcmp(target, list[i]) == 0)
192			return (i);
193	return (-1);
194}
195