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