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