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, "astat)) {
|
---|
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 = NULL;
|
---|
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 | if (testpath == NULL) {
|
---|
138 | errno = EINVAL;
|
---|
139 | goto out;
|
---|
140 | }
|
---|
141 | testpath++;
|
---|
142 | gq_args.gqa_pathp = testpath;
|
---|
143 | gq_args.gqa_uid = id.uid;
|
---|
144 |
|
---|
145 | DEBUG(10, ("sys_get_nfs_quotas: Asking for quota of path '%s' on "
|
---|
146 | "host '%s', rpcprog '%i', rpcvers '%i', network '%s'\n",
|
---|
147 | host, testpath+1, RQUOTAPROG, RQUOTAVERS, "udp"));
|
---|
148 |
|
---|
149 | clnt = clnt_create(host, RQUOTAPROG, RQUOTAVERS, "udp");
|
---|
150 | if (clnt == NULL) {
|
---|
151 | ret = -1;
|
---|
152 | goto out;
|
---|
153 | }
|
---|
154 |
|
---|
155 | clnt->cl_auth = authunix_create_default();
|
---|
156 | if (clnt->cl_auth == NULL) {
|
---|
157 | DEBUG(3, ("sys_get_nfs_quotas: authunix_create_default "
|
---|
158 | "failed\n"));
|
---|
159 | ret = -1;
|
---|
160 | goto out;
|
---|
161 | }
|
---|
162 |
|
---|
163 | clnt_stat = clnt_call(clnt,
|
---|
164 | RQUOTAPROC_GETQUOTA,
|
---|
165 | (const xdrproc_t) my_xdr_getquota_args,
|
---|
166 | (caddr_t)&gq_args,
|
---|
167 | (const xdrproc_t) my_xdr_getquota_rslt,
|
---|
168 | (caddr_t)&gq_rslt,
|
---|
169 | timeout);
|
---|
170 |
|
---|
171 | if (clnt_stat != RPC_SUCCESS) {
|
---|
172 | DEBUG(3, ("sys_get_nfs_quotas: clnt_call failed\n"));
|
---|
173 | ret = -1;
|
---|
174 | goto out;
|
---|
175 | }
|
---|
176 |
|
---|
177 | DEBUG(10, ("sys_get_nfs_quotas: getquota_rslt:\n"
|
---|
178 | "status : '%i'\n"
|
---|
179 | "bsize : '%i'\n"
|
---|
180 | "active : '%s'\n"
|
---|
181 | "bhardlimit : '%u'\n"
|
---|
182 | "bsoftlimit : '%u'\n"
|
---|
183 | "curblocks : '%u'\n"
|
---|
184 | "fhardlimit : '%u'\n"
|
---|
185 | "fsoftlimit : '%u'\n"
|
---|
186 | "curfiles : '%u'\n"
|
---|
187 | "btimeleft : '%u'\n"
|
---|
188 | "ftimeleft : '%u'\n",
|
---|
189 | gq_rslt.status,
|
---|
190 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize,
|
---|
191 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_active?"yes":"no",
|
---|
192 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit,
|
---|
193 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit,
|
---|
194 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks,
|
---|
195 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_fhardlimit,
|
---|
196 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_fsoftlimit,
|
---|
197 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_curfiles,
|
---|
198 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_btimeleft,
|
---|
199 | gq_rslt.getquota_rslt_u.gqr_rquota.rq_ftimeleft));
|
---|
200 |
|
---|
201 | /*
|
---|
202 | * gqr.status returns
|
---|
203 | * 0 if the rpc call fails,
|
---|
204 | * 1 if quotas exist,
|
---|
205 | * 2 if there is no quota set, and
|
---|
206 | * 3 if no permission to get the quota.
|
---|
207 | */
|
---|
208 |
|
---|
209 | switch (gq_rslt.status) {
|
---|
210 | case 0:
|
---|
211 | DEBUG(3, ("sys_get_nfs_quotas: Remote Quotas Failed! "
|
---|
212 | "Error '%i'\n", gq_rslt.status));
|
---|
213 | ret = -1;
|
---|
214 | goto out;
|
---|
215 |
|
---|
216 | case 1:
|
---|
217 | DEBUG(10, ("sys_get_nfs_quotas: Good quota data\n"));
|
---|
218 | dp->bsize = (uint64_t)gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize;
|
---|
219 | dp->softlimit = gq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit;
|
---|
220 | dp->hardlimit = gq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit;
|
---|
221 | dp->curblocks = gq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks;
|
---|
222 | break;
|
---|
223 |
|
---|
224 | case 2:
|
---|
225 | DEBUG(5, ("sys_get_nfs_quotas: No quota set\n"));
|
---|
226 | SMB_QUOTAS_SET_NO_LIMIT(dp);
|
---|
227 | break;
|
---|
228 |
|
---|
229 | case 3:
|
---|
230 | DEBUG(3, ("sys_get_nfs_quotas: no permission to get quota\n"));
|
---|
231 | errno = EPERM;
|
---|
232 | ret = -1;
|
---|
233 | goto out;
|
---|
234 |
|
---|
235 | default:
|
---|
236 | DEBUG(5, ("sys_get_nfs_quotas: Unknown remote quota status "
|
---|
237 | "code '%i'\n", gq_rslt.status));
|
---|
238 | ret = -1;
|
---|
239 | goto out;
|
---|
240 | break;
|
---|
241 | }
|
---|
242 |
|
---|
243 | dp->qflags = qflags;
|
---|
244 |
|
---|
245 | ret = 0;
|
---|
246 |
|
---|
247 | out:
|
---|
248 | if (clnt) {
|
---|
249 | if (clnt->cl_auth) {
|
---|
250 | auth_destroy(clnt->cl_auth);
|
---|
251 | }
|
---|
252 | clnt_destroy(clnt);
|
---|
253 | }
|
---|
254 |
|
---|
255 | SAFE_FREE(cutstr);
|
---|
256 |
|
---|
257 | DEBUG(10, ("sys_get_nfs_quotas: finished\n" ));
|
---|
258 | return ret;
|
---|
259 | }
|
---|
260 |
|
---|
261 | int sys_set_nfs_quota(const char *path, const char *bdev,
|
---|
262 | enum SMB_QUOTA_TYPE qtype,
|
---|
263 | unid_t id, SMB_DISK_QUOTA *dp)
|
---|
264 | {
|
---|
265 | DEBUG(1, ("sys_set_nfs_quota : not supported\n"));
|
---|
266 | errno = ENOSYS;
|
---|
267 | return -1;
|
---|
268 | }
|
---|
269 |
|
---|
270 | #else /* HAVE_NFS_QUOTAS */
|
---|
271 |
|
---|
272 | void dummy_sysquotas_nfs(void);
|
---|
273 | void dummy_sysquotas_nfs(void) {}
|
---|
274 |
|
---|
275 | #endif /* HAVE_NFS_QUOTAS */
|
---|