librpc: Shorten dcerpc_binding_handle_call a bit
[Samba/gebeck_regimport.git] / source3 / lib / sysquotas_nfs.c
blob63f114b057fdab3b4fc68cf42c9fce9dc8031e58
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/>.
21 #include "includes.h"
23 #undef DBGC_CLASS
24 #define DBGC_CLASS DBGC_QUOTA
26 #ifndef HAVE_SYS_QUOTAS
27 #ifdef HAVE_NFS_QUOTAS
28 #undef HAVE_NFS_QUOTAS
29 #endif
30 #endif
32 #ifdef HAVE_NFS_QUOTAS
35  * nfs quota support
36  * This is based on the FreeBSD / SUNOS5 section of quotas.c
37  */
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>
47 #ifndef RQ_PATHLEN
48 #define RQ_PATHLEN 1024
49 #endif
51 #ifdef HAVE_GETQUOTA_RSLT_GETQUOTA_RSLT_U
52 #define GQR_RQUOTA getquota_rslt_u.gqr_rquota
53 #define GQR_STATUS status
54 #else
55 #define GQR_RQUOTA gqr_rquota
56 #define GQR_STATUS gqr_status
57 #endif
59 static int my_xdr_getquota_args(XDR *xdrsp, struct getquota_args *args)
61         if (!xdr_string(xdrsp, &args->gqa_pathp, RQ_PATHLEN ))
62                 return(0);
63         if (!xdr_int(xdrsp, &args->gqa_uid))
64                 return(0);
65         return (1);
68 static int my_xdr_getquota_rslt(XDR *xdrsp, struct getquota_rslt *gqr)
70         int quotastat;
72         if (!xdr_int(xdrsp, &quotastat)) {
73                 DEBUG(6,("nfs_quotas: Status bad or zero\n"));
74                 return 0;
75         }
76         gqr->GQR_STATUS = quotastat;
78         if (!xdr_int(xdrsp, &gqr->GQR_RQUOTA.rq_bsize)) {
79                 DEBUG(6,("nfs_quotas: Block size bad or zero\n"));
80                 return 0;
81         }
82         if (!xdr_bool(xdrsp, &gqr->GQR_RQUOTA.rq_active)) {
83                 DEBUG(6,("nfs_quotas: Active bad or zero\n"));
84                 return 0;
85         }
86         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_bhardlimit)) {
87                 DEBUG(6,("nfs_quotas: Hardlimit bad or zero\n"));
88                 return 0;
89         }
90         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_bsoftlimit)) {
91                 DEBUG(6,("nfs_quotas: Softlimit bad or zero\n"));
92                 return 0;
93         }
94         if (!xdr_int(xdrsp, (int *)&gqr->GQR_RQUOTA.rq_curblocks)) {
95                 DEBUG(6,("nfs_quotas: Currentblocks bad or zero\n"));
96                 return 0;
97         }
98         return (1);
102 int sys_get_nfs_quota(const char *path, const char *bdev,
103                       enum SMB_QUOTA_TYPE qtype,
104                       unid_t id, SMB_DISK_QUOTA *dp)
106         CLIENT *clnt = NULL;
107         struct getquota_rslt gq_rslt;
108         struct getquota_args gq_args;
109         const char *mnttype;
110         char *cutstr, *host, *testpath;
111         int len;
112         static struct timeval timeout = {2,0};
113         enum clnt_stat clnt_stat;
115         int ret = -1;
116         uint32 qflags = 0;
118         if (!path || !bdev || !dp) {
119                 smb_panic("sys_get_nfs_quota: called with NULL pointer");
120         }
122         DEBUG(10, ("sys_get_nfs_quota: path[%s] bdev[%s] qtype[%d]\n",
123                    path, bdev, qtype));
125         ZERO_STRUCT(*dp);
127         dp->qtype = qtype;
129         if (qtype != SMB_USER_QUOTA_TYPE) {
130                 DEBUG(3, ("sys_get_nfs_quota: got unsupported quota type '%d', "
131                           "only supported type is '%d' (SMB_USER_QUOTA_TYPE)\n",
132                           qtype, SMB_USER_QUOTA_TYPE));
133                 errno = ENOSYS;
134                 return -1;
135         }
137         mnttype = bdev;
138         len = strcspn(mnttype, ":");
139         cutstr = (char *) SMB_MALLOC(len+1);
140         if (cutstr == NULL) {
141                 errno = ENOMEM;
142                 return -1;
143         }
145         memset(cutstr, '\0', len+1);
146         host = strncat(cutstr, mnttype, sizeof(char) * len);
147         testpath = strchr_m(mnttype, ':');
148         if (testpath == NULL) {
149                 errno = EINVAL;
150                 goto out;
151         }
152         testpath++;
153         gq_args.gqa_pathp = testpath;
154         gq_args.gqa_uid = id.uid;
156         DEBUG(10, ("sys_get_nfs_quotas: Asking for quota of path '%s' on "
157                    "host '%s', rpcprog '%i', rpcvers '%i', network '%s'\n",
158                     host, testpath+1, RQUOTAPROG, RQUOTAVERS, "udp"));
160         clnt = clnt_create(host, RQUOTAPROG, RQUOTAVERS, "udp");
161         if (clnt == NULL) {
162                 ret = -1;
163                 goto out;
164         }
166         clnt->cl_auth = authunix_create_default();
167         if (clnt->cl_auth == NULL) {
168                 DEBUG(3, ("sys_get_nfs_quotas: authunix_create_default "
169                           "failed\n"));
170                 ret = -1;
171                 goto out;
172         }
174         clnt_stat = clnt_call(clnt,
175                               RQUOTAPROC_GETQUOTA,
176                               (const xdrproc_t) my_xdr_getquota_args,
177                               (caddr_t)&gq_args,
178                               (const xdrproc_t) my_xdr_getquota_rslt,
179                               (caddr_t)&gq_rslt,
180                               timeout);
182         if (clnt_stat != RPC_SUCCESS) {
183                 DEBUG(3, ("sys_get_nfs_quotas: clnt_call failed\n"));
184                 ret = -1;
185                 goto out;
186         }
188         DEBUG(10, ("sys_get_nfs_quotas: getquota_rslt:\n"
189                    "status       : '%i'\n"
190                    "bsize        : '%i'\n"
191                    "active       : '%s'\n"
192                    "bhardlimit   : '%u'\n"
193                    "bsoftlimit   : '%u'\n"
194                    "curblocks    : '%u'\n"
195                    "fhardlimit   : '%u'\n"
196                    "fsoftlimit   : '%u'\n"
197                    "curfiles     : '%u'\n"
198                    "btimeleft    : '%u'\n"
199                    "ftimeleft    : '%u'\n",
200                    gq_rslt.GQR_STATUS,
201                    gq_rslt.GQR_RQUOTA.rq_bsize,
202                    gq_rslt.GQR_RQUOTA.rq_active?"yes":"no",
203                    gq_rslt.GQR_RQUOTA.rq_bhardlimit,
204                    gq_rslt.GQR_RQUOTA.rq_bsoftlimit,
205                    gq_rslt.GQR_RQUOTA.rq_curblocks,
206                    gq_rslt.GQR_RQUOTA.rq_fhardlimit,
207                    gq_rslt.GQR_RQUOTA.rq_fsoftlimit,
208                    gq_rslt.GQR_RQUOTA.rq_curfiles,
209                    gq_rslt.GQR_RQUOTA.rq_btimeleft,
210                    gq_rslt.GQR_RQUOTA.rq_ftimeleft));
212         /*
213          * gqr.status returns
214          *   0 if the rpc call fails,
215          *   1 if quotas exist,
216          *   2 if there is no quota set, and
217          *   3 if no permission to get the quota.
218          */
220         switch (gq_rslt.GQR_STATUS) {
221         case 0:
222                 DEBUG(3, ("sys_get_nfs_quotas: Remote Quotas Failed! "
223                           "Error '%i'\n", gq_rslt.GQR_STATUS));
224                 ret = -1;
225                 goto out;
227         case 1:
228                 DEBUG(10, ("sys_get_nfs_quotas: Good quota data\n"));
229                 dp->bsize = (uint64_t)gq_rslt.GQR_RQUOTA.rq_bsize;
230                 dp->softlimit = gq_rslt.GQR_RQUOTA.rq_bsoftlimit;
231                 dp->hardlimit = gq_rslt.GQR_RQUOTA.rq_bhardlimit;
232                 dp->curblocks = gq_rslt.GQR_RQUOTA.rq_curblocks;
233                 break;
235         case 2:
236                 DEBUG(5, ("sys_get_nfs_quotas: No quota set\n"));
237                 SMB_QUOTAS_SET_NO_LIMIT(dp);
238                 break;
240         case 3:
241                 DEBUG(3, ("sys_get_nfs_quotas: no permission to get quota\n"));
242                 errno = EPERM;
243                 ret = -1;
244                 goto out;
246         default:
247                 DEBUG(5, ("sys_get_nfs_quotas: Unknown remote quota status "
248                           "code '%i'\n", gq_rslt.GQR_STATUS));
249                 ret = -1;
250                 goto out;
251                 break;
252         }
254         dp->qflags = qflags;
256         ret = 0;
258 out:
259         if (clnt) {
260                 if (clnt->cl_auth) {
261                         auth_destroy(clnt->cl_auth);
262                 }
263                 clnt_destroy(clnt);
264         }
266         SAFE_FREE(cutstr);
268         DEBUG(10, ("sys_get_nfs_quotas: finished\n" ));
269         return ret;
272 int sys_set_nfs_quota(const char *path, const char *bdev,
273                       enum SMB_QUOTA_TYPE qtype,
274                       unid_t id, SMB_DISK_QUOTA *dp)
276         DEBUG(1, ("sys_set_nfs_quota : not supported\n"));
277         errno = ENOSYS;
278         return -1;
281 #else /* HAVE_NFS_QUOTAS */
283 void dummy_sysquotas_nfs(void);
284 void dummy_sysquotas_nfs(void) {}
286 #endif /* HAVE_NFS_QUOTAS */