11a3e5dd7SThomas Graichen /*
21a3e5dd7SThomas Graichen * by Manuel Bouyer (bouyer@ensta.fr)
31a3e5dd7SThomas Graichen *
41a3e5dd7SThomas Graichen * There is no copyright, you can use it as you want.
51a3e5dd7SThomas Graichen */
61a3e5dd7SThomas Graichen
71a3e5dd7SThomas Graichen #include <sys/param.h>
81a3e5dd7SThomas Graichen #include <sys/mount.h>
91a3e5dd7SThomas Graichen #include <sys/file.h>
101a3e5dd7SThomas Graichen #include <sys/stat.h>
111a3e5dd7SThomas Graichen #include <sys/socket.h>
121a3e5dd7SThomas Graichen
131a3e5dd7SThomas Graichen #include <ufs/ufs/quota.h>
141a3e5dd7SThomas Graichen #include <rpc/rpc.h>
151a3e5dd7SThomas Graichen #include <rpcsvc/rquota.h>
161a3e5dd7SThomas Graichen #include <arpa/inet.h>
17af37179bSAlfred Perlstein #include <netdb.h>
181a3e5dd7SThomas Graichen
193ebc6944SDag-Erling Smørgrav #include <ctype.h>
203ebc6944SDag-Erling Smørgrav #include <errno.h>
213ebc6944SDag-Erling Smørgrav #include <fstab.h>
223ebc6944SDag-Erling Smørgrav #include <grp.h>
231b3515f3SDag-Erling Smørgrav #include <libutil.h>
243ebc6944SDag-Erling Smørgrav #include <pwd.h>
253ebc6944SDag-Erling Smørgrav #include <signal.h>
263ebc6944SDag-Erling Smørgrav #include <stdio.h>
273ebc6944SDag-Erling Smørgrav #include <stdlib.h>
28aad5531eSSean Eric Fagan #include <err.h>
293ebc6944SDag-Erling Smørgrav #include <string.h>
303ebc6944SDag-Erling Smørgrav #include <syslog.h>
313ebc6944SDag-Erling Smørgrav #include <unistd.h>
323ebc6944SDag-Erling Smørgrav
33aad5531eSSean Eric Fagan static void rquota_service_1(struct svc_req *request, SVCXPRT *transp);
34aad5531eSSean Eric Fagan static void rquota_service_2(struct svc_req *request, SVCXPRT *transp);
352db2319dSDag-Erling Smørgrav static void sendquota(struct svc_req *request, SVCXPRT *transp);
36aad5531eSSean Eric Fagan static void sendquota_extended(struct svc_req *request, SVCXPRT *transp);
37aad5531eSSean Eric Fagan static int getfsquota(int type, long id, char *path, struct dqblk *dqblk);
381a3e5dd7SThomas Graichen
392db2319dSDag-Erling Smørgrav static int from_inetd = 1;
40aad5531eSSean Eric Fagan static int debug = 0;
411a3e5dd7SThomas Graichen
423ebc6944SDag-Erling Smørgrav static void
cleanup(int sig)43a174e5b1SWarner Losh cleanup(int sig)
441a3e5dd7SThomas Graichen {
453ebc6944SDag-Erling Smørgrav
463ebc6944SDag-Erling Smørgrav (void)sig;
47af37179bSAlfred Perlstein (void)rpcb_unset(RQUOTAPROG, RQUOTAVERS, NULL);
481a3e5dd7SThomas Graichen exit(0);
491a3e5dd7SThomas Graichen }
501a3e5dd7SThomas Graichen
511a3e5dd7SThomas Graichen int
main(int argc,char ** argv)52aad5531eSSean Eric Fagan main(int argc, char **argv)
531a3e5dd7SThomas Graichen {
541a3e5dd7SThomas Graichen SVCXPRT *transp;
55af37179bSAlfred Perlstein int ok;
56af37179bSAlfred Perlstein struct sockaddr_storage from;
5778e3eed0SStefan Farfeleder socklen_t fromlen;
58aad5531eSSean Eric Fagan int vers;
59aad5531eSSean Eric Fagan int ch;
60aad5531eSSean Eric Fagan
61aad5531eSSean Eric Fagan while ((ch = getopt(argc, argv, "d")) != -1) {
62aad5531eSSean Eric Fagan switch (ch) {
63aad5531eSSean Eric Fagan case 'd':
64aad5531eSSean Eric Fagan debug++;
65aad5531eSSean Eric Fagan break;
66aad5531eSSean Eric Fagan default:
67aad5531eSSean Eric Fagan break;
68aad5531eSSean Eric Fagan }
69aad5531eSSean Eric Fagan }
701a3e5dd7SThomas Graichen
711a3e5dd7SThomas Graichen fromlen = sizeof(from);
72ebb2b0f2SDag-Erling Smørgrav if (getsockname(0, (struct sockaddr *)&from, &fromlen) < 0)
731a3e5dd7SThomas Graichen from_inetd = 0;
741a3e5dd7SThomas Graichen
751a3e5dd7SThomas Graichen if (!from_inetd) {
76aad5531eSSean Eric Fagan if (!debug)
771a3e5dd7SThomas Graichen daemon(0, 0);
78af37179bSAlfred Perlstein (void)rpcb_unset(RQUOTAPROG, RQUOTAVERS, NULL);
791a3e5dd7SThomas Graichen (void)signal(SIGINT, cleanup);
801a3e5dd7SThomas Graichen (void)signal(SIGTERM, cleanup);
811a3e5dd7SThomas Graichen (void)signal(SIGHUP, cleanup);
821a3e5dd7SThomas Graichen }
831a3e5dd7SThomas Graichen
841a3e5dd7SThomas Graichen openlog("rpc.rquotad", LOG_CONS|LOG_PID, LOG_DAEMON);
851a3e5dd7SThomas Graichen
861a3e5dd7SThomas Graichen /* create and register the service */
87af37179bSAlfred Perlstein if (from_inetd) {
88af37179bSAlfred Perlstein transp = svc_tli_create(0, NULL, NULL, 0, 0);
891a3e5dd7SThomas Graichen if (transp == NULL) {
90af37179bSAlfred Perlstein syslog(LOG_ERR, "couldn't create udp service.");
911a3e5dd7SThomas Graichen exit(1);
921a3e5dd7SThomas Graichen }
93aad5531eSSean Eric Fagan vers = RQUOTAVERS;
94af37179bSAlfred Perlstein ok = svc_reg(transp, RQUOTAPROG, RQUOTAVERS,
95aad5531eSSean Eric Fagan rquota_service_1, NULL);
96aad5531eSSean Eric Fagan if (ok) {
97aad5531eSSean Eric Fagan vers = EXT_RQUOTAVERS;
98aad5531eSSean Eric Fagan ok = svc_reg(transp, RQUOTAPROG, EXT_RQUOTAVERS,
99aad5531eSSean Eric Fagan rquota_service_2, NULL);
100aad5531eSSean Eric Fagan }
101ebb2b0f2SDag-Erling Smørgrav } else {
102aad5531eSSean Eric Fagan vers = RQUOTAVERS;
103aad5531eSSean Eric Fagan ok = svc_create(rquota_service_1,
104af37179bSAlfred Perlstein RQUOTAPROG, RQUOTAVERS, "udp");
105aad5531eSSean Eric Fagan if (ok) {
106aad5531eSSean Eric Fagan vers = EXT_RQUOTAVERS;
107aad5531eSSean Eric Fagan ok = svc_create(rquota_service_2,
108aad5531eSSean Eric Fagan RQUOTAPROG, EXT_RQUOTAVERS, "udp");
109aad5531eSSean Eric Fagan
110aad5531eSSean Eric Fagan }
111ebb2b0f2SDag-Erling Smørgrav }
112af37179bSAlfred Perlstein if (!ok) {
1133ebc6944SDag-Erling Smørgrav syslog(LOG_ERR,
114aad5531eSSean Eric Fagan "unable to register (RQUOTAPROG, %s, %s)",
115aad5531eSSean Eric Fagan vers == RQUOTAVERS ? "RQUOTAVERS" : "EXT_RQUOTAVERS",
1163ebc6944SDag-Erling Smørgrav from_inetd ? "(inetd)" : "udp");
1171a3e5dd7SThomas Graichen exit(1);
1181a3e5dd7SThomas Graichen }
1191a3e5dd7SThomas Graichen
1201a3e5dd7SThomas Graichen svc_run();
1211a3e5dd7SThomas Graichen syslog(LOG_ERR, "svc_run returned");
1221a3e5dd7SThomas Graichen exit(1);
1231a3e5dd7SThomas Graichen }
1241a3e5dd7SThomas Graichen
1252db2319dSDag-Erling Smørgrav static void
rquota_service_2(struct svc_req * request,SVCXPRT * transp)126aad5531eSSean Eric Fagan rquota_service_2(struct svc_req *request, SVCXPRT *transp)
127aad5531eSSean Eric Fagan {
128aad5531eSSean Eric Fagan
129aad5531eSSean Eric Fagan switch (request->rq_proc) {
130aad5531eSSean Eric Fagan case NULLPROC:
131aad5531eSSean Eric Fagan (void)svc_sendreply(transp, (xdrproc_t)xdr_void, (char *)NULL);
132aad5531eSSean Eric Fagan break;
133aad5531eSSean Eric Fagan case RQUOTAPROC_GETQUOTA:
134aad5531eSSean Eric Fagan case RQUOTAPROC_GETACTIVEQUOTA:
135aad5531eSSean Eric Fagan sendquota_extended(request, transp);
136aad5531eSSean Eric Fagan break;
137aad5531eSSean Eric Fagan default:
138aad5531eSSean Eric Fagan svcerr_noproc(transp);
139aad5531eSSean Eric Fagan break;
140aad5531eSSean Eric Fagan }
141aad5531eSSean Eric Fagan if (from_inetd)
142aad5531eSSean Eric Fagan exit(0);
143aad5531eSSean Eric Fagan }
144aad5531eSSean Eric Fagan
145aad5531eSSean Eric Fagan static void
rquota_service_1(struct svc_req * request,SVCXPRT * transp)146aad5531eSSean Eric Fagan rquota_service_1(struct svc_req *request, SVCXPRT *transp)
1471a3e5dd7SThomas Graichen {
1483ebc6944SDag-Erling Smørgrav
1491a3e5dd7SThomas Graichen switch (request->rq_proc) {
1501a3e5dd7SThomas Graichen case NULLPROC:
151f249dbccSDag-Erling Smørgrav (void)svc_sendreply(transp, (xdrproc_t)xdr_void, (char *)NULL);
1521a3e5dd7SThomas Graichen break;
1531a3e5dd7SThomas Graichen case RQUOTAPROC_GETQUOTA:
1541a3e5dd7SThomas Graichen case RQUOTAPROC_GETACTIVEQUOTA:
1551a3e5dd7SThomas Graichen sendquota(request, transp);
1561a3e5dd7SThomas Graichen break;
1571a3e5dd7SThomas Graichen default:
1581a3e5dd7SThomas Graichen svcerr_noproc(transp);
1591a3e5dd7SThomas Graichen break;
1601a3e5dd7SThomas Graichen }
1611a3e5dd7SThomas Graichen if (from_inetd)
1621a3e5dd7SThomas Graichen exit(0);
1631a3e5dd7SThomas Graichen }
1641a3e5dd7SThomas Graichen
1651a3e5dd7SThomas Graichen /* read quota for the specified id, and send it */
1662db2319dSDag-Erling Smørgrav static void
sendquota(struct svc_req * request,SVCXPRT * transp)167a174e5b1SWarner Losh sendquota(struct svc_req *request, SVCXPRT *transp)
1681a3e5dd7SThomas Graichen {
1691a3e5dd7SThomas Graichen struct getquota_args getq_args;
1701a3e5dd7SThomas Graichen struct getquota_rslt getq_rslt;
1711a3e5dd7SThomas Graichen struct dqblk dqblk;
1721a3e5dd7SThomas Graichen struct timeval timev;
17350921097SKirk McKusick int scale;
1741a3e5dd7SThomas Graichen
175ebb2b0f2SDag-Erling Smørgrav bzero(&getq_args, sizeof(getq_args));
176f249dbccSDag-Erling Smørgrav if (!svc_getargs(transp, (xdrproc_t)xdr_getquota_args, &getq_args)) {
1771a3e5dd7SThomas Graichen svcerr_decode(transp);
1781a3e5dd7SThomas Graichen return;
1791a3e5dd7SThomas Graichen }
1801a3e5dd7SThomas Graichen if (request->rq_cred.oa_flavor != AUTH_UNIX) {
1811a3e5dd7SThomas Graichen /* bad auth */
1821a3e5dd7SThomas Graichen getq_rslt.status = Q_EPERM;
183aad5531eSSean Eric Fagan } else if (!getfsquota(USRQUOTA, getq_args.gqa_uid, getq_args.gqa_pathp, &dqblk)) {
1841a3e5dd7SThomas Graichen /* failed, return noquota */
1851a3e5dd7SThomas Graichen getq_rslt.status = Q_NOQUOTA;
1861a3e5dd7SThomas Graichen } else {
1871a3e5dd7SThomas Graichen gettimeofday(&timev, NULL);
1881a3e5dd7SThomas Graichen getq_rslt.status = Q_OK;
1891a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_active = TRUE;
19050921097SKirk McKusick scale = 1 << flsll(dqblk.dqb_bhardlimit >> 32);
19150921097SKirk McKusick getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize =
19250921097SKirk McKusick DEV_BSIZE * scale;
1931a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit =
19450921097SKirk McKusick dqblk.dqb_bhardlimit / scale;
1951a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit =
19650921097SKirk McKusick dqblk.dqb_bsoftlimit / scale;
1971a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks =
19850921097SKirk McKusick dqblk.dqb_curblocks / scale;
1991a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_fhardlimit =
2001a3e5dd7SThomas Graichen dqblk.dqb_ihardlimit;
2011a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_fsoftlimit =
2021a3e5dd7SThomas Graichen dqblk.dqb_isoftlimit;
2031a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_curfiles =
2041a3e5dd7SThomas Graichen dqblk.dqb_curinodes;
2051a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_btimeleft =
2061a3e5dd7SThomas Graichen dqblk.dqb_btime - timev.tv_sec;
2071a3e5dd7SThomas Graichen getq_rslt.getquota_rslt_u.gqr_rquota.rq_ftimeleft =
2081a3e5dd7SThomas Graichen dqblk.dqb_itime - timev.tv_sec;
2091a3e5dd7SThomas Graichen }
21095f1d628SDag-Erling Smørgrav if (!svc_sendreply(transp, (xdrproc_t)xdr_getquota_rslt, &getq_rslt))
2111a3e5dd7SThomas Graichen svcerr_systemerr(transp);
212f249dbccSDag-Erling Smørgrav if (!svc_freeargs(transp, (xdrproc_t)xdr_getquota_args, &getq_args)) {
2131a3e5dd7SThomas Graichen syslog(LOG_ERR, "unable to free arguments");
2141a3e5dd7SThomas Graichen exit(1);
2151a3e5dd7SThomas Graichen }
2161a3e5dd7SThomas Graichen }
2171a3e5dd7SThomas Graichen
2182db2319dSDag-Erling Smørgrav static void
sendquota_extended(struct svc_req * request,SVCXPRT * transp)219aad5531eSSean Eric Fagan sendquota_extended(struct svc_req *request, SVCXPRT *transp)
2201a3e5dd7SThomas Graichen {
221aad5531eSSean Eric Fagan struct ext_getquota_args getq_args;
222aad5531eSSean Eric Fagan struct getquota_rslt getq_rslt;
223aad5531eSSean Eric Fagan struct dqblk dqblk;
224aad5531eSSean Eric Fagan struct timeval timev;
225aad5531eSSean Eric Fagan int scale;
2261a3e5dd7SThomas Graichen
227aad5531eSSean Eric Fagan bzero(&getq_args, sizeof(getq_args));
228aad5531eSSean Eric Fagan if (!svc_getargs(transp, (xdrproc_t)xdr_ext_getquota_args, &getq_args)) {
229aad5531eSSean Eric Fagan svcerr_decode(transp);
2309dc9c85fSDag-Erling Smørgrav return;
231aad5531eSSean Eric Fagan }
232aad5531eSSean Eric Fagan if (request->rq_cred.oa_flavor != AUTH_UNIX) {
233aad5531eSSean Eric Fagan /* bad auth */
234aad5531eSSean Eric Fagan getq_rslt.status = Q_EPERM;
235aad5531eSSean Eric Fagan } else if (!getfsquota(getq_args.gqa_type, getq_args.gqa_id, getq_args.gqa_pathp, &dqblk)) {
236aad5531eSSean Eric Fagan /* failed, return noquota */
237aad5531eSSean Eric Fagan getq_rslt.status = Q_NOQUOTA;
238aad5531eSSean Eric Fagan } else {
239aad5531eSSean Eric Fagan gettimeofday(&timev, NULL);
240aad5531eSSean Eric Fagan getq_rslt.status = Q_OK;
241aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_active = TRUE;
242aad5531eSSean Eric Fagan scale = 1 << flsll(dqblk.dqb_bhardlimit >> 32);
243aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsize =
244aad5531eSSean Eric Fagan DEV_BSIZE * scale;
245aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_bhardlimit =
246aad5531eSSean Eric Fagan dqblk.dqb_bhardlimit / scale;
247aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_bsoftlimit =
248aad5531eSSean Eric Fagan dqblk.dqb_bsoftlimit / scale;
249aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_curblocks =
250aad5531eSSean Eric Fagan dqblk.dqb_curblocks / scale;
251aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_fhardlimit =
252aad5531eSSean Eric Fagan dqblk.dqb_ihardlimit;
253aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_fsoftlimit =
254aad5531eSSean Eric Fagan dqblk.dqb_isoftlimit;
255aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_curfiles =
256aad5531eSSean Eric Fagan dqblk.dqb_curinodes;
257aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_btimeleft =
258aad5531eSSean Eric Fagan dqblk.dqb_btime - timev.tv_sec;
259aad5531eSSean Eric Fagan getq_rslt.getquota_rslt_u.gqr_rquota.rq_ftimeleft =
260aad5531eSSean Eric Fagan dqblk.dqb_itime - timev.tv_sec;
261aad5531eSSean Eric Fagan }
262aad5531eSSean Eric Fagan if (!svc_sendreply(transp, (xdrproc_t)xdr_getquota_rslt, &getq_rslt))
263aad5531eSSean Eric Fagan svcerr_systemerr(transp);
264aad5531eSSean Eric Fagan if (!svc_freeargs(transp, (xdrproc_t)xdr_getquota_args, &getq_args)) {
265aad5531eSSean Eric Fagan syslog(LOG_ERR, "unable to free arguments");
2669dc9c85fSDag-Erling Smørgrav exit(1);
267aad5531eSSean Eric Fagan }
2681a3e5dd7SThomas Graichen }
2691a3e5dd7SThomas Graichen
2701a3e5dd7SThomas Graichen /*
2711a3e5dd7SThomas Graichen * gets the quotas for id, filesystem path.
2721a3e5dd7SThomas Graichen * Return 0 if fail, 1 otherwise
2731a3e5dd7SThomas Graichen */
2742db2319dSDag-Erling Smørgrav static int
getfsquota(int type,long id,char * path,struct dqblk * dqblk)275aad5531eSSean Eric Fagan getfsquota(int type, long id, char *path, struct dqblk *dqblk)
2761a3e5dd7SThomas Graichen {
277aad5531eSSean Eric Fagan struct quotafile *qf;
278aad5531eSSean Eric Fagan /*
279aad5531eSSean Eric Fagan * Remote quota checking is limited to mounted filesystems.
280aad5531eSSean Eric Fagan * Since UFS and ZFS support the quota system calls, we
281aad5531eSSean Eric Fagan * only need to make an fstab object that has the path, and
282aad5531eSSean Eric Fagan * a blank name for the filesystem type.
283aad5531eSSean Eric Fagan * This allows the quota_open() call to work the way we
284aad5531eSSean Eric Fagan * expect it to.
285aad5531eSSean Eric Fagan *
286aad5531eSSean Eric Fagan * The static char declaration is because compiler warnings
287aad5531eSSean Eric Fagan * don't allow passing a const char * to a char *.
288aad5531eSSean Eric Fagan */
289aad5531eSSean Eric Fagan int rv;
290aad5531eSSean Eric Fagan static char blank[] = "";
291aad5531eSSean Eric Fagan struct fstab fst;
2921a3e5dd7SThomas Graichen
293aad5531eSSean Eric Fagan fst.fs_file = path;
294aad5531eSSean Eric Fagan fst.fs_mntops = blank;
295aad5531eSSean Eric Fagan fst.fs_vfstype = blank;
296aad5531eSSean Eric Fagan
297aad5531eSSean Eric Fagan if (type != USRQUOTA && type != GRPQUOTA)
2981a3e5dd7SThomas Graichen return (0);
299aad5531eSSean Eric Fagan
300aad5531eSSean Eric Fagan qf = quota_open(&fst, type, O_RDONLY);
301aad5531eSSean Eric Fagan if (debug)
302aad5531eSSean Eric Fagan warnx("quota_open(<%s, %s>, %d) returned %p",
303aad5531eSSean Eric Fagan fst.fs_file, fst.fs_mntops, type,
304aad5531eSSean Eric Fagan qf);
305aad5531eSSean Eric Fagan if (qf == NULL)
306aad5531eSSean Eric Fagan return (0);
307aad5531eSSean Eric Fagan
308aad5531eSSean Eric Fagan rv = quota_read(qf, dqblk, id) == 0;
309aad5531eSSean Eric Fagan quota_close(qf);
310aad5531eSSean Eric Fagan if (debug)
311aad5531eSSean Eric Fagan warnx("getfsquota(%d, %ld, %s, %p) -> %d",
312aad5531eSSean Eric Fagan type, id, path, dqblk, rv);
313aad5531eSSean Eric Fagan return (rv);
3141a3e5dd7SThomas Graichen }
315