source3/lib/sysquotas_nfs.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    System QUOTA function wrappers for NFS
   4    Copyright (C) Michael Adam 2010
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20
  21 #include "includes.h"
  22
  23 #undef DBGC_CLASS
  24 #define DBGC_CLASS DBGC_QUOTA
  25
  26 #ifndef HAVE_SYS_QUOTAS
  27 #ifdef HAVE_NFS_QUOTAS
  28 #undef HAVE_NFS_QUOTAS
  29 #endif
  30 #endif
  31
  32 #ifdef HAVE_NFS_QUOTAS
  33
  34 /*
  35  * nfs quota support
  36  * This is based on the FreeBSD / SUNOS5 section of quotas.c
  37  */
  38
  39 /* <rpc/xdr.h> uses TRUE and FALSE */
  40 #ifdef TRUE
  41 #undef TRUE
  42 #endif
  43
  44 #ifdef FALSE
  45 #undef FALSE
  46 #endif
  47
  48 #include <rpc/rpc.h>
  49 #include <rpc/types.h>
  50 #include <rpc/xdr.h>
  51 #include <rpcsvc/rquota.h>
  52 #ifdef HAVE_RPC_NETTYPE_H
  53 #include <rpc/nettype.h>
  54 #endif
  55
  56 #ifndef RQ_PATHLEN
  57 #define RQ_PATHLEN 1024
  58 #endif
  59
  60 #ifdef HAVE_GETQUOTA_RSLT_GETQUOTA_RSLT_U
  61 #define GQR_RQUOTA getquota_rslt_u.gqr_rquota
  62 #define GQR_STATUS status
  63 #else
  64 #define GQR_RQUOTA gqr_rquota
  65 #define GQR_STATUS gqr_status
  66 #endif
  67
  68 static int my_xdr_getquota_args(XDR *xdrsp, struct getquota_args *args)
  69 {
  70         if (!xdr_string(xdrsp, &args->gqa_pathp, RQ_PATHLEN ))
  71                 return(0);
  72         if (!xdr_int(xdrsp, &args->gqa_uid))
  73                 return(0);
  74         return (1);
  75 }
  76
  77 static int my_xdr_getquota_rslt(XDR *xdrsp, struct getquota_rslt *gqr)
  78 {
  79         int quotastat;
  80
  81         if (!xdr_int(xdrsp, &quotastat)) {
  82                 DEBUG(6,("nfs_quotas: Status bad or zero\n"));
  83                 return 0;
  84         }
  85         gqr->GQR_STATUS = quotastat;
  86
  87         if (!xdr_int(xdrsp, &gqr->GQR_RQUOTA.rq_bsize)) {
  88                 DEBUG(6,("nfs_quotas: Block size bad or zero\n"));
  89                 return 0;
  90         }
  91         if (!xdr_bool(xdrsp, &gqr->GQR_RQUOTA.rq_active)) {
  92                 DEBUG(6,("nfs_quotas: Active bad or zero\n"));
  93                 return 0;
  94         }
  95         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_bhardlimit)) {
  96                 DEBUG(6,("nfs_quotas: Hardlimit bad or zero\n"));
  97                 return 0;
  98         }
  99         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_bsoftlimit)) {
 100                 DEBUG(6,("nfs_quotas: Softlimit bad or zero\n"));
 101                 return 0;
 102         }
 103         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_curblocks)) {
 104                 DEBUG(6,("nfs_quotas: Currentblocks bad or zero\n"));
 105                 return 0;
 106         }
 107         return (1);
 108 }
 109
 110
 111 int sys_get_nfs_quota(const char *path, const char *bdev,
 112                       enum SMB_QUOTA_TYPE qtype,
 113                       unid_t id, SMB_DISK_QUOTA *dp)
 114 {
 115         CLIENT *clnt = NULL;
 116         struct getquota_rslt gq_rslt;
 117         struct getquota_args gq_args;
 118         const char *mnttype;
 119         char *cutstr, *host, *testpath;
 120         int len;
 121         static struct timeval timeout = {2,0};
 122         enum clnt_stat clnt_stat;
 123
 124         int ret = -1;
 125         uint32_t qflags = 0;
 126
 127         if (!path || !bdev || !dp) {
 128                 smb_panic("sys_get_nfs_quota: called with NULL pointer");
 129         }
 130
 131         DEBUG(10, ("sys_get_nfs_quota: path[%s] bdev[%s] qtype[%d]\n",
 132                    path, bdev, qtype));
 133
 134         ZERO_STRUCT(*dp);
 135
 136         dp->qtype = qtype;
 137
 138         if (qtype != SMB_USER_QUOTA_TYPE) {
 139                 DEBUG(3, ("sys_get_nfs_quota: got unsupported quota type '%d', "
 140                           "only supported type is '%d' (SMB_USER_QUOTA_TYPE)\n",
 141                           qtype, SMB_USER_QUOTA_TYPE));
 142                 errno = ENOSYS;
 143                 return -1;
 144         }
 145
 146         mnttype = bdev;
 147         len = strcspn(mnttype, ":");
 148         cutstr = (char *) SMB_MALLOC(len+1);
 149         if (cutstr == NULL) {
 150                 errno = ENOMEM;
 151                 return -1;
 152         }
 153
 154         memset(cutstr, '\0', len+1);
 155         host = strncat(cutstr, mnttype, sizeof(char) * len);
 156         testpath = strchr_m(mnttype, ':');
 157         if (testpath == NULL) {
 158                 errno = EINVAL;
 159                 goto out;
 160         }
 161         testpath++;
 162         gq_args.gqa_pathp = testpath;
 163         gq_args.gqa_uid = id.uid;
 164
 165         DEBUG(10, ("sys_get_nfs_quotas: Asking for quota of path '%s' on "
 166                    "host '%s', rpcprog '%i', rpcvers '%i', network '%s'\n",
 167                    host, testpath+1, (int)RQUOTAPROG, (int)RQUOTAVERS, "udp"));
 168
 169         clnt = clnt_create(host, RQUOTAPROG, RQUOTAVERS, "udp");
 170         if (clnt == NULL) {
 171                 ret = -1;
 172                 goto out;
 173         }
 174
 175         clnt->cl_auth = authunix_create_default();
 176         if (clnt->cl_auth == NULL) {
 177                 DEBUG(3, ("sys_get_nfs_quotas: authunix_create_default "
 178                           "failed\n"));
 179                 ret = -1;
 180                 goto out;
 181         }
 182
 183         clnt_stat = clnt_call(clnt,
 184                               RQUOTAPROC_GETQUOTA,
 185                               (const xdrproc_t) my_xdr_getquota_args,
 186                               (caddr_t)&gq_args,
 187                               (const xdrproc_t) my_xdr_getquota_rslt,
 188                               (caddr_t)&gq_rslt,
 189                               timeout);
 190
 191         if (clnt_stat != RPC_SUCCESS) {
 192                 if (errno == ECONNREFUSED) {
 193                         /* If we cannot connect with rpc.quotad, it may
 194                          * simply be because there's no quota on the remote
 195                          * system
 196                          */
 197                         DBG_INFO("clnt_call failed with ECONNREFUSED - "
 198                                  "assuming no quotas on server\n");
 199                         ret = 0;
 200                 } else {
 201                         int save_errno = errno;
 202                         DBG_NOTICE("clnt_call failed - %s\n", strerror(errno));
 203                         errno = save_errno;
 204                         ret = -1;
 205                 }
 206                 goto out;
 207         }
 208
 209         DEBUG(10, ("sys_get_nfs_quotas: getquota_rslt:\n"
 210                    "status       : '%i'\n"
 211                    "bsize        : '%i'\n"
 212                    "active       : '%s'\n"
 213                    "bhardlimit   : '%u'\n"
 214                    "bsoftlimit   : '%u'\n"
 215                    "curblocks    : '%u'\n"
 216                    "fhardlimit   : '%u'\n"
 217                    "fsoftlimit   : '%u'\n"
 218                    "curfiles     : '%u'\n"
 219                    "btimeleft    : '%u'\n"
 220                    "ftimeleft    : '%u'\n",
 221                    gq_rslt.GQR_STATUS,
 222                    gq_rslt.GQR_RQUOTA.rq_bsize,
 223                    gq_rslt.GQR_RQUOTA.rq_active?"yes":"no",
 224                    gq_rslt.GQR_RQUOTA.rq_bhardlimit,
 225                    gq_rslt.GQR_RQUOTA.rq_bsoftlimit,
 226                    gq_rslt.GQR_RQUOTA.rq_curblocks,
 227                    gq_rslt.GQR_RQUOTA.rq_fhardlimit,
 228                    gq_rslt.GQR_RQUOTA.rq_fsoftlimit,
 229                    gq_rslt.GQR_RQUOTA.rq_curfiles,
 230                    gq_rslt.GQR_RQUOTA.rq_btimeleft,
 231                    gq_rslt.GQR_RQUOTA.rq_ftimeleft));
 232
 233         /*
 234          * gqr.status returns
 235          *   1 if quotas exist,
 236          *   2 if there is no quota set, and
 237          *   3 if no permission to get the quota.
 238          */
 239
 240         switch (gq_rslt.GQR_STATUS) {
 241         case 1:
 242                 DEBUG(10, ("sys_get_nfs_quotas: Good quota data\n"));
 243                 dp->bsize = (uint64_t)gq_rslt.GQR_RQUOTA.rq_bsize;
 244                 dp->softlimit = gq_rslt.GQR_RQUOTA.rq_bsoftlimit;
 245                 dp->hardlimit = gq_rslt.GQR_RQUOTA.rq_bhardlimit;
 246                 dp->curblocks = gq_rslt.GQR_RQUOTA.rq_curblocks;
 247                 break;
 248
 249         case 2:
 250                 DEBUG(5, ("sys_get_nfs_quotas: No quota set\n"));
 251                 SMB_QUOTAS_SET_NO_LIMIT(dp);
 252                 break;
 253
 254         case 3:
 255                 DEBUG(3, ("sys_get_nfs_quotas: no permission to get quota\n"));
 256                 errno = EPERM;
 257                 ret = -1;
 258                 goto out;
 259
 260         default:
 261                 DEBUG(5, ("sys_get_nfs_quotas: Unknown remote quota status "
 262                           "code '%i'\n", gq_rslt.GQR_STATUS));
 263                 ret = -1;
 264                 goto out;
 265                 break;
 266         }
 267
 268         dp->qflags = qflags;
 269
 270         ret = 0;
 271
 272 out:
 273         if (clnt) {
 274                 if (clnt->cl_auth) {
 275                         auth_destroy(clnt->cl_auth);
 276                 }
 277                 clnt_destroy(clnt);
 278         }
 279
 280         SAFE_FREE(cutstr);
 281
 282         DEBUG(10, ("sys_get_nfs_quotas: finished\n" ));
 283         return ret;
 284 }
 285
 286 int sys_set_nfs_quota(const char *path, const char *bdev,
 287                       enum SMB_QUOTA_TYPE qtype,
 288                       unid_t id, SMB_DISK_QUOTA *dp)
 289 {
 290         DEBUG(1, ("sys_set_nfs_quota : not supported\n"));
 291         errno = ENOSYS;
 292         return -1;
 293 }
 294
 295 #else /* HAVE_NFS_QUOTAS */
 296
 297 void dummy_sysquotas_nfs(void);
 298 void dummy_sysquotas_nfs(void) {}
 299
 300 #endif /* HAVE_NFS_QUOTAS */