11b3515f3SDag-Erling Smørgrav /*- 21b3515f3SDag-Erling Smørgrav * Copyright (c) 2008 Dag-Erling Coïdan Smørgrav 38bd6a3abSKirk McKusick * Copyright (c) 2008 Marshall Kirk McKusick 41b3515f3SDag-Erling Smørgrav * All rights reserved. 51b3515f3SDag-Erling Smørgrav * 61b3515f3SDag-Erling Smørgrav * Redistribution and use in source and binary forms, with or without 71b3515f3SDag-Erling Smørgrav * modification, are permitted provided that the following conditions 81b3515f3SDag-Erling Smørgrav * are met: 91b3515f3SDag-Erling Smørgrav * 1. Redistributions of source code must retain the above copyright 101b3515f3SDag-Erling Smørgrav * notice, this list of conditions and the following disclaimer 111b3515f3SDag-Erling Smørgrav * in this position and unchanged. 121b3515f3SDag-Erling Smørgrav * 2. Redistributions in binary form must reproduce the above copyright 131b3515f3SDag-Erling Smørgrav * notice, this list of conditions and the following disclaimer in the 141b3515f3SDag-Erling Smørgrav * documentation and/or other materials provided with the distribution. 151b3515f3SDag-Erling Smørgrav * 161b3515f3SDag-Erling Smørgrav * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 171b3515f3SDag-Erling Smørgrav * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 181b3515f3SDag-Erling Smørgrav * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 191b3515f3SDag-Erling Smørgrav * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 201b3515f3SDag-Erling Smørgrav * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 211b3515f3SDag-Erling Smørgrav * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 221b3515f3SDag-Erling Smørgrav * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 231b3515f3SDag-Erling Smørgrav * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 241b3515f3SDag-Erling Smørgrav * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 251b3515f3SDag-Erling Smørgrav * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 261b3515f3SDag-Erling Smørgrav * SUCH DAMAGE. 271b3515f3SDag-Erling Smørgrav * 281b3515f3SDag-Erling Smørgrav * $FreeBSD$ 291b3515f3SDag-Erling Smørgrav */ 301b3515f3SDag-Erling Smørgrav 311b3515f3SDag-Erling Smørgrav #include <sys/types.h> 321b3515f3SDag-Erling Smørgrav #include <sys/endian.h> 33916e406eSKirk McKusick #include <sys/mount.h> 341b3515f3SDag-Erling Smørgrav #include <sys/stat.h> 351b3515f3SDag-Erling Smørgrav 361b3515f3SDag-Erling Smørgrav #include <ufs/ufs/quota.h> 371b3515f3SDag-Erling Smørgrav 381b3515f3SDag-Erling Smørgrav #include <errno.h> 391b3515f3SDag-Erling Smørgrav #include <fcntl.h> 40916e406eSKirk McKusick #include <fstab.h> 411b3515f3SDag-Erling Smørgrav #include <grp.h> 421b3515f3SDag-Erling Smørgrav #include <pwd.h> 431b3515f3SDag-Erling Smørgrav #include <libutil.h> 441b3515f3SDag-Erling Smørgrav #include <stdint.h> 45916e406eSKirk McKusick #include <stdio.h> 461b3515f3SDag-Erling Smørgrav #include <stdlib.h> 471b3515f3SDag-Erling Smørgrav #include <string.h> 481b3515f3SDag-Erling Smørgrav #include <unistd.h> 491b3515f3SDag-Erling Smørgrav 501b3515f3SDag-Erling Smørgrav struct quotafile { 518bd6a3abSKirk McKusick int fd; /* -1 means using quotactl for access */ 525666aadbSDag-Erling Smørgrav int accmode; /* access mode */ 538bd6a3abSKirk McKusick int wordsize; /* 32-bit or 64-bit limits */ 548bd6a3abSKirk McKusick int quotatype; /* USRQUOTA or GRPQUOTA */ 555666aadbSDag-Erling Smørgrav dev_t dev; /* device */ 568bd6a3abSKirk McKusick char fsname[MAXPATHLEN + 1]; /* mount point of filesystem */ 578bd6a3abSKirk McKusick char qfname[MAXPATHLEN + 1]; /* quota file if not using quotactl */ 581b3515f3SDag-Erling Smørgrav }; 591b3515f3SDag-Erling Smørgrav 60916e406eSKirk McKusick static const char *qfextension[] = INITQFNAMES; 61916e406eSKirk McKusick 628bd6a3abSKirk McKusick /* 638bd6a3abSKirk McKusick * Check to see if a particular quota is to be enabled. 648bd6a3abSKirk McKusick */ 658bd6a3abSKirk McKusick static int 668bd6a3abSKirk McKusick hasquota(struct fstab *fs, int type, char *qfnamep, int qfbufsize) 678bd6a3abSKirk McKusick { 688bd6a3abSKirk McKusick char *opt; 698bd6a3abSKirk McKusick char *cp; 708bd6a3abSKirk McKusick struct statfs sfb; 718bd6a3abSKirk McKusick char buf[BUFSIZ]; 728bd6a3abSKirk McKusick static char initname, usrname[100], grpname[100]; 738bd6a3abSKirk McKusick 745666aadbSDag-Erling Smørgrav /* 755666aadbSDag-Erling Smørgrav * 1) we only need one of these 765666aadbSDag-Erling Smørgrav * 2) fstab may specify a different filename 775666aadbSDag-Erling Smørgrav */ 788bd6a3abSKirk McKusick if (!initname) { 798bd6a3abSKirk McKusick (void)snprintf(usrname, sizeof(usrname), "%s%s", 808bd6a3abSKirk McKusick qfextension[USRQUOTA], QUOTAFILENAME); 818bd6a3abSKirk McKusick (void)snprintf(grpname, sizeof(grpname), "%s%s", 828bd6a3abSKirk McKusick qfextension[GRPQUOTA], QUOTAFILENAME); 838bd6a3abSKirk McKusick initname = 1; 848bd6a3abSKirk McKusick } 858bd6a3abSKirk McKusick strcpy(buf, fs->fs_mntops); 868bd6a3abSKirk McKusick for (opt = strtok(buf, ","); opt; opt = strtok(NULL, ",")) { 878bd6a3abSKirk McKusick if ((cp = index(opt, '='))) 888bd6a3abSKirk McKusick *cp++ = '\0'; 898bd6a3abSKirk McKusick if (type == USRQUOTA && strcmp(opt, usrname) == 0) 908bd6a3abSKirk McKusick break; 918bd6a3abSKirk McKusick if (type == GRPQUOTA && strcmp(opt, grpname) == 0) 928bd6a3abSKirk McKusick break; 938bd6a3abSKirk McKusick } 948bd6a3abSKirk McKusick if (!opt) 958bd6a3abSKirk McKusick return (0); 968bd6a3abSKirk McKusick /* 978bd6a3abSKirk McKusick * Ensure that the filesystem is mounted. 988bd6a3abSKirk McKusick */ 998bd6a3abSKirk McKusick if (statfs(fs->fs_file, &sfb) != 0 || 1008bd6a3abSKirk McKusick strcmp(fs->fs_file, sfb.f_mntonname)) { 1018bd6a3abSKirk McKusick return (0); 1028bd6a3abSKirk McKusick } 1038bd6a3abSKirk McKusick if (cp) { 1048bd6a3abSKirk McKusick strncpy(qfnamep, cp, qfbufsize); 1058bd6a3abSKirk McKusick } else { 1068bd6a3abSKirk McKusick (void)snprintf(qfnamep, qfbufsize, "%s/%s.%s", fs->fs_file, 1078bd6a3abSKirk McKusick QUOTAFILENAME, qfextension[type]); 1088bd6a3abSKirk McKusick } 1098bd6a3abSKirk McKusick return (1); 1108bd6a3abSKirk McKusick } 1118bd6a3abSKirk McKusick 1121b3515f3SDag-Erling Smørgrav struct quotafile * 1138bd6a3abSKirk McKusick quota_open(struct fstab *fs, int quotatype, int openflags) 1141b3515f3SDag-Erling Smørgrav { 1151b3515f3SDag-Erling Smørgrav struct quotafile *qf; 1161b3515f3SDag-Erling Smørgrav struct dqhdr64 dqh; 1178bd6a3abSKirk McKusick struct group *grp; 1185666aadbSDag-Erling Smørgrav struct stat st; 1198bd6a3abSKirk McKusick int qcmd, serrno; 1201b3515f3SDag-Erling Smørgrav 121c3616249SKirk McKusick if (strcmp(fs->fs_vfstype, "ufs")) 122c3616249SKirk McKusick return (NULL); 1231b3515f3SDag-Erling Smørgrav if ((qf = calloc(1, sizeof(*qf))) == NULL) 1241b3515f3SDag-Erling Smørgrav return (NULL); 1255666aadbSDag-Erling Smørgrav qf->fd = -1; 1268bd6a3abSKirk McKusick qf->quotatype = quotatype; 1278bd6a3abSKirk McKusick strncpy(qf->fsname, fs->fs_file, sizeof(qf->fsname)); 1285666aadbSDag-Erling Smørgrav if (stat(qf->fsname, &st) != 0) 1295666aadbSDag-Erling Smørgrav goto error; 1305666aadbSDag-Erling Smørgrav qf->dev = st.st_dev; 1316197731dSKirk McKusick serrno = hasquota(fs, quotatype, qf->qfname, sizeof(qf->qfname)); 1328bd6a3abSKirk McKusick qcmd = QCMD(Q_GETQUOTA, quotatype); 1338bd6a3abSKirk McKusick if (quotactl(fs->fs_file, qcmd, 0, &dqh) == 0) { 1348bd6a3abSKirk McKusick qf->wordsize = 64; 1358bd6a3abSKirk McKusick return (qf); 1368bd6a3abSKirk McKusick } 1376197731dSKirk McKusick if (serrno == 0) { 1388bd6a3abSKirk McKusick errno = EOPNOTSUPP; 1395666aadbSDag-Erling Smørgrav goto error; 1408bd6a3abSKirk McKusick } 1415666aadbSDag-Erling Smørgrav qf->accmode = openflags & O_ACCMODE; 1425666aadbSDag-Erling Smørgrav if ((qf->fd = open(qf->qfname, qf->accmode)) < 0 && 1435666aadbSDag-Erling Smørgrav (openflags & O_CREAT) != O_CREAT) 1445666aadbSDag-Erling Smørgrav goto error; 1458bd6a3abSKirk McKusick /* File open worked, so process it */ 1468bd6a3abSKirk McKusick if (qf->fd != -1) { 1478bd6a3abSKirk McKusick qf->wordsize = 32; 1481b3515f3SDag-Erling Smørgrav switch (read(qf->fd, &dqh, sizeof(dqh))) { 1491b3515f3SDag-Erling Smørgrav case -1: 1505666aadbSDag-Erling Smørgrav goto error; 1511b3515f3SDag-Erling Smørgrav case sizeof(dqh): 1521b3515f3SDag-Erling Smørgrav if (strcmp(dqh.dqh_magic, Q_DQHDR64_MAGIC) != 0) { 1531b3515f3SDag-Erling Smørgrav /* no magic, assume 32 bits */ 1548bd6a3abSKirk McKusick qf->wordsize = 32; 1551b3515f3SDag-Erling Smørgrav return (qf); 1561b3515f3SDag-Erling Smørgrav } 1571b3515f3SDag-Erling Smørgrav if (be32toh(dqh.dqh_version) != Q_DQHDR64_VERSION || 1581b3515f3SDag-Erling Smørgrav be32toh(dqh.dqh_hdrlen) != sizeof(struct dqhdr64) || 1591b3515f3SDag-Erling Smørgrav be32toh(dqh.dqh_reclen) != sizeof(struct dqblk64)) { 1601b3515f3SDag-Erling Smørgrav /* correct magic, wrong version / lengths */ 1611b3515f3SDag-Erling Smørgrav errno = EINVAL; 1625666aadbSDag-Erling Smørgrav goto error; 1631b3515f3SDag-Erling Smørgrav } 1648bd6a3abSKirk McKusick qf->wordsize = 64; 1651b3515f3SDag-Erling Smørgrav return (qf); 1661b3515f3SDag-Erling Smørgrav default: 1678bd6a3abSKirk McKusick qf->wordsize = 32; 1681b3515f3SDag-Erling Smørgrav return (qf); 1691b3515f3SDag-Erling Smørgrav } 1701b3515f3SDag-Erling Smørgrav /* not reached */ 1711b3515f3SDag-Erling Smørgrav } 1725666aadbSDag-Erling Smørgrav /* open failed, but O_CREAT was specified, so create a new file */ 1735666aadbSDag-Erling Smørgrav if ((qf->fd = open(qf->qfname, O_RDWR|O_CREAT|O_TRUNC, 0)) < 0) 1745666aadbSDag-Erling Smørgrav goto error; 1758bd6a3abSKirk McKusick qf->wordsize = 64; 1761b3515f3SDag-Erling Smørgrav memset(&dqh, 0, sizeof(dqh)); 1771b3515f3SDag-Erling Smørgrav memcpy(dqh.dqh_magic, Q_DQHDR64_MAGIC, sizeof(dqh.dqh_magic)); 1781b3515f3SDag-Erling Smørgrav dqh.dqh_version = htobe32(Q_DQHDR64_VERSION); 1791b3515f3SDag-Erling Smørgrav dqh.dqh_hdrlen = htobe32(sizeof(struct dqhdr64)); 1801b3515f3SDag-Erling Smørgrav dqh.dqh_reclen = htobe32(sizeof(struct dqblk64)); 181c3616249SKirk McKusick if (write(qf->fd, &dqh, sizeof(dqh)) != sizeof(dqh)) { 182c3616249SKirk McKusick /* it was one we created ourselves */ 183c3616249SKirk McKusick unlink(qf->qfname); 1845666aadbSDag-Erling Smørgrav goto error; 185c3616249SKirk McKusick } 1861b3515f3SDag-Erling Smørgrav grp = getgrnam(QUOTAGROUP); 1871b3515f3SDag-Erling Smørgrav fchown(qf->fd, 0, grp ? grp->gr_gid : 0); 1881b3515f3SDag-Erling Smørgrav fchmod(qf->fd, 0640); 1891b3515f3SDag-Erling Smørgrav return (qf); 1905666aadbSDag-Erling Smørgrav error: 1915666aadbSDag-Erling Smørgrav serrno = errno; 1925666aadbSDag-Erling Smørgrav /* did we have an open file? */ 193c3616249SKirk McKusick if (qf->fd != -1) 1945666aadbSDag-Erling Smørgrav close(qf->fd); 1955666aadbSDag-Erling Smørgrav free(qf); 1965666aadbSDag-Erling Smørgrav errno = serrno; 1975666aadbSDag-Erling Smørgrav return (NULL); 1981b3515f3SDag-Erling Smørgrav } 1991b3515f3SDag-Erling Smørgrav 2001b3515f3SDag-Erling Smørgrav void 2011b3515f3SDag-Erling Smørgrav quota_close(struct quotafile *qf) 2021b3515f3SDag-Erling Smørgrav { 2031b3515f3SDag-Erling Smørgrav 2048bd6a3abSKirk McKusick if (qf->fd != -1) 2051b3515f3SDag-Erling Smørgrav close(qf->fd); 2061b3515f3SDag-Erling Smørgrav free(qf); 2071b3515f3SDag-Erling Smørgrav } 2081b3515f3SDag-Erling Smørgrav 209e525d16aSKirk McKusick int 210e525d16aSKirk McKusick quota_on(struct quotafile *qf) 211e525d16aSKirk McKusick { 212e525d16aSKirk McKusick int qcmd; 213e525d16aSKirk McKusick 214e525d16aSKirk McKusick qcmd = QCMD(Q_QUOTAON, qf->quotatype); 215e525d16aSKirk McKusick return (quotactl(qf->fsname, qcmd, 0, qf->qfname)); 216e525d16aSKirk McKusick } 217e525d16aSKirk McKusick 218e525d16aSKirk McKusick int 219e525d16aSKirk McKusick quota_off(struct quotafile *qf) 220e525d16aSKirk McKusick { 221e525d16aSKirk McKusick 222e525d16aSKirk McKusick return (quotactl(qf->fsname, QCMD(Q_QUOTAOFF, qf->quotatype), 0, 0)); 223e525d16aSKirk McKusick } 224e525d16aSKirk McKusick 2255666aadbSDag-Erling Smørgrav const char * 2265666aadbSDag-Erling Smørgrav quota_fsname(const struct quotafile *qf) 2275666aadbSDag-Erling Smørgrav { 2285666aadbSDag-Erling Smørgrav 2295666aadbSDag-Erling Smørgrav return (qf->fsname); 2305666aadbSDag-Erling Smørgrav } 2315666aadbSDag-Erling Smørgrav 2325666aadbSDag-Erling Smørgrav const char * 2335666aadbSDag-Erling Smørgrav quota_qfname(const struct quotafile *qf) 2345666aadbSDag-Erling Smørgrav { 2355666aadbSDag-Erling Smørgrav 2365666aadbSDag-Erling Smørgrav return (qf->qfname); 2375666aadbSDag-Erling Smørgrav } 2385666aadbSDag-Erling Smørgrav 2395666aadbSDag-Erling Smørgrav int 2405666aadbSDag-Erling Smørgrav quota_check_path(const struct quotafile *qf, const char *path) 2415666aadbSDag-Erling Smørgrav { 2425666aadbSDag-Erling Smørgrav struct stat st; 2435666aadbSDag-Erling Smørgrav 2445666aadbSDag-Erling Smørgrav if (stat(path, &st) == -1) 2455666aadbSDag-Erling Smørgrav return (-1); 2465666aadbSDag-Erling Smørgrav return (st.st_dev == qf->dev); 2475666aadbSDag-Erling Smørgrav } 2485666aadbSDag-Erling Smørgrav 2496197731dSKirk McKusick int 2506197731dSKirk McKusick quota_maxid(struct quotafile *qf) 2516197731dSKirk McKusick { 2526197731dSKirk McKusick struct stat st; 2536197731dSKirk McKusick 2546197731dSKirk McKusick if (stat(qf->qfname, &st) < 0) 2556197731dSKirk McKusick return (0); 2566197731dSKirk McKusick switch (qf->wordsize) { 2576197731dSKirk McKusick case 32: 258aee785baSKirk McKusick return (st.st_size / sizeof(struct dqblk32) - 1); 2596197731dSKirk McKusick case 64: 260aee785baSKirk McKusick return (st.st_size / sizeof(struct dqblk64) - 2); 2616197731dSKirk McKusick default: 2626197731dSKirk McKusick return (0); 2636197731dSKirk McKusick } 2646197731dSKirk McKusick /* not reached */ 2656197731dSKirk McKusick } 2666197731dSKirk McKusick 2671b3515f3SDag-Erling Smørgrav static int 2681b3515f3SDag-Erling Smørgrav quota_read32(struct quotafile *qf, struct dqblk *dqb, int id) 2691b3515f3SDag-Erling Smørgrav { 2701b3515f3SDag-Erling Smørgrav struct dqblk32 dqb32; 2711b3515f3SDag-Erling Smørgrav off_t off; 2721b3515f3SDag-Erling Smørgrav 2731b3515f3SDag-Erling Smørgrav off = id * sizeof(struct dqblk32); 2741b3515f3SDag-Erling Smørgrav if (lseek(qf->fd, off, SEEK_SET) == -1) 2751b3515f3SDag-Erling Smørgrav return (-1); 2761b3515f3SDag-Erling Smørgrav switch (read(qf->fd, &dqb32, sizeof(dqb32))) { 2771b3515f3SDag-Erling Smørgrav case 0: 2786197731dSKirk McKusick memset(dqb, 0, sizeof(*dqb)); 2791b3515f3SDag-Erling Smørgrav return (0); 2801b3515f3SDag-Erling Smørgrav case sizeof(dqb32): 2811b3515f3SDag-Erling Smørgrav dqb->dqb_bhardlimit = dqb32.dqb_bhardlimit; 2821b3515f3SDag-Erling Smørgrav dqb->dqb_bsoftlimit = dqb32.dqb_bsoftlimit; 2831b3515f3SDag-Erling Smørgrav dqb->dqb_curblocks = dqb32.dqb_curblocks; 2841b3515f3SDag-Erling Smørgrav dqb->dqb_ihardlimit = dqb32.dqb_ihardlimit; 2851b3515f3SDag-Erling Smørgrav dqb->dqb_isoftlimit = dqb32.dqb_isoftlimit; 2861b3515f3SDag-Erling Smørgrav dqb->dqb_curinodes = dqb32.dqb_curinodes; 2871b3515f3SDag-Erling Smørgrav dqb->dqb_btime = dqb32.dqb_btime; 2881b3515f3SDag-Erling Smørgrav dqb->dqb_itime = dqb32.dqb_itime; 2891b3515f3SDag-Erling Smørgrav return (0); 2901b3515f3SDag-Erling Smørgrav default: 2911b3515f3SDag-Erling Smørgrav return (-1); 2921b3515f3SDag-Erling Smørgrav } 2931b3515f3SDag-Erling Smørgrav } 2941b3515f3SDag-Erling Smørgrav 2951b3515f3SDag-Erling Smørgrav static int 2961b3515f3SDag-Erling Smørgrav quota_read64(struct quotafile *qf, struct dqblk *dqb, int id) 2971b3515f3SDag-Erling Smørgrav { 2981b3515f3SDag-Erling Smørgrav struct dqblk64 dqb64; 2991b3515f3SDag-Erling Smørgrav off_t off; 3001b3515f3SDag-Erling Smørgrav 3011b3515f3SDag-Erling Smørgrav off = sizeof(struct dqhdr64) + id * sizeof(struct dqblk64); 3021b3515f3SDag-Erling Smørgrav if (lseek(qf->fd, off, SEEK_SET) == -1) 3031b3515f3SDag-Erling Smørgrav return (-1); 3041b3515f3SDag-Erling Smørgrav switch (read(qf->fd, &dqb64, sizeof(dqb64))) { 3051b3515f3SDag-Erling Smørgrav case 0: 3066197731dSKirk McKusick memset(dqb, 0, sizeof(*dqb)); 3071b3515f3SDag-Erling Smørgrav return (0); 3081b3515f3SDag-Erling Smørgrav case sizeof(dqb64): 3091b3515f3SDag-Erling Smørgrav dqb->dqb_bhardlimit = be64toh(dqb64.dqb_bhardlimit); 3101b3515f3SDag-Erling Smørgrav dqb->dqb_bsoftlimit = be64toh(dqb64.dqb_bsoftlimit); 3111b3515f3SDag-Erling Smørgrav dqb->dqb_curblocks = be64toh(dqb64.dqb_curblocks); 3121b3515f3SDag-Erling Smørgrav dqb->dqb_ihardlimit = be64toh(dqb64.dqb_ihardlimit); 3131b3515f3SDag-Erling Smørgrav dqb->dqb_isoftlimit = be64toh(dqb64.dqb_isoftlimit); 3141b3515f3SDag-Erling Smørgrav dqb->dqb_curinodes = be64toh(dqb64.dqb_curinodes); 3151b3515f3SDag-Erling Smørgrav dqb->dqb_btime = be64toh(dqb64.dqb_btime); 3161b3515f3SDag-Erling Smørgrav dqb->dqb_itime = be64toh(dqb64.dqb_itime); 3171b3515f3SDag-Erling Smørgrav return (0); 3181b3515f3SDag-Erling Smørgrav default: 3191b3515f3SDag-Erling Smørgrav return (-1); 3201b3515f3SDag-Erling Smørgrav } 3211b3515f3SDag-Erling Smørgrav } 3221b3515f3SDag-Erling Smørgrav 3231b3515f3SDag-Erling Smørgrav int 3241b3515f3SDag-Erling Smørgrav quota_read(struct quotafile *qf, struct dqblk *dqb, int id) 3251b3515f3SDag-Erling Smørgrav { 3268bd6a3abSKirk McKusick int qcmd; 3271b3515f3SDag-Erling Smørgrav 3288bd6a3abSKirk McKusick if (qf->fd == -1) { 3298bd6a3abSKirk McKusick qcmd = QCMD(Q_GETQUOTA, qf->quotatype); 3308bd6a3abSKirk McKusick return (quotactl(qf->fsname, qcmd, id, dqb)); 3318bd6a3abSKirk McKusick } 3328bd6a3abSKirk McKusick switch (qf->wordsize) { 3331b3515f3SDag-Erling Smørgrav case 32: 334fdd356a8SDag-Erling Smørgrav return (quota_read32(qf, dqb, id)); 3351b3515f3SDag-Erling Smørgrav case 64: 336fdd356a8SDag-Erling Smørgrav return (quota_read64(qf, dqb, id)); 3371b3515f3SDag-Erling Smørgrav default: 3381b3515f3SDag-Erling Smørgrav errno = EINVAL; 3391b3515f3SDag-Erling Smørgrav return (-1); 3401b3515f3SDag-Erling Smørgrav } 3411b3515f3SDag-Erling Smørgrav /* not reached */ 3421b3515f3SDag-Erling Smørgrav } 3431b3515f3SDag-Erling Smørgrav 3441b3515f3SDag-Erling Smørgrav #define CLIP32(u64) ((u64) > UINT32_MAX ? UINT32_MAX : (uint32_t)(u64)) 3451b3515f3SDag-Erling Smørgrav 3461b3515f3SDag-Erling Smørgrav static int 3471b3515f3SDag-Erling Smørgrav quota_write32(struct quotafile *qf, const struct dqblk *dqb, int id) 3481b3515f3SDag-Erling Smørgrav { 3491b3515f3SDag-Erling Smørgrav struct dqblk32 dqb32; 3501b3515f3SDag-Erling Smørgrav off_t off; 3511b3515f3SDag-Erling Smørgrav 3521b3515f3SDag-Erling Smørgrav dqb32.dqb_bhardlimit = CLIP32(dqb->dqb_bhardlimit); 3531b3515f3SDag-Erling Smørgrav dqb32.dqb_bsoftlimit = CLIP32(dqb->dqb_bsoftlimit); 3541b3515f3SDag-Erling Smørgrav dqb32.dqb_curblocks = CLIP32(dqb->dqb_curblocks); 3551b3515f3SDag-Erling Smørgrav dqb32.dqb_ihardlimit = CLIP32(dqb->dqb_ihardlimit); 3561b3515f3SDag-Erling Smørgrav dqb32.dqb_isoftlimit = CLIP32(dqb->dqb_isoftlimit); 3571b3515f3SDag-Erling Smørgrav dqb32.dqb_curinodes = CLIP32(dqb->dqb_curinodes); 3581b3515f3SDag-Erling Smørgrav dqb32.dqb_btime = CLIP32(dqb->dqb_btime); 3591b3515f3SDag-Erling Smørgrav dqb32.dqb_itime = CLIP32(dqb->dqb_itime); 3601b3515f3SDag-Erling Smørgrav 3611b3515f3SDag-Erling Smørgrav off = id * sizeof(struct dqblk32); 3621b3515f3SDag-Erling Smørgrav if (lseek(qf->fd, off, SEEK_SET) == -1) 3631b3515f3SDag-Erling Smørgrav return (-1); 3648bd6a3abSKirk McKusick if (write(qf->fd, &dqb32, sizeof(dqb32)) == sizeof(dqb32)) 3658bd6a3abSKirk McKusick return (0); 3668bd6a3abSKirk McKusick return (-1); 3671b3515f3SDag-Erling Smørgrav } 3681b3515f3SDag-Erling Smørgrav 3691b3515f3SDag-Erling Smørgrav static int 3701b3515f3SDag-Erling Smørgrav quota_write64(struct quotafile *qf, const struct dqblk *dqb, int id) 3711b3515f3SDag-Erling Smørgrav { 3721b3515f3SDag-Erling Smørgrav struct dqblk64 dqb64; 3731b3515f3SDag-Erling Smørgrav off_t off; 3741b3515f3SDag-Erling Smørgrav 3751b3515f3SDag-Erling Smørgrav dqb64.dqb_bhardlimit = htobe64(dqb->dqb_bhardlimit); 3761b3515f3SDag-Erling Smørgrav dqb64.dqb_bsoftlimit = htobe64(dqb->dqb_bsoftlimit); 3771b3515f3SDag-Erling Smørgrav dqb64.dqb_curblocks = htobe64(dqb->dqb_curblocks); 3781b3515f3SDag-Erling Smørgrav dqb64.dqb_ihardlimit = htobe64(dqb->dqb_ihardlimit); 3791b3515f3SDag-Erling Smørgrav dqb64.dqb_isoftlimit = htobe64(dqb->dqb_isoftlimit); 3801b3515f3SDag-Erling Smørgrav dqb64.dqb_curinodes = htobe64(dqb->dqb_curinodes); 3811b3515f3SDag-Erling Smørgrav dqb64.dqb_btime = htobe64(dqb->dqb_btime); 3821b3515f3SDag-Erling Smørgrav dqb64.dqb_itime = htobe64(dqb->dqb_itime); 3831b3515f3SDag-Erling Smørgrav 3841b3515f3SDag-Erling Smørgrav off = sizeof(struct dqhdr64) + id * sizeof(struct dqblk64); 3851b3515f3SDag-Erling Smørgrav if (lseek(qf->fd, off, SEEK_SET) == -1) 3861b3515f3SDag-Erling Smørgrav return (-1); 3878bd6a3abSKirk McKusick if (write(qf->fd, &dqb64, sizeof(dqb64)) == sizeof(dqb64)) 3888bd6a3abSKirk McKusick return (0); 3898bd6a3abSKirk McKusick return (-1); 3901b3515f3SDag-Erling Smørgrav } 3911b3515f3SDag-Erling Smørgrav 3921b3515f3SDag-Erling Smørgrav int 3938bd6a3abSKirk McKusick quota_write_usage(struct quotafile *qf, struct dqblk *dqb, int id) 3941b3515f3SDag-Erling Smørgrav { 3958bd6a3abSKirk McKusick struct dqblk dqbuf; 3968bd6a3abSKirk McKusick int qcmd; 3971b3515f3SDag-Erling Smørgrav 3985666aadbSDag-Erling Smørgrav if ((qf->accmode & O_RDWR) != O_RDWR) { 3995666aadbSDag-Erling Smørgrav errno = EBADF; 4005666aadbSDag-Erling Smørgrav return (-1); 4015666aadbSDag-Erling Smørgrav } 4028bd6a3abSKirk McKusick if (qf->fd == -1) { 4038bd6a3abSKirk McKusick qcmd = QCMD(Q_SETUSE, qf->quotatype); 4048bd6a3abSKirk McKusick return (quotactl(qf->fsname, qcmd, id, dqb)); 4058bd6a3abSKirk McKusick } 4068bd6a3abSKirk McKusick /* 4078bd6a3abSKirk McKusick * Have to do read-modify-write of quota in file. 4088bd6a3abSKirk McKusick */ 4098bd6a3abSKirk McKusick if (quota_read(qf, &dqbuf, id) != 0) 4108bd6a3abSKirk McKusick return (-1); 4118bd6a3abSKirk McKusick /* 4128bd6a3abSKirk McKusick * Reset time limit if have a soft limit and were 4138bd6a3abSKirk McKusick * previously under it, but are now over it. 4148bd6a3abSKirk McKusick */ 4158bd6a3abSKirk McKusick if (dqbuf.dqb_bsoftlimit && id != 0 && 4168bd6a3abSKirk McKusick dqbuf.dqb_curblocks < dqbuf.dqb_bsoftlimit && 4178bd6a3abSKirk McKusick dqb->dqb_curblocks >= dqbuf.dqb_bsoftlimit) 4188bd6a3abSKirk McKusick dqbuf.dqb_btime = 0; 4198bd6a3abSKirk McKusick if (dqbuf.dqb_isoftlimit && id != 0 && 4208bd6a3abSKirk McKusick dqbuf.dqb_curinodes < dqbuf.dqb_isoftlimit && 4218bd6a3abSKirk McKusick dqb->dqb_curinodes >= dqbuf.dqb_isoftlimit) 4228bd6a3abSKirk McKusick dqbuf.dqb_itime = 0; 4238bd6a3abSKirk McKusick dqbuf.dqb_curinodes = dqb->dqb_curinodes; 4248bd6a3abSKirk McKusick dqbuf.dqb_curblocks = dqb->dqb_curblocks; 4258bd6a3abSKirk McKusick /* 4268bd6a3abSKirk McKusick * Write it back. 4278bd6a3abSKirk McKusick */ 4288bd6a3abSKirk McKusick switch (qf->wordsize) { 4298bd6a3abSKirk McKusick case 32: 430fdd356a8SDag-Erling Smørgrav return (quota_write32(qf, &dqbuf, id)); 4318bd6a3abSKirk McKusick case 64: 432fdd356a8SDag-Erling Smørgrav return (quota_write64(qf, &dqbuf, id)); 4338bd6a3abSKirk McKusick default: 4348bd6a3abSKirk McKusick errno = EINVAL; 4358bd6a3abSKirk McKusick return (-1); 4368bd6a3abSKirk McKusick } 4378bd6a3abSKirk McKusick /* not reached */ 4388bd6a3abSKirk McKusick } 4398bd6a3abSKirk McKusick 4408bd6a3abSKirk McKusick int 4418bd6a3abSKirk McKusick quota_write_limits(struct quotafile *qf, struct dqblk *dqb, int id) 4428bd6a3abSKirk McKusick { 4438bd6a3abSKirk McKusick struct dqblk dqbuf; 4448bd6a3abSKirk McKusick int qcmd; 4458bd6a3abSKirk McKusick 4465666aadbSDag-Erling Smørgrav if ((qf->accmode & O_RDWR) != O_RDWR) { 4475666aadbSDag-Erling Smørgrav errno = EBADF; 4485666aadbSDag-Erling Smørgrav return (-1); 4495666aadbSDag-Erling Smørgrav } 4508bd6a3abSKirk McKusick if (qf->fd == -1) { 4518bd6a3abSKirk McKusick qcmd = QCMD(Q_SETQUOTA, qf->quotatype); 4528bd6a3abSKirk McKusick return (quotactl(qf->fsname, qcmd, id, dqb)); 4538bd6a3abSKirk McKusick } 4548bd6a3abSKirk McKusick /* 4558bd6a3abSKirk McKusick * Have to do read-modify-write of quota in file. 4568bd6a3abSKirk McKusick */ 4578bd6a3abSKirk McKusick if (quota_read(qf, &dqbuf, id) != 0) 4588bd6a3abSKirk McKusick return (-1); 4598bd6a3abSKirk McKusick /* 4608bd6a3abSKirk McKusick * Reset time limit if have a soft limit and were 4618bd6a3abSKirk McKusick * previously under it, but are now over it 4628bd6a3abSKirk McKusick * or if there previously was no soft limit, but 4638bd6a3abSKirk McKusick * now have one and are over it. 4648bd6a3abSKirk McKusick */ 4658bd6a3abSKirk McKusick if (dqbuf.dqb_bsoftlimit && id != 0 && 4668bd6a3abSKirk McKusick dqbuf.dqb_curblocks < dqbuf.dqb_bsoftlimit && 4678bd6a3abSKirk McKusick dqbuf.dqb_curblocks >= dqb->dqb_bsoftlimit) 4688bd6a3abSKirk McKusick dqb->dqb_btime = 0; 4698bd6a3abSKirk McKusick if (dqbuf.dqb_bsoftlimit == 0 && id != 0 && 4708bd6a3abSKirk McKusick dqb->dqb_bsoftlimit > 0 && 4718bd6a3abSKirk McKusick dqbuf.dqb_curblocks >= dqb->dqb_bsoftlimit) 4728bd6a3abSKirk McKusick dqb->dqb_btime = 0; 4738bd6a3abSKirk McKusick if (dqbuf.dqb_isoftlimit && id != 0 && 4748bd6a3abSKirk McKusick dqbuf.dqb_curinodes < dqbuf.dqb_isoftlimit && 4758bd6a3abSKirk McKusick dqbuf.dqb_curinodes >= dqb->dqb_isoftlimit) 4768bd6a3abSKirk McKusick dqb->dqb_itime = 0; 4778bd6a3abSKirk McKusick if (dqbuf.dqb_isoftlimit == 0 && id !=0 && 4788bd6a3abSKirk McKusick dqb->dqb_isoftlimit > 0 && 4798bd6a3abSKirk McKusick dqbuf.dqb_curinodes >= dqb->dqb_isoftlimit) 4808bd6a3abSKirk McKusick dqb->dqb_itime = 0; 4818bd6a3abSKirk McKusick dqb->dqb_curinodes = dqbuf.dqb_curinodes; 4828bd6a3abSKirk McKusick dqb->dqb_curblocks = dqbuf.dqb_curblocks; 4838bd6a3abSKirk McKusick /* 4848bd6a3abSKirk McKusick * Write it back. 4858bd6a3abSKirk McKusick */ 4868bd6a3abSKirk McKusick switch (qf->wordsize) { 4871b3515f3SDag-Erling Smørgrav case 32: 488fdd356a8SDag-Erling Smørgrav return (quota_write32(qf, dqb, id)); 4891b3515f3SDag-Erling Smørgrav case 64: 490fdd356a8SDag-Erling Smørgrav return (quota_write64(qf, dqb, id)); 4911b3515f3SDag-Erling Smørgrav default: 4921b3515f3SDag-Erling Smørgrav errno = EINVAL; 4931b3515f3SDag-Erling Smørgrav return (-1); 4941b3515f3SDag-Erling Smørgrav } 4951b3515f3SDag-Erling Smørgrav /* not reached */ 4961b3515f3SDag-Erling Smørgrav } 497aee785baSKirk McKusick 498aee785baSKirk McKusick /* 499aee785baSKirk McKusick * Convert a quota file from one format to another. 500aee785baSKirk McKusick */ 501aee785baSKirk McKusick int 502aee785baSKirk McKusick quota_convert(struct quotafile *qf, int wordsize) 503aee785baSKirk McKusick { 504aee785baSKirk McKusick struct quotafile *newqf; 505aee785baSKirk McKusick struct dqhdr64 dqh; 506aee785baSKirk McKusick struct dqblk dqblk; 507aee785baSKirk McKusick struct group *grp; 508aee785baSKirk McKusick int serrno, maxid, id, fd; 509aee785baSKirk McKusick 510aee785baSKirk McKusick /* 511aee785baSKirk McKusick * Quotas must not be active and quotafile must be open 512aee785baSKirk McKusick * for reading and writing. 513aee785baSKirk McKusick */ 514aee785baSKirk McKusick if ((qf->accmode & O_RDWR) != O_RDWR || qf->fd == -1) { 515aee785baSKirk McKusick errno = EBADF; 516aee785baSKirk McKusick return (-1); 517aee785baSKirk McKusick } 518aee785baSKirk McKusick if ((wordsize != 32 && wordsize != 64) || 519aee785baSKirk McKusick wordsize == qf->wordsize) { 520aee785baSKirk McKusick errno = EINVAL; 521aee785baSKirk McKusick return (-1); 522aee785baSKirk McKusick } 523aee785baSKirk McKusick maxid = quota_maxid(qf); 524aee785baSKirk McKusick if ((newqf = calloc(1, sizeof(*qf))) == NULL) { 525aee785baSKirk McKusick errno = ENOMEM; 526aee785baSKirk McKusick return (-1); 527aee785baSKirk McKusick } 528aee785baSKirk McKusick *newqf = *qf; 529aee785baSKirk McKusick snprintf(newqf->qfname, MAXPATHLEN + 1, "%s_%d.orig", qf->qfname, 530aee785baSKirk McKusick qf->wordsize); 531aee785baSKirk McKusick if (rename(qf->qfname, newqf->qfname) < 0) { 532aee785baSKirk McKusick free(newqf); 533aee785baSKirk McKusick return (-1); 534aee785baSKirk McKusick } 535aee785baSKirk McKusick if ((newqf->fd = open(qf->qfname, O_RDWR|O_CREAT|O_TRUNC, 0)) < 0) { 536aee785baSKirk McKusick serrno = errno; 537aee785baSKirk McKusick goto error; 538aee785baSKirk McKusick } 539aee785baSKirk McKusick newqf->wordsize = wordsize; 540aee785baSKirk McKusick if (wordsize == 64) { 541aee785baSKirk McKusick memset(&dqh, 0, sizeof(dqh)); 542aee785baSKirk McKusick memcpy(dqh.dqh_magic, Q_DQHDR64_MAGIC, sizeof(dqh.dqh_magic)); 543aee785baSKirk McKusick dqh.dqh_version = htobe32(Q_DQHDR64_VERSION); 544aee785baSKirk McKusick dqh.dqh_hdrlen = htobe32(sizeof(struct dqhdr64)); 545aee785baSKirk McKusick dqh.dqh_reclen = htobe32(sizeof(struct dqblk64)); 546aee785baSKirk McKusick if (write(newqf->fd, &dqh, sizeof(dqh)) != sizeof(dqh)) { 547aee785baSKirk McKusick serrno = errno; 548aee785baSKirk McKusick goto error; 549aee785baSKirk McKusick } 550aee785baSKirk McKusick } 551aee785baSKirk McKusick grp = getgrnam(QUOTAGROUP); 552aee785baSKirk McKusick fchown(newqf->fd, 0, grp ? grp->gr_gid : 0); 553aee785baSKirk McKusick fchmod(newqf->fd, 0640); 554aee785baSKirk McKusick for (id = 0; id <= maxid; id++) { 555aee785baSKirk McKusick if ((quota_read(qf, &dqblk, id)) < 0) 556aee785baSKirk McKusick break; 557aee785baSKirk McKusick switch (newqf->wordsize) { 558aee785baSKirk McKusick case 32: 559aee785baSKirk McKusick if ((quota_write32(newqf, &dqblk, id)) < 0) 560aee785baSKirk McKusick break; 561aee785baSKirk McKusick continue; 562aee785baSKirk McKusick case 64: 563aee785baSKirk McKusick if ((quota_write64(newqf, &dqblk, id)) < 0) 564aee785baSKirk McKusick break; 565aee785baSKirk McKusick continue; 566aee785baSKirk McKusick default: 567aee785baSKirk McKusick errno = EINVAL; 568aee785baSKirk McKusick break; 569aee785baSKirk McKusick } 570aee785baSKirk McKusick } 571aee785baSKirk McKusick if (id < maxid) { 572aee785baSKirk McKusick serrno = errno; 573aee785baSKirk McKusick goto error; 574aee785baSKirk McKusick } 575aee785baSKirk McKusick /* 576aee785baSKirk McKusick * Update the passed in quotafile to reference the new file 577aee785baSKirk McKusick * of the converted format size. 578aee785baSKirk McKusick */ 579aee785baSKirk McKusick fd = qf->fd; 580aee785baSKirk McKusick qf->fd = newqf->fd; 581aee785baSKirk McKusick newqf->fd = fd; 582aee785baSKirk McKusick qf->wordsize = newqf->wordsize; 583aee785baSKirk McKusick quota_close(newqf); 584aee785baSKirk McKusick return (0); 585aee785baSKirk McKusick error: 586aee785baSKirk McKusick /* put back the original file */ 587aee785baSKirk McKusick (void) rename(newqf->qfname, qf->qfname); 588aee785baSKirk McKusick quota_close(newqf); 589aee785baSKirk McKusick errno = serrno; 590aee785baSKirk McKusick return (-1); 591aee785baSKirk McKusick } 592