source3/lib/sysquotas_4B.c

   1 /*
   2  * Unix SMB/CIFS implementation.
   3  * System QUOTA function wrappers for QUOTACTL_4B
   4
   5  * Copyright (C) 2011 James Peach.
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20  */
  21
  22 #include "includes.h"
  23
  24 #undef DBGC_CLASS
  25 #define DBGC_CLASS DBGC_QUOTA
  26
  27 #ifndef HAVE_SYS_QUOTAS
  28 #undef HAVE_QUOTACTL_4B
  29 #endif
  30
  31 #ifdef HAVE_QUOTACTL_4B
  32 /* int quotactl(const char *path, int cmd, int id, char *addr)
  33  *
  34  * This is used by many (all?) BSD-derived systems. This implementation has
  35  * been developed and tested on Darwin, but may also work on other BSD systems.
  36  */
  37
  38 #ifdef HAVE_SYS_TYPES_H
  39 #include <sys/types.h>
  40 #endif
  41
  42 #ifdef HAVE_SYS_QUOTA_H
  43 #include <sys/quota.h>
  44 #endif
  45
  46 #ifdef HAVE_UFS_UFS_QUOTA_H
  47 #include <ufs/ufs/quota.h>
  48 #endif
  49
  50 #if defined(DARWINOS)
  51 /* WorkARound broken HFS access checks in hfs_quotactl. Darwin only(?) */
  52 #define HFS_QUOTACTL_WAR 1
  53 #endif
  54
  55 static void xlate_qblk_to_smb(const struct dqblk * const qblk,
  56                         SMB_DISK_QUOTA *dp)
  57 {
  58         ZERO_STRUCTP(dp);
  59
  60         DEBUG(10, ("unix softlimit=%u hardlimit=%u curblock=%u\n",
  61             (unsigned)qblk->dqb_bsoftlimit, (unsigned)qblk->dqb_bhardlimit,
  62 #ifdef HAVE_STRUCT_DQBLK_DQB_CURBYTES
  63             (unsigned)qblk->dqb_curbytes));
  64 #else
  65             (unsigned)qblk->dqb_curblocks));
  66 #endif
  67
  68         DEBUGADD(10, ("unix softinodes=%u hardinodes=%u curinodes=%u\n",
  69             (unsigned)qblk->dqb_isoftlimit, (unsigned)qblk->dqb_ihardlimit,
  70             (unsigned)qblk->dqb_curinodes));
  71
  72 #ifdef HAVE_STRUCT_DQBLK_DQB_CURBYTES
  73         /* On Darwin, quotas are counted in bytes. We report them
  74          * in 512b blocks because various callers have assumptions
  75          * about the block size.
  76          */
  77 #define XLATE_TO_BLOCKS(bytes) (((bytes) + 1) / 512)
  78         dp->bsize = 512;
  79
  80         dp->softlimit = XLATE_TO_BLOCKS(qblk->dqb_bsoftlimit);
  81         dp->hardlimit = XLATE_TO_BLOCKS(qblk->dqb_bhardlimit);
  82         dp->curblocks = XLATE_TO_BLOCKS(qblk->dqb_curbytes);
  83 #undef XLATE_TO_BLOCKS
  84 #endif
  85
  86         dp->ihardlimit = qblk->dqb_ihardlimit;
  87         dp->isoftlimit = qblk->dqb_isoftlimit;
  88         dp->curinodes = qblk->dqb_curinodes;
  89
  90         dp->qflags = QUOTAS_ENABLED | QUOTAS_DENY_DISK;
  91
  92         DEBUG(10, ("softlimit=%u hardlimit=%u curblock=%u\n",
  93             (unsigned)dp->softlimit, (unsigned)dp->hardlimit,
  94             (unsigned)dp->curblocks));
  95
  96         DEBUGADD(10, ("softinodes=%u hardinodes=%u curinodes=%u\n",
  97             (unsigned)dp->isoftlimit, (unsigned)dp->ihardlimit,
  98             (unsigned)dp->curinodes));
  99
 100 }
 101
 102 static void xlate_smb_to_qblk(const SMB_DISK_QUOTA * const dp,
 103                         struct dqblk *qblk)
 104 {
 105         ZERO_STRUCTP(qblk);
 106
 107         qblk->dqb_bsoftlimit = dp->softlimit;
 108         qblk->dqb_bhardlimit = dp->hardlimit;
 109 #ifdef HAVE_STRUCT_DQBLK_DQB_CURBYTES
 110         /* On Darwin, quotas are counted in bytes. */
 111         qblk->dqb_bsoftlimit *= dp->bsize;
 112         qblk->dqb_bhardlimit *= dp->bsize;
 113 #endif
 114         qblk->dqb_ihardlimit = dp->ihardlimit;
 115         qblk->dqb_isoftlimit = dp->isoftlimit;
 116 }
 117
 118 static int sys_quotactl_4B(const char * path, int cmd,
 119                 int id, struct dqblk *qblk)
 120 {
 121         int ret;
 122
 123         /* NB: We must test GRPQUOTA here, because USRQUOTA is 0. */
 124         DEBUG(10, ("%s quota for %s ID %u on %s\n",
 125                     (cmd & QCMD(Q_GETQUOTA, 0)) ? "getting" : "setting",
 126                     (cmd & QCMD(0, GRPQUOTA)) ? "group" : "user",
 127                     (unsigned)id, path));
 128
 129 #ifdef HFS_QUOTACTL_WAR
 130         become_root();
 131 #endif  /* HFS_QUOTACTL_WAR */
 132
 133         ret = quotactl(path, cmd, id, qblk);
 134         if (ret == -1) {
 135                 /* ENOTSUP means quota support is not compiled in. EINVAL
 136                  * means that quotas are not configured (commonly).
 137                  */
 138                 if (errno != ENOTSUP && errno != EINVAL) {
 139                         DEBUG(0, ("failed to %s quota for %s ID %u on %s: %s\n",
 140                                     (cmd & QCMD(Q_GETQUOTA, 0)) ? "get" : "set",
 141                                     (cmd & QCMD(0, GRPQUOTA)) ? "group" : "user",
 142                                     (unsigned)id, path, strerror(errno)));
 143                 }
 144
 145 #ifdef HFS_QUOTACTL_WAR
 146                 unbecome_root();
 147 #endif  /* HFS_QUOTACTL_WAR */
 148
 149
 150                 return -1;
 151         }
 152
 153 #ifdef HFS_QUOTACTL_WAR
 154                 unbecome_root();
 155 #endif  /* HFS_QUOTACTL_WAR */
 156
 157         return 0;
 158 }
 159
 160 int sys_get_vfs_quota(const char *path, const char *bdev,
 161         enum SMB_QUOTA_TYPE qtype, unid_t id, SMB_DISK_QUOTA *dp)
 162 {
 163         int ret;
 164         struct dqblk qblk;
 165
 166         ZERO_STRUCT(qblk);
 167
 168         switch (qtype) {
 169         case SMB_USER_QUOTA_TYPE:
 170                 /* Get quota for provided UID. */
 171                 ret = sys_quotactl_4B(path, QCMD(Q_GETQUOTA, USRQUOTA),
 172                                         id.uid, &qblk);
 173                 break;
 174         case SMB_USER_FS_QUOTA_TYPE:
 175                 /* Get quota for current UID. */
 176                 ret = sys_quotactl_4B(path, QCMD(Q_GETQUOTA, USRQUOTA),
 177                                         geteuid(), &qblk);
 178                 break;
 179         case SMB_GROUP_QUOTA_TYPE:
 180                 /* Get quota for provided GID. */
 181                 ret = sys_quotactl_4B(path, QCMD(Q_GETQUOTA, GRPQUOTA),
 182                                         id.gid, &qblk);
 183                 break;
 184         case SMB_GROUP_FS_QUOTA_TYPE:
 185                 /* Get quota for current GID. */
 186                 ret = sys_quotactl_4B(path, QCMD(Q_GETQUOTA, GRPQUOTA),
 187                                         getegid(), &qblk);
 188                 break;
 189         default:
 190                 DEBUG(0, ("cannot get unsupported quota type: %u\n",
 191                             (unsigned)qtype));
 192                 errno = ENOSYS;
 193                 return -1;
 194         }
 195
 196         if (ret == -1) {
 197                 return -1;
 198         }
 199
 200         xlate_qblk_to_smb(&qblk, dp);
 201         dp->qtype = qtype;
 202
 203         return ret;
 204 }
 205
 206 int sys_set_vfs_quota(const char *path, const char *bdev,
 207         enum SMB_QUOTA_TYPE qtype, unid_t id, SMB_DISK_QUOTA *dp)
 208 {
 209         struct dqblk qblk;
 210
 211         xlate_smb_to_qblk(dp, &qblk);
 212
 213         switch (qtype) {
 214         case SMB_USER_QUOTA_TYPE:
 215                 /* Set quota for provided UID. */
 216                 return sys_quotactl_4B(path, QCMD(Q_SETQUOTA, USRQUOTA),
 217                                         id.uid, &qblk);
 218         case SMB_USER_FS_QUOTA_TYPE:
 219                 /* Set quota for current UID. */
 220                 return sys_quotactl_4B(path, QCMD(Q_SETQUOTA, USRQUOTA),
 221                                         geteuid(), &qblk);
 222         case SMB_GROUP_QUOTA_TYPE:
 223                 /* Set quota for provided GID. */
 224                 return sys_quotactl_4B(path, QCMD(Q_SETQUOTA, GRPQUOTA),
 225                                         id.gid, &qblk);
 226         case SMB_GROUP_FS_QUOTA_TYPE:
 227                 /* Set quota for current GID. */
 228                 return sys_quotactl_4B(path, QCMD(Q_SETQUOTA, GRPQUOTA),
 229                                         getegid(), &qblk);
 230         default:
 231                 DEBUG(0, ("cannot set unsupported quota type: %u\n",
 232                             (unsigned)qtype));
 233                 errno = ENOSYS;
 234                 return -1;
 235         }
 236 }
 237
 238 #endif /* HAVE_QUOTACTL_4B */