![[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] / 42BSD / etc|
Return to edquota.c CVS log | Up to [CSRG BSD Unix] / 42BSD / etc |
1.1 ! root 1: #ifndef lint ! 2: static char sccsid[] = "@(#)edquota.c 4.5 (Berkeley, from Melbourne) 6/12/83"; ! 3: #endif ! 4: ! 5: /* ! 6: * Disk quota editor. ! 7: */ ! 8: #include <stdio.h> ! 9: #include <signal.h> ! 10: #include <errno.h> ! 11: #include <pwd.h> ! 12: #include <ctype.h> ! 13: #include <fstab.h> ! 14: ! 15: #include <sys/param.h> ! 16: #include <sys/stat.h> ! 17: #include <sys/file.h> ! 18: #include <sys/quota.h> ! 19: ! 20: #define DEFEDITOR "/usr/ucb/vi" ! 21: ! 22: struct dquot dq[NMOUNT]; ! 23: struct dquot odq[NMOUNT]; ! 24: char dqf[NMOUNT][MAXPATHLEN + 1]; ! 25: char odqf[NMOUNT][MAXPATHLEN + 1]; ! 26: ! 27: char tmpfil[] = "/tmp/EdP.aXXXXX"; ! 28: char *qfname = "quotas"; ! 29: char *getenv(); ! 30: ! 31: main(argc, argv) ! 32: char **argv; ! 33: { ! 34: int uid; ! 35: char *arg0; ! 36: ! 37: mktemp(tmpfil); ! 38: close(creat(tmpfil, 0600)); ! 39: chown(tmpfil, getuid(), getgid()); ! 40: arg0 = *argv++; ! 41: if (argc < 2) { ! 42: fprintf(stderr, "Usage: %s [-p username] username ...\n", arg0); ! 43: unlink(tmpfil); ! 44: exit(1); ! 45: } ! 46: --argc; ! 47: if (getuid()) { ! 48: fprintf(stderr, "%s: permission denied\n", arg0); ! 49: unlink(tmpfil); ! 50: exit(1); ! 51: } ! 52: if (argc > 2 && strcmp(*argv, "-p") == 0) { ! 53: argc--, argv++; ! 54: uid = getentry(*argv++); ! 55: if (uid < 0) { ! 56: unlink(tmpfil); ! 57: exit(1); ! 58: } ! 59: getprivs(uid); ! 60: argc--; ! 61: while (argc-- > 0) { ! 62: uid = getentry(*argv++); ! 63: if (uid < 0) ! 64: continue; ! 65: getdiscq(uid, odq, odqf); ! 66: putprivs(uid); ! 67: } ! 68: unlink(tmpfil); ! 69: exit(0); ! 70: } ! 71: while (--argc >= 0) { ! 72: uid = getentry(*argv++); ! 73: if (uid < 0) ! 74: continue; ! 75: getprivs(uid); ! 76: if (editit()) ! 77: putprivs(uid); ! 78: } ! 79: unlink(tmpfil); ! 80: exit(0); ! 81: } ! 82: ! 83: getentry(name) ! 84: char *name; ! 85: { ! 86: struct passwd *pw; ! 87: int uid; ! 88: ! 89: if (alldigits(name)) ! 90: uid = atoi(name); ! 91: else if (pw = getpwnam(name)) ! 92: uid = pw->pw_uid; ! 93: else { ! 94: fprintf(stderr, "%s: no such user\n", name); ! 95: sleep(1); ! 96: return (-1); ! 97: } ! 98: return (uid); ! 99: } ! 100: ! 101: editit() ! 102: { ! 103: register pid, xpid; ! 104: int stat, omask; ! 105: ! 106: #define mask(s) (1<<((s)-1)) ! 107: omask = sigblock(mask(SIGINT)|mask(SIGQUIT)|mask(SIGHUP)); ! 108: top: ! 109: if ((pid = fork()) < 0) { ! 110: extern errno; ! 111: ! 112: if (errno == EPROCLIM) { ! 113: fprintf(stderr, "You have too many processes\n"); ! 114: return(0); ! 115: } ! 116: if (errno == EAGAIN) { ! 117: sleep(1); ! 118: goto top; ! 119: } ! 120: perror("fork"); ! 121: return (0); ! 122: } ! 123: if (pid == 0) { ! 124: register char *ed; ! 125: ! 126: sigsetmask(omask); ! 127: setgid(getgid()); ! 128: setuid(getuid()); ! 129: if ((ed = getenv("EDITOR")) == (char *)0) ! 130: ed = DEFEDITOR; ! 131: execlp(ed, ed, tmpfil, 0); ! 132: perror(ed); ! 133: exit(1); ! 134: } ! 135: while ((xpid = wait(&stat)) >= 0) ! 136: if (xpid == pid) ! 137: break; ! 138: sigsetmask(omask); ! 139: return (!stat); ! 140: } ! 141: ! 142: getprivs(uid) ! 143: register uid; ! 144: { ! 145: register i; ! 146: FILE *fd; ! 147: ! 148: getdiscq(uid, dq, dqf); ! 149: for (i = 0; i < NMOUNT; i++) { ! 150: odq[i] = dq[i]; ! 151: strcpy(odqf[i], dqf[i]); ! 152: } ! 153: if ((fd = fopen(tmpfil, "w")) == NULL) { ! 154: fprintf(stderr, "edquota: "); ! 155: perror(tmpfil); ! 156: exit(1); ! 157: } ! 158: for (i = 0; i < NMOUNT; i++) { ! 159: if (*dqf[i] == '\0') ! 160: continue; ! 161: fprintf(fd, ! 162: "fs %s blocks (soft = %d, hard = %d) inodes (soft = %d, hard = %d)\n" ! 163: , dqf[i] ! 164: , dq[i].dq_bsoftlimit / btodb(1024) ! 165: , dq[i].dq_bhardlimit / btodb(1024) ! 166: , dq[i].dq_isoftlimit ! 167: , dq[i].dq_ihardlimit ! 168: ); ! 169: } ! 170: fclose(fd); ! 171: } ! 172: ! 173: putprivs(uid) ! 174: register uid; ! 175: { ! 176: register i, j; ! 177: int n; ! 178: FILE *fd; ! 179: char line[BUFSIZ]; ! 180: ! 181: fd = fopen(tmpfil, "r"); ! 182: if (fd == NULL) { ! 183: fprintf(stderr, "Can't re-read temp file!!\n"); ! 184: return; ! 185: } ! 186: for (i = 0; i < NMOUNT; i++) { ! 187: char *cp, *dp, *next(); ! 188: ! 189: if (fgets(line, sizeof (line), fd) == NULL) ! 190: break; ! 191: cp = next(line, " \t"); ! 192: if (cp == NULL) ! 193: break; ! 194: *cp++ = '\0'; ! 195: while (*cp && *cp == '\t' && *cp == ' ') ! 196: cp++; ! 197: dp = cp, cp = next(cp, " \t"); ! 198: if (cp == NULL) ! 199: break; ! 200: *cp++ = '\0'; ! 201: while (*cp && *cp == '\t' && *cp == ' ') ! 202: cp++; ! 203: strcpy(dqf[i], dp); ! 204: n = sscanf(cp, ! 205: "blocks (soft = %d, hard = %d) inodes (soft = %hd, hard = %hd)\n" ! 206: , &dq[i].dq_bsoftlimit ! 207: , &dq[i].dq_bhardlimit ! 208: , &dq[i].dq_isoftlimit ! 209: , &dq[i].dq_ihardlimit ! 210: ); ! 211: if (n != 4) { ! 212: fprintf(stderr, "%s: bad format\n", cp); ! 213: continue; ! 214: } ! 215: dq[i].dq_bsoftlimit *= btodb(1024); ! 216: dq[i].dq_bhardlimit *= btodb(1024); ! 217: } ! 218: fclose(fd); ! 219: n = i; ! 220: for (i = 0; i < n; i++) { ! 221: if (*dqf[i] == '\0') ! 222: break; ! 223: for (j = 0; j < NMOUNT; j++) { ! 224: if (strcmp(dqf[i], odqf[j]) == 0) ! 225: break; ! 226: } ! 227: if (j >= NMOUNT) ! 228: continue; ! 229: *odqf[j] = '\0'; ! 230: if (dq[i].dq_isoftlimit == odq[j].dq_isoftlimit && ! 231: dq[i].dq_ihardlimit == odq[j].dq_ihardlimit && ! 232: dq[i].dq_bsoftlimit == odq[j].dq_bsoftlimit && ! 233: dq[i].dq_bhardlimit == odq[j].dq_bhardlimit) { ! 234: for (j = i; j < NMOUNT; j++) { ! 235: dq[j] = dq[j+1]; ! 236: strcpy(dqf[j], dqf[j+1]); ! 237: } ! 238: *dqf[j] = '\0'; ! 239: i--; ! 240: continue; ! 241: } ! 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: ! 322: setfsent(); ! 323: while (fs = getfsent()) { ! 324: struct stat statb; ! 325: struct dqblk dqblk; ! 326: dev_t fsdev; ! 327: ! 328: if (stat(fs->fs_spec, &statb) < 0) ! 329: continue; ! 330: fsdev = statb.st_rdev; ! 331: sprintf(qfilename, "%s/%s", fs->fs_file, qfname); ! 332: if (stat(qfilename, &statb) < 0 || statb.st_dev != fsdev) ! 333: continue; ! 334: if (quota(Q_GETDLIM, uid, fsdev, &dqblk) != 0) { ! 335: register fd = open(qfilename, O_RDONLY); ! 336: ! 337: if (fd < 0) ! 338: continue; ! 339: lseek(fd, (long)(uid * sizeof dqblk), L_SET); ! 340: if (read(fd, &dqblk, sizeof dqblk) != sizeof (dqblk)) { ! 341: close(fd); ! 342: continue; ! 343: } ! 344: close(fd); ! 345: } ! 346: dq->dq_dqb = dqblk; ! 347: dq->dq_dev = fsdev; ! 348: strcpy(*dqf, fs->fs_file); ! 349: dq++, dqf++; ! 350: } ! 351: endfsent(); ! 352: **dqf = '\0'; ! 353: } ! 354: ! 355: putdiscq(uid, dq, dqf) ! 356: register uid; ! 357: register struct dquot *dq; ! 358: register char (*dqf)[MAXPATHLEN + 1]; ! 359: { ! 360: register fd, cnt; ! 361: struct stat sb; ! 362: struct fstab *fs; ! 363: ! 364: cnt = 0; ! 365: for (cnt = 0; ++cnt <= NMOUNT && **dqf; dq++, dqf++) { ! 366: fs = getfsfile(*dqf); ! 367: if (fs == NULL) { ! 368: fprintf(stderr, "%s: not in /etc/fstab\n", *dqf); ! 369: continue; ! 370: } ! 371: strcat(*dqf, "/"); ! 372: strcat(*dqf, qfname); ! 373: if (stat(*dqf, &sb) >= 0) ! 374: quota(Q_SETDLIM, uid, sb.st_dev, &dq->dq_dqb); ! 375: if ((fd = open(*dqf, 1)) < 0) { ! 376: perror(*dqf); ! 377: } else { ! 378: lseek(fd, (long)uid * (long)sizeof (struct dqblk), 0); ! 379: if (write(fd, &dq->dq_dqb, sizeof (struct dqblk)) != ! 380: sizeof (struct dqblk)) { ! 381: fprintf(stderr, "edquota: "); ! 382: perror(*dqf); ! 383: } ! 384: close(fd); ! 385: } ! 386: } ! 387: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.