![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to edquota.c CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [CSRG BSD Unix] / 43BSD / etc|
Return to edquota.c CVS log | Up to [CSRG BSD Unix] / 43BSD / etc |
1.1 root 1: /*
2: * Copyright (c) 1980 Regents of the University of California.
3: * All rights reserved. The Berkeley software License Agreement
4: * specifies the terms and conditions for redistribution.
5: */
6:
7: #ifndef lint
8: char copyright[] =
9: "@(#) Copyright (c) 1980 Regents of the University of California.\n\
10: All rights reserved.\n";
11: #endif not lint
12:
13: #ifndef lint
14: static char sccsid[] = "@(#)edquota.c 5.3 (Berkeley) 11/4/85";
15: #endif not lint
16:
17: /*
18: * Disk quota editor.
19: */
20: #include <stdio.h>
21: #include <signal.h>
22: #include <errno.h>
23: #include <pwd.h>
24: #include <ctype.h>
25: #include <fstab.h>
26:
27: #include <sys/param.h>
28: #include <sys/stat.h>
29: #include <sys/file.h>
30: #include <sys/quota.h>
31:
32: #define DEFEDITOR "/usr/ucb/vi"
33:
34: struct dquot dq[NMOUNT];
35: struct dquot odq[NMOUNT];
36: char dqf[NMOUNT][MAXPATHLEN + 1];
37: char odqf[NMOUNT][MAXPATHLEN + 1];
38:
39: char tmpfil[] = "/tmp/EdP.aXXXXX";
40: char *qfname = "quotas";
41: char *getenv();
42:
43: main(argc, argv)
44: char **argv;
45: {
46: int uid;
47: char *arg0;
48:
49: mktemp(tmpfil);
50: close(creat(tmpfil, 0600));
51: chown(tmpfil, getuid(), getgid());
52: arg0 = *argv++;
53: if (argc < 2) {
54: fprintf(stderr, "Usage: %s [-p username] username ...\n", arg0);
55: unlink(tmpfil);
56: exit(1);
57: }
58: --argc;
59: if (getuid()) {
60: fprintf(stderr, "%s: permission denied\n", arg0);
61: unlink(tmpfil);
62: exit(1);
63: }
64: if (argc > 2 && strcmp(*argv, "-p") == 0) {
65: argc--, argv++;
66: uid = getentry(*argv++);
67: if (uid < 0) {
68: unlink(tmpfil);
69: exit(1);
70: }
71: getprivs(uid);
72: argc--;
73: while (argc-- > 0) {
74: uid = getentry(*argv++);
75: if (uid < 0)
76: continue;
77: getdiscq(uid, odq, odqf);
78: putprivs(uid);
79: }
80: unlink(tmpfil);
81: exit(0);
82: }
83: while (--argc >= 0) {
84: uid = getentry(*argv++);
85: if (uid < 0)
86: continue;
87: getprivs(uid);
88: if (editit())
89: putprivs(uid);
90: }
91: unlink(tmpfil);
92: exit(0);
93: }
94:
95: getentry(name)
96: char *name;
97: {
98: struct passwd *pw;
99: int uid;
100:
101: if (alldigits(name))
102: uid = atoi(name);
103: else if (pw = getpwnam(name))
104: uid = pw->pw_uid;
105: else {
106: fprintf(stderr, "%s: no such user\n", name);
107: sleep(1);
108: return (-1);
109: }
110: return (uid);
111: }
112:
113: editit()
114: {
115: register pid, xpid;
116: int stat, omask;
117:
118: #define mask(s) (1<<((s)-1))
119: omask = sigblock(mask(SIGINT)|mask(SIGQUIT)|mask(SIGHUP));
120: top:
121: if ((pid = fork()) < 0) {
122: extern errno;
123:
124: if (errno == EPROCLIM) {
125: fprintf(stderr, "You have too many processes\n");
126: return(0);
127: }
128: if (errno == EAGAIN) {
129: sleep(1);
130: goto top;
131: }
132: perror("fork");
133: return (0);
134: }
135: if (pid == 0) {
136: register char *ed;
137:
138: sigsetmask(omask);
139: setgid(getgid());
140: setuid(getuid());
141: if ((ed = getenv("EDITOR")) == (char *)0)
142: ed = DEFEDITOR;
143: execlp(ed, ed, tmpfil, 0);
144: perror(ed);
145: exit(1);
146: }
147: while ((xpid = wait(&stat)) >= 0)
148: if (xpid == pid)
149: break;
150: sigsetmask(omask);
151: return (!stat);
152: }
153:
154: getprivs(uid)
155: register uid;
156: {
157: register i;
158: FILE *fd;
159:
160: getdiscq(uid, dq, dqf);
161: for (i = 0; i < NMOUNT; i++) {
162: odq[i] = dq[i];
163: strcpy(odqf[i], dqf[i]);
164: }
165: if ((fd = fopen(tmpfil, "w")) == NULL) {
166: fprintf(stderr, "edquota: ");
167: perror(tmpfil);
168: exit(1);
169: }
170: for (i = 0; i < NMOUNT; i++) {
171: if (*dqf[i] == '\0')
172: continue;
173: fprintf(fd,
174: "fs %s blocks (soft = %d, hard = %d) inodes (soft = %d, hard = %d)\n"
175: , dqf[i]
176: , dbtob(dq[i].dq_bsoftlimit) / 1024
177: , dbtob(dq[i].dq_bhardlimit) / 1024
178: , dq[i].dq_isoftlimit
179: , dq[i].dq_ihardlimit
180: );
181: }
182: fclose(fd);
183: }
184:
185: putprivs(uid)
186: register uid;
187: {
188: register i, j;
189: int n;
190: FILE *fd;
191: char line[BUFSIZ];
192:
193: fd = fopen(tmpfil, "r");
194: if (fd == NULL) {
195: fprintf(stderr, "Can't re-read temp file!!\n");
196: return;
197: }
198: for (i = 0; i < NMOUNT; i++) {
199: char *cp, *dp, *next();
200:
201: if (fgets(line, sizeof (line), fd) == NULL)
202: break;
203: cp = next(line, " \t");
204: if (cp == NULL)
205: break;
206: *cp++ = '\0';
207: while (*cp && *cp == '\t' && *cp == ' ')
208: cp++;
209: dp = cp, cp = next(cp, " \t");
210: if (cp == NULL)
211: break;
212: *cp++ = '\0';
213: while (*cp && *cp == '\t' && *cp == ' ')
214: cp++;
215: strcpy(dqf[i], dp);
216: n = sscanf(cp,
217: "blocks (soft = %d, hard = %d) inodes (soft = %hd, hard = %hd)\n"
218: , &dq[i].dq_bsoftlimit
219: , &dq[i].dq_bhardlimit
220: , &dq[i].dq_isoftlimit
221: , &dq[i].dq_ihardlimit
222: );
223: if (n != 4) {
224: fprintf(stderr, "%s: bad format\n", cp);
225: continue;
226: }
227: dq[i].dq_bsoftlimit = btodb(dq[i].dq_bsoftlimit * 1024);
228: dq[i].dq_bhardlimit = btodb(dq[i].dq_bhardlimit * 1024);
229: }
230: fclose(fd);
231: n = i;
232: for (i = 0; i < n; i++) {
233: if (*dqf[i] == '\0')
234: break;
235: for (j = 0; j < NMOUNT; j++) {
236: if (strcmp(dqf[i], odqf[j]) == 0)
237: break;
238: }
239: if (j >= NMOUNT)
240: continue;
241: *odqf[j] = '\0';
242: /*
243: * This isn't really good enough, it is quite likely
244: * to have changed while we have been away editing,
245: * but it's not important enough to worry about at
246: * the minute.
247: */
248: dq[i].dq_curblocks = odq[j].dq_curblocks;
249: dq[i].dq_curinodes = odq[j].dq_curinodes;
250: /*
251: * If we've upped the inode or disk block limits
252: * and the guy is out of warnings, reinitialize.
253: */
254: if (dq[i].dq_bsoftlimit > odq[j].dq_bsoftlimit &&
255: dq[i].dq_bwarn == 0)
256: dq[i].dq_bwarn = MAX_DQ_WARN;
257: if (dq[i].dq_isoftlimit > odq[j].dq_isoftlimit &&
258: dq[i].dq_iwarn == 0)
259: dq[i].dq_iwarn = MAX_IQ_WARN;
260: }
261: if (i < NMOUNT) {
262: for (j = 0; j < NMOUNT; j++) {
263: if (*odqf[j] == '\0')
264: continue;
265: strcpy(dqf[i], odqf[j]);
266: dq[i].dq_isoftlimit = 0;
267: dq[i].dq_ihardlimit = 0;
268: dq[i].dq_bsoftlimit = 0;
269: dq[i].dq_bhardlimit = 0;
270: /*
271: * Same applies as just above
272: * but matters not at all, as we are just
273: * turning quota'ing off for this filesys.
274: */
275: dq[i].dq_curblocks = odq[j].dq_curblocks;
276: dq[i].dq_curinodes = odq[j].dq_curinodes;
277: if (++i >= NMOUNT)
278: break;
279: }
280: }
281: if (*dqf[0])
282: putdiscq(uid, dq, dqf);
283: }
284:
285: char *
286: next(cp, match)
287: register char *cp;
288: char *match;
289: {
290: register char *dp;
291:
292: while (cp && *cp) {
293: for (dp = match; dp && *dp; dp++)
294: if (*dp == *cp)
295: return (cp);
296: cp++;
297: }
298: return ((char *)0);
299: }
300:
301: alldigits(s)
302: register char *s;
303: {
304: register c;
305:
306: c = *s++;
307: do {
308: if (!isdigit(c))
309: return (0);
310: } while (c = *s++);
311: return (1);
312: }
313:
314: getdiscq(uid, dq, dqf)
315: register uid;
316: register struct dquot *dq;
317: register char (*dqf)[MAXPATHLEN + 1];
318: {
319: register struct fstab *fs;
320: char qfilename[MAXPATHLEN + 1];
321: struct stat statb;
322: struct dqblk dqblk;
323: dev_t fsdev;
324: int fd;
325: static int warned = 0;
326: extern int errno;
327:
328: setfsent();
329: while (fs = getfsent()) {
330: if (stat(fs->fs_spec, &statb) < 0)
331: continue;
332: fsdev = statb.st_rdev;
333: sprintf(qfilename, "%s/%s", fs->fs_file, qfname);
334: if (stat(qfilename, &statb) < 0 || statb.st_dev != fsdev)
335: continue;
336: if (quota(Q_GETDLIM, uid, fsdev, &dqblk) != 0) {
337: if (errno == EINVAL && !warned) {
338: warned++;
339: fprintf(stderr, "Warning: %s\n",
340: "Quotas are not compiled into this kernel");
341: sleep(3);
342: }
343: fd = open(qfilename, O_RDONLY);
344: if (fd < 0)
345: continue;
346: lseek(fd, (long)(uid * sizeof dqblk), L_SET);
347: switch (read(fd, &dqblk, sizeof dqblk)) {
348: case 0: /* EOF */
349: /*
350: * Convert implicit 0 quota (EOF)
351: * into an explicit one (zero'ed dqblk)
352: */
353: bzero((caddr_t)&dqblk, sizeof dqblk);
354: break;
355:
356: case sizeof dqblk: /* OK */
357: break;
358:
359: default: /* ERROR */
360: fprintf(stderr, "edquota: read error in ");
361: perror(qfilename);
362: close(fd);
363: continue;
364: }
365: close(fd);
366: }
367: dq->dq_dqb = dqblk;
368: dq->dq_dev = fsdev;
369: strcpy(*dqf, fs->fs_file);
370: dq++, dqf++;
371: }
372: endfsent();
373: **dqf = '\0';
374: }
375:
376: putdiscq(uid, dq, dqf)
377: register uid;
378: register struct dquot *dq;
379: register char (*dqf)[MAXPATHLEN + 1];
380: {
381: register fd, cnt;
382: struct stat sb;
383: struct fstab *fs;
384:
385: cnt = 0;
386: for (cnt = 0; ++cnt <= NMOUNT && **dqf; dq++, dqf++) {
387: fs = getfsfile(*dqf);
388: if (fs == NULL) {
389: fprintf(stderr, "%s: not in /etc/fstab\n", *dqf);
390: continue;
391: }
392: strcat(*dqf, "/");
393: strcat(*dqf, qfname);
394: if (stat(*dqf, &sb) >= 0)
395: quota(Q_SETDLIM, uid, sb.st_dev, &dq->dq_dqb);
396: if ((fd = open(*dqf, 1)) < 0) {
397: perror(*dqf);
398: } else {
399: lseek(fd, (long)uid * (long)sizeof (struct dqblk), 0);
400: if (write(fd, &dq->dq_dqb, sizeof (struct dqblk)) !=
401: sizeof (struct dqblk)) {
402: fprintf(stderr, "edquota: ");
403: perror(*dqf);
404: }
405: close(fd);
406: }
407: }
408: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.