s3:smbd: add a nfs backend for sysquotas.
[amitay/samba.git] / 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 #include <rpc/rpc.h>
40 #include <rpc/types.h>
41 #include <rpcsvc/rquota.h>
42 #ifdef HAVE_RPC_NETTYPE_H
43 #include <rpc/nettype.h>
44 #endif
45 #include <rpc/xdr.h>
46
47 static int my_xdr_getquota_args(XDR *xdrsp, struct getquota_args *args)
48 {
49         if (!xdr_string(xdrsp, &args->gqa_pathp, RQ_PATHLEN ))
50                 return(0);
51         if (!xdr_int(xdrsp, &args->gqa_uid))
52                 return(0);
53         return (1);
54 }
55
56 static int my_xdr_getquota_rslt(XDR *xdrsp, struct getquota_rslt *gqr)
57 {
58         int quotastat;
59
60         if (!xdr_int(xdrsp, &quotastat)) {
61                 DEBUG(6,("nfs_quotas: Status bad or zero\n"));
62                 return 0;
63         }
64         gqr->status = quotastat;
65
66         if (!xdr_int(xdrsp, &gqr->getquota_rslt_u.gqr_rquota.rq_bsize)) {
67                 DEBUG(6,("nfs_quotas: Block size bad or zero\n"));
68                 return 0;
69         }
70         if (!xdr_bool(xdrsp, &gqr->getquota_rslt_u.gqr_rquota.rq_active)) {
71                 DEBUG(6,("nfs_quotas: Active bad or zero\n"));
72                 return 0;
73         }
74         if (!xdr_int(xdrsp, (int *)&gqr->getquota_rslt_u.gqr_rquota.rq_bhardlimit)) {
75                 DEBUG(6,("nfs_quotas: Hardlimit bad or zero\n"));
76                 return 0;
77         }
78         if (!xdr_int(xdrsp, (int *)&gqr->getquota_rslt_u.gqr_rquota.rq_bsoftlimit)) {
79                 DEBUG(6,("nfs_quotas: Softlimit bad or zero\n"));
80                 return 0;
81         }
82         if (!xdr_int(xdrsp, (int *)&gqr->getquota_rslt_u.gqr_rquota.rq_curblocks)) {
83                 DEBUG(6,("nfs_quotas: Currentblocks bad or zero\n"));
84                 return 0;
85         }
86         return (1);
87 }
88
89
90 int sys_get_nfs_quota(const char *path, const char *bdev,
91                       enum SMB_QUOTA_TYPE qtype,
92                       unid_t id, SMB_DISK_QUOTA *dp)
93 {
94         CLIENT *clnt;
95         struct getquota_rslt gq_rslt;
96         struct getquota_args gq_args;
97         const char *mnttype;
98         char *cutstr, *pathname, *host, *testpath;
99         int len;
100         static struct timeval timeout = {2,0};
101         enum clnt_stat clnt_stat;
102
103         int ret = -1;
104         uint32 qflags = 0;
105
106         if (!path || !bdev || !dp) {
107                 smb_panic("sys_get_nfs_quota: called with NULL pointer");
108         }
109
110         DEBUG(10, ("sys_get_nfs_quota: path[%s] bdev[%s] qtype[%d]\n",
111                    path, bdev, qtype));
112
113         ZERO_STRUCT(*dp);
114
115         dp->qtype = qtype;
116
117         if (qtype != SMB_USER_QUOTA_TYPE) {
118                 DEBUG(3, ("sys_get_nfs_quota: got unsupported quota type '%d', "
119                           "only supported type is '%d' (SMB_USER_QUOTA_TYPE)\n",
120                           qtype, SMB_USER_QUOTA_TYPE));
121                 errno = ENOSYS;
122                 return -1;
123         }
124
125         mnttype = bdev;
126         len = strcspn(mnttype, ":");
127         pathname = strstr(mnttype, ":");
128         cutstr = (char *) SMB_MALLOC(len+1);
129         if (cutstr == NULL) {
130                 errno = ENOMEM;
131                 return -1;
132         }
133
134         memset(cutstr, '\0', len+1);
135         host = strncat(cutstr, mnttype, sizeof(char) * len);
136         testpath = strchr_m(mnttype, ':');
137         testpath++;
138         gq_args.gqa_pathp = testpath;
139         gq_args.gqa_uid = id.uid;
140
141         DEBUG(10, ("sys_get_nfs_quotas: Asking for quota of path '%s' on "
142                    "host '%s', rpcprog '%i', rpcvers '%i', network '%s'\n",
143                     host, testpath+1, RQUOTAPROG, RQUOTAVERS, "udp"));
144
145         clnt = clnt_create(host, RQUOTAPROG, RQUOTAVERS, "udp");
146         if (clnt == NULL) {
147                 ret = -1;
148                 goto out;
149         }
150
151         clnt->cl_auth = authunix_create_default();
152         if (clnt->cl_auth == NULL) {
153                 DEBUG(3, ("sys_get_nfs_quotas: authunix_create_default "
154                           "failed\n"));
155                 ret = -1;
156                 goto out;
157         }
158
159         clnt_stat = clnt_call(clnt,
160                               RQUOTAPROC_GETQUOTA,
161                               (const xdrproc_t) my_xdr_getquota_args,
162                               (caddr_t)&gq_args,
163                               (const xdrproc_t) my_xdr_getquota_rslt,
164                               (caddr_t)&gq_rslt,
165                               timeout);
166
167         if (clnt_stat != RPC_SUCCESS) {
168                 DEBUG(3, ("sys_get_nfs_quotas: clnt_call failed\n"));
169                 ret = -1;
170                 goto out;
171         }
172
173         DEBUG(10, ("sys_get_nfs_quotas: getquota_rslt:\n"
174                    "status       : '%i'\n"
175                    "bsize        : '%i'\n"
176                    "active       : '%s'\n"
177                    "bhardlimit   : '%u'\n"
178                    "bsoftlimit   : '%u'\n"
179                    "curblocks    : '%u'\n"
180                    "fhardlimit   : '%u'\n"
181                    "fsoftlimit   : '%u'\n"
182                    "curfiles     : '%u'\n"
183                    "btimeleft    : '%u'\n"
184                    "ftimeleft    : '%u'\n",
185                    gq_rslt.status,
186                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize,
187                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_active?"yes":"no",
188                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit,
189                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit,
190                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks,
191                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_fhardlimit,
192                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_fsoftlimit,
193                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_curfiles,
194                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_btimeleft,
195                    gq_rslt.getquota_rslt_u.gqr_rquota.rq_ftimeleft));
196
197         /*
198          * gqr.status returns
199          *   0 if the rpc call fails,
200          *   1 if quotas exist,
201          *   2 if there is no quota set, and
202          *   3 if no permission to get the quota.
203          */
204
205         switch (gq_rslt.status) {
206         case 0:
207                 DEBUG(3, ("sys_get_nfs_quotas: Remote Quotas Failed! "
208                           "Error '%i'\n", gq_rslt.status));
209                 ret = -1;
210                 goto out;
211
212         case 1:
213                 DEBUG(10, ("sys_get_nfs_quotas: Good quota data\n"));
214                 dp->bsize = (uint64_t)gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize;
215                 dp->softlimit = gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit;
216                 dp->hardlimit = gq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit;
217                 dp->curblocks = gq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks;
218                 break;
219
220         case 2:
221                 DEBUG(5, ("sys_get_nfs_quotas: No quota set\n"));
222                 SMB_QUOTAS_SET_NO_LIMIT(dp);
223                 break;
224
225         case 3:
226                 DEBUG(3, ("sys_get_nfs_quotas: no permission to get quota\n"));
227                 errno = EPERM;
228                 ret = -1;
229                 goto out;
230
231         default:
232                 DEBUG(5, ("sys_get_nfs_quotas: Unknown remote quota status "
233                           "code '%i'\n", gq_rslt.status));
234                 ret = -1;
235                 goto out;
236                 break;
237         }
238
239         dp->qflags = qflags;
240
241         ret = 0;
242
243 out:
244         if (clnt) {
245                 if (clnt->cl_auth) {
246                         auth_destroy(clnt->cl_auth);
247                 }
248                 clnt_destroy(clnt);
249         }
250
251         SAFE_FREE(cutstr);
252
253         DEBUG(10, ("sys_get_nfs_quotas: finished\n" ));
254         return ret;
255 }
256
257 int sys_set_nfs_quota(const char *path, const char *bdev,
258                       enum SMB_QUOTA_TYPE qtype,
259                       unid_t id, SMB_DISK_QUOTA *dp)
260 {
261         DEBUG(1, ("sys_set_nfs_quota : not supported\n"));
262         errno = ENOSYS;
263         return -1;
264 }
265
266 #else /* HAVE_NFS_QUOTAS */
267
268 void dummy_sysquotas_nfs(void);
269 void dummy_sysquotas_nfs(void) {}
270
271 #endif /* HAVE_NFS_QUOTAS */