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 * 3. 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$"); 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