4.3BSD/usr/src/etc/edquota.c
/*
* Copyright (c) 1980 Regents of the University of California.
* All rights reserved. The Berkeley software License Agreement
* specifies the terms and conditions for redistribution.
*/
#ifndef lint
char copyright[] =
"@(#) Copyright (c) 1980 Regents of the University of California.\n\
All rights reserved.\n";
#endif not lint
#ifndef lint
static char sccsid[] = "@(#)edquota.c 5.3 (Berkeley) 11/4/85";
#endif not lint
/*
* Disk quota editor.
*/
#include <stdio.h>
#include <signal.h>
#include <errno.h>
#include <pwd.h>
#include <ctype.h>
#include <fstab.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/file.h>
#include <sys/quota.h>
#define DEFEDITOR "/usr/ucb/vi"
struct dquot dq[NMOUNT];
struct dquot odq[NMOUNT];
char dqf[NMOUNT][MAXPATHLEN + 1];
char odqf[NMOUNT][MAXPATHLEN + 1];
char tmpfil[] = "/tmp/EdP.aXXXXX";
char *qfname = "quotas";
char *getenv();
main(argc, argv)
char **argv;
{
int uid;
char *arg0;
mktemp(tmpfil);
close(creat(tmpfil, 0600));
chown(tmpfil, getuid(), getgid());
arg0 = *argv++;
if (argc < 2) {
fprintf(stderr, "Usage: %s [-p username] username ...\n", arg0);
unlink(tmpfil);
exit(1);
}
--argc;
if (getuid()) {
fprintf(stderr, "%s: permission denied\n", arg0);
unlink(tmpfil);
exit(1);
}
if (argc > 2 && strcmp(*argv, "-p") == 0) {
argc--, argv++;
uid = getentry(*argv++);
if (uid < 0) {
unlink(tmpfil);
exit(1);
}
getprivs(uid);
argc--;
while (argc-- > 0) {
uid = getentry(*argv++);
if (uid < 0)
continue;
getdiscq(uid, odq, odqf);
putprivs(uid);
}
unlink(tmpfil);
exit(0);
}
while (--argc >= 0) {
uid = getentry(*argv++);
if (uid < 0)
continue;
getprivs(uid);
if (editit())
putprivs(uid);
}
unlink(tmpfil);
exit(0);
}
getentry(name)
char *name;
{
struct passwd *pw;
int uid;
if (alldigits(name))
uid = atoi(name);
else if (pw = getpwnam(name))
uid = pw->pw_uid;
else {
fprintf(stderr, "%s: no such user\n", name);
sleep(1);
return (-1);
}
return (uid);
}
editit()
{
register pid, xpid;
int stat, omask;
#define mask(s) (1<<((s)-1))
omask = sigblock(mask(SIGINT)|mask(SIGQUIT)|mask(SIGHUP));
top:
if ((pid = fork()) < 0) {
extern errno;
if (errno == EPROCLIM) {
fprintf(stderr, "You have too many processes\n");
return(0);
}
if (errno == EAGAIN) {
sleep(1);
goto top;
}
perror("fork");
return (0);
}
if (pid == 0) {
register char *ed;
sigsetmask(omask);
setgid(getgid());
setuid(getuid());
if ((ed = getenv("EDITOR")) == (char *)0)
ed = DEFEDITOR;
execlp(ed, ed, tmpfil, 0);
perror(ed);
exit(1);
}
while ((xpid = wait(&stat)) >= 0)
if (xpid == pid)
break;
sigsetmask(omask);
return (!stat);
}
getprivs(uid)
register uid;
{
register i;
FILE *fd;
getdiscq(uid, dq, dqf);
for (i = 0; i < NMOUNT; i++) {
odq[i] = dq[i];
strcpy(odqf[i], dqf[i]);
}
if ((fd = fopen(tmpfil, "w")) == NULL) {
fprintf(stderr, "edquota: ");
perror(tmpfil);
exit(1);
}
for (i = 0; i < NMOUNT; i++) {
if (*dqf[i] == '\0')
continue;
fprintf(fd,
"fs %s blocks (soft = %d, hard = %d) inodes (soft = %d, hard = %d)\n"
, dqf[i]
, dbtob(dq[i].dq_bsoftlimit) / 1024
, dbtob(dq[i].dq_bhardlimit) / 1024
, dq[i].dq_isoftlimit
, dq[i].dq_ihardlimit
);
}
fclose(fd);
}
putprivs(uid)
register uid;
{
register i, j;
int n;
FILE *fd;
char line[BUFSIZ];
fd = fopen(tmpfil, "r");
if (fd == NULL) {
fprintf(stderr, "Can't re-read temp file!!\n");
return;
}
for (i = 0; i < NMOUNT; i++) {
char *cp, *dp, *next();
if (fgets(line, sizeof (line), fd) == NULL)
break;
cp = next(line, " \t");
if (cp == NULL)
break;
*cp++ = '\0';
while (*cp && *cp == '\t' && *cp == ' ')
cp++;
dp = cp, cp = next(cp, " \t");
if (cp == NULL)
break;
*cp++ = '\0';
while (*cp && *cp == '\t' && *cp == ' ')
cp++;
strcpy(dqf[i], dp);
n = sscanf(cp,
"blocks (soft = %d, hard = %d) inodes (soft = %hd, hard = %hd)\n"
, &dq[i].dq_bsoftlimit
, &dq[i].dq_bhardlimit
, &dq[i].dq_isoftlimit
, &dq[i].dq_ihardlimit
);
if (n != 4) {
fprintf(stderr, "%s: bad format\n", cp);
continue;
}
dq[i].dq_bsoftlimit = btodb(dq[i].dq_bsoftlimit * 1024);
dq[i].dq_bhardlimit = btodb(dq[i].dq_bhardlimit * 1024);
}
fclose(fd);
n = i;
for (i = 0; i < n; i++) {
if (*dqf[i] == '\0')
break;
for (j = 0; j < NMOUNT; j++) {
if (strcmp(dqf[i], odqf[j]) == 0)
break;
}
if (j >= NMOUNT)
continue;
*odqf[j] = '\0';
/*
* This isn't really good enough, it is quite likely
* to have changed while we have been away editing,
* but it's not important enough to worry about at
* the minute.
*/
dq[i].dq_curblocks = odq[j].dq_curblocks;
dq[i].dq_curinodes = odq[j].dq_curinodes;
/*
* If we've upped the inode or disk block limits
* and the guy is out of warnings, reinitialize.
*/
if (dq[i].dq_bsoftlimit > odq[j].dq_bsoftlimit &&
dq[i].dq_bwarn == 0)
dq[i].dq_bwarn = MAX_DQ_WARN;
if (dq[i].dq_isoftlimit > odq[j].dq_isoftlimit &&
dq[i].dq_iwarn == 0)
dq[i].dq_iwarn = MAX_IQ_WARN;
}
if (i < NMOUNT) {
for (j = 0; j < NMOUNT; j++) {
if (*odqf[j] == '\0')
continue;
strcpy(dqf[i], odqf[j]);
dq[i].dq_isoftlimit = 0;
dq[i].dq_ihardlimit = 0;
dq[i].dq_bsoftlimit = 0;
dq[i].dq_bhardlimit = 0;
/*
* Same applies as just above
* but matters not at all, as we are just
* turning quota'ing off for this filesys.
*/
dq[i].dq_curblocks = odq[j].dq_curblocks;
dq[i].dq_curinodes = odq[j].dq_curinodes;
if (++i >= NMOUNT)
break;
}
}
if (*dqf[0])
putdiscq(uid, dq, dqf);
}
char *
next(cp, match)
register char *cp;
char *match;
{
register char *dp;
while (cp && *cp) {
for (dp = match; dp && *dp; dp++)
if (*dp == *cp)
return (cp);
cp++;
}
return ((char *)0);
}
alldigits(s)
register char *s;
{
register c;
c = *s++;
do {
if (!isdigit(c))
return (0);
} while (c = *s++);
return (1);
}
getdiscq(uid, dq, dqf)
register uid;
register struct dquot *dq;
register char (*dqf)[MAXPATHLEN + 1];
{
register struct fstab *fs;
char qfilename[MAXPATHLEN + 1];
struct stat statb;
struct dqblk dqblk;
dev_t fsdev;
int fd;
static int warned = 0;
extern int errno;
setfsent();
while (fs = getfsent()) {
if (stat(fs->fs_spec, &statb) < 0)
continue;
fsdev = statb.st_rdev;
sprintf(qfilename, "%s/%s", fs->fs_file, qfname);
if (stat(qfilename, &statb) < 0 || statb.st_dev != fsdev)
continue;
if (quota(Q_GETDLIM, uid, fsdev, &dqblk) != 0) {
if (errno == EINVAL && !warned) {
warned++;
fprintf(stderr, "Warning: %s\n",
"Quotas are not compiled into this kernel");
sleep(3);
}
fd = open(qfilename, O_RDONLY);
if (fd < 0)
continue;
lseek(fd, (long)(uid * sizeof dqblk), L_SET);
switch (read(fd, &dqblk, sizeof dqblk)) {
case 0: /* EOF */
/*
* Convert implicit 0 quota (EOF)
* into an explicit one (zero'ed dqblk)
*/
bzero((caddr_t)&dqblk, sizeof dqblk);
break;
case sizeof dqblk: /* OK */
break;
default: /* ERROR */
fprintf(stderr, "edquota: read error in ");
perror(qfilename);
close(fd);
continue;
}
close(fd);
}
dq->dq_dqb = dqblk;
dq->dq_dev = fsdev;
strcpy(*dqf, fs->fs_file);
dq++, dqf++;
}
endfsent();
**dqf = '\0';
}
putdiscq(uid, dq, dqf)
register uid;
register struct dquot *dq;
register char (*dqf)[MAXPATHLEN + 1];
{
register fd, cnt;
struct stat sb;
struct fstab *fs;
cnt = 0;
for (cnt = 0; ++cnt <= NMOUNT && **dqf; dq++, dqf++) {
fs = getfsfile(*dqf);
if (fs == NULL) {
fprintf(stderr, "%s: not in /etc/fstab\n", *dqf);
continue;
}
strcat(*dqf, "/");
strcat(*dqf, qfname);
if (stat(*dqf, &sb) >= 0)
quota(Q_SETDLIM, uid, sb.st_dev, &dq->dq_dqb);
if ((fd = open(*dqf, 1)) < 0) {
perror(*dqf);
} else {
lseek(fd, (long)uid * (long)sizeof (struct dqblk), 0);
if (write(fd, &dq->dq_dqb, sizeof (struct dqblk)) !=
sizeof (struct dqblk)) {
fprintf(stderr, "edquota: ");
perror(*dqf);
}
close(fd);
}
}
}